proj: Add PoC
This commit is contained in:
34
proj/server/src/server.py
Executable file
34
proj/server/src/server.py
Executable file
@@ -0,0 +1,34 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
from socketserver import UnixStreamServer, StreamRequestHandler, ThreadingMixIn
|
||||
import os
|
||||
|
||||
|
||||
class Handler(StreamRequestHandler):
|
||||
def handle(self):
|
||||
first = self.rfile.readline()
|
||||
pid = int(first.split(b':')[1])
|
||||
print(f'Process with PID {pid} connected')
|
||||
while True:
|
||||
msg = self.rfile.readline()
|
||||
if not msg:
|
||||
return
|
||||
timestamp, data = msg.split(b' ', 1)
|
||||
if not data.startswith(b'return '):
|
||||
print(data)
|
||||
#self.wfile.write(b'ok\n')
|
||||
else:
|
||||
print(data)
|
||||
|
||||
|
||||
class ThreadedUnixStreamServer(ThreadingMixIn, UnixStreamServer):
|
||||
pass
|
||||
|
||||
|
||||
def main() -> None:
|
||||
os.unlink('/tmp/test')
|
||||
with ThreadedUnixStreamServer('/tmp/test', Handler) as server:
|
||||
server.serve_forever()
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user