feat: regenerate routing tables on each change

Author: MegaRedHand

src/context.ts

@@ -19,7 +19,7 @@ export class GlobalContext {
   initialize(viewport: Viewport) {
     this.viewport = viewport;
 
-    const baseIp = "192.168.1.0";
+    const baseIp = "10.0.0.0";
     const mask = "255.255.255.255";
     this.ipGenerator = new IpAddressGenerator(baseIp, mask);
     loadFromLocalStorage(this);

src/types/graphs/datagraph.ts

@@ -19,7 +19,7 @@ interface RouterGraphNode extends CommonGraphNode {
 export interface RoutingTableEntry {
   ip: string;
   mask: string;
-  iface: number;
+  iface: DeviceId;
 }
 
 // Typescript type guard
@@ -181,12 +181,13 @@ export class DataGraph {
       `Connection created between devices ID: ${n1Id} and ID: ${n2Id}`,
     );
     this.notifyChanges();
+    this.regenerateAllRoutingTables();
   }
 
   updateDevicePosition(id: DeviceId, newValues: { x?: number; y?: number }) {
     const deviceGraphNode = this.devices.get(id);
     if (!deviceGraphNode) {
-      console.warn("Device’s id is not registered");
+      console.warn("Device's id is not registered");
       return;
     }
     this.devices.set(id, { ...deviceGraphNode, ...newValues });
@@ -237,6 +238,7 @@ export class DataGraph {
     this.devices.delete(id);
     console.log(`Device with ID ${id} and its connections were removed.`);
     this.notifyChanges();
+    this.regenerateAllRoutingTables();
   }
 
   // Method to remove a connection (edge) between two devices by their IDs
@@ -270,6 +272,7 @@ export class DataGraph {
       `Connection removed between devices ID: ${n1Id} and ID: ${n2Id}`,
     );
     this.notifyChanges();
+    this.regenerateAllRoutingTables();
   }
 
   subscribeChanges(callback: () => void) {
@@ -279,4 +282,56 @@ export class DataGraph {
   notifyChanges() {
     this.onChanges.forEach((callback) => callback());
   }
+
+  regenerateAllRoutingTables() {
+    console.log("Regenerating all routing tables");
+    this.devices.forEach((_, id) => this.regenerateRoutingTable(id));
+  }
+
+  regenerateRoutingTable(id: DeviceId) {
+    const router = this.devices.get(id);
+    if (!isRouter(router)) {
+      return;
+    }
+    console.log(`Regenerating routing table for ID ${id}`);
+    const parents = new Map<DeviceId, DeviceId>();
+    parents.set(id, id);
+    const queue = Array.from([id]);
+    while (queue.length > 0) {
+      const currentId = queue.shift();
+      const current = this.devices.get(currentId);
+      if (!isRouter(current)) {
+        // Don't route packets on hosts
+        continue;
+      }
+      current.connections.forEach((connectedId) => {
+        if (!parents.has(connectedId)) {
+          parents.set(connectedId, currentId);
+          queue.push(connectedId);
+        }
+      });
+    }
+
+    console.log(parents);
+
+    const table: RoutingTableEntry[] = [];
+    parents.forEach((currentId, childId) => {
+      const dstId = childId;
+      if (dstId === id) {
+        return;
+      }
+
+      while (currentId !== id) {
+        const parentId = parents.get(currentId);
+        childId = currentId;
+        currentId = parentId;
+      }
+      // Here the currentId is the router, and the childId
+      // is the first step towards dstId
+      const dst = this.devices.get(dstId);
+      const entry = { ip: dst.ip, mask: dst.mask, iface: childId };
+      table.push(entry);
+    });
+    router.routingTable = table;
+  }
 }