1 files changed, 10 insertions, 2 deletions
diff --git a/Swiften/Session/BasicSessionStream.h b/Swiften/Session/BasicSessionStream.h
index bf92bbb..d248ebc 100644
--- a/Swiften/Session/BasicSessionStream.h
+++ b/Swiften/Session/BasicSessionStream.h
@@ -1,6 +1,7 @@
 #pragma once
 
 #include <boost/shared_ptr.hpp>
+#include <boost/enable_shared_from_this.hpp>
 
 #include "Swiften/Network/Connection.h"
 #include "Swiften/Session/SessionStream.h"
@@ -17,7 +18,7 @@ namespace Swift {
 
   class BasicSessionStream : 
       public SessionStream, 
-      public boost::BOOST_SIGNALS_NAMESPACE::trackable {
+      public boost::enable_shared_from_this<BasicSessionStream> {
     public:
       BasicSessionStream(
 		    boost::shared_ptr<Connection> connection,
@@ -27,6 +28,8 @@ namespace Swift {
       );
       ~BasicSessionStream();
 
+			void initialize();
+
 			virtual void writeHeader(const ProtocolHeader& header);
 			virtual void writeElement(boost::shared_ptr<Element>);
 
@@ -40,12 +43,17 @@ namespace Swift {
     private:
       void handleXMPPError();
       void handleTLSError();
+			void handleStreamStartReceived(const ProtocolHeader&);
+			void handleElementReceived(boost::shared_ptr<Element>);
 
     private:
+			boost::shared_ptr<Connection> connection;
+			PayloadParserFactoryCollection* payloadParserFactories;
+			PayloadSerializerCollection* payloadSerializers;
+			TLSLayerFactory* tlsLayerFactory;
 			boost::shared_ptr<XMPPLayer> xmppLayer;
 			boost::shared_ptr<ConnectionLayer> connectionLayer;
 			StreamStack* streamStack;
-      TLSLayerFactory* tlsLayerFactory;
       boost::shared_ptr<TLSLayer> tlsLayer;
       boost::shared_ptr<WhitespacePingLayer> whitespacePingLayer;
   };