Hardware station lookup for pspoll frames can fail, which makes the driver
ignore ps-poll frames. Fix the resulting powersave issues by looking up
the station for pspoll frames in software
Signed-off-by: Felix Fietkau <nbd@nbd.name>
        struct mt76_wcid *wcid = status->wcid;
        bool ps;
 
+       if (ieee80211_is_pspoll(hdr->frame_control) && !wcid) {
+               sta = ieee80211_find_sta_by_ifaddr(dev->hw, hdr->addr2, NULL);
+               if (sta)
+                       wcid = status->wcid = (struct mt76_wcid *) sta->drv_priv;
+       }
+
        if (!wcid || !wcid->sta)
                return;