1 //===-- DomainSocket.cpp ----------------------------------------*- C++ -*-===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "lldb/Host/posix/DomainSocket.h"
11
12 #include "llvm/Support/FileSystem.h"
13
14 #include <stddef.h>
15 #include <sys/socket.h>
16 #include <sys/un.h>
17
18 using namespace lldb;
19 using namespace lldb_private;
20
21 #ifdef __ANDROID__
22 // Android does not have SUN_LEN
23 #ifndef SUN_LEN
24 #define SUN_LEN(ptr) \
25 (offsetof(struct sockaddr_un, sun_path) + strlen((ptr)->sun_path))
26 #endif
27 #endif // #ifdef __ANDROID__
28
29 namespace {
30
31 const int kDomain = AF_UNIX;
32 const int kType = SOCK_STREAM;
33
SetSockAddr(llvm::StringRef name,const size_t name_offset,sockaddr_un * saddr_un,socklen_t & saddr_un_len)34 bool SetSockAddr(llvm::StringRef name, const size_t name_offset,
35 sockaddr_un *saddr_un, socklen_t &saddr_un_len) {
36 if (name.size() + name_offset > sizeof(saddr_un->sun_path))
37 return false;
38
39 memset(saddr_un, 0, sizeof(*saddr_un));
40 saddr_un->sun_family = kDomain;
41
42 memcpy(saddr_un->sun_path + name_offset, name.data(), name.size());
43
44 // For domain sockets we can use SUN_LEN in order to calculate size of
45 // sockaddr_un, but for abstract sockets we have to calculate size manually
46 // because of leading null symbol.
47 if (name_offset == 0)
48 saddr_un_len = SUN_LEN(saddr_un);
49 else
50 saddr_un_len =
51 offsetof(struct sockaddr_un, sun_path) + name_offset + name.size();
52
53 #if defined(__APPLE__) || defined(__FreeBSD__) || defined(__NetBSD__)
54 saddr_un->sun_len = saddr_un_len;
55 #endif
56
57 return true;
58 }
59 } // namespace
60
DomainSocket(bool should_close,bool child_processes_inherit)61 DomainSocket::DomainSocket(bool should_close, bool child_processes_inherit)
62 : Socket(ProtocolUnixDomain, should_close, child_processes_inherit) {}
63
DomainSocket(SocketProtocol protocol,bool child_processes_inherit)64 DomainSocket::DomainSocket(SocketProtocol protocol,
65 bool child_processes_inherit)
66 : Socket(protocol, true, child_processes_inherit) {}
67
DomainSocket(NativeSocket socket,const DomainSocket & listen_socket)68 DomainSocket::DomainSocket(NativeSocket socket,
69 const DomainSocket &listen_socket)
70 : Socket(ProtocolUnixDomain, listen_socket.m_should_close_fd,
71 listen_socket.m_child_processes_inherit) {
72 m_socket = socket;
73 }
74
Connect(llvm::StringRef name)75 Status DomainSocket::Connect(llvm::StringRef name) {
76 sockaddr_un saddr_un;
77 socklen_t saddr_un_len;
78 if (!SetSockAddr(name, GetNameOffset(), &saddr_un, saddr_un_len))
79 return Status("Failed to set socket address");
80
81 Status error;
82 m_socket = CreateSocket(kDomain, kType, 0, m_child_processes_inherit, error);
83 if (error.Fail())
84 return error;
85 if (::connect(GetNativeSocket(), (struct sockaddr *)&saddr_un, saddr_un_len) <
86 0)
87 SetLastError(error);
88
89 return error;
90 }
91
Listen(llvm::StringRef name,int backlog)92 Status DomainSocket::Listen(llvm::StringRef name, int backlog) {
93 sockaddr_un saddr_un;
94 socklen_t saddr_un_len;
95 if (!SetSockAddr(name, GetNameOffset(), &saddr_un, saddr_un_len))
96 return Status("Failed to set socket address");
97
98 DeleteSocketFile(name);
99
100 Status error;
101 m_socket = CreateSocket(kDomain, kType, 0, m_child_processes_inherit, error);
102 if (error.Fail())
103 return error;
104 if (::bind(GetNativeSocket(), (struct sockaddr *)&saddr_un, saddr_un_len) ==
105 0)
106 if (::listen(GetNativeSocket(), backlog) == 0)
107 return error;
108
109 SetLastError(error);
110 return error;
111 }
112
Accept(Socket * & socket)113 Status DomainSocket::Accept(Socket *&socket) {
114 Status error;
115 auto conn_fd = AcceptSocket(GetNativeSocket(), nullptr, nullptr,
116 m_child_processes_inherit, error);
117 if (error.Success())
118 socket = new DomainSocket(conn_fd, *this);
119
120 return error;
121 }
122
GetNameOffset() const123 size_t DomainSocket::GetNameOffset() const { return 0; }
124
DeleteSocketFile(llvm::StringRef name)125 void DomainSocket::DeleteSocketFile(llvm::StringRef name) {
126 llvm::sys::fs::remove(name);
127 }
128