diff options
author | Maximilian Hils <git@maximilianhils.com> | 2019-11-12 05:04:05 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-11-12 05:04:05 +0100 |
commit | dac0bfe786a8d1dfdd97f29d6bb262ed258153fa (patch) | |
tree | 6a21ead3cbaab6f996c931d91fb071dc3c6b00ac /mitmproxy/platform | |
parent | e7c3975bfca8f0c1d48cef6a1263d579269157be (diff) | |
parent | bdc15cbe0c0cd5175af1b58078d65d400cca71d1 (diff) | |
download | mitmproxy-dac0bfe786a8d1dfdd97f29d6bb262ed258153fa.tar.gz mitmproxy-dac0bfe786a8d1dfdd97f29d6bb262ed258153fa.tar.bz2 mitmproxy-dac0bfe786a8d1dfdd97f29d6bb262ed258153fa.zip |
Merge pull request #3691 from mhils/sans-io-adjustments
Update mypy, sans-io adjustments
Diffstat (limited to 'mitmproxy/platform')
-rw-r--r-- | mitmproxy/platform/__init__.py | 28 |
1 files changed, 16 insertions, 12 deletions
diff --git a/mitmproxy/platform/__init__.py b/mitmproxy/platform/__init__.py index 61946ec4..7e690789 100644 --- a/mitmproxy/platform/__init__.py +++ b/mitmproxy/platform/__init__.py @@ -1,7 +1,7 @@ import re import socket import sys -from typing import Tuple +from typing import Callable, Optional, Tuple def init_transparent_mode() -> None: @@ -10,30 +10,34 @@ def init_transparent_mode() -> None: """ -def original_addr(csock: socket.socket) -> Tuple[str, int]: - """ - Get the original destination for the given socket. - This function will be None if transparent mode is not supported. - """ - +original_addr: Optional[Callable[[socket.socket], Tuple[str, int]]] +""" +Get the original destination for the given socket. +This function will be None if transparent mode is not supported. +""" if re.match(r"linux(?:2)?", sys.platform): from . import linux - original_addr = linux.original_addr # noqa + original_addr = linux.original_addr elif sys.platform == "darwin" or sys.platform.startswith("freebsd"): from . import osx - original_addr = osx.original_addr # noqa + original_addr = osx.original_addr elif sys.platform.startswith("openbsd"): from . import openbsd - original_addr = openbsd.original_addr # noqa + original_addr = openbsd.original_addr elif sys.platform == "win32": from . import windows resolver = windows.Resolver() init_transparent_mode = resolver.setup # noqa - original_addr = resolver.original_addr # noqa + original_addr = resolver.original_addr else: - original_addr = None # noqa + original_addr = None + +__all__ = [ + "original_addr", + "init_transparent_mode" +] |