blob: 51af81a03961ff500700b3497142fce8c25cf80a [file] [log] [blame]
// Copyright 2004 Google Inc. All Rights Reserved
#include <string>
#include <sstream>
#include <iostream>
#include "talk/base/common.h"
#include "talk/base/cryptstring.h"
#include "talk/base/gunit.h"
#include "talk/xmllite/xmlelement.h"
#include "talk/xmpp/util_unittest.h"
#include "talk/xmpp/constants.h"
#include "talk/xmpp/saslplainmechanism.h"
#include "talk/xmpp/plainsaslhandler.h"
#include "talk/xmpp/xmppengine.h"
using buzz::Jid;
using buzz::QName;
using buzz::XmlElement;
using buzz::XmppEngine;
using buzz::XmppTestHandler;
enum XlttStage {
XLTT_STAGE_CONNECT = 0,
XLTT_STAGE_STREAMSTART,
XLTT_STAGE_TLS_FEATURES,
XLTT_STAGE_TLS_PROCEED,
XLTT_STAGE_ENCRYPTED_START,
XLTT_STAGE_AUTH_FEATURES,
XLTT_STAGE_AUTH_SUCCESS,
XLTT_STAGE_AUTHENTICATED_START,
XLTT_STAGE_BIND_FEATURES,
XLTT_STAGE_BIND_SUCCESS,
XLTT_STAGE_SESSION_SUCCESS,
};
class XmppLoginTaskTest : public testing::Test {
public:
XmppEngine* engine() { return engine_.get(); }
XmppTestHandler* handler() { return handler_.get(); }
virtual void SetUp() {
engine_.reset(XmppEngine::Create());
handler_.reset(new XmppTestHandler(engine_.get()));
Jid jid("david@my-server");
talk_base::InsecureCryptStringImpl pass;
pass.password() = "david";
engine_->SetSessionHandler(handler_.get());
engine_->SetOutputHandler(handler_.get());
engine_->AddStanzaHandler(handler_.get());
engine_->SetUser(jid);
engine_->SetSaslHandler(
new buzz::PlainSaslHandler(jid, talk_base::CryptString(pass), true));
}
virtual void TearDown() {
handler_.reset();
engine_.reset();
}
void RunPartialLogin(XlttStage startstage, XlttStage endstage);
void SetTlsOptions(buzz::TlsOptions option);
private:
talk_base::scoped_ptr<XmppEngine> engine_;
talk_base::scoped_ptr<XmppTestHandler> handler_;
};
void XmppLoginTaskTest::SetTlsOptions(buzz::TlsOptions option) {
engine_->SetTls(option);
}
void XmppLoginTaskTest::RunPartialLogin(XlttStage startstage,
XlttStage endstage) {
std::string input;
switch (startstage) {
case XLTT_STAGE_CONNECT: {
engine_->Connect();
XmlElement appStanza(QName("test", "app-stanza"));
appStanza.AddText("this-is-a-test");
engine_->SendStanza(&appStanza);
EXPECT_EQ("<stream:stream to=\"my-server\" xml:lang=\"*\" "
"version=\"1.0\" xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">\r\n", handler_->OutputActivity());
EXPECT_EQ("[OPENING]", handler_->SessionActivity());
EXPECT_EQ("", handler_->StanzaActivity());
if (endstage == XLTT_STAGE_CONNECT)
return;
}
case XLTT_STAGE_STREAMSTART: {
input = "<stream:stream id=\"a5f2d8c9\" version=\"1.0\" "
"xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">";
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ("", handler_->SessionActivity());
EXPECT_EQ("", handler_->OutputActivity());
if (endstage == XLTT_STAGE_STREAMSTART)
return;
}
case XLTT_STAGE_TLS_FEATURES: {
input = "<stream:features>"
"<starttls xmlns='urn:ietf:params:xml:ns:xmpp-tls'/>"
"</stream:features>";
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<starttls xmlns=\"urn:ietf:params:xml:ns:xmpp-tls\"/>",
handler_->OutputActivity());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ("", handler_->SessionActivity());
if (endstage == XLTT_STAGE_TLS_FEATURES)
return;
}
case XLTT_STAGE_TLS_PROCEED: {
input = std::string("<proceed xmlns='urn:ietf:params:xml:ns:xmpp-tls'/>");
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[START-TLS my-server]"
"<stream:stream to=\"my-server\" xml:lang=\"*\" "
"version=\"1.0\" xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">\r\n", handler_->OutputActivity());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ("", handler_->SessionActivity());
if (endstage == XLTT_STAGE_TLS_PROCEED)
return;
}
case XLTT_STAGE_ENCRYPTED_START: {
input = std::string("<stream:stream id=\"01234567\" version=\"1.0\" "
"xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">");
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ("", handler_->SessionActivity());
EXPECT_EQ("", handler_->OutputActivity());
if (endstage == XLTT_STAGE_ENCRYPTED_START)
return;
}
case XLTT_STAGE_AUTH_FEATURES: {
input = "<stream:features>"
"<mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>"
"<mechanism>DIGEST-MD5</mechanism>"
"<mechanism>PLAIN</mechanism>"
"</mechanisms>"
"</stream:features>";
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" "
"mechanism=\"PLAIN\" "
"auth:allow-non-google-login=\"true\" "
"auth:client-uses-full-bind-result=\"true\" "
"xmlns:auth=\"http://www.google.com/talk/protocol/auth\""
">AGRhdmlkAGRhdmlk</auth>",
handler_->OutputActivity());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ("", handler_->SessionActivity());
if (endstage == XLTT_STAGE_AUTH_FEATURES)
return;
}
case XLTT_STAGE_AUTH_SUCCESS: {
input = "<success xmlns='urn:ietf:params:xml:ns:xmpp-sasl'/>";
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<stream:stream to=\"my-server\" xml:lang=\"*\" "
"version=\"1.0\" xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">\r\n", handler_->OutputActivity());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ("", handler_->SessionActivity());
if (endstage == XLTT_STAGE_AUTH_SUCCESS)
return;
}
case XLTT_STAGE_AUTHENTICATED_START: {
input = std::string("<stream:stream id=\"01234567\" version=\"1.0\" "
"xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">");
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ("", handler_->SessionActivity());
EXPECT_EQ("", handler_->OutputActivity());
if (endstage == XLTT_STAGE_AUTHENTICATED_START)
return;
}
case XLTT_STAGE_BIND_FEATURES: {
input = "<stream:features>"
"<bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'/>"
"<session xmlns='urn:ietf:params:xml:ns:xmpp-session'/>"
"</stream:features>";
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<iq type=\"set\" id=\"0\">"
"<bind xmlns=\"urn:ietf:params:xml:ns:xmpp-bind\"/></iq>",
handler_->OutputActivity());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ("", handler_->SessionActivity());
if (endstage == XLTT_STAGE_BIND_FEATURES)
return;
}
case XLTT_STAGE_BIND_SUCCESS: {
input = "<iq type='result' id='0'>"
"<bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'>"
"<jid>david@my-server/test</jid></bind></iq>";
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<iq type=\"set\" id=\"1\">"
"<session xmlns=\"urn:ietf:params:xml:ns:xmpp-session\"/></iq>",
handler_->OutputActivity());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ("", handler_->SessionActivity());
if (endstage == XLTT_STAGE_BIND_SUCCESS)
return;
}
case XLTT_STAGE_SESSION_SUCCESS: {
input = "<iq type='result' id='1'/>";
engine_->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<test:app-stanza xmlns:test=\"test\">this-is-a-test"
"</test:app-stanza>", handler_->OutputActivity());
EXPECT_EQ("[OPEN]", handler_->SessionActivity());
EXPECT_EQ("", handler_->StanzaActivity());
if (endstage == XLTT_STAGE_SESSION_SUCCESS)
return;
}
}
}
TEST_F(XmppLoginTaskTest, TestUtf8Good) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_CONNECT);
std::string input = "<?xml version='1.0' encoding='UTF-8'?>"
"<stream:stream id=\"a5f2d8c9\" version=\"1.0\" "
"xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestNonUtf8Bad) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_CONNECT);
std::string input = "<?xml version='1.0' encoding='ISO-8859-1'?>"
"<stream:stream id=\"a5f2d8c9\" version=\"1.0\" "
"xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-XML]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestNoFeatures) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_STREAMSTART);
std::string input = "<iq type='get' id='1'/>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-VERSION]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestTlsRequiredNotPresent) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_STREAMSTART);
std::string input = "<stream:features>"
"<mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>"
"<mechanism>DIGEST-MD5</mechanism>"
"<mechanism>PLAIN</mechanism>"
"</mechanisms>"
"</stream:features>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-TLS]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestTlsRequeiredAndPresent) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_STREAMSTART);
std::string input = "<stream:features>"
"<starttls xmlns='urn:ietf:params:xml:ns:xmpp-tls'>"
"<required/>"
"</starttls>"
"<mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>"
"<mechanism>X-GOOGLE-TOKEN</mechanism>"
"<mechanism>PLAIN</mechanism>"
"<mechanism>X-OAUTH2</mechanism>"
"</mechanisms>"
"</stream:features>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<starttls xmlns=\"urn:ietf:params:xml:ns:xmpp-tls\"/>",
handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestTlsEnabledNotPresent) {
SetTlsOptions(buzz::TLS_ENABLED);
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_STREAMSTART);
std::string input = "<stream:features>"
"<mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>"
"<mechanism>DIGEST-MD5</mechanism>"
"<mechanism>PLAIN</mechanism>"
"</mechanisms>"
"</stream:features>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" "
"mechanism=\"PLAIN\" auth:allow-non-google-login=\"true\" "
"auth:client-uses-full-bind-result=\"true\" "
"xmlns:auth=\"http://www.google.com/talk/protocol/auth\""
">AGRhdmlkAGRhdmlk</auth>", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestTlsEnabledAndPresent) {
SetTlsOptions(buzz::TLS_ENABLED);
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_STREAMSTART);
std::string input = "<stream:features>"
"<mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>"
"<mechanism>X-GOOGLE-TOKEN</mechanism>"
"<mechanism>PLAIN</mechanism>"
"<mechanism>X-OAUTH2</mechanism>"
"</mechanisms>"
"</stream:features>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" "
"mechanism=\"PLAIN\" auth:allow-non-google-login=\"true\" "
"auth:client-uses-full-bind-result=\"true\" "
"xmlns:auth=\"http://www.google.com/talk/protocol/auth\""
">AGRhdmlkAGRhdmlk</auth>", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestTlsDisabledNotPresent) {
SetTlsOptions(buzz::TLS_DISABLED);
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_STREAMSTART);
std::string input = "<stream:features>"
"<mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>"
"<mechanism>DIGEST-MD5</mechanism>"
"<mechanism>PLAIN</mechanism>"
"</mechanisms>"
"</stream:features>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" "
"mechanism=\"PLAIN\" auth:allow-non-google-login=\"true\" "
"auth:client-uses-full-bind-result=\"true\" "
"xmlns:auth=\"http://www.google.com/talk/protocol/auth\""
">AGRhdmlkAGRhdmlk</auth>", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestTlsDisabledAndPresent) {
SetTlsOptions(buzz::TLS_DISABLED);
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_STREAMSTART);
std::string input = "<stream:features>"
"<mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>"
"<mechanism>X-GOOGLE-TOKEN</mechanism>"
"<mechanism>PLAIN</mechanism>"
"<mechanism>X-OAUTH2</mechanism>"
"</mechanisms>"
"</stream:features>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" "
"mechanism=\"PLAIN\" auth:allow-non-google-login=\"true\" "
"auth:client-uses-full-bind-result=\"true\" "
"xmlns:auth=\"http://www.google.com/talk/protocol/auth\""
">AGRhdmlkAGRhdmlk</auth>", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestTlsFailure) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_TLS_FEATURES);
std::string input = "<failure xmlns='urn:ietf:params:xml:ns:xmpp-tls'/>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-TLS]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestTlsBadStream) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_TLS_PROCEED);
std::string input = "<wrongtag>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-VERSION]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestMissingSaslPlain) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_ENCRYPTED_START);
std::string input = "<stream:features>"
"<mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>"
"<mechanism>DIGEST-MD5</mechanism>"
"</mechanisms>"
"</stream:features>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-AUTH]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestWrongPassword) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_AUTH_FEATURES);
std::string input = "<failure xmlns='urn:ietf:params:xml:ns:xmpp-sasl'/>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-UNAUTHORIZED]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestAuthBadStream) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_AUTH_SUCCESS);
std::string input = "<wrongtag>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-VERSION]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestMissingBindFeature) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_AUTHENTICATED_START);
std::string input = "<stream:features>"
"<session xmlns='urn:ietf:params:xml:ns:xmpp-session'/>"
"</stream:features>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-BIND]", handler()->SessionActivity());
}
TEST_F(XmppLoginTaskTest, TestMissingSessionFeature) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_AUTHENTICATED_START);
std::string input = "<stream:features>"
"<bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'/>"
"</stream:features>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-BIND]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
/* TODO: Handle this case properly inside XmppLoginTask.
TEST_F(XmppLoginTaskTest, TestBindFailure1) {
// check wrong JID
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_BIND_FEATURES);
std::string input = "<iq type='result' id='0'>"
"<bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'>"
"<jid>davey@my-server/test</jid></bind></iq>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-BIND]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
*/
TEST_F(XmppLoginTaskTest, TestBindFailure2) {
// check missing JID
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_BIND_FEATURES);
std::string input = "<iq type='result' id='0'>"
"<bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'/></iq>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-BIND]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestBindFailure3) {
// check plain failure
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_BIND_FEATURES);
std::string input = "<iq type='error' id='0'/>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-BIND]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppLoginTaskTest, TestBindFailure4) {
// check wrong id to ignore
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_BIND_FEATURES);
std::string input = "<iq type='error' id='1'/>";
engine()->HandleInput(input.c_str(), input.length());
// continue after an ignored iq
RunPartialLogin(XLTT_STAGE_BIND_SUCCESS, XLTT_STAGE_SESSION_SUCCESS);
}
TEST_F(XmppLoginTaskTest, TestSessionFailurePlain1) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_BIND_SUCCESS);
std::string input = "<iq type='error' id='1'/>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-BIND]", handler()->SessionActivity());
}
TEST_F(XmppLoginTaskTest, TestSessionFailurePlain2) {
RunPartialLogin(XLTT_STAGE_CONNECT, XLTT_STAGE_BIND_SUCCESS);
// check reverse iq to ignore
// TODO: consider queueing or passing through?
std::string input = "<iq type='get' id='1'/>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
// continue after an ignored iq
RunPartialLogin(XLTT_STAGE_SESSION_SUCCESS, XLTT_STAGE_SESSION_SUCCESS);
}
TEST_F(XmppLoginTaskTest, TestBadXml) {
int errorKind = 0;
for (XlttStage stage = XLTT_STAGE_CONNECT;
stage <= XLTT_STAGE_SESSION_SUCCESS;
stage = static_cast<XlttStage>(stage + 1)) {
RunPartialLogin(XLTT_STAGE_CONNECT, stage);
std::string input;
switch (errorKind++ % 5) {
case 0: input = "&syntax;"; break;
case 1: input = "<nons:iq/>"; break;
case 2: input = "<iq a='b' a='dupe'/>"; break;
case 3: input = "<>"; break;
case 4: input = "<iq a='<wrong>'>"; break;
}
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-XML]", handler()->SessionActivity());
TearDown();
SetUp();
}
}
TEST_F(XmppLoginTaskTest, TestStreamError) {
for (XlttStage stage = XLTT_STAGE_CONNECT;
stage <= XLTT_STAGE_SESSION_SUCCESS;
stage = static_cast<XlttStage>(stage + 1)) {
switch (stage) {
case XLTT_STAGE_CONNECT:
case XLTT_STAGE_TLS_PROCEED:
case XLTT_STAGE_AUTH_SUCCESS:
continue;
default:
break;
}
RunPartialLogin(XLTT_STAGE_CONNECT, stage);
std::string input = "<stream:error>"
"<xml-not-well-formed xmlns='urn:ietf:params:xml:ns:xmpp-streams'/>"
"<text xml:lang='en' xmlns='urn:ietf:params:xml:ns:xmpp-streams'>"
"Some special application diagnostic information!"
"</text>"
"<escape-your-data xmlns='application-ns'/>"
"</stream:error>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-STREAM]", handler()->SessionActivity());
EXPECT_EQ("<str:error xmlns:str=\"http://etherx.jabber.org/streams\">"
"<xml-not-well-formed xmlns=\"urn:ietf:params:xml:ns:xmpp-streams\"/>"
"<text xml:lang=\"en\" xmlns=\"urn:ietf:params:xml:ns:xmpp-streams\">"
"Some special application diagnostic information!"
"</text>"
"<escape-your-data xmlns=\"application-ns\"/>"
"</str:error>", engine()->GetStreamError()->Str());
TearDown();
SetUp();
}
}