diff --git a/domi_owned/enumerate.py b/domi_owned/enumerate.py index 28c4a78..d85134c 100644 --- a/domi_owned/enumerate.py +++ b/domi_owned/enumerate.py @@ -97,14 +97,16 @@ def enum_dirs(self, urls): else: client = aiohttp.ClientSession(headers=self.utilities.HEADERS, loop=loop) - with client as session: - try: - loop.run_until_complete(self.query(session, urls)) - except asyncio.CancelledError: - sys.exit() - except Exception as error: - self.logger.error('An error occurred while enumerating Domino URLs') - sys.exit() + #with client as session: + try: + task = loop.create_task(self.query(client, urls)) + loop.run_until_complete(task) + loop.close() + except asyncio.CancelledError: + sys.exit() + except Exception as error: + self.logger.error('An error occurred while enumerating Domino URLs') + sys.exit() async def query(self, session, urls): """ diff --git a/domi_owned/hashdump.py b/domi_owned/hashdump.py index d99f3bc..065fb68 100644 --- a/domi_owned/hashdump.py +++ b/domi_owned/hashdump.py @@ -72,7 +72,7 @@ def get_accounts(self): if 'No documents found' in str(soup.findAll('h2')): break else: - links = [a.attrs.get('href') for a in soup.select('a[href^=/names.nsf/]')] + links = [a.attrs.get('href') for a in soup.select('a[href^="/names.nsf/"]')] for link in links: if self.utilities.ACCOUNT_REGEX.search(link): account_url = "{0}/names.nsf/{1}?OpenDocument".format(self.url, self.utilities.ACCOUNT_REGEX.search(link).group(1)) @@ -112,14 +112,16 @@ def get_hashes(self, urls): else: client = aiohttp.ClientSession(headers=self.utilities.HEADERS, loop=loop) - with client as session: - try: - loop.run_until_complete(self.query(session, urls)) - except asyncio.CancelledError: - sys.exit() - except Exception as error: - self.logger.error('An error occurred while dumping Domino account hashes') - sys.exit() + #with client as session: + try: + task = loop.create_task(self.query(client, urls)) + loop.run_until_complete(task) + loop.close() + except asyncio.CancelledError: + sys.exit() + except Exception as error: + self.logger.error('An error occurred while dumping Domino account hashes') + sys.exit() async def query(self, session, urls): """