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

   1 // SPDX-License-Identifier: ISC
   2 /*
   3  * Copyright (C) 2016 Felix Fietkau <nbd@nbd.name>
   4  */
   5
   6 #include <linux/module.h>
   7 #include "mt76.h"
   8
   9 bool __mt76_poll(struct mt76_dev *dev, u32 offset, u32 mask, u32 val,
  10                  int timeout)
  11 {
  12         u32 cur;
  13
  14         timeout /= 10;
  15         do {
  16                 cur = __mt76_rr(dev, offset) & mask;
  17                 if (cur == val)
  18                         return true;
  19
  20                 udelay(10);
  21         } while (timeout-- > 0);
  22
  23         return false;
  24 }
  25 EXPORT_SYMBOL_GPL(__mt76_poll);
  26
  27 bool ____mt76_poll_msec(struct mt76_dev *dev, u32 offset, u32 mask, u32 val,
  28                         int timeout, int tick)
  29 {
  30         u32 cur;
  31
  32         timeout /= tick;
  33         do {
  34                 cur = __mt76_rr(dev, offset) & mask;
  35                 if (cur == val)
  36                         return true;
  37
  38                 usleep_range(1000 * tick, 2000 * tick);
  39         } while (timeout-- > 0);
  40
  41         return false;
  42 }
  43 EXPORT_SYMBOL_GPL(____mt76_poll_msec);
  44
  45 int mt76_wcid_alloc(u32 *mask, int size)
  46 {
  47         int i, idx = 0, cur;
  48
  49         for (i = 0; i < DIV_ROUND_UP(size, 32); i++) {
  50                 idx = ffs(~mask[i]);
  51                 if (!idx)
  52                         continue;
  53
  54                 idx--;
  55                 cur = i * 32 + idx;
  56                 if (cur >= size)
  57                         break;
  58
  59                 mask[i] |= BIT(idx);
  60                 return cur;
  61         }
  62
  63         return -1;
  64 }
  65 EXPORT_SYMBOL_GPL(mt76_wcid_alloc);
  66
  67 int mt76_get_min_avg_rssi(struct mt76_dev *dev, bool ext_phy)
  68 {
  69         struct mt76_wcid *wcid;
  70         int i, j, min_rssi = 0;
  71         s8 cur_rssi;
  72
  73         local_bh_disable();
  74         rcu_read_lock();
  75
  76         for (i = 0; i < ARRAY_SIZE(dev->wcid_mask); i++) {
  77                 u32 mask = dev->wcid_mask[i];
  78                 u32 phy_mask = dev->wcid_phy_mask[i];
  79
  80                 if (!mask)
  81                         continue;
  82
  83                 for (j = i * 32; mask; j++, mask >>= 1, phy_mask >>= 1) {
  84                         if (!(mask & 1))
  85                                 continue;
  86
  87                         if (!!(phy_mask & 1) != ext_phy)
  88                                 continue;
  89
  90                         wcid = rcu_dereference(dev->wcid[j]);
  91                         if (!wcid)
  92                                 continue;
  93
  94                         spin_lock(&dev->rx_lock);
  95                         if (wcid->inactive_count++ < 5)
  96                                 cur_rssi = -ewma_signal_read(&wcid->rssi);
  97                         else
  98                                 cur_rssi = 0;
  99                         spin_unlock(&dev->rx_lock);
 100
 101                         if (cur_rssi < min_rssi)
 102                                 min_rssi = cur_rssi;
 103                 }
 104         }
 105
 106         rcu_read_unlock();
 107         local_bh_enable();
 108
 109         return min_rssi;
 110 }
 111 EXPORT_SYMBOL_GPL(mt76_get_min_avg_rssi);
 112
 113 int __mt76_worker_fn(void *ptr)
 114 {
 115         struct mt76_worker *w = ptr;
 116
 117         while (!kthread_should_stop()) {
 118                 set_current_state(TASK_INTERRUPTIBLE);
 119
 120                 if (kthread_should_park()) {
 121                         kthread_parkme();
 122                         continue;
 123                 }
 124
 125                 if (!test_and_clear_bit(MT76_WORKER_SCHEDULED, &w->state)) {
 126                         schedule();
 127                         continue;
 128                 }
 129
 130                 set_bit(MT76_WORKER_RUNNING, &w->state);
 131                 set_current_state(TASK_RUNNING);
 132                 w->fn(w);
 133                 cond_resched();
 134                 clear_bit(MT76_WORKER_RUNNING, &w->state);
 135         }
 136
 137         return 0;
 138 }
 139 EXPORT_SYMBOL_GPL(__mt76_worker_fn);
 140
 141 MODULE_DESCRIPTION("MediaTek MT76x helpers");
 142 MODULE_LICENSE("Dual BSD/GPL");