diff -r 000000000000 -r bde4ae8d615e os/mm/mmlibs/mmfw/tsrc/mmfunittest/avictrl/testplugins/aviplaycontroller/customaviplaycontroller.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/os/mm/mmlibs/mmfw/tsrc/mmfunittest/avictrl/testplugins/aviplaycontroller/customaviplaycontroller.cpp	Fri Jun 15 03:10:57 2012 +0200
@@ -0,0 +1,369 @@
+// Copyright (c) 2007-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 <mmf/server/mmffile.h>
+#include "customaviplaycontroller.h"
+#include "tsu_mmf_avictrltestids.h"
+
+#define KInitIncompleteCode KMaxTInt
+
+_LIT(KTestAviFile, "c:\\mm\\mmf\\testfiles\\avictrl\\aud_vid.avi");
+
+/**
+*
+* NewL
+*
+*/
+CMMFController* CCustomAviPlayController::NewL()
+	{
+    CCustomAviPlayController* self = new(ELeave) CCustomAviPlayController;
+	CleanupStack::PushL(self);
+	self->ConstructL();
+	CleanupStack::Pop( self );
+	
+	return STATIC_CAST( CMMFController*, self );
+	}
+
+/**
+*
+* CCustomAviPlayController
+*
+*/
+CCustomAviPlayController::~CCustomAviPlayController()
+	{
+	if (iDataSource)
+		{
+		delete iDataSource;
+		}
+	if (iDataSink)
+		{
+		delete iDataSink;
+		}
+		
+    }
+
+/**
+*
+* CCustomAviPlayController
+*
+*/
+CCustomAviPlayController::CCustomAviPlayController() :	CAviPlayController()
+	{
+	}
+
+/**
+*
+* Override base class ResetL
+*
+*/
+void CCustomAviPlayController::ResetL()
+	{
+	CAviPlayController::ResetL();
+	if (iDataSource)
+		{
+		delete iDataSource;
+		iDataSource = NULL;
+		}
+	if (iDataSink)
+		{
+		delete iDataSink;
+		iDataSink = NULL;
+		}
+	}
+	
+/**
+*
+* Override base class PrimeL
+*
+*/
+void CCustomAviPlayController::PrimeL(TMMFMessage& aMessage)
+	{
+	if (iMemAllocPrimeTestEnabled)
+		{
+		__UHEAP_SETFAIL(RHeap::EFailNext ,iMemAllocPrimeFailCount);
+		TRAPD(err, CAviPlayController::PrimeL(aMessage));
+		__UHEAP_SETFAIL(RHeap::ENone ,0);
+		User::LeaveIfError(err);
+		}
+	else
+		{
+		CAviPlayController::PrimeL(aMessage);
+		}
+	}
+
+/**
+*
+* Override base class CustomCommand
+*
+*/
+void CCustomAviPlayController::CustomCommand(TMMFMessage& aMessage)
+	{
+	//[ check if the command is for the custom plugin
+	// otherwise pass it on to the real audio controller ]
+	/*if (iPrimingMessage)
+		{
+		delete iPrimingMessage;
+		iPrimingMessage = NULL;
+		}
+	TRAPD(err, iPrimingMessage = CCustomMMFMessageHolder::NewL(aMessage));
+	if (err != KErrNone)
+		{
+		aMessage.Complete(err);
+		return;
+		}*/
+	
+	if (aMessage.Destination().InterfaceId() == KMemoryAllocationTestId)
+		{
+		switch (aMessage.Function())
+			{
+		case KMemoryAllocTest1:
+			DoMemoryAllocTest1(aMessage);
+			break;
+		case KMemoryAllocTest2:
+			DoPrepareMemoryAllocTest2(aMessage);
+			break;
+		case KMemoryAllocTest3:
+			DoMemoryAllocTest3(aMessage);
+			break;
+		case KOOMPrimeTestEnable:
+			DoSetPrimetTestEnable(aMessage);
+			break;
+		case KOOMPrimeTestFailCount:
+			DoSetPrimetTestFailCount(aMessage);
+			break;
+			}
+		}
+	else
+		{
+		//[ let the plugin process the message ]
+		CAviPlayController::CustomCommand(aMessage);
+		}
+	}
+
+static void CleanupController(TAny* ptr)
+	{
+	CCustomAviPlayController* controller = STATIC_CAST(CCustomAviPlayController*, ptr);
+	TRAP_IGNORE(controller->ResetL());
+	}
+	
+/**
+*
+* ConstructL
+*
+*/
+void CCustomAviPlayController::ConstructL()
+	{
+	CAviPlayController::ConstructL();
+	}
+
+/**
+*
+* DoMemoryAllocTest1
+*
+*/
+void CCustomAviPlayController::DoMemoryAllocTest1(TMMFMessage& aMessage)
+	{
+	//[ it is a alloc memory test command ]
+	// [ new algorithm adopted from M&G database ]
+	//
+	TInt errorCode = KErrNone;
+	TInt failCount  = 1;
+	TBool completed  = EFalse;
+	TBool badResult  = EFalse;
+	TBool reachedEnd = EFalse; 
+	
+	for( ; ; )
+		{
+		__UHEAP_SETFAIL(RHeap::EFailNext ,failCount);	// Leavescan will think __UHEAP_SETFAIL is a leaving function due to the macro ending in 'L'. Does not leave
+		//NB: Do not use __MM_HEAP_MARK macro's in this test, the cleaning up the CustomMMFAudioController
+		//is insufficient to remove all possible allocated memory from the framework.
+		//CMMFControllerProxyServer::DoStartThreadL has been updated to do heap checking for 
+		//the whole server thread.
+
+		//[ run a scenario of the major api 
+		//functions which alloc memory in the controller]
+		TRAP( errorCode, 			
+			TMMFFileConfig fileConfig; // video file 
+			fileConfig().iPath = KTestAviFile;
+
+			//[ lets manufacture a source and sink ]
+			//[Create the source]
+			MDataSource* source = MDataSource::NewSourceL(KUidMmfFileSource, fileConfig);
+			CleanupDeletePushL(source);
+
+			//[ Create the sink ]
+			MDataSink* sink = MDataSink::NewSinkL(KUidMmfAudioOutput, KNullDesC8);
+			CleanupDeletePushL(sink);
+
+			// Use a cleanup item to stop & reset the controller and so remove the data 
+			// sources/sinks automatically when this funtion leaves. This can be done
+			// before the sinks/sources are added.
+			TCleanupItem cleanupItem(CleanupController, this);
+			CleanupStack::PushL(cleanupItem);
+
+			//[ add the source ]
+			CAviPlayController::AddDataSourceL(*source);
+			 
+			//[ add the sink ]
+			CAviPlayController::AddDataSinkL(*sink);
+
+			CleanupStack::PopAndDestroy(3);// source, sink, cleanupItem
+		);
+		if( errorCode == KErrNone )
+			{
+			// [ check we have passed through all allocs in the test]
+			TAny* testAlloc = User::Alloc(1);
+			if( testAlloc == NULL )
+				{
+				reachedEnd = ETrue;
+				failCount -= 1; // Failcount of 1 equates to 0 successful allocs, etc.
+				}
+			else 
+				{
+				User::Free( testAlloc );
+				}
+
+			completed = reachedEnd || badResult;
+			
+			}
+		else if( errorCode != KErrNoMemory && errorCode != KErrCorrupt) 
+			{
+			// [ we failed for some reason other than memory
+			//     allocation, so fail the test ]
+			completed = ETrue;
+			badResult = ETrue;
+			}
+		
+		__UHEAP_SETFAIL(RHeap::ENone ,0);
+		
+		// [ exit the loop ]
+		if( completed )
+			{
+			break;
+			}
+
+		failCount +=1;
+		}
+	if(!badResult)
+		{
+		aMessage.Complete(KErrNone);
+		}
+	else
+		{
+		aMessage.Complete(errorCode);
+		}
+	}
+
+/**
+*
+* DoMemoryAllocTest2
+*
+*/
+void CCustomAviPlayController::DoPrepareMemoryAllocTest2(TMMFMessage& aMessage)
+	{
+	TRAPD(err, 
+		TMMFFileConfig fileConfig; // video file 
+		fileConfig().iPath = KTestAviFile;
+
+		// Use a cleanup item to stop & reset the controller and so remove the data 
+		// sources/sinks automatically when this funtion leaves. This can be done
+		// before the sinks/sources are added.
+		TCleanupItem cleanupItem(CleanupController, this);
+		CleanupStack::PushL(cleanupItem);
+		
+		//[ lets manufacture a source and sink ]
+		//[Create the source]
+		if (iDataSource)
+			{
+			delete iDataSource;
+			iDataSource = NULL;
+			}
+		iDataSource = MDataSource::NewSourceL(KUidMmfFileSource, fileConfig);
+		
+		//[ Create the sink ]
+		if (iDataSink)
+			{
+			delete iDataSink;
+			iDataSink = NULL;
+			}
+		iDataSink = MDataSink::NewSinkL(KUidMmfAudioOutput, KNullDesC8);
+		
+		//[ add the source ]
+		CAviPlayController::AddDataSourceL(*iDataSource);
+		 
+		//[ add the sink ]
+		CAviPlayController::AddDataSinkL(*iDataSink);
+
+		CleanupStack::Pop(1);	// cleanupItem
+		);
+	aMessage.Complete(err);
+	}
+
+/**
+*
+* DoMemoryAllocTest3
+*
+*/
+void CCustomAviPlayController::DoMemoryAllocTest3(TMMFMessage& aMessage)
+	{
+	__UHEAP_SETFAIL(RHeap::EFailNext ,iMemAllocPrimeFailCount);
+	TRAPD(err, 
+		// Use a cleanup item to stop & reset the controller and so remove the data 
+		// sources/sinks automatically when this funtion leaves. This can be done
+		// before the sinks/sources are added.
+		TCleanupItem cleanupItem(CleanupController, this);
+		CleanupStack::PushL(cleanupItem);
+		
+		//[ play ]
+		CAviPlayController::PlayL();
+		 
+		//[ stop ]
+		CAviPlayController::StopL();
+		 
+		//[ reset ]	
+		CAviPlayController::ResetL();
+
+		CleanupStack::PopAndDestroy(1);	// cleanupItem
+		);
+	__UHEAP_SETFAIL(RHeap::ENone ,0);
+	aMessage.Complete(err);
+	}
+	
+/**
+*
+* DoSetPrimetTestEnable
+*
+*/
+void CCustomAviPlayController::DoSetPrimetTestEnable(TMMFMessage& aMessage)
+	{
+	TPckgBuf<TBool> primeTestEnable;
+	TInt err = aMessage.ReadData1FromClient(primeTestEnable);
+	aMessage.Complete(err);
+	iMemAllocPrimeTestEnabled = primeTestEnable();
+	}
+
+/**
+*
+* DoSetPrimetTestFailCount
+*
+*/
+void CCustomAviPlayController::DoSetPrimetTestFailCount(TMMFMessage& aMessage)
+	{
+	TPckgBuf<TInt> primeTestFailCount;
+	TInt err = aMessage.ReadData1FromClient(primeTestFailCount);
+	aMessage.Complete(err);
+	iMemAllocPrimeFailCount = primeTestFailCount();
+	}
+