// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
// Copyright (c) 2018 Mellanox Technologies

#include <linux/hyperv.h>
#include "mlx5_core.h"
#include "lib/hv.h"

static int mlx5_hv_config_common(struct mlx5_core_dev *dev, void *buf, int len,
                                 int offset, bool read)
{
        int rc = -EOPNOTSUPP;
        int bytes_returned;
        int block_id;

        if (offset % HV_CONFIG_BLOCK_SIZE_MAX || len != HV_CONFIG_BLOCK_SIZE_MAX)
                return -EINVAL;

        block_id = offset / HV_CONFIG_BLOCK_SIZE_MAX;

        rc = read ?
             hyperv_read_cfg_blk(dev->pdev, buf,
                                 HV_CONFIG_BLOCK_SIZE_MAX, block_id,
                                 &bytes_returned) :
             hyperv_write_cfg_blk(dev->pdev, buf,
                                  HV_CONFIG_BLOCK_SIZE_MAX, block_id);

        /* Make sure len bytes were read successfully  */
        if (read && !rc && len != bytes_returned)
                rc = -EIO;

        if (rc) {
                mlx5_core_err(dev, "Failed to %s hv config, err = %d, len = %d, offset = %d\n",
                              read ? "read" : "write", rc, len,
                              offset);
                return rc;
        }

        return 0;
}

int mlx5_hv_read_config(struct mlx5_core_dev *dev, void *buf, int len,
                        int offset)
{
        return mlx5_hv_config_common(dev, buf, len, offset, true);
}

int mlx5_hv_write_config(struct mlx5_core_dev *dev, void *buf, int len,
                         int offset)
{
        return mlx5_hv_config_common(dev, buf, len, offset, false);
}

int mlx5_hv_register_invalidate(struct mlx5_core_dev *dev, void *context,
                                void (*block_invalidate)(void *context,
                                                         u64 block_mask))
{
        return hyperv_reg_block_invalidate(dev->pdev, context,
                                           block_invalidate);
}

void mlx5_hv_unregister_invalidate(struct mlx5_core_dev *dev)
{
        hyperv_reg_block_invalidate(dev->pdev, NULL, NULL);
}