//----------------------------------------------------------------------------
//
//  Copyright (C) 2008 Intel Corporation
//
//  File:       ACLSample.cpp 
//
//  Contents:   Sample code for an Intel� AMT Network client.
//
//----------------------------------------------------------------------------

#include "ACLFlow.h"
#include "CommonDefinitions.h"
#include "CmdLineArguments.h"
#include "CimOpenWsmanClient.h"		// Include from CIM Framework
#include "LogicException.h"

#include <Windows.h>

using namespace std;
using namespace Intel::Manageability::Exceptions;
using namespace Intel::WSManagement;
using namespace ExceptionNamespace;


int main(int argc, char* argv[])
{
	// set default dll lookup directory to system
	SetDefaultDllDirectories(LOAD_LIBRARY_SEARCH_SYSTEM32);

	PT_STATUS status = PT_STATUS_SUCCESS;

	CmdLineArguments::Format format;
	// Add the general arguments
	format.AddArg(CMD_VERBOSE, false, false, false, "");
	format.AddArg(CMD_USER, true, false, false, "");
	format.AddArg(CMD_PASS, true, false, false, "");
	format.AddArg(CMD_CERT_NAME, true, false, false, "");
	format.AddArg(CMD_HOST, true, true, false, "");
	format.AddArg(CMD_TLS, false, false, false, "");
	format.AddArg(CMD_KRB, false, false, false, "");
	format.AddArg(CMD_PROXY, true, false, false, "");
	format.AddArg(CMD_PROXY_USERNAME, true, false, false, "");
	format.AddArg(CMD_PROXY_PASS, true, false, false, "");
	format.AddArg(CMD_ACCEPT_SELF_CIGNED_CERT, false, false, false, "");

	if (argc == 1)
	{
		cout << format.CreateUsage(argv[0]);
		return status;
	}

	try
	{
		CmdLineArguments cmdLineArguments;
		cmdLineArguments.Parse(argc, argv, format, false);
		CimOpenWsmanClient wsmanClient(	cmdLineArguments.GetArgValue(CMD_HOST),
			cmdLineArguments.ArgExists(CMD_TLS) ? AMT_SECURE_PORT : AMT_UNSECURE_PORT,
			cmdLineArguments.ArgExists(CMD_TLS),	
			cmdLineArguments.ArgExists(CMD_KRB) ? KERBEROS : DIGEST,
			cmdLineArguments.GetArgValue(CMD_USER),
			cmdLineArguments.GetArgValue(CMD_PASS),
			cmdLineArguments.ArgExists(CMD_PROXY) ? cmdLineArguments.GetArgValue(CMD_PROXY): "",
			cmdLineArguments.ArgExists(CMD_PROXY_USERNAME) ? cmdLineArguments.GetArgValue(CMD_PROXY_USERNAME): "",
			cmdLineArguments.ArgExists(CMD_PROXY_PASS) ? cmdLineArguments.GetArgValue(CMD_PROXY_PASS): "",
			false,	// Local certificate store
			cmdLineArguments.ArgExists(CMD_CERT_NAME) ? cmdLineArguments.GetArgValue(CMD_CERT_NAME): "", 
			cmdLineArguments.ArgExists(CMD_LOCAL) ? OID_LOCAL : OID_REMOTE,
			cmdLineArguments.ArgExists(CMD_ACCEPT_SELF_CIGNED_CERT));


		bool verbose = cmdLineArguments.ArgExists(CMD_VERBOSE);
		string userRequest = cmdLineArguments.GetSelectedOptionName();

		int numOfAccounts = 0;

		EnumerateACLAccounts(&wsmanClient, numOfAccounts, verbose);
		GetACLAccounts(&wsmanClient, verbose);
		if (numOfAccounts < MAX_ACCOUNTS)
		{
			CreateACLAccount(&wsmanClient, verbose);				
			DeleteACLAccount(&wsmanClient, verbose);
		}
		else
		{
			cout << "\nWarning: Skipping the creation and deletion of an account since there are" << endl << "\t already maximum accounts." << endl;
		}
	}
	catch (CimException& exp)
	{
		cerr << endl << "--------------------CimException--------------------" << endl;
		cerr << exp.what() << endl;;
		status = PTSDK_STATUS_NETWORK_ERROR;
	}
	catch (ConversionException &exp)
	{
		cout << endl << "--------------------ConversionException--------------------" << endl;
		cout << exp.what() << endl;;
		status = PTSDK_STATUS_NETWORK_ERROR;
	}
	catch (CmdLineException &exp)
	{
		cout << endl << "--------------------CmdLineException--------------------" << endl;
		cout << exp.what() << endl;;
		status = PTSDK_STATUS_NETWORK_ERROR;
		cout << endl << format.CreateUsage(argv[0]);
	}
	catch (WSManException &exp)
	{
		cout << endl << "--------------------GeneralWsmanException--------------------" << endl;
		cout << "Exception string error: " << exp.what() << endl;
		status = PTSDK_STATUS_NETWORK_ERROR;
	}
	catch (HTTPException &exp)
	{
		cout << endl << "--------------------HTTPException--------------------" << endl;
		cout << "Exception string error: " << exp.what() << endl;
		status = PTSDK_STATUS_NETWORK_ERROR;
	}
	catch (LogicException &exp)
	{
		cerr << endl << "--------------------LogicException--------------------" << endl;
		cerr << exp.getDetail();
		if(exp.getErr() != 0)
			cerr << exp.getErr();
	}
	catch (const exception &exp)
	{
		cout << endl << "--------------------std::exception--------------------" << endl;
		cout << exp.what() << endl;;
		status = PTSDK_STATUS_NETWORK_ERROR;
	}	
	return status;
}