Make the device management service send a proper user agent.

chrome/browser/policy/device_management_service.cc

 // Copyright (c) 2010 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/policy/device_management_service.h"
 
 #include "chrome/browser/browser_thread.h"
 #include "net/base/cookie_monster.h"
 #include "net/base/host_resolver.h"
 #include "net/base/load_flags.h"
 #include "net/base/ssl_config_service_defaults.h"
 #include "net/http/http_auth_handler_factory.h"
 #include "net/http/http_network_layer.h"
 #include "net/proxy/proxy_service.h"
 #include "net/url_request/url_request_context.h"
 #include "net/url_request/url_request_status.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/io_thread.h"
 #include "chrome/browser/net/chrome_net_log.h"
 #include "chrome/browser/policy/device_management_backend_impl.h"
 #include "chrome/common/net/url_request_context_getter.h"
+#include "webkit/glue/webkit_glue.h"
 
 namespace policy {
 
 namespace {
 
 // Custom request context implementation that allows to override the user agent,
 // amongst others. Wraps a baseline request context from which we reuse the
 // networking components.
 class DeviceManagementRequestContext : public URLRequestContext {
  public:
   explicit DeviceManagementRequestContext(URLRequestContext* base_context);
   virtual ~DeviceManagementRequestContext();
+
+ private:
+  // Overriden from URLRequestContext.
+  virtual const std::string& GetUserAgent(const GURL& url) const;
 };
 
 DeviceManagementRequestContext::DeviceManagementRequestContext(
     URLRequestContext* base_context) {
   // Share resolver, proxy service and ssl bits with the baseline context. This
   // is important so we don't make redundant requests (e.g. when resolving proxy
   // auto configuration).
   net_log_ = base_context->net_log();
   host_resolver_ = base_context->host_resolver();
   proxy_service_ = base_context->proxy_service();
   ssl_config_service_ = base_context->ssl_config_service();
 
   // Share the http session.
   http_transaction_factory_ = net::HttpNetworkLayer::CreateFactory(
       base_context->http_transaction_factory()->GetSession());
 
   // No cookies, please.
   cookie_store_ = new net::CookieMonster(NULL, NULL);
 
   // Initialize these to sane values for our purposes.
   accept_language_ = "*";
   accept_charset_ = "*";
 }
 
 DeviceManagementRequestContext::~DeviceManagementRequestContext() {
   delete http_transaction_factory_;
   delete http_auth_handler_factory_;
 }
 
+const std::string& DeviceManagementRequestContext::GetUserAgent(
+    const GURL& url) const {
+  return webkit_glue::GetUserAgent(url);
+}
+
 // Request context holder.
 class DeviceManagementRequestContextGetter : public URLRequestContextGetter {
  public:
   DeviceManagementRequestContextGetter(
       URLRequestContextGetter* base_context_getter)
       : base_context_getter_(base_context_getter) {}
 
   // URLRequestContextGetter overrides.
   virtual URLRequestContext* GetURLRequestContext();
   virtual scoped_refptr<base::MessageLoopProxy> GetIOMessageLoopProxy() const;
@@ skipping 101 matching lines @@
     DeviceManagementJob* job = entry->second;
     job->HandleResponse(status, response_code, cookies, data);
     pending_jobs_.erase(entry);
   } else {
     NOTREACHED() << "Callback from foreign URL fetcher";
   }
   delete source;
 }
 
 }  // namespace policy