drivers/net/wireless/mediatek/mt76/util.h

   1 /* SPDX-License-Identifier: GPL-2.0-only */
   2 /*
   3  * Copyright (C) 2016 Felix Fietkau <nbd@nbd.name>
   4  * Copyright (C) 2004 - 2009 Ivo van Doorn <IvDoorn@gmail.com>
   5  */
   6
   7 #ifndef __MT76_UTIL_H
   8 #define __MT76_UTIL_H
   9
  10 #include <linux/skbuff.h>
  11 #include <linux/bitops.h>
  12 #include <linux/bitfield.h>
  13 #include <net/mac80211.h>
  14
  15 struct mt76_worker
  16 {
  17         struct task_struct *task;
  18         void (*fn)(struct mt76_worker *);
  19         unsigned long state;
  20 };
  21
  22 enum {
  23         MT76_WORKER_SCHEDULED,
  24         MT76_WORKER_RUNNING,
  25 };
  26
  27 #define MT76_INCR(_var, _size) \
  28         (_var = (((_var) + 1) % (_size)))
  29
  30 int mt76_wcid_alloc(u32 *mask, int size);
  31
  32 static inline bool
  33 mt76_wcid_mask_test(u32 *mask, int idx)
  34 {
  35         return mask[idx / 32] & BIT(idx % 32);
  36 }
  37
  38 static inline void
  39 mt76_wcid_mask_set(u32 *mask, int idx)
  40 {
  41         mask[idx / 32] |= BIT(idx % 32);
  42 }
  43
  44 static inline void
  45 mt76_wcid_mask_clear(u32 *mask, int idx)
  46 {
  47         mask[idx / 32] &= ~BIT(idx % 32);
  48 }
  49
  50 static inline void
  51 mt76_skb_set_moredata(struct sk_buff *skb, bool enable)
  52 {
  53         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
  54
  55         if (enable)
  56                 hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_MOREDATA);
  57         else
  58                 hdr->frame_control &= ~cpu_to_le16(IEEE80211_FCTL_MOREDATA);
  59 }
  60
  61 int __mt76_worker_fn(void *ptr);
  62
  63 static inline int
  64 mt76_worker_setup(struct ieee80211_hw *hw, struct mt76_worker *w,
  65                   void (*fn)(struct mt76_worker *),
  66                   const char *name)
  67 {
  68         const char *dev_name = wiphy_name(hw->wiphy);
  69         int ret;
  70
  71         if (fn)
  72                 w->fn = fn;
  73         w->task = kthread_create(__mt76_worker_fn, w, "mt76-%s %s",
  74                                  name, dev_name);
  75
  76         ret = PTR_ERR_OR_ZERO(w->task);
  77         if (ret) {
  78                 w->task = NULL;
  79                 return ret;
  80         }
  81
  82         wake_up_process(w->task);
  83
  84         return 0;
  85 }
  86
  87 static inline void mt76_worker_schedule(struct mt76_worker *w)
  88 {
  89         if (!w->task)
  90                 return;
  91
  92         if (!test_and_set_bit(MT76_WORKER_SCHEDULED, &w->state) &&
  93             !test_bit(MT76_WORKER_RUNNING, &w->state))
  94                 wake_up_process(w->task);
  95 }
  96
  97 static inline void mt76_worker_disable(struct mt76_worker *w)
  98 {
  99         if (!w->task)
 100                 return;
 101
 102         kthread_park(w->task);
 103         WRITE_ONCE(w->state, 0);
 104 }
 105
 106 static inline void mt76_worker_enable(struct mt76_worker *w)
 107 {
 108         if (!w->task)
 109                 return;
 110
 111         kthread_unpark(w->task);
 112         mt76_worker_schedule(w);
 113 }
 114
 115 static inline void mt76_worker_teardown(struct mt76_worker *w)
 116 {
 117         if (!w->task)
 118                 return;
 119
 120         kthread_stop(w->task);
 121         w->task = NULL;
 122 }
 123
 124 #endif