int need_align;
unsigned long offset;
int ret;
- size_t newsize;
#if DEBUG_SERVER_MST
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
(unsigned int) size,
(unsigned int) (mst->pos - mst->off));
#endif
- if ((size > mst->curr_buf) && (size < GNUNET_SERVER_MAX_MESSAGE_SIZE)) /* Received bigger message than we can currently handle! */
- {
- newsize = mst->curr_buf + size; /* How much space do we need? */
- if (newsize >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
- newsize = GNUNET_SERVER_MAX_MESSAGE_SIZE; /* Check it's not bigger than GNUNET_SERVER_MAX_MESSAGE_SIZE */
-
- mst->hdr = GNUNET_realloc(mst->hdr, newsize);
- mst->curr_buf = newsize;
- }
-
ret = GNUNET_OK;
ibuf = (char*)mst->hdr;
while (mst->pos > 0)
mst->pos);
mst->off = 0;
}
+ if (want > mst->curr_buf)
+ {
+ mst->hdr = GNUNET_realloc(mst->hdr, want);
+ ibuf = (char*)mst->hdr;
+ mst->curr_buf = want;
+ }
+ hdr = (const struct GNUNET_MessageHeader*) &ibuf[mst->off];
if (mst->pos - mst->off < want)
{
delta = GNUNET_MIN (want - (mst->pos - mst->off),
/* can try to do zero-copy and process directly from original buffer */
hdr = (const struct GNUNET_MessageHeader *) buf;
want = ntohs (hdr->size);
+ if (want < sizeof (struct GNUNET_MessageHeader))
+ {
+ GNUNET_break_op (0);
+ mst->off = 0;
+ return GNUNET_SYSERR;
+ }
if (size < want)
break; /* or not, buffer incomplete, so copy to private buffer... */
if (one_shot == GNUNET_SYSERR)
copy:
if ( (size > 0) && (! purge) )
{
+ if (size + mst->pos > mst->curr_buf)
+ {
+ mst->hdr = GNUNET_realloc(mst->hdr, size + mst->pos);
+ ibuf = (char*)mst->hdr;
+ mst->curr_buf = size + mst->pos;
+ }
GNUNET_assert (mst->pos + size <= mst->curr_buf);
memcpy (&ibuf[mst->pos], buf, size);
mst->pos += size;
void
GNUNET_SERVER_mst_destroy (struct GNUNET_SERVER_MessageStreamTokenizer *mst)
{
+ GNUNET_free (mst->hdr);
GNUNET_free (mst);
}