77 lines
1.6 KiB
C
77 lines
1.6 KiB
C
/*
|
|
* It tests the duplicate mlock result:
|
|
* - the ulimit of lock page is 64k
|
|
* - allocate address area 64k starting from p
|
|
* - mlock [p -- p + 30k]
|
|
* - Then mlock address [ p -- p + 40k ]
|
|
*
|
|
* It should succeed since totally we locked
|
|
* 40k < 64k limitation.
|
|
*
|
|
* It should not be run with CAP_IPC_LOCK.
|
|
*/
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <sys/resource.h>
|
|
#include <sys/capability.h>
|
|
#include <sys/mman.h>
|
|
#include "mlock2.h"
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
struct rlimit new;
|
|
char *p = NULL;
|
|
cap_t cap = cap_init();
|
|
int i;
|
|
|
|
/* drop capabilities including CAP_IPC_LOCK */
|
|
if (cap_set_proc(cap))
|
|
return -1;
|
|
|
|
/* set mlock limits to 64k */
|
|
new.rlim_cur = 65536;
|
|
new.rlim_max = 65536;
|
|
setrlimit(RLIMIT_MEMLOCK, &new);
|
|
|
|
/* test VM_LOCK */
|
|
p = malloc(1024 * 64);
|
|
if (mlock(p, 1024 * 30)) {
|
|
printf("mlock() 30k return failure.\n");
|
|
return -1;
|
|
}
|
|
for (i = 0; i < 10; i++) {
|
|
if (mlock(p, 1024 * 40)) {
|
|
printf("mlock() #%d 40k returns failure.\n", i);
|
|
return -1;
|
|
}
|
|
}
|
|
for (i = 0; i < 10; i++) {
|
|
if (mlock2_(p, 1024 * 40, MLOCK_ONFAULT)) {
|
|
printf("mlock2_() #%d 40k returns failure.\n", i);
|
|
return -1;
|
|
}
|
|
}
|
|
free(p);
|
|
|
|
/* Test VM_LOCKONFAULT */
|
|
p = malloc(1024 * 64);
|
|
if (mlock2_(p, 1024 * 30, MLOCK_ONFAULT)) {
|
|
printf("mlock2_() 30k return failure.\n");
|
|
return -1;
|
|
}
|
|
for (i = 0; i < 10; i++) {
|
|
if (mlock2_(p, 1024 * 40, MLOCK_ONFAULT)) {
|
|
printf("mlock2_() #%d 40k returns failure.\n", i);
|
|
return -1;
|
|
}
|
|
}
|
|
for (i = 0; i < 10; i++) {
|
|
if (mlock(p, 1024 * 40)) {
|
|
printf("mlock() #%d 40k returns failure.\n", i);
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|