def setUp(self): meta = {} run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-dir") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) page_dir = self.main_page_path.rsplit('/')[-1] args = run_args.parse_args(['--page-dir', page_dir]) self.loop = asyncio.new_event_loop() self.data = { 'method': 'GET', 'path': '/', 'headers': { 'Host': 'test_host', 'Connection': 'keep-alive', 'Upgrade-Insecure-Requests': '1', 'User-Agent': 'test_agent', 'Accept': 'text/html', 'Accept-Encoding': 'test_encoding', 'Accept-Language': 'test_lang', 'Cookie': 'test_cookie', 'uuid': 'test_uuid', 'peer': {'ip': '::1', 'port': 80}, 'status': 200, 'cookies': 'test_cookies', ' sess_uuid': 'test_uuid' } } aiohttp.ClientSession.post = AsyncMock( return_value=aiohttp.ClientResponse( url=yarl.URL("http://www.example.com"), method="GET", writer=None, continue100=1, timer=None, request_info=None, traces=None, loop=self.loop, session=None ) ) uuid = "test_uuid" args.tanner = "tanner.mushmush.org" args.no_dorks = True self.handler = TannerHandler(args, meta, uuid) self.result = None
def setUp(self): run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-dir") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) page_dir = self.main_page_path.rsplit('/')[-1] meta_content = {"/index.html": {"hash": "hash_name", "content_type": "text/html"}} self.page_content = "<html><body></body></html>" self.content_type = "text/html" with open(os.path.join(self.main_page_path, "hash_name"), 'w') as f: f.write(self.page_content) with open(os.path.join(self.main_page_path, "meta.json"), 'w') as f: json.dump(meta_content, f) self.args = run_args.parse_args(['--page-dir', page_dir]) self.args.index_page = '/index.html' self.args.no_dorks = True self.args.tanner = "tanner.mushmush.org" self.uuid = "test_uuid" self.handler = TannerHandler(self.args, meta_content, self.uuid) self.requested_name = '/' self.loop = asyncio.get_event_loop() self.handler.html_handler.handle_content = AsyncMock(return_value=self.page_content) self.res1 = None self.res2 = None self.res3 = None self.res4 = None self.detection = None self.expected_content = None self.call_content = None self.expected_header = None self.expected_status_code = None
def setUp(self): self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.content = '<html><head>title</head><body>sample</body></html>' self.page_dir = self.main_page_path.rsplit('/')[-1] self.index_page = "index.html" with open(os.path.join(self.main_page_path, 'index.html'), 'w') as f: f.write(self.content)
def setUp(self): self.url = 'http://example.com' self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.expected_new_url = yarl.URL('http://example.com') self.expected_err_url = yarl.URL('http://example.com/status_404') self.max_depth = sys.maxsize self.css_validate = 'false' self.handler = Cloner(self.url, self.max_depth, self.css_validate)
def setUp(self): meta = {} run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-dir") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.page_dir = self.main_page_path.rsplit('/')[-1] args = run_args.parse_args(['--page-dir', self.page_dir]) args_dict = vars(args) args_dict['full_page_path'] = self.main_page_path uuid = '9c10172f-7ce2-4fb4-b1c6-abc70141db56'.encode('utf-8') args.tanner = 'tanner.mushmush.org' args.no_dorks = True args.server_header = "test_server" args.slurp_enabled = True self.handler = HttpRequestHandler(meta, args, uuid) self.request_data = { 'method': 'GET', 'path': '/', 'headers': { 'Host': 'test_host', 'Content-Type': 'test_type', }, 'status': 200, 'cookies': { 'sess_uuid': 'prev_test_uuid', }, } self.loop = asyncio.new_event_loop() self.response_content = '<html><body></body></html>' self.response_headers = multidict.CIMultiDict([("Content-Type", "text/html")]) self.response_status = 200 event_result = dict(response=dict(message=dict(detection={'type': 1}, sess_uuid="test_uuid"))) RequestHandler = Mock() protocol = RequestHandler() message = RawRequestMessage( method='POST', path='/', version=HttpVersion(major=1, minor=1), headers=self.request_data['headers'], raw_headers=None, should_close=None, compression=None, upgrade=None, chunked=None, url=URL('http://test_url/') ) self.request = web.Request( message=message, payload=None, protocol=protocol, payload_writer=None, task='POST', loop=self.loop ) self.handler.tanner_handler.create_data = Mock(return_value=self.request_data) self.handler.tanner_handler.submit_data = AsyncMock(return_value=event_result) self.handler.submit_slurp = AsyncMock() web.Response.add_header = Mock() web.Response.write = Mock() web.Response.send_headers = Mock() web.Response.write_eof = AsyncMock() aiohttp.streams.EmptyStreamReader.read = AsyncMock( return_value=b'con1=test1&con2=test2') self.handler.tanner_handler.parse_tanner_response = AsyncMock( return_value=(self.response_content, self.response_headers, self.response_status))
def setUp(self): self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.url = yarl.URL('http://foo.com') self.root = 'http://example.com' self.max_depth = sys.maxsize self.loop = asyncio.new_event_loop() self.css_validate = 'false' self.handler = Cloner(self.root, self.max_depth, self.css_validate) self.filename = None self.hashname = None
def setUp(self): self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.root = "http://example.com" self.level = 0 self.max_depth = sys.maxsize self.loop = asyncio.new_event_loop() self.css_validate = "false" self.handler = Cloner(self.root, self.max_depth, self.css_validate) self.content = None self.expected_content = None self.return_content = None
def setUp(self): meta = {} run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-dir") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) page_dir = self.main_page_path.rsplit("/")[-1] args = run_args.parse_args(["--page-dir", page_dir]) args_dict = vars(args) args_dict["full_page_path"] = self.main_page_path self.loop = asyncio.new_event_loop() self.data = { "method": "GET", "path": "/", "headers": { "Host": "test_host", "Connection": "keep-alive", "Upgrade-Insecure-Requests": "1", "User-Agent": "test_agent", "Accept": "text/html", "Accept-Encoding": "test_encoding", "Accept-Language": "test_lang", "Cookie": "test_cookie", }, "uuid": "test_uuid", "peer": { "ip": "::1", "port": 80 }, "status": 200, "cookies": "test_cookies", "sess_uuid": "test_uuid", } aiohttp.ClientSession.post = AsyncMock( return_value=aiohttp.ClientResponse( url=yarl.URL("http://www.example.com"), method="GET", writer=None, continue100=1, timer=None, request_info=None, traces=None, loop=self.loop, session=None, )) uuid = "test_uuid" args.tanner = "tanner.mushmush.org" args.no_dorks = True self.handler = TannerHandler(args, meta, uuid) self.result = None
def setUp(self): meta = {} run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-location") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) page_location = self.main_page_path.rsplit('/')[-1] args = run_args.parse_args(['--page-location', page_location]) args_dict = vars(args) args_dict['full_page_path'] = self.main_page_path self.loop = asyncio.new_event_loop() self.data = { 'method': 'GET', 'path': '/', 'headers': { 'Host': 'test_host', 'Connection': 'keep-alive', 'Upgrade-Insecure-Requests': '1', 'User-Agent': 'test_agent', 'Accept': 'text/html', 'Accept-Encoding': 'test_encoding', 'Accept-Language': 'test_lang', 'Cookie': 'test_cookie' }, 'uuid': 'test_uuid', 'peer': { 'ip': '::1', 'port': 80 }, 'status': 200, 'cookies': 'test_cookies', 'sess_uuid': 'test_uuid' } aiohttp.ClientSession.post = AsyncMock( return_value=aiohttp.ClientResponse(url=yarl.URL( "http://www.example.com"), method="GET", writer=None, continue100=1, timer=None, request_info=None, traces=None, loop=self.loop, session=None)) uuid = "test_uuid" args.tanner = "tanner.mushmush.org" args.no_dorks = True self.handler = TannerHandlerDirectory(args, meta, uuid) self.result = None
def setUp(self): meta = {} run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-dir") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.page_dir = self.main_page_path.rsplit('/')[-1] args = run_args.parse_args(['--page-dir', self.page_dir]) uuid = '9c10172f-7ce2-4fb4-b1c6-abc70141db56'.encode('utf-8') args.tanner = 'tanner.mushmush.org' args.no_dorks = True args.server_header = "test_server" args.slurp_enabled = True self.handler = HttpRequestHandler(meta, args, uuid) self.data = { 'method': 'GET', 'path': '/', 'headers': { 'Host': 'test_host', 'status': 200 }, 'cookies': { 'sess_uuid': 'prev_test_uuid' } } self.loop = asyncio.new_event_loop() self.content = '<html><body></body></html>' self.content_type = 'test_type' event_result = dict(response=dict(message=dict(detection={'type': 1}, sess_uuid="test_uuid"))) RequestHandler = Mock() protocol = RequestHandler() message = RawRequestMessage( method='POST', path='/', version=HttpVersion(major=1, minor=1), headers=self.data['headers'], raw_headers=None, should_close=None, compression=None, upgrade=None, chunked=None, url=URL('http://test_url/') ) self.request = web.Request( message=message, payload=None, protocol=protocol, payload_writer=None, task='POST', loop=self.loop ) self.handler.tanner_handler.create_data = Mock(return_value=self.data) self.handler.tanner_handler.submit_data = AsyncMock(return_value=event_result) self.handler.submit_slurp = AsyncMock() web.Response.add_header = Mock() web.Response.write = Mock() web.Response.send_headers = Mock() web.Response.write_eof = AsyncMock() aiohttp.streams.EmptyStreamReader.read = AsyncMock(return_value=b'con1=test1&con2=test2') self.handler.tanner_handler.parse_tanner_response = AsyncMock( return_value=(self.content, self.content_type, self.data['headers'], self.data['headers']['status']))
def setUp(self): self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.dorks = dict(response={'dorks': "test_dorks"}) self.loop = asyncio.new_event_loop() aiohttp.ClientSession.get = AsyncMock( return_value=aiohttp.ClientResponse( url=yarl.URL("http://www.example.com"), method="GET", writer=None, continue100=1, timer=None, request_info=None, traces=None, loop=self.loop, session=None ) ) no_dorks = True tanner = "tanner.mushmush.org" self.handler = HtmlHandler(no_dorks, tanner) self.data = None
def setUp(self): self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.dorks = dict(response={'dorks': "test_dorks"}) self.loop = asyncio.new_event_loop() aiohttp.ClientSession.get = AsyncMock( return_value=aiohttp.ClientResponse( url=yarl.URL("http://www.example.com"), method="GET", writer=None, continue100=1, timer=None, request_info=None, traces=None, loop=self.loop, session=None ) ) no_dorks = True tanner = "tanner.mushmush.org" self.handler = HtmlHandler(no_dorks, tanner) self.data = None
def setUp(self): run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-location") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) page_location = self.main_page_path.rsplit('/')[-1] meta_content = { "/index.html": { "hash": "hash_name", "headers": [{ "Content-Type": "text/html" }] } } self.page_content = "<html><body></body></html>" self.headers = multidict.CIMultiDict([("Content-Type", "text/html")]) self.status_code = 200 self.content_type = "text/html" with open(os.path.join(self.main_page_path, "hash_name"), 'w') as f: f.write(self.page_content) with open(os.path.join(self.main_page_path, "meta.json"), 'w') as f: json.dump(meta_content, f) self.args = run_args.parse_args(['--page-location', page_location]) args_dict = vars(args) args_dict['full_page_path'] = self.main_page_path self.args.index_page = '/index.html' self.args.no_dorks = True self.args.tanner = "tanner.mushmush.org" self.uuid = "test_uuid" self.handler = TannerHandlerDirectory(self.args, meta_content, self.uuid) self.requested_name = '/' self.loop = asyncio.get_event_loop() self.handler.html_handler.handle_content = AsyncMock( return_value=self.page_content) self.res1 = None self.res2 = None self.res3 = None self.detection = None self.expected_content = None self.call_content = None
def setUp(self): self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.content = """ <html> <body> <p style="color:red;">A paragraph to be tested</p> </body> </html> """ self.expected_content = '<html>\n <body>\n <p style="color: red">\n' self.expected_content += ' <a href="test_dork1" style="color:red;text-decoration:none;cursor:text;">\n' self.expected_content += " A\n </a>\n paragraph to be tested\n </p>\n </body>\n</html>\n" self.no_dorks_content = '<html>\n <body>\n <p style="color:red;">\n A paragraph to be tested\n' self.no_dorks_content += " </p>\n </body>\n</html>\n" self.loop = asyncio.new_event_loop() self.return_content = None no_dorks = True tanner = "tanner.mushmush.org" self.handler = HtmlHandler(no_dorks, tanner)
def setUp(self): self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.content = ''' <html> <body> <p style="color:red;">A paragraph to be tested</p> </body> </html> ''' self.expected_content = '<html>\n <body>\n <p style="color: red">\n' self.expected_content += ' <a href="test_dork1" style="color:red;text-decoration:none;cursor:text;">\n' self.expected_content += ' A\n </a>\n paragraph to be tested\n </p>\n </body>\n</html>\n' self.no_dorks_content = '<html>\n <body>\n <p style="color:red;">\n A paragraph to be tested\n' self.no_dorks_content += ' </p>\n </body>\n</html>\n' self.loop = asyncio.new_event_loop() self.return_content = None no_dorks = True tanner = "tanner.mushmush.org" self.handler = HtmlHandler(no_dorks, tanner)
def setUp(self): meta = { "/status_404": { "hash": "bacfa45149ffbe8dbff34609bf56d748", "headers": [{ "Content-Type": "text/html; charset=UTF-8" }], } } run_args = argparse.ArgumentParser() self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) args = run_args.parse_args([]) args_dict = vars(args) args_dict["full_page_path"] = self.main_page_path uuid = "9c10172f-7ce2-4fb4-b1c6-abc70141db56".encode("utf-8") args.tanner = "tanner.mushmush.org" args.no_dorks = True args.host_ip = "127.0.0.1" args.port = "80" self.handler = HttpRequestHandler(meta, args, uuid) self.loop = asyncio.new_event_loop()
def setUp(self): run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-dir") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) page_dir = self.main_page_path.rsplit('/')[-1] meta_content = { "/index.html": { "hash": "hash_name", "content_type": "text/html" } } self.page_content = "<html><body></body></html>" self.content_type = "text/html" with open(os.path.join(self.main_page_path, "hash_name"), 'w') as f: f.write(self.page_content) with open(os.path.join(self.main_page_path, "meta.json"), 'w') as f: json.dump(meta_content, f) args = run_args.parse_args(['--page-dir', page_dir]) args.index_page = '/index.html' args.no_dorks = True args.tanner = "tanner.mushmush.org" uuid = "test_uuid" self.handler = TannerHandler(args, meta_content, uuid) self.requested_name = '/' self.loop = asyncio.get_event_loop() self.handler.html_handler.handle_content = AsyncMock( return_value=self.page_content) self.res1 = None self.res2 = None self.res3 = None self.res4 = None self.detection = None self.expected_content = None self.call_content = None
def setUp(self): meta = {} run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-dir") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) self.page_dir = self.main_page_path.rsplit("/")[-1] args = run_args.parse_args(["--page-dir", self.page_dir]) args_dict = vars(args) args_dict["full_page_path"] = self.main_page_path uuid = "9c10172f-7ce2-4fb4-b1c6-abc70141db56".encode("utf-8") args.tanner = "tanner.mushmush.org" args.no_dorks = True args.server_header = "test_server" args.slurp_enabled = True self.handler = HttpRequestHandler(meta, args, uuid) self.request_data = { "method": "GET", "path": "/", "headers": { "Host": "test_host", "Content-Type": "test_type", }, "status": 200, "cookies": { "sess_uuid": "prev_test_uuid", }, } self.loop = asyncio.new_event_loop() self.response_content = "<html><body></body></html>" self.response_headers = multidict.CIMultiDict([("Content-Type", "text/html")]) self.response_status = 200 event_result = dict(response=dict( message=dict(detection={"type": 1}, sess_uuid="test_uuid"))) RequestHandler = Mock() protocol = RequestHandler() message = RawRequestMessage( method="POST", path="/", version=HttpVersion(major=1, minor=1), headers=self.request_data["headers"], raw_headers=None, should_close=None, compression=None, upgrade=None, chunked=None, url=URL("http://test_url/"), ) self.request = web.Request( message=message, payload=None, protocol=protocol, payload_writer=None, task="POST", loop=self.loop, ) self.handler.tanner_handler.create_data = Mock( return_value=self.request_data) self.handler.tanner_handler.submit_data = AsyncMock( return_value=event_result) self.handler.submit_slurp = AsyncMock() web.Response.add_header = Mock() web.Response.write = Mock() web.Response.send_headers = Mock() web.Response.write_eof = AsyncMock() aiohttp.streams.EmptyStreamReader.read = AsyncMock( return_value=b"con1=test1&con2=test2") self.handler.tanner_handler.parse_tanner_response = AsyncMock( return_value=( self.response_content, self.response_headers, self.response_status, ))
def setUp(self): meta = {} run_args = argparse.ArgumentParser() run_args.add_argument("--tanner") run_args.add_argument("--page-dir") self.main_page_path = generate_unique_path() os.makedirs(self.main_page_path) page_dir = self.main_page_path.rsplit('/')[-1] args = run_args.parse_args(['--page-dir', page_dir]) args_dict = vars(args) args_dict['full_page_path'] = self.main_page_path snare_uuid = '9c10172f-7ce2-4fb4-b1c6-abc70141db56'.encode('utf-8') args.no_dorks = True self.handler = TannerHandler(args, meta, snare_uuid) headers = { 'Host': 'test_host', 'status': 200, 'Cookie': 'sess_uuid=prev_test_uuid; test_cookie=test' } message = RawRequestMessage(method='POST', path='/', version=HttpVersion(major=1, minor=1), headers=headers, raw_headers=None, should_close=None, compression=None, upgrade=None, chunked=None, url=URL('http://test_url/')) loop = asyncio.get_event_loop() RequestHandler = Mock() protocol = RequestHandler() self.request = web.Request(message=message, payload=None, protocol=protocol, payload_writer=None, task='POST', loop=loop) self.request.transport.get_extra_info = Mock( return_value=(['test_ip', 'test_port'])) self.response_status = "test_status" self.data = None self.expected_data = { 'method': 'POST', 'path': 'http://test_url/', 'headers': { 'Host': 'test_host', 'status': 200, 'Cookie': 'sess_uuid=prev_test_uuid; test_cookie=test' }, 'uuid': '9c10172f-7ce2-4fb4-b1c6-abc70141db56', 'peer': { 'ip': 'test_ip', 'port': 'test_port' }, 'status': 'test_status', 'cookies': { 'sess_uuid': 'prev_test_uuid', ' test_cookie': 'test' } }