zig

blob d7cfb4e1 (13828B) - Raw

---

 const builtin = @import("builtin");
 
 const std = @import("../../std.zig");
 const assert = std.debug.assert;
 const linux = std.os.linux;
 const mem = std.mem;
 const elf = std.elf;
 const expect = std.testing.expect;
 const expectEqual = std.testing.expectEqual;
 const fs = std.fs;
 
 test "fallocate" {
     if (builtin.cpu.arch.isMIPS64() and (builtin.abi == .gnuabin32 or builtin.abi == .muslabin32)) return error.SkipZigTest; // https://codeberg.org/ziglang/zig/issues/30220
 
     const io = std.testing.io;
 
     var tmp = std.testing.tmpDir(.{});
     defer tmp.cleanup();
 
     const path = "test_fallocate";
     const file = try tmp.dir.createFile(io, path, .{ .truncate = true, .mode = 0o666 });
     defer file.close(io);
 
     try expect((try file.stat()).size == 0);
 
     const len: i64 = 65536;
     switch (linux.errno(linux.fallocate(file.handle, 0, 0, len))) {
         .SUCCESS => {},
         .NOSYS => return error.SkipZigTest,
         .OPNOTSUPP => return error.SkipZigTest,
         else => |errno| std.debug.panic("unhandled errno: {}", .{errno}),
     }
 
     try expect((try file.stat()).size == len);
 }
 
 test "getpid" {
     try expect(linux.getpid() != 0);
 }
 
 test "getppid" {
     try expect(linux.getppid() != 0);
 }
 
 test "timer" {
     const epoll_fd = linux.epoll_create();
     var err: linux.E = linux.errno(epoll_fd);
     try expect(err == .SUCCESS);
 
     const timer_fd = linux.timerfd_create(linux.TIMERFD_CLOCK.MONOTONIC, .{});
     try expect(linux.errno(timer_fd) == .SUCCESS);
 
     const time_interval = linux.timespec{
         .sec = 0,
         .nsec = 2000000,
     };
 
     const new_time = linux.itimerspec{
         .it_interval = time_interval,
         .it_value = time_interval,
     };
 
     err = linux.errno(linux.timerfd_settime(@as(i32, @intCast(timer_fd)), .{}, &new_time, null));
     try expect(err == .SUCCESS);
 
     var event = linux.epoll_event{
         .events = linux.EPOLL.IN | linux.EPOLL.OUT | linux.EPOLL.ET,
         .data = linux.epoll_data{ .ptr = 0 },
     };
 
     err = linux.errno(linux.epoll_ctl(@as(i32, @intCast(epoll_fd)), linux.EPOLL.CTL_ADD, @as(i32, @intCast(timer_fd)), &event));
     try expect(err == .SUCCESS);
 
     const events_one: linux.epoll_event = undefined;
     var events = [_]linux.epoll_event{events_one} ** 8;
 
     err = linux.errno(linux.epoll_wait(@as(i32, @intCast(epoll_fd)), &events, 8, -1));
     try expect(err == .SUCCESS);
 }
 
 test "statx" {
     const io = std.testing.io;
 
     var tmp = std.testing.tmpDir(.{});
     defer tmp.cleanup();
 
     const tmp_file_name = "just_a_temporary_file.txt";
     var file = try tmp.dir.createFile(io, tmp_file_name, .{});
     defer file.close(io);
 
     var buf: linux.Statx = undefined;
     switch (linux.errno(linux.statx(file.handle, "", linux.AT.EMPTY_PATH, .BASIC_STATS, &buf))) {
         .SUCCESS => {},
         else => unreachable,
     }
 
     const uid = linux.getuid();
     const gid = linux.getgid();
     if (buf.mask.MODE)
         try expectEqual(@as(linux.mode_t, linux.S.IFREG), buf.mode & linux.S.IFMT);
     if (buf.mask.UID)
         try expectEqual(uid, buf.uid);
     if (buf.mask.GID)
         try expectEqual(gid, buf.gid);
     if (buf.mask.SIZE)
         try expectEqual(@as(u64, 0), buf.size);
 }
 
 test "user and group ids" {
     if (builtin.link_libc) return error.SkipZigTest;
     try expectEqual(linux.getauxval(elf.AT_UID), linux.getuid());
     try expectEqual(linux.getauxval(elf.AT_GID), linux.getgid());
     try expectEqual(linux.getauxval(elf.AT_EUID), linux.geteuid());
     try expectEqual(linux.getauxval(elf.AT_EGID), linux.getegid());
 }
 
 test "fadvise" {
     const io = std.testing.io;
 
     var tmp = std.testing.tmpDir(.{});
     defer tmp.cleanup();
 
     const tmp_file_name = "temp_posix_fadvise.txt";
     var file = try tmp.dir.createFile(io, tmp_file_name, .{});
     defer file.close(io);
 
     var buf: [2048]u8 = undefined;
     try file.writeAll(&buf);
 
     const ret = linux.fadvise(file.handle, 0, 0, linux.POSIX_FADV.SEQUENTIAL);
     try expectEqual(@as(usize, 0), ret);
 }
 
 test "sigset_t" {
     const SIG = linux.SIG;
     assert(@sizeOf(linux.sigset_t) == (linux.NSIG / 8));
 
     var sigset = linux.sigemptyset();
 
     // See that none are set, then set each one, see that they're all set, then
     // remove them all, and then see that none are set.
     for (1..linux.NSIG) |i| {
         const sig = std.enums.fromInt(SIG, i) orelse continue;
         try expectEqual(false, linux.sigismember(&sigset, sig));
     }
     for (1..linux.NSIG) |i| {
         const sig = std.enums.fromInt(SIG, i) orelse continue;
         linux.sigaddset(&sigset, sig);
     }
     for (1..linux.NSIG) |i| {
         const sig = std.enums.fromInt(SIG, i) orelse continue;
         try expectEqual(true, linux.sigismember(&sigset, sig));
     }
     for (1..linux.NSIG) |i| {
         const sig = std.enums.fromInt(SIG, i) orelse continue;
         linux.sigdelset(&sigset, sig);
     }
     for (1..linux.NSIG) |i| {
         const sig = std.enums.fromInt(SIG, i) orelse continue;
         try expectEqual(false, linux.sigismember(&sigset, sig));
     }
 }
 
 test "sigfillset" {
     // unlike the C library, all the signals are set in the kernel-level fillset
     const sigset = linux.sigfillset();
     for (1..linux.NSIG) |i| {
         const sig = std.enums.fromInt(linux.SIG, i) orelse continue;
         try expectEqual(true, linux.sigismember(&sigset, sig));
     }
 }
 
 test "sigemptyset" {
     const sigset = linux.sigemptyset();
     for (1..linux.NSIG) |i| {
         const sig = std.enums.fromInt(linux.SIG, i) orelse continue;
         try expectEqual(false, linux.sigismember(&sigset, sig));
     }
 }
 
 test "sysinfo" {
     var info: linux.Sysinfo = undefined;
     const result: usize = linux.sysinfo(&info);
     try expect(std.os.linux.errno(result) == .SUCCESS);
 
     try expect(info.mem_unit > 0);
     try expect(info.mem_unit <= std.heap.page_size_max);
 }
 
 comptime {
     assert(128 == @as(u32, @bitCast(linux.FUTEX_OP{ .cmd = @enumFromInt(0), .private = true, .realtime = false })));
     assert(256 == @as(u32, @bitCast(linux.FUTEX_OP{ .cmd = @enumFromInt(0), .private = false, .realtime = true })));
 
     // Check futex_param4 union is packed correctly
     const param_union = linux.futex_param4{
         .val2 = 0xaabbcc,
     };
     assert(@intFromPtr(param_union.timeout) == 0xaabbcc);
 }
 
 test "futex v1" {
     var lock: std.atomic.Value(u32) = std.atomic.Value(u32).init(1);
     var rc: usize = 0;
 
     // No-op wait, lock value is not expected value
     rc = linux.futex(&lock.raw, .{ .cmd = .WAIT, .private = true }, 2, .{ .timeout = null }, null, 0);
     try expectEqual(.AGAIN, linux.errno(rc));
 
     rc = linux.futex_4arg(&lock.raw, .{ .cmd = .WAIT, .private = true }, 2, null);
     try expectEqual(.AGAIN, linux.errno(rc));
 
     // Short-fuse wait, timeout kicks in
     rc = linux.futex(&lock.raw, .{ .cmd = .WAIT, .private = true }, 1, .{ .timeout = &.{ .sec = 0, .nsec = 2 } }, null, 0);
     try expectEqual(.TIMEDOUT, linux.errno(rc));
 
     rc = linux.futex_4arg(&lock.raw, .{ .cmd = .WAIT, .private = true }, 1, &.{ .sec = 0, .nsec = 2 });
     try expectEqual(.TIMEDOUT, linux.errno(rc));
 
     // Wakeup (no waiters)
     rc = linux.futex(&lock.raw, .{ .cmd = .WAKE, .private = true }, 2, .{ .timeout = null }, null, 0);
     try expectEqual(0, rc);
 
     rc = linux.futex_3arg(&lock.raw, .{ .cmd = .WAKE, .private = true }, 2);
     try expectEqual(0, rc);
 
     // CMP_REQUEUE - val3 mismatch
     rc = linux.futex(&lock.raw, .{ .cmd = .CMP_REQUEUE, .private = true }, 2, .{ .val2 = 0 }, null, 99);
     try expectEqual(.AGAIN, linux.errno(rc));
 
     // CMP_REQUEUE - requeue (but no waiters, so ... not much)
     {
         const val3 = 1;
         const wake_nr = 3;
         const requeue_max = std.math.maxInt(u31);
         var target_lock: std.atomic.Value(u32) = std.atomic.Value(u32).init(1);
         rc = linux.futex(&lock.raw, .{ .cmd = .CMP_REQUEUE, .private = true }, wake_nr, .{ .val2 = requeue_max }, &target_lock.raw, val3);
         try expectEqual(0, rc);
     }
 
     // WAKE_OP - just to see if we can construct the arguments ...
     {
         var lock2: std.atomic.Value(u32) = std.atomic.Value(u32).init(1);
         const wake1_nr = 2;
         const wake2_nr = 3;
         const wake_op = linux.FUTEX_WAKE_OP{
             .cmd = .ANDN,
             .arg_shift = true,
             .cmp = .LT,
             .oparg = 4,
             .cmdarg = 5,
         };
 
         rc = linux.futex(&lock.raw, .{ .cmd = .WAKE_OP, .private = true }, wake1_nr, .{ .val2 = wake2_nr }, &lock2.raw, @bitCast(wake_op));
         try expectEqual(0, rc);
     }
 
     // WAIT_BITSET
     {
         // val1 return early
         rc = linux.futex(&lock.raw, .{ .cmd = .WAIT_BITSET, .private = true }, 2, .{ .timeout = null }, null, 0xfff);
         try expectEqual(.AGAIN, linux.errno(rc));
 
         // timeout wait
         const timeout: linux.timespec = .{ .sec = 0, .nsec = 2 };
         rc = linux.futex(&lock.raw, .{ .cmd = .WAIT_BITSET, .private = true }, 1, .{ .timeout = &timeout }, null, 0xfff);
         try expectEqual(.TIMEDOUT, linux.errno(rc));
     }
 
     // WAKE_BITSET
     {
         rc = linux.futex(&lock.raw, .{ .cmd = .WAKE_BITSET, .private = true }, 2, .{ .timeout = null }, null, 0xfff000);
         try expectEqual(0, rc);
 
         // bitmask must have at least 1 bit set:
         rc = linux.futex(&lock.raw, .{ .cmd = .WAKE_BITSET, .private = true }, 2, .{ .timeout = null }, null, 0);
         try expectEqual(.INVAL, linux.errno(rc));
     }
 }
 
 comptime {
     assert(2 == @as(u32, @bitCast(linux.FUTEX2_FLAGS{ .size = .U32, .private = false })));
     assert(128 == @as(u32, @bitCast(linux.FUTEX2_FLAGS{ .size = @enumFromInt(0), .private = true })));
 }
 
 test "futex2_waitv" {
     const locks = [_]std.atomic.Value(u32){
         std.atomic.Value(u32).init(1),
         std.atomic.Value(u32).init(1),
         std.atomic.Value(u32).init(1),
     };
 
     const futexes = [_]linux.futex2_waitone{
         .{
             .val = 1,
             .uaddr = @intFromPtr(&locks[0].raw),
             .flags = .{ .size = .U32, .private = true },
         },
         .{
             .val = 1,
             .uaddr = @intFromPtr(&locks[1].raw),
             .flags = .{ .size = .U32, .private = true },
         },
         .{
             .val = 1,
             .uaddr = @intFromPtr(&locks[2].raw),
             .flags = .{ .size = .U32, .private = true },
         },
     };
 
     const timeout = linux.kernel_timespec{ .sec = 0, .nsec = 2 }; // absolute timeout, so this is 1970...
     const rc = linux.futex2_waitv(&futexes, futexes.len, .{}, &timeout, .MONOTONIC);
     switch (linux.errno(rc)) {
         .NOSYS => return error.SkipZigTest, // futex2_waitv added in kernel v5.16
         else => |err| try expectEqual(.TIMEDOUT, err),
     }
 }
 
 // Futex v2 API is only supported on recent kernels (v6.7), so skip tests if the syscalls
 // return ENOSYS.
 fn futex2_skip_if_unsupported() !void {
     const lock: u32 = 0;
     const rc = linux.futex2_wake(&lock, 0, 1, .{ .size = .U32, .private = true });
     if (linux.errno(rc) == .NOSYS) {
         return error.SkipZigTest;
     }
 }
 
 test "futex2_wait" {
     var lock: std.atomic.Value(u32) = std.atomic.Value(u32).init(1);
     var rc: usize = 0;
     const mask = 0x1;
 
     try futex2_skip_if_unsupported();
 
     // The API for 8,16,64 bit futexes is defined, but as of kernel v6.14
     // (at least) they're not implemented.
     if (false) {
         rc = linux.futex2_wait(&lock.raw, 1, mask, .{ .size = .U8, .private = true }, null, .MONOTONIC);
         try expectEqual(.INVAL, linux.errno(rc));
 
         rc = linux.futex2_wait(&lock.raw, 1, mask, .{ .size = .U16, .private = true }, null, .MONOTONIC);
         try expectEqual(.INVAL, linux.errno(rc));
 
         rc = linux.futex2_wait(&lock.raw, 1, mask, .{ .size = .U64, .private = true }, null, .MONOTONIC);
         try expectEqual(.INVAL, linux.errno(rc));
     }
 
     const flags = linux.FUTEX2_FLAGS{ .size = .U32, .private = true };
     // no-wait, lock state mismatch
     rc = linux.futex2_wait(&lock.raw, 2, mask, flags, null, .MONOTONIC);
     try expectEqual(.AGAIN, linux.errno(rc));
 
     // hit timeout on wait
     rc = linux.futex2_wait(&lock.raw, 1, mask, flags, &.{ .sec = 0, .nsec = 2 }, .MONOTONIC);
     try expectEqual(.TIMEDOUT, linux.errno(rc));
 
     // timeout is absolute
     {
         var curr: linux.timespec = undefined;
         rc = linux.clock_gettime(.MONOTONIC, &curr); // gettime() uses platform timespec
         try expectEqual(0, rc);
 
         // ... but futex2_wait always uses 64-bit timespec
         var timeout: linux.kernel_timespec = .{
             .sec = curr.sec,
             .nsec = curr.nsec + 2,
         };
         rc = linux.futex2_wait(&lock.raw, 1, mask, flags, &timeout, .MONOTONIC);
         try expectEqual(.TIMEDOUT, linux.errno(rc));
     }
 
     rc = linux.futex2_wait(&lock.raw, 1, mask, flags, &.{ .sec = 0, .nsec = 2 }, .REALTIME);
     try expectEqual(.TIMEDOUT, linux.errno(rc));
 }
 
 test "futex2_wake" {
     var lock: std.atomic.Value(u32) = std.atomic.Value(u32).init(1);
 
     try futex2_skip_if_unsupported();
 
     const rc = linux.futex2_wake(&lock.raw, 0xFF, 1, .{ .size = .U32, .private = true });
     try expectEqual(0, rc);
 }
 
 test "futex2_requeue" {
     try futex2_skip_if_unsupported();
 
     const locks = [_]std.atomic.Value(u32){
         std.atomic.Value(u32).init(1),
         std.atomic.Value(u32).init(1),
     };
 
     const futexes = [_]linux.futex2_waitone{
         .{
             .val = 1,
             .uaddr = @intFromPtr(&locks[0].raw),
             .flags = .{ .size = .U32, .private = true },
         },
         .{
             .val = 1,
             .uaddr = @intFromPtr(&locks[1].raw),
             .flags = .{ .size = .U32, .private = true },
         },
     };
 
     const rc = linux.futex2_requeue(&futexes, .{}, 2, 2);
     try expectEqual(0, rc);
 }
 
 test {
     _ = linux.IoUring;
 }