commit 6975128cc2d24806fbdc32a46690738228f84d71
author Ram Chandrasekar <rchandrasekar@google.com> Fri Feb 25 21:07:36 2022 +0000
committer Commit Bot <commit-bot@chromium.org> Thu Mar 03 17:54:01 2022 +0000
tree 2866bdd4b04841543474614265bb2e65214c8c6c
parent bd4110b66fe2c3f15865b6c173a72ae9f60e9599

cros_build_lib: Add root user check helpers

Add helper functions, to check root privilege for the user.

BUG=b:218521368
TEST=run_tests lib/ scripts/ service/ cli/

Change-Id: Idd0ad8be3764b323b140d3e334399eef4d21e8a6
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/3491293
Tested-by: Ram Chandrasekar <rchandrasekar@google.com>
Auto-Submit: Ram Chandrasekar <rchandrasekar@google.com>
Reviewed-by: Mike Frysinger <vapier@chromium.org>
Commit-Queue: Ram Chandrasekar <rchandrasekar@google.com>

scripts/cros_setup_toolchains.py

diff --git a/scripts/cros_setup_toolchains.py b/scripts/cros_setup_toolchains.py
index a6762fe..7fb323e 100644
--- a/scripts/cros_setup_toolchains.py
+++ b/scripts/cros_setup_toolchains.py
@@ -1389,7 +1389,7 @@
   else:
     cros_build_lib.AssertInsideChroot()
     # This has to be always run as root.
-    if os.geteuid() != 0:
+    if osutils.IsNonRootUser():
       cros_build_lib.Die('this script must be run as root')
 
     Crossdev.Load(options.reconfig)