fixes: https://github.com/iovisor/bpftrace/issues/293

99dd7733 · williangaspar · e653297a · 99dd7733
Commit 99dd7733 authored Dec 27, 2018 by williangaspar
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

src/main.cpp src/main.cpp +7 -0

No files found.
--- a/src/main.cpp
+++ b/src/main.cpp
 #include <iostream>
 #include <signal.h>
 #include <sys/resource.h>
+#include <unistd.h>

 #include "bpforc.h"
 #include "bpftrace.h"
@@ -54,6 +55,12 @@ static void enforce_infinite_rlimit() {

 int main(int argc, char *argv[])
 {
+  if (geteuid() != 0)
+  {
+    std::cerr << "ERROR: bpftrace currently only supports running as the root user." << std::endl;
+    exit(0);
+  }
+
  int err;
  Driver driver;
  char *pid_str = NULL;