2 files changed, 9 insertions, 3 deletions
diff --git a/Swiften/Session/BasicSessionStream.cpp b/Swiften/Session/BasicSessionStream.cpp
index 807470e..e0fbce7 100644
--- a/Swiften/Session/BasicSessionStream.cpp
+++ b/Swiften/Session/BasicSessionStream.cpp
@@ -110,15 +110,20 @@ void BasicSessionStream::handleTLSConnected() {
 
 void BasicSessionStream::handleTLSError() {
 	available = false;
 	onError(boost::shared_ptr<Error>(new Error(Error::TLSError)));
 }
 
-void BasicSessionStream::handleConnectionError(const boost::optional<Connection::Error>&) {
+void BasicSessionStream::handleConnectionError(const boost::optional<Connection::Error>& error) {
 	available = false;
-	onError(boost::shared_ptr<Error>(new Error(Error::ConnectionError)));
+	if (error == Connection::ReadError) {
+		onError(boost::shared_ptr<Error>(new Error(Error::ConnectionReadError)));
+	}
+	else {
+		onError(boost::shared_ptr<Error>(new Error(Error::ConnectionWriteError)));
+	}
 }
 
 void BasicSessionStream::handleDataRead(const ByteArray& data) {
 	onDataRead(String(data.getData(), data.getSize()));
 }
 
diff --git a/Swiften/Session/SessionStream.h b/Swiften/Session/SessionStream.h
index 6bba237..1252c5a 100644
--- a/Swiften/Session/SessionStream.h
+++ b/Swiften/Session/SessionStream.h
@@ -14,13 +14,14 @@ namespace Swift {
 			class Error : public Swift::Error {
 				public:
 					enum Type {
 						ParseError,
 						TLSError,
 						InvalidTLSCertificateError,
-						ConnectionError
+						ConnectionReadError,
+						ConnectionWriteError
 					};
 
 					Error(Type type) : type(type) {}
 
 					Type type;
 			};