// SPDX-License-Identifier: (GPL-2.0-only OR BSD-2-Clause)
/* Copyright (C) 2015-2018 Netronome Systems, Inc. */

#include <linux/delay.h>
#include <linux/device.h>
#include <linux/jiffies.h>
#include <linux/types.h>
#include <linux/slab.h>
#include <linux/wait.h>

#include "nfp_cpp.h"
#include "nfp6000/nfp6000.h"

struct nfp_cpp_mutex {
 struct nfp_cpp *cpp;
 int target;
 u16 depth;
 unsigned long long address;
 u32 key;
};

static u32 nfp_mutex_locked(u16 interface)
{
 return (u32)interface << 16 | 0x000f;
}

static u32 nfp_mutex_unlocked(u16 interface)
{
 return (u32)interface << 16 | 0x0000;
}

static u32 nfp_mutex_owner(u32 val)
{
 return val >> 16;
}

static bool nfp_mutex_is_locked(u32 val)
{
 return (val & 0xffff) == 0x000f;
}

static bool nfp_mutex_is_unlocked(u32 val)
{
 return (val & 0xffff) == 0000;
}

/* If you need more than 65536 recursive locks, please rethink your code. */
#define NFP_MUTEX_DEPTH_MAX         0xffff

static int
nfp_cpp_mutex_validate(u16 interface, int *target, unsigned long long address)
{
 /* Not permitted on invalid interfaces */
 if (NFP_CPP_INTERFACE_TYPE_of(interface) ==
     NFP_CPP_INTERFACE_TYPE_INVALID)
  return -EINVAL;

 /* Address must be 64-bit aligned */
 if (address & 7)
  return -EINVAL;

 if (*target != NFP_CPP_TARGET_MU)
  return -EINVAL;

 return 0;
}

/**
 * nfp_cpp_mutex_init() - Initialize a mutex location
 * @cpp: NFP CPP handle
 * @target: NFP CPP target ID (ie NFP_CPP_TARGET_CLS or NFP_CPP_TARGET_MU)
 * @address: Offset into the address space of the NFP CPP target ID
 * @key: Unique 32-bit value for this mutex
 *
 * The CPP target:address must point to a 64-bit aligned location, and
 * will initialize 64 bits of data at the location.
 *
 * This creates the initial mutex state, as locked by this
 * nfp_cpp_interface().
 *
 * This function should only be called when setting up
 * the initial lock state upon boot-up of the system.
 *
 * Return: 0 on success, or -errno on failure
 */
int nfp_cpp_mutex_init(struct nfp_cpp *cpp,
         int target, unsigned long long address, u32 key)
{
 const u32 muw = NFP_CPP_ID(target, 4, 0);    /* atomic_write */
 u16 interface = nfp_cpp_interface(cpp);
 int err;

 err = nfp_cpp_mutex_validate(interface, &target, address);
 if (err)
  return err;

 err = nfp_cpp_writel(cpp, muw, address + 4, key);
 if (err)
  return err;

 err = nfp_cpp_writel(cpp, muw, address, nfp_mutex_locked(interface));
 if (err)
  return err;

 return 0;
}

/**
 * nfp_cpp_mutex_alloc() - Create a mutex handle
 * @cpp: NFP CPP handle
 * @target: NFP CPP target ID (ie NFP_CPP_TARGET_CLS or NFP_CPP_TARGET_MU)
 * @address: Offset into the address space of the NFP CPP target ID
 * @key: 32-bit unique key (must match the key at this location)
 *
 * The CPP target:address must point to a 64-bit aligned location, and
 * reserve 64 bits of data at the location for use by the handle.
 *
 * Only target/address pairs that point to entities that support the
 * MU Atomic Engine's CmpAndSwap32 command are supported.
 *
 * Return: A non-NULL struct nfp_cpp_mutex * on success, NULL on failure.
 */
struct nfp_cpp_mutex *nfp_cpp_mutex_alloc(struct nfp_cpp *cpp, int target,
       unsigned long long address, u32 key)
{
 const u32 mur = NFP_CPP_ID(target, 3, 0);    /* atomic_read */
 u16 interface = nfp_cpp_interface(cpp);
 struct nfp_cpp_mutex *mutex;
 int err;
 u32 tmp;

 err = nfp_cpp_mutex_validate(interface, &target, address);
 if (err)
  return NULL;

 err = nfp_cpp_readl(cpp, mur, address + 4, &tmp);
 if (err < 0)
  return NULL;

 if (tmp != key)
  return NULL;

 mutex = kzalloc(sizeof(*mutex), GFP_KERNEL);
 if (!mutex)
  return NULL;

 mutex->cpp = cpp;
 mutex->target = target;
 mutex->address = address;
 mutex->key = key;
 mutex->depth = 0;

 return mutex;
}

/**
 * nfp_cpp_mutex_free() - Free a mutex handle - does not alter the lock state
 * @mutex: NFP CPP Mutex handle
 */
void nfp_cpp_mutex_free(struct nfp_cpp_mutex *mutex)
{
 kfree(mutex);
}

/**
 * nfp_cpp_mutex_lock() - Lock a mutex handle, using the NFP MU Atomic Engine
 * @mutex: NFP CPP Mutex handle
 *
 * Return: 0 on success, or -errno on failure
 */
int nfp_cpp_mutex_lock(struct nfp_cpp_mutex *mutex)
{
 unsigned long warn_at = jiffies + NFP_MUTEX_WAIT_FIRST_WARN * HZ;
 unsigned long err_at = jiffies + NFP_MUTEX_WAIT_ERROR * HZ;
 unsigned int timeout_ms = 1;
 int err;

 /* We can't use a waitqueue here, because the unlocker
  * might be on a separate CPU.
  *
  * So just wait for now.
 */
 for (;;) {
  err = nfp_cpp_mutex_trylock(mutex);
  if (err != -EBUSY)
   break;

  err = msleep_interruptible(timeout_ms);
  if (err != 0) {
   nfp_info(mutex->cpp,
     "interrupted waiting for NFP mutex\n");
   return -ERESTARTSYS;
  }

  if (time_is_before_eq_jiffies(warn_at)) {
   warn_at = jiffies + NFP_MUTEX_WAIT_NEXT_WARN * HZ;
   nfp_warn(mutex->cpp,
     "Warning: waiting for NFP mutex [depth:%hd target:%d addr:%llx key:%08x]\n",
     mutex->depth,
     mutex->target, mutex->address, mutex->key);
  }
  if (time_is_before_eq_jiffies(err_at)) {
   nfp_err(mutex->cpp, "Error: mutex wait timed out\n");
   return -EBUSY;
  }
 }

 return err;
}

/**
 * nfp_cpp_mutex_unlock() - Unlock a mutex handle, using the MU Atomic Engine
 * @mutex: NFP CPP Mutex handle
 *
 * Return: 0 on success, or -errno on failure
 */
int nfp_cpp_mutex_unlock(struct nfp_cpp_mutex *mutex)
{
 const u32 muw = NFP_CPP_ID(mutex->target, 4, 0);    /* atomic_write */
 const u32 mur = NFP_CPP_ID(mutex->target, 3, 0);    /* atomic_read */
 struct nfp_cpp *cpp = mutex->cpp;
 u32 key, value;
 u16 interface;
 int err;

 interface = nfp_cpp_interface(cpp);

 if (mutex->depth > 1) {
  mutex->depth--;
  return 0;
 }

 err = nfp_cpp_readl(mutex->cpp, mur, mutex->address + 4, &key);
 if (err < 0)
  return err;

 if (key != mutex->key)
  return -EPERM;

 err = nfp_cpp_readl(mutex->cpp, mur, mutex->address, &value);
 if (err < 0)
  return err;

 if (value != nfp_mutex_locked(interface))
  return -EACCES;

 err = nfp_cpp_writel(cpp, muw, mutex->address,
        nfp_mutex_unlocked(interface));
 if (err < 0)
  return err;

 mutex->depth = 0;
 return 0;
}

/**
 * nfp_cpp_mutex_trylock() - Attempt to lock a mutex handle
 * @mutex: NFP CPP Mutex handle
 *
 * Return:      0 if the lock succeeded, -errno on failure
 */
int nfp_cpp_mutex_trylock(struct nfp_cpp_mutex *mutex)
{
 const u32 muw = NFP_CPP_ID(mutex->target, 4, 0);    /* atomic_write */
 const u32 mus = NFP_CPP_ID(mutex->target, 5, 3);    /* test_set_imm */
 const u32 mur = NFP_CPP_ID(mutex->target, 3, 0);    /* atomic_read */
 struct nfp_cpp *cpp = mutex->cpp;
 u32 key, value, tmp;
 int err;

 if (mutex->depth > 0) {
  if (mutex->depth == NFP_MUTEX_DEPTH_MAX)
   return -E2BIG;
  mutex->depth++;
  return 0;
 }

 /* Verify that the lock marker is not damaged */
 err = nfp_cpp_readl(cpp, mur, mutex->address + 4, &key);
 if (err < 0)
  return err;

 if (key != mutex->key)
  return -EPERM;

 /* Compare against the unlocked state, and if true,
  * write the interface id into the top 16 bits, and
  * mark as locked.
 */
 value = nfp_mutex_locked(nfp_cpp_interface(cpp));

 /* We use test_set_imm here, as it implies a read
  * of the current state, and sets the bits in the
  * bytemask of the command to 1s. Since the mutex
  * is guaranteed to be 64-bit aligned, the bytemask
  * of this 32-bit command is ensured to be 8'b00001111,
  * which implies that the lower 4 bits will be set to
  * ones regardless of the initial state.
  *
  * Since this is a 'Readback' operation, with no Pull
  * data, we can treat this as a normal Push (read)
  * atomic, which returns the original value.
 */
 err = nfp_cpp_readl(cpp, mus, mutex->address, &tmp);
 if (err < 0)
  return err;

 /* Was it unlocked? */
 if (nfp_mutex_is_unlocked(tmp)) {
  /* The read value can only be 0x....0000 in the unlocked state.
   * If there was another contending for this lock, then
   * the lock state would be 0x....000f
 */

  /* Write our owner ID into the lock
   * While not strictly necessary, this helps with
   * debug and bookkeeping.
 */
  err = nfp_cpp_writel(cpp, muw, mutex->address, value);
  if (err < 0)
   return err;

  mutex->depth = 1;
  return 0;
 }

 return nfp_mutex_is_locked(tmp) ? -EBUSY : -EINVAL;
}

/**
 * nfp_cpp_mutex_reclaim() - Unlock mutex if held by local endpoint
 * @cpp: NFP CPP handle
 * @target: NFP CPP target ID (ie NFP_CPP_TARGET_CLS or NFP_CPP_TARGET_MU)
 * @address: Offset into the address space of the NFP CPP target ID
 *
 * Release lock if held by local system.  Extreme care is advised, call only
 * when no local lock users can exist.
 *
 * Return:      0 if the lock was OK, 1 if locked by us, -errno on invalid mutex
 */
int nfp_cpp_mutex_reclaim(struct nfp_cpp *cpp, int target,
     unsigned long long address)
{
 const u32 mur = NFP_CPP_ID(target, 3, 0); /* atomic_read */
 const u32 muw = NFP_CPP_ID(target, 4, 0); /* atomic_write */
 u16 interface = nfp_cpp_interface(cpp);
 int err;
 u32 tmp;

 err = nfp_cpp_mutex_validate(interface, &target, address);
 if (err)
  return err;

 /* Check lock */
 err = nfp_cpp_readl(cpp, mur, address, &tmp);
 if (err < 0)
  return err;

 if (nfp_mutex_is_unlocked(tmp) || nfp_mutex_owner(tmp) != interface)
  return 0;

 /* Bust the lock */
 err = nfp_cpp_writel(cpp, muw, address, nfp_mutex_unlocked(interface));
 if (err < 0)
  return err;

 return 1;
}