usbdiagsession.cpp

此程序是利用VC实现了USB的驱动程序。

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

#include "stdafx.h"
#include "UsbDiagSession.h"
#include "Ezusbsys.h"
#include <mmsystem.h>
#include <winioctl.h>
#include "Journal.h"

extern CJournal m_Journal;

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif
extern HANDLE hTimeDelayEvent;
extern HWND TheHandleFileMakerDlg;

BOOLEAN
bOpenDriver (HANDLE * phDeviceHandle, PCHAR devname)
{
   char completeDeviceName[64] = "";
   
   strcat (completeDeviceName,
      "\\\\.\\"
      );
   
   strcat (completeDeviceName,
		    devname
          );
   
   *phDeviceHandle = CreateFile(completeDeviceName,
      GENERIC_WRITE,
      FILE_SHARE_WRITE,
      NULL,
      OPEN_EXISTING,
      0,
      NULL);
   
   if (*phDeviceHandle == INVALID_HANDLE_VALUE) {
      return (FALSE);
   } else {
      return (TRUE);
   }  
}

UsbDiagSession::UsbDiagSession()
{
	USBIntied=FALSE;
	hDataBuffer=DataBuf;
}

UsbDiagSession::~UsbDiagSession() 
{
}

void UsbDiagSession::DoResetDevice()
{
	HANDLE  hDevice = NULL;
	DWORD     nBytes = 0;
	if (bOpenDriver (&hDevice, _T("Ezusb-0")) != TRUE)
	{
	  return;
	}
	BOOL bResult =DeviceIoControl (hDevice,
		   IOCTL_Myusb_RESET,
		   NULL,
		   0,
		   NULL,
		   0,
		   &nBytes,
		   NULL);
	CloseHandle(hDevice);
}

BOOL UsbDiagSession::Init()
{
	HANDLE  hDevice = NULL;
	if (bOpenDriver(&hDevice, _T("Ezusb-0")) != TRUE)
    {	USBIntied=FALSE;
        return FALSE;
    }
	CloseHandle(hDevice);
	USBIntied=TRUE;
    return TRUE;
}

void UsbDiagSession::DoResetPipe(ULONG nPipe)
{
	HANDLE  hDevice = NULL;
	DWORD     nBytes = 0;
	ULONG nPipeNum;
	if (bOpenDriver (&hDevice, _T("Ezusb-0")) != TRUE)
	{	USBIntied=FALSE;
		return;
	}
	nPipeNum=(nPipe&0x0F)-1;
	BOOL bResult =DeviceIoControl (hDevice,
		   IOCTL_Myusb_RESETPIPE,
		   &nPipeNum,
		   sizeof(ULONG),
		   NULL,
		   0,
		   &nBytes,
		   NULL);
	CloseHandle(hDevice);
}

BOOL UsbDiagSession::TransferSetupPacket(BYTE bt1, BYTE bt2, BYTE bt3, BYTE bt4,unsigned char * pDataBuf)
{
	HANDLE  hDevice = NULL;
	FOURBYTE_CTRL_TRANSFER inBuffer;
	DWORD     nBytes  = 0;
	inBuffer.byte1 =bt1;
	inBuffer.byte2 =bt2;
	inBuffer.byte3 =bt3;
	inBuffer.byte4 =bt4;
	inBuffer.bTimeOut  = 50;
	for(int i=0;i<3;i++)
	{
		for(int j=0;j<3;j++)
		{
			if (bOpenDriver (&hDevice, _T("Ezusb-0")) != TRUE)
			{	USBIntied=FALSE;
				return FALSE;
			}
			memset(pDataBuf,0,64);
			BOOL bResult = DeviceIoControl (hDevice,
					 IOCTL_CTRL_TRANS,
					 &inBuffer,
					 sizeof(FOURBYTE_CTRL_TRANSFER),
					 &pDataBuf[0],
					 64,
					 &nBytes,
					 NULL);
			CloseHandle(hDevice);
			if(bResult)
			{
				//char buf[100];
				//sprintf(buf,"returndata4-7: %X %X %X %X",pDataBuf[4],pDataBuf[5],pDataBuf[6],pDataBuf[7]);
				//m_Journal.AddJournalLog (buf);
				if( pDataBuf[0] == bt1 )
				{
					TransferTimeOut = ( (pDataBuf[1] & 0x40) > 0 );	//传输超时标志
					SampleFinished =  ( (pDataBuf[1] & 0x20) > 0 );	//采样完成标志
					HaveDataToSendBack = ( (pDataBuf[1] & 0x10) > 0 );	//有数据回传
					StateOfReadReady =  ( pDataBuf[1] & 0x0c ) >> 2;	//数据读取前准备状态
					StateOfDateCheck = ( pDataBuf[1] & 0x03 );	//数据写入后解析状态
					StateOfFLashPoc = ( pDataBuf[2] & 0x07 );	//数据更新状态

					UpdateDataMode = ( pDataBuf[2] & 0x80 ) > 0 ? 1:0;	//数据更新模式

					ErrorCommond = FALSE;	//指令传输标志
					if ( (pDataBuf[1] & 0x80) > 0 )
					{
						ErrorCommond = TRUE;
						return FALSE;
					}				

					switch(bt1)
					{
						case 0xc1:
							break;
						case 0xc2:
							break;
						default:
							if ( CanUseBufferCounter != pDataBuf[3] )		//可用缓冲个数
							{	
								CanUseBufferCounter = pDataBuf[3];
								PostMessage(TheHandleFileMakerDlg,MW_SOMEONEMSG, 22, CanUseBufferCounter);
							}

							SendBackPackageNum = pDataBuf[5];	//回送数据包数
							LogInfoPackageNum = pDataBuf[4];	//日志数据包数
							TotalGroupNum = pDataBuf[7];	//返回组个数
							break;
					}
				
					USBIntied=TRUE;		//已经初始化标志
					return TRUE;
				}
			}
			else
			{
				m_Journal.AddJournalLog ("Setup Transfer TimeOut.");
				DoResetDevice();
				Sleep(1000);
			}
			Sleep(10);
		}
		Sleep(30);
	}
	USBIntied=FALSE;
	return FALSE;
}


BOOL UsbDiagSession::DoRead(DWORD dwBytes, unsigned char *pData)
{
	HANDLE  hDevice = NULL;
	DWORD     nBytes   = 0;
    BULK_TRANSFER_CONTROL bulkControl;
	BOOL bResult=FALSE;

	if (bOpenDriver (&hDevice, _T("Ezusb-0")) != TRUE)
	{	USBIntied=FALSE;
		return FALSE; 
	}
	UCHAR* 	pBuffer=(UCHAR*)malloc(dwBytes);
	if(pBuffer)
	{
		memset(pBuffer,0,dwBytes);
		bulkControl.pipeNum = 4;
		bulkControl.bTimeOut = 500;
		bResult = DeviceIoControl (hDevice,
					 IOCTL_Myusb_BULK_READ,
					 &bulkControl,
					 sizeof(BULK_TRANSFER_CONTROL),
					 &pBuffer[0],
					 dwBytes,
					 &nBytes,
					 NULL);
		if(bResult)		//read success.
		{
			memcpy(pData,pBuffer,nBytes);
		}
		else
		{
			m_Journal.AddJournalLog ("Dataread TimeOut.");
		}
		free(pBuffer);
	}
	CloseHandle(hDevice);
	return bResult;
}

BOOL UsbDiagSession::DoWrite(DWORD dwBytes, unsigned char *pData)
{
	HANDLE  hDevice = NULL;
	DWORD     nBytes                          = 0;
    BULK_TRANSFER_CONTROL bulkControl;
	BOOL bResult=FALSE;

	if (bOpenDriver (&hDevice, _T("Ezusb-0")) != TRUE)
	{	USBIntied=FALSE;
		return FALSE; 
	}
	if(pData)
	{
		bulkControl.pipeNum = 5;
		bulkControl.bTimeOut = 500;
		bResult = DeviceIoControl (hDevice,
					 IOCTL_Myusb_BULK_WRITE,
					 &bulkControl,
					 sizeof(BULK_TRANSFER_CONTROL),
					 pData,
					 dwBytes,
					 &nBytes,
					 NULL);
	}
	CloseHandle(hDevice);
	return bResult;
}

BOOL UsbDiagSession::StartIsoStream(void)
{	
	BOOL	bResult = FALSE;
	DWORD	nBytes  = 0;
	ISO_TRANSFER_CONTROL 	IsoControl;
	HANDLE  hDevice = NULL;

	if (bOpenDriver (&hDevice, _T("Ezusb-0")) != TRUE)
	{	USBIntied=FALSE;
		return FALSE;
	}
    
	// reset the pipe
    DWORD	pipenum = 4;
    bResult = DeviceIoControl (hDevice,
             IOCTL_Myusb_RESETPIPE,
             &pipenum,
             sizeof(ULONG),
             NULL,
             0,
             &nBytes,
             NULL);

     if (bResult != TRUE)
     {
        CloseHandle (hDevice);
		USBIntied=FALSE;
        return FALSE;
     }

    // perform the ISO transfer
	IsoControl.PacketSize = 9;
	IsoControl.PacketCount = 64;
    IsoControl.PipeNum = 4;
    IsoControl.BufferCount = 2;
    IsoControl.FramesPerBuffer = 10;

     bResult = DeviceIoControl (hDevice,
                    IOCTL_Myusb_START_ISO_STREAM,
                    &IsoControl,
                    sizeof(ISO_TRANSFER_CONTROL),
                    NULL,
                    0,
                    &nBytes,
                    NULL);

     if (bResult != TRUE)
     {
        CloseHandle (hDevice);
		USBIntied=FALSE;
        return FALSE;
     }

    CloseHandle (hDevice);
	return TRUE;
}

BOOL UsbDiagSession::StopIsoStream(void)
{
	BOOL	bResult = FALSE;
	DWORD	nBytes  = 0;
	HANDLE  hDevice = NULL;

	if (bOpenDriver (&hDevice, _T("Ezusb-0")) != TRUE)
	{	USBIntied=FALSE;
		return FALSE;
	}

     bResult = DeviceIoControl (hDevice,
                    IOCTL_Myusb_STOP_ISO_STREAM,
                    NULL,
                    0,
                    NULL,
                    0,
                    &nBytes,
                    NULL);

     if (bResult != TRUE)
     {
        CloseHandle (hDevice);
		USBIntied=FALSE;
        return FALSE;
	 }

     CloseHandle (hDevice);
	 return TRUE;
}

/*
int UsbDiagSession::DataCRCcheck(int nByte,BYTE *pBufData)		//CRC校验
{
	int c,crc=0;
	int i,j;
	for( i=0; i< nByte; i++ )
	{
		c = *(pBufData + i);
		for(j=0;j<8;j++)
		{
			if ((crc ^ c) & 1)
				crc=(crc>>1)^0xA001;	// 0xA001 crc16 mask
			else
				crc>>=1;

			c>>=1;
		}
	}
	return crc;
}
*/

unsigned short UsbDiagSession::DataCRCcheck(int len,BYTE *ptr)		//CRC校验
{
	unsigned short crc;
	unsigned char da;
	unsigned short crc_ta[256]=
	{ /* CRC余式表 */
		0x0000, 0x1021, 0x2042, 0x3063, 0x4084, 0x50a5, 0x60c6, 0x70e7,
		0x8108, 0x9129, 0xa14a, 0xb16b, 0xc18c, 0xd1ad, 0xe1ce, 0xf1ef,
		0x1231, 0x0210, 0x3273, 0x2252, 0x52b5, 0x4294, 0x72f7, 0x62d6,
		0x9339, 0x8318, 0xb37b, 0xa35a, 0xd3bd, 0xc39c, 0xf3ff, 0xe3de,
		0x2462, 0x3443, 0x0420, 0x1401, 0x64e6, 0x74c7, 0x44a4, 0x5485,
		0xa56a, 0xb54b, 0x8528, 0x9509, 0xe5ee, 0xf5cf, 0xc5ac, 0xd58d,
		0x3653, 0x2672, 0x1611, 0x0630, 0x76d7, 0x66f6, 0x5695, 0x46b4,
		0xb75b, 0xa77a, 0x9719, 0x8738, 0xf7df, 0xe7fe, 0xd79d, 0xc7bc,
		0x48c4, 0x58e5, 0x6886, 0x78a7, 0x0840, 0x1861, 0x2802, 0x3823,
		0xc9cc, 0xd9ed, 0xe98e, 0xf9af, 0x8948, 0x9969, 0xa90a, 0xb92b,
		0x5af5, 0x4ad4, 0x7ab7, 0x6a96, 0x1a71, 0x0a50, 0x3a33, 0x2a12,
		0xdbfd, 0xcbdc, 0xfbbf, 0xeb9e, 0x9b79, 0x8b58, 0xbb3b, 0xab1a,
		0x6ca6, 0x7c87, 0x4ce4, 0x5cc5, 0x2c22, 0x3c03, 0x0c60, 0x1c41,
		0xedae, 0xfd8f, 0xcdec, 0xddcd, 0xad2a, 0xbd0b, 0x8d68, 0x9d49,
		0x7e97, 0x6eb6, 0x5ed5, 0x4ef4, 0x3e13, 0x2e32, 0x1e51, 0x0e70,
		0xff9f, 0xefbe, 0xdfdd, 0xcffc, 0xbf1b, 0xaf3a, 0x9f59, 0x8f78,
		0x9188, 0x81a9, 0xb1ca, 0xa1eb, 0xd10c, 0xc12d, 0xf14e, 0xe16f,
		0x1080, 0x00a1, 0x30c2, 0x20e3, 0x5004, 0x4025, 0x7046, 0x6067,
		0x83b9, 0x9398, 0xa3fb, 0xb3da, 0xc33d, 0xd31c, 0xe37f, 0xf35e,
		0x02b1, 0x1290, 0x22f3, 0x32d2, 0x4235, 0x5214, 0x6277, 0x7256,
		0xb5ea, 0xa5cb, 0x95a8, 0x8589, 0xf56e, 0xe54f, 0xd52c, 0xc50d,
		0x34e2, 0x24c3, 0x14a0, 0x0481, 0x7466, 0x6447, 0x5424, 0x4405,
		0xa7db, 0xb7fa, 0x8799, 0x97b8, 0xe75f, 0xf77e, 0xc71d, 0xd73c,
		0x26d3, 0x36f2, 0x0691, 0x16b0, 0x6657, 0x7676, 0x4615, 0x5634,
		0xd94c, 0xc96d, 0xf90e, 0xe92f, 0x99c8, 0x89e9, 0xb98a, 0xa9ab,
		0x5844, 0x4865, 0x7806, 0x6827, 0x18c0, 0x08e1, 0x3882, 0x28a3,
		0xcb7d, 0xdb5c, 0xeb3f, 0xfb1e, 0x8bf9, 0x9bd8, 0xabbb, 0xbb9a,
		0x4a75, 0x5a54, 0x6a37, 0x7a16, 0x0af1, 0x1ad0, 0x2ab3, 0x3a92,
		0xfd2e, 0xed0f, 0xdd6c, 0xcd4d, 0xbdaa, 0xad8b, 0x9de8, 0x8dc9,
		0x7c26, 0x6c07, 0x5c64, 0x4c45, 0x3ca2, 0x2c83, 0x1ce0, 0x0cc1,
		0xef1f, 0xff3e, 0xcf5d, 0xdf7c, 0xaf9b, 0xbfba, 0x8fd9, 0x9ff8,
		0x6e17, 0x7e36, 0x4e55, 0x5e74, 0x2e93, 0x3eb2, 0x0ed1, 0x1ef0
	};

	crc=0;
	while(len--!=0)
	{
		da=(unsigned char) (crc/256); /* 以8位二进制数的形式暂存CRC的高8位 */
		crc<<=8; /* 左移8位，相当于CRC的低8位乘以 */
		crc^=crc_ta[da^*ptr]; /* 高8位和当前字节相加后再查表求CRC ，再加上以前的CRC */
		ptr++;
	}

	return(crc);
}