/*

* Copyright (c) 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 "MemSpyDriverLogChanContainers.h"

// System includes

#include <memspy/driver/memspydriverconstants.h>

#include <memspy/driver/memspydriverobjectsshared.h>

// Shared includes

#include "MemSpyDriverOpCodes.h"

#include "MemSpyDriverObjectsInternal.h"

// User includes

#include "MemSpyDriverUtils.h"

#include "MemSpyDriverUserEventMonitor.h"

#include "MemSpyDriverSuspensionManager.h"

#include "MemSpyDriverOSAdaption.h"

// Constants

const TInt KMemSpyDriverLogChanContainersXferBufferSize = 1024 * 4;

DMemSpyDriverLogChanContainers::DMemSpyDriverLogChanContainers( DMemSpyDriverDevice& aDevice, DThread& aThread )

:   DMemSpyDriverLogChanContainerBase( aDevice, aThread )

    {

	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::DMemSpyDriverLogChanContainers() - this: 0x%08x", this ));

    }

DMemSpyDriverLogChanContainers::~DMemSpyDriverLogChanContainers()

	{

	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::~DMemSpyDriverLogChanContainers() - START - this: 0x%08x", this ));

	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::~DMemSpyDriverLogChanContainers() - END - this: 0x%08x", this ));

	}

TInt DMemSpyDriverLogChanContainers::Construct()

	{

	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::Construct() - START - this: 0x%08x", this ));

    const TInt ret = BaseConstruct( KMemSpyDriverLogChanContainersXferBufferSize );

	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::Construct() - END - this: 0x%08x, err: %d", this, ret ));

    return ret;

	}

TInt DMemSpyDriverLogChanContainers::Request( TInt aFunction, TAny* a1, TAny* a2 )

	{

	TInt r = DMemSpyDriverLogChanContainerBase::Request( aFunction, a1, a2 );

    if  ( r == KErrNone )

        {

	    switch( aFunction )

		    {

	    case EMemSpyDriverOpCodeContainersGetHandles:

            r = GetContainerHandles( (TMemSpyDriverInternalContainerHandleParams*) a1 );

            break;

	    case EMemSpyDriverOpCodeContainersGetHandleInfo:

            r = GetGenericHandleInfo( (TInt) a1, (TMemSpyDriverHandleInfoGeneric*) a2 );

            break;

        case EMemSpyDriverOpCodeContainersGetApproxSize:

            r = GetKernelObjectSizeApproximation( (TMemSpyDriverContainerType) ((TInt) a1), (TInt*) a2 );

            break;

        case EMemSpyDriverOpCodeContainersGetReferencesToMyThread:

            r = GetReferencesToMyThread( (TUint) a1, (TDes8*) a2 );

            break;

        case EMemSpyDriverOpCodeContainersGetReferencesToMyProcess:

            r = GetReferencesToMyProcess( (TUint) a1, (TDes8*) a2 );

            break;

        case EMemSpyDriverOpCodeContainersGetPAndSInfo:

            r = GetPAndSInfo( (DObject*) a1, (TMemSpyDriverPAndSInfo*) a2 );

            break;

        case EMemSpyDriverOpCodeContainersGetCondVarSuspendedThreads:

            r = GetCondVarSuspendedThreads( (TMemSpyDriverInternalCondVarSuspendedThreadParams*)a1 );

            break;

        case EMemSpyDriverOpCodeContainersGetCondVarSuspendedThreadInfo:

            r = GetCondVarSuspendedThreadInfo( a1, (TMemSpyDriverCondVarSuspendedThreadInfo*)a2 );

            break;

        default:

            r = KErrNotSupported;

		    break;

		    }

        }

    //

    return r;

	}

TBool DMemSpyDriverLogChanContainers::IsHandler( TInt aFunction ) const

    {

    return ( aFunction > EMemSpyDriverOpCodeContainersBase && aFunction < EMemSpyDriverOpCodeContainersEnd );

    }

TInt DMemSpyDriverLogChanContainers::GetContainerHandles( TMemSpyDriverInternalContainerHandleParams* aParams )

	{

    TMemSpyDriverInternalContainerHandleParams params;

    TInt r = Kern::ThreadRawRead( &ClientThread(), aParams, &params, sizeof(TMemSpyDriverInternalContainerHandleParams) );

    if  ( r != KErrNone )

        {

    	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles() - END - params read error: %d", r));

        return r;

        }

	const TInt maxCount = params.iMaxCount;

	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles() - START - tid or pid: %d, maxCount: %d, container: %d, handleSource: %d", params.iTidOrPid, maxCount, params.iContainer, params.iHandleSource ));

    DMemSpyDriverOSAdaptionDThread& threadAdaption = OSAdaption().DThread();

    DMemSpyDriverOSAdaptionDProcess& processAdaption = OSAdaption().DProcess();

    ResetTempHandles();

    if  ( params.iTidOrPid == KMemSpyDriverEnumerateContainerHandles )

        {

        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - enumerating container handles... " ));

	    NKern::ThreadEnterCS();

        // We are enumerating an entire container - not a thread-specific request

        const TObjectType type = ObjectTypeFromMemSpyContainerType( params.iContainer );

        DObjectCon* container = Kern::Containers()[type];

        container->Wait();

        NKern::LockSystem();

        const TInt count = container->Count();

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

            {

            DObject* object = (*container)[ i ];

            if  ( object != NULL )

                {

                AddTempHandle( object );

                }

            }

        NKern::UnlockSystem();

        container->Signal();

    	NKern::ThreadLeaveCS();

        }

    else

        {

        // Are we dealing with threads or processes?

        const TObjectType sourceContainerType = (TObjectType) params.iHandleSource;

        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - sourceContainerType: %d", sourceContainerType ));

        // We need to open the process or thread in order to get its handle

        r = OpenTempObject( params.iTidOrPid, sourceContainerType );

        if  ( r == KErrNone )

		    {

            TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - handle source object: %O", TempObject() ));

            // This is the source for the handles we want to check. Its either the process or thread's handle index...

            MemSpyObjectIx* handles = NULL;

            // Check that the process' thread's are suspended and get handle index

            if  ( params.iHandleSource == EMemSpyDriverThreadOrProcessTypeThread )

                {

                DThread* thread = (DThread*) TempObject();

                if  ( !SuspensionManager().IsSuspended( *thread ) )

                    {

                    TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - parent process not suspended => KErrAccessDenied"));

                    r = KErrAccessDenied;

                    }

                else

                    {

                    handles = threadAdaption.GetHandles( *thread );

                    TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - checking handles for thread: %O", thread ));

                    }

                }

            else if ( params.iHandleSource == EMemSpyDriverThreadOrProcessTypeProcess )

                {

                DProcess* process = (DProcess*) TempObject();

                if  ( !SuspensionManager().IsSuspended( *process ) )

                    {

                    TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - process not suspended => KErrAccessDenied"));

                    r =KErrAccessDenied;

                    }

                else

                    {

                    handles = processAdaption.GetHandles( *process );

                    TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - checking handles for process: %O", process ));

                    }

                }

            if  ( handles )

                {

	            NKern::ThreadEnterCS();

                // Iterate through each handle in the thread/process and add it to the temp handles container if

                // the handle is of the correct type.

                MemSpyObjectIx_Wait( handles );

	            TInt handleCount = handles->Count();

                TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - %d handles in index...", handleCount ));

                for( TInt handleIndex=0; handleIndex<handleCount; handleIndex++ )

    	            {

    	            // Get a handle from the container...

                    NKern::LockSystem();

    	            DObject* objectToSearchFor = (*handles)[ handleIndex ];

                    NKern::UnlockSystem();

                    if  ( objectToSearchFor != NULL )

                        {

                        // Check to see if this object is of the specified type. We can use quick mode

                        // because we know the object is valid since it's registered as a handle of the

                        // thread/process.

                        DObject* matchResult = CheckIfObjectIsInContainer( params.iContainer, objectToSearchFor, ETrue );

                        if  ( matchResult )

                            {

                            // Found a match in the specified container. Write the object's handle (aka the object address)

                            // back to the client address space

                            AddTempHandle( matchResult );

                            }

                        }

    	            }

                MemSpyObjectIx_Signal( handles );

                NKern::ThreadLeaveCS();

                }

            // Done with this now.

            CloseTempObject();

            }

        else

		    {

    	    Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - thread / process not found");

		    }

        }

    // Write back handles to client

    TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles - trying to write %d handles to client...", TempHandleCount() ));

    r = WriteToClient( params.iHandles, params.iCountPtr, maxCount );

	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetContainerHandles() - END - r: %d", r));

	return r;

    }

TInt DMemSpyDriverLogChanContainers::GetGenericHandleInfo( TInt aTid, TMemSpyDriverHandleInfoGeneric* aParams )

    {

	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - START" ));

    TMemSpyDriverHandleInfoGeneric params;

    TInt r = Kern::ThreadRawRead( &ClientThread(), aParams, &params, sizeof(TMemSpyDriverHandleInfoGeneric) );

    if  ( r != KErrNone )

        {

    	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - END - params read error: %d", r));

        return r;

        }

	TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo  - tid: %d, handle: 0x%08x", aTid, params.iHandle ));

    if  ( aTid == KMemSpyDriverEnumerateContainerHandles )

        {

	    TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo - getting info for non-thread specific handle..."));

        }

    else

        {

	    TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo - thread-specific request..."));

        r = OpenTempObject( aTid, EThread );

	    if (r != KErrNone)

		    {

    	    Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - END - thread not found");

		    return r;

		    }

        // Check that the process' thread's are suspended

        DThread* thread = (DThread*) TempObject();

	    TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo - opened thread: %O", thread));

        if  ( !SuspensionManager().IsSuspended( *thread ) )

            {

            TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - END - parent process not suspended => KErrAccessDenied"));

            CloseTempObject();

            return KErrAccessDenied;

            }

        }

	NKern::ThreadEnterCS();

    // First, locate the specific DObject in question. Cast the handle, but don't use the object...

    DObject* handleAsObject = (DObject*) params.iHandle;

    handleAsObject = CheckIfObjectIsInContainer( params.iType, handleAsObject );

    if  ( handleAsObject != NULL )

        {

        // We found the right object. First get generic info.

        handleAsObject->FullName( params.iName );

        handleAsObject->Name( params.iNameDetail );

        NKern::LockSystem();

        r = handleAsObject->Open();

        NKern::UnlockSystem();

        //

        if  ( r == KErrNone )

            {

            // Using threadAddaption to fetch generic info.

            // Implementations of following get functions are actually in DMemSpyDriverOSAdaptionDObject

            // so it does not matter what adaption to use for generic info.

            DMemSpyDriverOSAdaptionDThread& threadAddaption = OSAdaption().DThread();

            params.iAccessCount = threadAddaption.GetAccessCount( *handleAsObject );

            params.iUniqueID = threadAddaption.GetUniqueID( *handleAsObject );

            params.iProtection = threadAddaption.GetProtection( *handleAsObject );

            params.iAddressOfKernelOwner = threadAddaption.GetAddressOfKernelOwner( *handleAsObject );

            //

            handleAsObject->Close( NULL );

            }

        // Get type-specific info.

        if  ( params.iType == EMemSpyDriverContainerTypeThread )

            {

	        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - EMemSpyDriverContainerTypeThread" ));

            DThread* object = (DThread*) handleAsObject;

	        NKern::LockSystem();

            r = object->Open();

    	    NKern::UnlockSystem();

            //

            if  ( r == KErrNone )

                {

                DMemSpyDriverOSAdaptionDThread& threadAdaption = OSAdaption().DThread();

                //

                params.iId = threadAdaption.GetId( *object );

                params.iPriority = threadAdaption.GetPriority( *object );

                params.iAddressOfOwningProcess = threadAdaption.GetAddressOfOwningProcess( *object );

                threadAdaption.GetNameOfOwningProcess( *object, params.iNameOfOwner );

                //

                object->Close( NULL );

                }

            }

        else if ( params.iType == EMemSpyDriverContainerTypeProcess )

            {

	        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - EMemSpyDriverContainerTypeProcess" ));

            DProcess* object = (DProcess*) handleAsObject;

	        NKern::LockSystem();

            r = object->Open();

    	    NKern::UnlockSystem();

            //

            if  ( r == KErrNone )

                {

                DMemSpyDriverOSAdaptionDProcess& processAdaption = OSAdaption().DProcess();

                //

                params.iId = processAdaption.GetId( *object );

                //

                params.iPriority = processAdaption.GetPriority( *object );

                params.iAddressOfOwningProcess = processAdaption.GetAddressOfOwningProcess( *object );

                params.iCreatorId = processAdaption.GetCreatorId( *object );

                params.iSecurityZone = processAdaption.GetSecurityZone( *object );

                params.iAttributes = processAdaption.GetAttributes( *object );

                params.iAddressOfDataBssStackChunk = processAdaption.GetAddressOfDataBssStackChunk( *object );

                //

                object->Close( NULL );

                }

            }

        else if ( params.iType == EMemSpyDriverContainerTypeChunk )

            {

	        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - EMemSpyDriverContainerTypeChunk" ));

            DChunk* object = (DChunk*) handleAsObject;

	        NKern::LockSystem();

            r = object->Open();

    	    NKern::UnlockSystem();

            //

            if  ( r == KErrNone )

                {

                DMemSpyDriverOSAdaptionDChunk& ca = OSAdaption().DChunk();

                //

                params.iSize = ca.GetSize( *object );

                params.iId = ca.GetOwningProcessId( *object );

                params.iAddressOfOwningProcess = ca.GetAddressOfOwningProcess( *object );

                params.iMaxSize = ca.GetMaxSize( *object );

                params.iBottom = ca.GetBottom( *object );

                params.iTop = ca.GetTop( *object );

                params.iAttributes = ca.GetAttributes( *object );

                params.iStartPos = ca.GetStartPos( *object );

                params.iControllingOwner = ca.GetControllingOwnerId( *object );

                params.iRestrictions = ca.GetRestrictions( *object );

                params.iMapAttr = ca.GetMapAttr( *object );

                params.iChunkType = ca.GetType( *object );

                ca.GetNameOfOwningProcess( *object, params.iNameOfOwner );

                //

                object->Close( NULL );

                }

            }

        else if ( params.iType == EMemSpyDriverContainerTypeLibrary )

            {

	        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - EMemSpyDriverContainerTypeLibrary" ));

	        Kern::AccessCode();

            //

            DLibrary* object = (DLibrary*) handleAsObject;

	        NKern::LockSystem();

            r = object->Open();

    	    NKern::UnlockSystem();

            //

            if  ( r == KErrNone )

                {

                DMemSpyDriverOSAdaptionDCodeSeg& csa = OSAdaption().DCodeSeg();

                DCodeSeg* codeSeg = csa.GetCodeSeg( *object );

                params.iAddressOfCodeSeg = (TUint8*)codeSeg;

                params.iMapCount = csa.GetMapCount( *object );

                params.iState = csa.GetState( *object );

                //

                if  ( codeSeg )

                    {

                    params.iSize = csa.GetSize( *codeSeg );

                    }

                else

                    {

                    r = KErrNotFound;

                    }

                //

                object->Close( NULL );

                }

            //

	        Kern::EndAccessCode();

            }

        else if ( params.iType == EMemSpyDriverContainerTypeSemaphore )

            {

	        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - EMemSpyDriverContainerTypeSemaphore" ));

            DSemaphore* object = (DSemaphore*) handleAsObject;

	        NKern::LockSystem();

            r = object->Open();

    	    NKern::UnlockSystem();

            //

            if  ( r == KErrNone )

                {

                DMemSpyDriverOSAdaptionDSemaphore& sa = OSAdaption().DSemaphore();

                params.iCount = sa.GetCount( *object );

                params.iResetting = sa.GetResetting( *object );

                //

                object->Close( NULL );

                }

            }

        else if ( params.iType == EMemSpyDriverContainerTypeMutex )

            {

	        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - EMemSpyDriverContainerTypeMutex" ));

            DMutex* object = (DMutex*) handleAsObject;

	        NKern::LockSystem();

            r = object->Open();

    	    NKern::UnlockSystem();

            //

            if  ( r == KErrNone )

                {

                DMemSpyDriverOSAdaptionDMutex& ma = OSAdaption().DMutex();

                //

                params.iCount = ma.GetHoldCount( *object );

                params.iWaitCount = ma.GetWaitCount( *object );

                params.iResetting = ma.GetResetting( *object );

                params.iOrder = ma.GetOrder( *object );

                //

                object->Close( NULL );

                }

            }

        else if ( params.iType == EMemSpyDriverContainerTypeTimer )

            {

	        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - EMemSpyDriverContainerTypeTimer" ));

            // Get timer properties

            DTimer* object = (DTimer*) handleAsObject;

	        NKern::LockSystem();

            r = object->Open();

    	    NKern::UnlockSystem();

            //

            if  ( r == KErrNone )

                {

                DMemSpyDriverOSAdaptionDTimer& ta = OSAdaption().DTimer();

                //

                params.iTimerType = MapToMemSpyTimerType( ta.GetType( *object ) );

                params.iTimerState = MapToMemSpyTimerState( ta.GetState( *object ) );

                //

                object->Close( NULL );

                }

            }

        else if ( params.iType == EMemSpyDriverContainerTypeServer )

            {

	        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - EMemSpyDriverContainerTypeServer" ));

            DServer* object = (DServer*) handleAsObject;

	        NKern::LockSystem();

            r = object->Open();

    	    NKern::UnlockSystem();

            //

            if  ( r == KErrNone )

                {

                DMemSpyDriverOSAdaptionDServer& sa = OSAdaption().DServer();

                //

                params.iCount = sa.GetSessionCount( *object );

                params.iId = sa.GetOwningThreadId( *object );

                params.iSessionType = sa.GetSessionType( *object );

                params.iAddressOfOwningThread = sa.GetAddressOfOwningThread( *object );

                sa.GetNameOfOwningThread( *object, params.iNameOfOwner );

                //

                object->Close( NULL );

                }

            }

        else if ( params.iType == EMemSpyDriverContainerTypeSession )

            {

	        TRACE( Kern::Printf("DMemSpyDriverLogChanContainers::GetGenericHandleInfo() - EMemSpyDriverContainerTypeSession" ));