netgenie-server: Added a new server. todo - test.
This commit is contained in:
parent
8b73e6f5a6
commit
8f68432999
2
NetGuard
2
NetGuard
@ -1 +1 @@
|
|||||||
Subproject commit 0350e46d6dbe398dc395675e348121fe3ae6624a
|
Subproject commit e3bbfed2c3548f815d740c9ae261329d708f6d9c
|
@ -1,41 +1,89 @@
|
|||||||
#!/bin/env python3
|
import asyncio
|
||||||
|
|
||||||
def check_bridge_socket()::
|
class NGServer:
|
||||||
evt = None
|
def __init__(self):
|
||||||
return evt
|
self.incoming_queue = asyncio.Queue()
|
||||||
|
self.outgoing_queue = asyncio.Queue()
|
||||||
|
self.client_reader = None
|
||||||
|
self.client_writer = None
|
||||||
|
|
||||||
def handle_bridge_evt(evt):
|
async def handle_client(self, reader, writer):
|
||||||
"""
|
self.client_reader = reader
|
||||||
TODO: Whatever bridge events to handle
|
self.client_writer = writer
|
||||||
"""
|
|
||||||
|
|
||||||
|
|
||||||
def check_debug_socket():
|
|
||||||
dbg_evt = None
|
|
||||||
return dbg_evt
|
|
||||||
|
|
||||||
def handle_debug_evt(dbg_evt):
|
|
||||||
""" TODO: Do something with the events here."""
|
|
||||||
|
|
||||||
def run_loop():
|
|
||||||
""" """
|
|
||||||
while True:
|
while True:
|
||||||
# 1. Check the BridgeSocket
|
data = await reader.read(1024)
|
||||||
evt = check_bridge_socket()
|
if not data:
|
||||||
if evt != None:
|
writer.close()
|
||||||
handle_bridge_evt(evt)
|
break
|
||||||
dbg_evt = check_debug_socket()
|
await self.incoming_queue.put(data)
|
||||||
if dbg_evt != None:
|
|
||||||
handle_debug_evt(dbg_evt)
|
async def handle_incoming(self, reader, writer):
|
||||||
def main():
|
while True:
|
||||||
|
data = await self.incoming_queue.get()
|
||||||
|
writer.write(data)
|
||||||
|
await writer.drain()
|
||||||
|
|
||||||
|
async def handle_outgoing(self, reader, writer):
|
||||||
|
while True:
|
||||||
|
data = await self.outgoing_queue.get()
|
||||||
|
writer.write(data)
|
||||||
|
await writer.drain()
|
||||||
|
|
||||||
|
async def read_from_server(self):
|
||||||
|
while True:
|
||||||
|
data = await self.client_reader.read(1024)
|
||||||
|
if not data:
|
||||||
|
self.client_writer.close()
|
||||||
|
break
|
||||||
|
await self.outgoing_queue.put(data)
|
||||||
|
|
||||||
|
async def start_server(self):
|
||||||
|
server = await asyncio.start_server(self.handle_client, host='localhost', port=8000)
|
||||||
|
async with server:
|
||||||
|
await server.serve_forever()
|
||||||
|
|
||||||
|
async def start_incoming_server(self):
|
||||||
|
server = await asyncio.start_server(self.handle_incoming, host='localhost', port=8001)
|
||||||
|
async with server:
|
||||||
|
await server.serve_forever()
|
||||||
|
|
||||||
|
async def start_outgoing_server(self):
|
||||||
|
server = await asyncio.start_server(self.handle_outgoing, host='localhost', port=8002)
|
||||||
|
async with server:
|
||||||
|
await server.serve_forever()
|
||||||
|
|
||||||
|
async def do_async_stuff(ng):
|
||||||
|
await asyncio.gather(ng.start_server(),ng.start_queue_server())
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
"""
|
"""
|
||||||
1. TODO: Open Server port that the device connects to.
|
In this code, we're creating and running four separate coroutines:
|
||||||
2. TODO: Open Server port that an admin connects to, i.e., to test sending packets to
|
|
||||||
3. TODO (Optional): We might be able to start a Scapy [REPL](https://en.wikipedia.org/wiki/Read%E2%80%93eval%E2%80%93print_loop)
|
1. start_server
|
||||||
and then add these functions. That way the person in debugging the phone can do live testing.
|
2. start_incoming_server
|
||||||
|
3. start_outgoing_server
|
||||||
|
4. read_from_server.
|
||||||
|
|
||||||
|
We're also creating separate tasks for handling incoming and
|
||||||
|
outgoing data and adding them to the tasks list.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
ng = NGServer()
|
||||||
|
await asyncio.gather(
|
||||||
|
ng.start_server(),
|
||||||
|
ng.start_incoming_server(),
|
||||||
|
ng.start_outgoing_server(),
|
||||||
|
)
|
||||||
|
"""
|
||||||
|
tasks = [
|
||||||
|
ng.read_from_server(),
|
||||||
|
ng.handle_incoming(),
|
||||||
|
ng.handle_outgoing(),
|
||||||
|
]
|
||||||
|
await asyncio.gather(*tasks)
|
||||||
|
"""
|
||||||
|
|
||||||
|
def example():
|
||||||
if __name__ == '__main__':
|
""" """
|
||||||
main()
|
# asyncio.run(main())
|
||||||
|
Loading…
x
Reference in New Issue
Block a user