Make USB permissions work in the new permission message system

extensions/common/permissions/usb_device_permission.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 "extensions/common/permissions/usb_device_permission.h"
 
 #include <set>
 #include <string>
 
 #include "base/logging.h"
 #include "base/strings/string16.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "device/usb/usb_ids.h"
 #include "extensions/common/permissions/permissions_info.h"
 #include "grit/extensions_strings.h"
 #include "ui/base/l10n/l10n_util.h"
 
 namespace extensions {
 
-namespace {
+UsbDevicePermission::UsbDevicePermission(const APIPermissionInfo* info)
+    : SetDisjunctionPermission<UsbDevicePermissionData, UsbDevicePermission>(
+          info) {}
 
-// Adds the permissions from the |data_set| to |ids|.
-void AddPermissionsToLists(const std::set<UsbDevicePermissionData>& data_set,
-                           PermissionIDSet* ids) {
-  // TODO(sashab): Once GetMessages() is deprecated, move this logic back into
-  // GetPermissions().
-  // TODO(sashab, reillyg): Once GetMessages() is deprecated, rework the
-  // permission message logic for USB devices to generate more meaningful
-  // messages and better fit the current rules system.
-  if (data_set.size() == 1) {
-    const UsbDevicePermissionData& data = *data_set.begin();
+UsbDevicePermission::~UsbDevicePermission() {}
 
+PermissionIDSet UsbDevicePermission::GetPermissions() const {
+  PermissionIDSet ids;
+
+  std::set<uint16> unknown_product_vendors;
+  bool found_unknown_vendor = false;
+
+  for (const UsbDevicePermissionData& data : data_set_) {
     const char* vendor = device::UsbIds::GetVendorName(data.vendor_id());
     if (vendor) {
       const char* product =
           device::UsbIds::GetProductName(data.vendor_id(), data.product_id());
       if (product) {
         base::string16 product_name_and_vendor = l10n_util::GetStringFUTF16(
             IDS_EXTENSION_PROMPT_WARNING_USB_DEVICE_PRODUCT_NAME_AND_VENDOR,
             base::UTF8ToUTF16(product), base::UTF8ToUTF16(vendor));
-
-        ids->insert(APIPermission::kUsbDevice, product_name_and_vendor);
+        ids.insert(APIPermission::kUsbDevice, product_name_and_vendor);
       } else {
-        ids->insert(APIPermission::kUsbDeviceUnknownProduct,
-                    base::UTF8ToUTF16(vendor));
+        unknown_product_vendors.insert(data.vendor_id());
       }
     } else {
-      ids->insert(APIPermission::kUsbDeviceUnknownVendor);
+      found_unknown_vendor = true;
     }
-  } else if (data_set.size() > 1) {
-    std::vector<base::string16> details;
-    std::set<uint16> unknown_product_vendors;
-    bool found_unknown_vendor = false;
+  }
 
-    for (const UsbDevicePermissionData& data : data_set) {
-      const char* vendor = device::UsbIds::GetVendorName(data.vendor_id());
-      if (vendor) {
-        const char* product =
-            device::UsbIds::GetProductName(data.vendor_id(), data.product_id());
-        if (product) {
-          base::string16 product_name_and_vendor = l10n_util::GetStringFUTF16(
-              IDS_EXTENSION_PROMPT_WARNING_USB_DEVICE_PRODUCT_NAME_AND_VENDOR,
-              base::UTF8ToUTF16(product), base::UTF8ToUTF16(vendor));
-          details.push_back(l10n_util::GetStringFUTF16(
-              IDS_EXTENSION_PROMPT_WARNING_USB_DEVICE_LIST_ITEM,
-              product_name_and_vendor));
-        } else {
-          unknown_product_vendors.insert(data.vendor_id());
-        }
-      } else {
-        found_unknown_vendor = true;
-      }
-    }
+  for (const uint16& vendor_id : unknown_product_vendors) {

[Devlin, 2015/08/20 16:33:07]

hmm... a const& to a uint16 almost sounds _more_ expensive than a copy. ;)

[Marc Treib, 2015/08/21 08:00:48]

Indeed. That's a leftover from a previous version where I was storing strings...
Done.

+    const char* vendor = device::UsbIds::GetVendorName(vendor_id);

[Devlin, 2015/08/20 16:33:07]

Actually, looks like all of this could just go in the else above, and get rid of
unknown_product_vendors, right?

[Marc Treib, 2015/08/21 08:00:49]

No. The intermediate set is for deduping, since we could have multiple unknown
devices with the same vendor, and we only want it to show up once.

+    DCHECK(vendor);
+    ids.insert(APIPermission::kUsbDeviceUnknownProduct,
+               base::UTF8ToUTF16(vendor));
+  }
 
-    // List generic "devices from this vendor" entries after specific devices.
-    for (const uint16& vendor_id : unknown_product_vendors) {
-      const char* vendor = device::UsbIds::GetVendorName(vendor_id);
-      DCHECK(vendor);
-      details.push_back(l10n_util::GetStringFUTF16(
-          IDS_EXTENSION_PROMPT_WARNING_USB_DEVICE_LIST_ITEM_UNKNOWN_PRODUCT,
-          base::UTF8ToUTF16(vendor)));
-    }
+  if (found_unknown_vendor)
+    ids.insert(APIPermission::kUsbDeviceUnknownVendor);
 
-    // Display the catch all "device from an unknown vendor" last.
-    if (found_unknown_vendor) {
-      details.push_back(l10n_util::GetStringUTF16(
-          IDS_EXTENSION_PROMPT_WARNING_USB_DEVICE_LIST_ITEM_UNKNOWN_VENDOR));
-    }
-
-    for (const auto& detail : details)
-      ids->insert(APIPermission::kUsbDeviceList, detail);
-  }
-}
-
-}  // namespace
-
-UsbDevicePermission::UsbDevicePermission(const APIPermissionInfo* info)
-    : SetDisjunctionPermission<UsbDevicePermissionData, UsbDevicePermission>(
-          info) {
-}
-
-UsbDevicePermission::~UsbDevicePermission() {
-}
-
-PermissionIDSet UsbDevicePermission::GetPermissions() const {
-  PermissionIDSet ids;
-  AddPermissionsToLists(data_set_, &ids);
   return ids;
 }
 
 }  // namespace extensions