2 files changed, 22 insertions, 0 deletions

diff --git a/lib/utils_base.c b/lib/utils_base.c
index db21ea1..da28128 100644
--- a/lib/utils_base.c
+++ b/lib/utils_base.c
@@ -228,3 +228,18 @@ char *np_escaped_string (const char *string) {
         data[j] = '\0';
         return data;
 }
+
+int np_check_if_root(void) { return (geteuid() == 0); }
+
+int np_warn_if_not_root(void) {
+        int status = np_check_if_root();
+        if(!status) {
+                printf(_("Warning: "));
+                printf(_("This plugin must be either run as root or setuid root.\n"));
+                printf(_("To run as root, you can use a tool like sudo.\n"));
+                printf(_("To set the setuid permissions, use the command:\n"));
+                // XXX could we use something like progname?
+                printf("\tchmod u+s yourpluginfile\n");
+        }
+        return status;
+}
diff --git a/lib/utils_base.h b/lib/utils_base.h
index b722d0e..bda7659 100644
--- a/lib/utils_base.h
+++ b/lib/utils_base.h
@@ -43,4 +43,11 @@ void die (int, const char *, ...) __attribute__((noreturn,format(printf, 2, 3)))
 #define NP_RANGE_UNPARSEABLE 1
 #define NP_WARN_WITHIN_CRIT 2
 
+/* a simple check to see if we're running as root.  
+ * returns zero on failure, nonzero on success */
+int np_check_if_root(void);
+/* and a helpful wrapper around that.  it returns the same status
+ * code from the above function, in case it's helpful for testing */
+int np_warn_if_not_root(void);
+
 #endif /* _UTILS_BASE_ */