// SPDX-License-Identifier: GPL-2.0
// Copyright (c) Huawei Technologies Co., Ltd. 2025. All rights reserved.

#include <linux/device.h>

#include "hinic3_hwdev.h"
#include "hinic3_queue_common.h"

void hinic3_queue_pages_init(struct hinic3_queue_pages *qpages, u32 q_depth,
			     u32 page_size, u32 elem_size)
{
	u32 elem_per_page;

	elem_per_page = min(page_size / elem_size, q_depth);

	qpages->pages = NULL;
	qpages->page_size = page_size;
	qpages->num_pages = max(q_depth / elem_per_page, 1);
	qpages->elem_size_shift = ilog2(elem_size);
	qpages->elem_per_pg_shift = ilog2(elem_per_page);
}

static void __queue_pages_free(struct hinic3_hwdev *hwdev,
			       struct hinic3_queue_pages *qpages, u32 pg_cnt)
{
	while (pg_cnt > 0) {
		pg_cnt--;
		hinic3_dma_free_coherent_align(hwdev->dev,
					       qpages->pages + pg_cnt);
	}
	kfree(qpages->pages);
	qpages->pages = NULL;
}

void hinic3_queue_pages_free(struct hinic3_hwdev *hwdev,
			     struct hinic3_queue_pages *qpages)
{
	__queue_pages_free(hwdev, qpages, qpages->num_pages);
}

int hinic3_queue_pages_alloc(struct hinic3_hwdev *hwdev,
			     struct hinic3_queue_pages *qpages, u32 align)
{
	u32 pg_idx;
	int err;

	qpages->pages = kcalloc(qpages->num_pages, sizeof(qpages->pages[0]),
				GFP_KERNEL);
	if (!qpages->pages)
		return -ENOMEM;

	if (align == 0)
		align = qpages->page_size;

	for (pg_idx = 0; pg_idx < qpages->num_pages; pg_idx++) {
		err = hinic3_dma_zalloc_coherent_align(hwdev->dev,
						       qpages->page_size,
						       align,
						       GFP_KERNEL,
						       qpages->pages + pg_idx);
		if (err) {
			__queue_pages_free(hwdev, qpages, pg_idx);
			return err;
		}
	}

	return 0;
}