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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
//
// Copyright 2017 Garrett D'Amore <garrett@damore.org>
//
// This software is supplied under the terms of the MIT License, a
// copy of which should be located in the distribution where this
// file was obtained (LICENSE.txt). A copy of the license may also be
// found online at https://opensource.org/licenses/MIT.
//
#include "core/nng_impl.h"
#ifdef PLATFORM_POSIX_IPC
#include "platform/posix/posix_aio.h"
#include <errno.h>
#include <fcntl.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/uio.h>
#include <sys/un.h>
#include <unistd.h>
#ifdef SOCK_CLOEXEC
#define NNI_STREAM_SOCKTYPE (SOCK_STREAM | SOCK_CLOEXEC)
#else
#define NNI_STREAM_SOCKTYPE SOCK_STREAM
#endif
// Solaris/SunOS systems define this, which collides with our symbol
// names. Just undefine it now.
#ifdef sun
#undef sun
#endif
// We alias nni_posix_pipedesc to nni_plat_ipc_pipe.
// We alias nni_posix_epdesc to nni_plat_ipc_ep.
int
nni_plat_ipc_ep_init(nni_plat_ipc_ep **epp, const char *url, int mode)
{
nni_posix_epdesc * ed;
int rv;
struct sockaddr_un sun;
const char * path;
if (strncmp(url, "ipc://", strlen("ipc://")) != 0) {
return (NNG_EADDRINVAL);
}
path = url + strlen("ipc://"); // skip the prefix.
// prepare the sockaddr_un
sun.sun_family = AF_UNIX;
if (strlen(url) >= sizeof(sun.sun_path)) {
return (NNG_EADDRINVAL);
}
snprintf(sun.sun_path, sizeof(sun.sun_path), "%s", path);
if ((rv = nni_posix_epdesc_init(&ed, url)) != 0) {
return (rv);
}
switch (mode) {
case NNI_EP_MODE_DIAL:
nni_posix_epdesc_set_remote(ed, &sun, sizeof(sun));
break;
case NNI_EP_MODE_LISTEN:
nni_posix_epdesc_set_local(ed, &sun, sizeof(sun));
break;
default:
nni_posix_epdesc_fini(ed);
return (NNG_EINVAL);
}
*epp = (void *) ed;
return (0);
}
void
nni_plat_ipc_ep_fini(nni_plat_ipc_ep *ep)
{
nni_posix_epdesc_fini((void *) ep);
}
void
nni_plat_ipc_ep_close(nni_plat_ipc_ep *ep)
{
nni_posix_epdesc_close((void *) ep);
}
// UNIX DOMAIN SOCKETS -- these have names in the file namespace.
// We are going to check to see if there was a name already there.
// If there was, and nothing is listening (ECONNREFUSED), then we
// will just try to cleanup the old socket. Note that this is not
// perfect in all scenarios, so use this with caution.
static int
nni_plat_ipc_remove_stale(const char *path)
{
int fd;
int rv;
struct sockaddr_un sun;
sun.sun_family = AF_UNIX;
snprintf(sun.sun_path, sizeof(sun.sun_path), "%s", path);
if ((fd = socket(AF_UNIX, NNI_STREAM_SOCKTYPE, 0)) < 0) {
return (nni_plat_errno(errno));
}
// There is an assumption here that connect() returns immediately
// (even when non-blocking) when a server is absent. This seems
// to be true for the platforms we've tried. If it doesn't work,
// then the cleanup will fail. As this is supposed to be an
// exceptional case, don't worry.
(void) fcntl(fd, F_SETFL, O_NONBLOCK);
if (connect(fd, (void *) &sun, sizeof(sun)) < 0) {
if (errno == ECONNREFUSED) {
(void) unlink(path);
}
}
(void) close(fd);
return (0);
}
int
nni_plat_ipc_ep_listen(nni_plat_ipc_ep *ep)
{
const char * path;
nni_posix_epdesc * ed = (void *) ep;
struct sockaddr_un sun;
int rv;
path = nni_posix_epdesc_url(ed);
path += strlen("ipc://");
if ((rv = nni_plat_ipc_remove_stale(path)) != 0) {
return (rv);
}
return (nni_posix_epdesc_listen(ed));
}
void
nni_plat_ipc_ep_connect(nni_plat_ipc_ep *ep, nni_aio *aio)
{
nni_posix_epdesc_connect((void *) ep, aio);
}
void
nni_plat_ipc_ep_accept(nni_plat_ipc_ep *ep, nni_aio *aio)
{
nni_posix_epdesc_accept((void *) ep, aio);
}
void
nni_plat_ipc_pipe_fini(nni_plat_ipc_pipe *p)
{
nni_posix_pipedesc_fini((void *) p);
}
void
nni_plat_ipc_pipe_close(nni_plat_ipc_pipe *p)
{
nni_posix_pipedesc_close((void *) p);
}
void
nni_plat_ipc_pipe_send(nni_plat_ipc_pipe *p, nni_aio *aio)
{
nni_posix_pipedesc_send((void *) p, aio);
}
void
nni_plat_ipc_pipe_recv(nni_plat_ipc_pipe *p, nni_aio *aio)
{
nni_posix_pipedesc_recv((void *) p, aio);
}
#else
// Suppress empty symbols warnings in ranlib.
int nni_posix_ipc_not_used = 0;
#endif // PLATFORM_POSIX_IPC
|