remoting/signaling/xmpp_login_handler_unittest.cc

   1 // Copyright 2015 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 #include "remoting/signaling/xmpp_login_handler.h"
   6
   7 #include "base/base64.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "base/run_loop.h"
  10 #include "remoting/signaling/xmpp_stream_parser.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12 #include "third_party/webrtc/libjingle/xmllite/xmlelement.h"
  13
  14 #ifdef SendMessage
  15 #undef SendMessage
  16 #endif
  17
  18 #ifdef ERROR
  19 #undef ERROR
  20 #endif
  21
  22 namespace remoting {
  23
  24 char kTestUsername[] = "testUsername@gmail.com";
  25 char kTestToken[] = "testToken";
  26
  27 class XmppLoginHandlerTest : public testing::Test,
  28                              public XmppLoginHandler::Delegate {
  29  public:
  30   XmppLoginHandlerTest()
  31       : start_tls_called_(false), error_(SignalStrategy::OK) {}
  32
  33   void TearDown() override {
  34     login_handler_.reset();
  35     parser_.reset();
  36     base::RunLoop().RunUntilIdle();
  37   }
  38
  39   void SendMessage(const std::string& message) override {
  40     sent_data_ += message;
  41     if (delete_login_handler_from_delegate_)
  42       login_handler_.reset();
  43   }
  44
  45   void StartTls() override {
  46     start_tls_called_ = true;
  47     if (delete_login_handler_from_delegate_)
  48       login_handler_.reset();
  49   }
  50
  51   void OnHandshakeDone(const std::string& jid,
  52                        scoped_ptr<XmppStreamParser> parser) override {
  53     jid_ = jid;
  54     parser_ = parser.Pass();
  55     if (delete_login_handler_from_delegate_)
  56       login_handler_.reset();
  57   }
  58
  59   void OnLoginHandlerError(SignalStrategy::Error error) override {
  60     EXPECT_NE(error, SignalStrategy::OK);
  61     error_ = error;
  62     if (delete_login_handler_from_delegate_)
  63       login_handler_.reset();
  64   }
  65
  66  protected:
  67   void HandshakeBase();
  68
  69   base::MessageLoop message_loop_;
  70
  71   scoped_ptr<XmppLoginHandler> login_handler_;
  72   std::string sent_data_;
  73   bool start_tls_called_;
  74   std::string jid_;
  75   scoped_ptr<XmppStreamParser> parser_;
  76   SignalStrategy::Error error_;
  77   bool delete_login_handler_from_delegate_ = false;
  78 };
  79
  80 void XmppLoginHandlerTest::HandshakeBase() {
  81   login_handler_.reset(
  82       new XmppLoginHandler("google.com", kTestUsername, kTestToken,
  83                            XmppLoginHandler::TlsMode::WITHOUT_HANDSHAKE, this));
  84   login_handler_->Start();
  85   EXPECT_TRUE(start_tls_called_);
  86
  87   login_handler_->OnTlsStarted();
  88   std::string cookie;
  89   base::Base64Encode(
  90       std::string("\0", 1) + kTestUsername + std::string("\0", 1) + kTestToken,
  91       &cookie);
  92   EXPECT_EQ(
  93       sent_data_,
  94       "<stream:stream to=\"google.com\" version=\"1.0\" "
  95           "xmlns=\"jabber:client\" "
  96           "xmlns:stream=\"http://etherx.jabber.org/streams\">"
  97       "<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" mechanism=\"X-OAUTH2\" "
  98           "auth:service=\"oauth2\" auth:allow-generated-jid=\"true\" "
  99           "auth:client-uses-full-bind-result=\"true\" "
 100           "auth:allow-non-google-login=\"true\" "
 101           "xmlns:auth=\"http://www.google.com/talk/protocol/auth\">" + cookie +
 102       "</auth>");
 103   sent_data_.clear();
 104
 105   login_handler_->OnDataReceived(
 106       "<stream:stream from=\"google.com\" id=\"DCDDE5171CB2154A\" "
 107           "version=\"1.0\" "
 108           "xmlns:stream=\"http://etherx.jabber.org/streams\" "
 109           "xmlns=\"jabber:client\">"
 110         "<stream:features>"
 111           "<mechanisms xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">"
 112             "<mechanism>X-OAUTH2</mechanism>"
 113             "<mechanism>X-GOOGLE-TOKEN</mechanism>"
 114             "<mechanism>PLAIN</mechanism>"
 115           "</mechanisms>"
 116         "</stream:features>");
 117 }
 118
 119 TEST_F(XmppLoginHandlerTest, SuccessfulAuth) {
 120   HandshakeBase();
 121
 122   login_handler_->OnDataReceived(
 123       "<success xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\"/>");
 124   EXPECT_EQ(
 125       sent_data_,
 126       "<stream:stream to=\"google.com\" version=\"1.0\" "
 127           "xmlns=\"jabber:client\" "
 128           "xmlns:stream=\"http://etherx.jabber.org/streams\">"
 129         "<iq type=\"set\" id=\"0\">"
 130           "<bind xmlns=\"urn:ietf:params:xml:ns:xmpp-bind\">"
 131             "<resource>chromoting</resource>"
 132           "</bind>"
 133         "</iq>"
 134         "<iq type=\"set\" id=\"1\">"
 135           "<session xmlns=\"urn:ietf:params:xml:ns:xmpp-session\"/>"
 136         "</iq>");
 137   sent_data_.clear();
 138
 139   // |login_handler_| will call OnHandshakeDone() which will delete
 140   // |login_handler_|.
 141   delete_login_handler_from_delegate_ = true;
 142
 143   login_handler_->OnDataReceived(
 144       "<stream:stream from=\"google.com\" id=\"104FA10576E2AA80\" "
 145           "version=\"1.0\" "
 146           "xmlns:stream=\"http://etherx.jabber.org/streams\" "
 147           "xmlns=\"jabber:client\">"
 148         "<stream:features>"
 149           "<bind xmlns=\"urn:ietf:params:xml:ns:xmpp-bind\"/>"
 150           "<session xmlns=\"urn:ietf:params:xml:ns:xmpp-session\"/>"
 151         "</stream:features>"
 152         "<iq id=\"0\" type=\"result\">"
 153           "<bind xmlns=\"urn:ietf:params:xml:ns:xmpp-bind\">"
 154             "<jid>" + std::string(kTestUsername) + "/chromoting52B4920E</jid>"
 155           "</bind>"
 156         "</iq>"
 157         "<iq type=\"result\" id=\"1\"/>");
 158
 159   EXPECT_EQ(jid_, std::string(kTestUsername) + "/chromoting52B4920E");
 160   EXPECT_TRUE(parser_);
 161   EXPECT_FALSE(login_handler_);
 162 }
 163
 164 TEST_F(XmppLoginHandlerTest, StartTlsHandshake) {
 165   login_handler_.reset(
 166       new XmppLoginHandler("google.com", kTestUsername, kTestToken,
 167                            XmppLoginHandler::TlsMode::WITH_HANDSHAKE, this));
 168   login_handler_->Start();
 169   EXPECT_FALSE(start_tls_called_);
 170
 171   EXPECT_EQ(sent_data_,
 172             "<stream:stream to=\"google.com\" version=\"1.0\" "
 173             "xmlns=\"jabber:client\" "
 174             "xmlns:stream=\"http://etherx.jabber.org/streams\">"
 175             "<starttls xmlns=\"urn:ietf:params:xml:ns:xmpp-tls\"/>");
 176   sent_data_.clear();
 177
 178   login_handler_->OnDataReceived(
 179       "<stream:stream from=\"google.com\" id=\"78A87C70559EF28A\" "
 180           "version=\"1.0\" "
 181           "xmlns:stream=\"http://etherx.jabber.org/streams\" "
 182           "xmlns=\"jabber:client\">"
 183         "<stream:features>"
 184           "<starttls xmlns=\"urn:ietf:params:xml:ns:xmpp-tls\">"
 185             "<required/>"
 186           "</starttls>"
 187           "<mechanisms xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">"
 188             "<mechanism>X-OAUTH2</mechanism>"
 189             "<mechanism>X-GOOGLE-TOKEN</mechanism>"
 190           "</mechanisms>"
 191         "</stream:features>");
 192
 193   login_handler_->OnDataReceived(
 194       "<proceed xmlns=\"urn:ietf:params:xml:ns:xmpp-tls\"/>");
 195   EXPECT_TRUE(start_tls_called_);
 196 }
 197
 198 TEST_F(XmppLoginHandlerTest, AuthError) {
 199   HandshakeBase();
 200
 201   login_handler_->OnDataReceived(
 202       "<failure xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">"
 203       "<not-authorized/></failure>");
 204   EXPECT_EQ(error_, SignalStrategy::AUTHENTICATION_FAILED);
 205 }
 206
 207 TEST_F(XmppLoginHandlerTest, NoTls) {
 208   login_handler_.reset(
 209       new XmppLoginHandler("google.com", kTestUsername, kTestToken,
 210                            XmppLoginHandler::TlsMode::NO_TLS, this));
 211   login_handler_->Start();
 212
 213   EXPECT_FALSE(start_tls_called_);
 214   std::string cookie;
 215   base::Base64Encode(
 216       std::string("\0", 1) + kTestUsername + std::string("\0", 1) + kTestToken,
 217       &cookie);
 218   EXPECT_EQ(
 219       sent_data_,
 220       "<stream:stream to=\"google.com\" version=\"1.0\" "
 221           "xmlns=\"jabber:client\" "
 222           "xmlns:stream=\"http://etherx.jabber.org/streams\">"
 223       "<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" mechanism=\"X-OAUTH2\" "
 224           "auth:service=\"oauth2\" auth:allow-generated-jid=\"true\" "
 225           "auth:client-uses-full-bind-result=\"true\" "
 226           "auth:allow-non-google-login=\"true\" "
 227           "xmlns:auth=\"http://www.google.com/talk/protocol/auth\">" + cookie +
 228       "</auth>");
 229 }
 230
 231 TEST_F(XmppLoginHandlerTest, StreamParseError) {
 232   HandshakeBase();
 233   delete_login_handler_from_delegate_ = true;
 234   login_handler_->OnDataReceived("BAD DATA");
 235   EXPECT_EQ(error_, SignalStrategy::PROTOCOL_ERROR);
 236 }
 237
 238 // Verify that LoginHandler doesn't crash when destroyed from
 239 // Delegate::SendMessage().
 240 TEST_F(XmppLoginHandlerTest, DeleteInSendMessage) {
 241   login_handler_.reset(
 242       new XmppLoginHandler("google.com", kTestUsername, kTestToken,
 243                            XmppLoginHandler::TlsMode::WITHOUT_HANDSHAKE, this));
 244   login_handler_->Start();
 245   EXPECT_TRUE(start_tls_called_);
 246
 247   delete_login_handler_from_delegate_ = true;
 248   login_handler_->OnTlsStarted();
 249   EXPECT_FALSE(login_handler_);
 250 }
 251
 252 // Verify that LoginHandler doesn't crash when destroyed from
 253 // Delegate::StartTls().
 254 TEST_F(XmppLoginHandlerTest, DeleteInStartTls) {
 255   login_handler_.reset(
 256       new XmppLoginHandler("google.com", kTestUsername, kTestToken,
 257                            XmppLoginHandler::TlsMode::WITHOUT_HANDSHAKE, this));
 258   delete_login_handler_from_delegate_ = true;
 259   login_handler_->Start();
 260   EXPECT_TRUE(start_tls_called_);
 261   EXPECT_FALSE(login_handler_);
 262 }
 263
 264 }  // namespace remoting