hyprpm: move to system directories for storing plugins (#10211)

1 files changed, 110 insertions, 0 deletions

diff --git a/hyprpm/src/helpers/Sys.cpp b/hyprpm/src/helpers/Sys.cpp
new file mode 100644
index 00000000..3c6501ee
--- /dev/null
+++ b/hyprpm/src/helpers/Sys.cpp
@@ -0,0 +1,110 @@
+#include "Sys.hpp"
+#include "Die.hpp"
+#include "StringUtils.hpp"
+#include <pwd.h>
+#include <unistd.h>
+#include <print>
+#include <filesystem>
+
+#include <hyprutils/os/Process.hpp>
+#include <hyprutils/string/VarList.hpp>
+using namespace Hyprutils::OS;
+using namespace Hyprutils::String;
+
+static const std::vector<const char*> SUPERUSER_BINARIES = {
+    "sudo",
+    "doas",
+    "run0",
+};
+
+static bool executableExistsInPath(const std::string& exe) {
+    if (!getenv("PATH"))
+        return false;
+
+    static CVarList paths(getenv("PATH"), 0, ':', true);
+
+    for (auto& p : paths) {
+        std::string     path = p + std::string{"/"} + exe;
+        std::error_code ec;
+        if (!std::filesystem::exists(path, ec) || ec)
+            continue;
+
+        if (!std::filesystem::is_regular_file(path, ec) || ec)
+            continue;
+
+        auto stat = std::filesystem::status(path, ec);
+        if (ec)
+            continue;
+
+        auto perms = stat.permissions();
+
+        return std::filesystem::perms::none != (perms & std::filesystem::perms::others_exec);
+    }
+
+    return false;
+}
+
+static std::pair<std::string, int> execAndGet(std::string cmd, bool noRedirect = false) {
+    if (!noRedirect)
+        cmd += " 2>&1";
+
+    CProcess proc("/bin/sh", {"-c", cmd});
+
+    if (!proc.runSync())
+        return {"error", 1};
+
+    return {proc.stdOut(), proc.exitCode()};
+}
+
+int NSys::getUID() {
+    const auto UID   = getuid();
+    const auto PWUID = getpwuid(UID);
+    return PWUID ? PWUID->pw_uid : UID;
+}
+
+int NSys::getEUID() {
+    const auto UID   = geteuid();
+    const auto PWUID = getpwuid(UID);
+    return PWUID ? PWUID->pw_uid : UID;
+}
+
+bool NSys::isSuperuser() {
+    return getuid() != geteuid() || !geteuid();
+}
+
+std::string NSys::runAsSuperuser(const std::string& cmd) {
+    for (const auto& SB : SUPERUSER_BINARIES) {
+        if (!executableExistsInPath(SB))
+            continue;
+
+        const auto RESULT = execAndGet(std::string{SB} + " /bin/sh -c \"" + cmd + "\"", true);
+
+        if (RESULT.second != 0)
+            Debug::die("Failed to run a command as sudo. This could be due to an invalid password, or a hyprpm bug.");
+
+        return RESULT.first;
+    }
+
+    Debug::die("Failed to find a superuser binary. Supported: sudo, doas, run0.");
+    return "";
+}
+
+void NSys::cacheSudo() {
+    // "caches" the sudo so that the prompt later doesn't pop up in a weird spot
+    // sudo will not ask us again for a moment
+    runAsSuperuser("echo e > /dev/null");
+}
+
+void NSys::dropSudo() {
+    for (const auto& SB : SUPERUSER_BINARIES) {
+        if (!executableExistsInPath(SB))
+            continue;
+
+        if (SB == std::string_view{"sudo"})
+            execAndGet("sudo -k");
+        else
+            std::println("{}", infoString("Don't know how to drop timestamp for {}, ignoring.", SB));
+
+        return;
+    }
+}