summaryrefslogtreecommitdiffstats
path: root/firewall.py
blob: 00f3c819c22650327e3eb56858604c97ec8219ab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import os
import iutil
import string
from log import log
from flags import flags

class Firewall:
    def __init__ (self):
	self.enabled = -1
	self.ssh = 0
	self.telnet = 0
	self.smtp = 0
	self.http = 0
	self.ftp = 0
	self.portlist = ""
	self.ports = []
	self.policy = 0
	self.dhcp = 0
	self.trustdevs = []

    def writeKS(self, f):
	f.write("firewall")

	if self.enabled > 0:
	    for arg in self.getArgList():
		f.write(" " + arg)
	else:
	    f.write(" --disabled")

	f.write("\n")

    def getArgList(self):
	args = []

	if self.policy:
	    args.append ("--medium")
	else:
	    args.append ("--high")
	if self.dhcp:
	    args.append ("--dhcp")
	if self.portlist:
	    ports = string.split(self.portlist,',')
	    for port in ports:
		port = string.strip(port)
                try:
                    if not string.index(port,':'):
                        port = '%s:tcp' % port
                except:
                    pass
		self.ports.append(port)
	for port in self.ports:
	    args = args + [ "--port", port ]
	if self.smtp:
	    args = args + [ "--port","smtp:tcp" ]
	if self.http:
	    args = args + [ "--port","http:tcp" ]
	if self.ftp:
	    args = args + [ "--port","ftp:tcp" ]
	if self.ssh:
	    args = args + [ "--port","ssh:tcp" ]
	if self.telnet:
	    args = args + [ "--port","telnet:tcp" ]
	for dev in self.trustdevs:
	    args = args + [ "--trust", dev ]

	return args

    def write (self, instPath):
	args = [ "/usr/sbin/lokkit", "--quiet", "--nostart" ]

	if self.enabled > 0:
	    args = args + self.getArgList()

            try:
                if flags.setupFilesystems:
                    iutil.execWithRedirect(args[0], args, root = instPath,
                                           stdout = None, stderr = None)
                else:
                    log("would have run %s", args)
            except RuntimeError, msg:
                log ("lokkit run failed: %s", msg)
            except OSError, (errno, msg):
                log ("lokkit run failed: %s", msg)
        else:
            # remove /etc/sysconfig/ipchains
	    file = instPath + "/etc/sysconfig/ipchains"
	    if os.access(file, os.O_RDONLY):
                os.remove(file)