solenv/gcc-wrappers/wrapper.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3 * This file is part of the LibreOffice project.
   4 *
   5 * This Source Code Form is subject to the terms of the Mozilla Public
   6 * License, v. 2.0. If a copy of the MPL was not distributed with this
   7 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8 */
   9
  10 #include "wrapper.hxx"
  11
  12 #define WIN32_LEAN_AND_MEAN
  13
  14 #include <windows.h>
  15
  16 #define BUFLEN 2048
  17
  18 string getexe(string exename, bool maybeempty) {
  19     char* cmdbuf;
  20     size_t cmdlen;
  21     _dupenv_s(&cmdbuf,&cmdlen,exename.c_str());
  22     if(!cmdbuf) {
  23         if (maybeempty) {
  24             return string();
  25         }
  26         cout << "Error " << exename << " not defined. Did you forget to source the environment?" << endl;
  27         exit(1);
  28     }
  29     string command(cmdbuf);
  30     free(cmdbuf);
  31     return command;
  32 }
  33
  34 void setupccenv() {
  35     // Set-up library path
  36     string libpath="LIB=";
  37     char* libbuf;
  38     size_t liblen;
  39     _dupenv_s(&libbuf,&liblen,"ILIB");
  40     if (libbuf == nullptr) {
  41         std::cerr << "No environment variable ILIB" << std::endl;
  42         std::exit(EXIT_FAILURE);
  43     }
  44     libpath.append(libbuf);
  45     free(libbuf);
  46     if(_putenv(libpath.c_str())<0) {
  47         cerr << "Error: could not export LIB" << endl;
  48         exit(1);
  49     }
  50
  51     // Set-up include path
  52     string includepath="INCLUDE=.";
  53     char* incbuf;
  54     size_t inclen;
  55     _dupenv_s(&incbuf,&inclen,"SOLARINC");
  56     if (incbuf == nullptr) {
  57         std::cerr << "No environment variable SOLARINC" << std::endl;
  58         std::exit(EXIT_FAILURE);
  59     }
  60     string inctmp(incbuf);
  61     free(incbuf);
  62
  63     // 3 = strlen(" -I")
  64     for(size_t pos=0,len=0;pos<inctmp.length();) {
  65         size_t endpos=inctmp.find(" -I",pos+1);
  66         if(endpos==string::npos)
  67             endpos=inctmp.length();
  68         len=endpos-pos;
  69
  70         while(len>0&&inctmp[pos+len-1]==' ')
  71             --len;
  72
  73         if(len>3) {
  74             includepath.append(";");
  75             includepath.append(inctmp,pos+3,len-3);
  76         }
  77         pos=endpos;
  78     }
  79     if(_putenv(includepath.c_str())<0) {
  80         cerr << "Error: could not export INCLUDE" << endl;
  81         exit(1);
  82     }
  83 }
  84
  85 string processccargs(vector<string> rawargs) {
  86     // suppress the msvc banner
  87     string args=" -nologo";
  88     // TODO: should these options be enabled globally?
  89     args.append(" -EHsc");
  90     const char *const pDebugRuntime(getenv("MSVC_USE_DEBUG_RUNTIME"));
  91     if (pDebugRuntime && !strcmp(pDebugRuntime, "TRUE"))
  92         args.append(" -MDd");
  93     else
  94         args.append(" -MD");
  95     args.append(" -Gy");
  96     args.append(" -Ob1 -Oxs -Oy-");
  97
  98     // apparently these must be at the end
  99     // otherwise configure tests may fail
 100     // note: always use -debug so a PDB file is created
 101     string linkargs(" -link -debug");
 102
 103     for(vector<string>::iterator i = rawargs.begin(); i != rawargs.end(); ++i) {
 104         args.append(" ");
 105         if(*i == "-o") {
 106             // TODO: handle more than just exe output
 107             ++i;
 108             size_t dot=(*i).find_last_of(".");
 109             if(!(*i).compare(dot+1,3,"obj") || !(*i).compare(dot+1,1,"o"))
 110             {
 111                 args.append("-Fo");
 112                 args.append(*i);
 113             }
 114             else if(!(*i).compare(dot+1,3,"exe"))
 115             {
 116                 args.append("-Fe");
 117                 args.append(*i);
 118             }
 119             else if(!(*i).compare(dot+1,3,"dll"))
 120             {   // apparently cl.exe has no flag for dll?
 121                 linkargs.append(" -dll -out:");
 122                 linkargs.append(*i);
 123             }
 124             else
 125             {
 126                 cerr << "unknown -o argument - please adapt gcc-wrapper for \""
 127                      << (*i) << "\"";
 128                 exit(1);
 129             }
 130         }
 131         else if(*i == "-g" || !(*i).compare(0,5,"-ggdb")) {
 132             args.append("-Zi");
 133             args.append(" -FS");
 134         }
 135         else if(!(*i).compare(0,2,"-D")) {
 136             // need to re-escape strings for preprocessor
 137             for(size_t pos=(*i).find("\""); pos!=string::npos; pos=(*i).find("\"",pos)) {
 138                 (*i).replace(pos,0,"\\");
 139                 pos+=2;
 140             }
 141             args.append(*i);
 142         }
 143         else if(!(*i).compare(0,2,"-L")) {
 144             linkargs.append(" -LIBPATH:"+(*i).substr(2));
 145         }
 146         else if(!(*i).compare(0,2,"-l") && (*i).compare(0,5,"-link")) {
 147             linkargs.append(" "+(*i).substr(2)+".lib");
 148         }
 149         else if(!(*i).compare(0,5,"-def:") || !(*i).compare(0,5,"/def:")) {
 150             // why are we invoked with /def:? cl.exe should handle plain
 151             // "foo.def" by itself
 152             linkargs.append(" " + *i);
 153         }
 154         else if(!(*i).compare(0,12,"-fvisibility") || *i == "-fPIC") {
 155             //TODO: drop other gcc-specific options
 156         }
 157         else if(!(*i).compare(0,4,"-Wl,")) {
 158             //TODO: drop other gcc-specific options
 159         }
 160         else if(*i == "-Werror")
 161             args.append("-WX");
 162         else
 163             args.append(*i);
 164     }
 165     args.append(linkargs);
 166     return args;
 167 }
 168
 169 int startprocess(string command, string args) {
 170     STARTUPINFO si;
 171     PROCESS_INFORMATION pi;
 172     SECURITY_ATTRIBUTES sa;
 173
 174     HANDLE childout_read;
 175     HANDLE childout_write;
 176
 177     memset(&sa,0,sizeof(sa));
 178     memset(&si,0,sizeof(si));
 179     memset(&pi,0,sizeof(pi));
 180
 181     sa.nLength=sizeof(sa);
 182     sa.bInheritHandle=TRUE;
 183
 184     if(!CreatePipe(&childout_read,&childout_write,&sa,0)) {
 185         cerr << "Error: could not create stdout pipe" << endl;
 186         exit(1);
 187     }
 188
 189     si.cb=sizeof(si);
 190     si.dwFlags |= STARTF_USESTDHANDLES;
 191     si.hStdOutput=childout_write;
 192     si.hStdError=childout_write;
 193
 194     // support ccache
 195     size_t pos=command.find("ccache ");
 196     if(pos != string::npos) {
 197         args.insert(0,"cl.exe");
 198         command=command.substr(0,pos+strlen("ccache"))+".exe";
 199     }
 200
 201     auto cmdline = "\"" + command + "\" " + args;
 202
 203     //cerr << "CMD= " << command << " " << args << endl;
 204
 205     // Commandline may be modified by CreateProcess
 206     char* cmdlineBuf=_strdup(cmdline.c_str());
 207
 208     if(!CreateProcess(nullptr, // Process Name
 209         cmdlineBuf, // Command Line
 210         nullptr, // Process Handle not Inheritable
 211         nullptr, // Thread Handle not Inheritable
 212         TRUE, // Handles are Inherited
 213         0, // No creation flags
 214         nullptr, // Environment for process
 215         nullptr, // Use same starting directory
 216         &si, // Startup Info
 217         &pi) // Process Information
 218         ) {
 219             cerr << "Error: could not create process" << endl;
 220             exit(1);
 221     }
 222
 223     // if you don't close this the process will hang
 224     CloseHandle(childout_write);
 225
 226     // Get Process output
 227     char buffer[BUFLEN];
 228     DWORD readlen, writelen, ret;
 229     HANDLE stdout_handle=GetStdHandle(STD_OUTPUT_HANDLE);
 230     while(true) {
 231         int success=ReadFile(childout_read,buffer,BUFLEN,&readlen,nullptr);
 232         // check if the child process has exited
 233         if(GetLastError()==ERROR_BROKEN_PIPE)
 234             break;
 235         if(!success) {
 236             cerr << "Error: could not read from subprocess stdout" << endl;
 237             exit(1);
 238         }
 239         if(readlen!=0) {
 240             WriteFile(stdout_handle,buffer,readlen,&writelen,nullptr);
 241         }
 242     }
 243     WaitForSingleObject(pi.hProcess, INFINITE);
 244     GetExitCodeProcess(pi.hProcess, &ret);
 245     CloseHandle(pi.hThread);
 246     CloseHandle(pi.hProcess);
 247     return int(ret);
 248 }
 249
 250 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */