// Copyright (c) 2008-2009 Nokia Corporation and/or its subsidiary(-ies).
// All rights reserved.
// This component and the accompanying materials are made available
// under the terms of the License "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:
// Bootstrap Shadow Memory Region Test Driver
//


// -- INCLUDES ----------------------------------------------------------------

#include "d_trace.h"

#include <memmodel\epoc\platform.h>
#include <bootdefs.h>
#include <kernel/kernboot.h>
#include <kernel/kern_priv.h>
#include <platform.h>
#include <u32hal.h>
#include "d_smr.h"


// -- CLASSES -----------------------------------------------------------------


class DSMRTestFactory : public DLogicalDevice
	{
public:
	virtual TInt Install();
	virtual void GetCaps(TDes8& aDes) const;
	virtual TInt Create(DLogicalChannelBase*& aChannel);
	};


class DSMRTestChannel : public DLogicalChannelBase
	{
public:
	DSMRTestChannel();
	virtual ~DSMRTestChannel();
	
	//	Inherited from DObject
	virtual TInt RequestUserHandle(DThread* aThread, TOwnerType aType);
	
	// Inherited from DLogicalChannelBase
	virtual TInt DoCreate(TInt aUnit, const TDesC8* anInfo, const TVersion& aVer);
	virtual TInt Request(TInt aReqNo, TAny* a1, TAny* a2);
		
public:
	static void TestTrace(DSMRTestChannel* aSelf);
	
private:
	DThread* iClient;
    
	};

TInt OpenDumpCloseChunk(TUint32 aBase, TUint32 aSize);


// -- GLOBALS -----------------------------------------------------------------
//



// -- METHODS -----------------------------------------------------------------
//
// TEST FUNCTIONS
//

static TInt ECtrlCheckSMRIBPtr()
	{
	SMR_FUNC("DSMRTestChannel::Request::ECtrlCheckSMRIBPtr");

    TSuperPage& superPage = Kern::SuperPage();
    TLinAddr smrib = superPage.iSmrData;

    SMR_TRACE1("SMRIB - Virtual address %x", smrib);

    return smrib;
	}
	
static TInt ECtrlPrintSMRIB()
	{
	SMR_FUNC("DSMRTestChannel::Request::ECtrlPrintSMRIB");

    TSuperPage& superPage = Kern::SuperPage();
    TLinAddr smrib = superPage.iSmrData;

    SMR_TRACE1("SMRIB - Virtual address %x", smrib);

    if (smrib == KSuperPageAddressFieldUndefined)
    	SMR_LOGMSG_RETURN ("SMRIB Does not exist!", KErrBadHandle);
    
    SSmrBank* smrPtr = (SSmrBank*)(smrib);
    int x=0;
    while (smrPtr->iBase != 0)
        {
        SMR_TRACE6("SMRIB entry %d (0x%x): %x, %x, %x, %x", x, smrPtr, smrPtr->iBase, smrPtr->iSize, smrPtr->iPayloadUID, smrPtr->iPayloadFlags);
        x++;
        smrPtr++;
        }
    if (x==0)
    	SMR_TRACE0("SMRIB Zero, no valid entries");
    else
    	SMR_TRACE1("SMRIB Contained %d entries", x);

    return x;
	}


static TInt ECtrlAccessAllSMRs()
	{
	SMR_FUNC("DSMRTestChannel::Request::ECtrlAccessAllSMRs");

    TInt err=0; 
    TSuperPage& superPage = Kern::SuperPage();
    TLinAddr smrib = superPage.iSmrData;

    if (smrib == KSuperPageAddressFieldUndefined)
        return KErrBadHandle;
    
    SSmrBank* smrPtr = (SSmrBank*)(smrib);
    int x=0;
    while (smrPtr->iBase != 0)
        {
        SMR_TRACE6("SMRIB item %d (0x%x): %x, %x, %x, %x", x, smrPtr, smrPtr->iBase, smrPtr->iSize, smrPtr->iPayloadUID, smrPtr->iPayloadFlags);
        
        err = OpenDumpCloseChunk(smrPtr->iBase, smrPtr->iSize);
		if (err != KErrNone)
  			return err;
  			
        x++;
        smrPtr++;
        }
    if (x==0)
    	SMR_TRACE0("SMRIB Zero, no valid entries");
    else
    	SMR_TRACE1("SMRIB Contained %d entries", x);
   
	return x;
	}
	
	
static TInt ECtrlFreeHalfSMR1PhysicalRam()
	{
	SMR_FUNC("DSMRTestChannel::Request::ECtrlFreeHalfSMR1PhysicalRam");

    TInt err=0; 
    TSuperPage& superPage = Kern::SuperPage();
    TLinAddr smrib = superPage.iSmrData;
    

    if (smrib == KSuperPageAddressFieldUndefined)
        return KErrBadHandle;
    
    SSmrBank* smrPtr = (SSmrBank*)(smrib);
    int x=0;
    
    if ((smrPtr->iBase == 0) || (smrPtr->iSize == 0))
        SMR_LOGMSG_RETURN("SMRIB Does not contain one entry!", 0); 
        
    SMR_TRACE6("SMRIB item before %d (0x%x): %x, %x, %x, %x", x, smrPtr, smrPtr->iBase, smrPtr->iSize, smrPtr->iPayloadUID, smrPtr->iPayloadFlags);
	
	TInt halfSize = smrPtr->iSize >> 1;
	
	NKern::ThreadEnterCS();
    err = Epoc::FreePhysicalRam(smrPtr->iBase+halfSize, halfSize);
    NKern::ThreadLeaveCS();
    if (err != KErrNone)
        SMR_LOGMSG_RETURN("Epoc::FreePhysicalRam() gave error", err) 
    else
    	SMR_TRACE0("Success - half of physical ram freed for SMR 1");	
 
	smrPtr->iSize = halfSize;
	
    SMR_TRACE6("SMRIB item after %d (0x%x): %x, %x, %x, %x", x, smrPtr, smrPtr->iBase, smrPtr->iSize, smrPtr->iPayloadUID, smrPtr->iPayloadFlags);

    err = OpenDumpCloseChunk(smrPtr->iBase, smrPtr->iSize);
	if (err != KErrNone)
  		return err;

	return halfSize;
	}


static TInt ECtrlFreeAllSMR2PhysicalRam()
	{
	SMR_FUNC("DSMRTestChannel::Request::ECtrlFreeAllSMR2PhysicalRam");

    TInt err=0; 
    TSuperPage& superPage = Kern::SuperPage();
    TLinAddr smrib = superPage.iSmrData;

    if (smrib == KSuperPageAddressFieldUndefined)
        return KErrBadHandle;
    
    SSmrBank* smrPtr = (SSmrBank*)(smrib);
    int x=0;

	if ((smrPtr->iBase == 0) || (smrPtr->iSize == 0))
        SMR_LOGMSG_RETURN("SMRIB Does not contain first entry!", 0);
    
    smrPtr++; x++;
	if ((smrPtr->iBase == 0) || (smrPtr->iSize == 0))
        SMR_LOGMSG_RETURN("SMRIB Does not contain two entries!", 0);

    smrPtr++; x++;
	if ((smrPtr->iBase == 0) || (smrPtr->iSize == 0))
        SMR_LOGMSG_RETURN("SMRIB Does not contain three entries!", 0);

    SMR_TRACE6("SMRIB item before %d (0x%x): %x, %x, %x, %x", x, smrPtr, smrPtr->iBase, smrPtr->iSize, smrPtr->iPayloadUID, smrPtr->iPayloadFlags);        
   	TInt sizeToFree = smrPtr->iSize;
	
	NKern::ThreadEnterCS();
    err = Epoc::FreePhysicalRam(smrPtr->iBase, sizeToFree);
    NKern::ThreadLeaveCS();
    if (err != KErrNone)
        SMR_LOGMSG_RETURN("Epoc::FreePhysicalRam() gave error", err) 
    else
    	SMR_TRACE0("Success - all physical ram freed for SMR 2");	
        

	smrPtr->iBase = 0;
	smrPtr->iSize = 0;
	smrPtr->iPayloadUID = 0;
	smrPtr->iPayloadFlags = 0;
	
    SMR_TRACE6("SMRIB item after %d (0x%x): %x, %x, %x, %x", x, smrPtr, smrPtr->iBase, smrPtr->iSize, smrPtr->iPayloadUID, smrPtr->iPayloadFlags);
	return sizeToFree;
	}


TInt OpenDumpCloseChunk(TUint32 aBase, TUint32 aSize)
	{
	TInt err;
    TChunkCreateInfo cci;
    DChunk *chunkPtr;
    TLinAddr kernAddr = 0;
    TUint32 mapAttr = 0;
    
    cci.iType = TChunkCreateInfo::ESharedKernelSingle;
    cci.iMaxSize = 0x800000;
    cci.iMapAttr = EMapAttrCachedMax | EMapAttrSupRw;
    cci.iOwnsMemory = EFalse;
    cci.iDestroyedDfc = 0;
    
    NKern::ThreadEnterCS();
    err = Kern::ChunkCreate(cci, chunkPtr, kernAddr, mapAttr);
    NKern::ThreadLeaveCS();
    if (err != KErrNone)
    	SMR_LOGMSG_RETURN("Kern::ChunkCreate() gave error", err);
         
    NKern::ThreadEnterCS();
    err = Kern::ChunkCommitPhysical(chunkPtr, 0, aSize, aBase);
    NKern::ThreadLeaveCS();
    if (err != KErrNone)
    	SMR_LOGMSG_RETURN("Kern::ChunkCommitPhysical() gave error", err); 
        
    TUint32* setting = (TUint32*)(kernAddr); 
    SMR_TRACE1("SMR Image Memory Dump First Kb @ %08x", setting);
            
    for (TInt y=0; y < 0x80; y+=16)
        {
        SMR_TRACE5("  %08x:  %08x  %08x  %08x  %08x", setting, setting[0], setting[1], setting[2], setting[3]);
        setting+=4;
        }
        
    setting = (TUint32*)(kernAddr+aSize-0x80);
	SMR_TRACE1("SMR Image Memory Dump Last Kb @ %08x", setting);
            
    for (TInt y=0; y < 0x80; y+=16)
        {
        SMR_TRACE5("  %08x:  %08x  %08x  %08x  %08x", setting, setting[0], setting[1], setting[2], setting[3]);
        setting+=4;
        }
              
    NKern::ThreadEnterCS();
    TBool chunkRefCntZero = Kern::ChunkClose(chunkPtr);
    NKern::ThreadLeaveCS();
    if (chunkRefCntZero == 0)
		SMR_LOGMSG_RETURN("Kern::ChunkClose gave false result", KErrGeneral);
    
    return KErrNone;
	}

// -- METHODS -----------------------------------------------------------------
//
// DSMRTestFactory
//

TInt DSMRTestFactory::Install()
	{
    SMR_FUNC("DSMRTestFactory::Install");
	return SetName(&RSMRTest::Name());
	}

void DSMRTestFactory::GetCaps(TDes8& aDes) const
	{
    SMR_FUNC("DSMRTestFactory::GetCaps");
  	Kern::InfoCopy(aDes,0,0);
	}

TInt DSMRTestFactory::Create(DLogicalChannelBase*& aChannel)
	{
    SMR_FUNC("DSMRTestFactory::Create");
   
   	aChannel=new DSMRTestChannel();
	if(!aChannel)
		return KErrNoMemory;
	return KErrNone;
	}


// -- METHODS -----------------------------------------------------------------
//
// DSMRTestChannel
//

DSMRTestChannel::DSMRTestChannel()
	{
    SMR_FUNC("DSMRTestChannel");
   	}

DSMRTestChannel::~DSMRTestChannel()
	{
    SMR_FUNC("~DSMRTestChannel");
	}

TInt DSMRTestChannel::DoCreate(TInt /*aUnit*/, const TDesC8* /*aInfo*/, const TVersion& /*aVer*/)
	{
    SMR_FUNC("DSMRTestChannel::DoCreate");
   	
    iClient = &Kern::CurrentThread();
	return KErrNone;
	}

TInt DSMRTestChannel::RequestUserHandle(DThread* aThread, TOwnerType aType)
	{
    SMR_FUNC("DSMRTestChannel::RequestUserHandle");
    
	if (aType!=EOwnerThread || aThread!=iClient)
		return KErrAccessDenied;
	return KErrNone;
	}

TInt DSMRTestChannel::Request(TInt aReqNo, TAny* a1, TAny*)
	{
    SMR_FUNC("DSMRTestChannel::Request");
	TBool aEnforce = (TBool) a1;
	
	switch(aReqNo)
		{
		
	case RSMRTest::ECtrlCheckSMRIBPtr:
		{
		TInt rc = ECtrlCheckSMRIBPtr();
		if (rc < 0)
    		return rc;
		if (aEnforce && rc == 0)
    		return KErrBadHandle;
    		
		break; // fall through, return KErrNone
		}
		
	case RSMRTest::ECtrlPrintSMRIB:
		{
		TInt rc = ECtrlPrintSMRIB();
		if (rc < 0)
    		return rc;
		if (aEnforce && rc == 0)
    		return KErrNotFound;
    		
		break; // fall through, return KErrNone
		}
	
	case RSMRTest::ECtrlAccessAllSMRs:
		{
        TInt rc = ECtrlAccessAllSMRs();
		if (rc < 0)
    		return rc;
		if (aEnforce && rc == 0)
    		return KErrNotFound;
    		
    	break; // fall through, return KErrNone
		}
	
	case RSMRTest::ECtrlFreeHalfSMR1PhysicalRam:
		{
		TInt rc = ECtrlFreeHalfSMR1PhysicalRam();
		if (rc < 0)
    		return rc;
		if (aEnforce && rc == 0)
    		return KErrNotFound;
		
		break; // fall through, return KErrNone
		}
	
	case RSMRTest::ECtrlFreeAllSMR2PhysicalRam:
		{
		
		TInt rc = ECtrlFreeAllSMR2PhysicalRam();
		if (rc < 0)
    		return rc;
		if (aEnforce && rc == 0)
    		return KErrNotFound;
		
		break; // fall through, return KErrNone
		}
	
	default:
		return KErrNotSupported;
		}
		
	return KErrNone;
	}


// -- GLOBALS -----------------------------------------------------------------


DECLARE_STANDARD_LDD()
	{
    SMR_FUNC("D_SMR_DECLARE_STANDARD_LDD");

    const TRomHeader& romHdr = Epoc::RomHeader();
    
    TInt RHsize = sizeof(TRomHeader);
    SMR_TRACE2("RomHeader - addr %0x; size %d", &romHdr, RHsize);

    TSuperPage& superPage = Kern::SuperPage();
    TInt SPsize = sizeof(SSuperPageBase);
    
    TInt startupReason = superPage.iHwStartupReason;
    TLinAddr rootDirList = superPage.iRootDirList;
    
    SMR_TRACE2("SuperPage  - addr %0x; size %d", &superPage, SPsize);
    SMR_TRACE2("SuperPage - StartupReason: %0x; rootDirList %0x", startupReason, rootDirList);
                  
   	return new DSMRTestFactory;
	}