diff --git a/mtprotoproxy.py b/mtprotoproxy.py index c4aad14..195b5e6 100755 --- a/mtprotoproxy.py +++ b/mtprotoproxy.py @@ -150,6 +150,9 @@ class LayeredStreamWriterBase: def write(self, data): return self.upstream.write(data) + def write_eof(self): + return self.upstream.write_eof() + async def drain(self): return await self.upstream.drain()