#-*- coding:utf8 -*-  
  
import socket  
import os  
import struct  
import threading  
import time  
import sys  
from netaddr import IPNetwork,IPAddress  
from ctypes import *  
  
# 監聽主機，即監聽那個網路介面，下面的ip為我的ip  
#host = "172.xx.xx.xx" 
# 實訓ip
host = "10.0.2.10"
  
# 掃描的目標子網  
# subnet = "192.168.1.0/24"  
# 沒有命令列引數,預設192.168.1.0/24  
if len(sys.argv) == 1:# 掃描當前IP網段存在的主機
    #subnet = "172.19.65.0/24" 
     subnet = "10.0.2.0/24"
else:  
    subnet = sys.argv[1]  
  
# 自定義的字串,我們將在ICMP響應中進行核對  
magic_message = "PYTHONRULES!"  
  
# 批量傳送UDP資料包  
def udp_sender(subnet, magic_message):  
    time.sleep(5)   #可以說程式暫停5秒吧  
    # 建立一個socket物件(SOCK_DGRAM:UDP客戶端)  
    sender = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)  
  
    for ip in IPNetwork(subnet):  
        try:  
            # 嘗試傳送magic_message這個訊息到子網的每個ip,還用了個不怎麼可能用的65212埠  
            sender.sendto(magic_message, ("%s" % ip, 65212))  
        except:  
            pass    #代表什麼也不做  
  
# ip頭定義  
class IP(Structure):  
    _fields_ = [  
        ("ihl",             c_ubyte, 4),    #ip head length:頭長度  
        ("version",         c_ubyte, 4),    #版本  
        ("tos",             c_ubyte),       #服務型別  
        ("len",             c_ushort),      #ip資料包總長度  
        ("id",              c_ushort),       #識別符號  
        ("offset",          c_ushort),      #片偏移  
        ("ttl",             c_ubyte),       #生存時間  
        ("protocol_num",    c_ubyte),       #協議數字,應該是協議型別,這裡用數字來代表時哪個協議,下面建構函式有設定對映表  
        ("sum",             c_ushort),      #頭部校驗和  
        ("src",             c_uint32),       #源ip地址  
        ("dst",             c_uint32)        #目的ip地址  
    ]  
  
    # __new__(cls, *args, **kwargs)  建立物件時呼叫，返回當前物件的一個例項;注意：這裡的第一個引數是cls即class本身  
    def __new__(self, socket_buffer=None):  
        return  self.from_buffer_copy(socket_buffer)  
  
    # __init__(self, *args, **kwargs) 建立完物件後呼叫，對當前物件的例項的一些初始化，無返回值,即在呼叫__new__之後，根據返回的例項初始化；注意，這裡的第一個引數是self即物件本身【注意和new的區別】  
    def __init__(self, socket_buffer=None):  
        # 協議欄位與協議名稱的對應  
        self.protocol_map = {1:"ICMP", 6:"TCP", 17:"UDP"}  
  
        # 可讀性更強的ip地址(轉換32位打包的IPV4地址為IP地址的標準點號分隔字串表示。)  
        self.src_address = socket.inet_ntoa(struct.pack("<L", self.src))  
        self.dst_address = socket.inet_ntoa(struct.pack("<L", self.dst))  
  
        # 協議型別  
        try:  
            self.protocol = self.protocol_map[self.protocol_num]  
        except:  
            self.protocol = str(self.protocol_num)  
  
  
class ICMP(Structure):  
    #  
    _fields_ = [  
        ("type",            c_ubyte),       #型別  
        ("code",            c_ubyte),       #程式碼值  
        ("checksum",        c_ubyte),       #頭部校驗和  
        ("unused",          c_ubyte),       #未使用  
        ("next_hop_mtu",    c_ubyte)        #下一跳的MTU  
    ]  
  
    def __new__(self, socket_buffer):  
        return self.from_buffer_copy(socket_buffer)  
  
    def __init__(self, socket_buffer):  
        pass  
  
  
if  os.name == "nt":  
    socket_protocol = socket.IPPROTO_IP  
else:  
    socket_protocol = socket.IPPROTO_ICMP  
  
sniffer = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket_protocol)   #raw的中文是生的意思,大概就是原始套接字的意思吧  
  
sniffer.bind((host, 0)) #這裡埠為0,監聽所有埠吧~  
  
# 設定在捕獲的資料包中包含IP頭  
sniffer.setsockopt(socket.IPPROTO_IP, socket.IP_HDRINCL, 1)  
  
# 在Windows平臺上,我們需要設定IOCTL以啟用混雜模式  
if os.name == "nt":  
    sniffer.ioctl(socket.SIO_RCVALL, socket.RCVALL_ON)  
  
# 開啟多執行緒傳送udp資料包  
t = threading.Thread(target=udp_sender, args=(subnet, magic_message))  
t.start()  
  
try:  
    while True:  
        # 讀取資料包  
        raw_buffer =  sniffer.recvfrom(65565)[0]  
  
        # 將緩衝區的前20個位元組按IP頭進行解析  
        ip_header = IP(raw_buffer[0:20])  
  
        # 輸出協議和通訊雙方IP地址  
        #print  "Protocol: %s %s ->  %s" % (ip_header.protocol, ip_header.src_address, ip_header.dst_address)  
  
        # 如果為ICMP,進行處理  
        if ip_header.protocol == "ICMP":  
  
            # 計算ICMP包的起始位置,並獲取ICMP包的資料  
            offset = ip_header.ihl * 4      #ihl是頭部長度,代表32位(即4位元組)長的分片的個數 [我的理解是因為一個位元組表示一個符號,所以這裡的offset要搞成以位元組為單位的,為的是下一句的提取資料]  
            buf = raw_buffer[offset:offset+sizeof(ICMP)]  
  
            # 解析ICMP資料  
            icmp_header = ICMP(buf)  
  
            #print "ICMP -> Type: %d Code: %d" % (icmp_header.type, icmp_header.code)  
  
            # 檢查型別和程式碼值是否都為3  
            if icmp_header.type == 3 and icmp_header.code == 3:  
                # 確認響應的主機再我們的目標子網之內  
                if IPAddress(ip_header.src_address) in IPNetwork(subnet):  
                    # 確認ICMP包中包含我們傳送的自定義的字串  
                    if raw_buffer[len(raw_buffer) - len(magic_message):] == magic_message:  
                        print "Host Up: %s" % ip_header.src_address  
  
  
  
# 處理CTRL-C  
except  KeyboardInterrupt:  
  
    # 如果執行再Windows上,關閉混雜模式  
    if os.name == "nt":  
        sniffer.ioctl(socket.SIO_RCVALL, socket.RCVALL_OFF)