2 * Copyright (c) 2016 The FreeBSD Foundation
5 * This software was developed by Konstantin Belousov <kib@FreeBSD.org>
6 * under sponsorship from the FreeBSD Foundation.
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 #include <sys/cdefs.h>
31 __FBSDID("$FreeBSD$");
33 #include <sys/procctl.h>
58 str2pid(const char *str)
63 res = strtol(str, &tail, 0);
65 warnx("non-numeric pid");
72 #define KPTI_USAGE "|kpti"
77 #define LA_USAGE "|la48|la57"
86 fprintf(stderr, "Usage: proccontrol -m (aslr|protmax|trace|trapcap|"
87 "stackgap"KPTI_USAGE LA_USAGE") [-q] "
88 "[-s (enable|disable)] [-p pid | command]\n");
93 main(int argc, char *argv[])
95 int arg, ch, error, mode;
97 bool enable, do_command, query;
103 while ((ch = getopt(argc, argv, "m:qs:p:")) != -1) {
106 if (strcmp(optarg, "aslr") == 0)
108 else if (strcmp(optarg, "protmax") == 0)
110 else if (strcmp(optarg, "trace") == 0)
112 else if (strcmp(optarg, "trapcap") == 0)
114 else if (strcmp(optarg, "stackgap") == 0)
115 mode = MODE_STACKGAP;
117 else if (strcmp(optarg, "kpti") == 0)
121 else if (strcmp(optarg, "la57") == 0)
123 else if (strcmp(optarg, "la48") == 0)
130 if (strcmp(optarg, "enable") == 0)
132 else if (strcmp(optarg, "disable") == 0)
138 pid = str2pid(optarg);
151 do_command = argc != 0;
153 if (pid != -1 || query)
156 } else if (pid == -1) {
163 error = procctl(P_PID, pid, PROC_ASLR_STATUS, &arg);
166 error = procctl(P_PID, pid, PROC_TRACE_STATUS, &arg);
169 error = procctl(P_PID, pid, PROC_TRAPCAP_STATUS, &arg);
172 error = procctl(P_PID, pid, PROC_PROTMAX_STATUS, &arg);
175 error = procctl(P_PID, pid, PROC_STACKGAP_STATUS, &arg);
179 error = procctl(P_PID, pid, PROC_KPTI_STATUS, &arg);
185 error = procctl(P_PID, pid, PROC_LA_STATUS, &arg);
193 err(1, "procctl status");
196 switch (arg & ~PROC_ASLR_ACTIVE) {
197 case PROC_ASLR_FORCE_ENABLE:
198 printf("force enabled");
200 case PROC_ASLR_FORCE_DISABLE:
201 printf("force disabled");
203 case PROC_ASLR_NOFORCE:
204 printf("not forced");
207 if ((arg & PROC_ASLR_ACTIVE) != 0)
208 printf(", active\n");
210 printf(", not active\n");
214 printf("disabled\n");
216 printf("enabled, no debugger\n");
218 printf("enabled, traced by %d\n", arg);
222 case PROC_TRAPCAP_CTL_ENABLE:
225 case PROC_TRAPCAP_CTL_DISABLE:
226 printf("disabled\n");
231 switch (arg & ~PROC_PROTMAX_ACTIVE) {
232 case PROC_PROTMAX_FORCE_ENABLE:
233 printf("force enabled");
235 case PROC_PROTMAX_FORCE_DISABLE:
236 printf("force disabled");
238 case PROC_PROTMAX_NOFORCE:
239 printf("not forced");
242 if ((arg & PROC_PROTMAX_ACTIVE) != 0)
243 printf(", active\n");
245 printf(", not active\n");
248 switch (arg & (PROC_STACKGAP_ENABLE |
249 PROC_STACKGAP_DISABLE)) {
250 case PROC_STACKGAP_ENABLE:
253 case PROC_STACKGAP_DISABLE:
254 printf("disabled\n");
257 switch (arg & (PROC_STACKGAP_ENABLE_EXEC |
258 PROC_STACKGAP_DISABLE_EXEC)) {
259 case PROC_STACKGAP_ENABLE_EXEC:
260 printf("enabled after exec\n");
262 case PROC_STACKGAP_DISABLE_EXEC:
263 printf("disabled after exec\n");
269 switch (arg & ~PROC_KPTI_STATUS_ACTIVE) {
270 case PROC_KPTI_CTL_ENABLE_ON_EXEC:
273 case PROC_KPTI_CTL_DISABLE_ON_EXEC:
277 if ((arg & PROC_KPTI_STATUS_ACTIVE) != 0)
278 printf(", active\n");
280 printf(", not active\n");
286 switch (arg & ~(PROC_LA_STATUS_LA48 |
287 PROC_LA_STATUS_LA57)) {
288 case PROC_LA_CTL_LA48_ON_EXEC:
289 printf("la48 on exec");
291 case PROC_LA_CTL_LA57_ON_EXEC:
292 printf("la57 on exec");
294 case PROC_LA_CTL_DEFAULT_ON_EXEC:
295 printf("default on exec");
298 if ((arg & PROC_LA_STATUS_LA48) != 0)
299 printf(", la48 active\n");
300 else if ((arg & PROC_LA_STATUS_LA57) != 0)
301 printf(", la57 active\n");
308 arg = enable ? PROC_ASLR_FORCE_ENABLE :
309 PROC_ASLR_FORCE_DISABLE;
310 error = procctl(P_PID, pid, PROC_ASLR_CTL, &arg);
313 arg = enable ? PROC_TRACE_CTL_ENABLE :
314 PROC_TRACE_CTL_DISABLE;
315 error = procctl(P_PID, pid, PROC_TRACE_CTL, &arg);
318 arg = enable ? PROC_TRAPCAP_CTL_ENABLE :
319 PROC_TRAPCAP_CTL_DISABLE;
320 error = procctl(P_PID, pid, PROC_TRAPCAP_CTL, &arg);
323 arg = enable ? PROC_PROTMAX_FORCE_ENABLE :
324 PROC_PROTMAX_FORCE_DISABLE;
325 error = procctl(P_PID, pid, PROC_PROTMAX_CTL, &arg);
328 arg = enable ? PROC_STACKGAP_ENABLE_EXEC :
329 (PROC_STACKGAP_DISABLE |
330 PROC_STACKGAP_DISABLE_EXEC);
331 error = procctl(P_PID, pid, PROC_STACKGAP_CTL, &arg);
335 arg = enable ? PROC_KPTI_CTL_ENABLE_ON_EXEC :
336 PROC_KPTI_CTL_DISABLE_ON_EXEC;
337 error = procctl(P_PID, pid, PROC_KPTI_CTL, &arg);
342 arg = enable ? PROC_LA_CTL_LA57_ON_EXEC :
343 PROC_LA_CTL_DEFAULT_ON_EXEC;
344 error = procctl(P_PID, pid, PROC_LA_CTL, &arg);
347 arg = enable ? PROC_LA_CTL_LA48_ON_EXEC :
348 PROC_LA_CTL_DEFAULT_ON_EXEC;
349 error = procctl(P_PID, pid, PROC_LA_CTL, &arg);
357 err(1, "procctl ctl");
359 error = execvp(argv[0], argv);