zambyte/zaprus
1
0
Fork 0
mirror of https://git.robbyzambito.me/zaprus synced 2026-02-04 08:24:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Start adding connection message

Browse Source
This commit is contained in:
Robby Zambito
2026-01-19 14:01:57 -05:00
parent abf2cfb366
commit cc765dc91b
2 changed files with 73 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

99
src/main.zig
View File

@@ -90,10 +90,10 @@ pub fn main(init: std.process.Init) !void {
std.debug.print("dest: {s}\n", .{flags.dest orelse "<null>"});
std.debug.print("connect: {s}\n", .{flags.connect orelse "<null>"});
// const rand = blk: {
// const io_source: std.Random.IoSource = .{ .io = init.io };
// break :blk io_source.interface();
// };
const rand = blk: {
const io_source: std.Random.IoSource = .{ .io = init.io };
break :blk io_source.interface();
};
// const net_interface: std.Io.net.Interface = .{ .index = 1 };
// std.debug.print("Interface: {s}\n", .{(try net_interface.name(init.io)).toSlice()});
@@ -153,46 +153,89 @@ pub fn main(init: std.process.Init) !void {
};
var headers: EthIpUdp = .{
.src_mac = @splat(0x0e),
.src_mac = undefined, // TODO: REAL MAC
.ip = .{
.id = 0,
.src_addr = 0,
.src_addr = rand.int(u32),
.dst_addr = @bitCast([_]u8{ 255, 255, 255, 255 }),
.len = undefined,
},
.udp = .{
.src_port = undefined, // TODO: change this?
.src_port = rand.intRangeAtMost(u16, 1025, std.math.maxInt(u16)),
.dst_port = 8888,
.len = undefined,
},
};
std.debug.print("headers: {any}\n", .{&headers.toBytes()});
const relay: SaprusMessage = .{
.relay = .{
.dest = .fromBytes(&parseDest(flags.dest)),
.payload = flags.relay.?,
},
};
if (flags.relay != null) {
const relay: SaprusMessage = .{
.relay = .{
.dest = .fromBytes(&parseDest(flags.dest)),
.payload = flags.relay.?,
},
};
var relay_buf: [2048]u8 = undefined;
const relay_bytes = relay.toBytes(&relay_buf);
std.debug.print("payload: {any}\n", .{relay_bytes});
headers.setPayloadLen(relay_bytes.len);
var relay_buf: [2048]u8 = undefined;
const relay_bytes = relay.toBytes(&relay_buf);
std.debug.print("payload: {any}\n", .{relay_bytes});
headers.setPayloadLen(relay_bytes.len);
const full_msg = blk: {
var msg_buf: [2048]u8 = undefined;
var msg_w: Writer = .fixed(&msg_buf);
msg_w.writeAll(&headers.toBytes()) catch unreachable;
msg_w.writeAll(relay_bytes) catch unreachable;
break :blk msg_w.buffered();
};
const full_msg = blk: {
var msg_buf: [2048]u8 = undefined;
var msg_w: Writer = .fixed(&msg_buf);
msg_w.writeAll(&headers.toBytes()) catch unreachable;
msg_w.writeAll(relay_bytes) catch unreachable;
break :blk msg_w.buffered();
};
std.debug.print("full message = {any}\n", .{full_msg});
std.debug.print("full message = {any}\n", .{full_msg});
var socket: RawSocket = try .init("enp7s0");
defer socket.deinit();
try socket.send(full_msg);
var socket: RawSocket = try .init("enp7s0");
defer socket.deinit();
try socket.send(full_msg);
return;
}
if (flags.connect != null) {
const dest = rand.intRangeAtMost(u16, 1025, std.math.maxInt(u16));
const src = rand.intRangeAtMost(u16, 1025, std.math.maxInt(u16));
// udp dest port should not be 8888 after first
const udp_dest_port = rand.intRangeAtMost(u16, 1025, std.math.maxInt(u16));
const connection: SaprusMessage = .{
.connection = .{
.src = src,
.dest = dest,
.seq = undefined,
.id = undefined,
.payload = flags.connect.?,
},
};
_ = udp_dest_port;
var connection_buf: [2048]u8 = undefined;
const connection_bytes = connection.toBytes(&connection_buf);
std.debug.print("connection: {any}\n", .{connection_bytes});
headers.setPayloadLen(connection_bytes.len);
const full_msg = blk: {
var msg_buf: [2048]u8 = undefined;
var msg_w: Writer = .fixed(&msg_buf);
msg_w.writeAll(&headers.toBytes()) catch unreachable;
msg_w.writeAll(connection_bytes) catch unreachable;
break :blk msg_w.buffered();
};
std.debug.print("full message = {any}\n", .{full_msg});
var socket: RawSocket = try .init("enp7s0");
defer socket.deinit();
try socket.send(full_msg);
return;
}
unreachable;
}
fn parseDest(in: ?[]const u8) [4]u8 {