// 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 "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:
// UCC service to launch wireshark on remote PC.
// This allows controlled packet capture during each send receive test
// run by Netperf. 
// 
//

/**
 @file
 @internalTechnology
*/

#include <string>
#include <stdio.h>
#include "pcapService.h"

using std::string;

#define CONSOLE_NAME "Console - "
#define SERVICE_NAME "PcapService"
#define PCAPSVC_CFG   "PcapService.cfg"

CService* Service() {return CPcapService::Instance();}

// static data storage
CPcapService* CPcapService::iInstance = NULL;

/*static*/ CPcapService* CPcapService::Instance()
{
	if (iInstance==NULL)
	{
		iInstance = new CPcapService;
	}
	return iInstance;
}

CPcapService::~CPcapService()
{
	iInstance = NULL;
}

bool CPcapService::Setup()
{
	// quick config file read..
	FILE* cfgin = fopen(PCAPSVC_CFG,"r");
	if (!cfgin)
	{
		printf("Error reading config file PcapService.cfg\n");
		return false;
	}

	char line[200];
	while (!feof(cfgin))
	{
		*line = '\0';
		fgets(line, 200, cfgin);
		char* eolCh = line + strlen(line);
		while (*--eolCh == '\r' || *eolCh == '\n') // strip end of line chars
			*eolCh='\0';

		if (line == strstr(line,"Device="))
		{
			iDefaultDevice = line+7;
			printf("Device: [%s]\n",iDefaultDevice.c_str());
		}
		else if (line == strstr(line, "WiresharkPath="))
		{
			iWiresharkPath = line+14;
			printf("Wireshark Path: [%s]\n", iWiresharkPath.c_str());
		}
		else
		{
			printf("Config line ignored: %s\n", line);
		}
	}

	// check read configuration
	if (iDefaultDevice.empty() || iWiresharkPath.empty())
	{
		printf("Not all required parameters found in config file PcapService.cfg\n");
		return false;
	}

	fclose(cfgin);
	return true;
}


// entry point for received commands
int CPcapService::RunCommand(const CCall &aCall)
{
	int result = NOERROR;
	aCall.Dump();

	// handle call
	int callId;
	if (aCall.CallID(callId))
	{
		switch (callId)
		{
		case 1: // "StartServer"
		{
			// get & check args
			string outputFileName = "out.pcap";
			if (!aCall.Get("outputFileName", outputFileName))
			{
				printf("missing or invalid parameters; \"cmd\", \"args\"");
				return ERR_MISSING_PARAMETER;
			}

			string Device = iDefaultDevice;
			aCall.Get("device", Device);

			// issue cmd
			char args[256];
			sprintf(args, "-i \"%s\" -w \"%s\" -F libpcap",  Device.c_str(), outputFileName.c_str());
			puts(args);
			CProcess *process = new CProcess(iWiresharkPath, args, "");
			if (!process)
				return ERR_CANNOT_CREATE_NEW_INSTANCE;

			bool isDefault = true;
			if (!iProcesses.Add(*process, isDefault))
				return ERR_INVALIDSTATE;

			if (!process->Start())
				return ERR_CANNOT_CREATE_NEW_INSTANCE;

			result = (int) process;
			break;
		}
		case 2: // "Stop"
		{
			// issue cmd - stop & destruct operations have been combined into 1 client side request (for sake of simplicity)
		   // iProcesses.Remove(aCall);
			iProcesses.Remove(false); // remove default
			break;
		} 
		default:
			result = ERR_INVALID_CALL;
		}
	}
	else
		result = ERR_INVALID_CALL;

	return result;
}