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
OAuth2TokenFetcher(OAuth2TokenFetcher::Delegate * delegate,net::URLRequestContextGetter * context_getter)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
~OAuth2TokenFetcher()39 OAuth2TokenFetcher::~OAuth2TokenFetcher() {
40 }
41
StartExchangeFromCookies(const std::string & session_index)42 void OAuth2TokenFetcher::StartExchangeFromCookies(
43 const std::string& session_index) {
44 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
45 session_index_ = session_index;
46 // Delay the verification if the network is not connected or on a captive
47 // portal.
48 const NetworkState* default_network =
49 NetworkHandler::Get()->network_state_handler()->DefaultNetwork();
50 if (!default_network ||
51 default_network->connection_state() == shill::kStatePortal) {
52 // If network is offline, defer the token fetching until online.
53 VLOG(1) << "Network is offline. Deferring OAuth2 token fetch.";
54 BrowserThread::PostDelayedTask(
55 BrowserThread::UI,
56 FROM_HERE,
57 base::Bind(&OAuth2TokenFetcher::StartExchangeFromCookies,
58 AsWeakPtr(),
59 session_index),
60 base::TimeDelta::FromMilliseconds(kRequestRestartDelay));
61 return;
62 }
63 auth_fetcher_.StartCookieForOAuthLoginTokenExchange(session_index);
64 }
65
StartExchangeFromAuthCode(const std::string & auth_code)66 void OAuth2TokenFetcher::StartExchangeFromAuthCode(
67 const std::string& auth_code) {
68 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
69 auth_code_ = auth_code;
70 // Delay the verification if the network is not connected or on a captive
71 // portal.
72 const NetworkState* default_network =
73 NetworkHandler::Get()->network_state_handler()->DefaultNetwork();
74 if (!default_network ||
75 default_network->connection_state() == shill::kStatePortal) {
76 // If network is offline, defer the token fetching until online.
77 VLOG(1) << "Network is offline. Deferring OAuth2 token fetch.";
78 BrowserThread::PostDelayedTask(
79 BrowserThread::UI, FROM_HERE,
80 base::Bind(&OAuth2TokenFetcher::StartExchangeFromAuthCode,
81 AsWeakPtr(),
82 auth_code),
83 base::TimeDelta::FromMilliseconds(kRequestRestartDelay));
84 return;
85 }
86 auth_fetcher_.StartAuthCodeForOAuth2TokenExchange(auth_code);
87 }
88
OnClientOAuthSuccess(const GaiaAuthConsumer::ClientOAuthResult & oauth_tokens)89 void OAuth2TokenFetcher::OnClientOAuthSuccess(
90 const GaiaAuthConsumer::ClientOAuthResult& oauth_tokens) {
91 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
92 VLOG(1) << "Got OAuth2 tokens!";
93 retry_count_ = 0;
94 oauth_tokens_ = oauth_tokens;
95 delegate_->OnOAuth2TokensAvailable(oauth_tokens_);
96 }
97
OnClientOAuthFailure(const GoogleServiceAuthError & error)98 void OAuth2TokenFetcher::OnClientOAuthFailure(
99 const GoogleServiceAuthError& error) {
100 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
101 RetryOnError(error,
102 auth_code_.empty()
103 ? base::Bind(&OAuth2TokenFetcher::StartExchangeFromCookies,
104 AsWeakPtr(),
105 session_index_)
106 : base::Bind(&OAuth2TokenFetcher::StartExchangeFromAuthCode,
107 AsWeakPtr(),
108 auth_code_),
109 base::Bind(&Delegate::OnOAuth2TokensFetchFailed,
110 base::Unretained(delegate_)));
111 }
112
RetryOnError(const GoogleServiceAuthError & error,const base::Closure & task,const base::Closure & error_handler)113 void OAuth2TokenFetcher::RetryOnError(const GoogleServiceAuthError& error,
114 const base::Closure& task,
115 const base::Closure& error_handler) {
116 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
117 if ((error.state() == GoogleServiceAuthError::CONNECTION_FAILED ||
118 error.state() == GoogleServiceAuthError::SERVICE_UNAVAILABLE ||
119 error.state() == GoogleServiceAuthError::REQUEST_CANCELED) &&
120 retry_count_ < kMaxRequestAttemptCount) {
121 retry_count_++;
122 BrowserThread::PostDelayedTask(
123 BrowserThread::UI, FROM_HERE, task,
124 base::TimeDelta::FromMilliseconds(kRequestRestartDelay));
125 return;
126 }
127 LOG(ERROR) << "Unrecoverable error or retry count max reached. State: "
128 << error.state() << ", network error: " << error.network_error()
129 << ", message: " << error.error_message();
130 error_handler.Run();
131 }
132
133 } // namespace chromeos
134