1 files changed, 37 insertions, 34 deletions
diff --git a/Swiften/Client/ClientSessionStanzaChannel.h b/Swiften/Client/ClientSessionStanzaChannel.h
index 2743a16..c4ee393 100644
--- a/Swiften/Client/ClientSessionStanzaChannel.h
+++ b/Swiften/Client/ClientSessionStanzaChannel.h
@@ -1,49 +1,52 @@
 /*
- * Copyright (c) 2010 Remko Tronçon
- * Licensed under the GNU General Public License v3.
- * See Documentation/Licenses/GPLv3.txt for more information.
+ * Copyright (c) 2010-2016 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
  */
 
 #pragma once
 
-#include <boost/shared_ptr.hpp>
+#include <memory>
 
+#include <Swiften/Base/API.h>
 #include <Swiften/Base/IDGenerator.h>
 #include <Swiften/Client/ClientSession.h>
 #include <Swiften/Client/StanzaChannel.h>
-#include <Swiften/Elements/Message.h>
 #include <Swiften/Elements/IQ.h>
+#include <Swiften/Elements/Message.h>
 #include <Swiften/Elements/Presence.h>
 
 namespace Swift {
-	/**
-	 * StanzaChannel implementation around a ClientSession.
-	 */
-	class ClientSessionStanzaChannel : public StanzaChannel {
-		public:
-			void setSession(boost::shared_ptr<ClientSession> session);
-
-			void sendIQ(boost::shared_ptr<IQ> iq);
-			void sendMessage(boost::shared_ptr<Message> message);
-			void sendPresence(boost::shared_ptr<Presence> presence);
-			bool getStreamManagementEnabled() const;
-			virtual std::vector<Certificate::ref> getPeerCertificateChain() const;
-
-			bool isAvailable() const {
-				return session && session->getState() == ClientSession::Initialized;
-			}
-
-		private:
-			std::string getNewIQID();
-			void send(boost::shared_ptr<Stanza> stanza);
-			void handleSessionFinished(boost::shared_ptr<Error> error);
-			void handleStanza(boost::shared_ptr<Stanza> stanza);
-			void handleStanzaAcked(boost::shared_ptr<Stanza> stanza);
-			void handleSessionInitialized();
-
-		private:
-			IDGenerator idGenerator;
-			boost::shared_ptr<ClientSession> session;
-	};
+    /**
+     * StanzaChannel implementation around a ClientSession.
+     */
+    class SWIFTEN_API ClientSessionStanzaChannel : public StanzaChannel {
+        public:
+            virtual ~ClientSessionStanzaChannel();
+
+            void setSession(std::shared_ptr<ClientSession> session);
+
+            void sendIQ(std::shared_ptr<IQ> iq);
+            void sendMessage(std::shared_ptr<Message> message);
+            void sendPresence(std::shared_ptr<Presence> presence);
+            bool getStreamManagementEnabled() const;
+            virtual std::vector<Certificate::ref> getPeerCertificateChain() const;
+
+            bool isAvailable() const {
+                return session && session->getState() == ClientSession::State::Initialized;
+            }
+
+        private:
+            std::string getNewIQID();
+            void send(std::shared_ptr<Stanza> stanza);
+            void handleSessionFinished(std::shared_ptr<Error> error);
+            void handleStanza(std::shared_ptr<Stanza> stanza);
+            void handleStanzaAcked(std::shared_ptr<Stanza> stanza);
+            void handleSessionInitialized();
+
+        private:
+            IDGenerator idGenerator;
+            std::shared_ptr<ClientSession> session;
+    };
 
 }