From 4ab0d74250b3a69e3980ca961b9867bf365580ab Mon Sep 17 00:00:00 2001
From: Daniel Stenberg <daniel@haxx.se>
Date: Fri, 14 May 2004 08:40:33 +0000
Subject: enable memory debugging the same way the curl command line tool
 already does

---
 tests/libtest/first.c | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

(limited to 'tests/libtest/first.c')

diff --git a/tests/libtest/first.c b/tests/libtest/first.c
index d8bfdd19a..c4f35df22 100644
--- a/tests/libtest/first.c
+++ b/tests/libtest/first.c
@@ -3,6 +3,7 @@
 #ifdef CURLDEBUG
 /* provide a proto for this debug function */
 extern void curl_memdebug(const char *);
+extern void curl_memlimit(int);
 #endif
 
 /* test is provided in the test code file */
@@ -13,6 +14,21 @@ char *arg2=NULL;
 int main(int argc, char **argv)
 {
   char *URL;
+
+#ifdef CURLDEBUG
+  /* this sends all memory debug messages to a logfile named memdump */
+  char *env = curl_getenv("CURL_MEMDEBUG");
+  if(env) {
+    curl_free(env);
+    curl_memdebug("memdump");
+  }
+  /* this enables the fail-on-alloc-number-N functionality */
+  env = curl_getenv("CURL_MEMLIMIT");
+  if(env) {
+    curl_memlimit(atoi(env));
+    curl_free(env);
+  }
+#endif
   if(argc< 2 ) {
     fprintf(stderr, "Pass URL as argument please\n");
     return 1;
@@ -24,8 +40,5 @@ int main(int argc, char **argv)
 
   fprintf(stderr, "URL: %s\n", URL);
 
-#ifdef CURLDEBUG
-  curl_memdebug("memdump");
-#endif
   return test(URL);
 }
-- 
cgit v1.2.3