1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
//===-- 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 "lldb/Host/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;
}
}
DomainSocket::DomainSocket(NativeSocket socket)
: Socket(socket, ProtocolUnixDomain, true) {}
DomainSocket::DomainSocket(bool child_processes_inherit, Error &error)
: DomainSocket(
CreateSocket(kDomain, kType, 0, child_processes_inherit, error)) {}
DomainSocket::DomainSocket(SocketProtocol protocol,
bool child_processes_inherit, Error &error)
: Socket(CreateSocket(kDomain, kType, 0, child_processes_inherit, error),
protocol, true) {}
Error DomainSocket::Connect(llvm::StringRef name) {
sockaddr_un saddr_un;
socklen_t saddr_un_len;
if (!SetSockAddr(name, GetNameOffset(), &saddr_un, saddr_un_len))
return Error("Failed to set socket address");
Error error;
if (::connect(GetNativeSocket(), (struct sockaddr *)&saddr_un, saddr_un_len) <
0)
SetLastError(error);
return error;
}
Error 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 Error("Failed to set socket address");
DeleteSocketFile(name);
Error error;
if (::bind(GetNativeSocket(), (struct sockaddr *)&saddr_un, saddr_un_len) ==
0)
if (::listen(GetNativeSocket(), backlog) == 0)
return error;
SetLastError(error);
return error;
}
Error DomainSocket::Accept(llvm::StringRef name, bool child_processes_inherit,
Socket *&socket) {
Error error;
auto conn_fd = AcceptSocket(GetNativeSocket(), nullptr, nullptr,
child_processes_inherit, error);
if (error.Success())
socket = new DomainSocket(conn_fd);
return error;
}
size_t DomainSocket::GetNameOffset() const { return 0; }
void DomainSocket::DeleteSocketFile(llvm::StringRef name) {
FileSystem::Unlink(FileSpec{name, true});
}
|
