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_sysroot_utils.py

diff --git a/scripts/cros_sysroot_utils.py b/scripts/cros_sysroot_utils.py
index 330436b..d43af01 100644
--- a/scripts/cros_sysroot_utils.py
+++ b/scripts/cros_sysroot_utils.py
@@ -4,11 +4,11 @@
 
 """Collection of tools to create sysroots."""
 
-import os
 import sys
 
 from chromite.lib import commandline
 from chromite.lib import cros_build_lib
+from chromite.lib import osutils
 from chromite.lib import sysroot_lib
 
 
@@ -65,7 +65,7 @@
   if not cros_build_lib.IsInsideChroot():
     raise commandline.ChrootRequiredError(argv)
 
-  if os.geteuid() != 0:
+  if osutils.IsNonRootUser():
     cros_build_lib.sudo_run(sys.argv, print_cmd=False)
     return