chrome/browser/chromeos/login/signin/oauth2_token_fetcher.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/chromeos/login/signin/oauth2_token_fetcher.h"
   6
   7 #include "base/logging.h"
   8 #include "base/strings/string_util.h"
   9 #include "chromeos/network/network_handler.h"
  10 #include "chromeos/network/network_state.h"
  11 #include "chromeos/network/network_state_handler.h"
  12 #include "content/public/browser/browser_thread.h"
  13 #include "google_apis/gaia/gaia_constants.h"
  14 #include "google_apis/gaia/google_service_auth_error.h"
  15 #include "third_party/cros_system_api/dbus/service_constants.h"
  16
  17 using content::BrowserThread;
  18
  19 namespace {
  20
  21 // OAuth token request max retry count.
  22 const int kMaxRequestAttemptCount = 5;
  23 // OAuth token request retry delay in milliseconds.
  24 const int kRequestRestartDelay = 3000;
  25
  26 }  // namespace
  27
  28 namespace chromeos {
  29
  30 OAuth2TokenFetcher::OAuth2TokenFetcher(
  31     OAuth2TokenFetcher::Delegate* delegate,
  32     net::URLRequestContextGetter* context_getter)
  33     : delegate_(delegate),
  34       auth_fetcher_(this, GaiaConstants::kChromeSource, context_getter),
  35       retry_count_(0) {
  36   DCHECK(delegate);
  37 }
  38
  39 OAuth2TokenFetcher::~OAuth2TokenFetcher() {
  40 }
  41
  42 void OAuth2TokenFetcher::StartExchangeFromCookies(
  43     const std::string& session_index,
  44     const std::string& signin_scoped_device_id) {
  45   DCHECK_CURRENTLY_ON(BrowserThread::UI);
  46   session_index_ = session_index;
  47   signin_scoped_device_id_ = signin_scoped_device_id;
  48   // Delay the verification if the network is not connected or on a captive
  49   // portal.
  50   const NetworkState* default_network =
  51       NetworkHandler::Get()->network_state_handler()->DefaultNetwork();
  52   if (!default_network ||
  53       default_network->connection_state() == shill::kStatePortal) {
  54     // If network is offline, defer the token fetching until online.
  55     VLOG(1) << "Network is offline.  Deferring OAuth2 token fetch.";
  56     BrowserThread::PostDelayedTask(
  57         BrowserThread::UI,
  58         FROM_HERE,
  59         base::Bind(&OAuth2TokenFetcher::StartExchangeFromCookies,
  60                    AsWeakPtr(),
  61                    session_index,
  62                    signin_scoped_device_id),
  63         base::TimeDelta::FromMilliseconds(kRequestRestartDelay));
  64     return;
  65   }
  66   auth_fetcher_.StartCookieForOAuthLoginTokenExchangeWithDeviceId(
  67       session_index, signin_scoped_device_id);
  68 }
  69
  70 void OAuth2TokenFetcher::StartExchangeFromAuthCode(
  71     const std::string& auth_code,
  72     const std::string& signin_scoped_device_id) {
  73   DCHECK_CURRENTLY_ON(BrowserThread::UI);
  74   auth_code_ = auth_code;
  75   signin_scoped_device_id_ = signin_scoped_device_id;
  76   // Delay the verification if the network is not connected or on a captive
  77   // portal.
  78   const NetworkState* default_network =
  79       NetworkHandler::Get()->network_state_handler()->DefaultNetwork();
  80   if (!default_network ||
  81       default_network->connection_state() == shill::kStatePortal) {
  82     // If network is offline, defer the token fetching until online.
  83     VLOG(1) << "Network is offline.  Deferring OAuth2 token fetch.";
  84     BrowserThread::PostDelayedTask(
  85         BrowserThread::UI, FROM_HERE,
  86         base::Bind(&OAuth2TokenFetcher::StartExchangeFromAuthCode, AsWeakPtr(),
  87                    auth_code, signin_scoped_device_id),
  88         base::TimeDelta::FromMilliseconds(kRequestRestartDelay));
  89     return;
  90   }
  91   auth_fetcher_.StartAuthCodeForOAuth2TokenExchangeWithDeviceId(
  92       auth_code, signin_scoped_device_id);
  93 }
  94
  95 void OAuth2TokenFetcher::OnClientOAuthSuccess(
  96     const GaiaAuthConsumer::ClientOAuthResult& oauth_tokens) {
  97   DCHECK_CURRENTLY_ON(BrowserThread::UI);
  98   VLOG(1) << "Got OAuth2 tokens!";
  99   retry_count_ = 0;
 100   oauth_tokens_ = oauth_tokens;
 101   delegate_->OnOAuth2TokensAvailable(oauth_tokens_);
 102 }
 103
 104 void OAuth2TokenFetcher::OnClientOAuthFailure(
 105     const GoogleServiceAuthError& error) {
 106   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 107   RetryOnError(
 108       error,
 109       auth_code_.empty()
 110           ? base::Bind(&OAuth2TokenFetcher::StartExchangeFromCookies,
 111                        AsWeakPtr(), session_index_, signin_scoped_device_id_)
 112           : base::Bind(&OAuth2TokenFetcher::StartExchangeFromAuthCode,
 113                        AsWeakPtr(), auth_code_, signin_scoped_device_id_),
 114       base::Bind(&Delegate::OnOAuth2TokensFetchFailed,
 115                  base::Unretained(delegate_)));
 116 }
 117
 118 void OAuth2TokenFetcher::RetryOnError(const GoogleServiceAuthError& error,
 119                                       const base::Closure& task,
 120                                       const base::Closure& error_handler) {
 121   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 122   if (error.IsTransientError() && retry_count_ < kMaxRequestAttemptCount) {
 123     retry_count_++;
 124     BrowserThread::PostDelayedTask(
 125         BrowserThread::UI, FROM_HERE, task,
 126         base::TimeDelta::FromMilliseconds(kRequestRestartDelay));
 127     return;
 128   }
 129   LOG(ERROR) << "Unrecoverable error or retry count max reached. State: "
 130              << error.state() << ", network error: " << error.network_error()
 131              << ", message: " << error.error_message();
 132   error_handler.Run();
 133 }
 134
 135 }  // namespace chromeos