diff options
Diffstat (limited to 'tests/io_real.c')
-rw-r--r-- | tests/io_real.c | 17 |
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); } |