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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#! /usr/bin/env python
#
# ufw: front-end for Linux firewalling (cli)
#
# Copyright 2008-2011 Canonical Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 3,
# as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
import os
import re
import sys
import warnings
try:
import ufw.frontend
from ufw.common import UFWError
from ufw.util import error, warn, msg
except Exception:
print >> sys.stderr, "ERROR: Could not find required libraries. Aborting."
sys.exit(1)
import gettext
gettext.install(ufw.common.programName)
version = "#VERSION#"
# Internationalization
gettext.bindtextdomain(ufw.common.programName, \
os.path.join(ufw.common.trans_dir, 'messages'))
gettext.textdomain(ufw.common.programName)
_ = gettext.gettext
if sys.version_info[0] < 2 or \
(sys.version_info[0] == 2 and sys.version_info[1] < 5):
msg(ufw.common.programName + _(": Need at least python 2.5)\n"), \
sys.stderr)
sys.exit(1)
if sys.version_info[0] == 2 and sys.version_info[1] < 6:
def clean_warning(message, category, filename, lineno, file=None):
warn(message)
else:
def clean_warning(message, category, filename, lineno, file=None, line=""):
warn(message)
if __name__ == "__main__":
warnings.showwarning = clean_warning
app_action = False
pr = None
# Remember, will have to take --force into account if we use it with 'app'
idx = 1
if len(sys.argv) > 1 and sys.argv[1].lower() == "--dry-run":
idx += 1
if len(sys.argv) > idx and sys.argv[idx].lower() == "app":
app_action = True
try:
pr = ufw.frontend.parse_command(sys.argv)
except ValueError:
msg(ufw.frontend.get_command_help())
sys.exit(1)
except UFWError, e:
error(e.value)
except Exception:
raise
if pr.action == "help" or pr.action == "--help":
msg(ufw.frontend.get_command_help())
sys.exit(0)
elif pr.action == "version" or pr.action == "--version":
msg(ufw.common.programName + " " + version)
msg("Copyright 2008-2010 Canonical Ltd.")
sys.exit(0)
try:
ui = ufw.frontend.UFWFrontend(pr.dryrun)
except UFWError, e:
error(e.value)
except Exception:
raise
res = ""
try:
if app_action and pr.data.has_key('type') and pr.data['type'] == 'app':
res = ui.do_application_action(pr.action, pr.data['name'])
else:
bailout = False
if pr.action == "enable" and not pr.force and \
not ui.continue_under_ssh():
res = _("Aborted")
bailout = True
if not bailout:
if pr.data.has_key('rule'):
res = ui.do_action(pr.action, pr.data['rule'], \
pr.data['iptype'], pr.force)
else:
res = ui.do_action(pr.action, "", "", pr.force)
if res != "":
msg(res)
except UFWError, e:
error(e.value)
except Exception:
raise
sys.exit(0)
|