usr bin env python -coding utf-8- pyrt port litsening viewer Copyright

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
#!/usr/bin/env python
#-*-coding:utf-8-*-
#
# pyrt — port litsening viewer
#
# Copyright (c) 2009 knsd <http://knsd.net/contacts>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version. See the file LICENSE.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
#
name = "pyrt"
version = "0.0.2"
from ctypes import c_int, c_char, c_ulong, c_void_p, windll, byref, Structure, pointer, sizeof
import socket, struct
class MIB_TCPROW_OWNER_PID(Structure):
_fields_ = [
("dwState", c_ulong),
("dwLocalAddr", c_ulong),
("dwLocalPort", c_ulong),
("dwRemoteAddr", c_ulong),
("dwRemotePort", c_ulong),
("dwOwningPid", c_ulong),
]
class MIB_TCPTABLE_OWNER_PID(Structure):
_fields_ = [
("dwNumEntries", c_ulong),
("table", MIB_TCPROW_OWNER_PID * 512)
]
class PROCESSENTRY32(Structure):
_fields_ = [('dwSize', c_int),
('cntUsage', c_int),
('th32ProcessID', c_int),
('th32DefaultHeapID', c_int),
('th32ModuleID', c_int),
('cntThreads', c_int),
('th32ParentProcessID', c_int),
('pcPriClassBase', c_ulong),
('dwFlags', c_int),
('szExeFile', c_char * 260),
('th32MemoryBase', c_ulong),
('th32AccessKey', c_ulong)]
def getPidsPortsList():
port_list = []
tcp_table = MIB_TCPTABLE_OWNER_PID()
init_size = c_int()
af_inet = 2 #ipv4
tcp_table_owner_pid_listener = 3 #listeners
windll.iphlpapi.GetExtendedTcpTable(byref(tcp_table), byref(init_size), False, af_inet, tcp_table_owner_pid_listener, 0)
windll.iphlpapi.GetExtendedTcpTable(byref(tcp_table), byref(init_size), False, af_inet, tcp_table_owner_pid_listener, 0)
for itemnum in xrange(tcp_table.dwNumEntries):
item = tcp_table.table[itemnum]
host = socket.inet_ntoa(struct.pack("L", item.dwLocalAddr))
if host == '127.0.0.1': continue
port = socket.ntohs(item.dwLocalPort)
pid = item.dwOwningPid
port_list.append((pid, port))
return port_list
def getProcessDict():
process_dict = {}
hsnapshot = windll.kernel32.CreateToolhelp32Snapshot(2, 0)
process = PROCESSENTRY32()
process.dwSize = sizeof(PROCESSENTRY32)
r = windll.kernel32.Process32First(hsnapshot, pointer(process))
while r:
name = process.szExeFile
pid = process.th32ProcessID
process_dict[pid] = name
r = windll.kernel32.Process32Next(hsnapshot, pointer(process))
return process_dict
def main():
pids_ports_list = getPidsPortsList()
process_dict = getProcessDict()
pids_ports_list.sort()
for item in pids_ports_list:
print process_dict[item[0]], item[1]
print
raw_input('type any key to exit')
if __name__ == '__main__':
main()