Method main

examples/contrib/mitmproxywrapper.py:127–181

(cls)

    @classmethod
    def main(cls):
        parser = argparse.ArgumentParser(
            description="Helper tool for OS X proxy configuration and mitmproxy.",
            epilog="Any additional arguments will be passed on unchanged to mitmproxy/mitmweb.",
        )
        parser.add_argument(
            "-t",
            "--toggle",
            action="store_true",
            help="just toggle the proxy configuration",
        )
        # parser.add_argument('--honeyproxy', action='store_true', help='run honeyproxy instead of mitmproxy')
        parser.add_argument(
            "-p",
            "--port",
            type=int,
            help="override the default port of 8080",
            default=8080,
        )
        parser.add_argument(
            "-P",
            "--port-random",
            action="store_true",
            help="choose a random unused port",
        )
        parser.add_argument(
            "-w",
            "--web",
            action="store_true",
            help="web interface: run mitmweb instead of mitmproxy",
        )
        args, extra_arguments = parser.parse_known_args()
        port = args.port

        # Allocate a random unused port, and hope no other process steals it before mitmproxy/mitmweb uses it.
        # Passing the allocated socket to mitmproxy/mitmweb would be nicer of course.
        if args.port_random:
            with socketserver.TCPServer(("localhost", 0), None) as s:
                port = s.server_address[1]
                print(f"Using random port {port}...")

        wrapper = cls(port=port, use_mitmweb=args.web, extra_arguments=extra_arguments)

        def handler(signum, frame):
            print("Cleaning up proxy settings...")
            wrapper.toggle_proxy()

        signal.signal(signal.SIGINT, handler)

        if args.toggle:
            wrapper.toggle_proxy()
        # elif args.honeyproxy:
        #     wrapper.wrap_honeyproxy()
        else:
            wrapper.wrap_mitmproxy()


if __name__ == "__main__":