From f528b66a0ea74b306bed22d243495458c8cf2383 Mon Sep 17 00:00:00 2001 From: "Alex Xu (Hello71)" Date: Thu, 7 Jul 2016 19:46:47 -0400 Subject: Fix memory leaks, miscellaneous issues. --- src/udpastcp.c | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'src/udpastcp.c') diff --git a/src/udpastcp.c b/src/udpastcp.c index 7230c93..fa8fea2 100644 --- a/src/udpastcp.c +++ b/src/udpastcp.c @@ -2,9 +2,12 @@ #include #include #include +#include "common.h" #include "server.h" #include "client.h" +int free_mem_on_exit = 0; + int main(int argc, char *argv[]) { if (argc < 6) { puts("usage: udpintcp client|server LISTEN_HOST LISTEN_PORT REMOTE_HOST REMOTE_PORT"); @@ -13,9 +16,16 @@ int main(int argc, char *argv[]) { srandom((unsigned int)time(NULL)); + if (getenv("UDPASTCP_RELEASE_MEMORY")) { + DBG("UDPASTCP_RELEASE_MEMORY is set, will free all memory on exit."); + free_mem_on_exit = 1; + } + if (!strcmp(argv[1], "client")) { + DBG("starting client listening on [%s]:%s connecting to [%s]:%s", argv[2], argv[3], argv[4], argv[5]); return start_client(argv[2], argv[3], argv[4], argv[5]) == 0; } else if (!strcmp(argv[1], "server")) { + DBG("starting server listening on [%s]:%s connecting to [%s]:%s", argv[2], argv[3], argv[4], argv[5]); return start_server(argv[2], argv[3], argv[4], argv[5]) == 0; } else { fputs("invalid mode\n", stderr); -- cgit v1.2.3-54-g00ecf