mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-25 11:56:08 +00:00
1e246704e2
This simplifies the IOStream.cpp implementation by building on top of the existing lldb::IOObjectSP. Additionally, this should help ensure clients connected of a `--connection` specifier properly detect shutdown requests when the Socket is closed. Previously, the StreamDescriptor was just accessing the underlying native handle and was not aware of the `Close()` call to the Socket itself. This is both nice to have for simplifying the existing code and this unblocks an upcoming refactor to support the cancel request. --------- Co-authored-by: Jonas Devlieghere <jonas@devlieghere.com>
66 lines
1.6 KiB
C++
66 lines
1.6 KiB
C++
//===-- IOStream.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 "IOStream.h"
|
|
#include "lldb/Utility/IOObject.h"
|
|
#include "lldb/Utility/Status.h"
|
|
#include <fstream>
|
|
#include <string>
|
|
|
|
using namespace lldb_dap;
|
|
|
|
bool OutputStream::write_full(llvm::StringRef str) {
|
|
if (!descriptor)
|
|
return false;
|
|
|
|
size_t num_bytes = str.size();
|
|
auto status = descriptor->Write(str.data(), num_bytes);
|
|
return status.Success();
|
|
}
|
|
|
|
bool InputStream::read_full(std::ofstream *log, size_t length,
|
|
std::string &text) {
|
|
if (!descriptor)
|
|
return false;
|
|
|
|
std::string data;
|
|
data.resize(length);
|
|
|
|
auto status = descriptor->Read(data.data(), length);
|
|
if (status.Fail())
|
|
return false;
|
|
|
|
text += data;
|
|
return true;
|
|
}
|
|
|
|
bool InputStream::read_line(std::ofstream *log, std::string &line) {
|
|
line.clear();
|
|
while (true) {
|
|
if (!read_full(log, 1, line))
|
|
return false;
|
|
|
|
if (llvm::StringRef(line).ends_with("\r\n"))
|
|
break;
|
|
}
|
|
line.erase(line.size() - 2);
|
|
return true;
|
|
}
|
|
|
|
bool InputStream::read_expected(std::ofstream *log, llvm::StringRef expected) {
|
|
std::string result;
|
|
if (!read_full(log, expected.size(), result))
|
|
return false;
|
|
if (expected != result) {
|
|
if (log)
|
|
*log << "Warning: Expected '" << expected.str() << "', got '" << result
|
|
<< "\n";
|
|
}
|
|
return true;
|
|
}
|