X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=KernelLand%2FModules%2FIPStack%2Fbuffer.c;h=3fe8b895c072bbe591574ed69c26e4cc3b6c8312;hb=845b6f9d90bb87b5e760e4d49aa93b0e003ab750;hp=2555a6359d3b8a7ee214192aad8edef04a5e7f40;hpb=001546477549f71fd1af08e13c60e0a39397532a;p=tpg%2Facess2.git

diff --git a/KernelLand/Modules/IPStack/buffer.c b/KernelLand/Modules/IPStack/buffer.c
index 2555a635..3fe8b895 100644
--- a/KernelLand/Modules/IPStack/buffer.c
+++ b/KernelLand/Modules/IPStack/buffer.c
@@ -5,6 +5,7 @@
  * buffer.c
  * - Scatter-gather handling
  */
+#define DEBUG	0
 #include "ipstack.h"
 #include "include/buffer.h"
 
@@ -23,7 +24,6 @@ struct sIPStackBuffer
 		size_t	PostLength;
 		tIPStackBufferCb	Cb;
 		void	*CbArg;
-		// TODO: Callbacks?
 	} SubBuffers[];
 };
 
@@ -33,6 +33,7 @@ tIPStackBuffer *IPStack_Buffer_CreateBuffer(int MaxBuffers)
 	tIPStackBuffer *ret;
 	
 	ret = malloc( sizeof(*ret) + MaxBuffers * sizeof(ret->SubBuffers[0]) );
+	ASSERTR(ret, NULL);
 	ret->MaxSubBufffers = MaxBuffers;
 	ret->nSubBuffers = 0;
 	ret->TotalLength = 0;
@@ -43,6 +44,7 @@ tIPStackBuffer *IPStack_Buffer_CreateBuffer(int MaxBuffers)
 
 void IPStack_Buffer_ClearBuffer(tIPStackBuffer *Buffer)
 {
+	ASSERT(Buffer);
 	IPStack_Buffer_LockBuffer(Buffer);
 	for( int i = 0; i < Buffer->nSubBuffers; i ++ )
 	{
@@ -61,6 +63,8 @@ void IPStack_Buffer_ClearBuffer(tIPStackBuffer *Buffer)
 
 void IPStack_Buffer_DestroyBuffer(tIPStackBuffer *Buffer)
 {
+	LOG("Called with %p by %p", Buffer, __builtin_return_address(0));
+	ASSERT(Buffer);
 	IPStack_Buffer_ClearBuffer(Buffer);
 	Buffer->MaxSubBufffers = 0;
 	free(Buffer);
@@ -68,10 +72,12 @@ void IPStack_Buffer_DestroyBuffer(tIPStackBuffer *Buffer)
 
 void IPStack_Buffer_LockBuffer(tIPStackBuffer *Buffer)
 {
+	ASSERT(Buffer);
 	Mutex_Acquire(&Buffer->lBufferLock);
 }
 void IPStack_Buffer_UnlockBuffer(tIPStackBuffer *Buffer)
 {
+	ASSERT(Buffer);
 	Mutex_Release(&Buffer->lBufferLock);
 }
 
@@ -80,6 +86,7 @@ void IPStack_Buffer_AppendSubBuffer(tIPStackBuffer *Buffer,
 	tIPStackBufferCb Cb, void *Arg
 	)
 {
+	ASSERT(Buffer);
 	if( Buffer->nSubBuffers == Buffer->MaxSubBufffers ) {
 		// Ah, oops?
 		Log_Error("IPStack", "Buffer %p only had %i sub-buffers allocated, which was not enough",
@@ -99,11 +106,13 @@ void IPStack_Buffer_AppendSubBuffer(tIPStackBuffer *Buffer,
 
 size_t IPStack_Buffer_GetLength(tIPStackBuffer *Buffer)
 {
+	ASSERT(Buffer);
 	return Buffer->TotalLength;
 }
 
 size_t IPStack_Buffer_GetData(tIPStackBuffer *Buffer, void *Dest, size_t MaxBytes)
 {
+	ASSERT(Buffer);
 	Uint8	*dest = Dest;
 	size_t	rem_space = MaxBytes;
 	size_t	len;
@@ -111,6 +120,13 @@ size_t IPStack_Buffer_GetData(tIPStackBuffer *Buffer, void *Dest, size_t MaxByte
 	for( int i = Buffer->nSubBuffers; i -- && rem_space != 0; )
 	{
 		len = MIN(Buffer->SubBuffers[i].PreLength, rem_space);
+		#if !DISABLE_ASSERTS
+		if( !CheckMem(Buffer->SubBuffers[i].Data, len) ) {
+			Log_Error("IPStack", "Buffer pre %i invalid (%p+0x%x)",
+				i, Buffer->SubBuffers[i].Data, len);
+			return 0;
+		}
+		#endif
 		memcpy(dest,
 			Buffer->SubBuffers[i].Data,
 			len
@@ -124,10 +140,9 @@ size_t IPStack_Buffer_GetData(tIPStackBuffer *Buffer, void *Dest, size_t MaxByte
 			continue ;
 		
 		len = MIN(Buffer->SubBuffers[i].PostLength, rem_space);
-		memcpy(dest,
-			(Uint8*)Buffer->SubBuffers[i].Data + Buffer->SubBuffers[i].PreLength,
-			len
-			);
+		void *ptr = (Uint8*)Buffer->SubBuffers[i].Data + Buffer->SubBuffers[i].PreLength;
+		ASSERT( CheckMem(ptr, len) );
+		memcpy(dest, ptr, len);
 		dest += len;
 		rem_space -= len;
 	}
@@ -137,9 +152,8 @@ size_t IPStack_Buffer_GetData(tIPStackBuffer *Buffer, void *Dest, size_t MaxByte
 
 void *IPStack_Buffer_CompactBuffer(tIPStackBuffer *Buffer, size_t *Length)
 {
-	void	*ret;
-	
-	ret = malloc(Buffer->TotalLength);
+	ASSERT(Buffer);
+	void	*ret = malloc(Buffer->TotalLength);
 	if(!ret) {
 		*Length = 0;
 		return NULL;
@@ -154,6 +168,7 @@ void *IPStack_Buffer_CompactBuffer(tIPStackBuffer *Buffer, size_t *Length)
 
 int IPStack_Buffer_GetBuffer(tIPStackBuffer *Buffer, int Index, size_t *Length, const void **DataPtr)
 {
+	ASSERT(Buffer);
 	if( Index == -1 )	Index = 0;
 
 	if( Index >= Buffer->nSubBuffers*2 ) {
@@ -172,8 +187,10 @@ int IPStack_Buffer_GetBuffer(tIPStackBuffer *Buffer, int Index, size_t *Length,
 				return -1;
 		}
 
-		*DataPtr = (Uint8*)Buffer->SubBuffers[Index].Data + Buffer->SubBuffers[Index].PreLength;
-		*Length = Buffer->SubBuffers[Index].PostLength;
+		if( DataPtr )
+			*DataPtr = (Uint8*)Buffer->SubBuffers[Index].Data + Buffer->SubBuffers[Index].PreLength;
+		if( Length )
+			*Length = Buffer->SubBuffers[Index].PostLength;
 
 		return (Index + 1) + Buffer->nSubBuffers;
 	}
@@ -182,8 +199,10 @@ int IPStack_Buffer_GetBuffer(tIPStackBuffer *Buffer, int Index, size_t *Length,
 		 int	rv = Index + 1;
 		Index = Buffer->nSubBuffers - Index - 1;
 		// Prepended buffers
-		*DataPtr = Buffer->SubBuffers[Index].Data;
-		*Length = Buffer->SubBuffers[Index].PreLength;
+		if( DataPtr )
+			*DataPtr = Buffer->SubBuffers[Index].Data;
+		if( Length )
+			*Length = Buffer->SubBuffers[Index].PreLength;
 		return rv;
 	}
 }