add backward auth

qwj · qwj · commit 818ab9cc1056 · 2020-01-09T19:17:57.000+08:00
diff --git a/pproxy/__doc__.py b/pproxy/__doc__.py
@@ -1,5 +1,5 @@
 __title__       = "pproxy"
-__version__     = "2.3.1"
+__version__     = "2.3.2"
 __license__     = "MIT"
 __description__ = "Proxy server that can tunnel among remote servers by regex rules."
 __keywords__    = "proxy socks http shadowsocks shadowsocksr ssr redirect pf tunnel cipher ssl udp"
diff --git a/pproxy/server.py b/pproxy/server.py
@@ -243,8 +243,12 @@ async def server_run(self, handler):
                     errwait = min(errwait*1.3 + 0.1, 30)
     def client_run(self, args):
         async def handler(reader, writer):
-            if not self.uri.auth or self.uri.auth == (await reader.read_n(len(self.uri.auth))):
-                await self.conn.put((reader, writer))
+            if self.uri.auth:
+                try:
+                    assert self.uri.auth == (await reader.read_n(len(self.uri.auth)))
+                except Exception:
+                    return
+            await self.conn.put((reader, writer))
         if self.uri.unix:
             return asyncio.start_unix_server(handler, path=self.uri.bind)
         else:
