[rhmessaging-commits] rhmessaging commits: r2029 - mgmt/cumin/bin.

[rhmessaging-commits at lists.jboss.org]

Author: justi9
Date: 2008-05-09 11:13:34 -0400 (Fri, 09 May 2008)
New Revision: 2029

Modified:
   mgmt/cumin/bin/cumin-admin
Log:
Add internal permission check to cumin-admin

Modified: mgmt/cumin/bin/cumin-admin
===================================================================
--- mgmt/cumin/bin/cumin-admin	2008-05-09 14:43:35 UTC (rev 2028)
+++ mgmt/cumin/bin/cumin-admin	2008-05-09 15:13:34 UTC (rev 2029)
@@ -35,7 +35,21 @@
 
     return args
 
+def check_permission():
+    curr_uid = os.getuid()
+    file_uid = os.stat(__file__).st_uid
+
+    if curr_uid == 0 or curr_uid == file_uid:
+        pass
+    else:
+        print "You don't have sufficient privileges"
+        sys.exit(1)
+
 def main():
+    if "-h" in sys.argv or "--help" in sys.argv:
+        print_usage(config)
+        sys.exit(0)
+        
     config = CuminConfig()
     config.add_param("force", bool, False, "Don't complain and just do it")
 
@@ -43,11 +57,6 @@
 
     config.init()
 
-    if "-h" in sys.argv or "--help" in sys.argv:
-        print_usage(config)
-        sys.exit(0)
-        
-
     if config.debug:
         config.prt()
 
@@ -58,6 +67,8 @@
         print_usage(config)
         sys.exit(1)
 
+    check_permission()
+
     command = args[0]
 
     database = MintDatabase(config.data)