#
# Rajarshi Guha <rajarshi@presidency.com>
# 16/02/2005
#

import httplib, sys, getopt, string

def usage():
    print """
    Usage: refcheck.py -h HOSTNAME [OPTIONS]

    Command line parameters are:
    -h,--host       Webserver address or name
    -p,--path       Path on the webserver to request
    -r,--referer    Referer to use
    -u,--ua         User-Agent to use

    The host name is mandatory and User-Agent and Referer headers
    default to empty. The default path requested in '/'
    """

if __name__ == '__main__':

    referer = ''
    useragent = ''
    host = ''
    path = '/'
    
    if len(sys.argv) == 1:
        usage()
        sys.exit(0)
        
    try:
        opt,args = getopt.getopt(sys.argv[1:],'u:r:h:p:',
                ['ua=','referer=','host=','path='])
    except getopt.GetoptError:
        usage()
        sys.exit(0)

    for o,a in opt:
        if o in ('-u','--ua'):
            useragent = a
        if o in ('-r','--referer'):
            referer = a
        if o in ('-h','--host'):
            host = a
        if o in ('-p','--path'):
            path = a
    
    if host == '':
        usage()
        sys.exit(0)

    con = httplib.HTTPConnection(host)
    httpheaders = {'Referer':referer, 'User-Agent':useragent}
    con.request('GET', path, headers=httpheaders)
    response = con.getresponse()
    print response.status, response.reason