mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 00:01:36 +00:00
remove platform.setup
This commit is contained in:
parent
a9e6121a08
commit
a3c3e4e504
@ -1,7 +1,6 @@
|
||||
import sys
|
||||
|
||||
resolver = None
|
||||
setup = None
|
||||
|
||||
if sys.platform == "linux2":
|
||||
from . import linux
|
||||
@ -15,4 +14,3 @@ elif sys.platform == "freebsd10":
|
||||
elif sys.platform == "win32":
|
||||
from . import windows
|
||||
resolver = windows.Resolver
|
||||
setup = windows.TransparentProxy.setup
|
@ -18,6 +18,8 @@ PROXY_API_PORT = 8085
|
||||
class Resolver(object):
|
||||
|
||||
def __init__(self):
|
||||
TransparentProxy.setup()
|
||||
|
||||
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
self.socket.connect(("127.0.0.1", PROXY_API_PORT))
|
||||
|
||||
@ -140,13 +142,13 @@ class TransparentProxy(object):
|
||||
self.handle_icmp = self.driver.open_handle(filter="icmp", layer=Layer.NETWORK, flags=Flag.DROP)
|
||||
|
||||
@classmethod
|
||||
def setup(cls, options):
|
||||
def setup(cls):
|
||||
# TODO: Make sure that server can be killed cleanly. That's a bit difficult as we don't have access to
|
||||
# controller.should_exit when this is called.
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
server_unavailable = s.connect_ex(("127.0.0.1", PROXY_API_PORT))
|
||||
if server_unavailable:
|
||||
proxifier = TransparentProxy(proxy_addr=options.addr, proxy_port=options.port)
|
||||
proxifier = TransparentProxy()
|
||||
proxifier.start()
|
||||
|
||||
def start(self):
|
||||
|
@ -41,8 +41,6 @@ def process_proxy_options(parser, options):
|
||||
c += 1
|
||||
if not platform.resolver:
|
||||
return parser.error("Transparent mode not supported on this platform.")
|
||||
if platform.setup:
|
||||
platform.setup(options)
|
||||
get_upstream_server = TransparentUpstreamServerResolver(platform.resolver(), TRANSPARENT_SSL_PORTS)
|
||||
http_form_in, http_form_out = "relative", "relative"
|
||||
if options.reverse_proxy:
|
||||
|
@ -70,8 +70,7 @@ class TestProcessProxyOptions:
|
||||
self.assert_err("transparent mode not supported", "-T")
|
||||
|
||||
@mock.patch("libmproxy.platform.resolver")
|
||||
@mock.patch("libmproxy.platform.setup")
|
||||
def test_transparent_reverse(self, _, __):
|
||||
def test_transparent_reverse(self, _):
|
||||
self.assert_err("mutually exclusive", "-R", "http://localhost", "-T")
|
||||
self.assert_noerr("-T")
|
||||
self.assert_err("Invalid server specification", "-R", "reverse")
|
||||
|
Loading…
Reference in New Issue
Block a user