theHarvester/discovery/googleCSE.py

101 lines
3.5 KiB
Python

import sys
import myparser
import re
import time
import requests
class search_googleCSE:
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 = ""
self.cse_id = ""
self.lowRange = start
self.highRange = start+100
def do_search(self):
url = 'http://' + 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=%40\"" + 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 = 'http://' + 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()
#time.sleep(1)
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()
#print "\tSearching " + str(self.counter+self.lowRange) + " results...\t\t\t\t\t\r"
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...")