955a3f9ad5
This includes various fixes that I submitted recently such as updating the pdkill() tests for the actual implemented behaviour (https://github.com/google/capsicum-test/pull/53) and lots of changes to avoid calling sleep() and replacing it with reliable synchronization (pull requests 49,51,52,53,54). This should make the testsuite more reliable when running on Jenkins. Additionally, process status is now retrieved using libprocstat instead of running `ps` and parsing the output (https://github.com/google/capsicum-test/pull/50). This fixes one previously failing test and speeds up execution. Overall, this update reduces the total runtime from ~60s to about 4-5 seconds.
117 lines
2.8 KiB
C++
117 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(NULL, 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", 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
|
|
}
|