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