/*

* 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:  Defines the entry point for the console application.

*

*/

// Includes.

#include "../inc/ATCommonDefines.h"

#include "../inc/CATDatParser.h"

#include "../inc/CATParseTraceFile.h"

#include "../inc/CATProject.h"

//Debug logging parameters

//To enable debugging as argument to atool.exe use:

//--show_debug / -show_debug : Normal logging in console.

//--show_debug_all / -show_debug_all : Log all (low level functions also) to console.

//--show_dbgview / -show_dbgview : Normal logging to windows debug messages.

//--show_dbgview_all / -show_dbgview_all : Log all (low level functions also) to windows debug messages.

//Return codes (errorlevel) defined in AT_RETURN_CODE structure see ATCommonDefines.h.

extern bool g_bDebugConsole = false;

extern bool g_bDebugDbgView = false;

extern bool g_bDebugLowLevel = false;

//Argument parsing.

extern bool parseBaseArguments( vector<string>& vArgs, ARGUMENTS& args );

extern bool parseHookArguments( vector<string>& vArgs, ARGUMENTS& args );

extern bool parseOldHookArguments( vector<string>& vArgs, ARGUMENTS& args );

extern bool parseAnalyzeArguments( vector<string>& vArgs, ARGUMENTS& args );

extern bool parseHtiArguments( vector<string>& vArgs, ARGUMENTS& args );

extern bool parseParseArguments( vector<string>& vArgs, ARGUMENTS& args );

//Helps.

extern void print_help( void );

extern void print_syntax_examples( void );

//AT Library check functions

extern bool CheckATLibrariesArmv5( string sEpocRoot );

extern bool CheckATLibrariesArmv5Abiv2( string sEpocRoot );

extern bool CheckATLibrariesWinscw( string sEpocRoot );

//Hti functions.

#ifdef HTI_SUPPORT

extern int htiGetReport( int argc, _TCHAR* argv[] );

extern int htiDeleteFile( int argc, _TCHAR* argv[] );

#endif

//CLE version functions.

extern int showVersionInfo( void );

//Miscelllanaeous functions.

extern bool CheckSBS2Folder( void );

const char DEBUG_PARAMETER_CONSOLE[] = "-debug";

const char DEBUG_PARAMETER_DBGVIEW[] = "-dbgview";

const char SBS2_PARAMETER[] = "-sbs2";

char g_cCurrentDir[MAX_LINE_LENGTH];

//Global compile class objects are neededif ctrl+c is pressed mmp file must be restored.

CATProject project_module;

//Parse object.

CATParseTraceFile Parser;

/**

* Handle process control signals.

*/

BOOL WINAPI HandlerRoutine( DWORD dwCtrlType )

{

	//Run recovery and exit for project if user presses ctrl+c

	//or close signal is received.

	if( dwCtrlType == CTRL_C_EVENT || dwCtrlType == CTRL_CLOSE_EVENT )

	{

		int iMode = project_module.GetMode();

		if ( iMode == CATProject::COMPILE

			|| iMode == CATProject::INSTRUMENT

			|| iMode == CATProject::INSTRUMENT_CONSOLE )

			project_module.RunRecoveryAndExit();

	}

	//Return false so program execution is stopped.

	return false;

}

// TESTING

int _tmain( int argc, _TCHAR* argv[] )

{

	#ifdef MEM_LEAK_CHECK

	_CrtSetDbgFlag ( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF );

	#endif

	try {

		//Set function to handle process control signals.

		SetConsoleCtrlHandler( HandlerRoutine, true );

		vector<string> vArguments;

		for( int i = 1 ; i < argc ; i++ )

		{

			vArguments.push_back( argv[i] );

		}

		//Help variables.

		string sEpocRoot("");

		//Parse base arguments.

		ARGUMENTS args;

		parseBaseArguments( vArguments, args );

		// Debugging messages.

		if ( args.bDebugConsole == true )

			g_bDebugConsole = true;

		if ( args.bDebugDbgView == true )

			g_bDebugDbgView = true;

		if ( args.bDebugLowLevel == true )

			g_bDebugLowLevel = true;

		//According to main switch parse rest arguments.

		switch( args.eMainSwitch )

		{

		case SWITCH_UNKNOWN:

			print_help();

			return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

		case SWITCH_ANALYZE:

			if ( ! parseAnalyzeArguments( vArguments, args ) )

				return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

			// Get epocroot

			if( ! CATBase::GetEpocRoot( sEpocRoot ) )

				return AT_RETURN_CODE::CANNOT_FIND_EPOCROOT;

			project_module.SetEpocRoot( sEpocRoot );

			// project not uninstrumented run it first.

			if ( ! project_module.IsUninstrumented() )

			{

				project_module.SetMode( CATProject::UNINSTRUMENT_CONSOLE );

				project_module.Run();

			}

			// Set mode.

			project_module.SetMode( CATProject::ANALYZE );

			project_module.SetLogLevel( args.ANALYZE.iLoggingLevel );

			project_module.SetDataFile( args.ANALYZE.sDataFile );

			if ( args.ANALYZE.bSymbolFile )

				project_module.SetRomSymbolFiles( args.ANALYZE.vSymbolFiles );

			project_module.SetDataFileOutput( args.ANALYZE.sOutputFile);

			return project_module.Run();

		case SWITCH_HOOK:

			// Parse arguments related to hooking.

			if ( ! parseHookArguments( vArguments, args ) )

				return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

			// Set variables for project.

			if ( ! project_module.SetArguments( args ) )

				return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

			// Get epocroot

			if( ! CATBase::GetEpocRoot( sEpocRoot ) )

				return AT_RETURN_CODE::CANNOT_FIND_EPOCROOT;

			project_module.SetEpocRoot( sEpocRoot );

			// Check AnalyzeTool libraries

			if ( _stricmp( args.HOOK.sPlatform.c_str(), "winscw") == 0 )

			{

				// Emulator winscw platform

				if ( ! CheckATLibrariesWinscw(sEpocRoot) )

					return AT_RETURN_CODE::AT_LIBS_MISSING;

			}

			else

			{

				// Armv5

				if ( args.HOOK.iBuildSystem == 2 )

				{

					// Abiv2

					if ( ! CheckATLibrariesArmv5Abiv2(sEpocRoot) )

						return AT_RETURN_CODE::AT_LIBS_MISSING;

				}

				else

				{

					// Abiv1

					if( ! CheckATLibrariesArmv5(sEpocRoot) )

						return AT_RETURN_CODE::AT_LIBS_MISSING;

				}

			}

			// Run hooking.

			return project_module.Run();

			//Uninstrument

		case SWITCH_UNHOOK:

			// Set variables for project.

			if ( ! project_module.SetArguments( args ) )

				return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

			// Get epocroot

			if( ! CATBase::GetEpocRoot( sEpocRoot ) )

				return AT_RETURN_CODE::CANNOT_FIND_EPOCROOT;

			project_module.SetEpocRoot( sEpocRoot );

			return project_module.Run();

			//Support for old extensions.

		case SWITCH_OLD_HOOK:

			// Parse old style arguments related to hooking.

			if ( ! parseOldHookArguments( vArguments, args ) )

				return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

			// Set mode for project.

			if( args.eHookSwitch == HOOK_OLD_EXTENSION_INSTRUMENT )

				project_module.SetMode( CATProject::INSTRUMENT );

			else if( args.eHookSwitch == HOOK_OLD_EXTENSION_UNINSTRUMENT )

				project_module.SetMode( CATProject::UNINSTRUMENT );

			else if( args.eHookSwitch == HOOK_OLD_EXTENSION_FAILED )

				project_module.SetMode( CATProject::UNINSTRUMENT_FAILED );

			else

				return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

			// Get epocroot

			if( ! CATBase::GetEpocRoot( sEpocRoot ) )

				return AT_RETURN_CODE::CANNOT_FIND_EPOCROOT;

			project_module.SetEpocRoot( sEpocRoot );

			// Check AnalyzeTool libraries

			if ( _stricmp( args.HOOK.sPlatform.c_str(), "winscw") == 0 )

			{

				// Emulator winscw platform

				if ( ! CheckATLibrariesWinscw(sEpocRoot) )

					return AT_RETURN_CODE::AT_LIBS_MISSING;

			}

			else

			{

				// Armv5

				if ( args.bEnableSbs2 == true )

				{

					// Abiv2

					if ( ! CheckATLibrariesArmv5Abiv2(sEpocRoot) )

						return AT_RETURN_CODE::AT_LIBS_MISSING;

				}

				else

				{

					// Abiv1

					if( ! CheckATLibrariesArmv5(sEpocRoot) )

						return AT_RETURN_CODE::AT_LIBS_MISSING;

				}

			}

			// In uninstrument we check folder otherwise the switch for sbs version.

			if ( project_module.GetMode() == CATProject::UNINSTRUMENT )

			{

					if ( CheckSBS2Folder() )

						project_module.SetBuildSystem( CATProject::SBS_V2 );

					else

						project_module.SetBuildSystem( CATProject::SBS_V1 );

			}

			else

			{

				if ( args.bEnableSbs2 == true )

					project_module.SetBuildSystem( CATProject::SBS_V2 );

				else

					project_module.SetBuildSystem( CATProject::SBS_V1 );

			}

			// platform

			project_module.SetPlatform( args.HOOK.sPlatform );

			// build type

			if ( !_stricmp( args.HOOK.sBuildType.c_str(), "urel" ) )

				project_module.SetBuildType( CATProject::UREL );

			else if ( !_stricmp( args.HOOK.sBuildType.c_str(), "udeb" ) )

				project_module.SetBuildType( CATProject::UDEB );

			// Variant

			project_module.SetVariant( args.HOOK.sFeatureVariant );

			// If selected modules to be compiled

			project_module.SetTargetModules( args.HOOK.vTargetPrograms );

			// S60 data file name

			if ( args.HOOK.bDataFileName )

				project_module.SetS60FileName( args.HOOK.sDataFileName );

			// logging mode

			if ( args.HOOK.iLoggingMode == 2 )

				project_module.SetLoggingMode( CATProject::S60 );

			else

				project_module.SetLoggingMode( CATProject::XTI );

			// run

			return project_module.Run();

		case SWITCH_VERSION:

			return showVersionInfo();

		case SWITCH_CLEAN:

			project_module.SetMode( CATProject::CLEAN );

			if ( CheckSBS2Folder() )

				project_module.SetBuildSystem( CATProject::SBS_V2 );

			else

				project_module.SetBuildSystem( CATProject::SBS_V1 );

			return project_module.Run();

		case SWITCH_PARSE_TRACE:

			if ( ! parseParseArguments( vArguments, args ) )

				return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

			if (  CATBase::IsDataFile( args.PARSE.sDataFile ) )

			{

				cout << AT_MSG << "Error, " << args.PARSE.sDataFile << " is already parsed." << endl;

				return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

			}

			if ( args.PARSE.bOutputFile )

			{

				//Save data with name in arguments[3]

				Parser.StartParse( args.PARSE.sDataFile.c_str(), args.PARSE.sOutputFile.c_str() );

			}

			else

			{

				Parser.StartParse( args.PARSE.sDataFile.c_str(), NULL );

			}

			return AT_RETURN_CODE::OK;

		#ifdef HTI_SUPPORT

		case SWITCH_HTI_DELETE:

			return htiDeleteFile( argc, argv );

		case SWITCH_HTI_GET:

			return htiGetReport( argc, argv );

		#endif

		case SWITCH_HELP:

			print_help();

			print_syntax_examples();

			return AT_RETURN_CODE::OK;

		default:

			cout << AT_MSG << "Invalid parameters." << endl;

			return AT_RETURN_CODE::INVALID_ARGUMENT_ERROR;

		}

	} catch(...)

	{

		cout << AT_MSG << "Error, unhandled exception." << endl;

		return AT_RETURN_CODE::UNHANDLED_EXCEPTION;

	}

}

//EOF