Page MenuHomeFreeBSD
Differential D5603 Diff 21029 usr.bin/proccontrol/proccontrol.c

Changeset View

Standalone View

View Options

usr.bin/proccontrol/proccontrol.c

Show All 36 Lines
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <unistd.h> #include <unistd.h>
enum { enum {
MODE_INVALID, MODE_INVALID,
MODE_TRACE, MODE_TRACE,
MODE_TRAPCAP, MODE_TRAPCAP,
MODE_ASLR,
}; };
static pid_t static pid_t
str2pid(const char *str) str2pid(const char *str)
{ {
pid_t res; pid_t res;
char *tail; char *tail;
res = strtol(str, &tail, 0); res = strtol(str, &tail, 0);
if (*tail != '\0') { if (*tail != '\0') {
warnx("non-numeric pid"); warnx("non-numeric pid");
return (-1); return (-1);
} }
return (res); return (res);
} }
static void __dead2 static void __dead2
usage(void) usage(void)
{ {
fprintf(stderr, "Usage: proccontrol -m (trace|trapcap) [-q] " fprintf(stderr, "Usage: proccontrol -m (trace|trapcap|aslr) [-q] "
emasteUnsubmitted
Not Done
Inline Actions

The usage at least should be in alpha order, which will match the eventual man page.

emaste: The usage at least should be in alpha order, which will match the eventual man page.
"[-s (enable|disable)] [-p pid | command]\n"); "[-s (enable|disable)] [-p pid | command]\n");
exit(1); exit(1);
} }
int int
main(int argc, char *argv[]) main(int argc, char *argv[])
{ {
int arg, ch, error, mode; int arg, ch, error, mode;
pid_t pid; pid_t pid;
bool enable, do_command, query; bool enable, do_command, query;
mode = MODE_INVALID; mode = MODE_INVALID;
enable = true; enable = true;
pid = -1; pid = -1;
query = false; query = false;
while ((ch = getopt(argc, argv, "m:qs:p:")) != -1) { while ((ch = getopt(argc, argv, "m:qs:p:")) != -1) {
switch (ch) { switch (ch) {
case 'm': case 'm':
if (strcmp(optarg, "trace") == 0) if (strcmp(optarg, "trace") == 0)
markjUnsubmitted
Not Done
Inline Actions

I think I missed earlier discussion of this, but it seems wrong that we don't support multiple -m values.

markj: I think I missed earlier discussion of this, but it seems wrong that we don't support multiple…
kibAuthorUnsubmitted
Done
Inline Actions

We do not need to, you can chain proccontrol, same as e.g. nice/nohup/sudo etc.

kib: We do not need to, you can chain proccontrol, same as e.g. nice/nohup/sudo etc.
mode = MODE_TRACE; mode = MODE_TRACE;
else if (strcmp(optarg, "trapcap") == 0) else if (strcmp(optarg, "trapcap") == 0)
mode = MODE_TRAPCAP; mode = MODE_TRAPCAP;
else if (strcmp(optarg, "aslr") == 0)
mode = MODE_ASLR;
else else
usage(); usage();
break; break;
case 's': case 's':
if (strcmp(optarg, "enable") == 0) if (strcmp(optarg, "enable") == 0)
enable = true; enable = true;
else if (strcmp(optarg, "disable") == 0) else if (strcmp(optarg, "disable") == 0)
enable = false; enable = false;
Show All 26 Linesmain(int argc, char *argv[])
if (query) { if (query) {
switch (mode) { switch (mode) {
case MODE_TRACE: case MODE_TRACE:
error = procctl(P_PID, pid, PROC_TRACE_STATUS, &arg); error = procctl(P_PID, pid, PROC_TRACE_STATUS, &arg);
break; break;
case MODE_TRAPCAP: case MODE_TRAPCAP:
error = procctl(P_PID, pid, PROC_TRAPCAP_STATUS, &arg); error = procctl(P_PID, pid, PROC_TRAPCAP_STATUS, &arg);
break; break;
case MODE_ASLR:
error = procctl(P_PID, pid, PROC_ASLR_STATUS, &arg);
break;
default: default:
usage(); usage();
break; break;
} }
if (error != 0) if (error != 0)
err(1, "procctl status"); err(1, "procctl status");
switch (mode) { switch (mode) {
case MODE_TRACE: case MODE_TRACE:
Show All 9 Lines case MODE_TRAPCAP:
case PROC_TRAPCAP_CTL_ENABLE: case PROC_TRAPCAP_CTL_ENABLE:
printf("enabled\n"); printf("enabled\n");
break; break;
case PROC_TRAPCAP_CTL_DISABLE: case PROC_TRAPCAP_CTL_DISABLE:
printf("disabled\n"); printf("disabled\n");
break; break;
} }
break; break;
case MODE_ASLR:
switch (arg & ~PROC_ASLR_ACTIVE) {
case PROC_ASLR_FORCE_ENABLE:
printf("force enabled");
break;
case PROC_ASLR_FORCE_DISABLE:
printf("force disabled");
break;
case PROC_ASLR_NOFORCE:
printf("not forced");
break;
} }
if ((arg & PROC_ASLR_ACTIVE) != 0)
printf(", active\n");
else
printf(", not active\n");
break;
}
} else { } else {
switch (mode) { switch (mode) {
case MODE_TRACE: case MODE_TRACE:
arg = enable ? PROC_TRACE_CTL_ENABLE : arg = enable ? PROC_TRACE_CTL_ENABLE :
PROC_TRACE_CTL_DISABLE; PROC_TRACE_CTL_DISABLE;
error = procctl(P_PID, pid, PROC_TRACE_CTL, &arg); error = procctl(P_PID, pid, PROC_TRACE_CTL, &arg);
break; break;
case MODE_TRAPCAP: case MODE_TRAPCAP:
arg = enable ? PROC_TRAPCAP_CTL_ENABLE : arg = enable ? PROC_TRAPCAP_CTL_ENABLE :
PROC_TRAPCAP_CTL_DISABLE; PROC_TRAPCAP_CTL_DISABLE;
error = procctl(P_PID, pid, PROC_TRAPCAP_CTL, &arg); error = procctl(P_PID, pid, PROC_TRAPCAP_CTL, &arg);
break;
case MODE_ASLR:
arg = enable ? PROC_ASLR_FORCE_ENABLE :
PROC_ASLR_FORCE_DISABLE;
error = procctl(P_PID, pid, PROC_ASLR_CTL, &arg);
break; break;
default: default:
usage(); usage();
break; break;
} }
if (error != 0) if (error != 0)
err(1, "procctl ctl"); err(1, "procctl ctl");
if (do_command) { if (do_command) {
error = execvp(argv[0], argv); error = execvp(argv[0], argv);
err(1, "exec"); err(1, "exec");
} }
} }
exit(0); exit(0);
} }