/*

* Copyright (c) 2002-2009 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"

* 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:

*

*/

// [INCLUDE FILES] - do not remove

#include <e32svr.h>

#include <StifParser.h>

#include <Stiftestinterface.h>

#include "DRM_CAF.h"

#include <caf/caftypes.h>

#include <caf/caf.h>

#include <caf/data.h>

#include <f32file.h>

#include <bautils.h>

#include <TestclassAssert.h>

using namespace ContentAccess;

// LOCAL CONSTANTS AND MACROS

#define NUM_ELEMENTS(x) (sizeof(x) / sizeof(x[0]))

const TInt KAttributes[] =

    {

    EIsProtected,

    EIsForwardable,

    EIsModifyable,

    EIsCopyable,

    ECanPlay,

    ECanPrint,

    ECanExecute,

    ECanView,

    ERightsNone,

    ERightsPending,

    ERightsHaveExpired,

    EPreviewAvailable,

    EContentCDataInUse,

    ECanRewind,

    ECopyPaste,

    ERightsConsumable,

    ERightsStateless,

    ECanMove,

    ECanRename,

    ECanAutomaticConsume,

    EFileType,

    EPreviewType,

    ESilentRightsType

    };

const TInt KStringAttributes[] =

    {

    EEmbeddedRightsObject,

    EDomainRightsIssuerUrl,

    EDescription,

    EMimeType,

    ETitle,

    EAuthor,

    EIconURI,

    EPreviewURI,

    EContentURI,

    EContentID,

    EInfoURL,

    EPendingRightsETA,

    EInstantPreviewUri,

    EPreviewRightsUri,

    ERightsIssuerUrl,

    ESilentRightsUrl,

    ETransactionTrackingId,

    };

// Test logging macros(we only want line numbers)

//#undef test

//#undef __test

//#define __test(x, l) test(x, l)

//#define test(x) __test(x, __LINE__)

//#define FuncCount( a ) ( sizeof( a ) / sizeof( TBatTestFunc ) )

#define RUNCHECKED(function) \

    { \

    TInt c = function; \

    if (c != KErrNone) \

        { \

        STIF_ASSERT_TRUE(c); \

        } \

    else \

        { \

        GLastOk = __LINE__;\

        } \

    }

#define RUNCHECKEDL(function) \

    { \

    TInt c = function; \

    if (c != KErrNone) \

        { \

        STIF_ASSERT_TRUE(c); \

        User::Leave(c); \

        } \

    else \

        { \

        GLastOk = __LINE__;\

          } \

    }

#define RUN_TRAPPED(r, function) \

    { \

    r = KErrNone; TRAP(r, function); \

    STIF_ASSERT_EQUALS( KErrNone, r )\

    }

//RTest test(_L("c:\\OMADRM2BC.log"));

//#if 0

//RTest test(_L("c:\\OMADRM2BC.log"));

//TInt GLastOk;

//TInt GAllocatedCount;

//TInt GAllocatedSize;

//TBool flvariant;

//#endif

_LIT8(KROSimple, "<o-ex:rights xmlns:o-ex = \"http://odrl.net/1.1/ODRL-EX\" \

xmlns:o-dd = \"http://odrl.net/1.1/ODRL-DD\" \

xmlns:ds = \"http://www.w3.org/2000/09/xmldsig#/\">\

<o-ex:context><o-dd:version>1.0</o-dd:version></o-ex:context>\

<o-ex:agreement><o-ex:asset><o-ex:context>\

<o-dd:uid>cid:9999</o-dd:uid></o-ex:context></o-ex:asset><o-ex:permission>\

<o-dd:display/><o-dd:play/><o-dd:execute/><o-dd:print/>\

</o-ex:permission></o-ex:agreement></o-ex:rights>");

_LIT8(KDRCSimple,

"\003\016\152\000\305\005\205\006\206\007\207\001\106\107\003\061\056\060\000\001\

\001\111\112\106\110\003\143\151\144\072\062\060\060\063\060\064\062\071\061\062\

\060\067\065\063\055\063\064\064\067\060\071\066\063\066\060\100\167\167\167\056\

\147\157\157\147\154\145\056\143\157\155\000\001\001\113\114\303\020\063\323\173\

\130\367\145\145\367\077\137\105\057\120\076\265\377\001\001\001\115\116\122\123\

\003\065\000\001\001\001\117\122\124\125\003\062\060\060\063\055\060\064\055\062\

\071\124\061\062\072\060\064\072\064\067\000\001\126\003\062\060\060\063\055\060\

\065\055\062\061\124\061\062\072\060\064\072\064\067\000\001\001\001\001\001\001\

\001");

_LIT8(KDMHead, "--boundary-1\r\nContent-Type: text/plain\r\n\

Content-Transfer-Encoding: 8bit\r\n\

\r\n");

_LIT8(KDMTail, "\r\n--boundary-1--");

_LIT8(KDMContent, "0123456789");

_LIT8(KDMSimple, "--boundary-1\r\nContent-Type: text/plain\r\n\

Content-Transfer-Encoding: 8bit\r\n\

\r\n\

012345678911234567892123456789312345678941234567895123456789\

612345678971234567898123456789912345678901234567891123456789\

212345678931234567894123456789512345678961234567897123456789\

812345678991234567890123456789112345678921234567893123456789\

412345678951234567896123456789712345678981234567899123456789\

\r\n--boundary-1--\r\n");

_LIT8(KDMCDCount, "--random78o6bP%[GB6b/8&/45&%*^'?vS\r\n\

Content-Type: application/vnd.oma.drm.rights+xml\r\n\

Content-Transfer-Encoding: binary\r\n\

\r\n\

<o-ex:rights\

   xmlns:o-ex=\"http://odrl.net/1.1/ODRL-EX\"\

   xmlns:o-dd=\"http://odrl.net/1.1/ODRL-DD\"\

   xmlns:ds=\"http://www.w3.org/2000/09/xmldsig#/\"\

>\

<o-ex:context><o-dd:version>1.0</o-dd:version>\

</o-ex:context><o-ex:agreement><o-ex:asset>\

<o-ex:context><o-dd:uid>cid:20031217123451-8428794033@www.nokia.com</o-dd:uid>\

</o-ex:context></o-ex:asset>\

<o-ex:permission>\

<o-dd:display>\

<o-ex:constraint>\

<o-dd:count>3</o-dd:count>\

</o-ex:constraint>\

</o-dd:display>\

<o-dd:display>\

</o-ex:permission>\

</o-ex:agreement>\

</o-ex:rights>\r\n\

\r\n\

--random78o6bP%[GB6b/8&/45&%*^'?vS\r\n\

Content-Type: text/plain\r\n\

Content-ID: <20031217123451-8428794033@www.nokia.com>\r\n\

Content-Transfer-Encoding: binary\r\n\

\r\n\

Check:\r\n\

DbCnt: 109ms r:0\r\n\

Fixes:\r\n\

\r\n\

--random78o6bP%[GB6b/8&/45&%*^'?vS--");

_LIT(KOma1Content, "c:\\content.dcf");

_LIT(KOma2Content, "c:\\content.odf");

_LIT(KOma2ContentNoEnc, "c:\\content-noenc.odf");

_LIT(KOma2ContentNoEncTruncated, "c:\\content-noenc-trunc.odf");

_LIT(KEncryptedDir, "c:\\");

_LIT(KTempDcfName, "test.dcf");

_LIT(KTempDcfPathName, "c:\\test.dcf");

_LIT(KFVariantFile,"c:\\DrmCAFVarFile.txt");

// ============================= LOCAL FUNCTIONS ===============================

NONSHARABLE_CLASS( CActiveWaiter ): public CActive

    {

public:

    static CActiveWaiter* NewL();

    void StartWait();

    TRequestStatus& GetStatus();

    ~CActiveWaiter();

protected:     // from CActive

    /**

    * part of active object framework, provide default implementation

    */

    void RunL();

    /**

    * part of active object framework, provide default implementation

    */

    void DoCancel();

    /**

    * part of active object framework, provide default implementation

    */

    TInt RunError(TInt aError);

private:

    CActiveWaiter();

    void ConstructL();

    CActiveSchedulerWait iWait;

    };

CActiveWaiter::CActiveWaiter() : CActive(EPriorityStandard)

    {

    }

CActiveWaiter::~CActiveWaiter()

    {

    Deque();

    if( iWait.IsStarted() )

        {

        iWait.AsyncStop();

        }

    }

CActiveWaiter* CActiveWaiter::NewL() {

    CActiveWaiter* self = new (ELeave) CActiveWaiter();

    return self;

}

/**

* part of active object framework, provide default implementation

*/

void CActiveWaiter::RunL() {

    iWait.AsyncStop();

}

/**

* part of active object framework, provide default implementation

*/

void CActiveWaiter::DoCancel(){

}

/**

* part of active object framework, provide default implementation

*/

TInt CActiveWaiter::RunError(TInt /*aError*/){

    return KErrNone;

}

void CActiveWaiter::StartWait() {

    if(!IsAdded())

        {

        CActiveScheduler::Add(this);

        }

    if(!IsActive())

        {

        SetActive();

        }

    iStatus = KRequestPending;

    iWait.Start();

}

TRequestStatus& CActiveWaiter::GetStatus()

    {

    return iStatus;

    }

LOCAL_C CRightsManager* GetOmaDrmRightsManagerL()

    {

    RArray<TAgent> agents;

    CRightsManager* rm = NULL;

    TInt i;

    TInt r;

    CManager* manager = CManager::NewLC();

    //STIF_ASSERT_NOT_NULL( manager );

    manager->ListAgentsL(agents);

    for (i = 0; rm == NULL && i < agents.Count(); i++)

        {

        if (agents[i].Name().Compare(KOmaDrm2AgentName) == 0)

            {

            TRAP(r, rm = manager->CreateRightsManagerL(agents[i]));

            }

        }

    agents.Close();

    CleanupStack::PopAndDestroy();

    return rm;

    }

// ============================ MEMBER FUNCTIONS ===============================

// -----------------------------------------------------------------------------

// CDRM_CAF::Delete

// Delete here all resources allocated and opened from test methods.

// Called from destructor.

// -----------------------------------------------------------------------------

//

void CDRM_CAF::Delete()

    {

    }

// -----------------------------------------------------------------------------

// CDRM_CAF::RunMethodL

// Run specified method. Contains also table of test mothods and their names.

// -----------------------------------------------------------------------------

//

TInt CDRM_CAF::RunMethodL(

    CStifItemParser& aItem )

    {

    	TStifFunctionInfo const KFunctions[] =

        {

        // Copy this line for every implemented function.

        // First string is the function name used in TestScripter script file.

        // Second is the actual implementation member function.

        ENTRY( "CSupplier_Oma1XmlRoL", CDRM_CAF::CSupplier_Oma1XmlRoL ),

        ENTRY( "CSupplier_Oma1WbxmlRoL", CDRM_CAF::CSupplier_Oma1WbxmlRoL ),

        ENTRY( "CData_NewL", CDRM_CAF::CData_NewL ),

        ENTRY( "CData_Read_DataSize_SeekL", CDRM_CAF::CData_Read_DataSize_SeekL ),

        ENTRY( "CData_Read2L", CDRM_CAF::CData_Read2L ),

        ENTRY( "CData_SetPropertyL", CDRM_CAF::CData_SetPropertyL ),

        ENTRY( "CData_EvaluateIntentL", CDRM_CAF::CData_EvaluateIntentL ),

        ENTRY( "CData_ExecuteIntentL", CDRM_CAF::CData_ExecuteIntentL ),

        ENTRY( "CData_GetAttributeL", CDRM_CAF::CData_GetAttributeL ),

        ENTRY( "CData_GetAttributeSetL", CDRM_CAF::CData_GetAttributeSetL ),

        ENTRY( "CData_GetStringAttributeL", CDRM_CAF::CData_GetStringAttributeL ),

        ENTRY( "CData_GetStringAttributeSetL", CDRM_CAF::CData_GetStringAttributeSetL ),

        ENTRY( "CContent_NewL", CDRM_CAF::CContent_NewL ),

        ENTRY( "CContent_OpenCloseContainerL", CDRM_CAF::CContent_OpenCloseContainerL ),

        ENTRY( "CContent_GetEmbeddedObjectsL", CDRM_CAF::CContent_GetEmbeddedObjectsL ),

        ENTRY( "CContent_SearchL", CDRM_CAF::CContent_SearchL ),

        ENTRY( "CContent_GetAttributeL", CDRM_CAF::CContent_GetAttributeL ),

        ENTRY( "CContent_GetAttributeSetL", CDRM_CAF::CContent_GetAttributeSetL ),

        ENTRY( "CContent_GetStringAttributeL", CDRM_CAF::CContent_GetStringAttributeL ),

        ENTRY( "CContent_GetStringAttributeSetL", CDRM_CAF::CContent_GetStringAttributeL ),

        ENTRY( "CContent_AgentSpecificCommandL", CDRM_CAF::CContent_AgentSpecificCommandL ),

        ENTRY( "CContent_NotifyStatusChangeL", CDRM_CAF::CContent_NotifyStatusChangeL ),

        ENTRY( "CContent_CancelNotifyStatusChangeL", CDRM_CAF::CContent_CancelNotifyStatusChangeL ),

        ENTRY( "CContent_RequestRightsL", CDRM_CAF::CContent_RequestRightsL ),

        ENTRY( "CContent_DisplayInfoL", CDRM_CAF::CContent_DisplayInfoL ),

        ENTRY( "CContent_SetPropertyL", CDRM_CAF::CContent_SetPropertyL ),

        ENTRY( "CContent_OpenContentL", CDRM_CAF::CContent_OpenContentL ),

        ENTRY( "CManager_FileOperationsL", CDRM_CAF::CManager_FileOperationsL ),

        ENTRY( "CManager_GetAttributeL", CDRM_CAF::CManager_GetAttributeL ),

        ENTRY( "CManager_GetAttributeSetL", CDRM_CAF::CManager_GetAttributeSetL ),

        ENTRY( "CManager_GetStringAttributeL", CDRM_CAF::CManager_GetStringAttributeL ),

        ENTRY( "CManager_GetStringAttributeSetL", CDRM_CAF::CManager_GetStringAttributeSetL ),

        ENTRY( "CManager_SetPropertyL", CDRM_CAF::CManager_SetPropertyL ),

        ENTRY( "CManager_DisplayInfoL", CDRM_CAF::CManager_DisplayInfoL ),

        ENTRY( "CManager_ListAgentsL", CDRM_CAF::CManager_ListAgentsL ),

        ENTRY( "CManager_AgentSpecificCommandL", CDRM_CAF::CManager_AgentSpecificCommandL ),

        ENTRY( "CManager_CreateRightsManagerL", CDRM_CAF::CManager_CreateRightsManagerL ),

        ENTRY( "CSupplier_IsImportSupportedL", CDRM_CAF::CSupplier_IsImportSupportedL ),

        ENTRY( "CSupplier_PrepareHTTPRequestHeadersL", CDRM_CAF::CSupplier_PrepareHTTPRequestHeadersL ),

        ENTRY( "CSupplier_SetOutputDirectory_ImportFileL", CDRM_CAF::CSupplier_SetOutputDirectory_ImportFileL ),

        ENTRY( "CImportFile_WriteData_WriteDataComplete_OutputFileCountL", CDRM_CAF::CImportFile_WriteData_WriteDataComplete_OutputFileCountL ),

        ENTRY( "CImportFile_OutputFileL_GetImportStatusL", CDRM_CAF::CImportFile_OutputFileL_GetImportStatusL ),

        ENTRY( "CImportFile_GetSuggestedOutputFileExtensionL", CDRM_CAF::CImportFile_GetSuggestedOutputFileExtensionL ),

        ENTRY( "CImportFile_GetSuggestedOutputFileNameL", CDRM_CAF::CImportFile_GetSuggestedOutputFileNameL ),

        ENTRY( "CImportFile_ContinueWithNewOutputFileL", CDRM_CAF::CImportFile_ContinueWithNewOutputFileL ),

        ENTRY( "CImportFile_ContentMimeTypeL", CDRM_CAF::CImportFile_ContentMimeTypeL ),

        ENTRY( "CRightsManager_ListRightsL", CDRM_CAF::CRightsManager_ListRightsL ),

        ENTRY( "CRightsManager_ListAllRightsL", CDRM_CAF::CRightsManager_ListAllRightsL ),

        ENTRY( "CRightsManager_ListContentL", CDRM_CAF::CRightsManager_ListContentL ),

        ENTRY( "CRightsManager_DeleteRightsObjectL", CDRM_CAF::CRightsManager_DeleteRightsObjectL ),

        ENTRY( "CRightsManager_DeleteAllRightsObjectsL", CDRM_CAF::CRightsManager_DeleteAllRightsObjectsL ),

        ENTRY( "CRightsManager_SetPropertyL", CDRM_CAF::CRightsManager_SetPropertyL ),

        ENTRY( "CDRMLicenseChecker_CheckLicenseL", CDRM_CAF::CDRMLicenseChecker_CheckLicenseL ),

        //ADD NEW ENTRY HERE

        // [test cases entries] - Do not remove

        };

    const TInt count = sizeof( KFunctions ) /

                        sizeof( TStifFunctionInfo );

    return RunInternalL( KFunctions, count, aItem );

    }

TInt CDRM_CAF::CSupplier_Oma1XmlRoL()

    {

    CSupplier* supplier = CSupplier::NewLC();

    CMetaDataArray* metaData = CMetaDataArray::NewLC();

    CImportFile* file = supplier->ImportFileL(KOma1XmlRoContentType,

        *metaData);

    STIF_ASSERT_NOT_NULL( file );

    CleanupStack::PushL(file);

    RUNCHECKEDL(file->WriteData(KROSimple));

    RUNCHECKEDL(file->WriteDataComplete());

    CleanupStack::PopAndDestroy(3);

    return KErrNone;

    }

TInt CDRM_CAF::CSupplier_Oma1WbxmlRoL()

    {

    CSupplier* supplier = CSupplier::NewLC();

    CMetaDataArray* metaData = CMetaDataArray::NewLC();

    CImportFile* file = supplier->ImportFileL(KOma1WbxmlRoContentType,

        *metaData);

    STIF_ASSERT_NOT_NULL( file );

    CleanupStack::PushL(file);

    RUNCHECKEDL(file->WriteData(KDRCSimple));

    RUNCHECKEDL(file->WriteDataComplete());

    CleanupStack::PopAndDestroy(3);

    return KErrNone;

    }

TInt CDRM_CAF::CData_NewL()

    {

    CData* data;

    RFile file;

    TVirtualPathPtr dcf1(KOma1Content, KDefaultContentObject);

    TVirtualPathPtr dcf2(KOma2Content, KDefaultContentObject);

    TVirtualPathPtr dcf3(KOma2ContentNoEnc, KDefaultContentObject);

    data = CData::NewL(dcf1, EPeek, EContentShareReadOnly);

    STIF_ASSERT_NOT_NULL( data );

    delete data;

    data = CData::NewL(dcf2, EPeek, EContentShareReadOnly);

    STIF_ASSERT_NOT_NULL( data );

    delete data;

    data = CData::NewL(dcf3, EPeek, EContentShareReadOnly);

    STIF_ASSERT_NOT_NULL( data );

    delete data;

    TInt err = file.Open(iFs, KOma1Content, EFileRead);

    CleanupClosePushL(file);

    data = CData::NewL(file, KDefaultContentObject, EPeek);

    STIF_ASSERT_NOT_NULL( data );

    delete data;

    CleanupStack::PopAndDestroy();

    err = file.Open(iFs, KOma2Content, EFileRead);

    CleanupClosePushL(file);

    data = CData::NewL(file, KDefaultContentObject, EPeek);

    STIF_ASSERT_NOT_NULL( data );

    delete data;

    CleanupStack::PopAndDestroy();

    err = file.Open(iFs, KOma2ContentNoEnc, EFileRead);

    CleanupClosePushL(file);

    data = CData::NewL(file, KDefaultContentObject, EPeek);

    STIF_ASSERT_NOT_NULL( data );