diff options
Diffstat (limited to 'WebKitTools/MIDLWrapper/MIDLWrapper.cpp')
-rw-r--r-- | WebKitTools/MIDLWrapper/MIDLWrapper.cpp | 86 |
1 files changed, 86 insertions, 0 deletions
diff --git a/WebKitTools/MIDLWrapper/MIDLWrapper.cpp b/WebKitTools/MIDLWrapper/MIDLWrapper.cpp new file mode 100644 index 0000000..2132af8 --- /dev/null +++ b/WebKitTools/MIDLWrapper/MIDLWrapper.cpp @@ -0,0 +1,86 @@ +// MIDLWrapper.cpp : Just calls the built-in midl.exe with the given arguments. + +#define WIN32_LEAN_AND_MEAN // Exclude rarely-used stuff from Windows headers +#include <process.h> +#include <stdio.h> +#include <string> +#include <windows.h> + +using namespace std; + +int wmain(int argc, wchar_t* argv[], wchar_t* envp[]) +{ +#ifndef NDEBUG + fwprintf(stderr, L"######### im in ur IDE, compiling ur c0des ########\n"); +#endif + + int pathIndex = -1; + for (int i = 0; envp[i]; ++i) + if (!wcsncmp(envp[i], L"PATH=", 5)) { + pathIndex = i; + break; + } + + if (pathIndex == -1) { + fwprintf(stderr, L"Couldn't find PATH environment variable!\n"); + return -1; + } + + wchar_t* vcbin = wcsstr(envp[pathIndex], L"WebKitTools\\vcbin"); + if (!vcbin) { + fwprintf(stderr, L"Couldn't find WebKitTools\\vcbin in PATH!\n"); + return -1; + } + + wchar_t saved = *vcbin; + *vcbin = 0; + + wchar_t* afterLeadingSemiColon = wcsrchr(envp[pathIndex], ';'); + if (!afterLeadingSemiColon) + afterLeadingSemiColon = envp[pathIndex] + 5; // +5 for the length of "PATH=" + else + afterLeadingSemiColon++; + + *vcbin = saved; + + size_t pathLength = wcslen(envp[pathIndex]); + + wchar_t* trailingSemiColon = wcschr(vcbin, ';'); + if (!trailingSemiColon) + trailingSemiColon = envp[pathIndex] + pathLength; + + int vcbinLength = trailingSemiColon - afterLeadingSemiColon; + + size_t newPathLength = pathLength - vcbinLength; + + wchar_t* newPath = new wchar_t[newPathLength + 1]; + + // Copy everything before the vcbin path... + wchar_t* d = newPath; + wchar_t* s = envp[pathIndex]; + while (s < afterLeadingSemiColon) + *d++ = *s++; + + // Copy everything after the vcbin path... + s = trailingSemiColon; + while (*d++ = *s++); + + envp[pathIndex] = newPath; + +#ifndef NDEBUG + fwprintf(stderr, L"New path: %s\n", envp[pathIndex]); +#endif + + wchar_t** newArgv = new wchar_t*[argc + 1]; + for (int i = 0; i < argc; ++i) { + size_t length = wcslen(argv[i]); + newArgv[i] = new wchar_t[length + 3]; + *newArgv[i] = '\"'; + wcscpy_s(newArgv[i] + 1, length + 2, argv[i]); + *(newArgv[i] + 1 + length) = '\"'; + *(newArgv[i] + 2 + length) = 0; + } + newArgv[argc] = 0; + + return _wspawnvpe(_P_WAIT, L"midl", newArgv, envp); +} |