1 files changed, 18 insertions, 7 deletions
diff --git a/src/ninep/screen.zig b/src/ninep/screen.zig
index 5b68b34c..2684aa41 100644
--- a/src/ninep/screen.zig
+++ b/src/ninep/screen.zig
@@ -83,22 +83,33 @@ pub fn releaseSnapshot(p: *Pardes, req: Req) bool {
     return true;
 }
 
-pub fn readListeners(p: *Pardes, req: Req) Reply {
-    var buf: [512]u8 = undefined;
-    var text = std.Io.Writer.fixed(&buf);
+const listeners_cap = 512;
+
+fn listenersText(p: *Pardes, buf: []u8) []const u8 {
+    var text = std.Io.Writer.fixed(buf);
     if (p.fs.socket_path.len != 0)
-        text.print("unix!{s}\n", .{p.fs.socket_path}) catch return Reply.fail(req.tag, E.IO);
+        text.print("unix!{s}\n", .{p.fs.socket_path}) catch return text.buffered();
     for ([_]?std.Io.net.IpAddress{ p.fs.tcp_address, p.fs.quic_address }, [_][]const u8{ "tcp", "quic" }) |maybe, transport| {
         const address = maybe orelse continue;
         switch (address) {
             .ip4 => |ip| text.print("{s}!{d}.{d}.{d}.{d}!{d}\n", .{ transport, ip.bytes[0], ip.bytes[1], ip.bytes[2], ip.bytes[3], ip.port }) catch
-                return Reply.fail(req.tag, E.IO),
+                return text.buffered(),
             .ip6 => |ip| text.print("{s}!{f}!{d}\n", .{ transport, std.Io.net.Ip6Address.Unresolved{ .bytes = ip.bytes, .interface_name = null }, ip.port }) catch
-                return Reply.fail(req.tag, E.IO),
+                return text.buffered(),
         }
     }
+    return text.buffered();
+}
+
+pub fn listenersLen(p: *Pardes) u64 {
+    var buf: [listeners_cap]u8 = undefined;
+    return listenersText(p, &buf).len;
+}
+
+pub fn readListeners(p: *Pardes, req: Req) Reply {
+    var buf: [listeners_cap]u8 = undefined;
     const out = p.fs.stage(p.gpa);
-    out.appendSlice(p.gpa, text.buffered()) catch return Reply.fail(req.tag, E.NOMEM);
+    out.appendSlice(p.gpa, listenersText(p, &buf)) catch return Reply.fail(req.tag, E.NOMEM);
     return tree.stagedReply(p, req);
 }