aboutsummaryrefslogtreecommitdiff
path: root/vmime-master/src/vmime/utility/streamUtils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'vmime-master/src/vmime/utility/streamUtils.cpp')
-rw-r--r--vmime-master/src/vmime/utility/streamUtils.cpp130
1 files changed, 130 insertions, 0 deletions
diff --git a/vmime-master/src/vmime/utility/streamUtils.cpp b/vmime-master/src/vmime/utility/streamUtils.cpp
new file mode 100644
index 0000000..793973c
--- /dev/null
+++ b/vmime-master/src/vmime/utility/streamUtils.cpp
@@ -0,0 +1,130 @@
+//
+// VMime library (http://www.vmime.org)
+// Copyright (C) 2002 Vincent Richard <vincent@vmime.org>
+//
+// This program is free software; you can redistribute it and/or
+// modify it under the terms of the GNU General Public License as
+// published by the Free Software Foundation; either version 3 of
+// the License, or (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License along
+// with this program; if not, write to the Free Software Foundation, Inc.,
+// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+//
+// Linking this library statically or dynamically with other modules is making
+// a combined work based on this library. Thus, the terms and conditions of
+// the GNU General Public License cover the whole combination.
+//
+
+#include "vmime/utility/streamUtils.hpp"
+
+#include <algorithm> // for std::copy
+#include <iterator> // for std::back_inserter
+
+
+
+namespace vmime {
+namespace utility {
+
+
+outputStream& operator<<(outputStream& os, const byte_t c) {
+
+ os.write(&c, 1);
+ return os;
+}
+
+
+outputStream& operator<<(outputStream& os, const string& str) {
+
+ os.write(str.data(), str.length());
+ return os;
+}
+
+
+size_t bufferedStreamCopy(inputStream& is, outputStream& os) {
+
+ return bufferedStreamCopy(is, os, 0, NULL);
+}
+
+
+size_t bufferedStreamCopyRange(
+ inputStream& is,
+ outputStream& os,
+ const size_t start,
+ const size_t length
+) {
+
+ const size_t blockSize =
+ std::min(is.getBlockSize(), os.getBlockSize());
+
+ is.skip(start);
+
+ std::vector <byte_t> vbuffer(blockSize);
+
+ byte_t* buffer = &vbuffer.front();
+ size_t total = 0;
+
+ while (!is.eof() && total < length) {
+
+ const size_t remaining = std::min(length - total, blockSize);
+ const size_t read = is.read(buffer, remaining);
+
+ if (read != 0) {
+ os.write(buffer, read);
+ total += read;
+ }
+ }
+
+ return total;
+}
+
+
+size_t bufferedStreamCopy(
+ inputStream& is,
+ outputStream& os,
+ const size_t length,
+ progressListener* progress
+) {
+
+ const size_t blockSize =
+ std::min(is.getBlockSize(), os.getBlockSize());
+
+ std::vector <byte_t> vbuffer(blockSize);
+
+ byte_t* buffer = &vbuffer.front();
+ size_t total = 0;
+
+ if (progress != NULL) {
+ progress->start(length);
+ }
+
+ while (!is.eof()) {
+
+ const size_t read = is.read(buffer, blockSize);
+
+ if (read != 0) {
+
+ os.write(buffer, read);
+ total += read;
+
+ if (progress != NULL) {
+ progress->progress(total, std::max(total, length));
+ }
+ }
+ }
+
+ if (progress != NULL) {
+ progress->stop(total);
+ }
+
+ return total;
+}
+
+
+} // utility
+} // vmime