From b6099569e31705174920f44083f7f1bc22c445c7 Mon Sep 17 00:00:00 2001
From: Thomas Tsou <ttsou@vt.edu>
Date: Fri, 27 Aug 2010 13:27:32 -0700
Subject: usrp1: Cleanup libusb device handling

This patch limits all libusb device enumeration operations to FSF
(Vendor ID = 0xfffe) devices, which removes a lot of unncessary
libusb output when debug mode is enabled. The reference counts
held by the libusb device list are also reduced, which prevents
holding references to unused devices.
---
 host/lib/transport/libusb1_base.cpp | 50 ++++++++++++++++++++++++++++++-------
 1 file changed, 41 insertions(+), 9 deletions(-)

(limited to 'host/lib/transport/libusb1_base.cpp')

diff --git a/host/lib/transport/libusb1_base.cpp b/host/lib/transport/libusb1_base.cpp
index 6965de214..493d4eff3 100644
--- a/host/lib/transport/libusb1_base.cpp
+++ b/host/lib/transport/libusb1_base.cpp
@@ -16,10 +16,22 @@
 //
 
 #include "libusb1_base.hpp"
+#include <uhd/utils/assert.hpp>
 #include <iostream>
 
 using namespace uhd::transport;
 
+bool check_fsf_device(libusb_device *dev)
+{
+    libusb_device_descriptor desc;
+
+    if (libusb_get_device_descriptor(dev, &desc) < 0) {
+        UHD_ASSERT_THROW("USB: failed to get device descriptor");
+    }
+
+    return desc.idVendor == 0xfffe;
+}
+
 void libusb::init(libusb_context **ctx, int debug_level)
 {
     if (libusb_init(ctx) < 0)
@@ -28,27 +40,47 @@ void libusb::init(libusb_context **ctx, int debug_level)
     libusb_set_debug(*ctx, debug_level);
 }
 
+std::vector<libusb_device *> libusb::get_fsf_device_list(libusb_context *ctx)
+{
+    libusb_device **libusb_dev_list;
+    std::vector<libusb_device *> fsf_dev_list;
+
+    ssize_t dev_cnt = libusb_get_device_list(ctx, &libusb_dev_list);
+
+    //find the FSF devices 
+    for (ssize_t i = 0; i < dev_cnt; i++) {
+        libusb_device *dev = libusb_dev_list[i];
+
+        if (check_fsf_device(dev))
+            fsf_dev_list.push_back(dev);
+        else
+            libusb_unref_device(dev);
+    }
+
+    libusb_free_device_list(libusb_dev_list, 0);
+
+    return fsf_dev_list;
+}
 
 libusb_device_handle *libusb::open_device(libusb_context *ctx,
                                           usb_device_handle::sptr handle)
 {
-    libusb_device **dev_list;
-    libusb_device_handle *dev_handle;
+    libusb_device_handle *dev_handle = NULL;
+    std::vector<libusb_device *> fsf_dev_list = get_fsf_device_list(ctx);
 
-    ssize_t dev_cnt = libusb_get_device_list(ctx, &dev_list);
-
-    //find and open the receive device
-    for (ssize_t i = 0; i < dev_cnt; i++) {
-        libusb_device *dev = dev_list[i];
+    //find and open the USB device 
+    for (size_t i = 0; i < fsf_dev_list.size(); i++) {
+        libusb_device *dev = fsf_dev_list[i];
 
         if (compare_device(dev, handle)) {
             libusb_open(dev, &dev_handle);
+            libusb_unref_device(dev);
             break;
         }
+            
+        libusb_unref_device(dev);
     }
 
-    libusb_free_device_list(dev_list, 0);
-
     return dev_handle;
 }
 
-- 
cgit v1.2.3