subr_pool.c revision 1.11 1 1.11 thorpej /* $NetBSD: subr_pool.c,v 1.11 1998/08/28 20:05:48 thorpej Exp $ */
2 1.1 pk
3 1.1 pk /*-
4 1.1 pk * Copyright (c) 1997 The NetBSD Foundation, Inc.
5 1.1 pk * All rights reserved.
6 1.1 pk *
7 1.1 pk * This code is derived from software contributed to The NetBSD Foundation
8 1.1 pk * by Paul Kranenburg.
9 1.1 pk *
10 1.1 pk * Redistribution and use in source and binary forms, with or without
11 1.1 pk * modification, are permitted provided that the following conditions
12 1.1 pk * are met:
13 1.1 pk * 1. Redistributions of source code must retain the above copyright
14 1.1 pk * notice, this list of conditions and the following disclaimer.
15 1.1 pk * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 pk * notice, this list of conditions and the following disclaimer in the
17 1.1 pk * documentation and/or other materials provided with the distribution.
18 1.1 pk * 3. All advertising materials mentioning features or use of this software
19 1.1 pk * must display the following acknowledgement:
20 1.1 pk * This product includes software developed by the NetBSD
21 1.1 pk * Foundation, Inc. and its contributors.
22 1.1 pk * 4. Neither the name of The NetBSD Foundation nor the names of its
23 1.1 pk * contributors may be used to endorse or promote products derived
24 1.1 pk * from this software without specific prior written permission.
25 1.1 pk *
26 1.1 pk * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27 1.1 pk * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28 1.1 pk * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 1.1 pk * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30 1.1 pk * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31 1.1 pk * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32 1.1 pk * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33 1.1 pk * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34 1.1 pk * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35 1.1 pk * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36 1.1 pk * POSSIBILITY OF SUCH DAMAGE.
37 1.1 pk */
38 1.1 pk
39 1.1 pk #include <sys/param.h>
40 1.1 pk #include <sys/systm.h>
41 1.1 pk #include <sys/proc.h>
42 1.1 pk #include <sys/errno.h>
43 1.1 pk #include <sys/kernel.h>
44 1.1 pk #include <sys/malloc.h>
45 1.1 pk #include <sys/lock.h>
46 1.1 pk #include <sys/pool.h>
47 1.1 pk
48 1.3 pk #include <vm/vm.h>
49 1.3 pk #include <vm/vm_kern.h>
50 1.3 pk
51 1.3 pk #if defined(UVM)
52 1.3 pk #include <uvm/uvm.h>
53 1.3 pk #endif
54 1.3 pk
55 1.1 pk /*
56 1.1 pk * Pool resource management utility.
57 1.3 pk *
58 1.3 pk * Memory is allocated in pages which are split into pieces according
59 1.3 pk * to the pool item size. Each page is kept on a list headed by `pr_pagelist'
60 1.3 pk * in the pool structure and the individual pool items are on a linked list
61 1.3 pk * headed by `ph_itemlist' in each page header. The memory for building
62 1.3 pk * the page list is either taken from the allocated pages themselves (for
63 1.3 pk * small pool items) or taken from an internal pool of page headers (`phpool').
64 1.3 pk *
65 1.1 pk */
66 1.1 pk
67 1.3 pk /* List of all pools */
68 1.5 thorpej TAILQ_HEAD(,pool) pool_head = TAILQ_HEAD_INITIALIZER(pool_head);
69 1.3 pk
70 1.3 pk /* Private pool for page header structures */
71 1.3 pk static struct pool phpool;
72 1.3 pk
73 1.3 pk /* # of seconds to retain page after last use */
74 1.3 pk int pool_inactive_time = 10;
75 1.3 pk
76 1.3 pk /* Next candidate for drainage (see pool_drain()) */
77 1.3 pk static struct pool *drainpp = NULL;
78 1.3 pk
79 1.3 pk struct pool_item_header {
80 1.3 pk /* Page headers */
81 1.3 pk TAILQ_ENTRY(pool_item_header)
82 1.3 pk ph_pagelist; /* pool page list */
83 1.3 pk TAILQ_HEAD(,pool_item) ph_itemlist; /* chunk list for this page */
84 1.3 pk LIST_ENTRY(pool_item_header)
85 1.3 pk ph_hashlist; /* Off-page page headers */
86 1.3 pk int ph_nmissing; /* # of chunks in use */
87 1.3 pk caddr_t ph_page; /* this page's address */
88 1.3 pk struct timeval ph_time; /* last referenced */
89 1.3 pk };
90 1.3 pk
91 1.1 pk struct pool_item {
92 1.3 pk #ifdef DIAGNOSTIC
93 1.3 pk int pi_magic;
94 1.3 pk #define PI_MAGIC 0xdeadbeef
95 1.3 pk #endif
96 1.3 pk /* Other entries use only this list entry */
97 1.3 pk TAILQ_ENTRY(pool_item) pi_list;
98 1.3 pk };
99 1.3 pk
100 1.3 pk
101 1.3 pk #define PR_HASH_INDEX(pp,addr) \
102 1.3 pk (((u_long)(addr) >> (pp)->pr_pageshift) & (PR_HASHTABSIZE - 1))
103 1.3 pk
104 1.3 pk
105 1.3 pk
106 1.3 pk static struct pool_item_header
107 1.3 pk *pr_find_pagehead __P((struct pool *, caddr_t));
108 1.3 pk static void pr_rmpage __P((struct pool *, struct pool_item_header *));
109 1.3 pk static int pool_prime_page __P((struct pool *, caddr_t));
110 1.3 pk static void *pool_page_alloc __P((unsigned long, int, int));
111 1.3 pk static void pool_page_free __P((void *, unsigned long, int));
112 1.3 pk int pool_chk __P((struct pool *, char *));
113 1.3 pk
114 1.3 pk
115 1.3 pk #ifdef POOL_DIAGNOSTIC
116 1.3 pk /*
117 1.3 pk * Pool log entry. An array of these is allocated in pool_create().
118 1.3 pk */
119 1.3 pk struct pool_log {
120 1.3 pk const char *pl_file;
121 1.3 pk long pl_line;
122 1.3 pk int pl_action;
123 1.3 pk #define PRLOG_GET 1
124 1.3 pk #define PRLOG_PUT 2
125 1.3 pk void *pl_addr;
126 1.1 pk };
127 1.1 pk
128 1.3 pk /* Number of entries in pool log buffers */
129 1.3 pk int pool_logsize = 10;
130 1.1 pk
131 1.3 pk static void pr_log __P((struct pool *, void *, int, const char *, long));
132 1.3 pk static void pr_printlog __P((struct pool *));
133 1.3 pk
134 1.3 pk static __inline__ void
135 1.3 pk pr_log(pp, v, action, file, line)
136 1.3 pk struct pool *pp;
137 1.3 pk void *v;
138 1.3 pk int action;
139 1.3 pk const char *file;
140 1.3 pk long line;
141 1.3 pk {
142 1.3 pk int n = pp->pr_curlogentry;
143 1.3 pk struct pool_log *pl;
144 1.3 pk
145 1.3 pk if ((pp->pr_flags & PR_LOGGING) == 0)
146 1.3 pk return;
147 1.3 pk
148 1.3 pk /*
149 1.3 pk * Fill in the current entry. Wrap around and overwrite
150 1.3 pk * the oldest entry if necessary.
151 1.3 pk */
152 1.3 pk pl = &pp->pr_log[n];
153 1.3 pk pl->pl_file = file;
154 1.3 pk pl->pl_line = line;
155 1.3 pk pl->pl_action = action;
156 1.3 pk pl->pl_addr = v;
157 1.3 pk if (++n >= pp->pr_logsize)
158 1.3 pk n = 0;
159 1.3 pk pp->pr_curlogentry = n;
160 1.3 pk }
161 1.3 pk
162 1.3 pk static void
163 1.3 pk pr_printlog(pp)
164 1.3 pk struct pool *pp;
165 1.3 pk {
166 1.3 pk int i = pp->pr_logsize;
167 1.3 pk int n = pp->pr_curlogentry;
168 1.3 pk
169 1.3 pk if ((pp->pr_flags & PR_LOGGING) == 0)
170 1.3 pk return;
171 1.3 pk
172 1.3 pk pool_print(pp, "printlog");
173 1.3 pk
174 1.3 pk /*
175 1.3 pk * Print all entries in this pool's log.
176 1.3 pk */
177 1.3 pk while (i-- > 0) {
178 1.3 pk struct pool_log *pl = &pp->pr_log[n];
179 1.3 pk if (pl->pl_action != 0) {
180 1.3 pk printf("log entry %d:\n", i);
181 1.3 pk printf("\taction = %s, addr = %p\n",
182 1.3 pk pl->pl_action == PRLOG_GET ? "get" : "put",
183 1.3 pk pl->pl_addr);
184 1.3 pk printf("\tfile: %s at line %lu\n",
185 1.3 pk pl->pl_file, pl->pl_line);
186 1.3 pk }
187 1.3 pk if (++n >= pp->pr_logsize)
188 1.3 pk n = 0;
189 1.3 pk }
190 1.3 pk }
191 1.3 pk #else
192 1.3 pk #define pr_log(pp, v, action, file, line)
193 1.3 pk #define pr_printlog(pp)
194 1.3 pk #endif
195 1.3 pk
196 1.3 pk
197 1.3 pk /*
198 1.3 pk * Return the pool page header based on page address.
199 1.3 pk */
200 1.3 pk static __inline__ struct pool_item_header *
201 1.3 pk pr_find_pagehead(pp, page)
202 1.3 pk struct pool *pp;
203 1.3 pk caddr_t page;
204 1.3 pk {
205 1.3 pk struct pool_item_header *ph;
206 1.3 pk
207 1.3 pk if ((pp->pr_flags & PR_PHINPAGE) != 0)
208 1.3 pk return ((struct pool_item_header *)(page + pp->pr_phoffset));
209 1.3 pk
210 1.3 pk for (ph = LIST_FIRST(&pp->pr_hashtab[PR_HASH_INDEX(pp, page)]);
211 1.3 pk ph != NULL;
212 1.3 pk ph = LIST_NEXT(ph, ph_hashlist)) {
213 1.3 pk if (ph->ph_page == page)
214 1.3 pk return (ph);
215 1.3 pk }
216 1.3 pk return (NULL);
217 1.3 pk }
218 1.3 pk
219 1.3 pk /*
220 1.3 pk * Remove a page from the pool.
221 1.3 pk */
222 1.3 pk static __inline__ void
223 1.3 pk pr_rmpage(pp, ph)
224 1.3 pk struct pool *pp;
225 1.3 pk struct pool_item_header *ph;
226 1.3 pk {
227 1.3 pk
228 1.3 pk /*
229 1.7 thorpej * If the page was idle, decrement the idle page count.
230 1.3 pk */
231 1.6 thorpej if (ph->ph_nmissing == 0) {
232 1.6 thorpej #ifdef DIAGNOSTIC
233 1.6 thorpej if (pp->pr_nidle == 0)
234 1.6 thorpej panic("pr_rmpage: nidle inconsistent");
235 1.6 thorpej #endif
236 1.6 thorpej pp->pr_nidle--;
237 1.6 thorpej }
238 1.7 thorpej
239 1.7 thorpej /*
240 1.7 thorpej * Unlink a page from the pool and release it.
241 1.7 thorpej */
242 1.7 thorpej TAILQ_REMOVE(&pp->pr_pagelist, ph, ph_pagelist);
243 1.7 thorpej (*pp->pr_free)(ph->ph_page, pp->pr_pagesz, pp->pr_mtype);
244 1.7 thorpej pp->pr_npages--;
245 1.7 thorpej pp->pr_npagefree++;
246 1.6 thorpej
247 1.3 pk if ((pp->pr_flags & PR_PHINPAGE) == 0) {
248 1.3 pk LIST_REMOVE(ph, ph_hashlist);
249 1.3 pk pool_put(&phpool, ph);
250 1.3 pk }
251 1.3 pk
252 1.3 pk if (pp->pr_curpage == ph) {
253 1.3 pk /*
254 1.3 pk * Find a new non-empty page header, if any.
255 1.3 pk * Start search from the page head, to increase the
256 1.3 pk * chance for "high water" pages to be freed.
257 1.3 pk */
258 1.3 pk for (ph = TAILQ_FIRST(&pp->pr_pagelist); ph != NULL;
259 1.3 pk ph = TAILQ_NEXT(ph, ph_pagelist))
260 1.3 pk if (TAILQ_FIRST(&ph->ph_itemlist) != NULL)
261 1.3 pk break;
262 1.3 pk
263 1.3 pk pp->pr_curpage = ph;
264 1.3 pk }
265 1.3 pk }
266 1.3 pk
267 1.3 pk /*
268 1.3 pk * Allocate and initialize a pool.
269 1.3 pk */
270 1.1 pk struct pool *
271 1.3 pk pool_create(size, align, ioff, nitems, wchan, pagesz, alloc, release, mtype)
272 1.1 pk size_t size;
273 1.3 pk u_int align;
274 1.3 pk u_int ioff;
275 1.1 pk int nitems;
276 1.1 pk char *wchan;
277 1.3 pk size_t pagesz;
278 1.3 pk void *(*alloc) __P((unsigned long, int, int));
279 1.3 pk void (*release) __P((void *, unsigned long, int));
280 1.1 pk int mtype;
281 1.1 pk {
282 1.1 pk struct pool *pp;
283 1.3 pk int flags;
284 1.1 pk
285 1.3 pk pp = (struct pool *)malloc(sizeof(*pp), M_POOL, M_NOWAIT);
286 1.3 pk if (pp == NULL)
287 1.1 pk return (NULL);
288 1.3 pk
289 1.3 pk flags = PR_FREEHEADER;
290 1.3 pk #ifdef POOL_DIAGNOSTIC
291 1.3 pk if (pool_logsize != 0)
292 1.3 pk flags |= PR_LOGGING;
293 1.3 pk #endif
294 1.3 pk
295 1.3 pk pool_init(pp, size, align, ioff, flags, wchan, pagesz,
296 1.3 pk alloc, release, mtype);
297 1.3 pk
298 1.3 pk if (nitems != 0) {
299 1.3 pk if (pool_prime(pp, nitems, NULL) != 0) {
300 1.3 pk pool_destroy(pp);
301 1.3 pk return (NULL);
302 1.3 pk }
303 1.1 pk }
304 1.1 pk
305 1.3 pk return (pp);
306 1.3 pk }
307 1.3 pk
308 1.3 pk /*
309 1.3 pk * Initialize the given pool resource structure.
310 1.3 pk *
311 1.3 pk * We export this routine to allow other kernel parts to declare
312 1.3 pk * static pools that must be initialized before malloc() is available.
313 1.3 pk */
314 1.3 pk void
315 1.3 pk pool_init(pp, size, align, ioff, flags, wchan, pagesz, alloc, release, mtype)
316 1.3 pk struct pool *pp;
317 1.3 pk size_t size;
318 1.3 pk u_int align;
319 1.3 pk u_int ioff;
320 1.3 pk int flags;
321 1.3 pk char *wchan;
322 1.3 pk size_t pagesz;
323 1.3 pk void *(*alloc) __P((unsigned long, int, int));
324 1.3 pk void (*release) __P((void *, unsigned long, int));
325 1.3 pk int mtype;
326 1.3 pk {
327 1.3 pk int off, slack;
328 1.3 pk
329 1.3 pk /*
330 1.3 pk * Check arguments and construct default values.
331 1.3 pk */
332 1.3 pk if (!powerof2(pagesz) || pagesz > PAGE_SIZE)
333 1.3 pk panic("pool_init: page size invalid (%lx)\n", (u_long)pagesz);
334 1.3 pk
335 1.4 thorpej if (alloc == NULL && release == NULL) {
336 1.3 pk alloc = pool_page_alloc;
337 1.3 pk release = pool_page_free;
338 1.4 thorpej pagesz = PAGE_SIZE; /* Rounds to PAGE_SIZE anyhow. */
339 1.4 thorpej } else if ((alloc != NULL && release != NULL) == 0) {
340 1.4 thorpej /* If you specifiy one, must specify both. */
341 1.4 thorpej panic("pool_init: must specify alloc and release together");
342 1.4 thorpej }
343 1.4 thorpej
344 1.3 pk if (pagesz == 0)
345 1.3 pk pagesz = PAGE_SIZE;
346 1.3 pk
347 1.3 pk if (align == 0)
348 1.3 pk align = ALIGN(1);
349 1.3 pk
350 1.3 pk /*
351 1.3 pk * Initialize the pool structure.
352 1.3 pk */
353 1.3 pk TAILQ_INSERT_TAIL(&pool_head, pp, pr_poollist);
354 1.3 pk TAILQ_INIT(&pp->pr_pagelist);
355 1.3 pk pp->pr_curpage = NULL;
356 1.3 pk pp->pr_npages = 0;
357 1.3 pk pp->pr_minitems = 0;
358 1.3 pk pp->pr_minpages = 0;
359 1.3 pk pp->pr_maxpages = UINT_MAX;
360 1.3 pk pp->pr_flags = flags;
361 1.3 pk pp->pr_size = ALIGN(size);
362 1.3 pk pp->pr_align = align;
363 1.3 pk pp->pr_wchan = wchan;
364 1.3 pk pp->pr_mtype = mtype;
365 1.3 pk pp->pr_alloc = alloc;
366 1.3 pk pp->pr_free = release;
367 1.3 pk pp->pr_pagesz = pagesz;
368 1.3 pk pp->pr_pagemask = ~(pagesz - 1);
369 1.3 pk pp->pr_pageshift = ffs(pagesz) - 1;
370 1.3 pk
371 1.3 pk /*
372 1.3 pk * Decide whether to put the page header off page to avoid
373 1.3 pk * wasting too large a part of the page. Off-page page headers
374 1.3 pk * go on a hash table, so we can match a returned item
375 1.3 pk * with its header based on the page address.
376 1.3 pk * We use 1/16 of the page size as the threshold (XXX: tune)
377 1.3 pk */
378 1.3 pk if (pp->pr_size < pagesz/16) {
379 1.3 pk /* Use the end of the page for the page header */
380 1.3 pk pp->pr_flags |= PR_PHINPAGE;
381 1.3 pk pp->pr_phoffset = off =
382 1.3 pk pagesz - ALIGN(sizeof(struct pool_item_header));
383 1.2 pk } else {
384 1.3 pk /* The page header will be taken from our page header pool */
385 1.3 pk pp->pr_phoffset = 0;
386 1.3 pk off = pagesz;
387 1.9 perry memset(pp->pr_hashtab, 0, sizeof(pp->pr_hashtab));
388 1.2 pk }
389 1.1 pk
390 1.3 pk /*
391 1.3 pk * Alignment is to take place at `ioff' within the item. This means
392 1.3 pk * we must reserve up to `align - 1' bytes on the page to allow
393 1.3 pk * appropriate positioning of each item.
394 1.3 pk *
395 1.3 pk * Silently enforce `0 <= ioff < align'.
396 1.3 pk */
397 1.3 pk pp->pr_itemoffset = ioff = ioff % align;
398 1.3 pk pp->pr_itemsperpage = (off - ((align - ioff) % align)) / pp->pr_size;
399 1.3 pk
400 1.3 pk /*
401 1.3 pk * Use the slack between the chunks and the page header
402 1.3 pk * for "cache coloring".
403 1.3 pk */
404 1.3 pk slack = off - pp->pr_itemsperpage * pp->pr_size;
405 1.3 pk pp->pr_maxcolor = (slack / align) * align;
406 1.3 pk pp->pr_curcolor = 0;
407 1.3 pk
408 1.3 pk pp->pr_nget = 0;
409 1.3 pk pp->pr_nfail = 0;
410 1.3 pk pp->pr_nput = 0;
411 1.3 pk pp->pr_npagealloc = 0;
412 1.3 pk pp->pr_npagefree = 0;
413 1.1 pk pp->pr_hiwat = 0;
414 1.8 thorpej pp->pr_nidle = 0;
415 1.3 pk
416 1.3 pk #ifdef POOL_DIAGNOSTIC
417 1.3 pk if ((flags & PR_LOGGING) != 0) {
418 1.3 pk pp->pr_log = malloc(pool_logsize * sizeof(struct pool_log),
419 1.3 pk M_TEMP, M_NOWAIT);
420 1.3 pk if (pp->pr_log == NULL)
421 1.3 pk pp->pr_flags &= ~PR_LOGGING;
422 1.3 pk pp->pr_curlogentry = 0;
423 1.3 pk pp->pr_logsize = pool_logsize;
424 1.3 pk }
425 1.3 pk #endif
426 1.3 pk
427 1.1 pk simple_lock_init(&pp->pr_lock);
428 1.1 pk
429 1.3 pk /*
430 1.3 pk * Initialize private page header pool if we haven't done so yet.
431 1.3 pk */
432 1.3 pk if (phpool.pr_size == 0) {
433 1.3 pk pool_init(&phpool, sizeof(struct pool_item_header), 0, 0,
434 1.3 pk 0, "phpool", 0, 0, 0, 0);
435 1.1 pk }
436 1.1 pk
437 1.3 pk return;
438 1.1 pk }
439 1.1 pk
440 1.1 pk /*
441 1.1 pk * De-commision a pool resource.
442 1.1 pk */
443 1.1 pk void
444 1.1 pk pool_destroy(pp)
445 1.1 pk struct pool *pp;
446 1.1 pk {
447 1.3 pk struct pool_item_header *ph;
448 1.3 pk
449 1.3 pk #ifdef DIAGNOSTIC
450 1.3 pk if (pp->pr_nget - pp->pr_nput != 0) {
451 1.3 pk pr_printlog(pp);
452 1.3 pk panic("pool_destroy: pool busy: still out: %lu\n",
453 1.3 pk pp->pr_nget - pp->pr_nput);
454 1.3 pk }
455 1.3 pk #endif
456 1.1 pk
457 1.3 pk /* Remove all pages */
458 1.3 pk if ((pp->pr_flags & PR_STATIC) == 0)
459 1.3 pk while ((ph = pp->pr_pagelist.tqh_first) != NULL)
460 1.3 pk pr_rmpage(pp, ph);
461 1.3 pk
462 1.3 pk /* Remove from global pool list */
463 1.3 pk TAILQ_REMOVE(&pool_head, pp, pr_poollist);
464 1.3 pk drainpp = NULL;
465 1.3 pk
466 1.3 pk #ifdef POOL_DIAGNOSTIC
467 1.3 pk if ((pp->pr_flags & PR_LOGGING) != 0)
468 1.3 pk free(pp->pr_log, M_TEMP);
469 1.3 pk #endif
470 1.2 pk
471 1.3 pk if (pp->pr_flags & PR_FREEHEADER)
472 1.3 pk free(pp, M_POOL);
473 1.1 pk }
474 1.1 pk
475 1.1 pk
476 1.1 pk /*
477 1.3 pk * Grab an item from the pool; must be called at appropriate spl level
478 1.1 pk */
479 1.3 pk #ifdef POOL_DIAGNOSTIC
480 1.3 pk void *
481 1.3 pk _pool_get(pp, flags, file, line)
482 1.3 pk struct pool *pp;
483 1.3 pk int flags;
484 1.3 pk const char *file;
485 1.3 pk long line;
486 1.3 pk #else
487 1.1 pk void *
488 1.1 pk pool_get(pp, flags)
489 1.1 pk struct pool *pp;
490 1.1 pk int flags;
491 1.3 pk #endif
492 1.1 pk {
493 1.1 pk void *v;
494 1.1 pk struct pool_item *pi;
495 1.3 pk struct pool_item_header *ph;
496 1.1 pk
497 1.2 pk #ifdef DIAGNOSTIC
498 1.3 pk if ((pp->pr_flags & PR_STATIC) && (flags & PR_MALLOCOK)) {
499 1.3 pk pr_printlog(pp);
500 1.2 pk panic("pool_get: static");
501 1.3 pk }
502 1.2 pk #endif
503 1.2 pk
504 1.1 pk simple_lock(&pp->pr_lock);
505 1.3 pk if (curproc == NULL && (flags & PR_WAITOK) != 0)
506 1.3 pk panic("pool_get: must have NOWAIT");
507 1.1 pk
508 1.3 pk /*
509 1.3 pk * The convention we use is that if `curpage' is not NULL, then
510 1.3 pk * it points at a non-empty bucket. In particular, `curpage'
511 1.3 pk * never points at a page header which has PR_PHINPAGE set and
512 1.3 pk * has no items in its bucket.
513 1.3 pk */
514 1.3 pk again:
515 1.3 pk if ((ph = pp->pr_curpage) == NULL) {
516 1.3 pk void *v = (*pp->pr_alloc)(pp->pr_pagesz, flags, pp->pr_mtype);
517 1.1 pk if (v == NULL) {
518 1.3 pk if (flags & PR_URGENT)
519 1.3 pk panic("pool_get: urgent");
520 1.3 pk if ((flags & PR_WAITOK) == 0) {
521 1.3 pk pp->pr_nfail++;
522 1.3 pk simple_unlock(&pp->pr_lock);
523 1.1 pk return (NULL);
524 1.3 pk }
525 1.3 pk
526 1.1 pk pp->pr_flags |= PR_WANTED;
527 1.1 pk simple_unlock(&pp->pr_lock);
528 1.1 pk tsleep((caddr_t)pp, PSWP, pp->pr_wchan, 0);
529 1.3 pk simple_lock(&pp->pr_lock);
530 1.3 pk } else {
531 1.3 pk pp->pr_npagealloc++;
532 1.3 pk pool_prime_page(pp, v);
533 1.1 pk }
534 1.3 pk
535 1.3 pk goto again;
536 1.3 pk }
537 1.3 pk
538 1.3 pk if ((v = pi = TAILQ_FIRST(&ph->ph_itemlist)) == NULL)
539 1.3 pk panic("pool_get: %s: page empty", pp->pr_wchan);
540 1.3 pk
541 1.3 pk pr_log(pp, v, PRLOG_GET, file, line);
542 1.3 pk
543 1.3 pk #ifdef DIAGNOSTIC
544 1.3 pk if (pi->pi_magic != PI_MAGIC) {
545 1.3 pk pr_printlog(pp);
546 1.3 pk panic("pool_get(%s): free list modified: magic=%x; page %p;"
547 1.3 pk " item addr %p\n",
548 1.3 pk pp->pr_wchan, pi->pi_magic, ph->ph_page, pi);
549 1.3 pk }
550 1.3 pk #endif
551 1.3 pk
552 1.3 pk /*
553 1.3 pk * Remove from item list.
554 1.3 pk */
555 1.3 pk TAILQ_REMOVE(&ph->ph_itemlist, pi, pi_list);
556 1.6 thorpej if (ph->ph_nmissing == 0) {
557 1.6 thorpej #ifdef DIAGNOSTIC
558 1.6 thorpej if (pp->pr_nidle == 0)
559 1.6 thorpej panic("pool_get: nidle inconsistent");
560 1.6 thorpej #endif
561 1.6 thorpej pp->pr_nidle--;
562 1.6 thorpej }
563 1.3 pk ph->ph_nmissing++;
564 1.3 pk if (TAILQ_FIRST(&ph->ph_itemlist) == NULL) {
565 1.3 pk /*
566 1.3 pk * Find a new non-empty page header, if any.
567 1.3 pk * Start search from the page head, to increase
568 1.3 pk * the chance for "high water" pages to be freed.
569 1.3 pk *
570 1.3 pk * First, move the now empty page to the head of
571 1.3 pk * the page list.
572 1.3 pk */
573 1.3 pk TAILQ_REMOVE(&pp->pr_pagelist, ph, ph_pagelist);
574 1.3 pk TAILQ_INSERT_HEAD(&pp->pr_pagelist, ph, ph_pagelist);
575 1.3 pk while ((ph = TAILQ_NEXT(ph, ph_pagelist)) != NULL)
576 1.3 pk if (TAILQ_FIRST(&ph->ph_itemlist) != NULL)
577 1.3 pk break;
578 1.3 pk
579 1.3 pk pp->pr_curpage = ph;
580 1.1 pk }
581 1.3 pk
582 1.3 pk pp->pr_nget++;
583 1.1 pk simple_unlock(&pp->pr_lock);
584 1.1 pk return (v);
585 1.1 pk }
586 1.1 pk
587 1.1 pk /*
588 1.3 pk * Return resource to the pool; must be called at appropriate spl level
589 1.1 pk */
590 1.3 pk #ifdef POOL_DIAGNOSTIC
591 1.3 pk void
592 1.3 pk _pool_put(pp, v, file, line)
593 1.3 pk struct pool *pp;
594 1.3 pk void *v;
595 1.3 pk const char *file;
596 1.3 pk long line;
597 1.3 pk #else
598 1.1 pk void
599 1.1 pk pool_put(pp, v)
600 1.1 pk struct pool *pp;
601 1.1 pk void *v;
602 1.3 pk #endif
603 1.1 pk {
604 1.1 pk struct pool_item *pi = v;
605 1.3 pk struct pool_item_header *ph;
606 1.3 pk caddr_t page;
607 1.3 pk
608 1.3 pk page = (caddr_t)((u_long)v & pp->pr_pagemask);
609 1.1 pk
610 1.1 pk simple_lock(&pp->pr_lock);
611 1.3 pk
612 1.3 pk pr_log(pp, v, PRLOG_PUT, file, line);
613 1.3 pk
614 1.3 pk if ((ph = pr_find_pagehead(pp, page)) == NULL) {
615 1.3 pk pr_printlog(pp);
616 1.3 pk panic("pool_put: %s: page header missing", pp->pr_wchan);
617 1.3 pk }
618 1.3 pk
619 1.3 pk /*
620 1.3 pk * Return to item list.
621 1.3 pk */
622 1.2 pk #ifdef DIAGNOSTIC
623 1.3 pk pi->pi_magic = PI_MAGIC;
624 1.3 pk #endif
625 1.3 pk TAILQ_INSERT_HEAD(&ph->ph_itemlist, pi, pi_list);
626 1.3 pk ph->ph_nmissing--;
627 1.3 pk pp->pr_nput++;
628 1.3 pk
629 1.3 pk /* Cancel "pool empty" condition if it exists */
630 1.3 pk if (pp->pr_curpage == NULL)
631 1.3 pk pp->pr_curpage = ph;
632 1.3 pk
633 1.3 pk if (pp->pr_flags & PR_WANTED) {
634 1.3 pk pp->pr_flags &= ~PR_WANTED;
635 1.3 pk wakeup((caddr_t)pp);
636 1.3 pk simple_unlock(&pp->pr_lock);
637 1.3 pk return;
638 1.3 pk }
639 1.3 pk
640 1.3 pk /*
641 1.3 pk * If this page is now complete, move it to the end of the pagelist.
642 1.3 pk * If this page has just become un-empty, move it the head.
643 1.3 pk */
644 1.3 pk if (ph->ph_nmissing == 0) {
645 1.6 thorpej pp->pr_nidle++;
646 1.3 pk if (pp->pr_npages > pp->pr_maxpages) {
647 1.3 pk #if 0
648 1.3 pk timeout(pool_drain, 0, pool_inactive_time*hz);
649 1.3 pk #else
650 1.3 pk pr_rmpage(pp, ph);
651 1.2 pk #endif
652 1.3 pk } else {
653 1.3 pk TAILQ_REMOVE(&pp->pr_pagelist, ph, ph_pagelist);
654 1.3 pk TAILQ_INSERT_TAIL(&pp->pr_pagelist, ph, ph_pagelist);
655 1.3 pk ph->ph_time = time;
656 1.3 pk
657 1.3 pk /* XXX - update curpage */
658 1.3 pk for (ph = TAILQ_FIRST(&pp->pr_pagelist); ph != NULL;
659 1.3 pk ph = TAILQ_NEXT(ph, ph_pagelist))
660 1.3 pk if (TAILQ_FIRST(&ph->ph_itemlist) != NULL)
661 1.3 pk break;
662 1.1 pk
663 1.3 pk pp->pr_curpage = ph;
664 1.1 pk }
665 1.1 pk }
666 1.3 pk
667 1.1 pk simple_unlock(&pp->pr_lock);
668 1.1 pk }
669 1.1 pk
670 1.1 pk /*
671 1.3 pk * Add N items to the pool.
672 1.1 pk */
673 1.1 pk int
674 1.2 pk pool_prime(pp, n, storage)
675 1.1 pk struct pool *pp;
676 1.1 pk int n;
677 1.2 pk caddr_t storage;
678 1.1 pk {
679 1.3 pk caddr_t cp;
680 1.3 pk int newnitems, newpages;
681 1.2 pk
682 1.2 pk #ifdef DIAGNOSTIC
683 1.2 pk if (storage && !(pp->pr_flags & PR_STATIC))
684 1.2 pk panic("pool_prime: static");
685 1.2 pk /* !storage && static caught below */
686 1.2 pk #endif
687 1.1 pk
688 1.3 pk newnitems = pp->pr_minitems + n;
689 1.3 pk newpages =
690 1.3 pk roundup(pp->pr_itemsperpage,newnitems) / pp->pr_itemsperpage
691 1.3 pk - pp->pr_minpages;
692 1.3 pk
693 1.1 pk simple_lock(&pp->pr_lock);
694 1.3 pk while (newpages-- > 0) {
695 1.3 pk
696 1.2 pk if (pp->pr_flags & PR_STATIC) {
697 1.3 pk cp = storage;
698 1.3 pk storage += pp->pr_pagesz;
699 1.3 pk } else {
700 1.3 pk cp = (*pp->pr_alloc)(pp->pr_pagesz, 0, pp->pr_mtype);
701 1.3 pk }
702 1.2 pk
703 1.3 pk if (cp == NULL) {
704 1.1 pk simple_unlock(&pp->pr_lock);
705 1.1 pk return (ENOMEM);
706 1.1 pk }
707 1.1 pk
708 1.3 pk pool_prime_page(pp, cp);
709 1.3 pk pp->pr_minpages++;
710 1.1 pk }
711 1.3 pk
712 1.3 pk pp->pr_minitems = newnitems;
713 1.3 pk
714 1.3 pk if (pp->pr_minpages >= pp->pr_maxpages)
715 1.3 pk pp->pr_maxpages = pp->pr_minpages + 1; /* XXX */
716 1.3 pk
717 1.1 pk simple_unlock(&pp->pr_lock);
718 1.1 pk return (0);
719 1.1 pk }
720 1.3 pk
721 1.3 pk /*
722 1.3 pk * Add a page worth of items to the pool.
723 1.3 pk */
724 1.3 pk int
725 1.3 pk pool_prime_page(pp, storage)
726 1.3 pk struct pool *pp;
727 1.3 pk caddr_t storage;
728 1.3 pk {
729 1.3 pk struct pool_item *pi;
730 1.3 pk struct pool_item_header *ph;
731 1.3 pk caddr_t cp = storage;
732 1.3 pk unsigned int align = pp->pr_align;
733 1.3 pk unsigned int ioff = pp->pr_itemoffset;
734 1.3 pk int n;
735 1.3 pk
736 1.3 pk if ((pp->pr_flags & PR_PHINPAGE) != 0) {
737 1.3 pk ph = (struct pool_item_header *)(cp + pp->pr_phoffset);
738 1.3 pk } else {
739 1.3 pk ph = pool_get(&phpool, PR_URGENT);
740 1.3 pk LIST_INSERT_HEAD(&pp->pr_hashtab[PR_HASH_INDEX(pp, cp)],
741 1.3 pk ph, ph_hashlist);
742 1.3 pk }
743 1.3 pk
744 1.3 pk /*
745 1.3 pk * Insert page header.
746 1.3 pk */
747 1.3 pk TAILQ_INSERT_HEAD(&pp->pr_pagelist, ph, ph_pagelist);
748 1.3 pk TAILQ_INIT(&ph->ph_itemlist);
749 1.3 pk ph->ph_page = storage;
750 1.3 pk ph->ph_nmissing = 0;
751 1.3 pk ph->ph_time.tv_sec = ph->ph_time.tv_usec = 0;
752 1.3 pk
753 1.6 thorpej pp->pr_nidle++;
754 1.6 thorpej
755 1.3 pk /*
756 1.3 pk * Color this page.
757 1.3 pk */
758 1.3 pk cp = (caddr_t)(cp + pp->pr_curcolor);
759 1.3 pk if ((pp->pr_curcolor += align) > pp->pr_maxcolor)
760 1.3 pk pp->pr_curcolor = 0;
761 1.3 pk
762 1.3 pk /*
763 1.3 pk * Adjust storage to apply aligment to `pr_itemoffset' in each item.
764 1.3 pk */
765 1.3 pk if (ioff != 0)
766 1.3 pk cp = (caddr_t)(cp + (align - ioff));
767 1.3 pk
768 1.3 pk /*
769 1.3 pk * Insert remaining chunks on the bucket list.
770 1.3 pk */
771 1.3 pk n = pp->pr_itemsperpage;
772 1.3 pk
773 1.3 pk while (n--) {
774 1.3 pk pi = (struct pool_item *)cp;
775 1.3 pk
776 1.3 pk /* Insert on page list */
777 1.3 pk TAILQ_INSERT_TAIL(&ph->ph_itemlist, pi, pi_list);
778 1.3 pk #ifdef DIAGNOSTIC
779 1.3 pk pi->pi_magic = PI_MAGIC;
780 1.3 pk #endif
781 1.3 pk cp = (caddr_t)(cp + pp->pr_size);
782 1.3 pk }
783 1.3 pk
784 1.3 pk /*
785 1.3 pk * If the pool was depleted, point at the new page.
786 1.3 pk */
787 1.3 pk if (pp->pr_curpage == NULL)
788 1.3 pk pp->pr_curpage = ph;
789 1.3 pk
790 1.3 pk if (++pp->pr_npages > pp->pr_hiwat)
791 1.3 pk pp->pr_hiwat = pp->pr_npages;
792 1.3 pk
793 1.3 pk return (0);
794 1.3 pk }
795 1.3 pk
796 1.3 pk void
797 1.3 pk pool_setlowat(pp, n)
798 1.3 pk pool_handle_t pp;
799 1.3 pk int n;
800 1.3 pk {
801 1.3 pk pp->pr_minitems = n;
802 1.3 pk if (n == 0) {
803 1.3 pk pp->pr_minpages = 0;
804 1.3 pk return;
805 1.3 pk }
806 1.3 pk pp->pr_minpages =
807 1.3 pk roundup(pp->pr_itemsperpage,n) / pp->pr_itemsperpage;
808 1.3 pk }
809 1.3 pk
810 1.3 pk void
811 1.3 pk pool_sethiwat(pp, n)
812 1.3 pk pool_handle_t pp;
813 1.3 pk int n;
814 1.3 pk {
815 1.3 pk if (n == 0) {
816 1.3 pk pp->pr_maxpages = 0;
817 1.3 pk return;
818 1.3 pk }
819 1.3 pk pp->pr_maxpages =
820 1.3 pk roundup(pp->pr_itemsperpage,n) / pp->pr_itemsperpage;
821 1.3 pk }
822 1.3 pk
823 1.3 pk
824 1.3 pk /*
825 1.3 pk * Default page allocator.
826 1.3 pk */
827 1.3 pk static void *
828 1.3 pk pool_page_alloc(sz, flags, mtype)
829 1.3 pk unsigned long sz;
830 1.3 pk int flags;
831 1.3 pk int mtype;
832 1.3 pk {
833 1.11 thorpej boolean_t waitok = (flags & PR_WAITOK) ? TRUE : FALSE;
834 1.3 pk
835 1.3 pk #if defined(UVM)
836 1.11 thorpej return ((void *)uvm_km_alloc_poolpage(waitok));
837 1.4 thorpej #else
838 1.11 thorpej return ((void *)kmem_alloc_poolpage(waitok));
839 1.4 thorpej #endif
840 1.3 pk }
841 1.3 pk
842 1.3 pk static void
843 1.3 pk pool_page_free(v, sz, mtype)
844 1.3 pk void *v;
845 1.3 pk unsigned long sz;
846 1.3 pk int mtype;
847 1.3 pk {
848 1.3 pk
849 1.3 pk #if defined(UVM)
850 1.10 eeh uvm_km_free_poolpage((vaddr_t)v);
851 1.3 pk #else
852 1.10 eeh kmem_free_poolpage((vaddr_t)v);
853 1.3 pk #endif
854 1.3 pk }
855 1.3 pk
856 1.3 pk /*
857 1.3 pk * Release all complete pages that have not been used recently.
858 1.3 pk */
859 1.3 pk void
860 1.3 pk pool_reclaim (pp)
861 1.3 pk pool_handle_t pp;
862 1.3 pk {
863 1.3 pk struct pool_item_header *ph, *phnext;
864 1.3 pk struct timeval curtime = time;
865 1.3 pk
866 1.3 pk if (pp->pr_flags & PR_STATIC)
867 1.3 pk return;
868 1.3 pk
869 1.3 pk if (simple_lock_try(&pp->pr_lock) == 0)
870 1.3 pk return;
871 1.3 pk
872 1.3 pk for (ph = TAILQ_FIRST(&pp->pr_pagelist); ph != NULL; ph = phnext) {
873 1.3 pk phnext = TAILQ_NEXT(ph, ph_pagelist);
874 1.3 pk
875 1.3 pk /* Check our minimum page claim */
876 1.3 pk if (pp->pr_npages <= pp->pr_minpages)
877 1.3 pk break;
878 1.3 pk
879 1.3 pk if (ph->ph_nmissing == 0) {
880 1.3 pk struct timeval diff;
881 1.3 pk timersub(&curtime, &ph->ph_time, &diff);
882 1.3 pk if (diff.tv_sec < pool_inactive_time)
883 1.3 pk continue;
884 1.3 pk pr_rmpage(pp, ph);
885 1.3 pk }
886 1.3 pk }
887 1.3 pk
888 1.3 pk simple_unlock(&pp->pr_lock);
889 1.3 pk }
890 1.3 pk
891 1.3 pk
892 1.3 pk /*
893 1.3 pk * Drain pools, one at a time.
894 1.3 pk */
895 1.3 pk void
896 1.3 pk pool_drain(arg)
897 1.3 pk void *arg;
898 1.3 pk {
899 1.3 pk struct pool *pp;
900 1.3 pk int s = splimp();
901 1.3 pk
902 1.3 pk /* XXX:lock pool head */
903 1.3 pk if (drainpp == NULL && (drainpp = TAILQ_FIRST(&pool_head)) == NULL) {
904 1.3 pk splx(s);
905 1.3 pk return;
906 1.3 pk }
907 1.3 pk
908 1.3 pk pp = drainpp;
909 1.3 pk drainpp = TAILQ_NEXT(pp, pr_poollist);
910 1.3 pk /* XXX:unlock pool head */
911 1.3 pk
912 1.3 pk pool_reclaim(pp);
913 1.3 pk splx(s);
914 1.3 pk }
915 1.3 pk
916 1.3 pk
917 1.3 pk #ifdef DEBUG
918 1.3 pk /*
919 1.3 pk * Diagnostic helpers.
920 1.3 pk */
921 1.3 pk void
922 1.3 pk pool_print(pp, label)
923 1.3 pk struct pool *pp;
924 1.3 pk char *label;
925 1.3 pk {
926 1.3 pk
927 1.3 pk if (label != NULL)
928 1.3 pk printf("%s: ", label);
929 1.3 pk
930 1.3 pk printf("pool %s: nalloc %lu nfree %lu npagealloc %lu npagefree %lu\n"
931 1.6 thorpej " npages %u minitems %u itemsperpage %u itemoffset %u\n"
932 1.6 thorpej " nidle %lu\n",
933 1.3 pk pp->pr_wchan,
934 1.3 pk pp->pr_nget,
935 1.3 pk pp->pr_nput,
936 1.3 pk pp->pr_npagealloc,
937 1.3 pk pp->pr_npagefree,
938 1.3 pk pp->pr_npages,
939 1.3 pk pp->pr_minitems,
940 1.3 pk pp->pr_itemsperpage,
941 1.6 thorpej pp->pr_itemoffset,
942 1.6 thorpej pp->pr_nidle);
943 1.3 pk }
944 1.3 pk
945 1.3 pk int
946 1.3 pk pool_chk(pp, label)
947 1.3 pk struct pool *pp;
948 1.3 pk char *label;
949 1.3 pk {
950 1.3 pk struct pool_item_header *ph;
951 1.3 pk int r = 0;
952 1.3 pk
953 1.3 pk simple_lock(&pp->pr_lock);
954 1.3 pk
955 1.3 pk for (ph = TAILQ_FIRST(&pp->pr_pagelist); ph != NULL;
956 1.3 pk ph = TAILQ_NEXT(ph, ph_pagelist)) {
957 1.3 pk
958 1.3 pk struct pool_item *pi;
959 1.3 pk int n;
960 1.3 pk caddr_t page;
961 1.3 pk
962 1.3 pk page = (caddr_t)((u_long)ph & pp->pr_pagemask);
963 1.3 pk if (page != ph->ph_page) {
964 1.3 pk if (label != NULL)
965 1.3 pk printf("%s: ", label);
966 1.3 pk printf("pool(%s): page inconsistency: page %p;"
967 1.3 pk " at page head addr %p (p %p)\n",
968 1.3 pk pp->pr_wchan, ph->ph_page,
969 1.3 pk ph, page);
970 1.3 pk r++;
971 1.3 pk goto out;
972 1.3 pk }
973 1.3 pk
974 1.3 pk for (pi = TAILQ_FIRST(&ph->ph_itemlist), n = 0;
975 1.3 pk pi != NULL;
976 1.3 pk pi = TAILQ_NEXT(pi,pi_list), n++) {
977 1.3 pk
978 1.3 pk #ifdef DIAGNOSTIC
979 1.3 pk if (pi->pi_magic != PI_MAGIC) {
980 1.3 pk if (label != NULL)
981 1.3 pk printf("%s: ", label);
982 1.3 pk printf("pool(%s): free list modified: magic=%x;"
983 1.3 pk " page %p; item ordinal %d;"
984 1.3 pk " addr %p (p %p)\n",
985 1.3 pk pp->pr_wchan, pi->pi_magic, ph->ph_page,
986 1.3 pk n, pi, page);
987 1.3 pk panic("pool");
988 1.3 pk }
989 1.3 pk #endif
990 1.3 pk page = (caddr_t)((u_long)pi & pp->pr_pagemask);
991 1.3 pk if (page == ph->ph_page)
992 1.3 pk continue;
993 1.3 pk
994 1.3 pk if (label != NULL)
995 1.3 pk printf("%s: ", label);
996 1.3 pk printf("pool(%s): page inconsistency: page %p;"
997 1.3 pk " item ordinal %d; addr %p (p %p)\n",
998 1.3 pk pp->pr_wchan, ph->ph_page,
999 1.3 pk n, pi, page);
1000 1.3 pk r++;
1001 1.3 pk goto out;
1002 1.3 pk }
1003 1.3 pk }
1004 1.3 pk out:
1005 1.3 pk simple_unlock(&pp->pr_lock);
1006 1.3 pk return (r);
1007 1.3 pk }
1008 1.3 pk #endif
1009