| 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/protocol/fake_authenticator.h" | 5 #include "remoting/protocol/fake_authenticator.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "base/base64.h" | 9 #include "base/base64.h" |
| 10 #include "base/callback_helpers.h" | 10 #include "base/callback_helpers.h" |
| (...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 94 Action action, | 94 Action action, |
| 95 bool async) | 95 bool async) |
| 96 : type_(type), round_trips_(round_trips), action_(action), async_(async) {} | 96 : type_(type), round_trips_(round_trips), action_(action), async_(async) {} |
| 97 | 97 |
| 98 FakeAuthenticator::~FakeAuthenticator() {} | 98 FakeAuthenticator::~FakeAuthenticator() {} |
| 99 | 99 |
| 100 void FakeAuthenticator::set_messages_till_started(int messages) { | 100 void FakeAuthenticator::set_messages_till_started(int messages) { |
| 101 messages_till_started_ = messages; | 101 messages_till_started_ = messages; |
| 102 } | 102 } |
| 103 | 103 |
| 104 void FakeAuthenticator::Resume() { |
| 105 base::ResetAndReturn(&resume_closure_).Run(); |
| 106 } |
| 107 |
| 104 Authenticator::State FakeAuthenticator::state() const { | 108 Authenticator::State FakeAuthenticator::state() const { |
| 105 EXPECT_LE(messages_, round_trips_ * 2); | 109 EXPECT_LE(messages_, round_trips_ * 2); |
| 110 |
| 111 if (messages_ == pause_message_index_ && !resume_closure_.is_null()) |
| 112 return PROCESSING_MESSAGE; |
| 113 |
| 106 if (messages_ >= round_trips_ * 2) { | 114 if (messages_ >= round_trips_ * 2) { |
| 107 if (action_ == REJECT) { | 115 if (action_ == REJECT) { |
| 108 return REJECTED; | 116 return REJECTED; |
| 109 } else { | 117 } else { |
| 110 return ACCEPTED; | 118 return ACCEPTED; |
| 111 } | 119 } |
| 112 } | 120 } |
| 113 | 121 |
| 114 // Don't send the last message if this is a host that wants to | 122 // Don't send the last message if this is a host that wants to |
| 115 // reject a connection. | 123 // reject a connection. |
| (...skipping 29 matching lines...) Expand all Loading... |
| 145 | 153 |
| 146 // On the client receive the key in the last message. | 154 // On the client receive the key in the last message. |
| 147 if (type_ == CLIENT && messages_ == round_trips_ * 2 - 1) { | 155 if (type_ == CLIENT && messages_ == round_trips_ * 2 - 1) { |
| 148 std::string key_base64 = | 156 std::string key_base64 = |
| 149 message->TextNamed(buzz::QName(kChromotingXmlNamespace, "key")); | 157 message->TextNamed(buzz::QName(kChromotingXmlNamespace, "key")); |
| 150 EXPECT_TRUE(!key_base64.empty()); | 158 EXPECT_TRUE(!key_base64.empty()); |
| 151 EXPECT_TRUE(base::Base64Decode(key_base64, &auth_key_)); | 159 EXPECT_TRUE(base::Base64Decode(key_base64, &auth_key_)); |
| 152 } | 160 } |
| 153 | 161 |
| 154 ++messages_; | 162 ++messages_; |
| 163 if (messages_ == pause_message_index_) { |
| 164 resume_closure_ = resume_callback; |
| 165 return; |
| 166 } |
| 155 resume_callback.Run(); | 167 resume_callback.Run(); |
| 156 } | 168 } |
| 157 | 169 |
| 158 std::unique_ptr<buzz::XmlElement> FakeAuthenticator::GetNextMessage() { | 170 std::unique_ptr<buzz::XmlElement> FakeAuthenticator::GetNextMessage() { |
| 159 EXPECT_EQ(MESSAGE_READY, state()); | 171 EXPECT_EQ(MESSAGE_READY, state()); |
| 160 | 172 |
| 161 std::unique_ptr<buzz::XmlElement> result(new buzz::XmlElement( | 173 std::unique_ptr<buzz::XmlElement> result(new buzz::XmlElement( |
| 162 buzz::QName(kChromotingXmlNamespace, "authentication"))); | 174 buzz::QName(kChromotingXmlNamespace, "authentication"))); |
| 163 buzz::XmlElement* id = new buzz::XmlElement( | 175 buzz::XmlElement* id = new buzz::XmlElement( |
| 164 buzz::QName(kChromotingXmlNamespace, "id")); | 176 buzz::QName(kChromotingXmlNamespace, "id")); |
| (...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 209 const std::string& local_jid, | 221 const std::string& local_jid, |
| 210 const std::string& remote_jid) { | 222 const std::string& remote_jid) { |
| 211 std::unique_ptr<FakeAuthenticator> authenticator(new FakeAuthenticator( | 223 std::unique_ptr<FakeAuthenticator> authenticator(new FakeAuthenticator( |
| 212 FakeAuthenticator::HOST, round_trips_, action_, async_)); | 224 FakeAuthenticator::HOST, round_trips_, action_, async_)); |
| 213 authenticator->set_messages_till_started(messages_till_started_); | 225 authenticator->set_messages_till_started(messages_till_started_); |
| 214 return std::move(authenticator); | 226 return std::move(authenticator); |
| 215 } | 227 } |
| 216 | 228 |
| 217 } // namespace protocol | 229 } // namespace protocol |
| 218 } // namespace remoting | 230 } // namespace remoting |
| OLD | NEW |