1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
/* Distributed under the OSI-approved BSD 3-Clause License. See accompanying
file LICENSE.rst or https://cmake.org/licensing for details. */
#pragma once
#include "cmConfigure.h" // IWYU pragma: keep
#include <chrono>
#include <cstddef>
#include <cstdint>
#include <memory>
#include <string>
#include <utility>
#include <vector>
#include <cm/optional>
#include <cm3p/uv.h>
#include <sys/types.h>
#include "cmDuration.h"
#include "cmProcessOutput.h"
#include "cmUVHandlePtr.h"
class cmCTestRunTest;
/** \class cmProcess
* \brief run a process with c++
*
* cmProcess wraps the kwsys process stuff in a c++ class.
*/
class cmProcess
{
public:
explicit cmProcess(std::unique_ptr<cmCTestRunTest> runner);
~cmProcess();
void SetCommand(std::string const& command);
void SetCommandArguments(std::vector<std::string> const& arg);
void SetWorkingDirectory(std::string const& dir);
void SetTimeout(cmDuration t) { this->Timeout = t; }
void ChangeTimeout(cmDuration t);
void ResetStartTime();
// Return true if the process starts
bool StartProcess(uv_loop_t& loop, std::vector<size_t>* affinity);
enum class TimeoutReason
{
Normal,
StopTime,
};
void SetTimeoutReason(TimeoutReason r) { this->TimeoutReason_ = r; }
TimeoutReason GetTimeoutReason() const { return this->TimeoutReason_; }
enum class State
{
Starting,
Error,
Exception,
Executing,
Exited,
Expired,
Killed,
Disowned
};
State GetProcessStatus();
int GetId() const { return this->Id; }
void SetId(int id) { this->Id = id; }
int64_t GetExitValue() const { return this->ExitValue; }
cmDuration GetTotalTime() { return this->TotalTime; }
std::chrono::steady_clock::time_point GetStartTime()
{
return this->StartTime;
}
std::chrono::system_clock::time_point GetSystemStartTime()
{
return this->SystemStartTime;
}
enum class Exception
{
None,
Fault,
Illegal,
Interrupt,
Numerical,
Other
};
Exception GetExitException() const;
std::string GetExitExceptionString() const;
std::unique_ptr<cmCTestRunTest> GetRunner()
{
return std::move(this->Runner);
}
enum class Termination
{
Normal,
Custom,
Forced,
};
Termination GetTerminationStyle() const { return this->TerminationStyle; }
private:
cm::optional<cmDuration> Timeout;
TimeoutReason TimeoutReason_ = TimeoutReason::Normal;
std::chrono::steady_clock::time_point StartTime;
std::chrono::system_clock::time_point SystemStartTime;
cmDuration TotalTime;
bool ReadHandleClosed = false;
bool ProcessHandleClosed = false;
cm::uv_process_ptr Process;
cm::uv_pipe_ptr PipeReader;
cm::uv_timer_ptr Timer;
std::vector<char> Buf;
std::unique_ptr<cmCTestRunTest> Runner;
cmProcessOutput Conv;
int Signal = 0;
cmProcess::State ProcessState = cmProcess::State::Starting;
static void OnExitCB(uv_process_t* process, int64_t exit_status,
int term_signal);
static void OnTimeoutCB(uv_timer_t* timer);
static void OnReadCB(uv_stream_t* stream, ssize_t nread,
uv_buf_t const* buf);
static void OnAllocateCB(uv_handle_t* handle, size_t suggested_size,
uv_buf_t* buf);
void OnExit(int64_t exit_status, int term_signal);
void OnTimeout();
void OnRead(ssize_t nread, uv_buf_t const* buf);
void OnAllocate(size_t suggested_size, uv_buf_t* buf);
void StartTimer();
void Finish();
class Buffer : public std::vector<char>
{
// Half-open index range of partial line already scanned.
size_type First = 0;
size_type Last = 0;
public:
Buffer() = default;
bool GetLine(std::string& line);
bool GetLast(std::string& line);
};
Buffer Output;
std::string Command;
std::string WorkingDirectory;
std::vector<std::string> Arguments;
std::vector<char const*> ProcessArgs;
int Id;
int64_t ExitValue;
Termination TerminationStyle = Termination::Normal;
};
|
