41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from crypt_random import secure_random
|
|
from lora_handler import LoRaHandler, LoRaMessage
|
|
|
|
import time
|
|
import uasyncio as asyncio
|
|
|
|
|
|
class Connection:
|
|
def __init__(self):
|
|
self.load_public_constants()
|
|
|
|
# generate diffie-hellman keys
|
|
print("Generating keys...")
|
|
print(" - Generating private key...")
|
|
self.private_key = int.from_bytes(secure_random(32), "big") # 32 bytes = 256 bits
|
|
print(" - Generating public key...")
|
|
self.public_key = pow(self.G, self.private_key, self.P)
|
|
|
|
# start lora handler
|
|
self.lora = LoRaHandler()
|
|
|
|
async def advertising_loop(self):
|
|
while True:
|
|
self.lora.advertise()
|
|
await asyncio.sleep(5)
|
|
|
|
async def start(self):
|
|
print("Starting connection...")
|
|
asyncio.create_task(self.advertising_loop())
|
|
|
|
while True:
|
|
msg: LoRaMessage = await self.lora.message_queue.get()
|
|
|
|
print(f"Received message from {msg.sender_id}: {msg.payload}")
|
|
print("Sending response...")
|
|
self.lora.send(b"Pong!", msg.sender_id)
|
|
|
|
def load_public_constants(self):
|
|
with open("public_constants.txt", "r") as f:
|
|
self.P = int(f.readline(), 16)
|
|
self.G = int(f.readline(), 16) |