dosservices/tsrc/public/basic/dosservertestplugin/src/Tfdoscontrol.cpp
author Dremov Kirill (Nokia-D-MSW/Tampere) <kirill.dremov@nokia.com>
Sat, 20 Feb 2010 00:05:00 +0200
branchRCL_3
changeset 3 a811597961f0
parent 0 4e1aa6a622a0
permissions -rw-r--r--
Revision: 201003 Kit: 201007

/*
* Copyright (c) 2004 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
#include <ctffactory.h>
#include <tftypes.h>
#include <psvariables.h>
#include "siftrace.h"
#include "dosenums.h"
#include "ctfdoscontrol.h"
#include "ctfdoscontroltestcase.h"
#include "ctfeventlistener.h"
#include "CTFDosSharedDataTC.h"

const TUid KUidDosTestCategory = {0x10200C6A};

MTFDosControl* GetDosServerControl( void )
    {
    CTFStub* stub = CTFFactory::Stub( KTFStubTypeDosServerControl );
    return STATIC_CAST( CTFDosControl*, stub );
    }

MTFDosControl::MTFDosControl( void )
    {
    }
    
    
MTFDosControl::~MTFDosControl( void )
    {
    }
    

CTFDosControl::CTFDosControl( void )
: CTFStub( KTFStubTypeDosServerControl )
    {
    }
    
//lint -e1740 Pointer not directly deleted by destructor
CTFDosControl::~CTFDosControl( void )
    {
    ResetDosServices();
    COMPONENT_TRACE( ( _L( "DOSSERVERTESTPLUGIN - CTFDosServerControl::~CTFDosServerControl() - DosServer connection closed" ) ) );
    }
//lint +e1740    
    
void CTFDosControl::InitializeL( void )
    {
    COMPONENT_TRACE( ( _L( "DOSSERVERTESTPLUGIN - CTFDosServerControl::InitializeL()" ) ) );
    User::LeaveIfError( iDosServer.Connect() );            
    iInitialized = ETrue;
    COMPONENT_TRACE( ( _L( "DOSSERVERTESTPLUGIN - CTFDosServerControl::InitializeL - return" ) ) );
    }

void CTFDosControl::ResetDosServices( void )
    {
    COMPONENT_TRACE( ( _L( "CTFDosServerStub::ResetDosServices()" ) ) );
    if ( iEventListener != NULL )
        {
        iEventListener->Stop();
        delete iEventListener;
        iEventListener = NULL;
        }  
  
    if ( iDosSharedData != NULL )
        {
        iDosSharedData->Close();
        delete iDosSharedData;
        iDosSharedData = NULL;
        }      
      if ( iDosExtension != NULL )
        {
        iDosExtension->Close();
        delete iDosExtension;
        iDosExtension = NULL;
        }
          
    iDosServer.Close();
    iInitialized = EFalse;
    COMPONENT_TRACE( ( _L( "CTFDosServerStub::ResetDosServices() - Success" ) ) );
    }

void CTFDosControl::CallDosFunctionL( TTFDosServerControlTestCaseState& aParameter )
    {
    COMPONENT_TRACE( ( _L( "DOSSERVERTESTPLUGIN - CTFDosServerControl::CallDosFunctionL(%d, %d, %d)" ), aParameter.iDosFunction, aParameter.iArg1, aParameter.iArg2 ) );
    if ( !iInitialized )
        {
        InitializeL();
        }
    TInt retval = KTFErrDosServiceMissing;
    switch ( aParameter.iDosFunction )
        {      
         // SharedData TC
        case ETFDosSharedData_Open:
        	if ( iDosSharedData == NULL )
        		{
        		iDosSharedData = new ( ELeave ) RDosSharedData();
        		retval = iDosSharedData->Open ( iDosServer );
        		}
        	else
        		{
        		retval = KErrNone;
        		}
        	break;
        case ETFDosSharedData_Close:
            TRACE_ASSERT( iDosSharedData != NULL );
            if ( iDosSharedData != NULL )
                {
                iDosSharedData->Close();
                delete iDosSharedData;
                iDosSharedData = NULL;
                }
            retval = KErrNone;
            break;
        case ETFDosSharedData_FillDisk:
        	if ( iSdDisk == NULL )
        	    {
        		iSdDisk = CSdDisk::NewL();
        	    }
        	
       		iSdDisk->FillDisk(iFs, aParameter.iArg1);
       		delete iSdDisk;
       		iSdDisk = NULL;
				
			retval = KErrNone;
			break;
        case ETFDosSharedData_ReleaseDisk:
        	if ( iSdDisk == NULL )
        	    {
        		iSdDisk = CSdDisk::NewL();
        	    }
        	
       		iSdDisk->ReleaseDisk(iFs);
       		delete iSdDisk;
       		iSdDisk = NULL;
				
			retval = KErrNone;
    	    break;
        case ETFDosSharedData_CheckDiskSpace:
        	if ( iSdDisk == NULL )
        	    {
        		iSdDisk = CSdDisk::NewL();
        	    }

      		iFreeDiskSpace = iSdDisk->CheckDiskSpace( iFs );
     
   			RDebug::Print(_L("Shareddatatest ## CheckDiskSpace: %f Bytes"), iFreeDiskSpace);
       		        		      		
       		if ( aParameter.iArg1 != 0 )
       		    {       		    
       		    if ( aParameter.iArg1 == ( TInt(iFreeDiskSpace) / 1024 ) || 
       		         iSdDisk->CheckReserveFileSize( iFs ) == 0 )
       		        {
       			    retval = KErrNone;
       		        }
       		    else 
       		        {
       			    retval = KErrGeneral;
				    iSdDisk->ReleaseDisk(iFs);
	        	
	        	    if ( iDosSharedData != NULL )
	        	        {
					    iDosSharedData->Close();
               		    delete iDosSharedData;
               		    iDosSharedData = NULL;
       			        }
        		
        		    delete iSdDisk;
        		    iSdDisk = NULL;
       		        }
        	    }

   		    else if ( aParameter.iArg2 == 1 )
   		        {
   		        if ( iSdDisk->CheckReserveFileSize( iFs ) == 256*1024 )
   		            {
   		            retval = KErrNone;
   		            }
   		        else
   		            {
   		            retval = KErrGeneral;
   		            }
   		        }
        	    
        	else
        	    {
        	    retval = KErrNone;    
        	    }
    	    break;    	   
        case ETFDosSharedData_RequestFreeDiskSpace:
        	if ( iDosSharedData != NULL )
        	    {
        		RDebug::Print(_L("Shareddatatest ## RequestfreeDiskSpace: %d kB"), aParameter.iArg1 * 1024);
        		iDosSharedData->RequestFreeDiskSpace(aParameter.iArg1 * 1024);
        		retval = KErrNone;
        	    }
	        break;
        case ETFDosSharedData_RequestFreeDiskSpaceCancel:
        	if ( iDosSharedData != NULL )
        	    {
        		RDebug::Print(_L("Shareddatatest ## RequestfreeDiskSpaceCancel"));
        		iDosSharedData->RequestFreeDiskSpaceCancel();
        		retval = KErrNone;
        	    }    	    
	        break;        
	        
	      case ETFDosExtension_Open:
            if ( iDosExtension == NULL )
                {
                iDosExtension = new ( ELeave ) RTFDosExtension();
                retval = iDosExtension->Open( iDosServer );
                }
            else
                {
                retval = KErrNone;
                }
            break;
        case ETFDosExtension_CallFunction:
            TRACE_ASSERT( iDosExtension != NULL );
            if ( iDosExtension != NULL )
                {
                retval = iDosExtension->CallSyncDosExtFunction( aParameter.iArg1, ( TAny* )&aParameter.iArg2, sizeof ( aParameter.iArg2 ), ETrue );
                }
            break;
        case ETFDosExtension_CallFunctionAndComplete:
            TRACE_ASSERT( iDosExtension != NULL );
            if ( iDosExtension != NULL )
                {
                TRequestStatus status;
                retval = iDosExtension->CallAsyncDosExtFunction( status, aParameter.iArg1, ( TAny* )&aParameter.iArg2, sizeof ( TInt ), ETrue );                
                }
            break;
        
        case ETFDosExtension_GeneralEvent:
            TRACE_ASSERT( iDosExtension != NULL );
            
            iEventListener = new ( ELeave ) CTFEventListener( iDosServer );
            iEventListener->StartListeningL( KLDSYDummyEvent, sizeof(TInt), EQueue );
    
            if ( iDosExtension != NULL )
                {
                TRequestStatus status;
                iDosExtension->CallAsyncDosExtFunction( status, aParameter.iArg1, ( TAny* )&aParameter.iArg2, sizeof ( TInt ), EFalse );                
                retval = KErrNone;
                }
            break;    
            
        case ETFDosExtension_GeneralEventArray:
            TRACE_ASSERT( iDosExtension != NULL );
            
            iEventListener = new ( ELeave ) CTFEventListener( iDosServer );
            
            TRegisterEvent eventList[] = 
            {
               { KLDSYDummyEvent, sizeof ( TInt ), EQueue }    
            };
            
            iEventListener->StartListeningL( eventList, sizeof ( eventList ) / sizeof ( TRegisterEvent ) );
    
            if ( iDosExtension != NULL )
                {
                TRequestStatus status;
                iDosExtension->CallAsyncDosExtFunction( status, aParameter.iArg1, ( TAny* )&aParameter.iArg2, sizeof ( TInt ), EFalse );                
                retval = KErrNone;
                }
            break;    
                        
        case ETFDosExtension_Close:
            TRACE_ASSERT( iDosExtension != NULL );
            if ( iDosExtension != NULL )
                {
                iDosExtension->Close();
                delete iDosExtension;
                iDosExtension = NULL;
                }
                
            if ( iEventListener != NULL )
                {
                iEventListener->Stop();
                delete iEventListener;
                iEventListener = NULL;
                }
                    
            retval = KErrNone;
            break;
        
        case ETFRFs_Connect:
					{
					retval = iFs.Connect();	
					}
	        break;

				case ETFRFs_Close:
					{
					iFs.Close();	
					retval = KErrNone;
					}
	      break;
	              	        
        default:
            TRACE_ASSERT_ALWAYS;
        break;
        }
    User::LeaveIfError( retval );
    }

void CTFDosControl::NotifyDosEvent( TInt aEvent, TInt aParameter )
    {
    COMPONENT_TRACE( ( _L( "DOSSERVERTESTPLUGIN - CTFDosServerControl::NotifyDosEvent(%d, %d)" ), aEvent, aParameter ) );
    CTFDosControlTestCase* testCase = STATIC_CAST( CTFDosControlTestCase*, CurrentTestCase() );
    TRACE_ASSERT( testCase != NULL );
    if ( testCase != NULL )
        {
        testCase->NotifyDosEvent( aEvent, aParameter );
        }
    COMPONENT_TRACE( ( _L( "DOSSERVERTESTPLUGIN - CTFDosServerControl::NotifyDosEvent - return" ) ) );
    }
    
TInt RTFDosExtension::CallSyncDosExtFunction( const TInt& aFunc, TAny* aParam = NULL, TInt aParLength = 0, TBool aAutoComplete = ETrue )
    {
    COMPONENT_TRACE( ( _L( "    DSYTESTTOOL - RTFDosExtension::CallSyncDosExtFunction(0x%x, 0x%x, 0x%x, 0x%x)" ), aFunc, aParam, aParLength, aAutoComplete ) );
    TExtensionParPckg package;
    package().iFunc = aFunc;
    package().iParLength = aParLength;
    package().iAutoComplete = aAutoComplete;
    TPtr8 ptr( ( TUint8* )aParam, aParLength, aParLength );
    TInt result = CallFunction( package, ptr );
    COMPONENT_TRACE( ( _L( "    DSYTESTTOOL - RTFDosExtension::CallSyncDosExtFunction - return 0x%x" ), result ) );
    return result;
    }


TInt RTFDosExtension::CallAsyncDosExtFunction( TRequestStatus& aStatus, TInt aFunc, TAny* aParam = NULL, TInt aParLength = 0, TBool aAutoComplete = EFalse )
    {
    COMPONENT_TRACE( ( _L( "    DSYTESTTOOL - RTFDosExtension::CallAsyncDosExtFunction(0x%x, 0x%x, 0x%x, 0x%x, 0x%x)" ), &aStatus, aFunc, aParam, aParLength, aAutoComplete ) );
    TExtensionParPckg package;
    package().iFunc = aFunc;
    package().iParLength = aParLength;
    package().iAutoComplete = aAutoComplete;
    TPtr8 ptr( ( TUint8* )aParam, aParLength, aParLength );
    CallFunction( aStatus, package, ptr );
    User::WaitForRequest( aStatus );
    COMPONENT_TRACE( ( _L( "    DSYTESTTOOL - RTFDosExtension::CallAsyncDosExtFunction - return 0x%x" ), aStatus.Int() ) );
    return aStatus.Int();
    }