wrappers: Consistently quote all arguments when calling spawn or creating a command line
This commit is contained in:
1 parent
fa88c4ee1e
commit
750ab50433
3 files changed
+39
-29
No files matched your search
@@ -75,12 +75,13 @@ static void filter_stderr(char *buf, int n) {
|
||||
}
|
||||
|
||||
static int exec_filtered(const TCHAR **argv) {
|
||||
for (int i = 0; argv[i]; i++)
|
||||
argv[i] = escape(argv[i]);
|
||||
int len = 1;
|
||||
for (int i = 0; argv[i]; i++)
|
||||
len += _tcslen(argv[i]) + 1;
|
||||
TCHAR *cmdline = malloc(len * sizeof(*cmdline));
|
||||
int pos = 0;
|
||||
// On Windows, the arguments are already quoted and escaped properly.
|
||||
for (int i = 0; argv[i]; i++) {
|
||||
_tcscpy(&cmdline[pos], argv[i]);
|
||||
pos += _tcslen(argv[i]);
|
||||
@@ -165,7 +166,7 @@ int _tmain(int argc, TCHAR* argv[]) {
|
||||
int arg = 0;
|
||||
if (getenv("CCACHE"))
|
||||
exec_argv[arg++] = _T("ccache");
|
||||
exec_argv[arg++] = escape(concat(dir, _T(CLANG)));
|
||||
exec_argv[arg++] = concat(dir, _T(CLANG));
|
||||
|
||||
// If changing this wrapper, change clang-target-wrapper.sh accordingly.
|
||||
if (!_tcscmp(exe, _T("clang++")) || !_tcscmp(exe, _T("g++")) || !_tcscmp(exe, _T("c++")))
|
||||
@@ -204,7 +205,7 @@ int _tmain(int argc, TCHAR* argv[]) {
|
||||
exec_argv[arg++] = _T("-Qunused-arguments");
|
||||
|
||||
for (int i = 1; i < argc; i++)
|
||||
exec_argv[arg++] = escape(argv[i]);
|
||||
exec_argv[arg++] = argv[i];
|
||||
|
||||
exec_argv[arg] = NULL;
|
||||
if (arg > max_arg) {
|
||||
@@ -222,7 +223,7 @@ int _tmain(int argc, TCHAR* argv[]) {
|
||||
if (!_tcscmp(argv[i], _T("-v")))
|
||||
return exec_filtered(exec_argv);
|
||||
|
||||
int ret = _tspawnvp(_P_WAIT, exec_argv[0], exec_argv);
|
||||
int ret = _tspawnvp_escape(_P_WAIT, exec_argv[0], exec_argv);
|
||||
if (ret == -1) {
|
||||
_tperror(exec_argv[0]);
|
||||
return 1;
|
||||
|
||||
@@ -56,8 +56,8 @@ typedef char TCHAR;
|
||||
#define TS "%s"
|
||||
#endif
|
||||
|
||||
static TCHAR *escape(const TCHAR *str) {
|
||||
#ifdef _WIN32
|
||||
static TCHAR *escape(const TCHAR *str) {
|
||||
TCHAR *out = malloc((_tcslen(str) * 2 + 3) * sizeof(*out));
|
||||
TCHAR *ptr = out;
|
||||
int i;
|
||||
@@ -82,11 +82,20 @@ static TCHAR *escape(const TCHAR *str) {
|
||||
*ptr++ = '"';
|
||||
*ptr++ = '\0';
|
||||
return out;
|
||||
#else
|
||||
return _tcsdup(str);
|
||||
#endif
|
||||
}
|
||||
|
||||
static int _tspawnvp_escape(int mode, const TCHAR *filename, const TCHAR * const *argv) {
|
||||
int num_args = 0;
|
||||
while (argv[num_args])
|
||||
num_args++;
|
||||
const TCHAR **escaped_argv = malloc(num_args * sizeof(*escaped_argv));
|
||||
for (int i = 0; argv[i]; i++)
|
||||
escaped_argv[i] = escape(argv[i]);
|
||||
escaped_argv[num_args] = NULL;
|
||||
return _tspawnvp(mode, filename, escaped_argv);
|
||||
}
|
||||
#endif
|
||||
|
||||
static TCHAR *concat(const TCHAR *prefix, const TCHAR *suffix) {
|
||||
int prefixlen = _tcslen(prefix);
|
||||
int suffixlen = _tcslen(suffix);
|
||||
|
||||
+21
-21
@@ -37,7 +37,7 @@
|
||||
|
||||
#ifdef _WIN32
|
||||
#else
|
||||
#define _tspawnvp _spawnvp
|
||||
#define _tspawnvp_escape _spawnvp
|
||||
|
||||
#include <sys/wait.h>
|
||||
#include <errno.h>
|
||||
@@ -287,21 +287,21 @@ int _tmain(int argc, TCHAR* argv[]) {
|
||||
int arg = 0;
|
||||
|
||||
if (!_tcscmp(input_format, _T("rc"))) {
|
||||
exec_argv[arg++] = escape(concat(dir, CC));
|
||||
exec_argv[arg++] = concat(dir, CC);
|
||||
exec_argv[arg++] = _T("-E");
|
||||
for (int i = 0; i < nb_cpp_options; i++)
|
||||
exec_argv[arg++] = escape(cpp_options[i]);
|
||||
exec_argv[arg++] = cpp_options[i];
|
||||
exec_argv[arg++] = _T("-xc");
|
||||
exec_argv[arg++] = _T("-DRC_INVOKED=1");
|
||||
exec_argv[arg++] = escape(input);
|
||||
exec_argv[arg++] = input;
|
||||
exec_argv[arg++] = _T("-o");
|
||||
exec_argv[arg++] = escape(preproc_rc);
|
||||
exec_argv[arg++] = preproc_rc;
|
||||
exec_argv[arg] = NULL;
|
||||
|
||||
check_num_args(arg, max_arg);
|
||||
if (verbose)
|
||||
print_argv(exec_argv);
|
||||
int ret = _tspawnvp(_P_WAIT, exec_argv[0], exec_argv);
|
||||
int ret = _tspawnvp_escape(_P_WAIT, exec_argv[0], exec_argv);
|
||||
if (ret == -1) {
|
||||
_tperror(exec_argv[0]);
|
||||
return 1;
|
||||
@@ -312,25 +312,25 @@ int _tmain(int argc, TCHAR* argv[]) {
|
||||
}
|
||||
|
||||
arg = 0;
|
||||
exec_argv[arg++] = escape(concat(dir, _T("llvm-rc")));
|
||||
exec_argv[arg++] = concat(dir, _T("llvm-rc"));
|
||||
for (int i = 0; i < nb_rc_options; i++)
|
||||
exec_argv[arg++] = escape(rc_options[i]);
|
||||
exec_argv[arg++] = rc_options[i];
|
||||
exec_argv[arg++] = _T("-I");
|
||||
exec_argv[arg++] = escape(inputdir);
|
||||
exec_argv[arg++] = escape(preproc_rc);
|
||||
exec_argv[arg++] = inputdir;
|
||||
exec_argv[arg++] = preproc_rc;
|
||||
exec_argv[arg++] = _T("-c");
|
||||
exec_argv[arg++] = codepage;
|
||||
exec_argv[arg++] = _T("-fo");
|
||||
if (!_tcscmp(output_format, _T("res")))
|
||||
exec_argv[arg++] = escape(output);
|
||||
exec_argv[arg++] = output;
|
||||
else
|
||||
exec_argv[arg++] = escape(res);
|
||||
exec_argv[arg++] = res;
|
||||
exec_argv[arg] = NULL;
|
||||
|
||||
check_num_args(arg, max_arg);
|
||||
if (verbose)
|
||||
print_argv(exec_argv);
|
||||
ret = _tspawnvp(_P_WAIT, exec_argv[0], exec_argv);
|
||||
ret = _tspawnvp_escape(_P_WAIT, exec_argv[0], exec_argv);
|
||||
if (ret == -1) {
|
||||
_tperror(exec_argv[0]);
|
||||
return 1;
|
||||
@@ -346,16 +346,16 @@ int _tmain(int argc, TCHAR* argv[]) {
|
||||
// All done
|
||||
} else if (!_tcscmp(output_format, _T("coff"))) {
|
||||
arg = 0;
|
||||
exec_argv[arg++] = escape(concat(dir, _T("llvm-cvtres")));
|
||||
exec_argv[arg++] = escape(res);
|
||||
exec_argv[arg++] = concat(dir, _T("llvm-cvtres"));
|
||||
exec_argv[arg++] = res;
|
||||
exec_argv[arg++] = concat(_T("-machine:"), machine);
|
||||
exec_argv[arg++] = escape(concat(_T("-out:"), output));
|
||||
exec_argv[arg++] = concat(_T("-out:"), output);
|
||||
exec_argv[arg] = NULL;
|
||||
|
||||
check_num_args(arg, max_arg);
|
||||
if (verbose)
|
||||
print_argv(exec_argv);
|
||||
int ret = _tspawnvp(_P_WAIT, exec_argv[0], exec_argv);
|
||||
int ret = _tspawnvp_escape(_P_WAIT, exec_argv[0], exec_argv);
|
||||
if (ret == -1) {
|
||||
_tperror(exec_argv[0]);
|
||||
return 1;
|
||||
@@ -369,16 +369,16 @@ int _tmain(int argc, TCHAR* argv[]) {
|
||||
error(basename, _T("invalid output format: `"TS"'"), output_format);
|
||||
}
|
||||
} else if (!_tcscmp(input_format, _T("res"))) {
|
||||
exec_argv[arg++] = escape(concat(dir, _T("llvm-cvtres")));
|
||||
exec_argv[arg++] = escape(input);
|
||||
exec_argv[arg++] = concat(dir, _T("llvm-cvtres"));
|
||||
exec_argv[arg++] = input;
|
||||
exec_argv[arg++] = concat(_T("-machine:"), machine);
|
||||
exec_argv[arg++] = escape(concat(_T("-out:"), output));
|
||||
exec_argv[arg++] = concat(_T("-out:"), output);
|
||||
exec_argv[arg] = NULL;
|
||||
|
||||
check_num_args(arg, max_arg);
|
||||
if (verbose)
|
||||
print_argv(exec_argv);
|
||||
int ret = _tspawnvp(_P_WAIT, exec_argv[0], exec_argv);
|
||||
int ret = _tspawnvp_escape(_P_WAIT, exec_argv[0], exec_argv);
|
||||
if (ret == -1) {
|
||||
_tperror(exec_argv[0]);
|
||||
return 1;
|
||||
|
||||
Reference in new issue
Block a user