feat: add impl for node sockets (wip)

packages/preview2-shim/lib/common/assert.js

@@ -0,0 +1,7 @@
+export function assert(condition, tag, _val) {
+  if (condition) {
+    // TODO: throw meaningful errors
+    // NOTE: wasmtime conformance tests are expecting a string here (a tag)
+    throw tag;
+  }
+}

packages/preview2-shim/lib/io/calls.js

@@ -45,3 +45,8 @@ export const HTTP_CREATE_REQUEST = ++call_id << 24;
 export const CLOCKS_NOW = ++call_id << 24;
 export const CLOCKS_DURATION_SUBSCRIBE = ++call_id << 24;
 export const CLOCKS_INSTANT_SUBSCRIBE = ++call_id << 24;
+
+// Sockets
+export const SOCKET_RESOLVE_ADDRESS_CREATE_REQUEST = ++call_id << 24;
+export const SOCKET_RESOLVE_ADDRESS_GET_AND_DISPOSE_REQUEST = ++call_id << 24;
+export const SOCKET_RESOLVE_ADDRESS_DISPOSE_REQUEST = ++call_id << 24;

packages/preview2-shim/lib/io/stream-types.js

@@ -4,5 +4,6 @@ export const STDIN = ++cnt;
 export const STDOUT = ++cnt;
 export const STDERR = ++cnt;
 export const FILE = ++cnt;
+export const SOCKET = ++cnt;
 export const INCOMING_BODY = ++cnt;
 export const OUTGOING_BODY = ++cnt;

packages/preview2-shim/lib/io/worker-thread.js

@@ -1,17 +1,17 @@
-import { runAsWorker } from "../synckit/index.js";
-import { FILE, STDOUT, STDERR, STDIN } from "./stream-types.js";
+import { resolve } from "node:dns/promises";
 import { createReadStream, createWriteStream } from "node:fs";
-import { Readable } from "node:stream";
 import { hrtime } from "node:process";
+import { Readable } from "node:stream";
+import { runAsWorker } from "../synckit/index.js";
 import {
   CALL_MASK,
   CALL_SHIFT,
   CALL_TYPE_MASK,
   CLOCKS_DURATION_SUBSCRIBE,
   CLOCKS_INSTANT_SUBSCRIBE,
   CLOCKS_NOW,
-  FUTURE_DISPOSE_AND_GET_VALUE,
   FUTURE_DISPOSE,
+  FUTURE_DISPOSE_AND_GET_VALUE,
   HTTP_CREATE_REQUEST,
   INPUT_STREAM_BLOCKING_READ,
   INPUT_STREAM_BLOCKING_SKIP,
@@ -30,12 +30,16 @@ import {
   OUTPUT_STREAM_FLUSH,
   OUTPUT_STREAM_SPLICE,
   OUTPUT_STREAM_SUBSCRIBE,
-  OUTPUT_STREAM_WRITE_ZEROES,
   OUTPUT_STREAM_WRITE,
-  POLL_POLL_LIST,
+  OUTPUT_STREAM_WRITE_ZEROES,
   POLL_POLLABLE_BLOCK,
   POLL_POLLABLE_READY,
+  POLL_POLL_LIST,
+  SOCKET_RESOLVE_ADDRESS_CREATE_REQUEST,
+  SOCKET_RESOLVE_ADDRESS_DISPOSE_REQUEST,
+  SOCKET_RESOLVE_ADDRESS_GET_AND_DISPOSE_REQUEST,
 } from "./calls.js";
+import { FILE, SOCKET, STDERR, STDIN, STDOUT } from "./stream-types.js";
 
 let streamCnt = 0,
   pollCnt = 0;
@@ -120,6 +124,30 @@ function handle(call, id, payload) {
       return createFuture(createHttpRequest(method, url, headers, body));
     }
 
+    // Sockets
+    case SOCKET_RESOLVE_ADDRESS_CREATE_REQUEST:
+      return createFuture(resolve(payload.hostname));
+    case SOCKET_RESOLVE_ADDRESS_DISPOSE_REQUEST:
+      return void unfinishedFutures.delete(id);
+    case SOCKET_RESOLVE_ADDRESS_GET_AND_DISPOSE_REQUEST: {
+      const future = unfinishedFutures.get(id);
+      if (!future) {
+        // future not ready yet
+        if (unfinishedPolls.get(id)) {
+          throw 'would-block';
+        }
+        throw new Error("future already got and dropped");
+      }
+      unfinishedFutures.delete(id);
+      return future;
+    }
+    case OUTPUT_STREAM_CREATE | SOCKET: {
+      // TODO: implement
+    }
+    case INPUT_STREAM_CREATE | SOCKET: {
+      // TODO: implement
+    }
+
     // Stdio
     case OUTPUT_STREAM_DISPOSE | STDOUT:
     case OUTPUT_STREAM_DISPOSE | STDERR:
@@ -434,14 +462,16 @@ function handle(call, id, payload) {
 }
 
 // poll promises must always resolve and never error
+// once the future is resolved, it is removed from unfinishedPolls
 function createPoll(promise) {
   const pollId = ++pollCnt;
   unfinishedPolls.set(
     pollId,
     promise.then(
       () => void unfinishedPolls.delete(pollId),
-      () => {
+      (err) => {
         process._rawDebug("Unexpected poll error");
+        process._rawDebug(err);
         process.exit(1);
       }
     )

packages/preview2-shim/lib/nodejs/index.js

@@ -15,3 +15,6 @@ export {
   sockets,
   cli
 }
+
+export { WasiSockets } from "./sockets/wasi-sockets.js";
+

packages/preview2-shim/lib/nodejs/sockets.js

@@ -1,200 +1,13 @@
-export const instanceNetwork = {
-  instanceNetwork () {
-    console.log(`[sockets] instance network`);
-  }
-};
-
-export const ipNameLookup = {
-  dropResolveAddressStream () {
-
-  },
-  subscribe () {
-
-  },
-  resolveAddresses () {
-
-  },
-  resolveNextAddress () {
-
-  },
-  nonBlocking () {
-
-  },
-  setNonBlocking () {
-
-  },
-};
-
-export const network = {
-  dropNetwork () {
-
-  }
-};
-
-export const tcpCreateSocket = {
-  createTcpSocket () {
-
-  }
-};
-
-export const tcp = {
-  subscribe () {
-
-  },
-  dropTcpSocket() {
-
-  },
-  bind() {
-
-  },
-  connect() {
-
-  },
-  listen() {
-
-  },
-  accept() {
-
-  },
-  localAddress() {
-
-  },
-  remoteAddress() {
-
-  },
-  addressFamily() {
-
-  },
-  ipv6Only() {
-
-  },
-  setIpv6Only() {
-
-  },
-  setListenBacklogSize() {
-
-  },
-  keepAlive() {
-
-  },
-  setKeepAlive() {
-
-  },
-  noDelay() {
-
-  },
-  setNoDelay() {
-
-  },
-  unicastHopLimit() {
-
-  },
-  setUnicastHopLimit() {
-
-  },
-  receiveBufferSize() {
-
-  },
-  setReceiveBufferSize() {
-
-  },
-  sendBufferSize() {
-
-  },
-  setSendBufferSize() {
-
-  },
-  nonBlocking() {
-
-  },
-  setNonBlocking() {
-
-  },
-  shutdown() {
-
-  }
-};
-
-export const udp = {
-  subscribe () {
-
-  },
-
-  dropUdpSocket () {
-
-  },
-
-  bind () {
-
-  },
-
-  connect () {
-
-  },
-
-  receive () {
-
-  },
-
-  send () {
-
-  },
-
-  localAddress () {
-
-  },
-
-  remoteAddress () {
-
-  },
-
-  addressFamily () {
-
-  },
-
-  ipv6Only () {
-
-  },
-
-  setIpv6Only () {
-
-  },
-
-  unicastHopLimit () {
-
-  },
-
-  setUnicastHopLimit () {
-
-  },
-
-  receiveBufferSize () {
-
-  },
-
-  setReceiveBufferSize () {
-
-  },
-
-  sendBufferSize () {
-
-  },
-
-  setSendBufferSize () {
-
-  },
-
-  nonBlocking () {
-
-  },
-
-  setNonBlocking () {
-
-  }
-};
-
-export const udpCreateSocket = {
-  createTcpSocket () {
-
-  }
-};
+/* eslint-disable no-unused-vars */
+
+import { WasiSockets } from "./sockets/wasi-sockets.js";
+
+export const { 
+    ipNameLookup,
+    instanceNetwork, 
+    network, 
+    tcpCreateSocket,
+    udpCreateSocket,
+    tcp,
+    udp,
+} = new WasiSockets();;