From fffcd025adc717baf89bce78962300f90171344f Mon Sep 17 00:00:00 2001
From: Tony Garnock-Jones <tonyg@leastfixedpoint.com>
Date: Thu, 19 Aug 2021 18:40:05 -0400
Subject: [PATCH] Stdio transport

---
 syndicate/relay.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/syndicate/relay.py b/syndicate/relay.py
index 37a0c0c..20ded56 100644
--- a/syndicate/relay.py
+++ b/syndicate/relay.py
@@ -1,3 +1,4 @@
+import sys
 import asyncio
 import websockets
 import logging
@@ -429,3 +430,18 @@ class WebsocketTunnelRelay(TunnelRelay):
         self.loop = None
         self.ws = None
         return True
+
+@transport.address(transportAddress.Stdio)
+class PipeTunnelRelay(_StreamTunnelRelay):
+    def __init__(self, turn, address, input_fileobj = sys.stdin, output_fileobj = sys.stdout, **kwargs):
+        super().__init__(turn, address, **kwargs)
+        self.input_fileobj = input_fileobj
+        self.output_fileobj = output_fileobj
+        self.reader = asyncio.StreamReader()
+
+    async def _create_connection(self, loop):
+        return await loop.connect_read_pipe(lambda: self, self.input_fileobj)
+
+    def _send_bytes(self, bs):
+        self.output_fileobj.buffer.write(bs)
+        self.output_fileobj.buffer.flush()