tango-maat/scanner/ipport_matcher/ipport_matcher.cpp

/*
**********************************************************************************************
*	File: ipport_matcher.cpp
*   Description: 
*	Authors: Liu wentan <liuwentan@geedgenetworks.com>
*	Date:    2023-10-09
*   Copyright: (c) Since 2023 Geedge Networks, Ltd. All rights reserved.
***********************************************************************************************
*/

#include "uthash/utarray.h"
#include "uthash/uthash.h"
#include "maat_utils.h"
#include "maat_limits.h"
#include "ipport_matcher.h"

struct port_range {
    long long rule_id;
    void *tag;
    uint16_t min_port; /* host order */
    uint16_t max_port; /* host order */
};

struct ipport_node {
    int ip_type; //IPV4 or IPV6
    uint32_t ip_addr[4];
    UT_array *port_range_list; //array to store <struct port_range>
    UT_hash_handle hh;
};

struct ipport_matcher {
    struct ipport_node *ipport_hash;
};

UT_icd ut_port_range_icd = {sizeof(struct port_range), NULL, NULL, NULL};
static inline int compare_port_range_for_sort(const void *a, const void *b)
{
    struct port_range range_a = *(const struct port_range *)a;
    struct port_range range_b = *(const struct port_range *)b;

    int ret = range_a.min_port - range_b.min_port;
    if (0 == ret) {
        ret = range_a.max_port - range_b.max_port;
    }

    return ret;
}

static inline int compare_port_range_for_find(const void *a, const void *b)
{
    struct port_range range_a = *(const struct port_range *)a;
    struct port_range range_b = *(const struct port_range *)b;

    int ret = -1;
    if (range_a.min_port >= range_b.min_port &&
        range_a.min_port <= range_b.max_port) {
        ret = 0;
    } else if (range_a.max_port < range_b.min_port) {
        ret = -1;
    } else {
        ret = 1;
    }

    return ret;
}

struct ipport_matcher *ipport_matcher_new(struct ipport_rule *rules, size_t rule_num)
{
    if (NULL == rules || 0 == rule_num) {
        return NULL;
    }

    struct ipport_matcher *matcher = ALLOC(struct ipport_matcher, 1);
    struct ipport_node *node = NULL;
    char *key = NULL;
    size_t key_len = 0;

    for (size_t i = 0; i < rule_num; i++) {
        if (rules[i].ip.ip_type == IPV4) {
            key = (char *)&rules[i].ip.ipv4;
            key_len = 4;
        } else {
            key = (char *)rules[i].ip.ipv6;
            key_len = 16;
        }

        HASH_FIND(hh, matcher->ipport_hash, key, key_len, node);
        if (NULL == node) {
            node = ALLOC(struct ipport_node, 1);

            if (rules[i].ip.ip_type == IPV4) {
                node->ip_type = IPV4;
                node->ip_addr[0] = rules[i].ip.ipv4;
            } else {
                node->ip_type = IPV6;
                for (size_t j = 0; j < 4; j++) {
                    node->ip_addr[j] = rules[i].ip.ipv6[j];
                }
            }

            utarray_new(node->port_range_list, &ut_port_range_icd);
            HASH_ADD_KEYPTR(hh, matcher->ipport_hash, (char *)node->ip_addr, key_len, node);
        }
        
        struct port_range range;
        range.min_port = rules[i].min_port;
        range.max_port = rules[i].max_port;
        range.rule_id = rules[i].rule_id;
        range.tag = rules[i].user_tag;
        utarray_push_back(node->port_range_list, &range);
    }

    struct ipport_node *tmp_node = NULL;
    HASH_ITER(hh, matcher->ipport_hash, node, tmp_node) {
        utarray_sort(node->port_range_list, compare_port_range_for_sort);
    }

    return matcher;
}

int ipport_matcher_match(struct ipport_matcher *matcher, const struct ip_addr *ip_addr,
                         uint16_t port, struct ipport_result *result_array, size_t array_size)
{
    if (NULL == matcher || NULL == ip_addr || NULL == result_array || 0 == array_size) {
        return -1;
    }

    char *key = NULL;
    size_t key_len = 0;

    if (ip_addr->ip_type == IPV4) {
        key = (char *)&ip_addr->ipv4;
        key_len = 4;
    } else {
        key = (char *)ip_addr->ipv6;
        key_len = 16;
    }

    struct ipport_node *node = NULL;
    HASH_FIND(hh, matcher->ipport_hash, key, key_len, node);
    if (NULL == node) {
        return 0;
    }

    uint16_t host_port = ntohs(port);
    struct port_range range;
    range.min_port = host_port;
    range.max_port = host_port;

    struct port_range *tmp_range = NULL;
    tmp_range = (struct port_range *)utarray_find(node->port_range_list,
                                                  &range, compare_port_range_for_find);
    if (tmp_range != NULL) {
        result_array[0].rule_id = tmp_range->rule_id;
        result_array[0].tag = tmp_range->tag;
        return 1;
    }
    
    return 0;
}

void ipport_matcher_free(struct ipport_matcher *matcher)
{
    if (NULL == matcher) {
        return;
    }

    struct ipport_node *node = NULL, *tmp_node = NULL;
    HASH_ITER(hh, matcher->ipport_hash, node, tmp_node) {
        if (node->port_range_list != NULL) {
            utarray_free(node->port_range_list);
            node->port_range_list = NULL;
        }
        HASH_DEL(matcher->ipport_hash, node);

        FREE(node);
    }

    FREE(matcher);
}