╭─────────────╮ │scroll to top│ ╰─────────────╯ ╭ web.zig ──────────────────────────────────────────────────────╮ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ╰───────────────────────────────────────────────────────────────╯
╭────╮ ╭ clone ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╮ │http│ ssh │ │ ╰────╯ ╰ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╯
2440 // a cell's link as an href, when it's a web or mail url 2441 fn webLink(link: ?[]const u8) ?[]const u8 { 2442 const url = link orelse return null; 2443 for ([_][]const u8{ "http://", "https://", "mailto:" }) |scheme| { 2444 if (std.ascii.startsWithIgnoreCase(url, scheme)) return url; 2445 } 2446 return null; 2447 } 2448 2449 fn sameLink(a: ?[]const u8, b: ?[]const u8) bool { 2450 if (a) |x| return if (b) |y| std.mem.eql(u8, x, y) else false; 2451 return b == null; 2452 } 2453 2454 fn appendEscapedHtml(allocator: std.mem.Allocator, out: *std.ArrayList(u8), input: []const u8) !void { 2455 for (input) |ch| { 2456 switch (ch) { 2457 '&' => try out.appendSlice(allocator, "&amp;"), 2458 '<' => try out.appendSlice(allocator, "&lt;"), 2459 '>' => try out.appendSlice(allocator, "&gt;"), 2460 '"' => try out.appendSlice(allocator, "&quot;"), 2461 '\'' => try out.appendSlice(allocator, "&#39;"), 2462 else => try out.append(allocator, ch), 2463 } 2464 } 2465 } 2466 2467 // --- helpers -------------------------------------------------------------- 2468 2469 fn getCookieValue(request: *std.http.Server.Request, name: []const u8) ?[]const u8 { 2470 var iter = request.iterateHeaders(); 2471 while (iter.next()) |header| { 2472 if (!std.ascii.eqlIgnoreCase(header.name, "cookie")) continue; 2473 var pairs = std.mem.splitScalar(u8, header.value, ';'); 2474 while (pairs.next()) |pair| { 2475 const trimmed = std.mem.trim(u8, pair, " \t"); 2476 const eq = std.mem.indexOfScalar(u8, trimmed, '=') orelse continue; 2477 if (std.mem.eql(u8, trimmed[0..eq], name)) { 2478 return trimmed[eq + 1 ..]; 2479 } 2480 } 2481 } 2482 return null; 2483 } 2484 2485 fn parseFormField(allocator: std.mem.Allocator, body: []const u8, key: []const u8) !?[]u8 { 2486 var iter = std.mem.splitScalar(u8, body, '&'); 2487 while (iter.next()) |pair| { 2488 const eq = std.mem.indexOfScalar(u8, pair, '=') orelse continue; 2489 if (std.mem.eql(u8, pair[0..eq], key)) { 2490 return try decodeFormValue(allocator, pair[eq + 1 ..]); 2491 } 2492 } 2493 return null; 2494 } 2495 2496 fn readFormBody(request: *std.http.Server.Request, allocator: std.mem.Allocator) ![]u8 { 2497 var buffer: [256]u8 = undefined; 2498 return request.readerExpectNone(&buffer).allocRemaining(allocator, .limited(65536)); 2499 } 2500 2501 fn decodeFormValue(allocator: std.mem.Allocator, encoded: []const u8) ![]u8 { 2502 var out: std.ArrayList(u8) = .empty; 2503 errdefer out.deinit(allocator); 2504 var i: usize = 0; 2505 while (i < encoded.len) : (i += 1) { 2506 switch (encoded[i]) { 2507 '+' => try out.append(allocator, ' '), 2508 '%' => { 2509 if (i + 2 >= encoded.len) return error.InvalidPercentEncoding; 2510 const hi = std.fmt.charToDigit(encoded[i + 1], 16) catch return error.InvalidPercentEncoding; 2511 const lo = std.fmt.charToDigit(encoded[i + 2], 16) catch return error.InvalidPercentEncoding; 2512 try out.append(allocator, hi * 16 + lo); 2513 i += 2; 2514 }, 2515 else => try out.append(allocator, encoded[i]), 2516 } 2517 } 2518 return try out.toOwnedSlice(allocator); 2519 } 2520 2521 // disk-backed web session store. each login mints a random, opaque token 2522 // (not the user's id) that is stored as a file named by the token hex, whose 2523 // contents are the raw user_id bytes. authenticating a request is a lookup of 2524 // the cookie's token; logout deletes the file, revoking the session. there is 2525 // deliberately no expiry — a session lives until logout. form failures use 2526 // files in the flash subdirectory that the redirected GET consumes. the 2527 // directory is cleared whenever the server starts. 2528 pub const SessionStore = struct { 2529 io: std.Io, 2530 dir: std.Io.Dir, 2531 flash_dir: std.Io.Dir, 2532 auto_login: ?[token_hex_len]u8, 2533 2534 pub const token_hex_len = evt.event_id_size * 2; 2535 2536 pub fn init(io: std.Io, data_dir: std.Io.Dir) !SessionStore { 2537 try data_dir.createDirPath(io, "sessions"); 2538 const dir = try data_dir.openDir(io, "sessions", .{}); 2539 errdefer dir.close(io); 2540 try dir.deleteTree(io, "flash"); 2541 try dir.createDirPath(io, "flash"); 2542 const flash_dir = try dir.openDir(io, "flash", .{}); 2543 errdefer flash_dir.close(io); 2544 const auto_login = blk: { 2545 const file = dir.openFile(io, auto_login_name, .{ .mode = .read_only }) catch break :blk null; 2546 defer file.close(io); 2547 var token: [token_hex_len]u8 = undefined; 2548 var storage: [token_hex_len]u8 = undefined; 2549 var reader = file.reader(io, &storage); 2550 reader.interface.readSliceAll(&token) catch break :blk null; 2551 break :blk token; 2552 }; 2553 return .{ .io = io, .dir = dir, .flash_dir = flash_dir, .auto_login = auto_login }; 2554 } 2555 2556 pub fn deinit(self: SessionStore) void { 2557 self.flash_dir.close(self.io); 2558 self.dir.close(self.io); 2559 } 2560 2561 // mint a new session for user_id, returning the token hex to set as the 2562 // cookie value. 2563 pub fn create(self: SessionStore, user_id: *const [evt.event_id_size]u8) ![token_hex_len]u8 { 2564 var token: [evt.event_id_size]u8 = undefined; 2565 self.io.random(&token); 2566 const token_hex = std.fmt.bytesToHex(token, .lower); 2567 const file = try self.dir.createFile(self.io, &token_hex, .{}); 2568 defer file.close(self.io); 2569 try file.writeStreamingAll(self.io, user_id); 2570 return token_hex; 2571 } 2572 2573 fn createFlash(self: SessionStore, allocator: std.mem.Allocator, value: ui.Session.FormFeedback) ![token_hex_len]u8 { 2574 var token: [evt.event_id_size]u8 = undefined; 2575 self.io.random(&token); 2576 const token_hex = std.fmt.bytesToHex(token, .lower); 2577 const file = try self.flash_dir.createFile(self.io, &token_hex, .{}); 2578 errdefer self.flash_dir.deleteFile(self.io, &token_hex) catch {}; 2579 defer file.close(self.io); 2580 2581 var json: std.Io.Writer.Allocating = .init(allocator); 2582 defer json.deinit(); 2583 try std.json.Stringify.value(value, .{}, &json.writer); 2584 try file.writeStreamingAll(self.io, json.written()); 2585 return token_hex; 2586 } 2587 2588 fn takeFlash(self: SessionStore, allocator: std.mem.Allocator, token_hex: []const u8) ?ui.Session.FormFeedback { 2589 if (!isToken(token_hex)) return null; 2590 const file = self.flash_dir.openFile(self.io, token_hex, .{ .mode = .read_only }) catch return null; 2591 defer self.flash_dir.deleteFile(self.io, token_hex) catch {}; 2592 defer file.close(self.io); 2593 2594 var storage: [4096]u8 = undefined; 2595 var reader = file.reader(self.io, &storage); 2596 const bytes = reader.interface.allocRemaining(allocator, .limited(512 * 1024)) catch return null; 2597 return std.json.parseFromSliceLeaky(ui.Session.FormFeedback, allocator, bytes, .{}) catch null; 2598 } 2599 2600 // resolve a cookie's token to its user_id. returns false (logged out) for a 2601 // missing, malformed, or unknown token. 2602 pub fn lookup(self: SessionStore, token_hex: []const u8, out: *[evt.event_id_size]u8) bool { 2603 if (!isToken(token_hex)) return false; 2604 const file = self.dir.openFile(self.io, token_hex, .{ .mode = .read_only }) catch return false; 2605 defer file.close(self.io); 2606 var storage: [evt.event_id_size]u8 = undefined; 2607 var file_reader = file.reader(self.io, &storage); 2608 file_reader.interface.readSliceAll(out) catch return false; 2609 return true; 2610 } 2611 2612 // revoke a session. a no-op for a missing or malformed token. 2613 pub fn remove(self: SessionStore, token_hex: []const u8) void { 2614 if (!isToken(token_hex)) return; 2615 self.dir.deleteFile(self.io, token_hex) catch {}; 2616 } 2617 2618 // leave `token_hex`'s session for visitors arriving without one. only the 2619 // `try` fixture writes one. the name can't collide with a session token. 2620 pub fn offerAutoLogin(self: SessionStore, token_hex: *const [token_hex_len]u8) !void { 2621 const file = try self.dir.createFile(self.io, auto_login_name, .{}); 2622 defer file.close(self.io); 2623 try file.writeStreamingAll(self.io, token_hex); 2624 } 2625 2626 pub fn autoLogin(self: SessionStore, out: *[token_hex_len]u8) bool { 2627 out.* = self.auto_login orelse return false; 2628 return true; 2629 } 2630 2631 const auto_login_name = "auto-login"; 2632 2633 // a token is exactly token_hex_len hex chars. validating before using the 2634 // value as a path keeps attacker-supplied cookies from escaping the 2635 // sessions dir (e.g. via "/" or ".." bytes). 2636 fn isToken(token_hex: []const u8) bool { 2637 if (token_hex.len != token_hex_len) return false; 2638 for (token_hex) |c| if (!std.ascii.isHex(c)) return false; 2639 return true; 2640 } 2641 }; 2642