/*
 *   BSD LICENSE
 *
 *   Copyright(c) 2016-2023 Intel Corporation. All rights reserved.
 *
 *   Redistribution and use in source and binary forms, with or without
 *   modification, are permitted provided that the following conditions
 *   are met:
 *
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in
 *       the documentation and/or other materials provided with the
 *       distribution.
 *     * Neither the name of Intel Corporation nor the names of its
 *       contributors may be used to endorse or promote products derived
 *       from this software without specific prior written permission.
 *
 *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "rdt.h"

#include "cpu.h"
#include "mba_sc.h"
#include "pqos.h"

#include <ctype.h>
#include <errno.h>
#include <inttypes.h>
#include <limits.h>
#include <sched.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

static const struct pqos_cap *m_cap = NULL;
static const struct pqos_cpuinfo *m_cpu = NULL;
static const struct pqos_capability *m_cap_l2ca = NULL;
static const struct pqos_capability *m_cap_l3ca = NULL;
static const struct pqos_capability *m_cap_mba = NULL;
static const struct pqos_capability *m_cap_smba = NULL;

/**
 * @brief Prints L2, L3 or MBA configuration in \a cfg
 *
 * @param [in] stream to write output to
 * @param [in] cfg COS configuration
 */
static void
rdt_cfg_print(FILE *stream, const struct rdt_cfg cfg)
{
        if (NULL == cfg.u.generic_ptr)
                return;

        switch (cfg.type) {
        case PQOS_CAP_TYPE_L2CA:
                if (cfg.u.l2->cdp == 1)
                        fprintf(stream, "code MASK: 0x%llx, data MASK: 0x%llx",
                                (unsigned long long)cfg.u.l2->u.s.code_mask,
                                (unsigned long long)cfg.u.l2->u.s.data_mask);
                else
                        fprintf(stream, "MASK: 0x%llx",
                                (unsigned long long)cfg.u.l2->u.ways_mask);
                break;

        case PQOS_CAP_TYPE_L3CA:
                if (cfg.u.l3->cdp == 1)
                        fprintf(stream, "code MASK: 0x%llx, data MASK: 0x%llx",
                                (unsigned long long)cfg.u.l3->u.s.code_mask,
                                (unsigned long long)cfg.u.l3->u.s.data_mask);
                else
                        fprintf(stream, "MASK: 0x%llx",
                                (unsigned long long)cfg.u.l3->u.ways_mask);
                break;

        case PQOS_CAP_TYPE_MBA:
                fprintf(stream, "RATE: %u", cfg.u.mba->mb_max);
                break;

        default:
                break;
        }
}
/**
 * @brief Gets short string representation of configuration type of \a cfg
 *
 * @param [in] cfg COS allocation configuration
 *
 * @return String representation of configuration type
 */
static const char *
rdt_cfg_get_type_str(const struct rdt_cfg cfg)
{
        if (NULL == cfg.u.generic_ptr)
                return "ERROR!";

        switch (cfg.type) {
        case PQOS_CAP_TYPE_L2CA:
                return "L2";

        case PQOS_CAP_TYPE_L3CA:
                return "L3";

        case PQOS_CAP_TYPE_MBA:
                return "MBA";

        default:
                return "ERROR!";
        }
}

/**
 * @brief Validates configuration in \a cfg
 *
 * @param [in] cfg COS configuration
 *
 * @return Result
 * @retval 1 on success
 * @retval 0 on failure
 */
static int
rdt_cfg_is_valid(const struct rdt_cfg cfg)
{
        if (cfg.u.generic_ptr == NULL)
                return 0;

        switch (cfg.type) {
        case PQOS_CAP_TYPE_L2CA:
                /* Validate L2 CAT configuration.
                 *  - If CDP is enabled code_mask and data_mask must be set
                 *  - If CDP is not enabled ways_mask must be set
                 */
                return cfg.u.l2 != NULL &&
                       ((cfg.u.l2->cdp == 1 && cfg.u.l2->u.s.code_mask != 0 &&
                         cfg.u.l2->u.s.data_mask != 0) ||
                        (cfg.u.l2->cdp == 0 && cfg.u.l2->u.ways_mask != 0));

        case PQOS_CAP_TYPE_L3CA:
                /* Validate L3 CAT configuration.
                 *  - If CDP is enabled code_mask and data_mask must be set
                 *  - If CDP is not enabled ways_mask must be set
                 */
                return cfg.u.l3 != NULL &&
                       ((cfg.u.l3->cdp == 1 && cfg.u.l3->u.s.code_mask != 0 &&
                         cfg.u.l3->u.s.data_mask != 0) ||
                        (cfg.u.l3->cdp == 0 && cfg.u.l3->u.ways_mask != 0));

        case PQOS_CAP_TYPE_MBA:
                return cfg.u.mba != NULL && cfg.u.mba->mb_max > 0;

        case PQOS_CAP_TYPE_SMBA:
                return cfg.u.smba != NULL && cfg.u.smba->mb_max > 0;

        default:
                break;
        }

        return 0;
}

/**
 * @brief Function to get the highest resource ID (socket/cluster)
 *
 * @param [in] technology used to determine if res ID should be l3cat_id or
 *             cluster
 * @param [out] max_res_id the highest possible resource ID
 *
 * @return Result
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
get_max_res_id(unsigned technology, unsigned *max_res_id)
{
        unsigned num_ids, max = 0, *ids = NULL;
        unsigned i;

        if (m_cpu == NULL || technology == 0)
                return -EFAULT;

        /* get number of L2IDs */
        if (technology & (1 << PQOS_CAP_TYPE_L2CA)) {
                ids = pqos_cpu_get_l2ids(m_cpu, &num_ids);
                if (ids == NULL)
                        return -EFAULT;

                for (i = 0; i < num_ids; i++)
                        max = ids[i] > max ? ids[i] : max;

                free(ids);
        }
        /* get number of l3cat_ids */
        if (technology & (1 << PQOS_CAP_TYPE_L3CA)) {
                ids = pqos_cpu_get_l3cat_ids(m_cpu, &num_ids);
                if (ids == NULL)
                        return -EFAULT;

                for (i = 0; i < num_ids; i++)
                        max = ids[i] > max ? ids[i] : max;

                free(ids);
        }

        /* get number of mba_ids */
        if (technology & (1 << PQOS_CAP_TYPE_MBA)) {
                ids = pqos_cpu_get_mba_ids(m_cpu, &num_ids);
                if (ids == NULL)
                        return -EFAULT;

                for (i = 0; i < num_ids; i++)
                        max = ids[i] > max ? ids[i] : max;

                free(ids);
        }

        /* get number of smba_ids */
        if (technology & (1 << PQOS_CAP_TYPE_SMBA)) {
                ids = pqos_cpu_get_smba_ids(m_cpu, &num_ids);
                if (ids == NULL)
                        return -EFAULT;

                for (i = 0; i < num_ids; i++)
                        max = ids[i] > max ? ids[i] : max;

                free(ids);
        }

        *max_res_id = max;
        return 0;
}

/**
 * @brief Converts string \a str to UINT64
 *
 * @param [in] str string
 * @param [in] base numerical base
 * @param [out] value parsed value
 *
 * @return number of parsed characters
 * @retval positive on success
 * @retval negative on error (-errno)
 */
static int
str_to_uint64(const char *str, int base, uint64_t *value)
{
        const char *str_start = str;
        char *str_end = NULL;
        unsigned long long val;

        if (NULL == str || NULL == value)
                return -EINVAL;

        while (isblank(*str_start))
                str_start++;

        if (base == 10 && !isdigit(*str_start))
                return -EINVAL;

        if (base == 16 && !isxdigit(*str_start))
                return -EINVAL;

        errno = 0;
        val = strtoull(str_start, &str_end, base);
        if (errno != 0 || str_end == NULL || *str_end != '\0')
                return -EINVAL;
        if (val > UINT64_MAX)
                return -EINVAL;

        *value = val;

        return str_end - str;
}

/**
 * @brief Parses CBM (Capacity Bit Mask) string \a cbm
 *        and stores result in \a mask and \a cmask
 *
 * @param [in] cbm CBM string
 * @param [in] cdp two masks separated with ','
 * @param [out] data_mask common (L2 or L3 non CDP) or data mask (L3 CDP)
 * @param [out] code_mask code mask (L3 CDP)
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
parse_mask_set(char *cbm, int *cdp, uint64_t *data_mask, uint64_t *code_mask)
{
        char *p = cbm;
        char *saveptr = NULL;
        uint64_t mask[2] = {0};
        unsigned mask_count = 0;

        /* skip brackets */
        if (*p == '(') {
                char *q = strchr(p, ')');

                if (q == NULL)
                        return -EINVAL;
                ++p;
                *q = '\0';
        }

        for (;; p = NULL) {
                int ret;
                char *token = NULL;

                token = strtok_r(p, ",", &saveptr);
                if (token == NULL)
                        break;

                /* more than 2 masks */
                if (mask_count > 1)
                        return -EINVAL;

                ret = str_to_uint64(token, 16, &mask[mask_count++]);
                if (ret < 0)
                        return -EINVAL;
        }

        *data_mask = mask[0];
        *code_mask = mask[1];
        *cdp = mask_count > 1;

        return 0;
}

/**
 * @brief This function displays cores association
 *
 * @param [in] cos_id class of service (COS) identifier
 * @param [in] core_num number of cores
 * @param [in] core_array array with cores identifiers
 */
static void
print_core_association(const unsigned int cos_id,
                       const unsigned int core_num,
                       const unsigned int *core_array)
{
        unsigned i;

        for (i = 0; i < core_num; i++)
                printf("Core %u => COS%u\n", core_array[i], cos_id);
}

/**
 * @brief This function displays process identifier (PID) association
 *
 * @param [in] cos_id class of service identifier (COS)
 * @param [in] pid_num number of process identifier (PID)
 * @param [in] pid_array array with process identifiers (PID)
 */
static void
print_pid_association(const unsigned int cos_id,
                      const unsigned int pid_num,
                      const pid_t *pid_array)
{
        unsigned i;

        for (i = 0; i < pid_num; i++)
                printf("PID %i => COS%u\n", pid_array[i], cos_id);
}

int
parse_reset(const char *cpustr)
{
        int ret = 0;
        const unsigned cpustr_len = strnlen(cpustr, MAX_OPTARG_LEN);

        if (cpustr_len == MAX_OPTARG_LEN)
                return -EINVAL;

        ret = str_to_cpuset(cpustr, cpustr_len, &g_cfg.reset_cpuset);
        return ret > 0 ? 0 : ret;
}

/**
 * @brief Parses CBMs string \a param and stores in \a ca
 *
 * @param [in] param CBMs string
 * @param [out] ca to store result
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
rdt_ca_str_to_cbm(char *param, struct rdt_cfg ca)
{
        uint64_t mask = 0, mask2 = 0;
        int cdp;
        int ret;

        if (!(PQOS_CAP_TYPE_L2CA == ca.type || PQOS_CAP_TYPE_L3CA == ca.type) ||
            NULL == ca.u.generic_ptr || NULL == param)
                return -EINVAL;

        ret = parse_mask_set(param, &cdp, &mask, &mask2);
        if (ret < 0)
                return -EINVAL;

        if (mask == 0)
                return -EINVAL;

        if (cdp && mask2 == 0)
                return -EINVAL;

        if (PQOS_CAP_TYPE_L2CA == ca.type) {
                if (cdp) {
                        ca.u.l2->cdp = 1;
                        ca.u.l2->u.s.data_mask = mask;
                        ca.u.l2->u.s.code_mask = mask2;
                } else
                        ca.u.l2->u.ways_mask = mask;
        } else {
                if (cdp) {
                        ca.u.l3->cdp = 1;
                        ca.u.l3->u.s.data_mask = mask;
                        ca.u.l3->u.s.code_mask = mask2;
                } else
                        ca.u.l3->u.ways_mask = mask;
        }

        return 0;
}

/**
 * @brief Parses rate string \a param and stores in \a mba
 *
 * @param [in] param rate string
 * @param [out] mba to store result
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
rdt_mba_str_to_rate(const char *param, struct rdt_cfg mba)
{
        uint64_t rate;
        int ret;

        if (PQOS_CAP_TYPE_MBA != mba.type || NULL == mba.u.generic_ptr ||
            NULL == param)
                return -EINVAL;

        ret = str_to_uint64(param, 10, &rate);
        if (ret < 0 || rate == 0)
                return -EINVAL;

        mba.u.mba->ctrl = 0;
        mba.u.mba->mb_max = rate;

        return 0;
}

/**
 * @brief Parses rate string \a param and stores in \a mba
 *
 * @param [in] param rate string
 * @param [out] smba to store result
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
rdt_smba_str_to_rate(const char *param, struct rdt_cfg smba)
{
        uint64_t rate;
        int ret;

        if (PQOS_CAP_TYPE_SMBA != smba.type || NULL == smba.u.generic_ptr ||
            NULL == param)
                return -EINVAL;

        ret = str_to_uint64(param, 10, &rate);
        if (ret < 0 || rate == 0)
                return -EINVAL;

        smba.u.smba->ctrl = 0;
        smba.u.smba->mb_max = rate;
        smba.u.smba->smba = 1;

        return 0;
}

/**
 * @brief Parses mbps string \a param and stores in \a mba
 *
 * @param [in] param mbps string
 * @param [out] mba to store result
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
rdt_mba_str_to_mbps(const char *param, struct rdt_cfg mba)
{
        uint64_t mbps;
        int ret;

        if (PQOS_CAP_TYPE_MBA != mba.type || NULL == mba.u.generic_ptr ||
            NULL == param)
                return -EINVAL;

        ret = str_to_uint64(param, 10, &mbps);
        if (ret < 0 || mbps == 0)
                return -EINVAL;

        mba.u.mba->ctrl = 1;
        mba.u.mba->mb_max = mbps;

        return 0;
}

/*
 * @brief Simplifies feature string
 *
 * @param [in] feature feature string
 *
 * @return simplified feature as char
 * @retval char representation of feature
 * @retval '?' on error
 */
static char
simplify_feature_str(const char *feature)
{
        static const struct {
                const char *feature_long;
                const char feature_char;
        } feature_lut[] = {
            /* clang-format off */
            {"cpu", 'c'},
            {"l2", '2'},
            {"l3", '3'},
            {"mba", 'm'},
            {"mba_max", 'b'},
            {"smba", 's'},
            {NULL, 0}
            /* clang-format on */
        };

        if (feature == NULL)
                return '?';

        if (strlen(feature) > 1) {
                unsigned i = 0;

                while (feature_lut[i].feature_long != NULL) {
                        if (strcmp(feature, feature_lut[i].feature_long) == 0)
                                return feature_lut[i].feature_char;

                        i++;
                }
        } else
                return feature[0];

        return '?';
}

int
parse_rdt(char *rdtstr)
{
        char *rdtstr_saveptr = NULL;
        char *group = NULL;
        unsigned idx = g_cfg.config_count;
        const struct rdt_cfg l2ca = wrap_l2ca(&g_cfg.config[idx].l2);
        const struct rdt_cfg l3ca = wrap_l3ca(&g_cfg.config[idx].l3);
        const struct rdt_cfg mba = wrap_mba(&g_cfg.config[idx].mba);
        const struct rdt_cfg smba = wrap_smba(&g_cfg.config[idx].smba);
        const unsigned min_len_group = strlen("3=f");
        int ret;

        if (rdtstr == NULL)
                return -EINVAL;

        group = strtok_r(rdtstr, ";", &rdtstr_saveptr);

        while (group != NULL) {
                char *group_saveptr = NULL;

                /* Min len check, 1 feature "3=f" */
                if (strlen(group) < min_len_group) {
                        fprintf(stderr, "Invalid option: \"%s\"\n", group);
                        return -EINVAL;
                }

                const char *feature = strtok_r(group, "=", &group_saveptr);
                char *param = strtok_r(NULL, "=", &group_saveptr);

                if (feature == NULL || param == NULL) {
                        fprintf(stderr, "Invalid option: \"%s\"\n", group);
                        return -EINVAL;
                }

                switch (simplify_feature_str(feature)) {
                case '2':
                        if (rdt_cfg_is_valid(l2ca))
                                return -EINVAL;

                        ret = rdt_ca_str_to_cbm(param, l2ca);
                        if (ret < 0)
                                return ret;
                        break;

                case '3':
                        if (rdt_cfg_is_valid(l3ca))
                                return -EINVAL;

                        ret = rdt_ca_str_to_cbm(param, l3ca);
                        if (ret < 0)
                                return ret;
                        break;

                case 'c':
                        if (CPU_COUNT(&g_cfg.config[idx].cpumask) != 0)
                                return -EINVAL;

                        ret = str_to_cpuset(param, strlen(param),
                                            &g_cfg.config[idx].cpumask);
                        if (ret <= 0)
                                return -EINVAL;

                        if (CPU_COUNT(&g_cfg.config[idx].cpumask) == 0)
                                return -EINVAL;
                        break;

                case 'm':
                        if (rdt_cfg_is_valid(mba))
                                return -EINVAL;

                        ret = rdt_mba_str_to_rate(param, mba);
                        if (ret < 0)
                                return ret;
                        break;

                case 'b':
                        if (rdt_cfg_is_valid(mba))
                                return -EINVAL;

                        ret = rdt_mba_str_to_mbps(param, mba);
                        if (ret < 0)
                                return ret;
                        break;

                case 's':
                        if (rdt_cfg_is_valid(smba))
                                return -EINVAL;

                        ret = rdt_smba_str_to_rate(param, smba);
                        if (ret < 0)
                                return ret;
                        break;

                default:
                        fprintf(stderr, "Invalid option: \"%s\"\n", feature);
                        return -EINVAL;
                }

                group = strtok_r(NULL, ";", &rdtstr_saveptr);
        }

        /* if no cpus specified then set pid flag */
        if (CPU_COUNT(&g_cfg.config[idx].cpumask) == 0)
                g_cfg.config[idx].pid_cfg = 1;

        if (!(rdt_cfg_is_valid(l2ca) || rdt_cfg_is_valid(l3ca) ||
              rdt_cfg_is_valid(mba) || rdt_cfg_is_valid(smba)))
                return -EINVAL;

        g_cfg.config_count++;

        return 0;
}

int
parse_iface(const char *iface)
{
        if (strcasecmp(iface, "auto") == 0)
                g_cfg.interface = PQOS_INTER_AUTO;
        else if (strcasecmp(iface, "msr") == 0)
                g_cfg.interface = PQOS_INTER_MSR;
        else if (strcasecmp(iface, "os") == 0)
                g_cfg.interface = PQOS_INTER_OS;
        else {
                fprintf(stderr, "Invalid interface: \"%s\"\n", iface);
                return -EINVAL;
        }

        return 0;
}

/**
 * @brief Checks if configured CPU sets are overlapping
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
check_cpus_overlapping(void)
{
        unsigned i = 0;

        for (i = 0; i < g_cfg.config_count; i++) {
                unsigned j = 0;

                for (j = i + 1; j < g_cfg.config_count; j++) {
                        unsigned overlapping = 0;
#ifdef __linux__
                        cpu_set_t mask;

                        CPU_AND(&mask, &g_cfg.config[i].cpumask,
                                &g_cfg.config[j].cpumask);
                        overlapping = CPU_COUNT(&mask) != 0;
#endif
#ifdef __FreeBSD__
                        overlapping = CPU_OVERLAP(&g_cfg.config[i].cpumask,
                                                  &g_cfg.config[j].cpumask);
#endif
                        if (1 == overlapping) {
                                fprintf(stderr, "Allocation: Requested CPUs "
                                                "sets are overlapping.\n");
                                return -EINVAL;
                        }
                }
        }

        return 0;
}

/**
 * @brief Checks if configured CPUs are valid and have no COS associated
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
check_cpus(void)
{
        unsigned i = 0;
        int ret = 0;

        for (i = 0; i < g_cfg.config_count; i++) {
                unsigned cpu_id = 0;

                for (cpu_id = 0; cpu_id < CPU_SETSIZE; cpu_id++) {
                        unsigned cos_id = 0;

                        if (CPU_ISSET(cpu_id, &g_cfg.config[i].cpumask) == 0)
                                continue;

                        ret = pqos_cpu_check_core(m_cpu, cpu_id);
                        if (ret != PQOS_RETVAL_OK) {
                                fprintf(stderr,
                                        "Allocation: %u is not a valid "
                                        "logical core id.\n",
                                        cpu_id);
                                return -ENODEV;
                        }

                        ret = pqos_alloc_assoc_get(cpu_id, &cos_id);
                        if (ret != PQOS_RETVAL_OK) {
                                fprintf(stderr,
                                        "Allocation: Failed to read "
                                        "cpu %u COS association.\n",
                                        cpu_id);
                                return -EFAULT;
                        }

                        /*
                         * Check if COS assigned to lcore is different
                         * then default one (#0)
                         */
                        if (cos_id != 0) {
                                fprintf(stderr,
                                        "Allocation: cpu %u has "
                                        "already associated COS#%u. Please "
                                        "reset allocation."
                                        "\n",
                                        cpu_id, cos_id);
                                return -EBUSY;
                        }
                }
        }

        return 0;
}

/**
 * @brief Checks if CPU supports requested L3 CDP configuration
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
check_l3cdp_support(void)
{
        unsigned i = 0;
        int ret;
        int l3cdp_supported = 0;
        int l3cdp_enabled = 0;

        ret = pqos_l3ca_cdp_enabled(m_cap, &l3cdp_supported, &l3cdp_enabled);
        if (ret != PQOS_RETVAL_OK && ret != PQOS_RETVAL_RESOURCE)
                return -1;

        if (l3cdp_enabled)
                return 0;

        for (i = 0; i < g_cfg.config_count; i++) {
                if (0 == g_cfg.config[i].l3.cdp)
                        continue;

                if (!l3cdp_supported)
                        fprintf(stderr, "Allocation: L3 CDP requested but not "
                                        "supported.\n");
                else
                        fprintf(stderr, "Allocation: L3 CDP requested but not "
                                        "enabled. Please enable L3 CDP.\n");

                return -ENOTSUP;
        }

        return 0;
}

/**
 * @brief Checks if CPU supports requested L2 CDP configuration
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
check_l2cdp_support(void)
{
        unsigned i = 0;
        int ret;

        int l2cdp_supported = 0;
        int l2cdp_enabled = 0;

        ret = pqos_l2ca_cdp_enabled(m_cap, &l2cdp_supported, &l2cdp_enabled);
        if (ret != PQOS_RETVAL_OK && ret != PQOS_RETVAL_RESOURCE)
                return -1;

        if (l2cdp_enabled)
                return 0;

        for (i = 0; i < g_cfg.config_count; i++) {
                if (0 == g_cfg.config[i].l2.cdp)
                        continue;

                if (!l2cdp_supported)
                        fprintf(stderr, "Allocation: L2 CDP requested but not "
                                        "supported.\n");
                else
                        fprintf(stderr, "Allocation: L2 CDP requested but not "
                                        "enabled. Please enable L2 CDP.\n");

                return -ENOTSUP;
        }

        return 0;
}

/**
 * @brief Checks if CPU supports requested MBA CTRL configuration
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
check_mba_ctrl_support(void)
{
        unsigned i = 0;
        int ret;
        int ctrl_supported = 0;
        int ctrl_enabled = 0;

        /* MBA Software controller is available */
        if (g_cfg.interface == PQOS_INTER_MSR)
                return 0;

        ret = pqos_mba_ctrl_enabled(m_cap, &ctrl_supported, &ctrl_enabled);
        if (ret != PQOS_RETVAL_OK && ret != PQOS_RETVAL_RESOURCE)
                return -1;

        if (ctrl_enabled != 0)
                return 0;

        for (i = 0; i < g_cfg.config_count; i++) {
                if (0 == g_cfg.config[i].mba.ctrl)
                        continue;

                if (ctrl_supported == 0)
                        fprintf(stderr, "Allocation: MBA CTRL requested but "
                                        "not supported.\n");
                else
                        fprintf(stderr,
                                "Allocation: MBA CTRL requested but "
                                "not enabled. Please enable MBA CTRL.\n");

                return -ENOTSUP;
        }

        return 0;
}

/**
 * @brief Checks if CAT/MBA configuration requested by a user via cmd line
 *        is supported by the system.
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
check_supported(void)
{
        unsigned i = 0;

        for (i = 0; i < g_cfg.config_count; i++) {
                if (rdt_cfg_is_valid(wrap_l3ca(&g_cfg.config[i].l3)) &&
                    NULL == m_cap_l3ca) {
                        fprintf(stderr, "Allocation: L3CA requested but not "
                                        "supported by system!\n");
                        return -ENOTSUP;
                }

                if (rdt_cfg_is_valid(wrap_l2ca(&g_cfg.config[i].l2)) &&
                    NULL == m_cap_l2ca) {
                        fprintf(stderr, "Allocation: L2CA requested but not "
                                        "supported by system!\n");
                        return -ENOTSUP;
                }

                if (rdt_cfg_is_valid(wrap_mba(&g_cfg.config[i].mba)) &&
                    NULL == m_cap_mba) {
                        fprintf(stderr, "Allocation: MBA requested but not "
                                        "supported by system!\n");
                        return -ENOTSUP;
                }
        }

        return 0;
}

/**
 * @brief Returns negation of max CBM for requested \a type (L2 or L3)
 *
 * @param [in] type type of pqos CAT capability, L2 or L3
 *
 * @return mask
 * @retval UINT64_MAX on error
 */
static uint64_t
get_not_cbm(const enum pqos_cap_type type)
{
        if (PQOS_CAP_TYPE_L2CA == type && NULL != m_cap_l2ca)
                return (UINT64_MAX << (m_cap_l2ca->u.l2ca->num_ways));
        else if (PQOS_CAP_TYPE_L3CA == type && NULL != m_cap_l3ca)
                return (UINT64_MAX << (m_cap_l3ca->u.l3ca->num_ways));
        else
                return UINT64_MAX;
}

/**
 * @brief Returns contention mask for requested \a type (L2 or L3)
 *
 * @param [in] type type of pqos CAT capability, L2 or L3
 *
 * @return contention mask
 * @retval UINT64_MAX on error
 *
 */
static uint64_t
get_contention_mask(const enum pqos_cap_type type)
{
        if (PQOS_CAP_TYPE_L2CA == type && NULL != m_cap_l2ca)
                return m_cap_l2ca->u.l2ca->way_contention;
        else if (PQOS_CAP_TYPE_L3CA == type && NULL != m_cap_l3ca)
                return m_cap_l3ca->u.l3ca->way_contention;
        else
                return UINT64_MAX;
}

/**
 * @brief Returns cumulative mask for \a ca CAT config
 *
 * For L2/L3 CDP config returns code_mask | data_mask,
 * for L2/L3 non-CDP config returns ways_mask
 *
 * @param [in] ca CAT COS configuration
 *
 * @return cumulative mask
 * @retval UINT64_MAX on error
 */
static uint64_t
rdt_ca_get_cumulative_cbm(const struct rdt_cfg ca)
{
        if (!(PQOS_CAP_TYPE_L2CA == ca.type || PQOS_CAP_TYPE_L3CA == ca.type) ||
            NULL == ca.u.generic_ptr || 0 == rdt_cfg_is_valid(ca))
                return UINT64_MAX;

        if (PQOS_CAP_TYPE_L2CA == ca.type) {
                if (ca.u.l2->cdp == 1)
                        return ca.u.l2->u.s.code_mask | ca.u.l2->u.s.data_mask;
                else
                        return ca.u.l2->u.ways_mask;
        }

        if (ca.u.l3->cdp == 1)
                return ca.u.l3->u.s.code_mask | ca.u.l3->u.s.data_mask;

        return ca.u.l3->u.ways_mask;
}

/**
 * @brief Checks if requested CBMs of \a type are supported by system.
 *
 * Warn if CBM overlaps contention mask
 *
 * @param [in] type type of pqos CAT capability, L2 or L3
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
check_cbm_len_and_contention(const enum pqos_cap_type type)
{
        unsigned i = 0;
        struct rdt_cfg ca;
        uint64_t mask = 0;

        if (!(PQOS_CAP_TYPE_L2CA == type || PQOS_CAP_TYPE_L3CA == type))
                return -EINVAL;

        const uint64_t not_cbm = get_not_cbm(type);
        const uint64_t contention_cbm = get_contention_mask(type);

        if (UINT64_MAX == not_cbm || UINT64_MAX == contention_cbm)
                return -EINVAL;

        for (i = 0; i < g_cfg.config_count; i++) {
                if (PQOS_CAP_TYPE_L2CA == type)
                        ca = wrap_l2ca(&g_cfg.config[i].l2);
                else
                        ca = wrap_l3ca(&g_cfg.config[i].l3);

                if (!rdt_cfg_is_valid(ca))
                        continue;

                mask = rdt_ca_get_cumulative_cbm(ca);
                if (UINT64_MAX == mask)
                        return -EFAULT;

                if ((mask & not_cbm) != 0) {
                        fprintf(stderr,
                                "CAT: One or more of requested %s CBMs "
                                "(",
                                rdt_cfg_get_type_str(ca));
                        rdt_cfg_print(stderr, ca);
                        fprintf(stderr, ") not supported by system "
                                        "(too long).\n");
                        return -ENOTSUP;
                }

                /* Just a note */
                if ((mask & contention_cbm) != 0) {
                        printf("CAT: One or more of requested %s CBMs "
                               "(",
                               rdt_cfg_get_type_str(ca));
                        rdt_cfg_print(stdout, ca);
                        printf(") overlap contention mask.\n");
                }
        }

        return 0;
}

/**
 * @brief Checks if requested CBMs (of all types) are supported by system
 *
 * Warn if CBM overlaps contention mask
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
check_cbm_len_and_contention_all(void)
{
        int ret;

        if (NULL != m_cap_l2ca) {
                ret = check_cbm_len_and_contention(PQOS_CAP_TYPE_L2CA);
                if (ret < 0)
                        return ret;
        }

        if (NULL != m_cap_l3ca) {
                ret = check_cbm_len_and_contention(PQOS_CAP_TYPE_L3CA);
                if (ret < 0)
                        return ret;
        }

        return 0;
}

/**
 * @brief Validates requested CAT/MBA configuration
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
alloc_validate(void)
{
        int ret = 0;

        ret = check_cpus();
        if (ret != 0)
                return ret;

        ret = check_supported();
        if (ret != 0)
                return ret;

        ret = check_l3cdp_support();
        if (ret != 0)
                return ret;

        ret = check_l2cdp_support();
        if (ret != 0)
                return ret;

        ret = check_mba_ctrl_support();
        if (ret != 0)
                return ret;

        ret = check_cbm_len_and_contention_all();
        if (ret != 0)
                return ret;

        ret = check_cpus_overlapping();
        if (ret != 0)
                return ret;

        return 0;
}

/**
 * @brief Gets cores from \a cores set which belong to \a l3cat_id
 *
 * @param [in] cores set of cores
 * @param [in] l3cat_id to get cores from
 * @param [out] core_num number of cores in \a core_array
 * @param [out] core_array array of cores
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
get_l3cat_id_cores(const cpu_set_t *cores,
                   const unsigned l3cat_id,
                   unsigned *core_num,
                   unsigned core_array[CPU_SETSIZE])
{
        unsigned i;

        if (cores == NULL || core_num == NULL || core_array == NULL)
                return -EINVAL;

        if (m_cpu == NULL)
                return -EFAULT;

        if (CPU_COUNT(cores) == 0) {
                *core_num = 0;
                return 0;
        }

        for (i = 0, *core_num = 0; i < m_cpu->num_cores; i++) {
                if (m_cpu->cores[i].l3cat_id != l3cat_id ||
                    0 == CPU_ISSET(m_cpu->cores[i].lcore, cores))
                        continue;

                core_array[(*core_num)++] = m_cpu->cores[i].lcore;
        }

        return 0;
}

/**
 * @brief Gets cores from \a cores set which belong to \a mab_id
 *
 * @param [in] cores set of cores
 * @param [in] mba_id to get cores from
 * @param [out] core_num number of cores in \a core_array
 * @param [out] core_array array of cores
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
get_mba_id_cores(const cpu_set_t *cores,
                 const unsigned mba_id,
                 unsigned *core_num,
                 unsigned core_array[CPU_SETSIZE])
{
        unsigned i;

        if (cores == NULL || core_num == NULL || core_array == NULL)
                return -EINVAL;

        if (m_cpu == NULL)
                return -EFAULT;

        if (CPU_COUNT(cores) == 0) {
                *core_num = 0;
                return 0;
        }

        for (i = 0, *core_num = 0; i < m_cpu->num_cores; i++) {
                if (m_cpu->cores[i].mba_id != mba_id ||
                    0 == CPU_ISSET(m_cpu->cores[i].lcore, cores))
                        continue;

                core_array[(*core_num)++] = m_cpu->cores[i].lcore;
        }

        return 0;
}

/**
 * @brief Gets cores from \a cores set which belong to \a smab_id
 *
 * @param [in] cores set of cores
 * @param [in] smba_id to get cores from
 * @param [out] core_num number of cores in \a core_array
 * @param [out] core_array array of cores
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
get_smba_id_cores(const cpu_set_t *cores,
                  const unsigned smba_id,
                  unsigned *core_num,
                  unsigned core_array[CPU_SETSIZE])
{
        unsigned i;

        if (cores == NULL || core_num == NULL || core_array == NULL)
                return -EINVAL;

        if (m_cpu == NULL)
                return -EFAULT;

        if (CPU_COUNT(cores) == 0) {
                *core_num = 0;
                return 0;
        }

        for (i = 0, *core_num = 0; i < m_cpu->num_cores; i++) {
                if (m_cpu->cores[i].smba_id != smba_id ||
                    0 == CPU_ISSET(m_cpu->cores[i].lcore, cores))
                        continue;

                core_array[(*core_num)++] = m_cpu->cores[i].lcore;
        }

        return 0;
}

/**
 * @brief Gets cores from \a cores set which belong to \a l2_id
 *
 * @param [in] cores set of cores
 * @param [in] l2_id L2 cluster ID to get cores from
 * @param [out] core_num number of cores in \a core_array
 * @param [out] core_array array of cores
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
get_l2id_cores(const cpu_set_t *cores,
               const unsigned l2_id,
               unsigned *core_num,
               unsigned core_array[CPU_SETSIZE])
{
        unsigned i;

        if (cores == NULL || core_num == NULL || core_array == NULL)
                return -EINVAL;

        if (m_cpu == NULL)
                return -EFAULT;

        if (CPU_COUNT(cores) == 0) {
                *core_num = 0;
                return 0;
        }

        for (i = 0, *core_num = 0; i < m_cpu->num_cores; i++) {
                if (m_cpu->cores[i].l2_id != l2_id ||
                    0 == CPU_ISSET(m_cpu->cores[i].lcore, cores))
                        continue;

                core_array[(*core_num)++] = m_cpu->cores[i].lcore;
        }

        return 0;
}

/**
 * @brief pqos_alloc_release wrapper
 *
 * @param [in] cores set of cores
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
alloc_release(const cpu_set_t *cores)
{
        unsigned core_array[CPU_SETSIZE];
        unsigned core_num, i;
        int ret;

        if (cores == NULL)
                return -EINVAL;

        if (m_cpu == NULL)
                return -EFAULT;

        /* if no cores in set then return success */
        if (CPU_COUNT(cores) == 0)
                return 0;

        for (i = 0, core_num = 0; i < m_cpu->num_cores; i++) {
                if (0 == CPU_ISSET(m_cpu->cores[i].lcore, cores))
                        continue;

                core_array[core_num++] = m_cpu->cores[i].lcore;
        }

        ret = pqos_alloc_release(core_array, core_num);
        if (ret != PQOS_RETVAL_OK) {
                fprintf(stderr, "Failed to release COS!\n");
                return -EFAULT;
        }
        return 0;
}

/**
 * @brief Get default COS
 *
 * @param [out] l2_def L2 COS
 * @param [out] l3_def L3 COS
 * @param [out] mba_def MBA COS
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
alloc_get_default_cos(struct pqos_l2ca *l2_def,
                      struct pqos_l3ca *l3_def,
                      struct pqos_mba *mba_def,
                      struct pqos_mba *smba_def)
{
        if (m_cpu == NULL)
                return -EINVAL;

        if (l2_def == NULL && l3_def == NULL && mba_def == NULL &&
            smba_def == NULL)
                return -EINVAL;

        if (m_cap_l2ca == NULL && m_cap_l3ca == NULL && m_cap_mba == NULL &&
            m_cap_smba == NULL)
                return -EFAULT;

        if (m_cap_l2ca != NULL && l2_def != NULL) {
                const uint64_t def_mask =
                    (1ULL << m_cap_l2ca->u.l2ca->num_ways) - 1ULL;

                memset(l2_def, 0, sizeof(*l2_def));

                if (m_cap_l2ca->u.l2ca->cdp_on == 1) {
                        l2_def->cdp = 1;
                        l2_def->u.s.code_mask = def_mask;
                        l2_def->u.s.data_mask = def_mask;
                } else
                        l2_def->u.ways_mask = def_mask;
        }

        if (m_cap_l3ca != NULL && l3_def != NULL) {
                const uint64_t def_mask =
                    (1ULL << m_cap_l3ca->u.l3ca->num_ways) - 1ULL;

                memset(l3_def, 0, sizeof(*l3_def));

                if (m_cap_l3ca->u.l3ca->cdp_on == 1) {
                        l3_def->cdp = 1;
                        l3_def->u.s.code_mask = def_mask;
                        l3_def->u.s.data_mask = def_mask;
                } else
                        l3_def->u.ways_mask = def_mask;
        }

        if (m_cap_mba != NULL && mba_def != NULL) {
                memset(mba_def, 0, sizeof(*mba_def));
                if (m_cap_mba->u.mba->ctrl_on == 1) {
                        mba_def->ctrl = 1;
                        mba_def->mb_max = UINT32_MAX;
                } else
                        mba_def->mb_max = (m_cpu->vendor == PQOS_VENDOR_AMD)
                                              ? RDT_MAX_MBA_AMD
                                              : RDT_MAX_MBA;
        }

        if (m_cap_smba != NULL && smba_def != NULL) {
                memset(smba_def, 0, sizeof(*smba_def));
                smba_def->smba = 1;
                if (m_cap_smba->u.smba->ctrl_on == 1) {
                        smba_def->ctrl = 1;
                        smba_def->mb_max = UINT32_MAX;
                } else
                        smba_def->mb_max = (m_cpu->vendor == PQOS_VENDOR_AMD)
                                               ? RDT_MAX_MBA_AMD
                                               : RDT_MAX_MBA;
        }

        return 0;
}

/**
 * @brief Configures COS \a cos_id defined by \a l2ca, \a l3ca and \a mba on
 *        \a socket_id
 *
 * @param [in] l2ca L2 classes configuration
 * @param [in] l3ca L3 classes configuration
 * @param [in] mba MBA configuration
 * @param [in] core_id core belonging to a socket/L2 cluster to configure COS on
 * @param [in] cos_id id of COS to be configured
 *
 * @return status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
cfg_configure_cos(const struct pqos_l2ca *l2ca,
                  const struct pqos_l3ca *l3ca,
                  const struct pqos_mba *mba,
                  const struct pqos_mba *smba,
                  const unsigned core_id,
                  const unsigned cos_id)
{
        struct pqos_l2ca l2_defs;
        struct pqos_l3ca l3_defs;
        struct pqos_mba mba_defs;
        struct pqos_mba smba_defs;
        const struct pqos_coreinfo *ci = NULL;
        int ret;

        if (m_cpu == NULL)
                return -EINVAL;

        if (NULL == l2ca && NULL == l3ca && NULL == mba && NULL == smba)
                return -EINVAL;

        if (NULL == m_cap_l2ca && NULL == m_cap_l3ca && NULL == m_cap_mba &&
            NULL == smba)
                return -EFAULT;

        ci = pqos_cpu_get_core_info(m_cpu, core_id);
        if (ci == NULL) {
                fprintf(stderr, "Error getting information about core %u!\n",
                        core_id);
                return -EINVAL;
        }

        /* Get default COS values */
        ret = alloc_get_default_cos(&l2_defs, &l3_defs, &mba_defs, &smba_defs);
        if (ret != 0)
                return ret;

        /* Configure COS */
        if (l3ca != NULL && m_cap_l3ca != NULL &&
            m_cap_l3ca->u.l3ca->num_classes > cos_id) {
                const unsigned l3cat_id = ci->l3cat_id;
                struct pqos_l3ca ca = *l3ca;

                /* if COS is not configured, set it to default */
                if (!rdt_cfg_is_valid(wrap_l3ca(&ca)))
                        ca = l3_defs;

                /* set proper COS id */
                ca.class_id = cos_id;

                ret = pqos_l3ca_set(l3cat_id, 1, &ca);
                if (ret == PQOS_RETVAL_PARAM) {
                        fprintf(stderr,
                                "Invalid parameters for L3 CAT COS#%u "
                                "on L3ID %u!\n",
                                cos_id, l3cat_id);
                        return -EINVAL;
                } else if (ret != PQOS_RETVAL_OK) {
                        fprintf(stderr,
                                "Error setting L3 CAT COS#%u on L3ID %u!\n",
                                cos_id, l3cat_id);
                        return -EFAULT;
                }
                if (g_cfg.verbose) {
                        const char *package;

                        if (m_cpu->vendor == PQOS_VENDOR_AMD)
                                package = "Core Complex";
                        else
                                package = "SOCKET";
                        if (ca.cdp)
                                printf("%s %u L3CA COS%u => DATA 0x%lx,CODE "
                                       "0x%lx\n",
                                       package, l3cat_id, ca.class_id,
                                       (long)ca.u.s.data_mask,
                                       (long)ca.u.s.code_mask);
                        else
                                printf("%s %u L3CA COS%u => MASK 0x%lx\n",
                                       package, l3cat_id, ca.class_id,
                                       ca.u.ways_mask);
                }
        }

        if (l2ca != NULL && m_cap_l2ca != NULL &&
            m_cap_l2ca->u.l2ca->num_classes > cos_id) {
                const unsigned l2_id = ci->l2_id;
                struct pqos_l2ca ca = *l2ca;

                /* if COS is not configured, set it to default */
                if (!rdt_cfg_is_valid(wrap_l2ca(&ca)))
                        ca = l2_defs;

                /* set proper COS id */
                ca.class_id = cos_id;

                ret = pqos_l2ca_set(l2_id, 1, &ca);
                if (ret == PQOS_RETVAL_PARAM) {
                        fprintf(stderr,
                                "Invalid parameters for L2 CAT COS#%u "
                                "on L2ID %u!\n",
                                cos_id, l2_id);
                        return -EINVAL;
                } else if (ret != PQOS_RETVAL_OK) {
                        fprintf(stderr,
                                "Error setting L2 CAT COS#%u on L2ID %u!\n",
                                cos_id, l2_id);
                        return -EFAULT;
                }

                if (g_cfg.verbose) {
                        if (ca.cdp)
                                printf("L2ID %u L2CA COS%u => DATA 0x%lx"
                                       " CODE 0x%lx\n",
                                       l2_id, ca.class_id,
                                       (long)ca.u.s.data_mask,
                                       (long)ca.u.s.code_mask);
                        else
                                printf("L2ID %u L2CA COS%u => MASK 0x%lx\n",
                                       l2_id, ca.class_id,
                                       (long)ca.u.ways_mask);
                }
        }

        if (mba != NULL && m_cap_mba != NULL &&
            m_cap_mba->u.mba->num_classes > cos_id) {
                const unsigned mba_id = ci->mba_id;
                struct pqos_mba mba_requested = *mba;
                struct pqos_mba mba_actual;

                /* if COS is not configured, set it to default */
                if (!rdt_cfg_is_valid(wrap_mba(&mba_requested)))
                        mba_requested = mba_defs;
                else if (mba_sc_mode(&g_cfg)) {
                        mba_requested.mb_max =
                            (m_cpu->vendor == PQOS_VENDOR_AMD) ? RDT_MAX_MBA_AMD
                                                               : RDT_MAX_MBA;
                        mba_requested.ctrl = 0;
                }

                /* set proper COS id */
                mba_requested.class_id = cos_id;

                ret = pqos_mba_set(mba_id, 1, &mba_requested, &mba_actual);
                if (ret == PQOS_RETVAL_PARAM) {
                        fprintf(stderr, "Invalid RDT parameters!\n");
                        return -EINVAL;
                }
                if (ret != PQOS_RETVAL_OK) {
                        fprintf(stderr,
                                "Error setting MBA COS#%u on mba id %u!\n",
                                cos_id, mba_id);
                        return -EFAULT;
                }
                if (g_cfg.verbose) {
                        const char *unit;
                        const char *package;

                        if (m_cpu->vendor == PQOS_VENDOR_AMD) {
                                package = "Core Complex";
                                unit = "";
                        } else {
                                package = "SOCKET";
                                unit = "%";
                        }
                        printf("%s %u MBA COS%u => ", package, mba_id,
                               mba_actual.class_id);

                        if (mba_requested.ctrl == 1)
                                printf("%u MBps\n", mba_requested.mb_max);
                        else
                                printf("%u%s requested, %u%s applied\n",
                                       mba_requested.mb_max, unit,
                                       mba_actual.mb_max, unit);
                }
        }

        if (smba != NULL && m_cap_smba != NULL &&
            m_cap_smba->u.smba->num_classes > cos_id) {
                const unsigned smba_id = ci->smba_id;
                struct pqos_mba smba_requested = *smba;
                struct pqos_mba smba_actual;

                /* if COS is not configured, set it to default */
                if (!rdt_cfg_is_valid(wrap_smba(&smba_requested)))
                        smba_requested = smba_defs;

                /* set proper COS id */
                smba_requested.class_id = cos_id;

                ret = pqos_mba_set(smba_id, 1, &smba_requested, &smba_actual);
                if (ret == PQOS_RETVAL_PARAM) {
                        fprintf(stderr, "Invalid RDT parameters!\n");
                        return -EINVAL;
                }
                if (ret != PQOS_RETVAL_OK) {
                        fprintf(stderr,
                                "Error setting SMBA COS#%u on smba id %u!\n",
                                cos_id, smba_id);
                        return -EFAULT;
                }
                if (g_cfg.verbose) {
                        const char *unit;
                        const char *package;

                        if (m_cpu->vendor == PQOS_VENDOR_AMD) {
                                package = "Core Complex";
                                unit = "";
                        } else {
                                package = "SOCKET";
                                unit = "%";
                        }
                        printf("%s %u MBA COS%u => ", package, smba_id,
                               smba_actual.class_id);

                        if (smba_requested.ctrl == 1)
                                printf("%u MBps\n", smba_requested.mb_max);
                        else
                                printf("%u%s requested, %u%s applied\n",
                                       smba_requested.mb_max, unit,
                                       smba_actual.mb_max, unit);
                }
        }

        return 0;
}

/**
 * @brief Sets socket/cluster definitions for selected cores (OS interface)
 *        Note: Assigns COS on a system wide basis
 *
 * @param [in] technology configured RDT allocation technologies
 * @param [in] cores cores configuration table
 * @param [in] l2ca L2 CAT configuration table
 * @param [in] l3ca L3 CAT configuration table
 * @param [in] mba MBA configuration table
 *
 * @return Operation status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
cfg_set_cores_os(const unsigned technology,
                 const cpu_set_t *cores,
                 const struct pqos_l2ca *l2ca,
                 const struct pqos_l3ca *l3ca,
                 const struct pqos_mba *mba,
                 const struct pqos_mba *smba)
{
        int ret;
        unsigned i, max_id, core_num, cos_id;
        unsigned core_array[CPU_SETSIZE] = {0};

        /* Convert cpu set to core array */
        for (i = 0, core_num = 0; i < m_cpu->num_cores; i++) {
                if (0 == CPU_ISSET(m_cpu->cores[i].lcore, cores))
                        continue;
                core_array[core_num++] = m_cpu->cores[i].lcore;
        }
        if (core_num == 0)
                return -EFAULT;

        ret = get_max_res_id(technology, &max_id);
        if (ret != 0)
                return ret;

        /* Assign all cores to single group */
        ret = pqos_alloc_assign(technology, core_array, core_num, &cos_id);
        switch (ret) {
        case PQOS_RETVAL_OK:
                if (g_cfg.verbose)
                        print_core_association(cos_id, core_num, core_array);
                break;
        case PQOS_RETVAL_RESOURCE:
                fprintf(stderr, "No free COS available!\n");
                return -EBUSY;
        default:
                fprintf(stderr, "Unable to assign COS L2!\n");
                return -EFAULT;
        }

        /* Set COS definition on necessary sockets/clusters */
        for (i = 0; i <= max_id; i++) {
                if (technology & (1 << PQOS_CAP_TYPE_L2CA)) {
                        memset(core_array, 0, sizeof(core_array));

                        /* Get cores on res id i */
                        ret = get_l2id_cores(cores, i, &core_num, core_array);
                        if (ret != 0)
                                return ret;

                        if (core_num == 0)
                                continue;

                        /* Configure COS on res ID i */
                        ret = cfg_configure_cos(l2ca, NULL, NULL, NULL,
                                                core_array[0], cos_id);
                        if (ret != 0)
                                return ret;
                }
                if (technology & (1 << PQOS_CAP_TYPE_L3CA)) {
                        memset(core_array, 0, sizeof(core_array));

                        /* Get cores on res id i */
                        ret =
                            get_l3cat_id_cores(cores, i, &core_num, core_array);
                        if (ret != 0)
                                return ret;

                        if (core_num == 0)
                                continue;

                        /* Configure COS on res ID i */
                        ret = cfg_configure_cos(NULL, l3ca, NULL, NULL,
                                                core_array[0], cos_id);
                        if (ret != 0)
                                return ret;
                }
                if (technology & (1 << PQOS_CAP_TYPE_MBA)) {
                        memset(core_array, 0, sizeof(core_array));

                        /* Get cores on res id i */
                        ret = get_mba_id_cores(cores, i, &core_num, core_array);
                        if (ret != 0)
                                return ret;

                        if (core_num == 0)
                                continue;

                        /* Configure COS on res ID i */
                        ret = cfg_configure_cos(NULL, NULL, mba, NULL,
                                                core_array[0], cos_id);
                        if (ret != 0)
                                return ret;
                }

                if (technology & (1 << PQOS_CAP_TYPE_SMBA)) {
                        memset(core_array, 0, sizeof(core_array));

                        /* Get cores on res id i */
                        ret =
                            get_smba_id_cores(cores, i, &core_num, core_array);
                        if (ret != 0)
                                return ret;

                        if (core_num == 0)
                                continue;

                        /* Configure COS on res ID i */
                        ret = cfg_configure_cos(NULL, NULL, NULL, smba,
                                                core_array[0], cos_id);
                        if (ret != 0)
                                return ret;
                }
        }

        return 0;
}

/**
 * @brief Sets socket/cluster definitions for selected cores (MSR interface)
 *        Note: Assigns COS on a res ID basis
 *
 * @param [in] technology configured RDT allocation technologies
 * @param [in] cores cores configuration table
 * @param [in] l2ca L2 CAT configuration table
 * @param [in] l3ca L3 CAT configuration table
 * @param [in] mba MBA configuration table
 *
 * @return Operation status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
cfg_set_cores_msr(const unsigned technology,
                  const cpu_set_t *cores,
                  const struct pqos_l2ca *l2ca,
                  const struct pqos_l3ca *l3ca,
                  const struct pqos_mba *mba,
                  const struct pqos_mba *smba)
{
        int ret = PQOS_RETVAL_OK;
        unsigned res_id, max_id;

        ret = get_max_res_id(technology, &max_id);
        if (ret != 0)
                return ret;

        /* Assign new COS for all applicable res ids */
        for (res_id = 0; res_id <= max_id; res_id++) {
                unsigned core_array[CPU_SETSIZE] = {0};
                unsigned core_num = 0, cos_id;
                unsigned i;

                /* Get cores on res id i */
                if (technology & (1 << PQOS_CAP_TYPE_L3CA)) {
                        ret = get_l3cat_id_cores(cores, res_id, &core_num,
                                                 core_array);
                } else if (technology & (1 << PQOS_CAP_TYPE_MBA)) {
                        ret = get_mba_id_cores(cores, res_id, &core_num,
                                               core_array);
                } else if (technology & (1 << PQOS_CAP_TYPE_L2CA)) {
                        ret = get_l2id_cores(cores, res_id, &core_num,
                                             core_array);
                } else if (technology & (1 << PQOS_CAP_TYPE_SMBA)) {
                        ret = get_smba_id_cores(cores, res_id, &core_num,
                                                core_array);
                }
                if (ret != 0)
                        return ret;
                if (core_num == 0)
                        continue;

                /* Assign COS to those cores */
                ret = pqos_alloc_assign(technology, core_array, core_num,
                                        &cos_id);
                switch (ret) {
                case PQOS_RETVAL_OK:
                        if (g_cfg.verbose)
                                print_core_association(cos_id, core_num,
                                                       core_array);
                        break;
                case PQOS_RETVAL_RESOURCE:
                        fprintf(stderr, "No free COS available!\n");
                        return -EBUSY;
                default:
                        fprintf(stderr, "Unable to assign COS!\n");
                        return -EFAULT;
                }

                /* Configure COS on res ID i */
                for (i = 0; i < core_num; i++) {
                        ret = cfg_configure_cos(l2ca, l3ca, mba, smba,
                                                core_array[i], cos_id);
                        if (ret != 0)
                                return ret;
                }
        }

        return 0;
}

/**
 * @brief Sets socket/cluster definitions for selected PIDs
 *
 * @param [in] technology configured RDT allocation technologies
 * @param [in] l2ca L2 CAT configuration table
 * @param [in] l3ca L3 CAT configuration table
 * @param [in] mba MBA configuration table
 * @param [in] smba MBA configuration table
 *
 * @return Operation status
 * @retval 0 on success
 * @retval negative on error (-errno)
 */
static int
cfg_set_pids(const unsigned technology,
             const struct pqos_l3ca *l3ca,
             const struct pqos_l2ca *l2ca,
             const struct pqos_mba *mba,
             const struct pqos_mba *smba)
{
        int ret = 0;
        unsigned i, cos_id, num_pids = 1;
        pid_t pid = getpid(), *p = &pid;

        /* Assign COS to selected pids otherwise assign to this pid */
        if (g_cfg.pid_count) {
                p = g_cfg.pids;
                num_pids = g_cfg.pid_count;
        }

        ret = pqos_alloc_assign_pid(technology, p, num_pids, &cos_id);
        switch (ret) {
        case PQOS_RETVAL_OK:
                if (g_cfg.verbose)
                        print_pid_association(cos_id, num_pids, p);
                break;
        case PQOS_RETVAL_RESOURCE:
                fprintf(stderr, "No free COS available!.\n");
                ret = -EBUSY;
                goto set_pids_exit;
        default:
                fprintf(stderr, "Unable to assign task to COS!\n");
                ret = -EFAULT;
                goto set_pids_exit;
        }

        /* Set COS definitions across all res L2 IDs */
        if (technology & (1 << PQOS_CAP_TYPE_L2CA)) {
                unsigned num_ids, *ids;

                ids = pqos_cpu_get_l2ids(m_cpu, &num_ids);
                if (ids == NULL)
                        return -EFAULT;

                for (i = 0; i < num_ids; i++) {
                        unsigned core;

                        ret = pqos_cpu_get_one_by_l2id(m_cpu, ids[i], &core);
                        if (ret != 0)
                                break;

                        /* Configure COS on res L2 ID i */
                        ret = cfg_configure_cos(l2ca, NULL, NULL, NULL, core,
                                                cos_id);
                        if (ret != 0)
                                break;
                }
                free(ids);
        }

        /* Set COS definitions across all res L3 IDs */
        if (technology & (1 << PQOS_CAP_TYPE_L3CA)) {
                unsigned num_ids, *ids;

                ids = pqos_cpu_get_l3cat_ids(m_cpu, &num_ids);
                if (ids == NULL)
                        return -EFAULT;

                for (i = 0; i < num_ids; i++) {
                        unsigned core;

                        ret =
                            pqos_cpu_get_one_by_l3cat_id(m_cpu, ids[i], &core);
                        if (ret != 0)
                                break;

                        /* Configure COS on res L3 ID i */
                        ret = cfg_configure_cos(NULL, l3ca, NULL, NULL, core,
                                                cos_id);
                        if (ret != 0)
                                break;
                }
                free(ids);
        }

        /* Set COS definitions across all res L3 IDs */
        if (technology & (1 << PQOS_CAP_TYPE_MBA)) {
                unsigned num_ids, *ids;

                ids = pqos_cpu_get_mba_ids(m_cpu, &num_ids);
                if (ids == NULL)
                        return -EFAULT;

                for (i = 0; i < num_ids; i++) {
                        unsigned core;

                        ret = pqos_cpu_get_one_by_mba_id(m_cpu, ids[i], &core);
                        if (ret != 0)
                                break;

                        /* Configure COS on res L3 ID i */
                        ret = cfg_configure_cos(NULL, NULL, mba, NULL, core,
                                                cos_id);
                        if (ret != 0)
                                break;
                }
                free(ids);
        }

        /* Set COS definitions across all res L3 IDs */
        if (technology & (1 << PQOS_CAP_TYPE_SMBA)) {
                unsigned num_ids, *ids;

                ids = pqos_cpu_get_smba_ids(m_cpu, &num_ids);
                if (ids == NULL)
                        return -EFAULT;

                for (i = 0; i < num_ids; i++) {
                        unsigned core;

                        ret = pqos_cpu_get_one_by_smba_id(m_cpu, ids[i], &core);
                        if (ret != 0)
                                break;

                        /* Configure COS on res L3 ID i */
                        ret = cfg_configure_cos(NULL, NULL, NULL, smba, core,
                                                cos_id);
                        if (ret != 0)
                                break;
                }
                free(ids);
        }

set_pids_exit:
        if (ret != 0)
                (void)pqos_alloc_release_pid(p, num_pids);

        return ret;
}

int
alloc_configure(void)
{
        struct pqos_l3ca l3ca[g_cfg.config_count];
        struct pqos_l2ca l2ca[g_cfg.config_count];
        struct pqos_mba mba[g_cfg.config_count];
        struct pqos_mba smba[g_cfg.config_count];
        cpu_set_t cpu[g_cfg.config_count];
        int pid_cfg[g_cfg.config_count];
        unsigned i = 0;
        int ret = 0;

        memset(pid_cfg, 0, g_cfg.config_count * sizeof(pid_cfg[0]));

        /* Validate cmd line configuration */
        ret = alloc_validate();
        if (ret != 0) {
                fprintf(stderr, "Requested configuration is not valid!\n");
                return ret;
        }

        for (i = 0; i < g_cfg.config_count; i++) {
                l3ca[i] = g_cfg.config[i].l3;
                l2ca[i] = g_cfg.config[i].l2;
                mba[i] = g_cfg.config[i].mba;
                smba[i] = g_cfg.config[i].smba;
                cpu[i] = g_cfg.config[i].cpumask;
                pid_cfg[i] = g_cfg.config[i].pid_cfg;
        }

        for (i = 0; i < g_cfg.config_count; i++) {
                unsigned technology = 0;

                if (rdt_cfg_is_valid(wrap_l2ca(&l2ca[i])))
                        technology |= (1 << PQOS_CAP_TYPE_L2CA);

                if (rdt_cfg_is_valid(wrap_l3ca(&l3ca[i])))
                        technology |= (1 << PQOS_CAP_TYPE_L3CA);

                if (rdt_cfg_is_valid(wrap_mba(&mba[i])))
                        technology |= (1 << PQOS_CAP_TYPE_MBA);

                if (rdt_cfg_is_valid(wrap_smba(&smba[i])))
                        technology |= (1 << PQOS_CAP_TYPE_SMBA);

                /* If pid config selected then assign tasks otherwise cores */
                if (pid_cfg[i])
                        ret = cfg_set_pids(technology, &l3ca[i], &l2ca[i],
                                           &mba[i], &smba[i]);
                else if (g_cfg.interface == PQOS_INTER_MSR)
                        ret = cfg_set_cores_msr(technology, &cpu[i], &l2ca[i],
                                                &l3ca[i], &mba[i], &smba[i]);
                else
                        ret = cfg_set_cores_os(technology, &cpu[i], &l2ca[i],
                                               &l3ca[i], &mba[i], &smba[i]);

                /* If assign fails then free already assigned cpus */
                if (ret != 0) {
                        fprintf(stderr, "Allocation failed!\n");
                        printf("Reverting configuration of allocation...\n");
                        for (; (int)i >= 0; i--) {
                                if (pid_cfg[i])
                                        continue;
                                (void)alloc_release(&cpu[i]);
                        }
                        return ret;
                }
        }

        return 0;
}

int
alloc_reset(void)
{
        unsigned i = 0;
        int ret = 0;

        /* Associate COS#0 to cores */
        for (i = 0; i < CPU_SETSIZE; i++) {
                if (0 == CPU_ISSET(i, &g_cfg.reset_cpuset))
                        continue;

                ret = pqos_alloc_assoc_set(i, 0);
                if (ret != PQOS_RETVAL_OK) {
                        fprintf(stderr,
                                "Error associating COS,"
                                "core: %u, COS: 0!\n",
                                i);
                        return -EFAULT;
                }
        }

        return 0;
}

void
alloc_fini(void)
{
        int ret = 0;

        if (g_cfg.verbose)
                printf("Shutting down PQoS library...\n");

        /* deallocate all the resources */
        ret = pqos_fini();
        if (ret != PQOS_RETVAL_OK && ret != PQOS_RETVAL_INIT)
                fprintf(stderr, "Error shutting down PQoS library!\n");

        m_cap = NULL;
        m_cpu = NULL;
        m_cap_l2ca = NULL;
        m_cap_l3ca = NULL;
        m_cap_mba = NULL;
        m_cap_smba = NULL;
        memset(g_cfg.config, 0, sizeof(g_cfg.config));
        g_cfg.config_count = 0;
}

void
alloc_exit(void)
{
        unsigned i = 0;

        /* if lib is not initialized, do nothing */
        if (m_cap == NULL && m_cpu == NULL)
                return;

        if (g_cfg.verbose)
                printf("CAT: Reverting CAT configuration...\n");

        /* release resources */
        for (i = 0; i < g_cfg.config_count; i++) {
                /* release pids if selected */
                if (g_cfg.config[i].pid_cfg && g_cfg.pid_count != 0) {
                        if (pqos_alloc_release_pid(g_cfg.pids, g_cfg.pid_count))
                                fprintf(stderr, "Failed to release PID COS!\n");
                        continue;
                }
                /* release cores */
                if (alloc_release(&g_cfg.config[i].cpumask) != 0)
                        fprintf(stderr, "Failed to release cores COS!\n");
        }
}

int
alloc_init(void)
{
        int ret = 0;
        struct pqos_config cfg;

        if (m_cap != NULL || m_cpu != NULL) {
                fprintf(stderr, "Allocation: module already initialized!\n");
                return -EEXIST;
        }

        /* PQoS Initialization - Check and initialize allocation capabilities */
        memset(&cfg, 0, sizeof(cfg));
        cfg.fd_log = STDOUT_FILENO;
        cfg.verbose = g_cfg.verbose;
        cfg.interface = g_cfg.interface;
        ret = pqos_init(&cfg);
        if (ret != PQOS_RETVAL_OK) {
                fprintf(stderr, "Allocation: Error initializing PQoS "
                                "library!\n");
                ret = -EFAULT;
                goto err;
        }

        /* Obtain interface */
        ret = pqos_inter_get(&g_cfg.interface);
        if (ret != PQOS_RETVAL_OK) {
                fprintf(stderr, "Allocation: Error retrieving PQoS "
                                "interface!\n");
                ret = -EFAULT;
                goto err;
        }

        /* Get capability and CPU info pointer */
        ret = pqos_cap_get(&m_cap, &m_cpu);
        if (ret != PQOS_RETVAL_OK || m_cap == NULL || m_cpu == NULL) {
                fprintf(stderr, "Allocation: Error retrieving PQoS "
                                "capabilities!\n");
                ret = -EFAULT;
                goto err;
        }

        /* Get L2CA capabilities */
        ret = pqos_cap_get_type(m_cap, PQOS_CAP_TYPE_L2CA, &m_cap_l2ca);
        if (g_cfg.verbose && (ret != PQOS_RETVAL_OK || m_cap_l2ca == NULL))
                printf("Allocation: L2 CAT capability not supported.\n");

        /* Get L3CA capabilities */
        ret = pqos_cap_get_type(m_cap, PQOS_CAP_TYPE_L3CA, &m_cap_l3ca);
        if (g_cfg.verbose && (ret != PQOS_RETVAL_OK || m_cap_l3ca == NULL))
                printf("Allocation: L3 CAT capability not supported.\n");

        /* Get MBA capabilities */
        ret = pqos_cap_get_type(m_cap, PQOS_CAP_TYPE_MBA, &m_cap_mba);
        if (g_cfg.verbose && (ret != PQOS_RETVAL_OK || m_cap_mba == NULL))
                printf("Allocation: MBA capability not supported.\n");

        /* Get SMBA capabilities */
        ret = pqos_cap_get_type(m_cap, PQOS_CAP_TYPE_SMBA, &m_cap_smba);
        if (g_cfg.verbose && (ret != PQOS_RETVAL_OK || m_cap_smba == NULL))
                printf("Allocation: SMBA capability not supported.\n");

        if (m_cap_l3ca == NULL && m_cap_l2ca == NULL && m_cap_mba == NULL &&
            m_cap_smba == NULL) {
                fprintf(stderr, "Allocation capabilities not supported!\n");
                ret = -EFAULT;
                goto err;
        }

        return 0;

err:
        /* deallocate all the resources */
        alloc_fini();
        return ret;
}

void
print_cmd_line_rdt_config(void)
{
        char cpustr[CPU_SETSIZE * 3];
        unsigned i;

        if (CPU_COUNT(&g_cfg.reset_cpuset) != 0) {
                cpuset_to_str(cpustr, sizeof(cpustr), &g_cfg.reset_cpuset);
                printf("Allocation Reset: CPUs: %s\n", cpustr);
        }

        for (i = 0; i < g_cfg.config_count; i++) {
                struct rdt_cfg cfg_array[3];
                unsigned j;

                if (CPU_COUNT(&g_cfg.config[i].cpumask) == 0)
                        continue;

                cpuset_to_str(cpustr, sizeof(cpustr), &g_cfg.config[i].cpumask);

                cfg_array[0] = wrap_l2ca(&g_cfg.config[i].l2);
                cfg_array[1] = wrap_l3ca(&g_cfg.config[i].l3);
                cfg_array[2] = wrap_mba(&g_cfg.config[i].mba);

                for (j = 0; j < DIM(cfg_array); j++) {
                        if (!rdt_cfg_is_valid(cfg_array[j]))
                                continue;
                        printf("%s Allocation: CPUs: %s ",
                               rdt_cfg_get_type_str(cfg_array[j]), cpustr);
                        rdt_cfg_print(stdout, cfg_array[j]);
                        printf("\n");
                }
        }
}

void
print_lib_version(void)
{
        int major = m_cap->version / 10000;
        int minor = (m_cap->version % 10000) / 100;
        int patch = m_cap->version % 100;

        printf("PQoS Library version: %d.%d.%d\n", major, minor, patch);
}
