| // 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 "device/usb/usb_device_filter.h" |
| |
| #include <memory> |
| #include <utility> |
| |
| #include "base/values.h" |
| #include "device/usb/usb_descriptors.h" |
| #include "device/usb/usb_device.h" |
| |
| namespace device { |
| |
| namespace { |
| |
| const char kProductIdKey[] = "productId"; |
| const char kVendorIdKey[] = "vendorId"; |
| const char kInterfaceClassKey[] = "interfaceClass"; |
| const char kInterfaceSubclassKey[] = "interfaceSubclass"; |
| const char kInterfaceProtocolKey[] = "interfaceProtocol"; |
| |
| } // namespace |
| |
| UsbDeviceFilter::UsbDeviceFilter() |
| : vendor_id_set_(false), |
| product_id_set_(false), |
| interface_class_set_(false), |
| interface_subclass_set_(false), |
| interface_protocol_set_(false) { |
| } |
| |
| UsbDeviceFilter::UsbDeviceFilter(const UsbDeviceFilter& other) = default; |
| |
| UsbDeviceFilter::~UsbDeviceFilter() { |
| } |
| |
| void UsbDeviceFilter::SetVendorId(uint16_t vendor_id) { |
| vendor_id_set_ = true; |
| vendor_id_ = vendor_id; |
| } |
| |
| void UsbDeviceFilter::SetProductId(uint16_t product_id) { |
| product_id_set_ = true; |
| product_id_ = product_id; |
| } |
| |
| void UsbDeviceFilter::SetInterfaceClass(uint8_t interface_class) { |
| interface_class_set_ = true; |
| interface_class_ = interface_class; |
| } |
| |
| void UsbDeviceFilter::SetInterfaceSubclass(uint8_t interface_subclass) { |
| interface_subclass_set_ = true; |
| interface_subclass_ = interface_subclass; |
| } |
| |
| void UsbDeviceFilter::SetInterfaceProtocol(uint8_t interface_protocol) { |
| interface_protocol_set_ = true; |
| interface_protocol_ = interface_protocol; |
| } |
| |
| bool UsbDeviceFilter::Matches(scoped_refptr<UsbDevice> device) const { |
| if (vendor_id_set_) { |
| if (device->vendor_id() != vendor_id_) { |
| return false; |
| } |
| |
| if (product_id_set_ && device->product_id() != product_id_) { |
| return false; |
| } |
| } |
| |
| if (interface_class_set_) { |
| for (const UsbConfigDescriptor& config : device->configurations()) { |
| for (const UsbInterfaceDescriptor& iface : config.interfaces) { |
| if (iface.interface_class == interface_class_ && |
| (!interface_subclass_set_ || |
| (iface.interface_subclass == interface_subclass_ && |
| (!interface_protocol_set_ || |
| iface.interface_protocol == interface_protocol_)))) { |
| return true; |
| } |
| } |
| } |
| |
| return false; |
| } |
| |
| return true; |
| } |
| |
| std::unique_ptr<base::Value> UsbDeviceFilter::ToValue() const { |
| std::unique_ptr<base::DictionaryValue> obj(new base::DictionaryValue()); |
| |
| if (vendor_id_set_) { |
| obj->SetInteger(kVendorIdKey, vendor_id_); |
| if (product_id_set_) { |
| obj->SetInteger(kProductIdKey, product_id_); |
| } |
| } |
| |
| if (interface_class_set_) { |
| obj->SetInteger(kInterfaceClassKey, interface_class_); |
| if (interface_subclass_set_) { |
| obj->SetInteger(kInterfaceSubclassKey, interface_subclass_); |
| if (interface_protocol_set_) { |
| obj->SetInteger(kInterfaceProtocolKey, interface_protocol_); |
| } |
| } |
| } |
| |
| return std::move(obj); |
| } |
| |
| // static |
| bool UsbDeviceFilter::MatchesAny(scoped_refptr<UsbDevice> device, |
| const std::vector<UsbDeviceFilter>& filters) { |
| if (filters.empty()) |
| return true; |
| |
| for (std::vector<UsbDeviceFilter>::const_iterator i = filters.begin(); |
| i != filters.end(); |
| ++i) { |
| if (i->Matches(device)) { |
| return true; |
| } |
| } |
| return false; |
| } |
| |
| } // namespace device |