using System;
using System.Net;
using System.Net.Sockets;
using System.Net.NetworkInformation;
using System.Runtime.InteropServices;
using System.Collections.Concurrent;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using SharpPcap;
using SharpPcap.WinPcap;
using PacketDotNet;
using Demo.STP.Sniff.Packets;

namespace Demo.STP.Sniff
{
    /// <summary>
    /// 嗅探器,提供原始数据源
    /// </summary>
    public sealed class Sniffer
    {
        #region Instance
        private static readonly Lazy<Sniffer> _lazy =
            new Lazy<Sniffer>(() => new Sniffer());

        public static Sniffer Instance
        {
            get { return _lazy.Value; }
        }

        private Sniffer() { }
        #endregion Instance

        #region Init
        /// <summary>
        /// 初始化嗅探器
        /// </summary>
        /// <param name="ipAddress">需拦截网卡所配IP</param>
        /// <param name="filter">数据包过滤规则</param>
        public void Init(string ipAddress, string filter = "")
        {
            var network = Demo.STP.Common.NetHelper.GetNetwork(ipAddress);
            if (network == null)
                throw new ArgumentException(string.Format("Invalid IPAddress {0}.", ipAddress));

            this._currentDevice = WinPcapDeviceList.Instance.Where(deviceItem =>
                string.Equals(network.Name, deviceItem.Interface.FriendlyName, StringComparison.OrdinalIgnoreCase))
                .FirstOrDefault();

            if (this._currentDevice == null)
                throw new ArgumentException("无法从WinPcap找到对应网卡.");

            this._currentDevice.OnPacketArrival += device_OnPacketArrival;
            this._currentDevice.Open(DeviceMode.Normal);
            this._currentDevice.Filter = filter;
        }

        private void device_OnPacketArrival(object sender, CaptureEventArgs e)
        {
            this._packetQueue.Enqueue(e.Packet);
        }
        #endregion Init

        #region Start Hook
        /// <summary>
        /// 启动嗅探器
        /// </summary>
        public void StartHook()
        {
            if (this._currentDevice == null)
                throw new ArgumentException("请先初始化嗅探器.");

            KeepHook = true;
            this._currentDevice.StartCapture();

            RawCapture rawCapture = null;

            Task.Factory.StartNew(() =>
            {
                while (KeepHook)
                {
                    if (this._packetQueue.IsEmpty)
                    {
                        //理论上能轻松应对所有终端
                        //如果吞吐量达不到要求,请删除Sleep,但会占用一个核心的全部时间
                        Thread.Sleep();
                        continue;
                    }
                    if (this._packetQueue.TryDequeue(out rawCapture))
                    {
                        var basePacket = this.AnalysisPacket(rawCapture);
                        if (basePacket.HasValue)
                        {
                            OnPacketArrival(new PacketArrivedEventArgs { BasePacket = basePacket.Value });
                        }
                    }
                }
            }, TaskCreationOptions.LongRunning);
        }
        #endregion Start Hook

        #region Stop Hook
        /// <summary>
        /// 停止嗅探器
        /// </summary>
        public void StopHook()
        {
            if (this._currentDevice == null) return;

            KeepHook = false;

            if (this._currentDevice.Started)
                this._currentDevice.StopCapture();

            this._currentDevice.OnPacketArrival -= device_OnPacketArrival;
            this._currentDevice.Close();
        }
        #endregion Stop Hook

        #region 解析数据包
        private Nullable<IPBasePacket> AnalysisPacket(RawCapture rawCapture)
        {
            try
            {
                var rawPacket = rawCapture.ParseRawPacket();
                var ipPacket = rawPacket.ExtractTragetPacket<IpPacket>();
                if (ipPacket == null)
                    return null;

                EnumIPProtocol targetProtocol;
                var parseFlag = Enum.TryParse(ipPacket.Protocol.ToString(), true, out targetProtocol);
                if (!parseFlag)
                    return null;

                IPBasePacket basePacket = new IPBasePacket();
                basePacket.Protocol = targetProtocol;
                basePacket.IPVersion = ipPacket.Version.ToString();
                basePacket.SourceIP = ipPacket.SourceAddress.ToString();
                basePacket.TargetIP = ipPacket.DestinationAddress.ToString();

                dynamic tcp_udp_Packet = rawPacket.ExtractTragetPacket<TcpPacket>();
                if (tcp_udp_Packet == null)
                    tcp_udp_Packet = rawPacket.ExtractTragetPacket<UdpPacket>();

                if (tcp_udp_Packet != null)
                {
                    if (tcp_udp_Packet.PayloadData == null)
                        return null; 

                    basePacket.SourcePort = tcp_udp_Packet.SourcePort;
                    basePacket.TargetPort = tcp_udp_Packet.DestinationPort;

                    //basePacket.PacketLength = (uint)tcp_udp_Packet.Bytes.Length;
                    //basePacket.HeaderLength = (uint)tcp_udp_Packet.Header.Length;
                    //basePacket.MessageLength = (uint)tcp_udp_Packet.PayloadData.Length;

                    //basePacket.PacketBuffer = tcp_udp_Packet.Bytes;
                    //basePacket.HeaderBuffer = tcp_udp_Packet.Header;
                    //basePacket.MessageBuffer = tcp_udp_Packet.PayloadData;

                    basePacket.MessageLength = (uint)tcp_udp_Packet.PayloadData.Length;
                    basePacket.MessageBuffer = tcp_udp_Packet.PayloadData;
                }

                return basePacket;
            }
            catch
            {
                return null;
            }
        }
        #endregion 解析数据包

        #region PacketArrival Event
        public event EventHandler<PacketArrivedEventArgs> PacketArrival;

        internal void OnPacketArrival(PacketArrivedEventArgs e)
        {
            if (PacketArrival != null)
            {
                PacketArrival(this, e);
            }
        }

        public bool IsMonitorEmpty
        {
            get { return PacketArrival == null; }
        }
        #endregion PacketArrival Event

        #region Fields & Propertys
        public bool KeepHook { get; private set; }
        private WinPcapDevice _currentDevice;
        private ConcurrentQueue<RawCapture> _packetQueue = new ConcurrentQueue<RawCapture>();
        #endregion Fields & Propertys
    }

    public class PacketArrivedEventArgs : EventArgs
    {
        public IPBasePacket BasePacket { get; set; }
    }
}

Sniffer的完整代码,基于winpcap抓包统计吞吐量的更多相关文章

  1. winpcap抓包原理

    winpcap抓包原理 WinPcap 是由伯克利分组捕获库派生而来的分组捕获库,它是在Windows 操作平台上来实现对底层包的截取过滤.WinPcap 是 BPF 模型和 Libpcap 函数库在 ...

  2. 基于wireshark抓包分析TCP的三次握手

    1. TCP的三次握手 在TCP/IP协议通讯过程中,采用三次握手建立连接,从而保证连接的安全可靠. 所有基于TCP的通信都需要以两台主机的握手开始.这个握手过程主要是希望能达到以下不同的目的.[1] ...

  3. Python + winpcap抓包和发包

    winpcapy Python的winpcapy库可以简单地实现收发Layer2层(数据链路层,以太网)数据. winpcapy主页:https://github.com/orweis/winpcap ...

  4. NetAnalyzer笔记 之 三. 用C++做一个抓包程序

    [创建时间:2015-08-27 22:15:17] NetAnalyzer下载地址 经过前两篇的瞎扯,你是不是已经厌倦了呢,那么这篇让我们来点有意思的吧,什么,用C#.不,这篇我们先来C++的 Wi ...

  5. https wireshark抓包——要解密出原始数据光有ssl 证书还不行,还要有浏览器内的pre-master-secret(内存里)

    基于wireshark抓包的分析 首先使用wireshark并且打开浏览器,打开百度(百度使用的是HTTPS加密),随意输入关键词浏览. 我这里将抓到的包进行过滤.过滤规则如下 ip.addr == ...

  6. 软件测试必须掌握的抓包工具Wireshark,你会了么?

    作为软件测试工程师,大家在工作中肯定经常会用到各种抓包工具来辅助测试,比如浏览器自带的抓包工具-F12,方便又快捷:比如时下特别流行的Fiddler工具,使用各种web和APP测试的各种场景的抓包分析 ...

  7. charles 抓包 (二)

    本文基于charles 抓包 https (1)中的配置完成. 1.移动设备上的网络请求 打开要调试的APP,请求就会先发送到Charles,然后验证是否允许访问. 当点击允许后,可以在Proxy - ...

  8. 【毕业设计】基于Android的家校互动平台开发(内含完整代码和所有文档)——爱吖校推(你关注的,我们才推)

    ☆ 写在前面 之前答应大家的毕业答辩之后把所有文档贡献出来,现在答辩已过,LZ信守承诺,把所有文档开源到了GitHub(这个地址包含所有的代码和文档以及PPT,外层为简单的代码).还望喜欢的朋友们,不 ...

  9. 基于Linux C的socket抓包程序和Package分析 (一)

    版权声明:本文为博主原创文章,未经博主同意不得转载. https://blog.csdn.net/guankle/article/details/27538031  測试执行平台:CentOS 6 ...

随机推荐

  1. [sql查询] 重复数据只取一条

    SELECT * FROM tab_init WHERE id IN ( --根据Data分类获取数据最小ID列表 select max(id) from tab_init group by a,b ...

  2. SqlServer--查询案例

    use  MyDataBase1 -- * 表示显示所有列 -- 查询语句没有加where条件表示查询所有行 select *from TblStudent ---只查询表中的部分列 select t ...

  3. ORA-02266: unique/primary keys in table referenced by enabled foreign keys

    在数据库里面使用TRUNCATE命令截断一个表的数据时,遇到如下错误 SQL >TRUNCATE TABLE ESCMOWNER.SUBX_ITEM ORA-02266: unique/prim ...

  4. SQL SERVER 临时表导致存储过程重编译(recompile)的一些探讨

    SQLSERVER为了确保返回正确的值,或者处于性能上的顾虑,有意不重用缓存在内存里的执行计划,而重新编译执行计划的这种行为,被称为重编译(recompile).那么引发存储过程重编译的条件有哪一些呢 ...

  5. MS Sql Server 数据库或表修复(Log日志文件损坏的修复方法)

    ----------------- [1] use master go sp_configure reconfigure with override go ----------------- [2] ...

  6. 在服务器上发布MVC5的应用

    如果在Windows server 2012R2上发布MVC应用,步骤稍微简单一些: 安装Win Server2012R2 增加角色IIS和asp.net4.5, IIS里增加asp.net4.5支持 ...

  7. mysql5.7.1 zip版本安装记录

    很久没遇到使用mysql的项目了,今天下载了个,发现变了很多,搜索了下,发现按http://www.cnblogs.com/wenthink/p/MySQLInstall.html中提到的基本可以使用 ...

  8. 【转】Java并发编程:深入剖析ThreadLocal

    来自: http://www.importnew.com/17849.html 想必很多朋友对ThreadLocal并不陌生,今天我们就来一起探讨下ThreadLocal的使用方法和实现原理.首先,本 ...

  9. Java中怎么切换窗口

    首先,创建一个窗口对象: eg:  JFram frame = JFrame();//创建一个窗体 frame.setVisible(true);//设置窗体可见,默认不可见 然后编写相关应用或组件… ...

  10. C++的友元类和友元函数实例

    #include <math.h> #include<iostream> using namespace std; class Point { public: Point(do ...