// SPDX-License-Identifier: GPL-2.0
/* Copyright (C) B.A.T.M.A.N. contributors:
 *
 * Marek Lindner
 */

#include "gateway_common.h"
#include "main.h"

#include <linux/atomic.h>
#include <linux/byteorder/generic.h>
#include <linux/stddef.h>
#include <linux/types.h>
#include <uapi/linux/batadv_packet.h>
#include <uapi/linux/batman_adv.h>

#include "gateway_client.h"
#include "tvlv.h"

/**
 * batadv_gw_tvlv_container_update() - update the gw tvlv container after
 *  gateway setting change
 * @bat_priv: the bat priv with all the mesh interface information
 */
void batadv_gw_tvlv_container_update(struct batadv_priv *bat_priv)
{
        struct batadv_tvlv_gateway_data gw;
        u32 down, up;
        char gw_mode;

        gw_mode = atomic_read(&bat_priv->gw.mode);

        switch (gw_mode) {
        case BATADV_GW_MODE_OFF:
        case BATADV_GW_MODE_CLIENT:
                batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1);
                break;
        case BATADV_GW_MODE_SERVER:
                down = atomic_read(&bat_priv->gw.bandwidth_down);
                up = atomic_read(&bat_priv->gw.bandwidth_up);
                gw.bandwidth_down = htonl(down);
                gw.bandwidth_up = htonl(up);
                batadv_tvlv_container_register(bat_priv, BATADV_TVLV_GW, 1,
                                               &gw, sizeof(gw));
                break;
        }
}

/**
 * batadv_gw_tvlv_ogm_handler_v1() - process incoming gateway tvlv container
 * @bat_priv: the bat priv with all the mesh interface information
 * @orig: the orig_node of the ogm
 * @flags: flags indicating the tvlv state (see batadv_tvlv_handler_flags)
 * @tvlv_value: tvlv buffer containing the gateway data
 * @tvlv_value_len: tvlv buffer length
 */
static void batadv_gw_tvlv_ogm_handler_v1(struct batadv_priv *bat_priv,
                                          struct batadv_orig_node *orig,
                                          u8 flags,
                                          void *tvlv_value, u16 tvlv_value_len)
{
        struct batadv_tvlv_gateway_data gateway, *gateway_ptr;

        /* only fetch the tvlv value if the handler wasn't called via the
         * CIFNOTFND flag and if there is data to fetch
         */
        if (flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND ||
            tvlv_value_len < sizeof(gateway)) {
                gateway.bandwidth_down = 0;
                gateway.bandwidth_up = 0;
        } else {
                gateway_ptr = tvlv_value;
                gateway.bandwidth_down = gateway_ptr->bandwidth_down;
                gateway.bandwidth_up = gateway_ptr->bandwidth_up;
                if (gateway.bandwidth_down == 0 ||
                    gateway.bandwidth_up == 0) {
                        gateway.bandwidth_down = 0;
                        gateway.bandwidth_up = 0;
                }
        }

        batadv_gw_node_update(bat_priv, orig, &gateway);

        /* restart gateway selection */
        if (gateway.bandwidth_down != 0 &&
            atomic_read(&bat_priv->gw.mode) == BATADV_GW_MODE_CLIENT)
                batadv_gw_check_election(bat_priv, orig);
}

/**
 * batadv_gw_init() - initialise the gateway handling internals
 * @bat_priv: the bat priv with all the mesh interface information
 */
void batadv_gw_init(struct batadv_priv *bat_priv)
{
        if (bat_priv->algo_ops->gw.init_sel_class)
                bat_priv->algo_ops->gw.init_sel_class(bat_priv);
        else
                atomic_set(&bat_priv->gw.sel_class, 1);

        batadv_tvlv_handler_register(bat_priv, batadv_gw_tvlv_ogm_handler_v1,
                                     NULL, NULL, BATADV_TVLV_GW, 1,
                                     BATADV_TVLV_HANDLER_OGM_CIFNOTFND);
}

/**
 * batadv_gw_free() - free the gateway handling internals
 * @bat_priv: the bat priv with all the mesh interface information
 */
void batadv_gw_free(struct batadv_priv *bat_priv)
{
        batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1);
        batadv_tvlv_handler_unregister(bat_priv, BATADV_TVLV_GW, 1);
}