diff options
Diffstat (limited to 'Swiften/FileTransfer/FileReadBytestream.cpp')
-rw-r--r-- | Swiften/FileTransfer/FileReadBytestream.cpp | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/Swiften/FileTransfer/FileReadBytestream.cpp b/Swiften/FileTransfer/FileReadBytestream.cpp new file mode 100644 index 0000000..7ed33af --- /dev/null +++ b/Swiften/FileTransfer/FileReadBytestream.cpp @@ -0,0 +1,40 @@ +/* + * Copyright (c) 2010 Remko Tronçon + * Licensed under the GNU General Public License v3. + * See Documentation/Licenses/GPLv3.txt for more information. + */ + +#include <boost/filesystem/fstream.hpp> +#include <cassert> + +#include "Swiften/FileTransfer/FileReadBytestream.h" + +namespace Swift { + +FileReadBytestream::FileReadBytestream(const boost::filesystem::path& file) : file(file), stream(NULL) { +} + +FileReadBytestream::~FileReadBytestream() { + if (stream) { + stream->close(); + stream = NULL; + } +} + +ByteArray FileReadBytestream::read(size_t size) { + if (!stream) { + stream = new boost::filesystem::ifstream(file, std::ios_base::in|std::ios_base::binary); + } + ByteArray result; + result.resize(size); + assert(stream->good()); + stream->read(result.getData(), size); + result.resize(stream->gcount()); + return result; +} + +bool FileReadBytestream::isFinished() const { + return stream && !stream->good(); +} + +} |