First cut
This commit is contained in:
194
usb/descriptor.go
Normal file
194
usb/descriptor.go
Normal file
@@ -0,0 +1,194 @@
|
||||
package usb
|
||||
|
||||
// #cgo LDFLAGS: -lusb-1.0
|
||||
// #include <libusb-1.0/libusb.h>
|
||||
import "C"
|
||||
|
||||
type Descriptor struct {
|
||||
desc *C.struct_libusb_device_descriptor
|
||||
|
||||
Type DescriptorType // The type of this descriptor
|
||||
Spec BCD // USB Specification Release Number
|
||||
Class Class // The class of this device
|
||||
SubClass uint8 // The sub-class (within the class) of this device
|
||||
Protocol uint8 // The protocol (within the sub-class) of this device
|
||||
Vendor ID // The 8-bit Vendor identifer
|
||||
Product ID // The 8-bit Product identifier
|
||||
Device BCD // The device version
|
||||
Configs int // Number of configurations
|
||||
}
|
||||
|
||||
func newDescriptor(dev *C.libusb_device) (*Descriptor, error) {
|
||||
var desc C.struct_libusb_device_descriptor
|
||||
if errno := C.libusb_get_device_descriptor(dev, &desc); errno < 0 {
|
||||
return nil, usbError(errno)
|
||||
}
|
||||
return &Descriptor{
|
||||
desc: &desc,
|
||||
Type: DescriptorType(desc.bDescriptorType),
|
||||
Spec: BCD(desc.bcdUSB),
|
||||
Class: Class(desc.bDeviceClass),
|
||||
SubClass: uint8(desc.bDeviceSubClass),
|
||||
Protocol: uint8(desc.bDeviceProtocol),
|
||||
Vendor: ID(desc.idVendor),
|
||||
Product: ID(desc.idProduct),
|
||||
Device: BCD(desc.bcdDevice),
|
||||
Configs: int(desc.bNumConfigurations),
|
||||
}, nil
|
||||
}
|
||||
|
||||
/*
|
||||
func (d Descriptor) str(idx int) (string, error) {
|
||||
str := [64]byte{}
|
||||
n := C.libusb_get_string_descriptor_ascii(
|
||||
d.dev.handle,
|
||||
C.uint8_t(idx),
|
||||
(*C.uchar)(unsafe.Pointer(&str)),
|
||||
64,
|
||||
)
|
||||
if n < 0 {
|
||||
return "", usbError(n)
|
||||
}
|
||||
return string(str[:n]), nil
|
||||
}
|
||||
*/
|
||||
|
||||
type Class int
|
||||
|
||||
const (
|
||||
CLASS_PER_INTERFACE Class = C.LIBUSB_CLASS_PER_INTERFACE
|
||||
CLASS_AUDIO Class = C.LIBUSB_CLASS_AUDIO
|
||||
CLASS_COMM Class = C.LIBUSB_CLASS_COMM
|
||||
CLASS_HID Class = C.LIBUSB_CLASS_HID
|
||||
CLASS_PRINTER Class = C.LIBUSB_CLASS_PRINTER
|
||||
CLASS_PTP Class = C.LIBUSB_CLASS_PTP
|
||||
CLASS_MASS_STORAGE Class = C.LIBUSB_CLASS_MASS_STORAGE
|
||||
CLASS_HUB Class = C.LIBUSB_CLASS_HUB
|
||||
CLASS_DATA Class = C.LIBUSB_CLASS_DATA
|
||||
CLASS_WIRELESS Class = C.LIBUSB_CLASS_WIRELESS
|
||||
CLASS_APPLICATION Class = C.LIBUSB_CLASS_APPLICATION
|
||||
CLASS_VENDOR_SPEC Class = C.LIBUSB_CLASS_VENDOR_SPEC
|
||||
)
|
||||
|
||||
var classDescription = map[Class]string{
|
||||
CLASS_PER_INTERFACE: "per-interface",
|
||||
CLASS_AUDIO: "audio",
|
||||
CLASS_COMM: "communications",
|
||||
CLASS_HID: "human interface device",
|
||||
CLASS_PRINTER: "printer dclass",
|
||||
CLASS_PTP: "picture transfer protocol",
|
||||
CLASS_MASS_STORAGE: "mass storage",
|
||||
CLASS_HUB: "hub",
|
||||
CLASS_DATA: "data",
|
||||
CLASS_WIRELESS: "wireless",
|
||||
CLASS_APPLICATION: "application",
|
||||
CLASS_VENDOR_SPEC: "vendor-specific",
|
||||
}
|
||||
|
||||
func (c Class) String() string {
|
||||
return classDescription[c]
|
||||
}
|
||||
|
||||
type DescriptorType int
|
||||
|
||||
const (
|
||||
DT_DEVICE DescriptorType = C.LIBUSB_DT_DEVICE
|
||||
DT_CONFIG DescriptorType = C.LIBUSB_DT_CONFIG
|
||||
DT_STRING DescriptorType = C.LIBUSB_DT_STRING
|
||||
DT_INTERFACE DescriptorType = C.LIBUSB_DT_INTERFACE
|
||||
DT_ENDPOINT DescriptorType = C.LIBUSB_DT_ENDPOINT
|
||||
DT_HID DescriptorType = C.LIBUSB_DT_HID
|
||||
DT_REPORT DescriptorType = C.LIBUSB_DT_REPORT
|
||||
DT_PHYSICAL DescriptorType = C.LIBUSB_DT_PHYSICAL
|
||||
DT_HUB DescriptorType = C.LIBUSB_DT_HUB
|
||||
)
|
||||
|
||||
var descriptorTypeDescription = map[DescriptorType]string{
|
||||
DT_DEVICE: "device",
|
||||
DT_CONFIG: "configuration",
|
||||
DT_STRING: "string",
|
||||
DT_INTERFACE: "interface",
|
||||
DT_ENDPOINT: "endpoint",
|
||||
DT_HID: "HID",
|
||||
DT_REPORT: "HID report",
|
||||
DT_PHYSICAL: "physical",
|
||||
DT_HUB: "hub",
|
||||
}
|
||||
|
||||
func (dt DescriptorType) String() string {
|
||||
return descriptorTypeDescription[dt]
|
||||
}
|
||||
|
||||
type EndpointDirection int
|
||||
|
||||
const (
|
||||
ENDPOINT_IN EndpointDirection = C.LIBUSB_ENDPOINT_IN
|
||||
ENDPOINT_OUT EndpointDirection = C.LIBUSB_ENDPOINT_OUT
|
||||
)
|
||||
|
||||
var endpointDirectionDescription = map[EndpointDirection]string{
|
||||
ENDPOINT_IN: "device-to-host",
|
||||
ENDPOINT_OUT: "host-to-device",
|
||||
}
|
||||
|
||||
func (ed EndpointDirection) String() string {
|
||||
return endpointDirectionDescription[ed]
|
||||
}
|
||||
|
||||
type TransferType int
|
||||
|
||||
const (
|
||||
TRANSFER_TYPE_CONTROL TransferType = C.LIBUSB_TRANSFER_TYPE_CONTROL
|
||||
TRANSFER_TYPE_ISOCHRONOUS TransferType = C.LIBUSB_TRANSFER_TYPE_ISOCHRONOUS
|
||||
TRANSFER_TYPE_BULK TransferType = C.LIBUSB_TRANSFER_TYPE_BULK
|
||||
TRANSFER_TYPE_INTERRUPT TransferType = C.LIBUSB_TRANSFER_TYPE_INTERRUPT
|
||||
)
|
||||
|
||||
var transferTypeDescription = map[TransferType]string{
|
||||
TRANSFER_TYPE_CONTROL: "control",
|
||||
TRANSFER_TYPE_ISOCHRONOUS: "isochronous",
|
||||
TRANSFER_TYPE_BULK: "bulk",
|
||||
TRANSFER_TYPE_INTERRUPT: "interrupt",
|
||||
}
|
||||
|
||||
func (tt TransferType) String() string {
|
||||
return transferTypeDescription[tt]
|
||||
}
|
||||
|
||||
type IsoSyncType int
|
||||
|
||||
const (
|
||||
ISO_SYNC_TYPE_NONE IsoSyncType = C.LIBUSB_ISO_SYNC_TYPE_NONE
|
||||
ISO_SYNC_TYPE_ASYNC IsoSyncType = C.LIBUSB_ISO_SYNC_TYPE_ASYNC
|
||||
ISO_SYNC_TYPE_ADAPTIVE IsoSyncType = C.LIBUSB_ISO_SYNC_TYPE_ADAPTIVE
|
||||
ISO_SYNC_TYPE_SYNC IsoSyncType = C.LIBUSB_ISO_SYNC_TYPE_SYNC
|
||||
)
|
||||
|
||||
var isoSyncTypeDescription = map[IsoSyncType]string{
|
||||
ISO_SYNC_TYPE_NONE: "no synchronization",
|
||||
ISO_SYNC_TYPE_ASYNC: "asynchronous",
|
||||
ISO_SYNC_TYPE_ADAPTIVE: "adaptive",
|
||||
ISO_SYNC_TYPE_SYNC: "synchronous",
|
||||
}
|
||||
|
||||
func (ist IsoSyncType) String() string {
|
||||
return isoSyncTypeDescription[ist]
|
||||
}
|
||||
|
||||
type IsoUsageType int
|
||||
|
||||
const (
|
||||
ISO_USAGE_TYPE_DATA IsoUsageType = C.LIBUSB_ISO_USAGE_TYPE_DATA
|
||||
ISO_USAGE_TYPE_FEEDBACK IsoUsageType = C.LIBUSB_ISO_USAGE_TYPE_FEEDBACK
|
||||
ISO_USAGE_TYPE_IMPLICIT IsoUsageType = C.LIBUSB_ISO_USAGE_TYPE_IMPLICIT
|
||||
)
|
||||
|
||||
var isoUsageTypeDescription = map[IsoUsageType]string{
|
||||
ISO_USAGE_TYPE_DATA: "data",
|
||||
ISO_USAGE_TYPE_FEEDBACK: "feedback",
|
||||
ISO_USAGE_TYPE_IMPLICIT: "implicit data",
|
||||
}
|
||||
|
||||
func (iut IsoUsageType) String() string {
|
||||
return isoUsageTypeDescription[iut]
|
||||
}
|
90
usb/device.go
Normal file
90
usb/device.go
Normal file
@@ -0,0 +1,90 @@
|
||||
package usb
|
||||
|
||||
// #cgo LDFLAGS: -lusb-1.0
|
||||
// #include <libusb-1.0/libusb.h>
|
||||
import "C"
|
||||
|
||||
import (
|
||||
"log"
|
||||
"runtime"
|
||||
)
|
||||
|
||||
type DeviceInfo struct {
|
||||
dev *C.libusb_device
|
||||
}
|
||||
|
||||
func newDeviceInfo(dev *C.libusb_device) *DeviceInfo {
|
||||
d := &DeviceInfo{
|
||||
dev: dev,
|
||||
}
|
||||
|
||||
// Reference this device
|
||||
C.libusb_ref_device(dev)
|
||||
|
||||
// I still can't get this to be called
|
||||
runtime.SetFinalizer(d, (*DeviceInfo).Close)
|
||||
|
||||
log.Printf("deviceInfo %p initialized", d.dev)
|
||||
return d
|
||||
}
|
||||
|
||||
// Get the assigned Bus Number for this USB device.
|
||||
func (d *DeviceInfo) BusNumber() byte {
|
||||
return byte(C.libusb_get_bus_number(d.dev))
|
||||
}
|
||||
|
||||
// Get the assigned Bus Address for this USB device.
|
||||
func (d *DeviceInfo) Address() byte {
|
||||
return byte(C.libusb_get_device_address(d.dev))
|
||||
}
|
||||
|
||||
// Get a descriptor for the device.
|
||||
func (d *DeviceInfo) Descriptor() (*Descriptor, error) {
|
||||
return newDescriptor(d.dev)
|
||||
}
|
||||
|
||||
// Open the given device for I/O.
|
||||
func (d *DeviceInfo) Open() (*Device, error) {
|
||||
var handle *C.libusb_device_handle
|
||||
if errno := C.libusb_open(d.dev, &handle); errno != 0 {
|
||||
return nil, usbError(errno)
|
||||
}
|
||||
|
||||
return newDevice(handle), nil
|
||||
}
|
||||
|
||||
// Close decrements the reference count for the device in the libusb driver
|
||||
// code. It should be called exactly once!
|
||||
func (d *DeviceInfo) Close() error {
|
||||
if d.dev != nil {
|
||||
log.Printf("deviceInfo %p closed", d.dev)
|
||||
C.libusb_unref_device(d.dev)
|
||||
}
|
||||
d.dev = nil
|
||||
return nil
|
||||
}
|
||||
|
||||
type Device struct {
|
||||
handle *C.libusb_device_handle
|
||||
}
|
||||
|
||||
func newDevice(handle *C.libusb_device_handle) *Device {
|
||||
d := &Device{
|
||||
handle: handle,
|
||||
}
|
||||
|
||||
// :(
|
||||
runtime.SetFinalizer(d, (*Device).Close)
|
||||
|
||||
log.Printf("device %p initialized", d.handle)
|
||||
return d
|
||||
}
|
||||
|
||||
func (d *Device) Close() error {
|
||||
if d.handle != nil {
|
||||
log.Printf("device %p closed", d.handle)
|
||||
C.libusb_unref_device(d.handle)
|
||||
}
|
||||
d.handle = nil
|
||||
return nil
|
||||
}
|
45
usb/error.go
Normal file
45
usb/error.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package usb
|
||||
|
||||
// #cgo LDFLAGS: -lusb-1.0
|
||||
// #include <libusb-1.0/libusb.h>
|
||||
import "C"
|
||||
|
||||
type usbError C.int
|
||||
|
||||
func (e usbError) Error() string {
|
||||
return "libusb: " + usbErrorString[e]
|
||||
}
|
||||
|
||||
const (
|
||||
SUCCESS usbError = C.LIBUSB_SUCCESS
|
||||
ERROR_IO usbError = C.LIBUSB_ERROR_IO
|
||||
ERROR_INVALID_PARAM usbError = C.LIBUSB_ERROR_INVALID_PARAM
|
||||
ERROR_ACCESS usbError = C.LIBUSB_ERROR_ACCESS
|
||||
ERROR_NO_DEVICE usbError = C.LIBUSB_ERROR_NO_DEVICE
|
||||
ERROR_NOT_FOUND usbError = C.LIBUSB_ERROR_NOT_FOUND
|
||||
ERROR_BUSY usbError = C.LIBUSB_ERROR_BUSY
|
||||
ERROR_TIMEOUT usbError = C.LIBUSB_ERROR_TIMEOUT
|
||||
ERROR_OVERFLOW usbError = C.LIBUSB_ERROR_OVERFLOW
|
||||
ERROR_PIPE usbError = C.LIBUSB_ERROR_PIPE
|
||||
ERROR_INTERRUPTED usbError = C.LIBUSB_ERROR_INTERRUPTED
|
||||
ERROR_NO_MEM usbError = C.LIBUSB_ERROR_NO_MEM
|
||||
ERROR_NOT_SUPPORTED usbError = C.LIBUSB_ERROR_NOT_SUPPORTED
|
||||
ERROR_OTHER usbError = C.LIBUSB_ERROR_OTHER
|
||||
)
|
||||
|
||||
var usbErrorString = map[usbError]string{
|
||||
C.LIBUSB_SUCCESS: "success",
|
||||
C.LIBUSB_ERROR_IO: "i/o error",
|
||||
C.LIBUSB_ERROR_INVALID_PARAM: "invalid param",
|
||||
C.LIBUSB_ERROR_ACCESS: "bad access",
|
||||
C.LIBUSB_ERROR_NO_DEVICE: "no device",
|
||||
C.LIBUSB_ERROR_NOT_FOUND: "not found",
|
||||
C.LIBUSB_ERROR_BUSY: "device or resource busy",
|
||||
C.LIBUSB_ERROR_TIMEOUT: "timeout",
|
||||
C.LIBUSB_ERROR_OVERFLOW: "overflow",
|
||||
C.LIBUSB_ERROR_PIPE: "pipe error",
|
||||
C.LIBUSB_ERROR_INTERRUPTED: "interrupted",
|
||||
C.LIBUSB_ERROR_NO_MEM: "out of memory",
|
||||
C.LIBUSB_ERROR_NOT_SUPPORTED: "not supported",
|
||||
C.LIBUSB_ERROR_OTHER: "unknown error",
|
||||
}
|
33
usb/misc.go
Normal file
33
usb/misc.go
Normal file
@@ -0,0 +1,33 @@
|
||||
package usb
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
)
|
||||
|
||||
type BCD uint16
|
||||
|
||||
const (
|
||||
USB_2_0 BCD = 0x0200
|
||||
USB_1_1 BCD = 0x0110
|
||||
USB_1_0 BCD = 0x0100
|
||||
)
|
||||
|
||||
func (d BCD) Int() (i int) {
|
||||
ten := 1
|
||||
for o := uint(0); o < 4; o++ {
|
||||
n := ((0xF << (o*4)) & d) >> (o*4)
|
||||
i += int(n) * ten
|
||||
ten *= 10
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (d BCD) String() string {
|
||||
return fmt.Sprintf("%02x.%02x", int(d >> 8), int(d & 0xFF))
|
||||
}
|
||||
|
||||
type ID uint16
|
||||
|
||||
func (id ID) String() string {
|
||||
return fmt.Sprintf("%04x", int(id))
|
||||
}
|
24
usb/misc_test.go
Normal file
24
usb/misc_test.go
Normal file
@@ -0,0 +1,24 @@
|
||||
package usb
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestBCD(t *testing.T) {
|
||||
tests := []struct{
|
||||
BCD
|
||||
Int int
|
||||
Str string
|
||||
}{
|
||||
{0x1234, 1234, "12.34"},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
if got, want := test.BCD.Int(), test.Int; got != want {
|
||||
t.Errorf("Int(%x) = %d, want %d", test.BCD, got, want)
|
||||
}
|
||||
if got, want := test.BCD.String(), test.Str; got != want {
|
||||
t.Errorf("String(%x) = %q, want %q", test.BCD, got, want)
|
||||
}
|
||||
}
|
||||
}
|
76
usb/usb.go
Normal file
76
usb/usb.go
Normal file
@@ -0,0 +1,76 @@
|
||||
package usb
|
||||
|
||||
// #cgo LDFLAGS: -lusb-1.0
|
||||
// #include <libusb-1.0/libusb.h>
|
||||
import "C"
|
||||
|
||||
import (
|
||||
"log"
|
||||
"reflect"
|
||||
"runtime"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
type Context struct {
|
||||
ctx *C.libusb_context
|
||||
}
|
||||
|
||||
func (c *Context) Debug(level int) {
|
||||
C.libusb_set_debug(c.ctx, C.int(level))
|
||||
}
|
||||
|
||||
func NewContext() *Context {
|
||||
c := new(Context)
|
||||
|
||||
log.Printf("gousb initialized")
|
||||
if errno := C.libusb_init(&c.ctx); errno != 0 {
|
||||
panic(usbError(errno))
|
||||
}
|
||||
|
||||
// This doesn't seem to actually get called. Sigh.
|
||||
runtime.SetFinalizer(c, (*Context).Close)
|
||||
|
||||
return c
|
||||
}
|
||||
|
||||
// ListDevices calls each with each enumerated device. If the function returns
|
||||
// true, the device is added to the list of *DeviceInfo to return. All of
|
||||
// these must be Closed, even if an error is also returned.
|
||||
func (c *Context) ListDevices(each func(bus, addr int, desc *Descriptor) bool) ([]*DeviceInfo, error) {
|
||||
var list **C.libusb_device
|
||||
cnt := C.libusb_get_device_list(c.ctx, &list)
|
||||
if cnt < 0 {
|
||||
return nil, usbError(cnt)
|
||||
}
|
||||
defer C.libusb_free_device_list(list, 1)
|
||||
|
||||
var slice []*C.libusb_device
|
||||
*(*reflect.SliceHeader)(unsafe.Pointer(&slice)) = reflect.SliceHeader{
|
||||
Data: uintptr(unsafe.Pointer(list)),
|
||||
Len: int(cnt),
|
||||
Cap: int(cnt),
|
||||
}
|
||||
|
||||
ret := []*DeviceInfo{}
|
||||
for _, dev := range slice {
|
||||
bus := int(C.libusb_get_bus_number(dev))
|
||||
addr := int(C.libusb_get_device_address(dev))
|
||||
desc, err := newDescriptor(dev)
|
||||
if err != nil {
|
||||
return ret, err
|
||||
}
|
||||
if each(bus, addr, desc) {
|
||||
ret = append(ret, newDeviceInfo(dev))
|
||||
}
|
||||
}
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
func (c *Context) Close() error {
|
||||
if c.ctx != nil {
|
||||
C.libusb_exit(c.ctx)
|
||||
log.Printf("gousb finished")
|
||||
}
|
||||
c.ctx = nil
|
||||
return nil
|
||||
}
|
40
usb/usb_test.go
Normal file
40
usb/usb_test.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package usb_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
. "github.com/kylelemons/gousb/usb"
|
||||
"github.com/kylelemons/gousb/usbid"
|
||||
)
|
||||
|
||||
func TestNoop(t *testing.T) {
|
||||
c := NewContext()
|
||||
defer c.Close()
|
||||
c.Debug(3)
|
||||
}
|
||||
|
||||
func TestEnum(t *testing.T) {
|
||||
c := NewContext()
|
||||
defer c.Close()
|
||||
c.Debug(3)
|
||||
|
||||
devs, err := c.ListDevices(func(bus, addr int, desc *Descriptor) bool {
|
||||
t.Logf("%03d:%03d %+v", bus, addr, desc)
|
||||
if v, ok := usbid.Vendors[desc.Vendor]; ok {
|
||||
if p, ok := v.Devices[desc.Product]; ok {
|
||||
t.Logf(" - %s (%s) %s (%s)", v, desc.Vendor, p, desc.Product)
|
||||
} else {
|
||||
t.Logf(" - %s (%s) Unknown", v, desc.Vendor)
|
||||
}
|
||||
}
|
||||
return false
|
||||
})
|
||||
defer func() {
|
||||
for _, d := range devs {
|
||||
d.Close()
|
||||
}
|
||||
}()
|
||||
if err != nil {
|
||||
t.Fatalf("list: %s", err)
|
||||
}
|
||||
}
|
41
usbid/load.go
Normal file
41
usbid/load.go
Normal file
@@ -0,0 +1,41 @@
|
||||
package usbid
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strings"
|
||||
"log"
|
||||
|
||||
"github.com/kylelemons/gousb/usb"
|
||||
)
|
||||
|
||||
const (
|
||||
LinuxUsbDotOrg = "http://www.linux-usb.org/usb.ids"
|
||||
)
|
||||
|
||||
var Vendors map[usb.ID]*Vendor
|
||||
|
||||
func LoadFromURL(url string) error {
|
||||
resp, err := http.Get(url)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
ids, err := ParseIDs(resp.Body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
Vendors = ids
|
||||
return nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
ids, err := ParseIDs(strings.NewReader(usbIdListData))
|
||||
if err != nil {
|
||||
log.Printf("usbid: failed to parse: %s", err)
|
||||
return
|
||||
}
|
||||
|
||||
Vendors = ids
|
||||
}
|
16998
usbid/load_data.go
Normal file
16998
usbid/load_data.go
Normal file
File diff suppressed because it is too large
Load Diff
162
usbid/parse.go
Normal file
162
usbid/parse.go
Normal file
@@ -0,0 +1,162 @@
|
||||
package usbid
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"io"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/kylelemons/gousb/usb"
|
||||
)
|
||||
|
||||
type Vendor struct {
|
||||
Name string
|
||||
Devices map[usb.ID]*Device
|
||||
}
|
||||
|
||||
func (v Vendor) String() string {
|
||||
return v.Name
|
||||
}
|
||||
|
||||
// TODO(kevlar) s/device/
|
||||
type Device struct {
|
||||
Name string
|
||||
Interfaces map[usb.ID]string
|
||||
}
|
||||
|
||||
func (d Device) String() string {
|
||||
return d.Name
|
||||
}
|
||||
|
||||
/*
|
||||
type Class struct {
|
||||
Name string
|
||||
SubClass map[usb.ID]*SubClass
|
||||
}
|
||||
|
||||
type SubClass struct {
|
||||
Name string
|
||||
Protocol map[usb.ID]string
|
||||
}
|
||||
*/
|
||||
|
||||
func ParseIDs(r io.Reader) (map[usb.ID]*Vendor, error) {
|
||||
vendors := make(map[usb.ID]*Vendor, 2800)
|
||||
|
||||
split := func(s string) (kind string, level int, id usb.ID, name string, err error) {
|
||||
pieces := strings.SplitN(s, " ", 2)
|
||||
if len(pieces) != 2 {
|
||||
err = fmt.Errorf("malformatted line %q", s)
|
||||
return
|
||||
}
|
||||
|
||||
// Save the name
|
||||
name = pieces[1]
|
||||
|
||||
// Parse out the level
|
||||
for len(pieces[0]) > 0 && pieces[0][0] == '\t' {
|
||||
level, pieces[0] = level+1, pieces[0][1:]
|
||||
}
|
||||
|
||||
// Parse the first piece to see if it has a kind
|
||||
first := strings.SplitN(pieces[0], " ", 2)
|
||||
if len(first) == 2 {
|
||||
kind, pieces[0] = first[0], first[1]
|
||||
}
|
||||
|
||||
// Parse the ID
|
||||
i, err := strconv.ParseUint(pieces[0], 16, 16)
|
||||
if err != nil {
|
||||
err = fmt.Errorf("malformatted id %q: %s", pieces[0], err)
|
||||
return
|
||||
}
|
||||
id = usb.ID(i)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// Hold the interim values
|
||||
var vendor *Vendor
|
||||
var device *Device
|
||||
|
||||
parseVendor := func(level int, id usb.ID, name string) error {
|
||||
switch level {
|
||||
case 0:
|
||||
vendor = &Vendor{
|
||||
Name: name,
|
||||
}
|
||||
vendors[id] = vendor
|
||||
|
||||
case 1:
|
||||
if vendor == nil {
|
||||
return fmt.Errorf("product line without vendor line")
|
||||
}
|
||||
|
||||
device = &Device{
|
||||
Name: name,
|
||||
}
|
||||
if vendor.Devices == nil {
|
||||
vendor.Devices = make(map[usb.ID]*Device)
|
||||
}
|
||||
vendor.Devices[id] = device
|
||||
|
||||
case 2:
|
||||
if device == nil {
|
||||
return fmt.Errorf("interface line without device line")
|
||||
}
|
||||
|
||||
if device.Interfaces == nil {
|
||||
device.Interfaces = make(map[usb.ID]string)
|
||||
}
|
||||
device.Interfaces[id] = name
|
||||
|
||||
default:
|
||||
return fmt.Errorf("too many levels of nesting for vendor block")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// TODO(kevlar): Parse class information, etc
|
||||
//var class *Class
|
||||
//var subclass *SubClass
|
||||
|
||||
var kind string
|
||||
|
||||
lines := bufio.NewReaderSize(r, 512)
|
||||
parseLines:
|
||||
for lineno := 0; ; lineno++ {
|
||||
b, isPrefix, err := lines.ReadLine()
|
||||
switch {
|
||||
case err == io.EOF:
|
||||
break parseLines
|
||||
case err != nil:
|
||||
return nil, err
|
||||
case isPrefix:
|
||||
return nil, fmt.Errorf("line %d: line too long", lineno)
|
||||
}
|
||||
line := string(b)
|
||||
|
||||
if len(line) == 0 || line[0] == '#' {
|
||||
continue
|
||||
}
|
||||
|
||||
k, level, id, name, err := split(line)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("line %d: %s", lineno, err)
|
||||
}
|
||||
if k != "" {
|
||||
kind = k
|
||||
}
|
||||
|
||||
switch kind {
|
||||
case "": err = parseVendor(level, id, name)
|
||||
}
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("line %d: %s", lineno, err)
|
||||
}
|
||||
}
|
||||
|
||||
return vendors, nil
|
||||
}
|
88
usbid/parse_test.go
Normal file
88
usbid/parse_test.go
Normal file
@@ -0,0 +1,88 @@
|
||||
package usbid
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/kylelemons/gousb/usb"
|
||||
)
|
||||
|
||||
func TestParse(t *testing.T) {
|
||||
tests := []struct {
|
||||
Input string
|
||||
Output map[usb.ID]*Vendor
|
||||
}{{
|
||||
Input: `
|
||||
# Skip comment
|
||||
abcd Vendor One
|
||||
0123 Product One
|
||||
0124 Product Two
|
||||
efef Vendor Two
|
||||
0aba Product
|
||||
12 Interface One
|
||||
24 Interface Two
|
||||
0abb Product
|
||||
12 Interface
|
||||
|
||||
C 00 (Defined at Interface level)
|
||||
C 01 Audio
|
||||
01 Control Device
|
||||
02 Streaming
|
||||
03 MIDI Streaming
|
||||
C 02 Communications
|
||||
01 Direct Line
|
||||
02 Abstract (modem)
|
||||
00 None
|
||||
01 AT-commands (v.25ter)
|
||||
02 AT-commands (PCCA101)
|
||||
03 AT-commands (PCCA101 + wakeup)
|
||||
04 AT-commands (GSM)
|
||||
05 AT-commands (3G)
|
||||
06 AT-commands (CDMA)
|
||||
fe Defined by command set descriptor
|
||||
ff Vendor Specific (MSFT RNDIS?)
|
||||
`,
|
||||
Output: map[usb.ID]*Vendor{
|
||||
0xabcd: {
|
||||
Name: "Vendor One",
|
||||
Devices: map[usb.ID]*Device{
|
||||
0x0123: {Name: "Product One"},
|
||||
0x0124: {Name: "Product Two"},
|
||||
},
|
||||
},
|
||||
0xefef: {
|
||||
Name: "Vendor Two",
|
||||
Devices: map[usb.ID]*Device{
|
||||
0x0aba: {
|
||||
Name: "Product",
|
||||
Interfaces: map[usb.ID]string{
|
||||
0x12: "Interface One",
|
||||
0x24: "Interface Two",
|
||||
},
|
||||
},
|
||||
0x0abb: {
|
||||
Name: "Product",
|
||||
Interfaces: map[usb.ID]string{
|
||||
0x12: "Interface",
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
}}
|
||||
|
||||
for idx, test := range tests {
|
||||
vendors, err := ParseIDs(strings.NewReader(test.Input))
|
||||
if err != nil {
|
||||
t.Errorf("%d. ParseIDs: %s", idx, err)
|
||||
continue
|
||||
}
|
||||
if got, want := vendors, test.Output; !reflect.DeepEqual(got, want) {
|
||||
t.Errorf("%d. Parse mismatch", idx)
|
||||
t.Errorf(" - got: %v", idx, got)
|
||||
t.Errorf(" - want: %v", idx, want)
|
||||
continue
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user