ARK-OS/arkos/kernel/net/wireless/tests/util.h

/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * Utilities for cfg80211 unit testing
 *
 * Copyright (C) 2023 Intel Corporation
 */
#ifndef __CFG80211_UTILS_H
#define __CFG80211_UTILS_H

#define CHAN2G(_freq)  { \
	.band = NL80211_BAND_2GHZ, \
	.center_freq = (_freq), \
	.hw_value = (_freq), \
}

static const struct ieee80211_channel channels_2ghz[] = {
	CHAN2G(2412), /* Channel 1 */
	CHAN2G(2417), /* Channel 2 */
	CHAN2G(2422), /* Channel 3 */
	CHAN2G(2427), /* Channel 4 */
	CHAN2G(2432), /* Channel 5 */
	CHAN2G(2437), /* Channel 6 */
	CHAN2G(2442), /* Channel 7 */
	CHAN2G(2447), /* Channel 8 */
	CHAN2G(2452), /* Channel 9 */
	CHAN2G(2457), /* Channel 10 */
	CHAN2G(2462), /* Channel 11 */
	CHAN2G(2467), /* Channel 12 */
	CHAN2G(2472), /* Channel 13 */
	CHAN2G(2484), /* Channel 14 */
};

struct t_wiphy_priv {
	struct kunit *test;
	struct cfg80211_ops *ops;

	void *ctx;

	struct ieee80211_supported_band band_2ghz;
	struct ieee80211_channel channels_2ghz[ARRAY_SIZE(channels_2ghz)];
};

#define T_WIPHY(test, ctx) ({						\
		struct wiphy *__wiphy =					\
			kunit_alloc_resource(test, t_wiphy_init,	\
					     t_wiphy_exit,		\
					     GFP_KERNEL, &(ctx));	\
									\
		KUNIT_ASSERT_NOT_NULL(test, __wiphy);			\
		__wiphy;						\
	})
#define t_wiphy_ctx(wiphy) (((struct t_wiphy_priv *)wiphy_priv(wiphy))->ctx)

int t_wiphy_init(struct kunit_resource *resource, void *data);
void t_wiphy_exit(struct kunit_resource *resource);

#define t_skb_remove_member(skb, type, member)	do {				\
		memmove((skb)->data + (skb)->len - sizeof(type) +		\
			offsetof(type, member),					\
			(skb)->data + (skb)->len - sizeof(type) +		\
			offsetofend(type, member),				\
			offsetofend(type, member));				\
		skb_trim(skb, (skb)->len - sizeof_field(type, member));		\
	} while (0)

#endif /* __CFG80211_UTILS_H */