/* 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 */