[[ FoundationStream ]] Implemented MemoryOutputStream.

runrevmark · runrevmark · commit dcddc88d8558 · 2014-11-12T16:53:48.000Z
diff --git a/libfoundation/src/foundation-stream.cpp b/libfoundation/src/foundation-stream.cpp
@@ -148,14 +148,104 @@ bool MCMemoryInputStreamCreate(const void *p_block, size_t p_size, MCStreamRef&
 
 ////////////////////////////////////////////////////////////////////////////////
 
+struct __MCMemoryOutputStream
+{
+    void *buffer;
+	size_t length;
+	size_t capacity;
+};
+
+static void __MCMemoryOutputStreamDestroy(MCStreamRef p_stream)
+{
+	__MCMemoryOutputStream *self;
+	self = (__MCMemoryOutputStream *)MCStreamGetExtraBytesPtr(p_stream);
+    
+    free(self -> buffer);
+}
+
+static bool __MCMemoryOutputStreamIsFinished(MCStreamRef p_stream, bool& r_finished)
+{
+	__MCMemoryOutputStream *self;
+	self = (__MCMemoryOutputStream *)MCStreamGetExtraBytesPtr(p_stream);
+    r_finished = false;
+	return true;
+}
+
+static bool __MCMemoryOutputStreamGetAvailableForWrite(MCStreamRef p_stream, size_t& r_amount)
+{
+	__MCMemoryOutputStream *self;
+	self = (__MCMemoryOutputStream *)MCStreamGetExtraBytesPtr(p_stream);
+	r_amount = SIZE_MAX;
+	return true;
+}
+
+static bool __MCMemoryOutputStreamWrite(MCStreamRef p_stream, const void *p_buffer, size_t p_amount)
+{
+	__MCMemoryOutputStream *self;
+	self = (__MCMemoryOutputStream *)MCStreamGetExtraBytesPtr(p_stream);
+	if (p_amount > self -> capacity - self -> length)
+    {
+        size_t t_new_capacity;
+        t_new_capacity = (self -> length + p_amount + 65536) & ~65535;
+        
+        void *t_new_buffer;
+        t_new_buffer = realloc(self -> buffer, t_new_capacity);
+        if (t_new_buffer == nil)
+            return false;
+        
+        self -> buffer = t_new_buffer;
+        self -> capacity = t_new_capacity;
+    }
+	MCMemoryCopy((byte_t *)self -> buffer + self -> length, p_buffer, p_amount);
+	self -> length += p_amount;
+	return true;
+}
+
+static MCStreamCallbacks kMCMemoryOutputStreamCallbacks =
+{
+	__MCMemoryOutputStreamDestroy,
+	__MCMemoryOutputStreamIsFinished,
+	nil,
+	nil,
+	__MCMemoryOutputStreamGetAvailableForWrite,
+	__MCMemoryOutputStreamWrite,
+	nil,
+	nil,
+	nil,
+	nil,
+	nil,
+};
+
 bool MCMemoryOutputStreamCreate(MCStreamRef& r_stream)
 {
-	return false;
+	MCStreamRef t_stream;
+	if (!MCStreamCreate(&kMCMemoryOutputStreamCallbacks, sizeof(__MCMemoryOutputStream), t_stream))
+		return false;
+    
+	__MCMemoryOutputStream *self;
+	self = (__MCMemoryOutputStream *)MCStreamGetExtraBytesPtr(t_stream);
+	self -> buffer = nil;
+	self -> length = 0;
+	self -> capacity = 0;
+    
+	r_stream = t_stream;
+    
+	return true;
 }
 
-bool MCMemoryOutputStreamFinish(MCStreamRef stream, void*& r_buffer, size_t& r_size)
+bool MCMemoryOutputStreamFinish(MCStreamRef p_stream, void*& r_buffer, size_t& r_size)
 {
-	return false;
+	__MCMemoryOutputStream *self;
+	self = (__MCMemoryOutputStream *)MCStreamGetExtraBytesPtr(p_stream);
+    
+    r_buffer = realloc(self -> buffer, self -> length);
+    r_size = self -> length;
+    
+    self -> buffer = nil;
+    self -> length = 0;
+    self -> capacity = 0;
+    
+	return true;
 }
 
 ////////////////////////////////////////////////////////////////////////////////
@@ -427,7 +517,7 @@ bool MCStreamWriteUInt32(MCStreamRef self, uint32_t p_value)
 	return MCStreamWrite(self, &t_swapped_value, sizeof(uint32_t));
 }
 
-bool MCStreamReadUInt64(MCStreamRef self, uint64_t p_value)
+bool MCStreamWriteUInt64(MCStreamRef self, uint64_t p_value)
 {
     uint64_t t_swapped_value;
     t_swapped_value = MCSwapInt64NetworkToHost(p_value);
