// Copyright (c) 2001-2009 Nokia Corporation and/or its subsidiary(-ies).

// All rights reserved.

// This component and the accompanying materials are made available

// under the terms of the License "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:

// UsbcvControl.cpp: implementation of the UsbcvControl class.

//

#include "stdafx.h"

#include "t_usb_win.h"

#include "UsbcvControl.h"

#include "usbio.h"											// USBIO Dev Kit

#include "global.h"

#ifdef _DEBUG

#undef THIS_FILE

static char THIS_FILE[]=__FILE__;

#define new DEBUG_NEW

#endif

extern void PrintOut(BOOL screenFlag, BOOL logFlag, BOOL timeFlag, const char *format, ...);

extern BOOL gAbort;

// define the CreateProcess strings for the Usbcv test app

#define APPNAME "C:\\Program Files\\USB-IF Test Suite\\USBCommandVerifier\\UsbCV13.exe"

#define APPTITLE "Automated USBCV"

#define CMDLINE " /title \"Automated USBCV\" /drv hcdriver"

#define CURDIR "C:\\Program Files\\USB-IF Test Suite\\USBCommandVerifier\\lib"

#define DIALOG_CLASS "#32770"

#define BUTTON_CLASS "Button"

#define STATIC_CLASS "Static"

#define ID_COMPLIANCE 0x3EC

#define ID_RUN 0x3F5

#define ID_EXIT 0x1

#define DEVLIST_TITLE ""

#define ID_OKBUTTON 0x3E8

#define ID_TESTLIST 0x3E9

#define ID_DEVLIST 0x3EA

#define ID_TEXTBOX 0x3EB

#define ID_CONFIRM 0x3EC

#define ID_ABORT 0x3ED

#define TESTRESULT_TITLE "Test Results"

#define ID_RESULT_TEXT 0xFFFF

#define ID_RESULT_OK 0x2

#define BUFFER_SIZE 256

#define VIDPID_SIZE	16

#define TEST_PASSED "Passed"

#define TEST_FAILED "Failed"

// Wait sleep in milliseconds, other waits as repetitions of the sleep time

#define WAIT_SLEEP 250					

#define CREATE_WAIT 40

#define SELECT_WAIT 20

#define CONTINUE_WAIT 120

#define RESULT_WAIT	1200

#define EXIT_WAIT 40

#define USBMS_DEVICE_REQUEST_CONTINUES	6

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

// Construction/Destruction

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

UsbcvControl::UsbcvControl()

{

    ZeroMemory( &si, sizeof(si) );

    si.cb = sizeof(si);

    ZeroMemory( &pi, sizeof(pi) );

}

UsbcvControl::~UsbcvControl()

{

	DWORD exitCode;

	if (GetExitCodeProcess(pi.hProcess,(LPDWORD)&exitCode))

		{

		if (exitCode == STILL_ACTIVE)

			{

			SendMessage (mainWindow,WM_COMMAND,MAKEWPARAM(ID_EXIT,BN_CLICKED),(LPARAM)exitButton);

			}

		int i = 0;

		while (exitCode == STILL_ACTIVE && i++ < EXIT_WAIT)

			{

			Sleep (WAIT_SLEEP);

			GetExitCodeProcess(pi.hProcess,(LPDWORD)&exitCode);

			}

		// Force an unclean process termination only if necessary

		if (exitCode == STILL_ACTIVE)

			{

			TerminateProcess(pi.hProcess,0);

			}

		}	

}

DWORD UsbcvControl::Create()

{

	mainWindow = NULL;

	exitButton = NULL;

	if (!CreateProcess (APPNAME,CMDLINE,NULL,NULL,FALSE,0,NULL,CURDIR,&si,&pi))

		return USBIO_ERR_FAILED;

	HWND complianceButton;

	for (int i=0; i < CREATE_WAIT && !mainWindow; i++)

		{

		mainWindow = FindWindow(DIALOG_CLASS,APPTITLE);

		if (mainWindow)

			{

			complianceButton = GetDlgItem(mainWindow,ID_COMPLIANCE);

			runButton = GetDlgItem(mainWindow,ID_RUN);

			exitButton = GetDlgItem(mainWindow,ID_EXIT);	

			if (!complianceButton || !runButton || !exitButton)

				{

				mainWindow = NULL;

				}

			}

		if (!mainWindow)

			Sleep(WAIT_SLEEP);

		}

	if (mainWindow)

		{

		SendMessage (mainWindow,WM_COMMAND,MAKEWPARAM(ID_COMPLIANCE,BN_CLICKED),(LPARAM)complianceButton);

		return USBIO_ERR_SUCCESS;

		}

	else

		{

		return USBIO_ERR_FAILED;

		}

}

DWORD UsbcvControl::TestAllDevices(USHORT aVendorId, USHORT aProductId1, USHORT aProductId2)

{

	DWORD dwRC = USBIO_ERR_SUCCESS;

	HWND testList;

	int chapter9_TestIndex = -1;

	int MSC_TestIndex = -1;

	LRESULT lResult;

	testList = GetDlgItem(mainWindow,ID_TESTLIST);

	LRESULT itemCount = SendMessage (testList,LB_GETCOUNT,0,0);

	if (!itemCount)

		return USBIO_ERR_FAILED;

	char strBuffer[BUFFER_SIZE];

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

		{

		LRESULT lResult = SendMessage (testList,LB_GETTEXT,(WPARAM)i,(LPARAM)&strBuffer);

		if (lResult > 0)

			{

			if (strstr (strBuffer,"Chapter 9 Tests"))

				chapter9_TestIndex = i;

			if (strstr (strBuffer,"MSC Tests"))

				MSC_TestIndex = i;

			}

		}

	if (chapter9_TestIndex < 0 || MSC_TestIndex < 0)

		return USBIO_ERR_FAILED;

	int ch9DeviceIndex = 0;

	int numCh9Devices = 1;

	int mscDeviceIndex = 0;

	int numMscDevices = 1;

	int testResult;

	bool msDevice = false;

	while (numCh9Devices > 0 && dwRC == USBIO_ERR_SUCCESS)

		{

		// select the chapter 9 tests

		lResult = SendMessage (testList,LB_SETCURSEL,(WPARAM)chapter9_TestIndex,0);

		lResult = SendMessage (mainWindow,WM_COMMAND,MAKEWPARAM(ID_TESTLIST,LBN_SELCHANGE),(LPARAM)testList);

		// then run it

		lResult = SendMessage (mainWindow,WM_COMMAND,MAKEWPARAM(ID_RUN,BN_CLICKED),(LPARAM)runButton);

		dwRC = SelectDevice (aVendorId,aProductId1,aProductId2,&ch9DeviceIndex,&numCh9Devices,&msDevice);

		if (ch9DeviceIndex >= 0 && dwRC == USBIO_ERR_SUCCESS)

			{

			dwRC = WaitForTestResult(&testResult);

			if (dwRC == USBIO_ERR_SUCCESS)

				{

				if (!testResult)

					{

					gAbort = TRUE;

					}

				PRINT_TIME "USBCV Chapter 9 Test Complete - %s  ",testResult ? "Passed" : "Failed");

				ch9DeviceIndex++;

				}

			else

				{

				PRINT_ALWAYS "Error in Running USBCV\n");

				return dwRC;

				}

			}

		if (msDevice && numMscDevices > 0 && dwRC == USBIO_ERR_SUCCESS)

			{

			// select the mass storage tests

			lResult = SendMessage (testList,LB_SETCURSEL,(WPARAM)MSC_TestIndex,0);

			lResult = SendMessage (mainWindow,WM_COMMAND,MAKEWPARAM(ID_TESTLIST,LBN_SELCHANGE),(LPARAM)testList);

			// then run it

			lResult = SendMessage (mainWindow,WM_COMMAND,MAKEWPARAM(ID_RUN,BN_CLICKED),(LPARAM)runButton);

			if (dwRC == USBIO_ERR_SUCCESS)

				dwRC = SelectDevice (aVendorId,aProductId1,aProductId2,&mscDeviceIndex,&numMscDevices,&msDevice);

			if (mscDeviceIndex >= 0 && dwRC == USBIO_ERR_SUCCESS)

				{

				dwRC = ContinueOk ("The following test might destroy",ID_CONFIRM);

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

					{

					if (dwRC == USBIO_ERR_SUCCESS)

						dwRC = ContinueOk ("Processing device request",ID_OKBUTTON);

					}

				if (dwRC == USBIO_ERR_SUCCESS)

					dwRC = ContinueOk ("Disconnect and power off",ID_CONFIRM);

				if (dwRC == USBIO_ERR_SUCCESS)

					{

					dwRC = WaitForTestResult(&testResult);

					if (dwRC == USBIO_ERR_SUCCESS)

						{

						if (!testResult)

							{

							gAbort = TRUE;

							}

						PRINT_TIME "USBCV Mass Storage Class Test Complete - %s  ",testResult ? "Passed" : "Failed");

						mscDeviceIndex++;

						}

					else

						{

						PRINT_ALWAYS "Error in Running USBCV\n");

						return dwRC;

						}

					}

				}

			}

		}

	return dwRC;

}

DWORD UsbcvControl::ContinueOk (char * choiceText, int buttonId)

{

	HWND choiceDialog = NULL;

	HWND okButton;

	char strBuffer[BUFFER_SIZE];

	for (int i=0; i < CONTINUE_WAIT && !choiceDialog; i++)

		{

		choiceDialog = FindWindow(DIALOG_CLASS,"");

		if (choiceDialog)

			{

			UINT textSize = GetDlgItemText (choiceDialog,ID_TEXTBOX,strBuffer,BUFFER_SIZE);

			okButton = GetDlgItem(choiceDialog,buttonId);

			if (!textSize || !okButton || !strstr (strBuffer,choiceText))

				{

				choiceDialog = NULL;

				}

			}

		if (!choiceDialog)

			Sleep(WAIT_SLEEP);

		}

	if (!choiceDialog)

		return USBIO_ERR_FAILED;

	SendMessage (choiceDialog,WM_COMMAND,MAKEWPARAM(buttonId,BN_CLICKED),(LPARAM)okButton);

	return USBIO_ERR_SUCCESS;

}

DWORD UsbcvControl::SelectDevice(USHORT aVendorId, USHORT aProductId1, USHORT aProductId2, int * deviceIndex, int * numDevices, bool * msDevice)

{

	HWND devListDialog = NULL;

	HWND devList;

	HWND okButton;

	LRESULT lResult;

	int i;

	for (i=0; i < SELECT_WAIT && !devListDialog; i++)

		{

		devListDialog = FindWindow(DIALOG_CLASS,DEVLIST_TITLE);

		if (devListDialog)

			{

			devList = GetDlgItem(devListDialog,ID_DEVLIST);

			okButton = GetDlgItem(devListDialog,ID_OKBUTTON);

			if (!devList || !okButton)

				{

				devListDialog = NULL;

				}

			}

		if (!devListDialog)

			Sleep(WAIT_SLEEP);

		}

	if (!devListDialog)

		return USBIO_ERR_FAILED;

	LRESULT itemCount = SendMessage (devList,LB_GETCOUNT,0,0);

	if (!itemCount)

		return USBIO_ERR_FAILED;

	* numDevices = 0;

	if (* deviceIndex >= itemCount)

		return USBIO_ERR_SUCCESS;

	char strBuffer[BUFFER_SIZE];

	char strVID [VIDPID_SIZE];

	char strPID1 [VIDPID_SIZE];

	char strPID2 [VIDPID_SIZE];

	bool deviceFound = false;

	for (i = * deviceIndex; i < itemCount; i++)

		{

		lResult = SendMessage (devList,LB_GETTEXT,(WPARAM)i,(LPARAM)&strBuffer);

		if (lResult > 0)

			{

			sprintf (strVID,"VID=%04x",aVendorId);

			sprintf (strPID1,"PID=%04x",aProductId1);

			sprintf (strPID2,"PID=%04x",aProductId2);

			if (strstr (strBuffer,strVID) && (strstr (strBuffer,strPID1) || strstr (strBuffer,strPID2)))

				{

				if (deviceFound)

					{

					(* numDevices)++;

					}

				else

					{

					deviceFound = true;

					* msDevice = strstr (strBuffer,strPID2) != NULL;

					* deviceIndex = i;

					lResult = SendMessage (devList,LB_SETCURSEL,(WPARAM)i,0);

					}

				}

			}

		}

	if (deviceFound)

		{

		if (lResult == LB_ERR)

			{

			return USBIO_ERR_FAILED;

			}

		else

			{

			lResult = SendMessage (devListDialog,WM_COMMAND,MAKEWPARAM(ID_OKBUTTON,BN_CLICKED),(LPARAM)okButton);

			return USBIO_ERR_SUCCESS;

			}

		}

	else

		{

		* deviceIndex = -1;

		return USBIO_ERR_FAILED;

		}

}

DWORD UsbcvControl::WaitForTestResult(BOOL * passFail)

{

	HWND testResultDialog = NULL;

	HWND okButton;

	char strBuffer[BUFFER_SIZE];

	for (int i=0; i < RESULT_WAIT && !testResultDialog; i++)

		{

		testResultDialog = FindWindow(DIALOG_CLASS,TESTRESULT_TITLE);

		if (testResultDialog)

			{

			UINT textSize = GetDlgItemText(testResultDialog,ID_RESULT_TEXT,(LPTSTR)&strBuffer,BUFFER_SIZE);

			okButton = GetDlgItem(testResultDialog,ID_RESULT_OK);

			if (!textSize || !okButton)

				{

				testResultDialog = NULL;

				}

			}

		if (!testResultDialog)

			Sleep(WAIT_SLEEP);

		}

	if (!testResultDialog)

		return USBIO_ERR_FAILED;

	if (strstr (strBuffer,TEST_PASSED))

		{

		* passFail = TRUE;

		}

	else

		{

		if (strstr (strBuffer,TEST_FAILED))

			{

			* passFail = FALSE;

			}

		else

			{

			return USBIO_ERR_FAILED;

			}

		}

	SendMessage (testResultDialog,WM_COMMAND,MAKEWPARAM(ID_RESULT_OK,BN_CLICKED),(LPARAM)okButton);

	return USBIO_ERR_SUCCESS;

}