WooYun.org

#!/usr/bin/python
#coding:utf-8
import urllib
import urllib2
import cookielib
import sys 
import time
import datetime
import gc
import ssl
import chardet
from Queue import Queue
class ConnectTool:
	def  __init__(self,debuglevel=0):
		self.__null_proxy_handler= urllib2.ProxyHandler({})
		urllib2.socket.setdefaulttimeout(10)                                    #设置超时时间
		self.__headers = { 
			'User-Agent' :		'Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)',
			'Referer':		 'http://**.**.**.**',
			'Cookie':			'pyerpcookie=loginname=admin&truename=系统管理员&flag=1&datagroup=all&kefugroup=总经办&kpgroup=0&kpdian='
			 }
		self.__cookie=cookielib.CookieJar()
		self.__cJar=cookielib.LWPCookieJar()
		self.__httpcookieprocessor=urllib2.HTTPCookieProcessor(self.__cookie)
		self.__httpHandler= urllib2.HTTPHandler(debuglevel=debuglevel)
		self.__httpsHandler=urllib2.HTTPSHandler(debuglevel=debuglevel)
		self.__opener=''
		self.__opener=urllib2.build_opener(self.__httpcookieprocessor,self.__null_proxy_handler,self.__httpHandler,self.__httpsHandler)
		urllib2.install_opener(self.__opener)
	def  getHTML(self,URL,way='GET',params={},times=1):
		print datetime.datetime.now()
		data = 'fs=0&id=0&roles=22&username=testo&yusername=&pwd=testo&truename=testo&sex=男&age=1900-1-1&mob=&hk=&sfz=&zw=&kefugroup=总经办&week=&times='
		url=URL
		if way=='POST':
			req = urllib2.Request(url, data=data, headers=self.__headers)
		elif len(params)==0:
			req= urllib2.Request(url,headers=self.__headers)
		else :
			req= urllib2.Request(url+'?'+data,headers=self.__headers)
		response=None
		try:
#			gc.enable() 
#			gc.set_debug(gc.DEBUG_LEAK)
			context = ssl._create_unverified_context()
			response = urllib2.urlopen(req,context=context)
			temp=str(response.info())
# 			print 'cooke信息如下：'
# 			for item in self.__cookie:
# 				print 'Name = '+item.name
# 				print 'Value = '+item.value
			msg=response.read()
			
			chardit1 = chardet.detect(msg)
			the_page = str(msg)
			
			
			try:
				return temp.decode(chardit1['encoding']).encode('utf-8'),the_page.decode(chardit1['encoding']).encode('utf-8')
			except Exception,e:
				return temp,the_page
			
#		response = urllib2.urlopen('http://**.**.**.**',timeout=10)
#		print 'head is %s' % response.info()
		except Exception,e:
			msgg=None
			try:
				msgg= '错误码为: %s' % str(e).encode('utf-8')
			except Exception,e:
				msgg= '错误码为: %s' % str(e)
			print msgg
			if times <4:
				print '尝试第'+str(times)+'次'
				time.sleep(3)
				return self.getHTML(URL, way, params, times+1)
			else :
				print '失败次数过多，停止链接'
				the_page= msgg
				return '',the_page
		finally:
				if response:
					response.close()
# 					del response
		#response.close()
if __name__ == "__main__":		
	p=ConnectTool()
	w,e=p.getHTML('http://**.**.**.**/Ajax/users.ashx',way='POST')
 	print w,e