X-Git-Url: http://plrg.eecs.uci.edu/git/?p=c11tester.git;a=blobdiff_plain;f=common.h;h=01426a4d66bee75772507710221b3a99a02c49f0;hp=367dc6b2b00098b9380e400bea68e69771f91e8c;hb=bac8bc0661e904fc2b1b9aba07d239679483b2b4;hpb=c108b7d5709061ced258f5106ac67ada91242fc3

diff --git a/common.h b/common.h
index 367dc6b2..01426a4d 100644
--- a/common.h
+++ b/common.h
@@ -9,11 +9,12 @@
 #include "config.h"
 
 extern int model_out;
+extern int switch_alloc;
 
-#define model_print(fmt, ...) do { dprintf(model_out, fmt, ##__VA_ARGS__); } while (0)
+#define model_print(fmt, ...) do { switch_alloc = 1; dprintf(model_out, fmt, ## __VA_ARGS__); switch_alloc = 0; } while (0)
 
 #ifdef CONFIG_DEBUG
-#define DEBUG(fmt, ...) do { model_print("*** %15s:%-4d %25s() *** " fmt, __FILE__, __LINE__, __func__, ##__VA_ARGS__); } while (0)
+#define DEBUG(fmt, ...) do { model_print("*** %15s:%-4d %25s() *** " fmt, __FILE__, __LINE__, __func__, ## __VA_ARGS__); } while (0)
 #define DBG() DEBUG("\n")
 #define DBG_ENABLED() (1)
 #else
@@ -26,20 +27,20 @@ void assert_hook(void);
 
 #ifdef CONFIG_ASSERT
 #define ASSERT(expr) \
-do { \
-	if (!(expr)) { \
-		fprintf(stderr, "Error: assertion failed in %s at line %d\n", __FILE__, __LINE__); \
-		print_trace(); \
-		assert_hook();				 \
-		exit(EXIT_FAILURE); \
-	} \
-} while (0)
+	do { \
+		if (!(expr)) { \
+			fprintf(stderr, "Error: assertion failed in %s at line %d\n", __FILE__, __LINE__); \
+			/* print_trace(); // Trace printing may cause dynamic memory allocation */ \
+			assert_hook();                           \
+			_Exit(EXIT_FAILURE); \
+		} \
+	} while (0)
 #else
 #define ASSERT(expr) \
 	do { } while (0)
-#endif /* CONFIG_ASSERT */
+#endif	/* CONFIG_ASSERT */
 
 #define error_msg(...) fprintf(stderr, "Error: " __VA_ARGS__)
 
 void print_trace(void);
-#endif /* __COMMON_H__ */
+#endif	/* __COMMON_H__ */