1 files changed, 69 insertions, 0 deletions
diff --git a/Swiften/Session/SessionStream.h b/Swiften/Session/SessionStream.h
new file mode 100644
index 0000000..8c64ccf
--- /dev/null
+++ b/Swiften/Session/SessionStream.h
@@ -0,0 +1,69 @@
+#pragma once
+
+#include <boost/signal.hpp>
+#include <boost/shared_ptr.hpp>
+
+#include "Swiften/Elements/ProtocolHeader.h"
+#include "Swiften/Elements/Element.h"
+#include "Swiften/Base/Error.h"
+#include "Swiften/TLS/PKCS12Certificate.h"
+
+namespace Swift {
+	class SessionStream {
+		public:
+			class Error : public Swift::Error {
+				public:
+					enum Type {
+						ParseError,
+						TLSError,
+						InvalidTLSCertificateError,
+						ConnectionReadError,
+						ConnectionWriteError
+					};
+
+					Error(Type type) : type(type) {}
+
+					Type type;
+			};
+
+			virtual ~SessionStream();
+
+			virtual bool isAvailable() = 0;
+
+			virtual void writeHeader(const ProtocolHeader& header) = 0;
+			virtual void writeFooter() = 0;
+			virtual void writeElement(boost::shared_ptr<Element>) = 0;
+
+			virtual void addZLibCompression() = 0;
+
+			virtual bool supportsTLSEncryption() = 0;
+			virtual void addTLSEncryption() = 0;
+			virtual void setWhitespacePingEnabled(bool enabled) = 0;
+
+			virtual void resetXMPPParser() = 0;
+
+			void setTLSCertificate(const PKCS12Certificate& cert) {
+				certificate = cert;
+			}
+
+			virtual bool hasTLSCertificate() {
+				return !certificate.isNull();
+			}
+
+
+			boost::signal<void (const ProtocolHeader&)> onStreamStartReceived;
+			boost::signal<void (boost::shared_ptr<Element>)> onElementReceived;
+			boost::signal<void (boost::shared_ptr<Error>)> onError;
+			boost::signal<void ()> onTLSEncrypted;
+			boost::signal<void (const String&)> onDataRead;
+			boost::signal<void (const String&)> onDataWritten;
+
+		protected:
+			const PKCS12Certificate& getTLSCertificate() const {
+				return certificate;
+			}
+
+		private:
+			PKCS12Certificate certificate;
+	};
+}