diff options
author | Alex Xu (Hello71) <alex_y_xu@yahoo.ca> | 2016-07-07 19:46:47 -0400 |
---|---|---|
committer | Alex Xu (Hello71) <alex_y_xu@yahoo.ca> | 2016-07-07 19:46:47 -0400 |
commit | f528b66a0ea74b306bed22d243495458c8cf2383 (patch) | |
tree | 8f29e25b83ca88529da26e8b9e0e988005357f4a /src/udpastcp.c | |
parent | 03bdee772197c600b0066ebd1f27172332045da2 (diff) | |
download | udpastcp-f528b66a0ea74b306bed22d243495458c8cf2383.tar.xz udpastcp-f528b66a0ea74b306bed22d243495458c8cf2383.zip |
Fix memory leaks, miscellaneous issues.
Diffstat (limited to 'src/udpastcp.c')
-rw-r--r-- | src/udpastcp.c | 10 |
1 files changed, 10 insertions, 0 deletions
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 <string.h> #include <stdio.h> #include <time.h> +#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); |