/*
 * Copyright (c) 1996, 2003 VIA Networking Technologies, Inc.
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 *
 * File: bssdb.h
 *
 * Purpose: Handles the Basic Service Set & Node Database functions
 *
 * Author: Lyndon Chen
 *
 * Date: July 16, 2002
 *
 */

#ifndef __BSSDB_H__
#define __BSSDB_H__

#include <linux/skbuff.h>
#include "80211hdr.h"
#include "80211mgr.h"
#include "card.h"

/*---------------------  Export Definitions -------------------------*/

#define MAX_NODE_NUM             64
#define MAX_BSS_NUM              42
#define LOST_BEACON_COUNT      	 10   // 10 sec, XP defined
#define MAX_PS_TX_BUF            32   // sta max power saving tx buf
#define ADHOC_LOST_BEACON_COUNT  30   // 30 sec, beacon lost for adhoc only
#define MAX_INACTIVE_COUNT       300  // 300 sec, inactive STA node refresh

#define USE_PROTECT_PERIOD       10   // 10 sec, Use protect mode check period
#define ERP_RECOVER_COUNT        30   // 30 sec, ERP support callback check
#define BSS_CLEAR_COUNT           1

#define RSSI_STAT_COUNT          10
#define MAX_CHECK_RSSI_COUNT     8

// STA dwflags
#define WLAN_STA_AUTH            BIT0
#define WLAN_STA_ASSOC           BIT1
#define WLAN_STA_PS              BIT2
#define WLAN_STA_TIM             BIT3
// permanent; do not remove entry on expiration
#define WLAN_STA_PERM            BIT4
// If 802.1X is used, this flag is
// controlling whether STA is authorized to
// send and receive non-IEEE 802.1X frames
#define WLAN_STA_AUTHORIZED      BIT5

#define MAX_RATE            12

#define MAX_WPA_IE_LEN      64


/*---------------------  Export Classes  ----------------------------*/

/*---------------------  Export Variables  --------------------------*/


/*---------------------  Export Types  ------------------------------*/

//
// IEEE 802.11 Structures and definitions
//

typedef enum _NDIS_802_11_NETWORK_TYPE
{
    Ndis802_11FH,
    Ndis802_11DS,
    Ndis802_11OFDM5,
    Ndis802_11OFDM24,
    Ndis802_11NetworkTypeMax    // not a real type, defined as an upper bound
} NDIS_802_11_NETWORK_TYPE, *PNDIS_802_11_NETWORK_TYPE;


typedef struct tagSERPObject {
    bool bERPExist;
    unsigned char byERP;
}ERPObject, *PERPObject;


typedef struct tagSRSNCapObject {
    bool bRSNCapExist;
    unsigned short wRSNCap;
}SRSNCapObject, *PSRSNCapObject;

// BSS info(AP)
#pragma pack(1)
typedef struct tagKnownBSS {
    // BSS info
    bool bActive;
    unsigned char abyBSSID[WLAN_BSSID_LEN];
    unsigned int	uChannel;
    unsigned char abySuppRates[WLAN_IEHDR_LEN + WLAN_RATES_MAXLEN + 1];
    unsigned char abyExtSuppRates[WLAN_IEHDR_LEN + WLAN_RATES_MAXLEN + 1];
    unsigned int	uRSSI;
    unsigned char bySQ;
    unsigned short wBeaconInterval;
    unsigned short wCapInfo;
    unsigned char abySSID[WLAN_IEHDR_LEN + WLAN_SSID_MAXLEN + 1];
    unsigned char byRxRate;

//    unsigned short wATIMWindow;
    unsigned char byRSSIStatCnt;
    long            ldBmMAX;
    long            ldBmAverage[RSSI_STAT_COUNT];
    long            ldBmAverRange;
    //For any BSSID selection improvment
    bool bSelected;

    //++ WPA informations
    bool bWPAValid;
    unsigned char byGKType;
    unsigned char abyPKType[4];
    unsigned short wPKCount;
    unsigned char abyAuthType[4];
    unsigned short wAuthCount;
    unsigned char byDefaultK_as_PK;
    unsigned char byReplayIdx;
    //--

    //++ WPA2 informations
    bool bWPA2Valid;
    unsigned char byCSSGK;
    unsigned short wCSSPKCount;
    unsigned char abyCSSPK[4];
    unsigned short wAKMSSAuthCount;
    unsigned char abyAKMSSAuthType[4];

    //++  wpactl
    unsigned char byWPAIE[MAX_WPA_IE_LEN];
    unsigned char byRSNIE[MAX_WPA_IE_LEN];
    unsigned short wWPALen;
    unsigned short wRSNLen;

    // Clear count
    unsigned int	uClearCount;
//    unsigned char abyIEs[WLAN_BEACON_FR_MAXLEN];
    unsigned int	uIELength;
    QWORD           qwBSSTimestamp;
    QWORD           qwLocalTSF;     // local TSF timer

//    NDIS_802_11_NETWORK_TYPE    NetworkTypeInUse;
    CARD_PHY_TYPE   eNetworkTypeInUse;

    ERPObject       sERP;
    SRSNCapObject   sRSNCapObj;
    unsigned char abyIEs[1024];   // don't move this field !!

}__attribute__ ((__packed__))
KnownBSS , *PKnownBSS;

//2006-1116-01,<Add> by NomadZhao
#pragma pack()

typedef enum tagNODE_STATE {
    NODE_FREE,
    NODE_AGED,
    NODE_KNOWN,
    NODE_AUTH,
    NODE_ASSOC
} NODE_STATE, *PNODE_STATE;


// STA node info
typedef struct tagKnownNodeDB {
    // STA info
    bool bActive;
    unsigned char abyMACAddr[WLAN_ADDR_LEN];
    unsigned char abyCurrSuppRates[WLAN_IEHDR_LEN + WLAN_RATES_MAXLEN];
    unsigned char abyCurrExtSuppRates[WLAN_IEHDR_LEN + WLAN_RATES_MAXLEN];
    unsigned short wTxDataRate;
    bool bShortPreamble;
    bool bERPExist;
    bool bShortSlotTime;
    unsigned int	uInActiveCount;
    unsigned short wMaxBasicRate;     //Get from byTopOFDMBasicRate or byTopCCKBasicRate which depends on packetTyp.
    unsigned short wMaxSuppRate;      //Records the highest supported rate getting from SuppRates IE and ExtSuppRates IE in Beacon.
    unsigned short wSuppRate;
    unsigned char byTopOFDMBasicRate;//Records the highest basic rate in OFDM mode
    unsigned char byTopCCKBasicRate; //Records the highest basic rate in CCK mode

    // For AP mode
    struct sk_buff_head sTxPSQueue;
    unsigned short wCapInfo;
    unsigned short wListenInterval;
    unsigned short wAID;
    NODE_STATE      eNodeState;
    bool bPSEnable;
    bool bRxPSPoll;
    unsigned char byAuthSequence;
    unsigned long ulLastRxJiffer;
    unsigned char bySuppRate;
    unsigned long dwFlags;
    unsigned short wEnQueueCnt;

    bool bOnFly;
    unsigned long long       KeyRSC;
    unsigned char byKeyIndex;
    unsigned long dwKeyIndex;
    unsigned char byCipherSuite;
    unsigned long dwTSC47_16;
    unsigned short wTSC15_0;
    unsigned int	uWepKeyLength;
    unsigned char abyWepKey[WLAN_WEPMAX_KEYLEN];
    //
    // Auto rate fallback vars
    bool bIsInFallback;
    unsigned int	uAverageRSSI;
    unsigned int	uRateRecoveryTimeout;
    unsigned int	uRatePollTimeout;
    unsigned int	uTxFailures;
    unsigned int	uTxAttempts;

    unsigned int	uTxRetry;
    unsigned int	uFailureRatio;
    unsigned int	uRetryRatio;
    unsigned int	uTxOk[MAX_RATE+1];
    unsigned int	uTxFail[MAX_RATE+1];
    unsigned int	uTimeCount;

} KnownNodeDB, *PKnownNodeDB;


/*---------------------  Export Functions  --------------------------*/


PKnownBSS
BSSpSearchBSSList(
    void *hDeviceContext,
    unsigned char *pbyDesireBSSID,
    unsigned char *pbyDesireSSID,
    CARD_PHY_TYPE ePhyType
    );

PKnownBSS
BSSpAddrIsInBSSList(
    void *hDeviceContext,
    unsigned char *abyBSSID,
    PWLAN_IE_SSID pSSID
    );

void
BSSvClearBSSList(
    void *hDeviceContext,
    bool bKeepCurrBSSID
    );

bool
BSSbInsertToBSSList(
    void *hDeviceContext,
    unsigned char *abyBSSIDAddr,
    QWORD qwTimestamp,
    unsigned short wBeaconInterval,
    unsigned short wCapInfo,
    unsigned char byCurrChannel,
    PWLAN_IE_SSID pSSID,
    PWLAN_IE_SUPP_RATES pSuppRates,
    PWLAN_IE_SUPP_RATES pExtSuppRates,
    PERPObject psERP,
    PWLAN_IE_RSN pRSN,
    PWLAN_IE_RSN_EXT pRSNWPA,
    PWLAN_IE_COUNTRY pIE_Country,
    PWLAN_IE_QUIET pIE_Quiet,
    unsigned int uIELength,
    unsigned char *pbyIEs,
    void *pRxPacketContext
    );


bool
BSSbUpdateToBSSList(
    void *hDeviceContext,
    QWORD qwTimestamp,
    unsigned short wBeaconInterval,
    unsigned short wCapInfo,
    unsigned char byCurrChannel,
    bool bChannelHit,
    PWLAN_IE_SSID pSSID,
    PWLAN_IE_SUPP_RATES pSuppRates,
    PWLAN_IE_SUPP_RATES pExtSuppRates,
    PERPObject psERP,
    PWLAN_IE_RSN pRSN,
    PWLAN_IE_RSN_EXT pRSNWPA,
    PWLAN_IE_COUNTRY pIE_Country,
    PWLAN_IE_QUIET pIE_Quiet,
    PKnownBSS pBSSList,
    unsigned int uIELength,
    unsigned char *pbyIEs,
    void *pRxPacketContext
    );


bool
BSSDBbIsSTAInNodeDB(void *hDeviceContext, unsigned char *abyDstAddr,
		unsigned int *puNodeIndex);

void
BSSvCreateOneNode(void *hDeviceContext, unsigned int *puNodeIndex);

void
BSSvUpdateAPNode(
    void *hDeviceContext,
    unsigned short *pwCapInfo,
    PWLAN_IE_SUPP_RATES pItemRates,
    PWLAN_IE_SUPP_RATES pExtSuppRates
    );


void
BSSvSecondCallBack(
    void *hDeviceContext
    );


void
BSSvUpdateNodeTxCounter(
    void *hDeviceContext,
    unsigned char byTsr0,
    unsigned char byTsr1,
    unsigned char *pbyBuffer,
    unsigned int uFIFOHeaderSize
    );

void
BSSvRemoveOneNode(
    void *hDeviceContext,
    unsigned int uNodeIndex
    );

void
BSSvAddMulticastNode(
    void *hDeviceContext
    );


void
BSSvClearNodeDBTable(
    void *hDeviceContext,
    unsigned int uStartIndex
    );

void
BSSvClearAnyBSSJoinRecord(
    void *hDeviceContext
    );

#endif //__BSSDB_H__