VeraCrypt
aboutsummaryrefslogtreecommitdiff
path: root/src/Platform/MemoryStream.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/Platform/MemoryStream.cpp')
-rw-r--r--src/Platform/MemoryStream.cpp47
1 files changed, 47 insertions, 0 deletions
diff --git a/src/Platform/MemoryStream.cpp b/src/Platform/MemoryStream.cpp
new file mode 100644
index 00000000..42d38427
--- /dev/null
+++ b/src/Platform/MemoryStream.cpp
@@ -0,0 +1,47 @@
+/*
+ Copyright (c) 2008 TrueCrypt Developers Association. All rights reserved.
+
+ Governed by the TrueCrypt License 3.0 the full text of which is contained in
+ the file License.txt included in TrueCrypt binary and source code distribution
+ packages.
+*/
+
+#include "Exception.h"
+#include "MemoryStream.h"
+
+namespace TrueCrypt
+{
+ MemoryStream::MemoryStream (const ConstBufferPtr &data) :
+ ReadPosition (0)
+ {
+ Data = vector <byte> (data.Size());
+ BufferPtr (&Data[0], Data.size()).CopyFrom (data);
+ }
+
+ uint64 MemoryStream::Read (const BufferPtr &buffer)
+ {
+ if (Data.size() == 0)
+ throw ParameterIncorrect (SRC_POS);
+
+ ConstBufferPtr streamBuf (*this);
+ size_t len = buffer.Size();
+ if (streamBuf.Size() - ReadPosition < len)
+ len = streamBuf.Size() - ReadPosition;
+
+ BufferPtr(buffer).CopyFrom (streamBuf.GetRange (ReadPosition, len));
+ ReadPosition += len;
+ return len;
+ }
+
+ void MemoryStream::ReadCompleteBuffer (const BufferPtr &buffer)
+ {
+ if (Read (buffer) != buffer.Size())
+ throw InsufficientData (SRC_POS);
+ }
+
+ void MemoryStream::Write (const ConstBufferPtr &data)
+ {
+ for (uint64 i = 0; i < data.Size(); i++)
+ Data.push_back (data[i]);
+ }
+}