blob: d1b59eea8fea421843e6b062d2d853faa0fa5e22 [file] [log] [blame]
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001/*
2 *------------------------------------------------------------------
3 * Copyright (c) 2017 Cisco and/or its affiliates.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at:
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 *------------------------------------------------------------------
16 */
17
18#include <stdint.h>
19#include <net/if.h>
20#include <sys/types.h>
21#include <fcntl.h>
22#include <sys/ioctl.h>
23#include <sys/socket.h>
24#include <sys/un.h>
25#include <sys/uio.h>
26#include <sys/mman.h>
27#include <sys/prctl.h>
28#include <inttypes.h>
29#include <string.h>
30#include <stdio.h>
31#include <netdb.h>
32#include <linux/ip.h>
33#include <linux/icmp.h>
34#include <arpa/inet.h>
35#include <stdlib.h>
36#include <netinet/if_ether.h>
37#include <net/if_arp.h>
38#include <asm/byteorder.h>
39#include <byteswap.h>
40#include <string.h>
41#include <errno.h>
42#include <sys/stat.h>
43#include <sys/eventfd.h>
44#include <sys/timerfd.h>
45#include <sys/epoll.h>
46#include <signal.h>
47
48/* memif protocol msg, ring and descriptor definitions */
49#include <memif.h>
50/* memif api */
51#include <libmemif.h>
52/* socket messaging functions */
53#include <socket.h>
54/* private structs and functions */
55#include <memif_private.h>
56
57#define ERRLIST_LEN 36
58#define MAX_ERRBUF_LEN 256
59
60#if __x86_x64__
61#define MEMIF_MEMORY_BARRIER() __builtin_ia32_sfence ()
62#else
63#define MEMIF_MEORY_BARRIER() __sync_synchronize ()
64#endif /* __x86_x64__ */
65
66libmemif_main_t libmemif_main;
67int memif_epfd;
68
69static char memif_buf[MAX_ERRBUF_LEN];
70
71const char *memif_errlist[ERRLIST_LEN] = { /* MEMIF_ERR_SUCCESS */
72 "Success.",
73 /* MEMIF_ERR_SYSCALL */
74 "Unspecified syscall error (build with -DMEMIF_DBG or make debug).",
75 /* MEMIF_ERR_ACCES */
76 "Permission to resoure denied.",
77 /* MEMIF_ERR_NO_FILE */
78 "Socket file does not exist",
79 /* MEMIF_ERR_FILE_LIMIT */
80 "System limit on total numer of open files reached.",
81 /* MEMIF_ERR_PROC_FILE_LIMIT */
82 "Per-process limit on total number of open files reached.",
83 /* MEMIF_ERR_ALREADY */
84 "Connection already requested.",
85 /* MEMIF_ERR_AGAIN */
86 "File descriptor refers to file other than socket, or operation would block.",
87 /* MEMIF_ERR_BAD_FD */
88 "Bad file descriptor.",
89 /* MEMIF_ERR_NOMEM */
90 "Out of memory.",
91 /* MEMIF_ERR_INVAL_ARG */
92 "Invalid argument.",
93 /* MEMIF_ERR_NOCONN */
94 "Memif connection handle does not point to existing conenction",
95 /* MEMIF_ERR_CONN */
96 "Memif connection handle points to existing connection",
97 /* MEMIF_ERR_CB_FDUPDATE */
98 "Callback memif_control_fd_update_t returned error",
99 /* MEMIF_ERR_FILE_NOT_SOCK */
100 "File specified by socket filename exists and is not socket.",
101 /* MEMIF_ERR_NO_SHMFD */
102 "Missing shared memory file descriptor. (internal error)",
103 /* MEMIF_ERR_COOKIE */
104 "Invalid cookie on ring. (internal error)",
105 /* MEMIF_ERR_NOBUF_RING */
106 "Ring buffer full.",
107 /* MEMIF_ERR_NOBUF */
108 "Not enough memif buffers. There are unreceived data in shared memory.",
109 /* MEMIF_ERR_NOBUF_DET */
110 "Not enough space for memif details in suplied buffer. String data might be malformed.",
111 /* MEMIF_ERR_INT_WRITE */
112 "Send interrupt error.",
113 /* MEMIF_ERR_MFMSG */
114 "Malformed message received on control channel.",
115 /* MEMIF_ERR_QID */
116 "Invalid queue id",
117 /* MEMIF_ERR_PROTO */
118 "Incompatible memory interface protocol version.",
119 /* MEMIF_ERR_ID */
120 "Unmatched interface id.",
121 /* MEMIF_ERR_ACCSLAVE */
122 "Slave cannot accept connection reqest.",
123 /* MEMIF_ERR_ALRCONN */
124 "Interface is already connected.",
125 /* MEMIF_ERR_MODE */
126 "Mode mismatch.",
127 /* MEMIF_ERR_SECRET */
128 "Secret mismatch.",
129 /* MEMIF_ERR_NOSECRET */
130 "Secret required.",
131 /* MEMIF_ERR_MAXREG */
132 "Limit on total number of regions reached.",
133 /* MEMIF_ERR_MAXRING */
134 "Limit on total number of ring reached.",
135 /* MEMIF_ERR_NO_INTFD */
136 "Missing interrupt file descriptor. (internal error)",
137 /* MEMIF_ERR_DISCONNECT */
138 "Interface received disconnect request.",
139 /* MEMIF_ERR_DISCONNECTED */
140 "Interface is disconnected.",
141 /* MEMIF_ERR_UNKNOWN_MSG */
142 "Unknown message type received on control channel. (internal error)"
143};
144
145#define MEMIF_ERR_UNDEFINED "undefined error"
146
147char *
148memif_strerror (int err_code)
149{
150 if (err_code >= ERRLIST_LEN)
151 {
152 strncpy (memif_buf, MEMIF_ERR_UNDEFINED, strlen (MEMIF_ERR_UNDEFINED));
153 memif_buf[strlen (MEMIF_ERR_UNDEFINED)] = '\0';
154 }
155 else
156 {
157 strncpy (memif_buf, memif_errlist[err_code],
158 strlen (memif_errlist[err_code]));
159 memif_buf[strlen (memif_errlist[err_code])] = '\0';
160 }
161 return memif_buf;
162}
163
164#define DBG_TX_BUF (0)
165#define DBG_RX_BUF (1)
166
167#ifdef MEMIF_DBG_SHM
168static void
169print_bytes (void *data, uint16_t len, uint8_t q)
170{
171 if (q == DBG_TX_BUF)
172 printf ("\nTX:\n\t");
173 else
174 printf ("\nRX:\n\t");
175 int i;
176 for (i = 0; i < len; i++)
177 {
178 if (i % 8 == 0)
179 printf ("\n%d:\t", i);
180 printf ("%02X ", ((uint8_t *) (data))[i]);
181 }
182 printf ("\n\n");
183}
184#endif /* MEMIF_DBG */
185
186int
187memif_syscall_error_handler (int err_code)
188{
189 DBG_UNIX ("%s", strerror (err_code));
190
191 if (err_code == 0)
192 return MEMIF_ERR_SUCCESS;
193 if (err_code == EACCES)
194 return MEMIF_ERR_ACCES;
195 if (err_code == ENFILE)
196 return MEMIF_ERR_FILE_LIMIT;
197 if (err_code == EMFILE)
198 return MEMIF_ERR_PROC_FILE_LIMIT;
199 if (err_code == ENOMEM)
200 return MEMIF_ERR_NOMEM;
201/* connection refused if master dows not exist
202 this error would spam the user until master was created */
203 if (err_code == ECONNREFUSED)
204 return MEMIF_ERR_SUCCESS;
205 if (err_code == EALREADY)
206 return MEMIF_ERR_ALREADY;
207 if (err_code == EAGAIN)
208 return MEMIF_ERR_AGAIN;
209 if (err_code == EBADF)
210 return MEMIF_ERR_BAD_FD;
211 if (err_code == ENOENT)
212 return MEMIF_ERR_NO_FILE;
213
214 /* other syscall errors */
215 return MEMIF_ERR_SYSCALL;
216}
217
218static int
219memif_add_epoll_fd (int fd, uint32_t events)
220{
221 if (fd < 0)
222 {
223 DBG ("invalid fd %d", fd);
224 return -1;
225 }
226 struct epoll_event evt;
227 memset (&evt, 0, sizeof (evt));
228 evt.events = events;
229 evt.data.fd = fd;
230 if (epoll_ctl (memif_epfd, EPOLL_CTL_ADD, fd, &evt) < 0)
231 {
232 DBG ("epoll_ctl: %s fd %d", strerror (errno), fd);
233 return -1;
234 }
235 DBG ("fd %d added to epoll", fd);
236 return 0;
237}
238
239static int
240memif_mod_epoll_fd (int fd, uint32_t events)
241{
242 if (fd < 0)
243 {
244 DBG ("invalid fd %d", fd);
245 return -1;
246 }
247 struct epoll_event evt;
248 memset (&evt, 0, sizeof (evt));
249 evt.events = events;
250 evt.data.fd = fd;
251 if (epoll_ctl (memif_epfd, EPOLL_CTL_MOD, fd, &evt) < 0)
252 {
253 DBG ("epoll_ctl: %s fd %d", strerror (errno), fd);
254 return -1;
255 }
256 DBG ("fd %d moddified on epoll", fd);
257 return 0;
258}
259
260static int
261memif_del_epoll_fd (int fd)
262{
263 if (fd < 0)
264 {
265 DBG ("invalid fd %d", fd);
266 return -1;
267 }
268 struct epoll_event evt;
269 memset (&evt, 0, sizeof (evt));
270 if (epoll_ctl (memif_epfd, EPOLL_CTL_DEL, fd, &evt) < 0)
271 {
272 DBG ("epoll_ctl: %s fd %d", strerror (errno), fd);
273 return -1;
274 }
275 DBG ("fd %d removed from epoll", fd);
276 return 0;
277}
278
279int
280memif_control_fd_update (int fd, uint8_t events)
281{
282 if (events & MEMIF_FD_EVENT_DEL)
283 return memif_del_epoll_fd (fd);
284
285 uint32_t evt = 0;
286 if (events & MEMIF_FD_EVENT_READ)
287 evt |= EPOLLIN;
288 if (events & MEMIF_FD_EVENT_WRITE)
289 evt |= EPOLLOUT;
290
291 if (events & MEMIF_FD_EVENT_MOD)
292 return memif_mod_epoll_fd (fd, evt);
293
294 return memif_add_epoll_fd (fd, evt);
295}
296
297int
298add_list_elt (memif_list_elt_t * e, memif_list_elt_t ** list, uint16_t * len)
299{
300 libmemif_main_t *lm = &libmemif_main;
301
302 int i;
303 for (i = 0; i < *len; i++)
304 {
305 if ((*list)[i].data_struct == NULL)
306 {
307 (*list)[i].key = e->key;
308 (*list)[i].data_struct = e->data_struct;
309 return i;
310 }
311 }
312 memif_list_elt_t *tmp;
313 tmp = realloc (*list, sizeof (memif_list_elt_t) * *len * 2);
314 if (tmp == NULL)
315 return -1;
316
317 for (i = *len; i < *len * 2; i++)
318 {
319 tmp[i].key = -1;
320 tmp[i].data_struct = NULL;
321 }
322
323 tmp[*len].key = e->key;
324 tmp[*len].data_struct = e->data_struct;
325 i = *len;
326 *len = *len * 2;
327 *list = tmp;
328
329 return i;
330}
331
332int
333get_list_elt (memif_list_elt_t ** e, memif_list_elt_t * list, uint16_t len,
334 int key)
335{
336 if (key == -1)
337 {
338 *e = NULL;
339 return -1;
340 }
341 int i;
342 for (i = 0; i < len; i++)
343 {
344 if (list[i].key == key)
345 {
346 *e = &list[i];
347 return 0;
348 }
349 }
350 *e = NULL;
351 return -1;
352}
353
354/* does not free memory, only marks element as free */
355int
356free_list_elt (memif_list_elt_t * list, uint16_t len, int key)
357{
358 int i;
359 for (i = 0; i < len; i++)
360 {
361 if (list[i].key == key)
362 {
363 list[i].key = -1;
364 list[i].data_struct = NULL;
365 return 0;
366 }
367 }
368
369 return -1;
370}
371
372int
373free_list_elt_ctx (memif_list_elt_t * list, uint16_t len,
374 memif_connection_t * ctx)
375{
376 int i;
377 for (i = 0; i < len; i++)
378 {
379 if (list[i].key == -1)
380 {
381 if (list[i].data_struct == ctx)
382 {
383 list[i].data_struct = NULL;
384 return 0;
385 }
386 }
387 }
388
389 return -1;
390}
391
392static void
393memif_control_fd_update_register (memif_control_fd_update_t * cb)
394{
395 libmemif_main_t *lm = &libmemif_main;
396 lm->control_fd_update = cb;
397}
398
399int
400memif_init (memif_control_fd_update_t * on_control_fd_update, char *app_name)
401{
402 int err = MEMIF_ERR_SUCCESS; /* 0 */
403 libmemif_main_t *lm = &libmemif_main;
404
405 if (app_name)
406 {
407 lm->app_name = malloc (strlen (app_name) + sizeof (char));
408 memset (lm->app_name, 0, strlen (app_name) + sizeof (char));
409 strncpy ((char *) lm->app_name, app_name, strlen (app_name));
410 }
411 else
412 {
413 lm->app_name = malloc (strlen (MEMIF_DEFAULT_APP_NAME) + sizeof (char));
414 memset (lm->app_name, 0, strlen (app_name) + sizeof (char));
415 strncpy ((char *) lm->app_name, MEMIF_DEFAULT_APP_NAME,
416 strlen (MEMIF_DEFAULT_APP_NAME));
417 }
418
419 /* register control fd update callback */
420 if (on_control_fd_update != NULL)
421 memif_control_fd_update_register (on_control_fd_update);
422 else
423 {
424 memif_epfd = epoll_create (1);
425 memif_control_fd_update_register (memif_control_fd_update);
426 DBG ("libmemif event polling initialized");
427 }
428
429 memset (&lm->ms, 0, sizeof (memif_socket_t));
430
431 lm->control_list_len = 2;
432 lm->interrupt_list_len = 2;
433 lm->listener_list_len = 1;
434 lm->pending_list_len = 1;
435
436 lm->control_list =
437 malloc (sizeof (memif_list_elt_t) * lm->control_list_len);
438 lm->interrupt_list =
439 malloc (sizeof (memif_list_elt_t) * lm->interrupt_list_len);
440 lm->listener_list =
441 malloc (sizeof (memif_list_elt_t) * lm->listener_list_len);
442 lm->pending_list =
443 malloc (sizeof (memif_list_elt_t) * lm->pending_list_len);
444
445 int i;
446 for (i = 0; i < lm->control_list_len; i++)
447 {
448 lm->control_list[i].key = -1;
449 lm->control_list[i].data_struct = NULL;
450 }
451 for (i = 0; i < lm->interrupt_list_len; i++)
452 {
453 lm->interrupt_list[i].key = -1;
454 lm->interrupt_list[i].data_struct = NULL;
455 }
456 for (i = 0; i < lm->listener_list_len; i++)
457 {
458 lm->listener_list[i].key = -1;
459 lm->listener_list[i].data_struct = NULL;
460 }
461 for (i = 0; i < lm->pending_list_len; i++)
462 {
463 lm->pending_list[i].key = -1;
464 lm->pending_list[i].data_struct = NULL;
465 }
466
467 lm->disconn_slaves = 0;
468
469 lm->timerfd = timerfd_create (CLOCK_REALTIME, TFD_NONBLOCK);
470 if (lm->timerfd < 0)
471 {
472 err = errno;
473 DBG ("timerfd: %s", strerror (err));
474 return memif_syscall_error_handler (err);
475 }
476
477 lm->arm.it_value.tv_sec = 2;
478 lm->arm.it_value.tv_nsec = 0;
479 lm->arm.it_interval.tv_sec = 2;
480 lm->arm.it_interval.tv_nsec = 0;
481 memset (&lm->disarm, 0, sizeof (lm->disarm));
482
483 if (lm->control_fd_update (lm->timerfd, MEMIF_FD_EVENT_READ) < 0)
484 {
485 DBG ("callback type memif_control_fd_update_t error!");
486 return MEMIF_ERR_CB_FDUPDATE;
487 }
488
489 return 0;
490}
491
492static inline memif_ring_t *
493memif_get_ring (memif_connection_t * conn, memif_ring_type_t type,
494 uint16_t ring_num)
495{
496 if (&conn->regions[0] == NULL)
497 return NULL;
498 void *p = conn->regions[0].shm;
499 int ring_size =
500 sizeof (memif_ring_t) +
501 sizeof (memif_desc_t) * (1 << conn->run_args.log2_ring_size);
502 p += (ring_num + type * conn->run_args.num_s2m_rings) * ring_size;
503
504 return (memif_ring_t *) p;
505}
506
507int
508memif_set_rx_mode (memif_conn_handle_t c, memif_rx_mode_t rx_mode,
509 uint16_t qid)
510{
511 memif_connection_t *conn = (memif_connection_t *) c;
512 if (conn == NULL)
513 return MEMIF_ERR_NOCONN;
514 uint8_t num =
515 (conn->args.is_master) ? conn->run_args.num_s2m_rings : conn->run_args.
516 num_m2s_rings;
517 if (qid >= num)
518 return MEMIF_ERR_QID;
519
520 conn->rx_queues[qid].ring->flags = rx_mode;
521 DBG ("rx_mode flag: %u", conn->rx_queues[qid].ring->flags);
522 return MEMIF_ERR_SUCCESS;
523}
524
525int
526memif_create (memif_conn_handle_t * c, memif_conn_args_t * args,
527 memif_connection_update_t * on_connect,
528 memif_connection_update_t * on_disconnect,
529 memif_interrupt_t * on_interrupt, void *private_ctx)
530{
531 int err, i, index, sockfd = -1;
532 memif_list_elt_t list_elt;
533 memif_connection_t *conn = (memif_connection_t *) * c;
534 if (conn != NULL)
535 {
536 DBG ("This handle already points to existing memif.");
537 return MEMIF_ERR_CONN;
538 }
539 conn = (memif_connection_t *) malloc (sizeof (memif_connection_t));
540 if (conn == NULL)
541 {
542 err = memif_syscall_error_handler (errno);
543 goto error;
544 }
545 memset (conn, 0, sizeof (memif_connection_t));
546
547 libmemif_main_t *lm = &libmemif_main;
548
549 conn->args.interface_id = args->interface_id;
550
551 if (args->log2_ring_size == 0)
552 args->log2_ring_size = MEMIF_DEFAULT_LOG2_RING_SIZE;
553 if (args->buffer_size == 0)
554 args->buffer_size = MEMIF_DEFAULT_BUFFER_SIZE;
555 if (args->num_s2m_rings == 0)
556 args->num_s2m_rings = MEMIF_DEFAULT_TX_QUEUES;
557 if (args->num_m2s_rings == 0)
558 args->num_m2s_rings = MEMIF_DEFAULT_RX_QUEUES;
559
560 conn->args.num_s2m_rings = args->num_s2m_rings;
561 conn->args.num_m2s_rings = args->num_m2s_rings;
562 conn->args.buffer_size = args->buffer_size;
563 conn->args.log2_ring_size = args->log2_ring_size;
564 conn->args.is_master = args->is_master;
565 conn->args.mode = args->mode;
566 conn->msg_queue = NULL;
567 conn->regions = NULL;
568 conn->tx_queues = NULL;
569 conn->rx_queues = NULL;
570 conn->fd = -1;
571 conn->on_connect = on_connect;
572 conn->on_disconnect = on_disconnect;
573 conn->on_interrupt = on_interrupt;
574 conn->private_ctx = private_ctx;
575 memset (&conn->run_args, 0, sizeof (memif_conn_run_args_t));
576
577 uint8_t l = strlen ((char *) args->interface_name);
578 strncpy ((char *) conn->args.interface_name, (char *) args->interface_name,
579 l);
580
581 l = strlen ((char *) args->instance_name);
582 strncpy ((char *) conn->args.instance_name, (char *) args->instance_name,
583 l);
584
585 /* allocate and initialize socket_filename so it can be copyed to sun_path
586 without memory leaks */
587 conn->args.socket_filename = malloc (sizeof (char *) * 108);
588 memset (conn->args.socket_filename, 0, 108 * sizeof (char *));
589
590 if (args->socket_filename)
591 {
592 if (conn->args.socket_filename == NULL)
593 {
594 err = memif_syscall_error_handler (errno);
595 goto error;
596 }
597 strncpy ((char *) conn->args.socket_filename,
598 (char *) args->socket_filename,
599 strlen ((char *) args->socket_filename));
600 }
601 else
602 {
603 uint16_t sdl = strlen (MEMIF_DEFAULT_SOCKET_DIR);
604 uint16_t sfl = strlen (MEMIF_DEFAULT_SOCKET_FILENAME);
605 if (conn->args.socket_filename == NULL)
606 {
607 err = memif_syscall_error_handler (errno);
608 goto error;
609 }
610 strncpy ((char *) conn->args.socket_filename,
611 MEMIF_DEFAULT_SOCKET_DIR, sdl);
612 conn->args.socket_filename[sdl] = '/';
613 strncpy ((char *) (conn->args.socket_filename + 1 + sdl),
614 MEMIF_DEFAULT_SOCKET_FILENAME, sfl);
615 }
616
617 if (args->secret)
618 {
619 l = strlen ((char *) args->secret);
620 strncpy ((char *) conn->args.secret, (char *) args->secret, l);
621 }
622
623 if (conn->args.is_master)
624 {
625 conn->run_args.buffer_size = conn->args.buffer_size;
626 memif_socket_t *ms;
627 memif_list_elt_t elt;
628 for (i = 0; i < lm->listener_list_len; i++)
629 {
630 if ((ms =
631 (memif_socket_t *) lm->listener_list[i].data_struct) != NULL)
632 {
633 if (strncmp
634 ((char *) ms->filename, (char *) conn->args.socket_filename,
635 strlen ((char *) ms->filename)) == 0)
636 {
637 /* add interface to listener socket */
638 elt.key = conn->args.interface_id;
639 *c = elt.data_struct = conn;
640 add_list_elt (&elt, &ms->interface_list,
641 &ms->interface_list_len);
642 ms->use_count++;
643 conn->listener_fd = ms->fd;
644 break;
645 }
646 }
647 else
648 {
649 struct stat file_stat;
650 if (stat ((char *) conn->args.socket_filename, &file_stat) == 0)
651 {
652 if (S_ISSOCK (file_stat.st_mode))
653 unlink ((char *) conn->args.socket_filename);
654 else
655 return memif_syscall_error_handler (errno);
656 }
657 DBG ("creating socket file");
658 ms = malloc (sizeof (memif_socket_t));
659 ms->filename = malloc (strlen ((char *) conn->args.
660 socket_filename) +
661 sizeof (char));
662 memset (ms->filename, 0,
663 strlen ((char *) conn->args.socket_filename) +
664 sizeof (char));
665 strncpy ((char *) ms->filename,
666 (char *) conn->args.socket_filename,
667 strlen ((char *) conn->args.socket_filename));
668 ms->interface_list_len = 1;
669 ms->interface_list =
670 malloc (sizeof (memif_list_elt_t) * ms->interface_list_len);
671 ms->interface_list[0].key = -1;
672 ms->interface_list[0].data_struct = NULL;
673 struct sockaddr_un un = { 0 };
674 int on = 1;
675
676 ms->fd = socket (AF_UNIX, SOCK_SEQPACKET, 0);
677 if (ms->fd < 0)
678 {
679 err = memif_syscall_error_handler (errno);
680 goto error;
681 }
682 DBG ("socket %d created", ms->fd);
683 un.sun_family = AF_UNIX;
684 strncpy ((char *) un.sun_path, (char *) ms->filename,
685 sizeof (un.sun_path) - 1);
686 DBG ("sockopt");
687 if (setsockopt
688 (ms->fd, SOL_SOCKET, SO_PASSCRED, &on, sizeof (on)) < 0)
689 {
690 err = memif_syscall_error_handler (errno);
691 goto error;
692 }
693 DBG ("bind");
694 if (bind (ms->fd, (struct sockaddr *) &un, sizeof (un)) < 0)
695 {
696 err = memif_syscall_error_handler (errno);
697 goto error;
698 }
699 DBG ("listen");
700 if (listen (ms->fd, 1) < 0)
701 {
702 err = memif_syscall_error_handler (errno);
703 goto error;
704 }
705 DBG ("stat");
706 if (stat ((char *) ms->filename, &file_stat) < 0)
707 {
708 err = memif_syscall_error_handler (errno);
709 goto error;
710 }
711
712 /* add interface to listener socket */
713 elt.key = conn->args.interface_id;
714 *c = elt.data_struct = conn;
715 add_list_elt (&elt, &ms->interface_list,
716 &ms->interface_list_len);
717 ms->use_count = 1;
718 conn->listener_fd = ms->fd;
719
720 /* add listener socket to libmemif main */
721 elt.key = ms->fd;
722 elt.data_struct = ms;
723 add_list_elt (&elt, &lm->listener_list, &lm->listener_list_len);
724 lm->control_fd_update (ms->fd, MEMIF_FD_EVENT_READ);
725 break;
726 }
727 }
728 }
729 else
730 {
731 if (lm->disconn_slaves == 0)
732 {
733 if (timerfd_settime (lm->timerfd, 0, &lm->arm, NULL) < 0)
734 {
735 err = memif_syscall_error_handler (errno);
736 goto error;
737 }
738 }
739
740 lm->disconn_slaves++;
741
742 list_elt.key = -1;
743 *c = list_elt.data_struct = conn;
744 if ((index =
745 add_list_elt (&list_elt, &lm->control_list,
746 &lm->control_list_len)) < 0)
747 {
748 err = MEMIF_ERR_NOMEM;
749 goto error;
750 }
751 }
752
753 conn->index = index;
754
755 return 0;
756
757error:
758 if (sockfd > 0)
759 close (sockfd);
760 sockfd = -1;
761 if (conn->args.socket_filename)
762 free (conn->args.socket_filename);
763 if (conn != NULL)
764 free (conn);
765 *c = conn = NULL;
766 return err;
767}
768
769int
770memif_control_fd_handler (int fd, uint8_t events)
771{
772 int i, rv, sockfd = -1, err = MEMIF_ERR_SUCCESS; /* 0 */
773 uint16_t num;
774 memif_list_elt_t *e = NULL;
775 memif_connection_t *conn;
776 libmemif_main_t *lm = &libmemif_main;
777 if (fd == lm->timerfd)
778 {
779 uint64_t b;
780 ssize_t size;
781 size = read (fd, &b, sizeof (b));
782 for (i = 0; i < lm->control_list_len; i++)
783 {
784 if ((lm->control_list[i].key < 0)
785 && (lm->control_list[i].data_struct != NULL))
786 {
787 conn = lm->control_list[i].data_struct;
788 if (conn->args.is_master)
789 continue;
790
791 struct sockaddr_un sun;
792 sockfd = socket (AF_UNIX, SOCK_SEQPACKET, 0);
793 if (sockfd < 0)
794 {
795 err = memif_syscall_error_handler (errno);
796 goto error;
797 }
798
799 sun.sun_family = AF_UNIX;
800
801 strncpy (sun.sun_path, conn->args.socket_filename,
802 sizeof (sun.sun_path) - 1);
803
804 if (connect (sockfd, (struct sockaddr *) &sun,
805 sizeof (struct sockaddr_un)) == 0)
806 {
807 conn->fd = sockfd;
808 conn->read_fn = memif_conn_fd_read_ready;
809 conn->write_fn = memif_conn_fd_write_ready;
810 conn->error_fn = memif_conn_fd_error;
811
812 lm->control_list[conn->index].key = conn->fd;
813
814 lm->control_fd_update (sockfd,
815 MEMIF_FD_EVENT_READ |
816 MEMIF_FD_EVENT_WRITE);
817
818 lm->disconn_slaves--;
819 if (lm->disconn_slaves == 0)
820 {
821 if (timerfd_settime (lm->timerfd, 0, &lm->disarm, NULL)
822 < 0)
823 {
824 err = memif_syscall_error_handler (errno);
825 goto error;
826 }
827 }
828 }
829 else
830 {
831 err = memif_syscall_error_handler (errno);
832 goto error;
833 }
834 }
835 }
836 }
837 else
838 {
839 get_list_elt (&e, lm->interrupt_list, lm->interrupt_list_len, fd);
840 if (e != NULL)
841 {
842 if (((memif_connection_t *) e->data_struct)->on_interrupt != NULL)
843 {
844 num =
845 (((memif_connection_t *) e->data_struct)->args.
846 is_master) ? ((memif_connection_t *) e->data_struct)->
847 run_args.num_s2m_rings : ((memif_connection_t *) e->
848 data_struct)->run_args.
849 num_m2s_rings;
850 for (i = 0; i < num; i++)
851 {
852 if (((memif_connection_t *) e->data_struct)->rx_queues[i].
853 int_fd == fd)
854 {
855 ((memif_connection_t *) e->data_struct)->on_interrupt ((void *) e->data_struct, ((memif_connection_t *) e->data_struct)->private_ctx, i);
856 return MEMIF_ERR_SUCCESS;
857 }
858 }
859 }
860 return MEMIF_ERR_SUCCESS;
861 }
862 get_list_elt (&e, lm->listener_list, lm->listener_list_len, fd);
863 if (e != NULL)
864 {
865 memif_conn_fd_accept_ready ((memif_socket_t *) e->data_struct);
866 return MEMIF_ERR_SUCCESS;
867 }
868
869 get_list_elt (&e, lm->pending_list, lm->pending_list_len, fd);
870 if (e != NULL)
871 {
872 memif_read_ready (fd);
873 return MEMIF_ERR_SUCCESS;
874 }
875
876 get_list_elt (&e, lm->control_list, lm->control_list_len, fd);
877 if (e != NULL)
878 {
879 if (events & MEMIF_FD_EVENT_READ)
880 {
881 err =
882 ((memif_connection_t *) e->data_struct)->read_fn (e->
883 data_struct);
884 if (err != MEMIF_ERR_SUCCESS)
885 return err;
886 }
887 if (events & MEMIF_FD_EVENT_WRITE)
888 {
889 err =
890 ((memif_connection_t *) e->data_struct)->write_fn (e->
891 data_struct);
892 if (err != MEMIF_ERR_SUCCESS)
893 return err;
894 }
895 if (events & MEMIF_FD_EVENT_ERROR)
896 {
897 err =
898 ((memif_connection_t *) e->data_struct)->error_fn (e->
899 data_struct);
900 if (err != MEMIF_ERR_SUCCESS)
901 return err;
902 }
903 }
904 }
905
906 return MEMIF_ERR_SUCCESS; /* 0 */
907
908error:
909 if (sockfd > 0)
910 close (sockfd);
911 sockfd = -1;
912 return err;
913}
914
915int
916memif_poll_event (int timeout)
917{
918 libmemif_main_t *lm = &libmemif_main;
919 memif_list_elt_t *elt;
920 struct epoll_event evt, *e;
921 int en = 0, err = MEMIF_ERR_SUCCESS, i = 0; /* 0 */
922 uint16_t num;
923 uint32_t events = 0;
924 memset (&evt, 0, sizeof (evt));
925 evt.events = EPOLLIN | EPOLLOUT;
926 sigset_t sigset;
927 sigemptyset (&sigset);
928 en = epoll_pwait (memif_epfd, &evt, 1, timeout, &sigset);
929 if (en < 0)
930 {
931 DBG ("epoll_pwait: %s", strerror (errno));
932 return -1;
933 }
934 if (en > 0)
935 {
936 if (evt.events & EPOLLIN)
937 events |= MEMIF_FD_EVENT_READ;
938 if (evt.events & EPOLLOUT)
939 events |= MEMIF_FD_EVENT_WRITE;
940 if (evt.events & EPOLLERR)
941 events |= MEMIF_FD_EVENT_ERROR;
942 err = memif_control_fd_handler (evt.data.fd, events);
943 return err;
944 }
945 return 0;
946}
947
948static void
949memif_msg_queue_free (memif_msg_queue_elt_t ** e)
950{
951 if (*e == NULL)
952 return;
953 memif_msg_queue_free (&(*e)->next);
954 free (*e);
955 *e = NULL;
956 return;
957}
958
959/* send disconnect msg and close interface */
960int
961memif_disconnect_internal (memif_connection_t * c)
962{
963 if (c == NULL)
964 {
965 DBG ("no connection");
966 return MEMIF_ERR_NOCONN;
967 }
968 uint16_t num;
969 int err = MEMIF_ERR_SUCCESS, i; /* 0 */
970 memif_queue_t *mq;
971 libmemif_main_t *lm = &libmemif_main;
972 memif_list_elt_t *e;
973
974 c->on_disconnect ((void *) c, c->private_ctx);
975
976 if (c->fd > 0)
977 {
978 memif_msg_send_disconnect (c->fd, "interface deleted", 0);
979 lm->control_fd_update (c->fd, MEMIF_FD_EVENT_DEL);
980 close (c->fd);
981 }
982 get_list_elt (&e, lm->control_list, lm->control_list_len, c->fd);
983 if (e != NULL)
984 {
985 if (c->args.is_master)
986 free_list_elt (lm->control_list, lm->control_list_len, c->fd);
987 e->key = c->fd = -1;
988 }
989
990 if (c->tx_queues != NULL)
991 {
992 num =
993 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
994 num_s2m_rings;
995 for (i = 0; i < num; i++)
996 {
997 mq = &c->tx_queues[i];
998 if (mq != NULL)
999 {
1000 if (mq->int_fd > 0)
1001 close (mq->int_fd);
1002 free_list_elt (lm->interrupt_list, lm->interrupt_list_len,
1003 mq->int_fd);
1004 mq->int_fd = -1;
1005 }
1006 }
1007 free (c->tx_queues);
1008 c->tx_queues = NULL;
1009 }
1010
1011 if (c->rx_queues != NULL)
1012 {
1013 num =
1014 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1015 num_m2s_rings;
1016 for (i = 0; i < num; i++)
1017 {
1018 mq = &c->rx_queues[i];
1019 if (mq != NULL)
1020 {
1021 if (mq->int_fd > 0)
1022 {
1023 if (c->on_interrupt != NULL)
1024 lm->control_fd_update (mq->int_fd, MEMIF_FD_EVENT_DEL);
1025 close (mq->int_fd);
1026 }
1027 free_list_elt (lm->interrupt_list, lm->interrupt_list_len,
1028 mq->int_fd);
1029 mq->int_fd = -1;
1030 }
1031 }
1032 free (c->rx_queues);
1033 c->rx_queues = NULL;
1034 }
1035
1036 if (c->regions != NULL)
1037 {
1038 if (munmap (c->regions[0].shm, c->regions[0].region_size) < 0)
1039 return memif_syscall_error_handler (errno);
1040 if (c->regions[0].fd > 0)
1041 close (c->regions[0].fd);
1042 c->regions[0].fd = -1;
1043 free (c->regions);
1044 c->regions = NULL;
1045 }
1046
1047 memset (&c->run_args, 0, sizeof (memif_conn_run_args_t));
1048
1049 memif_msg_queue_free (&c->msg_queue);
1050
1051 if (!(c->args.is_master))
1052 {
1053 if (lm->disconn_slaves == 0)
1054 {
1055 if (timerfd_settime (lm->timerfd, 0, &lm->arm, NULL) < 0)
1056 {
1057 err = memif_syscall_error_handler (errno);
1058 DBG_UNIX ("timerfd_settime: arm");
1059 }
1060 }
1061 lm->disconn_slaves++;
1062 }
1063
1064 return err;
1065}
1066
1067int
1068memif_delete (memif_conn_handle_t * conn)
1069{
1070 memif_connection_t *c = (memif_connection_t *) * conn;
1071 if (c == NULL)
1072 {
1073 DBG ("no connection");
1074 return MEMIF_ERR_NOCONN;
1075 }
1076 libmemif_main_t *lm = &libmemif_main;
1077 memif_list_elt_t *e = NULL;
1078 memif_socket_t *ms = NULL;
1079
1080 int err = MEMIF_ERR_SUCCESS;
1081
1082 if (c->fd > 0)
1083 {
1084 DBG ("DISCONNECTING");
1085 err = memif_disconnect_internal (c);
1086 if (err == MEMIF_ERR_NOCONN)
1087 return err;
1088 }
1089
1090 free_list_elt_ctx (lm->control_list, lm->control_list_len, c);
1091
1092 if (c->args.is_master)
1093 {
1094 get_list_elt (&e, lm->listener_list, lm->listener_list_len,
1095 c->listener_fd);
1096 if (e != NULL)
1097 {
1098 ms = (memif_socket_t *) e->data_struct;
1099 ms->use_count--;
1100 free_list_elt (ms->interface_list, ms->interface_list_len,
1101 c->args.interface_id);
1102 if (ms->use_count <= 0)
1103 {
1104 lm->control_fd_update (c->listener_fd, MEMIF_FD_EVENT_DEL);
1105 free_list_elt (lm->listener_list, lm->listener_list_len,
1106 c->listener_fd);
1107 close (c->listener_fd);
1108 c->listener_fd = ms->fd = -1;
1109 free (ms->interface_list);
1110 ms->interface_list = NULL;
1111 free (ms->filename);
1112 ms->filename = NULL;
1113 free (ms);
1114 ms = NULL;
1115 }
1116 }
1117 }
1118 else
1119 {
1120 lm->disconn_slaves--;
1121 if (lm->disconn_slaves <= 0)
1122 {
1123 if (timerfd_settime (lm->timerfd, 0, &lm->disarm, NULL) < 0)
1124 {
1125 err = memif_syscall_error_handler (errno);
1126 DBG ("timerfd_settime: disarm");
1127 }
1128 }
1129 }
1130
1131 if (c->args.socket_filename)
1132 free (c->args.socket_filename);
1133 c->args.socket_filename = NULL;
1134
1135 free (c);
1136 c = NULL;
1137
1138 *conn = c;
1139 return err;
1140}
1141
1142int
1143memif_connect1 (memif_connection_t * c)
1144{
1145 libmemif_main_t *lm = &libmemif_main;
1146 memif_region_t *mr = c->regions;
1147 memif_queue_t *mq;
1148 int i;
1149 uint16_t num;
1150
1151 if (mr != NULL)
1152 {
1153 if (!mr->shm)
1154 {
1155 if (mr->fd < 0)
1156 return MEMIF_ERR_NO_SHMFD;
1157
1158 if ((mr->shm = mmap (NULL, mr->region_size, PROT_READ | PROT_WRITE,
1159 MAP_SHARED, mr->fd, 0)) == MAP_FAILED)
1160 {
1161 return memif_syscall_error_handler (errno);
1162 }
1163 }
1164 }
1165
1166 num =
1167 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
1168 num_s2m_rings;
1169 for (i = 0; i < num; i++)
1170 {
1171 mq = &c->tx_queues[i];
1172 if (mq != NULL)
1173 {
1174 mq->ring = c->regions[mq->region].shm + mq->offset;
1175 if (mq->ring->cookie != MEMIF_COOKIE)
1176 {
1177 DBG ("wrong cookie on tx ring %u", i);
1178 return MEMIF_ERR_COOKIE;
1179 }
1180 mq->ring->head = mq->ring->tail = mq->last_head = mq->alloc_bufs =
1181 0;
1182 }
1183 }
1184 num =
1185 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1186 num_m2s_rings;
1187 for (i = 0; i < num; i++)
1188 {
1189 mq = &c->rx_queues[i];
1190 if (mq != NULL)
1191 {
1192 mq->ring = c->regions[mq->region].shm + mq->offset;
1193 if (mq->ring->cookie != MEMIF_COOKIE)
1194 {
1195 DBG ("wrong cookie on rx ring %u", i);
1196 return MEMIF_ERR_COOKIE;
1197 }
1198 mq->ring->head = mq->ring->tail = mq->last_head = mq->alloc_bufs =
1199 0;
1200 }
1201 }
1202
1203 lm->control_fd_update (c->fd, MEMIF_FD_EVENT_READ | MEMIF_FD_EVENT_MOD);
1204
1205 return 0;
1206}
1207
1208int
1209memif_init_regions_and_queues (memif_connection_t * conn)
1210{
1211 memif_ring_t *ring = NULL;
1212 uint64_t buffer_offset;
1213 memif_region_t *r;
1214 int i, j;
1215 libmemif_main_t *lm = &libmemif_main;
1216 memif_list_elt_t e;
1217
1218 conn->regions = (memif_region_t *) malloc (sizeof (memif_region_t));
1219 if (conn->regions == NULL)
1220 return memif_syscall_error_handler (errno);
1221 r = conn->regions;
1222
1223 buffer_offset =
1224 (conn->run_args.num_s2m_rings +
1225 conn->run_args.num_m2s_rings) * (sizeof (memif_ring_t) +
1226 sizeof (memif_desc_t) *
1227 (1 << conn->run_args.log2_ring_size));
1228
1229 r->region_size = buffer_offset +
1230 conn->run_args.buffer_size * (1 << conn->run_args.log2_ring_size) *
1231 (conn->run_args.num_s2m_rings + conn->run_args.num_m2s_rings);
1232
1233 if ((r->fd = memfd_create ("memif region 0", MFD_ALLOW_SEALING)) == -1)
1234 return memif_syscall_error_handler (errno);
1235/*
1236 if ((fcntl (r->fd, F_ADD_SEALS, F_SEAL_SHRINK)) == -1)
1237 return memif_syscall_error_handler (errno);
1238*/
1239 if ((ftruncate (r->fd, r->region_size)) == -1)
1240 return memif_syscall_error_handler (errno);
1241
1242 if ((r->shm = mmap (NULL, r->region_size, PROT_READ | PROT_WRITE,
1243 MAP_SHARED, r->fd, 0)) == MAP_FAILED)
1244 return memif_syscall_error_handler (errno);
1245
1246 for (i = 0; i < conn->run_args.num_s2m_rings; i++)
1247 {
1248 ring = memif_get_ring (conn, MEMIF_RING_S2M, i);
1249 DBG ("RING: %p I: %d", ring, i);
1250 ring->head = ring->tail = 0;
1251 ring->cookie = MEMIF_COOKIE;
1252 ring->flags = 0;
1253 for (j = 0; j < (1 << conn->run_args.log2_ring_size); j++)
1254 {
1255 uint16_t slot = i * (1 << conn->run_args.log2_ring_size) + j;
1256 ring->desc[j].region = 0;
1257 ring->desc[j].offset = buffer_offset +
1258 (uint32_t) (slot * conn->run_args.buffer_size);
1259 ring->desc[j].buffer_length = conn->run_args.buffer_size;
1260 }
1261 }
1262 for (i = 0; i < conn->run_args.num_m2s_rings; i++)
1263 {
1264 ring = memif_get_ring (conn, MEMIF_RING_M2S, i);
1265 DBG ("RING: %p I: %d", ring, i);
1266 ring->head = ring->tail = 0;
1267 ring->cookie = MEMIF_COOKIE;
1268 ring->flags = 0;
1269 for (j = 0; j < (1 << conn->run_args.log2_ring_size); j++)
1270 {
1271 uint16_t slot =
1272 (i +
1273 conn->run_args.num_s2m_rings) *
1274 (1 << conn->run_args.log2_ring_size) + j;
1275 ring->desc[j].region = 0;
1276 ring->desc[j].offset = buffer_offset +
1277 (uint32_t) (slot * conn->run_args.buffer_size);
1278 ring->desc[j].buffer_length = conn->run_args.buffer_size;
1279 }
1280 }
1281 memif_queue_t *mq;
1282 mq =
1283 (memif_queue_t *) malloc (sizeof (memif_queue_t) *
1284 conn->run_args.num_s2m_rings);
1285 if (mq == NULL)
1286 return memif_syscall_error_handler (errno);
1287 int x;
1288 for (x = 0; x < conn->run_args.num_s2m_rings; x++)
1289 {
1290 if ((mq[x].int_fd = eventfd (0, EFD_NONBLOCK)) < 0)
1291 return memif_syscall_error_handler (errno);
1292 /* add int fd to interrupt fd list */
1293 e.key = mq[x].int_fd;
1294 e.data_struct = conn;
1295 add_list_elt (&e, &lm->interrupt_list, &lm->interrupt_list_len);
1296
1297 mq[x].ring = memif_get_ring (conn, MEMIF_RING_S2M, x);
1298 DBG ("RING: %p I: %d", mq[x].ring, x);
1299 mq[x].log2_ring_size = conn->run_args.log2_ring_size;
1300 mq[x].region = 0;
1301 mq[x].offset =
1302 (void *) mq[x].ring - (void *) conn->regions[mq->region].shm;
1303 mq[x].last_head = 0;
1304 mq[x].alloc_bufs = 0;
1305 }
1306 conn->tx_queues = mq;
1307
1308 mq =
1309 (memif_queue_t *) malloc (sizeof (memif_queue_t) *
1310 conn->run_args.num_m2s_rings);
1311 if (mq == NULL)
1312 return memif_syscall_error_handler (errno);
1313 for (x = 0; x < conn->run_args.num_m2s_rings; x++)
1314 {
1315 if ((mq[x].int_fd = eventfd (0, EFD_NONBLOCK)) < 0)
1316 return memif_syscall_error_handler (errno);
1317 /* add int fd to interrupt fd list */
1318 e.key = mq[x].int_fd;
1319 e.data_struct = conn;
1320 add_list_elt (&e, &lm->interrupt_list, &lm->interrupt_list_len);
1321
1322 mq[x].ring = memif_get_ring (conn, MEMIF_RING_M2S, x);
1323 DBG ("RING: %p I: %d", mq[x].ring, x);
1324 mq[x].log2_ring_size = conn->run_args.log2_ring_size;
1325 mq[x].region = 0;
1326 mq[x].offset =
1327 (void *) mq[x].ring - (void *) conn->regions[mq->region].shm;
1328 mq[x].last_head = 0;
1329 mq[x].alloc_bufs = 0;
1330 }
1331 conn->rx_queues = mq;
1332
1333 return 0;
1334}
1335
1336int
1337memif_buffer_alloc (memif_conn_handle_t conn, uint16_t qid,
1338 memif_buffer_t * bufs, uint16_t count,
1339 uint16_t * count_out)
1340{
1341 memif_connection_t *c = (memif_connection_t *) conn;
1342 if (c == NULL)
1343 return MEMIF_ERR_NOCONN;
1344 if (c->fd < 0)
1345 return MEMIF_ERR_DISCONNECTED;
1346 uint8_t num =
1347 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
1348 num_s2m_rings;
1349 if (qid >= num)
1350 return MEMIF_ERR_QID;
1351 memif_queue_t *mq = &c->tx_queues[qid];
1352 memif_ring_t *ring = mq->ring;
1353 memif_buffer_t *b0, *b1;
1354 uint16_t mask = (1 << mq->log2_ring_size) - 1;
1355 uint16_t s0, s1, ns;
1356 *count_out = 0;
1357 int err = MEMIF_ERR_SUCCESS; /* 0 */
1358
1359 if (ring->tail != ring->head)
1360 {
1361 if (ring->head > ring->tail)
1362 ns = (1 << mq->log2_ring_size) - ring->head + ring->tail;
1363 else
1364 ns = ring->tail - ring->head;
1365 }
1366 else
1367 ns = (1 << mq->log2_ring_size);
1368
1369 /* (head == tail) ? receive function will asume that no packets are available */
1370 ns -= 1;
1371
1372 while (count && ns)
1373 {
1374 while ((count > 2) && (ns > 2))
1375 {
1376 s0 = (ring->head + mq->alloc_bufs + *count_out) & mask;
1377 s1 = (ring->head + mq->alloc_bufs + *count_out + 1) & mask;
1378
1379 b0 = (bufs + *count_out);
1380 b1 = (bufs + *count_out + 1);
1381
1382 b0->desc_index = s0;
1383 b1->desc_index = s1;
1384 b0->buffer_len = ring->desc[s0].buffer_length;
1385 b1->buffer_len = ring->desc[s1].buffer_length;
1386 /* TODO: support multiple regions -> ring descriptor contains region index */
1387 b0->data = c->regions->shm + ring->desc[s0].offset;
1388 b1->data = c->regions->shm + ring->desc[s1].offset;
1389
1390 DBG ("allocated ring slots %u, %u", s0, s1);
1391 count -= 2;
1392 ns -= 2;
1393 *count_out += 2;
1394 }
1395 s0 = (ring->head + mq->alloc_bufs + *count_out) & mask;
1396
1397 b0 = (bufs + *count_out);
1398
1399 b0->desc_index = s0;
1400 b0->buffer_len = ring->desc[s0].buffer_length;
1401 b0->data = c->regions->shm + ring->desc[s0].offset;
1402
1403 DBG ("allocated ring slot %u", s0);
1404 count--;
1405 ns--;
1406 *count_out += 1;
1407 }
1408
1409 mq->alloc_bufs += *count_out;
1410 DBG ("allocated: %u/%u bufs. Total %u allocated bufs", *count_out, count,
1411 mq->alloc_bufs);
1412
1413 if (count)
1414 {
1415 DBG ("ring buffer full! qid: %u", qid);
1416 err = MEMIF_ERR_NOBUF_RING;
1417 }
1418
1419 return err;
1420}
1421
1422int
1423memif_buffer_free (memif_conn_handle_t conn, uint16_t qid,
1424 memif_buffer_t * bufs, uint16_t count,
1425 uint16_t * count_out)
1426{
1427 memif_connection_t *c = (memif_connection_t *) conn;
1428 if (c == NULL)
1429 return MEMIF_ERR_NOCONN;
1430 if (c->fd < 0)
1431 return MEMIF_ERR_DISCONNECTED;
1432 uint8_t num =
1433 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1434 num_m2s_rings;
1435 if (qid >= num)
1436 return MEMIF_ERR_QID;
1437 libmemif_main_t *lm = &libmemif_main;
1438 memif_queue_t *mq = &c->rx_queues[qid];
1439 memif_ring_t *ring = mq->ring;
1440 uint16_t tail = ring->tail;
1441 uint16_t mask = (1 << mq->log2_ring_size) - 1;
1442 memif_buffer_t *b0, *b1;
1443 *count_out = 0;
1444
1445 if (mq->alloc_bufs < count)
1446 count = mq->alloc_bufs;
1447
1448 while (count)
1449 {
1450 while (count > 2)
1451 {
1452 b0 = (bufs + *count_out);
1453 b1 = (bufs + *count_out + 1);
1454 tail = (b0->desc_index + 1) & mask;
1455 tail = (b1->desc_index + 1) & mask;
1456 b0->data = NULL;
1457 b1->data = NULL;
1458
1459 count -= 2;
1460 *count_out += 2;
1461 mq->alloc_bufs -= 2;
1462 }
1463 b0 = (bufs + *count_out);
1464 tail = (b0->desc_index + 1) & mask;
1465 b0->data = NULL;
1466
1467 count--;
1468 *count_out += 1;
1469 mq->alloc_bufs--;
1470 }
1471 MEMIF_MEORY_BARRIER ();
1472 ring->tail = tail;
1473
1474 return MEMIF_ERR_SUCCESS; /* 0 */
1475}
1476
1477int
1478memif_tx_burst (memif_conn_handle_t conn, uint16_t qid,
1479 memif_buffer_t * bufs, uint16_t count, uint16_t * tx)
1480{
1481 memif_connection_t *c = (memif_connection_t *) conn;
1482 if (c == NULL)
1483 return MEMIF_ERR_NOCONN;
1484 if (c->fd < 0)
1485 return MEMIF_ERR_DISCONNECTED;
1486 uint8_t num =
1487 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
1488 num_s2m_rings;
1489 if (qid >= num)
1490 return MEMIF_ERR_QID;
1491 memif_queue_t *mq = &c->tx_queues[qid];
1492 memif_ring_t *ring = mq->ring;
1493 uint16_t head = ring->head;
1494 uint16_t mask = (1 << mq->log2_ring_size) - 1;
1495 *tx = 0;
1496 memif_buffer_t *b0, *b1;
1497
1498 while (count)
1499 {
1500 while (count > 2)
1501 {
1502 b0 = (bufs + *tx);
1503 b1 = (bufs + *tx + 1);
1504 ring->desc[b0->desc_index].length = b0->data_len;
1505 ring->desc[b1->desc_index].length = b1->data_len;
1506
1507#ifdef MEMIF_DBG_SHM
1508 print_bytes (b0->data, b0->data_len, DBG_TX_BUF);
1509 print_bytes (b1->data, b1->data_len, DBG_TX_BUF);
1510#endif
1511
1512 head = (b0->desc_index + 1) & mask;
1513 head = (b1->desc_index + 1) & mask;
1514
1515 b0->data = NULL;
1516 b0->data_len = 0;
1517 b1->data = NULL;
1518 b1->data_len = 0;
1519
1520 count -= 2;
1521 *tx += 2;
1522 }
1523
1524 b0 = (bufs + *tx);
1525 ring->desc[b0->desc_index].length = b0->data_len;
1526
1527#ifdef MEMIF_DBG_SHM
1528 print_bytes (b0->data, b0->data_len, DBG_TX_BUF);
1529#endif
1530
1531 head = (b0->desc_index + 1) & mask;
1532
1533 b0->data = NULL;
1534 b0->data_len = 0;
1535
1536 count--;
1537 *tx += 1;
1538 }
1539 MEMIF_MEORY_BARRIER ();
1540 ring->head = head;
1541
1542 mq->alloc_bufs -= *tx;
1543
1544 if ((ring->flags & MEMIF_RING_FLAG_MASK_INT) == 0)
1545 {
1546 uint64_t a = 1;
1547 int r = write (mq->int_fd, &a, sizeof (a));
1548 if (r < 0)
1549 return MEMIF_ERR_INT_WRITE;
1550 }
1551
1552 return MEMIF_ERR_SUCCESS; /* 0 */
1553}
1554
1555int
1556memif_rx_burst (memif_conn_handle_t conn, uint16_t qid,
1557 memif_buffer_t * bufs, uint16_t count, uint16_t * rx)
1558{
1559 memif_connection_t *c = (memif_connection_t *) conn;
1560 if (c == NULL)
1561 return MEMIF_ERR_NOCONN;
1562 if (c->fd < 0)
1563 return MEMIF_ERR_DISCONNECTED;
1564 uint8_t num =
1565 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1566 num_m2s_rings;
1567 if (qid >= num)
1568 return MEMIF_ERR_QID;
1569 memif_queue_t *mq = &c->rx_queues[qid];
1570 memif_ring_t *ring = mq->ring;
1571 uint16_t head = ring->head;
1572 uint16_t ns;
1573 uint16_t mask = (1 << mq->log2_ring_size) - 1;
1574 memif_buffer_t *b0, *b1;
1575 *rx = 0;
1576
1577 uint64_t b;
1578 ssize_t r = read (mq->int_fd, &b, sizeof (b));
1579 if ((r == -1) && (errno != EAGAIN))
1580 return memif_syscall_error_handler (errno);
1581
1582 if (head == mq->last_head)
1583 return 0;
1584
1585 if (head > mq->last_head)
1586 ns = head - mq->last_head;
1587 else
1588 ns = (1 << mq->log2_ring_size) - mq->last_head + head;
1589
1590 while (ns && count)
1591 {
1592 while ((ns > 2) && (count > 2))
1593 {
1594 b0 = (bufs + *rx);
1595 b1 = (bufs + *rx + 1);
1596
1597 b0->desc_index = mq->last_head;
1598 b1->desc_index = mq->last_head + 1;
1599 b0->data = memif_get_buffer (conn, ring, mq->last_head);
1600 b1->data = memif_get_buffer (conn, ring, mq->last_head + 1);
1601 b0->data_len = ring->desc[mq->last_head].length;
1602 b1->data_len = ring->desc[mq->last_head + 1].length;
1603 b0->buffer_len = ring->desc[mq->last_head].buffer_length;
1604 b1->buffer_len = ring->desc[mq->last_head + 1].buffer_length;
1605
1606#ifdef MEMIF_DBG_SHM
1607 print_bytes (b0->data, b0->data_len, DBG_RX_BUF);
1608 print_bytes (b1->data, b1->data_len, DBG_RX_BUF);
1609#endif
1610
1611 mq->last_head = (mq->last_head + 2) & mask;
1612
1613 ns -= 2;
1614 count -= 2;
1615 *rx += 2;
1616 }
1617 b0 = (bufs + *rx);
1618
1619 b0->desc_index = mq->last_head;
1620 b0->data = memif_get_buffer (conn, ring, mq->last_head);
1621 b0->data_len = ring->desc[mq->last_head].length;
1622 b0->buffer_len = ring->desc[mq->last_head].buffer_length;
1623
1624#ifdef MEMIF_DBG_SHM
1625 print_bytes (b0->data, b0->data_len, DBG_RX_BUF);
1626#endif
1627
1628 mq->last_head = (mq->last_head + 1) & mask;
1629
1630 ns--;
1631 count--;
1632 *rx += 1;
1633 }
1634
1635 mq->alloc_bufs += *rx;
1636
1637 if (ns)
1638 {
1639 DBG ("not enough buffers!");
1640 return MEMIF_ERR_NOBUF;
1641 }
1642
1643 return MEMIF_ERR_SUCCESS; /* 0 */
1644}
1645
1646int
1647memif_get_details (memif_conn_handle_t conn, memif_details_t * md,
1648 char *buf, ssize_t buflen)
1649{
1650 memif_connection_t *c = (memif_connection_t *) conn;
1651 if (c == NULL)
1652 return MEMIF_ERR_NOCONN;
1653
1654 int err = MEMIF_ERR_SUCCESS, i;
1655 ssize_t l0, l1, total_l;
1656 l0 = 0;
1657
1658 l1 = strlen ((char *) c->args.interface_name);
1659 if (l0 + l1 <= buflen)
1660 {
1661 md->if_name = strncpy (buf + l0, (char *) c->args.interface_name, l1);
1662 md->if_name[l0 + l1] = '\0';
1663 l0 += l1 + 1;
1664 }
1665 else
1666 err = MEMIF_ERR_NOBUF_DET;
1667
1668 l1 = strlen ((char *) c->args.instance_name);
1669 if (l0 + l1 <= buflen)
1670 {
1671 md->inst_name = strncpy (buf + l0, (char *) c->args.instance_name, l1);
1672 md->inst_name[l0 + l1] = '\0';
1673 l0 += l1 + 1;
1674 }
1675 else
1676 err = MEMIF_ERR_NOBUF_DET;
1677
1678 l1 = strlen ((char *) c->remote_if_name);
1679 if (l0 + l1 <= buflen)
1680 {
1681 md->remote_if_name = strncpy (buf + l0, (char *) c->remote_if_name, l1);
1682 md->remote_if_name[l0 + l1] = '\0';
1683 l0 += l1 + 1;
1684 }
1685 else
1686 err = MEMIF_ERR_NOBUF_DET;
1687
1688 l1 = strlen ((char *) c->remote_name);
1689 if (l0 + l1 <= buflen)
1690 {
1691 md->remote_inst_name = strncpy (buf + l0, (char *) c->remote_name, l1);
1692 md->remote_inst_name[l0 + l1] = '\0';
1693 l0 += l1 + 1;
1694 }
1695 else
1696 err = MEMIF_ERR_NOBUF_DET;
1697
1698 md->id = c->args.interface_id;
1699
1700 if (c->args.secret)
1701 {
1702 l1 = strlen ((char *) c->args.secret);
1703 md->secret = strncpy (buf + l0, (char *) c->args.secret, l1);
1704 md->secret[l0 + l1] = '\0';
1705 l0 += l1 + 1;
1706 }
1707 else
1708 err = MEMIF_ERR_NOBUF_DET;
1709
1710 md->role = (c->args.is_master) ? 0 : 1;
1711 md->mode = c->args.mode;
1712
1713 l1 = strlen ((char *) c->args.socket_filename);
1714 if (l0 + l1 <= buflen)
1715 {
1716 md->socket_filename =
1717 strncpy (buf + l0, (char *) c->args.socket_filename, l1);
1718 md->socket_filename[l0 + l1] = '\0';
1719 l0 += l1 + 1;
1720 }
1721 else
1722 err = MEMIF_ERR_NOBUF_DET;
1723
1724 md->rx_queues_num =
1725 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1726 num_m2s_rings;
1727
1728 l1 = sizeof (memif_queue_details_t) * md->rx_queues_num;
1729 if (l0 + l1 <= buflen)
1730 {
1731 md->rx_queues = (memif_queue_details_t *) buf + l0;
1732 l0 = l1 + 1;
1733 }
1734 else
1735 err = MEMIF_ERR_NOBUF_DET;
1736
1737 for (i = 0; i < md->rx_queues_num; i++)
1738 {
1739 md->rx_queues[i].qid = i;
1740 md->rx_queues[i].ring_size = (1 << c->rx_queues[i].log2_ring_size);
1741 md->rx_queues[i].buffer_size = c->run_args.buffer_size;
1742 }
1743
1744 md->tx_queues_num =
1745 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
1746 num_s2m_rings;
1747
1748 l1 = sizeof (memif_queue_details_t) * md->tx_queues_num;
1749 if (l0 + l1 <= buflen)
1750 {
1751 md->tx_queues = (memif_queue_details_t *) buf + l0;
1752 l0 = l1 + 1;
1753 }
1754 else
1755 err = MEMIF_ERR_NOBUF_DET;
1756
1757 for (i = 0; i < md->tx_queues_num; i++)
1758 {
1759 md->tx_queues[i].qid = i;
1760 md->tx_queues[i].ring_size = (1 << c->tx_queues[i].log2_ring_size);
1761 md->tx_queues[i].buffer_size = c->run_args.buffer_size;
1762 }
1763
1764 md->link_up_down = (c->fd > 0) ? 1 : 0;
1765
1766 return err; /* 0 */
1767}
1768
1769int
1770memif_get_queue_efd (memif_conn_handle_t conn, uint16_t qid, int *efd)
1771{
1772 memif_connection_t *c = (memif_connection_t *) conn;
1773 *efd = -1;
1774 if (c == NULL)
1775 return MEMIF_ERR_NOCONN;
1776 if (c->fd < 0)
1777 return MEMIF_ERR_DISCONNECTED;
1778 uint8_t num =
1779 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1780 num_m2s_rings;
1781 if (qid >= num)
1782 return MEMIF_ERR_QID;
1783
1784 *efd = c->rx_queues[qid].int_fd;
1785
1786 return MEMIF_ERR_SUCCESS;
1787}
1788
1789int
1790memif_cleanup ()
1791{
1792 libmemif_main_t *lm = &libmemif_main;
1793 if (lm->app_name)
1794 free (lm->app_name);
1795 lm->app_name = NULL;
1796 if (lm->control_list)
1797 free (lm->control_list);
1798 lm->control_list = NULL;
1799 if (lm->interrupt_list)
1800 free (lm->interrupt_list);
1801 lm->interrupt_list = NULL;
1802 if (lm->listener_list)
1803 free (lm->listener_list);
1804 lm->listener_list = NULL;
1805 if (lm->pending_list)
1806 free (lm->pending_list);
1807 lm->pending_list = NULL;
1808
1809 return MEMIF_ERR_SUCCESS; /* 0 */
1810}