zig

commit 5aaa7d0fbb885db646ebd6557afc8def93ff860f (tree)
parent 805f9b309bee2d1a6c4fa34f1bc2746a8e60c35f
Author: LemonBoy <thatlemon@gmail.com>
Date:   Fri, 27 Sep 2019 18:18:38 +0200

Avoid truncating mmap2 offsets if not multiple of page size

Diffstat:
M
lib/std/os/linux.zig
 | 
24
++++++++++++++++++++++--

1 file changed, 22 insertions(+), 2 deletions(-)
diff --git a/lib/std/os/linux.zig b/lib/std/os/linux.zig
@@ -193,9 +193,29 @@ pub fn umount2(special: [*]const u8, flags: u32) usize {
 
 pub fn mmap(address: ?[*]u8, length: usize, prot: usize, flags: u32, fd: i32, offset: u64) usize {
     if (@hasDecl(@This(), "SYS_mmap2")) {
-        return syscall6(SYS_mmap2, @ptrToInt(address), length, prot, flags, @bitCast(usize, isize(fd)), @truncate(usize, offset / MMAP2_UNIT));
+        // Make sure the offset is also specified in multiples of page size
+        if ((offset & (MMAP2_UNIT - 1)) != 0)
+            return @bitCast(usize, isize(-EINVAL));
+
+        return syscall6(
+            SYS_mmap2,
+            @ptrToInt(address),
+            length,
+            prot,
+            flags,
+            @bitCast(usize, isize(fd)),
+            @truncate(usize, offset / MMAP2_UNIT),
+        );
     } else {
-        return syscall6(SYS_mmap, @ptrToInt(address), length, prot, flags, @bitCast(usize, isize(fd)), offset);
+        return syscall6(
+            SYS_mmap,
+            @ptrToInt(address),
+            length,
+            prot,
+            flags,
+            @bitCast(usize, isize(fd)),
+            offset,
+        );
     }
 }