]>
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
);
96 void * nsrealloc ( unsigned int poolid
, void * ptr
, size_t size
) {
97 struct nsminfo
* nsmp
, * nsmpn
;
100 return nsmalloc ( poolid
, size
);
107 if ( poolid
>= MAXPOOL
)
111 nsmp
=( struct nsminfo
*) ptr
- 1 ;
113 VALGRIND_MAKE_MEM_DEFINED ( nsmp
, sizeof ( struct nsminfo
));
115 if ( size
== nsmp
-> size
)
116 return ( void *) nsmp
-> data
;
118 nsmpn
=( struct nsminfo
*) realloc ( nsmp
, sizeof ( struct nsminfo
)+ size
);
122 VALGRIND_MOVE_MEMPOOL ( nsmp
, nsmpn
);
124 nsmpools
[ poolid
]. size
+= size
- nsmpn
-> size
;
128 nsmpn
-> prev
-> next
= nsmpn
;
130 nsmpools
[ poolid
]. blocks
= nsmpn
;
133 nsmpn
-> next
-> prev
= nsmpn
;
136 VALGRIND_MEMPOOL_CHANGE ( nsmpn
, nsmp
-> data
, nsmpn
-> data
, nsmpn
-> size
);
138 return ( void *) nsmpn
-> data
;
141 void nsfreeall ( unsigned int poolid
) {
142 struct nsminfo
* nsmp
, * nnsmp
;
144 if ( poolid
>= MAXPOOL
)
147 for ( nsmp
= nsmpools
[ poolid
]. blocks
; nsmp
; nsmp
= nnsmp
) {
149 VALGRIND_MEMPOOL_FREE ( nsmp
, nsmp
-> data
);
151 VALGRIND_DESTROY_MEMPOOL ( nsmp
);
154 nsmpools
[ poolid
]. blocks
= NULL
;
155 nsmpools
[ poolid
]. size
= 0 ;
156 nsmpools
[ poolid
]. count
= 0 ;
159 void nscheckfreeall ( unsigned int poolid
) {
160 if ( poolid
>= MAXPOOL
)
163 if ( nsmpools
[ poolid
]. blocks
) {
164 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
);
170 memset ( nsmpools
, 0 , sizeof ( nsmpools
));
176 for ( i
= 0 ; i
< MAXPOOL
; i
++)