// setcritical.cpp
// 
// Copyright (c) 2008 - 2010 Accenture. All rights reserved.
// This component and the accompanying materials are made available
// under the terms of the "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:
// Accenture - Initial contribution
//

#include <fshell/memoryaccesscmd.h>

using namespace IoUtils;

class CCmdSetcritical : public CMemoryAccessCommandBase
	{
public:
	static CCommandBase* NewLC();
	~CCmdSetcritical();
private:
	CCmdSetcritical();
private: // From CCommandBase.
	virtual const TDesC& Name() const;
	virtual void DoRunL();
	virtual void ArgumentsL(RCommandArgumentList& aArguments);
	virtual void OptionsL(RCommandOptionList& aOptions);
private:
	RArray<TUint> iTids;
	TBool iSystem, iPermanent, iProcess, iCritical;
	};


CCommandBase* CCmdSetcritical::NewLC()
	{
	CCmdSetcritical* self = new(ELeave) CCmdSetcritical();
	CleanupStack::PushL(self);
	self->BaseConstructL();
	return self;
	}

CCmdSetcritical::~CCmdSetcritical()
	{
	iTids.Close();
	}

CCmdSetcritical::CCmdSetcritical()
	{
	}

const TDesC& CCmdSetcritical::Name() const
	{
	_LIT(KName, "setcritical");	
	return KName;
	}

void CCmdSetcritical::ArgumentsL(RCommandArgumentList& aArguments)
	{
	aArguments.AppendUintL(iTids, _L("threadid"));
	}

void CCmdSetcritical::OptionsL(RCommandOptionList& aOptions)
	{
	aOptions.AppendBoolL(iSystem, _L("system"));
	aOptions.AppendBoolL(iProcess, _L("process"));
	aOptions.AppendBoolL(iCritical, _L("critical"));
	aOptions.AppendBoolL(iPermanent, _L("permanent"));
	}

void CCmdSetcritical::DoRunL()
	{
	LoadMemoryAccessL();

	TUint set = 0;
	if (iSystem && iCritical)
		set = KThreadFlagSystemCritical;
	else if (iSystem && iPermanent)
		set = KThreadFlagSystemPermanent;
	else if (iProcess && iCritical)
		set = KThreadFlagProcessCritical;
	else if (iProcess && iPermanent)
		set = KThreadFlagProcessPermanent;
	else if (!iSystem && !iCritical && !iProcess && !iPermanent)
		set = 0;
	else
		{
		LeaveIfErr(KErrArgument, _L("Incompatible options specified. You must either give no options (to specify ENotCritical) or one of --system/--process and one of --critical/--permanent"));
		}

	for (TInt i = 0; i < iTids.Count(); i++)
		{
		TUint id = iTids[i];
		RThread thread;
		TInt err = iMemAccess.RThreadForceOpen(thread, id);
		if (!err) err = iMemAccess.SetThreadCriticalFlags(thread, set);
		if (err)
			{
			PrintError(err, _L("Couldn't set flags for thread %u."), id);
			}
		thread.Close();
		}
	}

EXE_BOILER_PLATE(CCmdSetcritical)