[rhmessaging-commits] rhmessaging commits: r2029 - mgmt/cumin/bin.
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)