/*********************************
 * ICC wrapper for Mozilla Build *
 * by coolypf           20100127 *
 *********************************/

#include <stdio.h>
#include <stdlib.h>
#include <windows.h>
#include <string>

using std::string;

char cl[]="c:\\Program Files (x86)\\Microsoft Visual Studio 9.0\\VC\\BIN\\cl.exe";
char icl[]="c:\\Program Files (x86)\\Intel\\Compiler\\11.1\\051\\Bin\\ia32\\icl.exe";
char clflags[]="/O2";
char iclerr[]=": error: ";
PROCESS_INFORMATION pi;

HANDLE Exec(const char *cmd)
{
   SECURITY_ATTRIBUTES sa;
   HANDLE hRead,hWrite;
   sa.nLength = sizeof(SECURITY_ATTRIBUTES);
   sa.lpSecurityDescriptor = NULL;
   sa.bInheritHandle = TRUE;
   if (!CreatePipe(&hRead,&hWrite,&sa,0))
   {
      printf("Error On CreatePipe()");
      exit(-1);
   }
   STARTUPINFOA si;
   si.cb = sizeof(STARTUPINFO);
   GetStartupInfoA(&si);
   si.hStdError = hWrite;
   si.hStdOutput = hWrite;
   si.wShowWindow = SW_HIDE;
   si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;
   if (!CreateProcessA(NULL,_strdup(cmd),NULL,NULL,TRUE,NULL,NULL,NULL,&si,&pi))
   {
      printf("Error on CreateProcess()");
      exit(-1);
   }
   CloseHandle(hWrite);
   return hRead;
}

int main(int argc, char **argv)
{
   string arg;
   for(int i=1; i<argc; ++i)
   {
      arg.append(" ");
      arg.append(argv[i]);
   }
   string clcmd=cl+string(" ")+clflags+" "+arg;
   string iclcmd=icl+string(" ")+arg;
   char buf[10241];
   unsigned long count;
   HANDLE h;
   h=Exec(clcmd.c_str());
   while(ReadFile(h, buf, 10240, &count, 0))
   {
      buf[count]=0;
      printf("%s", buf);
   }
   WaitForSingleObject(pi.hProcess, INFINITE);
   unsigned long e1;
   if(!GetExitCodeProcess(pi.hProcess, &e1)) e1=0;
   h=Exec(iclcmd.c_str());
   bool r=false;
   while(ReadFile(h, buf, 10240, &count, 0))
   {
      buf[count]=0;
      if(strstr(buf, iclerr))
      {
         r=true;
         break;
      }
   }
   WaitForSingleObject(pi.hProcess, INFINITE);
   unsigned long e2;
   if(!GetExitCodeProcess(pi.hProcess, &e2)) e2=0;
   if(r) Exec(clcmd.c_str());
   if(e1==0 || e2==0) return 0;
   return e1;
}