*     offload. Frames to transmit on the off-channel channel are transmitted
  *     normally except for the %IEEE80211_TX_CTL_TX_OFFCHAN flag. When the
  *     duration (which will always be non-zero) expires, the driver must call
- *     ieee80211_remain_on_channel_expired(). This callback may sleep.
+ *     ieee80211_remain_on_channel_expired().
+ *     The driver must not call ieee80211_remain_on_channel_expired() before
+ *     the TX status for a frame that was sent off-channel, otherwise the TX
+ *     status is reported to userspace in an invalid way.
+ *     Note that this callback may be called while the device is in IDLE and
+ *     must be accepted in this case.
+ *     This callback may sleep.
  * @cancel_remain_on_channel: Requests that an ongoing off-channel period is
  *     aborted before it expires. This callback may sleep.
  *
 
        local->hw_roc_cookie = 0;
        local->hw_roc_channel = NULL;
 
-       ieee80211_recalc_idle(local);
-
        return 0;
 }
 
        struct ieee80211_work *wk;
        const struct ieee80211_mgmt *mgmt = (void *)buf;
        u32 flags;
-       bool is_offchan = false;
+       bool is_offchan = false, in_hw_roc = false;
 
        if (dont_wait_for_ack)
                flags = IEEE80211_TX_CTL_NO_ACK;
        if (chan == local->hw_roc_channel) {
                /* TODO: check channel type? */
                is_offchan = false;
+               in_hw_roc = true;
                flags |= IEEE80211_TX_CTL_TX_OFFCHAN;
        }
 
         * wait is involved, we might otherwise not be on
         * the right channel for long enough!
         */
-       if (!is_offchan && !wait && !sdata->vif.bss_conf.idle) {
+       if (!is_offchan && !wait && (in_hw_roc || !sdata->vif.bss_conf.idle)) {
                ieee80211_tx_skb(sdata, skb);
                return 0;
        }
 
 {
        struct ieee80211_sub_if_data *sdata;
        int count = 0;
-       bool working = false, scanning = false, hw_roc = false;
+       bool working = false, scanning = false;
        struct ieee80211_work *wk;
        unsigned int led_trig_start = 0, led_trig_stop = 0;
 
                count++;
        }
 
-       list_for_each_entry(wk, &local->work_list, list) {
-               working = true;
-               wk->sdata->vif.bss_conf.idle = false;
+       if (!local->ops->remain_on_channel) {
+               list_for_each_entry(wk, &local->work_list, list) {
+                       working = true;
+                       wk->sdata->vif.bss_conf.idle = false;
+               }
        }
 
        if (local->scan_sdata &&
                local->scan_sdata->vif.bss_conf.idle = false;
        }
 
-       if (local->hw_roc_channel)
-               hw_roc = true;
-
        list_for_each_entry(sdata, &local->interfaces, list) {
                if (sdata->vif.type == NL80211_IFTYPE_MONITOR ||
                    sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
                ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_IDLE);
        }
 
-       if (working || scanning || hw_roc)
+       if (working || scanning)
                led_trig_start |= IEEE80211_TPT_LEDTRIG_FL_WORK;
        else
                led_trig_stop |= IEEE80211_TPT_LEDTRIG_FL_WORK;
 
        ieee80211_mod_tpt_led_trig(local, led_trig_start, led_trig_stop);
 
-       if (hw_roc)
-               return ieee80211_idle_off(local, "hw remain-on-channel");
        if (working)
                return ieee80211_idle_off(local, "working");
        if (scanning)
 
                return;
        }
 
-       if (!list_empty(&local->work_list)) {
-               local->ps_sdata = NULL;
-               goto change;
-       }
-
        list_for_each_entry(sdata, &local->interfaces, list) {
                if (!ieee80211_sdata_running(sdata))
                        continue;
                local->ps_sdata = NULL;
        }
 
- change:
        ieee80211_change_ps(local);
 }
 
 
                                          GFP_KERNEL);
        }
 
-       ieee80211_recalc_idle(local);
-
        mutex_unlock(&local->mtx);
 }
 
        local->hw_roc_channel = NULL;
        local->hw_roc_cookie = 0;
 
-       ieee80211_recalc_idle(local);
-
        mutex_unlock(&local->mtx);
 }