diff -r 6aeb7a756187 -r 3c88a81ff781 utilities/downloadmanager/src/drmstorage.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/utilities/downloadmanager/src/drmstorage.cpp	Fri Oct 15 17:30:59 2010 -0400
@@ -0,0 +1,240 @@
+/**
+   This file is part of CWRT package **
+
+   Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies). **
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU (Lesser) General Public License as 
+   published by the Free Software Foundation, version 2.1 of the License. 
+   This program is distributed in the hope that it will be useful, but
+   WITHOUT ANY WARRANTY; without even the implied warranty of 
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 
+   (Lesser) General Public License for more details. You should have 
+   received a copy of the GNU (Lesser) General Public License along 
+   with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "drmstorage.h"
+#include "storageutility.h"
+#include "dmcommoninternal.h"
+#include <QFile>
+#include <QDir>
+#include <QVariant>
+#include <qregexp.h>
+#include <f32file.h>
+
+
+DrmStorage::DrmStorage(const QString& clientName, ClientDownload* download)
+{
+    m_file = 0;
+    m_clientName = clientName;
+    m_download = download;
+    // create temporary path
+    m_tempPath = StorageUtility::createTemporaryPath(clientName);
+}
+
+DrmStorage::~DrmStorage()
+{
+    if (m_file) {
+        close();
+        delete m_file;
+        m_file = 0;
+    }
+}
+
+int DrmStorage::createStore()
+{
+    // validate filename
+    QString validFilename;
+    QString filename = m_download->attributes().value(DlFileName).toString();
+    StorageUtility::validateFileName(filename, validFilename);
+    // generate unique filename 
+    QString permPath = m_download->attributes().value(DlDestPath).toString();
+    QString uniqueFilename;
+    StorageUtility::generateUniqueFileName(validFilename, m_tempPath, permPath, uniqueFilename);  
+    
+    // set the new filename
+    m_download->attributes().insert(DlFileName, uniqueFilename);
+    
+    // convert to symbian descriptors
+    QString path = m_tempPath;
+    path.replace(QString("/"), QString("\\"));
+    TPtrC ptrPath = SymbianStringToQtString(path);
+    TPtrC ptrFileName = SymbianStringToQtString(uniqueFilename);
+    _LIT8(KOmaDrm1DrmMessageContentType, OMA_DRM_MESSAGE_CONTENT_TYPE);
+    
+    // create supplier and get importfile which will actually write to the file
+    QT_TRAP_THROWING(ContentAccess::CSupplier* supplier = ContentAccess::CSupplier::NewLC();
+                     ContentAccess::CMetaDataArray* metaData = ContentAccess::CMetaDataArray::NewLC();
+                     supplier->SetOutputDirectoryL(ptrPath);
+                     m_file = supplier->ImportFileL(KOmaDrm1DrmMessageContentType(), *metaData, ptrFileName));
+
+    // perform cleanup
+    CleanupStack::PopAndDestroy(2); // metaData, supplier
+    return 0;
+}
+
+
+int DrmStorage::open(QIODevice::OpenMode /*mode*/)
+{
+    // we cannot open an existing file, so we delete the
+    // store and create a new one. Download will start/resume
+    // from beginning
+    deleteStore();
+    if (!m_file)
+        createStore();
+    return 0;
+}
+
+int DrmStorage::write(const QByteArray& data, bool lastChunk)
+{
+    if (!m_file)
+        return -1;
+    
+    // convert the data chunk to 8 bit descriptor
+    QString strChunk(data);
+    TPtrC ptrChunk = SymbianStringToQtString(strChunk);
+    HBufC8* buf8 = q_check_ptr(HBufC8::New(ptrChunk.Length()));
+    buf8->Des().Copy(ptrChunk);
+    TPtrC8 ptrData8(*buf8);
+    
+    if (m_file) {
+        // write the data chunk
+        m_file->WriteData(ptrData8);
+        if (lastChunk) {
+            // close the file
+            close();
+            // move the file to permanent destination path
+            // note that moveDlFile uses m_file
+            moveDlFile();
+            // clean up memory
+            delete m_file;
+            m_file = 0;
+        }
+    }
+    
+    delete buf8;
+    buf8 = NULL;
+    return 0;
+}
+
+int DrmStorage::close()
+{
+    if (m_file) {    
+        // close storage and get filename
+        m_file->WriteDataComplete();
+    }
+    return 0;
+}
+
+int DrmStorage::deleteStore()
+{
+    // create filename (with path)
+    QDir tempFilePath(m_tempPath);
+    QFileInfo tempFileInfo(tempFilePath, m_download->attributes().value(DlFileName).toString());
+    QString tempFileName = tempFileInfo.filePath();
+    
+    // remove the temporary file
+    if (QFile::exists(tempFileName)) {
+        // close the store
+        close();
+        if (m_file) {
+            delete m_file;
+            m_file = 0;
+        }
+        // remove file
+        QFile::remove(tempFileName);
+    }
+
+    return 0;
+}
+
+int DrmStorage::storedDataSize()
+{
+    // build temporary path with filename
+    QDir tempFilePath(m_tempPath);
+    QFileInfo tempFileinfo(tempFilePath, m_download->attributes().value(DlFileName).toString());
+    QString tempFilename = tempFileinfo.filePath();
+    QFile tempFile(tempFilename);
+
+    // return file size if file exists
+    if(tempFile.exists())
+        return tempFile.size();
+    else
+        return 0;
+}
+
+void DrmStorage::moveDlFile()
+{
+    // source file
+    QString fileName = m_download->attributes().value(DlFileName).toString();
+    QFileInfo srcFileInfo(m_tempPath, fileName);
+    QString srcFile = srcFileInfo.filePath();
+    
+    // if source file does not exist, return
+    if (!QFile::exists(srcFile))
+        return;
+
+    // create the permanent directory
+    QString permPath = m_download->attributes().value(DlDestPath).toString();
+    QDir permDir(permPath);
+    if(!permDir.exists()) {
+        if(!permDir.mkpath(permPath))
+            return;
+    }
+
+    if (m_file) {
+        // name of the file to which 
+        TPtrC outputFileName(KNullDesC);
+        QT_TRAP_THROWING(outputFileName.Set(m_file->OutputFileL(0).FileName()));
+        // convert filename to Qt equivalent
+        srcFile = SymbianStringToQtString(outputFileName);
+        // check if file exists and update filename attribute
+        if (!QFile::exists(srcFile))
+            return;
+        
+        // name without extension
+        TParse p;
+        p.SetNoWild(outputFileName, NULL, NULL);
+        TPtrC namePtr = p.Name();
+        // suggested extension
+        TBuf<4> extension;
+        m_file->GetSuggestedOutputFileExtension(extension);
+    
+        // build filename with new extension and convert to Qt equivalent 
+        HBufC* name = q_check_ptr(HBufC::New(namePtr.Length() + extension.Length()));
+        TPtr newNamePtr(name->Des());
+        newNamePtr.Format(_L("%S%S"), &namePtr, &extension);
+        QString newFileName = SymbianStringToQtString(newNamePtr); 
+        delete name;
+        name = NULL;
+    
+        // extension has changed, generate unique filename
+        StorageUtility::generateUniqueFileName(newFileName, m_tempPath, permPath, fileName);
+    }
+    
+    // remove the file if it already exists
+    QFileInfo fileInfo(permPath, fileName);
+    QString destFile = fileInfo.filePath();
+    if (QFile::exists(destFile))
+        QFile::remove(destFile);
+    
+    // Copy the file to destination and remove old file
+    QFile::copy(srcFile, destFile);
+    QFile::remove(srcFile);
+    
+    // set the new filename
+    m_download->attributes().insert(DlFileName, fileName);
+}
+
+QString DrmStorage::SymbianStringToQtString(TDesC& symbianString)
+{
+    // convert symbian string to Qt string
+    return QString::fromUtf16(symbianString.Ptr(), symbianString.Length());
+}
+
+TPtrC DrmStorage::SymbianStringToQtString(QString& qtString)
+{
+    // convert QString string to symbian string
+    return reinterpret_cast<const TText*> (qtString.constData());
+}