Mercurial > minori
annotate dep/animia/src/fd/libutil.cc @ 198:bc1ae1810855
dep/animia: switch from using classes to global functions
the old idea was ok, but sort of hackish; this method doesn't use classes
at all, and this way (especially important!) we can do wayland stuff AND x11
at the same time, which wasn't really possible without stupid workarounds in
the other method
| author | Paper <mrpapersonic@gmail.com> |
|---|---|
| date | Sun, 24 Dec 2023 02:59:42 -0500 |
| parents | 54c5d80a737e |
| children |
| rev | line source |
|---|---|
| 166 | 1 /* This file uses the FreeBSD-specific libprocstat |
| 2 */ | |
| 3 | |
| 4 #include "animia/fd/libutil.h" | |
| 5 #include "animia/fd.h" | |
| 6 #include "animia.h" | |
| 7 | |
| 8 #include <sys/types.h> | |
| 9 #include <sys/sysctl.h> | |
| 10 #include <sys/user.h> | |
| 11 #include <libutil.h> | |
| 12 | |
| 13 #include <memory> | |
| 14 | |
| 15 namespace animia::internal::libutil { | |
| 16 | |
| 17 static bool IsSystemFile(const std::string& file) { | |
| 18 return (!file.find("/usr") || !file.find("/lib") || !file.find("/dev") || | |
| 19 !file.find("/proc")); | |
| 20 } | |
| 21 | |
|
198
bc1ae1810855
dep/animia: switch from using classes to global functions
Paper <mrpapersonic@gmail.com>
parents:
166
diff
changeset
|
22 bool EnumerateOpenProcesses(process_proc_t process_proc) { |
| 166 | 23 static const int mib[] = {CTL_KERN, KERN_PROC, KERN_PROC_ALL, 0}; |
| 24 size_t length = 0; | |
| 25 | |
| 26 sysctl((int*)mib, (sizeof(mib) / sizeof(*mib)) - 1, NULL, &length, NULL, 0); | |
| 27 | |
| 28 std::unique_ptr<struct kinfo_proc[]> result; | |
| 29 result.reset(new struct kinfo_proc[length]); | |
| 30 | |
| 31 if (!result.get()) | |
| 32 return false; | |
| 33 | |
| 34 /* actually get our results */ | |
| 35 if (sysctl((const int*)mib, (sizeof(mib) / sizeof(*mib)) - 1, result.get(), &length, NULL, 0) == ENOMEM) { | |
| 36 result.reset(); | |
| 37 throw std::bad_alloc(); | |
| 38 } | |
| 39 | |
| 40 if (length < sizeof(struct kinfo_proc)) | |
| 41 return false; | |
| 42 | |
| 43 for (int i = 0; i < length / sizeof(result[0]); i++) | |
| 44 if (!process_proc({result[i].ki_pid, result[i].ki_comm})) | |
| 45 return false; | |
| 46 | |
| 47 return true; | |
| 48 } | |
| 49 | |
|
198
bc1ae1810855
dep/animia: switch from using classes to global functions
Paper <mrpapersonic@gmail.com>
parents:
166
diff
changeset
|
50 bool EnumerateOpenFiles(const std::set<pid_t>& pids, open_file_proc_t open_file_proc) { |
| 166 | 51 for (const auto& pid : pids) { |
| 52 int cnt; | |
| 53 std::unique_ptr<struct kinfo_file[]> files; | |
| 54 files.reset(kinfo_getfile(pid, &cnt)); | |
| 55 if (!files.get()) | |
| 56 return false; | |
| 57 | |
| 58 for (int i = 0; i < cnt; i++) { | |
| 59 const struct kinfo_file& current = files[i]; | |
| 60 if (current.kf_vnode_type != KF_VTYPE_VREG) | |
| 61 continue; | |
| 62 | |
| 63 if (!open_file_proc({pid, current.kf_path})) | |
| 64 return false; | |
| 65 } | |
| 66 } | |
| 67 | |
| 68 return true; | |
| 69 } | |
| 70 | |
|
198
bc1ae1810855
dep/animia: switch from using classes to global functions
Paper <mrpapersonic@gmail.com>
parents:
166
diff
changeset
|
71 } |