OLD | NEW |
| (Empty) |
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_MERGE_SESSION_THROTTLE_H_ | |
6 #define CHROME_BROWSER_CHROMEOS_LOGIN_MERGE_SESSION_THROTTLE_H_ | |
7 | |
8 #include <set> | |
9 | |
10 #include "base/atomic_ref_count.h" | |
11 #include "base/memory/ref_counted.h" | |
12 #include "base/memory/weak_ptr.h" | |
13 #include "content/public/browser/resource_throttle.h" | |
14 #include "net/base/completion_callback.h" | |
15 #include "webkit/common/resource_type.h" | |
16 | |
17 class Profile; | |
18 | |
19 namespace net { | |
20 class URLRequest; | |
21 } | |
22 | |
23 namespace chromeos { | |
24 class OAuth2LoginManager; | |
25 } | |
26 | |
27 // Used to show an interstitial page while merge session process (cookie | |
28 // reconstruction from OAuth2 refresh token in ChromeOS login) is still in | |
29 // progress while we are attempting to load a google property. | |
30 class MergeSessionThrottle | |
31 : public content::ResourceThrottle, | |
32 public base::SupportsWeakPtr<MergeSessionThrottle> { | |
33 public: | |
34 // Passed a boolean indicating whether or not it is OK to proceed with the | |
35 // page load. | |
36 typedef base::Closure CompletionCallback; | |
37 | |
38 explicit MergeSessionThrottle(net::URLRequest* request, | |
39 ResourceType::Type resource_type); | |
40 virtual ~MergeSessionThrottle(); | |
41 | |
42 // content::ResourceThrottle implementation: | |
43 virtual void WillStartRequest(bool* defer) OVERRIDE; | |
44 virtual const char* GetNameForLogging() const OVERRIDE; | |
45 | |
46 // Checks if session is already merged. | |
47 static bool AreAllSessionMergedAlready(); | |
48 | |
49 private: | |
50 | |
51 // MergeSessionLoadPage callback. | |
52 void OnBlockingPageComplete(); | |
53 | |
54 // Erase the state associated with a deferred load request. | |
55 void ClearRequestInfo(); | |
56 bool IsRemote(const GURL& url) const; | |
57 | |
58 // True if we |url| loading should be delayed. The function | |
59 // is safe to be called on any thread. | |
60 bool ShouldDelayUrl(const GURL& url) const; | |
61 | |
62 // Adds/removes |profile| to/from the blocking profiles set. | |
63 static void BlockProfile(Profile* profile); | |
64 static void UnblockProfile(Profile* profile); | |
65 | |
66 // Helper method that checks if we should delay reasource loading based on | |
67 // the state of the Profile that's derived from |render_process_id| and | |
68 // |render_view_id|. | |
69 static bool ShouldDelayRequest(int render_process_id, | |
70 int render_view_id); | |
71 | |
72 // Tests merge session status and if needed generates request | |
73 // waiter (for ResourceType::XHR content) or shows interstitial page | |
74 // (for ResourceType::MAIN_FRAME). | |
75 // The function must be called from UI thread. | |
76 static void DeleayResourceLoadingOnUIThread( | |
77 ResourceType::Type resource_type, | |
78 int render_process_id, | |
79 int render_view_id, | |
80 const GURL& url, | |
81 const MergeSessionThrottle::CompletionCallback& callback); | |
82 | |
83 net::URLRequest* request_; | |
84 ResourceType::Type resource_type_; | |
85 | |
86 // Global counter that keeps the track of session merge status for all | |
87 // encountered profiles. This is used to determine if a throttle should | |
88 // even be even added to new requests. Value of 0 (initial) means that we | |
89 // probably have some profiles to restore, while 1 means that all known | |
90 // profiles are restored. | |
91 static base::AtomicRefCount all_profiles_restored_; | |
92 | |
93 DISALLOW_COPY_AND_ASSIGN(MergeSessionThrottle); | |
94 }; | |
95 | |
96 #endif // CHROME_BROWSER_CHROMEOS_LOGIN_MERGE_SESSION_THROTTLE_H_ | |
OLD | NEW |