tcp端口检测

# coding=utf-8
import sys
import socket
import re


def check_server(address, port):
    s = socket.socket()
    print ‘Attempting to connect to %s on port %s‘ % (address, port)
    try:
        s.connect((address, port))
        print ‘Connected to %s on port %s‘ % (address, port)
        return True
    except socket.error as e:
        print ‘Connection to %s on port %s failed: %s‘ % (address, port, e)
        return False


if __name__ == ‘__main__‘:
    from argparse import ArgumentParser
    parser = ArgumentParser(description=u‘TCP端口检测‘)
    parser.add_argument(
        ‘-a‘,
        ‘--address‘,
        dest=‘address‘,
        default=‘localhost‘,
        help=‘address for the server‘)
    parser.add_argument(
        ‘-p‘,
        ‘--port‘,
        dest="port",
        default=80,
        type=int,
        help=‘port for the server‘)
    args = parser.parse_args()
    check = check_server(args.address, args.port)
    print ‘check_server returned %s‘ % check
    sys.exit(not check)

[hupeng@hupeng-vm Python]$python check_server.py && echo "SUCCESS"
Attempting to connect to localhost on port 80
Connected to localhost on port 80
check_server returned True
SUCCESS
[hupeng@hupeng-vm Python]$python check_server.py -p 81 && echo "Failure"
Attempting to connect to localhost on port 81
Connection to localhost on port 81 failed: [Errno 111] Connection refused
check_server returned False
[hupeng@hupeng-vm Python]$python check_server.py -p 81 || echo "Failure"
Attempting to connect to localhost on port 81
Connection to localhost on port 81 failed: [Errno 111] Connection refused
check_server returned False
Failure