mirror of
https://git.kernel.org/pub/scm/libs/libcap/libcap.git
synced 2026-01-27 09:54:36 +00:00
This is more general than just working with -lpthread threads, and addresses the following bug reports: https://bugzilla.kernel.org/show_bug.cgi?id=219174 https://bugzilla.kernel.org/show_bug.cgi?id=218607 https://bugzilla.kernel.org/show_bug.cgi?id=217476 Signed-off-by: Andrew G. Morgan <morgan@kernel.org>
75 lines
1.8 KiB
C
75 lines
1.8 KiB
C
#ifndef _DEFAULT_SOURCE
|
|
#define _DEFAULT_SOURCE
|
|
#endif
|
|
|
|
#include <errno.h>
|
|
#include <pthread.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sys/capability.h>
|
|
#include <sys/psx_syscall.h>
|
|
#include <sys/prctl.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <unistd.h>
|
|
|
|
static void *thread_fork_exit(void *data) {
|
|
usleep(1234);
|
|
pid_t pid = fork();
|
|
long int start = cap_prctl(PR_GET_KEEPCAPS, 0, 0, 0, 0, 0);
|
|
if (start == -1) {
|
|
perror("FAILED: unable to start");
|
|
exit(1);
|
|
}
|
|
if (pid == 0) {
|
|
if (cap_prctlw(PR_SET_KEEPCAPS, !start, 0, 0, 0, 0) != 0) {
|
|
perror("failed to set proc");
|
|
exit(1);
|
|
}
|
|
if (cap_prctlw(PR_GET_KEEPCAPS, 0, 0, 0, 0, 0) == start) {
|
|
perror("failed to have set forked proc");
|
|
exit(1);
|
|
}
|
|
exit(0);
|
|
}
|
|
int res;
|
|
if (waitpid(pid, &res, 0) != pid || res != 0) {
|
|
printf("FAILED: pid=%d wait returned %d and/or error: %d\n",
|
|
pid, res, errno);
|
|
exit(1);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
int i;
|
|
printf("hello libcap and libpsx ");
|
|
fflush(stdout);
|
|
long int start = cap_prctl(PR_GET_KEEPCAPS, 0, 0, 0, 0, 0);
|
|
if (start == -1) {
|
|
perror("FAILED: to actually start");
|
|
exit(1);
|
|
}
|
|
pthread_t ignored[10];
|
|
for (i = 0; i < 10; i++) {
|
|
pthread_create(&ignored[i], NULL, thread_fork_exit, NULL);
|
|
printf("."); /* because of fork, this may print double */
|
|
fflush(stdout); /* try to limit the above effect */
|
|
if (cap_prctlw(PR_SET_KEEPCAPS, i & 1, 0, 0, 0, 0)) {
|
|
perror("failed to set proc");
|
|
exit(1);
|
|
}
|
|
/*
|
|
* This should validate all threads think the same thing,
|
|
* and none of the fork() calls mess anything up.
|
|
*/
|
|
if (cap_prctlw(PR_GET_KEEPCAPS, 0, 0, 0, 0, 0) != (i & 1)) {
|
|
perror("failed to have set proc");
|
|
exit(1);
|
|
}
|
|
usleep(1000);
|
|
}
|
|
printf(" PASSED\n");
|
|
exit(0);
|
|
}
|