Add /usr/local/etc as possible location for files

This commit is contained in:
Alex Merkel 2020-10-05 16:33:35 +02:00
parent 177912431e
commit ddc2d96c93
No known key found for this signature in database
GPG key ID: B8F579CC2A6662A2
3 changed files with 44 additions and 79 deletions

View file

@ -31,11 +31,15 @@ def __init__(self, domain, dnsserver, verbose=False):
# self.dnsserver = [dnsserver] if isinstance(dnsserver, str) else dnsserver
self.dnsserver = list(map(str, dnsserver.split(','))) if isinstance(dnsserver, str) else dnsserver
try:
with open('wordlists/dns-names.txt', 'r') as file:
with open('/etc/theHarvester/wordlists/dns-names.txt', 'r') as file:
self.list = file.readlines()
except FileNotFoundError:
with open('/etc/theHarvester/dns-names.txt', 'r') as file:
self.list = file.readlines()
try:
with open('/usr/local/etc/theHarvester/wordlists/dns-names.txt', 'r') as file:
self.list = file.readlines()
except FileNotFoundError:
with open('wordlists/dns-names.txt', 'r') as file:
self.list = file.readlines()
self.domain = domain.replace('www.', '')
self.list = [f'{word.strip()}.{self.domain}' for word in self.list]

View file

@ -102,10 +102,18 @@ async def process_profiles(self):
async def append_dorks(self):
# Wrap in try-except incase filepaths are messed up.
try:
with open('wordlists/dorks.txt', mode='r') as fp:
with open('/etc/theHarvester/wordlists/dorks.txt', 'r') as fp:
self.dorks = [dork.strip() for dork in fp]
except FileNotFoundError as error:
print(error)
except FileNotFoundError:
try:
with open('/usr/local/etc/theHarvester/wordlists/dorks.txt', 'r') as fp:
self.dorks = [dork.strip() for dork in fp]
except FileNotFoundError:
try:
with open('wordlists/dorks.txt', 'r') as fp:
self.dorks = [dork.strip() for dork in fp]
except FileNotFoundError as error:
print(error)
async def construct_dorks(self):
# Format is: site:targetwebsite.com + space + inurl:admindork

View file

@ -15,103 +15,54 @@ def version() -> str:
return '3.2.0'
@staticmethod
def bing_key() -> str:
def api_keys() -> dict:
try:
with open('/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']['bing']['key']
return keys['apikeys']['bing']['key']
try:
with open('/usr/local/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']
@staticmethod
def bing_key() -> str:
return Core.api_keys()['bing']['key']
@staticmethod
def github_key() -> str:
try:
with open('/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']['github']['key']
return keys['apikeys']['github']['key']
return Core.api_keys()['github']['key']
@staticmethod
def hunter_key() -> str:
try:
with open('/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']['hunter']['key']
return keys['apikeys']['hunter']['key']
return Core.api_keys()['hunter']['key']
@staticmethod
def intelx_key() -> str:
try:
with open('/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']['intelx']['key']
return keys['apikeys']['intelx']['key']
return Core.api_keys()['intelx']['key']
@staticmethod
def pentest_tools_key() -> str:
try:
with open('/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']['pentestTools']['key']
return keys['apikeys']['pentestTools']['key']
return Core.api_keys()['pentestTools']['key']
@staticmethod
def projectdiscovery_key() -> str:
try:
with open('/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']['projectDiscovery']['key']
return keys['apikeys']['projectDiscovery']['key']
return Core.api_keys()['projectDiscovery']['key']
@staticmethod
def security_trails_key() -> str:
try:
with open('/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']['securityTrails']['key']
return keys['apikeys']['securityTrails']['key']
return Core.api_keys()['securityTrails']['key']
@staticmethod
def shodan_key() -> str:
try:
with open('/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']['shodan']['key']
return keys['apikeys']['shodan']['key']
return Core.api_keys()['shodan']['key']
@staticmethod
def spyse_key() -> str:
try:
with open('/etc/theHarvester/api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
except FileNotFoundError:
with open('api-keys.yaml', 'r') as api_keys:
keys = yaml.safe_load(api_keys)
return keys['apikeys']['spyse']['key']
return keys['apikeys']['spyse']['key']
return Core.api_keys()['spyse']['key']
@staticmethod
def proxy_list() -> List:
@ -119,10 +70,12 @@ def proxy_list() -> List:
with open('/etc/theHarvester/proxies.yaml', 'r') as proxy_file:
keys = yaml.safe_load(proxy_file)
except FileNotFoundError:
with open('proxies.yaml', 'r') as proxy_file:
keys = yaml.safe_load(proxy_file)
http_list = [f'http://{proxy}' for proxy in keys['http']] if keys['http'] is not None else []
return http_list
try:
with open('/usr/local/etc/theHarvester/proxies.yaml', 'r') as proxy_file:
keys = yaml.safe_load(proxy_file)
except FileNotFoundError:
with open('proxies.yaml', 'r') as proxy_file:
keys = yaml.safe_load(proxy_file)
http_list = [f'http://{proxy}' for proxy in keys['http']] if keys['http'] is not None else []
return http_list