proxy.py/proxy/socks/handler.py

# -*- coding: utf-8 -*-
"""
    proxy.py
    ~~~~~~~~
    ⚡⚡⚡ Fast, Lightweight, Pluggable, TLS interception capable proxy server focused on
    Network monitoring, controls & Application development, testing, debugging.

    :copyright: (c) 2013-present by Abhinav Singh and contributors.
    :license: BSD, see LICENSE for more details.
"""
from typing import Any, Optional

from .client import SocksClientConnection
from ..core.base import BaseTcpServerHandler


class SocksProtocolHandler(BaseTcpServerHandler[SocksClientConnection]):
    """Reference https://www.openssh.com/txt/socks4.protocol"""

    def __init__(self, *args: Any, **kwargs: Any) -> None:
        super().__init__(*args, **kwargs)

    @staticmethod
    def create(*args: Any) -> SocksClientConnection:
        return SocksClientConnection(*args)     # pragma: no cover

    def handle_data(self, data: memoryview) -> Optional[bool]:
        return super().handle_data(data)    # pragma: no cover