/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2013 Adrian Chadd <adrian@FreeBSD.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer,
 *    without modification.
 * 2. Redistributions in binary form must reproduce at minimum a disclaimer
 *    similar to the "NO WARRANTY" disclaimer below ("Disclaimer") and any
 *    redistribution must be conditioned upon including a substantially
 *    similar Disclaimer requirement for further binary redistribution.
 *
 * NO WARRANTY
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF NONINFRINGEMENT, MERCHANTIBILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
 * THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR SPECIAL, EXEMPLARY,
 * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 * THE POSSIBILITY OF SUCH DAMAGES.
 */
#include <sys/cdefs.h>
/*
 * Implement some basic spectral scan control logic.
 */
#include "opt_ath.h"
#include "opt_inet.h"
#include "opt_wlan.h"

#include <sys/param.h>
#include <sys/systm.h> 
#include <sys/sysctl.h>
#include <sys/kernel.h>
#include <sys/lock.h>
#include <sys/malloc.h>
#include <sys/mutex.h>
#include <sys/errno.h>

#include <machine/bus.h>
#include <machine/resource.h>
#include <sys/bus.h>

#include <sys/socket.h>

#include <net/if.h>
#include <net/if_var.h>
#include <net/if_media.h>
#include <net/if_arp.h>
#include <net/ethernet.h>               /* XXX for ether_sprintf */

#include <net80211/ieee80211_var.h>

#include <net/bpf.h>

#ifdef INET
#include <netinet/in.h>
#include <netinet/if_ether.h>
#endif

#include <dev/ath/if_athvar.h>
#include <dev/ath/if_ath_spectral.h>
#include <dev/ath/if_ath_misc.h>

#include <dev/ath/ath_hal/ah_desc.h>

struct ath_spectral_state {
        HAL_SPECTRAL_PARAM      spectral_state;

        /*
         * Should we enable spectral scan upon
         * each network interface reset/change?
         *
         * This is intended to allow spectral scan
         * frame reporting during channel scans.
         *
         * Later on it can morph into a larger
         * scale config method where it pushes
         * a "channel scan" config into the hardware
         * rather than just the spectral_state
         * config.
         */
        int spectral_enable_after_reset;
};

/*
 * Methods which are required
 */

/*
 * Attach spectral to the given interface
 */
int
ath_spectral_attach(struct ath_softc *sc)
{
        struct ath_spectral_state *ss;

        /*
         * If spectral isn't supported, don't error - just
         * quietly complete.
         */
        if (! ath_hal_spectral_supported(sc->sc_ah))
                return (0);

        ss = malloc(sizeof(struct ath_spectral_state),
            M_TEMP, M_WAITOK | M_ZERO);
        sc->sc_spectral = ss;

        (void) ath_hal_spectral_get_config(sc->sc_ah, &ss->spectral_state);

        return (0);
}

/*
 * Detach spectral from the given interface
 */
int
ath_spectral_detach(struct ath_softc *sc)
{

        if (! ath_hal_spectral_supported(sc->sc_ah))
                return (0);

        if (sc->sc_spectral != NULL) {
                free(sc->sc_spectral, M_TEMP);
        }
        return (0);
}

/*
 * Check whether spectral needs enabling and if so,
 * flip it on.
 */
int
ath_spectral_enable(struct ath_softc *sc, struct ieee80211_channel *ch)
{
        struct ath_spectral_state *ss = sc->sc_spectral;

        /* Default to disable spectral PHY reporting */
        sc->sc_dospectral = 0;

        if (ss == NULL)
                return (0);

        if (ss->spectral_enable_after_reset) {
                ath_hal_spectral_configure(sc->sc_ah,
                    &ss->spectral_state);
                (void) ath_hal_spectral_start(sc->sc_ah);
                sc->sc_dospectral = 1;
        }
        return (0);
}

/*
 * Handle ioctl requests from the diagnostic interface.
 *
 * The initial part of this code resembles ath_ioctl_diag();
 * it's likely a good idea to reduce duplication between
 * these two routines.
 */
int
ath_ioctl_spectral(struct ath_softc *sc, struct ath_diag *ad)
{
        unsigned int id = ad->ad_id & ATH_DIAG_ID;
        void *indata = NULL;
        void *outdata = NULL;
        u_int32_t insize = ad->ad_in_size;
        u_int32_t outsize = ad->ad_out_size;
        int error = 0;
        HAL_SPECTRAL_PARAM peout;
        HAL_SPECTRAL_PARAM *pe;
        struct ath_spectral_state *ss = sc->sc_spectral;
        int val;

        if (! ath_hal_spectral_supported(sc->sc_ah))
                return (EINVAL);

        ATH_LOCK(sc);
        ath_power_set_power_state(sc, HAL_PM_AWAKE);
        ATH_UNLOCK(sc);

        if (ad->ad_id & ATH_DIAG_IN) {
                /*
                 * Copy in data.
                 */
                indata = malloc(insize, M_TEMP, M_NOWAIT);
                if (indata == NULL) {
                        error = ENOMEM;
                        goto bad;
                }
                error = copyin(ad->ad_in_data, indata, insize);
                if (error)
                        goto bad;
        }
        if (ad->ad_id & ATH_DIAG_DYN) {
                /*
                 * Allocate a buffer for the results (otherwise the HAL
                 * returns a pointer to a buffer where we can read the
                 * results).  Note that we depend on the HAL leaving this
                 * pointer for us to use below in reclaiming the buffer;
                 * may want to be more defensive.
                 */
                outdata = malloc(outsize, M_TEMP, M_NOWAIT | M_ZERO);
                if (outdata == NULL) {
                        error = ENOMEM;
                        goto bad;
                }
        }
        switch (id) {
                case SPECTRAL_CONTROL_GET_PARAMS:
                        memset(&peout, 0, sizeof(peout));
                        outsize = sizeof(HAL_SPECTRAL_PARAM);
                        ath_hal_spectral_get_config(sc->sc_ah, &peout);
                        pe = (HAL_SPECTRAL_PARAM *) outdata;
                        memcpy(pe, &peout, sizeof(*pe));
                        break;
                case SPECTRAL_CONTROL_SET_PARAMS:
                        if (insize < sizeof(HAL_SPECTRAL_PARAM)) {
                                error = EINVAL;
                                break;
                        }
                        pe = (HAL_SPECTRAL_PARAM *) indata;
                        ath_hal_spectral_configure(sc->sc_ah, pe);
                        /* Save a local copy of the updated parameters */
                        ath_hal_spectral_get_config(sc->sc_ah,
                            &ss->spectral_state);
                        break;
                case SPECTRAL_CONTROL_START:
                        ath_hal_spectral_configure(sc->sc_ah,
                            &ss->spectral_state);
                        (void) ath_hal_spectral_start(sc->sc_ah);
                        sc->sc_dospectral = 1;
                        /* XXX need to update the PHY mask in the driver */
                        break;
                case SPECTRAL_CONTROL_STOP:
                        (void) ath_hal_spectral_stop(sc->sc_ah);
                        sc->sc_dospectral = 0;
                        /* XXX need to update the PHY mask in the driver */
                        break;
                case SPECTRAL_CONTROL_ENABLE_AT_RESET:
                        if (insize < sizeof(int)) {
                                device_printf(sc->sc_dev, "%d != %d\n",
                                    insize,
                                    (int) sizeof(int));
                                error = EINVAL;
                                break;
                        }
                        if (indata == NULL) {
                                device_printf(sc->sc_dev, "indata=NULL\n");
                                error = EINVAL;
                                break;
                        }
                        val = * ((int *) indata);
                        if (val == 0)
                                ss->spectral_enable_after_reset = 0;
                        else
                                ss->spectral_enable_after_reset = 1;
                        break;
                case SPECTRAL_CONTROL_ENABLE:
                        /* XXX TODO */
                case SPECTRAL_CONTROL_DISABLE:
                        /* XXX TODO */
                break;
                default:
                        error = EINVAL;
                        goto bad;
        }
        if (outsize < ad->ad_out_size)
                ad->ad_out_size = outsize;
        if (outdata && copyout(outdata, ad->ad_out_data, ad->ad_out_size))
                error = EFAULT;
bad:
        if ((ad->ad_id & ATH_DIAG_IN) && indata != NULL)
                free(indata, M_TEMP);
        if ((ad->ad_id & ATH_DIAG_DYN) && outdata != NULL)
                free(outdata, M_TEMP);
        ATH_LOCK(sc);
        ath_power_restore_power_state(sc);
        ATH_UNLOCK(sc);

        return (error);
}