os/security/crypto/weakcrypto/test/tasymmetric/tdsaprimegen.cpp
author sl
Tue, 10 Jun 2014 14:32:02 +0200
changeset 1 260cb5ec6c19
permissions -rw-r--r--
Update contrib.
     1 /*
     2 * Copyright (c) 1998-2009 Nokia Corporation and/or its subsidiary(-ies).
     3 * All rights reserved.
     4 * This component and the accompanying materials are made available
     5 * under the terms of the License "Eclipse Public License v1.0"
     6 * which accompanies this distribution, and is available
     7 * at the URL "http://www.eclipse.org/legal/epl-v10.html".
     8 *
     9 * Initial Contributors:
    10 * Nokia Corporation - initial contribution.
    11 *
    12 * Contributors:
    13 *
    14 * Description: 
    15 *
    16 */
    17 
    18 
    19 #include "tdsaprimegen.h"
    20 #include "t_input.h"
    21 #include <asymmetric.h>
    22 #include <random.h>
    23 #include <padding.h>
    24 #include <bigint.h>
    25 #include "tbrokenrandom.h"
    26 
    27 CTestAction* CDSAPrimeGen::NewL(RFs& aFs, CConsoleBase& aConsole, Output& aOut, 
    28 	const TTestActionSpec& aTestActionSpec)
    29 	{
    30 	CTestAction* self = CDSAPrimeGen::NewLC(aFs, aConsole,
    31 		aOut, aTestActionSpec);
    32 	CleanupStack::Pop();
    33 	return self;
    34 	}
    35 
    36 CTestAction* CDSAPrimeGen::NewLC(RFs& aFs, CConsoleBase& aConsole, Output& aOut, 
    37 	const TTestActionSpec& aTestActionSpec)
    38 	{
    39 	CDSAPrimeGen* self = new(ELeave) CDSAPrimeGen(aFs, aConsole, aOut);
    40 	CleanupStack::PushL(self);
    41 	self->ConstructL(aTestActionSpec);
    42 	return self;
    43 	}
    44 
    45 CDSAPrimeGen::~CDSAPrimeGen()
    46 	{
    47 	delete iBody;
    48 	iP.Close();
    49 	iQ.Close();
    50 	delete const_cast<CDSAPrimeCertificate*>(iPrimeCert);
    51 	}
    52 
    53 CDSAPrimeGen::CDSAPrimeGen(RFs& aFs, CConsoleBase& aConsole, Output& aOut)
    54 	: CTestAction(aConsole, aOut), iFs(aFs)
    55 	{
    56 	}
    57 
    58 void CDSAPrimeGen::ConstructL(const TTestActionSpec& aTestActionSpec)
    59 	{
    60 	CTestAction::ConstructL(aTestActionSpec);
    61 	iBody = HBufC8::NewL(aTestActionSpec.iActionBody.Length());
    62 	iBody->Des().Copy(aTestActionSpec.iActionBody);
    63 	
    64 
    65 	iKeyBits = Input::ParseIntElement(*iBody, _L8("<keybits>"), _L8("</keybits>"));
    66 
    67 	HBufC8* p = Input::ParseElementHexL(*iBody, _L8("<p>"));
    68 	CleanupStack::PushL(p);
    69 	iP = RInteger::NewL(*p);
    70 	CleanupStack::PopAndDestroy(p);
    71 
    72 	HBufC8* q = Input::ParseElementHexL(*iBody, _L8("<q>"));
    73 	CleanupStack::PushL(q);
    74 	iQ = RInteger::NewL(*q);
    75 	CleanupStack::PopAndDestroy(q);
    76 
    77 	TUint counter = Input::ParseIntElement(*iBody, _L8("<c>"), _L8("</c>"));
    78 	HBufC8* seed = Input::ParseElementHexL(*iBody, _L8("<seed>"));
    79 	CleanupStack::PushL(seed);
    80 	iPrimeCert = CDSAPrimeCertificate::NewL(*seed, counter);
    81 	CleanupStack::PopAndDestroy(seed);
    82 	}
    83 
    84 void CDSAPrimeGen::DoPerformPrerequisite(TRequestStatus& aStatus)
    85 	{
    86 	TRequestStatus* status = &aStatus;
    87 	User::RequestComplete(status, KErrNone);
    88 	iActionState = CTestAction::EAction;
    89 	}
    90 
    91 void CDSAPrimeGen::DoPerformPostrequisite(TRequestStatus& aStatus)
    92 	{
    93 	TRequestStatus* status = &aStatus;
    94 
    95 	iFinished = ETrue;
    96 	User::RequestComplete(status, KErrNone);
    97 	}
    98 
    99 void CDSAPrimeGen::DoReportAction(void)
   100 	{
   101 	}
   102 
   103 void CDSAPrimeGen::DoCheckResult(TInt)
   104 	{
   105 	if (iResult)
   106 		iConsole.Printf(_L("."));
   107 	else
   108 		iConsole.Printf(_L("X"));
   109 	}
   110 
   111 void CDSAPrimeGen::PerformAction(TRequestStatus& aStatus)
   112 	{
   113 	__UHEAP_MARK;
   114 	TRequestStatus* status = &aStatus;
   115 	iResult = EFalse;
   116 
   117 	CRandomSetSource* rng = new(ELeave)CRandomSetSource(iPrimeCert->Seed());
   118 	SetThreadRandomLC(rng);
   119 	CDSAKeyPair* dsaPair = CDSAKeyPair::NewLC(iKeyBits);
   120 	
   121 	//this is more or less just calling the prime generation routine over again
   122 	//but it does test if ValidatePrimesL and the NewLC generate the same primes
   123 	iResult = dsaPair->PublicKey().ValidatePrimesL(*iPrimeCert);
   124 
   125 	//we've already checked that the ValidatePrime + NewLC give same results
   126 	//this now checks against the vectors in the test scripts
   127 	if(iP != dsaPair->PublicKey().P() ||
   128 		iQ != dsaPair->PublicKey().Q() || 
   129 		iP != dsaPair->PrivateKey().P() ||
   130 		iQ != dsaPair->PrivateKey().Q() )
   131 		{
   132 		iResult = EFalse;
   133 		}
   134 
   135 	CleanupStack::PopAndDestroy(dsaPair); 
   136 	CleanupStack::PopAndDestroy(); //SetThreadRandomLC
   137 	User::RequestComplete(status, KErrNone);
   138 	iActionState = CTestAction::EPostrequisite;
   139 	__UHEAP_MARKEND;
   140 	}
   141