/*

* Copyright (c) 2002-2004 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:  Handles the custom WLANServiceTable  in CommsDat

*

*/

/*

* %version: 12 %

*/

//  INCLUDE FILES

#include <e32svr.h>

#include <commdb.h>

#include <commsdattypeinfov1_1.h>

#include "wdbifwlansettings.h"

#include "wdbifwlandefs.h"

#include "gendebug.h"

// LOCAL CONSTANTS

_LIT( KGenericTable, "WLANServiceTable");

const int KMaxEapDataBufLength = 128;

const int KOldStyleMaxEapTypeValue = 100;

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

// Class CWlanSettings

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

// MEMBER FUNCTIONS

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

// CWLanSettings::CWLanSettings()

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

// might leave.

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

//

EXPORT_C CWLanSettings::CWLanSettings() : iCurrentRecord( KErrNotFound),

                                           iWLANRecord( 0),

                                           iDBSession( 0)

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::CWlanSettings()")));    

    }

// Destructor

//

EXPORT_C CWLanSettings::~CWLanSettings() 

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::~CWlanSettings()")));    

    delete iDBSession;

    delete iWLANRecord;

    }

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

// TInt CWLanSettings::Connect()

// Connects to the CommsDat storage server

// Wraps around leavable function DoConnectL

// returns KErrNone on success

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

//

EXPORT_C TInt CWLanSettings::Connect()

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::Connect")));

    TRAPD(err,DoConnectL());

    return err;

    }

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

// void CWLanSettings::Disconnect()

// Disconnects from the CommsDat storage server

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

//

EXPORT_C void CWLanSettings::Disconnect()

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::Disconnect")));

    delete iDBSession;

    iDBSession = NULL;

    delete iWLANRecord;

    iWLANRecord = NULL;

    iCurrentRecord = KErrNotFound;

    }

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

// void CWLanSettings::GetDataL( SWLANSettings& aSettings)

// Reads data from a WLANServiceTable record which has previously

// been fetched (= current record) from the CommsDat.

// returns settings in aSettings

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

//

void CWLanSettings::GetDataL( SWLANSettings& aSettings)

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::GetDataL start")));

    ClearWepKeys( aSettings);

    aSettings.Id = iWLANRecord->RecordId();

    aSettings.Name = *((CMDBField<TDesC>*)iWLANRecord->GetFieldByIdL(KCDTIdRecordName));

    aSettings.ServiceID = *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanServiceId));

    aSettings.ConnectionMode = *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanConnMode));

    aSettings.SSID = *((CMDBField<TDesC>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanSSID));

    // new

    aSettings.UsedSSID = *((CMDBField<TDesC>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanUsedSSID));

    aSettings.ScanSSID = *((CMDBField<TBool>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanScanSSID));

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::GetDataL ScanSSID = %d"),aSettings.ScanSSID));

    aSettings.ChannelID = *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanChannelID));

    // end new

    aSettings.AuthenticationMode = EWepAuthModeOpen; // defaults to open (in case of wep, the value will be read from db later)

    aSettings.EnableWpaPsk = EFalse;

    aSettings.WPAKeyLength = 0;

    aSettings.SecurityMode = *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanSecMode));

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::GetDataL, SecurityMode = %d"),aSettings.SecurityMode));

    if( aSettings.SecurityMode == Wep)

        {    

        // authentication mode can be != open only when WEP in use

        aSettings.AuthenticationMode = *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanAuthMode));

        aSettings.WepIndex = *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanWepIndex));

        ReadWepKeysL(aSettings);

        TraceDump(WARNING_LEVEL,(_L("CWlanSettings::GetDataL, Wep keys succesfully read")));

        }

    else if( aSettings.SecurityMode > Wep)

        {

        aSettings.WPAKeyLength = *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanWpaKeyLength));

        aSettings.EnableWpaPsk = *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanEnableWpaPsk));

        aSettings.PresharedKeyFormat = EWlanPresharedKeyFormatAscii;

        aSettings.WPAPreSharedKey.Zero();

        CMDBField<TDesC8>* wpaKey = static_cast<CMDBField<TDesC8>*>(iWLANRecord->GetFieldByIdL(KCDTIdWlanWpaPreSharedKey));

        aSettings.WPAPreSharedKey.Append( *wpaKey);

        // In WPA-PSK case the preshared key can be either 8 - 63 characters in ASCII

        // or 64 characters in hex.

        //

        // In WAPI case the preshared key format is stored in WEP key #1 format field.

        if( aSettings.EnableWpaPsk )

            {

            if( aSettings.SecurityMode == Wapi )

                {

                TRAP_IGNORE(

                    aSettings.PresharedKeyFormat = *(CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL( KCDTIdWlanFormatKey1 ) ); 

                }

            else if( aSettings.WPAKeyLength == KWLMMaxWpaPskLength )

                {

                aSettings.PresharedKeyFormat = EWlanPresharedKeyFormatHex;

                }

            }            

        }

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::GetDataL end")));

    }

void CWLanSettings::ClearWepKeys( SWLANSettings &aSettings)

    {

    // it is enough to set the keylengths to zero, the key is not attempted to read if length == 0

    aSettings.WepKey1.KeyLength = 0;    

    aSettings.WepKey2.KeyLength = 0;

    aSettings.WepKey3.KeyLength = 0;

    aSettings.WepKey4.KeyLength = 0;

    }

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

// void CWLanSettings::WriteDataL( SWLANSettings& aSettings)

// Writes data from aSettings to a WLANServiceTable record which has 

// previously been fetched (or newly created) from the CommsDat.

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

//

void CWLanSettings::WriteDataL(SWLANSettings& aSettings)

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteDataL start")));

    CMDBField<TDesC>* textField = (CMDBField<TDesC>*)iWLANRecord->GetFieldByIdL(KCDTIdRecordName);

    textField->SetMaxLengthL(aSettings.Name.Length());

    textField->SetL( aSettings.Name);

    *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanServiceId)) = aSettings.ServiceID;

    *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanConnMode)) = aSettings.ConnectionMode;

    textField = (CMDBField<TDesC>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanSSID);

    textField->SetMaxLengthL( aSettings.SSID.Length());

    textField->SetL( aSettings.SSID);

    *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanAuthMode)) = aSettings.AuthenticationMode;

    *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanSecMode)) = aSettings.SecurityMode;

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteDataL, SecurityMode = %d"),aSettings.SecurityMode));

    if( aSettings.SecurityMode == Wep)

        {

        *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanWepIndex)) = aSettings.WepIndex;

        //

        // never write wep keys

        //

        //WriteWepKeysL( aSettings);

        }

    else if( aSettings.SecurityMode > Wep)

        {

        //

        // never write wpa psk key

        //

        /*

        TBuf8<KWlanWpaPskLength> buf;

        TInt len = aSettings.WPAPreSharedKey.Length();

        for( TInt i=0; i<len; i++)

            {

            buf.Append( aSettings.WPAPreSharedKey[i]);

            }

        CMDBField<TDesC8>* binField = (CMDBField<TDesC8>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanWpaPreSharedKey);

        binField->SetMaxLengthL( len);

        binField->SetL( buf);

        */

        *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanWpaKeyLength)) = aSettings.WPAKeyLength;

        *((CMDBField<TUint32>*)iWLANRecord->GetFieldByIdL(KCDTIdWlanEnableWpaPsk)) = aSettings.EnableWpaPsk;

        }

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteDataL end")));

    }

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

// TInt CWLanSettings::GetWlanSettings(TUint32 aId,SWLANSettings &aSettings)

// Gets a WLANServiceTable record from CommsDat with record ID aId

// writes record data to aSettings; the record ID of the fetched record is set as the 

// current record

// returns KErrNone on success

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

//

EXPORT_C TInt CWLanSettings::GetWlanSettings(TUint32 aId,SWLANSettings &aSettings)

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::GetWlanSettings %d"),aId));

    TRAPD( err, CreateWLANServiceRecordL( aId, EWlanRecordId));

    if( !err)

        {

        TRAP( err, GetDataL( aSettings));

        if( !err)

            {

            iCurrentRecord = aId;    

            }

        }

    return err;

    }

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

// TInt CWLanSettings::GetWlanSettingsForService(TUint32 aId,SWLANSettings &aSettings)

// Gets a WLANServiceTable record from CommsDat with service ID aId

// writes record data to aSettings; the record ID of the fetched record is set as the 

// current record

// returns KErrNone on success

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

//

EXPORT_C TInt CWLanSettings::GetWlanSettingsForService(TUint32 aId,SWLANSettings &aSettings)

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::GetWlanSettingsForService %d"),aId));

    TRAPD( err, CreateWLANServiceRecordL( aId, EWlanServiceId));

    if( !err)

        {

        TRAP( err, GetDataL( aSettings));

        if( !err)

            {

            iCurrentRecord = aSettings.Id;        

            }

        }

    return err;

    }

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

// TInt CWLanSettings::GetWlanSettingsForIap( const TUint32 aId, SWLANSettings &aSettings )

// Gets a WLANServiceTable record from CommsDat with IAP ID aId

// writes record data to aSettings; the record ID of the fetched record is set as the 

// current record

// returns KErrNone on success

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

//

EXPORT_C TInt CWLanSettings::GetWlanSettingsForIap(TUint32 aId,SWLANSettings &aSettings)

    {

    TraceDump( WARNING_LEVEL, ( _L("CWlanSettings::GetWlanSettingsForIap %d"), aId ) );

    TRAPD( err, CreateWLANServiceRecordL( aId, EWlanIapId ) );

    if ( !err )

        {

        TRAP( err, GetDataL( aSettings ) );

        if ( !err )

            {

            iCurrentRecord = aSettings.Id;

            }

        }

    return err;

    }

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

// TInt CWLanSettings::GetFirstSettings( SWLANSettings &aSettings )

// 

// Not implemented (never used)

//

// returns KErrNotSupported 

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

//

EXPORT_C TInt CWLanSettings::GetFirstSettings(SWLANSettings& /*aSettings*/) // implemented if needed

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::GetFirstSettings")));

    return KErrNotSupported;

    }

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

// TInt CWLanSettings::GetNextSettings( SWLANSettings &aSettings)

// 

// Not implemented (never used)

//

// returns KErrNotSupported 

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

//

EXPORT_C TInt CWLanSettings::GetNextSettings(SWLANSettings& /*aSettings*/) // implemented if needed

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::GetNextSettings")));

    return KErrNotSupported;

    }

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

// TInt CWLanSettings::AddWlanTable( )

// 

// Not implemented (WLAN table is created by UI)

//

// leaves with KErrNotSupported 

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

//

EXPORT_C void CWLanSettings::AddWlanTableL()

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::AddWlanTableL")));

    User::Leave( KErrNotSupported);    

    }

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

// TInt CWLanSettings::WriteWlanSettings( SWLANSettings &aSettings)

// Writes settings from aSettings to a WLANServiceTable record in CommsDat

// If aSettings.Id == 0, creates a new record, given new record ID 

// is returned in aSettings.Id; otherwise modifies an existing record

// returns KErrNone on success

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

//

EXPORT_C TInt CWLanSettings::WriteWlanSettings( SWLANSettings& aSettings)

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteWlanSettings start")));

    TInt id = ( aSettings.Id == 0 ? KCDNewRecordRequest : aSettings.Id);

    TRAPD( err, CreateWLANServiceRecordL( id, EWlanRecordId));

    if( !err)

        {

        if( aSettings.Id == 0) // create new

            {  // store first (placeholder for a new generic wlan record)

            TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteWlanSettings() creating new record")));

            TRAP( err, iWLANRecord->StoreL( *iDBSession)); 

            if( err)

                {

                return err;

                }

            aSettings.Id = iWLANRecord->RecordId(); // return new id to caller    

            TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteWlanSettings() new record id %d"),aSettings.Id));

            }

        TRAP( err, WriteDataL( aSettings)); // write from SWLANSettings to iWLANRecord

        if( err)

            {

            TRAP_IGNORE( iWLANRecord->DeleteL( *iDBSession)); // get rid of the placeholder

            TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteWlanSettings(), error writing to record; nothing stored")));

            return err;

            }

                // modify the placeholder or an existing record

        TRAP( err, iWLANRecord->ModifyL( *iDBSession));

        }

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteWlanSettings end")));

    return err;

    }

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

// void CWLanSettings::ReadWepKeysL( SWLANSettings &aSettings)

// Reads WEP keys from current WLANServiceTable record; the

// record from which data is read must be fetched from CommsDat

// before calling ReadWepKeysL()

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

//

void CWLanSettings::ReadWepKeysL( SWLANSettings &aSettings)

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::ReadWepKeysL start")));

    // 1    

    CMDBField<TDesC8>* wepKey = static_cast<CMDBField<TDesC8>*>(iWLANRecord->GetFieldByIdL(KCDTIdWlanWepKey1));

    UnPackWepKey( wepKey->GetL(), aSettings.WepKey1, 0);

    // 2

    wepKey = static_cast<CMDBField<TDesC8>*>(iWLANRecord->GetFieldByIdL(KCDTIdWlanWepKey2));

    UnPackWepKey( wepKey->GetL(), aSettings.WepKey2, 1);

    // 3

    wepKey = static_cast<CMDBField<TDesC8>*>(iWLANRecord->GetFieldByIdL(KCDTIdWlanWepKey3));

    UnPackWepKey( wepKey->GetL(), aSettings.WepKey3, 2);

    // 4

    wepKey = static_cast<CMDBField<TDesC8>*>(iWLANRecord->GetFieldByIdL(KCDTIdWlanWepKey4));

    UnPackWepKey( wepKey->GetL(), aSettings.WepKey4, 3);

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::ReadWepKeysL end")));

    }

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

// void CWLanSettings::WriteWepKeysL( SWLANSettings &aSettings)

// Writes WEP keys from aSettings to current WLANServiceTable record

// The record which the data is written to must be fetched from CommsDat

// before calling WriteWepKeysL()

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

//

void CWLanSettings::WriteWepKeysL( SWLANSettings &aSettings)

    {

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteWepKeys start")));

    // 1

    CMDBField<TDesC8>* wepKey = static_cast<CMDBField<TDesC8>*>(iWLANRecord->GetFieldByIdL(KCDTIdWlanWepKey1));

    ASSERT( aSettings.WepKey1.KeyLength <= KWLMMaxWEPKeyLength);

    PackWepKeyL( aSettings.WepKey1, wepKey);

    // 2

    wepKey = static_cast<CMDBField<TDesC8>*>(iWLANRecord->GetFieldByIdL(KCDTIdWlanWepKey2));

    ASSERT( aSettings.WepKey2.KeyLength <= KWLMMaxWEPKeyLength);

    PackWepKeyL( aSettings.WepKey2, wepKey);

    // 3

    wepKey = static_cast<CMDBField<TDesC8>*>(iWLANRecord->GetFieldByIdL(KCDTIdWlanWepKey3));

    ASSERT( aSettings.WepKey3.KeyLength <= KWLMMaxWEPKeyLength);

    PackWepKeyL( aSettings.WepKey3, wepKey);

    // 4

    wepKey = static_cast<CMDBField<TDesC8>*>(iWLANRecord->GetFieldByIdL(KCDTIdWlanWepKey4));

    ASSERT( aSettings.WepKey4.KeyLength <= KWLMMaxWEPKeyLength);

    PackWepKeyL( aSettings.WepKey4, wepKey);

    TraceDump(WARNING_LEVEL,(_L("CWlanSettings::WriteWepKeys end")));

    }

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

// void CWLanSettings::PackWepKeyL( const TWep& aInputKey, CMDBField<TDesC>& aPackedKey)

// Copies wepkey string from wepkey structs TUint8 buffer to a WLANServiceTable

// record's WEP key field descriptor

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

//

void CWLanSettings::PackWepKeyL( const TWep& aInputKey, CMDBField<TDesC8>* aPackedKey)

    {

    TraceDump(ERROR_LEVEL,(_L("CWlanSettings::PackWepKeyL %d"),aInputKey.KeyIndex));

    if( aInputKey.KeyLength > KWLMMaxWEPKeyLength)

        {

        User::Leave( KErrArgument);

        }

    TBuf8<KWLMMaxWEPKeyLength> wepBuf;

    for (TUint i = 0 ; i<aInputKey.KeyLength; i++)

        {

        wepBuf.Append(aInputKey.KeyMaterial[i]);

        }

    aPackedKey->SetL( wepBuf);

    }

void CWLanSettings::AsciiToHexValue( TUint8& aValue)

    {

    // 0-9 = [48(0x30) - 57(0x39)], A-F = [65(0x41) - 70(0x46)], a-f = [97(0x61)-102(0x66)]

    if( aValue >= '0' && aValue <= '9') // 0..9

        {

        aValue -= '0';

        }

    else if( aValue >= 'A' && aValue <= 'F')  // A..F

        {