120 lines
2.8 KiB
C++
120 lines
2.8 KiB
C++
#include "capsicum-test.h"
|
|
|
|
#ifdef __FreeBSD__
|
|
#include <sys/param.h>
|
|
#include <sys/proc.h>
|
|
#include <sys/queue.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/sysctl.h>
|
|
#include <sys/user.h>
|
|
#include <libprocstat.h>
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <signal.h>
|
|
|
|
#include <map>
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
bool verbose = false;
|
|
bool tmpdir_on_tmpfs = false;
|
|
bool force_mt = false;
|
|
bool force_nofork = false;
|
|
uid_t other_uid = 0;
|
|
|
|
namespace {
|
|
std::map<std::string, std::string> tmp_paths;
|
|
}
|
|
|
|
const char *TmpFile(const char *p) {
|
|
std::string pathname(p);
|
|
if (tmp_paths.find(pathname) == tmp_paths.end()) {
|
|
std::string fullname = tmpdir + "/" + pathname;
|
|
tmp_paths[pathname] = fullname;
|
|
}
|
|
return tmp_paths[pathname].c_str();
|
|
}
|
|
|
|
char ProcessState(int pid) {
|
|
#ifdef __linux__
|
|
// Open the process status file.
|
|
char s[1024];
|
|
snprintf(s, sizeof(s), "/proc/%d/status", pid);
|
|
FILE *f = fopen(s, "r");
|
|
if (f == NULL) return '\0';
|
|
|
|
// Read the file line by line looking for the state line.
|
|
const char *prompt = "State:\t";
|
|
while (!feof(f)) {
|
|
fgets(s, sizeof(s), f);
|
|
if (!strncmp(s, prompt, strlen(prompt))) {
|
|
fclose(f);
|
|
return s[strlen(prompt)];
|
|
}
|
|
}
|
|
fclose(f);
|
|
return '?';
|
|
#endif
|
|
#ifdef __FreeBSD__
|
|
// First check if the process exists/we have permission to see it. This
|
|
// Avoids warning messages being printed to stderr by libprocstat.
|
|
size_t len = 0;
|
|
int name[4];
|
|
name[0] = CTL_KERN;
|
|
name[1] = KERN_PROC;
|
|
name[2] = KERN_PROC_PID;
|
|
name[3] = pid;
|
|
if (sysctl(name, nitems(name), NULL, &len, NULL, 0) < 0 && errno == ESRCH) {
|
|
if (verbose) fprintf(stderr, "Process %d does not exist\n", pid);
|
|
return '\0'; // No such process.
|
|
}
|
|
unsigned int count = 0;
|
|
struct procstat *prstat = procstat_open_sysctl();
|
|
EXPECT_NE(nullptr, prstat) << "procstat_open_sysctl failed.";
|
|
errno = 0;
|
|
struct kinfo_proc *p = procstat_getprocs(prstat, KERN_PROC_PID, pid, &count);
|
|
if (p == NULL || count == 0) {
|
|
if (verbose) {
|
|
fprintf(stderr, "procstat_getprocs failed with %p/%d: %s\n", (void *)p,
|
|
count, strerror(errno));
|
|
}
|
|
procstat_close(prstat);
|
|
return '\0';
|
|
}
|
|
char result = '\0';
|
|
// See state() in bin/ps/print.c
|
|
switch (p->ki_stat) {
|
|
case SSTOP:
|
|
result = 'T';
|
|
break;
|
|
case SSLEEP:
|
|
if (p->ki_tdflags & TDF_SINTR) /* interruptable (long) */
|
|
result = 'S';
|
|
else
|
|
result = 'D';
|
|
break;
|
|
case SRUN:
|
|
case SIDL:
|
|
result = 'R';
|
|
break;
|
|
case SWAIT:
|
|
case SLOCK:
|
|
// We treat SWAIT/SLOCK as 'S' here (instead of 'W'/'L').
|
|
result = 'S';
|
|
break;
|
|
case SZOMB:
|
|
result = 'Z';
|
|
break;
|
|
default:
|
|
result = '?';
|
|
break;
|
|
}
|
|
procstat_freeprocs(prstat, p);
|
|
procstat_close(prstat);
|
|
if (verbose) fprintf(stderr, "Process %d in state '%c'\n", pid, result);
|
|
return result;
|
|
#endif
|
|
}
|