Mercurial > minori
annotate dep/animone/src/fd/proc.cc @ 270:0718f538c5f9
dep/animone: filter open files by access mode
| author | Paper <paper@paper.us.eu.org> |
|---|---|
| date | Fri, 12 Apr 2024 19:13:50 -0400 |
| parents | 862d0d8619f6 |
| children | b1f625b0227c |
| rev | line source |
|---|---|
| 258 | 1 #include "animone/fd/proc.h" |
| 2 #include "animone.h" | |
| 3 #include "animone/util.h" | |
| 4 | |
| 5 #include <filesystem> | |
| 6 #include <fstream> | |
| 7 #include <sstream> | |
| 8 #include <string> | |
| 9 | |
| 10 #include <fcntl.h> | |
| 11 #include <sys/stat.h> | |
| 12 #include <unistd.h> | |
| 13 | |
| 14 static constexpr std::string_view PROC_LOCATION = "/proc"; | |
| 15 | |
| 16 namespace animone::internal::proc { | |
| 17 | |
| 18 static bool IsRegularFile(std::string link) { | |
| 19 struct stat sb; | |
| 20 if (stat(link.c_str(), &sb) == -1) | |
| 21 return false; | |
| 22 | |
| 23 return S_ISREG(sb.st_mode); | |
| 24 } | |
| 25 | |
| 26 static bool AreFlagsOk(pid_t pid, int fd) { | |
| 27 const std::filesystem::path path = | |
| 28 std::filesystem::path(PROC_LOCATION) / std::to_string(pid) / "fdinfo" / std::to_string(fd); | |
| 29 | |
| 30 std::ifstream file(path); | |
| 31 if (!file) | |
| 32 return false; | |
| 33 | |
| 34 int flags = 0; | |
| 35 for (std::string line; std::getline(file, line);) | |
| 36 if (line.find("flags:", 0) == 0) | |
| 37 flags = util::StringToInt(line.substr(line.find_last_not_of("0123456789") + 1)); | |
| 38 | |
|
270
0718f538c5f9
dep/animone: filter open files by access mode
Paper <paper@paper.us.eu.org>
parents:
258
diff
changeset
|
39 /* check if the file was opened in a write mode */ |
|
0718f538c5f9
dep/animone: filter open files by access mode
Paper <paper@paper.us.eu.org>
parents:
258
diff
changeset
|
40 int accflags = flags & O_ACCMODE; |
|
0718f538c5f9
dep/animone: filter open files by access mode
Paper <paper@paper.us.eu.org>
parents:
258
diff
changeset
|
41 if (accflags == O_WRONLY || accflags == O_RDWR) |
| 258 | 42 return false; |
| 43 | |
| 44 return true; | |
| 45 } | |
| 46 | |
| 47 static bool GetFilenameFromFd(std::string link, std::string& out) { | |
| 48 /* /proc is a "virtual filesystem", so we have to guess the path size. yippee! */ | |
| 49 constexpr size_t OUT_MAX = (1ul << 15); // 32KiB | |
| 50 out.resize(32); | |
| 51 | |
| 52 ssize_t exe_used = 0; | |
| 53 do { | |
| 54 out.resize(out.length() * 2); | |
| 55 | |
| 56 exe_used = readlink(link.c_str(), &out.front(), out.length()); | |
| 57 if (exe_used == (ssize_t)-1 || exe_used < (ssize_t)1) | |
| 58 return false; // we got a bad result. SAD! | |
| 59 } while (out.length() < OUT_MAX && exe_used >= static_cast<ssize_t>(out.length())); | |
| 60 | |
| 61 out.resize(out.find('\0')); | |
| 62 | |
| 63 return true; | |
| 64 } | |
| 65 | |
| 66 static bool IsSystemFile(const std::string& path) { | |
| 67 static constexpr std::array<std::string_view, 9> invalid_paths = {"/boot", "/dev", "/bin", "/usr", "/opt", | |
| 68 "/proc", "/var", "/etc", "/dev"}; | |
| 69 | |
|
270
0718f538c5f9
dep/animone: filter open files by access mode
Paper <paper@paper.us.eu.org>
parents:
258
diff
changeset
|
70 for (const auto& invalid_path : invalid_paths) |
|
0718f538c5f9
dep/animone: filter open files by access mode
Paper <paper@paper.us.eu.org>
parents:
258
diff
changeset
|
71 if (!path.rfind(invalid_path, 0)) |
| 258 | 72 return true; |
| 73 | |
| 74 return false; | |
| 75 } | |
| 76 | |
| 77 bool GetProcessName(pid_t pid, std::string& result) { | |
| 78 const std::filesystem::path path = std::filesystem::path(PROC_LOCATION) / std::to_string(pid) / "comm"; | |
| 79 | |
| 80 if (!util::ReadFile(path, result)) | |
| 81 return false; | |
| 82 | |
| 83 result.erase(std::remove(result.begin(), result.end(), '\n'), result.end()); | |
| 84 return true; | |
| 85 } | |
| 86 | |
| 87 bool EnumerateOpenProcesses(process_proc_t process_proc) { | |
| 88 bool success = false; | |
| 89 | |
| 90 for (const auto& dir : std::filesystem::directory_iterator{PROC_LOCATION}) { | |
| 91 Process proc; | |
| 92 | |
| 93 try { | |
| 94 proc.pid = util::StringToInt(dir.path().stem()); | |
| 95 success = true; | |
| 96 } catch (std::invalid_argument const& ex) { | |
| 97 continue; | |
| 98 } | |
| 99 | |
| 100 if (!GetProcessName(proc.pid, proc.name)) | |
| 101 continue; | |
| 102 | |
| 103 if (!process_proc(proc)) | |
| 104 return false; | |
| 105 } | |
| 106 | |
| 107 return success; | |
| 108 } | |
| 109 | |
| 110 bool EnumerateOpenFiles(const std::set<pid_t>& pids, open_file_proc_t open_file_proc) { | |
| 111 if (!open_file_proc) | |
| 112 return false; | |
| 113 | |
| 114 for (const auto& pid : pids) { | |
| 115 const std::filesystem::path path = std::filesystem::path(PROC_LOCATION) / std::to_string(pid) / "fd"; | |
| 116 | |
| 117 for (const auto& dir : std::filesystem::directory_iterator{path}) { | |
| 118 if (!AreFlagsOk(pid, util::StringToInt(dir.path().stem()))) | |
| 119 continue; | |
| 120 | |
| 121 std::string name; | |
| 122 if (!GetFilenameFromFd(dir.path(), name)) | |
| 123 continue; | |
| 124 | |
| 125 if (!IsRegularFile(name)) | |
| 126 continue; | |
| 127 | |
| 128 if (IsSystemFile(name)) | |
| 129 continue; | |
| 130 | |
| 131 if (!open_file_proc({pid, name})) | |
| 132 return false; | |
| 133 } | |
| 134 } | |
| 135 return true; | |
| 136 } | |
| 137 | |
| 138 } // namespace animia::internal::proc |