mirror of
https://git.robbyzambito.me/zaprus
synced 2025-12-20 16:24:50 +00:00
Compare commits
1 Commits
Io-writer
...
push-kruxs
| Author | SHA1 | Date | |
|---|---|---|---|
| 59034f239b |
@@ -36,9 +36,30 @@ pub const ZeroCopyMessage = packed struct {
|
|||||||
dest: @Vector(4, u8),
|
dest: @Vector(4, u8),
|
||||||
payload: void,
|
payload: void,
|
||||||
|
|
||||||
pub fn getPayload(self: *align(1) Relay) []u8 {
|
pub fn getPayload(self: *align(@alignOf(ZeroCopyMessage)) Relay) []u8 {
|
||||||
const len: *u16 = @ptrFromInt(@intFromPtr(self) - @sizeOf(u16));
|
// Cast the 'self' pointer (which points to the Relay header,
|
||||||
return @as([*]u8, @ptrCast(&self.payload))[0 .. len.* - @sizeOf(Relay)];
|
// located at the same memory as the parent's 'bytes' field)
|
||||||
|
// to a pointer to void, as required by @fieldParentPtr for a void field.
|
||||||
|
// Preserve the known alignment.
|
||||||
|
const self_as_void_ptr: *align(@alignOf(ZeroCopyMessage)) void = @ptrCast(self);
|
||||||
|
|
||||||
|
// Cast the resulting *void pointer to the parent type *ZeroCopyMessage.
|
||||||
|
// This cast performs the necessary alignment check.
|
||||||
|
const parent: *ZeroCopyMessage = @alignCast(@fieldParentPtr("bytes", self_as_void_ptr));
|
||||||
|
|
||||||
|
// The 'length' field in the parent ZeroCopyMessage contains
|
||||||
|
// the size of the header (Relay) + payload length.
|
||||||
|
const total_len = parent.length;
|
||||||
|
|
||||||
|
// Payload length = total_len - size of the Relay header
|
||||||
|
const payload_len = total_len - @sizeOf(Relay);
|
||||||
|
|
||||||
|
// The payload starts immediately after the fixed fields of the Relay struct.
|
||||||
|
// The address of the 'payload' field represents this starting point.
|
||||||
|
const payload_start_ptr: [*]u8 = @ptrCast(&self.payload);
|
||||||
|
|
||||||
|
// Return a slice from the payload start address with the calculated length.
|
||||||
|
return payload_start_ptr[0..payload_len];
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
const Connection = packed struct {
|
const Connection = packed struct {
|
||||||
@@ -94,17 +115,17 @@ pub const ZeroCopyMessage = packed struct {
|
|||||||
allocator.free(self.asBytes());
|
allocator.free(self.asBytes());
|
||||||
}
|
}
|
||||||
|
|
||||||
fn getRelay(self: *Self) *align(1) Relay {
|
fn getRelay(self: *Self) *align(@alignOf(Self)) Relay {
|
||||||
return std.mem.bytesAsValue(Relay, &self.bytes);
|
return std.mem.bytesAsValue(Relay, &self.bytes);
|
||||||
}
|
}
|
||||||
fn getConnection(self: *Self) *align(1) Connection {
|
fn getConnection(self: *Self) *align(@alignOf(Self)) Connection {
|
||||||
return std.mem.bytesAsValue(Connection, &self.bytes);
|
return std.mem.bytesAsValue(Connection, &self.bytes);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn getSaprusTypePayload(self: *Self) Error!(union(PacketType) {
|
pub fn getSaprusTypePayload(self: *Self) Error!(union(PacketType) {
|
||||||
relay: *align(1) Relay,
|
relay: *align(@alignOf(Self)) Relay,
|
||||||
file_transfer: void,
|
file_transfer: void,
|
||||||
connection: *align(1) Connection,
|
connection: *align(@alignOf(Self)) Connection,
|
||||||
}) {
|
}) {
|
||||||
return switch (self.type) {
|
return switch (self.type) {
|
||||||
.relay => .{ .relay = self.getRelay() },
|
.relay => .{ .relay = self.getRelay() },
|
||||||
@@ -172,6 +193,7 @@ test "testing variable length zero copy struct" {
|
|||||||
// Create a view of the byte slice as a ZeroCopyMessage
|
// Create a view of the byte slice as a ZeroCopyMessage
|
||||||
const zcm: *ZeroCopyMessage = try .init(gpa, .relay, payload.len);
|
const zcm: *ZeroCopyMessage = try .init(gpa, .relay, payload.len);
|
||||||
defer zcm.deinit(gpa);
|
defer zcm.deinit(gpa);
|
||||||
|
std.debug.print("outer: {*}\n", .{zcm});
|
||||||
|
|
||||||
{
|
{
|
||||||
// Set the message values
|
// Set the message values
|
||||||
|
|||||||
Reference in New Issue
Block a user