X-Git-Url: https://git.ozlabs.org/?a=blobdiff_plain;f=ccan%2Ffailtest%2Ffailtest_override.h;h=0d71312c4a63f2820214231ea804988f30cd106f;hb=74257cee33ae3033f961d5f22a0313b8cb1b18d4;hp=236441fcfc78e74c6dd632308d5bed7eaf4f5223;hpb=6c02fd599f5b3f925197c3161ee186a3305fc963;p=ccan

diff --git a/ccan/failtest/failtest_override.h b/ccan/failtest/failtest_override.h
index 236441fc..0d71312c 100644
--- a/ccan/failtest/failtest_override.h
+++ b/ccan/failtest/failtest_override.h
@@ -1,6 +1,8 @@
+/* Licensed under LGPL - see LICENSE file for details */
 #ifndef CCAN_FAILTEST_OVERRIDE_H
 #define CCAN_FAILTEST_OVERRIDE_H
 /* This file is included before the source file to test. */
+#include "config.h"
 
 /* Replacement of allocators. */
 #include <stdlib.h>
@@ -17,6 +19,10 @@
 #define realloc(ptr, size)					\
 	failtest_realloc((ptr), (size), __FILE__, __LINE__)
 
+#undef free
+#define free(ptr) \
+	failtest_free(ptr)
+
 /* Replacement of I/O. */
 #include <sys/types.h>
 #include <sys/stat.h>
@@ -48,11 +54,15 @@
 	failtest_pwrite((fd), (buf), (count), (off), __FILE__, __LINE__)
 
 #undef close
-#define close(fd) failtest_close(fd)
+#define close(fd) failtest_close(fd, __FILE__, __LINE__)
 
 #undef fcntl
 #define fcntl(fd, ...) failtest_fcntl((fd), __FILE__, __LINE__, __VA_ARGS__)
 
+/* Replacement of getpid (since failtest will fork). */
+#undef getpid
+#define getpid() failtest_getpid(__FILE__, __LINE__)
+
 #include <ccan/failtest/failtest_proto.h>
 
 #endif /* CCAN_FAILTEST_OVERRIDE_H */