DevTools: Add geolocation override in browser

content/browser/devtools/protocol/page_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 "content/browser/devtools/protocol/page_handler.h"
 
 #include "base/base64.h"
+#include "content/browser/geolocation/geolocation_dispatcher_host.h"
 #include "content/browser/renderer_host/render_view_host_impl.h"
+#include "content/browser/web_contents/web_contents_impl.h"
 
 namespace content {
 namespace devtools {
 namespace page {
 
 typedef DevToolsProtocolClient::Response Response;
 
 PageHandler::PageHandler()
     : weak_factory_(this) {
 }
@@ skipping 31 matching lines @@
 Response PageHandler::GetNavigationHistory(
     int* current_index,
     std::vector<NavigationEntry>* entries) {
   return Response::FallThrough();
 }
 
 Response PageHandler::NavigateToHistoryEntry(int entry_id) {
   return Response::FallThrough();
 }
 
+Response PageHandler::SetGeolocationOverride(double* latitude,
+                                             double* longitude,
+                                             double* accuracy) {
+  WebContentsImpl* web_contents = static_cast<WebContentsImpl*>(
+      WebContents::FromRenderViewHost(host_));
+  if (!web_contents)
+    return Response::InternalError("No WebContents to override");
+  GeolocationDispatcherHost* geolocation_host =
+      web_contents->geolocation_dispatcher_host();
+  scoped_ptr<Geoposition> geoposition(new Geoposition());
+  if (latitude && longitude && accuracy) {
+    geoposition->latitude = *latitude;
+    geoposition->longitude = *longitude;
+    geoposition->accuracy = *accuracy;
+    geoposition->timestamp = base::Time::Now();
+  } else {
+    geoposition->error_code = Geoposition::ERROR_CODE_POSITION_UNAVAILABLE;
+  }
+  geolocation_host->SetOverride(geoposition.Pass());
+  return Response::OK();
+}
+
+Response PageHandler::ClearGeolocationOverride() {
+  WebContentsImpl* web_contents = static_cast<WebContentsImpl*>(
+      WebContents::FromRenderViewHost(host_));
+  if (!web_contents)
+    return Response::InternalError("No WebContents to override");
+  GeolocationDispatcherHost* geolocation_host =
+      web_contents->geolocation_dispatcher_host();
+  geolocation_host->ClearOverride();
+  return Response::OK();
+}
+
+
 Response PageHandler::SetTouchEmulationEnabled(bool enabled) {
   return Response::FallThrough();
 }
 
 scoped_refptr<DevToolsProtocol::Response> PageHandler::CaptureScreenshot(
     scoped_refptr<DevToolsProtocol::Command> command) {
   if (!host_ || !host_->GetView())
     return command->InternalErrorResponse("Could not connect to view");
 
   host_->GetSnapshotFromBrowser(
@@ skipping 52 matching lines @@
       &base_64_data);
 
   CaptureScreenshotResponse response;
   response.set_data(base_64_data);
   client_->SendCaptureScreenshotResponse(command, response);
 }
 
 }  // namespace page
 }  // namespace devtools
 }  // namespace content