Mercurial > minori
diff dep/animia/src/bsd.cpp @ 56:6ff7aabeb9d7
deps: add animia for open files detection
author | Paper <mrpapersonic@gmail.com> |
---|---|
date | Thu, 28 Sep 2023 12:35:21 -0400 |
parents | |
children | 4c6dd5999b39 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/dep/animia/src/bsd.cpp Thu Sep 28 12:35:21 2023 -0400 @@ -0,0 +1,102 @@ +/** + * bsd.cpp + * - provides support for most* versions of BSD + * - this *should* also work for OS X + * more technical details: this is essentially a wrapper + * around the very C-like BSD system functions that are... + * kind of unnatural to use in modern C++. +**/ +#include <vector> +#include <string> +#include <unordered_map> +#include <sys/types.h> +#include <sys/sysctl.h> +#include <sys/user.h> +#include <fcntl.h> +#include <iostream> +#ifdef __FreeBSD__ +#include <libutil.h> +#elif defined(__APPLE__) +#include <libproc.h> +#endif + +namespace Animia::Unix { + +/* this is a cleaned up version of a function from... Apple? + ...anyway, what it essentially does is gets the size and stuff from + sysctl() and reserves the space in a vector to store the PIDs */ +std::vector<int> get_all_pids() { + std::vector<int> ret; + struct kinfo_proc* result = NULL; + size_t length = 0; + static const int name[] = { CTL_KERN, KERN_PROC, KERN_PROC_ALL, 0 }; + + /* get appropriate length from sysctl() */ + sysctl((int*)name, (sizeof(name) / sizeof(*name)) - 1, NULL, &length, NULL, 0); + + result = (struct kinfo_proc*)malloc(length); + if (result == NULL) + return std::vector<int>(); + + /* actually get our results */ + if (sysctl((int*)name, (sizeof(name) / sizeof(*name)) - 1, result, &length, NULL, 0) == ENOMEM) { + assert(result != NULL); + free(result); + throw std::bad_alloc(); + } + + /* add pids to our vector */ + ret.reserve(length/sizeof(*result)); + for (int i = 0; i < length/sizeof(*result); i++) + ret.push_back(result[i].kp_proc.p_pid); + + return ret; +} + +std::string get_process_name(int pid) { + std::string ret; +#ifdef __FreeBSD__ + struct kinfo_proc* proc = kinfo_getproc(pid); + if (!proc) { + return ""; + ret = proc->ki_comm; + free(proc); +#elif defined(__APPLE__) + struct proc_bsdinfo proc; + + int st = proc_pidinfo(pid, PROC_PIDTBSDINFO, 0, &proc, PROC_PIDTBSDINFO_SIZE); + if (st != PROC_PIDTBSDINFO_SIZE) + return ""; + return proc.pbi_comm; +#endif + return ret; +} + +std::vector<std::string> get_open_files(int pid) { + // initialize buffer + std::vector<std::string> ret; + int bufsz = proc_pidinfo(pid, PROC_PIDLISTFDS, 0, NULL, 0); + struct proc_fdinfo *info = (struct proc_fdinfo *)malloc(bufsz); + proc_pidinfo(pid, PROC_PIDLISTFDS, 0, info, bufsz); + + // iterate over stuff + for (int i = 0; i < bufsz/sizeof(info[0]); i++) { + if (info[i].proc_fdtype == PROX_FDTYPE_VNODE) { + struct vnode_fdinfowithpath vnodeInfo; + proc_pidfdinfo(pid, info[i].proc_fd, PROC_PIDFDVNODEPATHINFO, &vnodeInfo, PROC_PIDFDVNODEPATHINFO_SIZE); + ret.push_back(vnodeInfo.pvip.vip_path); + } + } + return ret; +} + +std::unordered_map<int, std::vector<std::string>> get_all_open_files() { + std::unordered_map<int, std::vector<std::string>> map; + std::vector<int> pids = get_all_pids(); + for (int i: pids) { + map[i] = get_open_files(i); + } + return map; +} + +}