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); } diff --git a/tests/libtest/test.h b/tests/libtest/test.h index f0e13b20f..a442ba4bd 100644 --- a/tests/libtest/test.h +++ b/tests/libtest/test.h @@ -9,6 +9,7 @@ #include #include #include +#include #ifdef HAVE_SYS_SOCKET_H #include