chrome/browser/extensions/permission_messages_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/memory/scoped_ptr.h"
   6 #include "base/strings/utf_string_conversions.h"
   7 #include "chrome/browser/extensions/extension_service.h"
   8 #include "chrome/browser/extensions/permissions_updater.h"
   9 #include "chrome/browser/extensions/test_extension_environment.h"
  10 #include "chrome/common/extensions/permissions/chrome_permission_message_provider.h"
  11 #include "chrome/grit/generated_resources.h"
  12 #include "chrome/test/base/testing_profile.h"
  13 #include "components/crx_file/id_util.h"
  14 #include "extensions/browser/extension_prefs.h"
  15 #include "extensions/common/extension.h"
  16 #include "extensions/common/extension_builder.h"
  17 #include "extensions/common/manifest.h"
  18 #include "extensions/common/manifest_handlers/permissions_parser.h"
  19 #include "extensions/common/permissions/permission_set.h"
  20 #include "extensions/common/permissions/permissions_data.h"
  21 #include "extensions/common/permissions/permissions_info.h"
  22 #include "extensions/common/permissions/usb_device_permission.h"
  23 #include "extensions/common/permissions/usb_device_permission_data.h"
  24 #include "extensions/common/test_util.h"
  25 #include "extensions/common/value_builder.h"
  26 #include "testing/gtest/include/gtest/gtest.h"
  27 #include "ui/base/l10n/l10n_util.h"
  28
  29 namespace extensions {
  30
  31 // Tests that ChromePermissionMessageProvider provides not only correct, but
  32 // meaningful permission messages that coalesce correctly where appropriate.
  33 // There are 3 types of permission messages that need to be tested:
  34 //  1. The combined list of active permissions, displayed at install time (or
  35 //     when the app has been disabled automatically and needs to be re-enabled)
  36 //  2. The split list of active permissions, displayed in the App Info dialog,
  37 //     where the optional permissions are individually revokable
  38 //  3. The list of requested optional permissions, displayed in a prompt to the
  39 //     user when the app requests these during runtime
  40 // Some of these tests are prefixed AntiTest_, since they demonstrate existing
  41 // problematic functionality. These tests are prefixed with AntiTest_ and will
  42 // be changed as the correct behaviour is implemented. TODOs in the test explain
  43 // the currently problematic behaviour.
  44 class PermissionMessagesUnittest : public testing::Test {
  45  public:
  46   PermissionMessagesUnittest()
  47       : message_provider_(new ChromePermissionMessageProvider()) {}
  48   ~PermissionMessagesUnittest() override {}
  49
  50  protected:
  51   void CreateAndInstallAppWithPermissions(ListBuilder& required_permissions,
  52                                           ListBuilder& optional_permissions) {
  53     app_ = test_util::BuildApp(ExtensionBuilder().Pass())
  54                .MergeManifest(
  55                     DictionaryBuilder()
  56                         .Set("permissions", required_permissions)
  57                         .Set("optional_permissions", optional_permissions))
  58                .SetID(crx_file::id_util::GenerateId("app"))
  59                .SetLocation(Manifest::INTERNAL)
  60                .Build();
  61     env_.GetExtensionService()->AddExtension(app_.get());
  62   }
  63
  64   void CreateAndInstallExtensionWithPermissions(
  65       ListBuilder& required_permissions,
  66       ListBuilder& optional_permissions) {
  67     app_ = test_util::BuildExtension(ExtensionBuilder().Pass())
  68                .MergeManifest(
  69                     DictionaryBuilder()
  70                         .Set("permissions", required_permissions)
  71                         .Set("optional_permissions", optional_permissions))
  72                .SetID(crx_file::id_util::GenerateId("extension"))
  73                .SetLocation(Manifest::INTERNAL)
  74                .Build();
  75     env_.GetExtensionService()->AddExtension(app_.get());
  76   }
  77
  78   // Returns the permission messages that would display in the prompt that
  79   // requests all the optional permissions for the current |app_|.
  80   std::vector<base::string16> GetOptionalPermissionMessages() {
  81     scoped_refptr<const PermissionSet> granted_permissions =
  82         env_.GetExtensionPrefs()->GetGrantedPermissions(app_->id());
  83     scoped_refptr<const PermissionSet> optional_permissions =
  84         PermissionsParser::GetOptionalPermissions(app_.get());
  85     scoped_refptr<const PermissionSet> requested_permissions =
  86         PermissionSet::CreateDifference(optional_permissions.get(),
  87                                         granted_permissions.get());
  88     return GetMessages(requested_permissions);
  89   }
  90
  91   void GrantOptionalPermissions() {
  92     PermissionsUpdater perms_updater(env_.profile());
  93     perms_updater.AddPermissions(
  94         app_.get(),
  95         PermissionsParser::GetOptionalPermissions(app_.get()).get());
  96   }
  97
  98   std::vector<base::string16> active_permissions() {
  99     return GetMessages(app_->permissions_data()->active_permissions());
 100   }
 101
 102   std::vector<base::string16> required_permissions() {
 103     return GetMessages(PermissionsParser::GetRequiredPermissions(app_.get()));
 104   }
 105
 106   std::vector<base::string16> optional_permissions() {
 107     return GetMessages(PermissionsParser::GetOptionalPermissions(app_.get()));
 108   }
 109
 110  private:
 111   std::vector<base::string16> GetMessages(
 112       scoped_refptr<const PermissionSet> permissions) {
 113     std::vector<base::string16> messages;
 114     for (const PermissionMessage& msg :
 115          message_provider_->GetPermissionMessages(
 116              message_provider_->GetAllPermissionIDs(permissions.get(),
 117                                                     app_->GetType()))) {
 118       messages.push_back(msg.message());
 119     }
 120     return messages;
 121   }
 122
 123   extensions::TestExtensionEnvironment env_;
 124   scoped_ptr<ChromePermissionMessageProvider> message_provider_;
 125   scoped_refptr<const Extension> app_;
 126
 127   DISALLOW_COPY_AND_ASSIGN(PermissionMessagesUnittest);
 128 };
 129
 130 // If an app has both the 'history' and 'tabs' permission, one should hide the
 131 // other (the 'history' permission has superset permissions).
 132 TEST_F(PermissionMessagesUnittest, HistoryHidesTabsMessage) {
 133   CreateAndInstallExtensionWithPermissions(
 134       ListBuilder().Append("tabs").Append("history").Pass(),
 135       ListBuilder().Pass());
 136
 137   ASSERT_EQ(1U, required_permissions().size());
 138   EXPECT_EQ(
 139       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_WRITE),
 140       required_permissions()[0]);
 141
 142   ASSERT_EQ(0U, optional_permissions().size());
 143 }
 144
 145 // If an app requests the 'history' permission, but already has the 'tabs'
 146 // permission, only the new coalesced message is displayed.
 147 TEST_F(PermissionMessagesUnittest, MixedPermissionMessagesCoalesceOnceGranted) {
 148   CreateAndInstallExtensionWithPermissions(
 149       ListBuilder().Append("tabs").Pass(),
 150       ListBuilder().Append("history").Pass());
 151
 152   ASSERT_EQ(1U, required_permissions().size());
 153   EXPECT_EQ(
 154       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_READ),
 155       required_permissions()[0]);
 156
 157   ASSERT_EQ(1U, optional_permissions().size());
 158   EXPECT_EQ(
 159       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_WRITE),
 160       optional_permissions()[0]);
 161
 162   ASSERT_EQ(1U, active_permissions().size());
 163   EXPECT_EQ(
 164       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_READ),
 165       active_permissions()[0]);
 166
 167   ASSERT_EQ(1U, GetOptionalPermissionMessages().size());
 168   EXPECT_EQ(
 169       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_WRITE),
 170       GetOptionalPermissionMessages()[0]);
 171
 172   GrantOptionalPermissions();
 173
 174   ASSERT_EQ(1U, active_permissions().size());
 175   EXPECT_EQ(
 176       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_WRITE),
 177       active_permissions()[0]);
 178 }
 179
 180 // AntiTest: This behavior should be changed and improved.
 181 // If an app requests the 'tabs' permission but already has the 'history'
 182 // permission, a prompt is displayed. However, no prompt should appear at all,
 183 // since 'tabs' is a subset of 'history' and the final list of permissions are
 184 // not affected by this grant.
 185 TEST_F(PermissionMessagesUnittest,
 186        AntiTest_PromptCanRequestSubsetOfAlreadyGrantedPermissions) {
 187   CreateAndInstallExtensionWithPermissions(
 188       ListBuilder().Append("history").Pass(),
 189       ListBuilder().Append("tabs").Pass());
 190
 191   ASSERT_EQ(1U, required_permissions().size());
 192   EXPECT_EQ(
 193       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_WRITE),
 194       required_permissions()[0]);
 195
 196   ASSERT_EQ(1U, optional_permissions().size());
 197   EXPECT_EQ(
 198       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_READ),
 199       optional_permissions()[0]);
 200
 201   ASSERT_EQ(1U, active_permissions().size());
 202   EXPECT_EQ(
 203       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_WRITE),
 204       active_permissions()[0]);
 205
 206   // TODO(sashab): This prompt should display no permissions, since READ is a
 207   // subset permission of WRITE.
 208   ASSERT_EQ(1U, GetOptionalPermissionMessages().size());
 209   EXPECT_EQ(
 210       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_READ),
 211       GetOptionalPermissionMessages()[0]);
 212
 213   GrantOptionalPermissions();
 214
 215   ASSERT_EQ(1U, active_permissions().size());
 216   EXPECT_EQ(
 217       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_WRITE),
 218       active_permissions()[0]);
 219 }
 220
 221 // AntiTest: This behavior should be changed and improved.
 222 // If an app requests the 'sessions' permission, nothing is displayed in the
 223 // permission request prompt. However, the required permissions for the app are
 224 // actually modified, so the prompt *should* display a message to prevent this
 225 // permission from being granted for free.
 226 TEST_F(PermissionMessagesUnittest,
 227        AntiTest_PromptCanBeEmptyButCausesChangeInPermissions) {
 228   CreateAndInstallExtensionWithPermissions(
 229       ListBuilder().Append("tabs").Pass(),
 230       ListBuilder().Append("sessions").Pass());
 231
 232   ASSERT_EQ(1U, required_permissions().size());
 233   EXPECT_EQ(
 234       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_READ),
 235       required_permissions()[0]);
 236
 237   ASSERT_EQ(0U, optional_permissions().size());
 238
 239   ASSERT_EQ(1U, active_permissions().size());
 240   EXPECT_EQ(
 241       l10n_util::GetStringUTF16(IDS_EXTENSION_PROMPT_WARNING_HISTORY_READ),
 242       active_permissions()[0]);
 243
 244   // TODO(sashab): This prompt should display the sessions permission message,
 245   // as well as warn the user that it can affect the existing 'tab' permission.
 246   ASSERT_EQ(0U, GetOptionalPermissionMessages().size());
 247
 248   GrantOptionalPermissions();
 249
 250   ASSERT_EQ(1U, active_permissions().size());
 251   EXPECT_EQ(l10n_util::GetStringUTF16(
 252                 IDS_EXTENSION_PROMPT_WARNING_HISTORY_READ_AND_SESSIONS),
 253             active_permissions()[0]);
 254 }
 255
 256 class USBDevicePermissionMessagesTest : public testing::Test {
 257  public:
 258   USBDevicePermissionMessagesTest()
 259       : message_provider_(new ChromePermissionMessageProvider()) {}
 260   ~USBDevicePermissionMessagesTest() override {}
 261
 262   PermissionMessages GetMessages(const PermissionIDSet& permissions) {
 263     return message_provider_->GetPermissionMessages(permissions);
 264   }
 265
 266  private:
 267   scoped_ptr<ChromePermissionMessageProvider> message_provider_;
 268 };
 269
 270 TEST_F(USBDevicePermissionMessagesTest, SingleDevice) {
 271   {
 272     const char kMessage[] =
 273         "Access any PVR Mass Storage from HUMAX Co., Ltd. via USB";
 274
 275     scoped_ptr<base::ListValue> permission_list(new base::ListValue());
 276     permission_list->Append(
 277         UsbDevicePermissionData(0x02ad, 0x138c, -1).ToValue().release());
 278
 279     UsbDevicePermission permission(
 280         PermissionsInfo::GetInstance()->GetByID(APIPermission::kUsbDevice));
 281     ASSERT_TRUE(permission.FromValue(permission_list.get(), NULL, NULL));
 282
 283     PermissionMessages messages = GetMessages(permission.GetPermissions());
 284     ASSERT_EQ(1U, messages.size());
 285     EXPECT_EQ(base::ASCIIToUTF16(kMessage), messages.front().message());
 286   }
 287   {
 288     const char kMessage[] = "Access USB devices from HUMAX Co., Ltd.";
 289
 290     scoped_ptr<base::ListValue> permission_list(new base::ListValue());
 291     permission_list->Append(
 292         UsbDevicePermissionData(0x02ad, 0x138d, -1).ToValue().release());
 293
 294     UsbDevicePermission permission(
 295         PermissionsInfo::GetInstance()->GetByID(APIPermission::kUsbDevice));
 296     ASSERT_TRUE(permission.FromValue(permission_list.get(), NULL, NULL));
 297
 298     PermissionMessages messages = GetMessages(permission.GetPermissions());
 299     ASSERT_EQ(1U, messages.size());
 300     EXPECT_EQ(base::ASCIIToUTF16(kMessage), messages.front().message());
 301   }
 302   {
 303     const char kMessage[] = "Access USB devices from an unknown vendor";
 304
 305     scoped_ptr<base::ListValue> permission_list(new base::ListValue());
 306     permission_list->Append(
 307         UsbDevicePermissionData(0x02ae, 0x138d, -1).ToValue().release());
 308
 309     UsbDevicePermission permission(
 310         PermissionsInfo::GetInstance()->GetByID(APIPermission::kUsbDevice));
 311     ASSERT_TRUE(permission.FromValue(permission_list.get(), NULL, NULL));
 312
 313     PermissionMessages messages = GetMessages(permission.GetPermissions());
 314     ASSERT_EQ(1U, messages.size());
 315     EXPECT_EQ(base::ASCIIToUTF16(kMessage), messages.front().message());
 316   }
 317 }
 318
 319 TEST_F(USBDevicePermissionMessagesTest, MultipleDevice) {
 320   const char kMessage[] = "Access any of these USB devices";
 321   const char* kDetails[] = {
 322       "PVR Mass Storage from HUMAX Co., Ltd.",
 323       "unknown devices from HUMAX Co., Ltd.",
 324       "devices from an unknown vendor"
 325   };
 326
 327   // Prepare data set
 328   scoped_ptr<base::ListValue> permission_list(new base::ListValue());
 329   permission_list->Append(
 330       UsbDevicePermissionData(0x02ad, 0x138c, -1).ToValue().release());
 331   // This device's product ID is not in Chrome's database.
 332   permission_list->Append(
 333       UsbDevicePermissionData(0x02ad, 0x138d, -1).ToValue().release());
 334   // This additional unknown product will be collapsed into the entry above.
 335   permission_list->Append(
 336       UsbDevicePermissionData(0x02ad, 0x138e, -1).ToValue().release());
 337   // This device's vendor ID is not in Chrome's database.
 338   permission_list->Append(
 339       UsbDevicePermissionData(0x02ae, 0x138d, -1).ToValue().release());
 340   // This additional unknown vendor will be collapsed into the entry above.
 341   permission_list->Append(
 342       UsbDevicePermissionData(0x02af, 0x138d, -1).ToValue().release());
 343
 344   UsbDevicePermission permission(
 345       PermissionsInfo::GetInstance()->GetByID(APIPermission::kUsbDevice));
 346   ASSERT_TRUE(permission.FromValue(permission_list.get(), NULL, NULL));
 347
 348   PermissionMessages messages = GetMessages(permission.GetPermissions());
 349   ASSERT_EQ(1U, messages.size());
 350   EXPECT_EQ(base::ASCIIToUTF16(kMessage), messages.front().message());
 351   const std::vector<base::string16>& submessages =
 352       messages.front().submessages();
 353   ASSERT_EQ(arraysize(kDetails), submessages.size());
 354   for (size_t i = 0; i < submessages.size(); i++)
 355     EXPECT_EQ(base::ASCIIToUTF16(kDetails[i]), submessages[i]);
 356 }
 357
 358 }  // namespace extensions