// 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 = kzalloc_objs(qpages->pages[0], qpages->num_pages);
        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;
}