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
|
//===-- OutputRedirector.cpp -----------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===/
#if !defined(_WIN32)
#include <unistd.h>
#endif
#include "OutputRedirector.h"
using namespace llvm;
namespace lldb_vscode {
Error RedirectFd(int fd, std::function<void(llvm::StringRef)> callback) {
#if !defined(_WIN32)
int new_fd[2];
if (pipe(new_fd) == -1) {
int error = errno;
return createStringError(inconvertibleErrorCode(),
"Couldn't create new pipe for fd %d. %s", fd,
strerror(error));
}
if (dup2(new_fd[1], fd) == -1) {
int error = errno;
return createStringError(inconvertibleErrorCode(),
"Couldn't override the fd %d. %s", fd,
strerror(error));
}
int read_fd = new_fd[0];
std::thread t([read_fd, callback]() {
char buffer[4096];
while (true) {
ssize_t bytes_count = read(read_fd, &buffer, sizeof(buffer));
if (bytes_count == 0)
return;
if (bytes_count == -1) {
if (errno == EAGAIN || errno == EINTR)
continue;
break;
}
callback(StringRef(buffer, bytes_count).str());
}
});
t.detach();
#endif
return Error::success();
}
} // namespace lldb_vscode
|
