╭─────────────╮ │scroll to top│ ╰─────────────╯ ╭ web.zig ──────────────────────────────────────────────────────╮ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ╰───────────────────────────────────────────────────────────────╯
╭────╮ ╭ clone ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╮ │http│ ssh │ │ ╰────╯ ╰ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╯
2291 } 2292 } 2293 2294 // the 16 ansi colors as the browser shows them (xterm's defaults) 2295 const ansi_palette = [16]Grid.Color.Rgb{ 2296 .{ .r = 0x00, .g = 0x00, .b = 0x00 }, .{ .r = 0xcd, .g = 0x00, .b = 0x00 }, .{ .r = 0x00, .g = 0xcd, .b = 0x00 }, .{ .r = 0xcd, .g = 0xcd, .b = 0x00 }, 2297 .{ .r = 0x00, .g = 0x00, .b = 0xee }, .{ .r = 0xcd, .g = 0x00, .b = 0xcd }, .{ .r = 0x00, .g = 0xcd, .b = 0xcd }, .{ .r = 0xe5, .g = 0xe5, .b = 0xe5 }, 2298 .{ .r = 0x7f, .g = 0x7f, .b = 0x7f }, .{ .r = 0xff, .g = 0x00, .b = 0x00 }, .{ .r = 0x00, .g = 0xff, .b = 0x00 }, .{ .r = 0xff, .g = 0xff, .b = 0x00 }, 2299 .{ .r = 0x5c, .g = 0x5c, .b = 0xff }, .{ .r = 0xff, .g = 0x00, .b = 0xff }, .{ .r = 0x00, .g = 0xff, .b = 0xff }, .{ .r = 0xff, .g = 0xff, .b = 0xff }, 2300 }; 2301 2302 // resolve any color to rgb: palette colors through the ansi table and the 2303 // xterm 256-color layout (16 ansi, a 6x6x6 cube, then 24 grays) 2304 fn colorRgb(color: Grid.Color) Grid.Color.Rgb { 2305 return switch (color) { 2306 .rgb => |c| c, 2307 .ansi => |a| ansi_palette[@intFromEnum(a)], 2308 .indexed => |n| if (n < 16) 2309 ansi_palette[n] 2310 else if (n < 232) blk: { 2311 const levels = [6]u8{ 0, 0x5f, 0x87, 0xaf, 0xd7, 0xff }; 2312 const i = n - 16; 2313 break :blk .{ .r = levels[i / 36], .g = levels[(i / 6) % 6], .b = levels[i % 6] }; 2314 } else blk: { 2315 const v: u8 = 8 + (n - 232) * 10; 2316 break :blk .{ .r = v, .g = v, .b = v }; 2317 }, 2318 }; 2319 } 2320 2321 // builds an HTML ` style="color:…;background-color:…"` attribute from a 2322 // cell's style into `buf`, omitting whichever color is unset. an inverted 2323 // cell swaps the two, falling back to the page's --fg/--bg variables for 2324 // whichever side is unset. text attributes follow the colors. returns an 2325 // empty slice when nothing is set. 2326 fn styleAttr(buf: []u8, style: Grid.Style) []const u8 { 2327 if (style.eql(.{})) return buf[0..0]; 2328 const prefix = " style=\""; 2329 @memcpy(buf[0..prefix.len], prefix); 2330 var i: usize = prefix.len; 2331 const inverted = style.inverted; 2332 i = cssColor(buf, i, "color", if (inverted) style.bg else style.fg, if (inverted) "var(--bg)" else null) orelse return buf[0..0]; 2333 i = cssColor(buf, i, "background-color", if (inverted) style.fg else style.bg, if (inverted) "var(--fg)" else null) orelse return buf[0..0]; 2334 const decoration: []const u8 = if (style.underline and style.strikethrough) 2335 "text-decoration:underline line-through;" 2336 else if (style.underline) 2337 "text-decoration:underline;" 2338 else if (style.strikethrough) 2339 "text-decoration:line-through;" 2340 else 2341 ""; 2342 const attributes = std.fmt.bufPrint(buf[i..], "{s}{s}{s}{s}", .{ 2343 if (style.bold) "font-weight:bold;" else "", 2344 if (style.italic) "font-style:italic;" else "", 2345 decoration, 2346 if (style.dim) "opacity:.6;" else "", 2347 }) catch return buf[0..0]; 2348 i += attributes.len; 2349 if (i >= buf.len) return buf[0..0]; 2350 buf[i] = '"'; 2351 return buf[0 .. i + 1]; 2352 } 2353 2354 // append `prop:value;` at buf[i..] for the color, or for the fallback when 2355 // the color is unset (nothing when both are). returns the new end, or null 2356 // when it doesn't fit. 2357 fn cssColor(buf: []u8, i: usize, prop: []const u8, color: ?Grid.Color, fallback: ?[]const u8) ?usize { 2358 const s = if (color) |c| blk: { 2359 const v = colorRgb(c); 2360 break :blk std.fmt.bufPrint(buf[i..], "{s}:#{x:0>2}{x:0>2}{x:0>2};", .{ prop, v.r, v.g, v.b }) catch return null; 2361 } else if (fallback) |f| 2362 std.fmt.bufPrint(buf[i..], "{s}:{s};", .{ prop, f }) catch return null 2363 else 2364 return i; 2365 return i + s.len; 2366 } 2367 2368 // a cell's link as an href, when it's a web or mail url 2369 fn webLink(link: ?[]const u8) ?[]const u8 { 2370 const url = link orelse return null; 2371 for ([_][]const u8{ "http://", "https://", "mailto:" }) |scheme| { 2372 if (std.ascii.startsWithIgnoreCase(url, scheme)) return url; 2373 } 2374 return null; 2375 } 2376 2377 fn sameLink(a: ?[]const u8, b: ?[]const u8) bool { 2378 if (a) |x| return if (b) |y| std.mem.eql(u8, x, y) else false; 2379 return b == null; 2380 } 2381 2382 fn appendEscapedHtml(allocator: std.mem.Allocator, out: *std.ArrayList(u8), input: []const u8) !void { 2383 for (input) |ch| { 2384 switch (ch) { 2385 '&' => try out.appendSlice(allocator, "&amp;"), 2386 '<' => try out.appendSlice(allocator, "&lt;"), 2387 '>' => try out.appendSlice(allocator, "&gt;"), 2388 '"' => try out.appendSlice(allocator, "&quot;"), 2389 '\'' => try out.appendSlice(allocator, "&#39;"), 2390 else => try out.append(allocator, ch), 2391 } 2392 } 2393 } 2394 2395 // --- helpers -------------------------------------------------------------- 2396 2397 fn getCookieValue(request: *std.http.Server.Request, name: []const u8) ?[]const u8 { 2398 var iter = request.iterateHeaders(); 2399 while (iter.next()) |header| { 2400 if (!std.ascii.eqlIgnoreCase(header.name, "cookie")) continue; 2401 var pairs = std.mem.splitScalar(u8, header.value, ';'); 2402 while (pairs.next()) |pair| { 2403 const trimmed = std.mem.trim(u8, pair, " \t"); 2404 const eq = std.mem.indexOfScalar(u8, trimmed, '=') orelse continue; 2405 if (std.mem.eql(u8, trimmed[0..eq], name)) { 2406 return trimmed[eq + 1 ..]; 2407 } 2408 } 2409 } 2410 return null; 2411 } 2412 2413 fn parseFormField(allocator: std.mem.Allocator, body: []const u8, key: []const u8) !?[]u8 { 2414 var iter = std.mem.splitScalar(u8, body, '&'); 2415 while (iter.next()) |pair| { 2416 const eq = std.mem.indexOfScalar(u8, pair, '=') orelse continue; 2417 if (std.mem.eql(u8, pair[0..eq], key)) { 2418 return try decodeFormValue(allocator, pair[eq + 1 ..]); 2419 } 2420 } 2421 return null; 2422 } 2423 2424 fn readFormBody(request: *std.http.Server.Request, allocator: std.mem.Allocator) ![]u8 { 2425 var buffer: [256]u8 = undefined; 2426 return request.readerExpectNone(&buffer).allocRemaining(allocator, .limited(65536)); 2427 } 2428 2429 fn decodeFormValue(allocator: std.mem.Allocator, encoded: []const u8) ![]u8 { 2430 var out: std.ArrayList(u8) = .empty; 2431 errdefer out.deinit(allocator); 2432 var i: usize = 0; 2433 while (i < encoded.len) : (i += 1) { 2434 switch (encoded[i]) { 2435 '+' => try out.append(allocator, ' '), 2436 '%' => { 2437 if (i + 2 >= encoded.len) return error.InvalidPercentEncoding; 2438 const hi = std.fmt.charToDigit(encoded[i + 1], 16) catch return error.InvalidPercentEncoding; 2439 const lo = std.fmt.charToDigit(encoded[i + 2], 16) catch return error.InvalidPercentEncoding; 2440 try out.append(allocator, hi * 16 + lo); 2441 i += 2; 2442 }, 2443 else => try out.append(allocator, encoded[i]), 2444 } 2445 } 2446 return try out.toOwnedSlice(allocator); 2447 } 2448 2449 // disk-backed web session store. each login mints a random, opaque token 2450 // (not the user's id) that is stored as a file named by the token hex, whose 2451 // contents are the raw user_id bytes. authenticating a request is a lookup of 2452 // the cookie's token; logout deletes the file, revoking the session. there is 2453 // deliberately no expiry — a session lives until logout. form failures use 2454 // files in the flash subdirectory that the redirected GET consumes. the 2455 // directory is cleared whenever the server starts. 2456 pub const SessionStore = struct { 2457 io: std.Io, 2458 dir: std.Io.Dir, 2459 flash_dir: std.Io.Dir, 2460 auto_login: ?[token_hex_len]u8, 2461 2462 pub const token_hex_len = evt.event_id_size * 2; 2463 2464 pub fn init(io: std.Io, data_dir: std.Io.Dir) !SessionStore { 2465 try data_dir.createDirPath(io, "sessions"); 2466 const dir = try data_dir.openDir(io, "sessions", .{}); 2467 errdefer dir.close(io); 2468 try dir.deleteTree(io, "flash"); 2469 try dir.createDirPath(io, "flash"); 2470 const flash_dir = try dir.openDir(io, "flash", .{}); 2471 errdefer flash_dir.close(io); 2472 const auto_login = blk: { 2473 const file = dir.openFile(io, auto_login_name, .{ .mode = .read_only }) catch break :blk null; 2474 defer file.close(io); 2475 var token: [token_hex_len]u8 = undefined; 2476 var storage: [token_hex_len]u8 = undefined; 2477 var reader = file.reader(io, &storage); 2478 reader.interface.readSliceAll(&token) catch break :blk null; 2479 break :blk token; 2480 }; 2481 return .{ .io = io, .dir = dir, .flash_dir = flash_dir, .auto_login = auto_login }; 2482 } 2483 2484 pub fn deinit(self: SessionStore) void { 2485 self.flash_dir.close(self.io); 2486 self.dir.close(self.io); 2487 } 2488 2489 // mint a new session for user_id, returning the token hex to set as the 2490 // cookie value. 2491 pub fn create(self: SessionStore, user_id: *const [evt.event_id_size]u8) ![token_hex_len]u8 { 2492 var token: [evt.event_id_size]u8 = undefined; 2493 self.io.random(&token); 2494 const token_hex = std.fmt.bytesToHex(token, .lower); 2495 const file = try self.dir.createFile(self.io, &token_hex, .{}); 2496 defer file.close(self.io); 2497 try file.writeStreamingAll(self.io, user_id); 2498 return token_hex; 2499 } 2500 2501 fn createFlash(self: SessionStore, allocator: std.mem.Allocator, value: ui.Session.FormFeedback) ![token_hex_len]u8 { 2502 var token: [evt.event_id_size]u8 = undefined; 2503 self.io.random(&token); 2504 const token_hex = std.fmt.bytesToHex(token, .lower); 2505 const file = try self.flash_dir.createFile(self.io, &token_hex, .{}); 2506 errdefer self.flash_dir.deleteFile(self.io, &token_hex) catch {}; 2507 defer file.close(self.io); 2508 2509 var json: std.Io.Writer.Allocating = .init(allocator); 2510 defer json.deinit(); 2511 try std.json.Stringify.value(value, .{}, &json.writer); 2512 try file.writeStreamingAll(self.io, json.written()); 2513 return token_hex; 2514 } 2515 2516 fn takeFlash(self: SessionStore, allocator: std.mem.Allocator, token_hex: []const u8) ?ui.Session.FormFeedback { 2517 if (!isToken(token_hex)) return null; 2518 const file = self.flash_dir.openFile(self.io, token_hex, .{ .mode = .read_only }) catch return null; 2519 defer self.flash_dir.deleteFile(self.io, token_hex) catch {}; 2520 defer file.close(self.io); 2521 2522 var storage: [4096]u8 = undefined; 2523 var reader = file.reader(self.io, &storage); 2524 const bytes = reader.interface.allocRemaining(allocator, .limited(512 * 1024)) catch return null; 2525 return std.json.parseFromSliceLeaky(ui.Session.FormFeedback, allocator, bytes, .{}) catch null; 2526 } 2527 2528 // resolve a cookie's token to its user_id. returns false (logged out) for a 2529 // missing, malformed, or unknown token. 2530 pub fn lookup(self: SessionStore, token_hex: []const u8, out: *[evt.event_id_size]u8) bool { 2531 if (!isToken(token_hex)) return false; 2532 const file = self.dir.openFile(self.io, token_hex, .{ .mode = .read_only }) catch return false; 2533 defer file.close(self.io); 2534 var storage: [evt.event_id_size]u8 = undefined; 2535 var file_reader = file.reader(self.io, &storage); 2536 file_reader.interface.readSliceAll(out) catch return false; 2537 return true; 2538 } 2539 2540 // revoke a session. a no-op for a missing or malformed token. 2541 pub fn remove(self: SessionStore, token_hex: []const u8) void { 2542 if (!isToken(token_hex)) return; 2543 self.dir.deleteFile(self.io, token_hex) catch {}; 2544 } 2545 2546 // leave `token_hex`'s session for visitors arriving without one. only the 2547 // `try` fixture writes one. the name can't collide with a session token. 2548 pub fn offerAutoLogin(self: SessionStore, token_hex: *const [token_hex_len]u8) !void { 2549 const file = try self.dir.createFile(self.io, auto_login_name, .{}); 2550 defer file.close(self.io); 2551 try file.writeStreamingAll(self.io, token_hex); 2552 } 2553 2554 pub fn autoLogin(self: SessionStore, out: *[token_hex_len]u8) bool { 2555 out.* = self.auto_login orelse return false; 2556 return true; 2557 } 2558 2559 const auto_login_name = "auto-login"; 2560 2561 // a token is exactly token_hex_len hex chars. validating before using the 2562 // value as a path keeps attacker-supplied cookies from escaping the 2563 // sessions dir (e.g. via "/" or ".." bytes). 2564 fn isToken(token_hex: []const u8) bool { 2565 if (token_hex.len != token_hex_len) return false; 2566 for (token_hex) |c| if (!std.ascii.isHex(c)) return false; 2567 return true; 2568 } 2569 }; 2570