tests: Detect llvm coverage run and redirect to real I/O functions

Code coverage writes data to disk, we need to use real io functions at
this point so that the data is really written.

BUG=b:187647884
BRANCH=None
TEST=llvm-profdata merge -sparse default.profraw -o default.profdata
TEST=llvm-cov show ./flashrom_unit_tests
-instr-profile=default.profdata --format=html --output-dir=.

Change-Id: I21cc1d631e92fa19006b967e85676f108e80b307
Signed-off-by: Evan Benn <evanbenn@chromium.org>
Reviewed-on: https://review.coreboot.org/c/flashrom/+/69267
Reviewed-by: Anastasia Klimchuk <aklm@chromium.org>
Tested-by: build bot (Jenkins) <no-reply@coreboot.org>

1 files changed, 16 insertions, 1 deletions

diff --git a/tests/io_real.c b/tests/io_real.c
index 9f9e9d6d5..429027fb5 100644
--- a/tests/io_real.c
+++ b/tests/io_real.c
@@ -25,6 +25,12 @@ static int io_real_open(void *state, const char *pathname, int flags, mode_t mod
 	return __real_open(pathname, flags, mode);
 }
 
+static FILE *io_real_fopen(void *state, const char *pathname, const char *mode)
+{
+	LOG_ME;
+	return __real_fopen(pathname, mode);
+}
+
 static FILE *io_real_fdopen(void *state, int fd, const char *mode)
 {
 	LOG_ME;
@@ -36,13 +42,21 @@ static size_t io_real_fwrite(void *state, const void *ptr, size_t size, size_t n
 	return __real_fwrite(ptr, size, nmemb, fp);
 }
 
+static int io_real_fclose(void *state, FILE *fp)
+{
+	LOG_ME;
+	return __real_fclose(fp);
+}
+
 /* Mock ios that defer to the real io functions.
  * These exist so that code coverage can print to real files on disk.
  */
 static const struct io_mock real_io = {
 	.iom_open = io_real_open,
+	.iom_fopen = io_real_fopen,
 	.iom_fwrite = io_real_fwrite,
 	.iom_fdopen = io_real_fdopen,
+	.iom_fclose = io_real_fclose,
 };
 
 /* Return 0 if string ends with suffix. */
@@ -58,6 +72,7 @@ static int check_suffix(const char *string, const char *suffix)
 void maybe_unmock_io(const char *pathname)
 {
 	const char *gcov_suffix = ".gcda";
-	if (!check_suffix(pathname, gcov_suffix))
+	const char *llvm_cov_suffix = ".profraw";
+	if (!check_suffix(pathname, gcov_suffix) || !check_suffix(pathname, llvm_cov_suffix))
 		io_mock_register(&real_io);
 }