Beau Kujath
2 years ago
3 changed files with 0 additions and 250 deletions
@ -1,182 +0,0 @@ |
|||||
import asyncio |
|
||||
import threading |
|
||||
from scapy.all import * |
|
||||
import argparse |
|
||||
import base64 |
|
||||
|
|
||||
DEBUG=False |
|
||||
|
|
||||
def print_dbg(s): |
|
||||
if DEBUG: |
|
||||
print(s) |
|
||||
|
|
||||
class NGServer: |
|
||||
def __init__(self, bridge_ip='127.0.0.1', bridge_port=8000, |
|
||||
ioip='127.0.0.1', |
|
||||
incoming_port=8001, outgoing_port=8002): |
|
||||
self.incoming_queue = None # asyncio.Queue() |
|
||||
self.outgoing_queue = None # asyncio.Queue() |
|
||||
self.client_reader = None |
|
||||
self.client_writer = None |
|
||||
self.server_reader = None |
|
||||
self.server_writer = None |
|
||||
self.server_thread = None |
|
||||
|
|
||||
self.bridge_ip = bridge_ip |
|
||||
self.bridge_port = bridge_port |
|
||||
self.ioip = ioip |
|
||||
self.incoming_port = incoming_port |
|
||||
self.outgoing_port = outgoing_port |
|
||||
|
|
||||
async def handle_client(self, reader, writer): |
|
||||
self.client_reader = reader |
|
||||
self.client_writer = writer |
|
||||
while True: |
|
||||
data = await reader.read(1024) |
|
||||
if not data: |
|
||||
writer.close() |
|
||||
break |
|
||||
await self.incoming_queue.put(data) |
|
||||
|
|
||||
async def handle_incoming(self, reader, writer): |
|
||||
self.incoming_queue = asyncio.Queue() |
|
||||
while True: |
|
||||
data = await self.incoming_queue.get() |
|
||||
writer.write(data) |
|
||||
await writer.drain() |
|
||||
|
|
||||
async def handle_outgoing(self, reader, writer): |
|
||||
self.server_reader = reader |
|
||||
self.server_writer = writer |
|
||||
print_dbg("handle_outgoing - ENTER") |
|
||||
self.outgoing_queue = asyncio.Queue() |
|
||||
while True: |
|
||||
print_dbg("handle_outgoing - IN LOOP") |
|
||||
data = await reader.read(1024) |
|
||||
print_dbg("handle_outgoing - IP(data))={}".format( |
|
||||
IP(data))) |
|
||||
self.client_writer.write(data) |
|
||||
await self.client_writer.drain() |
|
||||
|
|
||||
async def read_from_server(self): |
|
||||
print_dbg("read_from_server - ENTER") |
|
||||
while True: |
|
||||
# Switch reader to writer |
|
||||
print_dbg("read_from_server - IN LOOP") |
|
||||
data = await self.client_reader.read(1024) |
|
||||
print_dbg("read_from_server - IP(data))={}".format( |
|
||||
IP(data))) |
|
||||
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=self.bridge_ip, |
|
||||
port=self.bridge_port) |
|
||||
async with server: |
|
||||
await server.serve_forever() |
|
||||
|
|
||||
async def start_incoming_server(self): |
|
||||
server = await asyncio.start_server( |
|
||||
self.handle_incoming, |
|
||||
host=self.ioip, |
|
||||
port=self.incoming_port) |
|
||||
async with server: |
|
||||
await server.serve_forever() |
|
||||
|
|
||||
async def start_outgoing_server(self): |
|
||||
server = await asyncio.start_server( |
|
||||
self.handle_outgoing, |
|
||||
host=self.ioip, |
|
||||
port=self.outgoing_port) |
|
||||
async with server: |
|
||||
await server.serve_forever() |
|
||||
|
|
||||
async def start_thread(ng): |
|
||||
""" """ |
|
||||
await asyncio.gather( |
|
||||
ng.start_server(), |
|
||||
ng.start_incoming_server(), |
|
||||
ng.start_outgoing_server(), |
|
||||
) |
|
||||
|
|
||||
await asyncio.gather( |
|
||||
ng.read_from_server(), |
|
||||
ng.handle_incoming(), |
|
||||
ng.handle_outgoing(), |
|
||||
) |
|
||||
|
|
||||
def run_thread(ng): |
|
||||
asyncio.run(start_thread(ng)) |
|
||||
|
|
||||
|
|
||||
def startServerAndWait(args): |
|
||||
ng = NGServer(bridge_ip=args.bridge_ip, |
|
||||
bridge_port=args.bridge_port, |
|
||||
ioip=args.ioip, |
|
||||
incoming_port=args.incoming_port, |
|
||||
outgoing_port=args.outgoing_port) |
|
||||
ng.server_thread = threading.Thread( |
|
||||
target=run_thread, |
|
||||
args=(ng,)) |
|
||||
ng.server_thread.start() |
|
||||
ng.server_thread.join() |
|
||||
|
|
||||
def startThread(ng): |
|
||||
|
|
||||
ng.server_thread = threading.Thread( |
|
||||
target=run_thread, |
|
||||
args=(ng,)) |
|
||||
ng.server_thread.start() |
|
||||
|
|
||||
def connectIO(inaddr='127.0.0.1', inport=8001, outaddr='127.0.0.1', outport=8002): |
|
||||
insock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, socket.IPPROTO_TCP) |
|
||||
insock.connect((inaddr, inport)) |
|
||||
outsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, socket.IPPROTO_TCP) |
|
||||
outsock.connect((outaddr, outport)) |
|
||||
return insock, outsock |
|
||||
|
|
||||
|
|
||||
def base64_decode_with_plus(data): |
|
||||
data = data.rstrip('+') |
|
||||
padding = 4 - len(data) % 4 |
|
||||
data += '=' * padding |
|
||||
return base64.b64decode(data) |
|
||||
|
|
||||
def base64_encode_with_plus(data): |
|
||||
encoded_data = base64.b64encode(data).rstrip(b'=') |
|
||||
padding = (4 - len(encoded_data) % 4) % 4 |
|
||||
encoded_data += b'+' * padding |
|
||||
return encoded_data.decode('ascii') |
|
||||
|
|
||||
|
|
||||
def parseArgs(): |
|
||||
args = argparse.ArgumentParser() |
|
||||
# The bridge ip and port |
|
||||
args.add_argument("-B","--bridge_ip", default='127.0.0.1', |
|
||||
help='The IP that listens for NetGenie mobile app client connections.') |
|
||||
args.add_argument("-b","--bridge_port", default=8000, help='The port that listens for NetGenie mobile app client connections.') |
|
||||
|
|
||||
# The IO ip and ports |
|
||||
args.add_argument("-I","--ioip", default='127.0.0.1', help='The IP that listens on the server for connections from whoever is debugging the phone.') |
|
||||
args.add_argument("-i","--incoming_port", default=8001, |
|
||||
help='The port that the person debugging will read packets from the device.') |
|
||||
args.add_argument("-o","--outgoing_port", default=8002, |
|
||||
help='The port that the person debugging will write packets to the device.') |
|
||||
return args.parse_args() |
|
||||
|
|
||||
|
|
||||
def main(): |
|
||||
""" |
|
||||
The NGServer class implements our NetGenie debug/bridge server. |
|
||||
""" |
|
||||
# Get the arguments |
|
||||
args = parseArgs() |
|
||||
# Start the server. |
|
||||
startServerAndWait(args) |
|
||||
|
|
||||
if __name__ == '__main__': |
|
||||
main() |
|
||||
@ -1,67 +0,0 @@ |
|||||
# Quick Start |
|
||||
|
|
||||
## Starting the server |
|
||||
|
|
||||
On the server, run the following |
|
||||
|
|
||||
```bash |
|
||||
$ python3 -m venv .example |
|
||||
$ activate .example/bin/activate |
|
||||
|
|
||||
$ python3 NetGenieSrv.py |
|
||||
``` |
|
||||
|
|
||||
This opens three sockets: |
|
||||
|
|
||||
1. The bridge socket that listens for the phone to connect to. |
|
||||
2. An ``incoming'' socket to read packets sent from the device to the server |
|
||||
3. An ``outgoing'' socket to write packets from the server to the device. |
|
||||
|
|
||||
On the phone, connect to the server and port specified by the `-B` and `-b` arguments. |
|
||||
|
|
||||
This can be specified by changing the source code or using the following command-line |
|
||||
arguments: |
|
||||
|
|
||||
```bash |
|
||||
$ python3 NetGenieSrv.py -h |
|
||||
usage: NetGenieSrv.py [-h] [-B BRIDGE_IP] [-b BRIDGE_PORT] [-I IOIP] [-i INCOMING_PORT] [-o OUTGOING_PORT] |
|
||||
|
|
||||
options: |
|
||||
-h, --help show this help message and exit |
|
||||
-B BRIDGE_IP, --bridge_ip BRIDGE_IP |
|
||||
The IP that listens for NetGenie mobile app client connections. |
|
||||
-b BRIDGE_PORT, --bridge_port BRIDGE_PORT |
|
||||
The port that listens for NetGenie mobile app client connections. |
|
||||
-I IOIP, --ioip IOIP The IP that listens on the server for connections from whoever is debugging the phone. |
|
||||
-i INCOMING_PORT, --incoming_port INCOMING_PORT |
|
||||
The port that the person debugging will read packets from the device. |
|
||||
-o OUTGOING_PORT, --outgoing_port OUTGOING_PORT |
|
||||
The port that the person debugging will write packets to the device. |
|
||||
``` |
|
||||
|
|
||||
You can also start NetGenieSrv from a python REPL as follows: |
|
||||
|
|
||||
```bash |
|
||||
In [1]: import NetGenieSrv as ngs |
|
||||
In [2]: ng = ngs.NGServer(bridge_ip='0.0.0.0') |
|
||||
In [3]: ngs.startThread(ng) |
|
||||
|
|
||||
``` |
|
||||
|
|
||||
## Using the server |
|
||||
|
|
||||
```bash |
|
||||
$ ipython3 |
|
||||
|
|
||||
import NetGenieSrv as ngs |
|
||||
from scapy.all import * |
|
||||
ip = IP(src='10.10.20.1', dst='255.255.4.20') |
|
||||
tcp = TCP(sport=8080, dport=8443) |
|
||||
pkt = ip/tcp |
|
||||
a = pkt.build() |
|
||||
|
|
||||
insock, outsock = ngs.connectIO() |
|
||||
pktIn = insock.read(40) # Read a packet send from the phone |
|
||||
outsock.send(a) # Send a packet to the NetGenie client. |
|
||||
|
|
||||
``` |
|
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue