Add mlock() and munlock()

libc/calls/calls.h

@@ -23,6 +23,7 @@
 #define _POSIX_READER_WRITER_LOCKS        _POSIX_VERSION
 #define _POSIX_SEMAPHORES                 _POSIX_VERSION
 #define _POSIX_SHARED_MEMORY_OBJECTS      _POSIX_VERSION
+#define _POSIX_MEMLOCK_RANGE              _POSIX_VERSION
 
 #define EOF      -1         /* end of file */
 #define WEOF     -1u        /* end of file (multibyte) */

libc/calls/mlock.c

@@ -0,0 +1,44 @@
+/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
+│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8                                :vi│
+╞══════════════════════════════════════════════════════════════════════════════╡
+│ Copyright 2023 Justine Alexandra Roberts Tunney                              │
+│                                                                              │
+│ Permission to use, copy, modify, and/or distribute this software for         │
+│ any purpose with or without fee is hereby granted, provided that the         │
+│ above copyright notice and this permission notice appear in all copies.      │
+│                                                                              │
+│ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL                │
+│ WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED                │
+│ WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE             │
+│ AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL         │
+│ DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR        │
+│ PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER               │
+│ TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR             │
+│ PERFORMANCE OF THIS SOFTWARE.                                                │
+╚─────────────────────────────────────────────────────────────────────────────*/
+#include "libc/calls/calls.h"
+#include "libc/calls/syscall_support-nt.internal.h"
+#include "libc/dce.h"
+#include "libc/nt/memory.h"
+#include "libc/runtime/runtime.h"
+
+static textwindows int sys_mlock_nt(const void *addr, size_t len) {
+  if (VirtualLock(addr, len)) {
+    return 0;
+  } else {
+    return __winerr();
+  }
+}
+
+/**
+ * Locks virtual memory interval into RAM, preventing it from swapping.
+ *
+ * @return 0 on success, or -1 w/ errno
+ */
+int mlock(const void *addr, size_t len) {
+  if (!IsWindows()) {
+    return sys_mlock(addr, len);
+  } else {
+    return sys_mlock_nt(addr, len);
+  }
+}

libc/calls/munlock.c

@@ -0,0 +1,44 @@
+/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
+│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8                                :vi│
+╞══════════════════════════════════════════════════════════════════════════════╡
+│ Copyright 2023 Justine Alexandra Roberts Tunney                              │
+│                                                                              │
+│ Permission to use, copy, modify, and/or distribute this software for         │
+│ any purpose with or without fee is hereby granted, provided that the         │
+│ above copyright notice and this permission notice appear in all copies.      │
+│                                                                              │
+│ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL                │
+│ WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED                │
+│ WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE             │
+│ AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL         │
+│ DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR        │
+│ PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER               │
+│ TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR             │
+│ PERFORMANCE OF THIS SOFTWARE.                                                │
+╚─────────────────────────────────────────────────────────────────────────────*/
+#include "libc/calls/calls.h"
+#include "libc/calls/syscall_support-nt.internal.h"
+#include "libc/dce.h"
+#include "libc/nt/memory.h"
+#include "libc/runtime/runtime.h"
+
+static textwindows int sys_munlock_nt(const void *addr, size_t len) {
+  if (VirtualUnlock(addr, len)) {
+    return 0;
+  } else {
+    return __winerr();
+  }
+}
+
+/**
+ * Unlocks virtual memory interval from RAM, to permit swapping.
+ *
+ * @return 0 on success, or -1 w/ errno
+ */
+int munlock(const void *addr, size_t len) {
+  if (!IsWindows()) {
+    return sys_munlock(addr, len);
+  } else {
+    return sys_munlock_nt(addr, len);
+  }
+}

libc/nt/kernel32/VirtualLock.s

@@ -1,2 +1,12 @@
 .include "o/libc/nt/codegen.inc"
 .imp	kernel32,__imp_VirtualLock,VirtualLock,0
+
+	.text.windows
+VirtualLock:
+	push	%rbp
+	mov	%rsp,%rbp
+	.profilable
+	mov	__imp_VirtualLock(%rip),%rax
+	jmp	__sysv2nt
+	.endfn	VirtualLock,globl
+	.previous

libc/nt/kernel32/VirtualUnlock.s

@@ -1,2 +1,12 @@
 .include "o/libc/nt/codegen.inc"
 .imp	kernel32,__imp_VirtualUnlock,VirtualUnlock,0
+
+	.text.windows
+VirtualUnlock:
+	push	%rbp
+	mov	%rsp,%rbp
+	.profilable
+	mov	__imp_VirtualUnlock(%rip),%rax
+	jmp	__sysv2nt
+	.endfn	VirtualUnlock,globl
+	.previous

libc/nt/master.sh

@@ -1046,11 +1046,11 @@ imp	'VirtualAllocEx'					VirtualAllocEx						kernel32	0	5
 imp	'VirtualAllocExNuma'					VirtualAllocExNuma					kernel32	0
 imp	'VirtualFree'						VirtualFree						kernel32	0	3
 imp	'VirtualFreeEx'						VirtualFreeEx						kernel32	0
-imp	'VirtualLock'						VirtualLock						kernel32	0
+imp	'VirtualLock'						VirtualLock						kernel32	0	2
+imp	'VirtualUnlock'						VirtualUnlock						kernel32	0	2
 imp	'VirtualProtectEx'					VirtualProtectEx					kernel32	0
 imp	'VirtualQuery'						VirtualQuery						kernel32	0	3
 imp	'VirtualQueryEx'					VirtualQueryEx						kernel32	0
-imp	'VirtualUnlock'						VirtualUnlock						kernel32	0
 imp	'WTSGetActiveConsoleSessionId'				WTSGetActiveConsoleSessionId				kernel32	1497
 imp	'WaitCommEvent'						WaitCommEvent						kernel32	0
 imp	'WaitForDebugEvent'					WaitForDebugEvent					kernel32	0

libc/nt/memory.h

@@ -66,6 +66,8 @@ void *VirtualAlloc(void *opt_lpAddress, uint64_t dwSize,
 bool32 VirtualFree(void *lpAddress, uint64_t dwSize, uint32_t dwFreeType);
 bool32 VirtualProtect(void *lpAddress, uint64_t dwSize, uint32_t flNewProtect,
                       uint32_t *lpflOldProtect) paramsnonnull();
+bool32 VirtualLock(void *lpAddress, size_t dwSize);
+bool32 VirtualUnlock(void *lpAddress, size_t dwSize);
 uint64_t VirtualQuery(const void *lpAddress,
                       struct NtMemoryBasicInformation *lpBuffer,
                       uint64_t dwLength);

libc/runtime/runtime.h

@@ -69,6 +69,8 @@ void *mremap(void *, size_t, size_t, int, ...);
 int munmap(void *, uint64_t);
 int mprotect(void *, uint64_t, int) privileged;
 int msync(void *, size_t, int);
+int mlock(const void *, size_t);
+int munlock(const void *, size_t);
 void *sbrk(intptr_t);
 int brk(void *);
 long gethostid(void);