blob: 6d0d432ecba4c77157ddd352a78b25e5934ae098 [file] [log] [blame]
Alexandre Lision8af73cb2013-12-10 14:11:20 -05001/* $Id: pool_signature.h 3553 2011-05-05 06:14:19Z nanang $ */
Tristan Matthews0a329cc2013-07-17 13:20:14 -04002/*
3 * Copyright (C) 2008-2011 Teluu Inc. (http://www.teluu.com)
4 * Copyright (C) 2003-2008 Benny Prijono <benny@prijono.org>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20#include <pj/assert.h>
21#include <pj/string.h>
22
23#if PJ_SAFE_POOL
24# define SIG_SIZE sizeof(pj_uint32_t)
25
26static void apply_signature(void *p, pj_size_t size);
27static void check_pool_signature(void *p, pj_size_t size);
28
29# define APPLY_SIG(p,sz) apply_signature(p,sz), \
30 p=(void*)(((char*)p)+SIG_SIZE)
31# define REMOVE_SIG(p,sz) check_pool_signature(p,sz), \
32 p=(void*)(((char*)p)-SIG_SIZE)
33
34# define SIG_BEGIN 0x600DC0DE
35# define SIG_END 0x0BADC0DE
36
37static void apply_signature(void *p, pj_size_t size)
38{
39 pj_uint32_t sig;
40
41 sig = SIG_BEGIN;
42 pj_memcpy(p, &sig, SIG_SIZE);
43
44 sig = SIG_END;
45 pj_memcpy(((char*)p)+SIG_SIZE+size, &sig, SIG_SIZE);
46}
47
48static void check_pool_signature(void *p, pj_size_t size)
49{
50 pj_uint32_t sig;
51 pj_uint8_t *mem = (pj_uint8_t*)p;
52
53 /* Check that signature at the start of the block is still intact */
54 sig = SIG_BEGIN;
55 pj_assert(!pj_memcmp(mem-SIG_SIZE, &sig, SIG_SIZE));
56
57 /* Check that signature at the end of the block is still intact.
58 * Note that "mem" has been incremented by SIG_SIZE
59 */
60 sig = SIG_END;
61 pj_assert(!pj_memcmp(mem+size, &sig, SIG_SIZE));
62}
63
64#else
65# define SIG_SIZE 0
66# define APPLY_SIG(p,sz)
67# define REMOVE_SIG(p,sz)
68#endif