/*
* Copyright (c) 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:  
*
*/




//  CLASS HEADER
#include "UT_CSRTPMasterSalt.h"

//  EXTERNAL INCLUDES
#include <digia/eunit/eunitmacros.h>


//  INTERNAL INCLUDES
#include "srtpmastersalt.h"
#include "srtputils.h"
#include "srtpdef.h"

_LIT8(KRFCTestSalt112bits,              "0EC675AD498AFEEBB6960B3AABE6");
_LIT8(KRFCTestSalt112bits2,              "0E0E0E0E0E0E0E0E0E0E0E0E");


// CONSTRUCTION
UT_CSRTPMasterSalt* UT_CSRTPMasterSalt::NewL()
    {
    UT_CSRTPMasterSalt* self = UT_CSRTPMasterSalt::NewLC(); 
    CleanupStack::Pop();

    return self;
    }

UT_CSRTPMasterSalt* UT_CSRTPMasterSalt::NewLC()
    {
    UT_CSRTPMasterSalt* self = new( ELeave ) UT_CSRTPMasterSalt();
    CleanupStack::PushL( self );

	self->ConstructL(); 

    return self;
    }

// Destructor (virtual by CBase)
UT_CSRTPMasterSalt::~UT_CSRTPMasterSalt()
    {
    }

// Default constructor
UT_CSRTPMasterSalt::UT_CSRTPMasterSalt() 
    {
    }

// Second phase construct
void UT_CSRTPMasterSalt::ConstructL()
    {
    // The ConstructL from the base class CEUnitTestSuiteClass must be called.
    // It generates the test case table.
    CEUnitTestSuiteClass::ConstructL();
    }


//  METHODS

void UT_CSRTPMasterSalt::SetupL(  )
    {
    HBufC8* masterSalt =HBufC8::NewL(KRFCTestSalt112bits().Length());
    CleanupStack::PushL( masterSalt);
	*masterSalt = KRFCTestSalt112bits;
	
	
	HBufC8* masterSalt2 =HBufC8::NewL(KRFCTestSalt112bits2().Length());
    CleanupStack::PushL( masterSalt2);
	*masterSalt2 = KRFCTestSalt112bits2;
	
    iMasterSalt = CSRTPMasterSalt::NewL( *masterSalt,KSRTPDefSessionSaltingKeyLength );
    iMasterSalt2 = CSRTPMasterSalt::NewL( *masterSalt,100 );
    iMasterSalt3 = CSRTPMasterSalt::NewL( *masterSalt2, KSRTPDefSessionSaltingKeyLength);
    iMasterSalt4 = CSRTPMasterSalt::NewL( *masterSalt);
    CleanupStack::PopAndDestroy( masterSalt2 );
    CleanupStack::PopAndDestroy( masterSalt );
    
    } 

void UT_CSRTPMasterSalt::Teardown(  )
    {
	delete iMasterSalt;
	delete iMasterSalt2;
	delete iMasterSalt3;
	delete iMasterSalt4;
	iRFCTestSalt112bits.Zero();
    iRFCTestSalt112bits_dehexed.Zero();      
    }

void UT_CSRTPMasterSalt::Hex(HBufC8& aString)
{
    TPtr8 ptr=aString.Des();
    TInt length = aString.Length();
    if (aString.Length()%2)
        {
        ptr.SetLength(0);
        return;
        }
    TInt i;
    for (i=0;i<aString.Length();i+=2)
        {
        TUint8 tmp;
        tmp=(TUint8)(aString[i]-(aString[i]>'9'?('A'-10):'0'));
        tmp*=16;
        tmp|=(TUint8)(aString[i+1]-(aString[i+1]>'9'?('A'-10):'0'));
        ptr[i/2]=tmp;
        }
    ptr.SetLength(aString.Length()/2);

}
void UT_CSRTPMasterSalt::UT_CSRTPMasterSalt_TSRTPMasterSaltL(  )
    {
    iSaltKey.Append(0x0E);
    iSaltKey.Append(0xC6);
    iSaltKey.Append(0x75);
    iSaltKey.Append(0xAD);
    iSaltKey.Append(0x49);
    iSaltKey.Append(0x8A);
    iSaltKey.Append(0xFE);
    iSaltKey.Append(0xEB);
    iSaltKey.Append(0xB6);
    iSaltKey.Append(0x96);
    iSaltKey.Append(0x0B);
    iSaltKey.Append(0x3A);
    iSaltKey.Append(0xAB);
    iSaltKey.Append(0xE6); 
    
    const TBuf8<32> salt = iMasterSalt->MasterSalt();      
    EUNIT_ASSERT( salt.Compare(iSaltKey) == 0);      
    
    EUNIT_ASSERT( iMasterSalt->SaltLength() == KSRTPDefaultMasterSaltLength );   
    iSaltKey.Zero(); 
    }

void UT_CSRTPMasterSalt::UT_CSRTPMasterSalt_MasterSaltL(  )
    {
//     initialize salting key (from RFC)
    iRFCTestSalt112bits.Copy(KRFCTestSalt112bits);
    TSRTPUtils::DeHex(iRFCTestSalt112bits, iRFCTestSalt112bits_dehexed); 
    
    const TBuf8<32> salt = iMasterSalt->MasterSalt();

    EUNIT_ASSERT( salt.Compare(iRFCTestSalt112bits_dehexed) == 0);      
      
//    EUNIT_ASSERT( salt.Compare(iSaltKey) == 0);      
    }

void UT_CSRTPMasterSalt::UT_CSRTPMasterSalt_SaltLengthL(  )
    {        
    EUNIT_ASSERT( iMasterSalt->SaltLength() == KSRTPDefaultMasterSaltLength );
    }

void UT_CSRTPMasterSalt::UT_CSRTPMasterSalt_ValidL(  )
    {            
    //this one should pass
    EUNIT_ASSERT( iMasterSalt->Valid());
    //invalid session salt length
    EUNIT_ASSERT( !iMasterSalt2->Valid());
    //invalid master salt length
    EUNIT_ASSERT( !iMasterSalt3->Valid());
    //test constructing with default values
    EUNIT_ASSERT( iMasterSalt4->Valid());
    }

void UT_CSRTPMasterSalt::TestEqualL(  )
	{
	HBufC8* masterSalt =HBufC8::NewL(KRFCTestSalt112bits().Length());
    CleanupStack::PushL( masterSalt);
	*masterSalt = KRFCTestSalt112bits;
	Hex(*masterSalt);
	const CSRTPMasterSalt& tempSalt=iMasterSalt2->operator=(*iMasterSalt2);

	EUNIT_ASSERT(tempSalt.MasterSalt() == *masterSalt);
	EUNIT_ASSERT(tempSalt.SaltLength()==100);
	CleanupStack::PopAndDestroy(masterSalt);
	}

void UT_CSRTPMasterSalt::UT_CSRTPMasterSalt_ConstructLL(  )
	{
	HBufC8* masterSalt =HBufC8::NewL(KRFCTestSalt112bits().Length());
    CleanupStack::PushL( masterSalt);
	*masterSalt = KRFCTestSalt112bits;
	Hex(*masterSalt);
	
	const CSRTPMasterSalt& masterSaltlength14 = iMasterSalt->operator=(*iMasterSalt);
 	
    EUNIT_ASSERT( masterSaltlength14.MasterSalt()== *masterSalt);  
    CleanupStack::PopAndDestroy(masterSalt);
 	 
 	
	}
//  TEST TABLE

EUNIT_BEGIN_TEST_TABLE( 
    UT_CSRTPMasterSalt,
    "CSRTPMasterSalt",
    "UNIT" )

EUNIT_TEST(
    "TSRTPMasterSalt - test ",
    "CSRTPMasterSalt",
    "CSRTPMasterSalt",
    "FUNCTIONALITY",
    SetupL, UT_CSRTPMasterSalt_TSRTPMasterSaltL, Teardown)

EUNIT_TEST(
    "MasterSalt - test ",
    "CSRTPMasterSalt",
    "MasterSalt",
    "FUNCTIONALITY",
    SetupL, UT_CSRTPMasterSalt_MasterSaltL, Teardown)

EUNIT_TEST(
    "SaltLength - test ",
    "CSRTPMasterSalt",
    "SaltLength",
    "FUNCTIONALITY",
    SetupL, UT_CSRTPMasterSalt_SaltLengthL, Teardown)

EUNIT_TEST(
    "Valid - test ",
    "CSRTPMasterSalt",
    "Valid",
    "FUNCTIONALITY",
    SetupL, UT_CSRTPMasterSalt_ValidL, Teardown)

EUNIT_TEST(
    "Test = ",
    "CSRTPMasterSalt",
    "Test Operator equal",
    "FUNCTIONALITY",
    SetupL, TestEqualL, Teardown)

EUNIT_TEST(
    "Test Construct",
    "CSRTPMasterSalt",
    "Test contruct keys with different length",
    "FUNCTIONALITY",
    SetupL, UT_CSRTPMasterSalt_ConstructLL, Teardown)
    
EUNIT_END_TEST_TABLE

//  END OF FILE