1 files changed, 8 insertions, 0 deletions
diff --git a/Swiften/FileTransfer/FileWriteBytestream.cpp b/Swiften/FileTransfer/FileWriteBytestream.cpp
index c38338a..6a22c6a 100644
--- a/Swiften/FileTransfer/FileWriteBytestream.cpp
+++ b/Swiften/FileTransfer/FileWriteBytestream.cpp
@@ -21,12 +21,20 @@ FileWriteBytestream::~FileWriteBytestream() {
 	}
 }
 
 void FileWriteBytestream::write(const std::vector<unsigned char>& data) {
 	if (!stream) {
 		stream = new boost::filesystem::ofstream(file, std::ios_base::out|std::ios_base::binary);
 	}
 	assert(stream->good());
 	stream->write(reinterpret_cast<const char*>(&data[0]), data.size());
+	onWrite(data);
+}
+
+void FileWriteBytestream::close() {
+	if (stream) {
+		stream->close();
+		stream = NULL;
+	}
 }
 
 }