OLD | NEW |
1 /* This Source Code Form is subject to the terms of the Mozilla Public | 1 /* This Source Code Form is subject to the terms of the Mozilla Public |
2 * License, v. 2.0. If a copy of the MPL was not distributed with this | 2 * License, v. 2.0. If a copy of the MPL was not distributed with this |
3 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ | 3 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ |
4 | 4 |
5 /* | 5 /* |
6 * arena.c | 6 * arena.c |
7 * | 7 * |
8 * This contains the implementation of NSS's thread-safe arenas. | 8 * This contains the implementation of NSS's thread-safe arenas. |
9 */ | 9 */ |
10 | 10 |
(...skipping 23 matching lines...) Expand all Loading... |
34 * NSS_ZRealloc | 34 * NSS_ZRealloc |
35 * NSS_ZFreeIf | 35 * NSS_ZFreeIf |
36 * | 36 * |
37 * The nonpublic methods relating to this type are: | 37 * The nonpublic methods relating to this type are: |
38 * | 38 * |
39 * nssArena_Create -- constructor | 39 * nssArena_Create -- constructor |
40 * nssArena_Destroy | 40 * nssArena_Destroy |
41 * nssArena_Mark | 41 * nssArena_Mark |
42 * nssArena_Release | 42 * nssArena_Release |
43 * nssArena_Unmark | 43 * nssArena_Unmark |
44 * | 44 * |
45 * nss_ZAlloc | 45 * nss_ZAlloc |
46 * nss_ZFreeIf | 46 * nss_ZFreeIf |
47 * nss_ZRealloc | 47 * nss_ZRealloc |
48 * | 48 * |
49 * In debug builds, the following calls are available: | 49 * In debug builds, the following calls are available: |
50 * | 50 * |
51 * nssArena_verifyPointer | 51 * nssArena_verifyPointer |
52 * nssArena_registerDestructor | 52 * nssArena_registerDestructor |
53 * nssArena_deregisterDestructor | 53 * nssArena_deregisterDestructor |
54 */ | 54 */ |
55 | 55 |
56 struct NSSArenaStr { | 56 struct NSSArenaStr { |
57 PLArenaPool pool; | 57 PLArenaPool pool; |
58 PRLock *lock; | 58 PRLock *lock; |
59 #ifdef ARENA_THREADMARK | 59 #ifdef ARENA_THREADMARK |
60 PRThread *marking_thread; | 60 PRThread *marking_thread; |
61 nssArenaMark *first_mark; | 61 nssArenaMark *first_mark; |
62 nssArenaMark *last_mark; | 62 nssArenaMark *last_mark; |
63 #endif /* ARENA_THREADMARK */ | 63 #endif /* ARENA_THREADMARK */ |
64 #ifdef ARENA_DESTRUCTOR_LIST | 64 #ifdef ARENA_DESTRUCTOR_LIST |
65 struct arena_destructor_node *first_destructor; | 65 struct arena_destructor_node *first_destructor; |
66 struct arena_destructor_node *last_destructor; | 66 struct arena_destructor_node *last_destructor; |
67 #endif /* ARENA_DESTRUCTOR_LIST */ | 67 #endif /* ARENA_DESTRUCTOR_LIST */ |
68 }; | 68 }; |
69 | 69 |
70 /* | 70 /* |
71 * nssArenaMark | 71 * nssArenaMark |
72 * | 72 * |
73 * This type is used to mark the current state of an NSSArena. | 73 * This type is used to mark the current state of an NSSArena. |
74 */ | 74 */ |
75 | 75 |
76 struct nssArenaMarkStr { | 76 struct nssArenaMarkStr { |
77 PRUint32 magic; | 77 PRUint32 magic; |
78 void *mark; | 78 void *mark; |
79 #ifdef ARENA_THREADMARK | 79 #ifdef ARENA_THREADMARK |
80 nssArenaMark *next; | 80 nssArenaMark *next; |
81 #endif /* ARENA_THREADMARK */ | 81 #endif /* ARENA_THREADMARK */ |
82 #ifdef ARENA_DESTRUCTOR_LIST | 82 #ifdef ARENA_DESTRUCTOR_LIST |
83 struct arena_destructor_node *next_destructor; | 83 struct arena_destructor_node *next_destructor; |
84 struct arena_destructor_node *prev_destructor; | 84 struct arena_destructor_node *prev_destructor; |
85 #endif /* ARENA_DESTRUCTOR_LIST */ | 85 #endif /* ARENA_DESTRUCTOR_LIST */ |
86 }; | 86 }; |
87 | 87 |
88 #define MARK_MAGIC 0x4d41524b /* "MARK" how original */ | 88 #define MARK_MAGIC 0x4d41524b /* "MARK" how original */ |
89 | 89 |
90 /* | 90 /* |
91 * But first, the pointer-tracking code | 91 * But first, the pointer-tracking code |
92 */ | 92 */ |
93 #ifdef DEBUG | 93 #ifdef DEBUG |
94 extern const NSSError NSS_ERROR_INTERNAL_ERROR; | 94 extern const NSSError NSS_ERROR_INTERNAL_ERROR; |
95 | 95 |
96 static nssPointerTracker arena_pointer_tracker; | 96 static nssPointerTracker arena_pointer_tracker; |
97 | 97 |
98 static PRStatus | 98 static PRStatus |
99 arena_add_pointer | 99 arena_add_pointer(const NSSArena *arena) |
100 ( | |
101 const NSSArena *arena | |
102 ) | |
103 { | 100 { |
104 PRStatus rv; | 101 PRStatus rv; |
105 | 102 |
106 rv = nssPointerTracker_initialize(&arena_pointer_tracker); | 103 rv = nssPointerTracker_initialize(&arena_pointer_tracker); |
107 if( PR_SUCCESS != rv ) { | 104 if (PR_SUCCESS != rv) { |
108 return rv; | 105 return rv; |
109 } | 106 } |
110 | 107 |
111 rv = nssPointerTracker_add(&arena_pointer_tracker, arena); | 108 rv = nssPointerTracker_add(&arena_pointer_tracker, arena); |
112 if( PR_SUCCESS != rv ) { | 109 if (PR_SUCCESS != rv) { |
113 NSSError e = NSS_GetError(); | 110 NSSError e = NSS_GetError(); |
114 if( NSS_ERROR_NO_MEMORY != e ) { | 111 if (NSS_ERROR_NO_MEMORY != e) { |
115 nss_SetError(NSS_ERROR_INTERNAL_ERROR); | 112 nss_SetError(NSS_ERROR_INTERNAL_ERROR); |
| 113 } |
| 114 |
| 115 return rv; |
| 116 } |
| 117 |
| 118 return PR_SUCCESS; |
| 119 } |
| 120 |
| 121 static PRStatus |
| 122 arena_remove_pointer(const NSSArena *arena) |
| 123 { |
| 124 PRStatus rv; |
| 125 |
| 126 rv = nssPointerTracker_remove(&arena_pointer_tracker, arena); |
| 127 if (PR_SUCCESS != rv) { |
| 128 nss_SetError(NSS_ERROR_INTERNAL_ERROR); |
116 } | 129 } |
117 | 130 |
118 return rv; | 131 return rv; |
119 } | |
120 | |
121 return PR_SUCCESS; | |
122 } | |
123 | |
124 static PRStatus | |
125 arena_remove_pointer | |
126 ( | |
127 const NSSArena *arena | |
128 ) | |
129 { | |
130 PRStatus rv; | |
131 | |
132 rv = nssPointerTracker_remove(&arena_pointer_tracker, arena); | |
133 if( PR_SUCCESS != rv ) { | |
134 nss_SetError(NSS_ERROR_INTERNAL_ERROR); | |
135 } | |
136 | |
137 return rv; | |
138 } | 132 } |
139 | 133 |
140 /* | 134 /* |
141 * nssArena_verifyPointer | 135 * nssArena_verifyPointer |
142 * | 136 * |
143 * This method is only present in debug builds. | 137 * This method is only present in debug builds. |
144 * | 138 * |
145 * If the specified pointer is a valid pointer to an NSSArena object, | 139 * If the specified pointer is a valid pointer to an NSSArena object, |
146 * this routine will return PR_SUCCESS. Otherwise, it will put an | 140 * this routine will return PR_SUCCESS. Otherwise, it will put an |
147 * error on the error stack and return PR_FAILURE. | 141 * error on the error stack and return PR_FAILURE. |
148 * | 142 * |
149 * The error may be one of the following values: | 143 * The error may be one of the following values: |
150 * NSS_ERROR_INVALID_ARENA | 144 * NSS_ERROR_INVALID_ARENA |
151 * | 145 * |
152 * Return value: | 146 * Return value: |
153 * PR_SUCCESS if the pointer is valid | 147 * PR_SUCCESS if the pointer is valid |
154 * PR_FAILURE if it isn't | 148 * PR_FAILURE if it isn't |
155 */ | 149 */ |
156 | 150 |
157 NSS_IMPLEMENT PRStatus | 151 NSS_IMPLEMENT PRStatus |
158 nssArena_verifyPointer | 152 nssArena_verifyPointer(const NSSArena *arena) |
159 ( | |
160 const NSSArena *arena | |
161 ) | |
162 { | 153 { |
163 PRStatus rv; | 154 PRStatus rv; |
164 | 155 |
165 rv = nssPointerTracker_initialize(&arena_pointer_tracker); | 156 rv = nssPointerTracker_initialize(&arena_pointer_tracker); |
166 if( PR_SUCCESS != rv ) { | 157 if (PR_SUCCESS != rv) { |
167 /* | 158 /* |
168 * This is a little disingenious. We have to initialize the | 159 * This is a little disingenious. We have to initialize the |
169 * tracker, because someone could "legitimately" try to verify | 160 * tracker, because someone could "legitimately" try to verify |
170 * an arena pointer before one is ever created. And this step | 161 * an arena pointer before one is ever created. And this step |
171 * might fail, due to lack of memory. But the only way that | 162 * might fail, due to lack of memory. But the only way that |
172 * this step can fail is if it's doing the call_once stuff, | 163 * this step can fail is if it's doing the call_once stuff, |
173 * (later calls just no-op). And if it didn't no-op, there | 164 * (later calls just no-op). And if it didn't no-op, there |
174 * aren't any valid arenas.. so the argument certainly isn't one. | 165 * aren't any valid arenas.. so the argument certainly isn't one. |
175 */ | 166 */ |
176 nss_SetError(NSS_ERROR_INVALID_ARENA); | 167 nss_SetError(NSS_ERROR_INVALID_ARENA); |
177 return PR_FAILURE; | 168 return PR_FAILURE; |
178 } | 169 } |
179 | 170 |
180 rv = nssPointerTracker_verify(&arena_pointer_tracker, arena); | 171 rv = nssPointerTracker_verify(&arena_pointer_tracker, arena); |
181 if( PR_SUCCESS != rv ) { | 172 if (PR_SUCCESS != rv) { |
182 nss_SetError(NSS_ERROR_INVALID_ARENA); | 173 nss_SetError(NSS_ERROR_INVALID_ARENA); |
183 return PR_FAILURE; | 174 return PR_FAILURE; |
184 } | 175 } |
185 | 176 |
186 return PR_SUCCESS; | 177 return PR_SUCCESS; |
187 } | 178 } |
188 #endif /* DEBUG */ | 179 #endif /* DEBUG */ |
189 | 180 |
190 #ifdef ARENA_DESTRUCTOR_LIST | 181 #ifdef ARENA_DESTRUCTOR_LIST |
191 | 182 |
192 struct arena_destructor_node { | 183 struct arena_destructor_node { |
193 struct arena_destructor_node *next; | 184 struct arena_destructor_node *next; |
194 struct arena_destructor_node *prev; | 185 struct arena_destructor_node *prev; |
195 void (*destructor)(void *argument); | 186 void (*destructor)(void *argument); |
196 void *arg; | 187 void *arg; |
197 }; | 188 }; |
198 | 189 |
199 /* | 190 /* |
200 * nssArena_registerDestructor | 191 * nssArena_registerDestructor |
201 * | 192 * |
202 * This routine stores a pointer to a callback and an arbitrary | 193 * This routine stores a pointer to a callback and an arbitrary |
203 * pointer-sized argument in the arena, at the current point in | 194 * pointer-sized argument in the arena, at the current point in |
204 * the mark stack. If the arena is destroyed, or an "earlier" | 195 * the mark stack. If the arena is destroyed, or an "earlier" |
205 * mark is released, then this destructor will be called at that | 196 * mark is released, then this destructor will be called at that |
206 * time. Note that the destructor will be called with the arena | 197 * time. Note that the destructor will be called with the arena |
207 * locked, which means the destructor may free memory in that | 198 * locked, which means the destructor may free memory in that |
208 * arena, but it may not allocate or cause to be allocated any | 199 * arena, but it may not allocate or cause to be allocated any |
209 * memory. This callback facility was included to support our | 200 * memory. This callback facility was included to support our |
210 * debug-version pointer-tracker feature; overuse runs counter to | 201 * debug-version pointer-tracker feature; overuse runs counter to |
211 * the the original intent of arenas. This routine returns a | 202 * the the original intent of arenas. This routine returns a |
212 * PRStatus value; if successful, it will return PR_SUCCESS. If | 203 * PRStatus value; if successful, it will return PR_SUCCESS. If |
213 * unsuccessful, it will set an error on the error stack and | 204 * unsuccessful, it will set an error on the error stack and |
214 * return PR_FAILURE. | 205 * return PR_FAILURE. |
215 * | 206 * |
216 * The error may be one of the following values: | 207 * The error may be one of the following values: |
217 * NSS_ERROR_INVALID_ARENA | 208 * NSS_ERROR_INVALID_ARENA |
218 * NSS_ERROR_NO_MEMORY | 209 * NSS_ERROR_NO_MEMORY |
219 * | 210 * |
220 * Return value: | 211 * Return value: |
221 * PR_SUCCESS | 212 * PR_SUCCESS |
222 * PR_FAILURE | 213 * PR_FAILURE |
223 */ | 214 */ |
224 | 215 |
225 NSS_IMPLEMENT PRStatus | 216 NSS_IMPLEMENT PRStatus |
226 nssArena_registerDestructor | 217 nssArena_registerDestructor(NSSArena *arena, void (*destructor)(void *argument), |
227 ( | 218 void *arg) |
228 NSSArena *arena, | |
229 void (*destructor)(void *argument), | |
230 void *arg | |
231 ) | |
232 { | 219 { |
233 struct arena_destructor_node *it; | 220 struct arena_destructor_node *it; |
234 | 221 |
235 #ifdef NSSDEBUG | 222 #ifdef NSSDEBUG |
236 if( PR_SUCCESS != nssArena_verifyPointer(arena) ) { | 223 if (PR_SUCCESS != nssArena_verifyPointer(arena)) { |
237 return PR_FAILURE; | 224 return PR_FAILURE; |
238 } | 225 } |
239 #endif /* NSSDEBUG */ | 226 #endif /* NSSDEBUG */ |
240 | |
241 it = nss_ZNEW(arena, struct arena_destructor_node); | |
242 if( (struct arena_destructor_node *)NULL == it ) { | |
243 return PR_FAILURE; | |
244 } | |
245 | 227 |
246 it->prev = arena->last_destructor; | 228 it = nss_ZNEW(arena, struct arena_destructor_node); |
247 arena->last_destructor->next = it; | 229 if ((struct arena_destructor_node *)NULL == it) { |
248 arena->last_destructor = it; | 230 return PR_FAILURE; |
249 it->destructor = destructor; | 231 } |
250 it->arg = arg; | |
251 | 232 |
252 if( (nssArenaMark *)NULL != arena->last_mark ) { | 233 it->prev = arena->last_destructor; |
253 arena->last_mark->prev_destructor = it->prev; | 234 arena->last_destructor->next = it; |
254 arena->last_mark->next_destructor = it->next; | 235 arena->last_destructor = it; |
255 } | 236 it->destructor = destructor; |
| 237 it->arg = arg; |
256 | 238 |
257 return PR_SUCCESS; | 239 if ((nssArenaMark *)NULL != arena->last_mark) { |
| 240 arena->last_mark->prev_destructor = it->prev; |
| 241 arena->last_mark->next_destructor = it->next; |
| 242 } |
| 243 |
| 244 return PR_SUCCESS; |
258 } | 245 } |
259 | 246 |
260 NSS_IMPLEMENT PRStatus | 247 NSS_IMPLEMENT PRStatus |
261 nssArena_deregisterDestructor | 248 nssArena_deregisterDestructor(NSSArena *arena, |
262 ( | 249 void (*destructor)(void *argument), void *arg) |
263 NSSArena *arena, | |
264 void (*destructor)(void *argument), | |
265 void *arg | |
266 ) | |
267 { | 250 { |
268 struct arena_destructor_node *it; | 251 struct arena_destructor_node *it; |
269 | 252 |
270 #ifdef NSSDEBUG | 253 #ifdef NSSDEBUG |
271 if( PR_SUCCESS != nssArena_verifyPointer(arena) ) { | 254 if (PR_SUCCESS != nssArena_verifyPointer(arena)) { |
272 return PR_FAILURE; | 255 return PR_FAILURE; |
273 } | 256 } |
274 #endif /* NSSDEBUG */ | 257 #endif /* NSSDEBUG */ |
275 | 258 |
276 for( it = arena->first_destructor; it; it = it->next ) { | 259 for (it = arena->first_destructor; it; it = it->next) { |
277 if( (it->destructor == destructor) && (it->arg == arg) ) { | 260 if ((it->destructor == destructor) && (it->arg == arg)) { |
278 break; | 261 break; |
| 262 } |
279 } | 263 } |
280 } | |
281 | 264 |
282 if( (struct arena_destructor_node *)NULL == it ) { | 265 if ((struct arena_destructor_node *)NULL == it) { |
283 nss_SetError(NSS_ERROR_NOT_FOUND); | 266 nss_SetError(NSS_ERROR_NOT_FOUND); |
284 return PR_FAILURE; | 267 return PR_FAILURE; |
285 } | 268 } |
286 | 269 |
287 if( it == arena->first_destructor ) { | 270 if (it == arena->first_destructor) { |
288 arena->first_destructor = it->next; | 271 arena->first_destructor = it->next; |
289 } | 272 } |
290 | 273 |
291 if( it == arena->last_destructor ) { | 274 if (it == arena->last_destructor) { |
292 arena->last_destructor = it->prev; | 275 arena->last_destructor = it->prev; |
293 } | 276 } |
294 | 277 |
295 if( (struct arena_destructor_node *)NULL != it->prev ) { | 278 if ((struct arena_destructor_node *)NULL != it->prev) { |
296 it->prev->next = it->next; | 279 it->prev->next = it->next; |
297 } | 280 } |
298 | 281 |
299 if( (struct arena_destructor_node *)NULL != it->next ) { | 282 if ((struct arena_destructor_node *)NULL != it->next) { |
300 it->next->prev = it->prev; | 283 it->next->prev = it->prev; |
301 } | 284 } |
302 | 285 |
303 { | 286 { |
304 nssArenaMark *m; | 287 nssArenaMark *m; |
305 for( m = arena->first_mark; m; m = m->next ) { | 288 for (m = arena->first_mark; m; m = m->next) { |
306 if( m->next_destructor == it ) { | 289 if (m->next_destructor == it) { |
307 m->next_destructor = it->next; | 290 m->next_destructor = it->next; |
308 } | 291 } |
309 if( m->prev_destructor == it ) { | 292 if (m->prev_destructor == it) { |
310 m->prev_destructor = it->prev; | 293 m->prev_destructor = it->prev; |
311 } | 294 } |
| 295 } |
312 } | 296 } |
313 } | |
314 | 297 |
315 nss_ZFreeIf(it); | 298 nss_ZFreeIf(it); |
316 return PR_SUCCESS; | 299 return PR_SUCCESS; |
317 } | 300 } |
318 | 301 |
319 static void | 302 static void |
320 nss_arena_call_destructor_chain | 303 nss_arena_call_destructor_chain(struct arena_destructor_node *it) |
321 ( | |
322 struct arena_destructor_node *it | |
323 ) | |
324 { | 304 { |
325 for( ; it ; it = it->next ) { | 305 for (; it; it = it->next) { |
326 (*(it->destructor))(it->arg); | 306 (*(it->destructor))(it->arg); |
327 } | 307 } |
328 } | 308 } |
329 | 309 |
330 #endif /* ARENA_DESTRUCTOR_LIST */ | 310 #endif /* ARENA_DESTRUCTOR_LIST */ |
331 | 311 |
332 /* | 312 /* |
333 * NSSArena_Create | 313 * NSSArena_Create |
334 * | 314 * |
335 * This routine creates a new memory arena. This routine may return | 315 * This routine creates a new memory arena. This routine may return |
336 * NULL upon error, in which case it will have created an error stack. | 316 * NULL upon error, in which case it will have created an error stack. |
337 * | 317 * |
338 * The top-level error may be one of the following values: | 318 * The top-level error may be one of the following values: |
339 * NSS_ERROR_NO_MEMORY | 319 * NSS_ERROR_NO_MEMORY |
340 * | 320 * |
341 * Return value: | 321 * Return value: |
342 * NULL upon error | 322 * NULL upon error |
343 * A pointer to an NSSArena upon success | 323 * A pointer to an NSSArena upon success |
344 */ | 324 */ |
345 | 325 |
346 NSS_IMPLEMENT NSSArena * | 326 NSS_IMPLEMENT NSSArena * |
347 NSSArena_Create | 327 NSSArena_Create(void) |
348 ( | |
349 void | |
350 ) | |
351 { | 328 { |
352 nss_ClearErrorStack(); | 329 nss_ClearErrorStack(); |
353 return nssArena_Create(); | 330 return nssArena_Create(); |
354 } | 331 } |
355 | 332 |
356 /* | 333 /* |
357 * nssArena_Create | 334 * nssArena_Create |
358 * | 335 * |
359 * This routine creates a new memory arena. This routine may return | 336 * This routine creates a new memory arena. This routine may return |
360 * NULL upon error, in which case it will have set an error on the | 337 * NULL upon error, in which case it will have set an error on the |
361 * error stack. | 338 * error stack. |
362 * | 339 * |
363 * The error may be one of the following values: | 340 * The error may be one of the following values: |
364 * NSS_ERROR_NO_MEMORY | 341 * NSS_ERROR_NO_MEMORY |
365 * | 342 * |
366 * Return value: | 343 * Return value: |
367 * NULL upon error | 344 * NULL upon error |
368 * A pointer to an NSSArena upon success | 345 * A pointer to an NSSArena upon success |
369 */ | 346 */ |
370 | 347 |
371 NSS_IMPLEMENT NSSArena * | 348 NSS_IMPLEMENT NSSArena * |
372 nssArena_Create | 349 nssArena_Create(void) |
373 ( | |
374 void | |
375 ) | |
376 { | 350 { |
377 NSSArena *rv = (NSSArena *)NULL; | 351 NSSArena *rv = (NSSArena *)NULL; |
378 | 352 |
379 rv = nss_ZNEW((NSSArena *)NULL, NSSArena); | 353 rv = nss_ZNEW((NSSArena *)NULL, NSSArena); |
380 if( (NSSArena *)NULL == rv ) { | 354 if ((NSSArena *)NULL == rv) { |
381 nss_SetError(NSS_ERROR_NO_MEMORY); | 355 nss_SetError(NSS_ERROR_NO_MEMORY); |
382 return (NSSArena *)NULL; | 356 return (NSSArena *)NULL; |
383 } | 357 } |
384 | 358 |
385 rv->lock = PR_NewLock(); | 359 rv->lock = PR_NewLock(); |
386 if( (PRLock *)NULL == rv->lock ) { | 360 if ((PRLock *)NULL == rv->lock) { |
387 (void)nss_ZFreeIf(rv); | 361 (void)nss_ZFreeIf(rv); |
388 nss_SetError(NSS_ERROR_NO_MEMORY); | 362 nss_SetError(NSS_ERROR_NO_MEMORY); |
389 return (NSSArena *)NULL; | 363 return (NSSArena *)NULL; |
390 } | 364 } |
391 | 365 |
392 /* | 366 /* |
393 * Arena sizes. The current security code has 229 occurrences of | 367 * Arena sizes. The current security code has 229 occurrences of |
394 * PORT_NewArena. The default chunksizes specified break down as | 368 * PORT_NewArena. The default chunksizes specified break down as |
395 * | 369 * |
396 * Size Mult. Specified as | 370 * Size Mult. Specified as |
397 * 512 1 512 | 371 * 512 1 512 |
398 * 1024 7 1024 | 372 * 1024 7 1024 |
399 * 2048 5 2048 | 373 * 2048 5 2048 |
400 * 2048 5 CRMF_DEFAULT_ARENA_SIZE | 374 * 2048 5 CRMF_DEFAULT_ARENA_SIZE |
401 * 2048 190 DER_DEFAULT_CHUNKSIZE | 375 * 2048 190 DER_DEFAULT_CHUNKSIZE |
402 * 2048 20 SEC_ASN1_DEFAULT_ARENA_SIZE | 376 * 2048 20 SEC_ASN1_DEFAULT_ARENA_SIZE |
403 * 4096 1 4096 | 377 * 4096 1 4096 |
404 * | 378 * |
405 * Obviously this "default chunksize" flexibility isn't very | 379 * Obviously this "default chunksize" flexibility isn't very |
406 * useful to us, so I'll just pick 2048. | 380 * useful to us, so I'll just pick 2048. |
407 */ | 381 */ |
408 | 382 |
409 PL_InitArenaPool(&rv->pool, "NSS", 2048, sizeof(double)); | 383 PL_InitArenaPool(&rv->pool, "NSS", 2048, sizeof(double)); |
410 | 384 |
411 #ifdef DEBUG | 385 #ifdef DEBUG |
412 { | 386 { |
413 PRStatus st; | 387 PRStatus st; |
414 st = arena_add_pointer(rv); | 388 st = arena_add_pointer(rv); |
415 if( PR_SUCCESS != st ) { | 389 if (PR_SUCCESS != st) { |
416 PL_FinishArenaPool(&rv->pool); | 390 PL_FinishArenaPool(&rv->pool); |
417 PR_DestroyLock(rv->lock); | 391 PR_DestroyLock(rv->lock); |
418 (void)nss_ZFreeIf(rv); | 392 (void)nss_ZFreeIf(rv); |
419 return (NSSArena *)NULL; | 393 return (NSSArena *)NULL; |
| 394 } |
420 } | 395 } |
421 } | |
422 #endif /* DEBUG */ | 396 #endif /* DEBUG */ |
423 | 397 |
424 return rv; | 398 return rv; |
425 } | 399 } |
426 | 400 |
427 /* | 401 /* |
428 * NSSArena_Destroy | 402 * NSSArena_Destroy |
429 * | 403 * |
430 * This routine will destroy the specified arena, freeing all memory | 404 * This routine will destroy the specified arena, freeing all memory |
431 * allocated from it. This routine returns a PRStatus value; if | 405 * allocated from it. This routine returns a PRStatus value; if |
432 * successful, it will return PR_SUCCESS. If unsuccessful, it will | 406 * successful, it will return PR_SUCCESS. If unsuccessful, it will |
433 * create an error stack and return PR_FAILURE. | 407 * create an error stack and return PR_FAILURE. |
434 * | 408 * |
435 * The top-level error may be one of the following values: | 409 * The top-level error may be one of the following values: |
436 * NSS_ERROR_INVALID_ARENA | 410 * NSS_ERROR_INVALID_ARENA |
437 * | 411 * |
438 * Return value: | 412 * Return value: |
439 * PR_SUCCESS upon success | 413 * PR_SUCCESS upon success |
440 * PR_FAILURE upon failure | 414 * PR_FAILURE upon failure |
441 */ | 415 */ |
442 | 416 |
443 NSS_IMPLEMENT PRStatus | 417 NSS_IMPLEMENT PRStatus |
444 NSSArena_Destroy | 418 NSSArena_Destroy(NSSArena *arena) |
445 ( | |
446 NSSArena *arena | |
447 ) | |
448 { | 419 { |
449 nss_ClearErrorStack(); | 420 nss_ClearErrorStack(); |
450 | 421 |
451 #ifdef DEBUG | 422 #ifdef DEBUG |
452 if( PR_SUCCESS != nssArena_verifyPointer(arena) ) { | 423 if (PR_SUCCESS != nssArena_verifyPointer(arena)) { |
453 return PR_FAILURE; | 424 return PR_FAILURE; |
454 } | 425 } |
455 #endif /* DEBUG */ | 426 #endif /* DEBUG */ |
456 | 427 |
457 return nssArena_Destroy(arena); | 428 return nssArena_Destroy(arena); |
458 } | 429 } |
459 | 430 |
460 /* | 431 /* |
461 * nssArena_Destroy | 432 * nssArena_Destroy |
462 * | 433 * |
463 * This routine will destroy the specified arena, freeing all memory | 434 * This routine will destroy the specified arena, freeing all memory |
464 * allocated from it. This routine returns a PRStatus value; if | 435 * allocated from it. This routine returns a PRStatus value; if |
465 * successful, it will return PR_SUCCESS. If unsuccessful, it will | 436 * successful, it will return PR_SUCCESS. If unsuccessful, it will |
466 * set an error on the error stack and return PR_FAILURE. | 437 * set an error on the error stack and return PR_FAILURE. |
467 * | 438 * |
468 * The error may be one of the following values: | 439 * The error may be one of the following values: |
469 * NSS_ERROR_INVALID_ARENA | 440 * NSS_ERROR_INVALID_ARENA |
470 * | 441 * |
471 * Return value: | 442 * Return value: |
472 * PR_SUCCESS | 443 * PR_SUCCESS |
473 * PR_FAILURE | 444 * PR_FAILURE |
474 */ | 445 */ |
475 | 446 |
476 NSS_IMPLEMENT PRStatus | 447 NSS_IMPLEMENT PRStatus |
477 nssArena_Destroy | 448 nssArena_Destroy(NSSArena *arena) |
478 ( | |
479 NSSArena *arena | |
480 ) | |
481 { | 449 { |
482 PRLock *lock; | 450 PRLock *lock; |
483 | 451 |
484 #ifdef NSSDEBUG | 452 #ifdef NSSDEBUG |
485 if( PR_SUCCESS != nssArena_verifyPointer(arena) ) { | 453 if (PR_SUCCESS != nssArena_verifyPointer(arena)) { |
486 return PR_FAILURE; | 454 return PR_FAILURE; |
487 } | 455 } |
488 #endif /* NSSDEBUG */ | 456 #endif /* NSSDEBUG */ |
489 | 457 |
490 if( (PRLock *)NULL == arena->lock ) { | 458 if ((PRLock *)NULL == arena->lock) { |
491 /* Just got destroyed */ | 459 /* Just got destroyed */ |
492 nss_SetError(NSS_ERROR_INVALID_ARENA); | 460 nss_SetError(NSS_ERROR_INVALID_ARENA); |
493 return PR_FAILURE; | 461 return PR_FAILURE; |
494 } | 462 } |
495 PR_Lock(arena->lock); | 463 PR_Lock(arena->lock); |
496 | 464 |
497 #ifdef DEBUG | 465 #ifdef DEBUG |
498 if( PR_SUCCESS != arena_remove_pointer(arena) ) { | 466 if (PR_SUCCESS != arena_remove_pointer(arena)) { |
499 PR_Unlock(arena->lock); | 467 PR_Unlock(arena->lock); |
500 return PR_FAILURE; | 468 return PR_FAILURE; |
501 } | 469 } |
502 #endif /* DEBUG */ | 470 #endif /* DEBUG */ |
503 | 471 |
504 #ifdef ARENA_DESTRUCTOR_LIST | 472 #ifdef ARENA_DESTRUCTOR_LIST |
505 /* Note that the arena is locked at this time */ | 473 /* Note that the arena is locked at this time */ |
506 nss_arena_call_destructor_chain(arena->first_destructor); | 474 nss_arena_call_destructor_chain(arena->first_destructor); |
507 #endif /* ARENA_DESTRUCTOR_LIST */ | 475 #endif /* ARENA_DESTRUCTOR_LIST */ |
508 | 476 |
509 PL_FinishArenaPool(&arena->pool); | 477 PL_FinishArenaPool(&arena->pool); |
510 lock = arena->lock; | 478 lock = arena->lock; |
511 arena->lock = (PRLock *)NULL; | 479 arena->lock = (PRLock *)NULL; |
512 PR_Unlock(lock); | 480 PR_Unlock(lock); |
513 PR_DestroyLock(lock); | 481 PR_DestroyLock(lock); |
514 (void)nss_ZFreeIf(arena); | 482 (void)nss_ZFreeIf(arena); |
515 return PR_SUCCESS; | 483 return PR_SUCCESS; |
516 } | 484 } |
517 | 485 |
518 static void *nss_zalloc_arena_locked(NSSArena *arena, PRUint32 size); | 486 static void *nss_zalloc_arena_locked(NSSArena *arena, PRUint32 size); |
519 | 487 |
520 /* | 488 /* |
521 * nssArena_Mark | 489 * nssArena_Mark |
522 * | 490 * |
523 * This routine "marks" the current state of an arena. Space | 491 * This routine "marks" the current state of an arena. Space |
524 * allocated after the arena has been marked can be freed by | 492 * allocated after the arena has been marked can be freed by |
525 * releasing the arena back to the mark with nssArena_Release, | 493 * releasing the arena back to the mark with nssArena_Release, |
526 * or committed by calling nssArena_Unmark. When successful, | 494 * or committed by calling nssArena_Unmark. When successful, |
527 * this routine returns a valid nssArenaMark pointer. This | 495 * this routine returns a valid nssArenaMark pointer. This |
528 * routine may return NULL upon error, in which case it will | 496 * routine may return NULL upon error, in which case it will |
529 * have set an error on the error stack. | 497 * have set an error on the error stack. |
530 * | 498 * |
531 * The error may be one of the following values: | 499 * The error may be one of the following values: |
532 * NSS_ERROR_INVALID_ARENA | 500 * NSS_ERROR_INVALID_ARENA |
533 * NSS_ERROR_NO_MEMORY | 501 * NSS_ERROR_NO_MEMORY |
534 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD | 502 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD |
535 * | 503 * |
536 * Return value: | 504 * Return value: |
537 * NULL upon failure | 505 * NULL upon failure |
538 * An nssArenaMark pointer upon success | 506 * An nssArenaMark pointer upon success |
539 */ | 507 */ |
540 | 508 |
541 NSS_IMPLEMENT nssArenaMark * | 509 NSS_IMPLEMENT nssArenaMark * |
542 nssArena_Mark | 510 nssArena_Mark(NSSArena *arena) |
543 ( | |
544 NSSArena *arena | |
545 ) | |
546 { | 511 { |
547 nssArenaMark *rv; | 512 nssArenaMark *rv; |
548 void *p; | 513 void *p; |
549 | 514 |
550 #ifdef NSSDEBUG | 515 #ifdef NSSDEBUG |
551 if( PR_SUCCESS != nssArena_verifyPointer(arena) ) { | 516 if (PR_SUCCESS != nssArena_verifyPointer(arena)) { |
552 return (nssArenaMark *)NULL; | 517 return (nssArenaMark *)NULL; |
553 } | 518 } |
554 #endif /* NSSDEBUG */ | 519 #endif /* NSSDEBUG */ |
555 | 520 |
556 if( (PRLock *)NULL == arena->lock ) { | 521 if ((PRLock *)NULL == arena->lock) { |
557 /* Just got destroyed */ | 522 /* Just got destroyed */ |
558 nss_SetError(NSS_ERROR_INVALID_ARENA); | 523 nss_SetError(NSS_ERROR_INVALID_ARENA); |
559 return (nssArenaMark *)NULL; | 524 return (nssArenaMark *)NULL; |
560 } | 525 } |
561 PR_Lock(arena->lock); | 526 PR_Lock(arena->lock); |
562 | 527 |
563 #ifdef ARENA_THREADMARK | 528 #ifdef ARENA_THREADMARK |
564 if( (PRThread *)NULL == arena->marking_thread ) { | 529 if ((PRThread *)NULL == arena->marking_thread) { |
565 /* Unmarked. Store our thread ID */ | 530 /* Unmarked. Store our thread ID */ |
566 arena->marking_thread = PR_GetCurrentThread(); | 531 arena->marking_thread = PR_GetCurrentThread(); |
567 /* This call never fails. */ | 532 /* This call never fails. */ |
568 } else { | 533 } else { |
569 /* Marked. Verify it's the current thread */ | 534 /* Marked. Verify it's the current thread */ |
570 if( PR_GetCurrentThread() != arena->marking_thread ) { | 535 if (PR_GetCurrentThread() != arena->marking_thread) { |
571 PR_Unlock(arena->lock); | 536 PR_Unlock(arena->lock); |
572 nss_SetError(NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD); | 537 nss_SetError(NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD); |
573 return (nssArenaMark *)NULL; | 538 return (nssArenaMark *)NULL; |
| 539 } |
574 } | 540 } |
575 } | |
576 #endif /* ARENA_THREADMARK */ | 541 #endif /* ARENA_THREADMARK */ |
577 | 542 |
578 p = PL_ARENA_MARK(&arena->pool); | 543 p = PL_ARENA_MARK(&arena->pool); |
579 /* No error possible */ | 544 /* No error possible */ |
580 | 545 |
581 /* Do this after the mark */ | 546 /* Do this after the mark */ |
582 rv = (nssArenaMark *)nss_zalloc_arena_locked(arena, sizeof(nssArenaMark)); | 547 rv = (nssArenaMark *)nss_zalloc_arena_locked(arena, sizeof(nssArenaMark)); |
583 if( (nssArenaMark *)NULL == rv ) { | 548 if ((nssArenaMark *)NULL == rv) { |
584 PR_Unlock(arena->lock); | 549 PR_Unlock(arena->lock); |
585 nss_SetError(NSS_ERROR_NO_MEMORY); | 550 nss_SetError(NSS_ERROR_NO_MEMORY); |
586 return (nssArenaMark *)NULL; | 551 return (nssArenaMark *)NULL; |
587 } | 552 } |
588 | 553 |
589 #ifdef ARENA_THREADMARK | 554 #ifdef ARENA_THREADMARK |
590 if ( (nssArenaMark *)NULL == arena->first_mark) { | 555 if ((nssArenaMark *)NULL == arena->first_mark) { |
591 arena->first_mark = rv; | 556 arena->first_mark = rv; |
592 arena->last_mark = rv; | 557 arena->last_mark = rv; |
593 } else { | 558 } else { |
594 arena->last_mark->next = rv; | 559 arena->last_mark->next = rv; |
595 arena->last_mark = rv; | 560 arena->last_mark = rv; |
596 } | 561 } |
597 #endif /* ARENA_THREADMARK */ | 562 #endif /* ARENA_THREADMARK */ |
598 | 563 |
599 rv->mark = p; | 564 rv->mark = p; |
600 rv->magic = MARK_MAGIC; | 565 rv->magic = MARK_MAGIC; |
601 | 566 |
602 #ifdef ARENA_DESTRUCTOR_LIST | 567 #ifdef ARENA_DESTRUCTOR_LIST |
603 rv->prev_destructor = arena->last_destructor; | 568 rv->prev_destructor = arena->last_destructor; |
604 #endif /* ARENA_DESTRUCTOR_LIST */ | 569 #endif /* ARENA_DESTRUCTOR_LIST */ |
605 | 570 |
606 PR_Unlock(arena->lock); | 571 PR_Unlock(arena->lock); |
607 | 572 |
608 return rv; | 573 return rv; |
609 } | 574 } |
610 | 575 |
611 /* | 576 /* |
612 * nss_arena_unmark_release | 577 * nss_arena_unmark_release |
613 * | 578 * |
614 * This static routine implements the routines nssArena_Release | 579 * This static routine implements the routines nssArena_Release |
615 * ans nssArena_Unmark, which are almost identical. | 580 * ans nssArena_Unmark, which are almost identical. |
616 */ | 581 */ |
617 | 582 |
618 static PRStatus | 583 static PRStatus |
619 nss_arena_unmark_release | 584 nss_arena_unmark_release(NSSArena *arena, nssArenaMark *arenaMark, |
620 ( | 585 PRBool release) |
621 NSSArena *arena, | |
622 nssArenaMark *arenaMark, | |
623 PRBool release | |
624 ) | |
625 { | 586 { |
626 void *inner_mark; | 587 void *inner_mark; |
627 | 588 |
628 #ifdef NSSDEBUG | 589 #ifdef NSSDEBUG |
629 if( PR_SUCCESS != nssArena_verifyPointer(arena) ) { | 590 if (PR_SUCCESS != nssArena_verifyPointer(arena)) { |
630 return PR_FAILURE; | 591 return PR_FAILURE; |
631 } | 592 } |
632 #endif /* NSSDEBUG */ | 593 #endif /* NSSDEBUG */ |
633 | 594 |
634 if( MARK_MAGIC != arenaMark->magic ) { | 595 if (MARK_MAGIC != arenaMark->magic) { |
635 nss_SetError(NSS_ERROR_INVALID_ARENA_MARK); | 596 nss_SetError(NSS_ERROR_INVALID_ARENA_MARK); |
636 return PR_FAILURE; | 597 return PR_FAILURE; |
637 } | 598 } |
638 | 599 |
639 if( (PRLock *)NULL == arena->lock ) { | 600 if ((PRLock *)NULL == arena->lock) { |
640 /* Just got destroyed */ | 601 /* Just got destroyed */ |
641 nss_SetError(NSS_ERROR_INVALID_ARENA); | 602 nss_SetError(NSS_ERROR_INVALID_ARENA); |
642 return PR_FAILURE; | 603 return PR_FAILURE; |
643 } | 604 } |
644 PR_Lock(arena->lock); | 605 PR_Lock(arena->lock); |
645 | 606 |
646 #ifdef ARENA_THREADMARK | 607 #ifdef ARENA_THREADMARK |
647 if( (PRThread *)NULL != arena->marking_thread ) { | 608 if ((PRThread *)NULL != arena->marking_thread) { |
648 if( PR_GetCurrentThread() != arena->marking_thread ) { | 609 if (PR_GetCurrentThread() != arena->marking_thread) { |
649 PR_Unlock(arena->lock); | 610 PR_Unlock(arena->lock); |
650 nss_SetError(NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD); | 611 nss_SetError(NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD); |
651 return PR_FAILURE; | 612 return PR_FAILURE; |
| 613 } |
652 } | 614 } |
653 } | |
654 #endif /* ARENA_THREADMARK */ | 615 #endif /* ARENA_THREADMARK */ |
655 | 616 |
656 if( MARK_MAGIC != arenaMark->magic ) { | 617 if (MARK_MAGIC != arenaMark->magic) { |
657 /* Just got released */ | 618 /* Just got released */ |
658 PR_Unlock(arena->lock); | 619 PR_Unlock(arena->lock); |
659 nss_SetError(NSS_ERROR_INVALID_ARENA_MARK); | 620 nss_SetError(NSS_ERROR_INVALID_ARENA_MARK); |
660 return PR_FAILURE; | 621 return PR_FAILURE; |
661 } | 622 } |
662 | 623 |
663 arenaMark->magic = 0; | 624 arenaMark->magic = 0; |
664 inner_mark = arenaMark->mark; | 625 inner_mark = arenaMark->mark; |
665 | 626 |
666 #ifdef ARENA_THREADMARK | 627 #ifdef ARENA_THREADMARK |
667 { | 628 { |
668 nssArenaMark **pMark = &arena->first_mark; | 629 nssArenaMark **pMark = &arena->first_mark; |
669 nssArenaMark *rest; | 630 nssArenaMark *rest; |
670 nssArenaMark *last = (nssArenaMark *)NULL; | 631 nssArenaMark *last = (nssArenaMark *)NULL; |
671 | 632 |
672 /* Find this mark */ | 633 /* Find this mark */ |
673 while( *pMark != arenaMark ) { | 634 while (*pMark != arenaMark) { |
674 last = *pMark; | 635 last = *pMark; |
675 pMark = &(*pMark)->next; | 636 pMark = &(*pMark)->next; |
| 637 } |
| 638 |
| 639 /* Remember the pointer, then zero it */ |
| 640 rest = (*pMark)->next; |
| 641 *pMark = (nssArenaMark *)NULL; |
| 642 |
| 643 arena->last_mark = last; |
| 644 |
| 645 /* Invalidate any later marks being implicitly released */ |
| 646 for (; (nssArenaMark *)NULL != rest; rest = rest->next) { |
| 647 rest->magic = 0; |
| 648 } |
| 649 |
| 650 /* If we just got rid of the first mark, clear the thread ID */ |
| 651 if ((nssArenaMark *)NULL == arena->first_mark) { |
| 652 arena->marking_thread = (PRThread *)NULL; |
| 653 } |
| 654 } |
| 655 #endif /* ARENA_THREADMARK */ |
| 656 |
| 657 if (release) { |
| 658 #ifdef ARENA_DESTRUCTOR_LIST |
| 659 if ((struct arena_destructor_node *)NULL != |
| 660 arenaMark->prev_destructor) { |
| 661 arenaMark->prev_destructor->next = |
| 662 (struct arena_destructor_node *)NULL; |
| 663 } |
| 664 arena->last_destructor = arenaMark->prev_destructor; |
| 665 |
| 666 /* Note that the arena is locked at this time */ |
| 667 nss_arena_call_destructor_chain(arenaMark->next_destructor); |
| 668 #endif /* ARENA_DESTRUCTOR_LIST */ |
| 669 |
| 670 PL_ARENA_RELEASE(&arena->pool, inner_mark); |
| 671 /* No error return */ |
676 } | 672 } |
677 | 673 |
678 /* Remember the pointer, then zero it */ | 674 PR_Unlock(arena->lock); |
679 rest = (*pMark)->next; | 675 return PR_SUCCESS; |
680 *pMark = (nssArenaMark *)NULL; | |
681 | |
682 arena->last_mark = last; | |
683 | |
684 /* Invalidate any later marks being implicitly released */ | |
685 for( ; (nssArenaMark *)NULL != rest; rest = rest->next ) { | |
686 rest->magic = 0; | |
687 } | |
688 | |
689 /* If we just got rid of the first mark, clear the thread ID */ | |
690 if( (nssArenaMark *)NULL == arena->first_mark ) { | |
691 arena->marking_thread = (PRThread *)NULL; | |
692 } | |
693 } | |
694 #endif /* ARENA_THREADMARK */ | |
695 | |
696 if( release ) { | |
697 #ifdef ARENA_DESTRUCTOR_LIST | |
698 if( (struct arena_destructor_node *)NULL != arenaMark->prev_destructor ) { | |
699 arenaMark->prev_destructor->next = (struct arena_destructor_node *)NULL; | |
700 } | |
701 arena->last_destructor = arenaMark->prev_destructor; | |
702 | |
703 /* Note that the arena is locked at this time */ | |
704 nss_arena_call_destructor_chain(arenaMark->next_destructor); | |
705 #endif /* ARENA_DESTRUCTOR_LIST */ | |
706 | |
707 PL_ARENA_RELEASE(&arena->pool, inner_mark); | |
708 /* No error return */ | |
709 } | |
710 | |
711 PR_Unlock(arena->lock); | |
712 return PR_SUCCESS; | |
713 } | 676 } |
714 | 677 |
715 /* | 678 /* |
716 * nssArena_Release | 679 * nssArena_Release |
717 * | 680 * |
718 * This routine invalidates and releases all memory allocated from | 681 * This routine invalidates and releases all memory allocated from |
719 * the specified arena after the point at which the specified mark | 682 * the specified arena after the point at which the specified mark |
720 * was obtained. This routine returns a PRStatus value; if successful, | 683 * was obtained. This routine returns a PRStatus value; if successful, |
721 * it will return PR_SUCCESS. If unsuccessful, it will set an error | 684 * it will return PR_SUCCESS. If unsuccessful, it will set an error |
722 * on the error stack and return PR_FAILURE. | 685 * on the error stack and return PR_FAILURE. |
723 * | 686 * |
724 * The error may be one of the following values: | 687 * The error may be one of the following values: |
725 * NSS_ERROR_INVALID_ARENA | 688 * NSS_ERROR_INVALID_ARENA |
726 * NSS_ERROR_INVALID_ARENA_MARK | 689 * NSS_ERROR_INVALID_ARENA_MARK |
727 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD | 690 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD |
728 * | 691 * |
729 * Return value: | 692 * Return value: |
730 * PR_SUCCESS | 693 * PR_SUCCESS |
731 * PR_FAILURE | 694 * PR_FAILURE |
732 */ | 695 */ |
733 | 696 |
734 NSS_IMPLEMENT PRStatus | 697 NSS_IMPLEMENT PRStatus |
735 nssArena_Release | 698 nssArena_Release(NSSArena *arena, nssArenaMark *arenaMark) |
736 ( | |
737 NSSArena *arena, | |
738 nssArenaMark *arenaMark | |
739 ) | |
740 { | 699 { |
741 return nss_arena_unmark_release(arena, arenaMark, PR_TRUE); | 700 return nss_arena_unmark_release(arena, arenaMark, PR_TRUE); |
742 } | 701 } |
743 | 702 |
744 /* | 703 /* |
745 * nssArena_Unmark | 704 * nssArena_Unmark |
746 * | 705 * |
747 * This routine "commits" the indicated mark and any marks after | 706 * This routine "commits" the indicated mark and any marks after |
748 * it, making them unreleasable. Note that any earlier marks can | 707 * it, making them unreleasable. Note that any earlier marks can |
749 * still be released, and such a release will invalidate these | 708 * still be released, and such a release will invalidate these |
750 * later unmarked regions. If an arena is to be safely shared by | 709 * later unmarked regions. If an arena is to be safely shared by |
751 * more than one thread, all marks must be either released or | 710 * more than one thread, all marks must be either released or |
752 * unmarked. This routine returns a PRStatus value; if successful, | 711 * unmarked. This routine returns a PRStatus value; if successful, |
753 * it will return PR_SUCCESS. If unsuccessful, it will set an error | 712 * it will return PR_SUCCESS. If unsuccessful, it will set an error |
754 * on the error stack and return PR_FAILURE. | 713 * on the error stack and return PR_FAILURE. |
755 * | 714 * |
756 * The error may be one of the following values: | 715 * The error may be one of the following values: |
757 * NSS_ERROR_INVALID_ARENA | 716 * NSS_ERROR_INVALID_ARENA |
758 * NSS_ERROR_INVALID_ARENA_MARK | 717 * NSS_ERROR_INVALID_ARENA_MARK |
759 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD | 718 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD |
760 * | 719 * |
761 * Return value: | 720 * Return value: |
762 * PR_SUCCESS | 721 * PR_SUCCESS |
763 * PR_FAILURE | 722 * PR_FAILURE |
764 */ | 723 */ |
765 | 724 |
766 NSS_IMPLEMENT PRStatus | 725 NSS_IMPLEMENT PRStatus |
767 nssArena_Unmark | 726 nssArena_Unmark(NSSArena *arena, nssArenaMark *arenaMark) |
768 ( | |
769 NSSArena *arena, | |
770 nssArenaMark *arenaMark | |
771 ) | |
772 { | 727 { |
773 return nss_arena_unmark_release(arena, arenaMark, PR_FALSE); | 728 return nss_arena_unmark_release(arena, arenaMark, PR_FALSE); |
774 } | 729 } |
775 | 730 |
776 /* | 731 /* |
777 * We prefix this header to all allocated blocks. It is a multiple | 732 * We prefix this header to all allocated blocks. It is a multiple |
778 * of the alignment size. Note that this usage of a header may make | 733 * of the alignment size. Note that this usage of a header may make |
779 * purify spew bogus warnings about "potentially leaked blocks" of | 734 * purify spew bogus warnings about "potentially leaked blocks" of |
780 * memory; if that gets too annoying we can add in a pointer to the | 735 * memory; if that gets too annoying we can add in a pointer to the |
781 * header in the header itself. There's not a lot of safety here; | 736 * header in the header itself. There's not a lot of safety here; |
782 * maybe we should add a magic value? | 737 * maybe we should add a magic value? |
783 */ | 738 */ |
784 struct pointer_header { | 739 struct pointer_header { |
785 NSSArena *arena; | 740 NSSArena *arena; |
786 PRUint32 size; | 741 PRUint32 size; |
787 }; | 742 }; |
788 | 743 |
789 static void * | 744 static void * |
790 nss_zalloc_arena_locked | 745 nss_zalloc_arena_locked(NSSArena *arena, PRUint32 size) |
791 ( | |
792 NSSArena *arena, | |
793 PRUint32 size | |
794 ) | |
795 { | 746 { |
796 void *p; | 747 void *p; |
797 void *rv; | 748 void *rv; |
798 struct pointer_header *h; | 749 struct pointer_header *h; |
799 PRUint32 my_size = size + sizeof(struct pointer_header); | 750 PRUint32 my_size = size + sizeof(struct pointer_header); |
800 PL_ARENA_ALLOCATE(p, &arena->pool, my_size); | 751 PL_ARENA_ALLOCATE(p, &arena->pool, my_size); |
801 if( (void *)NULL == p ) { | 752 if ((void *)NULL == p) { |
802 nss_SetError(NSS_ERROR_NO_MEMORY); | 753 nss_SetError(NSS_ERROR_NO_MEMORY); |
803 return (void *)NULL; | 754 return (void *)NULL; |
804 } | 755 } |
805 /* | 756 /* |
806 * Do this before we unlock. This way if the user is using | 757 * Do this before we unlock. This way if the user is using |
807 * an arena in one thread while destroying it in another, he'll | 758 * an arena in one thread while destroying it in another, he'll |
808 * fault/FMR in his code, not ours. | 759 * fault/FMR in his code, not ours. |
809 */ | 760 */ |
810 h = (struct pointer_header *)p; | 761 h = (struct pointer_header *)p; |
811 h->arena = arena; | 762 h->arena = arena; |
812 h->size = size; | 763 h->size = size; |
813 rv = (void *)((char *)h + sizeof(struct pointer_header)); | 764 rv = (void *)((char *)h + sizeof(struct pointer_header)); |
814 (void)nsslibc_memset(rv, 0, size); | 765 (void)nsslibc_memset(rv, 0, size); |
815 return rv; | 766 return rv; |
816 } | 767 } |
817 | 768 |
818 /* | 769 /* |
819 * NSS_ZAlloc | 770 * NSS_ZAlloc |
820 * | 771 * |
821 * This routine allocates and zeroes a section of memory of the | 772 * This routine allocates and zeroes a section of memory of the |
822 * size, and returns to the caller a pointer to that memory. If | 773 * size, and returns to the caller a pointer to that memory. If |
823 * the optional arena argument is non-null, the memory will be | 774 * the optional arena argument is non-null, the memory will be |
824 * obtained from that arena; otherwise, the memory will be obtained | 775 * obtained from that arena; otherwise, the memory will be obtained |
825 * from the heap. This routine may return NULL upon error, in | 776 * from the heap. This routine may return NULL upon error, in |
826 * which case it will have set an error upon the error stack. The | 777 * which case it will have set an error upon the error stack. The |
827 * value specified for size may be zero; in which case a valid | 778 * value specified for size may be zero; in which case a valid |
828 * zero-length block of memory will be allocated. This block may | 779 * zero-length block of memory will be allocated. This block may |
829 * be expanded by calling NSS_ZRealloc. | 780 * be expanded by calling NSS_ZRealloc. |
830 * | 781 * |
831 * The error may be one of the following values: | 782 * The error may be one of the following values: |
832 * NSS_ERROR_INVALID_ARENA | 783 * NSS_ERROR_INVALID_ARENA |
833 * NSS_ERROR_NO_MEMORY | 784 * NSS_ERROR_NO_MEMORY |
834 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD | 785 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD |
835 * | 786 * |
836 * Return value: | 787 * Return value: |
837 * NULL upon error | 788 * NULL upon error |
838 * A pointer to the new segment of zeroed memory | 789 * A pointer to the new segment of zeroed memory |
839 */ | 790 */ |
840 | 791 |
841 NSS_IMPLEMENT void * | 792 NSS_IMPLEMENT void * |
842 NSS_ZAlloc | 793 NSS_ZAlloc(NSSArena *arenaOpt, PRUint32 size) |
843 ( | |
844 NSSArena *arenaOpt, | |
845 PRUint32 size | |
846 ) | |
847 { | 794 { |
848 return nss_ZAlloc(arenaOpt, size); | 795 return nss_ZAlloc(arenaOpt, size); |
849 } | 796 } |
850 | 797 |
851 /* | 798 /* |
852 * nss_ZAlloc | 799 * nss_ZAlloc |
853 * | 800 * |
854 * This routine allocates and zeroes a section of memory of the | 801 * This routine allocates and zeroes a section of memory of the |
855 * size, and returns to the caller a pointer to that memory. If | 802 * size, and returns to the caller a pointer to that memory. If |
856 * the optional arena argument is non-null, the memory will be | 803 * the optional arena argument is non-null, the memory will be |
857 * obtained from that arena; otherwise, the memory will be obtained | 804 * obtained from that arena; otherwise, the memory will be obtained |
858 * from the heap. This routine may return NULL upon error, in | 805 * from the heap. This routine may return NULL upon error, in |
859 * which case it will have set an error upon the error stack. The | 806 * which case it will have set an error upon the error stack. The |
860 * value specified for size may be zero; in which case a valid | 807 * value specified for size may be zero; in which case a valid |
861 * zero-length block of memory will be allocated. This block may | 808 * zero-length block of memory will be allocated. This block may |
862 * be expanded by calling nss_ZRealloc. | 809 * be expanded by calling nss_ZRealloc. |
863 * | 810 * |
864 * The error may be one of the following values: | 811 * The error may be one of the following values: |
865 * NSS_ERROR_INVALID_ARENA | 812 * NSS_ERROR_INVALID_ARENA |
866 * NSS_ERROR_NO_MEMORY | 813 * NSS_ERROR_NO_MEMORY |
867 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD | 814 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD |
868 * | 815 * |
869 * Return value: | 816 * Return value: |
870 * NULL upon error | 817 * NULL upon error |
871 * A pointer to the new segment of zeroed memory | 818 * A pointer to the new segment of zeroed memory |
872 */ | 819 */ |
873 | 820 |
874 NSS_IMPLEMENT void * | 821 NSS_IMPLEMENT void * |
875 nss_ZAlloc | 822 nss_ZAlloc(NSSArena *arenaOpt, PRUint32 size) |
876 ( | |
877 NSSArena *arenaOpt, | |
878 PRUint32 size | |
879 ) | |
880 { | 823 { |
881 struct pointer_header *h; | 824 struct pointer_header *h; |
882 PRUint32 my_size = size + sizeof(struct pointer_header); | 825 PRUint32 my_size = size + sizeof(struct pointer_header); |
883 | 826 |
884 if( my_size < sizeof(struct pointer_header) ) { | 827 if (my_size < sizeof(struct pointer_header)) { |
885 /* Wrapped */ | 828 /* Wrapped */ |
886 nss_SetError(NSS_ERROR_NO_MEMORY); | 829 nss_SetError(NSS_ERROR_NO_MEMORY); |
887 return (void *)NULL; | 830 return (void *)NULL; |
888 } | |
889 | |
890 if( (NSSArena *)NULL == arenaOpt ) { | |
891 /* Heap allocation, no locking required. */ | |
892 h = (struct pointer_header *)PR_Calloc(1, my_size); | |
893 if( (struct pointer_header *)NULL == h ) { | |
894 nss_SetError(NSS_ERROR_NO_MEMORY); | |
895 return (void *)NULL; | |
896 } | 831 } |
897 | 832 |
898 h->arena = (NSSArena *)NULL; | 833 if ((NSSArena *)NULL == arenaOpt) { |
899 h->size = size; | 834 /* Heap allocation, no locking required. */ |
900 /* We used calloc: it's already zeroed */ | 835 h = (struct pointer_header *)PR_Calloc(1, my_size); |
| 836 if ((struct pointer_header *)NULL == h) { |
| 837 nss_SetError(NSS_ERROR_NO_MEMORY); |
| 838 return (void *)NULL; |
| 839 } |
901 | 840 |
902 return (void *)((char *)h + sizeof(struct pointer_header)); | 841 h->arena = (NSSArena *)NULL; |
903 } else { | 842 h->size = size; |
904 void *rv; | 843 /* We used calloc: it's already zeroed */ |
905 /* Arena allocation */ | 844 |
| 845 return (void *)((char *)h + sizeof(struct pointer_header)); |
| 846 } else { |
| 847 void *rv; |
| 848 /* Arena allocation */ |
906 #ifdef NSSDEBUG | 849 #ifdef NSSDEBUG |
907 if( PR_SUCCESS != nssArena_verifyPointer(arenaOpt) ) { | 850 if (PR_SUCCESS != nssArena_verifyPointer(arenaOpt)) { |
908 return (void *)NULL; | 851 return (void *)NULL; |
909 } | 852 } |
910 #endif /* NSSDEBUG */ | 853 #endif /* NSSDEBUG */ |
911 | 854 |
912 if( (PRLock *)NULL == arenaOpt->lock ) { | 855 if ((PRLock *)NULL == arenaOpt->lock) { |
913 /* Just got destroyed */ | 856 /* Just got destroyed */ |
914 nss_SetError(NSS_ERROR_INVALID_ARENA); | 857 nss_SetError(NSS_ERROR_INVALID_ARENA); |
915 return (void *)NULL; | 858 return (void *)NULL; |
916 } | 859 } |
917 PR_Lock(arenaOpt->lock); | 860 PR_Lock(arenaOpt->lock); |
918 | 861 |
919 #ifdef ARENA_THREADMARK | 862 #ifdef ARENA_THREADMARK |
920 if( (PRThread *)NULL != arenaOpt->marking_thread ) { | 863 if ((PRThread *)NULL != arenaOpt->marking_thread) { |
921 if( PR_GetCurrentThread() != arenaOpt->marking_thread ) { | 864 if (PR_GetCurrentThread() != arenaOpt->marking_thread) { |
922 nss_SetError(NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD); | 865 nss_SetError(NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD); |
923 PR_Unlock(arenaOpt->lock); | 866 PR_Unlock(arenaOpt->lock); |
924 return (void *)NULL; | 867 return (void *)NULL; |
925 } | 868 } |
926 } | 869 } |
927 #endif /* ARENA_THREADMARK */ | 870 #endif /* ARENA_THREADMARK */ |
928 | 871 |
929 rv = nss_zalloc_arena_locked(arenaOpt, size); | 872 rv = nss_zalloc_arena_locked(arenaOpt, size); |
930 | 873 |
931 PR_Unlock(arenaOpt->lock); | 874 PR_Unlock(arenaOpt->lock); |
932 return rv; | 875 return rv; |
933 } | 876 } |
934 /*NOTREACHED*/ | 877 /*NOTREACHED*/ |
935 } | 878 } |
936 | 879 |
937 /* | 880 /* |
938 * NSS_ZFreeIf | 881 * NSS_ZFreeIf |
939 * | 882 * |
940 * If the specified pointer is non-null, then the region of memory | 883 * If the specified pointer is non-null, then the region of memory |
941 * to which it points -- which must have been allocated with | 884 * to which it points -- which must have been allocated with |
942 * NSS_ZAlloc -- will be zeroed and released. This routine | 885 * NSS_ZAlloc -- will be zeroed and released. This routine |
943 * returns a PRStatus value; if successful, it will return PR_SUCCESS. | 886 * returns a PRStatus value; if successful, it will return PR_SUCCESS. |
944 * If unsuccessful, it will set an error on the error stack and return | 887 * If unsuccessful, it will set an error on the error stack and return |
945 * PR_FAILURE. | 888 * PR_FAILURE. |
946 * | 889 * |
947 * The error may be one of the following values: | 890 * The error may be one of the following values: |
948 * NSS_ERROR_INVALID_POINTER | 891 * NSS_ERROR_INVALID_POINTER |
949 * | 892 * |
950 * Return value: | 893 * Return value: |
951 * PR_SUCCESS | 894 * PR_SUCCESS |
952 * PR_FAILURE | 895 * PR_FAILURE |
953 */ | 896 */ |
954 NSS_IMPLEMENT PRStatus | 897 NSS_IMPLEMENT PRStatus |
955 NSS_ZFreeIf | 898 NSS_ZFreeIf(void *pointer) |
956 ( | |
957 void *pointer | |
958 ) | |
959 { | 899 { |
960 return nss_ZFreeIf(pointer); | 900 return nss_ZFreeIf(pointer); |
961 } | 901 } |
962 | 902 |
963 /* | 903 /* |
964 * nss_ZFreeIf | 904 * nss_ZFreeIf |
965 * | 905 * |
966 * If the specified pointer is non-null, then the region of memory | 906 * If the specified pointer is non-null, then the region of memory |
967 * to which it points -- which must have been allocated with | 907 * to which it points -- which must have been allocated with |
968 * nss_ZAlloc -- will be zeroed and released. This routine | 908 * nss_ZAlloc -- will be zeroed and released. This routine |
969 * returns a PRStatus value; if successful, it will return PR_SUCCESS. | 909 * returns a PRStatus value; if successful, it will return PR_SUCCESS. |
970 * If unsuccessful, it will set an error on the error stack and return | 910 * If unsuccessful, it will set an error on the error stack and return |
971 * PR_FAILURE. | 911 * PR_FAILURE. |
972 * | 912 * |
973 * The error may be one of the following values: | 913 * The error may be one of the following values: |
974 * NSS_ERROR_INVALID_POINTER | 914 * NSS_ERROR_INVALID_POINTER |
975 * | 915 * |
976 * Return value: | 916 * Return value: |
977 * PR_SUCCESS | 917 * PR_SUCCESS |
978 * PR_FAILURE | 918 * PR_FAILURE |
979 */ | 919 */ |
980 | 920 |
981 NSS_IMPLEMENT PRStatus | 921 NSS_IMPLEMENT PRStatus |
982 nss_ZFreeIf | 922 nss_ZFreeIf(void *pointer) |
983 ( | |
984 void *pointer | |
985 ) | |
986 { | 923 { |
987 struct pointer_header *h; | 924 struct pointer_header *h; |
988 | 925 |
989 if( (void *)NULL == pointer ) { | 926 if ((void *)NULL == pointer) { |
990 return PR_SUCCESS; | 927 return PR_SUCCESS; |
991 } | 928 } |
992 | 929 |
993 h = (struct pointer_header *)((char *)pointer | 930 h = (struct pointer_header *)((char *)pointer - |
994 - sizeof(struct pointer_header)); | 931 sizeof(struct pointer_header)); |
995 | 932 |
996 /* Check any magic here */ | 933 /* Check any magic here */ |
997 | 934 |
998 if( (NSSArena *)NULL == h->arena ) { | 935 if ((NSSArena *)NULL == h->arena) { |
999 /* Heap */ | 936 /* Heap */ |
1000 (void)nsslibc_memset(pointer, 0, h->size); | 937 (void)nsslibc_memset(pointer, 0, h->size); |
1001 PR_Free(h); | 938 PR_Free(h); |
1002 return PR_SUCCESS; | 939 return PR_SUCCESS; |
1003 } else { | 940 } else { |
1004 /* Arena */ | 941 /* Arena */ |
1005 #ifdef NSSDEBUG | 942 #ifdef NSSDEBUG |
1006 if( PR_SUCCESS != nssArena_verifyPointer(h->arena) ) { | 943 if (PR_SUCCESS != nssArena_verifyPointer(h->arena)) { |
1007 return PR_FAILURE; | 944 return PR_FAILURE; |
1008 } | 945 } |
1009 #endif /* NSSDEBUG */ | 946 #endif /* NSSDEBUG */ |
1010 | 947 |
1011 if( (PRLock *)NULL == h->arena->lock ) { | 948 if ((PRLock *)NULL == h->arena->lock) { |
1012 /* Just got destroyed.. so this pointer is invalid */ | 949 /* Just got destroyed.. so this pointer is invalid */ |
1013 nss_SetError(NSS_ERROR_INVALID_POINTER); | 950 nss_SetError(NSS_ERROR_INVALID_POINTER); |
1014 return PR_FAILURE; | 951 return PR_FAILURE; |
| 952 } |
| 953 PR_Lock(h->arena->lock); |
| 954 |
| 955 (void)nsslibc_memset(pointer, 0, h->size); |
| 956 |
| 957 /* No way to "free" it within an NSPR arena. */ |
| 958 |
| 959 PR_Unlock(h->arena->lock); |
| 960 return PR_SUCCESS; |
1015 } | 961 } |
1016 PR_Lock(h->arena->lock); | 962 /*NOTREACHED*/ |
1017 | |
1018 (void)nsslibc_memset(pointer, 0, h->size); | |
1019 | |
1020 /* No way to "free" it within an NSPR arena. */ | |
1021 | |
1022 PR_Unlock(h->arena->lock); | |
1023 return PR_SUCCESS; | |
1024 } | |
1025 /*NOTREACHED*/ | |
1026 } | 963 } |
1027 | 964 |
1028 /* | 965 /* |
1029 * NSS_ZRealloc | 966 * NSS_ZRealloc |
1030 * | 967 * |
1031 * This routine reallocates a block of memory obtained by calling | 968 * This routine reallocates a block of memory obtained by calling |
1032 * nss_ZAlloc or nss_ZRealloc. The portion of memory | 969 * nss_ZAlloc or nss_ZRealloc. The portion of memory |
1033 * between the new and old sizes -- which is either being newly | 970 * between the new and old sizes -- which is either being newly |
1034 * obtained or released -- is in either case zeroed. This routine | 971 * obtained or released -- is in either case zeroed. This routine |
1035 * may return NULL upon failure, in which case it will have placed | 972 * may return NULL upon failure, in which case it will have placed |
1036 * an error on the error stack. | 973 * an error on the error stack. |
1037 * | 974 * |
1038 * The error may be one of the following values: | 975 * The error may be one of the following values: |
1039 * NSS_ERROR_INVALID_POINTER | 976 * NSS_ERROR_INVALID_POINTER |
1040 * NSS_ERROR_NO_MEMORY | 977 * NSS_ERROR_NO_MEMORY |
1041 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD | 978 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD |
1042 * | 979 * |
1043 * Return value: | 980 * Return value: |
1044 * NULL upon error | 981 * NULL upon error |
1045 * A pointer to the replacement segment of memory | 982 * A pointer to the replacement segment of memory |
1046 */ | 983 */ |
1047 | 984 |
1048 NSS_EXTERN void * | 985 NSS_EXTERN void * |
1049 NSS_ZRealloc | 986 NSS_ZRealloc(void *pointer, PRUint32 newSize) |
1050 ( | |
1051 void *pointer, | |
1052 PRUint32 newSize | |
1053 ) | |
1054 { | 987 { |
1055 return nss_ZRealloc(pointer, newSize); | 988 return nss_ZRealloc(pointer, newSize); |
1056 } | 989 } |
1057 | 990 |
1058 /* | 991 /* |
1059 * nss_ZRealloc | 992 * nss_ZRealloc |
1060 * | 993 * |
1061 * This routine reallocates a block of memory obtained by calling | 994 * This routine reallocates a block of memory obtained by calling |
1062 * nss_ZAlloc or nss_ZRealloc. The portion of memory | 995 * nss_ZAlloc or nss_ZRealloc. The portion of memory |
1063 * between the new and old sizes -- which is either being newly | 996 * between the new and old sizes -- which is either being newly |
1064 * obtained or released -- is in either case zeroed. This routine | 997 * obtained or released -- is in either case zeroed. This routine |
1065 * may return NULL upon failure, in which case it will have placed | 998 * may return NULL upon failure, in which case it will have placed |
1066 * an error on the error stack. | 999 * an error on the error stack. |
1067 * | 1000 * |
1068 * The error may be one of the following values: | 1001 * The error may be one of the following values: |
1069 * NSS_ERROR_INVALID_POINTER | 1002 * NSS_ERROR_INVALID_POINTER |
1070 * NSS_ERROR_NO_MEMORY | 1003 * NSS_ERROR_NO_MEMORY |
1071 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD | 1004 * NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD |
1072 * | 1005 * |
1073 * Return value: | 1006 * Return value: |
1074 * NULL upon error | 1007 * NULL upon error |
1075 * A pointer to the replacement segment of memory | 1008 * A pointer to the replacement segment of memory |
1076 */ | 1009 */ |
1077 | 1010 |
1078 NSS_EXTERN void * | 1011 NSS_EXTERN void * |
1079 nss_ZRealloc | 1012 nss_ZRealloc(void *pointer, PRUint32 newSize) |
1080 ( | |
1081 void *pointer, | |
1082 PRUint32 newSize | |
1083 ) | |
1084 { | 1013 { |
1085 NSSArena *arena; | 1014 NSSArena *arena; |
1086 struct pointer_header *h, *new_h; | 1015 struct pointer_header *h, *new_h; |
1087 PRUint32 my_newSize = newSize + sizeof(struct pointer_header); | 1016 PRUint32 my_newSize = newSize + sizeof(struct pointer_header); |
1088 void *rv; | 1017 void *rv; |
1089 | 1018 |
1090 if( my_newSize < sizeof(struct pointer_header) ) { | 1019 if (my_newSize < sizeof(struct pointer_header)) { |
1091 /* Wrapped */ | 1020 /* Wrapped */ |
1092 nss_SetError(NSS_ERROR_NO_MEMORY); | 1021 nss_SetError(NSS_ERROR_NO_MEMORY); |
1093 return (void *)NULL; | 1022 return (void *)NULL; |
1094 } | |
1095 | |
1096 if( (void *)NULL == pointer ) { | |
1097 nss_SetError(NSS_ERROR_INVALID_POINTER); | |
1098 return (void *)NULL; | |
1099 } | |
1100 | |
1101 h = (struct pointer_header *)((char *)pointer | |
1102 - sizeof(struct pointer_header)); | |
1103 | |
1104 /* Check any magic here */ | |
1105 | |
1106 if( newSize == h->size ) { | |
1107 /* saves thrashing */ | |
1108 return pointer; | |
1109 } | |
1110 | |
1111 arena = h->arena; | |
1112 if (!arena) { | |
1113 /* Heap */ | |
1114 new_h = (struct pointer_header *)PR_Calloc(1, my_newSize); | |
1115 if( (struct pointer_header *)NULL == new_h ) { | |
1116 nss_SetError(NSS_ERROR_NO_MEMORY); | |
1117 return (void *)NULL; | |
1118 } | 1023 } |
1119 | 1024 |
1120 new_h->arena = (NSSArena *)NULL; | 1025 if ((void *)NULL == pointer) { |
1121 new_h->size = newSize; | 1026 nss_SetError(NSS_ERROR_INVALID_POINTER); |
1122 rv = (void *)((char *)new_h + sizeof(struct pointer_header)); | 1027 return (void *)NULL; |
1123 | |
1124 if( newSize > h->size ) { | |
1125 (void)nsslibc_memcpy(rv, pointer, h->size); | |
1126 (void)nsslibc_memset(&((char *)rv)[ h->size ], | |
1127 0, (newSize - h->size)); | |
1128 } else { | |
1129 (void)nsslibc_memcpy(rv, pointer, newSize); | |
1130 } | 1028 } |
1131 | 1029 |
1132 (void)nsslibc_memset(pointer, 0, h->size); | 1030 h = (struct pointer_header *)((char *)pointer - |
1133 h->size = 0; | 1031 sizeof(struct pointer_header)); |
1134 PR_Free(h); | |
1135 | 1032 |
1136 return rv; | 1033 /* Check any magic here */ |
1137 } else { | 1034 |
1138 void *p; | 1035 if (newSize == h->size) { |
1139 /* Arena */ | 1036 /* saves thrashing */ |
| 1037 return pointer; |
| 1038 } |
| 1039 |
| 1040 arena = h->arena; |
| 1041 if (!arena) { |
| 1042 /* Heap */ |
| 1043 new_h = (struct pointer_header *)PR_Calloc(1, my_newSize); |
| 1044 if ((struct pointer_header *)NULL == new_h) { |
| 1045 nss_SetError(NSS_ERROR_NO_MEMORY); |
| 1046 return (void *)NULL; |
| 1047 } |
| 1048 |
| 1049 new_h->arena = (NSSArena *)NULL; |
| 1050 new_h->size = newSize; |
| 1051 rv = (void *)((char *)new_h + sizeof(struct pointer_header)); |
| 1052 |
| 1053 if (newSize > h->size) { |
| 1054 (void)nsslibc_memcpy(rv, pointer, h->size); |
| 1055 (void)nsslibc_memset(&((char *)rv)[h->size], 0, |
| 1056 (newSize - h->size)); |
| 1057 } else { |
| 1058 (void)nsslibc_memcpy(rv, pointer, newSize); |
| 1059 } |
| 1060 |
| 1061 (void)nsslibc_memset(pointer, 0, h->size); |
| 1062 h->size = 0; |
| 1063 PR_Free(h); |
| 1064 |
| 1065 return rv; |
| 1066 } else { |
| 1067 void *p; |
| 1068 /* Arena */ |
1140 #ifdef NSSDEBUG | 1069 #ifdef NSSDEBUG |
1141 if (PR_SUCCESS != nssArena_verifyPointer(arena)) { | 1070 if (PR_SUCCESS != nssArena_verifyPointer(arena)) { |
1142 return (void *)NULL; | 1071 return (void *)NULL; |
1143 } | 1072 } |
1144 #endif /* NSSDEBUG */ | 1073 #endif /* NSSDEBUG */ |
1145 | 1074 |
1146 if (!arena->lock) { | 1075 if (!arena->lock) { |
1147 /* Just got destroyed.. so this pointer is invalid */ | 1076 /* Just got destroyed.. so this pointer is invalid */ |
1148 nss_SetError(NSS_ERROR_INVALID_POINTER); | 1077 nss_SetError(NSS_ERROR_INVALID_POINTER); |
1149 return (void *)NULL; | 1078 return (void *)NULL; |
1150 } | 1079 } |
1151 PR_Lock(arena->lock); | 1080 PR_Lock(arena->lock); |
1152 | 1081 |
1153 #ifdef ARENA_THREADMARK | 1082 #ifdef ARENA_THREADMARK |
1154 if (arena->marking_thread) { | 1083 if (arena->marking_thread) { |
1155 if (PR_GetCurrentThread() != arena->marking_thread) { | 1084 if (PR_GetCurrentThread() != arena->marking_thread) { |
1156 PR_Unlock(arena->lock); | 1085 PR_Unlock(arena->lock); |
1157 nss_SetError(NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD); | 1086 nss_SetError(NSS_ERROR_ARENA_MARKED_BY_ANOTHER_THREAD); |
1158 return (void *)NULL; | 1087 return (void *)NULL; |
1159 } | 1088 } |
1160 } | 1089 } |
1161 #endif /* ARENA_THREADMARK */ | 1090 #endif /* ARENA_THREADMARK */ |
1162 | 1091 |
1163 if( newSize < h->size ) { | 1092 if (newSize < h->size) { |
1164 /* | 1093 /* |
1165 * We have no general way of returning memory to the arena | 1094 * We have no general way of returning memory to the arena |
1166 * (mark/release doesn't work because things may have been | 1095 * (mark/release doesn't work because things may have been |
1167 * allocated after this object), so the memory is gone | 1096 * allocated after this object), so the memory is gone |
1168 * anyway. We might as well just return the same pointer to | 1097 * anyway. We might as well just return the same pointer to |
1169 * the user, saying "yeah, uh-hunh, you can only use less of | 1098 * the user, saying "yeah, uh-hunh, you can only use less of |
1170 * it now." We'll zero the leftover part, of course. And | 1099 * it now." We'll zero the leftover part, of course. And |
1171 * in fact we might as well *not* adjust h->size-- this way, | 1100 * in fact we might as well *not* adjust h->size-- this way, |
1172 * if the user reallocs back up to something not greater than | 1101 * if the user reallocs back up to something not greater than |
1173 * the original size, then voila, there's the memory! This | 1102 * the original size, then voila, there's the memory! This |
1174 * way a thrash big/small/big/small doesn't burn up the arena. | 1103 * way a thrash big/small/big/small doesn't burn up the arena. |
1175 */ | 1104 */ |
1176 char *extra = &((char *)pointer)[ newSize ]; | 1105 char *extra = &((char *)pointer)[newSize]; |
1177 (void)nsslibc_memset(extra, 0, (h->size - newSize)); | 1106 (void)nsslibc_memset(extra, 0, (h->size - newSize)); |
1178 PR_Unlock(arena->lock); | 1107 PR_Unlock(arena->lock); |
1179 return pointer; | 1108 return pointer; |
| 1109 } |
| 1110 |
| 1111 PL_ARENA_ALLOCATE(p, &arena->pool, my_newSize); |
| 1112 if ((void *)NULL == p) { |
| 1113 PR_Unlock(arena->lock); |
| 1114 nss_SetError(NSS_ERROR_NO_MEMORY); |
| 1115 return (void *)NULL; |
| 1116 } |
| 1117 |
| 1118 new_h = (struct pointer_header *)p; |
| 1119 new_h->arena = arena; |
| 1120 new_h->size = newSize; |
| 1121 rv = (void *)((char *)new_h + sizeof(struct pointer_header)); |
| 1122 if (rv != pointer) { |
| 1123 (void)nsslibc_memcpy(rv, pointer, h->size); |
| 1124 (void)nsslibc_memset(pointer, 0, h->size); |
| 1125 } |
| 1126 (void)nsslibc_memset(&((char *)rv)[h->size], 0, (newSize - h->size)); |
| 1127 h->arena = (NSSArena *)NULL; |
| 1128 h->size = 0; |
| 1129 PR_Unlock(arena->lock); |
| 1130 return rv; |
1180 } | 1131 } |
1181 | 1132 /*NOTREACHED*/ |
1182 PL_ARENA_ALLOCATE(p, &arena->pool, my_newSize); | |
1183 if( (void *)NULL == p ) { | |
1184 PR_Unlock(arena->lock); | |
1185 nss_SetError(NSS_ERROR_NO_MEMORY); | |
1186 return (void *)NULL; | |
1187 } | |
1188 | |
1189 new_h = (struct pointer_header *)p; | |
1190 new_h->arena = arena; | |
1191 new_h->size = newSize; | |
1192 rv = (void *)((char *)new_h + sizeof(struct pointer_header)); | |
1193 if (rv != pointer) { | |
1194 » (void)nsslibc_memcpy(rv, pointer, h->size); | |
1195 » (void)nsslibc_memset(pointer, 0, h->size); | |
1196 } | |
1197 (void)nsslibc_memset(&((char *)rv)[ h->size ], 0, (newSize - h->size)); | |
1198 h->arena = (NSSArena *)NULL; | |
1199 h->size = 0; | |
1200 PR_Unlock(arena->lock); | |
1201 return rv; | |
1202 } | |
1203 /*NOTREACHED*/ | |
1204 } | 1133 } |
1205 | 1134 |
1206 PRStatus | 1135 PRStatus |
1207 nssArena_Shutdown(void) | 1136 nssArena_Shutdown(void) |
1208 { | 1137 { |
1209 PRStatus rv = PR_SUCCESS; | 1138 PRStatus rv = PR_SUCCESS; |
1210 #ifdef DEBUG | 1139 #ifdef DEBUG |
1211 rv = nssPointerTracker_finalize(&arena_pointer_tracker); | 1140 rv = nssPointerTracker_finalize(&arena_pointer_tracker); |
1212 #endif | 1141 #endif |
1213 return rv; | 1142 return rv; |
1214 } | 1143 } |
OLD | NEW |