117 lines · c
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_H8#define __MT76_UTIL_H9 10#include <linux/skbuff.h>11#include <linux/bitops.h>12#include <linux/bitfield.h>13#include <net/mac80211.h>14 15struct mt76_worker16{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 void33mt76_wcid_mask_set(u32 *mask, int idx)34{35 mask[idx / 32] |= BIT(idx % 32);36}37 38static inline void39mt76_wcid_mask_clear(u32 *mask, int idx)40{41 mask[idx / 32] &= ~BIT(idx % 32);42}43 44static inline void45mt76_skb_set_moredata(struct sk_buff *skb, bool enable)46{47 struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;48 49 if (enable)50 hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_MOREDATA);51 else52 hdr->frame_control &= ~cpu_to_le16(IEEE80211_FCTL_MOREDATA);53}54 55int __mt76_worker_fn(void *ptr);56 57static inline int58mt76_worker_setup(struct ieee80211_hw *hw, struct mt76_worker *w,59 void (*fn)(struct mt76_worker *),60 const char *name)61{62 const char *dev_name = wiphy_name(hw->wiphy);63 int ret;64 65 if (fn)66 w->fn = fn;67 w->task = kthread_run(__mt76_worker_fn, w,68 "mt76-%s %s", name, dev_name);69 70 if (IS_ERR(w->task)) {71 ret = PTR_ERR(w->task);72 w->task = NULL;73 return ret;74 }75 76 return 0;77}78 79static inline void mt76_worker_schedule(struct mt76_worker *w)80{81 if (!w->task)82 return;83 84 if (!test_and_set_bit(MT76_WORKER_SCHEDULED, &w->state) &&85 !test_bit(MT76_WORKER_RUNNING, &w->state))86 wake_up_process(w->task);87}88 89static inline void mt76_worker_disable(struct mt76_worker *w)90{91 if (!w->task)92 return;93 94 kthread_park(w->task);95 WRITE_ONCE(w->state, 0);96}97 98static inline void mt76_worker_enable(struct mt76_worker *w)99{100 if (!w->task)101 return;102 103 kthread_unpark(w->task);104 mt76_worker_schedule(w);105}106 107static inline void mt76_worker_teardown(struct mt76_worker *w)108{109 if (!w->task)110 return;111 112 kthread_stop(w->task);113 w->task = NULL;114}115 116#endif117