Completed Step Decentralized-System#3 and Step Decentralized-System#5

Author: alesarritz

src/onionRouters/simpleOnionRouter.ts

@@ -1,17 +1,52 @@
 import bodyParser from "body-parser";
 import express, { Request, Response } from "express";
-import { BASE_ONION_ROUTER_PORT } from "../config";
+import { BASE_ONION_ROUTER_PORT, REGISTRY_PORT } from "../config";
+import { generateRsaKeyPair, exportPubKey, exportPrvKey } from "../crypto";
+import { webcrypto } from "node:crypto";
+
+type CK = webcrypto.CryptoKey;
 
 export async function simpleOnionRouter(nodeId: number) {
   const onionRouter = express();
   onionRouter.use(express.json());
   onionRouter.use(bodyParser.json());
 
+  let privateKey: CK;
+  let publicKey: CK;
+  let privateKeyStr: string;
+  let publicKeyStr: string;
+
   // Store the last received messages and destination (default: null)
   let lastReceivedEncryptedMessage: string | null = null;
   let lastReceivedDecryptedMessage: string | null = null;
   let lastMessageDestination: number | null = null;
 
+  try {
+    const keyPair = await generateRsaKeyPair();
+    privateKey = keyPair.privateKey;
+    publicKey = keyPair.publicKey;
+
+    // Convert keys to base64
+    publicKeyStr = await exportPubKey(publicKey);
+    privateKeyStr = await exportPrvKey(privateKey);
+
+    // Register with the registry
+    const response = await fetch(`http://localhost:${REGISTRY_PORT}/registerNode`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ nodeId, pubKey: publicKeyStr }),
+    });
+
+    if (!response.ok) {
+      throw new Error(`Failed to register node ${nodeId}`);
+    }
+
+    console.log(`Node ${nodeId} registered successfully`);
+  } catch (error) {
+    console.error("Failed to initialize node:", error);
+    throw error;
+  }
+
   // Status route
   onionRouter.get("/status", (req: Request, res: Response) => {
     res.send("live");
@@ -32,10 +67,13 @@ export async function simpleOnionRouter(nodeId: number) {
     res.json({ result: lastMessageDestination });
   });
 
+  // Provide private key for testing purposes
+  onionRouter.get("/getPrivateKey", (req: Request, res: Response) => {
+    res.json({ result: privateKeyStr });
+  });
+
   const server = onionRouter.listen(BASE_ONION_ROUTER_PORT + nodeId, () => {
-    console.log(
-      `Onion router ${nodeId} is listening on port ${BASE_ONION_ROUTER_PORT + nodeId}`
-    );
+    console.log(`Onion router ${nodeId} is listening on port ${BASE_ONION_ROUTER_PORT + nodeId}`);
   });
 
   return server;

src/registry/registry.ts

@@ -3,6 +3,7 @@ import express, { Request, Response } from "express";
 import { REGISTRY_PORT } from "../config";
 
 export type Node = { nodeId: number; pubKey: string };
+const nodes: Node[] = [];
 
 export type RegisterNodeBody = {
   nodeId: number;
@@ -18,12 +19,36 @@ export async function launchRegistry() {
   _registry.use(express.json());
   _registry.use(bodyParser.json());
 
-  // TODO implement the status route
-  // _registry.get("/status", (req, res) => {});
+  // Status route
   _registry.get("/status", (req: Request, res: Response) => {
     res.send("live");
   });
 
+  _registry.post("/registerNode", (req: Request, res: Response) => {
+    const { nodeId, pubKey } = req.body as RegisterNodeBody;
+
+    if (typeof nodeId !== "number" || typeof pubKey !== "string") {
+      return res.status(400).json({ error: "Invalid request body" });
+    }
+
+    // Check if the node already exists
+    const existingNode = nodes.find((n) => n.nodeId === nodeId);
+    if (existingNode) {
+      existingNode.pubKey = pubKey;
+    } else {
+      nodes.push({ nodeId, pubKey });
+    }
+
+    res.status(200).json({ success: true });
+    return res.status(200).json({ success: true });
+  });
+
+  // Implement /getNodeRegistry to return the list of registered nodes
+  _registry.get("/getNodeRegistry", (req: Request, res: Response) => {
+    res.json({ nodes });
+  });
+
+
   const server = _registry.listen(REGISTRY_PORT, () => {
     console.log(`registry is listening on port ${REGISTRY_PORT}`);
   });