/*

* Copyright (c) 2002-2006 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:   Implementation of the ScanInfoIe class.

*

*/

#include "hssscaninfoie.h"

// Defines the id byte of the RSN Information Element.

const TUint8 SCANINFOIE_RSNIE_ID = 48;

// Defines the static offsets for different fields in RSN IE.

const TUint32 SCANINFOIE_RSNIE_GROUP_SUITE_OFFSET = 2;

const TUint32 SCANINFOIE_RSNIE_PAIRWISE_SUITE_COUNT_OFFSET = 6;

const TUint32 SCANINFOIE_RSNIE_PAIRWISE_SUITE_OFFSET = 8;

// Defines the OUIs used in RSN IEs.

const TUint32 SCANINFOIE_OUI_LENGTH = 4;

const TUint8 SCANINFOIE_RSNIE_OUI_CCMP[] = { 0x00, 0x0F, 0xAC, 0x04 };

const TUint8 SCANINFOIE_RSNIE_OUI_EAP[] = { 0x00, 0x0F, 0xAC, 0x01 };

const TUint8 SCANINFOIE_RSNIE_OUI_PSK[] = { 0x00, 0x0F, 0xAC, 0x02 };

// Defines the static offsets for different fields in RSN IE.

const TUint32 SCANINFOIE_WPAIE_PAIRWISE_SUITE_COUNT_OFFSET = 10;

const TUint32 SCANINFOIE_WPAIE_PAIRWISE_SUITE_OFFSET = 12;

// Defines the OUIs used in WPA IEs.

const TUint8 SCANINFOIE_WPAIE_OUI_EAP[] = { 0x00, 0x50, 0xF2, 0x01 };

const TUint8 SCANINFOIE_WPAIE_OUI_PSK[] = { 0x00, 0x50, 0xF2, 0x02 };

// ============================ MEMBER FUNCTIONS ===============================

// -----------------------------------------------------------------------------

// ScanInfoIe::ScanInfoIe

// C++ default constructor can NOT contain any code, that

// might leave.

// -----------------------------------------------------------------------------

//

HssScanInfoIe::HssScanInfoIe()

    {

    }

// Destructor

HssScanInfoIe::~HssScanInfoIe()

    {    

    }

// -----------------------------------------------------------------------------

// ScanInfoIe::SecurityMode

// -----------------------------------------------------------------------------

//

HssSecurityMode HssScanInfoIe::SecurityMode(

    HssScanInfo& info )

    {

    TUint8 wpaie_length( 0 );

    const TUint8* wpaie_data = NULL;

    TUint8 rsnie_length( 0 );

    const TUint8* rsnie_data = NULL;

    info.InformationElement( SCANINFOIE_RSNIE_ID, rsnie_length, &rsnie_data );

    info.WpaIE( wpaie_length, &wpaie_data );

    if ( !info.Privacy() )

        {

        if ( !wpaie_length && !rsnie_length )

            {

            return HssSecurityModeOpen;

            }

        return HssSecurityMode802_1x;

        }

    if ( !wpaie_length && !rsnie_length )

        {

        return HssSecurityModeWep;

        }

    if ( rsnie_length )

        {        

        if ( IsKeyManagement(

            HssScanInfoIeTypeRsn,

            HssScanInfoIeKeyManagementPsk,

            rsnie_length,

            rsnie_data ) )

            {

            if ( IsWpa2Ciphers(

                HssScanInfoIeTypeRsn,

                rsnie_length,

                rsnie_data ) )

                {

                return HssSecurityModeWpa2Psk;

                }

            return HssSecurityModeWpaPsk;            

            }

        else if ( IsKeyManagement(

            HssScanInfoIeTypeRsn,

            HssScanInfoIeKeyManagementEap,

            rsnie_length,

            rsnie_data ) )

            {

            if ( IsWpa2Ciphers(

                HssScanInfoIeTypeRsn,

                rsnie_length,

                rsnie_data ) )

                {                

                return HssSecurityModeWpa2Eap;

                }

            return HssSecurityModeWpaEap;            

            }                        

        }

    if ( wpaie_length )

        {

        if ( IsKeyManagement(

            HssScanInfoIeTypeWpa,

            HssScanInfoIeKeyManagementPsk,

            wpaie_length,

            wpaie_data ) )

            {

            return HssSecurityModeWpaPsk;            

            }

        else if ( IsKeyManagement(

            HssScanInfoIeTypeWpa,

            HssScanInfoIeKeyManagementEap,

            wpaie_length,

            wpaie_data ) )

            {

            return HssSecurityModeWpaEap;            

            }        

        }

    return HssSecurityMode802_1x;

    }

// -----------------------------------------------------------------------------

// ScanInfoIe::Compare

// -----------------------------------------------------------------------------

//   

TInt HssScanInfoIe::Compare(

    const unsigned char* pl,

    int ll, 

    const unsigned char* pr, 

    int rl )

    {

    if ( ll != rl )

        return ll - rl;

    if ( pl == pr )

        return 0;

    for ( int i( 0 ); i < ll; ++i )

        if ( *(pl+i) != *(pr+i) )

            return *(pl+i) - *(pr+i);

    return 0;    

    }

// -----------------------------------------------------------------------------

// ScanInfoIe::IsKeyManagement

// -----------------------------------------------------------------------------

//

TBool HssScanInfoIe::IsKeyManagement(

    HssScanInfoIeType ie_type,

    HssScanInfoIeKeyManagement key_type,

    TUint8 /* ie_length */,

    const TUint8* ie_data )

    {

    const TUint8* key_data = NULL;

    const TUint8* key_comp = NULL;

    if ( ie_type == HssScanInfoIeTypeRsn )

        {

        if ( key_type == HssScanInfoIeKeyManagementEap )

            {

            key_comp = &SCANINFOIE_RSNIE_OUI_EAP[0];

            }

        else

            {

            key_comp = &SCANINFOIE_RSNIE_OUI_PSK[0];

            }

        TUint32 suites( *( ie_data + SCANINFOIE_RSNIE_PAIRWISE_SUITE_COUNT_OFFSET ) );

        key_data = ie_data + SCANINFOIE_RSNIE_PAIRWISE_SUITE_OFFSET +

            ( suites * SCANINFOIE_OUI_LENGTH );

        }

    else

        {

        if ( key_type == HssScanInfoIeKeyManagementEap )

            {

            key_comp = &SCANINFOIE_WPAIE_OUI_EAP[0];

            }

        else

            {

            key_comp = &SCANINFOIE_WPAIE_OUI_PSK[0];

            }

        TUint32 suites( *( ie_data + SCANINFOIE_WPAIE_PAIRWISE_SUITE_COUNT_OFFSET ) );

        key_data = ie_data + SCANINFOIE_WPAIE_PAIRWISE_SUITE_OFFSET +

            ( suites * SCANINFOIE_OUI_LENGTH );        

        }

    TUint16 key_suites( *key_data );

    key_data += sizeof( key_suites );

    while ( key_suites-- )

        {

        if ( !Compare(

            key_data + ( key_suites * SCANINFOIE_OUI_LENGTH ), SCANINFOIE_OUI_LENGTH,

            key_comp, SCANINFOIE_OUI_LENGTH ) )

            {

            return true;

            }

        }

    return false;

    };

// -----------------------------------------------------------------------------

// ScanInfoIe::IsWpa2Ciphers

// -----------------------------------------------------------------------------

//

TBool HssScanInfoIe::IsWpa2Ciphers(

    HssScanInfoIeType ie_type,

    TUint8 /* ie_length */,

    const TUint8* ie_data )

    {

    /**

    * WPA IE is not used in WPA2

    */    

    if ( ie_type == HssScanInfoIeTypeWpa )

        {

        return false;

        }

    const TUint8* cipher =

        ie_data + SCANINFOIE_RSNIE_GROUP_SUITE_OFFSET;

    if ( Compare(

        cipher, SCANINFOIE_OUI_LENGTH,

        &SCANINFOIE_RSNIE_OUI_CCMP[0], SCANINFOIE_OUI_LENGTH ) )

        {

        return false;

        }

    TUint32 suites( *( ie_data + SCANINFOIE_RSNIE_PAIRWISE_SUITE_COUNT_OFFSET ) );

    cipher = ie_data + SCANINFOIE_RSNIE_PAIRWISE_SUITE_OFFSET;

    while ( suites-- )

        {

        if ( !Compare(

            cipher + ( suites * SCANINFOIE_OUI_LENGTH ), SCANINFOIE_OUI_LENGTH,

            &SCANINFOIE_RSNIE_OUI_CCMP[0], SCANINFOIE_OUI_LENGTH ) )

            {

            return true;

            }

        }

    return false;

    }