python检测tcp端口_Python脚本之TCP端口检查

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

运行结果为：# 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