18 #include <e32property.h>                // RProperty

    19 #include <s32mem.h>                     // RDesRead/WriteStream

    20 

    28 #include "emailshutdownconst.h"

    31 const TInt KEmailUidExtraBuffer = 2 * KEmailPlatformApiUidItemSize;

    51     CleanupStack::PushL( self );

    52     self->ConstructL();

    53     CleanupStack::Pop( self );

    63     TRAP_IGNORE( AppendOrRemoveUidL( EEmailUidModeRemove ) );

   115 // ---------------------------------------------------------------------------

   116 // CEmailInterfaceFactoryImpl::AppendOrRemoveUidL

   117 // ---------------------------------------------------------------------------

   118 //

   119 void CEmailInterfaceFactoryImpl::AppendOrRemoveUidL(

   120         const TEmailUidAppendRemoveMode aMode )

   121     {

   122     // Read buffer length

   123     TInt bufLength( 0 );

   124     User::LeaveIfError( RProperty::Get( KEmailShutdownPsCategory,

   125                                         EEmailPsKeyPlatformApiAppsToCloseLength,

   126                                         bufLength ) );

   127 

   128     // Allocate buffer for reading and then read the list of UIDs from P&S.

   129     // Adding some extra buffer just in case the size key and actual list

   130     // are out of sync. This shouldn't happen, but you never know.

   131     HBufC8* readBuf = HBufC8::NewLC( bufLength + KEmailUidExtraBuffer );

   132     TPtr8 readPtr = readBuf->Des();

   133     

   134     User::LeaveIfError( RProperty::Get( KEmailShutdownPsCategory,

   135                                         EEmailPsKeyPlatformApiAppsToClose,

   136                                         readPtr ) );

   137     

   138     // For writing get the size of the original buffer + room for our own UID

   139     // if needed

   140     TInt writeBufSize = readPtr.Length();

   141     if( aMode == EEmailUidModeAppend )

   142         {

   143         writeBufSize += KEmailPlatformApiUidItemSize;

   144         }

   145     

   146     HBufC8* writeBuf = HBufC8::NewLC( writeBufSize );

   147     TPtr8 writePtr = writeBuf->Des();

   148 

   149     // Read and write streams used to read/write the UIDs from/to descriptors

   150     RDesReadStream readStream( readPtr );

   151     CleanupClosePushL( readStream );

   152 

   153     RDesWriteStream writeStream( writePtr );

   154     CleanupClosePushL( writeStream );

   155 

   156     // Get our own process UID

   157     RProcess ownProcess;

   158     TUid ownUid = ownProcess.SecureId();

   159     ownProcess.Close();

   160 

   161     TInt itemsCount = readPtr.Length() / KEmailPlatformApiUidItemSize;

   162 

   163     TBool ownUidFound = EFalse;

   164     TInt writeLength = 0;

   165     for ( TInt ii = 0;ii < itemsCount; ++ii )

   166         {

   167         // Read next UID from the stream

   168         TUid item = TUid::Uid( readStream.ReadInt32L() );

   169         

   170         // We can skip our own UID. If we are removing, then we don't want

   171         // our UID to be written. If we are adding, we don't need to set

   172         // the new values as our UID already exists in the list.

   173         if( item == ownUid )

   174             {

   175             ownUidFound = ETrue;

   176             if( aMode == EEmailUidModeAppend )

   177                 {

   178                 // Our own UID is already in the list, so no need to update

   179                 // the list. Hence we can quit here.

   180                 break;

   181                 }

   182             }

   183         else

   184             {

   185             writeStream.WriteInt32L( item.iUid );

   186             writeLength += KEmailPlatformApiUidItemSize;

   187             }

   188         }

   189     

   190     // If we are appending our UID and it wasn't found from the list,

   191     // write it to the stream

   192     if( aMode == EEmailUidModeAppend && !ownUidFound )

   193         {

   194         writeStream.WriteInt32L( ownUid.iUid );

   195         writeLength += KEmailPlatformApiUidItemSize;

   196         }

   197 

   198     // Set correct length for the write ptr buffer as it might not be

   199     // updated correctly by the write stream

   200     writePtr.SetLength( writeLength );

   201 

   202     // Set new values to P&S only if something has changed, so either:

   203     // 1) We are appending our UID and it didn't exist before

   204     // 2) We are removing our UID and it did exist before

   205     if( ( aMode == EEmailUidModeAppend && !ownUidFound ) ||

   206         ( aMode == EEmailUidModeRemove && ownUidFound ) )

   207         {

   208         // Write first the UID list as it is more probable to fail, writing

   209         // plain integer value shouldn't fail in any case. This way these

   210         // values stay in sync also in case of error, as the list length

   211         // gets updated only if the list itself is updated succesfully.

   212         User::LeaveIfError( RProperty::Set( KEmailShutdownPsCategory,

   213                         EEmailPsKeyPlatformApiAppsToClose,

   214                         writePtr ) );

   215 

   216         User::LeaveIfError( RProperty::Set( KEmailShutdownPsCategory,

   217                         EEmailPsKeyPlatformApiAppsToCloseLength,

   218                         writeLength ) );

   219         }

   220     

   221     CleanupStack::PopAndDestroy( 4, readBuf );

   222     }

   223