tests: Redirect to real I/O to support coverage run

Implement a check that redirects mock io functions to the real
implementations. Real I/O functions are needed for the coverage tool to
be able to create and write files.

BUG=None
BRANCH=None
TEST=None

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

1 files changed, 63 insertions, 0 deletions

diff --git a/tests/io_real.c b/tests/io_real.c
new file mode 100644
index 000000000..9f9e9d6d5
--- /dev/null
+++ b/tests/io_real.c
@@ -0,0 +1,63 @@
+/*
+ * This file is part of the flashrom project.
+ *
+ * Copyright 2022 Google LLC
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; version 2 of the License.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ */
+
+#include "io_mock.h"
+#include "wraps.h"
+
+#include "io_real.h"
+#include <string.h>
+
+static int io_real_open(void *state, const char *pathname, int flags, mode_t mode)
+{
+	LOG_ME;
+	return __real_open(pathname, flags, mode);
+}
+
+static FILE *io_real_fdopen(void *state, int fd, const char *mode)
+{
+	LOG_ME;
+	return __real_fdopen(fd, mode);
+}
+
+static size_t io_real_fwrite(void *state, const void *ptr, size_t size, size_t nmemb, FILE *fp)
+{
+	return __real_fwrite(ptr, size, nmemb, 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_fwrite = io_real_fwrite,
+	.iom_fdopen = io_real_fdopen,
+};
+
+/* Return 0 if string ends with suffix. */
+static int check_suffix(const char *string, const char *suffix)
+{
+	int len_l = strlen(string);
+	int len_r = strlen(suffix);
+	if (len_l > len_r)
+		return strcmp(string + len_l - len_r, suffix);
+	return 1;
+}
+
+void maybe_unmock_io(const char *pathname)
+{
+	const char *gcov_suffix = ".gcda";
+	if (!check_suffix(pathname, gcov_suffix))
+		io_mock_register(&real_io);
+}