Chinaunix首页 | 论坛 | 博客
  • 博客访问: 15315629
  • 博文数量: 112
  • 博客积分: 11195
  • 博客等级: 上将
  • 技术积分: 1989
  • 用 户 组: 普通用户
  • 注册时间: 2005-06-20 11:04
文章分类

全部博文(112)

文章存档

2013年(2)

2012年(27)

2011年(6)

2010年(11)

2009年(6)

2007年(7)

2006年(23)

2005年(30)

分类: Python/Ruby

2010-08-29 04:51:43

    某日无聊正在学习python,一网友问用vc如何编写监控windows系统各个进程的流量的程序,告其方法和思路以后,遂想以python实现,开始使用的pcap库和dpkt库,先是让人心里觉得膈蝇的是这两个库不再维护了,然后又被其超低的效率所折磨,正在考虑以ctypes调用windows下面dll实现的时候,新版python出来了!
 
代码如下:
 
 
程序要求使用python2.7版本

import socket
import struct
import thread
import threading
import time
import os

net_data = {}
d_net_info = {}
lock = threading.Lock()

def print_data():
    while True:
        lock.acquire()
        for key in net_data:
            print "%s %s\n"%(key, net_data[key])
        lock.release()
        time.sleep(5);
        
def get_net_info():
    net_info = os.popen('netstat -nbo').readlines()
    
    for l in net_info[4:]:
        s = l.split()
        if len(s)>2:
            key = "%s %s"%(s[1],s[2])
            key2 = "%s %s"%(s[2],s[1])
        else:
            if not d_net_info.has_key(key):
                d_net_info[key] = s[0]
                d_net_info[key2] = s[0]

def get_packet():
    HOST = socket.gethostbyname(socket.gethostname())
    s = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_IP)
    s.bind((HOST, 0))
    s.setsockopt(socket.IPPROTO_IP, socket.IP_HDRINCL, 1)
    s.ioctl(socket.SIO_RCVALL, socket.RCVALL_ON)
    net_data["unknow"] = 0

    while True:
        buf = s.recvfrom(65565)
        port = struct.unpack('HH', buf[0][20:24])
        
        src_ip = "%d.%d.%d.%d"%struct.unpack('BBBB', buf[0][12:16])
        dest_ip ="%d.%d.%d.%d"%struct.unpack('BBBB', buf[0][16:20])
        src_port = socket.htons(port[0])
        dest_port = socket.htons(port[1])
        
        data_len = len(buf[0])
        key="%s:%d %s:%d"%(src_ip,src_port,dest_ip,dest_port)
        if not d_net_info.has_key(key):
            get_net_info()

        if d_net_info.has_key(key):
            key2 ="%s %s"%(key,d_net_info[key])
            if net_data.has_key(key2):
                net_data[key2] =net_data[key2]+data_len
            else:
                net_data[key2] = data_len
        else:
             net_data["unknow"] =net_data["unknow"] + data_len
        
thread.start_new_thread(print_data,())
get_packet()
os.exit()

 

初学python,代码比较丑陋,可能效率和安全、稳定性方面也有很大问题。

阅读(68556) | 评论(5) | 转发(3) |
给主人留下些什么吧!~~

泉州俺最牛2011-05-10 09:05:52

没看懂,以后再看

ahhphei2011-04-18 14:52:28

楼主把C++的实现方法介绍一下吧!谢谢哦

chinaunix网友2010-09-11 13:18:06

http://www.jiaoyib2c.com.cn/ 机械包装 http://www.jiaoyib2c.com.cn/ 设备包装 http://www.001gx.cn/ 真空包装 http://www.jiaoyib2c.com.cn/ 木箱打包 http://www.jiaoyib2c.cn/ 出口包装箱 http://www.001gx.cn/ 机械真空包装ijn568h华宜物流公司 http://www.jiaoyib2c.cn/木制包装 http://www.jiaoyib2c.cn/出口木箱 http://www.001gx.cn/ 机械设备真空包装

chinaunix网友2010-09-06 15:21:36

谢谢楼主

chinaunix网友2010-09-06 15:21:01

双线服务器托管、租用、机柜租赁 郑州景安双线BGP机房, QQ:340699407 欢迎咨询洽谈