diff options
Diffstat (limited to 'extras/libmemif/examples/icmp_responder-mt_3-1/main.c')
| -rw-r--r-- | extras/libmemif/examples/icmp_responder-mt_3-1/main.c | 549 |
1 files changed, 549 insertions, 0 deletions
diff --git a/extras/libmemif/examples/icmp_responder-mt_3-1/main.c b/extras/libmemif/examples/icmp_responder-mt_3-1/main.c new file mode 100644 index 00000000000..a7b11bb1ebc --- /dev/null +++ b/extras/libmemif/examples/icmp_responder-mt_3-1/main.c @@ -0,0 +1,549 @@ +/* + *------------------------------------------------------------------ + * Copyright (c) 2019 Cisco and/or its affiliates. + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *------------------------------------------------------------------ + */ + +#include <stdlib.h> +#include <stdio.h> +#include <string.h> +#include <getopt.h> +#include <errno.h> +#include <inttypes.h> +#include <pthread.h> +#include <stdbool.h> +#include <unistd.h> + +#include <sys/epoll.h> +#include <sys/eventfd.h> + +#include <libmemif.h> +#include <icmp_proto.h> + + +#define APP_NAME "ICMP_Responder_mt_v3.1" +#define IF_NAME "memif_connection" + +#ifdef ICMP_DBG +#define DBG(...) do { \ + printf (APP_NAME":%s:%d: ", __func__, __LINE__); \ + printf (__VA_ARGS__); \ + printf ("\n"); \ + } while (0) +#else +#define DBG(...) +#endif + +#define ICMPR_BUFFER_LENGTH 32 +#define ICMPR_SOCKET_FILENAME_LEN 256 +#define ICMPR_MEMIF_BUFFER_NUM 256 + +static struct option options[] = { + {"threads", required_argument, 0, 't'}, + {"if_num", required_argument, 0, 'i'} +}; + +struct memif_connection +{ + uint16_t id; /* unique interface id */ + bool connected; /* is connected */ + struct per_thread_data *ptd; /* per thread data */ + memif_conn_handle_t handle; /* memif connection handle */ + uint8_t ip_addr[4]; /* ip4 address */ +}; + +struct per_thread_data +{ + bool running; /* is thread main loop running */ + uint8_t index; /* thread index */ + int epfd; /* epoll file descriptor */ + int pcfd; /* poll cancel file descriptor */ + uint16_t if_num; /* number of interfaces on this thread */ + struct memif_connection *conns; /* memif connections pool */ + memif_per_thread_main_handle_t pt_main; /* memif per thread main handle */ + memif_socket_handle_t socket_handle; /* memif socket handle */ +}; + +struct icmpr_main +{ + uint8_t threads; /* number of threads */ + uint16_t per_thread_if_num; /* number of interfaces per thread */ + struct per_thread_data *ptd; /* per thread data pool */ + pthread_t *pthread; /* thread pool */ +}; + +struct icmpr_main icmpr_main; + +int +add_epoll_fd (int epfd, int fd, uint32_t events) +{ + if (fd < 0) + { + DBG ("invalid fd %d", fd); + return -1; + } + struct epoll_event evt; + memset (&evt, 0, sizeof (evt)); + evt.events = events; + evt.data.fd = fd; + if (epoll_ctl (epfd, EPOLL_CTL_ADD, fd, &evt) < 0) + { + DBG ("epoll_ctl: %s fd %d", strerror (errno), fd); + return -1; + } + DBG ("fd %d added to epoll", fd); + return 0; +} + +int +mod_epoll_fd (int epfd, int fd, uint32_t events) +{ + if (fd < 0) + { + DBG ("invalid fd %d", fd); + return -1; + } + struct epoll_event evt; + memset (&evt, 0, sizeof (evt)); + evt.events = events; + evt.data.fd = fd; + if (epoll_ctl (epfd, EPOLL_CTL_MOD, fd, &evt) < 0) + { + DBG ("epoll_ctl: %s fd %d", strerror (errno), fd); + return -1; + } + DBG ("fd %d moddified on epoll", fd); + return 0; +} + +int +del_epoll_fd (int epfd, int fd) +{ + if (fd < 0) + { + DBG ("invalid fd %d", fd); + return -1; + } + struct epoll_event evt; + memset (&evt, 0, sizeof (evt)); + if (epoll_ctl (epfd, EPOLL_CTL_DEL, fd, &evt) < 0) + { + DBG ("epoll_ctl: %s fd %d", strerror (errno), fd); + return -1; + } + DBG ("fd %d removed from epoll", fd); + return 0; +} + +/* Called when libmemif requests an update on any of its file descriptors */ +static int +control_fd_update (int fd, uint8_t events, void *private_ctx) +{ + struct per_thread_data *ptd = (struct per_thread_data *) private_ctx; + uint32_t evt = 0; + + if (ptd == NULL) + return -1; + + /* convert memif event definitions to epoll events */ + if (events & MEMIF_FD_EVENT_DEL) + return del_epoll_fd (ptd->epfd, fd); + + if (events & MEMIF_FD_EVENT_READ) + evt |= EPOLLIN; + if (events & MEMIF_FD_EVENT_WRITE) + evt |= EPOLLOUT; + + if (events & MEMIF_FD_EVENT_MOD) + return mod_epoll_fd (ptd->epfd, fd, evt); + + return add_epoll_fd (ptd->epfd, fd, evt); +} + +static int +on_connect (memif_conn_handle_t conn, void *private_ctx) +{ + struct per_thread_data *ptd = (struct per_thread_data *) private_ctx; + struct memif_connection *c; + int i = 0; + + while (i < ptd->if_num && ptd->conns[i].handle != conn) + i++; + c = &ptd->conns[i]; + + c->connected = true; + DBG ("Connected: %u", c->id); + + memif_refill_queue (conn, 0, -1, 0); + + return 0; +} + +static int +on_disconnect (memif_conn_handle_t conn, void *private_ctx) +{ + struct per_thread_data *ptd = (struct per_thread_data *) private_ctx; + struct memif_connection *c; + int i = 0; + + while (i < ptd->if_num && ptd->conns[i].handle != conn) + i++; + c = &ptd->conns[i]; + + c->connected = false; + DBG ("Disconnected: %u", c->id); + + return 0; +} + +static int +on_interrupt (memif_conn_handle_t conn, void *private_ctx, uint16_t qid) +{ + struct per_thread_data *ptd = (struct per_thread_data *) private_ctx; + struct memif_connection *c; + memif_buffer_t mbufs[ICMPR_MEMIF_BUFFER_NUM]; + uint16_t rx = 0; + uint16_t tx = 0; + uint16_t ret; + memif_err_t err; + int i = 0; + + memset (mbufs, 0, sizeof (memif_buffer_t) * ICMPR_MEMIF_BUFFER_NUM); + + while (i < ptd->if_num && ptd->conns[i].handle != conn) + i++; + c = &ptd->conns[i]; + + /* receive data from shared memory buffers */ + err = memif_rx_burst (conn, qid, mbufs, ICMPR_MEMIF_BUFFER_NUM, &rx); + if (err != MEMIF_ERR_SUCCESS) + { + printf ("memif_rx_burst: %s\n", memif_strerror (err)); + goto error; + } + + /* resolve packet in place (zer-copy slave) */ + for (i = 0; i < rx; i++) + resolve_packet2 (mbufs[i].data, &mbufs[i].len, c->ip_addr); + + /* enqueue received buffers */ + err = memif_buffer_enq_tx (conn, qid, mbufs, i, &tx); + if (err != MEMIF_ERR_SUCCESS) + { + printf ("memif_rx_burst: %s\n", memif_strerror (err)); + goto error; + } + + /* mark shared memory buffers as free */ + err = memif_refill_queue (conn, qid, rx, 0); + if (err != MEMIF_ERR_SUCCESS) + { + printf ("memif_rx_burst: %s\n", memif_strerror (err)); + goto error; + } + + err = memif_tx_burst (conn, qid, mbufs, tx, &ret); + if (err != MEMIF_ERR_SUCCESS) + { + printf ("memif_rx_burst: %s\n", memif_strerror (err)); + goto error; + } + + return 0; + +error: + memif_refill_queue (conn, qid, -1, 0); + return -1; +} + +int +poll_event (memif_per_thread_main_handle_t pt_main, int pcfd, int epfd, + int timeout) +{ + struct epoll_event evt; + int en = 0; + uint8_t events = 0; + memset (&evt, 0, sizeof (evt)); + evt.events = EPOLLIN | EPOLLOUT; + + en = epoll_pwait (epfd, &evt, 1, timeout, NULL); + if (en < 0) + { + printf ("epoll_pwait: %s\n", strerror (errno)); + return -1; + } + + if (en > 0) + { + /* Cancel event polling */ + if (evt.data.fd == pcfd) + return 1; + + if (evt.events & EPOLLIN) + events |= MEMIF_FD_EVENT_READ; + if (evt.events & EPOLLOUT) + events |= MEMIF_FD_EVENT_WRITE; + if (evt.events & EPOLLERR) + events |= MEMIF_FD_EVENT_ERROR; + + /* No need to use locks, as the database is separated */ + memif_per_thread_control_fd_handler (pt_main, evt.data.fd, events); + } + + return 0; +} + +static void * +icmpr_thread_fn (void *data) +{ + struct per_thread_data *ptd = (struct per_thread_data *) data; + int rv; + uint16_t i; + char socket_filename[ICMPR_SOCKET_FILENAME_LEN] = "/run/vpp/memif"; + memif_conn_args_t args; + + ptd->epfd = epoll_create (1); + + ptd->conns = malloc (sizeof (struct memif_connection) * ptd->if_num); + if (ptd->conns == NULL) + { + printf ("%s\n", strerror (errno)); + return NULL; + } + + memset (ptd->conns, 0, sizeof (struct memif_connection) * ptd->if_num); + + /* Initialize memif database (per thread). */ + rv = + memif_per_thread_init (&ptd->pt_main, ptd, control_fd_update, APP_NAME, + NULL, NULL, NULL); + if (rv != MEMIF_ERR_SUCCESS) + { + printf ("memif_per_thread_init: %s\n", memif_strerror (rv)); + return NULL; + } + + /* Create unique socket. Each thread requires uniqueue socket. Interfaces created + * on the same thread can share one socket. + */ + socket_filename[strlen (socket_filename)] = '0' + ptd->index; + strncpy (socket_filename + strlen (socket_filename), ".sock", 5); + DBG ("socket_filename: %s", socket_filename); + + rv = memif_per_thread_create_socket (ptd->pt_main, &ptd->socket_handle, + socket_filename, ptd); + if (rv != MEMIF_ERR_SUCCESS) + { + printf ("memif_per_thread_create_socket: %s\n", memif_strerror (rv)); + return NULL; + } + + /* Create interfaces on this thread */ + for (i = 0; i < ptd->if_num; i++) + { + ptd->conns[i].ip_addr[0] = 192; + ptd->conns[i].ip_addr[1] = 168; + ptd->conns[i].ip_addr[2] = ptd->index + 1; + ptd->conns[i].ip_addr[3] = i * 2 + 2; + + memset (&args, 0, sizeof (args)); + + args.socket = ptd->socket_handle; + ptd->conns[i].id = i; + args.interface_id = i; + + rv = memif_create (&ptd->conns[i].handle, &args, on_connect, + on_disconnect, on_interrupt, ptd); + if (rv < 0) + { + printf ("%s\n", memif_strerror (rv)); + return NULL; + } + } + + /* Poll cancel file descriptor. When an event is received on this fd, exit thread + * loop in respective thread. + */ + ptd->pcfd = eventfd (0, EFD_NONBLOCK); + if (ptd->pcfd < 0) + { + printf ("eventfd: %s\n", strerror (errno)); + return NULL; + } + if (add_epoll_fd (ptd->epfd, ptd->pcfd, EPOLLIN) < 0) + { + printf ("Failed to add poll cancel fd to epfd."); + return NULL; + } + + /* Thread loop */ + ptd->running = true; + while (ptd->running) + { + rv = poll_event (ptd->pt_main, ptd->pcfd, ptd->epfd, -1); + if (rv != 0) + ptd->running = false; + } + + /* Clean up */ + for (i = 0; i < ptd->if_num; i++) + memif_delete (&ptd->conns[i].handle); + + memif_delete_socket (&ptd->socket_handle); + + memif_per_thread_cleanup (&ptd->pt_main); + + free (ptd->conns); + close (ptd->pcfd); + + return NULL; +} + +static void +icmpr_print_help () +{ + printf + ("exit - Exits the application.\nhelp - Print this help.\nshow - Show memif interfaces\n"); +} + +static void +icmpr_show_memifs () +{ + struct icmpr_main *im = &icmpr_main; + int i, j; + memif_socket_handle_t sh; + + printf ("%u Threads %u Memifs (per thread)\n", im->threads, + im->per_thread_if_num); + printf ("=================================\n"); + + for (i = 0; i < im->threads; i++) + { + sh = im->ptd[i].socket_handle; + printf ("Thread %u %s\n", i, memif_get_socket_filename (sh)); + for (j = 0; j < im->per_thread_if_num; j++) + { + printf ("\tMemif id %u\n\t%s\n", im->ptd[i].conns[j].id, + im->ptd[i].conns[j].connected ? "Link up" : "Link down"); + } + } +} + +int +main (int argc, char **argv) +{ + struct icmpr_main *im = &icmpr_main; + int rv, i; + int option_index = 0; + bool running; + char buffer[ICMPR_BUFFER_LENGTH]; + uint64_t b = 1; + + memset (im, 0, sizeof (struct icmpr_main)); + + /* Default args */ + im->threads = 4; + im->per_thread_if_num = 1; + + /* Parse args */ + while ((rv = + getopt_long (argc, argv, "t:i:", options, &option_index)) != (-1)) + { + switch (rv) + { + case 't': + im->threads = strtoul (optarg, NULL, 10); + break; + case 'i': + im->per_thread_if_num = strtoul (optarg, NULL, 10); + break; + default: + break; + } + } + + /* Check args */ + if (im->threads < 1) + { + printf ("threads < 1\n"); + exit (EXIT_FAILURE); + } + + if (im->per_thread_if_num < 1) + { + printf ("if_num < 1\n"); + exit (EXIT_FAILURE); + } + + /* Allocate memory */ + im->ptd = malloc (sizeof (struct per_thread_data) * im->threads); + if (im->ptd == NULL) + { + printf ("%s\n", strerror (errno)); + return -1; + } + im->pthread = malloc (sizeof (pthread_t) * im->threads); + if (im->pthread == NULL) + { + printf ("%s\n", strerror (errno)); + return -1; + } + + /* Initialize and create threads */ + for (i = 0; i < im->threads; i++) + { + im->ptd[i].index = i; + im->ptd[i].if_num = im->per_thread_if_num; + pthread_create (&im->pthread[i], NULL, icmpr_thread_fn, &im->ptd[i]); + } + + icmpr_print_help (); + + /* Main loop */ + running = true; + while (running) + { + printf ("cmd: "); + memset (buffer, 0, ICMPR_BUFFER_LENGTH); + if (fgets (buffer, ICMPR_BUFFER_LENGTH, stdin) != buffer) + { + printf ("%s\n", strerror (errno)); + running = false; + } + + if (strncmp (buffer, "exit", 4) == 0) + running = false; + else if (strncmp (buffer, "help", 4) == 0) + icmpr_print_help (); + else if (strncmp (buffer, "show", 4) == 0) + icmpr_show_memifs (); + } + + for (i = 0; i < im->threads; i++) + { + /* Stop polling */ + rv = write (im->ptd[i].pcfd, &b, sizeof (b)); + if (rv < 0) + { + printf ("Failed to cancel polling. %s\n", strerror (errno)); + exit (EXIT_FAILURE); + } + pthread_join (im->pthread[i], NULL); + } + + free (im->pthread); + free (im->ptd); + + return 0; +} |