X-Git-Url: http://git.ozlabs.org/?a=blobdiff_plain;f=ccan%2Ffailtest%2Ffailtest_override.h;h=dfcc62e0dd5f4ff5630cc727230a1361ce09ca0c;hb=371a31d946c1055e8b7f7d9cec2b7f617206aeaf;hp=236441fcfc78e74c6dd632308d5bed7eaf4f5223;hpb=6c02fd599f5b3f925197c3161ee186a3305fc963;p=ccan

diff --git a/ccan/failtest/failtest_override.h b/ccan/failtest/failtest_override.h
index 236441fc..dfcc62e0 100644
--- a/ccan/failtest/failtest_override.h
+++ b/ccan/failtest/failtest_override.h
@@ -1,6 +1,7 @@
 #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 +18,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 +53,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 */