Home | History | Annotate | Download | only in MIDLWrapper 
      1 // MIDLWrapper.cpp : Just calls the built-in midl.exe with the given arguments.
      2 
      3 #define WIN32_LEAN_AND_MEAN    // Exclude rarely-used stuff from Windows headers
      4 #include <process.h>
      5 #include <stdio.h>
      6 #include <string>
      7 #include <windows.h>
      8 
      9 using namespace std;
     10 
     11 int wmain(int argc, wchar_t* argv[], wchar_t* envp[])
     12 {
     13 #ifndef NDEBUG
     14     fwprintf(stderr, L"######### im in ur IDE, compiling ur c0des ########\n");
     15 #endif
     16 
     17     int pathIndex = -1;
     18     for (int i = 0; envp[i]; ++i)
     19         if (!wcsncmp(envp[i], L"PATH=", 5)) {
     20             pathIndex = i;
     21             break;
     22         }
     23 
     24     if (pathIndex == -1) {
     25         fwprintf(stderr, L"Couldn't find PATH environment variable!\n");
     26         return -1;
     27     }
     28 
     29     wchar_t* vcbin = wcsstr(envp[pathIndex], L"Tools\\vcbin");
     30     if (!vcbin) {
     31         fwprintf(stderr, L"Couldn't find Tools\\vcbin in PATH!\n");
     32         return -1;
     33     }
     34 
     35     wchar_t saved = *vcbin;
     36     *vcbin = 0;
     37 
     38     wchar_t* afterLeadingSemiColon = wcsrchr(envp[pathIndex], ';');
     39     if (!afterLeadingSemiColon)
     40         afterLeadingSemiColon = envp[pathIndex] + 5; // +5 for the length of "PATH="
     41     else
     42         afterLeadingSemiColon++;
     43 
     44     *vcbin = saved;
     45 
     46     size_t pathLength = wcslen(envp[pathIndex]);
     47 
     48     wchar_t* trailingSemiColon = wcschr(vcbin, ';');
     49     if (!trailingSemiColon)
     50         trailingSemiColon = envp[pathIndex] + pathLength;
     51 
     52     int vcbinLength = trailingSemiColon - afterLeadingSemiColon;
     53 
     54     size_t newPathLength = pathLength - vcbinLength;
     55 
     56     wchar_t* newPath = new wchar_t[newPathLength + 1];
     57 
     58     // Copy everything before the vcbin path...
     59     wchar_t* d = newPath;
     60     wchar_t* s = envp[pathIndex];
     61     while (s < afterLeadingSemiColon)
     62         *d++ = *s++;
     63 
     64     // Copy everything after the vcbin path...
     65     s = trailingSemiColon;
     66     while (*d++ = *s++);
     67 
     68     envp[pathIndex] = newPath;
     69 
     70 #ifndef NDEBUG
     71     fwprintf(stderr, L"New path: %s\n", envp[pathIndex]);
     72 #endif
     73 
     74     wchar_t** newArgv = new wchar_t*[argc + 1];
     75     for (int i = 0; i < argc; ++i) {
     76         size_t length = wcslen(argv[i]);
     77         newArgv[i] = new wchar_t[length + 3];
     78         *newArgv[i] = '\"';
     79         wcscpy_s(newArgv[i] + 1, length + 2, argv[i]);
     80         *(newArgv[i] + 1 + length) = '\"';
     81         *(newArgv[i] + 2 + length) = 0;
     82     }
     83     newArgv[argc] = 0;
     84 
     85     return _wspawnvpe(_P_WAIT, L"midl", newArgv, envp);
     86 }
     87