Mercurial > minori

comparison dep/animia/src/fd/bsd.cc @ 138:28842a8d0c6b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
dep/animia: huge refactor (again...) but this time, it actually compiles! and it WORKS! (on win32... not sure about other platforms...) configuring players is still not supported: at some point I'll prune something up...
author Paper <mrpapersonic@gmail.com>
date Sun, 12 Nov 2023 04:53:19 -0500
parents
children 478f3b366199
comparison
equal deleted inserted replaced
137:69db40272acd 138:28842a8d0c6b
1 /**
2 * bsd.cpp
3 * - provides support for most* versions of BSD
4 * - this also works for OS X :)
5 * more technical details: this is essentially a wrapper
6 * around the very C-like BSD system functions that are...
7 * kind of unnatural to use in modern C++.
8 **/
9 #include <fcntl.h>
10 #include <iostream>
11 #include <string>
12 #include <sys/sysctl.h>
13 #include <sys/types.h>
14 #include <sys/user.h>
15 #include <unordered_map>
16 #include <vector>
17 #ifdef __FreeBSD__
18 # include <libutil.h>
19 #elif defined(__APPLE__)
20 # include <libproc.h>
21 #endif
22
23 namespace animia::internal::unix {
24
25 /* this is a cleaned up version of a function from... Apple?
26 ...anyway, what it essentially does is gets the size and stuff from
27 sysctl() and reserves the space in a vector to store the PIDs */
28 bool GetAllPids(std::set<pid_t>& pids) {
29 struct kinfo_proc* result = NULL;
30 size_t length = 0;
31 static const int name[] = {CTL_KERN, KERN_PROC, KERN_PROC_ALL, 0};
32
33 /* get appropriate length from sysctl()
34 note: the reason this isn't checked is actually because this will
35 *always* return an error on OS X (or... maybe I'm doing it wrong :) ) */
36 sysctl((int*)name, (sizeof(name) / sizeof(*name)) - 1, NULL, &length, NULL, 0);
37
38 result = (struct kinfo_proc*)malloc(length);
39 if (result == NULL)
40 return std::vector<int>();
41
42 /* TODO: this might actually return ENOMEM if the amount of file handles changes between the
43 original sysctl() call and this one, which is technically possible */
44 if (sysctl((int*)name, (sizeof(name) / sizeof(*name)) - 1, result, &length, NULL, 0) == ENOMEM) {
45 assert(result != NULL);
46 free(result);
47 throw std::bad_alloc();
48 }
49
50 /* add pids to our vector */
51 pids.reserve(length / sizeof(*result));
52 for (int i = 0; i < length / sizeof(*result); i++)
53 pids.push_back(result[i].kp_proc.p_pid);
54 }
55
56 bool GetProcessName(pid_t pid, std::string& result) {
57 #ifdef __FreeBSD__
58 struct kinfo_proc* proc = kinfo_getproc(pid);
59 if (!proc)
60 return false;
61 result = proc->ki_comm;
62
63 /* FreeBSD manpage for kinfo_getproc():
64 "The pointer was obtained by an internal call to malloc(3) and
65 must be freed by the caller with a call to free(3)." */
66 free(proc);
67
68 return true;
69 #elif defined(__APPLE__)
70 struct proc_bsdinfo proc;
71
72 int st = proc_pidinfo(pid, PROC_PIDTBSDINFO, 0, &proc, PROC_PIDTBSDINFO_SIZE);
73 if (st != PROC_PIDTBSDINFO_SIZE)
74 return false;
75
76 /* fixme: is this right? pbi_comm is an alternative, but it reduces the string size to
77 16 chars. does pbi_name do the same, or is it different? */
78 result = proc.pbi_name;
79 return true;
80 #endif
81 }
82
83 /* this only works on OS X :( */
84 bool EnumerateOpenFiles(const std::set<pid_t>& pids, std::vector<std::tuple<pid_t, std::string>>& files) {
85 for (const auto& pid : pids) {
86 int bufsz = proc_pidinfo(pid, PROC_PIDLISTFDS, 0, NULL, 0);
87 if (bufsz == -1)
88 return false;
89
90 struct proc_fdinfo* info = (struct proc_fdinfo*)malloc(bufsz);
91 if (!info)
92 return false;
93
94 proc_pidinfo(pid, PROC_PIDLISTFDS, 0, info, bufsz);
95
96 for (int i = 0; i < bufsz / sizeof(info[0]); i++) {
97 if (info[i].proc_fdtype == PROX_FDTYPE_VNODE) {
98 struct vnode_fdinfowithpath vnodeInfo;
99
100 int sz = proc_pidfdinfo(pid, info[i].proc_fd, PROC_PIDFDVNODEPATHINFO, &vnodeInfo, PROC_PIDFDVNODEPATHINFO_SIZE);
101 if (sz != PROC_PIDFDVNODEPATHINFO_SIZE)
102 return false;
103
104 /* this doesn't work!
105 if (vnodeInfo.pfi.fi_openflags & O_WRONLY || vnodeInfo.pfi.fi_openflags & O_RDWR)
106 continue;
107 */
108
109 files.push_back({pid, vnodeInfo.pvip.vip_path});
110 }
111 }
112 }
113 return true;
114 }
115
116 } // namespace animia::internal::unix