tyler.durden/pixiewps
1
0
Fork 0
mirror of https://github.com/wiire-a/pixiewps.git synced 2025-07-27 15:54:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

Added memory cleanup in case of bad arguments

Browse Source
This commit is contained in:
wiire 2015-12-30 21:50:08 +01:00
parent 63450d6b2d
commit 64cec59d04
1 changed files with 27 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/pixiewps.c
View File

@ -175,18 +175,23 @@ memory_err:
break; break;
case 'V': case 'V':
{ {
struct timeval t_current; if (c > 1) { /* If --version is used then no other argument should be supplied */
gettimeofday(&t_current, 0); snprintf(wps->error, 256, "\n [!] Bad use of argument --version (-V)!\n\n");
time_t r_time; goto usage_err;
struct tm ts; } else {
char buffer[30]; struct timeval t_current;
r_time = t_current.tv_sec; gettimeofday(&t_current, 0);
ts = *localtime(&r_time); time_t r_time;
strftime(buffer, 30, "%c", &ts); struct tm ts;
fprintf(stderr, "\n Pixiewps %s\n\n [*] System time: %s\n\n", LONG_VERSION, buffer); char buffer[30];
free(wps->error); r_time = t_current.tv_sec;
free(wps); ts = *localtime(&r_time);
return ARG_ERROR; strftime(buffer, 30, "%c", &ts);
fprintf(stderr, "\n Pixiewps %s\n\n [*] System time: %s\n\n", LONG_VERSION, buffer);
free(wps->error);
free(wps);
return ARG_ERROR;
}
} }
case 'h': case 'h':
goto usage_err; goto usage_err;
@ -250,6 +255,16 @@ memory_err:
if (!c) { if (!c) {
usage_err: usage_err:
fprintf(stderr, usage, SHORT_VERSION, argv[0], wps->error); fprintf(stderr, usage, SHORT_VERSION, argv[0], wps->error);
free(wps->pke);
free(wps->pkr);
free(wps->e_hash1);
free(wps->e_hash2);
free(wps->authkey);
free(wps->e_nonce);
free(wps->r_nonce);
free(wps->e_bssid);
free(wps->error); free(wps->error);
free(wps); free(wps);
return ARG_ERROR; return ARG_ERROR;