Kernel/messages.c

   1 /*
   2  * Acess2 Kernel
   3  * - By John Hodge (thePowersGang)
   4  *
   5  * messages.c
   6  * - IPC Messages
   7  */
   8 #define DEBUG   0
   9 #include <acess.h>
  10 #include <threads.h>
  11 #include <threads_int.h>
  12 #include <errno.h>
  13
  14 // === CODE ===
  15 /**
  16  * \fn int Proc_SendMessage(Uint *Err, Uint Dest, int Length, void *Data)
  17  * \brief Send an IPC message
  18  * \param Err   Pointer to the errno variable
  19  * \param Dest  Destination Thread
  20  * \param Length        Length of the message
  21  * \param Data  Message data
  22  */
  23 int Proc_SendMessage(Uint *Err, Uint Dest, int Length, void *Data)
  24 {
  25         tThread *thread;
  26         tMsg    *msg;
  27
  28         ENTER("pErr iDest iLength pData", Err, Dest, Length, Data);
  29
  30         if(Length <= 0 || !Data) {
  31                 *Err = -EINVAL;
  32                 LEAVE_RET('i', -1);
  33         }
  34
  35         // Get thread
  36         thread = Threads_GetThread( Dest );
  37
  38         // Error check
  39         if(!thread)     LEAVE_RET('i', -1);
  40
  41         // Get Spinlock
  42         SHORTLOCK( &thread->IsLocked );
  43
  44         // Check if thread is still alive
  45         if(thread->Status == THREAD_STAT_DEAD) {
  46                 SHORTREL( &thread->IsLocked );
  47                 LEAVE_RET('i', -1);
  48         }
  49
  50         // Create message
  51         msg = malloc( sizeof(tMsg)+Length );
  52         msg->Next = NULL;
  53         msg->Source = Proc_GetCurThread()->TID;
  54         msg->Length = Length;
  55         memcpy(msg->Data, Data, Length);
  56
  57         // If there are already messages
  58         if(thread->LastMessage) {
  59                 thread->LastMessage->Next = msg;
  60                 thread->LastMessage = msg;
  61         } else {
  62                 thread->Messages = msg;
  63                 thread->LastMessage = msg;
  64         }
  65
  66         SHORTREL(&thread->IsLocked);
  67
  68         LOG("Waking %p (%i %s)", thread, thread->TID, thread->ThreadName);
  69         Threads_Wake( thread );
  70
  71         LEAVE_RET('i', 0);
  72 }
  73
  74 /**
  75  * \fn int Proc_GetMessage(Uint *Err, Uint *Source, void *Buffer)
  76  * \brief Gets a message
  77  * \param Err   Pointer to \a errno
  78  * \param Source        Where to put the source TID
  79  * \param Buffer        Buffer to place the message data (set to NULL to just get message length)
  80  */
  81 int Proc_GetMessage(Uint *Err, Uint *Source, void *Buffer)
  82 {
  83          int    ret;
  84         void    *tmp;
  85         tThread *cur = Proc_GetCurThread();
  86
  87         ENTER("pSource pBuffer", Source, Buffer);
  88
  89         // Check if queue has any items
  90         if(!cur->Messages) {
  91                 LEAVE('i', 0);
  92                 return 0;
  93         }
  94
  95         SHORTLOCK( &cur->IsLocked );
  96
  97         if(Source) {
  98                 *Source = cur->Messages->Source;
  99                 LOG("*Source = %i", *Source);
 100         }
 101
 102         // Get message length
 103         if( !Buffer ) {
 104                 ret = cur->Messages->Length;
 105                 SHORTREL( &cur->IsLocked );
 106                 LEAVE('i', ret);
 107                 return ret;
 108         }
 109
 110         // Get message
 111         if(Buffer != GETMSG_IGNORE)
 112         {
 113                 if( !CheckMem( Buffer, cur->Messages->Length ) )
 114                 {
 115                         LOG("Invalid buffer");
 116                         *Err = -EINVAL;
 117                         SHORTREL( &cur->IsLocked );
 118                         LEAVE('i', -1);
 119                         return -1;
 120                 }
 121                 LOG("Copied to buffer");
 122                 memcpy(Buffer, cur->Messages->Data, cur->Messages->Length);
 123         }
 124         ret = cur->Messages->Length;
 125
 126         // Remove from list
 127         tmp = cur->Messages;
 128         cur->Messages = cur->Messages->Next;
 129         if(cur->Messages == NULL)       cur->LastMessage = NULL;
 130
 131         SHORTREL( &cur->IsLocked );
 132
 133         free(tmp);      // Free outside of lock
 134
 135         LEAVE('i', ret);
 136         return ret;
 137 }