mirror of
https://github.com/laramies/theHarvester.git
synced 2024-09-22 00:06:30 +08:00
104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
from discovery.constants import *
|
|
from parsers import myparser
|
|
import re
|
|
import requests
|
|
import sys
|
|
import time
|
|
|
|
|
|
class SearchGoogleCSE:
|
|
|
|
def __init__(self, word, limit, start):
|
|
self.word = word
|
|
self.files = "pdf"
|
|
self.results = ""
|
|
self.totalresults = ""
|
|
self.server = "www.googleapis.com"
|
|
self.hostname = "www.googleapis.com"
|
|
self.userAgent = "(Mozilla/5.0 (Windows; U; Windows NT 6.0;en-US; rv:1.9.2) Gecko/20100115 Firefox/3.6"
|
|
self.quantity = "10"
|
|
self.limit = limit
|
|
self.counter = 1
|
|
self.api_key = googleCSEAPI_key
|
|
if self.api_key == "":
|
|
raise MissingKey(True)
|
|
self.cse_id = googleCSE_id
|
|
if self.cse_id == "":
|
|
raise MissingKey(False)
|
|
self.lowRange = start
|
|
self.highRange = start + 100
|
|
|
|
def do_search(self):
|
|
url = 'https://' + self.server + "/customsearch/v1?key=" + self.api_key + "&highrange=" + str(self.highRange) \
|
|
+ '&lowrange=' + str(self.lowRange) + '&cx=' + self.cse_id + "&start=" + str(self.counter) + \
|
|
"&q=" + self.word
|
|
headers = {
|
|
'Host': self.server,
|
|
'User-agent': self.userAgent
|
|
}
|
|
|
|
h = requests.get(url=url, headers=headers)
|
|
self.results = h.text
|
|
self.totalresults += self.results
|
|
|
|
def do_search_files(self, files):
|
|
url = 'https://' + self.server + "/customsearch/v1?key=" + self.api_key + "&highRange=" + str(self.highRange) \
|
|
+ '&lowRange=' + str(self.lowRange) + '&cx=' + self.cse_id + "&start=" + str(self.counter) + \
|
|
"&q=filetype:" + files + "%20site:" + self.word
|
|
headers = {
|
|
'Host': self.server,
|
|
'User-agent': self.userAgent
|
|
}
|
|
h = requests.get(url=url, headers=headers)
|
|
self.results = h.text
|
|
self.totalresults += self.results
|
|
|
|
def check_next(self):
|
|
renext = re.compile('> Next <')
|
|
nextres = renext.findall(self.results)
|
|
if nextres != []:
|
|
nexty = "1"
|
|
else:
|
|
nexty = "0"
|
|
return nexty
|
|
|
|
def get_emails(self):
|
|
rawres = myparser.Parser(self.totalresults, self.word)
|
|
return rawres.emails()
|
|
|
|
def get_hostnames(self):
|
|
rawres = myparser.Parser(self.totalresults, self.word)
|
|
return rawres.hostnames()
|
|
|
|
def get_files(self):
|
|
rawres = myparser.Parser(self.totalresults, self.word)
|
|
return rawres.fileurls(self.files)
|
|
|
|
def process(self):
|
|
tracker = self.counter + self.lowRange
|
|
while tracker <= self.limit:
|
|
self.do_search()
|
|
ESC = chr(27)
|
|
sys.stdout.write(ESC + '[2K' + ESC + '[G')
|
|
sys.stdout.write("\r\t" + "Searching " + str(self.counter + self.lowRange) + " results.")
|
|
sys.stdout.flush()
|
|
if self.counter == 101:
|
|
self.counter = 1
|
|
self.lowRange += 100
|
|
self.highRange += 100
|
|
else:
|
|
self.counter += 10
|
|
tracker = self.counter + self.lowRange
|
|
|
|
def store_results(self):
|
|
filename = "debug_results.txt"
|
|
file = open(filename, 'w')
|
|
file.write(self.totalresults)
|
|
|
|
def process_files(self, files):
|
|
while self.counter <= self.limit:
|
|
self.do_search_files(files)
|
|
time.sleep(1)
|
|
self.counter += 100
|
|
print("\tSearching " + str(self.counter) + " results.")
|