枚举系统中的串口

#ifndef EnumPortsEx_H
#define EnumPortsEx_H

 

//

 

#include <Windows.h>
#include <Shlwapi.h>
#include <WinSpool.h>

 

//

 

#pragma comment(lib, "Shlwapi")

 

//

 

template <typename CallbackT>
DWORD EnumPortsEx(CallbackT callback, DWORD dwContext)
{
 DWORD dwBufferSize  = 0;
 DWORD dwReturnSize  = 0;

 LPBYTE pBuffer   = NULL;
 PPORT_INFO_1 pPortInfo = NULL;
 TCHAR szPortName[128] = { 0 };

 SetLastError(ERROR_SUCCESS);
 EnumPorts(NULL, 1, NULL, 0, &dwBufferSize, &dwReturnSize);
 if (dwBufferSize == 0) {
  goto Exit;
 }

 pBuffer = new BYTE[dwBufferSize];
 if (!EnumPorts(NULL, 1, pBuffer, dwBufferSize, &dwBufferSize, &dwReturnSize)) {
  goto Exit;
 }
 pPortInfo = (PPORT_INFO_1) pBuffer;

 for (LONG i=0; i<dwReturnSize; ++i) {
  LPTSTR pszColon = StrChr(pPortInfo[i].pName, TEXT(':'));
  if (pszColon != NULL) {
   StrCpyN(szPortName, pPortInfo[i].pName, pszColon - pPortInfo[i].pName + 1);
  }
  else {
   StrCpy(szPortName, pPortInfo[i].pName);
  }

  callback(szPortName, dwContext);
 }

Exit:
 delete []pBuffer;
 pBuffer  = NULL;
 pPortInfo = NULL;

 return GetLastError();
}

 

//

 

#endif