new_client.py

#coding:utf-8
import sys
from selenium import webdriver
import json
import time
import urllib
import requests


server_ip = sys.argv[1]
server_port = int(sys.argv[2])
server_url = 'http://%s:%d'%(server_ip,server_port)
js = "document.getElementsByTagName('{0}')[0].innerHTML = '{1}'"

wd1 = webdriver.Firefox()
wd1.get(server_url)

while True:
	if len(wd1.current_url.split('/'))<5:
		time.sleep(0.5)
		continue
	url = '/'.join(wd1.current_url.split('/')[4:]).decode('base64')#是否需要join? 需要 ps:base64内容中有/
	h = json.loads(wd1.find_element_by_tag_name('pre').text)
	method = h['method']
	headers = h['headers']
	data = urllib.unquote(str(h['data'])) if method == 'POST' else None

	for i in headers.keys():
		if i.lower() in ['host','content-length']: 
			headers.pop(i) #?????

	profile = webdriver.FirefoxProfile()
	profile.add_extension('./modify_headers-0.7.1.1-fx.xpi')
	profile.set_preference('modifyheaders.headers.count',len(headers))
	for n,row in enumerate(headers.items()):
		profile.set_preference('modifyheaders.headers.action%d'%n,'Add')
		profile.set_preference('modifyheaders.headers.name%d'%n,row[0].lower())
		profile.set_preference('modifyheaders.headers.value%d'%n,row[1])
		profile.set_preference('modifyheaders.headers.enabled%d'n,True)
	profile.set_preference("modifyheaders.config.active", True)
    profile.set_preference("modifyheaders.config.alwaysOn", True)

    wd2 = webdriver.Firefox(firefox_profile=profile)
    if method == 'GET':
    	try:
    		wd2.get(url)
    	except Exception,e:
    		print e
    elif method =='POST':
    	resp = None
    	try:
    		resp = requests.post(url,headers=headers,data=data)
    	except Exception,e:
    		print e
    	if resp:
    		try:
    			wd2.get(url)
    		except Exception,e:
    			print e
    while True:
   		#处理异常
    	try:
    		wd2.current_url
    		time.sleep(0.5)
    	except Exception,e:
    		break
   	try:
   		wd2.quit()
   	except Exception,e:
   		print e
   	wd1.back()