From ce28b72f3b0fafc4579147e34fdae73fad120631 Mon Sep 17 00:00:00 2001 From: Philippe Pittoli Date: Wed, 4 Jan 2023 11:55:47 +0100 Subject: [PATCH] Pongd. --- zig-impl/src/pongd.zig | 112 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 112 insertions(+) create mode 100644 zig-impl/src/pongd.zig diff --git a/zig-impl/src/pongd.zig b/zig-impl/src/pongd.zig new file mode 100644 index 0000000..28409a5 --- /dev/null +++ b/zig-impl/src/pongd.zig @@ -0,0 +1,112 @@ +const std = @import("std"); +const hexdump = @import("./hexdump.zig"); +const net = std.net; +const fmt = std.fmt; +const os = std.os; + +const ipc = @import("./main.zig"); +const Message = ipc.Message; + +// Import send_fd this way in order to produce docs for exchange-fd functions. +const exchange_fd = @import("./exchange-fd.zig"); +const send_fd = exchange_fd.send_fd; + +const builtin = @import("builtin"); +const native_os = builtin.target.os.tag; +const print = std.debug.print; +const testing = std.testing; +const print_eq = @import("./util.zig").print_eq; + +fn create_service() !void { + const config = .{.safety = true}; + var gpa = std.heap.GeneralPurposeAllocator(config){}; + defer _ = gpa.deinit(); + const allocator = gpa.allocator(); + + var ctx = try ipc.Context.init(allocator); + defer ctx.deinit(); // There. Can't leak. Isn't Zig wonderful? + + // SERVER SIDE: creating a service. + _ = try ctx.server_init("pong"); + + // signal handler, to quit when asked + const S = struct { + var should_quit: bool = false; + + fn handler(sig: i32, info: *const os.siginfo_t, _: ?*const anyopaque) callconv(.C) void { + print ("A signal has been received: {}\n", .{sig}); + // Check that we received the correct signal. + switch (native_os) { + .netbsd => { + if (sig != os.SIG.HUP or sig != info.info.signo) + return; + }, + else => { + if (sig != os.SIG.HUP and sig != info.signo) + return; + }, + } + should_quit = true; + } + }; + + var sa = os.Sigaction{ + .handler = .{ .sigaction = &S.handler }, + .mask = os.empty_sigset, // Do not mask any signal. + .flags = os.SA.SIGINFO, + }; + + // Quit on SIGHUP (kill -1). + try os.sigaction(os.SIG.HUP, &sa, null); + + var some_event: ipc.Event = undefined; + ctx.timer = 10000; // 10 seconds + var count: u32 = 0; + while(! S.should_quit) { + some_event = try ctx.wait_event(); + switch (some_event.t) { + .TIMER => { + print("\rTimer! ({})", .{count}); + count += 1; + }, + + .CONNECTION => { + print("New connection: {} so far!\n", .{ctx.pollfd.items.len}); + }, + + .DISCONNECTION => { + print("User {} disconnected, {} remainaing.\n" + , .{some_event.origin, ctx.pollfd.items.len}); + }, + + .MESSAGE => { + if (some_event.m) |m| { + print("New message: {}\n", .{m}); + print("Let's echo it...\n", .{}); + try ctx.schedule(m); + } + else { + print("Error while receiving new message.\n", .{}); + print("Ignoring...\n", .{}); + } + }, + + .TX => { + print("Message sent.\n", .{}); + }, + + else => { + print("Error: unexpected event: {}\n", .{some_event}); + print("Let's suicide.\n", .{}); + break; + }, + } + } + + print("Goodbye\n", .{}); +} + +pub fn main() !u8 { + try create_service(); + return 0; +}