Fork/Fork.Net/Oreo.NetMonitor/Views/NetDetailForm.cs

using CapturePackage;
using Oreo.NetMonitor.Models;
using Oreo.NetMonitor.Services;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Diagnostics;
using System.Drawing;
using System.Linq;
using System.Net;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using System.Windows.Forms;
using Y.Utils.DataUtils.UnitConvertUtils;

namespace Oreo.NetMonitor.Views
{
    public partial class NetDetailForm : Form
    {

        NativeSocket2 _rs;
        long NowBag = 0;
        //string IP = "10.49.138.175";
        string IP = "";
        List<NetProcess> netProcesses = new List<NetProcess>();
        public NetDetailForm()
        {
            InitializeComponent();
        }
        private void NetDetailForm_Load(object sender, EventArgs e)
        {

            #region 启动任务
            IP = GetIP();
            Task.Factory.StartNew(() =>
            {
                try
                {
                    while (!this.IsDisposed)
                    {
                        GetNetProcess();
                        this.dataGridView1.Invoke(new Action(() =>
                        {
                            foreach (var p in netProcesses)
                            {
                                bool inView = false;
                                for (int i = 0; i < dataGridView1.Rows.Count; i++)
                                {
                                    if (dataGridView1.Rows[i].Cells["Column3"].Value.ToString() == p.ProcessID.ToString())
                                    {
                                        inView = true;
                                        dataGridView1.Rows[i].Cells["Column4"].Value = ByteConvertTool.Fmt(p.UpLoad) + "/s";
                                        dataGridView1.Rows[i].Cells["Column7"].Value = ByteConvertTool.Fmt(p.UpLoadCount);
                                        dataGridView1.Rows[i].Cells["Column5"].Value = ByteConvertTool.Fmt(p.DownLoad) + "/s";
                                        dataGridView1.Rows[i].Cells["Column8"].Value = ByteConvertTool.Fmt(p.DownLoadCount);
                                        dataGridView1.Rows[i].Cells["Column6"].Value = ByteConvertTool.Fmt(p.UpLoadCount + p.DownLoadCount);
                                    }
                                }
                                if (!inView)
                                    dataGridView1.Rows.Add(new object[] {
                                        p.ProcessICon,
                                        ProcessAPI.GetProcessNameByPID(p.ProcessID),
                                        p.ProcessID.ToString(),
                                        ByteConvertTool.Fmt(p.UpLoad)+ "/s",
                                        ByteConvertTool.Fmt(p.UpLoadCount),
                                        ByteConvertTool.Fmt(p.DownLoad)+ "/s",
                                        ByteConvertTool.Fmt(p.DownLoadCount),
                                        ByteConvertTool.Fmt(p.UpLoadCount + p.DownLoadCount)});
                            }
                        }));
                        CalcBagFlow();
                        this.status.Invoke(new Action(() =>
                        {
                            status.Text = string.Format("信息：IP：{0}，上传流量：{1}，下载流量：{2}",
                               IP, ByteConvertTool.Fmt(NetWorkService.NowSent), ByteConvertTool.Fmt(NetWorkService.NowReceived));
                            status.Text += string.Format("，单位时间：{0}分钟，上传流量：{1}，下载流量：{2}，下次刷新时间：{3}",
                                NetWorkService.ThresholdTime, ByteConvertTool.Fmt(NetWorkService.UnitSent), ByteConvertTool.Fmt(NetWorkService.UnitReceived), NetWorkService.CalcTime.ToString("yyyy-MM-dd HH:mm:ss"));
                        }));
                        Thread.Sleep(1000);
                    }
                }
                catch { }
            });
            CaptureInit();
            Task.Factory.StartNew(() => { try { _rs.Capture(); } catch { } });
            #endregion
        }

        #region 抓包过程
        private void CaptureInit()
        {
            _rs = new NativeSocket2(IPAddress.Parse(IP));
            _rs.IsStart = true;
            _rs.OnIPPacketCapure = (IPPacket tp) =>
            {
                this.Invoke(new Action(() =>
                {
                    try
                    {
                        if (tp.SrcAddr.ToString() == IP)
                        {
                            //源地址是本机-从本机发出
                            lock (netProcesses)
                            {
                                foreach (var item in netProcesses)
                                {
                                    int inPort = item.Ports.Where(x => x.Port == tp.SrcPort).Count(); ;
                                    if (inPort > 0)
                                    {
                                        item.UpBag++;
                                        if (item.UpBag > 1000) item.UpBag = 0;
                                        NowBag++;
                                    }
                                }
                            }
                        }
                        if (tp.DestAddr.ToString() == IP)
                        {
                            //目标地址是本机-本机接收
                            lock (netProcesses)
                            {
                                foreach (var item in netProcesses)
                                {
                                    int inPort = item.Ports.Where(x => x.Port == tp.DestPort).Count(); ;
                                    if (inPort > 0)
                                    {
                                        item.DownBag++;
                                        if (item.DownBag > 1000) item.DownBag = 0;
                                        NowBag++;
                                    }
                                }
                            }
                        }
                    }
                    catch { }
                }));

            };
        }
        #endregion
        #region 联网进程
        private void GetNetProcess()
        {
            foreach (var item in NetProcessAPI.GetAllTcpConnections())
            {
                AddTCPNetProcess(item);
            }
            foreach (var item in NetProcessAPI.GetAllUdpConnections())
            {
                AddUDPNetProcess(item);
            }
        }
        private void AddTCPNetProcess(TcpRow item)
        {
            try
            {
                var _netProcess = netProcesses.FirstOrDefault(x => x.ProcessID == item.owningPid);
                if (_netProcess == null)
                {
                    _netProcess = new NetProcess() { ProcessID = item.owningPid, ProcessName = ProcessAPI.GetProcessNameByPID(item.owningPid) };
                    netProcesses.Add(_netProcess);
                }
                if (_netProcess.Ports.FirstOrDefault(x => x.Port == item.LocalPort) == null)
                {
                    _netProcess.ProcessICon = ProcessAPI.GetIcon(item.owningPid, true);
                }
                _netProcess.Ports.Add(GetTcpProcessPort(item));
            }
            catch (Exception e)
            {

            }
        }
        private void AddUDPNetProcess(UdpRow item)
        {
            try
            {
                var _netProcess = netProcesses.FirstOrDefault(x => x.ProcessID == item.owningPid);
                if (_netProcess == null)
                {
                    _netProcess = new NetProcess() { ProcessID = item.owningPid, ProcessName = ProcessAPI.GetProcessNameByPID(item.owningPid) };
                    netProcesses.Add(_netProcess);
                }
                if (_netProcess.Ports.FirstOrDefault(x => x.Port == item.LocalPort) == null)
                {
                    _netProcess.ProcessICon = ProcessAPI.GetIcon(item.owningPid, true);
                    _netProcess.Ports.Add(GetUDPProcessPort(item));
                }
            }
            catch (Exception e)
            {

            }
        }
        private ProcessPort GetTcpProcessPort(TcpRow tr)
        {
            ProcessPort _pp = new ProcessPort()
            {
                LocalAddress = tr.LocalAddress.ToString(),
                Port = tr.LocalPort,
                RemoteAddress = tr.RemoteAddress.ToString(),
                Type = ProtocolType.TCPType,
                RemotePort = tr.RemotePort,

            };
            return _pp;
        }
        private ProcessPort GetUDPProcessPort(UdpRow tr)
        {
            ProcessPort _pp = new ProcessPort()
            {
                LocalAddress = tr.LocalAddress.ToString(),
                Port = tr.LocalPort,
                RemoteAddress = "",
                Type = ProtocolType.UDPType,
                RemotePort = 0,

            };
            return _pp;
        }
        #endregion
        #region 矫正流量
        private void CalcBagFlow()
        {
            lock (netProcesses)
            {
                foreach (var pro in netProcesses)
                {
                    long temp = pro.UpBag + pro.DownBag;
                    double rate = 0;
                    if (NowBag > 0 && temp > 0)
                        rate = (double)temp / (double)NowBag;
                    pro.UpLoad = (long)(NetWorkService.NowSent * rate);
                    pro.DownLoad = (long)(NetWorkService.NowReceived * rate);
                    pro.UpLoadCount += pro.UpLoad;
                    pro.DownLoadCount += pro.DownLoad;
                    pro.UpBag = 0;
                    pro.DownBag = 0;
                }
            }
            NowBag = 0;
        }
        #endregion
        #region 获取IP
        public string GetIP()
        {
            IPHostEntry myEntry = Dns.GetHostEntry(Dns.GetHostName());
            return myEntry.AddressList.FirstOrDefault<IPAddress>(e => e.AddressFamily.ToString().Equals("InterNetwork")).ToString();
        }
        #endregion

    }
}