Add data usage tracking for GAIA auth api

google_apis/gaia/gaia_auth_util.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 "google_apis/gaia/gaia_auth_util.h"
 
 #include <stddef.h>
 
 #include <memory>
 
 #include "base/json/json_reader.h"
 #include "base/logging.h"
 #include "base/strings/string_split.h"
 #include "base/strings/string_util.h"
 #include "base/values.h"
 #include "google_apis/gaia/gaia_urls.h"
+#include "net/url_request/url_fetcher.h"
+#include "net/url_request/url_request.h"
 #include "url/gurl.h"
 
 namespace gaia {
 
 namespace {
 
 const char kGmailDomain[] = "gmail.com";
 const char kGooglemailDomain[] = "googlemail.com";
 
+const void* kURLRequestUserDataKey =
+    static_cast<const void*>(&kURLRequestUserDataKey);
+
 std::string CanonicalizeEmailImpl(const std::string& email_address,
                                   bool change_googlemail_to_gmail) {
   std::vector<std::string> parts = base::SplitString(
       email_address, "@", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
   if (parts.size() != 2U) {
     NOTREACHED() << "expecting exactly one @, but got "
                  << (parts.empty() ? 0 : parts.size() - 1)
                  << " : " << email_address;
   } else {
     if (change_googlemail_to_gmail && parts[1] == kGooglemailDomain)
       parts[1] = kGmailDomain;
 
     if (parts[1] == kGmailDomain)  // only strip '.' for gmail accounts.
       base::RemoveChars(parts[0], ".", &parts[0]);
   }
 
   std::string new_email = base::ToLowerASCII(base::JoinString(parts, "@"));
   VLOG(1) << "Canonicalized " << email_address << " to " << new_email;
   return new_email;
 }
 
+class GaiaURLRequestUserData : public base::SupportsUserData::Data {
+ public:
+  static base::SupportsUserData::Data* Create() {
+    return new GaiaURLRequestUserData();

[tbansal1, 2017/01/23 21:03:05]

qq, who frees up this data? (I do not know enough about SupportsUserData).

[Raj, 2017/01/23 21:53:02]

It will be owned by SupportsUserData and freed in its destructor.

+  }
+};
+
 }  // namespace
 
 
 ListedAccount::ListedAccount() {}
 
 ListedAccount::ListedAccount(const ListedAccount& other) = default;
 
 ListedAccount::~ListedAccount() {}
 
 bool ListedAccount::operator==(const ListedAccount& other) const {
@@ skipping 116 matching lines @@
           if (list)
             list->push_back(listed_account);
         }
       }
     }
   }
 
   return true;
 }
 
+bool RequestOriginatedFromGaia(const net::URLRequest& request) {
+  return request.GetUserData(kURLRequestUserDataKey) != nullptr;
+}
+
+void MarkURLFetcherAsGaia(net::URLFetcher* fetcher) {
+  DCHECK(fetcher);
+  fetcher->SetURLRequestUserData(kURLRequestUserDataKey,
+                                 base::Bind(&GaiaURLRequestUserData::Create));

[tbansal1, 2017/01/23 21:03:05]

#include "base/bind.h"

[Raj, 2017/01/23 21:53:02]

Done.

+}
+
 }  // namespace gaia