From 6421cb6118c0bb95f29a7890015534ff1d41def3 Mon Sep 17 00:00:00 2001 From: jesopo Date: Mon, 28 Oct 2019 18:11:57 +0000 Subject: [PATCH] improve proxy.py socket override using nested function def --- modules/proxy.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/modules/proxy.py b/modules/proxy.py index c66710c8..1bcaebd1 100644 --- a/modules/proxy.py +++ b/modules/proxy.py @@ -32,8 +32,7 @@ class Module(ModuleManager.BaseModule): type, proxy_parsed.hostname, proxy_parsed.port) def _socket_factory(self, ptype, phost, pport): - return lambda host, port, bind, timeout: self._make_socket( - ptype, phost, pport, host, port, bind, timeout) - def _make_socket(self, ptype, phost, pport, host, port, bind, timeout): - return socks.create_connection((host, port), timeout, bind, - ptype, phost, pport) + def _(host, port, bind, timeout): + return socks.create_connection((host, port), timeout, bind, + ptype, phost, pport) + return _