// SPDX-License-Identifier: GPL-2.0+
// Copyright 2019 IBM Corp.
#include <linux/sched/mm.h>
#include "trace.h"
#include "ocxl_internal.h"

int ocxl_global_mmio_read32(struct ocxl_afu *afu, size_t offset,
    enum ocxl_endian endian, u32 *val)
{
 if (offset > afu->config.global_mmio_size - 4)
  return -EINVAL;

#ifdef __BIG_ENDIAN__
 if (endian == OCXL_HOST_ENDIAN)
  endian = OCXL_BIG_ENDIAN;
#endif

 switch (endian) {
 case OCXL_BIG_ENDIAN:
  *val = readl_be((char *)afu->global_mmio_ptr + offset);
  break;

 default:
  *val = readl((char *)afu->global_mmio_ptr + offset);
  break;
 }

 return 0;
}
EXPORT_SYMBOL_GPL(ocxl_global_mmio_read32);

int ocxl_global_mmio_read64(struct ocxl_afu *afu, size_t offset,
    enum ocxl_endian endian, u64 *val)
{
 if (offset > afu->config.global_mmio_size - 8)
  return -EINVAL;

#ifdef __BIG_ENDIAN__
 if (endian == OCXL_HOST_ENDIAN)
  endian = OCXL_BIG_ENDIAN;
#endif

 switch (endian) {
 case OCXL_BIG_ENDIAN:
  *val = readq_be((char *)afu->global_mmio_ptr + offset);
  break;

 default:
  *val = readq((char *)afu->global_mmio_ptr + offset);
  break;
 }

 return 0;
}
EXPORT_SYMBOL_GPL(ocxl_global_mmio_read64);

int ocxl_global_mmio_write32(struct ocxl_afu *afu, size_t offset,
    enum ocxl_endian endian, u32 val)
{
 if (offset > afu->config.global_mmio_size - 4)
  return -EINVAL;

#ifdef __BIG_ENDIAN__
 if (endian == OCXL_HOST_ENDIAN)
  endian = OCXL_BIG_ENDIAN;
#endif

 switch (endian) {
 case OCXL_BIG_ENDIAN:
  writel_be(val, (char *)afu->global_mmio_ptr + offset);
  break;

 default:
  writel(val, (char *)afu->global_mmio_ptr + offset);
  break;
 }


 return 0;
}
EXPORT_SYMBOL_GPL(ocxl_global_mmio_write32);

int ocxl_global_mmio_write64(struct ocxl_afu *afu, size_t offset,
    enum ocxl_endian endian, u64 val)
{
 if (offset > afu->config.global_mmio_size - 8)
  return -EINVAL;

#ifdef __BIG_ENDIAN__
 if (endian == OCXL_HOST_ENDIAN)
  endian = OCXL_BIG_ENDIAN;
#endif

 switch (endian) {
 case OCXL_BIG_ENDIAN:
  writeq_be(val, (char *)afu->global_mmio_ptr + offset);
  break;

 default:
  writeq(val, (char *)afu->global_mmio_ptr + offset);
  break;
 }


 return 0;
}
EXPORT_SYMBOL_GPL(ocxl_global_mmio_write64);

int ocxl_global_mmio_set32(struct ocxl_afu *afu, size_t offset,
    enum ocxl_endian endian, u32 mask)
{
 u32 tmp;

 if (offset > afu->config.global_mmio_size - 4)
  return -EINVAL;

#ifdef __BIG_ENDIAN__
 if (endian == OCXL_HOST_ENDIAN)
  endian = OCXL_BIG_ENDIAN;
#endif

 switch (endian) {
 case OCXL_BIG_ENDIAN:
  tmp = readl_be((char *)afu->global_mmio_ptr + offset);
  tmp |= mask;
  writel_be(tmp, (char *)afu->global_mmio_ptr + offset);
  break;

 default:
  tmp = readl((char *)afu->global_mmio_ptr + offset);
  tmp |= mask;
  writel(tmp, (char *)afu->global_mmio_ptr + offset);
  break;
 }

 return 0;
}
EXPORT_SYMBOL_GPL(ocxl_global_mmio_set32);

int ocxl_global_mmio_set64(struct ocxl_afu *afu, size_t offset,
    enum ocxl_endian endian, u64 mask)
{
 u64 tmp;

 if (offset > afu->config.global_mmio_size - 8)
  return -EINVAL;

#ifdef __BIG_ENDIAN__
 if (endian == OCXL_HOST_ENDIAN)
  endian = OCXL_BIG_ENDIAN;
#endif

 switch (endian) {
 case OCXL_BIG_ENDIAN:
  tmp = readq_be((char *)afu->global_mmio_ptr + offset);
  tmp |= mask;
  writeq_be(tmp, (char *)afu->global_mmio_ptr + offset);
  break;

 default:
  tmp = readq((char *)afu->global_mmio_ptr + offset);
  tmp |= mask;
  writeq(tmp, (char *)afu->global_mmio_ptr + offset);
  break;
 }

 return 0;
}
EXPORT_SYMBOL_GPL(ocxl_global_mmio_set64);

int ocxl_global_mmio_clear32(struct ocxl_afu *afu, size_t offset,
    enum ocxl_endian endian, u32 mask)
{
 u32 tmp;

 if (offset > afu->config.global_mmio_size - 4)
  return -EINVAL;

#ifdef __BIG_ENDIAN__
 if (endian == OCXL_HOST_ENDIAN)
  endian = OCXL_BIG_ENDIAN;
#endif

 switch (endian) {
 case OCXL_BIG_ENDIAN:
  tmp = readl_be((char *)afu->global_mmio_ptr + offset);
  tmp &= ~mask;
  writel_be(tmp, (char *)afu->global_mmio_ptr + offset);
  break;

 default:
  tmp = readl((char *)afu->global_mmio_ptr + offset);
  tmp &= ~mask;
  writel(tmp, (char *)afu->global_mmio_ptr + offset);
  break;
 }


 return 0;
}
EXPORT_SYMBOL_GPL(ocxl_global_mmio_clear32);

int ocxl_global_mmio_clear64(struct ocxl_afu *afu, size_t offset,
    enum ocxl_endian endian, u64 mask)
{
 u64 tmp;

 if (offset > afu->config.global_mmio_size - 8)
  return -EINVAL;

#ifdef __BIG_ENDIAN__
 if (endian == OCXL_HOST_ENDIAN)
  endian = OCXL_BIG_ENDIAN;
#endif

 switch (endian) {
 case OCXL_BIG_ENDIAN:
  tmp = readq_be((char *)afu->global_mmio_ptr + offset);
  tmp &= ~mask;
  writeq_be(tmp, (char *)afu->global_mmio_ptr + offset);
  break;

 default:
  tmp = readq((char *)afu->global_mmio_ptr + offset);
  tmp &= ~mask;
  writeq(tmp, (char *)afu->global_mmio_ptr + offset);
  break;
 }

 writeq(tmp, (char *)afu->global_mmio_ptr + offset);

 return 0;
}
EXPORT_SYMBOL_GPL(ocxl_global_mmio_clear64);