1 files changed, 17 insertions, 1 deletions
diff --git a/Swiften/Network/BOSHConnectionPool.h b/Swiften/Network/BOSHConnectionPool.h
index eaef56c..df116ff 100644
--- a/Swiften/Network/BOSHConnectionPool.h
+++ b/Swiften/Network/BOSHConnectionPool.h
@@ -9,30 +9,41 @@
 
 #include <vector>
 
 #include <Swiften/Base/API.h>
 #include <Swiften/Base/SafeString.h>
 #include <Swiften/Network/BOSHConnection.h>
+#include <Swiften/TLS/CertificateWithKey.h>
 #include <Swiften/TLS/TLSOptions.h>
 
 namespace Swift {
 	class CachingDomainNameResolver;
 	class EventLoop;
 	class HTTPConnectProxiedConnectionFactory;
 	class HTTPTrafficFilter;
-	class TLSConnectionFactory;
+	class TLSContextFactory;
+	class CachingDomainNameResolver;
+	class EventLoop;
 
 	class SWIFTEN_API BOSHConnectionPool : public boost::bsignals::trackable {
 		public:
 			BOSHConnectionPool(const URL& boshURL, DomainNameResolver* resolver, ConnectionFactory* connectionFactory, XMLParserFactory* parserFactory, TLSContextFactory* tlsFactory, TimerFactory* timerFactory, EventLoop* eventLoop, const std::string& to, unsigned long long initialRID, const URL& boshHTTPConnectProxyURL, const SafeString& boshHTTPConnectProxyAuthID, const SafeString& boshHTTPConnectProxyAuthPassword, const TLSOptions& tlsOptions, boost::shared_ptr<HTTPTrafficFilter> trafficFilter = boost::shared_ptr<HTTPTrafficFilter>());
 			~BOSHConnectionPool();
+
+			void open();
 			void write(const SafeByteArray& data);
 			void writeFooter();
 			void close();
 			void restartStream();
 
+			void setTLSCertificate(CertificateWithKey::ref certWithKey);
+			bool isTLSEncrypted() const;
+			Certificate::ref getPeerCertificate() const;
+			std::vector<Certificate::ref> getPeerCertificateChain() const;
+			boost::shared_ptr<CertificateVerificationError> getPeerCertificateVerificationError() const;
+
 			boost::signal<void (BOSHError::ref)> onSessionTerminated;
 			boost::signal<void ()> onSessionStarted;
 			boost::signal<void (const SafeByteArray&)> onXMPPDataRead;
 			boost::signal<void (const SafeByteArray&)> onBOSHDataRead;
 			boost::signal<void (const SafeByteArray&)> onBOSHDataWritten;
 
@@ -65,8 +76,13 @@ namespace Swift {
 			std::string to;
 			size_t requestLimit;
 			int restartCount;
 			bool pendingRestart;
 			std::vector<ConnectionFactory*> myConnectionFactories;
 			CachingDomainNameResolver* resolver;
+			CertificateWithKey::ref clientCertificate;
+			TLSContextFactory* tlsContextFactory_;
+			TLSOptions tlsOptions_;
+			std::vector<boost::shared_ptr<Certificate> > pinnedCertificateChain_;
+			boost::shared_ptr<CertificateVerificationError> lastVerificationError_;
 	};
 }