Last handshake packet uses non 8888 port

This commit is contained in:
2025-04-05 23:46:37 -04:00
parent ec94e85ab9
commit 6b38d5bb74

View File

@@ -14,7 +14,7 @@ pub fn deinit() void {
network.deinit();
}
fn broadcastSaprusMessage(msg: SaprusMessage, allocator: Allocator) !void {
fn broadcastSaprusMessage(msg: SaprusMessage, udp_port: u16, allocator: Allocator) !void {
const msg_bytes = try msg.toBytes(allocator);
defer allocator.free(msg_bytes);
@@ -31,7 +31,7 @@ fn broadcastSaprusMessage(msg: SaprusMessage, allocator: Allocator) !void {
const dest_addr = network.EndPoint{
.address = network.Address{ .ipv4 = network.Address.IPv4.broadcast },
.port = 8888,
.port = udp_port,
};
try sock.bind(bind_addr);
@@ -47,15 +47,20 @@ pub fn sendRelay(payload: []const u8, allocator: Allocator) !void {
},
};
try broadcastSaprusMessage(msg, allocator);
try broadcastSaprusMessage(msg, 8888, allocator);
}
fn randomPort() u16 {
var p: u16 = 0;
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 {
var dest_port: u16 = 0;
if (rand) |r| {
dest_port = r.intRangeAtMost(u16, 1024, 65000);
} else unreachable;
const dest_port = randomPort();
const msg = SaprusMessage{
.connection = .{
.header = .{
@@ -69,7 +74,7 @@ pub fn sendInitialConnection(payload: []const u8, initial_port: u16, allocator:
},
};
try broadcastSaprusMessage(msg, allocator);
try broadcastSaprusMessage(msg, 8888, allocator);
return msg;
}
@@ -90,7 +95,8 @@ fn awaitSentinelConnectionResponse(
.port = 8888,
};
try sock.setReadTimeout(1000);
// timeout 1s
try sock.setReadTimeout(1_000_000);
try sock.bind(bind_addr);
// Signal that the socket is ready to receive data.
@@ -128,7 +134,7 @@ pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusMessage {
);
// Block until the socket is set up.
try setting_up_socket.timedWait(500 * 1000 * 1000 * 1000);
try setting_up_socket.timedWait(5 * 1000 * 1000 * 1000 * 1000);
const msg = try sendInitialConnection(payload, initial_port, allocator);
@@ -136,7 +142,7 @@ pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusMessage {
response_thread.join();
// Complete handshake after awaiting response
try broadcastSaprusMessage(msg, allocator);
try broadcastSaprusMessage(msg, randomPort(), allocator);
return initial_conn_res;
}