#if defined(__APPLE__)
#define _DARWIN_UNLIMITED_SELECT
#endif
#include "lldb/Host/posix/ConnectionFileDescriptorPosix.h"
#include "lldb/Host/Config.h"
#include "lldb/Host/FileSystem.h"
#include "lldb/Host/Socket.h"
#include "lldb/Host/SocketAddress.h"
#include "lldb/Utility/LLDBLog.h"
#include "lldb/Utility/SelectHelper.h"
#include "lldb/Utility/Timeout.h"
#include <cerrno>
#include <cstdlib>
#include <cstring>
#include <fcntl.h>
#include <sys/types.h>
#if LLDB_ENABLE_POSIX
#include <termios.h>
#include <unistd.h>
#endif
#include <memory>
#include <sstream>
#include "llvm/Support/Errno.h"
#include "llvm/Support/ErrorHandling.h"
#if defined(__APPLE__)
#include "llvm/ADT/SmallVector.h"
#endif
#include "lldb/Host/Host.h"
#include "lldb/Host/Socket.h"
#include "lldb/Host/common/TCPSocket.h"
#include "lldb/Host/common/UDPSocket.h"
#include "lldb/Utility/Log.h"
#include "lldb/Utility/StreamString.h"
#include "lldb/Utility/Timer.h"
using namespace lldb;
using namespace lldb_private;
ConnectionFileDescriptor::ConnectionFileDescriptor(bool child_processes_inherit)
: Connection(), m_pipe(), m_mutex(), m_shutting_down(false),
m_child_processes_inherit(child_processes_inherit) {
Log *log(GetLog(LLDBLog::Connection | LLDBLog::Object));
LLDB_LOGF(log, "%p ConnectionFileDescriptor::ConnectionFileDescriptor ()",
static_cast<void *>(this));
}
ConnectionFileDescriptor::ConnectionFileDescriptor(int fd, bool owns_fd)
: Connection(), m_pipe(), m_mutex(), m_shutting_down(false),
m_child_processes_inherit(false) {
m_io_sp =
std::make_shared<NativeFile>(fd, File::eOpenOptionReadWrite, owns_fd);
Log *log(GetLog(LLDBLog::Connection | LLDBLog::Object));
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::ConnectionFileDescriptor (fd = "
"%i, owns_fd = %i)",
static_cast<void *>(this), fd, owns_fd);
OpenCommandPipe();
}
ConnectionFileDescriptor::ConnectionFileDescriptor(Socket *socket)
: Connection(), m_pipe(), m_mutex(), m_shutting_down(false),
m_child_processes_inherit(false) {
InitializeSocket(socket);
}
ConnectionFileDescriptor::~ConnectionFileDescriptor() {
Log *log(GetLog(LLDBLog::Connection | LLDBLog::Object));
LLDB_LOGF(log, "%p ConnectionFileDescriptor::~ConnectionFileDescriptor ()",
static_cast<void *>(this));
Disconnect(nullptr);
CloseCommandPipe();
}
void ConnectionFileDescriptor::OpenCommandPipe() {
CloseCommandPipe();
Log *log = GetLog(LLDBLog::Connection);
Status result = m_pipe.CreateNew(m_child_processes_inherit);
if (!result.Success()) {
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::OpenCommandPipe () - could not "
"make pipe: %s",
static_cast<void *>(this), result.AsCString());
} else {
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::OpenCommandPipe() - success "
"readfd=%d writefd=%d",
static_cast<void *>(this), m_pipe.GetReadFileDescriptor(),
m_pipe.GetWriteFileDescriptor());
}
}
void ConnectionFileDescriptor::CloseCommandPipe() {
Log *log = GetLog(LLDBLog::Connection);
LLDB_LOGF(log, "%p ConnectionFileDescriptor::CloseCommandPipe()",
static_cast<void *>(this));
m_pipe.Close();
}
bool ConnectionFileDescriptor::IsConnected() const {
return m_io_sp && m_io_sp->IsValid();
}
ConnectionStatus ConnectionFileDescriptor::Connect(llvm::StringRef path,
Status *error_ptr) {
return Connect(
path, [](llvm::StringRef) {}, error_ptr);
}
ConnectionStatus
ConnectionFileDescriptor::Connect(llvm::StringRef path,
socket_id_callback_type socket_id_callback,
Status *error_ptr) {
std::lock_guard<std::recursive_mutex> guard(m_mutex);
Log *log = GetLog(LLDBLog::Connection);
LLDB_LOGF(log, "%p ConnectionFileDescriptor::Connect (url = '%s')",
static_cast<void *>(this), path.str().c_str());
OpenCommandPipe();
if (path.empty()) {
if (error_ptr)
error_ptr->SetErrorString("invalid connect arguments");
return eConnectionStatusError;
}
llvm::StringRef scheme;
std::tie(scheme, path) = path.split("://");
if (!path.empty()) {
auto method =
llvm::StringSwitch<ConnectionStatus (ConnectionFileDescriptor::*)(
llvm::StringRef, socket_id_callback_type, Status *)>(scheme)
#ifndef MS_DEBUGGER
.Case("listen", &ConnectionFileDescriptor::AcceptTCP)
.Cases("accept", "unix-accept",
&ConnectionFileDescriptor::AcceptNamedSocket)
.Case("unix-abstract-accept",
&ConnectionFileDescriptor::AcceptAbstractSocket)
.Cases("connect", "tcp-connect",
&ConnectionFileDescriptor::ConnectTCP)
.Case("udp", &ConnectionFileDescriptor::ConnectUDP)
.Case("unix-connect", &ConnectionFileDescriptor::ConnectNamedSocket)
.Case("unix-abstract-connect",
&ConnectionFileDescriptor::ConnectAbstractSocket)
#endif
#if LLDB_ENABLE_POSIX
.Case("fd", &ConnectionFileDescriptor::ConnectFD)
#ifndef MS_DEBUGGER
.Case("file", &ConnectionFileDescriptor::ConnectFile)
.Case("serial", &ConnectionFileDescriptor::ConnectSerialPort)
#endif
#endif
.Default(nullptr);
if (method) {
if (error_ptr)
*error_ptr = Status();
return (this->*method)(path, socket_id_callback, error_ptr);
}
}
if (error_ptr)
error_ptr->SetErrorStringWithFormat("unsupported connection URL: '%s'",
path.str().c_str());
return eConnectionStatusError;
}
bool ConnectionFileDescriptor::InterruptRead() {
size_t bytes_written = 0;
Status result = m_pipe.Write("i", 1, bytes_written);
return result.Success();
}
ConnectionStatus ConnectionFileDescriptor::Disconnect(Status *error_ptr) {
Log *log = GetLog(LLDBLog::Connection);
LLDB_LOGF(log, "%p ConnectionFileDescriptor::Disconnect ()",
static_cast<void *>(this));
ConnectionStatus status = eConnectionStatusSuccess;
if (!IsConnected()) {
LLDB_LOGF(
log, "%p ConnectionFileDescriptor::Disconnect(): Nothing to disconnect",
static_cast<void *>(this));
return eConnectionStatusSuccess;
}
std::unique_lock<std::recursive_mutex> locker(m_mutex, std::defer_lock);
if (!locker.try_lock()) {
if (m_pipe.CanWrite()) {
size_t bytes_written = 0;
Status result = m_pipe.Write("q", 1, bytes_written);
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::Disconnect(): Couldn't get "
"the lock, sent 'q' to %d, error = '%s'.",
static_cast<void *>(this), m_pipe.GetWriteFileDescriptor(),
result.AsCString());
} else if (log) {
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::Disconnect(): Couldn't get the "
"lock, but no command pipe is available.",
static_cast<void *>(this));
}
locker.lock();
}
m_shutting_down = true;
Status error = m_io_sp->Close();
if (error.Fail())
status = eConnectionStatusError;
if (error_ptr)
*error_ptr = error;
m_pipe.Close();
m_uri.clear();
m_shutting_down = false;
return status;
}
size_t ConnectionFileDescriptor::Read(void *dst, size_t dst_len,
const Timeout<std::micro> &timeout,
ConnectionStatus &status,
Status *error_ptr) {
Log *log = GetLog(LLDBLog::Connection);
std::unique_lock<std::recursive_mutex> locker(m_mutex, std::defer_lock);
if (!locker.try_lock()) {
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::Read () failed to get the "
"connection lock.",
static_cast<void *>(this));
if (error_ptr)
error_ptr->SetErrorString("failed to get the connection lock for read.");
status = eConnectionStatusTimedOut;
return 0;
}
if (m_shutting_down) {
if (error_ptr)
error_ptr->SetErrorString("shutting down");
status = eConnectionStatusError;
return 0;
}
status = BytesAvailable(timeout, error_ptr);
if (status != eConnectionStatusSuccess)
return 0;
Status error;
size_t bytes_read = dst_len;
error = m_io_sp->Read(dst, bytes_read);
if (log) {
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::Read() fd = %" PRIu64
", dst = %p, dst_len = %" PRIu64 ") => %" PRIu64 ", error = %s",
static_cast<void *>(this),
static_cast<uint64_t>(m_io_sp->GetWaitableHandle()),
static_cast<void *>(dst), static_cast<uint64_t>(dst_len),
static_cast<uint64_t>(bytes_read), error.AsCString());
}
if (bytes_read == 0) {
error.Clear();
status = eConnectionStatusEndOfFile;
}
if (error_ptr)
*error_ptr = error;
if (error.Fail()) {
uint32_t error_value = error.GetError();
switch (error_value) {
case EAGAIN:
if (m_io_sp->GetFdType() == IOObject::eFDTypeSocket)
status = eConnectionStatusTimedOut;
else
status = eConnectionStatusSuccess;
return 0;
case EFAULT:
case EINTR:
case EINVAL:
case EIO:
case EISDIR:
case ENOBUFS:
case ENOMEM:
status = eConnectionStatusError;
break;
case ENOENT:
case EBADF:
case ENXIO:
case ECONNRESET:
case ENOTCONN:
status = eConnectionStatusLostConnection;
break;
case ETIMEDOUT:
status = eConnectionStatusTimedOut;
return 0;
default:
LLDB_LOG(log, "this = {0}, unexpected error: {1}", this,
llvm::sys::StrError(error_value));
status = eConnectionStatusError;
break;
}
return 0;
}
return bytes_read;
}
size_t ConnectionFileDescriptor::Write(const void *src, size_t src_len,
ConnectionStatus &status,
Status *error_ptr) {
Log *log = GetLog(LLDBLog::Connection);
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::Write (src = %p, src_len = %" PRIu64
")",
static_cast<void *>(this), static_cast<const void *>(src),
static_cast<uint64_t>(src_len));
if (!IsConnected()) {
if (error_ptr)
error_ptr->SetErrorString("not connected");
status = eConnectionStatusNoConnection;
return 0;
}
if (m_shutting_down) {
if (error_ptr)
error_ptr->SetErrorString("shutting down");
status = eConnectionStatusError;
return 0;
}
Status error;
size_t bytes_sent = src_len;
error = m_io_sp->Write(src, bytes_sent);
if (log) {
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::Write(fd = %" PRIu64
", src = %p, src_len = %" PRIu64 ") => %" PRIu64 " (error = %s)",
static_cast<void *>(this),
static_cast<uint64_t>(m_io_sp->GetWaitableHandle()),
static_cast<const void *>(src), static_cast<uint64_t>(src_len),
static_cast<uint64_t>(bytes_sent), error.AsCString());
}
if (error_ptr)
*error_ptr = error;
if (error.Fail()) {
switch (error.GetError()) {
case EAGAIN:
case EINTR:
status = eConnectionStatusSuccess;
return 0;
case ECONNRESET:
case ENOTCONN:
status = eConnectionStatusLostConnection;
break;
default:
status = eConnectionStatusError;
break;
}
return 0;
}
status = eConnectionStatusSuccess;
return bytes_sent;
}
std::string ConnectionFileDescriptor::GetURI() { return m_uri; }
ConnectionStatus
ConnectionFileDescriptor::BytesAvailable(const Timeout<std::micro> &timeout,
Status *error_ptr) {
Log *log = GetLog(LLDBLog::Connection);
LLDB_LOG(log, "this = {0}, timeout = {1}", this, timeout);
const IOObject::WaitableHandle handle = m_io_sp->GetWaitableHandle();
const int pipe_fd = m_pipe.GetReadFileDescriptor();
if (handle != IOObject::kInvalidHandleValue) {
SelectHelper select_helper;
if (timeout)
select_helper.SetTimeout(*timeout);
select_helper.FDSetRead(handle);
#if defined(_WIN32)
const bool have_pipe_fd = false;
#else
const bool have_pipe_fd = pipe_fd >= 0;
#endif
if (have_pipe_fd)
select_helper.FDSetRead(pipe_fd);
while (handle == m_io_sp->GetWaitableHandle()) {
Status error = select_helper.Select();
if (error_ptr)
*error_ptr = error;
if (error.Fail()) {
switch (error.GetError()) {
case EBADF:
return eConnectionStatusLostConnection;
case EINVAL:
default:
return eConnectionStatusError;
case ETIMEDOUT:
return eConnectionStatusTimedOut;
case EAGAIN:
case EINTR:
break;
}
} else {
if (select_helper.FDIsSetRead(handle))
return eConnectionStatusSuccess;
if (select_helper.FDIsSetRead(pipe_fd)) {
char c;
ssize_t bytes_read =
llvm::sys::RetryAfterSignal(-1, ::read, pipe_fd, &c, 1);
assert(bytes_read == 1);
UNUSED_IF_ASSERT_DISABLED(bytes_read);
switch (c) {
case 'q':
LLDB_LOGF(log,
"%p ConnectionFileDescriptor::BytesAvailable() "
"got data: %c from the command channel.",
static_cast<void *>(this), c);
return eConnectionStatusEndOfFile;
case 'i':
return eConnectionStatusInterrupted;
}
}
}
}
}
if (error_ptr)
error_ptr->SetErrorString("not connected");
return eConnectionStatusLostConnection;
}
lldb::ConnectionStatus ConnectionFileDescriptor::AcceptSocket(
Socket::SocketProtocol socket_protocol, llvm::StringRef socket_name,
llvm::function_ref<void(Socket &)> post_listen_callback,
Status *error_ptr) {
Status error;
std::unique_ptr<Socket> listening_socket =
Socket::Create(socket_protocol, m_child_processes_inherit, error);
Socket *accepted_socket;
if (!error.Fail())
error = listening_socket->Listen(socket_name, 5);
if (!error.Fail()) {
post_listen_callback(*listening_socket);
error = listening_socket->Accept(accepted_socket);
}
if (!error.Fail()) {
m_io_sp.reset(accepted_socket);
m_uri.assign(socket_name.str());
return eConnectionStatusSuccess;
}
if (error_ptr)
*error_ptr = error;
return eConnectionStatusError;
}
lldb::ConnectionStatus
ConnectionFileDescriptor::ConnectSocket(Socket::SocketProtocol socket_protocol,
llvm::StringRef socket_name,
Status *error_ptr) {
Status error;
std::unique_ptr<Socket> socket =
Socket::Create(socket_protocol, m_child_processes_inherit, error);
if (!error.Fail())
error = socket->Connect(socket_name);
if (!error.Fail()) {
m_io_sp = std::move(socket);
m_uri.assign(socket_name.str());
return eConnectionStatusSuccess;
}
if (error_ptr)
*error_ptr = error;
return eConnectionStatusError;
}
ConnectionStatus ConnectionFileDescriptor::AcceptNamedSocket(
llvm::StringRef socket_name, socket_id_callback_type socket_id_callback,
Status *error_ptr) {
return AcceptSocket(
Socket::ProtocolUnixDomain, socket_name,
[socket_id_callback, socket_name](Socket &listening_socket) {
socket_id_callback(socket_name);
},
error_ptr);
}
ConnectionStatus ConnectionFileDescriptor::ConnectNamedSocket(
llvm::StringRef socket_name, socket_id_callback_type socket_id_callback,
Status *error_ptr) {
return ConnectSocket(Socket::ProtocolUnixDomain, socket_name, error_ptr);
}
ConnectionStatus ConnectionFileDescriptor::AcceptAbstractSocket(
llvm::StringRef socket_name, socket_id_callback_type socket_id_callback,
Status *error_ptr) {
return AcceptSocket(
Socket::ProtocolUnixAbstract, socket_name,
[socket_id_callback, socket_name](Socket &listening_socket) {
socket_id_callback(socket_name);
},
error_ptr);
}
lldb::ConnectionStatus ConnectionFileDescriptor::ConnectAbstractSocket(
llvm::StringRef socket_name, socket_id_callback_type socket_id_callback,
Status *error_ptr) {
return ConnectSocket(Socket::ProtocolUnixAbstract, socket_name, error_ptr);
}
ConnectionStatus
ConnectionFileDescriptor::AcceptTCP(llvm::StringRef socket_name,
socket_id_callback_type socket_id_callback,
Status *error_ptr) {
ConnectionStatus ret = AcceptSocket(
Socket::ProtocolTcp, socket_name,
[socket_id_callback](Socket &listening_socket) {
uint16_t port =
static_cast<TCPSocket &>(listening_socket).GetLocalPortNumber();
socket_id_callback(std::to_string(port));
},
error_ptr);
if (ret == eConnectionStatusSuccess)
m_uri.assign(
static_cast<TCPSocket *>(m_io_sp.get())->GetRemoteConnectionURI());
return ret;
}
ConnectionStatus
ConnectionFileDescriptor::ConnectTCP(llvm::StringRef socket_name,
socket_id_callback_type socket_id_callback,
Status *error_ptr) {
return ConnectSocket(Socket::ProtocolTcp, socket_name, error_ptr);
}
ConnectionStatus
ConnectionFileDescriptor::ConnectUDP(llvm::StringRef s,
socket_id_callback_type socket_id_callback,
Status *error_ptr) {
if (error_ptr)
*error_ptr = Status();
llvm::Expected<std::unique_ptr<UDPSocket>> socket =
Socket::UdpConnect(s, m_child_processes_inherit);
if (!socket) {
if (error_ptr)
*error_ptr = socket.takeError();
else
LLDB_LOG_ERROR(GetLog(LLDBLog::Connection), socket.takeError(),
"tcp connect failed: {0}");
return eConnectionStatusError;
}
m_io_sp = std::move(*socket);
m_uri.assign(std::string(s));
return eConnectionStatusSuccess;
}
ConnectionStatus
ConnectionFileDescriptor::ConnectFD(llvm::StringRef s,
socket_id_callback_type socket_id_callback,
Status *error_ptr) {
#if LLDB_ENABLE_POSIX
int fd = -1;
if (!s.getAsInteger(0, fd)) {
errno = 0;
int flags = ::fcntl(fd, F_GETFL, 0);
if (flags == -1 || errno == EBADF) {
if (error_ptr)
error_ptr->SetErrorStringWithFormat("stale file descriptor: %s",
s.str().c_str());
m_io_sp.reset();
return eConnectionStatusError;
} else {
std::unique_ptr<TCPSocket> tcp_socket;
tcp_socket = std::make_unique<TCPSocket>(fd, false, false);
int resuse;
bool is_socket =
!!tcp_socket->GetOption(SOL_SOCKET, SO_REUSEADDR, resuse);
if (is_socket)
m_io_sp = std::move(tcp_socket);
else
m_io_sp =
std::make_shared<NativeFile>(fd, File::eOpenOptionReadWrite, false);
m_uri = s.str();
return eConnectionStatusSuccess;
}
}
if (error_ptr)
error_ptr->SetErrorStringWithFormat("invalid file descriptor: \"%s\"",
s.str().c_str());
m_io_sp.reset();
return eConnectionStatusError;
#endif
llvm_unreachable("this function should be only called w/ LLDB_ENABLE_POSIX");
}
ConnectionStatus ConnectionFileDescriptor::ConnectFile(
llvm::StringRef s, socket_id_callback_type socket_id_callback,
Status *error_ptr) {
#if LLDB_ENABLE_POSIX
std::string addr_str = s.str();
int fd = FileSystem::Instance().Open(addr_str.c_str(), O_RDWR);
if (fd == -1) {
if (error_ptr)
error_ptr->SetErrorToErrno();
return eConnectionStatusError;
}
if (::isatty(fd)) {
struct termios options;
::tcgetattr(fd, &options);
::cfsetospeed(&options, B115200);
::cfsetispeed(&options, B115200);
options.c_lflag &= ~(ICANON | ECHO | ECHOE | ISIG);
options.c_cc[VMIN] = 1;
options.c_cc[VTIME] = 0;
llvm::sys::RetryAfterSignal(-1, ::tcsetattr, fd, TCSANOW, &options);
}
m_io_sp = std::make_shared<NativeFile>(fd, File::eOpenOptionReadWrite, true);
return eConnectionStatusSuccess;
#endif
llvm_unreachable("this function should be only called w/ LLDB_ENABLE_POSIX");
}
ConnectionStatus ConnectionFileDescriptor::ConnectSerialPort(
llvm::StringRef s, socket_id_callback_type socket_id_callback,
Status *error_ptr) {
#if LLDB_ENABLE_POSIX
llvm::StringRef path, qs;
std::tie(path, qs) = s.split('?');
llvm::Expected<SerialPort::Options> serial_options =
SerialPort::OptionsFromURL(qs);
if (!serial_options) {
if (error_ptr)
*error_ptr = serial_options.takeError();
else
llvm::consumeError(serial_options.takeError());
return eConnectionStatusError;
}
int fd = FileSystem::Instance().Open(path.str().c_str(), O_RDWR);
if (fd == -1) {
if (error_ptr)
error_ptr->SetErrorToErrno();
return eConnectionStatusError;
}
llvm::Expected<std::unique_ptr<SerialPort>> serial_sp = SerialPort::Create(
fd, File::eOpenOptionReadWrite, serial_options.get(), true);
if (!serial_sp) {
if (error_ptr)
*error_ptr = serial_sp.takeError();
else
llvm::consumeError(serial_sp.takeError());
return eConnectionStatusError;
}
m_io_sp = std::move(serial_sp.get());
return eConnectionStatusSuccess;
#endif
llvm_unreachable("this function should be only called w/ LLDB_ENABLE_POSIX");
}
bool ConnectionFileDescriptor::GetChildProcessesInherit() const {
return m_child_processes_inherit;
}
void ConnectionFileDescriptor::SetChildProcessesInherit(
bool child_processes_inherit) {
m_child_processes_inherit = child_processes_inherit;
}
void ConnectionFileDescriptor::InitializeSocket(Socket *socket) {
m_io_sp.reset(socket);
m_uri = socket->GetRemoteConnectionURI();
}