blob: 39c18f27477ca399d469d1cf8cac95702a72882d [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
Milan Lenco0a47c992017-10-12 14:19:31 +020057#define ERRLIST_LEN 37
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +020058#define MAX_ERRBUF_LEN 256
59
60#if __x86_x64__
61#define MEMIF_MEMORY_BARRIER() __builtin_ia32_sfence ()
62#else
Milan Lenco0a47c992017-10-12 14:19:31 +020063#define MEMIF_MEMORY_BARRIER() __sync_synchronize ()
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +020064#endif /* __x86_x64__ */
65
66libmemif_main_t libmemif_main;
67int memif_epfd;
Milan Lenco0a47c992017-10-12 14:19:31 +020068int poll_cancel_fd = -1;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +020069
70static char memif_buf[MAX_ERRBUF_LEN];
71
72const char *memif_errlist[ERRLIST_LEN] = { /* MEMIF_ERR_SUCCESS */
73 "Success.",
74 /* MEMIF_ERR_SYSCALL */
75 "Unspecified syscall error (build with -DMEMIF_DBG or make debug).",
76 /* MEMIF_ERR_ACCES */
77 "Permission to resoure denied.",
78 /* MEMIF_ERR_NO_FILE */
79 "Socket file does not exist",
80 /* MEMIF_ERR_FILE_LIMIT */
81 "System limit on total numer of open files reached.",
82 /* MEMIF_ERR_PROC_FILE_LIMIT */
83 "Per-process limit on total number of open files reached.",
84 /* MEMIF_ERR_ALREADY */
85 "Connection already requested.",
86 /* MEMIF_ERR_AGAIN */
87 "File descriptor refers to file other than socket, or operation would block.",
88 /* MEMIF_ERR_BAD_FD */
89 "Bad file descriptor.",
90 /* MEMIF_ERR_NOMEM */
91 "Out of memory.",
92 /* MEMIF_ERR_INVAL_ARG */
93 "Invalid argument.",
94 /* MEMIF_ERR_NOCONN */
95 "Memif connection handle does not point to existing conenction",
96 /* MEMIF_ERR_CONN */
97 "Memif connection handle points to existing connection",
98 /* MEMIF_ERR_CB_FDUPDATE */
99 "Callback memif_control_fd_update_t returned error",
100 /* MEMIF_ERR_FILE_NOT_SOCK */
101 "File specified by socket filename exists and is not socket.",
102 /* MEMIF_ERR_NO_SHMFD */
103 "Missing shared memory file descriptor. (internal error)",
104 /* MEMIF_ERR_COOKIE */
105 "Invalid cookie on ring. (internal error)",
106 /* MEMIF_ERR_NOBUF_RING */
107 "Ring buffer full.",
108 /* MEMIF_ERR_NOBUF */
109 "Not enough memif buffers. There are unreceived data in shared memory.",
110 /* MEMIF_ERR_NOBUF_DET */
111 "Not enough space for memif details in suplied buffer. String data might be malformed.",
112 /* MEMIF_ERR_INT_WRITE */
113 "Send interrupt error.",
114 /* MEMIF_ERR_MFMSG */
115 "Malformed message received on control channel.",
116 /* MEMIF_ERR_QID */
117 "Invalid queue id",
118 /* MEMIF_ERR_PROTO */
119 "Incompatible memory interface protocol version.",
120 /* MEMIF_ERR_ID */
121 "Unmatched interface id.",
122 /* MEMIF_ERR_ACCSLAVE */
123 "Slave cannot accept connection reqest.",
124 /* MEMIF_ERR_ALRCONN */
125 "Interface is already connected.",
126 /* MEMIF_ERR_MODE */
127 "Mode mismatch.",
128 /* MEMIF_ERR_SECRET */
129 "Secret mismatch.",
130 /* MEMIF_ERR_NOSECRET */
131 "Secret required.",
132 /* MEMIF_ERR_MAXREG */
133 "Limit on total number of regions reached.",
134 /* MEMIF_ERR_MAXRING */
135 "Limit on total number of ring reached.",
136 /* MEMIF_ERR_NO_INTFD */
137 "Missing interrupt file descriptor. (internal error)",
138 /* MEMIF_ERR_DISCONNECT */
139 "Interface received disconnect request.",
140 /* MEMIF_ERR_DISCONNECTED */
141 "Interface is disconnected.",
142 /* MEMIF_ERR_UNKNOWN_MSG */
Milan Lenco0a47c992017-10-12 14:19:31 +0200143 "Unknown message type received on control channel. (internal error)",
144 /* MEMIF_ERR_POLL_CANCEL */
145 "Memif event polling was canceled."
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200146};
147
148#define MEMIF_ERR_UNDEFINED "undefined error"
149
150char *
151memif_strerror (int err_code)
152{
153 if (err_code >= ERRLIST_LEN)
154 {
155 strncpy (memif_buf, MEMIF_ERR_UNDEFINED, strlen (MEMIF_ERR_UNDEFINED));
156 memif_buf[strlen (MEMIF_ERR_UNDEFINED)] = '\0';
157 }
158 else
159 {
160 strncpy (memif_buf, memif_errlist[err_code],
161 strlen (memif_errlist[err_code]));
162 memif_buf[strlen (memif_errlist[err_code])] = '\0';
163 }
164 return memif_buf;
165}
166
167#define DBG_TX_BUF (0)
168#define DBG_RX_BUF (1)
169
170#ifdef MEMIF_DBG_SHM
171static void
172print_bytes (void *data, uint16_t len, uint8_t q)
173{
174 if (q == DBG_TX_BUF)
175 printf ("\nTX:\n\t");
176 else
177 printf ("\nRX:\n\t");
178 int i;
179 for (i = 0; i < len; i++)
180 {
181 if (i % 8 == 0)
182 printf ("\n%d:\t", i);
183 printf ("%02X ", ((uint8_t *) (data))[i]);
184 }
185 printf ("\n\n");
186}
Jakub Grajciarba3c4e82017-09-18 11:21:40 +0200187#endif /* MEMIF_DBG_SHM */
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200188
189int
190memif_syscall_error_handler (int err_code)
191{
192 DBG_UNIX ("%s", strerror (err_code));
193
194 if (err_code == 0)
195 return MEMIF_ERR_SUCCESS;
196 if (err_code == EACCES)
197 return MEMIF_ERR_ACCES;
198 if (err_code == ENFILE)
199 return MEMIF_ERR_FILE_LIMIT;
200 if (err_code == EMFILE)
201 return MEMIF_ERR_PROC_FILE_LIMIT;
202 if (err_code == ENOMEM)
203 return MEMIF_ERR_NOMEM;
204/* connection refused if master dows not exist
205 this error would spam the user until master was created */
206 if (err_code == ECONNREFUSED)
207 return MEMIF_ERR_SUCCESS;
208 if (err_code == EALREADY)
209 return MEMIF_ERR_ALREADY;
210 if (err_code == EAGAIN)
211 return MEMIF_ERR_AGAIN;
212 if (err_code == EBADF)
213 return MEMIF_ERR_BAD_FD;
214 if (err_code == ENOENT)
215 return MEMIF_ERR_NO_FILE;
216
217 /* other syscall errors */
218 return MEMIF_ERR_SYSCALL;
219}
220
221static int
222memif_add_epoll_fd (int fd, uint32_t events)
223{
224 if (fd < 0)
225 {
226 DBG ("invalid fd %d", fd);
227 return -1;
228 }
229 struct epoll_event evt;
230 memset (&evt, 0, sizeof (evt));
231 evt.events = events;
232 evt.data.fd = fd;
233 if (epoll_ctl (memif_epfd, EPOLL_CTL_ADD, fd, &evt) < 0)
234 {
235 DBG ("epoll_ctl: %s fd %d", strerror (errno), fd);
236 return -1;
237 }
238 DBG ("fd %d added to epoll", fd);
239 return 0;
240}
241
242static int
243memif_mod_epoll_fd (int fd, uint32_t events)
244{
245 if (fd < 0)
246 {
247 DBG ("invalid fd %d", fd);
248 return -1;
249 }
250 struct epoll_event evt;
251 memset (&evt, 0, sizeof (evt));
252 evt.events = events;
253 evt.data.fd = fd;
254 if (epoll_ctl (memif_epfd, EPOLL_CTL_MOD, fd, &evt) < 0)
255 {
256 DBG ("epoll_ctl: %s fd %d", strerror (errno), fd);
257 return -1;
258 }
259 DBG ("fd %d moddified on epoll", fd);
260 return 0;
261}
262
263static int
264memif_del_epoll_fd (int fd)
265{
266 if (fd < 0)
267 {
268 DBG ("invalid fd %d", fd);
269 return -1;
270 }
271 struct epoll_event evt;
272 memset (&evt, 0, sizeof (evt));
273 if (epoll_ctl (memif_epfd, EPOLL_CTL_DEL, fd, &evt) < 0)
274 {
275 DBG ("epoll_ctl: %s fd %d", strerror (errno), fd);
276 return -1;
277 }
278 DBG ("fd %d removed from epoll", fd);
279 return 0;
280}
281
282int
283memif_control_fd_update (int fd, uint8_t events)
284{
285 if (events & MEMIF_FD_EVENT_DEL)
286 return memif_del_epoll_fd (fd);
287
288 uint32_t evt = 0;
289 if (events & MEMIF_FD_EVENT_READ)
290 evt |= EPOLLIN;
291 if (events & MEMIF_FD_EVENT_WRITE)
292 evt |= EPOLLOUT;
293
294 if (events & MEMIF_FD_EVENT_MOD)
295 return memif_mod_epoll_fd (fd, evt);
296
297 return memif_add_epoll_fd (fd, evt);
298}
299
300int
301add_list_elt (memif_list_elt_t * e, memif_list_elt_t ** list, uint16_t * len)
302{
303 libmemif_main_t *lm = &libmemif_main;
304
305 int i;
306 for (i = 0; i < *len; i++)
307 {
308 if ((*list)[i].data_struct == NULL)
309 {
310 (*list)[i].key = e->key;
311 (*list)[i].data_struct = e->data_struct;
312 return i;
313 }
314 }
315 memif_list_elt_t *tmp;
316 tmp = realloc (*list, sizeof (memif_list_elt_t) * *len * 2);
317 if (tmp == NULL)
318 return -1;
319
320 for (i = *len; i < *len * 2; i++)
321 {
322 tmp[i].key = -1;
323 tmp[i].data_struct = NULL;
324 }
325
326 tmp[*len].key = e->key;
327 tmp[*len].data_struct = e->data_struct;
328 i = *len;
329 *len = *len * 2;
330 *list = tmp;
331
332 return i;
333}
334
335int
336get_list_elt (memif_list_elt_t ** e, memif_list_elt_t * list, uint16_t len,
337 int key)
338{
339 if (key == -1)
340 {
341 *e = NULL;
342 return -1;
343 }
344 int i;
345 for (i = 0; i < len; i++)
346 {
347 if (list[i].key == key)
348 {
349 *e = &list[i];
350 return 0;
351 }
352 }
353 *e = NULL;
354 return -1;
355}
356
357/* does not free memory, only marks element as free */
358int
359free_list_elt (memif_list_elt_t * list, uint16_t len, int key)
360{
361 int i;
362 for (i = 0; i < len; i++)
363 {
364 if (list[i].key == key)
365 {
366 list[i].key = -1;
367 list[i].data_struct = NULL;
368 return 0;
369 }
370 }
371
372 return -1;
373}
374
375int
376free_list_elt_ctx (memif_list_elt_t * list, uint16_t len,
377 memif_connection_t * ctx)
378{
379 int i;
380 for (i = 0; i < len; i++)
381 {
382 if (list[i].key == -1)
383 {
384 if (list[i].data_struct == ctx)
385 {
386 list[i].data_struct = NULL;
387 return 0;
388 }
389 }
390 }
391
392 return -1;
393}
394
395static void
396memif_control_fd_update_register (memif_control_fd_update_t * cb)
397{
398 libmemif_main_t *lm = &libmemif_main;
399 lm->control_fd_update = cb;
400}
401
402int
403memif_init (memif_control_fd_update_t * on_control_fd_update, char *app_name)
404{
405 int err = MEMIF_ERR_SUCCESS; /* 0 */
406 libmemif_main_t *lm = &libmemif_main;
407
408 if (app_name)
409 {
410 lm->app_name = malloc (strlen (app_name) + sizeof (char));
411 memset (lm->app_name, 0, strlen (app_name) + sizeof (char));
412 strncpy ((char *) lm->app_name, app_name, strlen (app_name));
413 }
414 else
415 {
416 lm->app_name = malloc (strlen (MEMIF_DEFAULT_APP_NAME) + sizeof (char));
417 memset (lm->app_name, 0, strlen (app_name) + sizeof (char));
418 strncpy ((char *) lm->app_name, MEMIF_DEFAULT_APP_NAME,
419 strlen (MEMIF_DEFAULT_APP_NAME));
420 }
421
422 /* register control fd update callback */
423 if (on_control_fd_update != NULL)
424 memif_control_fd_update_register (on_control_fd_update);
425 else
426 {
427 memif_epfd = epoll_create (1);
428 memif_control_fd_update_register (memif_control_fd_update);
Milan Lenco0a47c992017-10-12 14:19:31 +0200429 if ((poll_cancel_fd = eventfd (0, EFD_NONBLOCK)) < 0)
430 {
431 err = errno;
432 DBG ("eventfd: %s", strerror (err));
433 return memif_syscall_error_handler (err);
434 }
435 lm->control_fd_update (poll_cancel_fd, MEMIF_FD_EVENT_READ);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200436 DBG ("libmemif event polling initialized");
437 }
438
439 memset (&lm->ms, 0, sizeof (memif_socket_t));
440
441 lm->control_list_len = 2;
442 lm->interrupt_list_len = 2;
443 lm->listener_list_len = 1;
444 lm->pending_list_len = 1;
445
446 lm->control_list =
447 malloc (sizeof (memif_list_elt_t) * lm->control_list_len);
448 lm->interrupt_list =
449 malloc (sizeof (memif_list_elt_t) * lm->interrupt_list_len);
450 lm->listener_list =
451 malloc (sizeof (memif_list_elt_t) * lm->listener_list_len);
452 lm->pending_list =
453 malloc (sizeof (memif_list_elt_t) * lm->pending_list_len);
454
455 int i;
456 for (i = 0; i < lm->control_list_len; i++)
457 {
458 lm->control_list[i].key = -1;
459 lm->control_list[i].data_struct = NULL;
460 }
461 for (i = 0; i < lm->interrupt_list_len; i++)
462 {
463 lm->interrupt_list[i].key = -1;
464 lm->interrupt_list[i].data_struct = NULL;
465 }
466 for (i = 0; i < lm->listener_list_len; i++)
467 {
468 lm->listener_list[i].key = -1;
469 lm->listener_list[i].data_struct = NULL;
470 }
471 for (i = 0; i < lm->pending_list_len; i++)
472 {
473 lm->pending_list[i].key = -1;
474 lm->pending_list[i].data_struct = NULL;
475 }
476
477 lm->disconn_slaves = 0;
478
479 lm->timerfd = timerfd_create (CLOCK_REALTIME, TFD_NONBLOCK);
480 if (lm->timerfd < 0)
481 {
482 err = errno;
483 DBG ("timerfd: %s", strerror (err));
484 return memif_syscall_error_handler (err);
485 }
486
487 lm->arm.it_value.tv_sec = 2;
488 lm->arm.it_value.tv_nsec = 0;
489 lm->arm.it_interval.tv_sec = 2;
490 lm->arm.it_interval.tv_nsec = 0;
491 memset (&lm->disarm, 0, sizeof (lm->disarm));
492
493 if (lm->control_fd_update (lm->timerfd, MEMIF_FD_EVENT_READ) < 0)
494 {
495 DBG ("callback type memif_control_fd_update_t error!");
496 return MEMIF_ERR_CB_FDUPDATE;
497 }
498
499 return 0;
500}
501
502static inline memif_ring_t *
503memif_get_ring (memif_connection_t * conn, memif_ring_type_t type,
504 uint16_t ring_num)
505{
506 if (&conn->regions[0] == NULL)
507 return NULL;
508 void *p = conn->regions[0].shm;
509 int ring_size =
510 sizeof (memif_ring_t) +
511 sizeof (memif_desc_t) * (1 << conn->run_args.log2_ring_size);
512 p += (ring_num + type * conn->run_args.num_s2m_rings) * ring_size;
513
514 return (memif_ring_t *) p;
515}
516
517int
518memif_set_rx_mode (memif_conn_handle_t c, memif_rx_mode_t rx_mode,
519 uint16_t qid)
520{
521 memif_connection_t *conn = (memif_connection_t *) c;
522 if (conn == NULL)
523 return MEMIF_ERR_NOCONN;
524 uint8_t num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +0200525 (conn->args.is_master) ? conn->run_args.num_s2m_rings : conn->run_args.
526 num_m2s_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200527 if (qid >= num)
528 return MEMIF_ERR_QID;
529
530 conn->rx_queues[qid].ring->flags = rx_mode;
531 DBG ("rx_mode flag: %u", conn->rx_queues[qid].ring->flags);
532 return MEMIF_ERR_SUCCESS;
533}
534
535int
536memif_create (memif_conn_handle_t * c, memif_conn_args_t * args,
537 memif_connection_update_t * on_connect,
538 memif_connection_update_t * on_disconnect,
539 memif_interrupt_t * on_interrupt, void *private_ctx)
540{
541 int err, i, index, sockfd = -1;
542 memif_list_elt_t list_elt;
543 memif_connection_t *conn = (memif_connection_t *) * c;
544 if (conn != NULL)
545 {
546 DBG ("This handle already points to existing memif.");
547 return MEMIF_ERR_CONN;
548 }
549 conn = (memif_connection_t *) malloc (sizeof (memif_connection_t));
550 if (conn == NULL)
551 {
552 err = memif_syscall_error_handler (errno);
553 goto error;
554 }
555 memset (conn, 0, sizeof (memif_connection_t));
556
557 libmemif_main_t *lm = &libmemif_main;
558
559 conn->args.interface_id = args->interface_id;
560
561 if (args->log2_ring_size == 0)
562 args->log2_ring_size = MEMIF_DEFAULT_LOG2_RING_SIZE;
563 if (args->buffer_size == 0)
564 args->buffer_size = MEMIF_DEFAULT_BUFFER_SIZE;
565 if (args->num_s2m_rings == 0)
566 args->num_s2m_rings = MEMIF_DEFAULT_TX_QUEUES;
567 if (args->num_m2s_rings == 0)
568 args->num_m2s_rings = MEMIF_DEFAULT_RX_QUEUES;
569
570 conn->args.num_s2m_rings = args->num_s2m_rings;
571 conn->args.num_m2s_rings = args->num_m2s_rings;
572 conn->args.buffer_size = args->buffer_size;
573 conn->args.log2_ring_size = args->log2_ring_size;
574 conn->args.is_master = args->is_master;
575 conn->args.mode = args->mode;
576 conn->msg_queue = NULL;
577 conn->regions = NULL;
578 conn->tx_queues = NULL;
579 conn->rx_queues = NULL;
580 conn->fd = -1;
581 conn->on_connect = on_connect;
582 conn->on_disconnect = on_disconnect;
583 conn->on_interrupt = on_interrupt;
584 conn->private_ctx = private_ctx;
585 memset (&conn->run_args, 0, sizeof (memif_conn_run_args_t));
586
587 uint8_t l = strlen ((char *) args->interface_name);
588 strncpy ((char *) conn->args.interface_name, (char *) args->interface_name,
589 l);
590
591 l = strlen ((char *) args->instance_name);
592 strncpy ((char *) conn->args.instance_name, (char *) args->instance_name,
593 l);
594
595 /* allocate and initialize socket_filename so it can be copyed to sun_path
596 without memory leaks */
597 conn->args.socket_filename = malloc (sizeof (char *) * 108);
598 memset (conn->args.socket_filename, 0, 108 * sizeof (char *));
599
600 if (args->socket_filename)
601 {
602 if (conn->args.socket_filename == NULL)
603 {
604 err = memif_syscall_error_handler (errno);
605 goto error;
606 }
607 strncpy ((char *) conn->args.socket_filename,
608 (char *) args->socket_filename,
609 strlen ((char *) args->socket_filename));
610 }
611 else
612 {
613 uint16_t sdl = strlen (MEMIF_DEFAULT_SOCKET_DIR);
614 uint16_t sfl = strlen (MEMIF_DEFAULT_SOCKET_FILENAME);
615 if (conn->args.socket_filename == NULL)
616 {
617 err = memif_syscall_error_handler (errno);
618 goto error;
619 }
620 strncpy ((char *) conn->args.socket_filename,
621 MEMIF_DEFAULT_SOCKET_DIR, sdl);
622 conn->args.socket_filename[sdl] = '/';
623 strncpy ((char *) (conn->args.socket_filename + 1 + sdl),
624 MEMIF_DEFAULT_SOCKET_FILENAME, sfl);
625 }
626
627 if (args->secret)
628 {
629 l = strlen ((char *) args->secret);
630 strncpy ((char *) conn->args.secret, (char *) args->secret, l);
631 }
632
633 if (conn->args.is_master)
634 {
635 conn->run_args.buffer_size = conn->args.buffer_size;
636 memif_socket_t *ms;
637 memif_list_elt_t elt;
638 for (i = 0; i < lm->listener_list_len; i++)
639 {
640 if ((ms =
641 (memif_socket_t *) lm->listener_list[i].data_struct) != NULL)
642 {
643 if (strncmp
644 ((char *) ms->filename, (char *) conn->args.socket_filename,
645 strlen ((char *) ms->filename)) == 0)
646 {
647 /* add interface to listener socket */
648 elt.key = conn->args.interface_id;
649 *c = elt.data_struct = conn;
650 add_list_elt (&elt, &ms->interface_list,
651 &ms->interface_list_len);
652 ms->use_count++;
653 conn->listener_fd = ms->fd;
654 break;
655 }
656 }
657 else
658 {
659 struct stat file_stat;
660 if (stat ((char *) conn->args.socket_filename, &file_stat) == 0)
661 {
662 if (S_ISSOCK (file_stat.st_mode))
663 unlink ((char *) conn->args.socket_filename);
664 else
665 return memif_syscall_error_handler (errno);
666 }
667 DBG ("creating socket file");
668 ms = malloc (sizeof (memif_socket_t));
Jakub Grajciarb467b2a2017-09-14 14:12:10 +0200669 ms->filename =
670 malloc (strlen ((char *) conn->args.socket_filename) +
671 sizeof (char));
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200672 memset (ms->filename, 0,
673 strlen ((char *) conn->args.socket_filename) +
674 sizeof (char));
675 strncpy ((char *) ms->filename,
676 (char *) conn->args.socket_filename,
677 strlen ((char *) conn->args.socket_filename));
678 ms->interface_list_len = 1;
679 ms->interface_list =
680 malloc (sizeof (memif_list_elt_t) * ms->interface_list_len);
681 ms->interface_list[0].key = -1;
682 ms->interface_list[0].data_struct = NULL;
683 struct sockaddr_un un = { 0 };
684 int on = 1;
685
686 ms->fd = socket (AF_UNIX, SOCK_SEQPACKET, 0);
687 if (ms->fd < 0)
688 {
689 err = memif_syscall_error_handler (errno);
690 goto error;
691 }
692 DBG ("socket %d created", ms->fd);
693 un.sun_family = AF_UNIX;
694 strncpy ((char *) un.sun_path, (char *) ms->filename,
695 sizeof (un.sun_path) - 1);
696 DBG ("sockopt");
697 if (setsockopt
698 (ms->fd, SOL_SOCKET, SO_PASSCRED, &on, sizeof (on)) < 0)
699 {
700 err = memif_syscall_error_handler (errno);
701 goto error;
702 }
703 DBG ("bind");
704 if (bind (ms->fd, (struct sockaddr *) &un, sizeof (un)) < 0)
705 {
706 err = memif_syscall_error_handler (errno);
707 goto error;
708 }
709 DBG ("listen");
710 if (listen (ms->fd, 1) < 0)
711 {
712 err = memif_syscall_error_handler (errno);
713 goto error;
714 }
715 DBG ("stat");
716 if (stat ((char *) ms->filename, &file_stat) < 0)
717 {
718 err = memif_syscall_error_handler (errno);
719 goto error;
720 }
721
722 /* add interface to listener socket */
723 elt.key = conn->args.interface_id;
724 *c = elt.data_struct = conn;
725 add_list_elt (&elt, &ms->interface_list,
726 &ms->interface_list_len);
727 ms->use_count = 1;
728 conn->listener_fd = ms->fd;
729
730 /* add listener socket to libmemif main */
731 elt.key = ms->fd;
732 elt.data_struct = ms;
733 add_list_elt (&elt, &lm->listener_list, &lm->listener_list_len);
734 lm->control_fd_update (ms->fd, MEMIF_FD_EVENT_READ);
735 break;
736 }
737 }
738 }
739 else
740 {
741 if (lm->disconn_slaves == 0)
742 {
743 if (timerfd_settime (lm->timerfd, 0, &lm->arm, NULL) < 0)
744 {
745 err = memif_syscall_error_handler (errno);
746 goto error;
747 }
748 }
749
750 lm->disconn_slaves++;
751
752 list_elt.key = -1;
753 *c = list_elt.data_struct = conn;
754 if ((index =
755 add_list_elt (&list_elt, &lm->control_list,
756 &lm->control_list_len)) < 0)
757 {
758 err = MEMIF_ERR_NOMEM;
759 goto error;
760 }
761 }
762
763 conn->index = index;
764
765 return 0;
766
767error:
768 if (sockfd > 0)
769 close (sockfd);
770 sockfd = -1;
771 if (conn->args.socket_filename)
772 free (conn->args.socket_filename);
773 if (conn != NULL)
774 free (conn);
775 *c = conn = NULL;
776 return err;
777}
778
779int
780memif_control_fd_handler (int fd, uint8_t events)
781{
782 int i, rv, sockfd = -1, err = MEMIF_ERR_SUCCESS; /* 0 */
783 uint16_t num;
784 memif_list_elt_t *e = NULL;
785 memif_connection_t *conn;
786 libmemif_main_t *lm = &libmemif_main;
787 if (fd == lm->timerfd)
788 {
789 uint64_t b;
790 ssize_t size;
791 size = read (fd, &b, sizeof (b));
792 for (i = 0; i < lm->control_list_len; i++)
793 {
794 if ((lm->control_list[i].key < 0)
795 && (lm->control_list[i].data_struct != NULL))
796 {
797 conn = lm->control_list[i].data_struct;
798 if (conn->args.is_master)
799 continue;
800
801 struct sockaddr_un sun;
802 sockfd = socket (AF_UNIX, SOCK_SEQPACKET, 0);
803 if (sockfd < 0)
804 {
805 err = memif_syscall_error_handler (errno);
806 goto error;
807 }
808
809 sun.sun_family = AF_UNIX;
810
811 strncpy (sun.sun_path, conn->args.socket_filename,
812 sizeof (sun.sun_path) - 1);
813
814 if (connect (sockfd, (struct sockaddr *) &sun,
815 sizeof (struct sockaddr_un)) == 0)
816 {
817 conn->fd = sockfd;
818 conn->read_fn = memif_conn_fd_read_ready;
819 conn->write_fn = memif_conn_fd_write_ready;
820 conn->error_fn = memif_conn_fd_error;
821
822 lm->control_list[conn->index].key = conn->fd;
823
824 lm->control_fd_update (sockfd,
825 MEMIF_FD_EVENT_READ |
826 MEMIF_FD_EVENT_WRITE);
827
828 lm->disconn_slaves--;
829 if (lm->disconn_slaves == 0)
830 {
831 if (timerfd_settime (lm->timerfd, 0, &lm->disarm, NULL)
832 < 0)
833 {
834 err = memif_syscall_error_handler (errno);
835 goto error;
836 }
837 }
838 }
839 else
840 {
841 err = memif_syscall_error_handler (errno);
842 goto error;
843 }
844 }
845 }
846 }
847 else
848 {
849 get_list_elt (&e, lm->interrupt_list, lm->interrupt_list_len, fd);
850 if (e != NULL)
851 {
852 if (((memif_connection_t *) e->data_struct)->on_interrupt != NULL)
853 {
854 num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +0200855 (((memif_connection_t *) e->data_struct)->args.
856 is_master) ? ((memif_connection_t *) e->data_struct)->
857 run_args.num_s2m_rings : ((memif_connection_t *) e->
858 data_struct)->run_args.
859 num_m2s_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200860 for (i = 0; i < num; i++)
861 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +0200862 if (((memif_connection_t *) e->data_struct)->rx_queues[i].
863 int_fd == fd)
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200864 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +0200865 ((memif_connection_t *) e->
866 data_struct)->on_interrupt ((void *) e->data_struct,
867 ((memif_connection_t *)
868 e->data_struct)->
869 private_ctx, i);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200870 return MEMIF_ERR_SUCCESS;
871 }
872 }
873 }
874 return MEMIF_ERR_SUCCESS;
875 }
876 get_list_elt (&e, lm->listener_list, lm->listener_list_len, fd);
877 if (e != NULL)
878 {
879 memif_conn_fd_accept_ready ((memif_socket_t *) e->data_struct);
880 return MEMIF_ERR_SUCCESS;
881 }
882
883 get_list_elt (&e, lm->pending_list, lm->pending_list_len, fd);
884 if (e != NULL)
885 {
886 memif_read_ready (fd);
887 return MEMIF_ERR_SUCCESS;
888 }
889
890 get_list_elt (&e, lm->control_list, lm->control_list_len, fd);
891 if (e != NULL)
892 {
893 if (events & MEMIF_FD_EVENT_READ)
894 {
895 err =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +0200896 ((memif_connection_t *) e->data_struct)->read_fn (e->
897 data_struct);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200898 if (err != MEMIF_ERR_SUCCESS)
899 return err;
900 }
901 if (events & MEMIF_FD_EVENT_WRITE)
902 {
903 err =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +0200904 ((memif_connection_t *) e->data_struct)->write_fn (e->
905 data_struct);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200906 if (err != MEMIF_ERR_SUCCESS)
907 return err;
908 }
909 if (events & MEMIF_FD_EVENT_ERROR)
910 {
911 err =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +0200912 ((memif_connection_t *) e->data_struct)->error_fn (e->
913 data_struct);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200914 if (err != MEMIF_ERR_SUCCESS)
915 return err;
916 }
917 }
918 }
919
920 return MEMIF_ERR_SUCCESS; /* 0 */
921
922error:
923 if (sockfd > 0)
924 close (sockfd);
925 sockfd = -1;
926 return err;
927}
928
929int
930memif_poll_event (int timeout)
931{
932 libmemif_main_t *lm = &libmemif_main;
933 memif_list_elt_t *elt;
934 struct epoll_event evt, *e;
935 int en = 0, err = MEMIF_ERR_SUCCESS, i = 0; /* 0 */
936 uint16_t num;
937 uint32_t events = 0;
Milan Lenco0a47c992017-10-12 14:19:31 +0200938 uint64_t counter = 0;
939 ssize_t r = 0;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200940 memset (&evt, 0, sizeof (evt));
941 evt.events = EPOLLIN | EPOLLOUT;
942 sigset_t sigset;
943 sigemptyset (&sigset);
944 en = epoll_pwait (memif_epfd, &evt, 1, timeout, &sigset);
945 if (en < 0)
946 {
Milan Lenco0a47c992017-10-12 14:19:31 +0200947 err = errno;
948 DBG ("epoll_pwait: %s", strerror (err));
949 return memif_syscall_error_handler (err);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200950 }
951 if (en > 0)
952 {
Milan Lenco0a47c992017-10-12 14:19:31 +0200953 if (evt.data.fd == poll_cancel_fd)
954 {
955 r = read (evt.data.fd, &counter, sizeof (counter));
956 return MEMIF_ERR_POLL_CANCEL;
957 }
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200958 if (evt.events & EPOLLIN)
959 events |= MEMIF_FD_EVENT_READ;
960 if (evt.events & EPOLLOUT)
961 events |= MEMIF_FD_EVENT_WRITE;
962 if (evt.events & EPOLLERR)
963 events |= MEMIF_FD_EVENT_ERROR;
964 err = memif_control_fd_handler (evt.data.fd, events);
965 return err;
966 }
967 return 0;
968}
969
Milan Lenco0a47c992017-10-12 14:19:31 +0200970int
971memif_cancel_poll_event ()
972{
973 uint64_t counter = 1;
974 ssize_t w = 0;
975
976 if (poll_cancel_fd == -1)
977 return 0;
978 w = write (poll_cancel_fd, &counter, sizeof (counter));
979 if (w < sizeof (counter))
980 return MEMIF_ERR_INT_WRITE;
981
982 return 0;
983}
984
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +0200985static void
986memif_msg_queue_free (memif_msg_queue_elt_t ** e)
987{
988 if (*e == NULL)
989 return;
990 memif_msg_queue_free (&(*e)->next);
991 free (*e);
992 *e = NULL;
993 return;
994}
995
996/* send disconnect msg and close interface */
997int
998memif_disconnect_internal (memif_connection_t * c)
999{
1000 if (c == NULL)
1001 {
1002 DBG ("no connection");
1003 return MEMIF_ERR_NOCONN;
1004 }
1005 uint16_t num;
1006 int err = MEMIF_ERR_SUCCESS, i; /* 0 */
1007 memif_queue_t *mq;
1008 libmemif_main_t *lm = &libmemif_main;
1009 memif_list_elt_t *e;
1010
1011 c->on_disconnect ((void *) c, c->private_ctx);
1012
1013 if (c->fd > 0)
1014 {
1015 memif_msg_send_disconnect (c->fd, "interface deleted", 0);
1016 lm->control_fd_update (c->fd, MEMIF_FD_EVENT_DEL);
1017 close (c->fd);
1018 }
1019 get_list_elt (&e, lm->control_list, lm->control_list_len, c->fd);
1020 if (e != NULL)
1021 {
1022 if (c->args.is_master)
1023 free_list_elt (lm->control_list, lm->control_list_len, c->fd);
1024 e->key = c->fd = -1;
1025 }
1026
1027 if (c->tx_queues != NULL)
1028 {
1029 num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001030 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
1031 num_s2m_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001032 for (i = 0; i < num; i++)
1033 {
1034 mq = &c->tx_queues[i];
1035 if (mq != NULL)
1036 {
1037 if (mq->int_fd > 0)
1038 close (mq->int_fd);
1039 free_list_elt (lm->interrupt_list, lm->interrupt_list_len,
1040 mq->int_fd);
1041 mq->int_fd = -1;
1042 }
1043 }
1044 free (c->tx_queues);
1045 c->tx_queues = NULL;
1046 }
1047
1048 if (c->rx_queues != NULL)
1049 {
1050 num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001051 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1052 num_m2s_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001053 for (i = 0; i < num; i++)
1054 {
1055 mq = &c->rx_queues[i];
1056 if (mq != NULL)
1057 {
1058 if (mq->int_fd > 0)
1059 {
1060 if (c->on_interrupt != NULL)
1061 lm->control_fd_update (mq->int_fd, MEMIF_FD_EVENT_DEL);
1062 close (mq->int_fd);
1063 }
1064 free_list_elt (lm->interrupt_list, lm->interrupt_list_len,
1065 mq->int_fd);
1066 mq->int_fd = -1;
1067 }
1068 }
1069 free (c->rx_queues);
1070 c->rx_queues = NULL;
1071 }
1072
1073 if (c->regions != NULL)
1074 {
1075 if (munmap (c->regions[0].shm, c->regions[0].region_size) < 0)
1076 return memif_syscall_error_handler (errno);
1077 if (c->regions[0].fd > 0)
1078 close (c->regions[0].fd);
1079 c->regions[0].fd = -1;
1080 free (c->regions);
1081 c->regions = NULL;
1082 }
1083
1084 memset (&c->run_args, 0, sizeof (memif_conn_run_args_t));
1085
1086 memif_msg_queue_free (&c->msg_queue);
1087
1088 if (!(c->args.is_master))
1089 {
1090 if (lm->disconn_slaves == 0)
1091 {
1092 if (timerfd_settime (lm->timerfd, 0, &lm->arm, NULL) < 0)
1093 {
1094 err = memif_syscall_error_handler (errno);
1095 DBG_UNIX ("timerfd_settime: arm");
1096 }
1097 }
1098 lm->disconn_slaves++;
1099 }
1100
1101 return err;
1102}
1103
1104int
1105memif_delete (memif_conn_handle_t * conn)
1106{
1107 memif_connection_t *c = (memif_connection_t *) * conn;
1108 if (c == NULL)
1109 {
1110 DBG ("no connection");
1111 return MEMIF_ERR_NOCONN;
1112 }
1113 libmemif_main_t *lm = &libmemif_main;
1114 memif_list_elt_t *e = NULL;
1115 memif_socket_t *ms = NULL;
1116
1117 int err = MEMIF_ERR_SUCCESS;
1118
1119 if (c->fd > 0)
1120 {
1121 DBG ("DISCONNECTING");
1122 err = memif_disconnect_internal (c);
1123 if (err == MEMIF_ERR_NOCONN)
1124 return err;
1125 }
1126
1127 free_list_elt_ctx (lm->control_list, lm->control_list_len, c);
1128
1129 if (c->args.is_master)
1130 {
1131 get_list_elt (&e, lm->listener_list, lm->listener_list_len,
1132 c->listener_fd);
1133 if (e != NULL)
1134 {
1135 ms = (memif_socket_t *) e->data_struct;
1136 ms->use_count--;
1137 free_list_elt (ms->interface_list, ms->interface_list_len,
1138 c->args.interface_id);
1139 if (ms->use_count <= 0)
1140 {
1141 lm->control_fd_update (c->listener_fd, MEMIF_FD_EVENT_DEL);
1142 free_list_elt (lm->listener_list, lm->listener_list_len,
1143 c->listener_fd);
1144 close (c->listener_fd);
1145 c->listener_fd = ms->fd = -1;
1146 free (ms->interface_list);
1147 ms->interface_list = NULL;
1148 free (ms->filename);
1149 ms->filename = NULL;
1150 free (ms);
1151 ms = NULL;
1152 }
1153 }
1154 }
1155 else
1156 {
1157 lm->disconn_slaves--;
1158 if (lm->disconn_slaves <= 0)
1159 {
1160 if (timerfd_settime (lm->timerfd, 0, &lm->disarm, NULL) < 0)
1161 {
1162 err = memif_syscall_error_handler (errno);
1163 DBG ("timerfd_settime: disarm");
1164 }
1165 }
1166 }
1167
1168 if (c->args.socket_filename)
1169 free (c->args.socket_filename);
1170 c->args.socket_filename = NULL;
1171
1172 free (c);
1173 c = NULL;
1174
1175 *conn = c;
1176 return err;
1177}
1178
1179int
1180memif_connect1 (memif_connection_t * c)
1181{
1182 libmemif_main_t *lm = &libmemif_main;
1183 memif_region_t *mr = c->regions;
1184 memif_queue_t *mq;
1185 int i;
1186 uint16_t num;
1187
1188 if (mr != NULL)
1189 {
1190 if (!mr->shm)
1191 {
1192 if (mr->fd < 0)
1193 return MEMIF_ERR_NO_SHMFD;
1194
1195 if ((mr->shm = mmap (NULL, mr->region_size, PROT_READ | PROT_WRITE,
1196 MAP_SHARED, mr->fd, 0)) == MAP_FAILED)
1197 {
1198 return memif_syscall_error_handler (errno);
1199 }
1200 }
1201 }
1202
1203 num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001204 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
1205 num_s2m_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001206 for (i = 0; i < num; i++)
1207 {
1208 mq = &c->tx_queues[i];
1209 if (mq != NULL)
1210 {
1211 mq->ring = c->regions[mq->region].shm + mq->offset;
1212 if (mq->ring->cookie != MEMIF_COOKIE)
1213 {
1214 DBG ("wrong cookie on tx ring %u", i);
1215 return MEMIF_ERR_COOKIE;
1216 }
1217 mq->ring->head = mq->ring->tail = mq->last_head = mq->alloc_bufs =
1218 0;
1219 }
1220 }
1221 num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001222 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1223 num_m2s_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001224 for (i = 0; i < num; i++)
1225 {
1226 mq = &c->rx_queues[i];
1227 if (mq != NULL)
1228 {
1229 mq->ring = c->regions[mq->region].shm + mq->offset;
1230 if (mq->ring->cookie != MEMIF_COOKIE)
1231 {
1232 DBG ("wrong cookie on rx ring %u", i);
1233 return MEMIF_ERR_COOKIE;
1234 }
1235 mq->ring->head = mq->ring->tail = mq->last_head = mq->alloc_bufs =
1236 0;
1237 }
1238 }
1239
1240 lm->control_fd_update (c->fd, MEMIF_FD_EVENT_READ | MEMIF_FD_EVENT_MOD);
1241
1242 return 0;
1243}
1244
1245int
1246memif_init_regions_and_queues (memif_connection_t * conn)
1247{
1248 memif_ring_t *ring = NULL;
1249 uint64_t buffer_offset;
1250 memif_region_t *r;
1251 int i, j;
1252 libmemif_main_t *lm = &libmemif_main;
1253 memif_list_elt_t e;
1254
1255 conn->regions = (memif_region_t *) malloc (sizeof (memif_region_t));
1256 if (conn->regions == NULL)
1257 return memif_syscall_error_handler (errno);
1258 r = conn->regions;
1259
1260 buffer_offset =
1261 (conn->run_args.num_s2m_rings +
1262 conn->run_args.num_m2s_rings) * (sizeof (memif_ring_t) +
1263 sizeof (memif_desc_t) *
1264 (1 << conn->run_args.log2_ring_size));
1265
1266 r->region_size = buffer_offset +
1267 conn->run_args.buffer_size * (1 << conn->run_args.log2_ring_size) *
1268 (conn->run_args.num_s2m_rings + conn->run_args.num_m2s_rings);
1269
1270 if ((r->fd = memfd_create ("memif region 0", MFD_ALLOW_SEALING)) == -1)
1271 return memif_syscall_error_handler (errno);
1272/*
1273 if ((fcntl (r->fd, F_ADD_SEALS, F_SEAL_SHRINK)) == -1)
1274 return memif_syscall_error_handler (errno);
1275*/
1276 if ((ftruncate (r->fd, r->region_size)) == -1)
1277 return memif_syscall_error_handler (errno);
1278
1279 if ((r->shm = mmap (NULL, r->region_size, PROT_READ | PROT_WRITE,
1280 MAP_SHARED, r->fd, 0)) == MAP_FAILED)
1281 return memif_syscall_error_handler (errno);
1282
1283 for (i = 0; i < conn->run_args.num_s2m_rings; i++)
1284 {
1285 ring = memif_get_ring (conn, MEMIF_RING_S2M, i);
1286 DBG ("RING: %p I: %d", ring, i);
1287 ring->head = ring->tail = 0;
1288 ring->cookie = MEMIF_COOKIE;
1289 ring->flags = 0;
1290 for (j = 0; j < (1 << conn->run_args.log2_ring_size); j++)
1291 {
1292 uint16_t slot = i * (1 << conn->run_args.log2_ring_size) + j;
1293 ring->desc[j].region = 0;
1294 ring->desc[j].offset = buffer_offset +
1295 (uint32_t) (slot * conn->run_args.buffer_size);
1296 ring->desc[j].buffer_length = conn->run_args.buffer_size;
1297 }
1298 }
1299 for (i = 0; i < conn->run_args.num_m2s_rings; i++)
1300 {
1301 ring = memif_get_ring (conn, MEMIF_RING_M2S, i);
1302 DBG ("RING: %p I: %d", ring, i);
1303 ring->head = ring->tail = 0;
1304 ring->cookie = MEMIF_COOKIE;
1305 ring->flags = 0;
1306 for (j = 0; j < (1 << conn->run_args.log2_ring_size); j++)
1307 {
1308 uint16_t slot =
1309 (i +
1310 conn->run_args.num_s2m_rings) *
1311 (1 << conn->run_args.log2_ring_size) + j;
1312 ring->desc[j].region = 0;
1313 ring->desc[j].offset = buffer_offset +
1314 (uint32_t) (slot * conn->run_args.buffer_size);
1315 ring->desc[j].buffer_length = conn->run_args.buffer_size;
1316 }
1317 }
1318 memif_queue_t *mq;
1319 mq =
1320 (memif_queue_t *) malloc (sizeof (memif_queue_t) *
1321 conn->run_args.num_s2m_rings);
1322 if (mq == NULL)
1323 return memif_syscall_error_handler (errno);
1324 int x;
1325 for (x = 0; x < conn->run_args.num_s2m_rings; x++)
1326 {
1327 if ((mq[x].int_fd = eventfd (0, EFD_NONBLOCK)) < 0)
1328 return memif_syscall_error_handler (errno);
1329 /* add int fd to interrupt fd list */
1330 e.key = mq[x].int_fd;
1331 e.data_struct = conn;
1332 add_list_elt (&e, &lm->interrupt_list, &lm->interrupt_list_len);
1333
1334 mq[x].ring = memif_get_ring (conn, MEMIF_RING_S2M, x);
1335 DBG ("RING: %p I: %d", mq[x].ring, x);
1336 mq[x].log2_ring_size = conn->run_args.log2_ring_size;
1337 mq[x].region = 0;
1338 mq[x].offset =
1339 (void *) mq[x].ring - (void *) conn->regions[mq->region].shm;
1340 mq[x].last_head = 0;
1341 mq[x].alloc_bufs = 0;
1342 }
1343 conn->tx_queues = mq;
1344
1345 mq =
1346 (memif_queue_t *) malloc (sizeof (memif_queue_t) *
1347 conn->run_args.num_m2s_rings);
1348 if (mq == NULL)
1349 return memif_syscall_error_handler (errno);
1350 for (x = 0; x < conn->run_args.num_m2s_rings; x++)
1351 {
1352 if ((mq[x].int_fd = eventfd (0, EFD_NONBLOCK)) < 0)
1353 return memif_syscall_error_handler (errno);
1354 /* add int fd to interrupt fd list */
1355 e.key = mq[x].int_fd;
1356 e.data_struct = conn;
1357 add_list_elt (&e, &lm->interrupt_list, &lm->interrupt_list_len);
1358
1359 mq[x].ring = memif_get_ring (conn, MEMIF_RING_M2S, x);
1360 DBG ("RING: %p I: %d", mq[x].ring, x);
1361 mq[x].log2_ring_size = conn->run_args.log2_ring_size;
1362 mq[x].region = 0;
1363 mq[x].offset =
1364 (void *) mq[x].ring - (void *) conn->regions[mq->region].shm;
1365 mq[x].last_head = 0;
1366 mq[x].alloc_bufs = 0;
1367 }
1368 conn->rx_queues = mq;
1369
1370 return 0;
1371}
1372
1373int
1374memif_buffer_alloc (memif_conn_handle_t conn, uint16_t qid,
1375 memif_buffer_t * bufs, uint16_t count,
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001376 uint16_t * count_out, uint16_t size)
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001377{
1378 memif_connection_t *c = (memif_connection_t *) conn;
1379 if (c == NULL)
1380 return MEMIF_ERR_NOCONN;
1381 if (c->fd < 0)
1382 return MEMIF_ERR_DISCONNECTED;
1383 uint8_t num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001384 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
1385 num_s2m_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001386 if (qid >= num)
1387 return MEMIF_ERR_QID;
1388 memif_queue_t *mq = &c->tx_queues[qid];
1389 memif_ring_t *ring = mq->ring;
1390 memif_buffer_t *b0, *b1;
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001391 uint8_t chain_buf = 1;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001392 uint16_t mask = (1 << mq->log2_ring_size) - 1;
1393 uint16_t s0, s1, ns;
1394 *count_out = 0;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001395 int i, err = MEMIF_ERR_SUCCESS; /* 0 */
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001396
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001397 if (ring->tail > ring->head)
1398 ns = ring->tail - ring->head;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001399 else
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001400 ns = (1 << mq->log2_ring_size) - ring->head + ring->tail;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001401
1402 /* (head == tail) ? receive function will asume that no packets are available */
1403 ns -= 1;
1404
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001405 if (size > ring->desc[0].buffer_length)
1406 {
1407 chain_buf = size / ring->desc[s0].buffer_length;
1408 if (((size % ring->desc[s0].buffer_length) != 0) || (size == 0))
1409 chain_buf++;
1410 }
1411
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001412 while (count && ns)
1413 {
1414 while ((count > 2) && (ns > 2))
1415 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001416 s0 = (ring->head + mq->alloc_bufs) & mask;
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001417 s1 = (ring->head + mq->alloc_bufs + chain_buf) & mask;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001418
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001419 if ((2 * chain_buf) > ns)
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001420 break;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001421
1422 b0 = (bufs + *count_out);
1423 b1 = (bufs + *count_out + 1);
1424
1425 b0->desc_index = s0;
1426 b1->desc_index = s1;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001427 ring->desc[s0].flags = 0;
1428 ring->desc[s1].flags = 0;
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001429 b0->buffer_len = ring->desc[s0].buffer_length * chain_buf;
1430 b1->buffer_len = ring->desc[s1].buffer_length * chain_buf;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001431 /* TODO: support multiple regions -> ring descriptor contains region index */
1432 b0->data = c->regions->shm + ring->desc[s0].offset;
1433 b1->data = c->regions->shm + ring->desc[s1].offset;
1434
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001435 for (i = 0; i < (chain_buf - 1); i++)
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001436 {
1437 ring->desc[(s0 + i) & mask].flags |= MEMIF_DESC_FLAG_NEXT;
1438 ring->desc[(s1 + i) & mask].flags |= MEMIF_DESC_FLAG_NEXT;
1439 DBG ("allocating chained buffers");
1440 }
1441
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001442 mq->alloc_bufs += 2 * chain_buf;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001443
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001444 DBG ("allocated ring slots %u, %u", s0, s1);
1445 count -= 2;
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001446 ns -= (2 * chain_buf);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001447 *count_out += 2;
1448 }
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001449 s0 = (ring->head + mq->alloc_bufs) & mask;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001450
1451 b0 = (bufs + *count_out);
1452
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001453 if (chain_buf > ns)
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001454 break;
1455
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001456 b0->desc_index = s0;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001457 ring->desc[s0].flags = 0;
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001458 b0->buffer_len = ring->desc[s0].buffer_length * chain_buf;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001459 b0->data = c->regions->shm + ring->desc[s0].offset;
1460
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001461 for (i = 0; i < (chain_buf - 1); i++)
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001462 {
1463 ring->desc[(s0 + i) & mask].flags |= MEMIF_DESC_FLAG_NEXT;
1464 DBG ("allocating chained buffers");
1465 }
1466
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001467 mq->alloc_bufs += chain_buf;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001468
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001469 DBG ("allocated ring slot %u", s0);
1470 count--;
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001471 ns -= chain_buf;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001472 *count_out += 1;
1473 }
1474
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001475 DBG ("allocated: %u/%u bufs. Total %u allocated bufs", *count_out, count,
1476 mq->alloc_bufs);
1477
1478 if (count)
1479 {
1480 DBG ("ring buffer full! qid: %u", qid);
1481 err = MEMIF_ERR_NOBUF_RING;
1482 }
1483
1484 return err;
1485}
1486
1487int
1488memif_buffer_free (memif_conn_handle_t conn, uint16_t qid,
1489 memif_buffer_t * bufs, uint16_t count,
1490 uint16_t * count_out)
1491{
1492 memif_connection_t *c = (memif_connection_t *) conn;
1493 if (c == NULL)
1494 return MEMIF_ERR_NOCONN;
1495 if (c->fd < 0)
1496 return MEMIF_ERR_DISCONNECTED;
1497 uint8_t num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001498 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1499 num_m2s_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001500 if (qid >= num)
1501 return MEMIF_ERR_QID;
1502 libmemif_main_t *lm = &libmemif_main;
1503 memif_queue_t *mq = &c->rx_queues[qid];
1504 memif_ring_t *ring = mq->ring;
1505 uint16_t tail = ring->tail;
1506 uint16_t mask = (1 << mq->log2_ring_size) - 1;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001507 uint8_t chain_buf0, chain_buf1;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001508 memif_buffer_t *b0, *b1;
1509 *count_out = 0;
1510
1511 if (mq->alloc_bufs < count)
1512 count = mq->alloc_bufs;
1513
1514 while (count)
1515 {
1516 while (count > 2)
1517 {
1518 b0 = (bufs + *count_out);
1519 b1 = (bufs + *count_out + 1);
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001520 chain_buf0 =
1521 b0->buffer_len / ring->desc[b0->desc_index].buffer_length;
1522 if ((b0->buffer_len % ring->desc[b0->desc_index].buffer_length) !=
1523 0)
1524 chain_buf0++;
1525 chain_buf1 =
1526 b1->buffer_len / ring->desc[b1->desc_index].buffer_length;
1527 if ((b1->buffer_len % ring->desc[b1->desc_index].buffer_length) !=
1528 0)
1529 chain_buf1++;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001530 tail = (b1->desc_index + chain_buf1) & mask;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001531 b0->data = NULL;
1532 b1->data = NULL;
1533
1534 count -= 2;
1535 *count_out += 2;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001536 mq->alloc_bufs -= chain_buf0 + chain_buf1;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001537 }
1538 b0 = (bufs + *count_out);
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001539 chain_buf0 = b0->buffer_len / ring->desc[b0->desc_index].buffer_length;
1540 if ((b0->buffer_len % ring->desc[b0->desc_index].buffer_length) != 0)
1541 chain_buf0++;
1542 tail = (b0->desc_index + chain_buf0) & mask;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001543 b0->data = NULL;
1544
1545 count--;
1546 *count_out += 1;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001547 mq->alloc_bufs -= chain_buf0;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001548 }
Milan Lenco0a47c992017-10-12 14:19:31 +02001549 MEMIF_MEMORY_BARRIER ();
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001550 ring->tail = tail;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001551 DBG ("tail: %u", ring->tail);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001552
1553 return MEMIF_ERR_SUCCESS; /* 0 */
1554}
1555
1556int
1557memif_tx_burst (memif_conn_handle_t conn, uint16_t qid,
1558 memif_buffer_t * bufs, uint16_t count, uint16_t * tx)
1559{
1560 memif_connection_t *c = (memif_connection_t *) conn;
1561 if (c == NULL)
1562 return MEMIF_ERR_NOCONN;
1563 if (c->fd < 0)
1564 return MEMIF_ERR_DISCONNECTED;
1565 uint8_t num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001566 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
1567 num_s2m_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001568 if (qid >= num)
1569 return MEMIF_ERR_QID;
1570 memif_queue_t *mq = &c->tx_queues[qid];
1571 memif_ring_t *ring = mq->ring;
1572 uint16_t head = ring->head;
1573 uint16_t mask = (1 << mq->log2_ring_size) - 1;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001574 uint8_t chain_buf0, chain_buf1;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001575 *tx = 0;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001576 uint16_t curr_buf = 0;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001577 memif_buffer_t *b0, *b1;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001578 int i;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001579
1580 while (count)
1581 {
1582 while (count > 2)
1583 {
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001584 b0 = (bufs + curr_buf);
1585 b1 = (bufs + curr_buf + 1);
1586 chain_buf0 =
1587 b0->buffer_len / ring->desc[b0->desc_index].buffer_length;
1588 if ((b0->buffer_len % ring->desc[b0->desc_index].buffer_length) !=
1589 0)
1590 chain_buf0++;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001591
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001592 chain_buf1 =
1593 b1->buffer_len / ring->desc[b1->desc_index].buffer_length;
1594 if ((b1->buffer_len % ring->desc[b1->desc_index].buffer_length) !=
1595 0)
1596 chain_buf1++;
1597
1598 for (i = 0; i < memif_min (chain_buf0, chain_buf1); i++)
1599 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001600 /* b0 */
1601 if (b0->data_len >
1602 ring->desc[(b0->desc_index + i) & mask].buffer_length)
1603 {
1604 b0->data_len -=
1605 ring->desc[(b0->desc_index + i) & mask].length =
1606 ring->desc[(b0->desc_index + i) & mask].buffer_length;
1607 }
1608 else
1609 {
1610 ring->desc[(b0->desc_index + i) & mask].length =
1611 b0->data_len;
1612 b0->data_len = 0;
1613 }
1614 /* b1 */
1615 if (b1->data_len >
1616 ring->desc[(b1->desc_index + i) & mask].buffer_length)
1617 {
1618 b1->data_len -=
1619 ring->desc[(b1->desc_index + i) & mask].length =
1620 ring->desc[(b1->desc_index + i) & mask].buffer_length;
1621 }
1622 else
1623 {
1624 ring->desc[(b1->desc_index + i) & mask].length =
1625 b1->data_len;
1626 b1->data_len = 0;
1627 }
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001628#ifdef MEMIF_DBG_SHM
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001629 print_bytes (b0->data +
1630 ring->desc[(b0->desc_index +
1631 i) & mask].buffer_length *
1632 (chain_buf0 - 1),
1633 ring->desc[(b0->desc_index +
1634 i) & mask].buffer_length, DBG_TX_BUF);
1635 print_bytes (b1->data +
1636 ring->desc[(b1->desc_index +
1637 i) & mask].buffer_length *
1638 (chain_buf1 - 1),
1639 ring->desc[(b1->desc_index +
1640 i) & mask].buffer_length, DBG_TX_BUF);
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001641#endif /* MEMIF_DBG_SHM */
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001642 }
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001643
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001644 if (chain_buf0 > chain_buf1)
1645 {
1646 for (; i < chain_buf0; i++)
1647 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001648 if (b0->data_len >
1649 ring->desc[(b0->desc_index + i) & mask].buffer_length)
1650 {
1651 b0->data_len -=
1652 ring->desc[(b0->desc_index + i) & mask].length =
1653 ring->desc[(b0->desc_index + i) & mask].buffer_length;
1654 }
1655 else
1656 {
1657 ring->desc[(b0->desc_index + i) & mask].length =
1658 b0->data_len;
1659 b0->data_len = 0;
1660 }
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001661#ifdef MEMIF_DBG_SHM
1662 print_bytes (b0->data +
1663 ring->desc[(b0->desc_index +
1664 i) & mask].buffer_length *
1665 (chain_buf0 - 1),
1666 ring->desc[(b0->desc_index +
1667 i) & mask].buffer_length,
1668 DBG_TX_BUF);
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001669#endif /* MEMIF_DBG_SHM */
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001670 }
1671 }
1672 else
1673 {
1674 for (; i < chain_buf1; i++)
1675 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001676 if (b1->data_len >
1677 ring->desc[(b1->desc_index + i) & mask].buffer_length)
1678 {
1679 b1->data_len -=
1680 ring->desc[(b1->desc_index + i) & mask].length =
1681 ring->desc[(b1->desc_index + i) & mask].buffer_length;
1682 }
1683 else
1684 {
1685 ring->desc[(b1->desc_index + i) & mask].length =
1686 b1->data_len;
1687 b1->data_len = 0;
1688 }
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001689#ifdef MEMIF_DBG_SHM
1690 print_bytes (b1->data +
1691 ring->desc[(b1->desc_index +
1692 i) & mask].buffer_length *
1693 (chain_buf1 - 1),
1694 ring->desc[(b1->desc_index +
1695 i) & mask].buffer_length,
1696 DBG_TX_BUF);
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001697#endif /* MEMIF_DBG_SHM */
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001698 }
1699 }
1700
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001701 head = (b1->desc_index + chain_buf1) & mask;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001702
1703 b0->data = NULL;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001704#ifdef MEMIF_DBG
1705 if (b0->data_len != 0)
1706 DBG ("invalid b0 data length!");
1707#endif /* MEMIF_DBG */
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001708 b1->data = NULL;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001709#ifdef MEMIF_DBG
1710 if (b1->data_len != 0)
1711 DBG ("invalid b1 data length!");
1712#endif /* MEMIF_DBG */
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001713
1714 count -= 2;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001715 *tx += chain_buf0 + chain_buf1;
1716 curr_buf += 2;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001717 }
1718
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001719 b0 = (bufs + curr_buf);
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001720 chain_buf0 = b0->buffer_len / ring->desc[b0->desc_index].buffer_length;
1721 if ((b0->buffer_len % ring->desc[b0->desc_index].buffer_length) != 0)
1722 chain_buf0++;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001723
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001724 for (i = 0; i < chain_buf0; i++)
1725 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001726 if (b0->data_len >
1727 ring->desc[(b0->desc_index + i) & mask].buffer_length)
1728 {
1729 b0->data_len -= ring->desc[(b0->desc_index + i) & mask].length =
1730 ring->desc[(b0->desc_index + i) & mask].buffer_length;
1731 }
1732 else
1733 {
1734 ring->desc[(b0->desc_index + i) & mask].length = b0->data_len;
1735 b0->data_len = 0;
1736 }
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001737#ifdef MEMIF_DBG_SHM
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001738 print_bytes (b0->data +
1739 ring->desc[(b0->desc_index + i) & mask].buffer_length *
1740 (chain_buf0 - 1),
1741 ring->desc[(b0->desc_index + i) & mask].buffer_length,
1742 DBG_TX_BUF);
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001743#endif /* MEMIF_DBG_SHM */
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001744 }
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001745
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001746 head = (b0->desc_index + chain_buf0) & mask;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001747
1748 b0->data = NULL;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001749#ifdef MEMIF_DBG
1750 if (b0->data_len != 0)
1751 DBG ("invalid b0 data length!");
1752#endif /* MEMIF_DBG */
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001753
1754 count--;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001755 *tx += chain_buf0;
1756 curr_buf++;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001757 }
Milan Lenco0a47c992017-10-12 14:19:31 +02001758 MEMIF_MEMORY_BARRIER ();
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001759 ring->head = head;
1760
1761 mq->alloc_bufs -= *tx;
1762
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001763 /* TODO: return num of buffers and packets */
1764 *tx = curr_buf;
1765
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001766 if ((ring->flags & MEMIF_RING_FLAG_MASK_INT) == 0)
1767 {
1768 uint64_t a = 1;
1769 int r = write (mq->int_fd, &a, sizeof (a));
1770 if (r < 0)
1771 return MEMIF_ERR_INT_WRITE;
1772 }
1773
1774 return MEMIF_ERR_SUCCESS; /* 0 */
1775}
1776
1777int
1778memif_rx_burst (memif_conn_handle_t conn, uint16_t qid,
1779 memif_buffer_t * bufs, uint16_t count, uint16_t * rx)
1780{
1781 memif_connection_t *c = (memif_connection_t *) conn;
1782 if (c == NULL)
1783 return MEMIF_ERR_NOCONN;
1784 if (c->fd < 0)
1785 return MEMIF_ERR_DISCONNECTED;
1786 uint8_t num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001787 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
1788 num_m2s_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001789 if (qid >= num)
1790 return MEMIF_ERR_QID;
1791 memif_queue_t *mq = &c->rx_queues[qid];
1792 memif_ring_t *ring = mq->ring;
1793 uint16_t head = ring->head;
1794 uint16_t ns;
1795 uint16_t mask = (1 << mq->log2_ring_size) - 1;
1796 memif_buffer_t *b0, *b1;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001797 uint16_t curr_buf = 0;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001798 *rx = 0;
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001799#ifdef MEMIF_DBG_SHM
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001800 int i;
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001801#endif /* MEMIF_DBG_SHM */
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001802
1803 uint64_t b;
1804 ssize_t r = read (mq->int_fd, &b, sizeof (b));
1805 if ((r == -1) && (errno != EAGAIN))
1806 return memif_syscall_error_handler (errno);
1807
1808 if (head == mq->last_head)
1809 return 0;
1810
1811 if (head > mq->last_head)
1812 ns = head - mq->last_head;
1813 else
1814 ns = (1 << mq->log2_ring_size) - mq->last_head + head;
1815
1816 while (ns && count)
1817 {
1818 while ((ns > 2) && (count > 2))
1819 {
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001820 b0 = (bufs + curr_buf);
1821 b1 = (bufs + curr_buf + 1);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001822
1823 b0->desc_index = mq->last_head;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001824 b0->data = memif_get_buffer (conn, ring, mq->last_head);
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001825 b0->data_len = ring->desc[mq->last_head].length;
1826 b0->buffer_len = ring->desc[mq->last_head].buffer_length;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001827#ifdef MEMIF_DBG_SHM
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001828 i = 0;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001829 print_bytes (b0->data +
1830 ring->desc[b0->desc_index].buffer_length * i++,
1831 ring->desc[b0->desc_index].buffer_length, DBG_TX_BUF);
1832#endif /* MEMIF_DBG_SHM */
1833 ns--;
1834 *rx += 1;
1835 while (ring->desc[mq->last_head].flags & MEMIF_DESC_FLAG_NEXT)
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001836 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001837 ring->desc[mq->last_head].flags &= ~MEMIF_DESC_FLAG_NEXT;
1838 mq->last_head = (mq->last_head + 1) & mask;
1839 b0->data_len += ring->desc[mq->last_head].length;
1840 b0->buffer_len += ring->desc[mq->last_head].buffer_length;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001841#ifdef MEMIF_DBG_SHM
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001842 print_bytes (b0->data +
1843 ring->desc[b0->desc_index].buffer_length * i++,
1844 ring->desc[b0->desc_index].buffer_length,
1845 DBG_TX_BUF);
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001846#endif /* MEMIF_DBG_SHM */
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001847 ns--;
1848 *rx += 1;
1849 }
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001850 mq->last_head = (mq->last_head + 1) & mask;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001851
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001852 b1->desc_index = mq->last_head;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001853 b1->data = memif_get_buffer (conn, ring, mq->last_head);
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001854 b1->data_len = ring->desc[mq->last_head].length;
1855 b1->buffer_len = ring->desc[mq->last_head].buffer_length;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001856#ifdef MEMIF_DBG_SHM
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001857 i = 0;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001858 print_bytes (b1->data +
1859 ring->desc[b1->desc_index].buffer_length * i++,
1860 ring->desc[b1->desc_index].buffer_length, DBG_TX_BUF);
1861#endif /* MEMIF_DBG_SHM */
1862 ns--;
1863 *rx += 1;
1864 while (ring->desc[mq->last_head].flags & MEMIF_DESC_FLAG_NEXT)
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001865 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001866 ring->desc[mq->last_head].flags &= ~MEMIF_DESC_FLAG_NEXT;
1867 mq->last_head = (mq->last_head + 1) & mask;
1868 b1->data_len += ring->desc[mq->last_head].length;
1869 b1->buffer_len += ring->desc[mq->last_head].buffer_length;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001870#ifdef MEMIF_DBG_SHM
1871 print_bytes (b1->data +
1872 ring->desc[b1->desc_index].buffer_length * i++,
1873 ring->desc[b1->desc_index].buffer_length,
1874 DBG_TX_BUF);
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001875#endif /* MEMIF_DBG_SHM */
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001876 ns--;
1877 *rx += 1;
1878 }
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001879 mq->last_head = (mq->last_head + 1) & mask;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001880
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001881 count -= 2;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001882 curr_buf += 2;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001883 }
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001884 b0 = (bufs + curr_buf);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001885
1886 b0->desc_index = mq->last_head;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001887 b0->data = memif_get_buffer (conn, ring, mq->last_head);
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001888 b0->data_len = ring->desc[mq->last_head].length;
1889 b0->buffer_len = ring->desc[mq->last_head].buffer_length;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001890#ifdef MEMIF_DBG_SHM
Jakub Grajciar04b68bd2017-10-30 10:34:54 +01001891 i = 0;
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001892 print_bytes (b0->data +
1893 ring->desc[b0->desc_index].buffer_length * i++,
1894 ring->desc[b0->desc_index].buffer_length, DBG_TX_BUF);
1895#endif /* MEMIF_DBG_SHM */
1896 ns--;
1897 *rx += 1;
1898
1899 while (ring->desc[mq->last_head].flags & MEMIF_DESC_FLAG_NEXT)
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001900 {
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001901 ring->desc[mq->last_head].flags &= ~MEMIF_DESC_FLAG_NEXT;
1902 mq->last_head = (mq->last_head + 1) & mask;
1903 b0->data_len += ring->desc[mq->last_head].length;
1904 b0->buffer_len += ring->desc[mq->last_head].buffer_length;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001905#ifdef MEMIF_DBG_SHM
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001906 print_bytes (b0->data +
1907 ring->desc[b0->desc_index].buffer_length * i++,
1908 ring->desc[b0->desc_index].buffer_length, DBG_TX_BUF);
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001909#endif /* MEMIF_DBG_SHM */
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001910 ns--;
1911 *rx += 1;
1912 }
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02001913 mq->last_head = (mq->last_head + 1) & mask;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001914
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001915 count--;
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001916 curr_buf++;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001917 }
1918
1919 mq->alloc_bufs += *rx;
1920
Jakub Grajciarb467b2a2017-09-14 14:12:10 +02001921 /* TODO: return num of buffers and packets */
1922 *rx = curr_buf;
1923
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001924 if (ns)
1925 {
1926 DBG ("not enough buffers!");
1927 return MEMIF_ERR_NOBUF;
1928 }
1929
1930 return MEMIF_ERR_SUCCESS; /* 0 */
1931}
1932
1933int
1934memif_get_details (memif_conn_handle_t conn, memif_details_t * md,
1935 char *buf, ssize_t buflen)
1936{
1937 memif_connection_t *c = (memif_connection_t *) conn;
1938 if (c == NULL)
1939 return MEMIF_ERR_NOCONN;
1940
1941 int err = MEMIF_ERR_SUCCESS, i;
1942 ssize_t l0, l1, total_l;
1943 l0 = 0;
1944
1945 l1 = strlen ((char *) c->args.interface_name);
Milan Lenco0a47c992017-10-12 14:19:31 +02001946 if (l0 + l1 < buflen)
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001947 {
Milan Lenco0a47c992017-10-12 14:19:31 +02001948 md->if_name = strcpy (buf + l0, (char *) c->args.interface_name);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001949 l0 += l1 + 1;
1950 }
1951 else
1952 err = MEMIF_ERR_NOBUF_DET;
1953
1954 l1 = strlen ((char *) c->args.instance_name);
Milan Lenco0a47c992017-10-12 14:19:31 +02001955 if (l0 + l1 < buflen)
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001956 {
Milan Lenco0a47c992017-10-12 14:19:31 +02001957 md->inst_name = strcpy (buf + l0, (char *) c->args.instance_name);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001958 l0 += l1 + 1;
1959 }
1960 else
1961 err = MEMIF_ERR_NOBUF_DET;
1962
1963 l1 = strlen ((char *) c->remote_if_name);
Milan Lenco0a47c992017-10-12 14:19:31 +02001964 if (l0 + l1 < buflen)
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001965 {
Milan Lenco0a47c992017-10-12 14:19:31 +02001966 md->remote_if_name = strcpy (buf + l0, (char *) c->remote_if_name);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001967 l0 += l1 + 1;
1968 }
1969 else
1970 err = MEMIF_ERR_NOBUF_DET;
1971
1972 l1 = strlen ((char *) c->remote_name);
Milan Lenco0a47c992017-10-12 14:19:31 +02001973 if (l0 + l1 < buflen)
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001974 {
Milan Lenco0a47c992017-10-12 14:19:31 +02001975 md->remote_inst_name = strcpy (buf + l0, (char *) c->remote_name);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001976 l0 += l1 + 1;
1977 }
1978 else
1979 err = MEMIF_ERR_NOBUF_DET;
1980
1981 md->id = c->args.interface_id;
1982
1983 if (c->args.secret)
1984 {
1985 l1 = strlen ((char *) c->args.secret);
Milan Lenco0a47c992017-10-12 14:19:31 +02001986 if (l0 + l1 < buflen)
1987 {
1988 md->secret = strcpy (buf + l0, (char *) c->args.secret);
1989 l0 += l1 + 1;
1990 }
1991 else
1992 err = MEMIF_ERR_NOBUF_DET;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001993 }
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02001994
1995 md->role = (c->args.is_master) ? 0 : 1;
1996 md->mode = c->args.mode;
1997
1998 l1 = strlen ((char *) c->args.socket_filename);
Milan Lenco0a47c992017-10-12 14:19:31 +02001999 if (l0 + l1 < buflen)
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02002000 {
2001 md->socket_filename =
Milan Lenco0a47c992017-10-12 14:19:31 +02002002 strcpy (buf + l0, (char *) c->args.socket_filename);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02002003 l0 += l1 + 1;
2004 }
2005 else
2006 err = MEMIF_ERR_NOBUF_DET;
2007
2008 md->rx_queues_num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02002009 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
2010 num_m2s_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02002011
2012 l1 = sizeof (memif_queue_details_t) * md->rx_queues_num;
2013 if (l0 + l1 <= buflen)
2014 {
2015 md->rx_queues = (memif_queue_details_t *) buf + l0;
Milan Lenco0a47c992017-10-12 14:19:31 +02002016 l0 += l1;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02002017 }
2018 else
2019 err = MEMIF_ERR_NOBUF_DET;
2020
2021 for (i = 0; i < md->rx_queues_num; i++)
2022 {
2023 md->rx_queues[i].qid = i;
2024 md->rx_queues[i].ring_size = (1 << c->rx_queues[i].log2_ring_size);
2025 md->rx_queues[i].buffer_size = c->run_args.buffer_size;
2026 }
2027
2028 md->tx_queues_num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02002029 (c->args.is_master) ? c->run_args.num_m2s_rings : c->run_args.
2030 num_s2m_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02002031
2032 l1 = sizeof (memif_queue_details_t) * md->tx_queues_num;
2033 if (l0 + l1 <= buflen)
2034 {
2035 md->tx_queues = (memif_queue_details_t *) buf + l0;
Milan Lenco0a47c992017-10-12 14:19:31 +02002036 l0 += l1;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02002037 }
2038 else
2039 err = MEMIF_ERR_NOBUF_DET;
2040
2041 for (i = 0; i < md->tx_queues_num; i++)
2042 {
2043 md->tx_queues[i].qid = i;
2044 md->tx_queues[i].ring_size = (1 << c->tx_queues[i].log2_ring_size);
2045 md->tx_queues[i].buffer_size = c->run_args.buffer_size;
2046 }
2047
2048 md->link_up_down = (c->fd > 0) ? 1 : 0;
2049
2050 return err; /* 0 */
2051}
2052
2053int
2054memif_get_queue_efd (memif_conn_handle_t conn, uint16_t qid, int *efd)
2055{
2056 memif_connection_t *c = (memif_connection_t *) conn;
2057 *efd = -1;
2058 if (c == NULL)
2059 return MEMIF_ERR_NOCONN;
2060 if (c->fd < 0)
2061 return MEMIF_ERR_DISCONNECTED;
2062 uint8_t num =
Jakub Grajciarba3c4e82017-09-18 11:21:40 +02002063 (c->args.is_master) ? c->run_args.num_s2m_rings : c->run_args.
2064 num_m2s_rings;
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02002065 if (qid >= num)
2066 return MEMIF_ERR_QID;
2067
2068 *efd = c->rx_queues[qid].int_fd;
2069
2070 return MEMIF_ERR_SUCCESS;
2071}
2072
2073int
2074memif_cleanup ()
2075{
2076 libmemif_main_t *lm = &libmemif_main;
2077 if (lm->app_name)
2078 free (lm->app_name);
2079 lm->app_name = NULL;
2080 if (lm->control_list)
2081 free (lm->control_list);
2082 lm->control_list = NULL;
2083 if (lm->interrupt_list)
2084 free (lm->interrupt_list);
2085 lm->interrupt_list = NULL;
2086 if (lm->listener_list)
2087 free (lm->listener_list);
2088 lm->listener_list = NULL;
2089 if (lm->pending_list)
2090 free (lm->pending_list);
2091 lm->pending_list = NULL;
Milan Lenco0a47c992017-10-12 14:19:31 +02002092 if (poll_cancel_fd != -1)
2093 close (poll_cancel_fd);
Jakub Grajciar7c5c40d2017-08-30 10:13:25 +02002094
2095 return MEMIF_ERR_SUCCESS; /* 0 */
2096}