Use interface associations to check function permissions.

device/usb/usb_device_handle_unittest.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <stddef.h>
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
@@ skipping 114 matching lines @@
 
   TestOpenCallback open_device;
   gadget->GetDevice()->Open(open_device.callback());
   scoped_refptr<UsbDeviceHandle> handle = open_device.WaitForResult();
   ASSERT_TRUE(handle.get());
 
   TestResultCallback claim_interface;
   handle->ClaimInterface(0, claim_interface.callback());
   ASSERT_TRUE(claim_interface.WaitForResult());
 
+  const UsbInterfaceDescriptor* interface =
+      handle->FindInterfaceByEndpoint(0x81);
+  EXPECT_TRUE(interface);
+  EXPECT_EQ(0, interface->interface_number);
+  interface = handle->FindInterfaceByEndpoint(0x01);
+  EXPECT_TRUE(interface);
+  EXPECT_EQ(0, interface->interface_number);
+  EXPECT_FALSE(handle->FindInterfaceByEndpoint(0x82));
+  EXPECT_FALSE(handle->FindInterfaceByEndpoint(0x02));
+
   scoped_refptr<net::IOBufferWithSize> in_buffer(new net::IOBufferWithSize(64));
   TestCompletionCallback in_completion;
   handle->GenericTransfer(USB_DIRECTION_INBOUND, 0x81, in_buffer.get(),
                           in_buffer->size(),
                           5000,  // 5 second timeout
                           in_completion.callback());
 
   scoped_refptr<net::IOBufferWithSize> out_buffer(
       new net::IOBufferWithSize(in_buffer->size()));
   TestCompletionCallback out_completion;
@@ skipping 38 matching lines @@
 
   TestOpenCallback open_device;
   gadget->GetDevice()->Open(open_device.callback());
   scoped_refptr<UsbDeviceHandle> handle = open_device.WaitForResult();
   ASSERT_TRUE(handle.get());
 
   TestResultCallback claim_interface;
   handle->ClaimInterface(1, claim_interface.callback());
   ASSERT_TRUE(claim_interface.WaitForResult());
 
+  EXPECT_FALSE(handle->FindInterfaceByEndpoint(0x81));
+  EXPECT_FALSE(handle->FindInterfaceByEndpoint(0x01));
+  const UsbInterfaceDescriptor* interface =
+      handle->FindInterfaceByEndpoint(0x82);
+  EXPECT_TRUE(interface);
+  EXPECT_EQ(1, interface->interface_number);
+  interface = handle->FindInterfaceByEndpoint(0x02);
+  EXPECT_TRUE(interface);
+  EXPECT_EQ(1, interface->interface_number);
+
   scoped_refptr<net::IOBufferWithSize> in_buffer(
       new net::IOBufferWithSize(512));
   TestCompletionCallback in_completion;
   handle->GenericTransfer(USB_DIRECTION_INBOUND, 0x82, in_buffer.get(),
                           in_buffer->size(),
                           5000,  // 5 second timeout
                           in_completion.callback());
 
   scoped_refptr<net::IOBufferWithSize> out_buffer(
       new net::IOBufferWithSize(in_buffer->size()));
@@ skipping 53 matching lines @@
   TestResultCallback release_interface;
   handle->ReleaseInterface(2, release_interface.callback());
   ASSERT_TRUE(release_interface.WaitForResult());
 
   handle->Close();
 }
 
 }  // namespace
 
 }  // namespace device