chromeos/login/auth/online_attempt.h

   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 #ifndef CHROMEOS_LOGIN_AUTH_ONLINE_ATTEMPT_H_
   6 #define CHROMEOS_LOGIN_AUTH_ONLINE_ATTEMPT_H_
   7
   8 #include <string>
   9
  10 #include "base/compiler_specific.h"
  11 #include "base/gtest_prod_util.h"
  12 #include "base/memory/ref_counted.h"
  13 #include "base/memory/scoped_ptr.h"
  14 #include "base/memory/weak_ptr.h"
  15 #include "chromeos/chromeos_export.h"
  16 #include "chromeos/login/auth/auth_status_consumer.h"
  17 #include "google_apis/gaia/gaia_auth_consumer.h"
  18 #include "google_apis/gaia/google_service_auth_error.h"
  19
  20 class GaiaAuthFetcher;
  21
  22 namespace net {
  23 class URLRequestContextGetter;
  24 }
  25
  26 namespace chromeos {
  27 class AuthAttemptState;
  28 class AuthAttemptStateResolver;
  29
  30 class CHROMEOS_EXPORT OnlineAttempt : public GaiaAuthConsumer {
  31  public:
  32   OnlineAttempt(AuthAttemptState* current_attempt,
  33                 AuthAttemptStateResolver* callback);
  34   virtual ~OnlineAttempt();
  35
  36   // Initiate the online login attempt either through client or auth login.
  37   // Status will be recorded in |current_attempt|, and resolver_->Resolve() will
  38   // be called on the IO thread when useful state is available.
  39   // Must be called on the UI thread.
  40   void Initiate(net::URLRequestContextGetter* request_context);
  41
  42   // GaiaAuthConsumer overrides. Callbacks from GaiaAuthFetcher
  43   virtual void OnClientLoginFailure(
  44       const GoogleServiceAuthError& error) OVERRIDE;
  45   virtual void OnClientLoginSuccess(
  46       const GaiaAuthConsumer::ClientLoginResult& credentials) OVERRIDE;
  47
  48  private:
  49   FRIEND_TEST_ALL_PREFIXES(OnlineAttemptTest, LoginSuccess);
  50   FRIEND_TEST_ALL_PREFIXES(OnlineAttemptTest, TwoFactorSuccess);
  51
  52   // Milliseconds until we timeout our attempt to hit ClientLogin.
  53   static const int kClientLoginTimeoutMs;
  54
  55   void TryClientLogin();
  56   void CancelClientLogin();
  57
  58   void TriggerResolve(const AuthFailure& outcome);
  59
  60   bool HasPendingFetch();
  61   void CancelRequest();
  62
  63   scoped_refptr<base::MessageLoopProxy> message_loop_;
  64
  65   AuthAttemptState* const attempt_;
  66   AuthAttemptStateResolver* const resolver_;
  67
  68   // Handles ClientLogin communications with Gaia.
  69   scoped_ptr<GaiaAuthFetcher> client_fetcher_;
  70
  71   // Whether we're willing to re-try the ClientLogin attempt.
  72   bool try_again_;
  73
  74   // Used to cancel the CancelClientLogin closure.
  75   base::WeakPtrFactory<OnlineAttempt> weak_factory_;
  76
  77   friend class OnlineAttemptTest;
  78   DISALLOW_COPY_AND_ASSIGN(OnlineAttempt);
  79 };
  80
  81 }  // namespace chromeos
  82
  83 #endif  // CHROMEOS_LOGIN_AUTH_ONLINE_ATTEMPT_H_