OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "remoting/host/setup/win/auth_code_getter.h" | 5 #include "remoting/host/setup/win/auth_code_getter.h" |
6 | 6 |
7 #include "base/strings/utf_string_conversions.h" | 7 #include "base/strings/utf_string_conversions.h" |
8 #include "base/time/time.h" | 8 #include "base/time/time.h" |
9 #include "base/win/scoped_bstr.h" | 9 #include "base/win/scoped_bstr.h" |
10 #include "base/win/scoped_variant.h" | 10 #include "base/win/scoped_variant.h" |
11 #include "remoting/host/setup/oauth_helper.h" | 11 #include "remoting/host/setup/oauth_helper.h" |
12 | 12 |
13 namespace { | 13 namespace { |
14 const int kUrlPollIntervalMs = 100; | 14 const int kUrlPollIntervalMs = 100; |
15 } // namespace remoting | 15 } // namespace remoting |
16 | 16 |
17 namespace remoting { | 17 namespace remoting { |
18 | 18 |
19 AuthCodeGetter::AuthCodeGetter() : | 19 AuthCodeGetter::AuthCodeGetter() : |
20 browser_(NULL), | 20 browser_(NULL), |
21 timer_interval_(base::TimeDelta::FromMilliseconds(kUrlPollIntervalMs)) { | 21 timer_interval_(base::TimeDelta::FromMilliseconds(kUrlPollIntervalMs)) { |
22 } | 22 } |
23 | 23 |
24 AuthCodeGetter::~AuthCodeGetter() { | 24 AuthCodeGetter::~AuthCodeGetter() { |
25 KillBrowser(); | 25 KillBrowser(); |
26 } | 26 } |
27 | 27 |
28 void AuthCodeGetter::GetAuthCode( | 28 void AuthCodeGetter::GetAuthCode( |
29 base::Callback<void(const std::string&)> on_auth_code) { | 29 base::Callback<void(const std::string&)> on_auth_code) { |
30 if (browser_) { | 30 if (browser_.get()) { |
31 on_auth_code.Run(""); | 31 on_auth_code.Run(""); |
32 return; | 32 return; |
33 } | 33 } |
34 on_auth_code_ = on_auth_code; | 34 on_auth_code_ = on_auth_code; |
35 HRESULT hr = browser_.CreateInstance(CLSID_InternetExplorer, NULL, | 35 HRESULT hr = browser_.CreateInstance(CLSID_InternetExplorer, NULL, |
36 CLSCTX_LOCAL_SERVER); | 36 CLSCTX_LOCAL_SERVER); |
37 if (FAILED(hr)) { | 37 if (FAILED(hr)) { |
38 on_auth_code_.Run(""); | 38 on_auth_code_.Run(""); |
39 return; | 39 return; |
40 } | 40 } |
(...skipping 19 matching lines...) Expand all Loading... |
60 std::string auth_code; | 60 std::string auth_code; |
61 if (TestBrowserUrl(&auth_code)) { | 61 if (TestBrowserUrl(&auth_code)) { |
62 on_auth_code_.Run(auth_code); | 62 on_auth_code_.Run(auth_code); |
63 } else { | 63 } else { |
64 StartTimer(); | 64 StartTimer(); |
65 } | 65 } |
66 } | 66 } |
67 | 67 |
68 bool AuthCodeGetter::TestBrowserUrl(std::string* auth_code) { | 68 bool AuthCodeGetter::TestBrowserUrl(std::string* auth_code) { |
69 *auth_code = ""; | 69 *auth_code = ""; |
70 if (!browser_) { | 70 if (!browser_.get()) { |
71 return true; | 71 return true; |
72 } | 72 } |
73 base::win::ScopedBstr url; | 73 base::win::ScopedBstr url; |
74 HRESULT hr = browser_->get_LocationName(url.Receive()); | 74 HRESULT hr = browser_->get_LocationName(url.Receive()); |
75 if (!SUCCEEDED(hr)) { | 75 if (!SUCCEEDED(hr)) { |
76 KillBrowser(); | 76 KillBrowser(); |
77 return true; | 77 return true; |
78 } | 78 } |
79 *auth_code = GetOauthCodeInUrl(base::WideToUTF8(static_cast<BSTR>(url)), | 79 *auth_code = GetOauthCodeInUrl(base::WideToUTF8(static_cast<BSTR>(url)), |
80 GetDefaultOauthRedirectUrl()); | 80 GetDefaultOauthRedirectUrl()); |
81 if (!auth_code->empty()) { | 81 if (!auth_code->empty()) { |
82 KillBrowser(); | 82 KillBrowser(); |
83 return true; | 83 return true; |
84 } | 84 } |
85 return false; | 85 return false; |
86 } | 86 } |
87 | 87 |
88 void AuthCodeGetter::KillBrowser() { | 88 void AuthCodeGetter::KillBrowser() { |
89 if (browser_) { | 89 if (browser_.get()) { |
90 browser_->Quit(); | 90 browser_->Quit(); |
91 browser_.Release(); | 91 browser_.Release(); |
92 } | 92 } |
93 } | 93 } |
94 | 94 |
95 } // namespace remoting | 95 } // namespace remoting |
OLD | NEW |