// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
/*
 * Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved
 */

#include <rdma/uverbs_std_types.h>
#include <linux/pci-tph.h>
#include "dmah.h"

#define UVERBS_MODULE_NAME mlx5_ib
#include <rdma/uverbs_named_ioctl.h>

static int mlx5_ib_alloc_dmah(struct ib_dmah *ibdmah,
                              struct uverbs_attr_bundle *attrs)
{
        struct mlx5_core_dev *mdev = to_mdev(ibdmah->device)->mdev;
        struct mlx5_ib_dmah *dmah = to_mdmah(ibdmah);
        u16 st_bits = BIT(IB_DMAH_CPU_ID_EXISTS) |
                      BIT(IB_DMAH_MEM_TYPE_EXISTS);
        int err;

        /* PH is a must for TPH following PCIe spec 6.2-1.0 */
        if (!(ibdmah->valid_fields & BIT(IB_DMAH_PH_EXISTS)))
                return -EINVAL;

        /* ST is optional; however, partial data for it is not allowed */
        if (ibdmah->valid_fields & st_bits) {
                if ((ibdmah->valid_fields & st_bits) != st_bits)
                        return -EINVAL;
                err = mlx5_st_alloc_index(mdev, ibdmah->mem_type,
                                          ibdmah->cpu_id, &dmah->st_index);
                if (err)
                        return err;
        }

        return 0;
}

static int mlx5_ib_dealloc_dmah(struct ib_dmah *ibdmah,
                                struct uverbs_attr_bundle *attrs)
{
        struct mlx5_ib_dmah *dmah = to_mdmah(ibdmah);
        struct mlx5_core_dev *mdev = to_mdev(ibdmah->device)->mdev;

        if (ibdmah->valid_fields & BIT(IB_DMAH_CPU_ID_EXISTS))
                return mlx5_st_dealloc_index(mdev, dmah->st_index);

        return 0;
}

const struct ib_device_ops mlx5_ib_dev_dmah_ops = {
        .alloc_dmah = mlx5_ib_alloc_dmah,
        .dealloc_dmah = mlx5_ib_dealloc_dmah,
};