Python脚本之TCP端口检查

kghjtyhyt

脚本源文件为：

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