1 /** 2 * @file 3 * Network buffer management 4 * 5 */ 6 7 /* 8 * Copyright (c) 2001-2004 Swedish Institute of Computer Science. 9 * All rights reserved. 10 * 11 * Redistribution and use in source and binary forms, with or without modification, 12 * are permitted provided that the following conditions are met: 13 * 14 * 1. Redistributions of source code must retain the above copyright notice, 15 * this list of conditions and the following disclaimer. 16 * 2. Redistributions in binary form must reproduce the above copyright notice, 17 * this list of conditions and the following disclaimer in the documentation 18 * and/or other materials provided with the distribution. 19 * 3. The name of the author may not be used to endorse or promote products 20 * derived from this software without specific prior written permission. 21 * 22 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 23 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 24 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT 25 * SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 26 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT 27 * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING 30 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY 31 * OF SUCH DAMAGE. 32 * 33 * This file is part of the lwIP TCP/IP stack. 34 * 35 * Author: Adam Dunkels <adam@sics.se> 36 * 37 */ 38 39 #include "lwip/opt.h" 40 41 #if LWIP_NETCONN /* don't build if not configured for use in lwipopts.h */ 42 43 #include "lwip/netbuf.h" 44 #include "lwip/memp.h" 45 46 #include <string.h> 47 48 /** 49 * Create (allocate) and initialize a new netbuf. 50 * The netbuf doesn't yet contain a packet buffer! 51 * 52 * @return a pointer to a new netbuf 53 * NULL on lack of memory 54 */ 55 struct 56 netbuf *netbuf_new(void) 57 { 58 struct netbuf *buf; 59 60 buf = (struct netbuf *)memp_malloc(MEMP_NETBUF); 61 if (buf != NULL) { 62 buf->p = NULL; 63 buf->ptr = NULL; 64 ip_addr_set_any(&buf->addr); 65 buf->port = 0; 66 #if LWIP_NETBUF_RECVINFO || LWIP_CHECKSUM_ON_COPY 67 #if LWIP_CHECKSUM_ON_COPY 68 buf->flags = 0; 69 #endif /* LWIP_CHECKSUM_ON_COPY */ 70 buf->toport_chksum = 0; 71 #if LWIP_NETBUF_RECVINFO 72 ip_addr_set_any(&buf->toaddr); 73 #endif /* LWIP_NETBUF_RECVINFO */ 74 #endif /* LWIP_NETBUF_RECVINFO || LWIP_CHECKSUM_ON_COPY */ 75 return buf; 76 } else { 77 return NULL; 78 } 79 } 80 81 /** 82 * Deallocate a netbuf allocated by netbuf_new(). 83 * 84 * @param buf pointer to a netbuf allocated by netbuf_new() 85 */ 86 void 87 netbuf_delete(struct netbuf *buf) 88 { 89 if (buf != NULL) { 90 if (buf->p != NULL) { 91 pbuf_free(buf->p); 92 buf->p = buf->ptr = NULL; 93 } 94 memp_free(MEMP_NETBUF, buf); 95 } 96 } 97 98 /** 99 * Allocate memory for a packet buffer for a given netbuf. 100 * 101 * @param buf the netbuf for which to allocate a packet buffer 102 * @param size the size of the packet buffer to allocate 103 * @return pointer to the allocated memory 104 * NULL if no memory could be allocated 105 */ 106 void * 107 netbuf_alloc(struct netbuf *buf, u16_t size) 108 { 109 LWIP_ERROR("netbuf_alloc: invalid buf", (buf != NULL), return NULL;); 110 111 /* Deallocate any previously allocated memory. */ 112 if (buf->p != NULL) { 113 pbuf_free(buf->p); 114 } 115 buf->p = pbuf_alloc(PBUF_TRANSPORT, size, PBUF_RAM); 116 if (buf->p == NULL) { 117 return NULL; 118 } 119 LWIP_ASSERT("check that first pbuf can hold size", 120 (buf->p->len >= size)); 121 buf->ptr = buf->p; 122 return buf->p->payload; 123 } 124 125 /** 126 * Free the packet buffer included in a netbuf 127 * 128 * @param buf pointer to the netbuf which contains the packet buffer to free 129 */ 130 void 131 netbuf_free(struct netbuf *buf) 132 { 133 LWIP_ERROR("netbuf_free: invalid buf", (buf != NULL), return;); 134 if (buf->p != NULL) { 135 pbuf_free(buf->p); 136 } 137 buf->p = buf->ptr = NULL; 138 } 139 140 /** 141 * Let a netbuf reference existing (non-volatile) data. 142 * 143 * @param buf netbuf which should reference the data 144 * @param dataptr pointer to the data to reference 145 * @param size size of the data 146 * @return ERR_OK if data is referenced 147 * ERR_MEM if data couldn't be referenced due to lack of memory 148 */ 149 err_t 150 netbuf_ref(struct netbuf *buf, const void *dataptr, u16_t size) 151 { 152 LWIP_ERROR("netbuf_ref: invalid buf", (buf != NULL), return ERR_ARG;); 153 if (buf->p != NULL) { 154 pbuf_free(buf->p); 155 } 156 buf->p = pbuf_alloc(PBUF_TRANSPORT, 0, PBUF_REF); 157 if (buf->p == NULL) { 158 buf->ptr = NULL; 159 return ERR_MEM; 160 } 161 buf->p->payload = (void*)dataptr; 162 buf->p->len = buf->p->tot_len = size; 163 buf->ptr = buf->p; 164 return ERR_OK; 165 } 166 167 /** 168 * Chain one netbuf to another (@see pbuf_chain) 169 * 170 * @param head the first netbuf 171 * @param tail netbuf to chain after head, freed by this function, may not be reference after returning 172 */ 173 void 174 netbuf_chain(struct netbuf *head, struct netbuf *tail) 175 { 176 LWIP_ERROR("netbuf_ref: invalid head", (head != NULL), return;); 177 LWIP_ERROR("netbuf_chain: invalid tail", (tail != NULL), return;); 178 pbuf_cat(head->p, tail->p); 179 head->ptr = head->p; 180 memp_free(MEMP_NETBUF, tail); 181 } 182 183 /** 184 * Get the data pointer and length of the data inside a netbuf. 185 * 186 * @param buf netbuf to get the data from 187 * @param dataptr pointer to a void pointer where to store the data pointer 188 * @param len pointer to an u16_t where the length of the data is stored 189 * @return ERR_OK if the information was retreived, 190 * ERR_BUF on error. 191 */ 192 err_t 193 netbuf_data(struct netbuf *buf, void **dataptr, u16_t *len) 194 { 195 LWIP_ERROR("netbuf_data: invalid buf", (buf != NULL), return ERR_ARG;); 196 LWIP_ERROR("netbuf_data: invalid dataptr", (dataptr != NULL), return ERR_ARG;); 197 LWIP_ERROR("netbuf_data: invalid len", (len != NULL), return ERR_ARG;); 198 199 if (buf->ptr == NULL) { 200 return ERR_BUF; 201 } 202 *dataptr = buf->ptr->payload; 203 *len = buf->ptr->len; 204 return ERR_OK; 205 } 206 207 /** 208 * Move the current data pointer of a packet buffer contained in a netbuf 209 * to the next part. 210 * The packet buffer itself is not modified. 211 * 212 * @param buf the netbuf to modify 213 * @return -1 if there is no next part 214 * 1 if moved to the next part but now there is no next part 215 * 0 if moved to the next part and there are still more parts 216 */ 217 s8_t 218 netbuf_next(struct netbuf *buf) 219 { 220 LWIP_ERROR("netbuf_free: invalid buf", (buf != NULL), return -1;); 221 if (buf->ptr->next == NULL) { 222 return -1; 223 } 224 buf->ptr = buf->ptr->next; 225 if (buf->ptr->next == NULL) { 226 return 1; 227 } 228 return 0; 229 } 230 231 /** 232 * Move the current data pointer of a packet buffer contained in a netbuf 233 * to the beginning of the packet. 234 * The packet buffer itself is not modified. 235 * 236 * @param buf the netbuf to modify 237 */ 238 void 239 netbuf_first(struct netbuf *buf) 240 { 241 LWIP_ERROR("netbuf_free: invalid buf", (buf != NULL), return;); 242 buf->ptr = buf->p; 243 } 244 245 #endif /* LWIP_NETCONN */ 246