╭─────────────╮ │scroll to top│ ╰─────────────╯ ╭ serve_common.zig ─────────────────────────────────────────────╮ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ╰───────────────────────────────────────────────────────────────╯
╭────╮ ╭ clone ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─╮ │http│ ssh │ │ ╰────╯ ╰ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─╯
37 var log_mutex: std.Io.Mutex = .init; 38 39 pub fn logError(io: std.Io, err: *std.Io.Writer, comptime fmt: []const u8, args: anytype) void { 40 log_mutex.lockUncancelable(io); 41 defer log_mutex.unlock(io); 42 err.print(fmt, args) catch return; 43 err.flush() catch {}; 44 } 45 46 // accept connections forever, spawning `handleConn(context, stream)` as a task 47 // for each. the handler owns the stream (it closes it). `name` labels accept 48 // errors in the log. shared by the http, ssh, and web ui listeners. 49 pub fn runListener( 50 io: std.Io, 51 net_server: *std.Io.net.Server, 52 tasks: *std.Io.Group, 53 err: *std.Io.Writer, 54 name: []const u8, 55 context: anytype, 56 comptime handleConn: fn (@TypeOf(context), std.Io.net.Stream) void, 57 ) void { 58 const Context = @TypeOf(context); 59 60 const Conn = struct { 61 context: Context, 62 stream: std.Io.net.Stream, 63 64 fn run(c: @This()) void { 65 handleConn(c.context, c.stream); 66 } 67 }; 68 69 const Listener = struct { 70 io: std.Io, 71 net_server: *std.Io.net.Server, 72 tasks: *std.Io.Group, 73 err: *std.Io.Writer, 74 name: []const u8, 75 context: Context, 76 77 fn run(self: @This()) void { 78 while (true) { 79 const stream = self.net_server.accept(self.io) catch |accept_err| { 80 if (accept_err == error.Canceled) return; 81 logError(self.io, self.err, "{s} accept failed: {s}\n", .{ self.name, @errorName(accept_err) }); 82 continue; 83 }; 84 self.tasks.async(self.io, Conn.run, .{Conn{ .context = self.context, .stream = stream }}); 85 } 86 } 87 }; 88 89 tasks.async(io, Listener.run, .{Listener{ 90 .io = io, 91 .net_server = net_server, 92 .tasks = tasks, 93 .err = err, 94 .name = name, 95 .context = context, 96 }}); 97 } 98