一个检测TCP端口的Python脚本程序，默认需要两个参数，一个是-a参数，后面接要检测的IP地址，另一个是-p参数，后面接要检测的IP地址的端口，如果省略该选项，默认是IP地址80端口。以下是脚本源文件为：

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
# cat tcp_checker.py 
#!/usr/bin/env python
 
import  socket
import  re
import  sys
 
def check_server(address, port):
     # Create a TCP socket
     s = socket.socket()
     print  "\033[33mAttempting to connect to %s on port %s\033[0m"  % (address, port)
     try:
         s.connect((address, port))
         print  "\033[32mConnected to %s on port %s\033[0m"  % (address, port)
         return  True
     except socket.error, e:
         print  "\033[31mConnection to %s on port %s failed: %s\033[0m"  % (address, port, e)
         return  False
 
 
if  __name__ ==  "__main__" :
     from optparse  import  OptionParser
     parser = OptionParser()
 
     parser.add_option( "-a" ,  "--address" , dest= "address" , default= 'localhost' ,
                        help= "ADDRESS for server, default is localhost" , metavar= "ADDRESS" )
     parser.add_option( "-p" ,  "--port" , dest= "port" ,  type = "int" , default=80,
                       help= "PORT for server, default is 80" , metavar= "PORT" )
     (options,args) = parser.parse_args()
     print  "options: %s, args: %s"  % (options, args)
     check = check_server(options.address, options.port)
     print  "check_server returned %s"  % check
     sys. exit (not check)


运行结果为：

1
2
3
4
5
# python tcp_checker.py -a 192.168.56.108 -p 4000
options: { 'port' : 4000,  'address' :  '192.168.56.108' }, args: []
Attempting to connect to 192.168.56.108 on port 4000
Connected to 192.168.56.108 on port 4000
check_server returned True