1 /*
2 * Copyright (c) 2020-2025 Bjoern A. Zeeb <bz@FreeBSD.org>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7 #ifndef _MT76_UTIL_H
8 #define _MT76_UTIL_H
9
10 #include <linux/kthread.h>
11
12 struct mt76_worker
13 {
14 void(*fn)(struct mt76_worker *);
15 struct task_struct *task;
16 unsigned long state;
17 };
18
19 enum mt76_worker_state {
20 MT76_WORKER_SCHEDULED,
21 MT76_WORKER_RUNNING,
22 };
23
24 #if 0
25 bool __mt76_poll(struct mt76_dev *, u32, u32, u32, int);
26 bool __mt76_poll_msec(struct mt76_dev *, u32, u32, u32, int);
27 int mt76_get_min_avg_rssi(struct mt76_dev *, bool);
28 #endif
29 int mt76_wcid_alloc(u32 *, int);
30 int __mt76_worker_fn(void *);
31
32 /* wcid_phy_mask is [32] */
33 static inline void
mt76_wcid_mask_set(u32 * mask,u16 bit)34 mt76_wcid_mask_set(u32 *mask, u16 bit)
35 {
36
37 mask[bit / 32] |= BIT(bit % 32);
38 }
39
40 static inline void
mt76_wcid_mask_clear(u32 * mask,u16 bit)41 mt76_wcid_mask_clear(u32 *mask, u16 bit)
42 {
43
44 mask[bit / 32] &= ~BIT(bit % 32);
45 }
46
47 /* See, e.g., __mt76_worker_fn for some details. */
48 static inline int
mt76_worker_setup(struct ieee80211_hw * hw __unused,struct mt76_worker * w,void (* wfunc)(struct mt76_worker *),const char * name)49 mt76_worker_setup(struct ieee80211_hw *hw __unused, struct mt76_worker *w,
50 void (*wfunc)(struct mt76_worker *), const char *name)
51 {
52 int error;
53
54 if (wfunc)
55 w->fn = wfunc;
56
57 w->task = kthread_run(__mt76_worker_fn, w,
58 "mt76-%s", name);
59
60 if (!IS_ERR(w->task))
61 return (0);
62
63 error = PTR_ERR(w->task);
64 w->task = NULL;
65 return (error);
66 }
67
68 static inline void
mt76_worker_schedule(struct mt76_worker * w)69 mt76_worker_schedule(struct mt76_worker *w)
70 {
71
72 if (w->task == NULL)
73 return;
74
75 if (!test_and_set_bit(MT76_WORKER_SCHEDULED, &w->state) ||
76 !test_bit(MT76_WORKER_RUNNING, &w->state))
77 wake_up_process(w->task);
78 }
79
80 static inline void
mt76_worker_enable(struct mt76_worker * w)81 mt76_worker_enable(struct mt76_worker *w)
82 {
83
84 if (w->task == NULL)
85 return;
86
87 kthread_unpark(w->task);
88 mt76_worker_schedule(w);
89 }
90
91 static inline void
mt76_worker_disable(struct mt76_worker * w)92 mt76_worker_disable(struct mt76_worker *w)
93 {
94
95 if (w->task == NULL)
96 return;
97
98 kthread_park(w->task);
99 WRITE_ONCE(w->state, 0);
100 }
101
102 static inline void
mt76_worker_teardown(struct mt76_worker * w)103 mt76_worker_teardown(struct mt76_worker *w)
104 {
105
106 if (w->task == NULL)
107 return;
108
109 kthread_stop(w->task);
110 w->task = NULL;
111 }
112
113 static inline void
mt76_skb_set_moredata(struct sk_buff * skb,bool moredata)114 mt76_skb_set_moredata(struct sk_buff *skb, bool moredata)
115 {
116 /*
117 * This would be net80211::IEEE80211_FC1_MORE_DATA
118 * Implement it as mostly LinuxKPI 802.11 to avoid
119 * further header pollution and possible conflicts.
120 */
121 struct ieee80211_hdr *hdr;
122 uint16_t val;
123
124 hdr = (struct ieee80211_hdr *)skb->data;
125 val = cpu_to_le16(IEEE80211_FC1_MORE_DATA << 8);
126 if (!moredata)
127 hdr->frame_control &= ~val;
128 else
129 hdr->frame_control |= val;
130 }
131
132 #endif /* _MT76_UTIL_H */
133