From 448e900004053012e63078682e34b48799050794 Mon Sep 17 00:00:00 2001 From: Robby Zambito Date: Fri, 4 Apr 2025 22:09:06 -0400 Subject: Break relay into a specific program --- src/saprus_relay.zig | 71 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 71 insertions(+) create mode 100644 src/saprus_relay.zig (limited to 'src/saprus_relay.zig') diff --git a/src/saprus_relay.zig b/src/saprus_relay.zig new file mode 100644 index 0000000..0df66e9 --- /dev/null +++ b/src/saprus_relay.zig @@ -0,0 +1,71 @@ +const is_debug = builtin.mode == .Debug; + +/// This creates a debug allocator that can only be referenced in debug mode. +/// You should check for is_debug around every reference to dba. +var dba: DebugAllocator = + if (is_debug) + DebugAllocator.init + else + @compileError("Should not use debug allocator in release mode"); + +pub fn main() !void { + defer if (is_debug) { + _ = dba.deinit(); + }; + + var gpa = if (is_debug) dba.allocator() else std.heap.smp_allocator; + + const args = try std.process.argsAlloc(gpa); + defer std.process.argsFree(gpa, args); + + var message = ArrayList(u8).init(gpa); + defer message.deinit(); + + for (args[1..], 0..) |arg, i| { + try message.appendSlice(arg); + if (i < args.len - 1) + try message.append(' '); + } + + const msg = SaprusMessage{ + .relay = .{ + .header = .{ .dest = .{ 255, 255, 255, 255 } }, + .payload = if (message.items.len > 0) message.items else "Hello darkness my old friend", + }, + }; + + const msg_bytes = try msg.toBytes(gpa); + defer gpa.free(msg_bytes); + + try network.init(); + defer network.deinit(); + + var sock = try network.Socket.create(.ipv4, .udp); + defer sock.close(); + + try sock.setBroadcast(true); + + // Bind to 0.0.0.0:0 + const bind_addr = network.EndPoint{ + .address = network.Address{ .ipv4 = network.Address.IPv4.any }, + .port = 0, + }; + + const dest_addr = network.EndPoint{ + .address = network.Address{ .ipv4 = network.Address.IPv4.broadcast }, + .port = 8888, + }; + + try sock.bind(bind_addr); + + _ = try sock.sendTo(dest_addr, msg_bytes); +} + +const builtin = @import("builtin"); +const std = @import("std"); +const DebugAllocator = std.heap.DebugAllocator(.{}); +const ArrayList = std.ArrayList; + +const network = @import("network"); + +const SaprusMessage = @import("./saprus_message.zig").SaprusMessage; -- cgit