omadrm/drmengine/agentv2/src/Oma2AgentImportFile.cpp
changeset 0 95b198f216e5
child 18 8a03a285ab14
equal deleted inserted replaced
-1:000000000000 0:95b198f216e5
       
     1 /*
       
     2 * Copyright (c) 2002-2004 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 "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:  ?Description
       
    15 *
       
    16 */
       
    17 
       
    18 
       
    19 // INCLUDE FILES
       
    20 #include <e32std.h>
       
    21 #include <s32mem.h>
       
    22 #include <f32file.h>
       
    23 #include <caf/caf.h>
       
    24 #include <caf/caferr.h>
       
    25 #include <caf/SupplierOutputFile.h>
       
    26 #include "Oma2Agent.h"
       
    27 #include "Oma2AgentImportfile.h"
       
    28 #include "DcfCommon.h"
       
    29 #include "DrmRights.h"
       
    30 #include "DrmMessageParser.h"
       
    31 #include "Oma1DcfCreator.h"
       
    32 #include "DrmRightsParser.h"
       
    33 #include "DrmProtectedRoParser.h"
       
    34 
       
    35 using namespace ContentAccess;
       
    36 
       
    37 // LOCAL FUNCTION PROTOTYPES
       
    38 template<class S>
       
    39 LOCAL_C void PointerArrayResetDestroyAndClose(TAny* aPtr);
       
    40 
       
    41 // ============================= LOCAL FUNCTIONS ===============================
       
    42 
       
    43 // -----------------------------------------------------------------------------
       
    44 // ?function_name ?description.
       
    45 // ?description
       
    46 // Returns: ?value_1: ?description
       
    47 //          ?value_n: ?description_line1
       
    48 //                    ?description_line2
       
    49 // -----------------------------------------------------------------------------
       
    50 //
       
    51 template<class S>
       
    52 void PointerArrayResetDestroyAndClose(TAny* aPtr)
       
    53     {
       
    54     (reinterpret_cast<RPointerArray<S>*>(aPtr))->ResetAndDestroy();
       
    55     (reinterpret_cast<RPointerArray<S>*>(aPtr))->Close();
       
    56     }
       
    57 
       
    58 // ============================ MEMBER FUNCTIONS ===============================
       
    59 
       
    60 // -----------------------------------------------------------------------------
       
    61 // COma2AgentImportFile::
       
    62 // 
       
    63 // -----------------------------------------------------------------------------
       
    64 //
       
    65 COma2AgentImportFile* COma2AgentImportFile::NewL(
       
    66     const TDesC8& aMimeType,
       
    67     const CMetaDataArray& aMetaDataArray,
       
    68     const TDesC& aOutputDirectory,
       
    69     const TDesC& aSuggestedFileName)
       
    70 	{
       
    71 	COma2AgentImportFile* self=new(ELeave) COma2AgentImportFile(EFalse);
       
    72 	CleanupStack::PushL(self);
       
    73 	self->ConstructL(aMimeType, aMetaDataArray, aOutputDirectory,
       
    74 	    aSuggestedFileName);
       
    75 	CleanupStack::Pop(self);
       
    76 	return self;
       
    77 	}
       
    78 
       
    79 // -----------------------------------------------------------------------------
       
    80 // COma2AgentImportFile::
       
    81 // 
       
    82 // -----------------------------------------------------------------------------
       
    83 //
       
    84 COma2AgentImportFile* COma2AgentImportFile::NewL(
       
    85     const TDesC8& aMimeType,
       
    86     const CMetaDataArray& aMetaDataArray)
       
    87 	{
       
    88 	COma2AgentImportFile* self=new(ELeave) COma2AgentImportFile(EFalse);
       
    89 	CleanupStack::PushL(self);
       
    90 	self->ConstructL(aMimeType, aMetaDataArray, KNullDesC(), KNullDesC());
       
    91 	CleanupStack::Pop(self);
       
    92 	return self;
       
    93 	}
       
    94 	
       
    95 // -----------------------------------------------------------------------------
       
    96 // COma2AgentImportFile::
       
    97 // 
       
    98 // -----------------------------------------------------------------------------
       
    99 //
       
   100 COma2AgentImportFile::COma2AgentImportFile(
       
   101     TBool aAgentCreatesOutputFiles):
       
   102     iOutputDirectory(NULL),
       
   103     iSuggestedFileName(NULL),
       
   104     iOutputFileName(NULL),
       
   105     iLastWriteData(NULL),
       
   106     iAgentCreatesOutputFiles(aAgentCreatesOutputFiles),
       
   107     iMessageParser(NULL),
       
   108     iDcfCreator(NULL),
       
   109     iDcfRights(NULL),
       
   110     iDcfMimeType(NULL)
       
   111 	{
       
   112 	iImportStatus = EInProgress;
       
   113 	}
       
   114 
       
   115 // -----------------------------------------------------------------------------
       
   116 // COma2AgentImportFile::
       
   117 // 
       
   118 // -----------------------------------------------------------------------------
       
   119 //
       
   120 COma2AgentImportFile::~COma2AgentImportFile()
       
   121 	{
       
   122 	if (iFileOpen)
       
   123 		{
       
   124 		iFile.Close();
       
   125 		iFileOpen = EFalse;
       
   126 		}
       
   127 	iFs.Close();
       
   128 	delete iOutputDirectory;
       
   129 	delete iSuggestedFileName;
       
   130     delete iOutputFileName;
       
   131 	delete iLastWriteData;
       
   132 	iOutputFiles.ResetAndDestroy();
       
   133 	iOutputFiles.Close();
       
   134 	delete iMessageParser;
       
   135 	delete iDcfCreator;
       
   136 	delete iDcfRights;
       
   137 	delete iDcfMimeType;
       
   138 	}
       
   139   
       
   140 // -----------------------------------------------------------------------------
       
   141 // COma2AgentImportFile::
       
   142 // 
       
   143 // -----------------------------------------------------------------------------
       
   144 //
       
   145 void COma2AgentImportFile::ConstructL(
       
   146     const TDesC8& aMimeType,
       
   147     const CMetaDataArray& aMetaDataArray,
       
   148     const TDesC& aOutputDirectory,
       
   149     const TDesC& aSuggestedFileName)
       
   150 	{
       
   151 	RMemReadStream rights;
       
   152 	
       
   153 	if (aMimeType.CompareF(KOma1DrmMessageContentType) == 0 ||
       
   154 	    aMimeType.CompareF(KOmaImportContentType) == 0)
       
   155 	    {
       
   156 	    if (aMimeType.CompareF(KOma1DrmMessageContentType) == 0)
       
   157 	        {
       
   158             iMessageParser = CDRMMessageParser::NewL();
       
   159 	        iImportType = EOma1DrmMessage;
       
   160 	        }
       
   161 	    else
       
   162 	        {
       
   163 	        iDcfCreator = COma1DcfCreator::NewL();
       
   164     	    iImportType = EPlainContent;
       
   165             const TDesC8& mimeDes = aMetaDataArray.SearchL(
       
   166                 KOmaImportMimeTypeField);
       
   167             if (mimeDes.Length() == 0)
       
   168                 {
       
   169                 User::Leave(KErrArgument);
       
   170                 }
       
   171     	    iDcfMimeType = mimeDes.AllocL();
       
   172     	    const TDesC8& rightsDes = aMetaDataArray.SearchL(
       
   173     	        KOmaImportRightsField);
       
   174             if (rightsDes.Length() > 0)
       
   175     	        {
       
   176     	        rights.Open(rightsDes.Ptr(), rightsDes.Length());
       
   177     	        CleanupClosePushL(rights);
       
   178         	    iDcfRights = CDRMRights::NewL();
       
   179     	        iDcfRights->InternalizeL(rights);
       
   180     	        CleanupStack::PopAndDestroy(); // rights
       
   181     	        }
       
   182 	        }
       
   183     	iOutputDirectory = aOutputDirectory.AllocL();
       
   184     	iSuggestedFileName = aSuggestedFileName.AllocL();
       
   185     	iOutputFileName = HBufC::NewL(iOutputDirectory->Des().Length() +
       
   186             iSuggestedFileName->Des().Length());
       
   187 	    iAgentCreatesOutputFiles = ETrue;
       
   188 	    }
       
   189     else if (aMimeType.CompareF(KOma1XmlRoContentType) == 0)
       
   190 	    {
       
   191 	    iImportType = EOma1XmlRo;
       
   192 	    }
       
   193     else if (aMimeType.CompareF(KOma1WbxmlRoContentType) == 0)
       
   194 	    {
       
   195 	    iImportType = EOma1WbxmlRo;
       
   196 	    }
       
   197     else if (aMimeType.CompareF(KOma2RoContentType) == 0)
       
   198 	    {
       
   199 	    iImportType = EOma2Ro;
       
   200 	    }
       
   201     else
       
   202         {
       
   203 		User::Leave(KErrCANotSupported);
       
   204 		}
       
   205 	User::LeaveIfError(iFs.Connect());
       
   206     User::LeaveIfError(iFs.ShareAuto());	
       
   207 	}
       
   208 
       
   209 // -----------------------------------------------------------------------------
       
   210 // COma2AgentImportFile::
       
   211 // 
       
   212 // -----------------------------------------------------------------------------
       
   213 //
       
   214 TInt COma2AgentImportFile::OpenOutputFile()
       
   215 	{
       
   216 	TInt r = KErrNone;
       
   217 	TPtr fileNamePtr = iOutputFileName->Des();
       
   218 	
       
   219 	fileNamePtr.Copy(*iOutputDirectory);
       
   220 	fileNamePtr.Append(*iSuggestedFileName);
       
   221 
       
   222 	r = iFile.Create(iFs, fileNamePtr, EFileShareAny  | EFileStream |
       
   223 	    EFileWrite);
       
   224 	if (r == KErrNone)
       
   225 		{
       
   226 		iFileOpen = ETrue;
       
   227 		if (iImportType == EOma1DrmMessage)
       
   228 		    {
       
   229 		    TRAP(r, iMessageParser->InitializeMessageParserL(iFile));
       
   230 		    }
       
   231 		else
       
   232 		    {
       
   233 		    TRAP(r, iDcfCreator->EncryptInitializeL(iFile, *iDcfMimeType,
       
   234 		        iDcfRights));
       
   235 		    }
       
   236 		}
       
   237     else
       
   238         {
       
   239         r = KErrCANewFileHandleRequired;
       
   240         }
       
   241     return r;
       
   242 	}
       
   243 
       
   244 // -----------------------------------------------------------------------------
       
   245 // COma2AgentImportFile::
       
   246 // 
       
   247 // -----------------------------------------------------------------------------
       
   248 //
       
   249 void COma2AgentImportFile::WriteDataL(const TDesC8& aData)
       
   250     {
       
   251 	CDrmRightsParser* oma1Parser = NULL;
       
   252 	CDrmProtectedRoParser* oma2Parser = NULL;
       
   253 	RPointerArray<CDRMRights> rights;
       
   254     TCleanupItem listCleanup(PointerArrayResetDestroyAndClose<CDRMRights>,
       
   255         &rights);
       
   256 	TInt r;
       
   257 	
       
   258     switch (iImportType)
       
   259         {
       
   260         case EOma1DrmMessage:
       
   261         	if (!iFileOpen)
       
   262         		{
       
   263         		r = OpenOutputFile();
       
   264         		if (r == KErrNone)
       
   265         		    {
       
   266                     iMessageParser->ProcessMessageDataL(aData);
       
   267         		    }
       
   268         		else
       
   269         		    {
       
   270                     iLastWriteData = aData.AllocL();
       
   271                     User::Leave(r);
       
   272         		    }
       
   273                 }
       
   274             else
       
   275                 {
       
   276                 iMessageParser->ProcessMessageDataL(aData);
       
   277                 }
       
   278             break;
       
   279         case EPlainContent:
       
   280         	if (!iFileOpen)
       
   281         		{
       
   282         		r = OpenOutputFile();
       
   283         		if (r == KErrNone)
       
   284         		    {
       
   285                     iDcfCreator->EncryptUpdateL(aData);
       
   286         		    }
       
   287         		else
       
   288         		    {
       
   289                     iLastWriteData = aData.AllocL();
       
   290                     User::Leave(r);
       
   291         		    }
       
   292                 }
       
   293             else
       
   294                 {
       
   295                 iDcfCreator->EncryptUpdateL(aData);
       
   296                 }
       
   297             break;
       
   298         case EOma1XmlRo:
       
   299         case EOma1WbxmlRo:
       
   300             oma1Parser = CDrmRightsParser::NewL();
       
   301             CleanupStack::PushL(oma1Parser);
       
   302             CleanupStack::PushL(listCleanup);
       
   303             oma1Parser->ParseAndStoreL(aData, rights);
       
   304             CleanupStack::PopAndDestroy(2); // listCleanup, oma1Parser
       
   305             break;
       
   306         case EOma2Ro:
       
   307             __UHEAP_MARK;
       
   308             oma2Parser = CDrmProtectedRoParser::NewL();
       
   309             CleanupStack::PushL(oma2Parser);
       
   310             CleanupStack::PushL(listCleanup);
       
   311             oma2Parser->ParseAndStoreL(aData, rights);
       
   312             CleanupStack::PopAndDestroy(2); // listCleanup, oma2Parser
       
   313             __UHEAP_MARKEND;
       
   314             break;
       
   315         }
       
   316     }
       
   317     
       
   318 // -----------------------------------------------------------------------------
       
   319 // COma2AgentImportFile::
       
   320 // 
       
   321 // -----------------------------------------------------------------------------
       
   322 //
       
   323 TInt COma2AgentImportFile::WriteData(const TDesC8& aData)
       
   324 	{
       
   325 	TInt r = KErrNone;
       
   326 	TRAP(r, WriteDataL(aData));
       
   327 	return r;
       
   328 	}
       
   329 
       
   330 // -----------------------------------------------------------------------------
       
   331 // COma2AgentImportFile::
       
   332 // 
       
   333 // -----------------------------------------------------------------------------
       
   334 //
       
   335 TInt COma2AgentImportFile::WriteDataComplete()
       
   336 	{
       
   337 	TInt r = KErrNone;
       
   338 	CSupplierOutputFile *temp = NULL;
       
   339 	TPtrC8 type(NULL, 0);
       
   340 	
       
   341 	if (iFileOpen)
       
   342 		{
       
   343 		if (iImportType == EOma1DrmMessage)
       
   344 		    {
       
   345 		    TRAP(r, iMessageParser->FinalizeMessageParserL());
       
   346 		    }
       
   347 		else
       
   348 		    {
       
   349 		    TRAP(r, iDcfCreator->EncryptFinalizeL());
       
   350 		    }
       
   351 		iFile.Close();
       
   352 		iFileOpen = EFalse;
       
   353 		}
       
   354 	if( r ) 
       
   355 	    {
       
   356 		  return r;
       
   357       }
       
   358 	if (iOutputFileName)
       
   359 		{
       
   360 		switch (iImportType)
       
   361 		    {
       
   362 		    case EPlainContent:
       
   363             case EOma1DrmMessage:
       
   364                 type.Set(KOma1DcfContentType());
       
   365                 break;
       
   366             default:
       
   367                 return KErrGeneral;;
       
   368 		    }
       
   369 		TRAP(r, temp = CSupplierOutputFile::NewL(iOutputFileName->Des(),
       
   370 		    EContent, type));
       
   371 		if (r == KErrNone)
       
   372 			{
       
   373 			r = iOutputFiles.Append(temp);
       
   374 			if (r != KErrNone)
       
   375 				{
       
   376 				delete temp;
       
   377 				}
       
   378 			}
       
   379 		}
       
   380 	return r;
       
   381 	}
       
   382 		
       
   383 // -----------------------------------------------------------------------------
       
   384 // COma2AgentImportFile::
       
   385 // 
       
   386 // -----------------------------------------------------------------------------
       
   387 //
       
   388 void COma2AgentImportFile::WriteData(
       
   389     const TDesC8& aData,
       
   390     TRequestStatus &aStatus)
       
   391 	{
       
   392 	TRequestStatus *ptr = &aStatus;
       
   393 	TInt r = WriteData(aData);
       
   394 	User::RequestComplete(ptr,r);
       
   395 	}
       
   396 
       
   397 // -----------------------------------------------------------------------------
       
   398 // COma2AgentImportFile::
       
   399 // 
       
   400 // -----------------------------------------------------------------------------
       
   401 //
       
   402 void COma2AgentImportFile::WriteDataComplete(
       
   403     TRequestStatus &aStatus)
       
   404 	{
       
   405 	TRequestStatus *ptr = &aStatus;
       
   406 	TInt r = WriteDataComplete();
       
   407 	User::RequestComplete(ptr,r);
       
   408 	}
       
   409 	
       
   410 // -----------------------------------------------------------------------------
       
   411 // COma2AgentImportFile::
       
   412 // 
       
   413 // -----------------------------------------------------------------------------
       
   414 //
       
   415 TInt COma2AgentImportFile::OutputFileCountL() const
       
   416 	{
       
   417 	return iOutputFiles.Count();
       
   418 	}
       
   419 
       
   420 // -----------------------------------------------------------------------------
       
   421 // COma2AgentImportFile::
       
   422 // 
       
   423 // -----------------------------------------------------------------------------
       
   424 //
       
   425 CSupplierOutputFile& COma2AgentImportFile::OutputFileL(
       
   426     TInt aIndex)
       
   427 	{
       
   428 	return *iOutputFiles[aIndex];
       
   429 	}
       
   430 
       
   431 // -----------------------------------------------------------------------------
       
   432 // COma2AgentImportFile::
       
   433 // 
       
   434 // -----------------------------------------------------------------------------
       
   435 //
       
   436 TImportStatus COma2AgentImportFile::GetImportStatus() const
       
   437 	{
       
   438 	return iImportStatus;
       
   439 	}	
       
   440 
       
   441 // -----------------------------------------------------------------------------
       
   442 // COma2AgentImportFile::
       
   443 // 
       
   444 // -----------------------------------------------------------------------------
       
   445 //
       
   446 TInt COma2AgentImportFile::GetSuggestedOutputFileExtension(
       
   447     TDes& aFileExtension)
       
   448 	{
       
   449 	aFileExtension.Copy(_L(".dcf"));
       
   450 	return KErrNone;
       
   451 	}
       
   452 
       
   453 // -----------------------------------------------------------------------------
       
   454 // COma2AgentImportFile::
       
   455 // 
       
   456 // -----------------------------------------------------------------------------
       
   457 //
       
   458 TInt COma2AgentImportFile::GetSuggestedOutputFileName(
       
   459     TDes& aFileName)
       
   460 	{
       
   461 	aFileName.Append(_L(".dcf"));
       
   462 	return KErrNone;
       
   463 	}
       
   464 
       
   465 // -----------------------------------------------------------------------------
       
   466 // COma2AgentImportFile::
       
   467 // 
       
   468 // -----------------------------------------------------------------------------
       
   469 //
       
   470 TInt COma2AgentImportFile::ContinueWithNewOutputFile(
       
   471     RFile& aFile,
       
   472     const TDesC& aFileName)
       
   473 	{
       
   474 	RFile file;
       
   475 	TInt r = KErrNone;
       
   476 
       
   477 	if (iOutputFileName != NULL)
       
   478 	    {
       
   479 	    delete iOutputFileName;
       
   480 	    iOutputFileName = NULL;
       
   481 	    }
       
   482 	TRAP(r, iOutputFileName = aFileName.AllocL());
       
   483 	if (r == KErrNone)
       
   484 	    {
       
   485 	    file.Duplicate(aFile);
       
   486         iFile.Attach(file);
       
   487         iFileOpen = ETrue;
       
   488 		if (iImportType == EOma1DrmMessage)
       
   489 		    {
       
   490 		    TRAP(r, iMessageParser->InitializeMessageParserL(iFile));
       
   491 		    }
       
   492 		else
       
   493 		    {
       
   494 		    TRAP(r, iDcfCreator->EncryptInitializeL(iFile, *iDcfMimeType,
       
   495 		        iDcfRights));
       
   496 		    }
       
   497         if (r == KErrNone)
       
   498             {
       
   499             r = WriteData(*iLastWriteData);
       
   500             }
       
   501         delete iLastWriteData;
       
   502         iLastWriteData = NULL;
       
   503 	    }
       
   504 	return r;
       
   505 	}
       
   506 
       
   507 // -----------------------------------------------------------------------------
       
   508 // COma2AgentImportFile::
       
   509 // 
       
   510 // -----------------------------------------------------------------------------
       
   511 //
       
   512 void COma2AgentImportFile::ContinueWithNewOutputFile(
       
   513     RFile& aFile,
       
   514     const TDesC& aFileName,
       
   515     TRequestStatus& aStatus)
       
   516 	{
       
   517 	TRequestStatus *ptr = &aStatus;
       
   518 	TInt r = ContinueWithNewOutputFile(aFile, aFileName);
       
   519 	User::RequestComplete(ptr,r);
       
   520 	}
       
   521 
       
   522 // -----------------------------------------------------------------------------
       
   523 // COma2AgentImportFile::
       
   524 // 
       
   525 // -----------------------------------------------------------------------------
       
   526 //
       
   527 void COma2AgentImportFile::NewMimePartL(
       
   528     const TDesC8& /*aMimeType*/,
       
   529     const CMetaDataArray& /*aImportMetaData*/)
       
   530 	{
       
   531 	User::Leave(KErrCANotSupported);
       
   532 	}
       
   533 
       
   534 // -----------------------------------------------------------------------------
       
   535 // COma2AgentImportFile::
       
   536 // 
       
   537 // -----------------------------------------------------------------------------
       
   538 //
       
   539 void COma2AgentImportFile::EndMimePartL()
       
   540 	{
       
   541 	User::Leave(KErrCANotSupported);
       
   542 	}
       
   543 
       
   544 // End of file