@bytecodealliance/preview3-shim 0.1.2 → 0.2.0

package/lib/nodejs/sockets/tcp.js

@@ -1,801 +0,0 @@
-import { StreamReader, readableByteStreamFromReader } from "../stream.js";
-import { FutureReader } from "../future.js";
-import { ResourceWorker } from "../workers/resource-worker.js";
-import { SocketError } from "./error.js";
-
-import {
-  isWildcardIpAddress,
-  isUnicastIpAddress,
-  isMulticastIpAddress,
-  isIpv4MappedAddress,
-  isBindableIpAddress,
-  IP_ADDRESS_FAMILY,
-} from "./address.js";
-
-let WORKER = null;
-const symbolDispose = Symbol.dispose || Symbol.for("dispose");
-function worker() {
-  return (WORKER ??= new ResourceWorker(new URL("../workers/tcp-worker.js", import.meta.url)));
-}
-
-let TCP_CREATE_TOKEN = null;
-function token() {
-  return (TCP_CREATE_TOKEN ??= Symbol("TcpCreateToken"));
-}
-
-function invalidRecv() {
-  const stream = new ReadableStream({
-    start(controller) {
-      controller.close();
-    },
-  });
-  const promise = Promise.reject(new SocketError("invalid-state"));
-  return [new StreamReader(stream, { preventCancel: false }), new FutureReader(promise)];
-}
-
-const STATE = {
-  UNBOUND: "unbound",
-  BOUND: "bound",
-  LISTENING: "listening",
-  CONNECTING: "connecting",
-  CONNECTED: "connected",
-  CLOSED: "closed",
-};
-
-/**
- * Creates a new TCP socket
- * WIT: createTcpSocket: func(address-family: ip-address-family) -> result<tcp-socket, error-code>
- *
- * @param {IP_ADDRESS_FAMILY} addressFamily - The IP address family (IPv4 or IPv6)
- * @returns {TcpSocket>} A new TCP socket instance
- * @throws {SocketError} with payload.tag 'invalid-argument' if `addressFamily` is not IPV4 or IPV6
- * @throws {SocketError} for other errors, payload.tag maps the low-level system error_
- */
-export function createTcpSocket(addressFamily) {
-  if (addressFamily !== IP_ADDRESS_FAMILY.IPV4 && addressFamily !== IP_ADDRESS_FAMILY.IPV6) {
-    throw new SocketError("invalid-argument");
-  }
-
-  try {
-    const result = worker().runSync({
-      op: "tcp-create",
-      family: addressFamily,
-    });
-
-    return TcpSocket._create(token(), addressFamily, result.socketId);
-  } catch (error) {
-    throw SocketError.from(error);
-  }
-}
-
-export class TcpSocket {
-  #socketId = null;
-  #family = null;
-  #state = "unbound";
-  #sendStarted = false;
-  #receiveStarted = false;
-  #options = {
-    // defaults per https://nodejs.org/docs/latest/api/net.html#socketsetkeepaliveenable-initialdelay
-    keepAliveEnabled: false,
-
-    // Node.js doesn't give us the ability to detect the OS default,
-    // therefore we hardcode the default value instead of using the OS default,
-    // since we would never be able to report it as a return value otherwise.
-    // We could make this configurable as a global JCO implementation configuration
-    // instead.
-    keepAliveIdleTime: 7200_000_000_000n,
-
-    // The following options are NOT configurable in Node.js!
-    // Any configurations set will respond correctly, but underneath retain
-    // system / Node.js defaults.
-    keepAliveInterval: 1_000_000_000n,
-    keepAliveCount: 10,
-    hopLimit: 1,
-    receiveBufferSize: 65_536n,
-    sendBufferSize: 65_536n,
-  };
-
-  /**
-   * WIT static constructor for `wasi:sockets/types.tcp-socket`.
-   *
-   * WIT:
-   * ```
-   * create: static func(address-family: ip-address-family) -> result<tcp-socket, error-code>
-   * ```
-   *
-   * @static
-   * @param {IP_ADDRESS_FAMILY} addressFamily - The IP address family (IPv4 or IPv6)
-   * @returns {TcpSocket} A new TcpSocket instance
-   * @throws {SocketError} with payload.tag 'invalid-argument' if `addressFamily` is not IPV4 or IPV6
-   * @throws {SocketError} for other socket creation errors
-   */
-  static create(addressFamily) {
-    return createTcpSocket(addressFamily);
-  }
-
-  /**
-   * Creates a new TCP socket
-   * WIT:
-   * ```
-   * constructor(address-family: ip-address-family)
-   * ```
-   *
-   * @static
-   * @param {IP_ADDRESS_FAMILY} addressFamily - The IP address family (IPv4 or IPv6)
-   * @param {number} socketId - The internal socket identifier
-   * @returns {TcpSocket} A new TcpSocket instance
-   * @private
-   */
-  static _create(t, addressFamily, socketId) {
-    if (t !== token()) {
-      throw new Error("Use createTcpSocket to create a Socket");
-    }
-
-    if (addressFamily !== IP_ADDRESS_FAMILY.IPV4 && addressFamily !== IP_ADDRESS_FAMILY.IPV6) {
-      throw new SocketError("invalid-argument");
-    }
-
-    const socket = new TcpSocket();
-    socket.#family = addressFamily;
-    socket.#socketId = socketId;
-    return socket;
-  }
-
-  /**
-   * Binds the socket to a local address
-   * WIT:
-   * ```
-   * bind: func(local-address: ip-socket-address) -> result<_, error-code>
-   * ```
-   *
-   * @param {Object} localAddress - The local address to bind to
-   * @returns {void}
-   * @throws {SocketError} with payload.tag 'invalid-state' if the socket is not in ~STATE.UNBOUND~
-   * @throws {SocketError} with payload.tag 'invalid-argument' if ~localAddress~ is invalid
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  bind(localAddress) {
-    if (this.#state !== STATE.UNBOUND) {
-      throw new SocketError("invalid-state");
-    }
-
-    if (!this.#isValidLocalAddress(localAddress)) {
-      throw new SocketError("invalid-argument");
-    }
-    if (!isBindableIpAddress(localAddress)) {
-      throw new SocketError("address-not-bindable");
-    }
-    try {
-      worker().runSync({
-        op: "tcp-bind",
-        localAddress,
-        socketId: this.#socketId,
-      });
-
-      this.#state = STATE.BOUND;
-    } catch (error) {
-      throw SocketError.from(error);
-    }
-  }
-
-  /**
-   * Connects to a remote endpoint
-   * WIT:
-   * ```
-   * connect: async func(remote-address: ip-socket-address) -> result<_, error-code>
-   * ```
-   *
-   * @async
-   * @param {Object} remoteAddress - The remote address to connect to
-   * @returns {Promise<void>}
-   * @throws {SocketError} with payload.tag 'invalid-state' if socket is CONNECTING, CONNECTED, or LISTENING
-   * @throws {SocketError} with payload.tag 'invalid-argument' if ~remoteAddress~ is invalid or port == 0
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  async connect(remoteAddress) {
-    if (
-      this.#state === STATE.CONNECTING ||
-      this.#state === STATE.CONNECTED ||
-      this.#state === STATE.LISTENING
-    ) {
-      throw new SocketError("invalid-state");
-    }
-
-    if (!this.#isValidRemoteAddress(remoteAddress)) {
-      throw new SocketError("invalid-argument");
-    }
-
-    this.#state = STATE.CONNECTING;
-
-    try {
-      await worker().run({
-        op: "tcp-connect",
-        remoteAddress,
-        socketId: this.#socketId,
-      });
-
-      this.#state = STATE.CONNECTED;
-    } catch (error) {
-      this.#state = STATE.CLOSED;
-      throw SocketError.from(error);
-    }
-  }
-
-  /**
-   * Starts listening for connections
-   * WIT:
-   * ```
-   * listen: func() -> result<stream<tcp-socket>, error-code>
-   * ```
-   *
-   * @returns {StreamReader} A stream of incoming TCP socket connections
-   * @throws {SocketError} with payload.tag 'invalid-state' if socket is LISTENING or CONNECTED
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  listen() {
-    if (this.#state === STATE.LISTENING || this.#state === STATE.CONNECTED) {
-      throw new SocketError("invalid-state");
-    }
-    try {
-      // Convert incoming connections from worker to TcpSocket instances
-      const listener = this;
-      const transform = new TransformStream({
-        transform({ family, socketId }, controller) {
-          const socket = TcpSocket._create(token(), family, socketId);
-          socket.#state = STATE.CONNECTED;
-          socket.#options = { ...listener.#options };
-          controller.enqueue(socket);
-        },
-      });
-
-      worker().runSync(
-        {
-          op: "tcp-listen",
-          socketId: this.#socketId,
-          stream: transform.writable,
-        },
-        [transform.writable],
-      );
-
-      this.#state = STATE.LISTENING;
-      return new StreamReader(transform.readable);
-    } catch (error) {
-      this.#state = STATE.CLOSED;
-      throw SocketError.from(error);
-    }
-  }
-
-  /**
-   * Sends data to the connected peer
-   * WIT:
-   * ```
-   * send: async func(data: stream<u8>) -> result<_, error-code>
-   * ```
-   *
-   * @async
-   * @param {object} data - The byte stream to send.
-   * @returns {Promise<void>}
-   * @throws {SocketError} With payload.tag 'invalid-state' if socket is not CONNECTED
-   * @throws {SocketError} With payload.tag 'invalid-argument' if `data` is not a readable byte stream
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  async send(data) {
-    if (this.#state !== STATE.CONNECTED) {
-      throw new SocketError("invalid-state");
-    }
-    if (this.#sendStarted) {
-      data?.[symbolDispose]?.();
-      throw new SocketError("invalid-state");
-    }
-    let stream;
-    try {
-      stream = readableByteStreamFromReader(data, { name: "tcp send data" });
-    } catch (error) {
-      if (error instanceof TypeError) {
-        throw new SocketError("invalid-argument", error.message, undefined, { cause: error });
-      }
-      throw error;
-    }
-    this.#sendStarted = true;
-
-    try {
-      // Transfer the stream to the worker
-      await worker().run(
-        {
-          op: "tcp-send",
-          socketId: this.#socketId,
-          stream,
-        },
-        [stream],
-      );
-    } catch (error) {
-      throw SocketError.from(error);
-    }
-  }
-
-  /**
-   * Receives data from the connected peer
-   * WIT:
-   * ```
-   * receive: func() -> tuple<stream<u8>, future<result<_, error-code>>>
-   * ```
-   *
-   * @throws {SocketError} with payload.tag 'invalid-state' if socket is not CONNECTED
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   * @returns {{ stream: StreamReader, future: FutureReader }}
-   */
-  receive() {
-    if (this.#state !== STATE.CONNECTED) {
-      return invalidRecv();
-    }
-    if (this.#receiveStarted) {
-      return invalidRecv();
-    }
-    this.#receiveStarted = true;
-
-    const transform = new TransformStream();
-    const promise = worker()
-      .run(
-        {
-          op: "tcp-receive",
-          socketId: this.#socketId,
-          stream: transform.writable,
-        },
-        [transform.writable],
-      )
-      .catch((err) => {
-        throw SocketError.from(err);
-      });
-
-    return [
-      new StreamReader(transform.readable, { preventCancel: false }),
-      new FutureReader(promise),
-    ];
-  }
-
-  /**
-   * Gets the bound local address
-   * WIT:
-   * ```
-   * local-address: func() -> result<ip-socket-address, error-code>
-   * ```
-   *
-   * @returns {Object} The local socket address
-   * @throws {SocketError} with payload.tag 'invalid-state' if socket is not BOUND
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  getLocalAddress() {
-    if (this.#state === STATE.UNBOUND) {
-      throw new SocketError("invalid-state");
-    }
-
-    try {
-      return worker().runSync({
-        op: "tcp-get-local-address",
-        socketId: this.#socketId,
-      });
-    } catch (err) {
-      throw SocketError.from(err);
-    }
-  }
-
-  /**
-   * Gets the remote address
-   * WIT:
-   * ```
-   * remote-address: func() -> result<ip-socket-address, error-code>
-   * ```
-   *
-   * @returns {Object} The remote socket address
-   * @throws {SocketError} with payload.tag 'invalid-state' if socket is not CONNECTED
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  getRemoteAddress() {
-    if (this.#state !== STATE.CONNECTED) {
-      throw new SocketError("invalid-state");
-    }
-
-    try {
-      return worker().runSync({
-        op: "tcp-get-remote-address",
-        socketId: this.#socketId,
-      });
-    } catch (err) {
-      throw SocketError.from(err);
-    }
-  }
-
-  /**
-   * Checks if the socket is in listening state
-   * WIT:
-   * ```
-   * is-listening: func() -> bool
-   * ```
-   *
-   * @returns {boolean} True if socket is listening, false otherwise
-   */
-  getIsListening() {
-    return this.#state === STATE.LISTENING;
-  }
-
-  /**
-   * Gets the address family
-   * WIT:
-   * ```
-   * address-family: func() -> ip-address-family
-   * ```
-   *
-   * @returns {IP_ADDRESS_FAMILY} The IP address family (IPv4 or IPv6)
-   */
-  getAddressFamily() {
-    return this.#family;
-  }
-
-  /**
-   * Sets the listen backlog size
-   * WIT:
-   * ```
-   * set-listen-backlog-size: func(value: u64) -> result<_, error-code>
-   * ```
-   *
-   * @param {bigint} value - The backlog size
-   * @returns {void}
-   * @throws {SocketError} with payload.tag 'invalid-argument' if `value` is 0
-   * @throws {SocketError} with payload.tag 'not-supported' if socket is CONNECTING or CONNECTED
-   * @throws {SocketError} with payload.tag 'invalid-state' if socket is not UNBOUND or BOUND
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  setListenBacklogSize(value) {
-    if (value <= 0n) {
-      throw new SocketError("invalid-argument");
-    }
-
-    if (
-      this.#state === STATE.CONNECTING ||
-      this.#state === STATE.CONNECTED ||
-      this.#state === STATE.LISTENING
-    ) {
-      throw new SocketError("not-supported");
-    }
-
-    if (this.#state !== STATE.UNBOUND && this.#state !== STATE.BOUND) {
-      throw new SocketError("invalid-state");
-    }
-
-    this.#options.listenBacklogSize = value;
-    try {
-      worker().runSync({
-        op: "tcp-set-listen-backlog-size",
-        socketId: this.#socketId,
-        value,
-      });
-    } catch (err) {
-      throw SocketError.from(err);
-    }
-  }
-
-  /**
-   * Gets whether keep-alive is enabled
-   * WIT:
-   * ```
-   * keep-alive-enabled: func() -> result<bool, error-code>
-   * ```
-   *
-   * @returns {boolean} True if keep-alive is enabled, false otherwise
-   */
-  getKeepAliveEnabled() {
-    return this.#options.keepAliveEnabled;
-  }
-
-  /**
-   * Sets keep-alive enabled state
-   * WIT:
-   * ```
-   * set-keep-alive-enabled: func(value: bool) -> result<_, error-code>
-   * ```
-   *
-   * @async
-   * @param {boolean} value - Whether to enable keep-alive
-   * @returns {void}
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  setKeepAliveEnabled(value) {
-    try {
-      this.#options.keepAliveEnabled = value;
-
-      worker().runSync({
-        op: "tcp-set-keep-alive",
-        socketId: this.#socketId,
-        keepAliveEnabled: this.#options.keepAliveEnabled,
-        keepAliveIdleTime: this.#options.keepAliveIdleTime,
-      });
-    } catch (err) {
-      throw SocketError.from(err);
-    }
-  }
-
-  /**
-   * Gets the keep-alive idle time
-   * WIT:
-   * ```
-   * keep-alive-idle-time: func() -> result<duration, error-code>
-   * ```
-   *
-   * @returns {bigint} The keep-alive idle time in nanoseconds
-   */
-  getKeepAliveIdleTime() {
-    return this.#options.keepAliveIdleTime;
-  }
-
-  /**
-   * Sets the keep-alive idle time
-   * WIT:
-   * ```
-   * set-keep-alive-idle-time: func(value: duration) -> result<_, error-code>
-   * ```
-   *
-   * @param {bigint} value - The idle time in nanoseconds
-   * @returns {void}
-   * @throws {SocketError} with payload.tag 'invalid-argument' if `value` is less than 1
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  setKeepAliveIdleTime(value) {
-    if (value < 1n) {
-      throw new SocketError("invalid-argument");
-    }
-
-    // Ensure that the value passed to the worker never gets rounded down to 0.
-    if (value < 1_000_000_000n) {
-      value = 1_000_000_000n;
-    }
-
-    if (value === this.#options.keepAliveIdleTime || !this.#options.keepAliveEnabled) {
-      this.#options.keepAliveIdleTime = value;
-      return;
-    }
-
-    this.#options.keepAliveIdleTime = value;
-
-    try {
-      worker().runSync({
-        op: "tcp-set-keep-alive",
-        socketId: this.#socketId,
-        keepAliveEnabled: this.#options.keepAliveEnabled,
-        keepAliveIdleTime: this.#options.keepAliveIdleTime,
-      });
-    } catch (err) {
-      throw SocketError.from(err);
-    }
-  }
-
-  /**
-   * Gets the keep-alive interval
-   * WIT:
-   * ```
-   * keep-alive-interval: func() -> result<duration, error-code>
-   * ```
-   *
-   * @returns {bigint} The keep-alive interval in nanoseconds
-   */
-  getKeepAliveInterval() {
-    return this.#options.keepAliveInterval;
-  }
-
-  /**
-   * Sets the keep-alive interval
-   * WIT:
-   * ```
-   * set-keep-alive-interval: func(value: duration) -> result<_, error-code>
-   * ```
-   *
-   * @param {bigint} value - The interval in nanoseconds
-   * @returns {void}
-   * @throws {SocketError} with payload.tag 'invalid-argument' if `value` is less than 1
-   */
-  setKeepAliveInterval(value) {
-    if (value < 1n) {
-      throw new SocketError("invalid-argument");
-    }
-    this.#options.keepAliveInterval = value;
-  }
-
-  /**
-   * Gets the keep-alive count
-   * WIT:
-   * ```
-   * keep-alive-count: func() -> result<u32, error-code>
-   * ```
-   *
-   * @returns {number} The keep-alive count
-   */
-  getKeepAliveCount() {
-    return this.#options.keepAliveCount;
-  }
-
-  /**
-   * Sets the keep-alive count
-   * WIT:
-   * ```
-   * set-keep-alive-count: func(value: u32) -> result<_, error-code>
-   * ```
-   *
-   * @param {number} value - The keep-alive count
-   * @returns {void}
-   * @throws {SocketError} with payload.tag 'invalid-argument' if `value` is less than 1
-   */
-  setKeepAliveCount(value) {
-    if (value < 1) {
-      throw new SocketError("invalid-argument");
-    }
-    this.#options.keepAliveCount = value;
-  }
-
-  /**
-   * Gets the hop limit
-   * WIT:
-   * ```
-   * hop-limit: func() -> result<u8, error-code>
-   * ```
-   *
-   * @returns {number} The hop limit
-   */
-  getHopLimit() {
-    return this.#options.hopLimit;
-  }
-
-  /**
-   * Sets the hop limit
-   * WIT:
-   * ```
-   * set-hop-limit: func(value: u8) -> result<_, error-code>
-   * ```
-   *
-   * @param {number} value - The hop limit
-   * @returns {void}
-   * @throws {SocketError} with payload.tag 'invalid-argument' if `value` is less than 1
-   */
-  setHopLimit(value) {
-    if (value < 1) {
-      throw new SocketError("invalid-argument");
-    }
-    this.#options.hopLimit = value;
-  }
-
-  /**
-   * Gets the receive buffer size
-   * WIT:
-   * ```
-   * receive-buffer-size: func() -> result<u64, error-code>
-   * ```
-   *
-   * @returns {bigint} The receive buffer size in bytes
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  getReceiveBufferSize() {
-    if (this.#options.receiveBufferSize) {
-      return this.#options.receiveBufferSize;
-    }
-
-    try {
-      const result = worker().runSync({
-        op: "tcp-recv-buffer-size",
-        socketId: this.#socketId,
-      });
-
-      this.#options.receiveBufferSize = result;
-      return result;
-    } catch (err) {
-      throw SocketError.from(err);
-    }
-  }
-
-  /**
-   * Sets the receive buffer size
-   * WIT:
-   * ```
-   * set-receive-buffer-size: func(value: u64) -> result<_, error-code>
-   * ```
-   *
-   * @param {bigint} value - The receive buffer size in bytes
-   * @returns {void}
-   * @throws {SocketError} with payload.tag 'invalid-argument' if `value` is 0
-   */
-  setReceiveBufferSize(value) {
-    if (value <= 0n) {
-      throw new SocketError("invalid-argument");
-    }
-    this.#options.receiveBufferSize = value;
-  }
-
-  /**
-   * Gets the send buffer size
-   * WIT:
-   * ```
-   * send-buffer-size: func() -> result<u64, error-code>
-   * ```
-   *
-   * @returns {bigint} The send buffer size in bytes
-   * @throws {SocketError} for other errors, payload.tag maps the system error
-   */
-  getSendBufferSize() {
-    if (this.#options.sendBufferSize) {
-      return this.#options.sendBufferSize;
-    }
-
-    try {
-      const result = worker().runSync({
-        op: "tcp-send-buffer-size",
-        socketId: this.#socketId,
-      });
-
-      this.#options.sendBufferSize = result;
-      return result;
-    } catch (err) {
-      throw SocketError.from(err);
-    }
-  }
-
-  /**
-   * Sets the send buffer size
-   * WIT:
-   * ```
-   * set-send-buffer-size: func(value: u64) -> result<_, error-code>
-   * ```
-   *
-   * @param {bigint} value - The send buffer size in bytes
-   * @returns {void}
-   * @throws {SocketError} with payload.tag 'invalid-argument' if `value` is 0
-   */
-  setSendBufferSize(value) {
-    if (value <= 0n) {
-      throw new SocketError("invalid-argument");
-    }
-    this.#options.sendBufferSize = value;
-  }
-
-  /**
-   * Disposes the socket
-   */
-  [Symbol.dispose]() {
-    if (this.#socketId) {
-      worker().runSync({
-        op: "tcp-dispose",
-        socketId: this.#socketId,
-      });
-    }
-
-    this.#socketId = null;
-    this.#state = STATE.CLOSED;
-  }
-
-  /**
-   * Validates if a local address is valid for this socket
-   * @param {Object} localAddress - The local address to validate
-   * @returns {boolean} True if the address is valid, false otherwise
-   * @private
-   */
-  #isValidLocalAddress(localAddress) {
-    return (
-      this.#family === localAddress.tag &&
-      (isUnicastIpAddress(localAddress) || isWildcardIpAddress(localAddress)) &&
-      !isIpv4MappedAddress(localAddress)
-    );
-  }
-
-  /**
-   * Validates if a remote address is valid for this socket
-   * @param {Object} remoteAddress - The remote address to validate
-   * @returns {boolean} True if the address is valid, false otherwise
-   * @private
-   */
-  #isValidRemoteAddress(remoteAddress) {
-    return (
-      this.#family === remoteAddress.tag &&
-      remoteAddress.val.port !== 0 &&
-      isUnicastIpAddress(remoteAddress) &&
-      !isWildcardIpAddress(remoteAddress) &&
-      !isMulticastIpAddress(remoteAddress) &&
-      !isIpv4MappedAddress(remoteAddress)
-    );
-  }
-}