pkiutilities/CertSaver/src/CertSaverModel.cpp
branchRCL_3
changeset 67 79bd6bd83ec3
parent 54 94da73d93b58
--- a/pkiutilities/CertSaver/src/CertSaverModel.cpp	Wed Sep 15 13:09:05 2010 +0300
+++ b/pkiutilities/CertSaver/src/CertSaverModel.cpp	Wed Oct 13 15:30:47 2010 +0300
@@ -1,5 +1,5 @@
 /*
-* Copyright (c) 2003-2010 Nokia Corporation and/or its subsidiary(-ies).
+* Copyright (c) 2003-2007 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"
@@ -75,12 +75,7 @@
 _LIT( KEnter, "\n" );
 
 const TUid KTrustedServerTokenUid = { 0x101FB66F };
-const TUid KFileCertStoreTokenUid = { 0x101F501A };
-const TUid KDeviceCertStoreTokenUid = { 0x101FB668 };
-const TUid KFileKeyStoreTokenUid = { KTokenTypeFileKeystore };  // 0x101F7333
-const TUid KDeviceKeyStoreTokenUid = { 0x101FB66A };
-
-const TInt64 KZeroTime = 0;
+const TUid KFileTokensUid = { 0x101F501A };
 
 _LIT_SECURITY_POLICY_V1( KSymbianKeyStoreMgmtPolicy,
                          VID_DEFAULT, ECapabilityWriteUserData );
@@ -99,16 +94,16 @@
     }
 
 // ----------------------------------------------------------
-// CCertSaverModel::CCertSaverModel()
+// CCertSaverModel::CCertSaverModel(
+//    const CCertSaverDocument* aDocument, CCertSaverAppUi* aAppUi)
 // ----------------------------------------------------------
 //
-CCertSaverModel::CCertSaverModel( CCertSaverAppUi* aAppUi, const CCertParser& aParser ) :
+CCertSaverModel::CCertSaverModel(
+    CCertSaverAppUi* aAppUi,
+    const CCertParser& aParser ):
     iAppUi( aAppUi ), iFs( iAppUi->CoeEnv()->FsSession() ), iParser( aParser ),
     iSavedCACertsCount( 0 ), iSavedKeysCount( 0 ),
-    iSavedUserCertsCount( 0 ), iKeyAlreadyExists( EFalse ),
-    iSelectedKeyStoreToken( KFileKeyStoreTokenUid ),
-    iSelectedCertStoreToken( KFileCertStoreTokenUid ),
-    iSelectedKeyStoreIndex( KErrNotFound )
+    iSavedUserCertsCount( 0 ), iKeyAlreadyExists( EFalse )
     {
     }
 
@@ -124,8 +119,6 @@
     iCertOwnerType = aOwnerType;
     iCertFormat = aCertFormat;
     iNewCert = &aCertificate;
-    iSelectedKeyStoreToken = KFileKeyStoreTokenUid;
-    iSelectedCertStoreToken = KFileCertStoreTokenUid;
 
     if ( aOwnerType == EPeerCertificate )
         {
@@ -154,8 +147,8 @@
     MPKCS8DecodedKeyPairData* keyPairData = pkcs8Data->KeyPairData();
     keyPairData->GetKeyIdentifierL( keyIdentifier );
 
-    TTime startDate( KZeroTime );
-    TTime endDate( KZeroTime );
+    TTime startDate;
+    TTime endDate;
     GetKeyValidityPeriodL( startDate, endDate, keyIdentifier );
 
     TInt err = KErrNone;
@@ -163,9 +156,9 @@
     TBuf<KPrivaKeyLabelLength> keyLabel( KPrivateKeyLabel );
     TKeyUsagePKCS15 keyUsage = KeyUsageL( keyIdentifier, pkcs8Data->Algorithm() );
     CleanupStack::PopAndDestroy( pkcs8Data );
-    if ( KeyAlreadyExistsL( startDate, endDate, keyIdentifier, keyUsage) )
+    if (KeyAlreadyExistsL( startDate, endDate, keyIdentifier, keyUsage) )
         {
-        return;
+        User::Leave( KErrNone );
         }
 
     TInt accessType( 0 );
@@ -175,7 +168,8 @@
     for ( TInt i = 0; i < KAttempts; i++ )
         {
         CreateKeyLabelL( keyLabel );
-        err = iWrapper->AddKey( iSelectedKeyStoreIndex, aKey, keyUsage, keyLabel,
+        // Should not use hardcoded index
+        err = iWrapper->AddKey( 0, aKey, keyUsage, keyLabel,
             accessType, startDate, endDate, keyInfo );
         if ( err != KErrAlreadyExists )
             {
@@ -304,21 +298,24 @@
 //
 // ----------------------------------------------------------
 //
-void CCertSaverModel::GetKeyValidityPeriodL( TTime& aStartDate, TTime& aEndDate,
-        const TKeyIdentifier& aKeyIdentifier, const CArrayPtr<CX509Certificate>& aCertArray )
+void CCertSaverModel::GetKeyValidityPeriodL(
+    TTime& aStartDate,
+    TTime& aEndDate,
+    const TKeyIdentifier& aKeyIdentifier )
     {
-    const TTime timeNotSet( KZeroTime );
 
-    for( TInt index = 0; index < aCertArray.Count(); index++ )
+    for ( TInt i = 0; i < iParser.UserCertificates().Count(); i++ )
         {
-        const CX509Certificate* cert = aCertArray.At( index );
-        if( cert->KeyIdentifierL() == aKeyIdentifier )
+        const CX509Certificate* cert = iParser.UserCertificates().At( i );
+        if ( cert->KeyIdentifierL() == aKeyIdentifier )
             {
-            if ( aStartDate == timeNotSet || aStartDate > cert->ValidityPeriod().Start() )
+            // Associated certificate found
+            // In the first round aStartDate and aEndDate is initialised.
+            if ( i == 0 || aStartDate > cert->ValidityPeriod().Start() )
                 {
                 aStartDate = cert->ValidityPeriod().Start();
                 }
-            if ( aEndDate == timeNotSet || aEndDate < cert->ValidityPeriod().Finish() )
+            if ( i == 0 || aEndDate < cert->ValidityPeriod().Finish() )
                 {
                 aEndDate = cert->ValidityPeriod().Finish();
                 }
@@ -327,18 +324,6 @@
     }
 
 // ----------------------------------------------------------
-// CCertSaverModel::GetKeyValidityPeriodL()
-//
-// ----------------------------------------------------------
-//
-void CCertSaverModel::GetKeyValidityPeriodL( TTime& aStartDate, TTime& aEndDate,
-        const TKeyIdentifier& aKeyIdentifier )
-    {
-    GetKeyValidityPeriodL( aStartDate, aEndDate, aKeyIdentifier, iParser.UserCertificates() );
-    GetKeyValidityPeriodL( aStartDate, aEndDate, aKeyIdentifier, iParser.CACertificates() );
-    }
-
-// ----------------------------------------------------------
 // CCertSaverModel::CreateKeyLabelL()
 //
 // ----------------------------------------------------------
@@ -385,11 +370,11 @@
             {
             TUid uid =
             iWrapper->UnifiedKeyStore().KeyStoreManager(i).Token().TokenType().Type();
-            if ( uid == iSelectedKeyStoreToken )
+            if ( uid == TUid::Uid( KTokenTypeFileKeystore ) )
                 // if this is not found, we use the first one,
                 // which is already initialised
                 {
-                iSelectedKeyStoreIndex = i;
+                iSelectedKeyStore = i;
                 }
             }
         }
@@ -529,6 +514,7 @@
     CleanupStack::PopAndDestroy( message );
     CleanupStack::PushL( dlg );
 
+
     dlg->PrepareLC( R_MESSAGE_QUERY_DOSAVE );
 
     HBufC* header = StringLoader::LoadLC( R_CERTSAVER_DETAILS_HEADING );
@@ -634,7 +620,7 @@
         // If the certificate is already in CACerts.dat,
         // then don't save it
         if ( iNewCert->Fingerprint() == certificate->Fingerprint() &&
-            (( *iEntries[i]).Handle().iTokenHandle.iTokenTypeUid == iSelectedCertStoreToken ) )
+            (( *iEntries[i]).Handle().iTokenHandle.iTokenTypeUid == KFileTokensUid ) )
             {
             found = ETrue;
             }
@@ -680,26 +666,8 @@
             }
         }
 
-    // Select which store to use
-    TUid certstoreToken;
-    switch( iCertOwnerType )
-        {
-        case ECACertificate:
-            certstoreToken = KFileCertStoreTokenUid;
-            break;
-        case EUserCertificate:
-            certstoreToken = iSelectedCertStoreToken;
-            break;
-        case EPeerCertificate:
-            certstoreToken = KTrustedServerTokenUid;
-            break;
-        default:
-            certstoreToken = KFileCertStoreTokenUid;
-            break;
-        }
-
     // Get interface to writable store.
-    TInt certstoreIndex = -1;
+    TInt certstoreIndex( -1 );
     TInt count = iUnifiedCertStore->WritableCertStoreCount();
     if ( count > 0 )
         {
@@ -710,7 +678,9 @@
 
             MCTToken& token = writableCertStore.Token();
             TUid tokenuid = token.Handle().iTokenTypeUid;
-            if ( tokenuid == certstoreToken )
+            if ( ( tokenuid == KTrustedServerTokenUid ) && ( iCertOwnerType == EPeerCertificate ) ||
+                 ( tokenuid == KFileTokensUid ) && ( iCertOwnerType == ECACertificate ) ||
+                 ( tokenuid == KFileTokensUid ) && ( iCertOwnerType == EUserCertificate ) )
                 {
                 certstoreIndex = i;
                 break;
@@ -788,8 +758,7 @@
 
         // takes ownership of trusterUids
         status = iWrapper->SetApplicability(
-            iUnifiedCertStore->WritableCertStore( certstoreIndex ),
-            *( iEntries[ 0 ] ), iTrusterUids );
+            iUnifiedCertStore->WritableCertStore( certstoreIndex ), *(iEntries[0]), iTrusterUids );
 
         // If error happened, show error note and give up. Otherwise, continue
         if ( status )
@@ -800,8 +769,7 @@
 
         // Downloaded certificate is trusted by default
         status = iWrapper->SetTrust(
-            iUnifiedCertStore->WritableCertStore( certstoreIndex ),
-            *( iEntries [ 0 ] ), ETrue );
+            iUnifiedCertStore->WritableCertStore(certstoreIndex), *(iEntries[0]), ETrue );
         if ( status )
             {
             HandleSaveErrorL( status );
@@ -1140,7 +1108,7 @@
     // format the date to user readable format. The format is locale dependent
     finishValue.FormatL( finishString, *dateFormatString );
     AknTextUtils::DisplayTextLanguageSpecificNumberConversion( finishString );
-    CleanupStack::PopAndDestroy( dateFormatString );
+    CleanupStack::PopAndDestroy(); // dateFormatString
     aMessage.Append( finishString );
     AddNewlinesToMessage( aMessage );
     }
@@ -1526,12 +1494,12 @@
     {
     HBufC* message = HBufC::NewLC( KMaxLengthTextMeassageBody );
     TPtr msgPtr = message->Des();
-	TInt keysCount = iParser.Keys().Count();
-    TInt userCertificateCount = iParser.UserCertificates().Count();
-    TInt caCertificateCount = iParser.CACertificates().Count();
-    ConstructPKCS12QueryMsgL( msgPtr, keysCount, userCertificateCount, caCertificateCount );
-    TBool save = DoMessageQueryL( R_MESSAGE_QUERY_DOSAVE,
-            R_CERTSAVER_HEADER_PKCS12_FILE_CONTAINS, *message );
+    ConstructPKCS12QueryMsgL(msgPtr, iParser.Keys().Count(),
+                            iParser.UserCertificates().Count(),
+                            iParser.CACertificates().Count() );
+    TBool save = DoMessageQueryL(
+        R_MESSAGE_QUERY_DOSAVE, R_CERTSAVER_HEADER_PKCS12_FILE_CONTAINS,
+        *message );
     CleanupStack::PopAndDestroy( message );
     message = NULL;
     if ( !save )
@@ -1539,31 +1507,14 @@
         ShowInformationNoteL( R_CERTSAVER_PKCS12_DISCARDED );
         User::Leave( KErrExitApp );
         }
-
-    if ( keysCount )
-        {
-        CAknQueryDialog* protectWithPasswordQuery = CAknQueryDialog::NewL();
-        TInt buttonId = protectWithPasswordQuery->ExecuteLD( R_CERTSAVER_PROTECT_WITH_PASSWORD );
-        if ( buttonId == EAknSoftkeyYes )
-            {
-            iSelectedKeyStoreToken = KFileKeyStoreTokenUid;
-            iSelectedCertStoreToken = KFileCertStoreTokenUid;
-            }
-        else
-            {
-            iSelectedKeyStoreToken = KDeviceKeyStoreTokenUid;
-            iSelectedCertStoreToken = KDeviceCertStoreTokenUid;
-            }
-        }
-
     TInt status = KErrNone;
     // save private keys
     TRAP( status, SavePrivateKeyL() );
 
     // save user certificates if private key was saved.
-    if ( ( iSavedKeysCount > 0 || iKeyAlreadyExists ) && userCertificateCount > 0 )
+    if ( ( iSavedKeysCount > 0 || iKeyAlreadyExists ) && iParser.UserCertificates().Count() > 0 )
         {
-        for ( TInt i = 0; i < userCertificateCount; i++ )
+        for ( TInt i = 0; i < iParser.UserCertificates().Count(); i++ )
             {
             const CX509Certificate* cert = iParser.UserCertificates().At(i);
             iCertOwnerType = EUserCertificate;
@@ -1576,11 +1527,10 @@
                 }
             }
         }
-
     // save CA certificates
-    if ( caCertificateCount > 0 )
+    if ( iParser.CACertificates().Count() > 0 )
         {
-        for ( TInt i = 0; i < caCertificateCount; i++ )
+        for ( TInt i = 0; i < iParser.CACertificates().Count(); i++ )
             {
             const CX509Certificate* cert = iParser.CACertificates().At( i );
             iCertOwnerType = ECACertificate;
@@ -1593,7 +1543,6 @@
                 }
             }
         }
-
     if ( iSavedCACertsCount != 0 || iSavedKeysCount != 0
         || iSavedUserCertsCount != 0 )
         {
@@ -1601,7 +1550,8 @@
         TPtr msgPtr2 = message->Des();
         ConstructPKCS12QueryMsgL(
             msgPtr2, iSavedKeysCount, iSavedUserCertsCount, iSavedCACertsCount );
-        DoMessageQueryL( R_MESSAGE_QUERY_SAVED, R_CERTSAVER_HEADER_SAVED, *message );
+        DoMessageQueryL(
+            R_MESSAGE_QUERY_SAVED, R_CERTSAVER_HEADER_SAVED, *message );
         CleanupStack::PopAndDestroy( message );
         }
     else