LXR linux/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
  15struct mt76_worker
  16{
  17        struct task_struct *task;
  18        void (*fn)(struct mt76_worker *);
  19        unsigned long state;
  20};
  21
  22enum {
  23        MT76_WORKER_SCHEDULED,
  24        MT76_WORKER_RUNNING,
  25};
  26
  27#define MT76_INCR(_var, _size) \
  28        (_var = (((_var) + 1) % (_size)))
  29
  30int mt76_wcid_alloc(u32 *mask, int size);
  31
  32static inline bool
  33mt76_wcid_mask_test(u32 *mask, int idx)
  34{
  35        return mask[idx / 32] & BIT(idx % 32);
  36}
  37
  38static inline void
  39mt76_wcid_mask_set(u32 *mask, int idx)
  40{
  41        mask[idx / 32] |= BIT(idx % 32);
  42}
  43
  44static inline void
  45mt76_wcid_mask_clear(u32 *mask, int idx)
  46{
  47        mask[idx / 32] &= ~BIT(idx % 32);
  48}
  49
  50static inline void
  51mt76_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
  61int __mt76_worker_fn(void *ptr);
  62
  63static inline int
  64mt76_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
  87static 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
  97static 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
 106static 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
 115static 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
 125