Source/cmProcessTools.cxx

   1 /* Distributed under the OSI-approved BSD 3-Clause License.  See accompanying
   2    file Copyright.txt or https://cmake.org/licensing for details.  */
   3 #include "cmProcessTools.h"
   4
   5 #include <algorithm>
   6 #include <iterator>
   7 #include <ostream>
   8
   9 #include <cm3p/uv.h>
  10
  11 #include "cmProcessOutput.h"
  12 #include "cmUVHandlePtr.h"
  13 #include "cmUVStream.h"
  14
  15 std::vector<cmUVProcessChain::Status> cmProcessTools::RunProcess(
  16   cmUVProcessChainBuilder& builder, OutputParser* out, OutputParser* err,
  17   Encoding encoding)
  18 {
  19   cmProcessOutput processOutput(encoding);
  20
  21   builder.SetBuiltinStream(cmUVProcessChainBuilder::Stream_OUTPUT)
  22     .SetBuiltinStream(cmUVProcessChainBuilder::Stream_ERROR);
  23
  24   auto chain = builder.Start();
  25
  26   std::string strdata;
  27   cm::uv_pipe_ptr outputPipe;
  28   outputPipe.init(chain.GetLoop(), 0);
  29   uv_pipe_open(outputPipe, chain.OutputStream());
  30   auto outputHandle = cmUVStreamRead(
  31     outputPipe,
  32     [&out, &processOutput, &strdata](std::vector<char> data) {
  33       if (out) {
  34         processOutput.DecodeText(data.data(), data.size(), strdata, 1);
  35         if (!out->Process(strdata.c_str(), static_cast<int>(strdata.size()))) {
  36           out = nullptr;
  37         }
  38       }
  39     },
  40     [&out]() { out = nullptr; });
  41   cm::uv_pipe_ptr errorPipe;
  42   errorPipe.init(chain.GetLoop(), 0);
  43   uv_pipe_open(errorPipe, chain.ErrorStream());
  44   auto errorHandle = cmUVStreamRead(
  45     errorPipe,
  46     [&err, &processOutput, &strdata](std::vector<char> data) {
  47       if (err) {
  48         processOutput.DecodeText(data.data(), data.size(), strdata, 2);
  49         if (!err->Process(strdata.c_str(), static_cast<int>(strdata.size()))) {
  50           err = nullptr;
  51         }
  52       }
  53     },
  54     [&err]() { err = nullptr; });
  55   while (out || err || !chain.Finished()) {
  56     uv_run(&chain.GetLoop(), UV_RUN_ONCE);
  57   }
  58
  59   std::vector<cmUVProcessChain::Status> result;
  60   auto status = chain.GetStatus();
  61   std::transform(
  62     status.begin(), status.end(), std::back_inserter(result),
  63     [](const cmUVProcessChain::Status* s) -> cmUVProcessChain::Status {
  64       return *s;
  65     });
  66   return result;
  67 }
  68
  69 cmProcessTools::LineParser::LineParser(char sep, bool ignoreCR)
  70   : Separator(sep)
  71   , IgnoreCR(ignoreCR)
  72 {
  73 }
  74
  75 void cmProcessTools::LineParser::SetLog(std::ostream* log, const char* prefix)
  76 {
  77   this->Log = log;
  78   this->Prefix = prefix ? prefix : "";
  79 }
  80
  81 bool cmProcessTools::LineParser::ProcessChunk(const char* first, int length)
  82 {
  83   const char* last = first + length;
  84   for (const char* c = first; c != last; ++c) {
  85     if (*c == this->Separator || *c == '\0') {
  86       this->LineEnd = *c;
  87
  88       // Log this line.
  89       if (this->Log && this->Prefix) {
  90         *this->Log << this->Prefix << this->Line << "\n";
  91       }
  92
  93       // Hand this line to the subclass implementation.
  94       if (!this->ProcessLine()) {
  95         this->Line.clear();
  96         return false;
  97       }
  98
  99       this->Line.clear();
 100     } else if (*c != '\r' || !this->IgnoreCR) {
 101       // Append this character to the line under construction.
 102       this->Line.append(1, *c);
 103     }
 104   }
 105   return true;
 106 }