added pow

electron/src/lite-node/index.ts

@@ -2,6 +2,11 @@ import net from 'net';
 import crypto from 'crypto';
 import { ed25519, x25519 } from '@noble/curves/ed25519';
 import { sha512 } from '@noble/hashes/sha512';
+import util from 'util';
+import fs from 'fs';
+import path from 'path';
+const readFile = util.promisify(fs.readFile); // Promisify readFile for use with async/await
+// Adjust path if needed to your wasm file
 
 export const SEED_PEERS = ['127.0.0.1'];
 
@@ -11,6 +16,121 @@ export enum MessageType {
   RESPONSE = 3,
 }
 
+async function loadWebAssembly(memory) {
+  const importObject = {
+    env: {
+      memory: memory, // Pass the WebAssembly.Memory object to the module
+    },
+  };
+
+  // Correct the path to point to the specific location of the .wasm file
+  const filename = path.join(__dirname, './memory-pow.wasm.full');
+
+  try {
+    // Read the .wasm file from the filesystem
+    const buffer = await readFile(filename);
+    const module = await WebAssembly.compile(buffer);
+
+    // Create the WebAssembly instance with the compiled module and import object
+    const instance = new WebAssembly.Instance(module, importObject);
+
+    return instance; // Return the instance to be used elsewhere in your application
+  } catch (error) {
+    console.error('Error loading WebAssembly module:', error);
+    throw error; // Rethrow the error for further handling
+  }
+}
+const memory = new WebAssembly.Memory({ initial: 256, maximum: 256 });
+
+const heap = new Uint8Array(memory.buffer);
+
+const initialBrk = 512 * 1024;
+let brk = 512 * 1024; // Initialize brk outside to maintain state
+
+const waitingQueue = [];
+
+function sbrk(size) {
+  const oldBrk = brk;
+  if (brk + size > heap.length) {
+    // Check if the heap can accommodate the request
+    console.log('Not enough memory available, adding to waiting queue');
+    return null; // Not enough memory, return null
+  }
+  brk += size; // Advance brk by the size of the requested memory
+  return oldBrk; // Return the old break point (start of the newly allocated block)
+}
+
+function processWaitingQueue() {
+  console.log('Processing waiting queue...');
+  let i = 0;
+  while (i < waitingQueue.length) {
+    const request = waitingQueue[i];
+    const ptr = sbrk(request.size);
+    if (ptr !== null) {
+      // Check if memory was successfully allocated
+      request.resolve(ptr);
+      waitingQueue.splice(i, 1); // Remove the processed request
+    } else {
+      i++; // Continue if the current request cannot be processed
+    }
+  }
+}
+
+function requestMemory(size) {
+  return new Promise((resolve, reject) => {
+    const ptr = sbrk(size);
+    if (ptr !== null) {
+      resolve(ptr);
+    } else {
+      waitingQueue.push({ size, resolve, reject }); // Add to queue if not enough memory
+    }
+  });
+}
+
+interface WasmModule {
+  exports: WasmExports;
+}
+
+interface WasmExports {
+  compute2: (
+    hashPtr: number,
+    workBufferPtr: number,
+    workBufferLength: number,
+    difficulty: number
+  ) => number;
+}
+
+let response: number;
+
+const computePow = async (
+  memory,
+  hashPtr,
+  workBufferPtr,
+  workBufferLength,
+  difficulty
+) => {
+  let response = null;
+
+  await new Promise<void>((resolve) => {
+    loadWebAssembly(memory).then((wasmModule: any) => {
+      response = wasmModule.exports.compute2(
+        hashPtr,
+        workBufferPtr,
+        workBufferLength,
+        difficulty
+      );
+      resolve();
+    });
+  });
+
+  return response;
+};
+
+function resetMemory() {
+  brk = initialBrk; // Reset the break point
+  processWaitingQueue(); // Try to process any waiting memory requests
+}
+
 function parseMessage(buffer: Buffer) {
   if (buffer.length < 17) return null;
   if (buffer.subarray(0, 4).toString('ascii') !== 'QORT') return null;
@@ -89,10 +209,10 @@ export class LiteNodeClient {
   private socket: net.Socket | null = null;
   private buffer = Buffer.alloc(0);
 
-  private edPrivateKey: Uint8Array;
+  private edPrivateKey!: Uint8Array;
-  private edPublicKey: Uint8Array;
+  private edPublicKey!: Uint8Array;
-  private xPrivateKey: Uint8Array;
+  private xPrivateKey!: Uint8Array;
-  private xPublicKey: Uint8Array;
+  private xPublicKey!: Uint8Array;
 
   private theirEdPublicKey: Uint8Array | null = null;
   private theirXPublicKey: Uint8Array | null = null;
@@ -102,7 +222,10 @@ export class LiteNodeClient {
   constructor(
     private host: string,
     private port: number = 12392
-  ) {
+  ) {}
+
+  async init() {
+    // Generate keys
     const edSeed = ed25519.utils.randomPrivateKey(); // 32-byte seed
     const edPublicKey = ed25519.getPublicKey(edSeed);
 
@@ -117,7 +240,7 @@ export class LiteNodeClient {
     this.xPublicKey = x25519.getPublicKey(this.xPrivateKey);
   }
 
-  private handleChallenge(payload: Buffer) {
+  private async handleChallenge(payload: Buffer) {
     this.theirEdPublicKey = payload.subarray(0, 32);
     this.theirXPublicKey = ed25519ToX25519Public(this.theirEdPublicKey);
     this.theirChallenge = payload.subarray(32, 64);
@@ -126,25 +249,57 @@ export class LiteNodeClient {
       this.xPrivateKey,
       this.theirXPublicKey
     );
+
     const combined = Buffer.concat([
       Buffer.from(sharedSecret),
       this.theirChallenge,
     ]);
     const responseHash = crypto.createHash('sha256').update(combined).digest();
 
+    const hashPtr = sbrk(32);
+    const hashAry = new Uint8Array(memory.buffer, hashPtr, 32);
+    hashAry.set(responseHash);
+
+    // Use WASM for PoW
+    const powDifficulty = 2; // Difficulty bits
+    const workBufferLength = 2 * 1024 * 1024;
+    // const nonceValue = this.computePoWNonceWasm(
+    //   responseHash,
+    //   powBufferSize,
+    //   powDifficulty
+    // );
+    const workBufferPtr = await requestMemory(workBufferLength);
+
+    const nonceValue = await computePow(
+      memory,
+      hashPtr,
+      workBufferPtr,
+      workBufferLength,
+      powDifficulty
+    );
+    brk = initialBrk;
     const nonce = Buffer.alloc(4);
-    nonce.writeUInt32BE(0);
+    nonce.writeUInt32BE(nonceValue);
+
     const responsePayload = Buffer.concat([nonce, responseHash]);
 
+    console.log('🔨 PoW nonce computed:', nonceValue);
     console.log(
       '🔐 Shared secret (hex):',
       Buffer.from(sharedSecret).toString('hex')
     );
 
     this.sendMessage(MessageType.RESPONSE, responsePayload);
+    resetMemory();
   }
 
-  connect(): Promise<void> {
+  private async handleResponse(payload: Buffer) {
+    console.log('payload', payload);
+  }
+
+  async connect(): Promise<void> {
+    await this.init();
+
     return new Promise((resolve, reject) => {
       this.socket = net.createConnection(
         { host: this.host, port: this.port },
@@ -176,6 +331,9 @@ export class LiteNodeClient {
             case MessageType.CHALLENGE:
               this.handleChallenge(payload);
               break;
+            case MessageType.RESPONSE:
+              this.handleResponse(payload);
+              break;
 
             default:
               console.warn(`⚠️ Unhandled message type: ${messageType}`);

electron/tsconfig.json

@@ -4,7 +4,8 @@
   "compilerOptions": {
     "outDir": "./build",
     "importHelpers": true,
-    "target": "ES2017",
+    "target": "ES2020",
+    "lib": ["ES2020", "DOM"],
     "module": "CommonJS",
     "moduleResolution": "node",
     "esModuleInterop": true,