1 /*
2 * Copyright 2016 Jakub Klama <jceel@FreeBSD.org>
3 * All rights reserved
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted providing that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
15 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
16 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
18 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
22 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
23 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
24 * POSSIBILITY OF SUCH DAMAGE.
25 *
26 */
27
28 #include <stdlib.h>
29 #include <errno.h>
30 #include <string.h>
31 #include <unistd.h>
32 #include <pthread.h>
33 #include <assert.h>
34 #include <sys/types.h>
35 #ifdef __APPLE__
36 # include "../apple_endian.h"
37 #else
38 # include <sys/endian.h>
39 #endif
40 #include <sys/socket.h>
41 #include <sys/event.h>
42 #include <sys/uio.h>
43 #include <netdb.h>
44 #include "../lib9p.h"
45 #include "../lib9p_impl.h"
46 #include "../log.h"
47 #include "socket.h"
48
49 struct l9p_socket_softc
50 {
51 struct l9p_connection *ls_conn;
52 struct sockaddr ls_sockaddr;
53 socklen_t ls_socklen;
54 pthread_t ls_thread;
55 int ls_fd;
56 };
57
58 static int l9p_socket_readmsg(struct l9p_socket_softc *, void **, size_t *);
59 static int l9p_socket_get_response_buffer(struct l9p_request *,
60 struct iovec *, size_t *, void *);
61 static int l9p_socket_send_response(struct l9p_request *, const struct iovec *,
62 const size_t, const size_t, void *);
63 static void l9p_socket_drop_response(struct l9p_request *, const struct iovec *,
64 size_t, void *);
65 static void *l9p_socket_thread(void *);
66 static ssize_t xread(int, void *, size_t);
67 static ssize_t xwrite(int, void *, size_t);
68
69 int
l9p_start_server(struct l9p_server * server,const char * host,const char * port)70 l9p_start_server(struct l9p_server *server, const char *host, const char *port)
71 {
72 struct addrinfo *res, *res0, hints;
73 struct kevent kev[2];
74 struct kevent event[2];
75 int err, kq, i, val, evs, nsockets = 0;
76 int sockets[2];
77
78 memset(&hints, 0, sizeof(hints));
79 hints.ai_family = PF_UNSPEC;
80 hints.ai_socktype = SOCK_STREAM;
81 err = getaddrinfo(host, port, &hints, &res0);
82
83 if (err)
84 return (-1);
85
86 for (res = res0; res; res = res->ai_next) {
87 int s = socket(res->ai_family, res->ai_socktype,
88 res->ai_protocol);
89
90 val = 1;
91 setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));
92
93 if (s < 0)
94 continue;
95
96 if (bind(s, res->ai_addr, res->ai_addrlen) < 0) {
97 close(s);
98 continue;
99 }
100
101 sockets[nsockets] = s;
102 EV_SET(&kev[nsockets++], s, EVFILT_READ, EV_ADD | EV_ENABLE, 0,
103 0, 0);
104 listen(s, 10);
105 }
106
107 if (nsockets < 1) {
108 L9P_LOG(L9P_ERROR, "bind(): %s", strerror(errno));
109 return(-1);
110 }
111
112 kq = kqueue();
113
114 if (kevent(kq, kev, nsockets, NULL, 0, NULL) < 0) {
115 L9P_LOG(L9P_ERROR, "kevent(): %s", strerror(errno));
116 return (-1);
117 }
118
119 for (;;) {
120 evs = kevent(kq, NULL, 0, event, nsockets, NULL);
121 if (evs < 0) {
122 if (errno == EINTR)
123 continue;
124
125 L9P_LOG(L9P_ERROR, "kevent(): %s", strerror(errno));
126 return (-1);
127 }
128
129 for (i = 0; i < evs; i++) {
130 struct sockaddr client_addr;
131 socklen_t client_addr_len = sizeof(client_addr);
132 int news = accept((int)event[i].ident, &client_addr,
133 &client_addr_len);
134
135 if (news < 0) {
136 L9P_LOG(L9P_WARNING, "accept(): %s",
137 strerror(errno));
138 continue;
139 }
140
141 l9p_socket_accept(server, news, &client_addr,
142 client_addr_len);
143 }
144 }
145
146 }
147
148 void
l9p_socket_accept(struct l9p_server * server,int conn_fd,struct sockaddr * client_addr,socklen_t client_addr_len)149 l9p_socket_accept(struct l9p_server *server, int conn_fd,
150 struct sockaddr *client_addr, socklen_t client_addr_len)
151 {
152 struct l9p_socket_softc *sc;
153 struct l9p_connection *conn;
154 char host[NI_MAXHOST + 1];
155 char serv[NI_MAXSERV + 1];
156 int err;
157
158 err = getnameinfo(client_addr, client_addr_len, host, NI_MAXHOST, serv,
159 NI_MAXSERV, NI_NUMERICHOST | NI_NUMERICSERV);
160
161 if (err != 0) {
162 L9P_LOG(L9P_WARNING, "cannot look up client name: %s",
163 gai_strerror(err));
164 } else {
165 L9P_LOG(L9P_INFO, "new connection from %s:%s", host, serv);
166 }
167
168 if (l9p_connection_init(server, &conn) != 0) {
169 L9P_LOG(L9P_ERROR, "cannot create new connection");
170 return;
171 }
172
173 sc = l9p_calloc(1, sizeof(*sc));
174 sc->ls_conn = conn;
175 sc->ls_fd = conn_fd;
176
177 /*
178 * Fill in transport handler functions and aux argument.
179 */
180 conn->lc_lt.lt_aux = sc;
181 conn->lc_lt.lt_get_response_buffer = l9p_socket_get_response_buffer;
182 conn->lc_lt.lt_send_response = l9p_socket_send_response;
183 conn->lc_lt.lt_drop_response = l9p_socket_drop_response;
184
185 err = pthread_create(&sc->ls_thread, NULL, l9p_socket_thread, sc);
186 if (err) {
187 L9P_LOG(L9P_ERROR,
188 "pthread_create (for connection from %s:%s): error %s",
189 host, serv, strerror(err));
190 l9p_connection_close(sc->ls_conn);
191 free(sc);
192 }
193 }
194
195 static void *
l9p_socket_thread(void * arg)196 l9p_socket_thread(void *arg)
197 {
198 struct l9p_socket_softc *sc = (struct l9p_socket_softc *)arg;
199 struct iovec iov;
200 void *buf;
201 size_t length;
202
203 for (;;) {
204 if (l9p_socket_readmsg(sc, &buf, &length) != 0)
205 break;
206
207 iov.iov_base = buf;
208 iov.iov_len = length;
209 l9p_connection_recv(sc->ls_conn, &iov, 1, NULL);
210 free(buf);
211 }
212
213 L9P_LOG(L9P_INFO, "connection closed");
214 l9p_connection_close(sc->ls_conn);
215 free(sc);
216 return (NULL);
217 }
218
219 static int
l9p_socket_readmsg(struct l9p_socket_softc * sc,void ** buf,size_t * size)220 l9p_socket_readmsg(struct l9p_socket_softc *sc, void **buf, size_t *size)
221 {
222 uint32_t msize;
223 size_t toread;
224 ssize_t ret;
225 void *buffer;
226 int fd = sc->ls_fd;
227
228 assert(fd > 0);
229
230 buffer = l9p_malloc(sizeof(uint32_t));
231
232 ret = xread(fd, buffer, sizeof(uint32_t));
233 if (ret < 0) {
234 L9P_LOG(L9P_ERROR, "read(): %s", strerror(errno));
235 return (-1);
236 }
237
238 if (ret != sizeof(uint32_t)) {
239 if (ret == 0)
240 L9P_LOG(L9P_DEBUG, "%p: EOF", (void *)sc->ls_conn);
241 else
242 L9P_LOG(L9P_ERROR,
243 "short read: %zd bytes of %zd expected",
244 ret, sizeof(uint32_t));
245 return (-1);
246 }
247
248 msize = le32toh(*(uint32_t *)buffer);
249 toread = msize - sizeof(uint32_t);
250 buffer = l9p_realloc(buffer, msize);
251
252 ret = xread(fd, (char *)buffer + sizeof(uint32_t), toread);
253 if (ret < 0) {
254 L9P_LOG(L9P_ERROR, "read(): %s", strerror(errno));
255 return (-1);
256 }
257
258 if (ret != (ssize_t)toread) {
259 L9P_LOG(L9P_ERROR, "short read: %zd bytes of %zd expected",
260 ret, toread);
261 return (-1);
262 }
263
264 *size = msize;
265 *buf = buffer;
266 L9P_LOG(L9P_INFO, "%p: read complete message, buf=%p size=%d",
267 (void *)sc->ls_conn, buffer, msize);
268
269 return (0);
270 }
271
272 static int
l9p_socket_get_response_buffer(struct l9p_request * req,struct iovec * iov,size_t * niovp,void * arg __unused)273 l9p_socket_get_response_buffer(struct l9p_request *req, struct iovec *iov,
274 size_t *niovp, void *arg __unused)
275 {
276 size_t size = req->lr_conn->lc_msize;
277 void *buf;
278
279 buf = l9p_malloc(size);
280 iov[0].iov_base = buf;
281 iov[0].iov_len = size;
282
283 *niovp = 1;
284 return (0);
285 }
286
287 static int
l9p_socket_send_response(struct l9p_request * req __unused,const struct iovec * iov,const size_t niov __unused,const size_t iolen,void * arg)288 l9p_socket_send_response(struct l9p_request *req __unused,
289 const struct iovec *iov, const size_t niov __unused, const size_t iolen,
290 void *arg)
291 {
292 struct l9p_socket_softc *sc = (struct l9p_socket_softc *)arg;
293
294 assert(sc->ls_fd >= 0);
295
296 L9P_LOG(L9P_DEBUG, "%p: sending reply, buf=%p, size=%d", arg,
297 iov[0].iov_base, iolen);
298
299 if (xwrite(sc->ls_fd, iov[0].iov_base, iolen) != (int)iolen) {
300 L9P_LOG(L9P_ERROR, "short write: %s", strerror(errno));
301 return (-1);
302 }
303
304 free(iov[0].iov_base);
305 return (0);
306 }
307
308 static void
l9p_socket_drop_response(struct l9p_request * req __unused,const struct iovec * iov,size_t niov __unused,void * arg __unused)309 l9p_socket_drop_response(struct l9p_request *req __unused,
310 const struct iovec *iov, size_t niov __unused, void *arg __unused)
311 {
312
313 L9P_LOG(L9P_DEBUG, "%p: drop buf=%p", arg, iov[0].iov_base);
314 free(iov[0].iov_base);
315 }
316
317 static ssize_t
xread(int fd,void * buf,size_t count)318 xread(int fd, void *buf, size_t count)
319 {
320 size_t done = 0;
321 ssize_t ret;
322
323 while (done < count) {
324 ret = read(fd, (char *)buf + done, count - done);
325 if (ret < 0) {
326 if (errno == EINTR)
327 continue;
328
329 return (-1);
330 }
331
332 if (ret == 0)
333 return ((ssize_t)done);
334
335 done += (size_t)ret;
336 }
337
338 return ((ssize_t)done);
339 }
340
341 static ssize_t
xwrite(int fd,void * buf,size_t count)342 xwrite(int fd, void *buf, size_t count)
343 {
344 size_t done = 0;
345 ssize_t ret;
346
347 while (done < count) {
348 ret = write(fd, (char *)buf + done, count - done);
349 if (ret < 0) {
350 if (errno == EINTR)
351 continue;
352
353 return (-1);
354 }
355
356 if (ret == 0)
357 return ((ssize_t)done);
358
359 done += (size_t)ret;
360 }
361
362 return ((ssize_t)done);
363 }
364