|
@@ -69,12 +69,33 @@ static int parse_socks_client(const uint8_t *data, size_t datalen,
|
|
|
|
|
|
#define CHUNK_HEADER_LEN STRUCT_OFFSET(chunk_t, mem[0])
|
|
|
|
|
|
+/* We leave this many NUL bytes at the end of the buffer. */
|
|
|
+#define SENTINEL_LEN 4
|
|
|
+
|
|
|
+/* Header size plus NUL bytes at the end */
|
|
|
+#define CHUNK_OVERHEAD (CHUNK_HEADER_LEN + SENTINEL_LEN)
|
|
|
+
|
|
|
/** Return the number of bytes needed to allocate a chunk to hold
|
|
|
* <b>memlen</b> bytes. */
|
|
|
-#define CHUNK_ALLOC_SIZE(memlen) (CHUNK_HEADER_LEN + (memlen))
|
|
|
+#define CHUNK_ALLOC_SIZE(memlen) (CHUNK_OVERHEAD + (memlen))
|
|
|
/** Return the number of usable bytes in a chunk allocated with
|
|
|
* malloc(<b>memlen</b>). */
|
|
|
-#define CHUNK_SIZE_WITH_ALLOC(memlen) ((memlen) - CHUNK_HEADER_LEN)
|
|
|
+#define CHUNK_SIZE_WITH_ALLOC(memlen) ((memlen) - CHUNK_OVERHEAD)
|
|
|
+
|
|
|
+#define DEBUG_SENTINEL
|
|
|
+
|
|
|
+#ifdef DEBUG_SENTINEL
|
|
|
+#define DBG_S(s) s
|
|
|
+#else
|
|
|
+#define DBG_S(s) (void)0
|
|
|
+#endif
|
|
|
+
|
|
|
+#define CHUNK_SET_SENTINEL(chunk, alloclen) do { \
|
|
|
+ uint8_t *a = (uint8_t*) &(chunk)->mem[(chunk)->memlen]; \
|
|
|
+ DBG_S(uint8_t *b = &((uint8_t*)(chunk))[(alloclen)-SENTINEL_LEN]); \
|
|
|
+ DBG_S(tor_assert(a == b)); \
|
|
|
+ memset(a,0,SENTINEL_LEN); \
|
|
|
+ } while (0)
|
|
|
|
|
|
/** Return the next character in <b>chunk</b> onto which data can be appended.
|
|
|
* If the chunk is full, this might be off the end of chunk->mem. */
|
|
@@ -131,6 +152,7 @@ chunk_new_with_alloc_size(size_t alloc)
|
|
|
ch->memlen = CHUNK_SIZE_WITH_ALLOC(alloc);
|
|
|
total_bytes_allocated_in_chunks += alloc;
|
|
|
ch->data = &ch->mem[0];
|
|
|
+ CHUNK_SET_SENTINEL(ch, alloc);
|
|
|
return ch;
|
|
|
}
|
|
|
|
|
@@ -140,18 +162,20 @@ static INLINE chunk_t *
|
|
|
chunk_grow(chunk_t *chunk, size_t sz)
|
|
|
{
|
|
|
off_t offset;
|
|
|
- size_t memlen_orig = chunk->memlen;
|
|
|
+ const size_t memlen_orig = chunk->memlen;
|
|
|
+ const size_t orig_alloc = CHUNK_ALLOC_SIZE(memlen_orig);
|
|
|
+ const size_t new_alloc = CHUNK_ALLOC_SIZE(sz);
|
|
|
tor_assert(sz > chunk->memlen);
|
|
|
offset = chunk->data - chunk->mem;
|
|
|
- chunk = tor_realloc(chunk, CHUNK_ALLOC_SIZE(sz));
|
|
|
+ chunk = tor_realloc(chunk, new_alloc);
|
|
|
chunk->memlen = sz;
|
|
|
chunk->data = chunk->mem + offset;
|
|
|
#ifdef DEBUG_CHUNK_ALLOC
|
|
|
- tor_assert(chunk->DBG_alloc == CHUNK_ALLOC_SIZE(memlen_orig));
|
|
|
- chunk->DBG_alloc = CHUNK_ALLOC_SIZE(sz);
|
|
|
+ tor_assert(chunk->DBG_alloc == orig_alloc);
|
|
|
+ chunk->DBG_alloc = new_alloc;
|
|
|
#endif
|
|
|
- total_bytes_allocated_in_chunks +=
|
|
|
- CHUNK_ALLOC_SIZE(sz) - CHUNK_ALLOC_SIZE(memlen_orig);
|
|
|
+ total_bytes_allocated_in_chunks += new_alloc - orig_alloc;
|
|
|
+ CHUNK_SET_SENTINEL(chunk, new_alloc);
|
|
|
return chunk;
|
|
|
}
|
|
|
|