1 files changed, 2 insertions, 2 deletions
diff --git a/Swiften/Component/ComponentSession.cpp b/Swiften/Component/ComponentSession.cpp
index 7925b23..826451d 100644
--- a/Swiften/Component/ComponentSession.cpp
+++ b/Swiften/Component/ComponentSession.cpp
@@ -1,87 +1,87 @@
 /*
- * Copyright (c) 2010-2013 Remko Tronçon
+ * Copyright (c) 2010-2014 Remko Tronçon
  * Licensed under the GNU General Public License v3.
  * See Documentation/Licenses/GPLv3.txt for more information.
  */
 
 #include <Swiften/Component/ComponentSession.h>
 
 #include <boost/bind.hpp>
 #include <boost/smart_ptr/make_shared.hpp>
 
 #include <Swiften/Elements/ProtocolHeader.h>
 #include <Swiften/Elements/StreamFeatures.h>
 #include <Swiften/Elements/ComponentHandshake.h>
 #include <Swiften/Session/SessionStream.h>
 #include <Swiften/Component/ComponentHandshakeGenerator.h>
 
 namespace Swift {
 
 ComponentSession::ComponentSession(const JID& jid, const std::string& secret, boost::shared_ptr<SessionStream> stream, CryptoProvider* crypto) : jid(jid), secret(secret), stream(stream), crypto(crypto), state(Initial) {
 }
 
 ComponentSession::~ComponentSession() {
 }
 
 void ComponentSession::start() {
 	stream->onStreamStartReceived.connect(boost::bind(&ComponentSession::handleStreamStart, shared_from_this(), _1));
 	stream->onElementReceived.connect(boost::bind(&ComponentSession::handleElement, shared_from_this(), _1));
 	stream->onClosed.connect(boost::bind(&ComponentSession::handleStreamClosed, shared_from_this(), _1));
 
 	assert(state == Initial);
 	state = WaitingForStreamStart;
 	sendStreamHeader();
 }
 
 void ComponentSession::sendStreamHeader() {
 	ProtocolHeader header;
 	header.setTo(jid);
 	stream->writeHeader(header);
 }
 
 void ComponentSession::sendStanza(boost::shared_ptr<Stanza> stanza) {
 	stream->writeElement(stanza);
 }
 
 void ComponentSession::handleStreamStart(const ProtocolHeader& header) {
 	checkState(WaitingForStreamStart);
 	state = Authenticating;
 	stream->writeElement(ComponentHandshake::ref(new ComponentHandshake(ComponentHandshakeGenerator::getHandshake(header.getID(), secret, crypto))));
 }
 
-void ComponentSession::handleElement(boost::shared_ptr<Element> element) {
+void ComponentSession::handleElement(boost::shared_ptr<ToplevelElement> element) {
 	if (boost::shared_ptr<Stanza> stanza = boost::dynamic_pointer_cast<Stanza>(element)) {
 		if (getState() == Initialized) {
 			onStanzaReceived(stanza);
 		}
 		else {
 			finishSession(Error::UnexpectedElementError);
 		}
 	}
 	else if (boost::dynamic_pointer_cast<ComponentHandshake>(element)) {
 		if (!checkState(Authenticating)) {
 			return;
 		}
 		stream->setWhitespacePingEnabled(true);
 		state = Initialized;
 		onInitialized();
 	}
 	else if (getState() == Authenticating) {
 		if (boost::dynamic_pointer_cast<StreamFeatures>(element)) {
 			// M-Link sends stream features, so swallow that.
 		}
 		else {
 			// FIXME: We should actually check the element received
 			finishSession(Error::AuthenticationFailedError);
 		}
 	}
 	else {
 		finishSession(Error::UnexpectedElementError);
 	}
 }
 
 bool ComponentSession::checkState(State state) {
 	if (this->state != state) {
 		finishSession(Error::UnexpectedElementError);
 		return false;
 	}