zefie/llvm-mingw
1
0 Fork 0
Code Activity
Files
ee894b6f6365425ddafec6e8a01173be820f81bf
llvm-mingw/wrappers/windres-wrapper.c
T
Martin Storsjö ee894b6f63 windres: Accept the abbreviated form --include in addition to --include-dir 
As GNU binutils windres uses getopt for option parsing, it accepts
any unique abbreviation of the long options. For now just look
for this particular abbreviation instead of any potential one,
as there's at least one existing use in the wild of --include:

https://github.com/tcltk/tcl/blob/fe69f643609d633facf249c9c963ce121153a8ff/win/tcl.m4#L640
2020-01-05 22:09:31 +02:00

409 lines
14 KiB
C
Raw Blame History

/*
* author: Josh de Kock <josh@itanimul.li>, Martin Storsjo <martin@martin.st>
*
* This is free and unencumbered software released into the public domain.
*
* Anyone is free to copy, modify, publish, use, compile, sell, or
* distribute this software, either in source code form or as a compiled
* binary, for any purpose, commercial or non-commercial, and by any
* means.
*
* In jurisdictions that recognize copyright laws, the author or authors
* of this software dedicate any and all copyright interest in the
* software to the public domain. We make this dedication for the benefit
* of the public at large and to the detriment of our heirs and
* successors. We intend this dedication to be an overt act of
* relinquishment in perpetuity of all present and future rights to this
* software under copyright law.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*
* For more information, please refer to <http://unlicense.org/>
*/
#include "native-wrapper.h"
#ifndef DEFAULT_TARGET
#define DEFAULT_TARGET "x86_64-w64-mingw32"
#endif
#include <stdarg.h>
#ifdef _WIN32
#else
#define _tspawnvp_escape _spawnvp
#include <sys/wait.h>
#include <errno.h>
#define _P_WAIT 0
static int _spawnvp(int mode, const char *filename, const char * const *argv) {
pid_t pid;
if (!(pid = fork())) {
execvp(filename, (char **) argv);
perror(filename);
exit(1);
}
int stat = 0;
if (waitpid(pid, &stat, 0) == -1)
return -1;
if (WIFEXITED(stat))
return WEXITSTATUS(stat);
errno = EIO;
return -1;
}
#endif
// GNU binutils windres seem to require an extra level of escaping of
// -D options to the preprocessor, which we need to undo here.
// For defines that produce quoted strings, this windres frontend is
// called with parameters like this: -DSTRING=\\\"1.2.3\\\"
static const TCHAR *unescape_cpp(const TCHAR *str) {
TCHAR *out = _tcsdup(str);
int len = _tcslen(str);
int i, outpos = 0;
for (i = 0; i < len - 1; i++) {
if (str[i] == '\\' && str[i + 1] == '"')
continue;
out[outpos++] = str[i];
}
while (i < len)
out[outpos++] = str[i++];
out[outpos++] = '\0';
return out;
}
static void print_version(void) {
printf(
"version: LLVM windres (GNU windres compatible) 0.1\n"
);
exit(1);
}
static void print_help(void) {
printf(
"usage: llvm-windres <OPTION> [INPUT-FILE] [OUTPUT-FILE]\n"
"\n"
"LLVM Tool to manipulate Windows resources with a GNU windres interface.\n"
"\n"
"Options:\n"
" -i, --input <arg> Name of the input file.\n"
" -o, --output <arg> Name of the output file.\n"
" -J, --input-format <arg> Input format to read.\n"
" -O, --output-format <arg> Output format to generate.\n"
" --preprocessor <arg> Custom preprocessor command.\n"
" --preprocessor-arg <arg> Preprocessor command arguments.\n"
" -F, --target <arg> Target for COFF objects to be compiled for.\n"
" -I, --include-dir <arg> Include directory to pass to preprocessor and resource compiler.\n"
" -D, --define <arg[=val]> Define to pass to preprocessor.\n"
" -U, --undefine <arg[=val]> Undefine to pass to preprocessor.\n"
" -c, --codepage <arg> Default codepage to use when reading an rc file (0x0-0xffff).\n"
" --use-temp-file Use a temporary file for the preprocessing output.\n"
" -v, --verbose Enable verbose output.\n"
" -V, --version Display version.\n"
" -h, --help Display this message and exit.\n"
"Input Formats:\n"
" rc Text Windows Resource\n"
" res Binary Windows Resource\n"
"Output Formats:\n"
" res Binary Windows Resource\n"
" coff COFF object\n"
"Targets:\n"
" armv7-w64-mingw32\n"
" aarch64-w64-mingw32\n"
" i686-w64-mingw32\n"
" x86_86-w64-mingw32\n"
);
exit(1);
}
static void error(const TCHAR *basename, const TCHAR *fmt, ...) {
_ftprintf(stderr, _T(TS": error: "), basename);
va_list ap;
va_start(ap, fmt);
_vftprintf(stderr, fmt, ap);
_ftprintf(stderr, _T("\n"));
va_end(ap);
exit(1);
}
static void print_argv(const TCHAR **exec_argv) {
while (*exec_argv) {
_ftprintf(stderr, _T(TS" "), *exec_argv);
exec_argv++;
}
_ftprintf(stderr, _T("\n"));
}
static void check_num_args(int arg, int max_arg) {
if (arg > max_arg) {
fprintf(stderr, "Too many options added\n");
abort();
}
}
int _tmain(int argc, TCHAR* argv[]) {
const TCHAR *dir;
const TCHAR *basename;
const TCHAR *target;
split_argv(argv[0], &dir, &basename, &target, NULL);
if (!target)
target = _T(DEFAULT_TARGET);
const TCHAR *bfd_target = NULL;
const TCHAR *input = _T("-");
const TCHAR *output = _T("/dev/stdout");
const TCHAR *input_format = _T("rc");
const TCHAR *output_format = _T("coff");
const TCHAR **includes = malloc(argc * sizeof(*includes));
int nb_includes = 0;
const TCHAR *codepage = _T("1252");
const TCHAR **cpp_options = malloc(argc * sizeof(*cpp_options));
int nb_cpp_options = 0;
int verbose = 0;
#define _tcslen_const(a) (sizeof(a)/sizeof(TCHAR) - 1)
#define _tcsstart(a, b) !_tcsncmp(a, b, _tcslen_const(b))
#define IF_MATCH_EITHER(short, long) \
if (!_tcscmp(argv[i], _T(short)) || !_tcscmp(argv[i], _T(long)))
#define IF_MATCH_THREE(first, second, third) \
if (!_tcscmp(argv[i], _T(first)) || !_tcscmp(argv[i], _T(second)) || !_tcscmp(argv[i], _T(third)))
#define OPTION(short, long, var) \
if (_tcsstart(argv[i], _T(short)) && argv[i][_tcslen_const(_T(short))]) { \
var = argv[i] + _tcslen_const(_T(short)); \
} else if (_tcsstart(argv[i], _T(long "="))) { \
var = _tcschr(argv[i], '=') + 1; \
} else IF_MATCH_EITHER(short, long) { \
if (i + 1 < argc) \
var = argv[++i]; \
else \
error(basename, _T(TS" missing argument"), argv[i]); \
}
#define SEPARATE_ARG(var) do { \
if (i + 1 < argc) \
var = argv[++i]; \
else \
error(basename, _T(TS" missing argument"), argv[i]); \
} while (0)
#define SEPARATE_ARG_PREFIX(var, prefix) do { \
if (i + 1 < argc) \
var = concat(_T(prefix), argv[++i]); \
else \
error(basename, _T(TS" missing argument"), argv[i]); \
} while (0)
for (int i = 1; i < argc; i++) {
OPTION("-i", "--input", input)
else OPTION("-o", "--output", output)
else OPTION("-J", "--input-format", input_format)
else OPTION("-O", "--output-format", output_format)
else OPTION("-F", "--target", bfd_target)
else IF_MATCH_THREE("-I", "--include-dir", "--include") {
SEPARATE_ARG(includes[nb_includes++]);
} else if (_tcsstart(argv[i], _T("--include-dir=")) ||
_tcsstart(argv[i], _T("--include="))) {
includes[nb_includes++] = _tcschr(argv[i], '=') + 1;
} else if (_tcsstart(argv[i], _T("-I"))) {
includes[nb_includes++] = argv[i] + 2;
} else OPTION("-c", "--codepage", codepage)
else if (!_tcscmp(argv[i], _T("--preprocessor"))) {
error(basename, _T("ENOSYS"));
} else if (_tcsstart(argv[i], _T("--preprocessor-arg="))) {
cpp_options[nb_cpp_options++] = _tcschr(argv[i], '=') + 1;
} else if (!_tcscmp(argv[i], _T("--preprocessor-arg"))) {
SEPARATE_ARG(cpp_options[nb_cpp_options++]);
} else IF_MATCH_EITHER("-D", "--define") {
SEPARATE_ARG_PREFIX(cpp_options[nb_cpp_options++], "-D");
} else if (_tcsstart(argv[i], _T("-D"))) {
cpp_options[nb_cpp_options++] = argv[i];
} else IF_MATCH_EITHER("-U", "--undefine") {
SEPARATE_ARG_PREFIX(cpp_options[nb_cpp_options++], "-U");
} else if (_tcsstart(argv[i], _T("-U"))) {
cpp_options[nb_cpp_options++] = argv[i];
} else IF_MATCH_EITHER("-v", "--verbose") {
verbose = 1;
} else IF_MATCH_EITHER("-V", "--version") {
print_version();
} else IF_MATCH_EITHER("-h", "--help") {
print_help();
} else if (!_tcscmp(argv[i], _T("--use-temp-file"))) {
// No-op, we use a temp file by default.
} else if (_tcsstart(argv[i], _T("-"))) {
error(basename, _T("unrecognized option: `"TS"'"), argv[i]);
} else {
if (!_tcscmp(input, _T("-")))
input = argv[i];
else if (!_tcscmp(output, _T("/dev/stdout")))
output = argv[i];
else
error(basename, _T("rip: `"TS"'"), argv[i]);
}
}
if (bfd_target) {
if (!_tcscmp(bfd_target, _T("pe-x86-64")) ||
!_tcscmp(bfd_target, _T("pei-x86-64")))
target = _T("x86_64-w64-mingw32");
else if (!_tcscmp(bfd_target, _T("pe-i386")) ||
!_tcscmp(bfd_target, _T("pei-x86-64")))
target = _T("x86_64-w64-mingw32");
else
error(basename, _T("unsupported target: `"TS"'"), bfd_target);
}
TCHAR *arch = _tcsdup(target);
TCHAR *dash = _tcschr(arch, '-');
if (dash)
*dash = '\0';
const TCHAR *machine = _T("unknown");
if (!_tcscmp(arch, _T("i686")))
machine = _T("X86");
else if (!_tcscmp(arch, _T("x86_64")))
machine = _T("X64");
else if (!_tcscmp(arch, _T("armv7")))
machine = _T("ARM");
else if (!_tcscmp(arch, _T("aarch64")))
machine = _T("ARM64");
const TCHAR *CC = concat(target, _T("-clang"));
const TCHAR **rc_options = malloc(2 * argc * sizeof(*cpp_options));
int nb_rc_options = 0;
for (int i = 0; i < nb_includes; i++) {
cpp_options[nb_cpp_options++] = concat(_T("-I"), includes[i]);
rc_options[nb_rc_options++] = _T("-I");
rc_options[nb_rc_options++] = includes[i];
}
for (int i = 0; i < nb_cpp_options; i++)
cpp_options[i] = unescape_cpp(cpp_options[i]);
const TCHAR *preproc_rc = concat(output, _T(".preproc.rc"));
const TCHAR *res = concat(output, _T(".out.res"));
TCHAR *inputdir = _tcsdup(input);
{
TCHAR *sep = _tcsrchrs(inputdir, '/', '\\');
if (sep)
*sep = '\0';
else
inputdir = _tcsdup(_T("."));
}
int max_arg = 2 * argc + 20;
const TCHAR **exec_argv = malloc((max_arg + 1) * sizeof(*exec_argv));
int arg = 0;
if (!_tcscmp(input_format, _T("rc"))) {
exec_argv[arg++] = concat(dir, CC);
exec_argv[arg++] = _T("-E");
for (int i = 0; i < nb_cpp_options; i++)
exec_argv[arg++] = cpp_options[i];
exec_argv[arg++] = _T("-xc");
exec_argv[arg++] = _T("-DRC_INVOKED=1");
exec_argv[arg++] = input;
exec_argv[arg++] = _T("-o");
exec_argv[arg++] = preproc_rc;
exec_argv[arg] = NULL;
check_num_args(arg, max_arg);
if (verbose)
print_argv(exec_argv);
int ret = _tspawnvp_escape(_P_WAIT, exec_argv[0], exec_argv);
if (ret == -1) {
_tperror(exec_argv[0]);
return 1;
}
if (ret != 0) {
error(basename, _T("preprocessor failed"));
return ret;
}
arg = 0;
exec_argv[arg++] = concat(dir, _T("llvm-rc"));
for (int i = 0; i < nb_rc_options; i++)
exec_argv[arg++] = rc_options[i];
exec_argv[arg++] = _T("-I");
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++] = output;
else
exec_argv[arg++] = res;
exec_argv[arg] = NULL;
check_num_args(arg, max_arg);
if (verbose)
print_argv(exec_argv);
ret = _tspawnvp_escape(_P_WAIT, exec_argv[0], exec_argv);
if (ret == -1) {
_tperror(exec_argv[0]);
return 1;
}
if (ret != 0) {
error(basename, _T("llvm-rc failed"));
if (!verbose)
_tunlink(preproc_rc);
return ret;
}
if (!_tcscmp(output_format, _T("res"))) {
// All done
} else if (!_tcscmp(output_format, _T("coff"))) {
arg = 0;
exec_argv[arg++] = concat(dir, _T("llvm-cvtres"));
exec_argv[arg++] = res;
exec_argv[arg++] = concat(_T("-machine:"), machine);
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_escape(_P_WAIT, exec_argv[0], exec_argv);
if (ret == -1) {
_tperror(exec_argv[0]);
return 1;
}
if (!verbose) {
_tunlink(preproc_rc);
_tunlink(res);
}
return ret;
} else {
error(basename, _T("invalid output format: `"TS"'"), output_format);
}
} else if (!_tcscmp(input_format, _T("res"))) {
exec_argv[arg++] = concat(dir, _T("llvm-cvtres"));
exec_argv[arg++] = input;
exec_argv[arg++] = concat(_T("-machine:"), machine);
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_escape(_P_WAIT, exec_argv[0], exec_argv);
if (ret == -1) {
_tperror(exec_argv[0]);
return 1;
}
return ret;
} else {
error(basename, _T("invalid input format: `"TS"'"), input_format);
}
return 0;
}
View Git Blame Copy Permalink