Have Permission{Manager,Service} use Origin.

content/browser/permissions/permission_service_impl.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 "content/browser/permissions/permission_service_impl.h"
 
 #include <stddef.h>
 #include <utility>
 
 #include "base/bind.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/permission_manager.h"
 #include "content/public/browser/permission_type.h"
+#include "url/origin.h"
 
 namespace content {
 
 namespace {
 
 PermissionType PermissionNameToPermissionType(mojom::PermissionName name) {
   switch(name) {
     case mojom::PermissionName::GEOLOCATION:
       return PermissionType::GEOLOCATION;
     case mojom::PermissionName::NOTIFICATIONS:
@@ skipping 42 matching lines @@
 
   mojo::Array<mojom::PermissionStatus> result =
       mojo::Array<mojom::PermissionStatus>::New(request_count);
   for (int i = 0; i < request_count; ++i)
     result[i] = mojom::PermissionStatus::DENIED;
   callback.Run(std::move(result));
 }
 
 PermissionServiceImpl::PendingSubscription::PendingSubscription(
     PermissionType permission,
-    const GURL& origin,
+    const url::Origin& origin,
     const PermissionStatusCallback& callback)
-    : id(-1),
-      permission(permission),
-      origin(origin),
-      callback(callback) {
-}
+    : id(-1), permission(permission), origin(origin), callback(callback) {}
 
 PermissionServiceImpl::PendingSubscription::~PendingSubscription() {
   if (!callback.is_null())
     callback.Run(mojom::PermissionStatus::ASK);
 }
 
 PermissionServiceImpl::PermissionServiceImpl(
     PermissionServiceContext* context,
     mojo::InterfaceRequest<mojom::PermissionService> request)
     : context_(context),
       binding_(this, std::move(request)),
       weak_factory_(this) {
   binding_.set_connection_error_handler(
       base::Bind(&PermissionServiceImpl::OnConnectionError,
                  base::Unretained(this)));
 }
 
 PermissionServiceImpl::~PermissionServiceImpl() {
   DCHECK(pending_requests_.IsEmpty());
 }
 
 void PermissionServiceImpl::OnConnectionError() {
   context_->ServiceHadConnectionError(this);
   // After that call, |this| will be deleted.
 }
 
 void PermissionServiceImpl::RequestPermission(
     mojom::PermissionName permission,
-    const mojo::String& origin,
+    const mojo::String& origin_string,
     const PermissionStatusCallback& callback) {
   // This condition is valid if the call is coming from a ChildThread instead of
   // a RenderFrame. Some consumers of the service run in Workers and some in
   // Frames. In the context of a Worker, it is not possible to show a
   // permission prompt because there is no tab. In the context of a Frame, we
   // can. Even if the call comes from a context where it is not possible to show
   // any UI, we want to still return something relevant so the current
   // permission status is returned.
   BrowserContext* browser_context = context_->GetBrowserContext();
   DCHECK(browser_context);
+  url::Origin origin(origin_string.get());
   if (!context_->render_frame_host() ||
       !browser_context->GetPermissionManager()) {
-    callback.Run(GetPermissionStatusFromName(permission, GURL(origin.get())));
+    callback.Run(GetPermissionStatusFromName(permission, origin));
     return;
   }
 
   int pending_request_id = pending_requests_.Add(new PendingRequest(
       base::Bind(&PermissionRequestResponseCallbackWrapper, callback), 1));
   int id = browser_context->GetPermissionManager()->RequestPermission(
-      PermissionNameToPermissionType(permission),
-      context_->render_frame_host(),
-      GURL(origin.get()),
-      base::Bind(&PermissionServiceImpl::OnRequestPermissionResponse,
-                 weak_factory_.GetWeakPtr(),
-                 pending_request_id));
+      PermissionNameToPermissionType(permission), context_->render_frame_host(),
+      origin, base::Bind(&PermissionServiceImpl::OnRequestPermissionResponse,
+                         weak_factory_.GetWeakPtr(), pending_request_id));
 
   // Check if the request still exists. It might have been removed by the
   // callback if it was run synchronously.
   PendingRequest* pending_request = pending_requests_.Lookup(
       pending_request_id);
   if (!pending_request)
       return;
   pending_request->id = id;
 }
 
 void PermissionServiceImpl::OnRequestPermissionResponse(
     int pending_request_id,
     mojom::PermissionStatus status) {
   OnRequestPermissionsResponse(pending_request_id,
                                std::vector<mojom::PermissionStatus>(1, status));
 }
 
 void PermissionServiceImpl::RequestPermissions(
     mojo::Array<mojom::PermissionName> permissions,
-    const mojo::String& origin,
+    const mojo::String& origin_string,
     const PermissionsStatusCallback& callback) {
   if (permissions.is_null()) {
     callback.Run(mojo::Array<mojom::PermissionStatus>());
     return;
   }
 
   // This condition is valid if the call is coming from a ChildThread instead of
   // a RenderFrame. Some consumers of the service run in Workers and some in
   // Frames. In the context of a Worker, it is not possible to show a
   // permission prompt because there is no tab. In the context of a Frame, we
   // can. Even if the call comes from a context where it is not possible to show
   // any UI, we want to still return something relevant so the current
   // permission status is returned for each permission.
   BrowserContext* browser_context = context_->GetBrowserContext();
   DCHECK(browser_context);
+  const url::Origin origin(origin_string.get());
   if (!context_->render_frame_host() ||
       !browser_context->GetPermissionManager()) {
     mojo::Array<mojom::PermissionStatus> result(permissions.size());
     for (size_t i = 0; i < permissions.size(); ++i) {
-      result[i] =
-          GetPermissionStatusFromName(permissions[i], GURL(origin.get()));
+      result[i] = GetPermissionStatusFromName(permissions[i], origin);
     }
     callback.Run(std::move(result));
     return;
   }
 
   std::vector<PermissionType> types(permissions.size());
   for (size_t i = 0; i < types.size(); ++i)
     types[i] = PermissionNameToPermissionType(permissions[i]);
 
   int pending_request_id = pending_requests_.Add(
       new PendingRequest(callback, permissions.size()));
   int id = browser_context->GetPermissionManager()->RequestPermissions(
-      types,
-      context_->render_frame_host(),
-      GURL(origin.get()),
+      types, context_->render_frame_host(), origin,
       base::Bind(&PermissionServiceImpl::OnRequestPermissionsResponse,
-                 weak_factory_.GetWeakPtr(),
-                 pending_request_id));
+                 weak_factory_.GetWeakPtr(), pending_request_id));
 
   // Check if the request still exists. It may have been removed by the
   // the response callback.
   PendingRequest* pending_request = pending_requests_.Lookup(
       pending_request_id);
   if (!pending_request)
       return;
   pending_request->id = id;
 }
 
@@ skipping 30 matching lines @@
         it.GetCurrentValue()->permission, it.GetCurrentValue()->origin));
     it.GetCurrentValue()->callback.reset();
     permission_manager->UnsubscribePermissionStatusChange(
         it.GetCurrentValue()->id);
   }
   pending_subscriptions_.Clear();
 }
 
 void PermissionServiceImpl::HasPermission(
     mojom::PermissionName permission,
-    const mojo::String& origin,
+    const mojo::String& origin_string,
     const PermissionStatusCallback& callback) {
-  callback.Run(GetPermissionStatusFromName(permission, GURL(origin.get())));
+  callback.Run(GetPermissionStatusFromName(permission,
+                                           url::Origin(origin_string.get())));
 }
 
 void PermissionServiceImpl::RevokePermission(
     mojom::PermissionName permission,
-    const mojo::String& origin,
+    const mojo::String& origin_string,
     const PermissionStatusCallback& callback) {
-  GURL origin_url(origin.get());
+  const url::Origin origin(origin_string.get());
   PermissionType permission_type = PermissionNameToPermissionType(permission);
   mojom::PermissionStatus status =
-      GetPermissionStatusFromType(permission_type, origin_url);
+      GetPermissionStatusFromType(permission_type, origin);
 
   // Resetting the permission should only be possible if the permission is
   // already granted.
   if (status != mojom::PermissionStatus::GRANTED) {
     callback.Run(status);
     return;
   }
 
-  ResetPermissionStatus(permission_type, origin_url);
+  ResetPermissionStatus(permission_type, origin);
 
-  callback.Run(GetPermissionStatusFromType(permission_type, origin_url));
+  callback.Run(GetPermissionStatusFromType(permission_type, origin));
 }
 
 void PermissionServiceImpl::GetNextPermissionChange(
     mojom::PermissionName permission,
-    const mojo::String& mojo_origin,
+    const mojo::String& origin_string,
     mojom::PermissionStatus last_known_status,
     const PermissionStatusCallback& callback) {
-  GURL origin(mojo_origin.get());
+  const url::Origin origin(origin_string.get());
   mojom::PermissionStatus current_status =
       GetPermissionStatusFromName(permission, origin);
   if (current_status != last_known_status) {
     callback.Run(current_status);
     return;
   }
 
   BrowserContext* browser_context = context_->GetBrowserContext();
   DCHECK(browser_context);
   if (!browser_context->GetPermissionManager()) {
     callback.Run(current_status);
     return;
   }
 
   PermissionType permission_type = PermissionNameToPermissionType(permission);
 
   // We need to pass the id of PendingSubscription in pending_subscriptions_
   // to the callback but SubscribePermissionStatusChange() will also return an
   // id which is different.
   PendingSubscription* subscription =
       new PendingSubscription(permission_type, origin, callback);
   int pending_subscription_id = pending_subscriptions_.Add(subscription);
 
-  GURL embedding_origin = context_->GetEmbeddingOrigin();
+  const url::Origin embedding_origin = context_->GetEmbeddingOrigin();
   subscription->id =
       browser_context->GetPermissionManager()->SubscribePermissionStatusChange(
-          permission_type,
-          origin,
-          // If the embedding_origin is empty, we,ll use the |origin| instead.
-          embedding_origin.is_empty() ? origin : embedding_origin,
+          permission_type, origin,
+          embedding_origin.empty() ? origin : embedding_origin,
           base::Bind(&PermissionServiceImpl::OnPermissionStatusChanged,
-                     weak_factory_.GetWeakPtr(),
-                     pending_subscription_id));
+                     weak_factory_.GetWeakPtr(), pending_subscription_id));
 }
 
 mojom::PermissionStatus PermissionServiceImpl::GetPermissionStatusFromName(
     mojom::PermissionName permission,
-    const GURL& origin) {
+    const url::Origin& origin) {
   return GetPermissionStatusFromType(PermissionNameToPermissionType(permission),
                                      origin);
 }
 
 mojom::PermissionStatus PermissionServiceImpl::GetPermissionStatusFromType(
     PermissionType type,
-    const GURL& origin) {
+    const url::Origin& origin) {
   BrowserContext* browser_context = context_->GetBrowserContext();
   DCHECK(browser_context);
   if (!browser_context->GetPermissionManager())
     return mojom::PermissionStatus::DENIED;
 
-  // If the embedding_origin is empty we'll use |origin| instead.
-  GURL embedding_origin = context_->GetEmbeddingOrigin();
+  const url::Origin embedding_origin = context_->GetEmbeddingOrigin();
   return browser_context->GetPermissionManager()->GetPermissionStatus(
-      type, origin, embedding_origin.is_empty() ? origin : embedding_origin);
+      type, origin, embedding_origin.empty() ? origin : embedding_origin);
 }
 
 void PermissionServiceImpl::ResetPermissionStatus(PermissionType type,
-                                                  const GURL& origin) {
+                                                  const url::Origin& origin) {
   BrowserContext* browser_context = context_->GetBrowserContext();
   DCHECK(browser_context);
   if (!browser_context->GetPermissionManager())
     return;
 
-  // If the embedding_origin is empty we'll use |origin| instead.
-  GURL embedding_origin = context_->GetEmbeddingOrigin();
+  const url::Origin embedding_origin = context_->GetEmbeddingOrigin();
   browser_context->GetPermissionManager()->ResetPermission(
-      type, origin, embedding_origin.is_empty() ? origin : embedding_origin);
+      type, origin, embedding_origin.empty() ? origin : embedding_origin);
 }
 
 void PermissionServiceImpl::OnPermissionStatusChanged(
     int pending_subscription_id,
     mojom::PermissionStatus status) {
   PendingSubscription* subscription =
       pending_subscriptions_.Lookup(pending_subscription_id);
 
   BrowserContext* browser_context = context_->GetBrowserContext();
   DCHECK(browser_context);
   if (browser_context->GetPermissionManager()) {
     browser_context->GetPermissionManager()->UnsubscribePermissionStatusChange(
         subscription->id);
   }
 
   PermissionStatusCallback callback = subscription->callback;
 
   subscription->callback.reset();
   pending_subscriptions_.Remove(pending_subscription_id);
 
   callback.Run(status);
 }
 
 }  // namespace content