Add link URL and success/failure callback parameters to chrome.webstore.install()

chrome/browser/extensions/webstore_inline_installer.cc

 // Copyright (c) 2011 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 "chrome/browser/extensions/webstore_inline_installer.h"
 
 #include <vector>
 
 #include "base/string_util.h"
 #include "base/values.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/crx_installer.h"
 #include "chrome/browser/extensions/extension_install_dialog.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/chrome_utility_messages.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_constants.h"
+#include "chrome/common/extensions/url_pattern.h"
 #include "content/browser/tab_contents/tab_contents.h"
 #include "content/browser/utility_process_host.h"
-#include "googleurl/src/gurl.h"
 #include "net/base/escape.h"
 #include "net/url_request/url_request_status.h"
 
 const char kManifestKey[] = "manifest";
 const char kIconUrlKey[] = "icon_url";
 const char kLocalizedNameKey[] = "localized_name";
 const char kLocalizedDescriptionKey[] = "localized_description";
 const char kUsersKey[] = "users";
 const char kAverageRatingKey[] = "average_rating";
 const char kRatingCountKey[] = "rating_count";
+const char kVerifiedSiteKey[] = "verified_site";
 
-const char kInvalidWebstoreItemId[] = "Invalid webstore item ID";
-const char kWebstoreRequestError[] = "Could not fetch data from webstore";
-const char kInvalidWebstoreResponseError[] = "Invalid webstore reponse";
+const char kInvalidWebstoreItemId[] = "Invalid Chrome Web Store item ID";
+const char kWebstoreRequestError[] =
+    "Could not fetch data from the Chrome Web Store";
+const char kInvalidWebstoreResponseError[] = "Invalid Chrome Web Store reponse";
 const char kInvalidManifestError[] = "Invalid manifest";
 const char kUserCancelledError[] = "User cancelled install";
+const char kNotFromVerifiedSite[] =
+    "Installs can only be initiated by the Chrome Web Store item's verified "
+    "site";
 
 class SafeWebstoreResponseParser : public UtilityProcessHost::Client {
  public:
   SafeWebstoreResponseParser(WebstoreInlineInstaller *client,
                              const std::string& webstore_data)
       : client_(client),
         webstore_data_(webstore_data),
         utility_host_(NULL) {}
 
   void Start() {
@@ skipping 73 matching lines @@
 
   std::string webstore_data_;
 
   UtilityProcessHost* utility_host_;
 
   std::string error_;
   scoped_ptr<DictionaryValue> parsed_webstore_data_;
 };
 
 WebstoreInlineInstaller::WebstoreInlineInstaller(TabContents* tab_contents,
+                                                 int install_id,
                                                  std::string webstore_item_id,
+                                                 GURL requestor_url,
                                                  Delegate* delegate)
     : tab_contents_(tab_contents),
+      install_id_(install_id),
       id_(webstore_item_id),
+      requestor_url_(requestor_url),
       delegate_(delegate) {}
 
 WebstoreInlineInstaller::~WebstoreInlineInstaller() {
 }
 
 void WebstoreInlineInstaller::BeginInstall() {
   AddRef(); // Balanced in CompleteInstall.
 
   if (!Extension::IdIsValid(id_)) {
     CompleteInstall(kInvalidWebstoreItemId);
@@ skipping 74 matching lines @@
       return;
     }
     icon_url = GURL(extension_urls::GetWebstoreLaunchURL()).Resolve(
         icon_url_string);
     if (!icon_url.is_valid()) {
       CompleteInstall(kInvalidWebstoreResponseError);
       return;
     }
   }
 
+  // Verified site is optional
+  if (webstore_data->HasKey(kVerifiedSiteKey)) {
+    std::string verified_site_domain;
+    if (!webstore_data->GetString(kVerifiedSiteKey, &verified_site_domain)) {
+      CompleteInstall(kInvalidWebstoreResponseError);
+      return;
+    }
+
+    URLPattern verified_site_pattern(URLPattern::SCHEME_ALL);
+    verified_site_pattern.SetScheme("*");
+    verified_site_pattern.SetHost(verified_site_domain);
+    verified_site_pattern.SetMatchSubdomains(true);
+    verified_site_pattern.SetPath("/*");
+
+    if (!verified_site_pattern.MatchesURL(requestor_url_)) {
+      CompleteInstall(kNotFromVerifiedSite);
+      return;
+    }
+  }
+
   scoped_refptr<WebstoreInstallHelper> helper = new WebstoreInstallHelper(
       this,
       manifest,
       "", // We don't have any icon data.
       icon_url,
       Profile::FromBrowserContext(tab_contents_->browser_context())->
           GetRequestContext());
   // The helper will call us back via OnWebstoreParseSucces or
   // OnWebstoreParseFailure.
   helper->Start();
@@ skipping 67 matching lines @@
   // or NOTIFICATION_EXTENSION_INSTALL_ERROR fire).
   CompleteInstall("");
 }
 
 void WebstoreInlineInstaller::InstallUIAbort(bool user_initiated) {
   CompleteInstall(kUserCancelledError);
 }
 
 void WebstoreInlineInstaller::CompleteInstall(const std::string& error) {
   if (error.empty()) {
-    delegate_->OnInlineInstallSuccess();
+    delegate_->OnInlineInstallSuccess(install_id_);
   } else {
-    delegate_->OnInlineInstallFailure(error);
+    delegate_->OnInlineInstallFailure(install_id_, error);
   }
   Release(); // Matches the AddRef in BeginInstall.
 }