/** @file wlan_mac_station_info.c
 *  @brief Station Information Metadata Subsystem
 *
 *  This contains code tracking metadata about stations.
 *
 *  @copyright Copyright 2014-2019, Mango Communications. All rights reserved.
 *          Distributed under the Mango Communications Reference Design License
 *              See LICENSE.txt included in the design archive or
 *              at http://mangocomm.com/802.11/license
 *
 *  This file is part of the Mango 802.11 Reference Design (https://mangocomm.com/802.11)
 */

#ifndef WLAN_MAC_STATION_INFO_H_
#define WLAN_MAC_STATION_INFO_H_

/***************************** Include Files *********************************/

#include "wlan_mac_high_sw_config.h"
#include "xil_types.h"
#include "wlan_common_types.h"
#include "wlan_high_types.h"


/*************************** Constant Definitions ****************************/

//-----------------------------------------------
// Timeout used to remove inactive BSS infos
//     - Part of bss_info_timestamp_check()
//
#define STATION_INFO_TIMEOUT_USEC                           600000000

/********************************************************************
 * @brief Tx/Rx Counts Sub-structure
 *
 * This struct contains counts about the communications link. It is intended to
 * be instantiated multiple times in the broader txrx_counts_t struct so that
 * different packet types can be individually tracked.
 *
 ********************************************************************/
typedef struct txrx_counts_sub_t{
    u64        rx_num_bytes;                ///< # of successfully received bytes (de-duplicated)
    u64        rx_num_bytes_total;          ///< # of successfully received bytes (including duplicates)
    u64        tx_num_bytes_success;        ///< # of successfully transmitted bytes (high-level MPDUs)
    u64        tx_num_bytes_total;          ///< Total # of transmitted bytes (high-level MPDUs)
    u32        rx_num_packets;              ///< # of successfully received packets (de-duplicated)
    u32        rx_num_packets_total;        ///< # of successfully received packets (including duplicates)
    u32        tx_num_packets_success;      ///< # of successfully transmitted packets (high-level MPDUs)
    u32        tx_num_packets_total;        ///< Total # of transmitted packets (high-level MPDUs)
    u64        tx_num_attempts;             ///< # of low-level attempts (including retransmissions)
} txrx_counts_sub_t;
CASSERT(sizeof(txrx_counts_sub_t) == 56,txrx_counts_sub_alignment_check);

/********************************************************************
 * @brief Station Counts Structure
 *
 * This struct contains counts about the communications link.  Additionally,
 * counting different parameters can be decoupled from station_info structs
 * entirely to enable promiscuous counts about unassociated devices seen in
 * the network.
 *
 ********************************************************************/
typedef struct station_txrx_counts_t{
	txrx_counts_sub_t   data;                          /* Counts about data types	*/
	 /*----- 8-byte boundary ------*/
	txrx_counts_sub_t   mgmt;                          /* Counts about management types */
	 /*----- 8-byte boundary ------*/
} station_txrx_counts_t;

ASSERT_TYPE_SIZE(station_txrx_counts_t, 112);


/********************************************************************
 * @brief Rate Selection Information
 *
 * This structure contains information about the rate selection scheme.
 *
 ********************************************************************/
typedef struct rate_selection_info_t{
    u16 rate_selection_scheme;
    u8	reserved[6];
} rate_selection_info_t;

#define RATE_SELECTION_SCHEME_STATIC                       0

/********************************************************************
 * @brief Station Information Structure
 *
 * This struct contains information about associated stations (or, in the
 * case of a station, information about the associated access point).
 *
 * NOTE:  The reason that the reference design uses a #define for fields in
 *     two different structs is so that fields that must be in two different
 *     structs stay in sync and so there is not another level of indirection
 *     by using nested structs.
 *
 ********************************************************************/
#define STATION_INFO_HOSTNAME_MAXLEN                       19


typedef struct station_info_t{
    u8                 addr[MAC_ADDR_LEN];                         /* HW Address */
    s16                ID;                                         /* Identification Index for this station */
    char               hostname[STATION_INFO_HOSTNAME_MAXLEN+1];   /* Hostname from DHCP requests */
    u8                 flags;                                      /* 1-bit flags */
	u8                 ps_state;                                   /* Power saving state */
	u16                capabilities;                               /* Capabilities */
    u64                latest_rx_timestamp;               		   /* Timestamp of most recent reception */
	u64                latest_txrx_timestamp;               	   /* Timestamp of most recent reception or transmission */
    u16                latest_rx_seq;                              /* Sequence number of the last MPDU reception */
    s16				   QID;										   /* Queue ID used by this station */
    int				   num_linked_queue_buffer;					   /* Number of queue buffers linked to this station_info_t */
    tx_params_t        tx_params_data;	 		 				   /* Transmission Parameters Structure for Data */
    tx_params_t        tx_params_mgmt;	 		 				   /* Transmission Parameters Structure for Management */
#if WLAN_SW_CONFIG_ENABLE_TXRX_COUNTS
    station_txrx_counts_t		txrx_counts;                        			/* Tx/Rx Counts */
#endif
    rate_selection_info_t		rate_info;
} station_info_t;
#if WLAN_SW_CONFIG_ENABLE_TXRX_COUNTS
ASSERT_TYPE_SIZE(station_info_t, 192);

// For wlan_exp retrieval purposes, we need to make sure that the txrx_counter field is located
// at the end of the struct with rate_info. Here we can construct a compile time assertion that
// will fail if that assumption is broken.
ASSERT_FIELD_END_ALIGNMENT(station_info_t, txrx_counts, sizeof(station_txrx_counts_t) + sizeof(rate_selection_info_t));

#else
ASSERT_TYPE_SIZE(station_info_t, 80);
#endif

// For wlan_exp retrieval purposes, we need to make sure that the rate_info field is located
// at the end of the struct Here we can construct a compile time assertion that will fail if
// that assumption is broken.
ASSERT_FIELD_END_ALIGNMENT(station_info_t, rate_info, sizeof(rate_selection_info_t));


#define STATION_INFO_FLAG_KEEP                             0x01 ///< Prevent MAC High Framework from deleting this station_infO
#define STATION_INFO_FLAG_DISABLE_ASSOC_CHECK              0x02 ///< Mask for flag in station_info -- disable association check

#define STATION_INFO_PS_STATE_DOZE 						   0x01 ///< Mask to sleeping stations (if STA supports PS)

#define STATION_INFO_CAPABILITIES_HT_CAPABLE               0x0001 ///< Station is capable of HT Tx and Rx

#define RX_PROCESS_COUNTS_OPTION_FLAG_IS_DUPLICATE		   0x00000001

#define STATION_INFO_PRINT_OPTION_FLAG_INCLUDE_COUNTS	   0x00000001


//Define a new type of dl_entry for pointing to station_info_t
// structs that contains some extra fields for faster searching
// without needing to jump to DRAM.
typedef struct station_info_entry_t station_info_entry_t;
struct station_info_entry_t{
	station_info_entry_t* next;
	station_info_entry_t* prev;
	station_info_t*     data;
	u8				    addr[6];
	u16					reserved;
};
ASSERT_TYPE_SIZE(station_info_entry_t, 20);

typedef enum default_tx_param_sel_t{
	unicast_mgmt,
	unicast_data,
	mcast_mgmt,
	mcast_data,
} default_tx_param_sel_t;


// Forward declarations -- these must be defined elsewhere
struct dl_entry;
struct dl_list;
struct wlan_mac_low_tx_details_t;

/*************************** Function Prototypes *****************************/

void             station_info_init();
void             station_info_init_finish();

station_info_entry_t* station_info_checkout();
void             station_info_checkin(struct dl_entry* entry);

station_info_t*  station_info_posttx_process(void* pkt_buf_addr);
station_info_t*  station_info_txreport_process(void* pkt_buf_addr, struct wlan_mac_low_tx_details_t* wlan_mac_low_tx_details);
station_info_t*	 station_info_postrx_process(void* pkt_buf_addr);
void 			 station_info_rx_process_hostname(void* pkt_buf_addr, station_info_t* station_info);
#if WLAN_SW_CONFIG_ENABLE_TXRX_COUNTS
void   			 station_info_rx_process_counts(void* pkt_buf_addr, station_info_t* station_info, u32 option_flags);
#endif

void             station_info_print(struct dl_list* list, u32 option_flags);

#if WLAN_SW_CONFIG_ENABLE_TXRX_COUNTS
void             txrx_counts_zero_all();
void 			 station_info_clear_txrx_counts(station_txrx_counts_t* txrx_counts);
#endif

void             station_info_timestamp_check();

station_info_t*  station_info_create(u8* addr);
void             station_info_reset_all();
void 			 station_info_clear(station_info_t* station_info);

struct dl_list*  station_info_get_list();

station_info_entry_t* station_info_find_by_id(u32 id, struct dl_list* list);
station_info_entry_t* station_info_find_by_addr(u8* addr, struct dl_list* list);

station_info_t*  station_info_add_to_list(dl_list* app_station_info_list, u8* addr);
int 			 station_info_remove_from_list(dl_list* app_station_info_list, u8* addr);

u8               station_info_is_member(struct dl_list* app_station_info_list, station_info_t* station_info);

tx_params_t		 wlan_mac_get_default_tx_params(default_tx_param_sel_t default_tx_param_sel);
void		 	 wlan_mac_set_default_tx_params(default_tx_param_sel_t default_tx_param_sel, tx_params_t* tx_params);
void 			 wlan_mac_reapply_default_tx_params();

#endif