rebootFunction: Allow reboot function in ADB without root

USAGE.md

@@ -187,7 +187,7 @@ pmt type partition(s) [OPTIONS]
 ---
 
 ### 9. `reboot`
-Reboot the device. Default reboot target is normal. General syntax:
+Reboot the device. Default reboot target is normal. If you are using it via ADB terminal, you **DO NOT** need root to use this feature. General syntax:
 ```bash
 pmt reboot [rebootTarget] [OPTIONS]
 ```

include/PartitionManager/PartitionManager.hpp

@@ -55,6 +55,7 @@ private:
 public:
   void registerFunction(std::unique_ptr<basic_function> _func, CLI::App &_app);
 
+  [[nodiscard]] bool isUsed(std::string name) const;
   [[nodiscard]] bool handleAll() const;
 };
 

src/FunctionManager.cpp

@@ -69,6 +69,14 @@ void basic_function_manager::registerFunction(
                    << std::endl;
 }
 
+bool basic_function_manager::isUsed(const std::string name) const {
+  if (_functions.empty()) return false;
+  for (const auto &func : _functions) {
+    if (func->name() == name) return func->isUsed();
+  }
+  return false;
+}
+
 bool basic_function_manager::handleAll() const {
   LOGN(PMTF, INFO) << "running caught function commands in command-line."
                    << std::endl;

src/PartitionManager.cpp

@@ -112,9 +112,11 @@ int Main(int argc, char **argv) {
                   "directory with -S "
                   "(--search-path)");
 
-    if (!Helper::hasSuperUser())
+    if (!Helper::hasSuperUser()) {
+      if (!(FuncManager.isUsed("reboot") && Helper::hasAdbPermissions()))
         throw Error(
             "Partition Manager Tool is requires super-user privileges!\n");
+    }
 
     return FuncManager.handleAll() == true ? EXIT_SUCCESS : EXIT_FAILURE;
   } catch (Helper::Error &error) {

srclib/libhelper/include/libhelper/lib.hpp

@@ -105,6 +105,7 @@ void reset();
 
 // Checkers
 bool hasSuperUser();
+bool hasAdbPermissions();
 bool isExists(std::string_view entry);
 bool fileIsExists(std::string_view file);
 bool directoryIsExists(std::string_view directory);

srclib/libhelper/src/Checkers.cpp

@@ -23,6 +23,7 @@
 
 namespace Helper {
 bool hasSuperUser() { return (getuid() == AID_ROOT); }
+bool hasAdbPermissions() { return (getuid() == AID_ADB); }
 
 bool isExists(const std::string_view entry) {
   struct stat st{};