| ... | @@ -794,6 +794,21 @@ pub const IO_Uring = struct { | ... | @@ -794,6 +794,21 @@ pub const IO_Uring = struct { |
| 794 | return sqe; | 794 | return sqe; |
| 795 | } | 795 | } |
| 796 | | 796 | |
| | 797 | /// Queues (but does not submit) an SQE to perform a `symlinkat(2)`. |
| | 798 | /// Returns a pointer to the SQE. |
| | 799 | pub fn symlinkat( |
| | 800 | self: *IO_Uring, |
| | 801 | user_data: u64, |
| | 802 | target: [*:0]const u8, |
| | 803 | new_dir_fd: os.fd_t, |
| | 804 | link_path: [*:0]const u8, |
| | 805 | ) !*io_uring_sqe { |
| | 806 | const sqe = try self.get_sqe(); |
| | 807 | io_uring_prep_symlinkat(sqe, target, new_dir_fd, link_path); |
| | 808 | sqe.user_data = user_data; |
| | 809 | return sqe; |
| | 810 | } |
| | 811 | |
| 797 | /// Registers an array of file descriptors. | 812 | /// Registers an array of file descriptors. |
| 798 | /// Every time a file descriptor is put in an SQE and submitted to the kernel, the kernel must | 813 | /// Every time a file descriptor is put in an SQE and submitted to the kernel, the kernel must |
| 799 | /// retrieve a reference to the file, and once I/O has completed the file reference must be | 814 | /// retrieve a reference to the file, and once I/O has completed the file reference must be |
| ... | @@ -1389,6 +1404,22 @@ pub fn io_uring_prep_mkdirat( | ... | @@ -1389,6 +1404,22 @@ pub fn io_uring_prep_mkdirat( |
| 1389 | io_uring_prep_rw(.MKDIRAT, sqe, dir_fd, @ptrToInt(path), mode, 0); | 1404 | io_uring_prep_rw(.MKDIRAT, sqe, dir_fd, @ptrToInt(path), mode, 0); |
| 1390 | } | 1405 | } |
| 1391 | | 1406 | |
| | 1407 | pub fn io_uring_prep_symlinkat( |
| | 1408 | sqe: *io_uring_sqe, |
| | 1409 | target: [*:0]const u8, |
| | 1410 | new_dir_fd: os.fd_t, |
| | 1411 | link_path: [*:0]const u8, |
| | 1412 | ) void { |
| | 1413 | io_uring_prep_rw( |
| | 1414 | .SYMLINKAT, |
| | 1415 | sqe, |
| | 1416 | new_dir_fd, |
| | 1417 | @ptrToInt(target), |
| | 1418 | 0, |
| | 1419 | @ptrToInt(link_path), |
| | 1420 | ); |
| | 1421 | } |
| | 1422 | |
| 1392 | test "structs/offsets/entries" { | 1423 | test "structs/offsets/entries" { |
| 1393 | if (builtin.os.tag != .linux) return error.SkipZigTest; | 1424 | if (builtin.os.tag != .linux) return error.SkipZigTest; |
| 1394 | | 1425 | |
| ... | @@ -2523,3 +2554,52 @@ test "mkdirat" { | ... | @@ -2523,3 +2554,52 @@ test "mkdirat" { |
| 2523 | // Validate that the directory exist | 2554 | // Validate that the directory exist |
| 2524 | _ = try std.fs.cwd().openDir(path, .{}); | 2555 | _ = try std.fs.cwd().openDir(path, .{}); |
| 2525 | } | 2556 | } |
| | 2557 | |
| | 2558 | test "symlinkat" { |
| | 2559 | if (builtin.os.tag != .linux) return error.SkipZigTest; |
| | 2560 | |
| | 2561 | var ring = IO_Uring.init(1, 0) catch |err| switch (err) { |
| | 2562 | error.SystemOutdated => return error.SkipZigTest, |
| | 2563 | error.PermissionDenied => return error.SkipZigTest, |
| | 2564 | else => return err, |
| | 2565 | }; |
| | 2566 | defer ring.deinit(); |
| | 2567 | |
| | 2568 | const path = "test_io_uring_symlinkat"; |
| | 2569 | const link_path = "test_io_uring_symlinkat_link"; |
| | 2570 | |
| | 2571 | const file = try std.fs.cwd().createFile(path, .{ .truncate = true, .mode = 0o666 }); |
| | 2572 | defer { |
| | 2573 | file.close(); |
| | 2574 | std.fs.cwd().deleteFile(path) catch {}; |
| | 2575 | std.fs.cwd().deleteFile(link_path) catch {}; |
| | 2576 | } |
| | 2577 | |
| | 2578 | // Submit symlinkat |
| | 2579 | |
| | 2580 | var sqe = try ring.symlinkat( |
| | 2581 | 0x12121212, |
| | 2582 | path, |
| | 2583 | linux.AT.FDCWD, |
| | 2584 | link_path, |
| | 2585 | ); |
| | 2586 | try testing.expectEqual(linux.IORING_OP.SYMLINKAT, sqe.opcode); |
| | 2587 | try testing.expectEqual(@as(i32, linux.AT.FDCWD), sqe.fd); |
| | 2588 | try testing.expectEqual(@as(u32, 1), try ring.submit()); |
| | 2589 | |
| | 2590 | const cqe = try ring.copy_cqe(); |
| | 2591 | switch (cqe.err()) { |
| | 2592 | .SUCCESS => {}, |
| | 2593 | // This kernel's io_uring does not yet implement symlinkat (kernel version < 5.15) |
| | 2594 | .INVAL => return error.SkipZigTest, |
| | 2595 | else => |errno| std.debug.panic("unhandled errno: {}", .{errno}), |
| | 2596 | } |
| | 2597 | try testing.expectEqual(linux.io_uring_cqe{ |
| | 2598 | .user_data = 0x12121212, |
| | 2599 | .res = 0, |
| | 2600 | .flags = 0, |
| | 2601 | }, cqe); |
| | 2602 | |
| | 2603 | // Validate that the symlink exist |
| | 2604 | _ = try std.fs.cwd().openFile(link_path, .{}); |
| | 2605 | } |
Vincent Rischmann <vincent@rischmann.fr> 2021-12-12 18:43:31+01:00
lib