21.4 Python 使用GeoIP2地圖定位

2023-10-24 21:01:44

GeoIP2是一種IP地址定位庫,它允許開發人員根據IP地址查詢有關位置和地理位置的資訊。它使用MaxMind公司的IP地址資料庫,並提供一個方便的Python API。GeoIP2可以用於許多不同的應用程式,例如網站分析、廣告定位和身份驗證。GeoIP2提供了許多不同的資訊,例如國家、城市、郵政編碼、經緯度、時區等等。它還可以使用IPv6地址進行查詢。

讀者可自行執行pip install geoip2命令安裝這個第三方庫,並自行下載對應免費版本的GeoLite2-City.mmdb主機資料庫檔案,當一切準備就緒以後我們就可以使用該資料庫定位位置了,如下程式碼是一個演示案例,首先通過GetPcap將封包解析並儲存值ret變數內返回,當返回後通過geoip2.database載入資料庫檔案,並通過迴圈的方式以此查詢reader.city資料,並將詳細地址輸出。

import dpkt
import socket
import geoip2.database

def GetPcap(pcap):
    ret = []
    for timestamp,packet in pcap:
        try:
            eth = dpkt.ethernet.Ethernet(packet)
            ip = eth.data
            src = socket.inet_ntoa(ip.src)
            dst = socket.inet_ntoa(ip.dst)
            ret.append(dst)
        except:
            pass
    return set(ret)

if __name__ == '__main__':
    fp = open('d://lyshark.pcap','rb')
    pcap = dpkt.pcap.Reader(fp)
    addr = GetPcap(pcap)
    reader = geoip2.database.Reader("d://GeoLite2-City.mmdb")
    for item in addr:
        try:
            response = reader.city(item)
            print("IP地址: %-16s --> " %item,end="")
            print("網段: %-16s --> " %response.traits.network,end="")
            print("經度: %-10s 緯度: %-10s --> " %(response.location.latitude, response.location.longitude),end="")
            print("地區: {}".format(response.country.names["zh-CN"]),end="\n")
        except Exception:
            pass

當讀者執行上述程式碼後,則可輸出當前抓到封包的詳細資訊,其中包括了網段,經緯度,地區等敏感資料,如下圖所示;

當獲取當經緯度詳細資訊後,則下一步就是要生成一個Google地圖模板,一般Google底部模板採用KML檔案格式,如下這段程式碼則是生成一個對應的地圖模板,通過傳入所需分析的pcap封包以及資料庫資訊,則可以輸出一個GoogleEarth.kml模板檔案。

import dpkt
import socket
import geoip2.database
from optparse import OptionParser

def GetPcap(pcap):
    ret = []
    for timestamp,packet in pcap:
        try:
            eth = dpkt.ethernet.Ethernet(packet)
            ip = eth.data
            src = socket.inet_ntoa(ip.src)
            dst = socket.inet_ntoa(ip.dst)
            ret.append(dst)
        except:
            pass
    return set(ret)

def retKML(addr,longitude,latitude):
    kml = (
              '<Placemark>\n'
              '<name>%s</name>\n'
              '<Point>\n'
              '<coordinates>%6f,%6f</coordinates>\n'
              '</Point>\n'
              '</Placemark>\n'
          ) %(addr, longitude, latitude)
    return kml

if __name__ == '__main__':
    parser = OptionParser()
    parser.add_option("-p", "--pcap", dest="pcap_file", help="set -p *.pcap")
    parser.add_option("-d", "--mmdb", dest="mmdb_file", help="set -d *.mmdb")
    (options, args) = parser.parse_args()
    if options.pcap_file and options.mmdb_file:
        fp = open(options.pcap_file,'rb')
        pcap = dpkt.pcap.Reader(fp)
        addr = GetPcap(pcap)
        reader = geoip2.database.Reader(options.mmdb_file)

        kmlheader = '<?xml version="1.0" encoding="UTF-8"?>\
        \n<kml xmlns="http://www.opengis.net/kml/2.2">\n<Document>\n'
        with open("GoogleEarth.kml", "w") as f:
            f.write(kmlheader)
            f.close()

        for item in addr:
            try:
                response = reader.city(item)
                print("IP地址: %-16s --> " %item,end="")
                print("網段: %-16s --> " %response.traits.network,end="")
                print("經度: %-10s 緯度: %-10s --> " %(response.location.latitude, response.location.longitude),end="")
                print("地區: {}".format(response.country.names["zh-CN"]),end="\n")

                with open("GoogleEarth.kml","a+") as f:
                    f.write(retKML(item,response.location.latitude, response.location.longitude))
                    f.close()
            except Exception:
                pass

        kmlfooter = '</Document>\n</kml>\n'
        with open("GoogleEarth.kml", "a+") as f:
            f.write(kmlfooter)
            f.close()
    else:
        parser.print_help()

此時開啟谷歌地圖,並選擇左側的專案選單,選擇從計算機中匯入KML檔案,並自行將googleearth.kml檔案匯入到地圖檔案內,如下圖所示;

當匯入成功後,此時在地圖左側將會出現一些座標資訊,此時讀者可自行點選這些座標以確定當前IP地址的詳細位置,當然該地址僅供參考,因為某些主機的地址可能會使用隱藏IP的方式並不一定確保一定準確。

當然了上述程式碼只是一個演示案例,在實際過濾中我們可以會進行多層解析,例如如下這段程式碼,其中AnalysisIP_To_Address函數就可用於接收一個過濾規則,程式碼中dport ==80 or dport == 443則用於只過濾出目標埠是80443的主機資訊,讀者也可自行增加一條符合規則的過濾條件進行自定義捕捉,此處僅僅只是一個演示案例。

import argparse
import socket,dpkt
import geoip2.database

def AnalysisPace(DpktPack,Filter):
    respon = []
    with open(DpktPack,"rb") as fp:
        pcap = dpkt.pcap.Reader(fp)
        for timestamp, packet in pcap:
            try:
                eth = dpkt.ethernet.Ethernet(packet)
                # 解析過濾出網路層(三層)中的IP封包
                if eth.data.__class__.__name__ == "IP":
                    ip = eth.data
                    src = socket.inet_ntoa(ip.src)
                    dst = socket.inet_ntoa(ip.dst)
                    # 解析過濾出傳輸層(四層)中的TCP封包
                    if eth.data.data.__class__.__name__ == "TCP":
                        sport = eth.data.data.sport
                        dport = eth.data.data.dport
                        # 過濾出源地址是192.168.1.2且目的埠是80或者443的流量
                        # if src == "192.168.1.2" and dport == 80 or dport == 443:
                        if eval(Filter):
                            dic = { "src":"None","sport":0 , "dst":"None","dport":0 }
                            #print("[+] 時間戳: %-17s 源地址: %-14s:%-2s ---> 目標地址: %-16s:%-2s" %(timestamp,src, sport, dst, dport))
                            RecvData = eth.data.data.data
                            if len(RecvData) and b"GET" in RecvData:
                                #print("[*] 時間戳: {} 源地址: {} <--- 存取網頁: {}".format(timestamp,src,bytes.decode(RecvData).split("\n")[1]))
                                pass
                            dic['src'] = src
                            dic['dst'] = dst
                            dic['sport'] = sport
                            dic['dport'] = dport
                            respon.append(dic)
            except Exception:
                pass
    return respon

def AnalysisIP_To_Address(PcapFile,MmdbFile):
    IPDict = AnalysisPace(PcapFile,"dport ==80 or dport == 443")
    NoRepeat = []

    for item in range(len(IPDict)):
        NoRepeat.append(IPDict[item].get("dst"))
    NoRepeat = set(NoRepeat)

    reader = geoip2.database.Reader(MmdbFile)
    for item in NoRepeat:
        try:
            response = reader.city(item)
            print("[+] IP地址: %-16s --> " %item,end="")
            print("網段: %-16s --> " %response.traits.network,end="")
            print("經度: %-10s 緯度: %-10s --> " %(response.location.latitude, response.location.longitude),end="")
            print("定位: {} {} {}".format(response.country.names["zh-CN"],response.subdivisions.most_specific.name,response.city.name),end="\n")
        except Exception:
            print("定位: None None None")
            pass

if __name__ == '__main__':
    Banner()
    parser = argparse.ArgumentParser()
    parser.add_argument("-p", "--pcap", dest="pcap", help="設定抓到的封包 *.pcap")
    parser.add_argument("-d", "--mmdb", dest="mmdb", help="設定城市資料庫 GeoLite2-City.mmdb")
    
    args = parser.parse_args()
    # 使用方式: main.py -p data.pcap -d GeoLite2-City.mmdb (分析封包中IP)
    if args.pcap and args.mmdb:
        AnalysisIP_To_Address(args.pcap,args.mmdb)
    else:
        parser.print_help()

本文作者: 王瑞
本文連結: https://www.lyshark.com/post/647f46e.html
版權宣告: 本部落格所有文章除特別宣告外,均採用 BY-NC-SA 許可協定。轉載請註明出處!