mirror of
https://github.com/the-tcpdump-group/tcpdump.git
synced 2024-11-23 18:14:29 +08:00
65 lines
1.8 KiB
C
65 lines
1.8 KiB
C
/*
|
|
* Copyright (c) 2018 The TCPDUMP project
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that: (1) source code
|
|
* distributions retain the above copyright notice and this paragraph
|
|
* in its entirety, and (2) distributions including binary code include
|
|
* the above copyright notice and this paragraph in its entirety in
|
|
* the documentation or other materials provided with the distribution.
|
|
* THIS SOFTWARE IS PROVIDED ``AS IS'' AND
|
|
* WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, WITHOUT
|
|
* LIMITATION, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
|
* FOR A PARTICULAR PURPOSE.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif
|
|
|
|
#include <stdlib.h>
|
|
#include "netdissect-alloc.h"
|
|
|
|
static void nd_add_alloc_list(netdissect_options *, nd_mem_chunk_t *);
|
|
|
|
/*
|
|
* nd_free_all() is intended to be used after a packet printing
|
|
*/
|
|
|
|
/* Add a memory chunk in allocation linked list */
|
|
static void
|
|
nd_add_alloc_list(netdissect_options *ndo, nd_mem_chunk_t *chunkp)
|
|
{
|
|
if (ndo->ndo_last_mem_p == NULL) /* first memory allocation */
|
|
chunkp->prev_mem_p = NULL;
|
|
else /* previous memory allocation */
|
|
chunkp->prev_mem_p = ndo->ndo_last_mem_p;
|
|
ndo->ndo_last_mem_p = chunkp;
|
|
}
|
|
|
|
/* malloc replacement, with tracking in a linked list */
|
|
void *
|
|
nd_malloc(netdissect_options *ndo, size_t size)
|
|
{
|
|
nd_mem_chunk_t *chunkp = malloc(sizeof(nd_mem_chunk_t) + size);
|
|
if (chunkp == NULL)
|
|
return NULL;
|
|
nd_add_alloc_list(ndo, chunkp);
|
|
return chunkp + 1;
|
|
}
|
|
|
|
/* Free chunks in allocation linked list from last to first */
|
|
void
|
|
nd_free_all(netdissect_options *ndo)
|
|
{
|
|
nd_mem_chunk_t *current, *previous;
|
|
current = ndo->ndo_last_mem_p;
|
|
while (current != NULL) {
|
|
previous = current->prev_mem_p;
|
|
free(current);
|
|
current = previous;
|
|
}
|
|
ndo->ndo_last_mem_p = NULL;
|
|
}
|