drive: Set a custom User-Agent header for Drive operations

chrome/browser/google_apis/operations_base.cc

 // Copyright (c) 2012 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/google_apis/operations_base.h"
 
 #include "base/json/json_reader.h"
 #include "base/metrics/histogram.h"
 #include "base/string_number_conversions.h"
 #include "base/stringprintf.h"
 #include "base/values.h"
 #include "chrome/browser/browser_process.h"
+#include "chrome/common/chrome_version_info.h"
 #include "chrome/common/net/url_util.h"
 #include "content/public/browser/browser_thread.h"
 #include "google_apis/gaia/gaia_urls.h"
 #include "google_apis/gaia/google_service_auth_error.h"
 #include "google_apis/gaia/oauth2_access_token_fetcher.h"
 #include "net/base/load_flags.h"
 #include "net/http/http_util.h"
 #include "net/url_request/url_fetcher.h"
 #include "net/url_request/url_request_status.h"
+#include "webkit/user_agent/user_agent_util.h"
 
 using content::BrowserThread;
 using net::URLFetcher;
 
 namespace {
 
 // Used for success ratio histograms. 0 for failure, 1 for success,
 // 2 for no connection (likely offline).
 const int kSuccessRatioHistogramFailure = 0;
 const int kSuccessRatioHistogramSuccess = 1;
@@ skipping 23 matching lines @@
   if (!value->get()) {
     LOG(ERROR) << "Error while parsing entry response: "
                << error_message
                << ", code: "
                << error_code
                << ", data:\n"
                << data;
   }
 }
 
+// Returns a user agent string used for communicating with the Drive backend,
+// both WAPI and Drive API.  The user agent looks like:
+//
+// chromedrive-<VERSION> chrome-cc/none (<OS_CPU_INFO>)
+// chromedrive-24.0.1274.0 chrome-cc/none (CrOS x86_64 0.4.0)
+//
+// TODO(satorux): Move this function to somewhere else: crbug.com/151605
+std::string GetDriveUserAgent() {
+  const char kDriveClientName[] = "chromedrive";
+
+  chrome::VersionInfo version_info;
+  const std::string version = (version_info.is_valid() ?
+                               version_info.Version() :
+                               std::string("unknown"));
+
+  // This part is <client_name>/<version>.
+  const std::string library = "chrome-cc/none";

[Daniel Erat, 2012/09/21 21:11:12]

nit: use const char kLibraryName[] here?

[satorux1, 2012/09/21 22:41:22]

Done.

+
+  const std::string os_cpu = webkit_glue::BuildOSCpuInfo();
+
+  return base::StringPrintf("%s-%s %s (%s)",
+                            kDriveClientName,
+                            version.c_str(),
+                            library.c_str(),
+                            os_cpu.c_str());
+}
+
 }  // namespace
 
 namespace gdata {
 
 //================================ AuthOperation ===============================
 
 AuthOperation::AuthOperation(OperationRegistry* registry,
                              const AuthStatusCallback& callback,
                              const std::vector<std::string>& scopes,
                              const std::string& refresh_token)
@@ skipping 100 matching lines @@
         BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE));
   } else if (!output_file_path_.empty()) {
     url_fetcher_->SaveResponseToFileAtPath(output_file_path_,
         BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE));
   }
 
   // Add request headers.
   // Note that SetExtraRequestHeaders clears the current headers and sets it
   // to the passed-in headers, so calling it for each header will result in
   // only the last header being set in request headers.
+  //
+  // TODO(satorux): The custom user-agent should be set only for Drive
+  // operations. crbug.com/151605
+  url_fetcher_->AddExtraRequestHeader("User-Agent: " + GetDriveUserAgent());
   url_fetcher_->AddExtraRequestHeader(kGDataVersionHeader);
   url_fetcher_->AddExtraRequestHeader(
         base::StringPrintf(kAuthorizationHeaderFormat, auth_token.data()));
   std::vector<std::string> headers = GetExtraRequestHeaders();
   for (size_t i = 0; i < headers.size(); ++i) {
     url_fetcher_->AddExtraRequestHeader(headers[i]);
     DVLOG(1) << "Extra header: " << headers[i];
   }
 
   // Set upload data if available.
@@ skipping 222 matching lines @@
 }
 
 void GetDataOperation::RunCallback(GDataErrorCode fetch_error_code,
                                    scoped_ptr<base::Value> value) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   if (!callback_.is_null())
     callback_.Run(fetch_error_code, value.Pass());
 }
 
 }  // namespace gdata