aboutsummaryrefslogtreecommitdiff
path: root/src/librc
diff options
context:
space:
mode:
Diffstat (limited to 'src/librc')
-rw-r--r--src/librc/librc-daemon.c14
-rw-r--r--src/librc/librc.h1
2 files changed, 13 insertions, 2 deletions
diff --git a/src/librc/librc-daemon.c b/src/librc/librc-daemon.c
index e345f5f7..7ea66faa 100644
--- a/src/librc/librc-daemon.c
+++ b/src/librc/librc-daemon.c
@@ -182,6 +182,7 @@ librc_hidden_def(rc_find_pids)
# define _GET_KINFO_UID(kp) (kp.ki_ruid)
# define _GET_KINFO_COMM(kp) (kp.ki_comm)
# define _GET_KINFO_PID(kp) (kp.ki_pid)
+# define _KVM_PATH _PATH_DEVNULL
# else
# define _KVM_GETPROC2
# define _KINFO_PROC kinfo_proc2
@@ -189,6 +190,7 @@ librc_hidden_def(rc_find_pids)
# define _GET_KINFO_UID(kp) (kp.p_ruid)
# define _GET_KINFO_COMM(kp) (kp.p_comm)
# define _GET_KINFO_PID(kp) (kp.p_pid)
+# define _KVM_PATH NULL
# endif
pid_t *rc_find_pids (const char *exec, const char *cmd,
@@ -205,8 +207,10 @@ pid_t *rc_find_pids (const char *exec, const char *cmd,
pid_t *tmp;
int npids = 0;
- if ((kd = kvm_openfiles (NULL, NULL, NULL, O_RDONLY, errbuf)) == NULL) {
- fprintf (stderr, "kvm_open: %s", errbuf);
+ if ((kd = kvm_openfiles (_KVM_PATH, _KVM_PATH,
+ NULL, O_RDONLY, errbuf)) == NULL)
+ {
+ fprintf (stderr, "kvm_open: %s\n", errbuf);
return (NULL);
}
@@ -215,6 +219,12 @@ pid_t *rc_find_pids (const char *exec, const char *cmd,
#else
kp = kvm_getprocs (kd, KERN_PROC_PROC, 0, &processes);
#endif
+ if ((kp == NULL && processes > 0) || (kp != NULL && processes < 0)) {
+ fprintf (stderr, "kvm_getprocs: %s\n", kvm_geterr (kd));
+ kvm_close (kd);
+ return (NULL);
+ }
+
for (i = 0; i < processes; i++) {
pid_t p = _GET_KINFO_PID (kp[i]);
if (pid != 0 && pid != p)
diff --git a/src/librc/librc.h b/src/librc/librc.h
index 0f6212fe..7a5d9a0e 100644
--- a/src/librc/librc.h
+++ b/src/librc/librc.h
@@ -45,6 +45,7 @@
#include <fcntl.h>
#include <libgen.h>
#include <limits.h>
+#include <paths.h>
#include <regex.h>
#include <stdarg.h>
#include <stdbool.h>