Python 简单的模拟wireshark抓包工具

来源:互联网 发布:皮皮麻将算法 编辑:程序博客网 时间:2024/06/13 21:14

混杂模式是指路过目标机器的数据包也包含在内

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)


原创粉丝点击