Add Async Socket API

Author: yne

examples/http_client.js

@@ -0,0 +1,18 @@
+#!/usr/bin/env qjs
+///@ts-check
+/// <reference path="../doc/globals.d.ts" />
+/// <reference path="../doc/os.d.ts" />
+import * as os from "os";
+/** @template T @param {os.Result<T>} result */
+function must(result) {
+    if (typeof result === "number" && result < 0) throw result;
+    return /** @type {T} */ (result)
+}
+
+const sockfd = must(os.socket(os.AF_INET, os.SOCK_STREAM));
+await os.connect(sockfd, os.getaddrinfo("bellard.org",'80')[0]);
+const httpReq = ["GET / HTTP/1.0", "", ""].join('\r\n')
+must(await os.send(sockfd, Uint8Array.from(httpReq, c => c.charCodeAt(0)).buffer) > 0);
+const chunk = new Uint8Array(512);
+const recvd = await os.recv(sockfd, chunk.buffer);
+console.log([...chunk.slice(0,recvd)].map(c => String.fromCharCode(c)).join(''));

examples/http_server.js

@@ -0,0 +1,94 @@
+#!/usr/bin/env qjs
+///@ts-check
+/// <reference path="../doc/globals.d.ts" />
+/// <reference path="../doc/os.d.ts" />
+import * as os from "os";
+
+const MIMES = new Map([
+    ['html', 'text/html'],
+    ['txt', 'text/plain'],
+    ['css', 'text/css'],
+    ['c', 'text/plain'],
+    ['h', 'text/plain'],
+    ['json', 'application/json'],
+    ['mjs', 'application/javascript'],
+    ['js', 'application/javascript'],
+    ['', 'application/octet-stream'],
+]);
+/** @template T @param {os.Result<T>} result */
+function must(result) {
+    if (typeof result === "number" && result < 0) throw result;
+    return /** @type {T} */ (result)
+}
+/**@param {os.FileDescriptor} fd */
+async function* recvLines(fd) {
+    const chunk = new Uint8Array(1);
+    let line = '';
+    while (await os.recv(fd, chunk.buffer) > 0) {
+        const char = String.fromCharCode(...chunk);
+        if (char == '\n') {
+            yield line;
+            line = '';
+        } else line += char;
+    }
+    if (line) yield line;
+}
+/** @param {os.FileDescriptor} fd @param {string[]} lines */
+function sendLines(fd, lines) {
+    const buf = Uint8Array.from(lines.join('\r\n'), c => c.charCodeAt(0));
+    return os.send(fd, buf.buffer);
+}
+//USAGE: qjs http_server.js [PORT=8080 [HOST=localhost]]
+const [port = "8080", host = "localhost"] = scriptArgs.slice(1);
+const [ai] = os.getaddrinfo(host, port);
+//if (!ai.length) throw `Unable to getaddrinfo(${host}, ${port})`;
+const sock_srv = must(os.socket(os.AF_INET, os.SOCK_STREAM));
+must(os.setsockopt(sock_srv, os.SO_REUSEADDR, new Uint32Array([1]).buffer));
+must(os.bind(sock_srv, ai));
+must(os.listen(sock_srv));
+//os.signal(os.SIGINT, ()=>os.close(sock_srv)); // don't work
+console.log(`Listening on http://${host}:${port} (${ai.addr}:${ai.port}) ...`);
+const openCmd = { linux: "xdg-open", darwin: "open", win32: "start" }[os.platform];
+if (openCmd) os.exec([openCmd, `http://${host}:${port}`]);
+while (true) { // TODO: break on SIG*
+    const [sock_cli] = await os.accept(sock_srv);
+
+    const lines = recvLines(sock_cli);
+    const [method, path, http_ver] = ((await lines.next()).value || '').split(' ');
+    let safe_path = '.' + path.replaceAll(/\.+/g, '.'); // may += index.html later
+    console.log(method, safe_path, http_ver);
+
+    const headers = new Map()
+    for await (const line of lines) {
+        const header = line.trimEnd();
+        if (!header) break;
+        const sepIdx = header.indexOf(': ');
+        headers.set(header.slice(0, sepIdx), header.slice(sepIdx + 2));
+    }
+
+    let [obj, err] = os.stat(safe_path);
+    if (obj?.mode & os.S_IFDIR && safe_path.endsWith('/') && os.stat(safe_path + 'index.html')[0]) {
+        safe_path += 'index.html';
+        [obj, err] = os.stat(safe_path);
+    }
+    if (err) {
+        await sendLines(sock_cli, ['HTTP/1.1 404', '', safe_path, 'errno:' + err])
+    } else if (obj?.mode & os.S_IFDIR) {
+        if (!safe_path.endsWith('/'))
+            await sendLines(sock_cli, ['HTTP/1.1 301', `Location: ${safe_path}/`, '']);
+        else
+            await sendLines(sock_cli, ['HTTP/1.1 200', 'Content-Type: text/html', '',
+                os.readdir(safe_path)[0]?.filter(e => e[0] != '.').map(e => `<li><a href="${e}">${e}</a></li>`).join('')
+            ]);
+    } else {
+        const mime = MIMES.get(safe_path.split('.').at(-1) || '') || MIMES.get('');
+        await sendLines(sock_cli, ['HTTP/1.1 200', `Content-Type: ${mime}`, '', '']);
+        const fd = must(os.open(safe_path));
+        const fbuf = new Uint8Array(4096);
+        for (let got = 0; (got = os.read(fd, fbuf.buffer, 0, fbuf.byteLength)) > 0;) {
+            await os.send(sock_cli, fbuf.buffer, got);
+        }
+    }
+
+    os.close(sock_cli);
+}