mirror/lwip
1
0
Fork 0
mirror of https://github.com/lwip-tcpip/lwip.git synced 2024-12-26 03:16:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

pbuf: Simplify pbuf_get_contiguous implementation

Browse Source 
Use pbuf_skip_const() to simplify the implementation.

Signed-off-by: Axel Lin <axel.lin@ingics.com>
This commit is contained in:
Axel Lin 2017-10-27 11:37:29 +08:00 committed by Dirk Ziegelmeier
parent c1efb9e296
commit 41cf4012af
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/core/pbuf.c
View File

@ -110,6 +110,9 @@
volatile u8_t pbuf_free_ooseq_pending; volatile u8_t pbuf_free_ooseq_pending;
#define PBUF_POOL_IS_EMPTY() pbuf_pool_is_empty() #define PBUF_POOL_IS_EMPTY() pbuf_pool_is_empty()
static const struct pbuf *
pbuf_skip_const(const struct pbuf *in, u16_t in_offset, u16_t *out_offset);
/** /**
* Attempt to reclaim some memory from queued out-of-sequence TCP segments * Attempt to reclaim some memory from queued out-of-sequence TCP segments
* if we run out of pool pbufs. It's better to give priority to new packets * if we run out of pool pbufs. It's better to give priority to new packets
@ -1071,28 +1074,25 @@ void *
pbuf_get_contiguous(const struct pbuf *p, void *buffer, size_t bufsize, u16_t len, u16_t offset) pbuf_get_contiguous(const struct pbuf *p, void *buffer, size_t bufsize, u16_t len, u16_t offset)
{ {
const struct pbuf *q; const struct pbuf *q;
uint16_t out_offset;
LWIP_ERROR("pbuf_get_contiguous: invalid buf", (p != NULL), return NULL;); LWIP_ERROR("pbuf_get_contiguous: invalid buf", (p != NULL), return NULL;);
LWIP_ERROR("pbuf_get_contiguous: invalid dataptr", (buffer != NULL), return NULL;); LWIP_ERROR("pbuf_get_contiguous: invalid dataptr", (buffer != NULL), return NULL;);
LWIP_ERROR("pbuf_get_contiguous: invalid dataptr", (bufsize >= len), return NULL;); LWIP_ERROR("pbuf_get_contiguous: invalid dataptr", (bufsize >= len), return NULL;);
for (q = p; q != NULL; q = q->next) { q = pbuf_skip_const(p, offset, &out_offset);
if ((offset != 0) && (offset >= q->len)) { if (q != NULL) {
/* don't copy from this buffer -> on to the next */ if (q->len >= (out_offset + len)) {
offset = (u16_t)(offset - q->len);
} else {
if (q->len >= (offset + len)) {
/* all data in this pbuf, return zero-copy */ /* all data in this pbuf, return zero-copy */
return (u8_t *)q->payload + offset; return (u8_t *)q->payload + out_offset;
} }
/* need to copy */ /* need to copy */
if (pbuf_copy_partial(q, buffer, len, offset) != len) { if (pbuf_copy_partial(q, buffer, len, out_offset) != len) {
/* copying failed: pbuf is too short */ /* copying failed: pbuf is too short */
return NULL; return NULL;
} }
return buffer; return buffer;
} }
}
/* pbuf is too short (offset does not fit in) */ /* pbuf is too short (offset does not fit in) */
return NULL; return NULL;
} }