/*

* Copyright (c) 2005-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:  

*

*/

// dlltestDlg.cpp : implementation file

//

#include "stdafx.h"

#include "dlltest.h"

#include "dlltestDlg.h"

#include <utils.h>

#include <statexp.h>

#ifdef _DEBUG

#define new DEBUG_NEW

#undef THIS_FILE

static char THIS_FILE[] = __FILE__;

#endif

// function pointer prototypes

typedef	char * (*LPFN_Version)();

typedef	char * (*LPFN_GetError)(const int handle);

typedef	int (*LPFN_SetConnectionLogging)(const char * filename);

typedef	void (*LPFN_CloseConnectionLogging)();

typedef	int (*LPFN_Connect)(const STATCONNECTTYPE iConnectType, const char * pszPlatformType);

typedef	int (*LPFN_Disconnect)(const int handle);

typedef	int (*LPFN_SetCommandDelay)(const int handle, const unsigned int iMillisecondDelay);

typedef	int (*LPFN_SetCommandLogging)(const int handle, const char * pszLogPath, HWND messageWindow, const STATLOGLEVEL iLevel, const bool bAppend);

typedef	int (*LPFN_SetImageVerification)(const int handle, const char * pszRefDir, const bool bRemoveOldImages, const int iFactor);

typedef	int (*LPFN_SendRawCommand)(const int handle, const char * pszText);

typedef	int (*LPFN_SendCommandFile)(const int handle, const char * pszFile);

typedef	int (*LPFN_GetSnapshot)(const int handle, TBitmapFileHeader **ppFile, TBitmapInfoHeader **ppBitmap, char **ppData, unsigned long *pSize);

// initialise pointers

LPFN_Version pfn_Version = NULL;

LPFN_GetError pfn_GetError = NULL;

LPFN_SetConnectionLogging pfn_SetConnectionLogging = NULL;

LPFN_CloseConnectionLogging pfn_CloseConnectionLogging = NULL;

LPFN_Connect pfn_Connect = NULL;

LPFN_Disconnect pfn_Disconnect = NULL;

LPFN_SetCommandDelay pfn_SetCommandDelay = NULL;

LPFN_SetCommandLogging pfn_SetCommandLogging = NULL;

LPFN_SetImageVerification pfn_SetImageVerification = NULL;

LPFN_SendRawCommand pfn_SendRawCommand = NULL;

LPFN_SendCommandFile pfn_SendCommandFile = NULL;

LPFN_GetSnapshot pfn_GetSnapshot = NULL;

//////////////////////////////////////////////////////////////////////////

// This is the main thread function which listens on a socket and

// processes any incoming scripts

DWORD WINAPI

ThreadProc(LPVOID lpParameter)

{

	CDlltestDlg *pWorker = (CDlltestDlg *)lpParameter;

	pWorker->Run();

	return 1;

}

/////////////////////////////////////////////////////////////////////////////

// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog

{

public:

	CAboutDlg();

// Dialog Data

	//{{AFX_DATA(CAboutDlg)

	enum { IDD = IDD_ABOUTBOX };

	//}}AFX_DATA

	// ClassWizard generated virtual function overrides

	//{{AFX_VIRTUAL(CAboutDlg)

	protected:

	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support

	//}}AFX_VIRTUAL

// Implementation

protected:

	//{{AFX_MSG(CAboutDlg)

	//}}AFX_MSG

	DECLARE_MESSAGE_MAP()

};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)

{

	//{{AFX_DATA_INIT(CAboutDlg)

	//}}AFX_DATA_INIT

}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)

{

	CDialog::DoDataExchange(pDX);

	//{{AFX_DATA_MAP(CAboutDlg)

	//}}AFX_DATA_MAP

}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)

	//{{AFX_MSG_MAP(CAboutDlg)

		// No message handlers

	//}}AFX_MSG_MAP

END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////

// CDlltestDlg dialog

CDlltestDlg::CDlltestDlg(CWnd* pParent /*=NULL*/)

	: CDialog(CDlltestDlg::IDD, pParent)

{

	//{{AFX_DATA_INIT(CDlltestDlg)

		// NOTE: the ClassWizard will add member initialization here

	//}}AFX_DATA_INIT

	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);

}

void CDlltestDlg::DoDataExchange(CDataExchange* pDX)

{

	CDialog::DoDataExchange(pDX);

	//{{AFX_DATA_MAP(CDlltestDlg)

	DDX_Control(pDX, IDC_OVERWRITE, m_RemoveImages);

	DDX_Control(pDX, IDC_RUN, m_Run);

	DDX_Control(pDX, IDC_CHKLOGTOFILE, m_ChkLogtofile);

	DDX_Control(pDX, IDC_BROWSESNAPSHOT, m_BrowseSnapshot);

	DDX_Control(pDX, IDC_BROWSEREFIMAGES, m_BrowseRefImages);

	DDX_Control(pDX, IDC_BROWSELOG, m_BrowseLog);

	DDX_Control(pDX, IDC_BROWSECMDFILE, m_BrowseCmdFile);

	DDX_Control(pDX, IDC_CHECKCMDFILE, m_CheckCmdFile);

	DDX_Control(pDX, IDC_CHECKVERIFICATION, m_CheckVerification);

	DDX_Control(pDX, IDC_CHECKSNAPSHOT, m_CheckSnapshot);

	DDX_Control(pDX, IDC_CHECKRAWCMD, m_CheckRawCmd);

	DDX_Control(pDX, IDC_ADDRESS, m_Address);

	DDX_Control(pDX, IDC_CONNECTION, m_Connection);

	DDX_Control(pDX, IDC_PROGRESS, m_Progress);

	DDX_Control(pDX, IDC_SNAPSHOT, m_Snapshot);

	DDX_Control(pDX, IDC_DELAY, m_Delay);

	DDX_Control(pDX, IDC_FUDGE, m_Fudge);

	DDX_Control(pDX, IDC_REFDIR, m_RefDir);

	DDX_Control(pDX, IDC_RAWCMD, m_RawCmd);

	DDX_Control(pDX, IDC_LOGFILE, m_LogFile);

	DDX_Control(pDX, IDC_CMDFILE, m_CmdFile);

	DDX_Control(pDX, IDC_ITERATIONS, m_Iterations);

	DDX_Control(pDX, IDC_PLATFORM, m_Platform);

	DDX_Control(pDX, IDC_LINK, m_Link);

	//}}AFX_DATA_MAP

}

BEGIN_MESSAGE_MAP(CDlltestDlg, CDialog)

	//{{AFX_MSG_MAP(CDlltestDlg)

	ON_WM_SYSCOMMAND()

	ON_WM_PAINT()

	ON_WM_QUERYDRAGICON()

	ON_BN_CLICKED(IDC_RUN, OnRun)

	ON_CBN_SELCHANGE(IDC_CONNECTION, OnChangeConnection)

	ON_BN_CLICKED(IDC_RESETSETTINGS, OnResetsettings)

	ON_BN_CLICKED(IDC_RESETTESTS, OnResettests)

	ON_BN_CLICKED(IDC_BROWSELOG, OnBrowselog)

	ON_BN_CLICKED(IDC_CHECKCMDFILE, OnCheckcmdfile)

	ON_BN_CLICKED(IDC_CHECKRAWCMD, OnCheckrawcmd)

	ON_BN_CLICKED(IDC_CHECKSNAPSHOT, OnChecksnapshot)

	ON_BN_CLICKED(IDC_CHECKVERIFICATION, OnCheckverification)

	ON_BN_CLICKED(IDC_BROWSECMDFILE, OnBrowsecmdfile)

	ON_BN_CLICKED(IDC_BROWSESNAPSHOT, OnBrowsesnapshot)

	ON_BN_CLICKED(IDC_BROWSEREFIMAGES, OnBrowserefimages)

	ON_BN_CLICKED(IDC_CHKLOGTOFILE, OnChklogtofile)

	//}}AFX_MSG_MAP

END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////

// CDlltestDlg message handlers

BOOL CDlltestDlg::OnInitDialog()

{

	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.

	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);

	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);

	if (pSysMenu != NULL)

	{

		CString strAboutMenu;

		strAboutMenu.LoadString(IDS_ABOUTBOX);

		if (!strAboutMenu.IsEmpty())

		{

			pSysMenu->AppendMenu(MF_SEPARATOR);

			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);

		}

	}

	SetIcon(m_hIcon, TRUE);			// Set big icon

	SetIcon(m_hIcon, FALSE);		// Set small icon

	m_Snapshot.SetLimitText(sizeof(szBuffer));

	m_RefDir.SetLimitText(sizeof(szBuffer));

	m_RawCmd.SetLimitText(sizeof(szBuffer));

	m_LogFile.SetLimitText(sizeof(szBuffer));

	m_CmdFile.SetLimitText(sizeof(szBuffer));

	// initialise our settings

	OnResetsettings();

	OnResettests();

	ReadSettings();

	return TRUE;  // return TRUE  unless you set the focus to a control

}

//////////////////////////////////////////////////////////////////////////////////////

void CDlltestDlg::OnSysCommand(UINT nID, LPARAM lParam)

{

	if ((nID & 0xFFF0) == IDM_ABOUTBOX)

	{

		CAboutDlg dlgAbout;

		dlgAbout.DoModal();

	}

	else

	{

		CDialog::OnSysCommand(nID, lParam);

	}

}

//////////////////////////////////////////////////////////////////////////////////////

// If you add a minimize button to your dialog, you will need the code below

//  to draw the icon.  For MFC applications using the document/view model,

//  this is automatically done for you by the framework.

void CDlltestDlg::OnPaint() 

{

	if (IsIconic())

	{

		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle

		int cxIcon = GetSystemMetrics(SM_CXICON);

		int cyIcon = GetSystemMetrics(SM_CYICON);

		CRect rect;

		GetClientRect(&rect);

		int x = (rect.Width() - cxIcon + 1) / 2;

		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon

		dc.DrawIcon(x, y, m_hIcon);

	}

	else

	{

		CDialog::OnPaint();

	}

}

//////////////////////////////////////////////////////////////////////////////////////

HCURSOR CDlltestDlg::OnQueryDragIcon()

{

	return (HCURSOR) m_hIcon;

}

//////////////////////////////////////////////////////////////////////////////////////

void CDlltestDlg::OnRun() 

{

	if (bRunning)

	{

		if (IDYES == AfxMessageBox(_T("The work task has not completed its current job and stopping it now may cause it to be left in an unstable state.  Do you wish to abort the current task?"), MB_ICONWARNING | MB_YESNO | MB_DEFBUTTON2))

		{

			TerminateThread(hThreadHandle, (unsigned long)-1);

			EnableFields(TRUE);

		}

	}

	else

	{

		//call Run to run through the script (with no appending to log file in this case) - spawn a thread...

		DWORD dwThreadID;

		if (hThreadHandle = CreateThread( NULL,					// security attributes

												0,				// stack size

												ThreadProc,		// proc to call

												this,			// proc parameter

												0,				// creation flags

												&dwThreadID))	// thread identifier

		{

			EnableFields(FALSE);

			MSG msg;

			while(WaitForSingleObject(hThreadHandle, STAT_THREAD_SLEEP_TIME) == WAIT_TIMEOUT)

			{

				PeekMessage(&msg, m_hWnd, 0, 0, PM_REMOVE);

				DispatchMessage(&msg);

			}

			EnableFields(TRUE);

		}

	}

}

//////////////////////////////////////////////////////////////////////////////////////

void CDlltestDlg::Run()

{

	struct TBitmapFileHeader *pFile = NULL;

	struct TBitmapInfoHeader *pInfo = NULL;

	char *pData = NULL;

	char szPrevious[10] = {0};

	char szRemaining[10] = {0};

	unsigned long lSize = 0;

	bErrors = false;

	m_Progress.SetRange(0, 100);

	m_Progress.SetPos(0);

	m_Progress.SetStep(GetIncrement());

	CString logFile;		// We use this file name throughout the

							// function and we should save it.

	CString logFilePath;	// Also, we have a bit of a mixture

							// of calls to make to set up logging

							// and some require a file name (and assume a

							// folder) and some require a full path.

							// Generate a full path (based on the file name

							// above) and use whichever variable suits

							// each function call.

	if (m_ChkLogtofile.GetCheck())

	{

		m_LogFile.GetWindowText(logFile);

		GetLogFilePath(logFile,logFilePath);

	}

	// connect to runtime dll

	if (m_Link.GetCurSel() == 1)

	{

		// display version

		MessageBox(Version(), "STAT DLL Version", MB_OK);

		CWaitCursor oWait;

		// log connections

		if (m_ChkLogtofile.GetCheck())

		{

			if (0 != logFilePath.GetLength())

			{

				SetConnectionLogging(logFilePath);

			}

		}

		// get the connect string

		switch(m_Connection.GetCurSel())

		{

		case 1:	// serial

		case 2:	// infra-red

		case 3:	// bluetooth

		case 4:	// USB

			m_Platform.GetLBText(m_Platform.GetCurSel(), szBuffer);

			break;

		case 0:	// socket

			m_Address.GetWindowText(szBuffer, sizeof(szBuffer));

			break;

		}

		// connect

		Evaluate(h = Connect((STATCONNECTTYPE)(m_Connection.GetCurSel() + 1), szBuffer, NULL, NULL), h);

		if (h)

		{

			m_Progress.StepIt();

			UpdateWindow();

			// log commands

			if (m_ChkLogtofile.GetCheck())

			{

				if (0 != logFile.GetLength())

					Evaluate(SetCommandLogging(h, logFile, NULL, EVerbose, true, NULL, NULL), h);

			}

			if (!bErrors)

			{

				// set command delay

				m_Delay.GetWindowText(szBuffer, sizeof(szBuffer));

				iValue = atoi(szBuffer);

				Evaluate(SetCommandDelay(h, iValue), h);

			}

			if (!bErrors)

			{

				// set image verification

				if (m_CheckVerification.GetCheck())

				{

					m_Fudge.GetWindowText(szBuffer, sizeof(szBuffer));

					iValue = atoi(szBuffer);

					m_RefDir.GetWindowText(szBuffer, sizeof(szBuffer));

					if (*szBuffer)

						Evaluate(SetImageVerification(h, szBuffer, m_RemoveImages.GetCheck()?true:false, iValue), h);

				}

			}

			m_Iterations.GetWindowText(szPrevious, sizeof(szPrevious));

			int iMax = atoi(szPrevious);

			for (int i=0;i<iMax;i++)

			{

				if (!bErrors)

				{

					// execute raw command

					if (m_CheckRawCmd.GetCheck())

					{

						m_RawCmd.GetWindowText(szBuffer, sizeof(szBuffer));

						if (*szBuffer)

						{

							Evaluate(SendRawCommand(h, szBuffer), h);

							m_Progress.StepIt();

							UpdateWindow();

						}

					}

				}

				if (!bErrors)

				{

					// execute command file

					if (m_CheckCmdFile.GetCheck())

					{

						m_CmdFile.GetWindowText(szBuffer, sizeof(szBuffer));

						if (*szBuffer)

						{

							Evaluate(SendCommandFile(h, szBuffer), h);

							m_Progress.StepIt();

							UpdateWindow();

						}

					}

				}

				if (!bErrors)

				{

					// get a snapshot

					if (m_CheckSnapshot.GetCheck())

					{

						m_Snapshot.GetWindowText(szBuffer, sizeof(szBuffer));

						if (*szBuffer)

						{

							DeleteFile(szBuffer);

							Evaluate(GetSnapshot(h, &pFile, &pInfo, &pData, &lSize), h);

							if (!bErrors)

								if (!WriteBitmap(szBuffer, pFile, pInfo, pData, lSize))

									bErrors = true;

							m_Progress.StepIt();

							UpdateWindow();

						}

					}

				}

				// decrement the iteration count

				itoa(iMax - i - 1, szRemaining, 10);

				m_Iterations.SetWindowText(szRemaining);

				UpdateWindow();

			}

			// disconnect

			Evaluate(Disconnect(h), h);

			m_Progress.SetPos(100);

			m_Iterations.SetWindowText(szPrevious);

		}

		if (m_ChkLogtofile.GetCheck())

			CloseConnectionLogging();

	}

	else

	{

		// connect to loaded dll

#if defined _DEBUG

		HINSTANCE inst = LoadLibrary("statd.dll");

#else // defined _DEBUG

		HINSTANCE inst = LoadLibrary("stat.dll");

#endif // defined _DEBUG

		if (inst && LoadFunctionEntrypoints(inst))

		{

			// display version

			MessageBox((*pfn_Version)(), "STAT DLL Version", MB_OK);

			CWaitCursor oWait;

			// log connections

			if (m_ChkLogtofile.GetCheck())

			{

				if (0 != logFilePath.GetLength())

				{

					(*pfn_SetConnectionLogging)(logFilePath);

				}

			}

			// get the connect string

			switch(m_Connection.GetCurSel())

			{

			case 1:	// serial

			case 2:	// infra-red

			case 3:	// bluetooth

			case 4:	// USB

				m_Platform.GetLBText(m_Platform.GetCurSel(), szBuffer);

				break;

			case 0:	// socket

				m_Address.GetWindowText(szBuffer, sizeof(szBuffer));

				break;

			}

			// connect