def dork_generator_chain(self, dbtype): """ Helper method to constructs chain of objects to satify dependencies for the dork_generator. Returns an instance of dork_page_generator. """ if dbtype == "sql": engine = create_engine('sqlite:///') #Create mock of empty main db helpers.populate_main_sql_testdatabase(engine) db = database_sqla.Database(engine) elif dbtype == "mongodb": conn_string = helpers.create_mongo_database(fill=True) db = database_mongo.Database(helpers.create_mongo_database) else: raise Exception("Unsupported database type: {0}".format(dbtype)) reduced_dorks_file = os.path.join( os.path.split(os.path.abspath(__file__))[0], 'data/dorks_reduced.txt') file_processor = DorkFileProcessor(db, dorks_file=reduced_dorks_file) #setting the bar low for testing clusterer = cluster.Cluster("/\w+", 1, 1, 1, min_df=0.0) dork_generator = DorkPageGenerator(db, file_processor, clusterer, self.datadir) return db, engine, dork_generator
def test_honeypot_sql(self): """Objective: Testing overall Honeypot integration. Input: Loads the honeypot module with mongodb as main database. Expected Response: Honeypot responses with a non-empty HTTP response. Note: This test verifies the overall functionality.""" db_file = tempfile.mkstemp()[1] conn_string = "sqlite:///{0}".format(db_file) sql_engine = create_engine(conn_string) helpers.populate_main_sql_testdatabase(sql_engine) config_file = tempfile.mkstemp()[1] with open(config_file, 'w') as f: f.writelines(helpers.gen_config(conn_string)) try: raw_request = "GET /honeypot_test HTTP/1.1\r\nHost: honeypot\r\n\r\n" source_address = ["127.0.0.1", "12345"] sensor_address = ["1.2.3.4", "8080"] GlastopfHoneypot.prepare_environment(self.tmpdir) self.glastopf = GlastopfHoneypot(work_dir=self.tmpdir, config=config_file) self.glastopf.options["enabled"] = "False" print "Sending request: http://localhost:8080/" response = self.glastopf.handle_request(raw_request, source_address, sensor_address) self.assertIsNot(response, None) finally: if os.path.isfile(config_file): os.remove(config_file) if os.path.isfile(db_file): os.remove(db_file)
def test_honeypot_sql(self): """Objective: Testing overall Honeypot integration. Input: Loads the honeypot module with mongodb as main database. Expected Response: Honeypot responses with a non-empty HTTP response. Note: This test verifies the overall functionality.""" db_file = tempfile.mkstemp()[1] conn_string = "sqlite:///{0}".format(db_file) sql_engine = create_engine(conn_string) helpers.populate_main_sql_testdatabase(sql_engine) config_file = tempfile.mkstemp()[1] with open(config_file, "w") as f: f.writelines(helpers.gen_config(conn_string)) try: raw_request = "GET /honeypot_test HTTP/1.1\r\nHost: honeypot\r\n\r\n" source_address = ["127.0.0.1", "12345"] sensor_address = ["1.2.3.4", "8080"] GlastopfHoneypot.prepare_environment(self.tmpdir) self.glastopf = GlastopfHoneypot(work_dir=self.tmpdir, config=config_file) self.glastopf.options["enabled"] = "False" print "Sending request: http://localhost:8080/" response = self.glastopf.handle_request(raw_request, source_address, sensor_address) self.assertIsNot(response, None) finally: if os.path.isfile(config_file): os.remove(config_file) if os.path.isfile(db_file): os.remove(db_file)
def dork_generator_chain(self, dbtype): """ Helper method to constructs chain of objects to satify dependencies for the dork_generator. Returns an instance of dork_page_generator. """ if dbtype == "sql": engine = create_engine('sqlite:///') #Create mock of empty main db helpers.populate_main_sql_testdatabase(engine) db = database_sqla.Database(engine) elif dbtype == "mongodb": conn_string = helpers.create_mongo_database(fill=True) db = database_mongo.Database(helpers.create_mongo_database) else: raise Exception("Unsupported database type: {0}".format(dbtype)) reduced_dorks_file = os.path.join(os.path.split(os.path.abspath(__file__))[0], 'data/dorks_reduced.txt') file_processor = DorkFileProcessor(db, dorks_file=reduced_dorks_file) dork_generator = DorkPageGenerator(db, file_processor, self.datadir) return db, engine, dork_generator
def dork_generator_chain(self, dbtype, pages_dir): """ Helper method to constructs chain of objects to satify dependencies for the dork_generator. Returns an instance of dork_page_generator. """ if dbtype == "sql": engine = create_engine('sqlite:///') #Create mock of empty main db helpers.populate_main_sql_testdatabase(engine) db = database_sqla.Database(engine) elif dbtype == "mongodb": conn_string = helpers.create_mongo_database(fill=True) db = database_mongo.Database(helpers.create_mongo_database) else: raise Exception("Unsupported database type: {0}".format(dbtype)) reduced_dorks_file = os.path.join(os.path.split(os.path.abspath(__file__))[0], 'data/dorks_reduced.txt') file_processor = DorkFileProcessor(db, dorks_file=reduced_dorks_file) #setting the bar low for testing clusterer = cluster.Cluster("/\w+", 1, 1, 1, min_df=0.0) data_dir = os.getcwd() + "/modules/handlers/emulators/data" dork_generator = DorkPageGenerator(db, file_processor, clusterer, data_dir=data_dir, pages_dir=pages_dir) return db, engine, dork_generator