NetBuffer: Check result of FlattenedSize() for error

CID 610923
haiku · Jul 3, 2015 · a495c5e · a495c5e
1 parent 5889eb7
commit a495c5e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/kits/network/libnetapi/NetBuffer.cpp b/src/kits/network/libnetapi/NetBuffer.cpp
@@ -202,7 +202,11 @@ BNetBuffer::AppendMessage(const BMessage& data)
 {
 	char stackFlattenedData[STACK_BUFFER_SIZE];
 
-	size_t dataSize = data.FlattenedSize();
+	ssize_t dataSize = data.FlattenedSize();
+
+	if (dataSize < B_OK)
+		return dataSize;
+
 	if (dataSize == 0)
 		return B_ERROR;