• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Copyright (c) 2019 Richard Palethorpe <rpalethorpe@suse.com>
4  */
5 
6 #include <string.h>
7 
8 #define TST_NO_DEFAULT_MAIN
9 #include "tst_test.h"
10 #include "tst_capability.h"
11 
12 #include "lapi/syscalls.h"
13 
tst_capget(struct tst_cap_user_header * hdr,struct tst_cap_user_data * data)14 int tst_capget(struct tst_cap_user_header *hdr,
15 	       struct tst_cap_user_data *data)
16 {
17 	return tst_syscall(__NR_capget, hdr, data);
18 }
19 
tst_capset(struct tst_cap_user_header * hdr,const struct tst_cap_user_data * data)20 int tst_capset(struct tst_cap_user_header *hdr,
21 	       const struct tst_cap_user_data *data)
22 {
23 	return tst_syscall(__NR_capset, hdr, data);
24 }
25 
do_cap_drop(uint32_t * set,uint32_t mask,const struct tst_cap * cap)26 static void do_cap_drop(uint32_t *set, uint32_t mask, const struct tst_cap *cap)
27 {
28 	if (*set & mask) {
29 		tst_res(TINFO, "Dropping %s(%d)", cap->name, cap->id);
30 		*set &= ~mask;
31 	}
32 }
33 
do_cap_req(uint32_t * permitted,uint32_t * effective,uint32_t mask,const struct tst_cap * cap)34 static void do_cap_req(uint32_t *permitted, uint32_t *effective, uint32_t mask,
35 		       const struct tst_cap *cap)
36 {
37 	if (!(*permitted & mask))
38 		tst_brk(TCONF, "Need %s(%d)", cap->name, cap->id);
39 
40 	if (!(*effective & mask)) {
41 		tst_res(TINFO, "Permitting %s(%d)", cap->name, cap->id);
42 		*effective |= mask;
43 	}
44 }
45 
tst_cap_action(struct tst_cap * cap)46 void tst_cap_action(struct tst_cap *cap)
47 {
48 	struct tst_cap_user_header hdr = {
49 		.version = 0x20080522,
50 		.pid = tst_syscall(__NR_gettid),
51 	};
52 	struct tst_cap_user_data cur[2] = { {0} };
53 	struct tst_cap_user_data new[2] = { {0} };
54 	uint32_t act = cap->action;
55 	uint32_t *pE = &new[CAP_TO_INDEX(cap->id)].effective;
56 	uint32_t *pP = &new[CAP_TO_INDEX(cap->id)].permitted;
57 	uint32_t mask = CAP_TO_MASK(cap->id);
58 
59 	if (tst_capget(&hdr, cur))
60 		tst_brk(TBROK | TTERRNO, "tst_capget()");
61 
62 	memcpy(new, cur, sizeof(new));
63 
64 	switch (act) {
65 	case TST_CAP_DROP:
66 		do_cap_drop(pE, mask, cap);
67 		break;
68 	case TST_CAP_REQ:
69 		do_cap_req(pP, pE, mask, cap);
70 		break;
71 	default:
72 		tst_brk(TBROK, "Unrecognised action %d", cap->action);
73 	}
74 
75 	if (!memcmp(cur, new, sizeof(new)))
76 		return;
77 
78 	if (tst_capset(&hdr, new))
79 		tst_brk(TBROK | TERRNO, "tst_capset(%s)", cap->name);
80 }
81 
tst_cap_setup(struct tst_cap * caps,unsigned int action_mask)82 void tst_cap_setup(struct tst_cap *caps, unsigned int action_mask)
83 {
84 	struct tst_cap *cap;
85 
86 	for (cap = caps; cap->action; cap++) {
87 		if (cap->action & action_mask)
88 			tst_cap_action(cap);
89 	}
90 }
91