File: pidfd_of_thread.c

package info (click to toggle)
criu 4.2-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 11,500 kB
  • sloc: ansic: 139,280; python: 7,484; sh: 3,824; java: 2,799; makefile: 2,659; asm: 1,137; perl: 206; xml: 117; exp: 45
file content (114 lines) | stat: -rw-r--r-- 2,350 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#include <sched.h>
#include <sys/stat.h>
#include <sys/statfs.h>
#include <fcntl.h>

#include "zdtmtst.h"
#include "lock.h"

const char *test_doc = "Check C/R of pidfds that point to threads\n";
const char *test_author = "Bhavik Sachdev <b.sachdev1904@gmail.com>";

/* see also: https://codebrowser.dev/glibc/glibc/sysdeps/unix/sysv/linux/tst-clone3.c.html */

#ifndef PIDFD_THREAD
#define PIDFD_THREAD	O_EXCL
#endif

#ifndef PIDFD_SIGNAL_THREAD
#define PIDFD_SIGNAL_THREAD		(1UL << 0)
#endif

#ifndef PID_FS_MAGIC
#define PID_FS_MAGIC 0x50494446
#endif

static long get_fs_type(int lfd)
{
	struct statfs fst;

	if (fstatfs(lfd, &fst)) {
		return -1;
	}
	return fst.f_type;
}

static int pidfd_open(pid_t pid, unsigned int flags)
{
	return syscall(__NR_pidfd_open, pid, flags);
}

static int pidfd_send_signal(int pidfd, int sig, siginfo_t* info, unsigned int flags)
{
	return syscall(__NR_pidfd_send_signal, pidfd, sig, info, flags);
}

static int thread_func(void *a)
{
	test_waitsig();
	return 0;
}

#define CTID_INIT_VAL 1

int main(int argc, char* argv[])
{
	char st[64 * 1024] __attribute__ ((aligned));
	pid_t tid;
	int pidfd, test_pidfd;
	futex_t exited;

	int clone_flags = CLONE_THREAD;
	clone_flags |= CLONE_VM | CLONE_SIGHAND;
	clone_flags |= CLONE_CHILD_CLEARTID;

	test_init(argc, argv);

	test_pidfd = pidfd_open(getpid(), 0);
	if (test_pidfd < 0) {
		pr_perror("pidfd_open() failed");
		return 1;
	}

	/* PIDFD_THREAD, PIDFD_SIGNAL_THREAD are supported only with pidfs */
	if (get_fs_type(test_pidfd) != PID_FS_MAGIC) {
		test_daemon();
		test_waitsig();
		skip("pidfs not supported.");
		close(test_pidfd);
		return 0;
	}
	close(test_pidfd);

	futex_set(&exited, CTID_INIT_VAL);

	tid = clone(thread_func, st + sizeof(st), clone_flags, NULL, NULL, NULL, &(exited.raw));
	if (tid == -1) {
	    pr_perror("clone() failed");
	    return 1;
	}

	test_msg("Successfully created a thread with tid: %d\n", tid);
	pidfd = pidfd_open(tid, PIDFD_THREAD);
	if (pidfd < 0) {
	    pr_perror("pidfd_open() failed");
	    return 1;
	}

	test_daemon();
	test_waitsig();

	if (pidfd_send_signal(pidfd, SIGTERM, NULL, PIDFD_SIGNAL_THREAD)) {
	    pr_perror("pidfd_send_signal() failed");
	    fail();
	    close(pidfd);
	    return 1;
	}

	test_msg("Waiting for thread to exit\n");
	futex_wait_until(&exited, 0);

	pass();
	close(pidfd);
	return 0;
}