Oct: is_admin()

Author: scivision

+stdlib/is_admin.m

@@ -1 +1,5 @@
 %% IS_ADMIN is the process run as root / admin (requires MEX)
+
+function is_admin()
+error("need to 'buildtool mex' or 'legacy_mex_build()' first")
+end

+stdlib/set_permissions.m

@@ -9,3 +9,7 @@
 % * ok (1,1) logical
 %
 % This function is written in C++ using STL <filesystem> and is only available in Matlab.
+
+function set_permissions(~, ~, ~, ~)
+error("need to 'buildtool mex' or 'legacy_mex_build()' first")
+end

octave_build.m

@@ -10,6 +10,7 @@
 %% specific source
 srcs = {
 fullfile(d, "is_rosetta.cpp"), ...
+fullfile(d, "is_admin.cpp"), ...
 };
 
 
@@ -18,5 +19,6 @@
   src = s{1};
   [~, n] = fileparts(src);
 
-  mkoctfile(src,"-v", ["-I", inc], "--output", fullfile(t, n))
+  disp(["mkoctfile: ", src, " => ", n, ".oct"])
+  mkoctfile(src, ["-I", inc], "--output", fullfile(t, n))
 end

private/get_mex_sources.m

@@ -16,7 +16,7 @@
 
 
 srcs = {
-fullfile(top, "src/is_admin.cpp"), ...
+[fullfile(top, "src/is_admin.cpp"), fullfile(top, "src/admin_fs.cpp")] ...
 fullfile(top, "src/is_char_device.cpp"), ...
 fullfile(top, "src/set_permissions.cpp"), ...
 [fullfile(top, "src/is_rosetta.cpp"), mac], ...

src/admin_fs.cpp

@@ -0,0 +1,31 @@
+#if defined(_WIN32)
+#define WIN32_LEAN_AND_MEAN
+#include <windows.h>  // GetTokenInformation
+#else
+#include <unistd.h>  // geteuid
+#include <sys/types.h>  // geteuid, pid_t
+#endif
+
+#include "admin_fs.h"
+
+
+bool fs_is_admin(){
+  // running as admin / root / superuser
+#if defined(_WIN32)
+  HANDLE hToken = nullptr;
+  TOKEN_ELEVATION elevation;
+  DWORD dwSize;
+
+  const bool ok = (OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken) &&
+     GetTokenInformation(hToken, TokenElevation, &elevation, sizeof(elevation), &dwSize));
+
+  if(hToken)
+    CloseHandle(hToken);
+  if(ok)
+    return elevation.TokenIsElevated;
+
+  return false;
+#else
+  return geteuid() == 0;
+#endif
+}

src/admin_fs.h

@@ -0,0 +1 @@
+bool fs_is_admin();

src/is_admin.cpp

@@ -4,35 +4,7 @@
 #include <vector>
 #include <memory>
 
-#if defined(_WIN32)
-#define WIN32_LEAN_AND_MEAN
-#include <windows.h>  // GetTokenInformation
-#else
-#include <unistd.h>  // geteuid
-#include <sys/types.h>  // geteuid, pid_t
-#endif
-
-
-static bool fs_is_admin(){
-  // running as admin / root / superuser
-#if defined(_WIN32)
-  HANDLE hToken = nullptr;
-  TOKEN_ELEVATION elevation;
-  DWORD dwSize;
-
-  const bool ok = (OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken) &&
-     GetTokenInformation(hToken, TokenElevation, &elevation, sizeof(elevation), &dwSize));
-
-  if(hToken)
-    CloseHandle(hToken);
-  if(ok)
-    return elevation.TokenIsElevated;
-
-  return false;
-#else
-  return geteuid() == 0;
-#endif
-}
+#include "admin_fs.h"
 
 
 class MexFunction : public matlab::mex::Function {

src/octave/is_admin.cpp

@@ -0,0 +1,17 @@
+#include <octave/oct.h>
+
+#include "admin_fs.cpp"
+
+
+DEFUN_DLD (is_admin, args, nargout,
+           "is the process running as admin / root / superuser")
+{
+  if (args.length() != 0){
+    octave_stdout << "Oct: No input required\n";
+    return octave_value(false);
+  }
+
+  bool y = fs_is_admin();
+
+  return octave_value(y);
+}

src/octave/is_rosetta.cpp

@@ -4,7 +4,7 @@
 
 
 DEFUN_DLD (is_rosetta, args, nargout,
-           "Hello World Help String")
+           "is the process running under macOS Rosetta")
 {
   if (args.length() != 0){
     octave_stdout << "is_rosetta: No input required\n";