]>
jfr.im git - irc/quakenet/newserv.git/blob - core/nsmalloc.c
1 /* nsmalloc: Simple pooled malloc() thing. */
12 #include "../lib/valgrind.h"
13 #include "../lib/memcheck.h"
14 #include "../core/hooks.h"
15 #include "../core/error.h"
17 struct nsmpool nsmpools
[ MAXPOOL
];
19 void * nsmalloc ( unsigned int poolid
, size_t size
) {
22 if ( poolid
>= MAXPOOL
)
25 /* Allocate enough for the structure and the required data */
26 nsmp
=( struct nsminfo
*) malloc ( sizeof ( struct nsminfo
)+ size
);
31 VALGRIND_CREATE_MEMPOOL ( nsmp
, 0 , 0 );
34 nsmpools
[ poolid
]. size
+= size
;
35 nsmpools
[ poolid
]. count
++;
37 if ( nsmpools
[ poolid
]. blocks
) {
38 nsmpools
[ poolid
]. blocks
-> prev
= nsmp
;
40 nsmp
-> next
= nsmpools
[ poolid
]. blocks
;
42 nsmpools
[ poolid
]. blocks
= nsmp
;
44 VALGRIND_MEMPOOL_ALLOC ( nsmp
, nsmp
-> data
, nsmp
-> size
);
46 nsmp
-> redzone
= REDZONE_MAGIC
;
47 VALGRIND_MAKE_MEM_NOACCESS (& nsmp
-> redzone
, sizeof ( nsmp
-> redzone
));
49 return ( void *) nsmp
-> data
;
52 void * nscalloc ( unsigned int poolid
, size_t nmemb
, size_t size
) {
53 size_t total
= nmemb
* size
;
56 m
= nsmalloc ( poolid
, total
);
65 void nsfree ( unsigned int poolid
, void * ptr
) {
68 if (! ptr
|| poolid
>= MAXPOOL
)
72 nsmp
=( struct nsminfo
*) ptr
- 1 ;
74 VALGRIND_MAKE_MEM_DEFINED (& nsmp
-> redzone
, sizeof ( nsmp
-> redzone
));
75 assert ( nsmp
-> redzone
== REDZONE_MAGIC
);
78 nsmp
-> prev
-> next
= nsmp
-> next
;
80 nsmpools
[ poolid
]. blocks
= NULL
;
83 nsmp
-> next
-> prev
= nsmp
-> prev
;
86 nsmpools
[ poolid
]. size
-= nsmp
-> size
;
87 nsmpools
[ poolid
]. count
--;
89 VALGRIND_MEMPOOL_FREE ( nsmp
, nsmp
-> data
);
91 VALGRIND_DESTROY_MEMPOOL ( nsmp
);
97 void * nsrealloc ( unsigned int poolid
, void * ptr
, size_t size
) {
98 struct nsminfo
* nsmp
, * nsmpn
;
101 return nsmalloc ( poolid
, size
);
108 if ( poolid
>= MAXPOOL
)
112 nsmp
=( struct nsminfo
*) ptr
- 1 ;
114 VALGRIND_MAKE_MEM_DEFINED ( nsmp
, sizeof ( struct nsminfo
));
116 if ( size
== nsmp
-> size
)
117 return ( void *) nsmp
-> data
;
119 nsmpn
=( struct nsminfo
*) realloc ( nsmp
, sizeof ( struct nsminfo
)+ size
);
123 VALGRIND_MOVE_MEMPOOL ( nsmp
, nsmpn
);
125 nsmpools
[ poolid
]. size
+= size
- nsmpn
-> size
;
129 nsmpn
-> prev
-> next
= nsmpn
;
131 nsmpools
[ poolid
]. blocks
= nsmpn
;
134 nsmpn
-> next
-> prev
= nsmpn
;
137 VALGRIND_MEMPOOL_CHANGE ( nsmpn
, nsmp
-> data
, nsmpn
-> data
, nsmpn
-> size
);
139 return ( void *) nsmpn
-> data
;
142 void nsfreeall ( unsigned int poolid
) {
143 struct nsminfo
* nsmp
, * nnsmp
;
145 if ( poolid
>= MAXPOOL
)
148 for ( nsmp
= nsmpools
[ poolid
]. blocks
; nsmp
; nsmp
= nnsmp
) {
150 VALGRIND_MEMPOOL_FREE ( nsmp
, nsmp
-> data
);
152 VALGRIND_DESTROY_MEMPOOL ( nsmp
);
156 nsmpools
[ poolid
]. blocks
= NULL
;
157 nsmpools
[ poolid
]. size
= 0 ;
158 nsmpools
[ poolid
]. count
= 0 ;
161 void nscheckfreeall ( unsigned int poolid
) {
162 if ( poolid
>= MAXPOOL
)
165 if ( nsmpools
[ poolid
]. blocks
) {
166 Error ( "core" , ERR_INFO
, "nsmalloc: Blocks still allocated in pool # %d ( %s ): %z ub, %l u items" , poolid
, nsmpoolnames
[ poolid
]? nsmpoolnames
[ poolid
]: "??" , nsmpools
[ poolid
]. size
, nsmpools
[ poolid
]. count
);
172 memset ( nsmpools
, 0 , sizeof ( nsmpools
));
178 for ( i
= 0 ; i
< MAXPOOL
; i
++)