X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Kernel%2Fmessages.c;h=32e579b3a9d50704edd60cd35e08eb46a146b8db;hb=e29b02ca55d580b2f7f10d1093c3d6ad1bc59458;hp=057f99791ab808fa59ca9627c7c8d59ae06d2e23;hpb=95a7eaaa4a1065334125b65130866f8d1048ddb7;p=tpg%2Facess2.git

diff --git a/Kernel/messages.c b/Kernel/messages.c
index 057f9979..32e579b3 100644
--- a/Kernel/messages.c
+++ b/Kernel/messages.c
@@ -2,8 +2,8 @@
  * AcessOS Microkernel Version
  * messages.c
  */
-#include <common.h>
-#include <proc.h>
+#include <acess.h>
+#include <threads.h>
 #include <errno.h>
 
 // === CODE ===
@@ -94,12 +94,20 @@ int Proc_GetMessage(Uint *Err, Uint *Source, void *Buffer)
 	
 	// Get message
 	if(Buffer != GETMSG_IGNORE)
+	{
+		if( !CheckMem( Buffer, cur->Messages->Length ) )
+		{
+			*Err = -EINVAL;
+			RELEASE( &cur->IsLocked );
+			return -1;
+		}
 		memcpy(Buffer, cur->Messages->Data, cur->Messages->Length);
+	}
 	ret = cur->Messages->Length;
 	
 	// Remove from list
 	tmp = cur->Messages->Next;
-	free(cur->Messages);
+	free( (void*)cur->Messages );
 	cur->Messages = tmp;
 	
 	RELEASE( &cur->IsLocked );