1 files changed, 85 insertions, 79 deletions
diff --git a/Swiften/Network/BoostNetworkFactories.h b/Swiften/Network/BoostNetworkFactories.h
index 9c3bab1..33a3584 100644
--- a/Swiften/Network/BoostNetworkFactories.h
+++ b/Swiften/Network/BoostNetworkFactories.h
@@ -1,89 +1,95 @@
 /*
- * Copyright (c) 2010-2013 Remko Tronçon
- * Licensed under the GNU General Public License v3.
- * See Documentation/Licenses/GPLv3.txt for more information.
+ * Copyright (c) 2010-2019 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
  */
 
 #pragma once
 
+#include <memory>
+
 #include <Swiften/Base/API.h>
-#include <Swiften/Base/Override.h>
-#include <Swiften/Network/NetworkFactories.h>
 #include <Swiften/Network/BoostIOServiceThread.h>
+#include <Swiften/Network/NetworkFactories.h>
 
 namespace Swift {
-	class EventLoop;
-	class NATTraverser;
-	class PlatformTLSFactories;
-
-	class SWIFTEN_API BoostNetworkFactories : public NetworkFactories {
-		public:
-			BoostNetworkFactories(EventLoop* eventLoop);
-			~BoostNetworkFactories();
-
-			virtual TimerFactory* getTimerFactory() const SWIFTEN_OVERRIDE {
-				return timerFactory;
-			}
-
-			virtual ConnectionFactory* getConnectionFactory() const SWIFTEN_OVERRIDE {
-				return connectionFactory;
-			}
-
-			BoostIOServiceThread* getIOServiceThread() {
-				return &ioServiceThread;
-			}
-
-			DomainNameResolver* getDomainNameResolver() const SWIFTEN_OVERRIDE {
-				return domainNameResolver;
-			}
-
-			ConnectionServerFactory* getConnectionServerFactory() const SWIFTEN_OVERRIDE {
-				return connectionServerFactory;
-			}
-
-			NetworkEnvironment* getNetworkEnvironment() const SWIFTEN_OVERRIDE {
-				return networkEnvironment;
-			}
-
-			NATTraverser* getNATTraverser() const SWIFTEN_OVERRIDE {
-				return natTraverser;
-			}
-
-			virtual XMLParserFactory* getXMLParserFactory() const SWIFTEN_OVERRIDE {
-				return xmlParserFactory;
-			}
-
-			virtual TLSContextFactory* getTLSContextFactory() const SWIFTEN_OVERRIDE;
-
-			virtual ProxyProvider* getProxyProvider() const SWIFTEN_OVERRIDE {
-				return proxyProvider;
-			}
-
-			virtual EventLoop* getEventLoop() const SWIFTEN_OVERRIDE {
-				return eventLoop;
-			}
-
-			virtual IDNConverter* getIDNConverter() const SWIFTEN_OVERRIDE {
-				return idnConverter;
-			}
-
-			virtual CryptoProvider* getCryptoProvider() const SWIFTEN_OVERRIDE {
-				return cryptoProvider;
-			}
-
-		private:
-			BoostIOServiceThread ioServiceThread;
-			TimerFactory* timerFactory;
-			ConnectionFactory* connectionFactory;
-			DomainNameResolver* domainNameResolver;
-			ConnectionServerFactory* connectionServerFactory;
-			NATTraverser* natTraverser;
-			NetworkEnvironment* networkEnvironment;
-			XMLParserFactory* xmlParserFactory;
-			PlatformTLSFactories* tlsFactories;
-			ProxyProvider* proxyProvider;
-			EventLoop* eventLoop;
-			IDNConverter* idnConverter;
-			CryptoProvider* cryptoProvider;
-	};
+    class EventLoop;
+    class NATTraverser;
+    class PlatformTLSFactories;
+
+    class SWIFTEN_API BoostNetworkFactories : public NetworkFactories {
+        public:
+            /**
+             * Construct the network factories, using the provided EventLoop.
+             * @param ioService If this optional parameter is provided, it will be
+             * used for the construction of the BoostIOServiceThread.
+             */
+            BoostNetworkFactories(EventLoop* eventLoop, std::shared_ptr<boost::asio::io_service> ioService = std::shared_ptr<boost::asio::io_service>());
+            virtual ~BoostNetworkFactories() override;
+
+            virtual TimerFactory* getTimerFactory() const override {
+                return timerFactory;
+            }
+
+            virtual ConnectionFactory* getConnectionFactory() const override {
+                return connectionFactory;
+            }
+
+            BoostIOServiceThread* getIOServiceThread() {
+                return &ioServiceThread;
+            }
+
+            DomainNameResolver* getDomainNameResolver() const override {
+                return domainNameResolver;
+            }
+
+            ConnectionServerFactory* getConnectionServerFactory() const override {
+                return connectionServerFactory;
+            }
+
+            NetworkEnvironment* getNetworkEnvironment() const override {
+                return networkEnvironment;
+            }
+
+            NATTraverser* getNATTraverser() const override {
+                return natTraverser;
+            }
+
+            virtual XMLParserFactory* getXMLParserFactory() const override {
+                return xmlParserFactory;
+            }
+
+            virtual TLSContextFactory* getTLSContextFactory() const override;
+
+            virtual ProxyProvider* getProxyProvider() const override {
+                return proxyProvider;
+            }
+
+            virtual EventLoop* getEventLoop() const override {
+                return eventLoop;
+            }
+
+            virtual IDNConverter* getIDNConverter() const override {
+                return idnConverter.get();
+            }
+
+            virtual CryptoProvider* getCryptoProvider() const override {
+                return cryptoProvider;
+            }
+
+        private:
+            BoostIOServiceThread ioServiceThread;
+            TimerFactory* timerFactory;
+            ConnectionFactory* connectionFactory;
+            DomainNameResolver* domainNameResolver;
+            ConnectionServerFactory* connectionServerFactory;
+            NATTraverser* natTraverser;
+            NetworkEnvironment* networkEnvironment;
+            XMLParserFactory* xmlParserFactory;
+            PlatformTLSFactories* tlsFactories;
+            ProxyProvider* proxyProvider;
+            EventLoop* eventLoop;
+            std::unique_ptr<IDNConverter> idnConverter;
+            CryptoProvider* cryptoProvider;
+    };
 }