git.sur5r.net Git - freertos/blob - Demo/Common/ethernet/lwIP/core/memp.c

   1 /*\r
   2  * Copyright (c) 2001-2004 Swedish Institute of Computer Science.\r
   3  * All rights reserved. \r
   4  * \r
   5  * Redistribution and use in source and binary forms, with or without modification, \r
   6  * are permitted provided that the following conditions are met:\r
   7  *\r
   8  * 1. Redistributions of source code must retain the above copyright notice,\r
   9  *    this list of conditions and the following disclaimer.\r
  10  * 2. Redistributions in binary form must reproduce the above copyright notice,\r
  11  *    this list of conditions and the following disclaimer in the documentation\r
  12  *    and/or other materials provided with the distribution.\r
  13  * 3. The name of the author may not be used to endorse or promote products\r
  14  *    derived from this software without specific prior written permission. \r
  15  *\r
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED \r
  17  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF \r
  18  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT \r
  19  * SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, \r
  20  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT \r
  21  * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS \r
  22  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN \r
  23  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING \r
  24  * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY \r
  25  * OF SUCH DAMAGE.\r
  26  *\r
  27  * This file is part of the lwIP TCP/IP stack.\r
  28  * \r
  29  * Author: Adam Dunkels <adam@sics.se>\r
  30  *\r
  31  */\r
  32 \r
  33 #include "lwip/opt.h"\r
  34 \r
  35 #include "lwip/memp.h"\r
  36 \r
  37 #include "lwip/pbuf.h"\r
  38 #include "lwip/udp.h"\r
  39 #include "lwip/raw.h"\r
  40 #include "lwip/tcp.h"\r
  41 #include "lwip/api.h"\r
  42 #include "lwip/api_msg.h"\r
  43 #include "lwip/tcpip.h"\r
  44 \r
  45 #include "lwip/sys.h"\r
  46 #include "lwip/stats.h"\r
  47 \r
  48 struct memp {\r
  49   struct memp *next;\r
  50 };\r
  51 \r
  52 #define MEMP_SIZE MEM_ALIGN_SIZE(sizeof(struct memp))\r
  53 \r
  54 static struct memp *memp_tab[MEMP_MAX];\r
  55 \r
  56 static const u16_t memp_sizes[MEMP_MAX] = {\r
  57   MEM_ALIGN_SIZE(sizeof(struct pbuf)),\r
  58   MEM_ALIGN_SIZE(sizeof(struct raw_pcb)),\r
  59   MEM_ALIGN_SIZE(sizeof(struct udp_pcb)),\r
  60   MEM_ALIGN_SIZE(sizeof(struct tcp_pcb)),\r
  61   MEM_ALIGN_SIZE(sizeof(struct tcp_pcb_listen)),\r
  62   MEM_ALIGN_SIZE(sizeof(struct tcp_seg)),\r
  63   MEM_ALIGN_SIZE(sizeof(struct netbuf)),\r
  64   MEM_ALIGN_SIZE(sizeof(struct netconn)),\r
  65   MEM_ALIGN_SIZE(sizeof(struct api_msg)),\r
  66   MEM_ALIGN_SIZE(sizeof(struct tcpip_msg)),\r
  67   MEM_ALIGN_SIZE(sizeof(struct sys_timeo))\r
  68 };\r
  69 \r
  70 static const u16_t memp_num[MEMP_MAX] = {\r
  71   MEMP_NUM_PBUF,\r
  72   MEMP_NUM_RAW_PCB,\r
  73   MEMP_NUM_UDP_PCB,\r
  74   MEMP_NUM_TCP_PCB,\r
  75   MEMP_NUM_TCP_PCB_LISTEN,\r
  76   MEMP_NUM_TCP_SEG,\r
  77   MEMP_NUM_NETBUF,\r
  78   MEMP_NUM_NETCONN,\r
  79   MEMP_NUM_API_MSG,\r
  80   MEMP_NUM_TCPIP_MSG,\r
  81   MEMP_NUM_SYS_TIMEOUT\r
  82 };\r
  83 \r
  84 #define MEMP_TYPE_SIZE(qty, type) \\r
  85   ((qty) * (MEMP_SIZE + MEM_ALIGN_SIZE(sizeof(type))))\r
  86 \r
  87 static u8_t memp_memory[MEM_ALIGNMENT - 1 + \r
  88   MEMP_TYPE_SIZE(MEMP_NUM_PBUF, struct pbuf) +\r
  89   MEMP_TYPE_SIZE(MEMP_NUM_RAW_PCB, struct raw_pcb) +\r
  90   MEMP_TYPE_SIZE(MEMP_NUM_UDP_PCB, struct udp_pcb) +\r
  91   MEMP_TYPE_SIZE(MEMP_NUM_TCP_PCB, struct tcp_pcb) +\r
  92   MEMP_TYPE_SIZE(MEMP_NUM_TCP_PCB_LISTEN, struct tcp_pcb_listen) +\r
  93   MEMP_TYPE_SIZE(MEMP_NUM_TCP_SEG, struct tcp_seg) +\r
  94   MEMP_TYPE_SIZE(MEMP_NUM_NETBUF, struct netbuf) +\r
  95   MEMP_TYPE_SIZE(MEMP_NUM_NETCONN, struct netconn) +\r
  96   MEMP_TYPE_SIZE(MEMP_NUM_API_MSG, struct api_msg) +\r
  97   MEMP_TYPE_SIZE(MEMP_NUM_TCPIP_MSG, struct tcpip_msg) +\r
  98   MEMP_TYPE_SIZE(MEMP_NUM_SYS_TIMEOUT, struct sys_timeo)];\r
  99 \r
 100 #if !SYS_LIGHTWEIGHT_PROT\r
 101 static sys_sem_t mutex;\r
 102 #endif\r
 103 \r
 104 #if MEMP_SANITY_CHECK\r
 105 static int\r
 106 memp_sanity(void)\r
 107 {\r
 108   s16_t i, c;\r
 109   struct memp *m, *n;\r
 110 \r
 111   for (i = 0; i < MEMP_MAX; i++) {\r
 112     for (m = memp_tab[i]; m != NULL; m = m->next) {\r
 113       c = 1;\r
 114       for (n = memp_tab[i]; n != NULL; n = n->next) {\r
 115         if (n == m && --c < 0) {\r
 116           return 0; /* LW was: abort(); */\r
 117         }\r
 118       }\r
 119     }\r
 120   }\r
 121   return 1;\r
 122 }\r
 123 #endif /* MEMP_SANITY_CHECK*/\r
 124 \r
 125 void\r
 126 memp_init(void)\r
 127 {\r
 128   struct memp *memp;\r
 129   u16_t i, j;\r
 130 \r
 131 #if MEMP_STATS\r
 132   for (i = 0; i < MEMP_MAX; ++i) {\r
 133     lwip_stats.memp[i].used = lwip_stats.memp[i].max =\r
 134       lwip_stats.memp[i].err = 0;\r
 135     lwip_stats.memp[i].avail = memp_num[i];\r
 136   }\r
 137 #endif /* MEMP_STATS */\r
 138 \r
 139   memp = MEM_ALIGN(memp_memory);\r
 140   for (i = 0; i < MEMP_MAX; ++i) {\r
 141     memp_tab[i] = NULL;\r
 142     for (j = 0; j < memp_num[i]; ++j) {\r
 143       memp->next = memp_tab[i];\r
 144       memp_tab[i] = memp;\r
 145       memp = (struct memp *)((u8_t *)memp + MEMP_SIZE + memp_sizes[i]);\r
 146     }\r
 147   }\r
 148 \r
 149 #if !SYS_LIGHTWEIGHT_PROT\r
 150   mutex = sys_sem_new(1);\r
 151 #endif\r
 152 }\r
 153 \r
 154 void *\r
 155 memp_malloc(memp_t type)\r
 156 {\r
 157   struct memp *memp;\r
 158   void *mem;\r
 159 #if SYS_LIGHTWEIGHT_PROT\r
 160   SYS_ARCH_DECL_PROTECT(old_level);\r
 161 #endif\r
 162  \r
 163   LWIP_ASSERT("memp_malloc: type < MEMP_MAX", type < MEMP_MAX);\r
 164 \r
 165 #if SYS_LIGHTWEIGHT_PROT\r
 166   SYS_ARCH_PROTECT(old_level);\r
 167 #else /* SYS_LIGHTWEIGHT_PROT */  \r
 168   sys_sem_wait(mutex);\r
 169 #endif /* SYS_LIGHTWEIGHT_PROT */  \r
 170 \r
 171   memp = memp_tab[type];\r
 172   \r
 173   if (memp != NULL) {    \r
 174     memp_tab[type] = memp->next;    \r
 175     memp->next = NULL;\r
 176 #if MEMP_STATS\r
 177     ++lwip_stats.memp[type].used;\r
 178     if (lwip_stats.memp[type].used > lwip_stats.memp[type].max) {\r
 179       lwip_stats.memp[type].max = lwip_stats.memp[type].used;\r
 180     }\r
 181 #endif /* MEMP_STATS */\r
 182     mem = (u8_t *)memp + MEMP_SIZE;\r
 183     LWIP_ASSERT("memp_malloc: memp properly aligned",\r
 184                 ((mem_ptr_t)memp % MEM_ALIGNMENT) == 0);\r
 185   } else {\r
 186     LWIP_DEBUGF(MEMP_DEBUG | 2, ("memp_malloc: out of memory in pool %"S16_F"\n", type));\r
 187 #if MEMP_STATS\r
 188     ++lwip_stats.memp[type].err;\r
 189 #endif /* MEMP_STATS */\r
 190     mem = NULL;\r
 191   }\r
 192 \r
 193 #if SYS_LIGHTWEIGHT_PROT\r
 194   SYS_ARCH_UNPROTECT(old_level);\r
 195 #else /* SYS_LIGHTWEIGHT_PROT */\r
 196   sys_sem_signal(mutex);\r
 197 #endif /* SYS_LIGHTWEIGHT_PROT */  \r
 198 \r
 199   return mem;\r
 200 }\r
 201 \r
 202 void\r
 203 memp_free(memp_t type, void *mem)\r
 204 {\r
 205   struct memp *memp;\r
 206 #if SYS_LIGHTWEIGHT_PROT\r
 207   SYS_ARCH_DECL_PROTECT(old_level);\r
 208 #endif /* SYS_LIGHTWEIGHT_PROT */  \r
 209 \r
 210   if (mem == NULL) {\r
 211     return;\r
 212   }\r
 213 \r
 214   memp = (struct memp *)((u8_t *)mem - MEMP_SIZE);\r
 215 \r
 216 #if SYS_LIGHTWEIGHT_PROT\r
 217   SYS_ARCH_PROTECT(old_level);\r
 218 #else /* SYS_LIGHTWEIGHT_PROT */  \r
 219   sys_sem_wait(mutex);\r
 220 #endif /* SYS_LIGHTWEIGHT_PROT */  \r
 221 \r
 222 #if MEMP_STATS\r
 223   lwip_stats.memp[type].used--; \r
 224 #endif /* MEMP_STATS */\r
 225   \r
 226   memp->next = memp_tab[type]; \r
 227   memp_tab[type] = memp;\r
 228 \r
 229 #if MEMP_SANITY_CHECK\r
 230   LWIP_ASSERT("memp sanity", memp_sanity());\r
 231 #endif  \r
 232 \r
 233 #if SYS_LIGHTWEIGHT_PROT\r
 234   SYS_ARCH_UNPROTECT(old_level);\r
 235 #else /* SYS_LIGHTWEIGHT_PROT */\r
 236   sys_sem_signal(mutex);\r
 237 #endif /* SYS_LIGHTWEIGHT_PROT */  \r
 238 }\r