Mercurial > minori
annotate dep/animia/src/fd/proc.cc @ 172:45a0967485f1
graph, statistics: make my code a little less messy
| author | Paper <mrpapersonic@gmail.com> |
|---|---|
| date | Tue, 28 Nov 2023 13:22:35 -0500 |
| parents | 54c5d80a737e |
| children | 5be17d636aee |
| rev | line source |
|---|---|
| 166 | 1 #include "animia/fd/proc.h" |
|
152
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
2 #include "animia.h" |
|
156
cdf79282d647
dep/animia: add VERY early x11 window stuff
Paper <mrpapersonic@gmail.com>
parents:
154
diff
changeset
|
3 #include "animia/util.h" |
|
140
1e696863b54c
dep/animia: remove superfluous includes
Paper <mrpapersonic@gmail.com>
parents:
139
diff
changeset
|
4 |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
5 #include <algorithm> |
|
156
cdf79282d647
dep/animia: add VERY early x11 window stuff
Paper <mrpapersonic@gmail.com>
parents:
154
diff
changeset
|
6 #include <cstring> |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
7 #include <filesystem> |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
8 #include <fstream> |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
9 #include <sstream> |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
10 #include <string> |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
11 #include <unordered_map> |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
12 #include <vector> |
|
140
1e696863b54c
dep/animia: remove superfluous includes
Paper <mrpapersonic@gmail.com>
parents:
139
diff
changeset
|
13 |
|
156
cdf79282d647
dep/animia: add VERY early x11 window stuff
Paper <mrpapersonic@gmail.com>
parents:
154
diff
changeset
|
14 #include <dirent.h> |
|
140
1e696863b54c
dep/animia: remove superfluous includes
Paper <mrpapersonic@gmail.com>
parents:
139
diff
changeset
|
15 #include <fcntl.h> |
|
1e696863b54c
dep/animia: remove superfluous includes
Paper <mrpapersonic@gmail.com>
parents:
139
diff
changeset
|
16 #include <sys/stat.h> |
|
1e696863b54c
dep/animia: remove superfluous includes
Paper <mrpapersonic@gmail.com>
parents:
139
diff
changeset
|
17 #include <unistd.h> |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
18 |
| 166 | 19 #ifdef FREEBSD |
| 20 # include <sys/types.h> | |
| 21 # include <sys/user.h> | |
| 22 # include <libutil.h> | |
| 23 #endif | |
| 24 | |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
25 #define PROC_LOCATION "/proc" |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
26 |
| 166 | 27 namespace animia::internal::proc { |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
28 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
29 /* this uses dirent instead of std::filesystem; it would make a bit |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
30 more sense to use the latter, but this is platform dependent already :) */ |
|
163
44c5e6dd9488
dep/animia/osx: move GetProcessName to util/osx so we can use it in quartz
Paper <mrpapersonic@gmail.com>
parents:
156
diff
changeset
|
31 static std::vector<std::string> GetAllFilesInDir(const std::string& _dir) { |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
32 std::vector<std::string> ret; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
33 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
34 DIR* dir = opendir(_dir.c_str()); |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
35 if (!dir) |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
36 return ret; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
37 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
38 struct dirent* dp; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
39 while ((dp = readdir(dir)) != NULL) { |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
40 if (!(!strcmp(dp->d_name, ".") || !strcmp(dp->d_name, ".."))) |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
41 ret.push_back(_dir + "/" + dp->d_name); |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
42 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
43 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
44 closedir(dir); |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
45 return ret; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
46 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
47 |
|
163
44c5e6dd9488
dep/animia/osx: move GetProcessName to util/osx so we can use it in quartz
Paper <mrpapersonic@gmail.com>
parents:
156
diff
changeset
|
48 static std::string Basename(const std::string& path) { |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
49 return path.substr(path.find_last_of("/") + 1, path.length()); |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
50 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
51 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
52 static bool IsRegularFile(std::string link) { |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
53 struct stat sb; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
54 if (stat(link.c_str(), &sb) == -1) |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
55 return false; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
56 return S_ISREG(sb.st_mode); |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
57 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
58 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
59 static bool AreFlagsOk(pid_t pid, int fd) { |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
60 const std::string path = PROC_LOCATION "/" + std::to_string(pid) + "/fdinfo/" + std::to_string(fd); |
|
164
99fdf5a90b0f
fd/linux: avoid reading buffers multiple times
Paper <mrpapersonic@gmail.com>
parents:
163
diff
changeset
|
61 |
|
99fdf5a90b0f
fd/linux: avoid reading buffers multiple times
Paper <mrpapersonic@gmail.com>
parents:
163
diff
changeset
|
62 std::ifstream file(path.c_str()); |
|
99fdf5a90b0f
fd/linux: avoid reading buffers multiple times
Paper <mrpapersonic@gmail.com>
parents:
163
diff
changeset
|
63 if (!file) |
|
99fdf5a90b0f
fd/linux: avoid reading buffers multiple times
Paper <mrpapersonic@gmail.com>
parents:
163
diff
changeset
|
64 return false; |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
65 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
66 int flags = 0; |
|
164
99fdf5a90b0f
fd/linux: avoid reading buffers multiple times
Paper <mrpapersonic@gmail.com>
parents:
163
diff
changeset
|
67 for (std::string line; std::getline(file, line); ) |
|
99fdf5a90b0f
fd/linux: avoid reading buffers multiple times
Paper <mrpapersonic@gmail.com>
parents:
163
diff
changeset
|
68 if (line.find("flags:", 0) == 0) |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
69 flags = std::stoi(line.substr(line.find_last_not_of("0123456789") + 1)); |
|
164
99fdf5a90b0f
fd/linux: avoid reading buffers multiple times
Paper <mrpapersonic@gmail.com>
parents:
163
diff
changeset
|
70 |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
71 if (flags & O_WRONLY || flags & O_RDWR) |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
72 return false; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
73 return true; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
74 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
75 |
| 166 | 76 static bool GetFilenameFromFd(std::string link, std::string& out) { |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
77 /* gets around stupid linux limitation where /proc doesn't |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
78 give actual filesize readings */ |
| 166 | 79 constexpr size_t OUT_MAX = (1 << 15); // 32KiB |
| 80 out.resize(1024); | |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
81 |
| 166 | 82 for (ssize_t exe_used = 0; |
| 83 out.length() < OUT_MAX && exe_used >= (ssize_t)(out.length() - 1); | |
| 84 out.resize(out.length() * 2)) { | |
| 85 exe_used = readlink(link.c_str(), &out.front(), out.length()); | |
| 86 if (exe_used == (ssize_t)-1 || exe_used < (ssize_t)1) | |
| 87 return false; | |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
88 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
89 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
90 |
| 154 | 91 static std::string GetProcessName(pid_t pid) { |
| 166 | 92 std::string result; |
|
152
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
93 |
| 166 | 94 #ifdef FREEBSD |
| 95 struct kinfo_proc* proc = kinfo_getproc(pid); | |
| 96 if (!proc) | |
| 97 return ""; | |
| 98 result = proc->ki_comm; | |
| 99 free(proc); | |
| 100 #elif defined(LINUX) | |
| 101 const std::string path = PROC_LOCATION "/" + std::to_string(pid) + "/comm"; | |
| 154 | 102 |
|
152
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
103 if (!util::ReadFile(path, result)) |
| 154 | 104 return ""; |
|
152
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
105 |
|
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
106 result.erase(std::remove(result.begin(), result.end(), '\n'), result.end()); |
| 166 | 107 #endif |
| 154 | 108 |
| 109 return result; | |
|
152
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
110 } |
|
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
111 |
| 166 | 112 bool ProcFdTools::EnumerateOpenProcesses(process_proc_t process_proc) { |
|
144
e6668085e24d
dep/animia: fix many bugs in the linux code
Paper <mrpapersonic@gmail.com>
parents:
143
diff
changeset
|
113 bool success = false; |
|
e6668085e24d
dep/animia: fix many bugs in the linux code
Paper <mrpapersonic@gmail.com>
parents:
143
diff
changeset
|
114 for (const auto& dir : GetAllFilesInDir(PROC_LOCATION)) { |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
115 pid_t pid; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
116 try { |
|
144
e6668085e24d
dep/animia: fix many bugs in the linux code
Paper <mrpapersonic@gmail.com>
parents:
143
diff
changeset
|
117 pid = std::stoul(Basename(dir)); |
|
e6668085e24d
dep/animia: fix many bugs in the linux code
Paper <mrpapersonic@gmail.com>
parents:
143
diff
changeset
|
118 success = true; |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
119 } catch (std::invalid_argument) { |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
120 continue; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
121 } |
|
152
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
122 if (!process_proc({pid, GetProcessName(pid)})) |
|
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
123 return false; |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
124 } |
|
144
e6668085e24d
dep/animia: fix many bugs in the linux code
Paper <mrpapersonic@gmail.com>
parents:
143
diff
changeset
|
125 return success; |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
126 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
127 |
| 166 | 128 bool ProcFdTools::EnumerateOpenFiles(const std::set<pid_t>& pids, open_file_proc_t open_file_proc) { |
|
152
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
129 if (!open_file_proc) |
|
144
e6668085e24d
dep/animia: fix many bugs in the linux code
Paper <mrpapersonic@gmail.com>
parents:
143
diff
changeset
|
130 return false; |
|
e6668085e24d
dep/animia: fix many bugs in the linux code
Paper <mrpapersonic@gmail.com>
parents:
143
diff
changeset
|
131 |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
132 for (const auto& pid : pids) { |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
133 const std::string path = PROC_LOCATION "/" + std::to_string(pid) + "/fd"; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
134 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
135 for (const auto& dir : GetAllFilesInDir(path)) { |
|
144
e6668085e24d
dep/animia: fix many bugs in the linux code
Paper <mrpapersonic@gmail.com>
parents:
143
diff
changeset
|
136 if (!AreFlagsOk(pid, std::stoi(Basename(dir)))) |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
137 continue; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
138 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
139 std::string name = GetFilenameFromFd(dir); |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
140 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
141 if (!IsRegularFile(name)) |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
142 continue; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
143 |
|
152
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
144 if (!open_file_proc({pid, name})) |
|
8700806c2cc2
dep/animia: awesome new breaking changes!
Paper <mrpapersonic@gmail.com>
parents:
151
diff
changeset
|
145 return false; |
|
138
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
146 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
147 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
148 return true; |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
149 } |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
150 |
|
28842a8d0c6b
dep/animia: huge refactor (again...)
Paper <mrpapersonic@gmail.com>
parents:
diff
changeset
|
151 } // namespace animia::internal::linux |
