2010-07-16 18:01:53 +02:00
|
|
|
/* @file pjrc_rawhid_windows.cpp
|
|
|
|
* @addtogroup GCSPlugins GCS Plugins
|
|
|
|
* @{
|
|
|
|
* @addtogroup RawHIDPlugin Raw HID Plugin
|
|
|
|
* @{
|
|
|
|
* @brief Impliments a HID USB connection to the flight hardware as a QIODevice
|
|
|
|
*****************************************************************************/
|
|
|
|
|
2010-03-21 14:44:00 +01:00
|
|
|
/* Simple Raw HID functions for Windows - for use with Teensy RawHID example
|
|
|
|
* http://www.pjrc.com/teensy/rawhid.html
|
|
|
|
* Copyright (c) 2009 PJRC.COM, LLC
|
|
|
|
*
|
|
|
|
* rawhid_open - open 1 or more devices
|
|
|
|
* rawhid_recv - receive a packet
|
|
|
|
* rawhid_send - send a packet
|
|
|
|
* rawhid_close - close a device
|
|
|
|
*
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
|
|
* in the Software without restriction, including without limitation the rights
|
|
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
|
|
* furnished to do so, subject to the following conditions:
|
|
|
|
*
|
|
|
|
* The above description, website URL and copyright notice and this permission
|
|
|
|
* notice shall be included in all copies or substantial portions of the Software.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
|
|
* THE SOFTWARE.
|
|
|
|
*
|
|
|
|
* Version 1.0: Initial Release
|
|
|
|
*/
|
|
|
|
/* See: http://msdn.microsoft.com/en-us/library/ms794141.aspx */
|
|
|
|
|
|
|
|
#include "pjrc_rawhid.h"
|
|
|
|
|
|
|
|
#include <windows.h>
|
|
|
|
#include <setupapi.h>
|
|
|
|
#include <ddk/hidsdi.h>
|
|
|
|
#include <ddk/hidclass.h>
|
2010-07-29 21:36:42 +02:00
|
|
|
#include <QString>
|
2010-03-21 14:44:00 +01:00
|
|
|
|
2010-04-10 18:14:29 +02:00
|
|
|
#define printf qDebug
|
2010-03-21 14:44:00 +01:00
|
|
|
|
|
|
|
typedef struct hid_struct hid_t;
|
|
|
|
struct hid_struct {
|
|
|
|
HANDLE handle;
|
|
|
|
int open;
|
|
|
|
struct hid_struct *prev;
|
|
|
|
struct hid_struct *next;
|
|
|
|
};
|
|
|
|
hid_t *first_hid;
|
|
|
|
hid_t *last_hid;
|
|
|
|
HANDLE rx_event;
|
|
|
|
HANDLE tx_event;
|
|
|
|
CRITICAL_SECTION rx_mutex;
|
|
|
|
CRITICAL_SECTION tx_mutex;
|
|
|
|
|
|
|
|
static void add_hid(hid_t *h);
|
|
|
|
static hid_t* get_hid(int num);
|
|
|
|
static void free_all_hid(void);
|
|
|
|
static void hid_close(hid_t *hid);
|
|
|
|
static void print_win32_err(void);
|
|
|
|
|
|
|
|
pjrc_rawhid::pjrc_rawhid()
|
|
|
|
{
|
|
|
|
first_hid = NULL;
|
|
|
|
last_hid = NULL;
|
|
|
|
rx_event = NULL;
|
|
|
|
tx_event = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
// open - open 1 or more devices
|
|
|
|
//
|
|
|
|
// Inputs:
|
|
|
|
// max = maximum number of devices to open
|
|
|
|
// vid = Vendor ID, or -1 if any
|
|
|
|
// pid = Product ID, or -1 if any
|
|
|
|
// usage_page = top level usage page, or -1 if any
|
|
|
|
// usage = top level usage number, or -1 if any
|
|
|
|
// Output:
|
|
|
|
// actual number of devices opened
|
|
|
|
//
|
|
|
|
int pjrc_rawhid::open(int max, int vid, int pid, int usage_page, int usage)
|
|
|
|
{
|
|
|
|
GUID guid;
|
|
|
|
HDEVINFO info;
|
|
|
|
DWORD index=0, reqd_size;
|
|
|
|
SP_DEVICE_INTERFACE_DATA iface;
|
|
|
|
SP_DEVICE_INTERFACE_DETAIL_DATA *details;
|
|
|
|
HIDD_ATTRIBUTES attrib;
|
|
|
|
PHIDP_PREPARSED_DATA hid_data;
|
|
|
|
HIDP_CAPS capabilities;
|
|
|
|
HANDLE h;
|
|
|
|
BOOL ret;
|
|
|
|
hid_t *hid;
|
|
|
|
int count=0;
|
|
|
|
|
|
|
|
if (first_hid) free_all_hid();
|
|
|
|
if (max < 1) return 0;
|
|
|
|
if (!rx_event) {
|
|
|
|
rx_event = CreateEvent(NULL, TRUE, TRUE, NULL);
|
|
|
|
tx_event = CreateEvent(NULL, TRUE, TRUE, NULL);
|
|
|
|
InitializeCriticalSection(&rx_mutex);
|
|
|
|
InitializeCriticalSection(&tx_mutex);
|
|
|
|
}
|
|
|
|
HidD_GetHidGuid(&guid);
|
|
|
|
info = SetupDiGetClassDevs(&guid, NULL, NULL, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE);
|
|
|
|
if (info == INVALID_HANDLE_VALUE) return 0;
|
|
|
|
for (index=0; 1 ;index++) {
|
|
|
|
iface.cbSize = sizeof(SP_DEVICE_INTERFACE_DATA);
|
|
|
|
ret = SetupDiEnumDeviceInterfaces(info, NULL, &guid, index, &iface);
|
|
|
|
if (!ret) return count;
|
|
|
|
SetupDiGetInterfaceDeviceDetail(info, &iface, NULL, 0, &reqd_size, NULL);
|
|
|
|
details = (SP_DEVICE_INTERFACE_DETAIL_DATA *)malloc(reqd_size);
|
|
|
|
if (details == NULL) continue;
|
|
|
|
|
|
|
|
memset(details, 0, reqd_size);
|
|
|
|
details->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);
|
|
|
|
ret = SetupDiGetDeviceInterfaceDetail(info, &iface, details,
|
|
|
|
reqd_size, NULL, NULL);
|
|
|
|
if (!ret) {
|
|
|
|
free(details);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
h = CreateFile(details->DevicePath, GENERIC_READ|GENERIC_WRITE,
|
|
|
|
FILE_SHARE_READ|FILE_SHARE_WRITE, NULL,
|
|
|
|
OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL);
|
|
|
|
free(details);
|
|
|
|
if (h == INVALID_HANDLE_VALUE) continue;
|
|
|
|
attrib.Size = sizeof(HIDD_ATTRIBUTES);
|
|
|
|
ret = HidD_GetAttributes(h, &attrib);
|
2010-04-11 23:08:46 +02:00
|
|
|
//printf("vid: %4x\n", attrib.VendorID);
|
2010-03-21 14:44:00 +01:00
|
|
|
if (!ret || (vid > 0 && attrib.VendorID != vid) ||
|
|
|
|
(pid > 0 && attrib.ProductID != pid) ||
|
|
|
|
!HidD_GetPreparsedData(h, &hid_data)) {
|
|
|
|
CloseHandle(h);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (!HidP_GetCaps(hid_data, &capabilities) ||
|
|
|
|
(usage_page > 0 && capabilities.UsagePage != usage_page) ||
|
|
|
|
(usage > 0 && capabilities.Usage != usage)) {
|
|
|
|
HidD_FreePreparsedData(hid_data);
|
|
|
|
CloseHandle(h);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
HidD_FreePreparsedData(hid_data);
|
|
|
|
hid = (struct hid_struct *)malloc(sizeof(struct hid_struct));
|
|
|
|
if (!hid) {
|
|
|
|
CloseHandle(h);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
hid->handle = h;
|
|
|
|
hid->open = 1;
|
|
|
|
add_hid(hid);
|
|
|
|
count++;
|
|
|
|
if (count >= max) return count;
|
|
|
|
}
|
|
|
|
return count;
|
|
|
|
}
|
|
|
|
|
|
|
|
// recveive - receive a packet
|
|
|
|
// Inputs:
|
|
|
|
// num = device to receive from (zero based)
|
|
|
|
// buf = buffer to receive packet
|
|
|
|
// len = buffer's size
|
|
|
|
// timeout = time to wait, in milliseconds
|
|
|
|
// Output:
|
|
|
|
// number of bytes received, or -1 on error
|
|
|
|
//
|
|
|
|
int pjrc_rawhid::receive(int num, void *buf, int len, int timeout)
|
|
|
|
{
|
|
|
|
hid_t *hid;
|
2010-04-10 18:14:29 +02:00
|
|
|
unsigned char tmpbuf[516]={0};
|
2010-03-21 14:44:00 +01:00
|
|
|
OVERLAPPED ov;
|
|
|
|
DWORD n, r;
|
|
|
|
|
2010-07-29 22:00:17 +02:00
|
|
|
if (sizeof(tmpbuf) < len) return -1;
|
2010-03-21 14:44:00 +01:00
|
|
|
hid = get_hid(num);
|
|
|
|
if (!hid || !hid->open) return -1;
|
|
|
|
EnterCriticalSection(&rx_mutex);
|
|
|
|
ResetEvent(&rx_event);
|
|
|
|
memset(&ov, 0, sizeof(ov));
|
|
|
|
ov.hEvent = rx_event;
|
2010-07-29 22:00:17 +02:00
|
|
|
if (!ReadFile(hid->handle, tmpbuf, len, NULL, &ov)) {
|
2010-03-21 14:44:00 +01:00
|
|
|
if (GetLastError() != ERROR_IO_PENDING) goto return_error;
|
|
|
|
r = WaitForSingleObject(rx_event, timeout);
|
|
|
|
if (r == WAIT_TIMEOUT) goto return_timeout;
|
|
|
|
if (r != WAIT_OBJECT_0) goto return_error;
|
|
|
|
}
|
|
|
|
if (!GetOverlappedResult(hid->handle, &ov, &n, FALSE)) goto return_error;
|
|
|
|
LeaveCriticalSection(&rx_mutex);
|
|
|
|
if (n <= 0) return -1;
|
|
|
|
n--;
|
|
|
|
if (n > len) n = len;
|
2010-07-29 22:00:17 +02:00
|
|
|
memcpy(buf, tmpbuf, n);
|
2010-03-21 14:44:00 +01:00
|
|
|
return n;
|
|
|
|
return_timeout:
|
|
|
|
CancelIo(hid->handle);
|
|
|
|
LeaveCriticalSection(&rx_mutex);
|
|
|
|
return 0;
|
|
|
|
return_error:
|
|
|
|
print_win32_err();
|
|
|
|
LeaveCriticalSection(&rx_mutex);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
// send - send a packet
|
|
|
|
// Inputs:
|
|
|
|
// num = device to transmit to (zero based)
|
|
|
|
// buf = buffer containing packet to send
|
|
|
|
// len = number of bytes to transmit
|
|
|
|
// timeout = time to wait, in milliseconds
|
|
|
|
// Output:
|
|
|
|
// number of bytes sent, or -1 on error
|
|
|
|
//
|
|
|
|
int pjrc_rawhid::send(int num, void *buf, int len, int timeout)
|
|
|
|
{
|
|
|
|
hid_t *hid;
|
2010-04-10 18:14:29 +02:00
|
|
|
unsigned char tmpbuf[64]={0};
|
2010-03-21 14:44:00 +01:00
|
|
|
OVERLAPPED ov;
|
|
|
|
DWORD n, r;
|
|
|
|
|
2010-07-29 22:02:40 +02:00
|
|
|
if (sizeof(tmpbuf) < len) return -1;
|
2010-03-21 14:44:00 +01:00
|
|
|
hid = get_hid(num);
|
|
|
|
if (!hid || !hid->open) return -1;
|
|
|
|
EnterCriticalSection(&tx_mutex);
|
|
|
|
ResetEvent(&tx_event);
|
|
|
|
memset(&ov, 0, sizeof(ov));
|
|
|
|
ov.hEvent = tx_event;
|
2010-07-29 22:02:40 +02:00
|
|
|
memcpy(tmpbuf, buf, len);
|
2010-04-10 18:14:29 +02:00
|
|
|
if (!WriteFile(hid->handle, tmpbuf, 64, NULL, &ov)) {
|
2010-03-21 14:44:00 +01:00
|
|
|
if (GetLastError() != ERROR_IO_PENDING) goto return_error;
|
|
|
|
r = WaitForSingleObject(tx_event, timeout);
|
|
|
|
if (r == WAIT_TIMEOUT) goto return_timeout;
|
|
|
|
if (r != WAIT_OBJECT_0) goto return_error;
|
|
|
|
}
|
|
|
|
if (!GetOverlappedResult(hid->handle, &ov, &n, FALSE)) goto return_error;
|
|
|
|
LeaveCriticalSection(&tx_mutex);
|
|
|
|
if (n <= 0) return -1;
|
|
|
|
return n - 1;
|
|
|
|
return_timeout:
|
|
|
|
CancelIo(hid->handle);
|
|
|
|
LeaveCriticalSection(&tx_mutex);
|
|
|
|
return 0;
|
|
|
|
return_error:
|
|
|
|
print_win32_err();
|
|
|
|
LeaveCriticalSection(&tx_mutex);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2010-07-29 21:36:42 +02:00
|
|
|
QString pjrc_rawhid::getserial(int num)
|
2010-03-21 14:44:00 +01:00
|
|
|
{
|
|
|
|
hid_t *hid;
|
|
|
|
char temp[126];
|
|
|
|
|
|
|
|
hid = get_hid(num);
|
2010-07-29 21:48:51 +02:00
|
|
|
if (!hid || !hid->open)
|
|
|
|
return "";
|
2010-03-21 14:44:00 +01:00
|
|
|
|
|
|
|
/* Should we do some "critical section" stuff here?? */
|
|
|
|
if(!HidD_GetSerialNumberString(hid->handle, temp, sizeof(temp))) {
|
|
|
|
print_win32_err();
|
2010-07-29 21:36:42 +02:00
|
|
|
return QString("Error");
|
2010-03-21 14:44:00 +01:00
|
|
|
}
|
|
|
|
|
2010-07-29 21:48:51 +02:00
|
|
|
return QString().fromUtf16((ushort*)temp,-1);
|
2010-03-21 14:44:00 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// close - close a device
|
|
|
|
//
|
|
|
|
// Inputs:
|
|
|
|
// num = device to close (zero based)
|
|
|
|
// Output
|
|
|
|
// (nothing)
|
|
|
|
//
|
|
|
|
void pjrc_rawhid::close(int num)
|
|
|
|
{
|
|
|
|
hid_t *hid;
|
|
|
|
|
|
|
|
hid = get_hid(num);
|
|
|
|
if (!hid || !hid->open) return;
|
|
|
|
hid_close(hid);
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
//
|
|
|
|
// Private Functions
|
|
|
|
//
|
|
|
|
//
|
|
|
|
static void add_hid(hid_t *h)
|
|
|
|
{
|
|
|
|
if (!first_hid || !last_hid) {
|
|
|
|
first_hid = last_hid = h;
|
|
|
|
h->next = h->prev = NULL;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
last_hid->next = h;
|
|
|
|
h->prev = last_hid;
|
|
|
|
h->next = NULL;
|
|
|
|
last_hid = h;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static hid_t * get_hid(int num)
|
|
|
|
{
|
|
|
|
hid_t *p;
|
|
|
|
for (p = first_hid; p && num > 0; p = p->next, num--) ;
|
|
|
|
return p;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static void free_all_hid(void)
|
|
|
|
{
|
|
|
|
hid_t *p, *q;
|
|
|
|
|
|
|
|
for (p = first_hid; p; p = p->next) {
|
|
|
|
hid_close(p);
|
|
|
|
}
|
|
|
|
p = first_hid;
|
|
|
|
while (p) {
|
|
|
|
q = p;
|
|
|
|
p = p->next;
|
|
|
|
free(q);
|
|
|
|
}
|
|
|
|
first_hid = last_hid = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static void hid_close(hid_t *hid)
|
|
|
|
{
|
|
|
|
CloseHandle(hid->handle);
|
|
|
|
hid->handle = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void print_win32_err(void)
|
|
|
|
{
|
|
|
|
char buf[256];
|
2010-04-10 18:14:29 +02:00
|
|
|
char temp[256];
|
2010-03-21 14:44:00 +01:00
|
|
|
DWORD err;
|
|
|
|
|
|
|
|
err = GetLastError();
|
2010-04-10 18:14:29 +02:00
|
|
|
//FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, NULL, err, 0, (WCHAR*)buf, sizeof(buf), NULL);
|
|
|
|
FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, NULL, err, MAKELANGID(LANG_ENGLISH, SUBLANG_DEFAULT), (WCHAR*)buf, sizeof(buf), NULL);
|
|
|
|
WideCharToMultiByte( CP_ACP, 0, (WCHAR*)buf, sizeof(buf), temp, sizeof(temp), NULL, NULL );
|
|
|
|
printf("err %ld: %s\n", err, temp);
|
2010-03-21 14:44:00 +01:00
|
|
|
}
|