╭─────────────╮ │scroll to top│ ╰─────────────╯ ╭ web.zig ──────────────────────────────────────────────────────╮ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ╰───────────────────────────────────────────────────────────────╯
╭────╮ ╭ clone ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─╮ │http│ ssh │ │ ╰────╯ ╰ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─╯
2849 } else blk: { 2850 const v: u8 = 8 + (n - 232) * 10; 2851 break :blk .{ .r = v, .g = v, .b = v }; 2852 }, 2853 }; 2854 } 2855 2856 // builds an HTML ` style="color:…;background-color:…"` attribute from a 2857 // cell's style into `buf`, omitting whichever color is unset. an inverted 2858 // cell swaps the two, falling back to the page's --fg/--bg variables for 2859 // whichever side is unset. text attributes follow the colors. returns an 2860 // empty slice when nothing is set. 2861 fn styleAttr(buf: []u8, style: Grid.Style) []const u8 { 2862 if (style.eql(.{})) return buf[0..0]; 2863 const prefix = " style=\""; 2864 @memcpy(buf[0..prefix.len], prefix); 2865 var i: usize = prefix.len; 2866 const inverted = style.inverted; 2867 i = cssColor(buf, i, "color", if (inverted) style.bg else style.fg, if (inverted) "var(--bg)" else null) orelse return buf[0..0]; 2868 i = cssColor(buf, i, "background-color", if (inverted) style.fg else style.bg, if (inverted) "var(--fg)" else null) orelse return buf[0..0]; 2869 const decoration: []const u8 = if (style.underline and style.strikethrough) 2870 "text-decoration:underline line-through;" 2871 else if (style.underline) 2872 "text-decoration:underline;" 2873 else if (style.strikethrough) 2874 "text-decoration:line-through;" 2875 else 2876 ""; 2877 const attributes = std.fmt.bufPrint(buf[i..], "{s}{s}{s}{s}", .{ 2878 if (style.bold) "font-weight:bold;" else "", 2879 if (style.italic) "font-style:italic;" else "", 2880 decoration, 2881 if (style.dim) "opacity:.6;" else "", 2882 }) catch return buf[0..0]; 2883 i += attributes.len; 2884 if (i >= buf.len) return buf[0..0]; 2885 buf[i] = '"'; 2886 return buf[0 .. i + 1]; 2887 } 2888 2889 // append `prop:value;` at buf[i..] for the color, or for the fallback when 2890 // the color is unset (nothing when both are). returns the new end, or null 2891 // when it doesn't fit. 2892 fn cssColor(buf: []u8, i: usize, prop: []const u8, color: ?Grid.Color, fallback: ?[]const u8) ?usize { 2893 const s = if (color) |c| blk: { 2894 const v = colorRgb(c); 2895 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; 2896 } else if (fallback) |f| 2897 std.fmt.bufPrint(buf[i..], "{s}:{s};", .{ prop, f }) catch return null 2898 else 2899 return i; 2900 return i + s.len; 2901 } 2902 2903 // a cell's link as an href, when it's a page path or a web or mail url 2904 fn webLink(link: ?[]const u8) ?[]const u8 { 2905 const url = link orelse return null; 2906 // a page path, not a protocol-relative url 2907 if (url.len > 1 and url[0] == '/' and url[1] != '/') return url; 2908 for ([_][]const u8{ "http://", "https://", "mailto:" }) |scheme| { 2909 if (std.ascii.startsWithIgnoreCase(url, scheme)) return url; 2910 } 2911 return null; 2912 } 2913 2914 fn sameLink(a: ?[]const u8, b: ?[]const u8) bool { 2915 if (a) |x| return if (b) |y| std.mem.eql(u8, x, y) else false; 2916 return b == null; 2917 } 2918 2919 fn appendEscapedHtml(allocator: std.mem.Allocator, out: *std.ArrayList(u8), input: []const u8) !void { 2920 for (input) |ch| { 2921 switch (ch) { 2922 '&' => try out.appendSlice(allocator, "&amp;"), 2923 '<' => try out.appendSlice(allocator, "&lt;"), 2924 '>' => try out.appendSlice(allocator, "&gt;"), 2925 '"' => try out.appendSlice(allocator, "&quot;"), 2926 '\'' => try out.appendSlice(allocator, "&#39;"), 2927 else => try out.append(allocator, ch), 2928 } 2929 } 2930 } 2931 2932 // --- helpers -------------------------------------------------------------- 2933 2934 fn getCookieValue(request: *std.http.Server.Request, name: []const u8) ?[]const u8 { 2935 var iter = request.iterateHeaders(); 2936 while (iter.next()) |header| { 2937 if (!std.ascii.eqlIgnoreCase(header.name, "cookie")) continue; 2938 var pairs = std.mem.splitScalar(u8, header.value, ';'); 2939 while (pairs.next()) |pair| { 2940 const trimmed = std.mem.trim(u8, pair, " \t"); 2941 const eq = std.mem.indexOfScalar(u8, trimmed, '=') orelse continue; 2942 if (std.mem.eql(u8, trimmed[0..eq], name)) { 2943 return trimmed[eq + 1 ..]; 2944 } 2945 } 2946 } 2947 return null; 2948 } 2949 2950 fn parseFormField(allocator: std.mem.Allocator, body: []const u8, key: []const u8) !?[]u8 { 2951 var iter = std.mem.splitScalar(u8, body, '&'); 2952 while (iter.next()) |pair| { 2953 const eq = std.mem.indexOfScalar(u8, pair, '=') orelse continue; 2954 if (std.mem.eql(u8, pair[0..eq], key)) { 2955 return try decodeFormValue(allocator, pair[eq + 1 ..]); 2956 } 2957 } 2958 return null; 2959 } 2960 2961 fn readFormBody(request: *std.http.Server.Request, allocator: std.mem.Allocator) ![]u8 { 2962 var buffer: [256]u8 = undefined; 2963 return request.readerExpectNone(&buffer).allocRemaining(allocator, .limited(65536)); 2964 } 2965 2966 fn decodeFormValue(allocator: std.mem.Allocator, encoded: []const u8) ![]u8 { 2967 var out: std.ArrayList(u8) = .empty; 2968 errdefer out.deinit(allocator); 2969 var i: usize = 0; 2970 while (i < encoded.len) : (i += 1) { 2971 switch (encoded[i]) { 2972 '+' => try out.append(allocator, ' '), 2973 '%' => { 2974 if (i + 2 >= encoded.len) return error.InvalidPercentEncoding; 2975 const hi = std.fmt.charToDigit(encoded[i + 1], 16) catch return error.InvalidPercentEncoding; 2976 const lo = std.fmt.charToDigit(encoded[i + 2], 16) catch return error.InvalidPercentEncoding; 2977 try out.append(allocator, hi * 16 + lo); 2978 i += 2; 2979 }, 2980 else => try out.append(allocator, encoded[i]), 2981 } 2982 } 2983 return try out.toOwnedSlice(allocator); 2984 } 2985 2986 // disk-backed web session store. each login mints a random, opaque token 2987 // (not the user's id) that is stored as a file named by the token hex, whose 2988 // contents are the raw user_id bytes. authenticating a request is a lookup of 2989 // the cookie's token; logout deletes the file, revoking the session. there is 2990 // deliberately no expiry — a session lives until logout. form failures use 2991 // files in the flash subdirectory that the redirected GET consumes. the 2992 // directory is cleared whenever the server starts. 2993 pub const SessionStore = struct { 2994 io: std.Io, 2995 dir: std.Io.Dir, 2996 flash_dir: std.Io.Dir, 2997 auto_login: ?[token_hex_len]u8, 2998 2999 pub const token_hex_len = evt.event_id_size * 2; 3000 3001 pub fn init(io: std.Io, data_dir: std.Io.Dir) !SessionStore { 3002 try data_dir.createDirPath(io, "sessions"); 3003 const dir = try data_dir.openDir(io, "sessions", .{}); 3004 errdefer dir.close(io); 3005 try dir.deleteTree(io, "flash"); 3006 try dir.createDirPath(io, "flash"); 3007 const flash_dir = try dir.openDir(io, "flash", .{}); 3008 errdefer flash_dir.close(io); 3009 const auto_login = blk: { 3010 const file = dir.openFile(io, auto_login_name, .{ .mode = .read_only }) catch break :blk null; 3011 defer file.close(io); 3012 var token: [token_hex_len]u8 = undefined; 3013 var storage: [token_hex_len]u8 = undefined; 3014 var reader = file.reader(io, &storage); 3015 reader.interface.readSliceAll(&token) catch break :blk null; 3016 break :blk token; 3017 }; 3018 return .{ .io = io, .dir = dir, .flash_dir = flash_dir, .auto_login = auto_login }; 3019 } 3020 3021 pub fn deinit(self: SessionStore) void { 3022 self.flash_dir.close(self.io); 3023 self.dir.close(self.io); 3024 } 3025 3026 // mint a new session for user_id, returning the token hex to set as the 3027 // cookie value. 3028 pub fn create(self: SessionStore, user_id: *const [evt.event_id_size]u8) ![token_hex_len]u8 { 3029 var token: [evt.event_id_size]u8 = undefined; 3030 self.io.random(&token); 3031 const token_hex = std.fmt.bytesToHex(token, .lower); 3032 const file = try self.dir.createFile(self.io, &token_hex, .{}); 3033 defer file.close(self.io); 3034 try file.writeStreamingAll(self.io, user_id); 3035 return token_hex; 3036 } 3037 3038 fn createFlash(self: SessionStore, allocator: std.mem.Allocator, value: ui.Session.FormFeedback) ![token_hex_len]u8 { 3039 var token: [evt.event_id_size]u8 = undefined; 3040 self.io.random(&token); 3041 const token_hex = std.fmt.bytesToHex(token, .lower); 3042 const file = try self.flash_dir.createFile(self.io, &token_hex, .{}); 3043 errdefer self.flash_dir.deleteFile(self.io, &token_hex) catch {}; 3044 defer file.close(self.io); 3045 3046 var json: std.Io.Writer.Allocating = .init(allocator); 3047 defer json.deinit(); 3048 try std.json.Stringify.value(value, .{}, &json.writer); 3049 try file.writeStreamingAll(self.io, json.written()); 3050 return token_hex; 3051 } 3052 3053 fn takeFlash(self: SessionStore, allocator: std.mem.Allocator, token_hex: []const u8) ?ui.Session.FormFeedback { 3054 if (!isToken(token_hex)) return null; 3055 const file = self.flash_dir.openFile(self.io, token_hex, .{ .mode = .read_only }) catch return null; 3056 defer self.flash_dir.deleteFile(self.io, token_hex) catch {}; 3057 defer file.close(self.io); 3058 3059 var storage: [4096]u8 = undefined; 3060 var reader = file.reader(self.io, &storage); 3061 const bytes = reader.interface.allocRemaining(allocator, .limited(512 * 1024)) catch return null; 3062 return std.json.parseFromSliceLeaky(ui.Session.FormFeedback, allocator, bytes, .{}) catch null; 3063 } 3064 3065 // resolve a cookie's token to its user_id. returns false (logged out) for a 3066 // missing, malformed, or unknown token. 3067 pub fn lookup(self: SessionStore, token_hex: []const u8, out: *[evt.event_id_size]u8) bool { 3068 if (!isToken(token_hex)) return false; 3069 const file = self.dir.openFile(self.io, token_hex, .{ .mode = .read_only }) catch return false; 3070 defer file.close(self.io); 3071 var storage: [evt.event_id_size]u8 = undefined; 3072 var file_reader = file.reader(self.io, &storage); 3073 file_reader.interface.readSliceAll(out) catch return false; 3074 return true; 3075 } 3076 3077 // revoke a session. a no-op for a missing or malformed token. 3078 pub fn remove(self: SessionStore, token_hex: []const u8) void { 3079 if (!isToken(token_hex)) return; 3080 self.dir.deleteFile(self.io, token_hex) catch {}; 3081 } 3082 3083 // leave `token_hex`'s session for visitors arriving without one. only the 3084 // `try` fixture writes one. the name can't collide with a session token. 3085 pub fn offerAutoLogin(self: SessionStore, token_hex: *const [token_hex_len]u8) !void { 3086 const file = try self.dir.createFile(self.io, auto_login_name, .{}); 3087 defer file.close(self.io); 3088 try file.writeStreamingAll(self.io, token_hex); 3089 } 3090 3091 pub fn autoLogin(self: SessionStore, out: *[token_hex_len]u8) bool { 3092 out.* = self.auto_login orelse return false; 3093 return true; 3094 } 3095 3096 const auto_login_name = "auto-login"; 3097 3098 // a token is exactly token_hex_len hex chars. validating before using the 3099 // value as a path keeps attacker-supplied cookies from escaping the 3100 // sessions dir (e.g. via "/" or ".." bytes). 3101 fn isToken(token_hex: []const u8) bool { 3102 if (token_hex.len != token_hex_len) return false; 3103 for (token_hex) |c| if (!std.ascii.isHex(c)) return false; 3104 return true; 3105 } 3106 }; 3107