Networking

sinxadmin

Example 5-1. TCP port checker#!/usr/bin/env pythonimport socketimport reimport sysdef check_server(address, port):    #create a TCP socket    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, e:        print "Connection to %s on port %s failed: %s" % (address, port, e)    return Falseif __name__ == '__main__':    from optparse import OptionParser    parser = OptionParser()    parser.add_option("-a", "--address", dest="address", default='localhost',        help="ADDRESS for server", metavar="ADDRESS")    parser.add_option("-p", "--port", dest="port", type="int", default=80,        help="PORT for server", 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)Example 5-2. Socket-based web server checker#!/usr/bin/env pythonimport socketimport reimport sysdef check_webserver(address, port, resource):    #build up HTTP request string    if not resource.startswith('/'):        resource = '/' + resource    request_string = "GET %s HTTP/1.1\r\nHost: %s\r\n\r\n" % (resource, address)    print 'HTTP request:'    print '