mirrors/cosmopolitan
1
0
Fork 0
mirror of https://github.com/jart/cosmopolitan.git synced 2025-02-07 15:03:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add mlock() and munlock()

Browse source
This commit is contained in:
Justine Tunney 2023-04-27 10:42:52 -07:00
parent 7c9ef924bf
commit fec139f423
No known key found for this signature in database
GPG key ID: BE714B4575D6E328
8 changed files with 115 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
libc/calls/calls.h
View file

@ -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) */

44
libc/calls/mlock.c Normal file
View file

@ -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);
}
}

44
libc/calls/munlock.c Normal file
View file

@ -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);
}
}

10
libc/nt/kernel32/VirtualLock.s
View file

@ -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

10
libc/nt/kernel32/VirtualUnlock.s
View file

@ -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

4
libc/nt/master.sh
View file

@ -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

2
libc/nt/memory.h
View file

@ -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);

2
libc/runtime/runtime.h
View file

@ -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);