1 files changed, 11 insertions, 3 deletions
diff --git a/Swiften/FileTransfer/FileWriteBytestream.cpp b/Swiften/FileTransfer/FileWriteBytestream.cpp
index 4d29bd1..6a22c6a 100644
--- a/Swiften/FileTransfer/FileWriteBytestream.cpp
+++ b/Swiften/FileTransfer/FileWriteBytestream.cpp
@@ -7,7 +7,7 @@
 #include <boost/filesystem/fstream.hpp>
 #include <cassert>
 
-#include "Swiften/FileTransfer/FileWriteBytestream.h"
+#include <Swiften/FileTransfer/FileWriteBytestream.h>
 
 namespace Swift {
 
@@ -21,12 +21,20 @@ FileWriteBytestream::~FileWriteBytestream() {
 	}
 }
 
-void FileWriteBytestream::write(const ByteArray& data) {
+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.getData()), data.getSize());
+	stream->write(reinterpret_cast<const char*>(&data[0]), data.size());
+	onWrite(data);
+}
+
+void FileWriteBytestream::close() {
+	if (stream) {
+		stream->close();
+		stream = NULL;
+	}
 }
 
 }