mirror/xdp-project-bpf-examples
1
0
Fork 0
mirror of https://github.com/xdp-project/bpf-examples.git synced 2024-05-06 15:54:53 +00:00
Code Issues Projects Releases Wiki Activity

AF_XDP-interaction: Add common_params.c+h from XDP-tutorial

Browse Source 
Signed-off-by: Jesper Dangaard Brouer <brouer@redhat.com>
This commit is contained in:
Jesper Dangaard Brouer
2021-10-26 15:17:39 +02:00
parent e9002588ef
commit 118bb615b3
2 changed files with 220 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

198
AF_XDP-interaction/common_params.c Normal file
View File

@@ -0,0 +1,198 @@
#include <stddef.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <stdbool.h>
#include <getopt.h>
#include <errno.h>
#include <net/if.h>
#include <linux/if_link.h> /* XDP_FLAGS_* depend on kernel-headers installed */
#include <linux/if_xdp.h>
#include "common_params.h"
int verbose = 1;
#define BUFSIZE 30
void _print_options(const struct option_wrapper *long_options, bool required)
{
int i, pos;
char buf[BUFSIZE];
for (i = 0; long_options[i].option.name != 0; i++) {
if (long_options[i].required != required)
continue;
if (long_options[i].option.val > 64) /* ord('A') = 65 */
printf(" -%c,", long_options[i].option.val);
else
printf(" ");
pos = snprintf(buf, BUFSIZE, " --%s", long_options[i].option.name);
if (long_options[i].metavar)
snprintf(&buf[pos], BUFSIZE-pos, " %s", long_options[i].metavar);
printf("%-22s", buf);
printf(" %s", long_options[i].help);
printf("\n");
}
}
void usage(const char *prog_name, const char *doc,
const struct option_wrapper *long_options, bool full)
{
printf("Usage: %s [options]\n", prog_name);
if (!full) {
printf("Use --help (or -h) to see full option list.\n");
return;
}
printf("\nDOCUMENTATION:\n %s\n", doc);
printf("Required options:\n");
_print_options(long_options, true);
printf("\n");
printf("Other options:\n");
_print_options(long_options, false);
printf("\n");
}
int option_wrappers_to_options(const struct option_wrapper *wrapper,
struct option **options)
{
int i, num;
struct option *new_options;
for (i = 0; wrapper[i].option.name != 0; i++) {}
num = i;
new_options = malloc(sizeof(struct option) * num);
if (!new_options)
return -1;
for (i = 0; i < num; i++) {
memcpy(&new_options[i], &wrapper[i], sizeof(struct option));
}
*options = new_options;
return 0;
}
void parse_cmdline_args(int argc, char **argv,
const struct option_wrapper *options_wrapper,
struct config *cfg, const char *doc)
{
struct option *long_options;
bool full_help = false;
int longindex = 0;
char *dest;
int opt;
if (option_wrappers_to_options(options_wrapper, &long_options)) {
fprintf(stderr, "Unable to malloc()\n");
exit(EXIT_FAIL_OPTION);
}
/* Parse commands line args */
while ((opt = getopt_long(argc, argv, "hd:r:L:R:ASNFUMQ:czpq",
long_options, &longindex)) != -1) {
switch (opt) {
case 'd':
if (strlen(optarg) >= IF_NAMESIZE) {
fprintf(stderr, "ERR: --dev name too long\n");
goto error;
}
cfg->ifname = (char *)&cfg->ifname_buf;
strncpy(cfg->ifname, optarg, IF_NAMESIZE);
cfg->ifindex = if_nametoindex(cfg->ifname);
if (cfg->ifindex == 0) {
fprintf(stderr,
"ERR: --dev name unknown err(%d):%s\n",
errno, strerror(errno));
goto error;
}
break;
case 'r':
if (strlen(optarg) >= IF_NAMESIZE) {
fprintf(stderr, "ERR: --redirect-dev name too long\n");
goto error;
}
cfg->redirect_ifname = (char *)&cfg->redirect_ifname_buf;
strncpy(cfg->redirect_ifname, optarg, IF_NAMESIZE);
cfg->redirect_ifindex = if_nametoindex(cfg->redirect_ifname);
if (cfg->redirect_ifindex == 0) {
fprintf(stderr,
"ERR: --redirect-dev name unknown err(%d):%s\n",
errno, strerror(errno));
goto error;
}
break;
case 'A':
cfg->xdp_flags &= ~XDP_FLAGS_MODES; /* Clear flags */
break;
case 'S':
cfg->xdp_flags &= ~XDP_FLAGS_MODES; /* Clear flags */
cfg->xdp_flags |= XDP_FLAGS_SKB_MODE; /* Set flag */
cfg->xsk_bind_flags &= XDP_ZEROCOPY;
cfg->xsk_bind_flags |= XDP_COPY;
break;
case 'N':
cfg->xdp_flags &= ~XDP_FLAGS_MODES; /* Clear flags */
cfg->xdp_flags |= XDP_FLAGS_DRV_MODE; /* Set flag */
break;
case 3: /* --offload-mode */
cfg->xdp_flags &= ~XDP_FLAGS_MODES; /* Clear flags */
cfg->xdp_flags |= XDP_FLAGS_HW_MODE; /* Set flag */
break;
case 'F':
cfg->xdp_flags &= ~XDP_FLAGS_UPDATE_IF_NOEXIST;
break;
case 'M':
cfg->reuse_maps = true;
break;
case 'U':
cfg->do_unload = true;
break;
case 'p':
cfg->xsk_poll_mode = true;
break;
case 'q':
verbose = false;
break;
case 'Q':
cfg->xsk_if_queue = atoi(optarg);
break;
case 1: /* --filename */
dest = (char *)&cfg->filename;
strncpy(dest, optarg, sizeof(cfg->filename));
break;
case 2: /* --progsec */
dest = (char *)&cfg->progsec;
strncpy(dest, optarg, sizeof(cfg->progsec));
break;
case 'L': /* --src-mac */
dest = (char *)&cfg->src_mac;
strncpy(dest, optarg, sizeof(cfg->src_mac));
break;
case 'R': /* --dest-mac */
dest = (char *)&cfg->dest_mac;
strncpy(dest, optarg, sizeof(cfg->dest_mac));
break;
case 'c':
cfg->xsk_bind_flags &= XDP_ZEROCOPY;
cfg->xsk_bind_flags |= XDP_COPY;
break;
case 'z':
cfg->xsk_bind_flags &= XDP_COPY;
cfg->xsk_bind_flags |= XDP_ZEROCOPY;
break;
case 'h':
full_help = true;
/* fall-through */
error:
default:
usage(argv[0], doc, options_wrapper, full_help);
free(long_options);
exit(EXIT_FAIL_OPTION);
}
}
free(long_options);
}

22
AF_XDP-interaction/common_params.h Normal file
View File

@@ -0,0 +1,22 @@
/* This common_user.h is used by userspace programs */
#ifndef __COMMON_PARAMS_H
#define __COMMON_PARAMS_H
#include <getopt.h>
#include "common_defines.h"
struct option_wrapper {
struct option option;
char *help;
char *metavar;
bool required;
};
void usage(const char *prog_name, const char *doc,
const struct option_wrapper *long_options, bool full);
void parse_cmdline_args(int argc, char **argv,
const struct option_wrapper *long_options,
struct config *cfg, const char *doc);
#endif /* __COMMON_PARAMS_H */