def set_ignore_ereg_path(self, ignore_ereg_path):
ValueChecker.is_string(ignore_ereg_path)
if None is self.__ignore_ereg_path:
self.__ignore_ereg_path = []
self.__ignore_ereg_path.append(ignore_ereg_path)
return self
def set_ereg_path(self, ereg_path):
ValueChecker.is_string(ereg_path)
if None is self.__ereg_path:
self.__ereg_path = []
self.__ereg_path.append(ereg_path)
return self
def add_exclude(self, exclude):
ValueChecker.is_string(exclude)
if exclude not in self.__exclude:
self.__exclude.append(exclude)
return self
def set_proxy_authorization(self, username, password):
ValueChecker.is_string(username)
ValueChecker.is_string(password)
self.__proxy_authorization = True
self.__proxy_authorization_username = username
self.__proxy_authorization_password = password
return self
def add_D(self, D):
ValueChecker.is_string(D)
if D not in self.__D:
self.__D.append(D)
return self
def add_p(self, p):
ValueChecker.is_string(p)
if p not in self.__p:
self.__p.append(p)
return self
def add_script(self, script):
ValueChecker.is_string(script)
if script not in self.__script:
self.__script.append(script)
return self
def add_script_args(self, script_args):
ValueChecker.is_string(script_args)
if script_args not in self.__script_args:
self.__script_args.append(script_args)
return self
def add_host(self, host):
ValueChecker.is_string(host)
if host not in self.__host:
self.__host.append(host)
return self
def add_exclude_ports(self, exclude_ports):
ValueChecker.is_string(exclude_ports)
if exclude_ports not in self.__exclude_ports:
self.__exclude_ports.append(exclude_ports)
return self
def set_exclude_type(self, exclude_type):
ValueChecker.is_string(exclude_type)
if None is self.__exclude_type:
self.__exclude_type = []
self.__exclude_type.append(exclude_type)
return self
def add_dns_servers(self, dns_servers):
ValueChecker.is_string(dns_servers)
if dns_servers not in self.__dns_servers:
self.__dns_servers.append(dns_servers)
return self
def add_proxies(self, proxies):
ValueChecker.is_string(proxies)
if proxies not in self.__proxies:
self.__proxies.append(proxies)
return self
def remove_var(self, key):
ValueChecker.is_string(key)
var = self.get_var(key)
if None != var:
self.__vars.remove((key, var))
return self
def get_var(self, key):
ValueChecker.is_string(key)
for var in self.__vars:
if var[0] == key:
return var[0]
return None
def remove_least_port_strength(self, ip, port, strength):
ValueChecker.is_number(port)
ValueChecker.is_string(ip)
self.validate_strength(strength)
config = ip + '/' + str(port) + '/' + '/' + strength
self.__least_port_strength.remove(config)
return self
def remove_allowed_port(self, port, protocol):
ValueChecker.is_number(port)
ValueChecker.is_string(protocol)
protocol = protocol.lower()
config = (port, protocol)
self.__open_ports.remove(config)
return self
def add_partition(self, id, path, warning_percent, critical_percent):
ValueChecker.is_string(id)
ValueChecker.is_string(path)
ValueChecker.is_number(warning_percent)
ValueChecker.is_number(critical_percent)
self.__check_partitions.append(
(id, path, warning_percent, critical_percent))
return self
def add_least_port_strength(self, ip, port, strength):
ValueChecker.is_number(port)
ValueChecker.is_string(ip)
self.validate_strength(strength)
config = ip + '/' + str(port) + '/' + strength
if config not in self.__least_port_strength:
self.__least_port_strength.append(config)
return self
def add_allowed_port(self, port, protocol):
ValueChecker.is_number(port)
ValueChecker.is_string(protocol)
protocol = protocol.lower()
config = (port, protocol)
if config not in self.__open_ports:
self.__open_ports.append(config)
return self
def remove_allowed_cipher(self, ip, port, protocol, cipher_names):
ValueChecker.is_number(port)
ValueChecker.is_string(protocol)
ValueChecker.is_string(ip)
ValueChecker.is_array(cipher_names)
config = ip + '/' + str(port) + '/' + protocol + '/' + ','.join(
cipher_names)
self.__allowed_ciphers.remove(config)
return self
def add_allowed_cipher(self, ip, port, protocol, cipher_names):
ValueChecker.is_number(port)
ValueChecker.is_string(ip)
ValueChecker.is_string(protocol)
ValueChecker.is_array(cipher_names)
config = ip + '/' + str(port) + '/' + protocol + '/' + ','.join(
cipher_names)
if config not in self.__allowed_ciphers:
self.__allowed_ciphers.append(config)
return self
def remove_allowed_port(self, port, protocol):
ValueChecker.is_number(port)
ValueChecker.is_string(protocol)
protocol = protocol.lower()
if protocol not in ['udp', 'tcp']:
raise Exception('Protocol must be UDP or TCP in ' + self.get_id())
config = str(port) + '/' + protocol
self.__allowed_ports.remove(config)
return self
def add_ufw_rule(self, policy_from, policy_to, policy_action):
ValueChecker.is_string(policy_from)
ValueChecker.is_string(policy_to)
ValueChecker.is_string(policy_action)
self.__ufw_rules.append(
(policy_from.replace(' ', '-'), policy_to.replace(' ', '-'),
policy_action.replace(' ', '-')))
return self
def set_send(self, send):
ValueChecker.is_string(send)
self.__send = send
return self
def set_host(self, host):
ValueChecker.is_string(host)
self.__host = host
return self
def set_sni(self, sni):
ValueChecker.is_string(sni)
self.__sni = sni
return self
def set_mismatch_state(self, mismatch_state):
ValueChecker.is_string(mismatch_state)
self.__mismatch_state = mismatch_state
return self
def set_refuse_state(self, refuse_state):
ValueChecker.is_string(refuse_state)
self.__refuse_state = refuse_state
return self
def set_quit(self, quit):
ValueChecker.is_string(quit)
self.__quit = quit
return self
