blob: 46b79c6d31c497cb11c9b7205950c32141b31c3b [file] [log] [blame]
// Copyright 2004 Google Inc. All Rights Reserved
// Author: David Bau
#include <string>
#include <sstream>
#include <iostream>
#include "talk/base/common.h"
#include "talk/base/gunit.h"
#include "talk/xmllite/xmlelement.h"
#include "talk/xmpp/constants.h"
#include "talk/xmpp/util_unittest.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::XmppIqCookie;
using buzz::XmppIqHandler;
using buzz::XmppTestHandler;
using buzz::QN_ID;
using buzz::QN_IQ;
using buzz::QN_TYPE;
using buzz::QN_ROSTER_QUERY;
using buzz::XMPP_RETURN_OK;
using buzz::XMPP_RETURN_BADARGUMENT;
// XmppEngineTestIqHandler
// This class grabs the response to an IQ stanza and stores it in a string.
class XmppEngineTestIqHandler : public XmppIqHandler {
public:
virtual void IqResponse(XmppIqCookie, const XmlElement * stanza) {
ss_ << stanza->Str();
}
std::string IqResponseActivity() {
std::string result = ss_.str();
ss_.str("");
return result;
}
private:
std::stringstream ss_;
};
class XmppEngineTest : 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 RunLogin();
private:
talk_base::scoped_ptr<XmppEngine> engine_;
talk_base::scoped_ptr<XmppTestHandler> handler_;
};
void XmppEngineTest::RunLogin() {
// Connect
EXPECT_EQ(XmppEngine::STATE_START, engine()->GetState());
engine()->Connect();
EXPECT_EQ(XmppEngine::STATE_OPENING, engine()->GetState());
EXPECT_EQ("[OPENING]", handler_->SessionActivity());
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());
std::string 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());
input =
"<stream:features>"
"<starttls xmlns='urn:ietf:params:xml:ns:xmpp-tls'>"
"<required/>"
"</starttls>"
"<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("<starttls xmlns=\"urn:ietf:params:xml:ns:xmpp-tls\"/>",
handler_->OutputActivity());
EXPECT_EQ("", handler_->SessionActivity());
EXPECT_EQ("", handler_->StanzaActivity());
input = "<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_->SessionActivity());
EXPECT_EQ("", handler_->StanzaActivity());
input = "<stream:stream id=\"01234567\" version=\"1.0\" "
"xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">";
engine()->HandleInput(input.c_str(), input.length());
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());
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_->SessionActivity());
EXPECT_EQ("", handler_->StanzaActivity());
input = "<stream:stream id=\"01234567\" version=\"1.0\" "
"xmlns:stream=\"http://etherx.jabber.org/streams\" "
"xmlns=\"jabber:client\">";
engine()->HandleInput(input.c_str(), input.length());
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_->SessionActivity());
EXPECT_EQ("", handler_->StanzaActivity());
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_->SessionActivity());
EXPECT_EQ("", handler_->StanzaActivity());
input = "<iq type='result' id='1'/>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[OPEN]", handler_->SessionActivity());
EXPECT_EQ("", handler_->StanzaActivity());
EXPECT_EQ(Jid("david@my-server/test"), engine()->FullJid());
}
// TestSuccessfulLogin()
// This function simply tests to see if a login works. This includes
// encryption and authentication
TEST_F(XmppEngineTest, TestSuccessfulLoginAndDisconnect) {
RunLogin();
engine()->Disconnect();
EXPECT_EQ("</stream:stream>[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
TEST_F(XmppEngineTest, TestSuccessfulLoginAndConnectionClosed) {
RunLogin();
engine()->ConnectionClosed(0);
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[CLOSED][ERROR-CONNECTION-CLOSED]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
// TestNotXmpp()
// This tests the error case when connecting to a non XMPP service
TEST_F(XmppEngineTest, TestNotXmpp) {
// Connect
engine()->Connect();
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());
// Send garbage response (courtesy of apache)
std::string input = "<!DOCTYPE HTML PUBLIC \"-//IETF//DTD HTML 2.0//EN\">";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("[OPENING][CLOSED][ERROR-XML]", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
// TestPassthrough()
// This tests that arbitrary stanzas can be passed to the server through
// the engine.
TEST_F(XmppEngineTest, TestPassthrough) {
// Queue up an app stanza
XmlElement application_stanza(QName("test", "app-stanza"));
application_stanza.AddText("this-is-a-test");
engine()->SendStanza(&application_stanza);
// Do the whole login handshake
RunLogin();
EXPECT_EQ("<test:app-stanza xmlns:test=\"test\">this-is-a-test"
"</test:app-stanza>", handler()->OutputActivity());
// do another stanza
XmlElement roster_get(QN_IQ);
roster_get.AddAttr(QN_TYPE, "get");
roster_get.AddAttr(QN_ID, engine()->NextId());
roster_get.AddElement(new XmlElement(QN_ROSTER_QUERY, true));
engine()->SendStanza(&roster_get);
EXPECT_EQ("<iq type=\"get\" id=\"2\"><query xmlns=\"jabber:iq:roster\"/>"
"</iq>", handler()->OutputActivity());
// now say the server ends the stream
engine()->HandleInput("</stream:stream>", 16);
EXPECT_EQ("[CLOSED][ERROR-DOCUMENT-CLOSED]", handler()->SessionActivity());
EXPECT_EQ("[CLOSED]", handler()->OutputActivity());
EXPECT_EQ("", handler()->StanzaActivity());
}
// TestIqCallback()
// This tests the routing of Iq stanzas and responses.
TEST_F(XmppEngineTest, TestIqCallback) {
XmppEngineTestIqHandler iq_response;
XmppIqCookie cookie;
// Do the whole login handshake
RunLogin();
// Build an iq request
XmlElement roster_get(QN_IQ);
roster_get.AddAttr(QN_TYPE, "get");
roster_get.AddAttr(QN_ID, engine()->NextId());
roster_get.AddElement(new XmlElement(QN_ROSTER_QUERY, true));
engine()->SendIq(&roster_get, &iq_response, &cookie);
EXPECT_EQ("<iq type=\"get\" id=\"2\"><query xmlns=\"jabber:iq:roster\"/>"
"</iq>", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
EXPECT_EQ("", iq_response.IqResponseActivity());
// now say the server responds to the iq
std::string input = "<iq type='result' id='2'>"
"<query xmlns='jabber:iq:roster'><item>foo</item>"
"</query></iq>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
EXPECT_EQ("<cli:iq type=\"result\" id=\"2\" xmlns:cli=\"jabber:client\">"
"<query xmlns=\"jabber:iq:roster\"><item>foo</item></query>"
"</cli:iq>", iq_response.IqResponseActivity());
EXPECT_EQ(XMPP_RETURN_BADARGUMENT, engine()->RemoveIqHandler(cookie, NULL));
// Do it again with another id to test cancel
roster_get.SetAttr(QN_ID, engine()->NextId());
engine()->SendIq(&roster_get, &iq_response, &cookie);
EXPECT_EQ("<iq type=\"get\" id=\"3\"><query xmlns=\"jabber:iq:roster\"/>"
"</iq>", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
EXPECT_EQ("", handler()->StanzaActivity());
EXPECT_EQ("", iq_response.IqResponseActivity());
// cancel the handler this time
EXPECT_EQ(XMPP_RETURN_OK, engine()->RemoveIqHandler(cookie, NULL));
// now say the server responds to the iq: the iq handler should not get it.
input = "<iq type='result' id='3'><query xmlns='jabber:iq:roster'><item>bar"
"</item></query></iq>";
engine()->HandleInput(input.c_str(), input.length());
EXPECT_EQ("<cli:iq type=\"result\" id=\"3\" xmlns:cli=\"jabber:client\">"
"<query xmlns=\"jabber:iq:roster\"><item>bar</item></query>"
"</cli:iq>", handler()->StanzaActivity());
EXPECT_EQ("", iq_response.IqResponseActivity());
EXPECT_EQ("", handler()->OutputActivity());
EXPECT_EQ("", handler()->SessionActivity());
}