def SetCurrentFile(self, filename):
self.current_file_uri = 'file://' + os.path.abspath(filename)
self.pipeline.set_property('uri', self.current_file_uri)
discoverer = Discoverer(os.path.abspath(filename))
discoverer.connect('discovered', self._OnDiscovered)
self.discovery_start_time = time.time()
discoverer.discover()
self.ready_to_play = False #will be ready when discoverer has finished
def verify_node_test():
disco = Discoverer()
disco.get_list_of_nodes()
n1 = disco.nodes_list[0]
ret = disco.verify_node(n1)
def __init__(self, config=None):
self.config_mdb_atlas = config.get('mdb_atlas')
self.config_email = config.get('email')
self.config_sms = config.get('sms')
self._discoverer = Discoverer(None, None)
self._nmapper = Nmapper()
self._alerter = Alerter()
self._scan_db = MongoDBClient(
self.config_mdb_atlas.get('scan_cluster_name'),
self.config_mdb_atlas.get('scan_username'),
self.config_mdb_atlas.get('scan_password'),
self.config_mdb_atlas.get('scan_db'))
self._running_event = None
self._previous_scan = {'hosts_list': [], 'hosts_ports_scan': []}
self._email_client = EmailClient(
self.config_email.get('email_from_addr'),
self.config_email.get('email_password'),
self.config_email.get('email_smtp_server'),
self.config_email.get('email_smtp_server_port'))
self._sms_client = SmsClient(self.config_sms.get('sms_account_sid'),
self.config_sms.get('sms_auth_token'))
if not os.path.exists(output_home):
logger.info('Output path does not exists and created.')
os.makedirs(output_home)
threshold_parameter = namedtuple(
'threshold_parameter', ['tf', 'agg_coef', 'max_entropy', 'min_entropy'])
threshold_parameters = dict()
threshold_parameters['bigram'] = threshold_parameter(*args.bigram)
threshold_parameters['latin'] = threshold_parameter(*args.latin)
threshold_parameters[2] = threshold_parameter(*args.unigram_2)
threshold_parameters[3] = threshold_parameter(*args.unigram_3)
dictionary = load_dictionary(args.dictionary_path)
discoverer = Discoverer(save_segmentation=False)
# Used to store stats generated in each iteration.
stats_ind = list()
import time
for iteration in range(args.iteration):
time.sleep(1)
logger.info("""
**********************************************************************
commencing iteration {}...
**********************************************************************
""".format(iteration + 1))
