#include <net/mac80211.h>
#include <net/rtnetlink.h>

#include "ieee80211_i.h"
#include "mesh.h"
#include "driver-ops.h"
#include "led.h"

static void ieee80211_sched_scan_cancel(struct ieee80211_local *local)
{
        if (ieee80211_request_sched_scan_stop(local))
                return;
        cfg80211_sched_scan_stopped_rtnl(local->hw.wiphy, 0);
}

int __ieee80211_suspend(struct ieee80211_hw *hw, struct cfg80211_wowlan *wowlan)
{
        struct ieee80211_local *local = hw_to_local(hw);
        struct ieee80211_sub_if_data *sdata;
        struct sta_info *sta;

        if (!local->open_count)
                goto suspend;

        ieee80211_scan_cancel(local);

        ieee80211_dfs_cac_cancel(local);

        ieee80211_roc_purge(local, NULL);

        ieee80211_del_virtual_monitor(local);

        if (ieee80211_hw_check(hw, AMPDU_AGGREGATION) &&
            !(wowlan && wowlan->any)) {
                mutex_lock(&local->sta_mtx);
                list_for_each_entry(sta, &local->sta_list, list) {
                        set_sta_flag(sta, WLAN_STA_BLOCK_BA);
                        ieee80211_sta_tear_down_BA_sessions(
                                        sta, AGG_STOP_LOCAL_REQUEST);
                }
                mutex_unlock(&local->sta_mtx);
        }

        /* keep sched_scan only in case of 'any' trigger */
        if (!(wowlan && wowlan->any))
                ieee80211_sched_scan_cancel(local);

        ieee80211_stop_queues_by_reason(hw,
                                        IEEE80211_MAX_QUEUE_MAP,
                                        IEEE80211_QUEUE_STOP_REASON_SUSPEND,
                                        false);

        /* flush out all packets */
        synchronize_net();

        ieee80211_flush_queues(local, NULL, true);

        local->quiescing = true;
        /* make quiescing visible to timers everywhere */
        mb();

        flush_workqueue(local->workqueue);

        /* Don't try to run timers while suspended. */
        del_timer_sync(&local->sta_cleanup);

         /*
         * Note that this particular timer doesn't need to be
         * restarted at resume.
         */
        cancel_work_sync(&local->dynamic_ps_enable_work);
        del_timer_sync(&local->dynamic_ps_timer);

        local->wowlan = wowlan;
        if (local->wowlan) {
                int err;

                /* Drivers don't expect to suspend while some operations like
                 * authenticating or associating are in progress. It doesn't
                 * make sense anyway to accept that, since the authentication
                 * or association would never finish since the driver can't do
                 * that on its own.
                 * Thus, clean up in-progress auth/assoc first.
                 */
                list_for_each_entry(sdata, &local->interfaces, list) {
                        if (!ieee80211_sdata_running(sdata))
                                continue;
                        if (sdata->vif.type != NL80211_IFTYPE_STATION)
                                continue;
                        ieee80211_mgd_quiesce(sdata);
                        /* If suspended during TX in progress, and wowlan
                         * is enabled (connection will be active) there
                         * can be a race where the driver is put out
                         * of power-save due to TX and during suspend
                         * dynamic_ps_timer is cancelled and TX packet
                         * is flushed, leaving the driver in ACTIVE even
                         * after resuming until dynamic_ps_timer puts
                         * driver back in DOZE.
                         */
                        if (sdata->u.mgd.associated &&
                            sdata->u.mgd.powersave &&
                             !(local->hw.conf.flags & IEEE80211_CONF_PS)) {
                                local->hw.conf.flags |= IEEE80211_CONF_PS;
                                ieee80211_hw_config(local,
                                                    IEEE80211_CONF_CHANGE_PS);
                        }
                }

                err = drv_suspend(local, wowlan);
                if (err < 0) {
                        local->quiescing = false;
                        local->wowlan = false;
                        if (ieee80211_hw_check(hw, AMPDU_AGGREGATION)) {
                                mutex_lock(&local->sta_mtx);
                                list_for_each_entry(sta,
                                                    &local->sta_list, list) {
                                        clear_sta_flag(sta, WLAN_STA_BLOCK_BA);
                                }
                                mutex_unlock(&local->sta_mtx);
                        }
                        ieee80211_wake_queues_by_reason(hw,
                                        IEEE80211_MAX_QUEUE_MAP,
                                        IEEE80211_QUEUE_STOP_REASON_SUSPEND,
                                        false);
                        return err;
                } else if (err > 0) {
                        WARN_ON(err != 1);
                        /* cfg80211 will call back into mac80211 to disconnect
                         * all interfaces, allow that to proceed properly
                         */
                        ieee80211_wake_queues_by_reason(hw,
                                        IEEE80211_MAX_QUEUE_MAP,
                                        IEEE80211_QUEUE_STOP_REASON_SUSPEND,
                                        false);
                        return err;
                } else {
                        goto suspend;
                }
        }

        /* remove all interfaces that were created in the driver */
        list_for_each_entry(sdata, &local->interfaces, list) {
                if (!ieee80211_sdata_running(sdata))
                        continue;
                switch (sdata->vif.type) {
                case NL80211_IFTYPE_AP_VLAN:
                case NL80211_IFTYPE_MONITOR:
                        continue;
                case NL80211_IFTYPE_STATION:
                        ieee80211_mgd_quiesce(sdata);
                        break;
                case NL80211_IFTYPE_WDS:
                        /* tear down aggregation sessions and remove STAs */
                        mutex_lock(&local->sta_mtx);
                        sta = sdata->u.wds.sta;
                        if (sta && sta->uploaded) {
                                enum ieee80211_sta_state state;

                                state = sta->sta_state;
                                for (; state > IEEE80211_STA_NOTEXIST; state--)
                                        WARN_ON(drv_sta_state(local, sta->sdata,
                                                              sta, state,
                                                              state - 1));
                        }
                        mutex_unlock(&local->sta_mtx);
                        break;
                default:
                        break;
                }

                flush_delayed_work(&sdata->dec_tailroom_needed_wk);
                drv_remove_interface(local, sdata);
        }

        /*
         * We disconnected on all interfaces before suspend, all channel
         * contexts should be released.
         */
        WARN_ON(!list_empty(&local->chanctx_list));

        /* stop hardware - this must stop RX */
        ieee80211_stop_device(local);

 suspend:
        local->suspended = true;
        /* need suspended to be visible before quiescing is false */
        barrier();
        local->quiescing = false;

        return 0;
}

/*
 * __ieee80211_resume() is a static inline which just calls
 * ieee80211_reconfig(), which is also needed for hardware
 * hang/firmware failure/etc. recovery.
 */

void ieee80211_report_wowlan_wakeup(struct ieee80211_vif *vif,
                                    struct cfg80211_wowlan_wakeup *wakeup,
                                    gfp_t gfp)
{
        struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);

        cfg80211_report_wowlan_wakeup(&sdata->wdev, wakeup, gfp);
}
EXPORT_SYMBOL(ieee80211_report_wowlan_wakeup);