Update main.c

Added check at top of main() to test if user is running as sudo. Print message and exit if not run as sudo.

Update main.c
Added check at top of main() to test if user is running as sudo. Print message and exit if not run as sudo.
tmcolby · Gordon Hollingworth
1 parent f4ff1f08
Showing 1 changed file with 8 additions and 0 deletions
main.c
@@ -96,6 +96,14 @@ int main(int argc, char *argv[])
 		int length;
 		unsigned char signature[20];
 	} message;
+	
+	//exit if not run as sudo
+	int user = getuid();
+	if(user != 0)
+	{
+		printf("Must be run with sudo...\n");
+		exit(-1);
+	}
  
 	// Skip the command name
 	argv++; argc--;