lib/CompilerDriver/Action.cpp

   1 //===--- Action.cpp - The LLVM Compiler Driver ------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open
   6 // Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 //  Action class - implementation and auxiliary functions.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "llvm/CompilerDriver/Action.h"
  15 #include "llvm/CompilerDriver/BuiltinOptions.h"
  16
  17 #include "llvm/Support/raw_ostream.h"
  18 #include "llvm/System/Program.h"
  19 #include "llvm/System/TimeValue.h"
  20
  21 #include <stdexcept>
  22 #include <string>
  23
  24 using namespace llvm;
  25 using namespace llvmc;
  26
  27 namespace {
  28   int ExecuteProgram(const std::string& name,
  29                      const StrVector& args) {
  30     sys::Path prog = sys::Program::FindProgramByName(name);
  31
  32     if (prog.isEmpty())
  33       throw std::runtime_error("Can't find program '" + name + "'");
  34     if (!prog.canExecute())
  35       throw std::runtime_error("Program '" + name + "' is not executable.");
  36
  37     // Build the command line vector and the redirects array.
  38     const sys::Path* redirects[3] = {0,0,0};
  39     sys::Path stdout_redirect;
  40
  41     std::vector<const char*> argv;
  42     argv.reserve((args.size()+2));
  43     argv.push_back(name.c_str());
  44
  45     for (StrVector::const_iterator B = args.begin(), E = args.end();
  46          B!=E; ++B) {
  47       if (*B == ">") {
  48         ++B;
  49         stdout_redirect.set(*B);
  50         redirects[1] = &stdout_redirect;
  51       }
  52       else {
  53         argv.push_back((*B).c_str());
  54       }
  55     }
  56     argv.push_back(0);  // null terminate list.
  57
  58     // Invoke the program.
  59     return sys::Program::ExecuteAndWait(prog, &argv[0], 0, &redirects[0]);
  60   }
  61
  62   void print_string (const std::string& str) {
  63     errs() << str << ' ';
  64   }
  65 }
  66
  67 namespace llvmc {
  68   void AppendToGlobalTimeLog(const std::string& cmd, double time);
  69 }
  70
  71 int llvmc::Action::Execute() const {
  72   if (DryRun || VerboseMode) {
  73     errs() << Command_ << " ";
  74     std::for_each(Args_.begin(), Args_.end(), print_string);
  75     errs() << '\n';
  76   }
  77   if (!DryRun) {
  78     if (Time) {
  79       sys::TimeValue now = sys::TimeValue::now();
  80       int ret = ExecuteProgram(Command_, Args_);
  81       sys::TimeValue now2 = sys::TimeValue::now();
  82       now2 -= now;
  83       double elapsed = now2.seconds()  + now2.microseconds()  / 1000000.0;
  84       AppendToGlobalTimeLog(Command_, elapsed);
  85
  86       return ret;
  87     }
  88     else {
  89       return ExecuteProgram(Command_, Args_);
  90     }
  91   }
  92
  93   return 0;
  94 }