// 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);