/*
 * AP mode helper functions
 * Copyright (c) 2009, Jouni Malinen <j@w1.fi>
 *
 * This software may be distributed under the terms of the BSD license.
 * See README for more details.
 */

#include "includes.h"

#include "common.h"
#include "common/ieee802_11_defs.h"
#include "fst/fst.h"
#include "sta_info.h"
#include "hostapd.h"


int hostapd_register_probereq_cb(struct hostapd_data *hapd,
				 int (*cb)(void *ctx, const u8 *sa,
					   const u8 *da, const u8 *bssid,
					   const u8 *ie, size_t ie_len,
					   int ssi_signal),
				 void *ctx)
{
	struct hostapd_probereq_cb *n;

	n = os_realloc_array(hapd->probereq_cb, hapd->num_probereq_cb + 1,
			     sizeof(struct hostapd_probereq_cb));
	if (n == NULL)
		return -1;

	hapd->probereq_cb = n;
	n = &hapd->probereq_cb[hapd->num_probereq_cb];
	hapd->num_probereq_cb++;

	n->cb = cb;
	n->ctx = ctx;

	return 0;
}


struct prune_data {
	struct hostapd_data *hapd;
	const u8 *addr;
	int mld_assoc_link_id;
};

static int prune_associations(struct hostapd_iface *iface, void *ctx)
{
	struct prune_data *data = ctx;
	struct sta_info *osta;
	struct hostapd_data *ohapd;
	size_t j;

	for (j = 0; j < iface->num_bss; j++) {
		ohapd = iface->bss[j];
		if (ohapd == data->hapd)
			continue;
#ifdef CONFIG_TESTING_OPTIONS
		if (ohapd->conf->skip_prune_assoc)
			continue;
#endif /* CONFIG_TESTING_OPTIONS */
#ifdef CONFIG_FST
		/* Don't prune STAs belong to same FST */
		if (ohapd->iface->fst &&
		    data->hapd->iface->fst &&
		    fst_are_ifaces_aggregated(ohapd->iface->fst,
					      data->hapd->iface->fst))
			continue;
#endif /* CONFIG_FST */
		osta = ap_get_sta(ohapd, data->addr);
		if (!osta)
			continue;

#ifdef CONFIG_IEEE80211BE
		if (data->mld_assoc_link_id >= 0 &&
		    osta->mld_assoc_link_id == data->mld_assoc_link_id)
			continue;
#endif /* CONFIG_IEEE80211BE */

		wpa_printf(MSG_INFO, "%s: Prune association for " MACSTR,
			   ohapd->conf->iface, MAC2STR(osta->addr));
		ap_sta_disassociate(ohapd, osta, WLAN_REASON_UNSPECIFIED);
	}

	return 0;
}

/**
 * hostapd_prune_associations - Remove extraneous associations
 * @hapd: Pointer to BSS data for the most recent association
 * @addr: Associated STA address
 * @mld_assoc_link_id: MLD link id used for association or -1 for non MLO
 *
 * This function looks through all radios and BSS's for previous
 * (stale) associations of STA. If any are found they are removed.
 */
void hostapd_prune_associations(struct hostapd_data *hapd, const u8 *addr,
				int mld_assoc_link_id)
{
	struct prune_data data;

	data.hapd = hapd;
	data.addr = addr;
	data.mld_assoc_link_id = mld_assoc_link_id;

	if (hapd->iface->interfaces &&
	    hapd->iface->interfaces->for_each_interface)
		hapd->iface->interfaces->for_each_interface(
			hapd->iface->interfaces, prune_associations, &data);
}