add backward auth

qwj · qwj · commit b947d0586d84 · 2020-01-09T19:05:43.000+08:00
diff --git a/pproxy/__doc__.py b/pproxy/__doc__.py
@@ -1,5 +1,5 @@
 __title__       = "pproxy"
-__version__     = "2.3.0"
+__version__     = "2.3.1"
 __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
@@ -221,6 +221,7 @@ async def server_run(self, handler):
                 wait = asyncio.open_connection(host=self.uri.host_name, port=self.uri.port, local_addr=(self.uri.lbind, 0) if self.uri.lbind else None)
             try:
                 reader, writer = await asyncio.wait_for(wait, timeout=SOCKET_TIMEOUT)
+                writer.write(self.uri.auth)
                 self.writer = writer
                 try:
                     data = await reader.read_n(1)
@@ -242,7 +243,8 @@ async def server_run(self, handler):
                     errwait = min(errwait*1.3 + 0.1, 30)
     def client_run(self, args):
         async def handler(reader, writer):
-            await self.conn.put((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.unix:
             return asyncio.start_unix_server(handler, path=self.uri.bind)
         else:
