基於socket的web伺服器檢測

標籤：ret   pre   add   with   art   res   基於   split   byte   

 

# coding=utf-8import sysimport socketimport redef check_webserver(address, port, resource):    address = socket.gethostbyname(address)    if not resource.startswith(‘/‘):        resource = ‘/‘ + resource    request_string = ‘GET %s HTTP/1.0\r\n\r\n‘ % (resource)    print ‘HTTP request: |||%s||||‘ % request_string    s = socket.socket()    failed = False    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)        s.send(request_string)        resp = s.recv(100)        print ‘Received 100 bytes of HTTP response‘        print ‘|||%s|||‘ % resp    except socket.error as e:        print ‘Connection to %s on port %s failed: %s‘ % (address, port, e)        failed = True    finally:        print ‘Close the connection‘        s.close()    if failed:        return False    lines = resp.splitlines()    print ‘First line of HTTP response: %s‘ % lines[0]    try:        version, status, message = re.split(‘\s+‘, lines[0], 2)    except ValueError:        print ‘Failed to split status line‘        return False    if status in [‘200‘, ‘301‘]:        print ‘Success - status was %s‘ % status        return True    else:        print ‘Failure - Status was %s‘ % status        return Falseif __name__ == ‘__main__‘:    from argparse import ArgumentParser    parser = ArgumentParser(description=u"基於socket的web伺服器檢測")    parser.add_argument(        ‘-a‘,        ‘--address‘,        dest="address",        default="localhost",        help="address for the server")    parser.add_argument(        "-p",        "--port",        dest="port",        type=int,        default=80,        help="port for the server")    parser.add_argument(        "-r",        ‘--resource‘,        dest=‘resource‘,        default="index.html",        help="resouce to check")    args = parser.parse_args()    check = check_webserver(args.address, args.port, args.resource)    print ‘check_webserver returned: %s‘ % check    sys.exit(not check)

 

基於socket的web伺服器檢測