学步园

   //========================================================================
  //TITLE:
  //    WinCE虚拟串口驱动(二)
  //AUTHOR:
  //    norains
  //DATE:
  //    Saturday 28-March-2009
  //Environment:
  //    WINDOWS CE 5.0
  //========================================================================
  
  虚拟串口驱动的完整代码如下：
  
// VirtualSerial.cpp : Defines the entry point for the DLL application.<br /> //</p> <p>#include "windows.h"<br /> #include "reg.h"<br /> #include <vector><br /> #include <Pegdser.h><br /> #include "algorithm"</p> <p>//--------------------------------------------------------------------------<br /> //Macro<br /> #define REG_ROOT_KEY HKEY_LOCAL_MACHINE<br /> #define REG_DEVICE_SUB_KEY TEXT("Drivers//Builtin//VirtualSerial")<br /> #define REG_MAP_PORT_NAME TEXT("Map_Port")</p> <p>//The buffer length for storing the read data.<br /> #define READ_BUFFER_LENGTH MAX_PATH<br /> //--------------------------------------------------------------------------<br /> //Gloabal variable<br /> HANDLE g_hCom = INVALID_HANDLE_VALUE;<br /> unsigned int g_uiOpenCount = 0;<br /> CRITICAL_SECTION g_csOpen;<br /> CRITICAL_SECTION g_csRead;<br /> CRITICAL_SECTION g_csWrite;<br /> std::vector<BYTE> g_vtBufRead(READ_BUFFER_LENGTH,0);<br /> DWORD g_dwLenReadBuf = 0;<br /> DWORD g_dwEvtMask = 0;<br /> DWORD g_dwWaitMask = 0;<br /> HANDLE g_hEventComm = NULL;<br /> BOOL g_bMonitorProcRunning = FALSE;<br /> BOOL g_bExitMonitorProc = FALSE;<br /> BOOL g_bReaded = FALSE;<br /> //--------------------------------------------------------------------------</p> <p>BOOL WINAPI DllEntry(HANDLE hInstDll, DWORD dwReason, LPVOID lpvReserved)<br /> {<br /> switch ( dwReason )<br /> {<br /> case DLL_PROCESS_ATTACH:<br /> break;<br /> }<br /> return TRUE;<br /> }</p> <p>DWORD MonitorCommEventProc(LPVOID pParam)<br /> {<br /> InterlockedExchange(reinterpret_cast<LONG *>(&g_bMonitorProcRunning),TRUE);</p> <p> RETAILMSG(TRUE,(TEXT("[VSP]:MonitorCommEventProc Running!/r/n")));</p> <p> std::vector<BYTE> vtBufRead(g_vtBufRead.size(),0);<br /> while(TRUE)<br /> {<br /> DWORD dwEvtMask = 0;<br /> BOOL bWaitRes = WaitCommEvent(g_hCom,&dwEvtMask,NULL); </p> <p> if(g_bExitMonitorProc != FALSE)<br /> {<br /> break;<br /> } </p> <p> if(bWaitRes == FALSE)<br /> {<br /> continue;<br /> } </p> <p> DWORD dwRead = 0;<br /> if(dwEvtMask & EV_RXCHAR)<br /> {<br /> EnterCriticalSection(&g_csRead); </p> <p> ReadFile(g_hCom,&g_vtBufRead[0],vtBufRead.size(),&dwRead,NULL);<br /> if(dwRead == vtBufRead.size() || g_bReaded != FALSE)<br /> {<br /> g_dwLenReadBuf = dwRead;<br /> g_vtBufRead.swap(vtBufRead);<br /> }<br /> else if(dwRead != 0)<br /> {<br /> if(g_dwLenReadBuf + dwRead <= g_vtBufRead.size())<br /> {<br /> g_dwLenReadBuf += dwRead;<br /> g_vtBufRead.insert(g_vtBufRead.end(),vtBufRead.begin(),vtBufRead.begin() + dwRead);<br /> }<br /> else<br /> {<br /> DWORD dwCover = g_dwLenReadBuf + dwRead - g_vtBufRead.size();<br /> std::copy(g_vtBufRead.begin() + dwCover,g_vtBufRead.begin() + g_dwLenReadBuf,g_vtBufRead.begin());<br /> std::copy(vtBufRead.begin(),vtBufRead.begin() + dwRead,g_vtBufRead.begin() + (g_dwLenReadBuf - dwCover));<br /> g_dwLenReadBuf = g_vtBufRead.size();<br /> }<br /> }</p> <p> g_bReaded = FALSE;</p> <p> DEBUGMSG(TRUE,(TEXT("[VSP]:Read data : %d/r/n"),dwRead)); </p> <p> LeaveCriticalSection(&g_csRead);<br /> }</p> <p> if(dwEvtMask == EV_RXCHAR && ((g_dwWaitMask & EV_RXCHAR) == 0 || dwRead == 0))<br /> {<br /> //The return event mask is only EV_RXCHAR and there is not EV_RXCHAR in the wait mask.<br /> continue;<br /> }</p> <p> InterlockedExchange(reinterpret_cast<LONG *>(&g_dwEvtMask),dwEvtMask);<br /> PulseEvent(g_hEventComm); </p> <p> //Sleep for other thread to respond to the event<br /> Sleep(100);</p> <p> DEBUGMSG(TRUE,(TEXT("[VSP]:PulseEvent! The event-mask is 0x%x/r/n"),dwEvtMask)); </p> <p> }</p> <p> RETAILMSG(TRUE,(TEXT("[VSP]:Exit the MonitorCommEventProc/r/n")));<br /> InterlockedExchange(reinterpret_cast<LONG *>(&g_bMonitorProcRunning),FALSE);</p> <p> return 0;<br /> }</p> <p>BOOL VSP_Close(DWORD dwHandle)<br /> {<br /> EnterCriticalSection(&g_csOpen);</p> <p> g_uiOpenCount --;<br /> if(g_uiOpenCount == 0)<br /> {<br /> //Notify the monitor thread to exit.<br /> InterlockedExchange(reinterpret_cast<LONG *>(&g_bExitMonitorProc),TRUE);<br /> DWORD dwMask = 0;<br /> GetCommMask(g_hCom,&dwMask);<br /> SetCommMask(g_hCom,dwMask); </p> <p> while(InterlockedExchange(reinterpret_cast<LONG *>(&g_bMonitorProcRunning),TRUE) == TRUE)<br /> {<br /> Sleep(20);<br /> }<br /> InterlockedExchange(reinterpret_cast<LONG *>(&g_bMonitorProcRunning),FALSE);</p> <p> CloseHandle(g_hCom);<br /> g_hCom = NULL;<br /> }</p> <p> LeaveCriticalSection(&g_csOpen);</p> <p> return TRUE;<br /> }</p> <p>DWORD VSP_Init(DWORD dwContext)<br /> {<br /> RETAILMSG(TRUE,(TEXT("[+VSP_Init]/r/n"))); </p> <p> InitializeCriticalSection(&g_csOpen);<br /> InitializeCriticalSection(&g_csRead);<br /> InitializeCriticalSection(&g_csWrite);</p> <p> g_hEventComm = CreateEvent(NULL,TRUE,FALSE,NULL);</p> <p> RETAILMSG(TRUE,(TEXT("[-VSP_Init]/r/n")));</p> <p> return TRUE;<br /> }</p> <p>BOOL VSP_Deinit(<br /> DWORD dwContext // future: pointer to the per disk structure<br /> )<br /> {<br /> RETAILMSG(TRUE,(TEXT("[+VSP_Deinit]/r/n"))); </p> <p> CloseHandle(g_hEventComm);<br /> g_hEventComm = NULL;</p> <p> DeleteCriticalSection(&g_csOpen);<br /> DeleteCriticalSection(&g_csRead);<br /> DeleteCriticalSection(&g_csWrite);</p> <p> RETAILMSG(TRUE,(TEXT("[-VSP_Deinit]/r/n")));<br /> return TRUE;<br /> }</p> <p>DWORD VSP_Open(<br /> DWORD dwData,<br /> DWORD dwAccess,<br /> DWORD dwShareMode<br /> )<br /> {<br /> BOOL bResult = FALSE;</p> <p> EnterCriticalSection(&g_csOpen);</p> <p> //The variable<br /> CReg reg;<br /> std::vector<TCHAR> vtBuf(MAX_PATH,0);<br /> COMMPROP commProp = {0};</p> <p> if(g_uiOpenCount != 0)<br /> {<br /> goto SET_SUCCEED_FLAG;<br /> } </p> <p> if(reg.Open(REG_ROOT_KEY,REG_DEVICE_SUB_KEY) == FALSE)<br /> {<br /> RETAILMSG(TRUE,(TEXT("[VSP]:Failed to open the registry/r/n")));<br /> goto LEAVE_CRITICAL_SECTION;<br /> }</p> <p> //Get the MAP_PORT name<br /> reg.GetValueSZ(REG_MAP_PORT_NAME,&vtBuf[0],vtBuf.size());</p> <p> g_hCom = CreateFile(&vtBuf[0],GENERIC_READ | GENERIC_WRITE ,0,NULL,OPEN_EXISTING,0,NULL);<br /> if(g_hCom == INVALID_HANDLE_VALUE )<br /> {<br /> RETAILMSG(TRUE,(TEXT("[VSP]Failed to map to %s/r/n"),&vtBuf[0]));<br /> goto LEAVE_CRITICAL_SECTION;<br /> }<br /> else<br /> {<br /> RETAILMSG(TRUE,(TEXT("[VSP]Succeed to map to %s/r/n"),&vtBuf[0]));<br /> } </p> <p> InterlockedExchange(reinterpret_cast<LONG *>(&g_bExitMonitorProc),FALSE);<br /> CloseHandle(CreateThread(NULL,NULL,MonitorCommEventProc,NULL,NULL,NULL));</p> <p>SET_SUCCEED_FLAG:</p> <p> g_uiOpenCount ++;<br /> bResult = TRUE;</p> <p>LEAVE_CRITICAL_SECTION: </p> <p> LeaveCriticalSection(&g_csOpen);</p> <p> return bResult;<br /> }</p> <p>BOOL VSP_IOControl(<br /> DWORD dwHandle,<br /> DWORD dwIoControlCode,<br /> PBYTE pBufIn,<br /> DWORD dwBufInSize,<br /> PBYTE pBufOut,<br /> DWORD dwBufOutSize,<br /> PDWORD pBytesReturned<br /> )<br /> {<br /> switch(dwIoControlCode)<br /> {<br /> case IOCTL_SERIAL_SET_DCB:<br /> {<br /> return SetCommState(g_hCom,reinterpret_cast<DCB *>(pBufIn));<br /> }<br /> case IOCTL_SERIAL_GET_DCB:<br /> {<br /> return GetCommState(g_hCom,reinterpret_cast<DCB *>(pBufOut));<br /> }<br /> case IOCTL_SERIAL_WAIT_ON_MASK:<br /> {<br /> if(dwBufOutSize < sizeof(DWORD) || WaitForSingleObject(g_hEventComm,INFINITE) == WAIT_TIMEOUT)<br /> {<br /> *pBytesReturned = 0;<br /> return FALSE;<br /> }<br /> else<br /> {<br /> InterlockedExchange(reinterpret_cast<LONG *>(pBufOut),g_dwEvtMask);<br /> *pBytesReturned = sizeof(DWORD);<br /> return TRUE;<br /> }<br /> }<br /> case IOCTL_SERIAL_SET_WAIT_MASK:<br /> {<br /> g_dwWaitMask = *reinterpret_cast<DWORD *>(pBufIn);<br /> return SetCommMask(g_hCom,g_dwWaitMask | EV_RXCHAR); //The driver need the EV_RXCHAR notify event.<br /> }<br /> case IOCTL_SERIAL_GET_WAIT_MASK:<br /> {<br /> if(dwBufOutSize < sizeof(DWORD) || GetCommMask(g_hCom,reinterpret_cast<DWORD *>(pBufOut)) == FALSE)<br /> {<br /> *pBytesReturned = 0;<br /> return FALSE;<br /> }<br /> else<br /> {<br /> *pBytesReturned = sizeof(DWORD);<br /> return TRUE;<br /> }<br /> }<br /> }</p> <p> return FALSE;<br /> }</p> <p>DWORD VSP_Read(DWORD dwHandle, LPVOID pBuffer, DWORD dwNumBytes)<br /> {<br /> EnterCriticalSection(&g_csRead); </p> <p> //The g_dwLenReadBuf must be less than or equal to g_vtBufRead.size(), so needn't compare with each other.<br /> DWORD dwCopy = g_dwLenReadBuf > dwNumBytes ? dwNumBytes : g_dwLenReadBuf;<br /> if(dwCopy != 0)<br /> {<br /> memcpy(pBuffer,&g_vtBufRead[0],dwCopy);<br /> }<br /> DEBUGMSG(TRUE,(TEXT("[VSP]:Copy cout:%d/r/n"),dwCopy));</p> <p> g_bReaded = TRUE;</p> <p> LeaveCriticalSection(&g_csRead);</p> <p> //Sleep for other thread to entry the function.<br /> Sleep(10); </p> <p> return dwCopy;<br /> }</p> <p>DWORD VSP_Write(DWORD dwHandle, LPCVOID pBuffer, DWORD dwNumBytes)<br /> {<br /> EnterCriticalSection(&g_csWrite);<br /> DWORD dwWrite = 0;<br /> WriteFile(g_hCom,pBuffer,dwNumBytes,&dwWrite,NULL);<br /> LeaveCriticalSection(&g_csWrite);<br /> return dwWrite;<br /> }</p> <p>DWORD VSP_Seek(DWORD dwHandle, long lDistance, DWORD dwMoveMethod)<br /> {<br /> return FALSE;<br /> }</p> <p>void VSP_PowerUp(void)<br /> {<br /> return;<br /> }</p> <p>void VSP_PowerDown(void)<br /> {<br /> return;<br /> }<br />

  
  不过该驱动代码是作者量身定做的，像IOControl就简单地实现了几个，其余的因为在实际使用中本人没用到，所以都没实现，只是简单地返回了FALSE。如果有朋友对此有兴趣，并且实际中也使用到，可以自行调用原生函数实现。
  
  最后，是能让驱动正常挂载的注册表设置：
  
[HKEY_LOCAL_MACHINE/Drivers/Builtin/VirtualSerial]
    "Prefix"="VSP"
    "Dll"="VirtualSerial.dll"
    "Order"=dword:0
    "Index"=dword:1
    "Map_Port"="COM1:"