diff --git a/plugins/task_manager/process_network_total.py b/plugins/task_manager/process_network_total.py index 11584b5d8..53ad15f0d 100644 --- a/plugins/task_manager/process_network_total.py +++ b/plugins/task_manager/process_network_total.py @@ -56,13 +56,7 @@ class process_network_total: __last_write_time = 0 __end_time = 0 - def start(self,timeout = 0): - ''' - @name 启动进程网络监控 - @author hwliang<2021-09-13> - @param timeout 结束时间(秒),0表示持久运行,默认为0 - @return void - ''' + def start(self,timeout = 0): stime = time.time() self.__end_time = timeout + stime self.__last_stat = stime @@ -80,12 +74,6 @@ class process_network_total: self.rm_pid_file() def handle_packet(self, pcap_data): - ''' - @name 处理pcap数据包 - @author hwliang<2021-09-12> - @param pcap_data pcap数据包 - @return void - ''' # 获取IP协议头 ip_header = pcap_data[14:34] # 解析src/dst地址 @@ -103,14 +91,6 @@ class process_network_total: self.total_net_process(dst,src,pack_size) def total_net_process(self,dst,src,pack_size): - ''' - @name 统计进程流量 - @author hwliang<2021-09-13> - @param dst 目标地址 - @param src 源地址 - @param pack_size 数据包长度 - @return void - ''' self.get_tcp_stat() direction = None mtime = time.time() @@ -149,12 +129,7 @@ class process_network_total: self.write_net_process() def write_net_process(self): - ''' - @name 写入进程流量 - @author hwliang<2021-09-13> - @return void - ''' - w_file = '/dev/shm/bt_net_process' + w_file = '/dev/shm/mw_net_process' process_size = copy.deepcopy(self.__net_process_size) net_process = [] for pid in process_size.keys(): @@ -165,24 +140,12 @@ class process_network_total: f.close() def hex_to_ip(self, hex_ip): - ''' - @name 将16进制的IP地址转换为字符串IP地址 - @author hwliang<2021-09-13> - @param hex_ip 16进制的IP地址:16进程端口 - @return tuple(ip,port) IP地址,端口 - ''' hex_ip,hex_port = hex_ip.split(':') ip = '.'.join([str(int(hex_ip[i:i+2], 16)) for i in range(0, len(hex_ip), 2)][::-1]) port = int(hex_port, 16) return ip,port def get_tcp_stat(self,force = False): - ''' - @name 获取当前TCP连接状态表 - @author hwliang<2021-09-13> - @param force 是否强制刷新 - @return dict - ''' if not force and self.__net_process_list: return self.__net_process_list self.__net_process_list = {} tcp_stat_file = '/proc/net/tcp' @@ -211,21 +174,9 @@ class process_network_total: def get_port_pack(self,port): - ''' - @name 将端口转换为字节流 - @author hwliang<2021-09-13> - @param port 端口 - @return bytes - ''' return struct.pack('H',int(port))[::-1] def get_ip_pack(self,ip): - ''' - @name 将IP地址转换为字节流 - @author hwliang<2021-09-13> - @param ip IP地址 - @return bytes - ''' ip_arr = ip.split('.') ip_pack = b'' for i in ip_arr: @@ -233,25 +184,12 @@ class process_network_total: return ip_pack def inode_to_pid(self,inode,force = False): - ''' - @name 将inode转换为进程ID - @author hwliang<2021-09-13> - @param inode inode - @param force 是否强制刷新 - @return int - ''' inode_list = self.get_process_inodes() if inode in inode_list: return inode_list[inode] return None def get_process_inodes(self,force = False): - ''' - @name 获取进程inode列表 - @author hwliang<2021-09-13> - @param force 是否强制刷新 - @return dict - ''' if not force and self.__inode_list: return self.__inode_list proc_path = '/proc' inode_list = {} @@ -274,12 +212,6 @@ class process_network_total: return inode_list def get_process_name(self,pid): - ''' - @name 获取进程名称 - @author hwliang<2021-09-13> - @param pid 进程ID - @return str - ''' pid_path = '/proc/' + pid + '/comm' if not os.path.exists(pid_path): return '' pid_file = open(pid_path, 'rb') @@ -289,22 +221,12 @@ class process_network_total: def write_pid(self): - ''' - @name 写入进程ID到PID文件 - @author hwliang<2021-09-13> - @return void - ''' self_pid = os.getpid() pid_file = open(self.__pid_file,'w') pid_file.write(str(self_pid)) pid_file.close() def rm_pid_file(self): - ''' - @name 删除进程pid文件 - @author hwliang<2021-09-13> - @return void - ''' if os.path.exists(self.__pid_file): os.remove(self.__pid_file)