--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/devencdiskutils/DevEncCommonUtils/src/DevEncSession.cpp	Tue Jan 26 15:20:08 2010 +0200
@@ -0,0 +1,288 @@
+/*
+* Copyright (c) 2007 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:
+*
+*/
+
+
+// Class include
+#include <DevEncSession.h>
+
+// User includes
+//#include "DevEncDiskUtils.h"
+#include "DevEncLog.h"
+#include "DevEncProtectedPSKey.h"
+
+// System includes
+#include <DevEncEngineBase.h>
+#include <DevEncEngineConstants.h>
+#include <e32property.h>
+
+#include <centralrepository.h>
+#include <DevEncExternalCRKeys.h>
+
+EXPORT_C CDevEncSessionBase* NewSessionL()
+    {
+    return new (ELeave) CDevEncSession;
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::CDevEncSession()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C CDevEncSession::CDevEncSession():
+iDriveNumber( EDriveC )
+	{
+	
+	}
+	
+EXPORT_C CDevEncSession::CDevEncSession( TDriveNumber aNumber ):
+    iConnected (EFalse), iDriveNumber( aNumber ), iDevEncEngine (NULL)
+    {
+    // Derived from CBase -> members zeroed.
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::~CDevEncSession()
+// 
+// --------------------------------------------------------------------------
+CDevEncSession::~CDevEncSession()
+    {
+    if (iConnected)
+        UnloadDevEncEngine();
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::Close()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C void CDevEncSession::Close()
+    {
+    UnloadDevEncEngine();
+    iConnected = EFalse;
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::Connect()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C TInt CDevEncSession::Connect()
+    {
+    TRAPD(err,LoadDevEncEngineL());
+    return err;
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::StartDiskEncrypt()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C TInt CDevEncSession::StartDiskEncrypt()
+    {
+    FLOG("CDevEncSession::StartDiskEncrypt >>");
+     
+     ASSERT( iConnected );
+
+     TInt err = KErrNone;
+     TInt diskStatus = 0;
+
+     err = DiskStatus( diskStatus );
+
+     if( err )
+         {
+         DFLOG2( "ERROR: Disk status query failed, err %d", err);
+         }
+     else if( EDecrypted == diskStatus )
+         {
+         DFLOG( "Starting encryption" );
+         err = iDevEncEngine->StartDiskOperation( EEncryptDisk );
+         }
+     else
+         {
+         DFLOG2( "Encryption not possible, disk status %d", diskStatus);
+         err = KErrNotReady;
+         }
+     FLOG2("CDevEncSession::StartDiskEncrypt, err = %d <<", err);
+     return err;
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::StartDiskDecrypt()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C TInt CDevEncSession::StartDiskDecrypt()
+    {
+    ASSERT( iConnected );
+
+    TInt err = KErrNone;
+    TInt diskStatus = 0;
+
+    err = DiskStatus( diskStatus );
+
+    if( err )
+        {
+        DFLOG2( "ERROR: Disk status query failed, err %d", err);
+        }
+    else if( EEncrypted == diskStatus )
+        {
+        DFLOG( "Starting decryption" );
+        err = iDevEncEngine->StartDiskOperation( EDecryptDisk );
+        }
+    else
+        {
+        DFLOG2( "Encryption not possible, disk status %d", diskStatus);
+        err = KErrNotReady;
+        }
+
+    return err;
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::StartDiskWipe()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C TInt CDevEncSession::StartDiskWipe()
+    {
+    ASSERT( iConnected );
+
+    DFLOG( "Starting wiping" );
+    TInt err = iDevEncEngine->StartDiskOperation( EWipeDisk );
+
+    return err;
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::DiskStatus()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C TInt CDevEncSession::DiskStatus( TInt& aStatus ) const
+    {
+    ASSERT( iConnected );
+    TInt err = iDevEncEngine->DiskStatus(aStatus);
+    return err;
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::Progress()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C TInt CDevEncSession::Progress( TInt& aProgress ) const
+    {
+    ASSERT( iConnected );
+    // Progress status must be polled, because after the extension has stopped
+    // updating the status there will be no new progress updates --> will wait
+    // indefinitely.
+    TInt err = iDevEncEngine->Progress(aProgress);
+
+    return err;
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::Connected()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C TBool CDevEncSession::Connected() const
+    {
+    return iConnected;
+    }
+
+// --------------------------------------------------------------------------
+// CDevEncSession::DriveNumber()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C TDriveNumber CDevEncSession::DriveNumber() const
+    {
+    return iDriveNumber;
+    }
+	
+// --------------------------------------------------------------------------
+// CDevEncSession::SetDrive()
+// 
+// --------------------------------------------------------------------------
+EXPORT_C void CDevEncSession::SetDrive( TDriveNumber aNumber )
+    {
+    iDriveNumber = aNumber;
+    }
+
+void CDevEncSession::LoadDevEncEngineL()
+    {
+    FLOG(" CDevEncSession::LoadDevEncEngineL >> ");
+    
+    if (!iDevEncEngine)
+        {
+         TInt err = iLibrary.Load(KEncryptionDll);
+         if (err != KErrNone)
+             {
+             FLOG2("Error in finding the library... %d", err);
+             if (err == KErrNotFound)
+                 err = KErrNotSupported;
+             User::Leave(err);
+             }
+         TLibraryFunction entry = iLibrary.Lookup(1);
+         
+         if (!entry)
+             {
+             FLOG("Error in loading the library...");
+             User::Leave(KErrBadLibraryEntryPoint);
+             }
+         iDevEncEngine = (CDevEncEngineBase *) entry();
+         TInt errconnection = iDevEncEngine->Connect(iDriveNumber);
+         if( errconnection != KErrNone )
+        	 {
+             User::Leave(errconnection);
+        	 }
+         iConnected = ETrue;
+        }
+    FLOG(" CDevEncSession::LoadDevEncEngineL << ");
+    }
+
+void CDevEncSession::UnloadDevEncEngine()
+    {
+    FLOG(" CDevEncSession::UnloadDevEncEngineL >> ");
+    
+    if (iDevEncEngine)
+        {
+        iDevEncEngine->Close();
+        delete iDevEncEngine;
+        iDevEncEngine = NULL;
+        iLibrary.Close();
+        }
+    iConnected = EFalse;
+    
+    FLOG(" CDevEncSession::UnloadDevEncEngineL << ");
+    }
+
+EXPORT_C TInt CDevEncSession::SetDevEncControlL(TInt aValue)
+    {
+    DFLOG2("CDevEncSession SetDevEncControl, value = %d >>", aValue);
+    
+    CRepository* rep = CRepository::NewLC(TUid::Uid(KCRDevEncUiSettings));
+    TInt err = rep->Set(KDevEncUiDmControl, aValue);
+    CleanupStack::PopAndDestroy(rep);
+    
+    DFLOG2("CDevEncSession SetDevEncControl, err = %d <<", err);
+    return err;
+    }
+
+EXPORT_C TInt CDevEncSession::GetDevEncControlL(TInt& aValue)
+    {
+    DFLOG("CDevEncSession GetDevEncControl >>");
+    
+    CRepository* rep = CRepository::NewLC(TUid::Uid(KCRDevEncUiSettings));
+    TInt err = rep->Get(KDevEncUiDmControl, aValue);
+    CleanupStack::PopAndDestroy(rep);
+    
+    DFLOG2("CDevEncSession GetDevEncControl, err = %d <<", err);
+    return err;
+    }
+// End of file