view/rangesCtrl/IsUsbUrgCom.cpp

/*!
  \file
  \brief URG の USB 接続ポートかを判断する

  \author Satofumi KAMIMURA

  $Id$
*/

#include "IsUsbUrgCom.h"
#include "DetectOS.h"
#ifdef WINDOWS_OS
#include <windows.h>
#include <string>
#endif

#ifdef WINDOWS_OS
static bool existRegValue(HKEY hkey, LPCSTR subkey, const char* find_value) {

  HKEY next_hkey;
  if (RegOpenKeyExA(hkey, subkey, 0, KEY_READ, &next_hkey) != ERROR_SUCCESS) {
    return false;
  }
  enum { MaxLength = 255 };

  // find_value が存在するかを探索
  CHAR device[MaxLength];
  char name[MaxLength];

  DWORD ret = ERROR_SUCCESS;
  for (int i = 0; ret == ERROR_SUCCESS; ++i) {
    DWORD dl = MaxLength, nl = MaxLength;
    ret = RegEnumValueA(hkey, i, device, &dl, NULL, NULL, (BYTE*)name, &nl);
    if (! strcmp(name, find_value)) {
      RegCloseKey(next_hkey);
      return true;
    }
  }

  // find_value が存在しなければ、再帰探索を行う
  char next_subkey[MaxLength];
  FILETIME filetime;

  ret = ERROR_SUCCESS;
  for (int i = 0; ret == ERROR_SUCCESS; ++i) {
    DWORD dl = MaxLength, nl = MaxLength;
    ret = RegEnumKeyExA(next_hkey, i, next_subkey,
                       &dl, NULL, NULL, &nl, &filetime);

    bool value_exist = existRegValue(next_hkey, next_subkey, find_value);
    if (value_exist) {
      RegCloseKey(next_hkey);
      return true;
    }
  }

  RegCloseKey(next_hkey);
  return false;
}


bool beego::isUsbUrgCom(const char* com_port) {

  // "URG Series USB Device Driver (COMx)" が Value に含まれているか検索
  std::string value_pattern =
    std::string("URG Series USB Device Driver (") + com_port + ")";

  bool ret = existRegValue(HKEY_LOCAL_MACHINE,
                           "SYSTEM\\CurrentControlSet\\Enum\\USB",
                           value_pattern.c_str());
  if (ret) {
    // 見つかったら戻る
    return ret;
  }

  // "URG-X002 USB Device Driver (COMx)" が Value に含まれているか検索
  value_pattern = std::string("URG-X002 USB Device Driver (") + com_port + ")";
  
  return existRegValue(HKEY_LOCAL_MACHINE,
                       "SYSTEM\\CurrentControlSet\\Enum\\USB",
                       value_pattern.c_str());
}
#else

bool beego::isUsbUrgCom(const char* com_port) {
  return false;
}
#endif