1 files changed, 41 insertions, 35 deletions
diff --git a/Swiften/Network/TLSConnection.h b/Swiften/Network/TLSConnection.h
index 60f73ea..1ab1ec6 100644
--- a/Swiften/Network/TLSConnection.h
+++ b/Swiften/Network/TLSConnection.h
@@ -1,46 +1,52 @@
 /*
- * Copyright (c) 2011 Kevin Smith
- * Licensed under the GNU General Public License v3.
- * See Documentation/Licenses/GPLv3.txt for more information.
+ * Copyright (c) 2011-2018 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
  */
 
 #pragma once
 
-#include <boost/shared_ptr.hpp>
-#include <boost/enable_shared_from_this.hpp>
-#include <Swiften/Base/boost_bsignals.h>
+#include <memory>
 
+#include <boost/signals2.hpp>
+
+#include <Swiften/Base/API.h>
 #include <Swiften/Base/SafeByteArray.h>
 #include <Swiften/Network/Connection.h>
+#include <Swiften/TLS/TLSOptions.h>
 
 namespace Swift {
-	class HostAddressPort;
-	class TLSContextFactory;
-	class TLSContext;
-
-	class TLSConnection : public Connection {
-		public:
-
-			TLSConnection(Connection::ref connection, TLSContextFactory* tlsFactory);
-			virtual ~TLSConnection();
-
-			virtual void listen() {assert(false);}
-			virtual void connect(const HostAddressPort& address);
-			virtual void disconnect();
-			virtual void write(const SafeByteArray& data);
-
-			virtual HostAddressPort getLocalAddress() const;
-
-		private:
-			void handleRawConnectFinished(bool error);
-			void handleRawDisconnected(const boost::optional<Error>& error);
-			void handleRawDataRead(boost::shared_ptr<SafeByteArray> data);
-			void handleRawDataWritten();
-			void handleTLSConnectFinished(bool error);
-			void handleTLSDataForNetwork(const SafeByteArray& data);
-			void handleTLSDataForApplication(const SafeByteArray& data);
-		private:
-			TLSContext* context;
-			Connection::ref connection;
-	};
+    class HostAddressPort;
+    class TLSContextFactory;
+    class TLSContext;
+
+    class SWIFTEN_API TLSConnection : public Connection {
+        public:
+
+            TLSConnection(Connection::ref connection, TLSContextFactory* tlsFactory, const TLSOptions&);
+            virtual ~TLSConnection();
+
+            virtual void listen() {assert(false);}
+            virtual void connect(const HostAddressPort& address);
+            virtual void disconnect();
+            virtual void write(const SafeByteArray& data);
+
+            virtual HostAddressPort getLocalAddress() const;
+            virtual HostAddressPort getRemoteAddress() const;
+
+            TLSContext* getTLSContext() const;
+
+        private:
+            void handleRawConnectFinished(bool error);
+            void handleRawDisconnected(const boost::optional<Error>& error);
+            void handleRawDataRead(std::shared_ptr<SafeByteArray> data);
+            void handleRawDataWritten();
+            void handleTLSConnectFinished(bool error);
+            void handleTLSDataForNetwork(const SafeByteArray& data);
+            void handleTLSDataForApplication(const SafeByteArray& data);
+
+        private:
+            std::unique_ptr<TLSContext> context;
+            Connection::ref connection;
+    };
 }