HoHCI-BikeSim/External/BikePowerRecordingLibrary/ANT_LIB/software/USB/devices/usb_device_libusb.cpp

/*
 * Dynastream Innovations Inc.
 * Cochrane, AB, CANADA
 *
 * Copyright © 1998-2008 Dynastream Innovations Inc.
 * All rights reserved. This software may not be reproduced by
 * any means without express written approval of Dynastream
 * Innovations Inc.
 */


#include "types.h"
#if defined(DSI_TYPES_WINDOWS)

#if defined(_MSC_VER)
   #include "WinDevice.h"
#endif

#include "usb_device_libusb.hpp"

#include "macros.h"

#include "dsi_libusb_library.hpp"

#include <memory>

using namespace std;


USBDeviceLibusb::USBDeviceLibusb(struct usb_device& stDevice_)
:
   pstDevice(&stDevice_),
   usVid(stDevice_.descriptor.idVendor),
   usPid(stDevice_.descriptor.idProduct),
   ulSerialNumber(0)
{

   szProductDescription[0] = '\0';
   szSerialString[0] = '\0';

   //Get a reference to library
   auto_ptr<const LibusbLibrary> pclAutoLibusbLibrary(NULL);
   if(LibusbLibrary::Load(pclAutoLibusbLibrary) == FALSE)
      return;
   const LibusbLibrary& clLibusbLibrary = *pclAutoLibusbLibrary;


   usb_dev_handle* pstTempDeviceHandle = clLibusbLibrary.Open(&stDevice_);  //We can open the device to get the info even if someone else is using it, so this is okay.
   if(pstTempDeviceHandle == NULL)
      return;

   int ret = clLibusbLibrary.GetStringSimple(pstTempDeviceHandle, (clLibusbLibrary.Device(pstTempDeviceHandle))->descriptor.iProduct, (char*)szProductDescription, sizeof(szProductDescription));
   if(ret < 0)
   {
      szProductDescription[0] = '\0';
   }

   ret = clLibusbLibrary.GetStringSimple(pstTempDeviceHandle, (clLibusbLibrary.Device(pstTempDeviceHandle))->descriptor.iSerialNumber, (char*)szSerialString, sizeof(szSerialString));
   if(ret < 0)
   {
      szSerialString[0] = '\0';
      ulSerialNumber = 0;
   }
   else
   {
      USBDeviceLibusb::GetDeviceSerialNumber(ulSerialNumber);
   }

   clLibusbLibrary.Close(pstTempDeviceHandle);

   return;
}

USBDeviceLibusb::USBDeviceLibusb(const USBDeviceLibusb& clDevice_)
:
   pstDevice(clDevice_.pstDevice),
   usVid(clDevice_.pstDevice->descriptor.idVendor),
   usPid(clDevice_.pstDevice->descriptor.idProduct),
   ulSerialNumber(clDevice_.ulSerialNumber)
{
   STRNCPY((char*)szProductDescription, (char*)clDevice_.szProductDescription, sizeof(szProductDescription));
   memcpy(szSerialString, clDevice_.szSerialString, sizeof(szSerialString));
   return;
}


USBDeviceLibusb& USBDeviceLibusb::operator=(const USBDeviceLibusb& clDevice_)
{
   if(this == &clDevice_)
      return *this;

   pstDevice = clDevice_.pstDevice;
   usVid = clDevice_.pstDevice->descriptor.idVendor;
   usPid = clDevice_.pstDevice->descriptor.idProduct;
   ulSerialNumber = clDevice_.ulSerialNumber;
   STRNCPY((char*)szProductDescription, (char*)clDevice_.szProductDescription, sizeof(szProductDescription));
   memcpy(szSerialString, clDevice_.szSerialString, sizeof(szSerialString));

   return *this;
}

BOOL USBDeviceLibusb::USBReset() const  //!!make static?
{
#if defined(_MSC_VER)
   // !! Borland builder chokes on cfgmgr32
   TCHAR line1[64];
   TCHAR* argv_[2];

   //Only do the soft reset if we fail to open a device or it seems like theres no deivces to open (device in a "hosed" state)
   //The soft reset will have no effect on devices currently opened by other applications.
   argv_[0] = line1;
   SNPRINTF(&(line1[0]),sizeof(line1), "@USB\\VID_%04X&PID_%04X\\*", usVid, usPid);
   WinDevice_Disable(1,argv_);
   WinDevice_Enable(1,argv_);

#endif

   return TRUE;
}

BOOL USBDeviceLibusb::GetProductDescription(UCHAR* pucProductDescription_, USHORT usBufferSize_) const
{
   return(STRNCPY((char*) pucProductDescription_, (char*) szProductDescription, usBufferSize_));
}

BOOL USBDeviceLibusb::GetSerialString(UCHAR* pucSerialString_, USHORT usBufferSize_) const
{
   if(sizeof(szSerialString) > usBufferSize_)
   {
      memcpy(pucSerialString_, szSerialString, usBufferSize_);
      return FALSE;
   }

   memcpy(pucSerialString_, szSerialString, sizeof(szSerialString));
   return TRUE;
}

//The serial number actually is not limited to a ULONG by USB specs,
//so, our range here is determined by whatever we do in our products.
//For now we have it defined as 1 to (ULONG_MAX-1)
BOOL USBDeviceLibusb::GetDeviceSerialNumber(ULONG& ulSerialNumber_)
{
   ULONG ulSerial = strtoul((char*)szSerialString, NULL, 10);
   if(ulSerial == 0 || ulSerial == ULONG_MAX)
      return FALSE;

   ulSerialNumber_ = ulSerial;
   return TRUE;
}



#endif //defined(DSI_TYPES_WINDOWS)