// 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:
// e32test\win32\usbrflct\usbrflct.cpp
// Win32 USB test program USBRFLCT: performs I/O with a device running the
// Symbian OS test program T_USB, using the generic USB device driver USBIO.
// === Includes ===
// 
//

#include <windows.h>
#include <stdio.h>
#include <iostream.h>
#include <time.h>

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


// === Global Defines ===

// The version of this program
#define VERSION_MAJOR  1
#ifdef VERSION_MINOR
# undef VERSION_MINOR						  // VERSION_MINOR sometimes yields funny compiler warning (c4005)
#endif
#define VERSION_MINOR  6
#ifdef VERSION_MICRO
# undef VERSION_MICRO						  // ditto
#endif
#define VERSION_MICRO  0

// The version of the USBIO driver this program is compiled against
#define USBIO_VERSION_MAJOR  2
#define USBIO_VERSION_MINOR  41

#define MAX_DESCRIPTOR_BUFFER_SIZE 2047

// === Global Vars ===

// Our own private GUID (also used in the .inf file as 'DriverUserInterfaceGuid')
// {55606403-E62D-4707-9F56-40D48C6736D0}
static const GUID g_UsbioID =
	{0x55606403, 0xe62d, 0x4707, {0x9f, 0x56, 0x40, 0xd4, 0x8c, 0x67, 0x36, 0xd0}};

// Handle for USB device list
static HDEVINFO g_DevList = NULL;

// USBIO supported device
static CUsbIo g_UsbDev;
static CUsbIoPipe g_BulkOutPipe;
static CUsbIoPipe g_BulkInPipe;

// 2 Bulk endpoints
static UCHAR g_ucBulkOutEndpoint = 0;
static UCHAR g_ucBulkInEndpoint = 0;

// Read/write buffer
static const DWORD KBufferSize = 1024 * 1024;
static const DWORD KPreambleLength = 8;
static BYTE Data[KBufferSize];								// the read/write buffer
static DWORD Length;										// Length of a transfer
static time_t T_0;											// the starting time

static CUsbIoBuf g_Buf((VOID*) Data, KBufferSize);		// the data buffer
static CUsbIoBuf g_ZlpBuf (NULL, 0);					// the data buffer

static DWORD dwRC = USBIO_ERR_SUCCESS;						// global error indicator

enum
	{
	ELoop,
	ELoopDebug,
	EReceiveOnly,
	ETransmitOnly
	};

static int TransferMode = ELoop;
static bool VerboseMode = false;
static bool ZlpMode = false;

static unsigned int maxOutPacketSize = 0;

// The version of T_USB we require (at least)
static const DWORD KTusbVersion = 20070524;

// After how many iterations to update the CRT:
static const int KLoopModeDisplayUpdate = 1024;
// After how many iterations (= MBytes) to update the CRT:
static const int KUniModeDisplayUpdate = 10;

// Helper #defines

#define PRINT_IF_VERBOSE(string) \
		do { \
		if (VerboseMode) \
			{ \
			printf(string); \
			} \
		} while (0)

#define PRINT_IF_VERBOSE1(string, a) \
		do { \
		if (VerboseMode) \
			{ \
			printf(string, a); \
			} \
		} while (0)


// === Functions ===

//
// Process the command line arguments, printing a helpful message
// if none are supplied.
//
static int ProcessCmdLine(int argc, char* argv[])
	{
	char help_text[] =
		"* Syntax: usbrflct [options]\n"
		"* Options:\n"
		"* /[r|t]    receive|transmit only; default is to loop\n"
		"* /z        zero length packet at end of a transmission\n"
		"*	    (only if last packet is full length)\n"
		"* /l        loop mode with stats printed for every iteration\n"
		"* /v        verbose driver & program output\n"
		"* /[h|?]    displays this help text\n"
		"\n";

	for (int i = 1; i < argc; i++)
		{
		strupr(argv[i]);
		if ((argv[i][0] == '-') || (argv[i][0] == '/'))
			{
			switch (argv[i][1])
				{
			case 'R':
			case 'r':
				TransferMode = EReceiveOnly;
				break;
			case 'T':
			case 't':
				TransferMode = ETransmitOnly;
				break;
			case 'L':
			case 'l':
				TransferMode = ELoopDebug;
				break;
			case 'V':
			case 'v':
				VerboseMode = true;
				break;
			case 'Z':
			case 'z':
				ZlpMode = true;
				break;
			case '?':
			case 'H':
			case 'h':
				cout << help_text;
				return -1;
			default:
				cout << "* Invalid argument: " << argv[i] << "\n";
				cout << help_text;
				return -1;
				}
			}
		}
	return 0;
	}


static void OpenUsbDevice()
	{
	CUsbIo::DestroyDeviceList(g_DevList);

	// Enumerate attached USB devices supported by USBIO
	g_DevList = CUsbIo::CreateDeviceList(&g_UsbioID);

	// Open first device in list
	dwRC = g_UsbDev.Open(0, g_DevList, &g_UsbioID);

	PRINT_IF_VERBOSE1("\nCUsbIo::Open returned <0x%X>\n", dwRC);

	if (dwRC == USBIO_ERR_VERSION_MISMATCH)
		{
		printf("\n* Error: \"The API version reported by the USBRFLCT driver\n" \
			   "*         does not match the expected version.\"\n");
		printf("* The driver will need to be updated as follows:\n");
		printf("* 1. Connect the device to the PC & start T_USB,\n" \
			   "* then find the USB device in the Windows Device Manager\n" \
			   "* ('Control Panel'->'System'->'Hardware'->'Device Manager').\n" \
			   "* Right click on the device name and choose 'Uninstall...'.\n");
		printf("* 2. In c:\\winnt\\inf\\, find (by searching for \"Symbian\") and\n" \
			   "* delete the *.INF file that was used to install the existing\n" \
			   "* version of USBRFLCT.SYS. Make sure to also delete the\n" \
			   "* precompiled version of that file (<samename>.PNF).\n");
		printf("* 3. In c:\\winnt\\system32\\drivers\\, delete the file USBRFLCT.SYS.\n");
		printf("* Then unplug & reconnect the USB device and, when prompted, install\n" \
			   "* the new USBRFLCT.SYS driver using the .INF file from this distribution.\n" \
			   "* (All files can be found under e32test\\win32\\usbrflct_distribution\\.)\n");
		}
	}


static void CloseUsbDevice()
	{
	// Close the device
	g_UsbDev.Close();
	PRINT_IF_VERBOSE("CUsbIo::Close called\n");
	}


static void GetDeviceDescriptor()
	{
	USB_DEVICE_DESCRIPTOR DeviceDescriptor;

	memset(&DeviceDescriptor, 0, sizeof(USB_DEVICE_DESCRIPTOR));

	// Get device descriptor
	dwRC = g_UsbDev.GetDeviceDescriptor(&DeviceDescriptor);
	PRINT_IF_VERBOSE1("CUsbIo::GetDeviceDescriptor returned <0x%X>\n", dwRC);

	if (VerboseMode && (dwRC == USBIO_ERR_SUCCESS))
		{
		printf("\nDEVICE DESCRIPTOR:\n"
			   "bLength = <%u>\n"
			   "bDescriptorType = <%u>\n"
			   "bcdUSB = <%u>\n"
			   "bDeviceClass = <%u>\n"
			   "bDeviceSubClass = <%u>\n"
			   "bDeviceProtocol = <%u>\n"
			   "bMaxPacketSize0 = <%u>\n"
			   "idVendor = <%u>\n"
			   "idProduct = <%u>\n"
			   "bcdDevice = <%u>\n"
			   "iManufacturer = <%u>\n"
			   "iProduct = <%u>\n"
			   "iSerialNumber = <%u>\n"
			   "bNumConfigurations = <%u>\n\n",
			   DeviceDescriptor.bLength,
			   DeviceDescriptor.bDescriptorType,
			   DeviceDescriptor.bcdUSB,
			   DeviceDescriptor.bDeviceClass,
			   DeviceDescriptor.bDeviceSubClass,
			   DeviceDescriptor.bDeviceProtocol,
			   DeviceDescriptor.bMaxPacketSize0,
			   DeviceDescriptor.idVendor,
			   DeviceDescriptor.idProduct,
			   DeviceDescriptor.bcdDevice,
			   DeviceDescriptor.iManufacturer,
			   DeviceDescriptor.iProduct,
			   DeviceDescriptor.iSerialNumber,
			   DeviceDescriptor.bNumConfigurations);
		}
	}


static void GetConfigurationDescriptor()
	{
	CHAR szBuffer[MAX_DESCRIPTOR_BUFFER_SIZE] = "";
	USB_CONFIGURATION_DESCRIPTOR* pConfigDescriptor = NULL;
	USB_INTERFACE_DESCRIPTOR* pInterfaceDescriptor = NULL;
	USB_ENDPOINT_DESCRIPTOR* pEndpointDescriptor = NULL;

	DWORD dwByteCount = MAX_DESCRIPTOR_BUFFER_SIZE;

	memset(szBuffer, 0, sizeof(szBuffer));

	// Get first configuration descriptor
	dwRC =
		g_UsbDev.GetConfigurationDescriptor((USB_CONFIGURATION_DESCRIPTOR*) szBuffer,
											dwByteCount, 0);
	PRINT_IF_VERBOSE1("CUsbIo::GetConfigurationDescriptor returned <0x%X>\n", dwRC);

	if (dwRC == USBIO_ERR_SUCCESS)
		{
		USB_COMMON_DESCRIPTOR* Desc;
		ULONG rl = dwByteCount;
		ULONG ulDescLength = 0;
		CHAR* data = szBuffer;

		while (rl > 0)
			{
			Desc = (USB_COMMON_DESCRIPTOR*) data;
			ulDescLength = Desc->bLength;
			if ((ulDescLength > rl) || (ulDescLength == 0))
				{
				printf("Length remaining too short!\n");
				rl = 0;
				}
			else
				{
				switch (Desc->bDescriptorType)
					{
				case USB_CONFIGURATION_DESCRIPTOR_TYPE:
					pConfigDescriptor =
						(USB_CONFIGURATION_DESCRIPTOR*) data;
					if (VerboseMode)
						{
						printf("\nCONFIGURATION DESCRIPTOR:\n"
							   "bLength = <%u>\n"
							   "bDescriptorType = <%u>\n"
							   "wTotalLength = <%u>\n"
							   "bNumInterfaces = <%u>\n"
							   "bConfigurationValue = <%u>\n"
							   "iConfiguration = <%u>\n"
							   "bmAttributes = <%u>\n"
							   "MaxPower = <%u>\n",
							   pConfigDescriptor->bLength,
							   pConfigDescriptor->bDescriptorType,
							   pConfigDescriptor->wTotalLength,
							   pConfigDescriptor->bNumInterfaces,
							   pConfigDescriptor->bConfigurationValue,
							   pConfigDescriptor->iConfiguration,
							   pConfigDescriptor->bmAttributes,
							   pConfigDescriptor->MaxPower);
						}
					break;
				case USB_INTERFACE_DESCRIPTOR_TYPE:
					pInterfaceDescriptor =
						(USB_INTERFACE_DESCRIPTOR*) data;
					if (VerboseMode)
						{
						printf("\nINTERFACE DESCRIPTOR: \n"
							   "bLength = <%u>\n"
							   "bDescriptorType = <%u>\n"
							   "bInterfaceNumber = <%u>\n"
							   "bAlternateSetting = <%u>\n"
							   "bNumEndpoints = <%u>\n"
							   "bInterfaceClass = <%u>\n"
							   "bInterfaceSubClass = <%u>\n"
							   "bInterfaceProtocol = <%u>\n"
							   "iInterface = <%u>\n",
							   pInterfaceDescriptor->bLength,
							   pInterfaceDescriptor->bDescriptorType,
							   pInterfaceDescriptor->bInterfaceNumber,
							   pInterfaceDescriptor->bAlternateSetting,
							   pInterfaceDescriptor->bNumEndpoints,
							   pInterfaceDescriptor->bInterfaceClass,
							   pInterfaceDescriptor->bInterfaceSubClass,
							   pInterfaceDescriptor->bInterfaceProtocol,
							   pInterfaceDescriptor->iInterface);
						}
					break;
				case USB_ENDPOINT_DESCRIPTOR_TYPE:
					pEndpointDescriptor =
						(USB_ENDPOINT_DESCRIPTOR*) data;
					if (VerboseMode)
						{
						printf("\nENDPOINT DESCRIPTOR: \n"
							   "bLength = <%u>\n"
							   "bDescriptorType = <%u>\n"
							   "bEndpointAddress = <%u>\n"
							   "bmAttributes = <%u>\n"
							   "wMaxPacketSize = <%u>\n"
							   "bInterval = <%u>\n",
							   pEndpointDescriptor->bLength,
							   pEndpointDescriptor->bDescriptorType,
							   pEndpointDescriptor->bEndpointAddress,
							   pEndpointDescriptor->bmAttributes,
							   pEndpointDescriptor->wMaxPacketSize,
							   pEndpointDescriptor->bInterval);
						}
					break;
				default:
					break;
					}
				}
			data += ulDescLength;
			rl -= ulDescLength;
			}
		}
	PRINT_IF_VERBOSE("\n");
	}


static void GetStringDescriptor()
	{
	CHAR szBuffer[MAX_DESCRIPTOR_BUFFER_SIZE] = "";
	USB_STRING_DESCRIPTOR* pStringDescriptor = NULL;
	DWORD dwByteCount = MAX_DESCRIPTOR_BUFFER_SIZE;

	memset(szBuffer, 0, sizeof(szBuffer));

	// Get string descriptor
	dwRC = g_UsbDev.GetStringDescriptor((USB_STRING_DESCRIPTOR*) szBuffer,
										dwByteCount, 1, 0);
	PRINT_IF_VERBOSE1("CUsbIo::GetStringDescriptor returned <0x%X>\n", dwRC);

	if (dwRC == USBIO_ERR_SUCCESS)
		{
		pStringDescriptor = (USB_STRING_DESCRIPTOR*) szBuffer;
		if (VerboseMode)
			{
			printf("\nSTRING DESCRIPTOR:\n"
				   "bLength = <%u>\n"
				   "bDescriptorType = <%u>\n"
				   "bString = <",							// output continues below!
				   pStringDescriptor->bLength,
				   pStringDescriptor->bDescriptorType);
			}
		INT i = 0;
		CHAR* Ptr = szBuffer;
		for (i = 2, Ptr += 2;
			 i < pStringDescriptor->bLength;
			 i += 2, Ptr += 2)
			{
			PRINT_IF_VERBOSE1("%c", *Ptr);
			}
		PRINT_IF_VERBOSE(">\n\n");
		}
	}


static void SetConfiguration()
	{
	USBIO_SET_CONFIGURATION SetConfig;

	memset(&SetConfig, 0, sizeof(USBIO_SET_CONFIGURATION));

	// Set the first configuration as active
	SetConfig.ConfigurationIndex = 0;
	SetConfig.NbOfInterfaces = 1;
	SetConfig.InterfaceList[0].InterfaceIndex = 0;
	SetConfig.InterfaceList[0].AlternateSettingIndex = 0;
	SetConfig.InterfaceList[0].MaximumTransferSize = KBufferSize;
	dwRC = g_UsbDev.SetConfiguration(&SetConfig);
	PRINT_IF_VERBOSE1("CUsbIo::SetConfiguration returned <0x%X>\n", dwRC);
	}


static void GetConfigurationInfo()
	{
	USBIO_CONFIGURATION_INFO ConfigInfo;
	USHORT i = 0;

	memset(&ConfigInfo, 0, sizeof(USBIO_CONFIGURATION_INFO));

	dwRC = g_UsbDev.GetConfigurationInfo(&ConfigInfo);
	PRINT_IF_VERBOSE1("CUsbIo::GetConfigurationInfo returned <0x%X>\n", dwRC);

	if (dwRC == USBIO_ERR_SUCCESS)
		{
		if (VerboseMode)
			{
			printf("\nCONFIGURATION INFO:\n"
				   "NbOfInterfaces = <%lu>\n"
				   "NbOfPipes = <%lu>\n",
				   ConfigInfo.NbOfInterfaces,
				   ConfigInfo.NbOfPipes);
			}
		for (i = 0; i < ConfigInfo.NbOfInterfaces; i++)
			{
			if (VerboseMode)
				{
				printf("\nINTERFACE <%u>:\n", i + 1);
				printf("InterfaceNumber = <%u>\n"
					   "AlternateSetting = <%u>\n"
					   "Class = <%u>\n"
					   "SubClass = <%u>\n"
					   "Protocol = <%u>\n"
					   "NumberOfPipes = <%u>\n"
					   "reserved1 = <%u>\n"
					   "reserved2 = <%u>\n",
					   ConfigInfo.InterfaceInfo[i].InterfaceNumber,
					   ConfigInfo.InterfaceInfo[i].AlternateSetting,
					   ConfigInfo.InterfaceInfo[i].Class,
					   ConfigInfo.InterfaceInfo[i].SubClass,
					   ConfigInfo.InterfaceInfo[i].Protocol,
					   ConfigInfo.InterfaceInfo[i].NumberOfPipes,
					   ConfigInfo.InterfaceInfo[i].reserved1,
					   ConfigInfo.InterfaceInfo[i].reserved2);
				}
			}
		for (i = 0; i < ConfigInfo.NbOfPipes; i++)
			{
			PRINT_IF_VERBOSE("\n");
			if ((ConfigInfo.PipeInfo[i].PipeType == PipeTypeBulk) &&
				!(ConfigInfo.PipeInfo[i].EndpointAddress & 0x80))
				{
				PRINT_IF_VERBOSE("Bulk OUT pipe found:\n");
				g_ucBulkOutEndpoint = ConfigInfo.PipeInfo[i].EndpointAddress;
				maxOutPacketSize = ConfigInfo.PipeInfo[i].MaximumPacketSize;
				}
			else if ((ConfigInfo.PipeInfo[i].PipeType == PipeTypeBulk) &&
					 (ConfigInfo.PipeInfo[i].EndpointAddress & 0x80))
				{
				PRINT_IF_VERBOSE("Bulk IN pipe found:\n");
				g_ucBulkInEndpoint = ConfigInfo.PipeInfo[i].EndpointAddress;
				}
			if (VerboseMode)
				{
				printf("PIPE <%u>\n", i + 1);
				printf("PipeType = <%d>\n"
					   "MaximumTransferSize = <%lu>\n"
					   "MaximumPacketSize = <%u>\n"
					   "EndpointAddress = <%u>\n"
					   "Interval = <%u>\n"
					   "InterfaceNumber = <%u>\n"
					   "reserved1 = <%u>\n"
					   "reserved2 = <%u>\n"
					   "reserved3 = <%u>\n",
					   ConfigInfo.PipeInfo[i].PipeType,
					   ConfigInfo.PipeInfo[i].MaximumTransferSize,
					   ConfigInfo.PipeInfo[i].MaximumPacketSize,
					   ConfigInfo.PipeInfo[i].EndpointAddress,
					   ConfigInfo.PipeInfo[i].Interval,
					   ConfigInfo.PipeInfo[i].InterfaceNumber,
					   ConfigInfo.PipeInfo[i].reserved1,
					   ConfigInfo.PipeInfo[i].reserved2,
					   ConfigInfo.PipeInfo[i].reserved3);
				}
			}
		}
	PRINT_IF_VERBOSE("\n");
	}


static void OpenPipes()
	{
	CUsbIo::DestroyDeviceList(g_DevList);

	// Enumerate attached USB devices supported by USBIO
	g_DevList = CUsbIo::CreateDeviceList(&g_UsbioID);

	// Create the bulk OUT pipe
	dwRC = g_BulkOutPipe.Bind(0, g_ucBulkOutEndpoint, g_DevList, &g_UsbioID);
	PRINT_IF_VERBOSE1("CUsbIoPipe::Bind (Bulk OUT) returned <0x%X>\n", dwRC);

	if (dwRC == USBIO_ERR_SUCCESS)
		{
		// Create the bulk IN pipe
		dwRC = g_BulkInPipe.Bind(0, g_ucBulkInEndpoint, g_DevList, &g_UsbioID);
		PRINT_IF_VERBOSE1("CUsbIoPipe::Bind (Bulk IN) returned <0x%X>\n", dwRC);
		}
	PRINT_IF_VERBOSE("\n");
	}


static void ClosePipes()
	{
	// Close down the bulk OUT pipe
	dwRC = g_BulkOutPipe.Unbind();
	PRINT_IF_VERBOSE1("CUsbIoPipe::Unbind (Bulk OUT) returned <0x%X>\n", dwRC);

	if (dwRC == USBIO_ERR_SUCCESS)
		{
		// Close down the bulk IN pipe
		dwRC = g_BulkInPipe.Unbind();
		PRINT_IF_VERBOSE1("CUsbIoPipe::Unbind (Bulk IN) returned <0x%X>\n", dwRC);
		}
	}


static void ReceiveVersion()
	{
	// Here we (hope to) read an 8 byte packet containing the T_USB version.
	printf("* Waiting for T_USB version packet to arrive...");

	// The first 4 bytes are interpreted as an int32 value.
	DWORD bytes_read = 0;
	g_Buf.NumberOfBytesToTransfer = KPreambleLength;
	g_BulkInPipe.Read(&g_Buf);
	dwRC = g_BulkInPipe.WaitForCompletion(&g_Buf, INFINITE);
	printf(" done.\n");
	bytes_read = g_Buf.BytesTransferred;
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		if (bytes_read < KPreambleLength)
			{
			printf("* Read less bytes (%d) than expected (%d).\n",
				   bytes_read, KPreambleLength);
			dwRC = USBIO_ERR_FAILED;
			return;
			}
		}
	else
		{
		printf("\n* Error: CUsbIoPipe::Read (version) returned <0x%X>\n", dwRC);
		return;
		}
	// First make sure it's actually the version packet, and not
	// a data preamble packet of an old T_USB.
	if (!(Data[4] == 'V' &&
		  Data[5] == 'e' &&
		  Data[6] == 'r' &&
		  Data[7] == 's'))
		{
		printf("* Inadequate version of T_USB: no version packet was sent (we need at least %d)\n",
			   KTusbVersion);
		dwRC = USBIO_ERR_FAILED;
		return;
		}
	DWORD tusb_version = *((ULONG*) Data);						// first 4 bytes
	if (tusb_version < KTusbVersion)
		{
		printf("* Inadequate version of T_USB: %d (we need at least %d)\n",
			   tusb_version, KTusbVersion);
		dwRC = USBIO_ERR_FAILED;
		return;
		}
	printf("* Suitable version of T_USB found: %d.\n", tusb_version);
	}


static void SendVersion()
	{
	// Here we send an 8 byte packet containing USBRFLCT's + USBIO's versions.
	printf("* Sending our version packet to T_USB...");

	DWORD bytes_written = 0;
	g_Buf.NumberOfBytesToTransfer = KPreambleLength;
	Data[0] = VERSION_MAJOR;
	Data[1] = VERSION_MINOR;
	Data[2] = VERSION_MICRO;
	Data[3] = USBIO_VERSION_MAJOR;
	Data[4] = USBIO_VERSION_MINOR;
	g_BulkOutPipe.Write(&g_Buf);
	dwRC = g_BulkOutPipe.WaitForCompletion(&g_Buf, INFINITE);
	printf(" done.\n");
	bytes_written = g_Buf.BytesTransferred;
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		if (bytes_written < KPreambleLength)
			{
			printf("* Wrote less bytes (%d) than requested (%d).\n",
				   bytes_written, KPreambleLength);
			dwRC = USBIO_ERR_FAILED;
			}
		}
	else
		{
		printf("\n* Error: CUsbIoPipe::Write (version) returned <0x%X>\n", dwRC);
		}
	}


static void ExchangeVersions()
	{
	SendVersion();
	if (dwRC != USBIO_ERR_SUCCESS)
		return;
	ReceiveVersion();
	}


static void GetLength()
	{
	// The first two bytes are interpreted as a length value.
	DWORD bytes_read = KPreambleLength;
	g_Buf.NumberOfBytesToTransfer = KPreambleLength;

	g_BulkInPipe.Read(&g_Buf);
	dwRC = g_BulkInPipe.WaitForCompletion(&g_Buf, INFINITE);
	bytes_read = g_Buf.BytesTransferred;

	if (dwRC == USBIO_ERR_SUCCESS)
		{
		if (bytes_read < KPreambleLength)
			{
			printf("* Read less bytes (%d) than expected (%d).\n",
				   bytes_read, KPreambleLength);
			dwRC = USBIO_ERR_FAILED;
			return;
			}
		}
	else
		{
		printf("\n* Error: CUsbIoPipe::Read (length) returned <0x%X>\n", dwRC);
		return;
		}
	Length = *((ULONG*) Data);								// first 4 bytes
	if (Length > KBufferSize)
		{
		printf("* This is too much: %d (our buffer is too small: %d)\n",
			   Length, KBufferSize);
		dwRC = USBIO_ERR_FAILED;
		}
	if (VerboseMode)
		{
		printf("* Just read %d bytes, now assuming transfer length is %d bytes.\n",
			   bytes_read, Length);
		}
	else if (TransferMode == EReceiveOnly || TransferMode == ETransmitOnly)
		{
		printf("* Single transfer size: %d bytes.\n", Length);
		}
	}


static void ReadData()
	{
	// We have to setup a read for at least one byte in order to get
	// the host to issue IN tokens for our zero-byte read:
	if (Length == 0)
		g_Buf.NumberOfBytesToTransfer = 1;
	else
		g_Buf.NumberOfBytesToTransfer = Length;

	g_BulkInPipe.Read(&g_Buf);
	dwRC = g_BulkInPipe.WaitForCompletion(&g_Buf, INFINITE);
	DWORD bytes_read = g_Buf.BytesTransferred;

	if (dwRC != USBIO_ERR_SUCCESS)
		{
		printf("\n* Error: CUsbIoPipe::Read (data) returned <0x%X>\n", dwRC);
		}
	else
		{
		if (bytes_read != Length)
			{
			printf("* Read more/less bytes (%d) than expected (%d).\n",
				   bytes_read, Length);
			dwRC = USBIO_ERR_FAILED;
			}
		else
			{
			PRINT_IF_VERBOSE1("* Read %d bytes.\n", Length);
			}
		}
	}


static void WriteData()
	{
	DWORD bytes_written = Length;

	g_Buf.NumberOfBytesToTransfer = bytes_written;

	g_BulkOutPipe.Write(&g_Buf);
	dwRC = g_BulkOutPipe.WaitForCompletion(&g_Buf, INFINITE);
	if (ZlpMode && (Length >= maxOutPacketSize) && ((Length % maxOutPacketSize) == 0))
	{
		// writes a zero length packet
		g_BulkOutPipe.Write(&g_ZlpBuf);
		dwRC = g_BulkOutPipe.WaitForCompletion(&g_ZlpBuf, INFINITE);
	}


	bytes_written = g_Buf.BytesTransferred;

	if (dwRC == USBIO_ERR_SUCCESS)
		{
		PRINT_IF_VERBOSE1("* Wrote %d bytes.\n", bytes_written);
		}
	else
		{
		printf("\n* Error: CUsbIoPipe::Write returned <0x%X>\n", dwRC);
		}
	}


void PrintStats()
	{
	static DWORD loop = 0;									// the loop counter
	static double xfer_size = 0;							// the total transfer amount so far
	time_t t_1 = time(NULL);								// current time
	double t_diff = difftime(t_1, T_0);						// this yields current seconds since start
	xfer_size += (KPreambleLength + (2 * Length)) * KLoopModeDisplayUpdate;	//
	double xfer_rate = xfer_size / t_diff;					// mean transfer rate since start
	loop += KLoopModeDisplayUpdate;
	printf("* Iter: %d (%d bytes) Total: %.0f bytes Rate: %.0f bytes/s  \r",
		   loop, Length, xfer_size, xfer_rate);
	}


void PrintStatsEveryLoop()
	{
	static DWORD loop = 0;									// the loop counter
	static double xfer_size = 0;							// the total transfer amount so far
	time_t t_1 = time(NULL);								// current time
	double t_diff = difftime(t_1, T_0);						// this yields current seconds since start
	xfer_size += (KPreambleLength + (2 * Length));			//
	double xfer_rate = xfer_size / t_diff;					// mean transfer rate since start
	printf("* Iter: %d (%d bytes) Total: %.0f bytes Rate: %.0f bytes/s  \r",
		   ++loop, Length, xfer_size, xfer_rate);
	}


void PrintUnidirStats()
	{
	static DWORD loop = 0;									// the loop counter
	static double xfer_size = 0;							// the total transfer amount so far
	time_t t_1 = time(NULL);								// current time
	double t_diff = difftime(t_1, T_0);						// this yields current seconds since start
	xfer_size += Length * KUniModeDisplayUpdate;
	double xfer_rate = xfer_size / t_diff;					// mean transfer rate since start
	loop += KUniModeDisplayUpdate;
	printf("* Iter: %d (%d bytes) Total: %.0f bytes Rate: %.0f bytes/s  \r",
		   loop, Length, xfer_size, xfer_rate);
	}


static void LoopTransfer()
	{
	printf("* Loop Transfers -- reading & writing alternately.\n");

	T_0 = time(NULL);										// starting time

	while (dwRC == USBIO_ERR_SUCCESS)
		{
		static DWORD n = 0;

		// First we get the length (+ the packet size)
		GetLength();

		if (dwRC == USBIO_ERR_SUCCESS)
			{
			// Then we read 'Length' bytes
			ReadData();
			}

		if (dwRC == USBIO_ERR_SUCCESS)
			{
			// Now we send the received data back to the client.
			WriteData();
			}

		if (dwRC == USBIO_ERR_SUCCESS)
			{
			// Finally, sometimes we print some statistics
			if (TransferMode == ELoopDebug)
				PrintStatsEveryLoop();
			else if ((++n % KLoopModeDisplayUpdate) == 0)
				PrintStats();
			}
		}
	}


static void ReceiveOnlyTransfer()
	{
	printf("* Receive-only transfers (IN).\n");

	// First (and only once) we get the transfer length (+ the packet size)
	GetLength();

	T_0 = time(NULL);										// starting time

	while (dwRC == USBIO_ERR_SUCCESS)
		{
		static DWORD n = -1;
		static DWORD pktNum;

		// Then we read 'Length' bytes
		ReadData();
		pktNum = *(DWORD *)&Data;
		if (pktNum != ++n)
			{
			printf ("\n* Error: rcv'd wrong pkt number: 0x%x (expected: 0x%x)\n", pktNum, n);
			// reset from the received packet number, so that ...
			// if a packet is lost or duplicated a single error is reported
			n = pktNum;
			}
		// Finally, sometimes we print some statistics
		if ((n % KUniModeDisplayUpdate) == 0)
			PrintUnidirStats();
		}
	}


static void TransmitOnlyTransfer()
	{
	printf("* Transmit-only transfers (OUT).\n");

	// First (and only once) we get the transfer length (+ the packet size)
	GetLength();

	T_0 = time(NULL);										// starting time

	while (dwRC == USBIO_ERR_SUCCESS)
		{
		static DWORD n = 0;
		// First the packet number is put into the first four bytes
		*(DWORD *)&Data = n++;
		// Then we write 'Length' bytes
		WriteData();

		// Finally, sometimes we print some statistics
		if ((n % KUniModeDisplayUpdate) == 0)
			PrintUnidirStats();
		}
	}


static void DoTransfers()
	{
	switch (TransferMode)
		{
	case ELoop:
	case ELoopDebug:
		LoopTransfer();
		break;
	case EReceiveOnly:
		ReceiveOnlyTransfer();
		break;
	case ETransmitOnly:
		TransmitOnlyTransfer();
		break;
	default:
		dwRC = -1;
		break;
		}
	}


static void Delay(int milliseconds)
	{
	printf("* Short wait... ");
	Sleep(milliseconds);
	printf("done.\n");
	}


static void PrintHello()
	{
	printf("*--------------------------------------------------\n");
	printf("* USBRFLCT v%d.%d.%d (for use with USBRFLCT.SYS v%d.%d)\n",
		   VERSION_MAJOR, VERSION_MINOR, VERSION_MICRO,
		   USBIO_VERSION_MAJOR, USBIO_VERSION_MINOR);
	printf("* USB Reflector Test Program / T_USB Host-side Part\n");
	printf("*   Copyright (c) 2001-2009 Nokia Corporation and/or its subsidiary(-ies).\n");
	printf("*--------------------------------------------------\n");
	}


int main(int argc, char* argv[])
	{
	PrintHello();

	if (ProcessCmdLine(argc, argv) != 0)
		return -1;

	OpenUsbDevice();

	if (dwRC == USBIO_ERR_SUCCESS)
		{
		GetDeviceDescriptor();
		}
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		GetConfigurationDescriptor();
		}
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		GetStringDescriptor();
		}
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		SetConfiguration();
		}
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		// In order to give the USB device-side program (t_usb)
		// enough time after getting configured to carry out
		// some device tests, we wait here for a short while
		// before proceeding:
		Delay(2000);
		GetConfigurationInfo();
		}
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		OpenPipes();
		}
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		ExchangeVersions();
		}
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		DoTransfers();
		}
	if (dwRC == USBIO_ERR_SUCCESS)
		{
		ClosePipes();
		}

	CloseUsbDevice();

	return 0;
	}


// --eof