/*

 * Copyright (c) 2005-2009 Nokia Corporation and/or its subsidiary(-ies).

 * All rights reserved.

 * This component and the accompanying materials are made available

 * under the terms of the License "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: 

 * (c) 1999 Symbian Ltd

 *

 */

 /**

  @file

 */

 #include <e32base.h>

 #include <e32std.h>

 #include <e32test.h>

 #include <random.h>

 #include <f32file.h>

 RTest test(_L("Random Number Generator"));

 TInt gTestPassCount = 0;

 TInt gTestRequestCount = 10;

 TInt gNumberOfRandomNumbers=10000;

 /** Wraps a console and logs output to a file. */

 class CTestConsole:public CConsoleBase

 	{

 	public:

 		static CTestConsole* NewL(CConsoleBase* aCon, const TDesC& aFilename);

 		TInt Create(const TDesC16& aTitle,TSize aSize) {return iCon->Create(aTitle,aSize);};

 		void Read(TRequestStatus& aStatus) {iCon->Read(aStatus);};

 		void ReadCancel(void) {iCon->ReadCancel();};

 		void Write(const TDesC16& aString);

 		TPoint CursorPos(void) const {return iCon->CursorPos();};

 		void SetCursorPosAbs(const TPoint& aPos) {iCon->SetCursorPosAbs(aPos);};

 		void SetCursorPosRel(const TPoint& aPos) {iCon->SetCursorPosRel(aPos);};

 		void SetCursorHeight(TInt aHeight) {iCon->SetCursorHeight(aHeight);};

 		void SetTitle(const TDesC16& aTitle) {iCon->SetTitle(aTitle);};

 		void ClearScreen(void) {iCon->ClearScreen();};

 		void ClearToEndOfLine(void) {iCon->ClearToEndOfLine();};

 		TSize ScreenSize(void) const {return iCon->ScreenSize();};

 		TKeyCode KeyCode(void) const {return iCon->KeyCode();};

 		TUint KeyModifiers(void) const {return iCon->KeyModifiers();};

 		~CTestConsole(void);

 	private:

 		CTestConsole(CConsoleBase* aCon);

 	    void ConstructL(const TDesC& aFilename);

 	    CConsoleBase* iCon;  ///< Pointer to wrapped console, we don't own this

 	    RFs iFs;

 	    RFile iFile;		 ///< Log file

 	};

 CTestConsole* CTestConsole::NewL(CConsoleBase* aCon, const TDesC& aFilename)

 	{

 	CTestConsole* self;

 	self=new (ELeave) CTestConsole(aCon);

 	CleanupStack::PushL(self);

 	self->ConstructL(aFilename);

 	CleanupStack::Pop(self);

 	return self;

 	}

 CTestConsole::CTestConsole(CConsoleBase* aCon) :

 	CConsoleBase(), iCon(aCon)

 	{

 	}

 void CTestConsole::ConstructL(const TDesC& aFilename)

 	{

 	User::LeaveIfError(iFs.Connect());

 	User::LeaveIfError(iFile.Replace(iFs,aFilename,EFileShareAny|EFileWrite));

 	}

 CTestConsole::~CTestConsole(void)

 	{

 	iFile.Close();

 	iFs.Close();

 	}

 void CTestConsole::Write(const TDesC16& aString)

 	{

 	iCon->Write(aString);

 	TUint8 space[200];

 	TPtr8 ptr(space,200);

 	ptr.Copy(aString);

 	iFile.Write(ptr);

 	}

 void Monobit(const TUint8* aData)

 	{

 	const TInt bitcount[256]=

 		{ 0,1,1,2,1,2,2,3,		// 00-07

 		  1,2,2,3,2,3,3,4,		// 08-0f

 		  1,2,2,3,2,3,3,4,		// 10-17

 		  2,3,3,4,3,4,4,5,		// 18-1f

 		  1,2,2,3,2,3,3,4,		// 20-27

 		  2,3,3,4,3,4,4,5,		// 28-2f

 		  2,3,3,4,3,4,4,5,		// 30-37

 		  3,4,4,5,4,5,5,6,		// 38-3f

 		  1,2,2,3,2,3,3,4,		// 40-47

 		  2,3,3,4,3,4,4,5,		// 48-4f

 		  2,3,3,4,3,4,4,5,		// 50-57

 		  3,4,4,5,4,5,5,6,		// 58-5f

 		  2,3,3,4,3,4,4,5,		// 60-67

 		  3,4,4,5,4,5,5,6,		// 68-6f

 		  3,4,4,5,4,5,5,6,		// 70-77

 		  4,5,5,6,5,6,6,7,		// 78-7f

 		  1,2,2,3,2,3,3,4,		// 80-87

 		  2,3,3,4,3,4,4,5,		// 88-8f

 		  2,3,3,4,3,4,4,5,		// 90-97

 		  3,4,4,5,4,5,5,6,		// 98-9f

 		  2,3,3,4,3,4,4,5,		// a0-a7

 		  3,4,4,5,4,5,5,6,		// a8-af

 		  3,4,4,5,4,5,5,6,		// b0-b7

 		  4,5,5,6,5,6,6,7,		// b8-bf

 		  2,3,3,4,3,4,4,5,		// c0-c7

 		  3,4,4,5,4,5,5,6,		// c8-cf

 		  3,4,4,5,4,5,5,6,		// d0-d7

 		  4,5,5,6,5,6,6,7,		// d8-df

 		  3,4,4,5,4,5,5,6,		// e0-e7

 		  4,5,5,6,5,6,6,7,		// e8-ef

 		  4,5,5,6,5,6,6,7,		// f0-f7

 		  5,6,6,7,6,7,7,8		// f8-ff

 		};

 	TInt total=0;

 	TInt i;

 	for (i=0;i<2500;i++)

 		{

 		total+=bitcount[aData[i]];

 		}

 	test.Printf(_L("    Total bitcount %d\r\n"),total);

 	if ((total>9654)&&(total<10346))

 		{

 		test.Printf(_L("    Passed Monobit\r\n"));

 		}

 	else

 		{

 		test.Printf(_L("    ***FAILED!\r\n"));

 		User::Panic(_L("t_random.exe"), KErrGeneral);

 		}

 	}

 void Poker(const TUint8* aData)

 	{

 	TInt f[16]= {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};

 	TInt i;

 	for (i=0;i<2500;i++)

 		{

 		f[(aData[i]&0x0f)]++;

 		f[((aData[i]&0xf0)>>4)]++;

 		}

 	TReal x=0;

 	for (i=0;i<16;i++)

 		{

 		x+=f[i]*f[i];

 		}

 	x*=16;

 	x/=5000;

 	x-=5000;

 	if ((x>1.03)&&(x<57.4))

 		{

 		test.Printf(_L("    Passed poker test\r\n"));

 		}

 	else

 		{

 		test.Printf(_L("    ***FAILED poker test\r\n"));

 		User::Panic(_L("t_random.exe"), KErrGeneral);

 		}

 	}

 void Runs(const TUint8* aData)

 	{

 	TInt i;

 	TInt lastbit=0;

 	TInt count[7][2]={

 		{ 0,0 },

 		{ 0,0 },

 		{ 0,0 },

 		{ 0,0 },

 		{ 0,0 },

 		{ 0,0 },

 		{ 0,0 }

 		};

 	TInt thisrun=0;

 	TInt longrun=0;

 	for (i=0;i<2500;i++)

 		{

 		TInt bit;

 		for (bit=0;bit<8;bit++)

 			{

 			if (((aData[i]>>bit)&1)==lastbit)

 				{

 				thisrun++;

 				}

 			else

 				{

 				if (thisrun<5)

 					{

 					count[thisrun][lastbit]++;

 					}

 				else

 					{

 					count[5][lastbit]++;

 					}

 				lastbit^=1;

 				if (thisrun>longrun)

 					{

 					longrun=thisrun;

 					}

 				thisrun=0;

 				}

 			}

 		}

 	TInt bound[6][2]=

 		{

 			{2267,2733},

 			{1079,1421},

 			{ 502, 748},

 			{ 223, 402},

 			{  90, 223},

 			{  90, 223}

 		};

 	TBool failed=EFalse;

 	for (i=0;i<6;i++)

 		{

 		if (!((count[i][0]>bound[i][0])&&(count[i][0]<bound[i][1])))

 			{

 			test.Printf(_L("    ***FAILED runs test\r\n"));

 			failed=ETrue;

 			}

 		if (!((count[i][1]>bound[i][0])&&(count[i][1]<bound[i][1])))

 			{

 			test.Printf(_L("    ***FAILED runs test\r\n"));

 			failed=ETrue;

 			}

 		}

 	if (!failed)

 		{

 		test.Printf(_L("    Passed runs test\r\n"));

 		}

 	if ( (longrun>34) || (failed) )

 		{

 		test.Printf(_L("    ***FAILED longrun test\r\n"));

 		User::Panic(_L("t_random.exe"), KErrGeneral);

 		}

 	}

 void FIPSTest(const TUint8* aData)

 //	Run some basic tests to check it's returned some numbers

 //	These will panic if a failure is detected

 	{

 	Monobit(aData);

 	Poker(aData);

 	Runs(aData);

 	}

 void WriteFile(const TUint8* aData,const TDesC& aFileName)

 	{

 	RFs fs;

 	fs.Connect();

 	RFile file;

 	TInt err;

 	err=file.Open(fs,aFileName,EFileShareAny|EFileWrite);

 	if (err)

 		{

 		if (file.Create(fs,aFileName,EFileShareAny|EFileWrite))

 			{

 			return;

 			}

 		}

 	TPtrC8 ptr(aData,gNumberOfRandomNumbers);

 	TInt size;

 	file.Size(size);

 	file.Write(size,ptr);

 	file.Close();

 	fs.Close();

 	FIPSTest(aData);

 	}

 void DoTestsL(void)

 	{

 	test.Printf(_L("Run random tests with normal salting\r\n"));    

 	TInt i;

 	TBuf8<16> buf2;

 	for (i=0;i<16;i++)

 		{

 		buf2.SetLength(i);

 		TRandom::RandomL(buf2);

 		}

 	HBufC8* buf=HBufC8::NewMaxL(gNumberOfRandomNumbers);

 	CleanupStack::PushL(buf);

 	TPtr8 buffer=buf->Des();

 	test.Printf(_L("\nRequesting for random numbers.\n"));

 	for (i=0;i<gTestRequestCount;i++)

 		{

 		TPtr8 thePtr(buf->Des());		

 		//	Generate the random data	

 		TRandom::RandomL(buffer);

 		if (buf->Length()!=gNumberOfRandomNumbers)

 		    {

 			User::Leave(KErrGeneral);

 		    }

 		++gTestPassCount;

 	    test.Printf(_L("."));

 		}	

 	// Request for Secure Random numbers.

 	test.Printf(_L("\nRequesting for cryptographically secure random numbers.\n"));

     for (i=0;i<gTestRequestCount;i++)

         {

         TPtr8 thePtr(buf->Des());        

 		// Generate the random data    

         TRAP_IGNORE(TRandom::SecureRandomL(buffer));

         if (buf->Length() != gNumberOfRandomNumbers)

             {

             User::Leave(KErrGeneral);

             }

         ++gTestPassCount;

         test.Printf(_L("."));

         }

 	CleanupStack::PopAndDestroy(buf);

 	}

 void TestsL(void)

 	{

 	TDriveUnit sysDrive (RFs::GetSystemDrive());

 	TBuf<64> logFile (sysDrive.Name());

 	logFile.Append(_L("\\t_random.log"));

 	CTestConsole* con = CTestConsole::NewL(test.Console(), logFile);

 	test.SetConsole(con);

 	DoTestsL();

 	}

 GLDEF_C TInt E32Main(void)

 	{

 	CTrapCleanup* cleanup;

 	cleanup=CTrapCleanup::New();

 	__UHEAP_MARK;

 	test.Start(_L(" @SYMTestCaseID:SEC-CRYPTOSPI-RANDOM-0001 Starting random number generator tests\r\n"));

 	CConsoleBase* originalConsole = test.Console();

 	TRAPD(ret,TestsL());   

     // Infor the user about the test cases' status.

     test.Printf(_L("\n%d tests failed out of %d\r\n"), ((2*gTestRequestCount) - gTestPassCount), 2*gTestRequestCount);

     test (ret == KErrNone);

 	test.End();

 	if (test.Console() != originalConsole)

 		{

 		delete test.Console();

 		test.SetConsole(originalConsole);

 		}	

 	test.Close();

 	__UHEAP_MARKEND;

 	delete cleanup;

 	return(KErrNone);

 	}