258
+ − 1 /* kvm.cc: provides support for *BSD.
+ − 2 */
+ − 3
+ − 4 #include "animone/fd/kvm.h"
+ − 5 #include "animone.h"
+ − 6 #include "animone/fd.h"
+ − 7
+ − 8 #include <sys/file.h>
+ − 9 #include <sys/filedesc.h>
+ − 10 #include <sys/param.h>
+ − 11 #include <sys/queue.h>
+ − 12 #include <sys/sysctl.h>
+ − 13 #include <sys/types.h>
+ − 14 #include <sys/user.h>
+ − 15 #include <sys/vnode.h>
+ − 16
+ − 17 #include <kvm.h>
+ − 18 #ifdef LIBUTIL
+ − 19 # include <libutil.h>
+ − 20 #endif
+ − 21
+ − 22 #include <string>
+ − 23
+ − 24 namespace animone::internal::kvm {
+ − 25
+ − 26 bool GetProcessName(pid_t pid, std::string& name) {
+ − 27 char errbuf[_POSIX2_LINE_MAX];
+ − 28 kvm_t* kernel = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, errbuf);
+ − 29 if (!kernel)
+ − 30 return false;
+ − 31
+ − 32 int entries = 0;
+ − 33 struct kinfo_proc* kinfo = kvm_getprocs(kernel, KERN_PROC_PID, pid, &entries);
+ − 34 if (!kinfo) {
+ − 35 kvm_close(kernel);
+ − 36 return false;
+ − 37 }
+ − 38
+ − 39 if (entries < 1) {
+ − 40 kvm_close(kernel);
+ − 41 return false;
+ − 42 }
+ − 43
+ − 44 name = kinfo[0].ki_paddr->p_comm;
+ − 45
+ − 46 return true;
+ − 47 }
+ − 48
+ − 49 /* Most of the BSDs share the common kvm library,
+ − 50 * so accessing this information can be trivial.
+ − 51 */
+ − 52 bool EnumerateOpenProcesses(process_proc_t process_proc) {
+ − 53 char errbuf[_POSIX2_LINE_MAX];
+ − 54 kvm_t* kernel = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, errbuf);
+ − 55 if (!kernel)
+ − 56 return false;
+ − 57
+ − 58 int entries = 0;
+ − 59 struct kinfo_proc* kinfo = kvm_getprocs(kernel, KERN_PROC_ALL, 0, &entries);
+ − 60 if (!kinfo) {
+ − 61 kvm_close(kernel);
+ − 62 return false;
+ − 63 }
+ − 64
+ − 65 for (int i = 0; i < entries; i++) {
+ − 66 if (!process_proc({kinfo[i].ki_paddr->p_pid, kinfo[i].ki_paddr->p_comm})) {
+ − 67 kvm_close(kernel);
+ − 68 return false;
+ − 69 }
+ − 70 }
+ − 71
+ − 72 kvm_close(kernel);
+ − 73
+ − 74 return true;
+ − 75 }
+ − 76
+ − 77 bool EnumerateOpenFiles(std::set<pid_t>& pids, open_file_proc_t open_file_proc) {
+ − 78 #ifdef __OpenBSD__
+ − 79 char errbuf[_POSIX2_LINE_MAX];
+ − 80 kvm_t* kernel = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, errbuf);
+ − 81 if (!kernel)
+ − 82 return false;
+ − 83
+ − 84 for (const auto& pid : pids) {
+ − 85 int cnt;
+ − 86 struct kinfo_file* kfile = kvm_getfiles(kernel, KERN_FILE_BYPID, pid, &cnt);
+ − 87 if (!kfile) {
+ − 88 kvm_close(kernel);
+ − 89 return false;
+ − 90 }
+ − 91
+ − 92 for (int i = 0; i < cnt; i++) {
+ − 93 if (!open_file_proc({pid, kfile[i].kf_path})) {
+ − 94 kvm_close(kernel);
+ − 95 return false;
+ − 96 }
+ − 97 }
+ − 98 }
+ − 99
+ − 100 kvm_close(kernel);
+ − 101
+ − 102 return true;
+ − 103 #elif defined(LIBUTIL)
+ − 104 /* does this code even work? */
+ − 105 for (const auto& pid : pids) {
+ − 106 int cnt;
+ − 107 std::unique_ptr<struct kinfo_file[]> files(kinfo_getfile(pid, &cnt));
+ − 108 if (!files)
+ − 109 return false;
+ − 110
+ − 111 for (int i = 0; i < cnt; i++) {
+ − 112 const struct kinfo_file& current = files[i];
+ − 113 if (current.kf_vnode_type != KF_VTYPE_VREG)
+ − 114 continue;
+ − 115
+ − 116 if (!open_file_proc({pid, current.kf_path}))
+ − 117 return false;
+ − 118 }
+ − 119 }
+ − 120
+ − 121 return true;
+ − 122 #elif defined(__NetBSD__)
+ − 123 for (const auto& pid : pids) {
+ − 124 int mib[6] = {CTL_KERN, KERN_FILE2, KERN_FILE_BYPID, pid, sizeof(struct kinfo_file), 0};
+ − 125
+ − 126 size_t len = 0;
+ − 127 if (sysctl(mib, sizeof(mib) / sizeof(mib[0]), NULL, &len, NULL, 0) == -1)
+ − 128 return false;
+ − 129
+ − 130 mib[5] = len / sizeof(struct kinfo_file);
+ − 131
+ − 132 std::unique_ptr<struct kinfo_file[]> buf(new struct kinfo_file[mib[5]]);
+ − 133 if (!buf)
+ − 134 return false;
+ − 135
+ − 136 if (sysctl(mib, sizeof(mib) / sizeof(mib[0]), buf.get(), &len, NULL, 0) == -1)
+ − 137 return false;
+ − 138
+ − 139 /* TODO: check kfile[i].ki_ofileflags */
+ − 140 for (size_t i = 0; i < mib[5]; i++)
+ − 141 if (!open_file_proc({pid, kfile[i].kf_path}))
+ − 142 return false;
+ − 143 }
+ − 144
+ − 145 return true;
+ − 146 #else
+ − 147 return false;
+ − 148 #endif
+ − 149 }
+ − 150
+ − 151 } // namespace animone::internal::kvm
