From 545fea574acb0a51815229c9ef955c2b2fb4cb0b Mon Sep 17 00:00:00 2001
From: Thulio Ferraz Assis <3149049+f0rmiga@users.noreply.github.com>
Date: Tue, 12 Jul 2022 13:01:10 -0700
Subject: [PATCH] fix: fail if the user is root

Also, add a new attribute for the brave to ignore the check.

Signed-off-by: Thulio Ferraz Assis <3149049+f0rmiga@users.noreply.github.com>
---
 python/repositories.bzl | 32 +++++++++++++++++++++++++-------
 1 file changed, 25 insertions(+), 7 deletions(-)

diff --git a/python/repositories.bzl b/python/repositories.bzl
index b0af316419..09b05fa4ac 100644
--- a/python/repositories.bzl
+++ b/python/repositories.bzl
@@ -108,13 +108,26 @@ def _python_repository_impl(rctx):
         rctx.file(distutils_path, rctx.attr.distutils_content)
 
     # Make the Python installation read-only.
-    if "windows" not in rctx.os.name:
-        exec_result = rctx.execute(["chmod", "-R", "ugo-w", "lib"])
-        if exec_result.return_code:
-            fail_msg = "Failed to make interpreter installation read-only. 'chmod' error msg: {}".format(
-                exec_result.stderr,
-            )
-            fail(fail_msg)
+    if not rctx.attr.ignore_root_user_error:
+        if "windows" not in rctx.os.name:
+            exec_result = rctx.execute(["chmod", "-R", "ugo-w", "lib"])
+            if exec_result.return_code != 0:
+                fail_msg = "Failed to make interpreter installation read-only. 'chmod' error msg: {}".format(
+                    exec_result.stderr,
+                )
+                fail(fail_msg)
+            exec_result = rctx.execute(["touch", "lib/.test"])
+            if exec_result.return_code == 0:
+                exec_result = rctx.execute(["id", "-u"])
+                if exec_result.return_code != 0:
+                    fail("Could not determine current user ID. 'id -u' error msg: {}".format(
+                        exec_result.stderr,
+                    ))
+                uid = int(exec_result.stdout.strip())
+                if uid == 0:
+                    fail("The current user is root, please run as non-root when using the hermetic Python interpreter. See https://github.com/bazelbuild/rules_python/pull/713.")
+                else:
+                    fail("The current user has CAP_DAC_OVERRIDE set, please drop this capability when using the hermetic Python interpreter. See https://github.com/bazelbuild/rules_python/pull/713.")
 
     python_bin = "python.exe" if ("windows" in platform) else "bin/python3"
 
@@ -227,6 +240,11 @@ python_repository = repository_rule(
                   "Either distutils or distutils_content can be specified, but not both.",
             mandatory = False,
         ),
+        "ignore_root_user_error": attr.bool(
+            default = False,
+            doc = "Whether the check for root should be ignored or not. This causes cache misses with .pyc files.",
+            mandatory = False,
+        ),
         "platform": attr.string(
             doc = "The platform name for the Python interpreter tarball.",
             mandatory = True,