@steve02081504/fount-p2p 0.0.5 → 0.0.7

package/link/providers/lan_tcp.mjs

@@ -0,0 +1,342 @@
+import { Buffer } from 'node:buffer'
+import { randomBytes } from 'node:crypto'
+import net from 'node:net'
+
+import { normalizeHex64 } from '../../core/hexIds.mjs'
+import { getLanPeerHint } from '../../discovery/lan_peer_hints.mjs'
+import { asLinkHandle, coercePipeInbound, createLinkPipe } from '../pipe.mjs'
+
+import { LINK_LEVEL_LAN_TCP } from './levels.mjs'
+
+const MAX_FRAME_BYTES = 1 << 20
+
+/**
+ * 在 socket 上挂 length-prefix 编解码（u32be + payload）。
+ * @param {import('node:net').Socket} socket TCP socket
+ * @param {(payload: Buffer) => void} onPayload 完整帧回调
+ * @returns {{ write: (payload: Buffer | Uint8Array | string) => void, destroy: () => void }} 编解码句柄
+ */
+function attachLengthPrefix(socket, onPayload) {
+	/** @type {Buffer[]} */
+	const chunks = []
+	let buffered = 0
+	/**
+	 * @param {number} n 需要的字节数
+	 * @returns {Buffer | null} 凑齐则返回，否则 null
+	 */
+	function take(n) {
+		if (buffered < n) return null
+		const out = Buffer.allocUnsafe(n)
+		let offset = 0
+		while (offset < n) {
+			const head = chunks[0]
+			const copy = Math.min(head.length, n - offset)
+			head.copy(out, offset, 0, copy)
+			offset += copy
+			buffered -= copy
+			if (copy === head.length) chunks.shift()
+			else chunks[0] = head.subarray(copy)
+		}
+		return out
+	}
+	/**
+	 * @param {Buffer} chunk 入站数据
+	 * @returns {void}
+	 */
+	function onData(chunk) {
+		chunks.push(chunk)
+		buffered += chunk.length
+		while (buffered >= 4) {
+			const header = take(4)
+			const len = header.readUInt32BE(0)
+			if (len > MAX_FRAME_BYTES) {
+				socket.destroy(new Error('p2p: lan_tcp frame too large'))
+				return
+			}
+			if (buffered < len) {
+				chunks.unshift(header)
+				buffered += 4
+				return
+			}
+			onPayload(take(len))
+		}
+	}
+	socket.on('data', onData)
+	return {
+		/**
+		 * @param {Buffer | Uint8Array | string} payload 出站载荷
+		 * @returns {void}
+		 */
+		write(payload) {
+			const body = Buffer.isBuffer(payload) ? payload : Buffer.from(payload)
+			const header = Buffer.allocUnsafe(4)
+			header.writeUInt32BE(body.length)
+			socket.write(Buffer.concat([header, body]))
+		},
+		/**
+		 * @returns {void}
+		 */
+		destroy() {
+			socket.off('data', onData)
+		},
+	}
+}
+
+/**
+ * 在已挂 codec 的 socket 上创建 pipe。
+ * @param {object} options 配置
+ * @returns {ReturnType<typeof createLinkPipe>} pipe 句柄
+ */
+function createTcpPipe(options) {
+	const linkId = normalizeHex64(options.linkId)
+	if (!linkId) throw new Error('p2p: lan_tcp linkId required')
+	const { socket, codec } = options
+	const pipe = createLinkPipe({
+		providerId: 'lan_tcp',
+		level: LINK_LEVEL_LAN_TCP,
+		initiator: !!options.initiator,
+		nodeHash: options.nodeHash,
+		localIdentity: options.localIdentity,
+		/** @returns {string} 本端 binding（linkId） */
+		getLocalBinding: () => linkId,
+		/** @returns {string} 对端 binding（linkId） */
+		getRemoteBinding: () => linkId,
+		/**
+		 * @param {string} text control JSON
+		 * @returns {void}
+		 */
+		sendControlText(text) {
+			codec.write(text)
+		},
+		/**
+		 * @param {string} _action action
+		 * @param {Uint8Array} frame 帧
+		 * @returns {void}
+		 */
+		sendFrame(_action, frame) {
+			codec.write(frame)
+		},
+		/**
+		 * @returns {void}
+		 */
+		closeTransport() {
+			codec.destroy()
+			socket.destroy()
+		},
+		/**
+		 * @returns {object} 附加 stats
+		 */
+		extraStats() {
+			return { host: options.host || null, port: options.port || null }
+		},
+	})
+	socket.on('close', () => { void pipe.close('socket-close') })
+	socket.on('error', () => { void pipe.close('socket-error') })
+	return pipe
+}
+
+/**
+ * 在已连接 socket 上建立 pipe（入站帧经 pending 缓冲，避免 link-open/hello 竞态丢包）。
+ * @param {object} options 配置
+ * @returns {Promise<import('./index.mjs').LinkHandle>} 已启动握手的 link
+ */
+async function openTcpPipe(options) {
+	const socket = options.socket
+	/** @type {ReturnType<typeof createLinkPipe> | null} */
+	let pipe = null
+	/** @type {Buffer[]} */
+	const pending = []
+	const codec = attachLengthPrefix(socket, payload => {
+		if (!pipe) {
+			pending.push(payload)
+			return
+		}
+		pipe.handleInbound(coercePipeInbound(payload))
+	})
+
+	pipe = createTcpPipe({ ...options, codec, socket })
+	for (const payload of pending.splice(0))
+		pipe.handleInbound(coercePipeInbound(payload))
+
+	if (options.initiator)
+		codec.write(JSON.stringify({
+			type: 'link-open',
+			linkId: normalizeHex64(options.linkId),
+			from: options.localIdentity?.nodeHash || '',
+		}))
+
+	await pipe.startHandshake()
+	return asLinkHandle(pipe)
+}
+
+/**
+ * 拨号到已有 LAN hint。
+ * @param {object} options dial 选项
+ * @returns {Promise<import('./index.mjs').LinkHandle>} 已就绪的 link
+ */
+async function dialLanTcp(options) {
+	const remoteNodeHash = normalizeHex64(options.nodeHash)
+	const hint = getLanPeerHint(remoteNodeHash)
+	if (!hint) throw new Error('p2p: lan_tcp no peer hint')
+
+	const socket = await new Promise((resolve, reject) => {
+		const conn = net.createConnection({ host: hint.host, port: hint.port })
+		/**
+		 * @param {Error} error 连接错误
+		 * @returns {void}
+		 */
+		const onError = error => {
+			conn.off('connect', onConnect)
+			reject(error)
+		}
+		/**
+		 * @returns {void}
+		 */
+		function onConnect() {
+			conn.off('error', onError)
+			resolve(conn)
+		}
+		conn.once('error', onError)
+		conn.once('connect', onConnect)
+	})
+
+	const link = await openTcpPipe({
+		initiator: true,
+		linkId: randomBytes(32).toString('hex'),
+		nodeHash: remoteNodeHash,
+		localIdentity: options.localIdentity,
+		socket,
+		host: hint.host,
+		port: hint.port,
+	})
+	await link.ready
+	return link
+}
+
+/**
+ * 创建 lan_tcp LinkProvider。
+ * 每个实例独立 listen socket；注册 id 唯一，避免同进程多 registry 互相覆盖。
+ * 链路上的 `providerId` 仍为 `lan_tcp`（见 createTcpPipe）。
+ * @returns {import('./index.mjs').LinkProvider & { ensureListening?: Function, localEndpoint?: Function }} LAN TCP provider
+ */
+export function createLanTcpLinkProvider() {
+	const instanceId = `lan_tcp:${randomBytes(4).toString('hex')}`
+	/** @type {((link: import('./index.mjs').LinkHandle) => void) | null} */
+	let onInbound = null
+	/** @type {object | null} */
+	let localIdentity = null
+	/** @type {import('node:net').Server | null} */
+	let server = null
+	let listenPort = 0
+
+	/**
+	 * @param {import('node:net').Socket} socket 入站连接
+	 * @returns {void}
+	 */
+	function acceptConnection(socket) {
+		if (!onInbound || !localIdentity) {
+			socket.destroy()
+			return
+		}
+		/** @type {ReturnType<typeof createLinkPipe> | null} */
+		let pipe = null
+		const codec = attachLengthPrefix(socket, payload => {
+			if (pipe) {
+				pipe.handleInbound(coercePipeInbound(payload))
+				return
+			}
+			let parsed
+			try {
+				parsed = JSON.parse(payload.toString('utf8'))
+			}
+			catch {
+				socket.destroy()
+				return
+			}
+			if (parsed?.type !== 'link-open' || !parsed.linkId) {
+				socket.destroy()
+				return
+			}
+			try {
+				pipe = createTcpPipe({
+					initiator: false,
+					linkId: parsed.linkId,
+					nodeHash: normalizeHex64(parsed.from) || null,
+					localIdentity,
+					socket,
+					codec,
+					port: listenPort,
+				})
+			}
+			catch {
+				socket.destroy()
+				return
+			}
+			onInbound?.(asLinkHandle(pipe))
+			void pipe.startHandshake().catch(() => {
+				socket.destroy()
+			})
+		})
+		socket.on('error', () => { /* ignore */ })
+	}
+
+	return {
+		id: instanceId,
+		level: LINK_LEVEL_LAN_TCP,
+		caps: { needsOfferAnswer: false, needsDiscoverySignal: false },
+		/**
+		 * @returns {boolean} 始终可用
+		 */
+		isAvailable() {
+			return true
+		},
+		/**
+		 * @param {{ nodeHash: string }} remote 远端
+		 * @returns {boolean} 是否有 LAN peer hint
+		 */
+		canReach(remote) {
+			return !!getLanPeerHint(remote.nodeHash)
+		},
+		/**
+		 * @returns {{ port: number } | null} 本机 listen 端点
+		 */
+		localEndpoint() {
+			return listenPort > 0 ? { port: listenPort } : null
+		},
+		/**
+		 * @param {object} options dial 选项
+		 * @returns {Promise<import('./index.mjs').LinkHandle>} 已就绪的 link
+		 */
+		async dial(options) {
+			return dialLanTcp(options)
+		},
+		/**
+		 * @param {{ onInbound: (link: import('./index.mjs').LinkHandle) => void, localIdentity: object }} handlers 回调
+		 * @returns {Promise<() => void>} 停止 listening
+		 */
+		async ensureListening(handlers) {
+			onInbound = handlers.onInbound
+			localIdentity = handlers.localIdentity
+			if (!server) {
+				server = net.createServer(acceptConnection)
+				await new Promise((resolve, reject) => {
+					server.once('error', reject)
+					server.listen(0, '0.0.0.0', () => {
+						server.off('error', reject)
+						const addr = server.address()
+						listenPort = typeof addr === 'object' && addr ? addr.port : 0
+						resolve()
+					})
+				})
+			}
+			return () => {
+				onInbound = null
+				if (server) {
+					server.close()
+					server = null
+					listenPort = 0
+				}
+			}
+		},
+	}
+}

package/link/providers/levels.mjs

@@ -0,0 +1,9 @@
+/**
+ * LinkProvider level 约定：数值越大越优先（建链时降序尝试）。
+ * Discovery 的 priority 仍为升序，两套语义故意分名。
+ */
+export const LINK_LEVEL_LAN_TCP = 80
+/** WebRTC 链路 level。 */
+export const LINK_LEVEL_WEBRTC = 70
+/** BLE GATT 链路 level。 */
+export const LINK_LEVEL_BLE_GATT = 40