// 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;

 }