// SPDX-License-Identifier: GPL-2.0
/*
 * IEEE 802.15.4 PAN management
 *
 * Copyright (C) 2023 Qorvo US, Inc
 * Authors:
 *   - David Girault <david.girault@qorvo.com>
 *   - Miquel Raynal <miquel.raynal@bootlin.com>
 */

#include <linux/kernel.h>
#include <net/cfg802154.h>
#include <net/af_ieee802154.h>

/* Checks whether a device address matches one from the PAN list.
 * This helper is meant to be used only during PAN management, when we expect
 * extended addresses to be used.
 */
static bool cfg802154_pan_device_is_matching(struct ieee802154_pan_device *pan_dev,
          struct ieee802154_addr *ext_dev)
{
 if (!pan_dev || !ext_dev)
  return false;

 if (ext_dev->mode == IEEE802154_ADDR_SHORT)
  return false;

 return pan_dev->extended_addr == ext_dev->extended_addr;
}

bool cfg802154_device_is_associated(struct wpan_dev *wpan_dev)
{
 bool is_assoc;

 mutex_lock(&wpan_dev->association_lock);
 is_assoc = !list_empty(&wpan_dev->children) || wpan_dev->parent;
 mutex_unlock(&wpan_dev->association_lock);

 return is_assoc;
}

bool cfg802154_device_is_parent(struct wpan_dev *wpan_dev,
    struct ieee802154_addr *target)
{
 lockdep_assert_held(&wpan_dev->association_lock);

 return cfg802154_pan_device_is_matching(wpan_dev->parent, target);
}
EXPORT_SYMBOL_GPL(cfg802154_device_is_parent);

struct ieee802154_pan_device *
cfg802154_device_is_child(struct wpan_dev *wpan_dev,
     struct ieee802154_addr *target)
{
 struct ieee802154_pan_device *child;

 lockdep_assert_held(&wpan_dev->association_lock);

 list_for_each_entry(child, &wpan_dev->children, node)
  if (cfg802154_pan_device_is_matching(child, target))
   return child;

 return NULL;
}
EXPORT_SYMBOL_GPL(cfg802154_device_is_child);

__le16 cfg802154_get_free_short_addr(struct wpan_dev *wpan_dev)
{
 struct ieee802154_pan_device *child;
 __le16 addr;

 lockdep_assert_held(&wpan_dev->association_lock);

 do {
  get_random_bytes(&addr, 2);
  if (addr == cpu_to_le16(IEEE802154_ADDR_SHORT_BROADCAST) ||
      addr == cpu_to_le16(IEEE802154_ADDR_SHORT_UNSPEC))
   continue;

  if (wpan_dev->short_addr == addr)
   continue;

  if (wpan_dev->parent && wpan_dev->parent->short_addr == addr)
   continue;

  list_for_each_entry(child, &wpan_dev->children, node)
   if (child->short_addr == addr)
    continue;

  break;
 } while (1);

 return addr;
}
EXPORT_SYMBOL_GPL(cfg802154_get_free_short_addr);

unsigned int cfg802154_set_max_associations(struct wpan_dev *wpan_dev,
         unsigned int max)
{
 unsigned int old_max;

 lockdep_assert_held(&wpan_dev->association_lock);

 old_max = wpan_dev->max_associations;
 wpan_dev->max_associations = max;

 return old_max;
}
EXPORT_SYMBOL_GPL(cfg802154_set_max_associations);