#!/usr/bin/python
import random
import sys
import urllib2
import base64
verbose = 0
charset = "0123456789"
method = 'GET'
content_type = 'text/html'
def randomKey(charset, len):
randKey = []
for i in xrange(0, len):
sampl = random.choice(charset)
randKey.append(sampl)
return ''.join(randKey)
def checkResponse(resp_content, failure_keyword):
print resp_content + "{OK}"
if not failure_keyword in resp_content:
return 1
return 0
def sendKey(url, param, key):
if verbose:
print key
clen = 0
data = ""
if (method == "GET"):
url = url + "?" + param + "=" + key
else:
data = key
clen = len(data)
print "url : " + url
request = urllib2.Request(url, data, {'Content-Type': content_type, 'Content-Length': clen})
request.get_method = lambda: method
try:
resp = urllib2.urlopen(request)
rcode = resp.getcode()
resp_content = resp.read()
if (checkResponse(resp_content,"better") == 1) :
print "GOT IT:" + resp_content
exit
else :
print "FAIL"
resp.close()
except urllib2.HTTPError, e:
print str(e)
rcode = e.getcode()
return rcode
def main():
ARG_NUM = 3
argc = sys.argv.__len__()
rcode = 200
if (argc < ARG_NUM):
print "Expected args: <url> <key_len>"
exit (2)
v_url = sys.argv[1]
len = int(sys.argv[2])
key = randomKey(charset, len)
rcode = sendKey(v_url, "param", key)
print "DONE: " + str(rcode)
return(0)
if __name__ == "__main__":
sys.exit(main())