Use Client as var type instead of singleton

Use this from the relay message

build.zig.zon

@@ -45,8 +45,8 @@
             .hash = "clap-0.10.0-oBajB434AQBDh-Ei3YtoKIRxZacVPF1iSwp3IX_ZB8f0",
         },
         .gatorcat = .{
-            .url = "git+https://github.com/kj4tmp/gatorcat#bb1847f6c95852e7a0ec8c07870a948c171d5f98",
+            .url = "git+https://github.com/kj4tmp/gatorcat.git#0a97b666677501db4939e3e8245f88a19e015893",
-            .hash = "gatorcat-0.3.2-WcrpTf1mBwDrmPaIhKCfLJO064v8Sjjn7DBq4CKZSgHH",
+            .hash = "gatorcat-0.3.4-WcrpTcleBwCta_9TjomuIGb3bdg2Pke_FXI_WkMTEivH",
         },
     },
     .paths = .{

src/Client.zig

@@ -1,28 +1,86 @@
 const base64Enc = std.base64.Base64Encoder.init(std.base64.standard_alphabet_chars, '=');
 const base64Dec = std.base64.Base64Decoder.init(std.base64.standard_alphabet_chars, '=');
 
-var rand: ?Random = null;
+rand: Random,
+socket: gcat.nic.RawSocket,
 
-pub fn init() !void {
+const Self = @This();
+
+const max_message_size = 2048;
+
+pub fn init(interface_name: [:0]const u8) !Self {
     var prng = Random.DefaultPrng.init(blk: {
         var seed: u64 = undefined;
         try posix.getrandom(mem.asBytes(&seed));
         break :blk seed;
     });
-    rand = prng.random();
+    const rand = prng.random();
-    try network.init();
+
+    const socket: gcat.nic.RawSocket = try .init(interface_name);
+
+    return .{
+        .rand = rand,
+        .socket = socket,
+    };
 }
 
-pub fn deinit() void {
+pub fn deinit(self: *Self) void {
-    network.deinit();
+    self.socket.deinit();
 }
 
-fn broadcastSaprusMessage(msg: *SaprusMessage, udp_port: u16) !void {
+/// Used for relay messages and connection handshake.
+/// Assumes Client .init has been called.
+fn broadcastInitialInterestMessage(self: *Self, msg_bytes: []align(@alignOf(SaprusMessage)) u8) !void {
+    var packet_bytes: [max_message_size]u8 = comptime blk: {
+        var b: [max_message_size]u8 = @splat(0);
+
+        // Destination MAC addr to FF:FF:FF:FF:FF:FF
+        for (0..6) |i| {
+            b[i] = 0xff;
+        }
+
+        // Set Ethernet type to IPv4
+        b[0x0c] = 0x08;
+        b[0x0d] = 0x00;
+
+        // Set IPv4 version to 4
+        b[0x0e] = 0x45;
+
+        // Destination broadcast
+        for (0x1e..0x22) |i| {
+            b[i] = 0xff;
+        }
+
+        // Set TTL
+        b[0x16] = 0x40;
+
+        // Set IPv4 protocol to UDP
+        b[0x17] = 0x11;
+
+        // Set interest filter value to 8888.
+        b[0x24] = 0x22;
+        b[0x25] = 0xb8;
+        break :blk b;
+    };
+    var msg: *SaprusMessage = try .bytesAsValue(msg_bytes);
+    try msg.networkFromNativeEndian();
+    defer msg.nativeFromNetworkEndian() catch unreachable;
+
+    // The byte position within the packet that the saprus message starts at.
+    const saprus_start_byte = 42;
+    @memcpy(packet_bytes[saprus_start_byte .. saprus_start_byte + msg_bytes.len], msg_bytes);
+
+    try self.socket.linkLayer().send(packet_bytes[0 .. saprus_start_byte + msg_bytes.len]);
+}
+
+// fn broadcastSaprusMessage(msg_bytes: []align(@alignOf(SaprusMessage)) u8) !void {}
+
+fn broadcastSaprusMessage(msg_bytes: []align(@alignOf(SaprusMessage)) u8, udp_port: u16) !void {
     if (false) {
         var foo: gcat.nic.RawSocket = try .init("enp7s0"); // /proc/net/dev
         defer foo.deinit();
     }
-    const msg_bytes = msg.asBytes();
+    const msg: *SaprusMessage = try .bytesAsValue(msg_bytes);
     try msg.networkFromNativeEndian();
     defer msg.nativeFromNetworkEndian() catch unreachable;
 
@@ -44,52 +102,57 @@ fn broadcastSaprusMessage(msg: *SaprusMessage, udp_port: u16) !void {
 
     try sock.bind(bind_addr);
 
+    std.debug.print("{x}\n", .{msg_bytes});
+
     _ = try sock.sendTo(dest_addr, msg_bytes);
 }
 
-pub fn sendRelay(payload: []const u8, dest: [4]u8, allocator: Allocator) !void {
+pub fn sendRelay(self: *Self, payload: []const u8, dest: [4]u8) !void {
-    const msg: *SaprusMessageNew(.relay) = try .init(
+    var buf: [max_message_size]u8 align(@alignOf(SaprusMessage)) = undefined;
-        allocator,
+    const msg_bytes = buf[0..try SaprusMessage.calcSize(
-        @intCast(base64Enc.calcSize(payload.len)),
+        .relay,
-    );
+        base64Enc.calcSize(payload.len),
-    defer msg.deinit(allocator);
+    )];
-    msg.dest = dest;
+    const msg: *SaprusMessage = .init(.relay, msg_bytes);
-    _ = base64Enc.encode(msg.getPayload(), payload);
 
-    try broadcastSaprusMessage(try SaprusMessage.bytesAsValue(msg.asBytes()), 8888);
+    const relay = (try msg.getSaprusTypePayload()).relay;
+    relay.dest = dest;
+    _ = base64Enc.encode(relay.getPayload(), payload);
+
+    try self.broadcastInitialInterestMessage(msg_bytes);
 }
 
-fn randomPort() u16 {
+fn randomPort(self: Self) u16 {
-    var p: u16 = 0;
+    return self.rand.intRangeAtMost(u16, 1024, 65000);
-    if (rand) |r| {
-        p = r.intRangeAtMost(u16, 1024, 65000);
-    } else unreachable;
-
-    return p;
 }
 
-pub fn sendInitialConnection(payload: []const u8, initial_port: u16, allocator: Allocator) !*SaprusMessage {
+pub fn sendInitialConnection(
-    const dest_port = randomPort();
+    self: Self,
-    const msg: *SaprusMessage = try .init(allocator, .connection, @intCast(payload.len));
+    payload: []const u8,
-    defer msg.deinit(allocator);
+    output_bytes: []align(@alignOf(SaprusMessage)) u8,
+    initial_port: u16,
+) !*SaprusMessage {
+    const dest_port = self.randomPort();
+    const msg_bytes = output_bytes[0..try SaprusMessage.calcSize(
+        .connection,
+        base64Enc.calcSize(payload.len),
+    )];
+    const msg: *SaprusMessage = .init(.connection, msg_bytes);
+
     const connection = (try msg.getSaprusTypePayload()).connection;
     connection.src_port = initial_port;
     connection.dest_port = dest_port;
-    @memcpy(connection.getPayload(), payload);
+    _ = base64Enc.encode(connection.getPayload(), payload);
 
-    try broadcastSaprusMessage(msg, 8888);
+    try broadcastSaprusMessage(msg_bytes, 8888);
 
     return msg;
 }
 
-pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusConnection {
+pub fn connect(self: Self, payload: []const u8) !?SaprusConnection {
-    var initial_port: u16 = 0;
+    const initial_port = self.randomPort();
-    if (rand) |r| {
-        initial_port = r.intRangeAtMost(u16, 1024, 65000);
-    } else unreachable;
 
-    var initial_conn_res: ?SaprusMessage = null;
+    var initial_conn_res: ?*SaprusMessage = null;
-    errdefer if (initial_conn_res) |*c| c.deinit(allocator);
 
     var sock = try network.Socket.create(.ipv4, .udp);
     defer sock.close();
@@ -104,17 +167,17 @@ pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusConnection {
     try sock.setReadTimeout(1 * std.time.us_per_s);
     try sock.bind(bind_addr);
 
-    const msg = try sendInitialConnection(payload, initial_port, allocator);
+    var sent_msg_bytes: [max_message_size]u8 align(@alignOf(SaprusMessage)) = undefined;
+    const msg = try self.sendInitialConnection(payload, &sent_msg_bytes, initial_port);
 
-    var response_buf: [4096]u8 align(4) = @splat(0);
+    var response_buf: [max_message_size]u8 align(@alignOf(SaprusMessage)) = undefined;
     _ = try sock.receive(&response_buf); // Ignore message that I sent.
     const len = try sock.receive(&response_buf);
 
-    std.debug.print("response bytes: {x}\n", .{response_buf});
+    initial_conn_res = try .networkBytesAsValue(response_buf[0..len]);
-    initial_conn_res = (try SaprusMessage.bytesAsValue(response_buf[0..len])).*;
 
     // Complete handshake after awaiting response
-    try broadcastSaprusMessage(msg, randomPort());
+    try broadcastSaprusMessage(msg.asBytes(), self.randomPort());
 
     if (false) {
         return initial_conn_res.?;
@@ -123,7 +186,6 @@ pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusConnection {
 }
 
 const SaprusMessage = @import("message.zig").Message;
-const SaprusMessageNew = @import("message.zig").MessageNew;
 const SaprusConnection = @import("Connection.zig");
 
 const std = @import("std");
@@ -133,5 +195,3 @@ const mem = std.mem;
 
 const network = @import("network");
 const gcat = @import("gatorcat");
-
-const Allocator = mem.Allocator;

src/main.zig

@@ -42,24 +42,23 @@ pub fn main() !void {
     };
     defer res.deinit();
 
-    try SaprusClient.init();
-    defer SaprusClient.deinit();
-
     if (res.args.help != 0) {
         return clap.help(std.io.getStdErr().writer(), clap.Help, &params, .{});
     }
 
+    var client = try SaprusClient.init("enp7s0");
+    defer client.deinit();
+
     if (res.args.relay) |r| {
         const dest = parseDest(res.args.dest);
-        try SaprusClient.sendRelay(
+        try client.sendRelay(
             if (r.len > 0) r else "Hello darkness my old friend",
             dest,
-            gpa,
         );
         // std.debug.print("Sent: {s}\n", .{r});
         return;
     } else if (res.args.connect) |c| {
-        _ = SaprusClient.connect(if (c.len > 0) c else "Hello darkness my old friend", gpa) catch |err| switch (err) {
+        _ = client.connect(if (c.len > 0) c else "Hello darkness my old friend") catch |err| switch (err) {
             error.WouldBlock => null,
             else => return err,
         };

src/message.zig

@@ -20,130 +20,14 @@ pub const ConnectionOptions = packed struct(u8) {
     opt8: bool = false,
 };
 
-pub const Error = error{
+pub const MessageTypeError = error{
     NotImplementedSaprusType,
     UnknownSaprusType,
+};
+pub const MessageParseError = MessageTypeError || error{
     InvalidMessage,
 };
 
-pub fn MessageNew(comptime packet_type: PacketType) type {
-    comptime {
-        if (packet_type == .file_transfer)
-            @compileError("File transfer not implemented");
-        if (packet_type != .relay and packet_type != .connection)
-            @compileError("Unkown message type");
-    }
-
-    return packed struct {
-        const Self = @This();
-        const SelfBytes = []align(@alignOf(Self)) u8;
-
-        const Relay = struct {
-            pub fn getPayload(self: *Self) []u8 {
-                return @as([*]align(@alignOf(Self)) u8, @ptrCast(&self.payload))[0 .. self.length - 4];
-            }
-        };
-        const Connection = packed struct {
-            pub fn getPayload(self: Self) []u8 {
-                return @as([*]u8, &self.payload)[0 .. self.length - 4];
-            }
-        };
-
-        type: PacketType = packet_type,
-        length: u16,
-
-        // Relay
-        dest: if (packet_type == .relay) @Vector(4, u8) else void,
-
-        // Connection
-        src_port: if (packet_type == .connection) u16 else void, // random number > 1024
-        dest_port: if (packet_type == .connection) u16 else void, // random number > 1024
-        seq_num: if (packet_type == .connection) u32 else void,
-        msg_id: if (packet_type == .connection) u32 else void,
-        reserved: if (packet_type == .connection) u8 else void,
-        options: if (packet_type == .connection) ConnectionOptions else void = if (packet_type == .connection) .{} else {},
-
-        // Relay or Connection
-        payload: switch (packet_type) {
-            .relay, .connection => void,
-            else => noreturn,
-        },
-
-        pub usingnamespace switch (packet_type) {
-            .relay => Relay,
-            .connection => Connection,
-            .file_transfer => @compileError("File Transfer message type not implemented"),
-            else => @compileError("Unknown message type"),
-        };
-
-        pub fn init(allocator: Allocator, payload_len: u16) !*Self {
-            const size = payload_len + @sizeOf(Self);
-            const bytes = try allocator.alignedAlloc(u8, @alignOf(Self), size);
-            const res: *Self = @ptrCast(bytes.ptr);
-            res.type = packet_type;
-            res.length = payload_len;
-            return res;
-        }
-
-        pub fn deinit(self: *Self, allocator: Allocator) void {
-            allocator.free(self.asBytes());
-        }
-
-        pub fn nativeFromNetworkEndian(self: *Self) void {
-            self.type = @enumFromInt(bigToNative(
-                @typeInfo(@TypeOf(self.type)).@"enum".tag_type,
-                @intFromEnum(self.type),
-            ));
-            self.length = bigToNative(@TypeOf(self.length), self.length);
-
-            if (packet_type == .connection) {
-                self.src_port = bigToNative(@TypeOf(self.src_port), self.src_port);
-                self.dest_port = bigToNative(@TypeOf(self.dest_port), self.dest_port);
-                self.seq_num = bigToNative(@TypeOf(self.seq_num), self.seq_num);
-                self.msg_id = bigToNative(@TypeOf(self.msg_id), self.msg_id);
-            }
-        }
-
-        pub fn networkFromNativeEndian(self: *Self) void {
-            self.type = @enumFromInt(bigToNative(
-                @typeInfo(@TypeOf(self.type)).@"enum".tag_type,
-                @intFromEnum(self.type),
-            ));
-            self.length = bigToNative(@TypeOf(self.length), self.length);
-
-            if (packet_type == .connection) {
-                self.src_port = nativeToBig(@TypeOf(self.src_port), self.src_port);
-                self.dest_port = nativeToBig(@TypeOf(self.dest_port), self.dest_port);
-                self.seq_num = nativeToBig(@TypeOf(self.seq_num), self.seq_num);
-                self.msg_id = nativeToBig(@TypeOf(self.msg_id), self.msg_id);
-            }
-        }
-
-        pub fn asBytes(self: *Self) SelfBytes {
-            const size = @sizeOf(Self) + self.length;
-            return @as([*]align(@alignOf(Self)) u8, @ptrCast(self))[0..size];
-        }
-    };
-}
-
-test MessageNew {
-    comptime for (@typeInfo(MessageNew(.connection)).@"struct".decls) |field| {
-        @compileLog(field);
-    };
-}
-
-// pub fn bytesAsMessage(bytes: []const u8) !*Self {
-//     const res = std.mem.bytesAsValue(Self, bytes);
-//     return switch (res.type) {
-//         .relay, .connection => if (bytes.len == res.length + @sizeOf(Self))
-//             res
-//         else
-//             Error.InvalidMessage,
-//         .file_transfer => Error.NotImplementedSaprusType,
-//         else => Error.UnknownSaprusType,
-//     };
-// }
-
 // ZERO COPY STUFF
 // &payload could be a void value that is treated as a pointer to a [*]u8
 /// All Saprus messages
@@ -154,7 +38,7 @@ pub const Message = packed struct {
 
         pub fn getPayload(self: *align(1) Relay) []u8 {
             const len: *u16 = @ptrFromInt(@intFromPtr(self) - @sizeOf(u16));
-            return @as([*]u8, @ptrCast(&self.payload))[0 .. len.* - @sizeOf(Relay)];
+            return @as([*]u8, @ptrCast(&self.payload))[0 .. len.* - @bitSizeOf(Relay) / 8];
         }
     };
     const Connection = packed struct {
@@ -168,7 +52,7 @@ pub const Message = packed struct {
 
         pub fn getPayload(self: *align(1) Connection) []u8 {
             const len: *u16 = @ptrFromInt(@intFromPtr(self) - @sizeOf(u16));
-            return @as([*]u8, @ptrCast(&self.payload))[0 .. len.* - @sizeOf(Connection)];
+            return @as([*]u8, @ptrCast(&self.payload))[0 .. len.* - @bitSizeOf(Connection) / 8];
         }
 
         fn nativeFromNetworkEndian(self: *align(1) Connection) void {
@@ -193,23 +77,27 @@ pub const Message = packed struct {
     length: u16,
     bytes: void = {},
 
-    pub fn init(allocator: Allocator, comptime @"type": PacketType, payload_len: u16) !*Self {
+    /// Takes a byte slice, and returns a Message struct backed by the slice.
-        const header_size = @sizeOf(switch (@"type") {
+    /// This properly initializes the top level headers within the slice.
-            .relay => Relay,
+    /// This is used for creating new messages. For reading messages from the network,
-            .connection => Connection,
+    /// see: networkBytesAsValue.
-            .file_transfer => return Error.NotImplementedSaprusType,
+    pub fn init(@"type": PacketType, bytes: []align(@alignOf(Self)) u8) *Self {
-            else => return Error.UnknownSaprusType,
+        std.debug.assert(bytes.len >= @sizeOf(Self));
-        });
-        const size = payload_len + @sizeOf(Self) + header_size;
-        const bytes = try allocator.alignedAlloc(u8, @alignOf(Self), size);
         const res: *Self = @ptrCast(bytes.ptr);
         res.type = @"type";
-        res.length = payload_len + header_size;
+        res.length = @intCast(bytes.len - @sizeOf(Self));
         return res;
     }
 
-    pub fn deinit(self: *Self, allocator: Allocator) void {
+    /// Compute the number of bytes required to store a given payload size for a given message type.
-        allocator.free(self.asBytes());
+    pub fn calcSize(comptime @"type": PacketType, payload_len: usize) MessageTypeError!u16 {
+        const header_size = @bitSizeOf(switch (@"type") {
+            .relay => Relay,
+            .connection => Connection,
+            .file_transfer => return MessageTypeError.NotImplementedSaprusType,
+            else => return MessageTypeError.UnknownSaprusType,
+        }) / 8;
+        return @intCast(payload_len + @sizeOf(Self) + header_size);
     }
 
     fn getRelay(self: *Self) *align(1) Relay {
@@ -219,7 +107,8 @@ pub const Message = packed struct {
         return std.mem.bytesAsValue(Connection, &self.bytes);
     }
 
-    pub fn getSaprusTypePayload(self: *Self) Error!(union(PacketType) {
+    /// Access the message Saprus payload.
+    pub fn getSaprusTypePayload(self: *Self) MessageTypeError!(union(PacketType) {
         relay: *align(1) Relay,
         file_transfer: void,
         connection: *align(1) Connection,
@@ -227,12 +116,13 @@ pub const Message = packed struct {
         return switch (self.type) {
             .relay => .{ .relay = self.getRelay() },
             .connection => .{ .connection = self.getConnection() },
-            .file_transfer => Error.NotImplementedSaprusType,
+            .file_transfer => MessageTypeError.NotImplementedSaprusType,
-            else => Error.UnknownSaprusType,
+            else => MessageTypeError.UnknownSaprusType,
         };
     }
 
-    pub fn nativeFromNetworkEndian(self: *Self) Error!void {
+    /// Convert the message to native endianness from network endianness in-place.
+    pub fn nativeFromNetworkEndian(self: *Self) MessageTypeError!void {
         self.type = @enumFromInt(bigToNative(
             @typeInfo(@TypeOf(self.type)).@"enum".tag_type,
             @intFromEnum(self.type),
@@ -255,12 +145,13 @@ pub const Message = packed struct {
         }
     }
 
-    pub fn networkFromNativeEndian(self: *Self) Error!void {
+    /// Convert the message to network endianness from native endianness in-place.
+    pub fn networkFromNativeEndian(self: *Self) MessageTypeError!void {
         try switch (try self.getSaprusTypePayload()) {
             .relay => {},
             .connection => |*con| con.*.networkFromNativeEndian(),
-            .file_transfer => Error.NotImplementedSaprusType,
+            .file_transfer => MessageTypeError.NotImplementedSaprusType,
-            else => Error.UnknownSaprusType,
+            else => MessageTypeError.UnknownSaprusType,
         };
         self.type = @enumFromInt(nativeToBig(
             @typeInfo(@TypeOf(self.type)).@"enum".tag_type,
@@ -269,18 +160,29 @@ pub const Message = packed struct {
         self.length = nativeToBig(@TypeOf(self.length), self.length);
     }
 
-    pub fn bytesAsValue(bytes: SelfBytes) !*Self {
+    /// Convert network endian bytes to a native endian value in-place.
+    pub fn networkBytesAsValue(bytes: SelfBytes) MessageParseError!*Self {
+        const res = std.mem.bytesAsValue(Self, bytes);
+        try res.nativeFromNetworkEndian();
+        return .bytesAsValue(bytes);
+    }
+
+    /// Create a structured view of the bytes without initializing the length or type,
+    /// and without converting the endianness.
+    pub fn bytesAsValue(bytes: SelfBytes) MessageParseError!*Self {
         const res = std.mem.bytesAsValue(Self, bytes);
         return switch (res.type) {
             .relay, .connection => if (bytes.len == res.length + @sizeOf(Self))
                 res
             else
-                Error.InvalidMessage,
+                MessageParseError.InvalidMessage,
-            .file_transfer => Error.NotImplementedSaprusType,
+            .file_transfer => MessageParseError.NotImplementedSaprusType,
-            else => Error.UnknownSaprusType,
+            else => MessageParseError.UnknownSaprusType,
         };
     }
 
+    /// Deprecated.
+    /// If I need the bytes, I should just pass around the slice that is backing this to begin with.
     pub fn asBytes(self: *Self) SelfBytes {
         const size = @sizeOf(Self) + self.length;
         return @as([*]align(@alignOf(Self)) u8, @ptrCast(self))[0..size];
@@ -288,43 +190,74 @@ pub const Message = packed struct {
 };
 
 test "testing variable length zero copy struct" {
-    const gpa = std.testing.allocator;
-    const payload = "Hello darkness my old friend";
-
-    // Create a view of the byte slice as a Message
-    const msg: *Message = try .init(gpa, .relay, payload.len);
-    defer msg.deinit(gpa);
-
     {
-        // Set the message values
+        // Relay test
+        const payload = "Hello darkness my old friend";
+        var msg_bytes: [try Message.calcSize(.relay, payload.len)]u8 align(@alignOf(Message)) = undefined;
+
+        // Create a view of the byte slice as a Message
+        const msg: *Message = .init(.relay, &msg_bytes);
+
         {
-            // These are both set by the init call.
+            // Set the message values
-            // msg.type = .relay;
+            {
-            // msg.length = payload_len;
+                // These are both set by the init call.
+                // msg.type = .relay;
+                // msg.length = payload_len;
+            }
+            const relay = (try msg.getSaprusTypePayload()).relay;
+            relay.dest = .{ 1, 2, 3, 4 };
+            @memcpy(relay.getPayload(), payload);
         }
-        const relay = (try msg.getSaprusTypePayload()).relay;
+
-        relay.dest = .{ 1, 2, 3, 4 };
+        {
-        @memcpy(relay.getPayload(), payload);
+            // Print the message as hex using the network byte order
+            try msg.networkFromNativeEndian();
+            // We know the error from nativeFromNetworkEndian is unreachable because
+            // it would have returned an error from networkFromNativeEndian.
+            defer msg.nativeFromNetworkEndian() catch unreachable;
+            std.debug.print("relay network bytes: {x}\n", .{msg_bytes});
+            std.debug.print("bytes len: {d}\n", .{msg_bytes.len});
+        }
+
+        if (false) {
+            // Illegal behavior
+            std.debug.print("{any}\n", .{(try msg.getSaprusTypePayload()).connection});
+        }
+
+        try std.testing.expectEqualDeep(msg, try Message.bytesAsValue(msg.asBytes()));
     }
 
     {
-        const bytes = msg.asBytes();
+        // Connection test
+        const payload = "Hello darkness my old friend";
+        var msg_bytes: [try Message.calcSize(.connection, payload.len)]u8 align(@alignOf(Message)) = undefined;
 
-        // Print the message as hex using the network byte order
+        // Create a view of the byte slice as a Message
-        try msg.networkFromNativeEndian();
+        const msg: *Message = .init(.connection, &msg_bytes);
-        // We know the error from nativeFromNetworkEndian is unreachable because
+
-        // it would have returned an error from networkFromNativeEndian.
+        {
-        defer msg.nativeFromNetworkEndian() catch unreachable;
+            // Initializing connection header values
-        std.debug.print("network bytes: {x}\n", .{bytes});
+            const connection = (try msg.getSaprusTypePayload()).connection;
-        std.debug.print("bytes len: {d}\n", .{bytes.len});
+            connection.src_port = 1;
+            connection.dest_port = 2;
+            connection.seq_num = 3;
+            connection.msg_id = 4;
+            connection.reserved = 5;
+            connection.options = @bitCast(@as(u8, 6));
+            @memcpy(connection.getPayload(), payload);
+        }
+
+        {
+            // Print the message as hex using the network byte order
+            try msg.networkFromNativeEndian();
+            // We know the error from nativeFromNetworkEndian is unreachable because
+            // it would have returned an error from networkFromNativeEndian.
+            defer msg.nativeFromNetworkEndian() catch unreachable;
+            std.debug.print("connection network bytes: {x}\n", .{msg_bytes});
+            std.debug.print("bytes len: {d}\n", .{msg_bytes.len});
+        }
     }
-
-    if (false) {
-        // Illegal behavior
-        std.debug.print("{any}\n", .{(try msg.getSaprusTypePayload()).connection});
-    }
-
-    try std.testing.expectEqualDeep(msg, try Message.bytesAsValue(msg.asBytes()));
 }
 
 const std = @import("std");
@@ -334,6 +267,51 @@ const asBytes = std.mem.asBytes;
 const nativeToBig = std.mem.nativeToBig;
 const bigToNative = std.mem.bigToNative;
 
+test "Round trip Relay toBytes and fromBytes" {
+    if (false) {
+        const gpa = std.testing.allocator;
+        const msg = Message{
+            .relay = .{
+                .header = .{ .dest = .{ 255, 255, 255, 255 } },
+                .payload = "Hello darkness my old friend",
+            },
+        };
+
+        const to_bytes = try msg.toBytes(gpa);
+        defer gpa.free(to_bytes);
+
+        const from_bytes = try Message.fromBytes(to_bytes, gpa);
+        defer from_bytes.deinit(gpa);
+
+        try std.testing.expectEqualDeep(msg, from_bytes);
+    }
+    return error.SkipZigTest;
+}
+
+test "Round trip Connection toBytes and fromBytes" {
+    if (false) {
+        const gpa = std.testing.allocator;
+        const msg = Message{
+            .connection = .{
+                .header = .{
+                    .src_port = 0,
+                    .dest_port = 0,
+                },
+                .payload = "Hello darkness my old friend",
+            },
+        };
+
+        const to_bytes = try msg.toBytes(gpa);
+        defer gpa.free(to_bytes);
+
+        const from_bytes = try Message.fromBytes(to_bytes, gpa);
+        defer from_bytes.deinit(gpa);
+
+        try std.testing.expectEqualDeep(msg, from_bytes);
+    }
+    return error.SkipZigTest;
+}
+
 test {
     std.testing.refAllDeclsRecursive(@This());
 }