/*

* Copyright (c) 2003-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:  DRM Encryption tool.

*

*/

#include "drmencryptor.h"

//#include "ui_dialog.h"

#include <QMessageBox>

#include <QDoubleValidator>

#include <QMenu>

#include <QHBoxLayout>

#include <QString>

#include <QLabel>

#include <QPlainTextEdit>

#include <QVBoxLayout>

#include <QWidget>

#include <QPalette>

#ifdef RD_MULTIPLE_DRIVE

#include <driveinfo.h>

#endif

#include <e32std.h>

#include <e32base.h>

#include <e32cmn.h>

#include <Oma1DcfCreator.h>

#include <s32file.h>

#include <DRMMessageParser.h>

#include <DcfRep.h>

#include <DcfEntry.h>

#include "drmserviceapi.h"

const TInt KBufferSize = 20000;

DRMEncryptor::DRMEncryptor()

{   

    QWidget *widget = new QWidget();

    setCentralWidget(widget);

    setBackgroundRole( QPalette::Window );

    QVBoxLayout *layout = new QVBoxLayout;

    layout->setAlignment(Qt::AlignCenter);

    QLabel *mainWindowLabel = new QLabel("DRM Encryptor");

    layout->addWidget(mainWindowLabel);

    //layout->addStretch();

    //ui->setupUi(this);

    //iOptionsMenu = menuBar()->addMenu(tr("&Options"));

    // Create menu options

    iSetDRMClockAct = new QAction(tr("&Set DRM Clock"), this);

    iSetDRMClockAct->setStatusTip(tr("Set DRM secure clock time."));

    iGetDRMClockAct = new QAction(tr("&Get DRM Clock Time"), this);

    iGetDRMClockAct->setStatusTip(tr("Get DRM secure clock time."));

    iEncryptAct = new QAction(tr("&Encrypt"), this);

    iEncryptAct->setStatusTip(tr("Encrypt"));

    iDeleteWMDRMDBAct = new QAction(tr("&Delete WMDRM DB"), this);

    iDeleteWMDRMDBAct->setStatusTip(tr("Delete WMDRM Database rights."));

    menuBar()->addAction(iSetDRMClockAct);

    menuBar()->addAction(iGetDRMClockAct);

    menuBar()->addAction(iEncryptAct);

    menuBar()->addAction(iDeleteWMDRMDBAct);

    connect(iSetDRMClockAct, SIGNAL(triggered()), this, SLOT(setDRMClock()));

    connect(iGetDRMClockAct, SIGNAL(triggered()), this, SLOT(getDRMClock()));

    connect(iEncryptAct, SIGNAL(triggered()), this, SLOT(startEncrypt()));

    connect(iDeleteWMDRMDBAct, SIGNAL(triggered()), this, SLOT(deleteWmDrmDB()));

    //setDRMClockButton = new QPushButton("Set DRM Clock", this);

    //connect(setDRMClockButton, SIGNAL(clicked()), this, SLOT(setDRMClock()));

    //setDRMClockButton->show();

/*  ui->pushButton->setEnabled(false);

    ui->lineEdit->setValidator(new QDoubleValidator(this));

    ui->lineEdit_2->setValidator(new QDoubleValidator(this));

*/

    //connect(ui->mainMenu,SIGNAL(onChanged()),this,SLOT(onLaske()));

    //connect(ui->lineEdit,SIGNAL(textChanged(const QString &)),this,SLOT(onTextChanged(const QString &)));

    //connect(ui->mainMenu,SIGNAL(textChanged(const QString &)),this,SLOT(onTextChanged(const QString &)));

    widget->setLayout(layout);

}

DRMEncryptor::~DRMEncryptor()

{

    //delete ui;

}

void DRMEncryptor::setDRMClock()

{

    DateTimeDialog *dateTimeDialog = new DateTimeDialog();

    dateTimeDialog->show();

}

void DRMEncryptor::getDRMClock()

{

    TInt timeZone;

    TDateTime date;

    DRMClock::ESecurityLevel level;

    TInt error( KErrNone );

    TBuf< 80 > buf;

    error = service->GetSecureTime( time, timeZone, level );

    date = time.DateTime();

    if(level == DRMClock::KSecure)

    {

        _LIT(KFormatTxt,"DRMClock Time:\n%d/%d/%d\n%d:%d:%d\nNitz available");

        buf.Format( KFormatTxt,

                date.Day()+1,

                TInt(date.Month()+1),

                date.Year(),

                date.Hour(),

                date.Minute(),

                date.Second());

    }

    else

    {

        _LIT(KFormatTxt,"DRMClock Time:\n%d/%d/%d\n%d:%d:%d\nNitz unavailable");

        buf.Format( KFormatTxt,

                date.Day()+1,

                TInt(date.Month()+1),

                date.Year(),

                date.Hour(),

                date.Minute(),

                date.Second());

    }

    QGridLayout *layout = new QGridLayout;

    layout->setAlignment(Qt::AlignCenter);

    setLayout(layout);

    QString datetimeString((QChar*)buf.Ptr(),buf.Length());

    QMessageBox *drmClockTime = new QMessageBox(this);

    drmClockTime->setWindowTitle(tr("DRM Clock"));

    drmClockTime->setText(datetimeString);

    layout->addWidget(drmClockTime);

    drmClockTime->show();

}

LOCAL_C void ReadFileL(HBufC8*& aContent, const TDesC& aName, RFs& aFs)

    {

    TInt size = 0;

    RFile file;

    User::LeaveIfError(file.Open(aFs, aName, EFileRead));

    User::LeaveIfError(file.Size(size));

    aContent = HBufC8::NewLC(size);

    TPtr8 ptr(aContent->Des());

    User::LeaveIfError(file.Read(ptr, size));

    CleanupStack::Pop(); //aContent

    }

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

// UpdateDCFRepositoryL()

// Update saved file to DCFRepository

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

//

LOCAL_C void UpdateDCFRepositoryL( const TDesC& aFileName)

    {

    CDcfEntry* dcf( NULL );

    CDcfRep* dcfRep( NULL );

    dcf = CDcfEntry::NewL();

    CleanupStack::PushL( dcf );

    dcfRep = CDcfRep::NewL();

    CleanupStack::PushL( dcfRep );

    dcf->SetLocationL( aFileName, 0 );

    dcfRep->UpdateL( dcf );

    CleanupStack::PopAndDestroy(2); // dcf, dcfRep

    }

void ProcessRightsL(const TDesC& aFile)

    {

    CDRMMessageParser* c = NULL;

    HBufC8* d = NULL;

    RFs fs;

    RPointerArray<CDRMRights> rights;

    User::LeaveIfError(fs.Connect());

    c = CDRMMessageParser::NewL();

    ReadFileL(d, aFile, fs);

    c->ProcessRightsObject(*d, rights);

    rights.ResetAndDestroy();

    delete d;

    delete c;

    fs.Close();

    }

void DRMEncryptor::ProcessMessageL(const TDesC& aFile, const TDesC& aOutput)

    {

    CDRMMessageParser* c = NULL;

    HBufC8* d = NULL;

    RFs fs;

    TPtr8 inRead(NULL, 0);

    TInt error = 1;

    __UHEAP_MARK;

    User::LeaveIfError(fs.Connect());

    CleanupClosePushL(fs);

    c = CDRMMessageParser::NewL();

    CleanupStack::PushL(c);

    d = HBufC8::NewLC( KBufferSize );

    RFile input;

    User::LeaveIfError(input.Open( fs, aFile, EFileRead ));

    CleanupClosePushL( input );

    RFileWriteStream output;

    output.Replace( fs, aOutput, EFileWrite );

    CleanupClosePushL( output );

    c->InitializeMessageParserL( output );

    while( error )

        {

        inRead.Set( const_cast<TUint8*>(d->Ptr()),0,KBufferSize);

        error = input.Read( inRead );

        if( error )

            {

            c->FinalizeMessageParserL();

            User::Leave( error );

            }

        else

            {

            error = inRead.Length();

            }

        if( error )

            {

            c->ProcessMessageDataL(inRead);

            }

        }

    c->FinalizeMessageParserL();

    CleanupStack::PopAndDestroy( 5 ); // fs, c, d, input, output

    UpdateDCFRepositoryL( aOutput );

    __UHEAP_MARKEND;

    }

void EncryptFileL(const TDesC& aFile, TDesC& aOutput, TInt aMultiplier)

    {

    COma1DcfCreator* c = NULL;

    CDRMRights* rights = NULL;

    TBuf8<64> mime;

    RFs fs;

    TFileName aDcfFile;

    TInt aOriginalFileNameLength(aOutput.Length() - 4);

    User::LeaveIfError(fs.Connect());

    if (aFile.Right(3).CompareF(_L("amr")) == 0) //AMR

        {

        mime.Copy(_L8("audio/amr"));

        }

    else if (aFile.Right(3).CompareF(_L("awb")) == 0) //AMR-AWB

        {

        mime.Copy(_L8("audio/amr-wb"));

        }

    else if (aFile.Right(3).CompareF(_L("mp3")) == 0) //MP3

        {

        mime.Copy(_L8("audio/mpeg"));

        }

    else if (aFile.Right(3).CompareF(_L("mp4")) == 0) //MP4

        {

        mime.Copy(_L8("audio/mp4"));

        }

    else if (aFile.Right(3).CompareF(_L("m4a")) == 0) //M4A

        {

        mime.Copy(_L8("audio/mp4"));

        }

    else if (aFile.Right(3).CompareF(_L("3gp")) == 0) //3GPP

        {

        mime.Copy(_L8("audio/3gpp"));

        }

    else if (aFile.Right(3).CompareF(_L("3g2")) == 0) //3GPP2

        {

        mime.Copy(_L8("audio/3gpp2"));

        }

    else if (aFile.Right(3).CompareF(_L("aac")) == 0) //AAC

        {

        mime.Copy(_L8("audio/aac"));

        }

    else if (aFile.Right(3).CompareF(_L("mid")) == 0) //MIDI

        {

        mime.Copy(_L8("audio/midi"));

        }

    else if (aFile.Right(5).CompareF(_L(".spmid")) == 0) //SP-MIDI

        {

        mime.Copy(_L8("audio/sp-midi"));

        }

    else if (aFile.Right(3).CompareF(_L("rmf")) == 0) //RMF

        {

        mime.Copy(_L8("audio/rmf"));

        }

    else if (aFile.Right(4).CompareF(_L("mxmf")) == 0) //Mobile-XMF

        {

        mime.Copy(_L8("audio/mobile-xmf"));

        }

    else if (aFile.Right(3).CompareF(_L("wav")) == 0) //WAV

        {

        mime.Copy(_L8("audio/x-wav"));

        }

    else if (aFile.Right(3).CompareF(_L("gif")) == 0) // GIF

        {

        mime.Copy(_L8("image/gif"));

        }

    else if (aFile.Right(3).CompareF(_L("jpg")) == 0) // JPEG

        {

        mime.Copy(_L8("image/jpeg"));

        }

    else if (aFile.Right(3).CompareF(_L("txt")) == 0) // text

        {

        mime.Copy(_L8("text/plain"));

        }

    else if (aFile.Right(3).CompareF(_L("pip")) == 0) // PIP

        {

        mime.Copy(_L8("application/x-pip"));

        }

    aDcfFile.Append(aOutput);

    for(TInt i = 0; i < aMultiplier ; ++i)

        {

        aDcfFile.Delete(aOriginalFileNameLength, aDcfFile.Length());

        aDcfFile.Append(_L("-"));

        aDcfFile.AppendNum(i);

        aDcfFile.Append(_L(".dcf"));

        c = COma1DcfCreator::NewL();

        CleanupStack::PushL(c);

        fs.Delete(aOutput);

        c->EncryptFileL(aFile, aDcfFile, mime, rights);

        UpdateDCFRepositoryL( aDcfFile );

        delete rights;

        CleanupStack::PopAndDestroy(); // c

        }

    fs.Close();

    }

void DRMEncryptor::startEncrypt()

    {

        TUint result = 0;

        TUint aEncryptedCount = 0;

        TUint aRightsCount = 0;

        TUint aMessagesProcessed = 0;

        TBuf<128> buffer;

        result = EncryptL(aEncryptedCount, aRightsCount, aMessagesProcessed);

        buffer.Append(_L("Encryption done\n"));

        if(result < 1) 

        {

            buffer.Append(_L("No files found"));

        }

        if(aMessagesProcessed != 0) 

        {

            buffer.AppendNum(aMessagesProcessed);

            buffer.Append(_L(" messages processed\n"));

        }

        if(aRightsCount != 0) 

        {

            buffer.AppendNum(aRightsCount);

            buffer.Append(_L(" rights objects processed\n"));

        }

        if(aMessagesProcessed != 0) 

        {

            buffer.AppendNum(aEncryptedCount);

            buffer.Append(_L(" files encrypted\n"));

        }

        QGridLayout *layout = new QGridLayout;

        layout->setAlignment(Qt::AlignCenter);

        setLayout(layout);

        QString encryptString((QChar*)buffer.Ptr(),buffer.Length());

        QMessageBox *encrpytMessage = new QMessageBox(this);

        encrpytMessage->setWindowTitle(tr("Encrypt"));

        encrpytMessage->setText(encryptString);

        layout->addWidget(encrpytMessage);

        encrpytMessage->show();

    }

TUint DRMEncryptor::EncryptL(TUint& aEncryptedCount, TUint& aRightsCount, TUint& aMessagesProcessed)

    {

    TInt i;

    CDir* files;

    TFileName input;

    TFileName output;

    TUint inputNameSize = 0;

    TUint outputNameSize = 0;

    RFs fs;

    User::LeaveIfError(fs.Connect());

    TInt aMultiplier(1);

#ifdef __WINS__

    input.Append(_L("c:\\data\\DRM\\"));

    output.Append(_L("c:\\data\\Others\\"));

#else

#ifndef RD_MULTIPLE_DRIVE

    input.Append(_L("e:\\DRM\\"));

    output.Append(_L("e:\\Others\\"));

#else //RD_MULTIPLE_DRIVE

    TInt driveNumber( -1 );

    TChar driveLetter;

    DriveInfo::GetDefaultDrive( DriveInfo::EDefaultMassStorage, driveNumber );

    fs.DriveToChar( driveNumber, driveLetter );

    _LIT( KdrmDir, "%c:\\DRM\\" );

    input.Format( KdrmDir, (TUint)driveLetter );

    _LIT( KothersDir, "%c:\\Others\\" );

    output.Format( KothersDir, (TUint)driveLetter );

#endif

#endif

    inputNameSize = input.Length();

    outputNameSize = output.Length();

    fs.MkDir(input);

    fs.MkDir(output);

    fs.GetDir(input, KEntryAttNormal, ESortNone, files);

    for (i = 0; i < files->Count(); i++)

        {

        input.Append((*files)[i].iName);

        output.Append((*files)[i].iName);

        if (input.Right(2).CompareF(_L("dm")) == 0)

            {

            for(TInt ii = 0; ii < aMultiplier ; ++ii)

                {

                output.Delete(outputNameSize +(*files)[i].iName.Length() , output.Length()-1);

                output.Append(_L("-"));

                output.AppendNum(ii);

                output.Append(_L(".dcf"));

                ProcessMessageL(input, output);

                ++aMessagesProcessed;

                }

            }

        else if (input.Right(3).CompareF(_L("oro")) == 0 ||

                input.Right(3).CompareF(_L("drc")) == 0 ||

                input.Right(2).CompareF(_L("ro")) == 0 ||

                input.Right(2).CompareF(_L("dr")) == 0 )

            {

            for (TInt iii = 0; iii < aMultiplier; ++iii)

                {

                ProcessRightsL(input);

                ++aRightsCount;

                }