/*
 * Copyright 2006, Johannes Berg <johannes@sipsolutions.net>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

/* just for IFNAMSIZ */
#include <linux/if.h>
#include <linux/slab.h>
#include <linux/export.h>
#include "led.h"

void ieee80211_led_rx(struct ieee80211_local *local)
{
        if (unlikely(!local->rx_led))
                return;
        if (local->rx_led_counter++ % 2 == 0)
                led_trigger_event(local->rx_led, LED_OFF);
        else
                led_trigger_event(local->rx_led, LED_FULL);
}

/* q is 1 if a packet was enqueued, 0 if it has been transmitted */
void ieee80211_led_tx(struct ieee80211_local *local, int q)
{
        if (unlikely(!local->tx_led))
                return;
        /* not sure how this is supposed to work ... */
        local->tx_led_counter += 2*q-1;
        if (local->tx_led_counter % 2 == 0)
                led_trigger_event(local->tx_led, LED_OFF);
        else
                led_trigger_event(local->tx_led, LED_FULL);
}

void ieee80211_led_assoc(struct ieee80211_local *local, bool associated)
{
        if (unlikely(!local->assoc_led))
                return;
        if (associated)
                led_trigger_event(local->assoc_led, LED_FULL);
        else
                led_trigger_event(local->assoc_led, LED_OFF);
}

void ieee80211_led_radio(struct ieee80211_local *local, bool enabled)
{
        if (unlikely(!local->radio_led))
                return;
        if (enabled)
                led_trigger_event(local->radio_led, LED_FULL);
        else
                led_trigger_event(local->radio_led, LED_OFF);
}

void ieee80211_led_names(struct ieee80211_local *local)
{
        snprintf(local->rx_led_name, sizeof(local->rx_led_name),
                 "%srx", wiphy_name(local->hw.wiphy));
        snprintf(local->tx_led_name, sizeof(local->tx_led_name),
                 "%stx", wiphy_name(local->hw.wiphy));
        snprintf(local->assoc_led_name, sizeof(local->assoc_led_name),
                 "%sassoc", wiphy_name(local->hw.wiphy));
        snprintf(local->radio_led_name, sizeof(local->radio_led_name),
                 "%sradio", wiphy_name(local->hw.wiphy));
}

void ieee80211_led_init(struct ieee80211_local *local)
{
        local->rx_led = kzalloc(sizeof(struct led_trigger), GFP_KERNEL);
        if (local->rx_led) {
                local->rx_led->name = local->rx_led_name;
                if (led_trigger_register(local->rx_led)) {
                        kfree(local->rx_led);
                        local->rx_led = NULL;
                }
        }

        local->tx_led = kzalloc(sizeof(struct led_trigger), GFP_KERNEL);
        if (local->tx_led) {
                local->tx_led->name = local->tx_led_name;
                if (led_trigger_register(local->tx_led)) {
                        kfree(local->tx_led);
                        local->tx_led = NULL;
                }
        }

        local->assoc_led = kzalloc(sizeof(struct led_trigger), GFP_KERNEL);
        if (local->assoc_led) {
                local->assoc_led->name = local->assoc_led_name;
                if (led_trigger_register(local->assoc_led)) {
                        kfree(local->assoc_led);
                        local->assoc_led = NULL;
                }
        }

        local->radio_led = kzalloc(sizeof(struct led_trigger), GFP_KERNEL);
        if (local->radio_led) {
                local->radio_led->name = local->radio_led_name;
                if (led_trigger_register(local->radio_led)) {
                        kfree(local->radio_led);
                        local->radio_led = NULL;
                }
        }

        if (local->tpt_led_trigger) {
                if (led_trigger_register(&local->tpt_led_trigger->trig)) {
                        kfree(local->tpt_led_trigger);
                        local->tpt_led_trigger = NULL;
                }
        }
}

void ieee80211_led_exit(struct ieee80211_local *local)
{
        if (local->radio_led) {
                led_trigger_unregister(local->radio_led);
                kfree(local->radio_led);
        }
        if (local->assoc_led) {
                led_trigger_unregister(local->assoc_led);
                kfree(local->assoc_led);
        }
        if (local->tx_led) {
                led_trigger_unregister(local->tx_led);
                kfree(local->tx_led);
        }
        if (local->rx_led) {
                led_trigger_unregister(local->rx_led);
                kfree(local->rx_led);
        }

        if (local->tpt_led_trigger) {
                led_trigger_unregister(&local->tpt_led_trigger->trig);
                kfree(local->tpt_led_trigger);
        }
}

char *__ieee80211_get_radio_led_name(struct ieee80211_hw *hw)
{
        struct ieee80211_local *local = hw_to_local(hw);

        return local->radio_led_name;
}
EXPORT_SYMBOL(__ieee80211_get_radio_led_name);

char *__ieee80211_get_assoc_led_name(struct ieee80211_hw *hw)
{
        struct ieee80211_local *local = hw_to_local(hw);

        return local->assoc_led_name;
}
EXPORT_SYMBOL(__ieee80211_get_assoc_led_name);

char *__ieee80211_get_tx_led_name(struct ieee80211_hw *hw)
{
        struct ieee80211_local *local = hw_to_local(hw);

        return local->tx_led_name;
}
EXPORT_SYMBOL(__ieee80211_get_tx_led_name);

char *__ieee80211_get_rx_led_name(struct ieee80211_hw *hw)
{
        struct ieee80211_local *local = hw_to_local(hw);

        return local->rx_led_name;
}
EXPORT_SYMBOL(__ieee80211_get_rx_led_name);

static unsigned long tpt_trig_traffic(struct ieee80211_local *local,
                                      struct tpt_led_trigger *tpt_trig)
{
        unsigned long traffic, delta;

        traffic = tpt_trig->tx_bytes + tpt_trig->rx_bytes;

        delta = traffic - tpt_trig->prev_traffic;
        tpt_trig->prev_traffic = traffic;
        return DIV_ROUND_UP(delta, 1024 / 8);
}

static void tpt_trig_timer(unsigned long data)
{
        struct ieee80211_local *local = (void *)data;
        struct tpt_led_trigger *tpt_trig = local->tpt_led_trigger;
        struct led_classdev *led_cdev;
        unsigned long on, off, tpt;
        int i;

        if (!tpt_trig->running)
                return;

        mod_timer(&tpt_trig->timer, round_jiffies(jiffies + HZ));

        tpt = tpt_trig_traffic(local, tpt_trig);

        /* default to just solid on */
        on = 1;
        off = 0;

        for (i = tpt_trig->blink_table_len - 1; i >= 0; i--) {
                if (tpt_trig->blink_table[i].throughput < 0 ||
                    tpt > tpt_trig->blink_table[i].throughput) {
                        off = tpt_trig->blink_table[i].blink_time / 2;
                        on = tpt_trig->blink_table[i].blink_time - off;
                        break;
                }
        }

        read_lock(&tpt_trig->trig.leddev_list_lock);
        list_for_each_entry(led_cdev, &tpt_trig->trig.led_cdevs, trig_list)
                led_blink_set(led_cdev, &on, &off);
        read_unlock(&tpt_trig->trig.leddev_list_lock);
}

char *__ieee80211_create_tpt_led_trigger(struct ieee80211_hw *hw,
                                unsigned int flags,
                                const struct ieee80211_tpt_blink *blink_table,
                                unsigned int blink_table_len)
{
        struct ieee80211_local *local = hw_to_local(hw);
        struct tpt_led_trigger *tpt_trig;

        if (WARN_ON(local->tpt_led_trigger))
                return NULL;

        tpt_trig = kzalloc(sizeof(struct tpt_led_trigger), GFP_KERNEL);
        if (!tpt_trig)
                return NULL;

        snprintf(tpt_trig->name, sizeof(tpt_trig->name),
                 "%stpt", wiphy_name(local->hw.wiphy));

        tpt_trig->trig.name = tpt_trig->name;

        tpt_trig->blink_table = blink_table;
        tpt_trig->blink_table_len = blink_table_len;
        tpt_trig->want = flags;

        setup_timer(&tpt_trig->timer, tpt_trig_timer, (unsigned long)local);

        local->tpt_led_trigger = tpt_trig;

        return tpt_trig->name;
}
EXPORT_SYMBOL(__ieee80211_create_tpt_led_trigger);

static void ieee80211_start_tpt_led_trig(struct ieee80211_local *local)
{
        struct tpt_led_trigger *tpt_trig = local->tpt_led_trigger;

        if (tpt_trig->running)
                return;

        /* reset traffic */
        tpt_trig_traffic(local, tpt_trig);
        tpt_trig->running = true;

        tpt_trig_timer((unsigned long)local);
        mod_timer(&tpt_trig->timer, round_jiffies(jiffies + HZ));
}

static void ieee80211_stop_tpt_led_trig(struct ieee80211_local *local)
{
        struct tpt_led_trigger *tpt_trig = local->tpt_led_trigger;
        struct led_classdev *led_cdev;

        if (!tpt_trig->running)
                return;

        tpt_trig->running = false;
        del_timer_sync(&tpt_trig->timer);

        read_lock(&tpt_trig->trig.leddev_list_lock);
        list_for_each_entry(led_cdev, &tpt_trig->trig.led_cdevs, trig_list)
                led_set_brightness(led_cdev, LED_OFF);
        read_unlock(&tpt_trig->trig.leddev_list_lock);
}

void ieee80211_mod_tpt_led_trig(struct ieee80211_local *local,
                                unsigned int types_on, unsigned int types_off)
{
        struct tpt_led_trigger *tpt_trig = local->tpt_led_trigger;
        bool allowed;

        WARN_ON(types_on & types_off);

        if (!tpt_trig)
                return;

        tpt_trig->active &= ~types_off;
        tpt_trig->active |= types_on;

        /*
         * Regardless of wanted state, we shouldn't blink when
         * the radio is disabled -- this can happen due to some
         * code ordering issues with __ieee80211_recalc_idle()
         * being called before the radio is started.
         */
        allowed = tpt_trig->active & IEEE80211_TPT_LEDTRIG_FL_RADIO;

        if (!allowed || !(tpt_trig->active & tpt_trig->want))
                ieee80211_stop_tpt_led_trig(local);
        else
                ieee80211_start_tpt_led_trig(local);
}