forked from osmocom/wireshark
![Moshe Kaplan](/assets/img/avatar_default.png)
free() memory allocated for argv to fix Coverity 1108127, which detected it as a resource leak.
79 lines
1.9 KiB
C
79 lines
1.9 KiB
C
/*
|
|
* Compile and link this with all CLI programs where the main routine
|
|
* should get UTF-8 arguments on Windows. In those programs, include the
|
|
* cli_main.h header to rename main to real_main on Windows.
|
|
*
|
|
* This is used in software licensed under the GPLv2, and its license MUST
|
|
* be compatible with that license.
|
|
*
|
|
* This is used in software licensed under the Apache 2.0 license, and its
|
|
* license MUST be compatible with that license.
|
|
*
|
|
* For that purpose, we use the MIT (X11) license.
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*/
|
|
|
|
#include "cli_main.h"
|
|
|
|
#ifdef _WIN32
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <windows.h>
|
|
|
|
int
|
|
wmain(int argc, wchar_t *wc_argv[])
|
|
{
|
|
char **argv;
|
|
int i;
|
|
int return_code;
|
|
|
|
argv = (char **)malloc((argc + 1) * sizeof(char *));
|
|
if (argv == NULL) {
|
|
fprintf(stderr, "Out of memory for converted argument list\n");
|
|
return 2;
|
|
}
|
|
for (i = 0; i < argc; i++) {
|
|
/*
|
|
* XXX = use WC_ERR_INVALID_CHARS rather than 0, and fail if
|
|
* the argument isn't valid UTF-16?
|
|
*/
|
|
int width;
|
|
char *utf8_string;
|
|
|
|
width = WideCharToMultiByte(CP_UTF8, 0, wc_argv[i], -1, NULL, 0,
|
|
NULL, NULL);
|
|
if (width == 0) {
|
|
fprintf(stderr, "WideCharToMultiByte failed: %d\n",
|
|
width);
|
|
return 2;
|
|
}
|
|
utf8_string = malloc(width);
|
|
if (utf8_string == NULL) {
|
|
fprintf(stderr,
|
|
"Out of memory for converted argument list\n");
|
|
return 2;
|
|
}
|
|
if (WideCharToMultiByte(CP_UTF8, 0, wc_argv[i], -1, utf8_string,
|
|
width, NULL, NULL) == 0) {
|
|
fprintf(stderr, "WideCharToMultiByte failed: %d\n",
|
|
width);
|
|
return 2;
|
|
}
|
|
argv[i] = utf8_string;
|
|
}
|
|
argv[i] = NULL;
|
|
/*
|
|
* The original "main" routine was renamed to "real_main" via a macro in
|
|
* the cli_main.h header file since either "main" or "wmain" can be
|
|
* defined on Windows, but not both.
|
|
*/
|
|
return_code = real_main(argc, argv);
|
|
for (i = 0; i < argc; i++) {
|
|
free(argv[i]);
|
|
}
|
|
free(argv);
|
|
return return_code;
|
|
}
|
|
#endif
|