最近老板接了一个中船重工的项目,需要做一个有关海军软件系统的组件评估项目,项目中有一个子项目需要获取特定进程的各种系统参数,项目使用.NET平台。在获取特定进程各种系统参数时,其它诸如进程ID,进程名,进程私有工作集,进程工作集,进程IO吞吐量,进程CPU占用率等都可以直接使用.NET中的相关API直接获取,例如使用PerformanceCounter对象可以获取进程私有工作集、进程工作集、进程IO吞吐量等,但是对于特定进程的网络上行流量和下行流量却没有办法直接使用.NET中API获取,网上也找了很多资料,了解到只能通过对特定进程使用的端口号进行抓包来获取进程网络流量,具体实现时自己也想过使用C#调用winpcap库,但由于实现比较复杂,所以就放弃了使用该方法。后来在网上找到了一个国外使用C#实现的winpcap库SharpPcap库,于是使用SharpPcap库实现了获取特定进程网络流量功能。准备工作是需要将SharpPcap库下载到计算机,然后再项目中引用PacketDotNet.dll和SharpPcap.dll,然后添加引用:
1 using SharpPcap;
2 using PacketDotNet;
具体实现比较复杂,以下介绍一下实现中的核心部分:
首先定义ProcessPerformanceInfo类,用来记录进程相关信息,ProcessPerformanceInfo定义如下:
复制代码
1 //记录特定进程性能信息的类
2 public class ProcessPerformanceInfo : IDisposable
3 {
4 public int ProcessID { get; set; }//进程ID
5 public string ProcessName { get; set; }//进程名
6 public float PrivateWorkingSet { get; set; }//私有工作集(KB)
7 public float WorkingSet { get; set; }//工作集(KB)
8 public float CpuTime { get; set; }//CPU占用率(%)
9 public float IOOtherBytes { get; set; }//每秒IO操作(不包含控制操作)读写数据的字节数(KB)
10 public int IOOtherOperations { get; set; }//每秒IO操作数(不包括读写)(个数)
11 public long NetSendBytes { get; set; }//网络发送数据字节数
12 public long NetRecvBytes { get; set; }//网络接收数据字节数
13 public long NetTotalBytes { get; set; }//网络数据总字节数
14 public List<ICaptureDevice> dev = new List<ICaptureDevice>();
15
16 /// <summary>
17 /// 实现IDisposable的方法
18 /// </summary>
19 public void Dispose()
20 {
21 foreach (ICaptureDevice d in dev)
22 {
23 d.StopCapture();
24 d.Close();
25 }
26 }
27 }
复制代码
定义一个ProcessPerformanceInfo类型的属性ProcInfo:
1 public ProcessPerformanceInfo ProcInfo { get; set; }
第一步:获取指定进程使用的所有端口号
由于一个进程可能使用多个端口号,因此监视一个进程流量时必须监视该进程使用的所有端口号,具体可以通过cmd执行命令netstat -ano并对结果进行分析。代码如下:
复制代码
1 //进程id
2 int pid = ProcInfo.ProcessID;
3 //存放进程使用的端口号链表
4 List<int> ports = new List<int>();
5
6 #region 获取指定进程对应端口号
7 Process pro = new Process();
8 pro.StartInfo.FileName = "cmd.exe";
9 pro.StartInfo.UseShellExecute = false;
10 pro.StartInfo.RedirectStandardInput = true;
11 pro.StartInfo.RedirectStandardOutput = true;
12 pro.StartInfo.RedirectStandardError = true;
13 pro.StartInfo.CreateNoWindow = true;
14 pro.Start();
15 pro.StandardInput.WriteLine("netstat -ano");
16 pro.StandardInput.WriteLine("exit");
17 Regex reg = new Regex("\\s+", RegexOptions.Compiled);
18 string line = null;
19 ports.Clear();
20 while ((line = pro.StandardOutput.ReadLine()) != null)
21 {
22 line = line.Trim();
23 if (line.StartsWith("TCP", StringComparison.OrdinalIgnoreCase))
24 {
25 line = reg.Replace(line, ",");
26 string[] arr = line.Split(',');
27 if (arr[4] == pid.ToString())
28 {
29 string soc = arr[1];
30 int pos = soc.LastIndexOf(':');
31 int pot = int.Parse(soc.Substring(pos + 1));
32 ports.Add(pot);
33 }
34 }
35 else if (line.StartsWith("UDP", StringComparison.OrdinalIgnoreCase))
36 {
37 line = reg.Replace(line, ",");
38 string[] arr = line.Split(',');
39 if (arr[3] == pid.ToString())
40 {
41 string soc = arr[1];
42 int pos = soc.LastIndexOf(':');
43 int pot = int.Parse(soc.Substring(pos + 1));
44 ports.Add(pot);
45 }
46 }
47 }
48 pro.Close();
49 #endregion
复制代码
所获取的端口号都存放在ports里。
第二步:获取本机IP地址和本机网络设备(即网卡)
复制代码
1 //获取本机IP地址
2 IPAddress[] addrList = Dns.GetHostByName(Dns.GetHostName()).AddressList;
3 string IP = addrList[0].ToString();
4 //获取本机网络设备
5 var devices = CaptureDeviceList.Instance;
6 int count = devices.Count;
7 if (count < 1)
8 {
9 Console.WriteLine("No device found on this machine");
10 return;
11 }
复制代码
第三步:开始抓包,至于sharppcap库的使用方法,在官方有很详细的介绍:http://www.codeproject.com/Articles/12458/SharpPcap-A-Packet-Capture-Framework-for-NET
实现代码如下:
复制代码
1 //开始抓包
2 for (int i = 0; i < count; ++i)
3 {
4 for (int j = 0; j < ports.Count; ++j)
5 {
6 CaptureFlowRecv(IP, ports[j], i);
7 CaptureFlowSend(IP, ports[j], i);
8 }
9 }
复制代码
CaptureFlowRecv和CaptureFlowSend函数定义如下:
复制代码
1 public void CaptureFlowSend(string IP, int portID, int deviceID)
2 {
3 ICaptureDevice device = (ICaptureDevice)CaptureDeviceList.New()[deviceID];
4
5 device.OnPacketArrival += new PacketArrivalEventHandler(device_OnPacketArrivalSend);
6
7 int readTimeoutMilliseconds = 1000;
8 device.Open(DeviceMode.Promiscuous, readTimeoutMilliseconds);
9
10 string filter = "src host " + IP + " and src port " + portID;
11 device.Filter = filter;
12 device.StartCapture();
13 ProcInfo.dev.Add(device);
14 }
15
16 public void CaptureFlowRecv(string IP, int portID, int deviceID)
17 {
18 ICaptureDevice device = CaptureDeviceList.New()[deviceID];
19 device.OnPacketArrival += new PacketArrivalEventHandler(device_OnPacketArrivalRecv);
20
21 int readTimeoutMilliseconds = 1000;
22 device.Open(DeviceMode.Promiscuous, readTimeoutMilliseconds);
23
24 string filter = "dst host " + IP + " and dst port " + portID;
25 device.Filter = filter;
26 device.StartCapture();
27 ProcInfo.dev.Add(device);
28 }
复制代码
复制代码
1 private void device_OnPacketArrivalSend(object sender, CaptureEventArgs e)
2 {
3 var len = e.Packet.Data.Length;
4 ProcInfo.NetSendBytes += len;
5 }
6
7 private void device_OnPacketArrivalRecv(object sender, CaptureEventArgs e)
8 {
9 var len = e.Packet.Data.Length;
10 ProcInfo.NetRecvBytes += len;
11 }
复制代码
第四步:设置每秒刷新上行下行流量
复制代码
1 /// <summary>
2 /// 实时刷新性能参数
3 /// </summary>
4 public void RefershInfo()
5 {
6 ProcInfo.NetRecvBytes = 0;
7 ProcInfo.NetSendBytes = 0;
8 ProcInfo.NetTotalBytes = 0;
9 Thread.Sleep(1000);
10 ProcInfo.NetTotalBytes = ProcInfo.NetRecvBytes + ProcInfo.NetSendBytes;
11 }
复制代码
第五步:测试部分代码
复制代码
1 while (true)
2 {
3 Console.WriteLine("proc NetTotalBytes : " + ProcInfo.NetTotalBytes);
4 Console.WriteLine("proc NetSendBytes : " + ProcInfo.NetSendBytes);
5 Console.WriteLine("proc NetRecvBytes : " + ProcInfo.NetRecvBytes);
6
7 //每隔1s调用刷新函数对性能参数进行刷新
8 RefershInfo();
9 }
10 //最后要记得调用Dispose方法停止抓包并关闭设备
11 Proc.Dispose();
复制代码
以上仅仅是核心部分实现的代码简介,旨在为读者提供思路,具体实现还需要添加很多接口和类来实现。最后附上自己运行后的截图:
希望能对大家有所帮助!
Copyright © 广州京杭网络科技有限公司 2005-2024 版权所有 粤ICP备16019765号
广州京杭网络科技有限公司 版权所有