Cancel the permission request

android_webview/native/permission/permission_request_handler.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 "android_webview/native/permission/permission_request_handler.h"
 
 #include "android_webview/native/permission/aw_permission_request.h"
 #include "android_webview/native/permission/aw_permission_request_delegate.h"
 #include "android_webview/native/permission/permission_request_handler_client.h"
 #include "base/android/scoped_java_ref.h"
 #include "base/bind.h"
+#include "content/public/browser/navigation_details.h"
+#include "content/public/browser/navigation_entry.h"
+#include "content/public/browser/web_contents.h"
 
 using base::android::ScopedJavaLocalRef;
 
 namespace android_webview {
 
+namespace {
+
+int GetActiveEntryID(content::WebContents* web_contents) {
+  if (!web_contents) return 0;
+
+  content::NavigationEntry* active_entry =
+      web_contents->GetController().GetActiveEntry();
+  return active_entry ? active_entry->GetUniqueID() : 0;
+}
+
+}  // namespace
+
 PermissionRequestHandler::PermissionRequestHandler(
-    PermissionRequestHandlerClient* client)
-    : client_(client) {
+    PermissionRequestHandlerClient* client, content::WebContents* web_contents)
+    : content::WebContentsObserver(web_contents),
+      client_(client),
+      contents_unique_id_(GetActiveEntryID(web_contents)) {
 }
 
 PermissionRequestHandler::~PermissionRequestHandler() {
-  for (RequestIterator i = requests_.begin(); i != requests_.end(); ++i)
-    CancelRequest(i);
+  CancelAllRequests();
 }
 
 void PermissionRequestHandler::SendRequest(
     scoped_ptr<AwPermissionRequestDelegate> request) {
   if (Preauthorized(request->GetOrigin(), request->GetResources())) {
     request->NotifyRequestResult(true);
     return;
   }
 
   AwPermissionRequest* aw_request = new AwPermissionRequest(request.Pass());
@@ skipping 22 matching lines @@
 
   std::string key = origin.GetOrigin().spec();
   if (key.empty()) {
     LOG(ERROR) << "The origin of preauthorization is empty, ignore it.";
     return;
   }
 
   preauthorized_permission_[key] |= resources;
 }
 
+void PermissionRequestHandler::NavigationEntryCommitted(
+    const content::LoadCommittedDetails& details) {
+  const content::PageTransition transition = details.entry->GetTransitionType();
+  if (details.is_navigation_to_different_page() ||
+      content::PageTransitionStripQualifier(transition) ==
+      content::PAGE_TRANSITION_RELOAD ||
+      contents_unique_id_ != details.entry->GetUniqueID()) {
+    CancelAllRequests();
+    contents_unique_id_ = details.entry->GetUniqueID();
+  }
+}
+
 PermissionRequestHandler::RequestIterator
 PermissionRequestHandler::FindRequest(const GURL& origin,
                                       int64 resources) {
   RequestIterator i;
   for (i = requests_.begin(); i != requests_.end(); ++i) {
     if (i->get() && i->get()->GetOrigin() == origin &&
         i->get()->GetResources() == resources) {
       break;
     }
   }
   return i;
 }
 
 void PermissionRequestHandler::CancelRequest(RequestIterator i) {
   if (i->get())
     client_->OnPermissionRequestCanceled(i->get());
   // The request's grant()/deny() could be called upon
   // OnPermissionRequestCanceled. Delete AwPermissionRequest if it still
   // exists.
   if (i->get())
     delete i->get();
 }
 
+void PermissionRequestHandler::CancelAllRequests() {
+  for (RequestIterator i = requests_.begin(); i != requests_.end(); ++i)
+    CancelRequest(i);
+}
+
 void PermissionRequestHandler::PruneRequests() {
   for (RequestIterator i = requests_.begin(); i != requests_.end();) {
     if (!i->get())
       i = requests_.erase(i);
     else
       ++i;
   }
 }
 
 bool PermissionRequestHandler::Preauthorized(const GURL& origin,
                                               int64 resources) {
   std::map<std::string, int64>::iterator i =
       preauthorized_permission_.find(origin.GetOrigin().spec());
 
   return i != preauthorized_permission_.end() &&
       (resources & i->second) == resources;
 }
 
 }  // namespace android_webivew