stdbuf: Track buffer size

2 files changed, 7 insertions, 2 deletions

diff --git a/include/object/stdbuf.h b/include/object/stdbuf.h
index db30621..271c3ea 100644
--- a/include/object/stdbuf.h
+++ b/include/object/stdbuf.h
@@ -8,12 +8,17 @@ class BufferOutput : public T {
 	private:
 		BufferOutput(const BufferOutput &copy);
 		char *buffer;
+		uint16_t length;
 
 	public:
-		BufferOutput(char *target) { buffer = target; }
+		BufferOutput(char *target) : buffer(target), length(0) {}
 		virtual void put(char c) {
 			*buffer = c;
 			buffer++;
+			length++;
+		}
+		inline uint16_t size() {
+			return length;
 		}
 };
 
diff --git a/include/object/xdrstream.h b/include/object/xdrstream.h
index 55bef5b..e8d4ddc 100644
--- a/include/object/xdrstream.h
+++ b/include/object/xdrstream.h
@@ -6,7 +6,7 @@
 class XDRStream {
  private:
 	XDRStream(const XDRStream& copy);
-	uint32_t next_array_len;
+	uint16_t next_array_len;
 	bool is_fixed_length;
 
  public: