28
28
self.rules_list = []
30
self.frontend = Frontend()
32
self.variable = Variable()
30
self.frontend = Frontend()
32
self.variable = Variable()
34
self.status = self.frontend.get_status()
35
self.default = self.frontend.get_default()
36
self.ufw_log = self.frontend.get_ufw_log()
37
self.gufw_log = self.frontend.get_gufw_log()
38
self.wrap = self.log.get_wrapping()
34
self.status = self.frontend.get_status()
35
self.default = self.frontend.get_default()
36
self.ufw_log = self.frontend.get_ufw_log()
37
self.gufw_log = self.frontend.get_gufw_log()
38
self.wrap = self.log.get_wrapping()
39
self.maximized = False
41
42
# Return status ufw
120
121
# Add rule to ufw
121
122
def add_rule(self, service, action, protocol, fromip, fromport, toip, toport):
122
l_rule = self.frontend.add_rule_component(service, action, protocol, fromip, fromport, toip, toport)
123
self.log.add_log(self.gufw_log, l_rule)
124
return self.frontend.add_rule(l_rule)
123
rule = self.frontend.add_rule_component(service, action, protocol, fromip, fromport, toip, toport)
124
self.log.add_log(self.gufw_log, rule)
125
return self.frontend.add_rule(rule)
127
128
# Remove rule from ufw
128
129
def remove_rule(self, rule_to, action, rule_from):
129
l_rule = self.frontend.remove_rule_component(rule_to, action, rule_from)
130
self.log.add_log(self.gufw_log, l_rule)
131
return self.frontend.remove_rule(l_rule)
130
rule = self.frontend.remove_rule_component(rule_to, action, rule_from)
131
self.log.add_log(self.gufw_log, rule)
132
return self.frontend.remove_rule(rule)
134
135
# Split rule by action
140
141
if rule.find(self.variable.get_constant("deny_upper")) != -1:
141
l_split_rule = rule.split(self.variable.get_constant("deny_upper"))
142
return_split_rule.append(l_split_rule[0].strip())
142
split_rule = rule.split(self.variable.get_constant("deny_upper"))
143
return_split_rule.append(split_rule[0].strip())
143
144
return_split_rule.append(self.variable.get_constant("deny_upper"))
144
return_split_rule.append(l_split_rule[1].strip())
145
return_split_rule.append(split_rule[1].strip())
147
148
elif rule.find(self.variable.get_constant("allow_upper")) != -1:
148
l_split_rule = rule.split(self.variable.get_constant("allow_upper"))
149
return_split_rule.append(l_split_rule[0].strip())
149
split_rule = rule.split(self.variable.get_constant("allow_upper"))
150
return_split_rule.append(split_rule[0].strip())
150
151
return_split_rule.append(self.variable.get_constant("allow_upper"))
151
return_split_rule.append(l_split_rule[1].strip())
152
return_split_rule.append(split_rule[1].strip())
154
155
elif rule.find(self.variable.get_constant("limit_upper")) != -1:
155
l_split_rule = rule.split(self.variable.get_constant("limit_upper"))
156
return_split_rule.append(l_split_rule[0].strip())
156
split_rule = rule.split(self.variable.get_constant("limit_upper"))
157
return_split_rule.append(split_rule[0].strip())
157
158
return_split_rule.append(self.variable.get_constant("limit_upper"))
158
return_split_rule.append(l_split_rule[1].strip())
159
return_split_rule.append(split_rule[1].strip())
161
162
elif rule.find(self.variable.get_constant("reject_upper")) != -1:
162
l_split_rule = rule.split(self.variable.get_constant("reject_upper"))
163
return_split_rule.append(l_split_rule[0].strip())
163
split_rule = rule.split(self.variable.get_constant("reject_upper"))
164
return_split_rule.append(split_rule[0].strip())
164
165
return_split_rule.append(self.variable.get_constant("reject_upper"))
165
return_split_rule.append(l_split_rule[1].strip())
166
return_split_rule.append(split_rule[1].strip())
167
168
return return_split_rule
172
def get_old_size_window(self):
173
width,height = self.frontend.get_old_size_window()
177
# Save actual size window
178
def save_size_window(self, win_width, win_height):
179
self.frontend.save_size_window(win_width, win_height)