1. 程式人生 > >Python 簡單的模擬wireshark抓包工具

Python 簡單的模擬wireshark抓包工具


混雜模式是指路過目標機器的資料包也包含在內

0埠指系統自動分配


#coding=utf-8
import socket
import os


#The host be listended
host = "192.168.1.100"


#it means windows ,posix stands for Linux
if os.name == "nt":
    socket_protocal = socket.IPPROTO_IP
else:
    socket_protocal = socket.IPPROTO_ICMP


sniffer = socket.socket(socket.AF_INET ,socket.SOCK_RAW ,socket_protocal)


sniffer.bind((host, 0))
#include the ip header into packet that captured
sniffer.setsockopt(socket.IPPROTO_IP,socket.IP_HDRINCL,1)


#Promiscuous Mode start
if os.name =="nt":
    sniffer.ioctl(socket.SIO_RCVALL,socket.RCVALL_ON)


print sniffer.recvfrom(65565)


#Promiscuous Mode end
if os.name =="nt":

    sniffer.ioctl(socket.SIO_RCVALL,socket.RCVALL_OFF)


然後解析以上資料包

#!/usr/bin/python
# coding=utf-8
import socket
import os
import struct
from ctypes import *


# 監聽的主機
host = "192.168.1.100"




# IP頭定義
class IP(Structure):
    """docstring for IP"""
    _fields_ = [
        ("ihl", c_ubyte, 4),
        ("version", c_ubyte, 4),
        ("tos", c_ubyte),
        ("len", c_ushort),
        ("id", c_ushort),
        ("offset", c_ushort),
        ("ttl", c_ubyte),
        ("protocol_num", c_ubyte),
        ("sum", c_ushort),
        ("src", c_ulong),
        ("dst", c_ulong)
    ]


    def __new__(self, socket_buffer=None):
        return self.from_buffer_copy(socket_buffer)


    def __init__(self, socket_buffer=None):
        # 協議欄位與協議名稱對應
        self.protocol_map = {1: "ICMP", 6: "TCP", 17: "UDP"}


        # 可讀性更強的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)


            # 下面的程式碼類似於之前的例子




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)


sniffer.bind((host, 0))
sniffer.setsockopt(socket.IPPROTO_IP, socket.IP_HDRINCL, 1)


if os.name == "nt":
    sniffer.ioctl(socket.SIO_RCVALL, socket.RCVALL_ON)


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)
        # 處理CTRL-C
except KeyboardInterrupt:


    # 如果執行在Windows上,關閉混雜模式
    if os.name == "nt":
        sniffer.ioctl(socket.SIO_RCVALL, socket.RCVALL_OFF)