chrome/browser/chromeos/login/oauth2_login_verifier.h

   1 // Copyright (c) 2013 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 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_OAUTH2_LOGIN_VERIFIER_H_
   6 #define CHROME_BROWSER_CHROMEOS_LOGIN_OAUTH2_LOGIN_VERIFIER_H_
   7
   8 #include <string>
   9
  10 #include "base/basictypes.h"
  11 #include "base/callback_forward.h"
  12 #include "base/compiler_specific.h"
  13 #include "base/memory/ref_counted.h"
  14 #include "base/memory/scoped_ptr.h"
  15 #include "base/memory/weak_ptr.h"
  16 #include "chrome/browser/profiles/profile.h"
  17 #include "google_apis/gaia/gaia_auth_consumer.h"
  18 #include "google_apis/gaia/gaia_auth_fetcher.h"
  19 #include "google_apis/gaia/oauth2_access_token_fetcher.h"
  20 #include "google_apis/gaia/oauth2_token_service.h"
  21 #include "net/url_request/url_request_context_getter.h"
  22
  23 namespace chromeos {
  24
  25 // Given the OAuth2 refresh token, this class will try to exchange it for GAIA
  26 // credentials (SID+LSID) and populate current session's cookie jar.
  27 class OAuth2LoginVerifier : public base::SupportsWeakPtr<OAuth2LoginVerifier>,
  28                             public GaiaAuthConsumer,
  29                             public OAuth2TokenService::Consumer {
  30  public:
  31   typedef base::Callback<void(bool connection_error)> ErrorHandler;
  32
  33   class Delegate {
  34    public:
  35     virtual ~Delegate() {}
  36     // Invoked when cookie session is successfully merged.
  37     virtual void OnSessionMergeSuccess() = 0;
  38
  39     // Invoked when cookie session can not be merged.
  40     virtual void OnSessionMergeFailure(bool connection_error) = 0;
  41
  42     // Invoked when account list is retrieved during post-merge session
  43     // verification.
  44     virtual void OnListAccountsSuccess(const std::string& data) = 0;
  45
  46     // Invoked when post-merge session verification fails.
  47     virtual void OnListAccountsFailure(bool connection_error) = 0;
  48   };
  49
  50   OAuth2LoginVerifier(OAuth2LoginVerifier::Delegate* delegate,
  51                       net::URLRequestContextGetter* system_request_context,
  52                       net::URLRequestContextGetter* user_request_context,
  53                       const std::string& oauthlogin_access_token);
  54   virtual ~OAuth2LoginVerifier();
  55
  56   // Attempts to restore session from OAuth2 refresh token minting all necesarry
  57   // tokens along the way (OAuth2 access token, SID/LSID, GAIA service token).
  58   void VerifyProfileTokens(Profile* profile);
  59
  60  private:
  61   enum SessionRestoreType {
  62     RESTORE_UNDEFINED = 0,
  63     RESTORE_FROM_GAIA_TOKEN = 1,
  64     RESTORE_FROM_OAUTH2_REFRESH_TOKEN = 2,
  65   };
  66   // GaiaAuthConsumer overrides.
  67   virtual void OnUberAuthTokenSuccess(const std::string& token) OVERRIDE;
  68   virtual void OnUberAuthTokenFailure(
  69       const GoogleServiceAuthError& error) OVERRIDE;
  70   virtual void OnMergeSessionSuccess(const std::string& data) OVERRIDE;
  71   virtual void OnMergeSessionFailure(
  72       const GoogleServiceAuthError& error) OVERRIDE;
  73   virtual void OnListAccountsSuccess(const std::string& data) OVERRIDE;
  74   virtual void OnListAccountsFailure(
  75       const GoogleServiceAuthError& error) OVERRIDE;
  76
  77   // OAuth2TokenService::Consumer overrides.
  78   virtual void OnGetTokenSuccess(const OAuth2TokenService::Request* request,
  79                                  const std::string& access_token,
  80                                  const base::Time& expiration_time) OVERRIDE;
  81   virtual void OnGetTokenFailure(const OAuth2TokenService::Request* request,
  82                                  const GoogleServiceAuthError& error) OVERRIDE;
  83
  84   // Starts fetching OAuth1 access token for OAuthLogin call.
  85   void StartFetchingOAuthLoginAccessToken(Profile* profile);
  86
  87   // Starts OAuthLogin request for GAIA uber-token.
  88   void StartOAuthLoginForUberToken();
  89
  90   // Attempts to merge session from present |gaia_token_|.
  91   void StartMergeSession();
  92
  93   // Schedules post merge verification to ensure that browser session restore
  94   // hasn't stumped over SID/LSID.
  95   void SchedulePostMergeVerification();
  96
  97   // Starts post merge request verification.
  98   void StartPostRestoreVerification();
  99
 100   // Decides how to proceed on GAIA |error|. If the error looks temporary,
 101   // retries |task| after certain delay until max retry count is reached.
 102   void RetryOnError(const char* operation_id,
 103                     const GoogleServiceAuthError& error,
 104                     const base::Closure& task_to_retry,
 105                     const ErrorHandler& error_handler);
 106
 107   OAuth2LoginVerifier::Delegate* delegate_;
 108   scoped_refptr<net::URLRequestContextGetter> system_request_context_;
 109   scoped_refptr<net::URLRequestContextGetter> user_request_context_;
 110   scoped_ptr<GaiaAuthFetcher> gaia_fetcher_;
 111   std::string access_token_;
 112   std::string gaia_token_;
 113   scoped_ptr<OAuth2TokenService::Request> login_token_request_;
 114   // The retry counter. Increment this only when failure happened.
 115   int retry_count_;
 116
 117   DISALLOW_COPY_AND_ASSIGN(OAuth2LoginVerifier);
 118 };
 119
 120 }  // namespace chromeos
 121
 122 #endif  // CHROME_BROWSER_CHROMEOS_LOGIN_OAUTH2_LOGIN_VERIFIER_H_