WebKit merge 41660:41709 (WebKit side)....

third_party/WebKit/WebCore/page/Geolocation.cpp

 /*
  * Copyright (C) 2008, 2009 Apple Inc. All Rights Reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
@@ skipping 39 matching lines @@
     m_timer.stop();
 
     RefPtr<PositionError> error = PositionError::create(PositionError::TIMEOUT, "Timed out");
     m_errorCallback->handleEvent(error.get());
 }
 
 Geolocation::Geolocation(Frame* frame)
     : m_frame(frame)
     , m_service(GeolocationService::create(this))
     , m_allowGeolocation(Unknown)
+    , m_shouldClearCache(false)
 {
+    if (!m_frame)
+        return;
     ASSERT(m_frame->document());
     m_frame->document()->setUsingGeolocation(true);
 }
 
 void Geolocation::disconnectFrame()
 {
     m_service->stopUpdating();
     m_frame = 0;
 }
 
 void Geolocation::getCurrentPosition(PassRefPtr<PositionCallback> successCallback, PassRefPtr<PositionErrorCallback> errorCallback, PositionOptions* options)
 {
     RefPtr<GeoNotifier> notifier = GeoNotifier::create(successCallback, errorCallback, options);
 
-    if (!shouldAllowGeolocation()) {
-        if (notifier->m_errorCallback) {
-            RefPtr<PositionError> error = WebCore::PositionError::create(PositionError::PERMISSION_DENIED, "Disallowed Geolocation");
-            notifier->m_errorCallback->handleEvent(error.get());
-        }
-        return;
-    }
-
     if (!m_service->startUpdating(options)) {
         if (notifier->m_errorCallback) {
             RefPtr<PositionError> error = PositionError::create(PositionError::PERMISSION_DENIED, "Unable to Start");
             notifier->m_errorCallback->handleEvent(error.get());
         }
         return;
     }
 
     m_oneShots.add(notifier);
 }
 
 int Geolocation::watchPosition(PassRefPtr<PositionCallback> successCallback, PassRefPtr<PositionErrorCallback> errorCallback, PositionOptions* options)
 {
     RefPtr<GeoNotifier> notifier = GeoNotifier::create(successCallback, errorCallback, options);
 
-    if (!shouldAllowGeolocation()) {
-        if (notifier->m_errorCallback) {
-            RefPtr<PositionError> error = WebCore::PositionError::create(PositionError::PERMISSION_DENIED, "Disallowed Geolocation");
-            notifier->m_errorCallback->handleEvent(error.get());
-        }
-        return 0;
-    }
-
     if (!m_service->startUpdating(options)) {
         if (notifier->m_errorCallback) {
             RefPtr<PositionError> error = PositionError::create(PositionError::PERMISSION_DENIED, "Unable to Start");
             notifier->m_errorCallback->handleEvent(error.get());
         }
         return 0;
     }
     
     static int sIdentifier = 0;
     
@@ skipping 15 matching lines @@
     if (hasListeners())
         m_service->suspend();
 }
 
 void Geolocation::resume()
 {
     if (hasListeners())
         m_service->resume();
 }
 
+void Geolocation::setIsAllowed(bool allowed)
+{
+    m_allowGeolocation = allowed ? Yes : No;
+    
+    if (isAllowed())
+        geolocationServicePositionChanged(m_service.get());
+    else {
+        WTF::RefPtr<WebCore::PositionError> error = WebCore::PositionError::create(PositionError::PERMISSION_DENIED, "User disallowed GeoLocation");
+        handleError(error.get());
+    }
+}
+
 void Geolocation::sendErrorToOneShots(PositionError* error)
 {
     Vector<RefPtr<GeoNotifier> > copy;
     copyToVector(m_oneShots, copy);
 
     Vector<RefPtr<GeoNotifier> >::const_iterator end = copy.end();
     for (Vector<RefPtr<GeoNotifier> >::const_iterator it = copy.begin(); it != end; ++it) {
         RefPtr<GeoNotifier> notifier = *it;
         
         if (notifier->m_errorCallback)
@@ skipping 58 matching lines @@
 void Geolocation::handleError(PositionError* error)
 {
     ASSERT(error);
     
     sendErrorToOneShots(error);    
     sendErrorToWatchers(error);
 
     m_oneShots.clear();
 }
 
+void Geolocation::requestPermission()
+{
+    if (m_allowGeolocation > Unknown)
+        return;
+
+    if (!m_frame)
+        return;
+    
+    Page* page = m_frame->page();
+    if (!page)
+        return;
+    
+    // Ask the chrome: it maintains the geolocation challenge policy itself.
+    page->chrome()->requestGeolocationPermissionForFrame(m_frame, this);
+    
+    m_allowGeolocation = InProgress;
+}
+
 void Geolocation::geolocationServicePositionChanged(GeolocationService* service)
 {
     ASSERT(service->lastPosition());
     
+    requestPermission();
+    if (!isAllowed())
+        return;
+    
     sendPositionToOneShots(service->lastPosition());
     sendPositionToWatchers(service->lastPosition());
         
     m_oneShots.clear();
 
     if (!hasListeners())
         m_service->stopUpdating();
 }
 
 void Geolocation::geolocationServiceErrorOccurred(GeolocationService* service)
 {
     ASSERT(service->lastError());
     
     handleError(service->lastError());
 }
 
-bool Geolocation::shouldAllowGeolocation()
-{
-    if (!m_frame)
-        return false;
-
-    Page* page = m_frame->page();
-    if (!page)
-        return false;
-
-    if (m_allowGeolocation == Unknown)
-        m_allowGeolocation = page->chrome()->shouldAllowGeolocationForFrame(m_frame) ? Yes : No;
-    return m_allowGeolocation == Yes;
-}
-
 } // namespace WebCore