--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/kerneltest/e32test/usb/t_usb_win/src/UsbcvControl.cpp	Mon Oct 19 15:55:17 2009 +0100
@@ -0,0 +1,433 @@
+// 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;
+
+}