forked from OSchip/llvm-project
128 lines
3.7 KiB
C++
128 lines
3.7 KiB
C++
//===-- DomainSocket.cpp ----------------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "lldb/Host/posix/DomainSocket.h"
|
|
|
|
#include "llvm/Support/FileSystem.h"
|
|
|
|
#include <stddef.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
|
|
using namespace lldb;
|
|
using namespace lldb_private;
|
|
|
|
#ifdef __ANDROID__
|
|
// Android does not have SUN_LEN
|
|
#ifndef SUN_LEN
|
|
#define SUN_LEN(ptr) \
|
|
(offsetof(struct sockaddr_un, sun_path) + strlen((ptr)->sun_path))
|
|
#endif
|
|
#endif // #ifdef __ANDROID__
|
|
|
|
namespace {
|
|
|
|
const int kDomain = AF_UNIX;
|
|
const int kType = SOCK_STREAM;
|
|
|
|
bool SetSockAddr(llvm::StringRef name, const size_t name_offset,
|
|
sockaddr_un *saddr_un, socklen_t &saddr_un_len) {
|
|
if (name.size() + name_offset > sizeof(saddr_un->sun_path))
|
|
return false;
|
|
|
|
memset(saddr_un, 0, sizeof(*saddr_un));
|
|
saddr_un->sun_family = kDomain;
|
|
|
|
memcpy(saddr_un->sun_path + name_offset, name.data(), name.size());
|
|
|
|
// For domain sockets we can use SUN_LEN in order to calculate size of
|
|
// sockaddr_un, but for abstract sockets we have to calculate size manually
|
|
// because of leading null symbol.
|
|
if (name_offset == 0)
|
|
saddr_un_len = SUN_LEN(saddr_un);
|
|
else
|
|
saddr_un_len =
|
|
offsetof(struct sockaddr_un, sun_path) + name_offset + name.size();
|
|
|
|
#if defined(__APPLE__) || defined(__FreeBSD__) || defined(__NetBSD__)
|
|
saddr_un->sun_len = saddr_un_len;
|
|
#endif
|
|
|
|
return true;
|
|
}
|
|
} // namespace
|
|
|
|
DomainSocket::DomainSocket(bool should_close, bool child_processes_inherit)
|
|
: Socket(ProtocolUnixDomain, should_close, child_processes_inherit) {}
|
|
|
|
DomainSocket::DomainSocket(SocketProtocol protocol,
|
|
bool child_processes_inherit)
|
|
: Socket(protocol, true, child_processes_inherit) {}
|
|
|
|
DomainSocket::DomainSocket(NativeSocket socket,
|
|
const DomainSocket &listen_socket)
|
|
: Socket(ProtocolUnixDomain, listen_socket.m_should_close_fd,
|
|
listen_socket.m_child_processes_inherit) {
|
|
m_socket = socket;
|
|
}
|
|
|
|
Status DomainSocket::Connect(llvm::StringRef name) {
|
|
sockaddr_un saddr_un;
|
|
socklen_t saddr_un_len;
|
|
if (!SetSockAddr(name, GetNameOffset(), &saddr_un, saddr_un_len))
|
|
return Status("Failed to set socket address");
|
|
|
|
Status error;
|
|
m_socket = CreateSocket(kDomain, kType, 0, m_child_processes_inherit, error);
|
|
if (error.Fail())
|
|
return error;
|
|
if (::connect(GetNativeSocket(), (struct sockaddr *)&saddr_un, saddr_un_len) <
|
|
0)
|
|
SetLastError(error);
|
|
|
|
return error;
|
|
}
|
|
|
|
Status DomainSocket::Listen(llvm::StringRef name, int backlog) {
|
|
sockaddr_un saddr_un;
|
|
socklen_t saddr_un_len;
|
|
if (!SetSockAddr(name, GetNameOffset(), &saddr_un, saddr_un_len))
|
|
return Status("Failed to set socket address");
|
|
|
|
DeleteSocketFile(name);
|
|
|
|
Status error;
|
|
m_socket = CreateSocket(kDomain, kType, 0, m_child_processes_inherit, error);
|
|
if (error.Fail())
|
|
return error;
|
|
if (::bind(GetNativeSocket(), (struct sockaddr *)&saddr_un, saddr_un_len) ==
|
|
0)
|
|
if (::listen(GetNativeSocket(), backlog) == 0)
|
|
return error;
|
|
|
|
SetLastError(error);
|
|
return error;
|
|
}
|
|
|
|
Status DomainSocket::Accept(Socket *&socket) {
|
|
Status error;
|
|
auto conn_fd = AcceptSocket(GetNativeSocket(), nullptr, nullptr,
|
|
m_child_processes_inherit, error);
|
|
if (error.Success())
|
|
socket = new DomainSocket(conn_fd, *this);
|
|
|
|
return error;
|
|
}
|
|
|
|
size_t DomainSocket::GetNameOffset() const { return 0; }
|
|
|
|
void DomainSocket::DeleteSocketFile(llvm::StringRef name) {
|
|
llvm::sys::fs::remove(name);
|
|
}
|