diff --git a/Networking/udp2file.c b/Networking/udp2file.c index 6426a83..22b18f1 100644 --- a/Networking/udp2file.c +++ b/Networking/udp2file.c @@ -114,8 +114,10 @@ int main(int argc, char **argv) // ------------------------------------------------------------------------- // check for help argument - if ( (strncmp ( argv[1], "-h", strlen("-h")) == 0) - || (strncmp ( argv[1], "--help", strlen("--help")) == 0) + if ( (argc == 2) + && ( (strncmp ( argv[1], "-h", strlen("-h")) == 0) + || (strncmp ( argv[1], "--help", strlen("--help")) == 0) + ) ) { usage(); @@ -433,11 +435,14 @@ int main(int argc, char **argv) // ------------------------------------------------------------------------- // clean up // ------------------------------------------------------------------------- + if (buffer_check == 1) + { + fclose(pBufStatFile); + } fclose(pDestFile); - fclose(pBufStatFile); close(rx_fd); - LOGMSG_INFO("\n######################\n\tProgram exit\n######################") + LOGMSG_INFO("\n############################\n\tProgram Exit\n############################") return 0; } \ No newline at end of file