web服务器检测

# coding=utf-8

import sys

import socket

import re

def 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 ['', '']:

        print 'Success - status was %s' % status

        return True

    else:

        print 'Failure - Status was %s' % status

        return False

if __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)

def check_webserver_by_httplib(address, port, resource):

    address = socket.gethostbyname(address)

    failed = False

    if not resource.startswith('/'):

        resource = '/' + resource

    try:

        conn = httplib.HTTPConnection(address, port)

        print 'HTTP connection created successfully'

        # make request

        conn.request('GET', resource)

        print 'request for %s successful' % (resource)

        # get response

        resp = conn.getresponse()

        print 'response status: %s' % resp.status

    except socket.error as e:

        print 'HTTP connection failed: %s' % e

        return False

    finally:

        conn.close()

        print 'HTTP connection closed successfully'

    if resp.status in [200, 301]:

        return True

    else:

        return False