forked from OSchip/llvm-project
92 lines
2.9 KiB
C++
92 lines
2.9 KiB
C++
//===-- FifoFiles.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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "FifoFiles.h"
|
|
|
|
#if !defined(_WIN32)
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#include <chrono>
|
|
#include <fstream>
|
|
#include <future>
|
|
#include <thread>
|
|
|
|
#include "llvm/Support/FileSystem.h"
|
|
|
|
#include "lldb/lldb-defines.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace lldb_vscode {
|
|
|
|
FifoFile::FifoFile(StringRef path) : m_path(path) {}
|
|
|
|
FifoFile::~FifoFile() {
|
|
#if !defined(_WIN32)
|
|
unlink(m_path.c_str());
|
|
#endif
|
|
}
|
|
|
|
Expected<std::shared_ptr<FifoFile>> CreateFifoFile(StringRef path) {
|
|
#if defined(_WIN32)
|
|
return createStringError(inconvertibleErrorCode(), "Unimplemented");
|
|
#else
|
|
if (int err = mkfifo(path.data(), 0600))
|
|
return createStringError(std::error_code(err, std::generic_category()),
|
|
"Couldn't create fifo file: %s", path.data());
|
|
return std::make_shared<FifoFile>(path);
|
|
#endif
|
|
}
|
|
|
|
FifoFileIO::FifoFileIO(StringRef fifo_file, StringRef other_endpoint_name)
|
|
: m_fifo_file(fifo_file), m_other_endpoint_name(other_endpoint_name) {}
|
|
|
|
Expected<json::Value> FifoFileIO::ReadJSON(std::chrono::milliseconds timeout) {
|
|
// We use a pointer for this future, because otherwise its normal destructor
|
|
// would wait for the getline to end, rendering the timeout useless.
|
|
Optional<std::string> line;
|
|
std::future<void> *future =
|
|
new std::future<void>(std::async(std::launch::async, [&]() {
|
|
std::ifstream reader(m_fifo_file, std::ifstream::in);
|
|
std::string buffer;
|
|
std::getline(reader, buffer);
|
|
if (!buffer.empty())
|
|
line = buffer;
|
|
}));
|
|
if (future->wait_for(timeout) == std::future_status::timeout ||
|
|
!line.hasValue())
|
|
return createStringError(inconvertibleErrorCode(),
|
|
"Timed out trying to get messages from the " +
|
|
m_other_endpoint_name);
|
|
delete future;
|
|
return json::parse(*line);
|
|
}
|
|
|
|
Error FifoFileIO::SendJSON(const json::Value &json,
|
|
std::chrono::milliseconds timeout) {
|
|
bool done = false;
|
|
std::future<void> *future =
|
|
new std::future<void>(std::async(std::launch::async, [&]() {
|
|
std::ofstream writer(m_fifo_file, std::ofstream::out);
|
|
writer << JSONToString(json) << std::endl;
|
|
done = true;
|
|
}));
|
|
if (future->wait_for(timeout) == std::future_status::timeout || !done) {
|
|
return createStringError(inconvertibleErrorCode(),
|
|
"Timed out trying to send messages to the " +
|
|
m_other_endpoint_name);
|
|
}
|
|
delete future;
|
|
return Error::success();
|
|
}
|
|
|
|
} // namespace lldb_vscode
|