// SPDX-License-Identifier: GPL-2.0+
#include "sparx5_main.h"

void sparx5_pgid_init(struct sparx5 *spx5)
{
        int i;

        for (i = 0; i < spx5->data->consts->n_pgids; i++)
                spx5->pgid_map[i] = SPX5_PGID_FREE;

        /* Reserved for unicast, flood control, broadcast, and CPU.
         * These cannot be freed.
         */
        for (i = 0; i <= sparx5_get_pgid(spx5, PGID_CPU); i++)
                spx5->pgid_map[i] = SPX5_PGID_RESERVED;
}

int sparx5_pgid_alloc_mcast(struct sparx5 *spx5, u16 *idx)
{
        int i;

        /* The multicast area starts at index 65, but the first 7
         * are reserved for flood masks and CPU. Start alloc after that.
         */
        for (i = sparx5_get_pgid(spx5, PGID_MCAST_START);
             i < spx5->data->consts->n_pgids; i++) {
                if (spx5->pgid_map[i] == SPX5_PGID_FREE) {
                        spx5->pgid_map[i] = SPX5_PGID_MULTICAST;
                        *idx = i;
                        return 0;
                }
        }

        return -EBUSY;
}

int sparx5_pgid_free(struct sparx5 *spx5, u16 idx)
{
        if (idx <= sparx5_get_pgid(spx5, PGID_CPU) ||
            idx >= spx5->data->consts->n_pgids)
                return -EINVAL;

        if (spx5->pgid_map[idx] == SPX5_PGID_FREE)
                return -EINVAL;

        spx5->pgid_map[idx] = SPX5_PGID_FREE;
        return 0;
}

int sparx5_get_pgid(struct sparx5 *sparx5, int pgid)
{
        return sparx5->data->consts->n_ports + pgid;
}