1// SPDX-License-Identifier: GPL-2.0+
 2/* Microchip Sparx5 Switch driver
 3 *
 4 * Copyright (c) 2023 Microchip Technology Inc. and its subsidiaries.
 5 */
 6
 7#include "sparx5_main_regs.h"
 8#include "sparx5_main.h"
 9
10static u32 sparx5_pool_id_to_idx(u32 id)
11{
12	return --id;
13}
14
15u32 sparx5_pool_idx_to_id(u32 idx)
16{
17	return ++idx;
18}
19
20/* Release resource from pool.
21 * Return reference count on success, otherwise return error.
22 */
23int sparx5_pool_put(struct sparx5_pool_entry *pool, int size, u32 id)
24{
25	struct sparx5_pool_entry *e_itr;
26
27	e_itr = (pool + sparx5_pool_id_to_idx(id));
28	if (e_itr->ref_cnt == 0)
29		return -EINVAL;
30
31	return --e_itr->ref_cnt;
32}
33
34/* Get resource from pool.
35 * Return reference count on success, otherwise return error.
36 */
37int sparx5_pool_get(struct sparx5_pool_entry *pool, int size, u32 *id)
38{
39	struct sparx5_pool_entry *e_itr;
40	int i;
41
42	for (i = 0, e_itr = pool; i < size; i++, e_itr++) {
43		if (e_itr->ref_cnt == 0) {
44			*id = sparx5_pool_idx_to_id(i);
45			return ++e_itr->ref_cnt;
46		}
47	}
48
49	return -ENOSPC;
50}
51
52/* Get resource from pool that matches index.
53 * Return reference count on success, otherwise return error.
54 */
55int sparx5_pool_get_with_idx(struct sparx5_pool_entry *pool, int size, u32 idx,
56			     u32 *id)
57{
58	struct sparx5_pool_entry *e_itr;
59	int i, ret = -ENOSPC;
60
61	for (i = 0, e_itr = pool; i < size; i++, e_itr++) {
62		/* Pool index of first free entry */
63		if (e_itr->ref_cnt == 0 && ret == -ENOSPC)
64			ret = i;
65		/* Tc index already in use ? */
66		if (e_itr->idx == idx && e_itr->ref_cnt > 0) {
67			ret = i;
68			break;
69		}
70	}
71
72	/* Did we find a free entry? */
73	if (ret >= 0) {
74		*id = sparx5_pool_idx_to_id(ret);
75		e_itr = (pool + ret);
76		e_itr->idx = idx;
77		return ++e_itr->ref_cnt;
78	}
79
80	return ret;
81}