/*
 * Copyright (c) 2013 Qualcomm Atheros, Inc.
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <linux/export.h>
#include <linux/relay.h>
#include <linux/random.h>
#include "ath9k.h"

static s8 fix_rssi_inv_only(u8 rssi_val)
{
 if (rssi_val == 128)
  rssi_val = 0;
 return (s8) rssi_val;
}

static void ath_debug_send_fft_sample(struct ath_spec_scan_priv *spec_priv,
          struct fft_sample_tlv *fft_sample_tlv)
{
 int length;
 if (!spec_priv->rfs_chan_spec_scan)
  return;

 length = __be16_to_cpu(fft_sample_tlv->length) +
   sizeof(*fft_sample_tlv);
 relay_write(spec_priv->rfs_chan_spec_scan, fft_sample_tlv, length);
}

typedef int (ath_cmn_fft_idx_validator) (u8 *sample_end, int bytes_read);

static int
ath_cmn_max_idx_verify_ht20_fft(u8 *sample_end, int bytes_read)
{
 struct ath_ht20_mag_info *mag_info;
 u8 *sample;
 u16 max_magnitude;
 u8 max_index;
 u8 max_exp;

 /* Sanity check so that we don't read outside the read
  * buffer
 */
 if (bytes_read < SPECTRAL_HT20_SAMPLE_LEN - 1)
  return -1;

 mag_info = (struct ath_ht20_mag_info *) (sample_end -
    sizeof(struct ath_ht20_mag_info) + 1);

 sample = sample_end - SPECTRAL_HT20_SAMPLE_LEN + 1;

 max_index = spectral_max_index_ht20(mag_info->all_bins);
 max_magnitude = spectral_max_magnitude(mag_info->all_bins);

 max_exp = mag_info->max_exp & 0xf;

 /* Don't try to read something outside the read buffer
  * in case of a missing byte (so bins[0] will be outside
  * the read buffer)
 */
 if (bytes_read < SPECTRAL_HT20_SAMPLE_LEN && max_index < 1)
  return -1;

 if ((sample[max_index] & 0xf8) != ((max_magnitude >> max_exp) & 0xf8))
  return -1;
 else
  return 0;
}

static int
ath_cmn_max_idx_verify_ht20_40_fft(u8 *sample_end, int bytes_read)
{
 struct ath_ht20_40_mag_info *mag_info;
 u8 *sample;
 u16 lower_mag, upper_mag;
 u8 lower_max_index, upper_max_index;
 u8 max_exp;
 int dc_pos = SPECTRAL_HT20_40_NUM_BINS / 2;

 /* Sanity check so that we don't read outside the read
  * buffer
 */
 if (bytes_read < SPECTRAL_HT20_40_SAMPLE_LEN - 1)
  return -1;

 mag_info = (struct ath_ht20_40_mag_info *) (sample_end -
    sizeof(struct ath_ht20_40_mag_info) + 1);

 sample = sample_end - SPECTRAL_HT20_40_SAMPLE_LEN + 1;

 lower_mag = spectral_max_magnitude(mag_info->lower_bins);
 lower_max_index = spectral_max_index_ht40(mag_info->lower_bins);

 upper_mag = spectral_max_magnitude(mag_info->upper_bins);
 upper_max_index = spectral_max_index_ht40(mag_info->upper_bins);

 max_exp = mag_info->max_exp & 0xf;

 /* Don't try to read something outside the read buffer
  * in case of a missing byte (so bins[0] will be outside
  * the read buffer)
 */
 if (bytes_read < SPECTRAL_HT20_40_SAMPLE_LEN &&
    ((upper_max_index < 1) || (lower_max_index < 1)))
  return -1;

 if (((sample[upper_max_index + dc_pos] & 0xf8) !=
      ((upper_mag >> max_exp) & 0xf8)) ||
     ((sample[lower_max_index] & 0xf8) !=
      ((lower_mag >> max_exp) & 0xf8)))
  return -1;
 else
  return 0;
}

typedef int (ath_cmn_fft_sample_handler) (struct ath_rx_status *rs,
   struct ath_spec_scan_priv *spec_priv,
   u8 *sample_buf, u64 tsf, u16 freq, int chan_type);

static int
ath_cmn_process_ht20_fft(struct ath_rx_status *rs,
   struct ath_spec_scan_priv *spec_priv,
   u8 *sample_buf,
   u64 tsf, u16 freq, int chan_type)
{
 struct fft_sample_ht20 fft_sample_20;
 struct ath_common *common = ath9k_hw_common(spec_priv->ah);
 struct ath_hw *ah = spec_priv->ah;
 struct ath_ht20_mag_info *mag_info;
 struct fft_sample_tlv *tlv;
 int i = 0;
 int ret = 0;
 int dc_pos = SPECTRAL_HT20_NUM_BINS / 2;
 u16 magnitude, tmp_mag, length;
 u8 max_index, bitmap_w, max_exp;

 length = sizeof(fft_sample_20) - sizeof(struct fft_sample_tlv);
 fft_sample_20.tlv.type = ATH_FFT_SAMPLE_HT20;
 fft_sample_20.tlv.length = __cpu_to_be16(length);
 fft_sample_20.freq = __cpu_to_be16(freq);
 fft_sample_20.rssi = fix_rssi_inv_only(rs->rs_rssi_ctl[0]);
 fft_sample_20.noise = ah->noise;

 mag_info = (struct ath_ht20_mag_info *) (sample_buf +
     SPECTRAL_HT20_NUM_BINS);

 magnitude = spectral_max_magnitude(mag_info->all_bins);
 fft_sample_20.max_magnitude = __cpu_to_be16(magnitude);

 max_index = spectral_max_index_ht20(mag_info->all_bins);
 fft_sample_20.max_index = max_index;

 bitmap_w = spectral_bitmap_weight(mag_info->all_bins);
 fft_sample_20.bitmap_weight = bitmap_w;

 max_exp = mag_info->max_exp & 0xf;
 fft_sample_20.max_exp = max_exp;

 fft_sample_20.tsf = __cpu_to_be64(tsf);

 memcpy(fft_sample_20.data, sample_buf, SPECTRAL_HT20_NUM_BINS);

 ath_dbg(common, SPECTRAL_SCAN, "FFT HT20 frame: max mag 0x%X,"
     "max_mag_idx %i\n",
     magnitude >> max_exp,
     max_index);

 if ((fft_sample_20.data[max_index] & 0xf8) !=
     ((magnitude >> max_exp) & 0xf8)) {
  ath_dbg(common, SPECTRAL_SCAN, "Magnitude mismatch !\n");
  ret = -1;
 }

 /* DC value (value in the middle) is the blind spot of the spectral
  * sample and invalid, interpolate it.
 */
 fft_sample_20.data[dc_pos] = (fft_sample_20.data[dc_pos + 1] +
     fft_sample_20.data[dc_pos - 1]) / 2;

 /* Check if the maximum magnitude is indeed maximum,
  * also if the maximum value was at dc_pos, calculate
  * a new one (since value at dc_pos is invalid).
 */
 if (max_index == dc_pos) {
  tmp_mag = 0;
  for (i = 0; i < dc_pos; i++) {
   if (fft_sample_20.data[i] > tmp_mag) {
    tmp_mag = fft_sample_20.data[i];
    fft_sample_20.max_index = i;
   }
  }

  magnitude = tmp_mag << max_exp;
  fft_sample_20.max_magnitude = __cpu_to_be16(magnitude);

  ath_dbg(common, SPECTRAL_SCAN,
   "Calculated new lower max 0x%X at %i\n",
   tmp_mag, fft_sample_20.max_index);
 } else
 for (i = 0; i < SPECTRAL_HT20_NUM_BINS; i++) {
  if (fft_sample_20.data[i] == (magnitude >> max_exp))
   ath_dbg(common, SPECTRAL_SCAN,
    "Got max: 0x%X at index %i\n",
    fft_sample_20.data[i], i);

  if (fft_sample_20.data[i] > (magnitude >> max_exp)) {
   ath_dbg(common, SPECTRAL_SCAN,
    "Got bin %i greater than max: 0x%X\n",
    i, fft_sample_20.data[i]);
   ret = -1;
  }
 }

 if (ret < 0)
  return ret;

 tlv = (struct fft_sample_tlv *)&fft_sample_20;

 ath_debug_send_fft_sample(spec_priv, tlv);

 return 0;
}

static int
ath_cmn_process_ht20_40_fft(struct ath_rx_status *rs,
   struct ath_spec_scan_priv *spec_priv,
   u8 *sample_buf,
   u64 tsf, u16 freq, int chan_type)
{
 struct fft_sample_ht20_40 fft_sample_40;
 struct ath_common *common = ath9k_hw_common(spec_priv->ah);
 struct ath_hw *ah = spec_priv->ah;
 struct ath9k_hw_cal_data *caldata = ah->caldata;
 struct ath_ht20_40_mag_info *mag_info;
 struct fft_sample_tlv *tlv;
 int dc_pos = SPECTRAL_HT20_40_NUM_BINS / 2;
 int i = 0;
 int ret = 0;
 s16 ext_nf;
 u16 lower_mag, upper_mag, tmp_mag, length;
 s8 lower_rssi, upper_rssi;
 u8 lower_max_index, upper_max_index;
 u8 lower_bitmap_w, upper_bitmap_w, max_exp;

 if (caldata)
  ext_nf = ath9k_hw_getchan_noise(ah, ah->curchan,
    caldata->nfCalHist[3].privNF);
 else
  ext_nf = ATH_DEFAULT_NOISE_FLOOR;

 length = sizeof(fft_sample_40) - sizeof(struct fft_sample_tlv);
 fft_sample_40.tlv.type = ATH_FFT_SAMPLE_HT20_40;
 fft_sample_40.tlv.length = __cpu_to_be16(length);
 fft_sample_40.freq = __cpu_to_be16(freq);
 fft_sample_40.channel_type = chan_type;

 if (chan_type == NL80211_CHAN_HT40PLUS) {
  lower_rssi = fix_rssi_inv_only(rs->rs_rssi_ctl[0]);
  upper_rssi = fix_rssi_inv_only(rs->rs_rssi_ext[0]);

  fft_sample_40.lower_noise = ah->noise;
  fft_sample_40.upper_noise = ext_nf;
 } else {
  lower_rssi = fix_rssi_inv_only(rs->rs_rssi_ext[0]);
  upper_rssi = fix_rssi_inv_only(rs->rs_rssi_ctl[0]);

  fft_sample_40.lower_noise = ext_nf;
  fft_sample_40.upper_noise = ah->noise;
 }

 fft_sample_40.lower_rssi = lower_rssi;
 fft_sample_40.upper_rssi = upper_rssi;

 mag_info = (struct ath_ht20_40_mag_info *) (sample_buf +
     SPECTRAL_HT20_40_NUM_BINS);

 lower_mag = spectral_max_magnitude(mag_info->lower_bins);
 fft_sample_40.lower_max_magnitude = __cpu_to_be16(lower_mag);

 upper_mag = spectral_max_magnitude(mag_info->upper_bins);
 fft_sample_40.upper_max_magnitude = __cpu_to_be16(upper_mag);

 lower_max_index = spectral_max_index_ht40(mag_info->lower_bins);
 fft_sample_40.lower_max_index = lower_max_index;

 upper_max_index = spectral_max_index_ht40(mag_info->upper_bins);
 fft_sample_40.upper_max_index = upper_max_index;

 lower_bitmap_w = spectral_bitmap_weight(mag_info->lower_bins);
 fft_sample_40.lower_bitmap_weight = lower_bitmap_w;

 upper_bitmap_w = spectral_bitmap_weight(mag_info->upper_bins);
 fft_sample_40.upper_bitmap_weight = upper_bitmap_w;

 max_exp = mag_info->max_exp & 0xf;
 fft_sample_40.max_exp = max_exp;

 fft_sample_40.tsf = __cpu_to_be64(tsf);

 memcpy(fft_sample_40.data, sample_buf, SPECTRAL_HT20_40_NUM_BINS);

 ath_dbg(common, SPECTRAL_SCAN, "FFT HT20/40 frame: lower mag 0x%X,"
     "lower_mag_idx %i, upper mag 0x%X,"
     "upper_mag_idx %i\n",
     lower_mag >> max_exp,
     lower_max_index,
     upper_mag >> max_exp,
     upper_max_index);

 /* Check if we got the expected magnitude values at
  * the expected bins
 */
 if (((fft_sample_40.data[upper_max_index + dc_pos] & 0xf8)
     != ((upper_mag >> max_exp) & 0xf8)) ||
    ((fft_sample_40.data[lower_max_index] & 0xf8)
     != ((lower_mag >> max_exp) & 0xf8))) {
  ath_dbg(common, SPECTRAL_SCAN, "Magnitude mismatch !\n");
  ret = -1;
 }

 /* DC value (value in the middle) is the blind spot of the spectral
  * sample and invalid, interpolate it.
 */
 fft_sample_40.data[dc_pos] = (fft_sample_40.data[dc_pos + 1] +
     fft_sample_40.data[dc_pos - 1]) / 2;

 /* Check if the maximum magnitudes are indeed maximum,
  * also if the maximum value was at dc_pos, calculate
  * a new one (since value at dc_pos is invalid).
 */
 if (lower_max_index == dc_pos) {
  tmp_mag = 0;
  for (i = 0; i < dc_pos; i++) {
   if (fft_sample_40.data[i] > tmp_mag) {
    tmp_mag = fft_sample_40.data[i];
    fft_sample_40.lower_max_index = i;
   }
  }

  lower_mag = tmp_mag << max_exp;
  fft_sample_40.lower_max_magnitude = __cpu_to_be16(lower_mag);

  ath_dbg(common, SPECTRAL_SCAN,
   "Calculated new lower max 0x%X at %i\n",
   tmp_mag, fft_sample_40.lower_max_index);
 } else
 for (i = 0; i < dc_pos; i++) {
  if (fft_sample_40.data[i] == (lower_mag >> max_exp))
   ath_dbg(common, SPECTRAL_SCAN,
    "Got lower mag: 0x%X at index %i\n",
    fft_sample_40.data[i], i);

  if (fft_sample_40.data[i] > (lower_mag >> max_exp)) {
   ath_dbg(common, SPECTRAL_SCAN,
    "Got lower bin %i higher than max: 0x%X\n",
    i, fft_sample_40.data[i]);
   ret = -1;
  }
 }

 if (upper_max_index == dc_pos) {
  tmp_mag = 0;
  for (i = dc_pos; i < SPECTRAL_HT20_40_NUM_BINS; i++) {
   if (fft_sample_40.data[i] > tmp_mag) {
    tmp_mag = fft_sample_40.data[i];
    fft_sample_40.upper_max_index = i;
   }
  }
  upper_mag = tmp_mag << max_exp;
  fft_sample_40.upper_max_magnitude = __cpu_to_be16(upper_mag);

  ath_dbg(common, SPECTRAL_SCAN,
   "Calculated new upper max 0x%X at %i\n",
   tmp_mag, fft_sample_40.upper_max_index);
 } else
 for (i = dc_pos; i < SPECTRAL_HT20_40_NUM_BINS; i++) {
  if (fft_sample_40.data[i] == (upper_mag >> max_exp))
   ath_dbg(common, SPECTRAL_SCAN,
    "Got upper mag: 0x%X at index %i\n",
    fft_sample_40.data[i], i);

  if (fft_sample_40.data[i] > (upper_mag >> max_exp)) {
   ath_dbg(common, SPECTRAL_SCAN,
    "Got upper bin %i higher than max: 0x%X\n",
    i, fft_sample_40.data[i]);

   ret = -1;
  }
 }

 if (ret < 0)
  return ret;

 tlv = (struct fft_sample_tlv *)&fft_sample_40;

 ath_debug_send_fft_sample(spec_priv, tlv);

 return 0;
}

static inline void
ath_cmn_copy_fft_frame(u8 *in, u8 *out, int sample_len, int sample_bytes)
{
 switch (sample_bytes - sample_len) {
 case -1:
  /* First byte missing */
  memcpy(&out[1], in,
         sample_len - 1);
  break;
 case 0:
  /* Length correct, nothing to do. */
  memcpy(out, in, sample_len);
  break;
 case 1:
  /* MAC added 2 extra bytes AND first byte
   * is missing.
 */
  memcpy(&out[1], in, 30);
  out[31] = in[31];
  memcpy(&out[32], &in[33],
         sample_len - 32);
  break;
 case 2:
  /* MAC added 2 extra bytes at bin 30 and 32,
   * remove them.
 */
  memcpy(out, in, 30);
  out[30] = in[31];
  memcpy(&out[31], &in[33],
         sample_len - 31);
  break;
 default:
  break;
 }
}

static int
ath_cmn_is_fft_buf_full(struct ath_spec_scan_priv *spec_priv)
{
 int i = 0;
 int ret = 0;
 struct rchan_buf *buf;
 struct rchan *rc = spec_priv->rfs_chan_spec_scan;

 for_each_possible_cpu(i) {
  if ((buf = *per_cpu_ptr(rc->buf, i))) {
   ret += relay_buf_full(buf);
  }
 }

 if (ret)
  return 1;
 else
  return 0;
}

/* returns 1 if this was a spectral frame, even if not handled. */
int ath_cmn_process_fft(struct ath_spec_scan_priv *spec_priv, struct ieee80211_hdr *hdr,
      struct ath_rx_status *rs, u64 tsf)
{
 u8 sample_buf[SPECTRAL_SAMPLE_MAX_LEN] = {0};
 struct ath_hw *ah = spec_priv->ah;
 struct ath_common *common = ath9k_hw_common(spec_priv->ah);
 struct ath_softc *sc = common->priv;
 u8 num_bins, *vdata = (u8 *)hdr;
 struct ath_radar_info *radar_info;
 int len = rs->rs_datalen;
 int i;
 int got_slen = 0;
 u8  *sample_start;
 int sample_bytes = 0;
 int ret = 0;
 u16 fft_len, sample_len, freq = ah->curchan->chan->center_freq;
 enum nl80211_channel_type chan_type;
 ath_cmn_fft_idx_validator *fft_idx_validator;
 ath_cmn_fft_sample_handler *fft_handler;

 /* AR9280 and before report via ATH9K_PHYERR_RADAR, AR93xx and newer
  * via ATH9K_PHYERR_SPECTRAL. Haven't seen ATH9K_PHYERR_FALSE_RADAR_EXT
  * yet, but this is supposed to be possible as well.
 */
 if (rs->rs_phyerr != ATH9K_PHYERR_RADAR &&
     rs->rs_phyerr != ATH9K_PHYERR_FALSE_RADAR_EXT &&
     rs->rs_phyerr != ATH9K_PHYERR_SPECTRAL)
  return 0;

 /* check if spectral scan bit is set. This does not have to be checked
  * if received through a SPECTRAL phy error, but shouldn't hurt.
 */
 radar_info = ((struct ath_radar_info *)&vdata[len]) - 1;
 if (!(radar_info->pulse_bw_info & SPECTRAL_SCAN_BITMASK))
  return 0;

 if (!spec_priv->rfs_chan_spec_scan)
  return 1;

 /* Output buffers are full, no need to process anything
  * since there is no space to put the result anyway
 */
 ret = ath_cmn_is_fft_buf_full(spec_priv);
 if (ret == 1) {
  ath_dbg(common, SPECTRAL_SCAN, "FFT report ignored, no space "
      "left on output buffers\n");
  return 1;
 }

 chan_type = cfg80211_get_chandef_type(&common->hw->conf.chandef);
 if ((chan_type == NL80211_CHAN_HT40MINUS) ||
     (chan_type == NL80211_CHAN_HT40PLUS)) {
  fft_len = SPECTRAL_HT20_40_TOTAL_DATA_LEN;
  sample_len = SPECTRAL_HT20_40_SAMPLE_LEN;
  num_bins = SPECTRAL_HT20_40_NUM_BINS;
  fft_idx_validator = &ath_cmn_max_idx_verify_ht20_40_fft;
  fft_handler = &ath_cmn_process_ht20_40_fft;
 } else {
  fft_len = SPECTRAL_HT20_TOTAL_DATA_LEN;
  sample_len = SPECTRAL_HT20_SAMPLE_LEN;
  num_bins = SPECTRAL_HT20_NUM_BINS;
  fft_idx_validator = ath_cmn_max_idx_verify_ht20_fft;
  fft_handler = &ath_cmn_process_ht20_fft;
 }

 ath_dbg(common, SPECTRAL_SCAN, "Got radar dump bw_info: 0x%X,"
     "len: %i fft_len: %i\n",
     radar_info->pulse_bw_info,
     len,
     fft_len);
 sample_start = vdata;
 for (i = 0; i < len - 2; i++) {
  sample_bytes++;

  /* Only a single sample received, no need to look
   * for the sample's end, do the correction based
   * on the packet's length instead. Note that hw
   * will always put the radar_info structure on
   * the end.
 */
  if (len <= fft_len + 2) {
   sample_bytes = len - sizeof(struct ath_radar_info);
   got_slen = 1;
  }

  /* Search for the end of the FFT frame between
   * sample_len - 1 and sample_len + 2. exp_max is 3
   * bits long and it's the only value on the last
   * byte of the frame so since it'll be smaller than
   * the next byte (the first bin of the next sample)
   * 90% of the time, we can use it as a separator.
 */
  if (vdata[i] <= 0x7 && sample_bytes >= sample_len - 1) {

   /* Got a frame length within boundaries, there are
    * four scenarios here:
    *
    * a) sample_len -> We got the correct length
    * b) sample_len + 2 -> 2 bytes added around bin[31]
    * c) sample_len - 1 -> The first byte is missing
    * d) sample_len + 1 -> b + c at the same time
    *
    * When MAC adds 2 extra bytes, bin[31] and bin[32]
    * have the same value, so we can use that for further
    * verification in cases b and d.
 */

   /* Did we go too far ? If so we couldn't determine
    * this sample's boundaries, discard any further
    * data
 */
   if ((sample_bytes > sample_len + 2) ||
      ((sample_bytes > sample_len) &&
      (sample_start[31] != sample_start[32])))
    break;

   /* See if we got a valid frame by checking the
    * consistency of mag_info fields. This is to
    * prevent from "fixing" a correct frame.
    * Failure is non-fatal, later frames may
    * be valid.
 */
   if (!fft_idx_validator(&vdata[i], i)) {
    ath_dbg(common, SPECTRAL_SCAN,
     "Found valid fft frame at %i\n", i);
    got_slen = 1;
   }

   /* We expect 1 - 2 more bytes */
   else if ((sample_start[31] == sample_start[32]) &&
    (sample_bytes >= sample_len) &&
    (sample_bytes < sample_len + 2) &&
    (vdata[i + 1] <= 0x7))
    continue;

   /* Try to distinguish cases a and c */
   else if ((sample_bytes == sample_len - 1) &&
    (vdata[i + 1] <= 0x7))
    continue;

   got_slen = 1;
  }

  if (got_slen) {
   ath_dbg(common, SPECTRAL_SCAN, "FFT frame len: %i\n",
    sample_bytes);

   /* Only try to fix a frame if it's the only one
    * on the report, else just skip it.
 */
   if (sample_bytes != sample_len && len <= fft_len + 2) {
    ath_cmn_copy_fft_frame(sample_start,
             sample_buf, sample_len,
             sample_bytes);

    ret = fft_handler(rs, spec_priv, sample_buf,
        tsf, freq, chan_type);

    if (ret == 0)
     RX_STAT_INC(sc, rx_spectral_sample_good);
    else
     RX_STAT_INC(sc, rx_spectral_sample_err);

    /* Mix the received bins to the /dev/random
     * pool
 */
    add_device_randomness(sample_buf, num_bins);

    memset(sample_buf, 0, SPECTRAL_SAMPLE_MAX_LEN);
   }

   /* Process a normal frame */
   if (sample_bytes == sample_len) {
    ret = fft_handler(rs, spec_priv, sample_start,
        tsf, freq, chan_type);

    if (ret == 0)
     RX_STAT_INC(sc, rx_spectral_sample_good);
    else
     RX_STAT_INC(sc, rx_spectral_sample_err);

    /* Mix the received bins to the /dev/random
     * pool
 */
    add_device_randomness(sample_start, num_bins);
   }

   /* Short report processed, break out of the
    * loop.
 */
   if (len <= fft_len + 2)
    return 1;

   sample_start = &vdata[i + 1];

   /* -1 to grab sample_len -1, -2 since
    * they 'll get increased by one. In case
    * of failure try to recover by going byte
    * by byte instead.
 */
   if (ret == 0) {
    i += num_bins - 2;
    sample_bytes = num_bins - 2;
   }
   got_slen = 0;
  }
 }

 i -= num_bins - 2;
 if (len - i != sizeof(struct ath_radar_info))
  ath_dbg(common, SPECTRAL_SCAN, "FFT report truncated"
      "(bytes left: %i)\n",
      len - i);
 return 1;
}
EXPORT_SYMBOL(ath_cmn_process_fft);

/*********************/
/* spectral_scan_ctl */
/*********************/

static ssize_t read_file_spec_scan_ctl(struct file *file, char __user *user_buf,
           size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 char *mode = "";
 unsigned int len;

 switch (spec_priv->spectral_mode) {
 case SPECTRAL_DISABLED:
  mode = "disable";
  break;
 case SPECTRAL_BACKGROUND:
  mode = "background";
  break;
 case SPECTRAL_CHANSCAN:
  mode = "chanscan";
  break;
 case SPECTRAL_MANUAL:
  mode = "manual";
  break;
 }
 len = strlen(mode);
 return simple_read_from_buffer(user_buf, count, ppos, mode, len);
}

void ath9k_cmn_spectral_scan_trigger(struct ath_common *common,
     struct ath_spec_scan_priv *spec_priv)
{
 struct ath_hw *ah = spec_priv->ah;
 u32 rxfilter;

 if (IS_ENABLED(CONFIG_ATH9K_TX99))
  return;

 if (!ath9k_hw_ops(ah)->spectral_scan_trigger) {
  ath_err(common, "spectrum analyzer not implemented on this hardware\n");
  return;
 }

 if (!spec_priv->spec_config.enabled)
  return;

 ath_ps_ops(common)->wakeup(common);
 rxfilter = ath9k_hw_getrxfilter(ah);
 ath9k_hw_setrxfilter(ah, rxfilter |
     ATH9K_RX_FILTER_PHYRADAR |
     ATH9K_RX_FILTER_PHYERR);

 /* TODO: usually this should not be necessary, but for some reason
  * (or in some mode?) the trigger must be called after the
  * configuration, otherwise the register will have its values reset
  * (on my ar9220 to value 0x01002310)
 */
 ath9k_cmn_spectral_scan_config(common, spec_priv, spec_priv->spectral_mode);
 ath9k_hw_ops(ah)->spectral_scan_trigger(ah);
 ath_ps_ops(common)->restore(common);
}
EXPORT_SYMBOL(ath9k_cmn_spectral_scan_trigger);

int ath9k_cmn_spectral_scan_config(struct ath_common *common,
          struct ath_spec_scan_priv *spec_priv,
          enum spectral_mode spectral_mode)
{
 struct ath_hw *ah = spec_priv->ah;

 if (!ath9k_hw_ops(ah)->spectral_scan_trigger) {
  ath_err(common, "spectrum analyzer not implemented on this hardware\n");
  return -1;
 }

 switch (spectral_mode) {
 case SPECTRAL_DISABLED:
  spec_priv->spec_config.enabled = 0;
  break;
 case SPECTRAL_BACKGROUND:
  /* send endless samples.
   * TODO: is this really useful for "background"?
 */
  spec_priv->spec_config.endless = 1;
  spec_priv->spec_config.enabled = 1;
  break;
 case SPECTRAL_CHANSCAN:
 case SPECTRAL_MANUAL:
  spec_priv->spec_config.endless = 0;
  spec_priv->spec_config.enabled = 1;
  break;
 default:
  return -1;
 }

 ath_ps_ops(common)->wakeup(common);
 ath9k_hw_ops(ah)->spectral_scan_config(ah, &spec_priv->spec_config);
 ath_ps_ops(common)->restore(common);

 spec_priv->spectral_mode = spectral_mode;

 return 0;
}
EXPORT_SYMBOL(ath9k_cmn_spectral_scan_config);

static ssize_t write_file_spec_scan_ctl(struct file *file,
     const char __user *user_buf,
     size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 struct ath_common *common = ath9k_hw_common(spec_priv->ah);
 char buf[32];
 ssize_t len;

 if (IS_ENABLED(CONFIG_ATH9K_TX99))
  return -EOPNOTSUPP;

 len = min(count, sizeof(buf) - 1);
 if (copy_from_user(buf, user_buf, len))
  return -EFAULT;

 buf[len] = '\0';

 if (strncmp("trigger", buf, 7) == 0) {
  ath9k_cmn_spectral_scan_trigger(common, spec_priv);
 } else if (strncmp("background", buf, 10) == 0) {
  ath9k_cmn_spectral_scan_config(common, spec_priv, SPECTRAL_BACKGROUND);
  ath_dbg(common, CONFIG, "spectral scan: background mode enabled\n");
 } else if (strncmp("chanscan", buf, 8) == 0) {
  ath9k_cmn_spectral_scan_config(common, spec_priv, SPECTRAL_CHANSCAN);
  ath_dbg(common, CONFIG, "spectral scan: channel scan mode enabled\n");
 } else if (strncmp("manual", buf, 6) == 0) {
  ath9k_cmn_spectral_scan_config(common, spec_priv, SPECTRAL_MANUAL);
  ath_dbg(common, CONFIG, "spectral scan: manual mode enabled\n");
 } else if (strncmp("disable", buf, 7) == 0) {
  ath9k_cmn_spectral_scan_config(common, spec_priv, SPECTRAL_DISABLED);
  ath_dbg(common, CONFIG, "spectral scan: disabled\n");
 } else {
  return -EINVAL;
 }

 return count;
}

static const struct file_operations fops_spec_scan_ctl = {
 .read = read_file_spec_scan_ctl,
 .write = write_file_spec_scan_ctl,
 .open = simple_open,
 .owner = THIS_MODULE,
 .llseek = default_llseek,
};

/*************************/
/* spectral_short_repeat */
/*************************/

static ssize_t read_file_spectral_short_repeat(struct file *file,
            char __user *user_buf,
            size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 char buf[32];
 unsigned int len;

 len = sprintf(buf, "%d\n", spec_priv->spec_config.short_repeat);
 return simple_read_from_buffer(user_buf, count, ppos, buf, len);
}

static ssize_t write_file_spectral_short_repeat(struct file *file,
      const char __user *user_buf,
      size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 unsigned long val;
 ssize_t ret;

 ret = kstrtoul_from_user(user_buf, count, 0, &val);
 if (ret)
  return ret;

 if (val > 1)
  return -EINVAL;

 spec_priv->spec_config.short_repeat = val;
 return count;
}

static const struct file_operations fops_spectral_short_repeat = {
 .read = read_file_spectral_short_repeat,
 .write = write_file_spectral_short_repeat,
 .open = simple_open,
 .owner = THIS_MODULE,
 .llseek = default_llseek,
};

/******************/
/* spectral_count */
/******************/

static ssize_t read_file_spectral_count(struct file *file,
     char __user *user_buf,
     size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 char buf[32];
 unsigned int len;

 len = sprintf(buf, "%d\n", spec_priv->spec_config.count);
 return simple_read_from_buffer(user_buf, count, ppos, buf, len);
}

static ssize_t write_file_spectral_count(struct file *file,
      const char __user *user_buf,
      size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 unsigned long val;
 ssize_t ret;

 ret = kstrtoul_from_user(user_buf, count, 0, &val);
 if (ret)
  return ret;
 if (val > 255)
  return -EINVAL;

 spec_priv->spec_config.count = val;
 return count;
}

static const struct file_operations fops_spectral_count = {
 .read = read_file_spectral_count,
 .write = write_file_spectral_count,
 .open = simple_open,
 .owner = THIS_MODULE,
 .llseek = default_llseek,
};

/*******************/
/* spectral_period */
/*******************/

static ssize_t read_file_spectral_period(struct file *file,
      char __user *user_buf,
      size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 char buf[32];
 unsigned int len;

 len = sprintf(buf, "%d\n", spec_priv->spec_config.period);
 return simple_read_from_buffer(user_buf, count, ppos, buf, len);
}

static ssize_t write_file_spectral_period(struct file *file,
       const char __user *user_buf,
       size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 unsigned long val;
 ssize_t ret;

 ret = kstrtoul_from_user(user_buf, count, 0, &val);
 if (ret)
  return ret;

 if (val > 255)
  return -EINVAL;

 spec_priv->spec_config.period = val;
 return count;
}

static const struct file_operations fops_spectral_period = {
 .read = read_file_spectral_period,
 .write = write_file_spectral_period,
 .open = simple_open,
 .owner = THIS_MODULE,
 .llseek = default_llseek,
};

/***********************/
/* spectral_fft_period */
/***********************/

static ssize_t read_file_spectral_fft_period(struct file *file,
          char __user *user_buf,
          size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 char buf[32];
 unsigned int len;

 len = sprintf(buf, "%d\n", spec_priv->spec_config.fft_period);
 return simple_read_from_buffer(user_buf, count, ppos, buf, len);
}

static ssize_t write_file_spectral_fft_period(struct file *file,
           const char __user *user_buf,
           size_t count, loff_t *ppos)
{
 struct ath_spec_scan_priv *spec_priv = file->private_data;
 unsigned long val;
 ssize_t ret;

 ret = kstrtoul_from_user(user_buf, count, 0, &val);
 if (ret)
  return ret;

 if (val > 15)
  return -EINVAL;

 spec_priv->spec_config.fft_period = val;
 return count;
}

static const struct file_operations fops_spectral_fft_period = {
 .read = read_file_spectral_fft_period,
 .write = write_file_spectral_fft_period,
 .open = simple_open,
 .owner = THIS_MODULE,
 .llseek = default_llseek,
};

/*******************/
/* Relay interface */
/*******************/

static struct dentry *create_buf_file_handler(const char *filename,
           struct dentry *parent,
           umode_t mode,
           struct rchan_buf *buf,
           int *is_global)
{
 struct dentry *buf_file;

 buf_file = debugfs_create_file(filename, mode, parent, buf,
           &relay_file_operations);
 if (IS_ERR(buf_file))
  return NULL;

 *is_global = 1;
 return buf_file;
}

static int remove_buf_file_handler(struct dentry *dentry)
{
 debugfs_remove(dentry);

 return 0;
}

static const struct rchan_callbacks rfs_spec_scan_cb = {
 .create_buf_file = create_buf_file_handler,
 .remove_buf_file = remove_buf_file_handler,
};

/*********************/
/* Debug Init/Deinit */
/*********************/

void ath9k_cmn_spectral_deinit_debug(struct ath_spec_scan_priv *spec_priv)
{
 if (spec_priv->rfs_chan_spec_scan) {
  relay_close(spec_priv->rfs_chan_spec_scan);
  spec_priv->rfs_chan_spec_scan = NULL;
 }
}
EXPORT_SYMBOL(ath9k_cmn_spectral_deinit_debug);

void ath9k_cmn_spectral_init_debug(struct ath_spec_scan_priv *spec_priv,
       struct dentry *debugfs_phy)
{
 spec_priv->rfs_chan_spec_scan = relay_open("spectral_scan",
         debugfs_phy,
         1024, 256, &rfs_spec_scan_cb,
         NULL);
 if (!spec_priv->rfs_chan_spec_scan)
  return;

 debugfs_create_file("spectral_scan_ctl",
       0600,
       debugfs_phy, spec_priv,
       &fops_spec_scan_ctl);
 debugfs_create_file("spectral_short_repeat",
       0600,
       debugfs_phy, spec_priv,
       &fops_spectral_short_repeat);
 debugfs_create_file("spectral_count",
       0600,
       debugfs_phy, spec_priv,
       &fops_spectral_count);
 debugfs_create_file("spectral_period",
       0600,
       debugfs_phy, spec_priv,
       &fops_spectral_period);
 debugfs_create_file("spectral_fft_period",
       0600,
       debugfs_phy, spec_priv,
       &fops_spectral_fft_period);
}
EXPORT_SYMBOL(ath9k_cmn_spectral_init_debug);