Files
bun.sh/src/js/builtins/Ipc.ts
2025-05-06 22:12:24 -07:00

237 lines
7.5 KiB
TypeScript

// for net.Server, get ._handle
// const handleConversion = {
// "net.Server": {
// simultaneousAccepts: true,
// send(message, server, options) {
// return server._handle;
// },
// got(message, handle, emit) {
// const server = new net.Server();
// server.listen(handle, () => {
// emit(server);
// });
// },
// },
// "net.Socket": {
// send(message, socket, options) {
// if (!socket._handle) return;
// // If the socket was created by net.Server
// if (socket.server) {
// // The worker should keep track of the socket
// message.key = socket.server._connectionKey;
// const firstTime = !this[kChannelHandle].sockets.send[message.key];
// const socketList = getSocketList("send", this, message.key);
// // The server should no longer expose a .connection property
// // and when asked to close it should query the socket status from
// // the workers
// if (firstTime) socket.server._setupWorker(socketList);
// // Act like socket is detached
// if (!options.keepOpen) socket.server._connections--;
// }
// const handle = socket._handle;
// // Remove handle from socket object, it will be closed when the socket
// // will be sent
// if (!options.keepOpen) {
// handle.onread = nop;
// socket._handle = null;
// socket.setTimeout(0);
// if (freeParser === undefined) freeParser = require("_http_common").freeParser;
// if (HTTPParser === undefined) HTTPParser = require("_http_common").HTTPParser;
// // In case of an HTTP connection socket, release the associated
// // resources
// if (socket.parser && socket.parser instanceof HTTPParser) {
// freeParser(socket.parser, null, socket);
// if (socket._httpMessage) socket._httpMessage.detachSocket(socket);
// }
// }
// return handle;
// },
// postSend(message, handle, options, callback, target) {
// // Store the handle after successfully sending it, so it can be closed
// // when the NODE_HANDLE_ACK is received. If the handle could not be sent,
// // just close it.
// if (handle && !options.keepOpen) {
// if (target) {
// // There can only be one _pendingMessage as passing handles are
// // processed one at a time: handles are stored in _handleQueue while
// // waiting for the NODE_HANDLE_ACK of the current passing handle.
// assert(!target._pendingMessage);
// target._pendingMessage = { callback, message, handle, options, retransmissions: 0 };
// } else {
// handle.close();
// }
// }
// // NOTE that another function will call _pendingMessage.handle.close() and set _pendingMessage to null
// },
// got(message, handle, emit) {
// const socket = new net.Socket({
// handle: handle,
// readable: true,
// writable: true,
// });
// // If the socket was created by net.Server we will track the socket
// if (message.key) {
// // Add socket to connections list
// const socketList = getSocketList("got", this, message.key);
// socketList.add({
// socket: socket,
// });
// }
// emit(socket);
// },
// },
// "dgram.Native": {
// simultaneousAccepts: false,
// send(message, handle, options) {
// return handle;
// },
// got(message, handle, emit) {
// emit(handle);
// },
// },
// "dgram.Socket": {
// simultaneousAccepts: false,
// send(message, socket, options) {
// message.dgramType = socket.type;
// return socket[kStateSymbol].handle;
// },
// got(message, handle, emit) {
// const socket = new dgram.Socket(message.dgramType);
// socket.bind(handle, () => {
// emit(socket);
// });
// },
// },
// };
// have to use jsdoc type definitions because bundle-functions is based on regex
/**
* @typedef {Object} Serialized
* @property {"NODE_HANDLE"} cmd
* @property {unknown} message
* @property {"net.Socket" | "net.Server" | "dgram.Socket"} type
*/
/**
* @typedef {import("node:net").Server | import("node:net").Socket | import("node:dgram").Socket} Handle
*/
/**
* @param {unknown} message
* @param {Handle} handle
* @param {{ keepOpen?: boolean } | undefined} options
* @returns {[unknown, Serialized] | null}
*/
export function serialize(_message, _handle, _options) {
// sending file descriptors is not supported yet
return null; // send the message without the file descriptor
/*
const net = require("node:net");
const dgram = require("node:dgram");
if (handle instanceof net.Server) {
// this one doesn't need a close function, but the fd needs to be kept alive until it is sent
const server = handle as unknown as (typeof net)["Server"] & { _handle: Bun.TCPSocketListener<unknown> };
return [server._handle, { cmd: "NODE_HANDLE", message, type: "net.Server" }];
} else if (handle instanceof net.Socket) {
const new_message: { cmd: "NODE_HANDLE"; message: unknown; type: "net.Socket"; key?: string } = {
cmd: "NODE_HANDLE",
message,
type: "net.Socket",
};
const socket = handle as unknown as (typeof net)["Socket"] & {
_handle: Bun.Socket;
server: (typeof net)["Server"] | null;
setTimeout(timeout: number): void;
};
if (!socket._handle) return null; // failed
// If the socket was created by net.Server
if (socket.server) {
// The worker should keep track of the socket
new_message.key = socket.server._connectionKey;
const firstTime = !this[kChannelHandle].sockets.send[message.key];
const socketList = getSocketList("send", this, message.key);
// The server should no longer expose a .connection property
// and when asked to close it should query the socket status from
// the workers
if (firstTime) socket.server._setupWorker(socketList);
// Act like socket is detached
if (!options?.keepOpen) socket.server._connections--;
}
const internal_handle = socket._handle;
// Remove handle from socket object, it will be closed when the socket
// will be sent
if (!options?.keepOpen) {
// we can use a $newZigFunction to have it unset the callback
internal_handle.onread = nop;
socket._handle = null;
socket.setTimeout(0);
}
return [internal_handle, new_message];
} else if (handle instanceof dgram.Socket) {
// this one doesn't need a close function, but the fd needs to be kept alive until it is sent
throw new Error("todo serialize dgram.Socket");
} else {
throw $ERR_INVALID_HANDLE_TYPE();
}
*/
}
/**
* @param {Serialized} serialized
* @param {unknown} handle
* @param {(handle: Handle) => void} emit
* @returns {void}
*/
export function parseHandle(target, serialized, fd) {
const emit = $newZigFunction("ipc.zig", "emitHandleIPCMessage", 3);
const net = require("node:net");
// const dgram = require("node:dgram");
switch (serialized.type) {
case "net.Server": {
const server = new net.Server();
server.listen({ fd }, () => {
emit(target, serialized.message, server);
});
return;
}
case "net.Socket": {
throw new Error("TODO case net.Socket");
}
case "dgram.Socket": {
throw new Error("TODO case dgram.Socket");
}
default: {
throw new Error("failed to parse handle");
}
}
}