大家都知道，在進行滲透測試最重要的一個環(huán)節(jié)就是信息收集。說到信息收集怎么能不讓能聯(lián)想到谷歌以及谷歌語法呢？如果是人工的搜索并且提取相關信息無疑是一個工作量很大的一個事情。并且也是相當浪費時間的一個事情。那么我們如何快速地去除一些重復性工作呢？通過代碼語言無疑是我們一個很好的選擇，接下來我將介紹使用Python3 + Selenium + pymysql + Google來完成這樣一個重復性任務。下面我們將以提取站點域名以及URL為例。
首先我先介紹一下Selenium庫以及pymysql是什么？Python3 以及Google我就不贅述了。簡單地說明一下，selenium 是一個用于Web應用程序測試的工具。Selenium測試直接運行在瀏覽器中，就像真正的用戶在操作一樣。那么我們可以利用該庫去操作WEB瀏覽器。pymysql 是python3當中操控數(shù)據(jù)庫的一個組件。
下面我們來分析，我們需要利用Selenium來自動運行WEB瀏覽器。我們需要在瀏覽器上做什么樣的操作呢？輸入谷歌url，在谷歌搜索欄輸入谷歌語法，然后提取我們需要的內容。對不對？好的，那么我們腦子里面已經(jīng)有了一個初步的概念。接下來，我們需要做的是告訴Selenium我們的想法，并且它需要做的事情。接下來，我們來看下面的代碼：
#提取數(shù)據(jù)class driver_1(object): def __init__(self): self.driver = webdriver.Chrome(‘D:chromedriver_win32chromedriver.exe’) #訪問谷歌，搜索信息 def logload_Chrome(self,search): self.driver.get(‘https://www.google.com/’) #定位搜索欄位置 Search = self.driver.find_element_by_name(‘q’) self.driver.implicitly_wait(30) Search.send_keys(search) Search.send_keys(Keys.ENTER) #提取站點信息 def Search_site(self,site,num): dict_name_url = {} now_site = ‘site:’ + ”’ + site + ”’ self.logload_Chrome(now_site) #提取url和name for a in range(0,num): if a > 0 and a a[href]’) i = 0 while i a[href]’) i = 0 while i < len(Result_name): dict_name_url[str(Result_name[i].text)] = str(Result_url[i].get_attribute('href')) i += 1 return dict_name_url
上述代碼的作用是：登錄谷歌并且輸入搜索內容，并且通過循環(huán)來提取Google搜索出來的內容。
當我們順利地提取出內容的時候就可以將內容放入數(shù)據(jù)庫。這個時候，我們需要利用到pymysql庫。通過它我們可以操控數(shù)據(jù)庫。詳細情況請看下面代碼：
#保存數(shù)據(jù)信息class Save_to_DB(object): #鏈接出入數(shù)據(jù) def connect(self,i,key,value): conn = pymysql.connect( host = ‘localhost’, user = ‘root’, password = ‘Password@2020’, database = ‘suda’ ) #創(chuàng)建游標對象 cursor = conn.cursor() try: cursor.execute(“insert into information_site_suda values(%s,’%s’,’%s’);” %(int(i),key,value)) conn.commit() conn.close() except: conn.ping() cursor = conn.cursor() cursor.execute(“insert into information_site values(%s,’%s’,’%s’);” % (int(i), key, value)) conn.commit() conn.close()
上述代碼的作用是將查詢出來的數(shù)據(jù)存放入數(shù)據(jù)庫當中。
既然功能模塊已經(jīng)實現(xiàn)，接下來就是調用了。詳細代碼可以查看附件（注意：中的一些部分需要修改才能運行，不如，MYSQL數(shù)據(jù)庫名等等）。該文章或許一些人看不懂。沒有關系，文章當中的模塊可以自己學習并不難。主要是希望各位了解到盡量不要去做重復性的勞動。學會利用工具。我一直認為代碼功底是區(qū)分腳本小子和大牛的分水線。請各位一起加油?。?！
