/*-
 * Copyright (c) 2014 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 <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <fcntl.h>
#include <unistd.h>
#include <string.h>
#include <err.h>
#include <net/if.h>
#include <sys/endian.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/sysctl.h>

#include "net80211/ieee80211_ioctl.h"
#include "net80211/ieee80211_radiotap.h"

#include "if_iwn_ioctl.h"
#include "if_iwnreg.h"
#include "iwnstats.h"
#include "iwn_ioctl.h"

#define	IWN_DEFAULT_IF		"iwn0"

static struct iwnstats *
iwnstats_new(const char *ifname)
{
	struct iwnstats *is;
	char buf[128];

	is = calloc(1, sizeof(struct iwnstats));
	if (is == NULL)
		return (NULL);

	snprintf(buf, sizeof(buf), "/dev/%s", ifname);
	is->s = open(buf, O_RDWR);
	if (is->s < 0)
		err(1, "open");

	return (is);
}

static void
iwn_stats_phy_print(struct iwn_rx_phy_stats *rxphy, const char *prefix)
{

	printf("%s: %s: ina=%d, fina=%d, bad_plcp=%d, bad_crc32=%d, overrun=%d, eoverrun=%d\n",
	        __func__,
		prefix,
		le32toh(rxphy->ina),
		le32toh(rxphy->fina),
		le32toh(rxphy->bad_plcp),
		le32toh(rxphy->bad_crc32),
		le32toh(rxphy->overrun),
		le32toh(rxphy->eoverrun));

	printf("%s: %s: fa=%d, bad_fina_sync=%d, sfd_timeout=%d, fina_timeout=%d, no_rts_ack=%d\n",
	        __func__,
		prefix,
		le32toh(rxphy->fa),
		le32toh(rxphy->bad_fina_sync),
		le32toh(rxphy->sfd_timeout),
		le32toh(rxphy->fina_timeout),
		le32toh(rxphy->no_rts_ack));

	printf("%s: %s: rxe_limit=%d, ack=%d, cts=%d, ba_resp=%d, dsp_kill=%d, bad_mh=%d, rssi_sum=%d\n",
	        __func__,
		prefix,
		le32toh(rxphy->rxe_limit),
		le32toh(rxphy->ack),
		le32toh(rxphy->cts),
		le32toh(rxphy->ba_resp),
		le32toh(rxphy->dsp_kill),
		le32toh(rxphy->bad_mh),
		le32toh(rxphy->rssi_sum));
}

static void
iwn_stats_rx_general_print(struct iwn_rx_general_stats *g)
{

	printf("%s: bad_cts=%d, bad_ack=%d, not_bss=%d, filtered=%d, bad_chan=%d, beacons=%d\n",
	    __func__,
	    le32toh(g->bad_cts),
	    le32toh(g->bad_ack),
	    le32toh(g->not_bss),
	    le32toh(g->filtered),
	    le32toh(g->bad_chan),
	    le32toh(g->beacons));

	/* XXX it'd be nice to have adc/ina saturated as a % of time */
	printf("%s: missed_beacons=%d, adc_saturated=%d, ina_searched=%d\n",
	    __func__,
	    le32toh(g->missed_beacons),
	    le32toh(g->adc_saturated),
	    le32toh(g->ina_searched));

	printf("%s: noise=[%d, %d, %d] flags=0x%08x, load=%d, fa=%d\n",
	    __func__,
	    le32toh(g->noise[0]),
	    le32toh(g->noise[1]),
	    le32toh(g->noise[2]),
	    le32toh(g->flags),
	    le32toh(g->load),
	    le32toh(g->fa));

	printf("%s: rssi=[%d, %d, %d] energy=[%d %d %d]\n",
	    __func__,
	    le32toh(g->rssi[0]),
	    le32toh(g->rssi[1]),
	    le32toh(g->rssi[2]),
	    le32toh(g->energy[0]),
	    le32toh(g->energy[1]),
	    le32toh(g->energy[2]));
}

static void
iwn_stats_tx_print(struct iwn_tx_stats *tx)
{

	printf("%s: preamble=%d, rx_detected=%d, bt_defer=%d, bt_kill=%d, short_len=%d\n",
	    __func__,
	    le32toh(tx->preamble),
	    le32toh(tx->rx_detected),
	    le32toh(tx->bt_defer),
	    le32toh(tx->bt_kill),
	    le32toh(tx->short_len));

	printf("%s: cts_timeout=%d, ack_timeout=%d, exp_ack=%d, ack=%d, msdu=%d\n",
	    __func__,
	    le32toh(tx->cts_timeout),
	    le32toh(tx->ack_timeout),
	    le32toh(tx->exp_ack),
	    le32toh(tx->ack),
	    le32toh(tx->msdu));

	printf("%s: burst_err1=%d, burst_err2=%d, cts_collision=%d, ack_collision=%d\n",
	    __func__,
	    le32toh(tx->burst_err1),
	    le32toh(tx->burst_err2),
	    le32toh(tx->cts_collision),
	    le32toh(tx->ack_collision));

	printf("%s: ba_timeout=%d, ba_resched=%d, query_ampdu=%d, query=%d, query_ampdu_frag=%d\n",
	    __func__,
	    le32toh(tx->ba_timeout),
	    le32toh(tx->ba_resched),
	    le32toh(tx->query_ampdu),
	    le32toh(tx->query),
	    le32toh(tx->query_ampdu_frag));

	printf("%s: query_mismatch=%d, not_ready=%d, underrun=%d, bt_ht_kill=%d, rx_ba_resp=%d\n",
	    __func__,
	    le32toh(tx->query_mismatch),
	    le32toh(tx->not_ready),
	    le32toh(tx->underrun),
	    le32toh(tx->bt_ht_kill),
	    le32toh(tx->rx_ba_resp));
}

static void
iwn_stats_ht_phy_print(struct iwn_rx_ht_phy_stats *ht)
{

	printf("%s: bad_plcp=%d, overrun=%d, eoverrun=%d, good_crc32=%d, bad_crc32=%d\n",
	    __func__,
	    le32toh(ht->bad_plcp),
	    le32toh(ht->overrun),
	    le32toh(ht->eoverrun),
	    le32toh(ht->good_crc32),
	    le32toh(ht->bad_crc32));

	printf("%s: bad_mh=%d, good_ampdu_crc32=%d, ampdu=%d, fragment=%d\n",
	    __func__,
	    le32toh(ht->bad_plcp),
	    le32toh(ht->good_ampdu_crc32),
	    le32toh(ht->ampdu),
	    le32toh(ht->fragment));
}


static void
iwn_stats_general_print(struct iwn_stats *stats)
{

	/* General */
	printf("%s: temp=%d, temp_m=%d, burst_check=%d, burst=%d, sleep=%d, slot_out=%d, slot_idle=%d\n",
	        __func__,
		le32toh(stats->general.temp),
		le32toh(stats->general.temp_m),
		le32toh(stats->general.burst_check),
		le32toh(stats->general.burst),
		le32toh(stats->general.sleep),
		le32toh(stats->general.slot_out),
		le32toh(stats->general.slot_idle));
	printf("%s: slot_out=%d, ttl_tstamp=0x%08x, tx_ant_a=%d, tx_ant_b=%d, exec=%d, probe=%d\n",
	        __func__,
		le32toh(stats->general.slot_out),
		le32toh(stats->general.ttl_tstamp),
		le32toh(stats->general.tx_ant_a),
		le32toh(stats->general.tx_ant_b),
		le32toh(stats->general.exec),
		le32toh(stats->general.probe));
	printf("%s: rx_enabled=%d\n",
	        __func__,
		le32toh(stats->general.rx_enabled));
}

static void
iwn_print(struct iwnstats *is)
{
	struct iwn_stats *s;
	struct timeval tv;

	s = &is->st;

	gettimeofday(&tv, NULL);
	printf("time=%ld.%.6ld\n", (long)tv.tv_sec, (long)tv.tv_usec);

	iwn_stats_general_print(s);

	/* RX */
	iwn_stats_phy_print(&s->rx.ofdm, "ofdm");
	iwn_stats_phy_print(&s->rx.cck, "cck");
	iwn_stats_ht_phy_print(&s->rx.ht);
	iwn_stats_rx_general_print(&s->rx.general);

	/* TX */
	iwn_stats_tx_print(&s->tx);
	printf("--\n");
}

static void
usage(void)
{
	printf("Usage: iwnstats [-h] [-i ifname]\n");
	printf("    -h:			Help\n");
	printf("    -i <ifname>:	Use ifname (default %s)\n",
	    IWN_DEFAULT_IF);
}

int
main(int argc, char *argv[])
{
	struct iwnstats *is;
	int ch;
	char *ifname;
	bool first;

	ifname = strdup(IWN_DEFAULT_IF);

	/* Parse command line arguments */
	while ((ch = getopt(argc, argv,
	    "hi:")) != -1) {
		switch (ch) {
		case 'i':
			if (ifname)
				free(ifname);
			ifname = strdup(optarg);
			break;
		default:
		case '?':
		case 'h':
			usage();
			exit(1);
		}
	}

	is = iwnstats_new(ifname);

	if (is == NULL) {
		fprintf(stderr, "%s: couldn't allocate new stats structure\n",
		    argv[0]);
		exit(127);
	}

	/* begin fetching data */
	first = true;
	while (1) {
		if (iwn_collect(is) != 0) {
			fprintf(stderr, "%s: fetch failed\n", argv[0]);
			if (first)
				return 1;
			goto next;
		}

		iwn_print(is);

	next:
		usleep(100 * 1000);
		first = false;
	}

	exit(0);
}