Avoid incorrectly setting LD_LIBRARY_PATH

replit · Oct 16, 2024 · 1a58ce8 · 1a58ce8
1 parent 458dcf0
commit 1a58ce8
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 11 deletions.
diff --git a/pkgs/python-utils/default.nix b/pkgs/python-utils/default.nix
@@ -32,19 +32,19 @@ let
       ldLibraryPathConvertWrapper = pkgs.writeShellApplication {
         inherit name;
         text = ''
-          if test "''${REPLIT_RTLD_LOADER:-}" = "1" && test "''${REPLIT_NIX_CHANNEL:-}" != "legacy"
+          if test "''${REPLIT_NIX_CHANNEL:-}" = "legacy"
           then
-            # activate RTLD loader!
-            export LD_AUDIT="${pkgs.replit-rtld-loader}/rtld_loader.so"
-            export REPLIT_LD_LIBRARY_PATH=${python-ld-library-path}:''${REPLIT_LD_LIBRARY_PATH:-}
-          else
             export LD_LIBRARY_PATH=${python-ld-library-path}
             if [ -n "''${PYTHON_LD_LIBRARY_PATH-}" ]; then
               export LD_LIBRARY_PATH=''${PYTHON_LD_LIBRARY_PATH}:$LD_LIBRARY_PATH
             fi
             if [ -n "''${REPLIT_LD_LIBRARY_PATH-}" ]; then
               export LD_LIBRARY_PATH=''${REPLIT_LD_LIBRARY_PATH}:$LD_LIBRARY_PATH
             fi
+          else
+            # activate RTLD loader!
+            export LD_AUDIT="${pkgs.replit-rtld-loader}/rtld_loader.so"
+            export REPLIT_LD_LIBRARY_PATH=${python-ld-library-path}:''${REPLIT_LD_LIBRARY_PATH:-}
           fi
           exec "${bin}" "$@"
         '';

diff --git a/pkgs/python-wrapped/main.go b/pkgs/python-wrapped/main.go
@@ -9,12 +9,8 @@ import (
 var PythonExePath string
 var ReplitPythonLdLibraryPath string
 
-func main() {
-	if ldAudit := os.Getenv("REPLIT_LD_AUDIT"); ldAudit != "" {
-		os.Setenv("LD_AUDIT", ldAudit)
-	}
-	os.Unsetenv("PYTHONNOUSERSITE")
-
+// Set up environment for legacy nixpkgs
+func legacy() {
 	ldLibraryPath := []string{}
 	for _, key := range []string{
 		"REPLIT_LD_LIBRARY_PATH",
@@ -30,7 +26,26 @@ func main() {
 	if len(ldLibraryPath) > 0 {
 		os.Setenv("LD_LIBRARY_PATH", strings.Join(ldLibraryPath, ":"))
 	}
+}
 
+// Set up environment for non-legacy nixpkgs
+func modern() {
+	if ldAudit := os.Getenv("REPLIT_LD_AUDIT"); ldAudit != "" {
+		os.Setenv("LD_AUDIT", ldAudit)
+	}
+	if val, ok := os.LookupEnv("REPLIT_LD_LIBRARY_PATH"); ok && val != "" {
+		os.Setenv("REPLIT_LD_LIBRARY_PATH", strings.Join([]string{ReplitPythonLdLibraryPath, val}, ":"))
+	}
+}
+
+func main() {
+	os.Unsetenv("PYTHONNOUSERSITE")
+
+	if val, ok := os.LookupEnv("REPLIT_NIX_CHANNEL"); !ok || val == "legacy" || val == "" {
+		legacy()
+	} else {
+		modern()
+	}
 
 	if err := syscall.Exec(PythonExePath, os.Args, os.Environ()); err != nil {
 		panic(err)