examples/l3fwd/em_route_parse.c

/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright(c) 2022 Intel Corporation
 */

#include <stdio.h>
#include <stdint.h>
#include <errno.h>
#include <sys/socket.h>

#include "l3fwd.h"
#include "l3fwd_route.h"

static struct em_rule *em_route_base_v4;
static struct em_rule *em_route_base_v6;

enum {
    CB_FLD_DST_ADDR,
    CB_FLD_SRC_ADDR,
    CB_FLD_DST_PORT,
    CB_FLD_SRC_PORT,
    CB_FLD_PROTO,
    CB_FLD_IF_OUT,
    CB_FLD_MAX
};

static int
em_parse_v6_net(const char *in, uint8_t *v)
{
    int32_t rc;

    /* get address. */
    rc = inet_pton(AF_INET6, in, v);
    if (rc != 1)
        return -EINVAL;

    return 0;
}

static int
em_parse_v6_rule(char *str, struct em_rule *v)
{
    int i, rc;
    char *s, *sp, *in[CB_FLD_MAX];
    static const char *dlm = " \t\n";
    int dim = CB_FLD_MAX;
    s = str;

    for (i = 0; i != dim; i++, s = NULL) {
        in[i] = strtok_r(s, dlm, &sp);
        if (in[i] == NULL)
            return -EINVAL;
    }

    rc = em_parse_v6_net(in[CB_FLD_DST_ADDR], v->v6_key.ip_dst);
    if (rc != 0)
        return rc;
    rc = em_parse_v6_net(in[CB_FLD_SRC_ADDR], v->v6_key.ip_src);
    if (rc != 0)
        return rc;

    /* source port. */
    GET_CB_FIELD(in[CB_FLD_SRC_PORT], v->v6_key.port_src, 0, UINT16_MAX, 0);
    /* destination port. */
    GET_CB_FIELD(in[CB_FLD_DST_PORT], v->v6_key.port_dst, 0, UINT16_MAX, 0);
    /* protocol. */
    GET_CB_FIELD(in[CB_FLD_PROTO], v->v6_key.proto, 0, UINT8_MAX, 0);
    /* out interface. */
    GET_CB_FIELD(in[CB_FLD_IF_OUT], v->if_out, 0, UINT8_MAX, 0);

    return 0;
}

static int
em_parse_v4_rule(char *str, struct em_rule *v)
{
    int i, rc;
    char *s, *sp, *in[CB_FLD_MAX];
    static const char *dlm = " \t\n";
    int dim = CB_FLD_MAX;
    s = str;

    for (i = 0; i != dim; i++, s = NULL) {
        in[i] = strtok_r(s, dlm, &sp);
        if (in[i] == NULL)
            return -EINVAL;
    }

    rc = inet_pton(AF_INET, in[CB_FLD_DST_ADDR], &(v->v4_key.ip_dst));
    v->v4_key.ip_dst = ntohl(v->v4_key.ip_dst);
    if (rc != 1)
        return rc;

    rc = inet_pton(AF_INET, in[CB_FLD_SRC_ADDR], &(v->v4_key.ip_src));
    v->v4_key.ip_src = ntohl(v->v4_key.ip_src);
    if (rc != 1)
        return rc;

    /* source port. */
    GET_CB_FIELD(in[CB_FLD_SRC_PORT], v->v4_key.port_src, 0, UINT16_MAX, 0);
    /* destination port. */
    GET_CB_FIELD(in[CB_FLD_DST_PORT], v->v4_key.port_dst, 0, UINT16_MAX, 0);
    /* protocol. */
    GET_CB_FIELD(in[CB_FLD_PROTO], v->v4_key.proto, 0, UINT8_MAX, 0);
    /* out interface. */
    GET_CB_FIELD(in[CB_FLD_IF_OUT], v->if_out, 0, UINT8_MAX, 0);

    return 0;
}

static int
em_add_rules(const char *rule_path,
        struct em_rule **proute_base,
        int (*parser)(char *, struct em_rule *))
{
    struct em_rule *route_rules;
    struct em_rule *next;
    unsigned int route_num = 0;
    unsigned int route_cnt = 0;
    char buff[LINE_MAX];
    FILE *fh;
    unsigned int i = 0, rule_size = sizeof(*next);
    int val;

    *proute_base = NULL;
    fh = fopen(rule_path, "rb");
    if (fh == NULL)
        return -EINVAL;

    while ((fgets(buff, LINE_MAX, fh) != NULL)) {
        if (buff[0] == ROUTE_LEAD_CHAR)
            route_num++;
    }

    if (route_num == 0) {
        fclose(fh);
        return -EINVAL;
    }

    val = fseek(fh, 0, SEEK_SET);
    if (val < 0) {
        fclose(fh);
        return -EINVAL;
    }

    route_rules = calloc(route_num, rule_size);

    if (route_rules == NULL) {
        fclose(fh);
        return -EINVAL;
    }

    i = 0;
    while (fgets(buff, LINE_MAX, fh) != NULL) {
        i++;
        if (is_bypass_line(buff))
            continue;

        char s = buff[0];

        /* Route entry */
        if (s == ROUTE_LEAD_CHAR)
            next = &route_rules[route_cnt];

        /* Illegal line */
        else {
            RTE_LOG(ERR, L3FWD,
                "%s Line %u: should start with leading "
                "char %c\n",
                rule_path, i, ROUTE_LEAD_CHAR);
            fclose(fh);
            free(route_rules);
            return -EINVAL;
        }

        if (parser(buff + 1, next) != 0) {
            RTE_LOG(ERR, L3FWD,
                "%s Line %u: parse rules error\n",
                rule_path, i);
            fclose(fh);
            free(route_rules);
            return -EINVAL;
        }

        route_cnt++;
    }

    fclose(fh);

    *proute_base = route_rules;

    return route_cnt;
}

static int
em_add_default_v4_rules(void)
{
    /* populate the LPM IPv4 table */
    unsigned int i, rule_size = sizeof(*em_route_base_v4);
    route_num_v4 = RTE_DIM(ipv4_l3fwd_em_route_array);

    em_route_base_v4 = calloc(route_num_v4, rule_size);

    for (i = 0; i < (unsigned int)route_num_v4; i++) {
        em_route_base_v4[i].v4_key.ip_dst = ipv4_l3fwd_em_route_array[i].key.ip_dst;
        em_route_base_v4[i].v4_key.ip_src = ipv4_l3fwd_em_route_array[i].key.ip_src;
        em_route_base_v4[i].v4_key.port_dst = ipv4_l3fwd_em_route_array[i].key.port_dst;
        em_route_base_v4[i].v4_key.port_src = ipv4_l3fwd_em_route_array[i].key.port_src;
        em_route_base_v4[i].v4_key.proto = ipv4_l3fwd_em_route_array[i].key.proto;
        em_route_base_v4[i].if_out = ipv4_l3fwd_em_route_array[i].if_out;
    }
    return 0;
}

static int
em_add_default_v6_rules(void)
{
    /* populate the LPM IPv6 table */
    unsigned int i, rule_size = sizeof(*em_route_base_v6);
    route_num_v6 = RTE_DIM(ipv6_l3fwd_em_route_array);

    em_route_base_v6 = calloc(route_num_v6, rule_size);

    for (i = 0; i < (unsigned int)route_num_v6; i++) {
        memcpy(em_route_base_v6[i].v6_key.ip_dst, ipv6_l3fwd_em_route_array[i].key.ip_dst,
               sizeof(em_route_base_v6[i].v6_key.ip_dst));
        memcpy(em_route_base_v6[i].v6_key.ip_src, ipv6_l3fwd_em_route_array[i].key.ip_src,
               sizeof(em_route_base_v6[i].v6_key.ip_src));
        em_route_base_v6[i].v6_key.port_dst = ipv6_l3fwd_em_route_array[i].key.port_dst;
        em_route_base_v6[i].v6_key.port_src = ipv6_l3fwd_em_route_array[i].key.port_src;
        em_route_base_v6[i].v6_key.proto = ipv6_l3fwd_em_route_array[i].key.proto;
        em_route_base_v6[i].if_out = ipv6_l3fwd_em_route_array[i].if_out;
    }
    return 0;
}

void
em_free_routes(void)
{
    free(em_route_base_v4);
    free(em_route_base_v6);
    em_route_base_v4 = NULL;
    em_route_base_v6 = NULL;
    route_num_v4 = 0;
    route_num_v6 = 0;
}

/* Load rules from the input file */
void
read_config_files_em(void)
{
    /* ipv4 check */
    if (parm_config.rule_ipv4_name != NULL &&
            parm_config.rule_ipv6_name != NULL) {
        /* ipv4 check */
        route_num_v4 = em_add_rules(parm_config.rule_ipv4_name,
                    &em_route_base_v4, &em_parse_v4_rule);
        if (route_num_v4 < 0) {
            em_free_routes();
            rte_exit(EXIT_FAILURE, "Failed to add EM IPv4 rules\n");
        }

        /* ipv6 check */
        route_num_v6 = em_add_rules(parm_config.rule_ipv6_name,
                    &em_route_base_v6, &em_parse_v6_rule);
        if (route_num_v6 < 0) {
            em_free_routes();
            rte_exit(EXIT_FAILURE, "Failed to add EM IPv6 rules\n");
        }
    } else {
        RTE_LOG(INFO, L3FWD, "Missing 1 or more rule files, using default instead\n");
        if (em_add_default_v4_rules() < 0) {
            em_free_routes();
            rte_exit(EXIT_FAILURE, "Failed to add default IPv4 rules\n");
        }
        if (em_add_default_v6_rules() < 0) {
            em_free_routes();
            rte_exit(EXIT_FAILURE, "Failed to add default IPv6 rules\n");
        }
    }
}