diff --git a/std/c.zig b/std/c.zig index 45d93bfd9f..dcde3b8ebb 100644 --- a/std/c.zig +++ b/std/c.zig @@ -107,6 +107,7 @@ pub extern "c" fn sysctl(name: [*]const c_int, namelen: c_uint, oldp: ?*c_void, pub extern "c" fn sysctlbyname(name: [*]const u8, oldp: ?*c_void, oldlenp: ?*usize, newp: ?*c_void, newlen: usize) c_int; pub extern "c" fn sysctlnametomib(name: [*]const u8, mibp: ?*c_int, sizep: ?*usize) c_int; +pub extern "c" fn gethostname(name: [*]u8, len: usize) c_int; pub extern "c" fn bind(socket: fd_t, address: ?*const sockaddr, address_len: socklen_t) c_int; pub extern "c" fn socket(domain: c_uint, sock_type: c_uint, protocol: c_uint) c_int; pub extern "c" fn listen(sockfd: fd_t, backlog: c_uint) c_int; diff --git a/std/os.zig b/std/os.zig index b32c8a0b7e..a8677666ad 100644 --- a/std/os.zig +++ b/std/os.zig @@ -2688,6 +2688,38 @@ pub fn futimens(fd: fd_t, times: *const [2]timespec) FutimensError!void { } } +pub const GetHostNameError = error{ + PermissionDenied, + Unexpected, +}; + +pub fn gethostname(name_buffer: *[HOST_NAME_MAX]u8) GetHostNameError![]u8 { + if (builtin.link_libc) { + switch (errno(system.gethostname(name_buffer, name_buffer.len))) { + 0 => return mem.toSlice(u8, name_buffer), + EFAULT => unreachable, + ENAMETOOLONG => unreachable, // HOST_NAME_MAX prevents this + EPERM => return error.PermissionDenied, + else => |err| return unexpectedErrno(err), + } + } + if (linux.is_the_target) { + var uts: utsname = undefined; + switch (errno(system.uname(&uts))) { + 0 => { + const hostname = mem.toSlice(u8, &uts.nodename); + mem.copy(u8, name_buffer, hostname); + return name_buffer[0..hostname.len]; + }, + EFAULT => unreachable, + EPERM => return error.PermissionDenied, + else => |err| return unexpectedErrno(err), + } + } + + @compileError("TODO implement gethostname for this OS"); +} + test "" { _ = @import("os/darwin.zig"); _ = @import("os/freebsd.zig"); diff --git a/std/os/bits/darwin.zig b/std/os/bits/darwin.zig index 483d4cda90..b21498ba2d 100644 --- a/std/os/bits/darwin.zig +++ b/std/os/bits/darwin.zig @@ -1175,3 +1175,4 @@ pub fn S_ISSOCK(m: u32) bool { pub fn S_IWHT(m: u32) bool { return m & S_IFMT == S_IFWHT; } +pub const HOST_NAME_MAX = 72; diff --git a/std/os/bits/freebsd.zig b/std/os/bits/freebsd.zig index 45432a6c07..f94ff07f58 100644 --- a/std/os/bits/freebsd.zig +++ b/std/os/bits/freebsd.zig @@ -935,3 +935,5 @@ pub fn S_ISSOCK(m: u32) bool { pub fn S_IWHT(m: u32) bool { return m & S_IFMT == S_IFWHT; } + +pub const HOST_NAME_MAX = 255; diff --git a/std/os/bits/linux.zig b/std/os/bits/linux.zig index c489592c4b..939d203e8d 100644 --- a/std/os/bits/linux.zig +++ b/std/os/bits/linux.zig @@ -1175,3 +1175,13 @@ pub const IORING_REGISTER_FILES = 2; pub const IORING_UNREGISTER_FILES = 3; pub const IORING_REGISTER_EVENTFD = 4; pub const IORING_UNREGISTER_EVENTFD = 5; + +pub const utsname = extern struct { + sysname: [65]u8, + nodename: [65]u8, + release: [65]u8, + version: [65]u8, + machine: [65]u8, + domainname: [65]u8, +}; +pub const HOST_NAME_MAX = 64; diff --git a/std/os/bits/netbsd.zig b/std/os/bits/netbsd.zig index ff19d090af..5036e1e9a1 100644 --- a/std/os/bits/netbsd.zig +++ b/std/os/bits/netbsd.zig @@ -819,3 +819,5 @@ pub fn S_ISSOCK(m: u32) bool { pub fn S_IWHT(m: u32) bool { return m & S_IFMT == S_IFWHT; } + +pub const HOST_NAME_MAX = 255; diff --git a/std/os/linux.zig b/std/os/linux.zig index 74c71ee2c4..47fc1841e9 100644 --- a/std/os/linux.zig +++ b/std/os/linux.zig @@ -965,6 +965,10 @@ pub fn sigaltstack(ss: ?*stack_t, old_ss: ?*stack_t) usize { return syscall2(SYS_sigaltstack, @ptrToInt(ss), @ptrToInt(old_ss)); } +pub fn uname(uts: *utsname) usize { + return syscall1(SYS_uname, @ptrToInt(uts)); +} + // XXX: This should be weak extern const __ehdr_start: elf.Ehdr = undefined; diff --git a/std/os/test.zig b/std/os/test.zig index a821c5dd9f..d886597a38 100644 --- a/std/os/test.zig +++ b/std/os/test.zig @@ -210,3 +210,12 @@ test "dl_iterate_phdr" { expect(os.dl_iterate_phdr(usize, iter_fn, &counter) != 0); expect(counter != 0); } + +test "gethostname" { + if (os.windows.is_the_target) + return error.SkipZigTest; + + var buf: [os.HOST_NAME_MAX]u8 = undefined; + const hostname = try os.gethostname(&buf); + expect(hostname.len != 0); +}