[compiler-rt] [Fuzzer] Passthrough zlib CMake paths into the test (PR #107926)

[Petr Hosek via llvm-commits]

https://github.com/petrhosek created https://github.com/llvm/llvm-project/pull/107926

We shouldn't assume that we're using system zlib installation

>From 4e7146ceada7540370e6a792114c1b49d3931e13 Mon Sep 17 00:00:00 2001
From: Petr Hosek <phosek at google.com>
Date: Mon, 9 Sep 2024 22:08:54 +0000
Subject: [PATCH] [Fuzzer] Passthrough zlib CMake paths into the test

We shouldn't assume that we're using system zlib installation
---
 compiler-rt/test/fuzzer/compressed.test   | 4 ++--
 compiler-rt/test/lit.common.cfg.py        | 2 ++
 compiler-rt/test/lit.common.configured.in | 2 ++
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/compiler-rt/test/fuzzer/compressed.test b/compiler-rt/test/fuzzer/compressed.test
index 284f7ed96ca851..59a43cc807337f 100644
--- a/compiler-rt/test/fuzzer/compressed.test
+++ b/compiler-rt/test/fuzzer/compressed.test
@@ -4,7 +4,7 @@ REQUIRES: zlib
 # unsupported by this test.
 UNSUPPORTED: i386, target=arm{{.*}}
 # Custom mutator should find this bug, w/o custom -- no chance.
-RUN: %cpp_compiler %S/CompressedTest.cpp -o %t-CompressedTestCustom -DCUSTOM_MUTATOR -lz
-RUN: %cpp_compiler %S/CompressedTest.cpp -o %t-CompressedTestPlain -lz
+RUN: %cpp_compiler %S/CompressedTest.cpp -o %t-CompressedTestCustom -DCUSTOM_MUTATOR -I%zlib_include_dir %zlib_library
+RUN: %cpp_compiler %S/CompressedTest.cpp -o %t-CompressedTestPlain -I%zlib_include_dir %zlib_library
 RUN: not %run %t-CompressedTestCustom -seed=1 -runs=1000000
 RUN:     %run %t-CompressedTestPlain  -seed=1 -runs=1000000
diff --git a/compiler-rt/test/lit.common.cfg.py b/compiler-rt/test/lit.common.cfg.py
index 0690c3a18efdbc..1c6fbc80cd4137 100644
--- a/compiler-rt/test/lit.common.cfg.py
+++ b/compiler-rt/test/lit.common.cfg.py
@@ -312,6 +312,8 @@ def push_dynamic_library_lookup_path(config, new_path):
 
 if config.have_zlib:
     config.available_features.add("zlib")
+    config.substitutions.append(("%zlib_include_dir", config.zlib_include_dir))
+    config.substitutions.append(("%zlib_library", config.zlib_library))
 
 if config.have_internal_symbolizer:
     config.available_features.add("internal_symbolizer")
diff --git a/compiler-rt/test/lit.common.configured.in b/compiler-rt/test/lit.common.configured.in
index 8889b816b149fc..049133dc965c31 100644
--- a/compiler-rt/test/lit.common.configured.in
+++ b/compiler-rt/test/lit.common.configured.in
@@ -71,6 +71,8 @@ else:
 
 set_default("have_internal_symbolizer", @COMPILER_RT_ENABLE_INTERNAL_SYMBOLIZER_PYBOOL@)
 set_default("have_zlib", @ZLIB_FOUND_PYBOOL@)
+set_default("zlib_include_dir", "@ZLIB_INCLUDE_DIR@")
+set_default("zlib_library", "@ZLIB_LIBRARY@")
 set_default("libcxx_used", "@LLVM_LIBCXX_USED@")
 
 # LLVM tools dir can be passed in lit parameters, so try to