Make StreamServer return address of accecpted client
parent
ed956b5812
commit
aa4e92f3b3
|
@ -1360,14 +1360,22 @@ pub const StreamServer = struct {
|
|||
BlockedByFirewall,
|
||||
} || os.UnexpectedError;
|
||||
|
||||
/// If this function succeeds, the returned `fs.File` is a caller-managed resource.
|
||||
pub fn accept(self: *StreamServer) AcceptError!fs.File {
|
||||
pub const Connection = struct {
|
||||
file: fs.File,
|
||||
address: Address
|
||||
};
|
||||
|
||||
/// If this function succeeds, the returned `Connection` is a caller-managed resource.
|
||||
pub fn accept(self: *StreamServer) AcceptError!Connection {
|
||||
const nonblock = if (std.io.is_async) os.SOCK_NONBLOCK else 0;
|
||||
const accept_flags = nonblock | os.SOCK_CLOEXEC;
|
||||
var accepted_addr: Address = undefined;
|
||||
var adr_len: os.socklen_t = @sizeOf(Address);
|
||||
if (os.accept4(self.sockfd.?, &accepted_addr.any, &adr_len, accept_flags)) |fd| {
|
||||
return fs.File.openHandle(fd);
|
||||
return Connection{
|
||||
.file = fs.File.openHandle(fd),
|
||||
.address = accepted_addr,
|
||||
};
|
||||
} else |err| switch (err) {
|
||||
// We only give SOCK_NONBLOCK when I/O mode is async, in which case this error
|
||||
// is handled by os.accept4.
|
||||
|
|
|
@ -115,8 +115,8 @@ fn testClient(addr: net.Address) anyerror!void {
|
|||
}
|
||||
|
||||
fn testServer(server: *net.StreamServer) anyerror!void {
|
||||
var client_file = try server.accept();
|
||||
var client = try server.accept();
|
||||
|
||||
const stream = &client_file.outStream().stream;
|
||||
const stream = &client.file.outStream().stream;
|
||||
try stream.print("hello from server\n");
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue