simple-proxy 0.0.5__tar.gz → 0.0.7__tar.gz

{simple-proxy-0.0.5/simple_proxy.egg-info → simple-proxy-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: simple-proxy
-Version: 0.0.5
+Version: 0.0.7
 Summary: A very simple NIO TCP proxy server
 Home-page: https://github.com/ruanhao/simple-proxy
 Author: Hao Ruan

{simple-proxy-0.0.5 → simple-proxy-0.0.7}/setup.py

@@ -22,7 +22,7 @@ config = {
     'keywords': ['network', 'tcp', 'non-blocking', 'proxy'],
     'version': __version__,
     'packages': find_packages(),
-    'install_requires': ['click', 'py-netty', 'cryptography'],
+    'install_requires': ['click', 'py-netty', 'cryptography~=42.0.0', 'attrs~=22.1.0'],
     'python_requires': ">=3.7, <4",
     'setup_requires': ['wheel'],
     'package_data': {'simple_proxy': ['*']},

{simple-proxy-0.0.5 → simple-proxy-0.0.7}/simple_proxy/__init__.py

@@ -4,6 +4,7 @@ import ssl
 from functools import wraps
 from py_netty.handler import LoggingChannelHandler
 from py_netty import Bootstrap, ServerBootstrap, EventLoopGroup
+from py_netty.channel import NioSocketChannel
 import traceback
 import sys
 import os
@@ -188,12 +189,15 @@ def sneaky():
 
 
 @sneaky()
-def _handle(buffer, direction, src, dst, print_content, to_file):
-    try:
-        src_ip, src_port = src.getpeername()[:2]
-        dst_ip, dst_port = dst.getpeername()[:2]
-    except OSError:
-        return buffer
+def _handle(buffer: bytes, direction: bool, src: NioSocketChannel, dst: NioSocketChannel, print_content: bool, to_file: bool):
+    # try:
+    #     src_ip, src_port = src.getpeername()[:2]
+    #     dst_ip, dst_port = dst.getpeername()[:2]
+    # except OSError:
+    #     return buffer
+
+    src_ip, src_port = src.channelinfo().peername
+    dst_ip, dst_port = dst.channelinfo().peername
 
     raddr = (src_ip, src_port) if direction else (dst_ip, dst_port)
 
@@ -297,7 +301,7 @@ class ProxyChannelHandler(LoggingChannelHandler):
         class _ChannelHandler(LoggingChannelHandler):
 
             def channel_read(this, ctx, bytebuf):
-                _handle(bytebuf, False, ctx.channel().socket(), ctx0.channel().socket(), self._content, self._to_file)
+                _handle(bytebuf, False, ctx.channel(), ctx0.channel(), self._content, self._to_file)
                 ctx0.write(bytebuf)
 
             def channel_inactive(this, ctx):
@@ -339,7 +343,7 @@ class ProxyChannelHandler(LoggingChannelHandler):
                 self._client_channel(ctx, self._remote_host, self._remote_port)
             _clients[self.raddr].proxy_socket = self._client.socket()
 
-        _handle(bytebuf, True, ctx.channel().socket(), self._client.socket(), self._content, self._to_file)
+        _handle(bytebuf, True, ctx.channel(), self._client, self._content, self._to_file)
         self._client.write(bytebuf)
 
     def channel_inactive(self, ctx):

simple-proxy-0.0.7/simple_proxy/version.py

@@ -0,0 +1 @@
+__version__ = "0.0.7"

{simple-proxy-0.0.5 → simple-proxy-0.0.7/simple_proxy.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: simple-proxy
-Version: 0.0.5
+Version: 0.0.7
 Summary: A very simple NIO TCP proxy server
 Home-page: https://github.com/ruanhao/simple-proxy
 Author: Hao Ruan

simple-proxy-0.0.7/simple_proxy.egg-info/requires.txt

@@ -0,0 +1,4 @@
+click
+py-netty
+cryptography~=42.0.0
+attrs~=22.1.0

simple-proxy-0.0.5/simple_proxy/version.py

@@ -1 +0,0 @@
-__version__ = "0.0.5"

simple-proxy-0.0.5/simple_proxy.egg-info/requires.txt

@@ -1,3 +0,0 @@
-click
-py-netty
-cryptography