1 files changed, 13 insertions, 2 deletions
diff --git a/Swiften/FileTransfer/ByteArrayReadBytestream.h b/Swiften/FileTransfer/ByteArrayReadBytestream.h
index a6945c3..6cbdef0 100644
--- a/Swiften/FileTransfer/ByteArrayReadBytestream.h
+++ b/Swiften/FileTransfer/ByteArrayReadBytestream.h
@@ -2,38 +2,49 @@
  * Copyright (c) 2010 Remko Tronçon
  * Licensed under the GNU General Public License v3.
  * See Documentation/Licenses/GPLv3.txt for more information.
  */
 
 #pragma once
 
 #include <vector>
 
+#include <Swiften/Base/Algorithm.h>
 #include <Swiften/FileTransfer/ReadBytestream.h>
 
 namespace Swift {
 	class ByteArrayReadBytestream : public ReadBytestream {
 		public:
-			ByteArrayReadBytestream(const std::vector<unsigned char>& data) : data(data), position(0) {
+			ByteArrayReadBytestream(const std::vector<unsigned char>& data) : data(data), position(0), dataComplete(true) {
 			}
 
 			virtual std::vector<unsigned char> read(size_t size) {
 				size_t readSize = size;
 				if (position + readSize > data.size()) {
 					readSize = data.size() - position;
 				}
 				std::vector<unsigned char> result(data.begin() + position, data.begin() + position + readSize);
 
 				onRead(result);
 				position += readSize;
 				return result;
 			}
 
 			virtual bool isFinished() const {
-				return position >= data.size();
+				return position >= data.size() && dataComplete;
+			}
+
+			virtual void setDataComplete(bool b) {
+				dataComplete = b;
+			}
+
+			void addData(const std::vector<unsigned char>& moreData) {
+				append(data, moreData);
+				onDataAvailable();
 			}
 
 		private:
 			std::vector<unsigned char> data;
 			size_t position;
+			bool dataComplete;
 	};
 }