// SPDX-License-Identifier: MIT
/*
 * Copyright 2022 Advanced Micro Devices, Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 *
 */

#include "amdgpu.h"

/**
 * amdgpu_mm_rdoorbell - read a doorbell dword
 *
 * @adev: amdgpu_device pointer
 * @index: doorbell index
 *
 * Returns the value in the doorbell aperture at the
 * requested doorbell index (CIK).
 */
u32 amdgpu_mm_rdoorbell(struct amdgpu_device *adev, u32 index)
{
 if (amdgpu_device_skip_hw_access(adev))
  return 0;

 if (index < adev->doorbell.num_kernel_doorbells)
  return readl(adev->doorbell.cpu_addr + index);

 dev_err(adev->dev, "reading beyond doorbell aperture: 0x%08x!\n",
  index);
 return 0;
}

/**
 * amdgpu_mm_wdoorbell - write a doorbell dword
 *
 * @adev: amdgpu_device pointer
 * @index: doorbell index
 * @v: value to write
 *
 * Writes @v to the doorbell aperture at the
 * requested doorbell index (CIK).
 */
void amdgpu_mm_wdoorbell(struct amdgpu_device *adev, u32 index, u32 v)
{
 if (amdgpu_device_skip_hw_access(adev))
  return;

 if (index < adev->doorbell.num_kernel_doorbells)
  writel(v, adev->doorbell.cpu_addr + index);
 else
  dev_err(adev->dev,
   "writing beyond doorbell aperture: 0x%08x!\n", index);
}

/**
 * amdgpu_mm_rdoorbell64 - read a doorbell Qword
 *
 * @adev: amdgpu_device pointer
 * @index: doorbell index
 *
 * Returns the value in the doorbell aperture at the
 * requested doorbell index (VEGA10+).
 */
u64 amdgpu_mm_rdoorbell64(struct amdgpu_device *adev, u32 index)
{
 if (amdgpu_device_skip_hw_access(adev))
  return 0;

 if (index < adev->doorbell.num_kernel_doorbells)
  return atomic64_read((atomic64_t *)(adev->doorbell.cpu_addr + index));

 dev_err(adev->dev, "reading beyond doorbell aperture: 0x%08x!\n",
  index);
 return 0;
}

/**
 * amdgpu_mm_wdoorbell64 - write a doorbell Qword
 *
 * @adev: amdgpu_device pointer
 * @index: doorbell index
 * @v: value to write
 *
 * Writes @v to the doorbell aperture at the
 * requested doorbell index (VEGA10+).
 */
void amdgpu_mm_wdoorbell64(struct amdgpu_device *adev, u32 index, u64 v)
{
 if (amdgpu_device_skip_hw_access(adev))
  return;

 if (index < adev->doorbell.num_kernel_doorbells)
  atomic64_set((atomic64_t *)(adev->doorbell.cpu_addr + index), v);
 else
  dev_err(adev->dev,
   "writing beyond doorbell aperture: 0x%08x!\n", index);
}

/**
 * amdgpu_doorbell_index_on_bar - Find doorbell's absolute offset in BAR
 *
 * @adev: amdgpu_device pointer
 * @db_bo: doorbell object's bo
 * @doorbell_index: doorbell relative index in this doorbell object
 * @db_size: doorbell size is in byte
 *
 * returns doorbell's absolute index in BAR
 */
uint32_t amdgpu_doorbell_index_on_bar(struct amdgpu_device *adev,
          struct amdgpu_bo *db_bo,
          uint32_t doorbell_index,
          uint32_t db_size)
{
 int db_bo_offset;

 db_bo_offset = amdgpu_bo_gpu_offset_no_check(db_bo);

 /* doorbell index is 32 bit but doorbell's size can be 32 bit
  * or 64 bit, so *db_size(in byte)/4 for alignment.
 */
 return db_bo_offset / sizeof(u32) + doorbell_index *
        DIV_ROUND_UP(db_size, 4);
}

/**
 * amdgpu_doorbell_create_kernel_doorbells - Create kernel doorbells for graphics
 *
 * @adev: amdgpu_device pointer
 *
 * Creates doorbells for graphics driver usages.
 * returns 0 on success, error otherwise.
 */
int amdgpu_doorbell_create_kernel_doorbells(struct amdgpu_device *adev)
{
 int r;
 int size;

 /* SI HW does not have doorbells, skip allocation */
 if (adev->doorbell.num_kernel_doorbells == 0)
  return 0;

 /* Reserve first num_kernel_doorbells (page-aligned) for kernel ops */
 size = ALIGN(adev->doorbell.num_kernel_doorbells * sizeof(u32), PAGE_SIZE);

 /* Allocate an extra page for MES kernel usages (ring test) */
 adev->mes.db_start_dw_offset = size / sizeof(u32);
 size += PAGE_SIZE;

 r = amdgpu_bo_create_kernel(adev,
        size,
        PAGE_SIZE,
        AMDGPU_GEM_DOMAIN_DOORBELL,
        &adev->doorbell.kernel_doorbells,
        NULL,
        (void **)&adev->doorbell.cpu_addr);
 if (r) {
  dev_err(adev->dev,
   "Failed to allocate kernel doorbells, err=%d\n", r);
  return r;
 }

 adev->doorbell.num_kernel_doorbells = size / sizeof(u32);
 return 0;
}

/*
 * GPU doorbell aperture helpers function.
 */
/**
 * amdgpu_doorbell_init - Init doorbell driver information.
 *
 * @adev: amdgpu_device pointer
 *
 * Init doorbell driver information (CIK)
 * Returns 0 on success, error on failure.
 */
int amdgpu_doorbell_init(struct amdgpu_device *adev)
{

 /* No doorbell on SI hardware generation */
 if (adev->asic_type < CHIP_BONAIRE) {
  adev->doorbell.base = 0;
  adev->doorbell.size = 0;
  adev->doorbell.num_kernel_doorbells = 0;
  return 0;
 }

 if (pci_resource_flags(adev->pdev, 2) & IORESOURCE_UNSET)
  return -EINVAL;

 amdgpu_asic_init_doorbell_index(adev);

 /* doorbell bar mapping */
 adev->doorbell.base = pci_resource_start(adev->pdev, 2);
 adev->doorbell.size = pci_resource_len(adev->pdev, 2);

 adev->doorbell.num_kernel_doorbells =
  min_t(u32, adev->doorbell.size / sizeof(u32),
        adev->doorbell_index.max_assignment + 1);
 if (adev->doorbell.num_kernel_doorbells == 0)
  return -EINVAL;

 /*
  * For Vega, reserve and map two pages on doorbell BAR since SDMA
  * paging queue doorbell use the second page. The
  * AMDGPU_DOORBELL64_MAX_ASSIGNMENT definition assumes all the
  * doorbells are in the first page. So with paging queue enabled,
  * the max num_kernel_doorbells should + 1 page (0x400 in dword)
 */
 if (adev->asic_type >= CHIP_VEGA10)
  adev->doorbell.num_kernel_doorbells += 0x400;

 return 0;
}

/**
 * amdgpu_doorbell_fini - Tear down doorbell driver information.
 *
 * @adev: amdgpu_device pointer
 *
 * Tear down doorbell driver information (CIK)
 */
void amdgpu_doorbell_fini(struct amdgpu_device *adev)
{
 amdgpu_bo_free_kernel(&adev->doorbell.kernel_doorbells,
         NULL,
         (void **)&adev->doorbell.cpu_addr);
}