libdb/uIP/memb.c

   1 #include <stdlib.h>
   2 #include <string.h>
   3
   4 #include "uip.h"
   5 #include "memb.h"
   6
   7 void memb_init(struct memb_blks *blk)
   8 {
   9         UIP_MEMSET(blk->mem,0,(MEM_ALIGN_SIZE(blk->size)+sizeof(u32))*blk->num);
  10 }
  11
  12 void* memb_alloc(struct memb_blks *blk)
  13 {
  14         s32 i;
  15         u32 *ptr;
  16
  17         ptr = (u32*)blk->mem;
  18         for(i=0;i<blk->num;i++) {
  19                 if(*ptr==0) {
  20                         ++(*ptr);
  21                         return (void*)(ptr+1);
  22                 }
  23                 ptr = (u32*)(u8*)ptr+(MEM_ALIGN_SIZE(blk->size)+sizeof(u32));
  24         }
  25         return NULL;
  26 }
  27
  28 u8 memb_free(struct memb_blks *blk,void *ptr)
  29 {
  30         s32 i;
  31         u32 *ptr2,*ptr1;
  32
  33         ptr1 = ptr;
  34         ptr2 = (u32*)blk->mem;
  35         for(i=0;i<blk->num;i++) {
  36                 if(ptr2==(ptr1 - 1)) {
  37                         return --(*ptr2);
  38                 }
  39                 ptr2 = (u32*)(u8*)ptr2+(MEM_ALIGN_SIZE(blk->size)+sizeof(u32));
  40         }
  41         return -1;
  42 }
  43
  44 u8 memb_ref(struct memb_blks *blk,void *ptr)
  45 {
  46         u32 *pref = ptr-sizeof(u32);
  47         return ++(*pref);
  48 }