From 06c89243670c2662d0aedead76ec7230f6ff332a Mon Sep 17 00:00:00 2001 From: r00t Date: Tue, 11 Apr 2017 01:34:14 -0400 Subject: [PATCH] root user check --- bdisk/bdisk.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/bdisk/bdisk.py b/bdisk/bdisk.py index 7de3efb..40655bd 100755 --- a/bdisk/bdisk.py +++ b/bdisk/bdisk.py @@ -8,12 +8,15 @@ import bSSL import ipxe import bsync import bGPG +import os # we need to: # we also need to figure out how to implement "mentos" (old bdisk) like functionality, letting us reuse an existing chroot install if possible to save time for future builds. # if not, though, it's no big deal. # still on the todo: iPXE if __name__ == '__main__': + if os.getuid() != 0: + exit('{0}: ERROR: BDisk *must* be run as the root user or with sudo!'.format(datetime.datetime.now())) print('{0}: Starting.'.format(datetime.datetime.now())) conf = host.parseConfig(host.getConfig())[1] prep.dirChk(conf)