subr_extent.c revision 1.51 1 1.51 bouyer /* $NetBSD: subr_extent.c,v 1.51 2005/03/15 18:22:24 bouyer Exp $ */
2 1.1 thorpej
3 1.1 thorpej /*-
4 1.11 thorpej * Copyright (c) 1996, 1998 The NetBSD Foundation, Inc.
5 1.1 thorpej * All rights reserved.
6 1.1 thorpej *
7 1.1 thorpej * This code is derived from software contributed to The NetBSD Foundation
8 1.1 thorpej * by Jason R. Thorpe and Matthias Drochner.
9 1.1 thorpej *
10 1.1 thorpej * Redistribution and use in source and binary forms, with or without
11 1.1 thorpej * modification, are permitted provided that the following conditions
12 1.1 thorpej * are met:
13 1.1 thorpej * 1. Redistributions of source code must retain the above copyright
14 1.1 thorpej * notice, this list of conditions and the following disclaimer.
15 1.1 thorpej * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 thorpej * notice, this list of conditions and the following disclaimer in the
17 1.1 thorpej * documentation and/or other materials provided with the distribution.
18 1.1 thorpej * 3. All advertising materials mentioning features or use of this software
19 1.1 thorpej * must display the following acknowledgement:
20 1.20 christos * This product includes software developed by the NetBSD
21 1.20 christos * Foundation, Inc. and its contributors.
22 1.1 thorpej * 4. Neither the name of The NetBSD Foundation nor the names of its
23 1.1 thorpej * contributors may be used to endorse or promote products derived
24 1.1 thorpej * from this software without specific prior written permission.
25 1.1 thorpej *
26 1.1 thorpej * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27 1.1 thorpej * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28 1.1 thorpej * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 1.10 jtc * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30 1.10 jtc * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31 1.1 thorpej * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32 1.1 thorpej * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33 1.1 thorpej * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34 1.1 thorpej * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35 1.1 thorpej * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36 1.1 thorpej * POSSIBILITY OF SUCH DAMAGE.
37 1.1 thorpej */
38 1.1 thorpej
39 1.1 thorpej /*
40 1.1 thorpej * General purpose extent manager.
41 1.1 thorpej */
42 1.44 lukem
43 1.44 lukem #include <sys/cdefs.h>
44 1.51 bouyer __KERNEL_RCSID(0, "$NetBSD: subr_extent.c,v 1.51 2005/03/15 18:22:24 bouyer Exp $");
45 1.1 thorpej
46 1.7 cgd #ifdef _KERNEL
47 1.49 martin #include "opt_lockdebug.h"
48 1.49 martin
49 1.1 thorpej #include <sys/param.h>
50 1.1 thorpej #include <sys/extent.h>
51 1.1 thorpej #include <sys/malloc.h>
52 1.14 pk #include <sys/pool.h>
53 1.1 thorpej #include <sys/time.h>
54 1.1 thorpej #include <sys/systm.h>
55 1.1 thorpej #include <sys/proc.h>
56 1.12 thorpej #include <sys/lock.h>
57 1.34 mrg
58 1.34 mrg #include <uvm/uvm_extern.h>
59 1.29 thorpej
60 1.29 thorpej #define KMEM_IS_RUNNING (kmem_map != NULL)
61 1.18 pk #elif defined(_EXTENT_TESTING)
62 1.7 cgd /*
63 1.7 cgd * user-land definitions, so it can fit into a testing harness.
64 1.7 cgd */
65 1.7 cgd #include <sys/param.h>
66 1.18 pk #include <sys/pool.h>
67 1.7 cgd #include <sys/extent.h>
68 1.7 cgd #include <errno.h>
69 1.7 cgd #include <stdlib.h>
70 1.7 cgd #include <stdio.h>
71 1.28 ross #include <string.h>
72 1.7 cgd
73 1.30 jhawk /*
74 1.30 jhawk * Use multi-line #defines to avoid screwing up the kernel tags file;
75 1.30 jhawk * without this, ctags produces a tags file where panic() shows up
76 1.30 jhawk * in subr_extent.c rather than subr_prf.c.
77 1.30 jhawk */
78 1.30 jhawk #define \
79 1.30 jhawk malloc(s, t, flags) malloc(s)
80 1.30 jhawk #define \
81 1.30 jhawk free(p, t) free(p)
82 1.30 jhawk #define \
83 1.30 jhawk tsleep(chan, pri, str, timo) (EWOULDBLOCK)
84 1.38 augustss #define \
85 1.38 augustss ltsleep(chan,pri,str,timo,lck) (EWOULDBLOCK)
86 1.30 jhawk #define \
87 1.30 jhawk wakeup(chan) ((void)0)
88 1.30 jhawk #define \
89 1.41 thorpej pool_get(pool, flags) malloc((pool)->pr_size,0,0)
90 1.30 jhawk #define \
91 1.30 jhawk pool_put(pool, rp) free(rp,0)
92 1.30 jhawk #define \
93 1.30 jhawk panic(a) printf(a)
94 1.30 jhawk #define \
95 1.30 jhawk splhigh() (1)
96 1.30 jhawk #define \
97 1.30 jhawk splx(s) ((void)(s))
98 1.29 thorpej
99 1.31 thorpej #define \
100 1.31 thorpej simple_lock_init(l) ((void)(l))
101 1.31 thorpej #define \
102 1.31 thorpej simple_lock(l) ((void)(l))
103 1.31 thorpej #define \
104 1.31 thorpej simple_unlock(l) ((void)(l))
105 1.29 thorpej #define KMEM_IS_RUNNING (1)
106 1.7 cgd #endif
107 1.1 thorpej
108 1.50 junyoung static void extent_insert_and_optimize(struct extent *, u_long, u_long,
109 1.50 junyoung int, struct extent_region *, struct extent_region *);
110 1.1 thorpej static struct extent_region *extent_alloc_region_descriptor
111 1.50 junyoung (struct extent *, int);
112 1.50 junyoung static void extent_free_region_descriptor(struct extent *,
113 1.50 junyoung struct extent_region *);
114 1.1 thorpej
115 1.41 thorpej static struct pool expool;
116 1.41 thorpej static struct simplelock expool_init_slock = SIMPLELOCK_INITIALIZER;
117 1.41 thorpej static int expool_initialized;
118 1.14 pk
119 1.1 thorpej /*
120 1.6 thorpej * Macro to align to an arbitrary power-of-two boundary.
121 1.6 thorpej */
122 1.19 pk #define EXTENT_ALIGN(_start, _align, _skew) \
123 1.19 pk (((((_start) - (_skew)) + ((_align) - 1)) & (-(_align))) + (_skew))
124 1.6 thorpej
125 1.6 thorpej /*
126 1.15 sommerfe * Create the extent_region pool.
127 1.15 sommerfe * (This is deferred until one of our callers thinks we can malloc()).
128 1.15 sommerfe */
129 1.15 sommerfe
130 1.41 thorpej static __inline void
131 1.41 thorpej expool_init(void)
132 1.15 sommerfe {
133 1.41 thorpej
134 1.41 thorpej simple_lock(&expool_init_slock);
135 1.41 thorpej if (expool_initialized) {
136 1.41 thorpej simple_unlock(&expool_init_slock);
137 1.41 thorpej return;
138 1.41 thorpej }
139 1.41 thorpej
140 1.18 pk #if defined(_KERNEL)
141 1.41 thorpej pool_init(&expool, sizeof(struct extent_region), 0, 0, 0,
142 1.46 thorpej "extent", NULL);
143 1.18 pk #else
144 1.41 thorpej expool.pr_size = sizeof(struct extent_region);
145 1.18 pk #endif
146 1.41 thorpej
147 1.41 thorpej expool_initialized = 1;
148 1.41 thorpej simple_unlock(&expool_init_slock);
149 1.15 sommerfe }
150 1.15 sommerfe
151 1.15 sommerfe /*
152 1.1 thorpej * Allocate and initialize an extent map.
153 1.1 thorpej */
154 1.1 thorpej struct extent *
155 1.1 thorpej extent_create(name, start, end, mtype, storage, storagesize, flags)
156 1.11 thorpej const char *name;
157 1.1 thorpej u_long start, end;
158 1.48 thorpej struct malloc_type *mtype;
159 1.1 thorpej caddr_t storage;
160 1.1 thorpej size_t storagesize;
161 1.1 thorpej int flags;
162 1.1 thorpej {
163 1.1 thorpej struct extent *ex;
164 1.1 thorpej caddr_t cp = storage;
165 1.1 thorpej size_t sz = storagesize;
166 1.1 thorpej struct extent_region *rp;
167 1.1 thorpej int fixed_extent = (storage != NULL);
168 1.26 pk int s;
169 1.1 thorpej
170 1.6 thorpej #ifdef DIAGNOSTIC
171 1.1 thorpej /* Check arguments. */
172 1.1 thorpej if (name == NULL)
173 1.1 thorpej panic("extent_create: name == NULL");
174 1.1 thorpej if (end < start) {
175 1.5 christos printf("extent_create: extent `%s', start 0x%lx, end 0x%lx\n",
176 1.1 thorpej name, start, end);
177 1.1 thorpej panic("extent_create: end < start");
178 1.1 thorpej }
179 1.1 thorpej if (fixed_extent && (storagesize < sizeof(struct extent_fixed)))
180 1.22 thorpej panic("extent_create: fixed extent, bad storagesize 0x%lx",
181 1.22 thorpej (u_long)storagesize);
182 1.6 thorpej if (fixed_extent == 0 && (storagesize != 0 || storage != NULL))
183 1.6 thorpej panic("extent_create: storage provided for non-fixed");
184 1.6 thorpej #endif
185 1.1 thorpej
186 1.1 thorpej /* Allocate extent descriptor. */
187 1.1 thorpej if (fixed_extent) {
188 1.1 thorpej struct extent_fixed *fex;
189 1.1 thorpej
190 1.16 perry memset(storage, 0, storagesize);
191 1.1 thorpej
192 1.1 thorpej /*
193 1.1 thorpej * Align all descriptors on "long" boundaries.
194 1.1 thorpej */
195 1.1 thorpej fex = (struct extent_fixed *)cp;
196 1.1 thorpej ex = (struct extent *)fex;
197 1.6 thorpej cp += ALIGN(sizeof(struct extent_fixed));
198 1.6 thorpej sz -= ALIGN(sizeof(struct extent_fixed));
199 1.1 thorpej fex->fex_storage = storage;
200 1.1 thorpej fex->fex_storagesize = storagesize;
201 1.1 thorpej
202 1.1 thorpej /*
203 1.1 thorpej * In a fixed extent, we have to pre-allocate region
204 1.1 thorpej * descriptors and place them in the extent's freelist.
205 1.1 thorpej */
206 1.1 thorpej LIST_INIT(&fex->fex_freelist);
207 1.6 thorpej while (sz >= ALIGN(sizeof(struct extent_region))) {
208 1.1 thorpej rp = (struct extent_region *)cp;
209 1.6 thorpej cp += ALIGN(sizeof(struct extent_region));
210 1.6 thorpej sz -= ALIGN(sizeof(struct extent_region));
211 1.1 thorpej LIST_INSERT_HEAD(&fex->fex_freelist, rp, er_link);
212 1.1 thorpej }
213 1.1 thorpej } else {
214 1.26 pk s = splhigh();
215 1.41 thorpej if (expool_initialized == 0)
216 1.41 thorpej expool_init();
217 1.26 pk splx(s);
218 1.15 sommerfe
219 1.1 thorpej ex = (struct extent *)malloc(sizeof(struct extent),
220 1.1 thorpej mtype, (flags & EX_WAITOK) ? M_WAITOK : M_NOWAIT);
221 1.1 thorpej if (ex == NULL)
222 1.1 thorpej return (NULL);
223 1.1 thorpej }
224 1.1 thorpej
225 1.1 thorpej /* Fill in the extent descriptor and return it to the caller. */
226 1.12 thorpej simple_lock_init(&ex->ex_slock);
227 1.1 thorpej LIST_INIT(&ex->ex_regions);
228 1.1 thorpej ex->ex_name = name;
229 1.1 thorpej ex->ex_start = start;
230 1.1 thorpej ex->ex_end = end;
231 1.1 thorpej ex->ex_mtype = mtype;
232 1.1 thorpej ex->ex_flags = 0;
233 1.1 thorpej if (fixed_extent)
234 1.1 thorpej ex->ex_flags |= EXF_FIXED;
235 1.6 thorpej if (flags & EX_NOCOALESCE)
236 1.6 thorpej ex->ex_flags |= EXF_NOCOALESCE;
237 1.1 thorpej return (ex);
238 1.1 thorpej }
239 1.1 thorpej
240 1.1 thorpej /*
241 1.1 thorpej * Destroy an extent map.
242 1.21 chs * Since we're freeing the data, there can't be any references
243 1.21 chs * so we don't need any locking.
244 1.1 thorpej */
245 1.1 thorpej void
246 1.1 thorpej extent_destroy(ex)
247 1.1 thorpej struct extent *ex;
248 1.1 thorpej {
249 1.1 thorpej struct extent_region *rp, *orp;
250 1.1 thorpej
251 1.6 thorpej #ifdef DIAGNOSTIC
252 1.1 thorpej /* Check arguments. */
253 1.1 thorpej if (ex == NULL)
254 1.1 thorpej panic("extent_destroy: NULL extent");
255 1.6 thorpej #endif
256 1.12 thorpej
257 1.1 thorpej /* Free all region descriptors in extent. */
258 1.47 matt for (rp = LIST_FIRST(&ex->ex_regions); rp != NULL; ) {
259 1.1 thorpej orp = rp;
260 1.47 matt rp = LIST_NEXT(rp, er_link);
261 1.1 thorpej LIST_REMOVE(orp, er_link);
262 1.1 thorpej extent_free_region_descriptor(ex, orp);
263 1.1 thorpej }
264 1.1 thorpej
265 1.1 thorpej /* If we're not a fixed extent, free the extent descriptor itself. */
266 1.1 thorpej if ((ex->ex_flags & EXF_FIXED) == 0)
267 1.6 thorpej free(ex, ex->ex_mtype);
268 1.1 thorpej }
269 1.1 thorpej
270 1.1 thorpej /*
271 1.1 thorpej * Insert a region descriptor into the sorted region list after the
272 1.1 thorpej * entry "after" or at the head of the list (if "after" is NULL).
273 1.6 thorpej * The region descriptor we insert is passed in "rp". We must
274 1.6 thorpej * allocate the region descriptor before calling this function!
275 1.6 thorpej * If we don't need the region descriptor, it will be freed here.
276 1.1 thorpej */
277 1.6 thorpej static void
278 1.6 thorpej extent_insert_and_optimize(ex, start, size, flags, after, rp)
279 1.1 thorpej struct extent *ex;
280 1.1 thorpej u_long start, size;
281 1.1 thorpej int flags;
282 1.6 thorpej struct extent_region *after, *rp;
283 1.1 thorpej {
284 1.7 cgd struct extent_region *nextr;
285 1.1 thorpej int appended = 0;
286 1.1 thorpej
287 1.1 thorpej if (after == NULL) {
288 1.1 thorpej /*
289 1.1 thorpej * We're the first in the region list. If there's
290 1.1 thorpej * a region after us, attempt to coalesce to save
291 1.1 thorpej * descriptor overhead.
292 1.1 thorpej */
293 1.6 thorpej if (((ex->ex_flags & EXF_NOCOALESCE) == 0) &&
294 1.47 matt (LIST_FIRST(&ex->ex_regions) != NULL) &&
295 1.47 matt ((start + size) == LIST_FIRST(&ex->ex_regions)->er_start)) {
296 1.1 thorpej /*
297 1.1 thorpej * We can coalesce. Prepend us to the first region.
298 1.1 thorpej */
299 1.47 matt LIST_FIRST(&ex->ex_regions)->er_start = start;
300 1.6 thorpej extent_free_region_descriptor(ex, rp);
301 1.6 thorpej return;
302 1.1 thorpej }
303 1.1 thorpej
304 1.1 thorpej /*
305 1.6 thorpej * Can't coalesce. Fill in the region descriptor
306 1.1 thorpej * in, and insert us at the head of the region list.
307 1.1 thorpej */
308 1.1 thorpej rp->er_start = start;
309 1.1 thorpej rp->er_end = start + (size - 1);
310 1.1 thorpej LIST_INSERT_HEAD(&ex->ex_regions, rp, er_link);
311 1.6 thorpej return;
312 1.1 thorpej }
313 1.1 thorpej
314 1.1 thorpej /*
315 1.6 thorpej * If EXF_NOCOALESCE is set, coalescing is disallowed.
316 1.1 thorpej */
317 1.6 thorpej if (ex->ex_flags & EXF_NOCOALESCE)
318 1.6 thorpej goto cant_coalesce;
319 1.1 thorpej
320 1.1 thorpej /*
321 1.1 thorpej * Attempt to coalesce with the region before us.
322 1.1 thorpej */
323 1.1 thorpej if ((after->er_end + 1) == start) {
324 1.1 thorpej /*
325 1.1 thorpej * We can coalesce. Append ourselves and make
326 1.1 thorpej * note of it.
327 1.1 thorpej */
328 1.1 thorpej after->er_end = start + (size - 1);
329 1.1 thorpej appended = 1;
330 1.1 thorpej }
331 1.1 thorpej
332 1.1 thorpej /*
333 1.1 thorpej * Attempt to coalesce with the region after us.
334 1.1 thorpej */
335 1.47 matt if ((LIST_NEXT(after, er_link) != NULL) &&
336 1.47 matt ((start + size) == LIST_NEXT(after, er_link)->er_start)) {
337 1.1 thorpej /*
338 1.1 thorpej * We can coalesce. Note that if we appended ourselves
339 1.1 thorpej * to the previous region, we exactly fit the gap, and
340 1.1 thorpej * can free the "next" region descriptor.
341 1.1 thorpej */
342 1.1 thorpej if (appended) {
343 1.1 thorpej /*
344 1.1 thorpej * Yup, we can free it up.
345 1.1 thorpej */
346 1.47 matt after->er_end = LIST_NEXT(after, er_link)->er_end;
347 1.47 matt nextr = LIST_NEXT(after, er_link);
348 1.7 cgd LIST_REMOVE(nextr, er_link);
349 1.7 cgd extent_free_region_descriptor(ex, nextr);
350 1.1 thorpej } else {
351 1.1 thorpej /*
352 1.1 thorpej * Nope, just prepend us to the next region.
353 1.1 thorpej */
354 1.47 matt LIST_NEXT(after, er_link)->er_start = start;
355 1.1 thorpej }
356 1.6 thorpej
357 1.6 thorpej extent_free_region_descriptor(ex, rp);
358 1.6 thorpej return;
359 1.1 thorpej }
360 1.1 thorpej
361 1.1 thorpej /*
362 1.1 thorpej * We weren't able to coalesce with the next region, but
363 1.1 thorpej * we don't need to allocate a region descriptor if we
364 1.1 thorpej * appended ourselves to the previous region.
365 1.1 thorpej */
366 1.6 thorpej if (appended) {
367 1.6 thorpej extent_free_region_descriptor(ex, rp);
368 1.6 thorpej return;
369 1.6 thorpej }
370 1.1 thorpej
371 1.6 thorpej cant_coalesce:
372 1.1 thorpej
373 1.1 thorpej /*
374 1.6 thorpej * Fill in the region descriptor and insert ourselves
375 1.1 thorpej * into the region list.
376 1.1 thorpej */
377 1.1 thorpej rp->er_start = start;
378 1.1 thorpej rp->er_end = start + (size - 1);
379 1.1 thorpej LIST_INSERT_AFTER(after, rp, er_link);
380 1.1 thorpej }
381 1.1 thorpej
382 1.1 thorpej /*
383 1.1 thorpej * Allocate a specific region in an extent map.
384 1.1 thorpej */
385 1.1 thorpej int
386 1.1 thorpej extent_alloc_region(ex, start, size, flags)
387 1.1 thorpej struct extent *ex;
388 1.1 thorpej u_long start, size;
389 1.1 thorpej int flags;
390 1.1 thorpej {
391 1.6 thorpej struct extent_region *rp, *last, *myrp;
392 1.1 thorpej u_long end = start + (size - 1);
393 1.1 thorpej int error;
394 1.1 thorpej
395 1.6 thorpej #ifdef DIAGNOSTIC
396 1.1 thorpej /* Check arguments. */
397 1.1 thorpej if (ex == NULL)
398 1.1 thorpej panic("extent_alloc_region: NULL extent");
399 1.1 thorpej if (size < 1) {
400 1.5 christos printf("extent_alloc_region: extent `%s', size 0x%lx\n",
401 1.1 thorpej ex->ex_name, size);
402 1.1 thorpej panic("extent_alloc_region: bad size");
403 1.1 thorpej }
404 1.1 thorpej if (end < start) {
405 1.5 christos printf(
406 1.1 thorpej "extent_alloc_region: extent `%s', start 0x%lx, size 0x%lx\n",
407 1.1 thorpej ex->ex_name, start, size);
408 1.1 thorpej panic("extent_alloc_region: overflow");
409 1.1 thorpej }
410 1.6 thorpej #endif
411 1.42 thorpej #ifdef LOCKDEBUG
412 1.42 thorpej if (flags & EX_WAITSPACE)
413 1.42 thorpej simple_lock_only_held(NULL,
414 1.42 thorpej "extent_alloc_region(EX_WAITSPACE)");
415 1.42 thorpej #endif
416 1.6 thorpej
417 1.1 thorpej /*
418 1.1 thorpej * Make sure the requested region lies within the
419 1.1 thorpej * extent.
420 1.12 thorpej *
421 1.12 thorpej * We don't lock to check the range, because those values
422 1.12 thorpej * are never modified, and if another thread deletes the
423 1.12 thorpej * extent, we're screwed anyway.
424 1.1 thorpej */
425 1.1 thorpej if ((start < ex->ex_start) || (end > ex->ex_end)) {
426 1.6 thorpej #ifdef DIAGNOSTIC
427 1.5 christos printf("extent_alloc_region: extent `%s' (0x%lx - 0x%lx)\n",
428 1.1 thorpej ex->ex_name, ex->ex_start, ex->ex_end);
429 1.5 christos printf("extent_alloc_region: start 0x%lx, end 0x%lx\n",
430 1.1 thorpej start, end);
431 1.1 thorpej panic("extent_alloc_region: region lies outside extent");
432 1.6 thorpej #else
433 1.6 thorpej return (EINVAL);
434 1.6 thorpej #endif
435 1.6 thorpej }
436 1.6 thorpej
437 1.6 thorpej /*
438 1.6 thorpej * Allocate the region descriptor. It will be freed later
439 1.12 thorpej * if we can coalesce with another region. Don't lock before
440 1.12 thorpej * here! This could block.
441 1.6 thorpej */
442 1.6 thorpej myrp = extent_alloc_region_descriptor(ex, flags);
443 1.6 thorpej if (myrp == NULL) {
444 1.6 thorpej #ifdef DIAGNOSTIC
445 1.6 thorpej printf(
446 1.6 thorpej "extent_alloc_region: can't allocate region descriptor\n");
447 1.6 thorpej #endif
448 1.6 thorpej return (ENOMEM);
449 1.1 thorpej }
450 1.1 thorpej
451 1.1 thorpej alloc_start:
452 1.12 thorpej simple_lock(&ex->ex_slock);
453 1.12 thorpej
454 1.1 thorpej /*
455 1.1 thorpej * Attempt to place ourselves in the desired area of the
456 1.1 thorpej * extent. We save ourselves some work by keeping the list sorted.
457 1.1 thorpej * In other words, if the start of the current region is greater
458 1.1 thorpej * than the end of our region, we don't have to search any further.
459 1.1 thorpej */
460 1.1 thorpej
461 1.1 thorpej /*
462 1.1 thorpej * Keep a pointer to the last region we looked at so
463 1.1 thorpej * that we don't have to traverse the list again when
464 1.1 thorpej * we insert ourselves. If "last" is NULL when we
465 1.1 thorpej * finally insert ourselves, we go at the head of the
466 1.1 thorpej * list. See extent_insert_and_optimize() for details.
467 1.1 thorpej */
468 1.1 thorpej last = NULL;
469 1.1 thorpej
470 1.47 matt LIST_FOREACH(rp, &ex->ex_regions, er_link) {
471 1.1 thorpej if (rp->er_start > end) {
472 1.1 thorpej /*
473 1.1 thorpej * We lie before this region and don't
474 1.1 thorpej * conflict.
475 1.1 thorpej */
476 1.9 thorpej break;
477 1.1 thorpej }
478 1.1 thorpej
479 1.1 thorpej /*
480 1.1 thorpej * The current region begins before we end.
481 1.1 thorpej * Check for a conflict.
482 1.1 thorpej */
483 1.1 thorpej if (rp->er_end >= start) {
484 1.1 thorpej /*
485 1.6 thorpej * We conflict. If we can (and want to) wait,
486 1.6 thorpej * do so.
487 1.1 thorpej */
488 1.6 thorpej if (flags & EX_WAITSPACE) {
489 1.1 thorpej ex->ex_flags |= EXF_WANTED;
490 1.37 sommerfe error = ltsleep(ex,
491 1.37 sommerfe PNORELOCK | PRIBIO | ((flags & EX_CATCH) ? PCATCH : 0),
492 1.37 sommerfe "extnt", 0, &ex->ex_slock);
493 1.1 thorpej if (error)
494 1.1 thorpej return (error);
495 1.1 thorpej goto alloc_start;
496 1.1 thorpej }
497 1.6 thorpej extent_free_region_descriptor(ex, myrp);
498 1.12 thorpej simple_unlock(&ex->ex_slock);
499 1.1 thorpej return (EAGAIN);
500 1.1 thorpej }
501 1.1 thorpej /*
502 1.1 thorpej * We don't conflict, but this region lies before
503 1.1 thorpej * us. Keep a pointer to this region, and keep
504 1.1 thorpej * trying.
505 1.1 thorpej */
506 1.1 thorpej last = rp;
507 1.1 thorpej }
508 1.1 thorpej
509 1.1 thorpej /*
510 1.1 thorpej * We don't conflict with any regions. "last" points
511 1.1 thorpej * to the region we fall after, or is NULL if we belong
512 1.1 thorpej * at the beginning of the region list. Insert ourselves.
513 1.1 thorpej */
514 1.6 thorpej extent_insert_and_optimize(ex, start, size, flags, last, myrp);
515 1.12 thorpej simple_unlock(&ex->ex_slock);
516 1.6 thorpej return (0);
517 1.1 thorpej }
518 1.1 thorpej
519 1.1 thorpej /*
520 1.1 thorpej * Macro to check (x + y) <= z. This check is designed to fail
521 1.1 thorpej * if an overflow occurs.
522 1.1 thorpej */
523 1.1 thorpej #define LE_OV(x, y, z) ((((x) + (y)) >= (x)) && (((x) + (y)) <= (z)))
524 1.1 thorpej
525 1.1 thorpej /*
526 1.1 thorpej * Allocate a region in an extent map subregion.
527 1.1 thorpej *
528 1.1 thorpej * If EX_FAST is specified, we return the first fit in the map.
529 1.1 thorpej * Otherwise, we try to minimize fragmentation by finding the
530 1.1 thorpej * smallest gap that will hold the request.
531 1.1 thorpej *
532 1.1 thorpej * The allocated region is aligned to "alignment", which must be
533 1.1 thorpej * a power of 2.
534 1.1 thorpej */
535 1.1 thorpej int
536 1.19 pk extent_alloc_subregion1(ex, substart, subend, size, alignment, skew, boundary,
537 1.1 thorpej flags, result)
538 1.1 thorpej struct extent *ex;
539 1.19 pk u_long substart, subend, size, alignment, skew, boundary;
540 1.1 thorpej int flags;
541 1.1 thorpej u_long *result;
542 1.1 thorpej {
543 1.6 thorpej struct extent_region *rp, *myrp, *last, *bestlast;
544 1.43 enami u_long newstart, newend, exend, beststart, bestovh, ovh;
545 1.23 mycroft u_long dontcross;
546 1.1 thorpej int error;
547 1.1 thorpej
548 1.6 thorpej #ifdef DIAGNOSTIC
549 1.12 thorpej /*
550 1.12 thorpej * Check arguments.
551 1.12 thorpej *
552 1.12 thorpej * We don't lock to check these, because these values
553 1.12 thorpej * are never modified, and if another thread deletes the
554 1.12 thorpej * extent, we're screwed anyway.
555 1.12 thorpej */
556 1.1 thorpej if (ex == NULL)
557 1.1 thorpej panic("extent_alloc_subregion: NULL extent");
558 1.1 thorpej if (result == NULL)
559 1.1 thorpej panic("extent_alloc_subregion: NULL result pointer");
560 1.8 thorpej if ((substart < ex->ex_start) || (substart > ex->ex_end) ||
561 1.8 thorpej (subend > ex->ex_end) || (subend < ex->ex_start)) {
562 1.5 christos printf("extent_alloc_subregion: extent `%s', ex_start 0x%lx, ex_end 0x%lx\n",
563 1.1 thorpej ex->ex_name, ex->ex_start, ex->ex_end);
564 1.5 christos printf("extent_alloc_subregion: substart 0x%lx, subend 0x%lx\n",
565 1.1 thorpej substart, subend);
566 1.1 thorpej panic("extent_alloc_subregion: bad subregion");
567 1.1 thorpej }
568 1.8 thorpej if ((size < 1) || ((size - 1) > (subend - substart))) {
569 1.5 christos printf("extent_alloc_subregion: extent `%s', size 0x%lx\n",
570 1.1 thorpej ex->ex_name, size);
571 1.1 thorpej panic("extent_alloc_subregion: bad size");
572 1.1 thorpej }
573 1.1 thorpej if (alignment == 0)
574 1.1 thorpej panic("extent_alloc_subregion: bad alignment");
575 1.1 thorpej if (boundary && (boundary < size)) {
576 1.5 christos printf(
577 1.40 marcus "extent_alloc_subregion: extent `%s', size 0x%lx, "
578 1.40 marcus "boundary 0x%lx\n", ex->ex_name, size, boundary);
579 1.1 thorpej panic("extent_alloc_subregion: bad boundary");
580 1.1 thorpej }
581 1.42 thorpej #endif
582 1.42 thorpej #ifdef LOCKDEBUG
583 1.42 thorpej if (flags & EX_WAITSPACE)
584 1.42 thorpej simple_lock_only_held(NULL,
585 1.42 thorpej "extent_alloc_subregion1(EX_WAITSPACE)");
586 1.6 thorpej #endif
587 1.6 thorpej
588 1.6 thorpej /*
589 1.6 thorpej * Allocate the region descriptor. It will be freed later
590 1.12 thorpej * if we can coalesce with another region. Don't lock before
591 1.12 thorpej * here! This could block.
592 1.6 thorpej */
593 1.6 thorpej myrp = extent_alloc_region_descriptor(ex, flags);
594 1.6 thorpej if (myrp == NULL) {
595 1.6 thorpej #ifdef DIAGNOSTIC
596 1.6 thorpej printf(
597 1.6 thorpej "extent_alloc_subregion: can't allocate region descriptor\n");
598 1.6 thorpej #endif
599 1.6 thorpej return (ENOMEM);
600 1.6 thorpej }
601 1.1 thorpej
602 1.1 thorpej alloc_start:
603 1.12 thorpej simple_lock(&ex->ex_slock);
604 1.12 thorpej
605 1.1 thorpej /*
606 1.1 thorpej * Keep a pointer to the last region we looked at so
607 1.1 thorpej * that we don't have to traverse the list again when
608 1.1 thorpej * we insert ourselves. If "last" is NULL when we
609 1.1 thorpej * finally insert ourselves, we go at the head of the
610 1.1 thorpej * list. See extent_insert_and_optimize() for deatails.
611 1.1 thorpej */
612 1.1 thorpej last = NULL;
613 1.1 thorpej
614 1.1 thorpej /*
615 1.1 thorpej * Keep track of size and location of the smallest
616 1.1 thorpej * chunk we fit in.
617 1.1 thorpej *
618 1.1 thorpej * Since the extent can be as large as the numeric range
619 1.1 thorpej * of the CPU (0 - 0xffffffff for 32-bit systems), the
620 1.1 thorpej * best overhead value can be the maximum unsigned integer.
621 1.1 thorpej * Thus, we initialize "bestovh" to 0, since we insert ourselves
622 1.1 thorpej * into the region list immediately on an exact match (which
623 1.1 thorpej * is the only case where "bestovh" would be set to 0).
624 1.1 thorpej */
625 1.1 thorpej bestovh = 0;
626 1.1 thorpej beststart = 0;
627 1.1 thorpej bestlast = NULL;
628 1.1 thorpej
629 1.1 thorpej /*
630 1.43 enami * Keep track of end of free region. This is either the end of extent
631 1.43 enami * or the start of a region past the subend.
632 1.43 enami */
633 1.43 enami exend = ex->ex_end;
634 1.43 enami
635 1.43 enami /*
636 1.1 thorpej * For N allocated regions, we must make (N + 1)
637 1.1 thorpej * checks for unallocated space. The first chunk we
638 1.1 thorpej * check is the area from the beginning of the subregion
639 1.9 thorpej * to the first allocated region after that point.
640 1.1 thorpej */
641 1.19 pk newstart = EXTENT_ALIGN(substart, alignment, skew);
642 1.1 thorpej if (newstart < ex->ex_start) {
643 1.6 thorpej #ifdef DIAGNOSTIC
644 1.5 christos printf(
645 1.1 thorpej "extent_alloc_subregion: extent `%s' (0x%lx - 0x%lx), alignment 0x%lx\n",
646 1.1 thorpej ex->ex_name, ex->ex_start, ex->ex_end, alignment);
647 1.12 thorpej simple_unlock(&ex->ex_slock);
648 1.1 thorpej panic("extent_alloc_subregion: overflow after alignment");
649 1.6 thorpej #else
650 1.6 thorpej extent_free_region_descriptor(ex, myrp);
651 1.12 thorpej simple_unlock(&ex->ex_slock);
652 1.6 thorpej return (EINVAL);
653 1.6 thorpej #endif
654 1.1 thorpej }
655 1.1 thorpej
656 1.9 thorpej /*
657 1.9 thorpej * Find the first allocated region that begins on or after
658 1.9 thorpej * the subregion start, advancing the "last" pointer along
659 1.9 thorpej * the way.
660 1.9 thorpej */
661 1.47 matt LIST_FOREACH(rp, &ex->ex_regions, er_link) {
662 1.9 thorpej if (rp->er_start >= newstart)
663 1.9 thorpej break;
664 1.9 thorpej last = rp;
665 1.9 thorpej }
666 1.17 pk
667 1.17 pk /*
668 1.25 drochner * Relocate the start of our candidate region to the end of
669 1.25 drochner * the last allocated region (if there was one overlapping
670 1.25 drochner * our subrange).
671 1.17 pk */
672 1.25 drochner if (last != NULL && last->er_end >= newstart)
673 1.19 pk newstart = EXTENT_ALIGN((last->er_end + 1), alignment, skew);
674 1.9 thorpej
675 1.47 matt for (; rp != NULL; rp = LIST_NEXT(rp, er_link)) {
676 1.1 thorpej /*
677 1.43 enami * If the region pasts the subend, bail out and see
678 1.43 enami * if we fit against the subend.
679 1.43 enami */
680 1.51 bouyer if (rp->er_start > subend) {
681 1.43 enami exend = rp->er_start;
682 1.43 enami break;
683 1.43 enami }
684 1.43 enami
685 1.43 enami /*
686 1.1 thorpej * Check the chunk before "rp". Note that our
687 1.1 thorpej * comparison is safe from overflow conditions.
688 1.1 thorpej */
689 1.1 thorpej if (LE_OV(newstart, size, rp->er_start)) {
690 1.1 thorpej /*
691 1.1 thorpej * Do a boundary check, if necessary. Note
692 1.1 thorpej * that a region may *begin* on the boundary,
693 1.1 thorpej * but it must end before the boundary.
694 1.1 thorpej */
695 1.1 thorpej if (boundary) {
696 1.1 thorpej newend = newstart + (size - 1);
697 1.1 thorpej
698 1.1 thorpej /*
699 1.23 mycroft * Calculate the next boundary after the start
700 1.23 mycroft * of this region.
701 1.23 mycroft */
702 1.50 junyoung dontcross = EXTENT_ALIGN(newstart+1, boundary,
703 1.23 mycroft (flags & EX_BOUNDZERO) ? 0 : ex->ex_start)
704 1.23 mycroft - 1;
705 1.23 mycroft
706 1.24 mycroft #if 0
707 1.33 pk printf("newstart=%lx newend=%lx ex_start=%lx ex_end=%lx boundary=%lx dontcross=%lx\n",
708 1.24 mycroft newstart, newend, ex->ex_start, ex->ex_end,
709 1.24 mycroft boundary, dontcross);
710 1.24 mycroft #endif
711 1.24 mycroft
712 1.32 mrg /* Check for overflow */
713 1.32 mrg if (dontcross < ex->ex_start)
714 1.32 mrg dontcross = ex->ex_end;
715 1.32 mrg else if (newend > dontcross) {
716 1.24 mycroft /*
717 1.24 mycroft * Candidate region crosses boundary.
718 1.24 mycroft * Throw away the leading part and see
719 1.24 mycroft * if we still fit.
720 1.24 mycroft */
721 1.24 mycroft newstart = dontcross + 1;
722 1.24 mycroft newend = newstart + (size - 1);
723 1.24 mycroft dontcross += boundary;
724 1.24 mycroft if (!LE_OV(newstart, size, rp->er_start))
725 1.45 bouyer goto skip;
726 1.24 mycroft }
727 1.24 mycroft
728 1.23 mycroft /*
729 1.23 mycroft * If we run past the end of
730 1.23 mycroft * the extent or the boundary
731 1.23 mycroft * overflows, then the request
732 1.23 mycroft * can't fit.
733 1.1 thorpej */
734 1.36 mrg if (newstart + size - 1 > ex->ex_end ||
735 1.23 mycroft dontcross < newstart)
736 1.23 mycroft goto fail;
737 1.1 thorpej }
738 1.1 thorpej
739 1.1 thorpej /*
740 1.1 thorpej * We would fit into this space. Calculate
741 1.1 thorpej * the overhead (wasted space). If we exactly
742 1.1 thorpej * fit, or we're taking the first fit, insert
743 1.1 thorpej * ourselves into the region list.
744 1.1 thorpej */
745 1.1 thorpej ovh = rp->er_start - newstart - size;
746 1.1 thorpej if ((flags & EX_FAST) || (ovh == 0))
747 1.1 thorpej goto found;
748 1.1 thorpej
749 1.1 thorpej /*
750 1.1 thorpej * Don't exactly fit, but check to see
751 1.1 thorpej * if we're better than any current choice.
752 1.1 thorpej */
753 1.1 thorpej if ((bestovh == 0) || (ovh < bestovh)) {
754 1.1 thorpej bestovh = ovh;
755 1.1 thorpej beststart = newstart;
756 1.1 thorpej bestlast = last;
757 1.1 thorpej }
758 1.1 thorpej }
759 1.1 thorpej
760 1.45 bouyer skip:
761 1.1 thorpej /*
762 1.1 thorpej * Skip past the current region and check again.
763 1.1 thorpej */
764 1.19 pk newstart = EXTENT_ALIGN((rp->er_end + 1), alignment, skew);
765 1.1 thorpej if (newstart < rp->er_end) {
766 1.1 thorpej /*
767 1.1 thorpej * Overflow condition. Don't error out, since
768 1.1 thorpej * we might have a chunk of space that we can
769 1.1 thorpej * use.
770 1.1 thorpej */
771 1.1 thorpej goto fail;
772 1.1 thorpej }
773 1.50 junyoung
774 1.1 thorpej last = rp;
775 1.1 thorpej }
776 1.1 thorpej
777 1.1 thorpej /*
778 1.1 thorpej * The final check is from the current starting point to the
779 1.1 thorpej * end of the subregion. If there were no allocated regions,
780 1.1 thorpej * "newstart" is set to the beginning of the subregion, or
781 1.1 thorpej * just past the end of the last allocated region, adjusted
782 1.1 thorpej * for alignment in either case.
783 1.1 thorpej */
784 1.1 thorpej if (LE_OV(newstart, (size - 1), subend)) {
785 1.24 mycroft /*
786 1.24 mycroft * Do a boundary check, if necessary. Note
787 1.24 mycroft * that a region may *begin* on the boundary,
788 1.24 mycroft * but it must end before the boundary.
789 1.24 mycroft */
790 1.24 mycroft if (boundary) {
791 1.24 mycroft newend = newstart + (size - 1);
792 1.24 mycroft
793 1.24 mycroft /*
794 1.24 mycroft * Calculate the next boundary after the start
795 1.24 mycroft * of this region.
796 1.24 mycroft */
797 1.50 junyoung dontcross = EXTENT_ALIGN(newstart+1, boundary,
798 1.24 mycroft (flags & EX_BOUNDZERO) ? 0 : ex->ex_start)
799 1.24 mycroft - 1;
800 1.24 mycroft
801 1.24 mycroft #if 0
802 1.33 pk printf("newstart=%lx newend=%lx ex_start=%lx ex_end=%lx boundary=%lx dontcross=%lx\n",
803 1.24 mycroft newstart, newend, ex->ex_start, ex->ex_end,
804 1.24 mycroft boundary, dontcross);
805 1.24 mycroft #endif
806 1.24 mycroft
807 1.32 mrg /* Check for overflow */
808 1.32 mrg if (dontcross < ex->ex_start)
809 1.32 mrg dontcross = ex->ex_end;
810 1.32 mrg else if (newend > dontcross) {
811 1.24 mycroft /*
812 1.24 mycroft * Candidate region crosses boundary.
813 1.24 mycroft * Throw away the leading part and see
814 1.24 mycroft * if we still fit.
815 1.24 mycroft */
816 1.24 mycroft newstart = dontcross + 1;
817 1.24 mycroft newend = newstart + (size - 1);
818 1.24 mycroft dontcross += boundary;
819 1.24 mycroft if (!LE_OV(newstart, (size - 1), subend))
820 1.24 mycroft goto fail;
821 1.24 mycroft }
822 1.24 mycroft
823 1.24 mycroft /*
824 1.24 mycroft * If we run past the end of
825 1.24 mycroft * the extent or the boundary
826 1.24 mycroft * overflows, then the request
827 1.24 mycroft * can't fit.
828 1.24 mycroft */
829 1.36 mrg if (newstart + size - 1 > ex->ex_end ||
830 1.24 mycroft dontcross < newstart)
831 1.24 mycroft goto fail;
832 1.24 mycroft }
833 1.24 mycroft
834 1.1 thorpej /*
835 1.1 thorpej * We would fit into this space. Calculate
836 1.1 thorpej * the overhead (wasted space). If we exactly
837 1.1 thorpej * fit, or we're taking the first fit, insert
838 1.1 thorpej * ourselves into the region list.
839 1.1 thorpej */
840 1.43 enami ovh = exend - newstart - (size - 1);
841 1.1 thorpej if ((flags & EX_FAST) || (ovh == 0))
842 1.1 thorpej goto found;
843 1.1 thorpej
844 1.1 thorpej /*
845 1.1 thorpej * Don't exactly fit, but check to see
846 1.1 thorpej * if we're better than any current choice.
847 1.1 thorpej */
848 1.1 thorpej if ((bestovh == 0) || (ovh < bestovh)) {
849 1.1 thorpej bestovh = ovh;
850 1.1 thorpej beststart = newstart;
851 1.1 thorpej bestlast = last;
852 1.1 thorpej }
853 1.1 thorpej }
854 1.1 thorpej
855 1.1 thorpej fail:
856 1.1 thorpej /*
857 1.1 thorpej * One of the following two conditions have
858 1.1 thorpej * occurred:
859 1.1 thorpej *
860 1.1 thorpej * There is no chunk large enough to hold the request.
861 1.1 thorpej *
862 1.1 thorpej * If EX_FAST was not specified, there is not an
863 1.1 thorpej * exact match for the request.
864 1.1 thorpej *
865 1.1 thorpej * Note that if we reach this point and EX_FAST is
866 1.1 thorpej * set, then we know there is no space in the extent for
867 1.1 thorpej * the request.
868 1.1 thorpej */
869 1.1 thorpej if (((flags & EX_FAST) == 0) && (bestovh != 0)) {
870 1.1 thorpej /*
871 1.1 thorpej * We have a match that's "good enough".
872 1.1 thorpej */
873 1.1 thorpej newstart = beststart;
874 1.1 thorpej last = bestlast;
875 1.1 thorpej goto found;
876 1.1 thorpej }
877 1.1 thorpej
878 1.1 thorpej /*
879 1.1 thorpej * No space currently available. Wait for it to free up,
880 1.1 thorpej * if possible.
881 1.1 thorpej */
882 1.6 thorpej if (flags & EX_WAITSPACE) {
883 1.1 thorpej ex->ex_flags |= EXF_WANTED;
884 1.37 sommerfe error = ltsleep(ex,
885 1.37 sommerfe PNORELOCK | PRIBIO | ((flags & EX_CATCH) ? PCATCH : 0),
886 1.37 sommerfe "extnt", 0, &ex->ex_slock);
887 1.1 thorpej if (error)
888 1.1 thorpej return (error);
889 1.1 thorpej goto alloc_start;
890 1.1 thorpej }
891 1.1 thorpej
892 1.6 thorpej extent_free_region_descriptor(ex, myrp);
893 1.12 thorpej simple_unlock(&ex->ex_slock);
894 1.1 thorpej return (EAGAIN);
895 1.1 thorpej
896 1.1 thorpej found:
897 1.1 thorpej /*
898 1.1 thorpej * Insert ourselves into the region list.
899 1.1 thorpej */
900 1.6 thorpej extent_insert_and_optimize(ex, newstart, size, flags, last, myrp);
901 1.12 thorpej simple_unlock(&ex->ex_slock);
902 1.6 thorpej *result = newstart;
903 1.6 thorpej return (0);
904 1.1 thorpej }
905 1.1 thorpej
906 1.1 thorpej int
907 1.1 thorpej extent_free(ex, start, size, flags)
908 1.1 thorpej struct extent *ex;
909 1.1 thorpej u_long start, size;
910 1.1 thorpej int flags;
911 1.1 thorpej {
912 1.12 thorpej struct extent_region *rp, *nrp = NULL;
913 1.1 thorpej u_long end = start + (size - 1);
914 1.12 thorpej int exflags;
915 1.1 thorpej
916 1.6 thorpej #ifdef DIAGNOSTIC
917 1.12 thorpej /*
918 1.12 thorpej * Check arguments.
919 1.12 thorpej *
920 1.12 thorpej * We don't lock to check these, because these values
921 1.12 thorpej * are never modified, and if another thread deletes the
922 1.12 thorpej * extent, we're screwed anyway.
923 1.12 thorpej */
924 1.1 thorpej if (ex == NULL)
925 1.1 thorpej panic("extent_free: NULL extent");
926 1.51 bouyer if ((start < ex->ex_start) || (end > ex->ex_end)) {
927 1.1 thorpej extent_print(ex);
928 1.5 christos printf("extent_free: extent `%s', start 0x%lx, size 0x%lx\n",
929 1.1 thorpej ex->ex_name, start, size);
930 1.1 thorpej panic("extent_free: extent `%s', region not within extent",
931 1.1 thorpej ex->ex_name);
932 1.1 thorpej }
933 1.1 thorpej /* Check for an overflow. */
934 1.1 thorpej if (end < start) {
935 1.1 thorpej extent_print(ex);
936 1.5 christos printf("extent_free: extent `%s', start 0x%lx, size 0x%lx\n",
937 1.1 thorpej ex->ex_name, start, size);
938 1.1 thorpej panic("extent_free: overflow");
939 1.1 thorpej }
940 1.6 thorpej #endif
941 1.1 thorpej
942 1.1 thorpej /*
943 1.12 thorpej * If we're allowing coalescing, we must allocate a region
944 1.12 thorpej * descriptor now, since it might block.
945 1.12 thorpej *
946 1.12 thorpej * XXX Make a static, create-time flags word, so we don't
947 1.12 thorpej * XXX have to lock to read it!
948 1.12 thorpej */
949 1.12 thorpej simple_lock(&ex->ex_slock);
950 1.12 thorpej exflags = ex->ex_flags;
951 1.12 thorpej simple_unlock(&ex->ex_slock);
952 1.14 pk
953 1.12 thorpej if ((exflags & EXF_NOCOALESCE) == 0) {
954 1.12 thorpej /* Allocate a region descriptor. */
955 1.12 thorpej nrp = extent_alloc_region_descriptor(ex, flags);
956 1.12 thorpej if (nrp == NULL)
957 1.12 thorpej return (ENOMEM);
958 1.12 thorpej }
959 1.12 thorpej
960 1.12 thorpej simple_lock(&ex->ex_slock);
961 1.12 thorpej
962 1.12 thorpej /*
963 1.1 thorpej * Find region and deallocate. Several possibilities:
964 1.1 thorpej *
965 1.1 thorpej * 1. (start == er_start) && (end == er_end):
966 1.1 thorpej * Free descriptor.
967 1.1 thorpej *
968 1.1 thorpej * 2. (start == er_start) && (end < er_end):
969 1.1 thorpej * Adjust er_start.
970 1.1 thorpej *
971 1.1 thorpej * 3. (start > er_start) && (end == er_end):
972 1.1 thorpej * Adjust er_end.
973 1.1 thorpej *
974 1.1 thorpej * 4. (start > er_start) && (end < er_end):
975 1.1 thorpej * Fragment region. Requires descriptor alloc.
976 1.1 thorpej *
977 1.6 thorpej * Cases 2, 3, and 4 require that the EXF_NOCOALESCE flag
978 1.1 thorpej * is not set.
979 1.1 thorpej */
980 1.47 matt LIST_FOREACH(rp, &ex->ex_regions, er_link) {
981 1.1 thorpej /*
982 1.1 thorpej * Save ourselves some comparisons; does the current
983 1.1 thorpej * region end before chunk to be freed begins? If so,
984 1.1 thorpej * then we haven't found the appropriate region descriptor.
985 1.1 thorpej */
986 1.1 thorpej if (rp->er_end < start)
987 1.1 thorpej continue;
988 1.1 thorpej
989 1.1 thorpej /*
990 1.1 thorpej * Save ourselves some traversal; does the current
991 1.1 thorpej * region begin after the chunk to be freed ends? If so,
992 1.1 thorpej * then we've already passed any possible region descriptors
993 1.1 thorpej * that might have contained the chunk to be freed.
994 1.1 thorpej */
995 1.1 thorpej if (rp->er_start > end)
996 1.1 thorpej break;
997 1.1 thorpej
998 1.1 thorpej /* Case 1. */
999 1.1 thorpej if ((start == rp->er_start) && (end == rp->er_end)) {
1000 1.1 thorpej LIST_REMOVE(rp, er_link);
1001 1.1 thorpej extent_free_region_descriptor(ex, rp);
1002 1.1 thorpej goto done;
1003 1.1 thorpej }
1004 1.1 thorpej
1005 1.1 thorpej /*
1006 1.6 thorpej * The following cases all require that EXF_NOCOALESCE
1007 1.1 thorpej * is not set.
1008 1.1 thorpej */
1009 1.6 thorpej if (ex->ex_flags & EXF_NOCOALESCE)
1010 1.1 thorpej continue;
1011 1.1 thorpej
1012 1.1 thorpej /* Case 2. */
1013 1.1 thorpej if ((start == rp->er_start) && (end < rp->er_end)) {
1014 1.1 thorpej rp->er_start = (end + 1);
1015 1.1 thorpej goto done;
1016 1.1 thorpej }
1017 1.1 thorpej
1018 1.1 thorpej /* Case 3. */
1019 1.1 thorpej if ((start > rp->er_start) && (end == rp->er_end)) {
1020 1.1 thorpej rp->er_end = (start - 1);
1021 1.1 thorpej goto done;
1022 1.1 thorpej }
1023 1.1 thorpej
1024 1.1 thorpej /* Case 4. */
1025 1.1 thorpej if ((start > rp->er_start) && (end < rp->er_end)) {
1026 1.1 thorpej /* Fill in new descriptor. */
1027 1.1 thorpej nrp->er_start = end + 1;
1028 1.1 thorpej nrp->er_end = rp->er_end;
1029 1.1 thorpej
1030 1.1 thorpej /* Adjust current descriptor. */
1031 1.1 thorpej rp->er_end = start - 1;
1032 1.1 thorpej
1033 1.13 pk /* Insert new descriptor after current. */
1034 1.1 thorpej LIST_INSERT_AFTER(rp, nrp, er_link);
1035 1.13 pk
1036 1.13 pk /* We used the new descriptor, so don't free it below */
1037 1.13 pk nrp = NULL;
1038 1.1 thorpej goto done;
1039 1.1 thorpej }
1040 1.1 thorpej }
1041 1.1 thorpej
1042 1.1 thorpej /* Region not found, or request otherwise invalid. */
1043 1.12 thorpej simple_unlock(&ex->ex_slock);
1044 1.1 thorpej extent_print(ex);
1045 1.5 christos printf("extent_free: start 0x%lx, end 0x%lx\n", start, end);
1046 1.1 thorpej panic("extent_free: region not found");
1047 1.1 thorpej
1048 1.1 thorpej done:
1049 1.13 pk if (nrp != NULL)
1050 1.13 pk extent_free_region_descriptor(ex, nrp);
1051 1.1 thorpej if (ex->ex_flags & EXF_WANTED) {
1052 1.1 thorpej ex->ex_flags &= ~EXF_WANTED;
1053 1.1 thorpej wakeup(ex);
1054 1.1 thorpej }
1055 1.12 thorpej simple_unlock(&ex->ex_slock);
1056 1.1 thorpej return (0);
1057 1.1 thorpej }
1058 1.1 thorpej
1059 1.12 thorpej /*
1060 1.12 thorpej * Allocate an extent region descriptor. EXTENT MUST NOT BE LOCKED,
1061 1.12 thorpej * AS THIS FUNCTION MAY BLOCK! We will handle any locking we may need.
1062 1.12 thorpej */
1063 1.1 thorpej static struct extent_region *
1064 1.1 thorpej extent_alloc_region_descriptor(ex, flags)
1065 1.1 thorpej struct extent *ex;
1066 1.1 thorpej int flags;
1067 1.1 thorpej {
1068 1.1 thorpej struct extent_region *rp;
1069 1.12 thorpej int exflags;
1070 1.26 pk int s;
1071 1.29 thorpej
1072 1.29 thorpej /*
1073 1.29 thorpej * If the kernel memory allocator is not yet running, we can't
1074 1.29 thorpej * use it (obviously).
1075 1.29 thorpej */
1076 1.29 thorpej if (KMEM_IS_RUNNING == 0)
1077 1.29 thorpej flags &= ~EX_MALLOCOK;
1078 1.12 thorpej
1079 1.12 thorpej /*
1080 1.12 thorpej * XXX Make a static, create-time flags word, so we don't
1081 1.12 thorpej * XXX have to lock to read it!
1082 1.12 thorpej */
1083 1.12 thorpej simple_lock(&ex->ex_slock);
1084 1.12 thorpej exflags = ex->ex_flags;
1085 1.12 thorpej simple_unlock(&ex->ex_slock);
1086 1.1 thorpej
1087 1.12 thorpej if (exflags & EXF_FIXED) {
1088 1.1 thorpej struct extent_fixed *fex = (struct extent_fixed *)ex;
1089 1.1 thorpej
1090 1.12 thorpej for (;;) {
1091 1.12 thorpej simple_lock(&ex->ex_slock);
1092 1.47 matt if ((rp = LIST_FIRST(&fex->fex_freelist)) != NULL) {
1093 1.12 thorpej /*
1094 1.12 thorpej * Don't muck with flags after pulling it off
1095 1.12 thorpej * the freelist; it may have been dynamically
1096 1.12 thorpej * allocated, and kindly given to us. We
1097 1.12 thorpej * need to remember that information.
1098 1.12 thorpej */
1099 1.12 thorpej LIST_REMOVE(rp, er_link);
1100 1.12 thorpej simple_unlock(&ex->ex_slock);
1101 1.12 thorpej return (rp);
1102 1.12 thorpej }
1103 1.12 thorpej if (flags & EX_MALLOCOK) {
1104 1.12 thorpej simple_unlock(&ex->ex_slock);
1105 1.1 thorpej goto alloc;
1106 1.12 thorpej }
1107 1.12 thorpej if ((flags & EX_WAITOK) == 0) {
1108 1.12 thorpej simple_unlock(&ex->ex_slock);
1109 1.1 thorpej return (NULL);
1110 1.12 thorpej }
1111 1.1 thorpej ex->ex_flags |= EXF_FLWANTED;
1112 1.37 sommerfe if (ltsleep(&fex->fex_freelist,
1113 1.37 sommerfe PNORELOCK| PRIBIO | ((flags & EX_CATCH) ? PCATCH : 0),
1114 1.37 sommerfe "extnt", 0, &ex->ex_slock))
1115 1.1 thorpej return (NULL);
1116 1.1 thorpej }
1117 1.1 thorpej }
1118 1.1 thorpej
1119 1.1 thorpej alloc:
1120 1.26 pk s = splhigh();
1121 1.41 thorpej if (expool_initialized == 0)
1122 1.41 thorpej expool_init();
1123 1.41 thorpej rp = pool_get(&expool, (flags & EX_WAITOK) ? PR_WAITOK : 0);
1124 1.26 pk splx(s);
1125 1.1 thorpej
1126 1.7 cgd if (rp != NULL)
1127 1.7 cgd rp->er_flags = ER_ALLOC;
1128 1.1 thorpej
1129 1.1 thorpej return (rp);
1130 1.1 thorpej }
1131 1.1 thorpej
1132 1.12 thorpej /*
1133 1.12 thorpej * Free an extent region descriptor. EXTENT _MUST_ BE LOCKED! This
1134 1.12 thorpej * is safe as we do not block here.
1135 1.12 thorpej */
1136 1.1 thorpej static void
1137 1.1 thorpej extent_free_region_descriptor(ex, rp)
1138 1.1 thorpej struct extent *ex;
1139 1.1 thorpej struct extent_region *rp;
1140 1.1 thorpej {
1141 1.26 pk int s;
1142 1.1 thorpej
1143 1.6 thorpej if (ex->ex_flags & EXF_FIXED) {
1144 1.1 thorpej struct extent_fixed *fex = (struct extent_fixed *)ex;
1145 1.1 thorpej
1146 1.6 thorpej /*
1147 1.6 thorpej * If someone's waiting for a region descriptor,
1148 1.6 thorpej * be nice and give them this one, rather than
1149 1.6 thorpej * just free'ing it back to the system.
1150 1.6 thorpej */
1151 1.6 thorpej if (rp->er_flags & ER_ALLOC) {
1152 1.6 thorpej if (ex->ex_flags & EXF_FLWANTED) {
1153 1.6 thorpej /* Clear all but ER_ALLOC flag. */
1154 1.6 thorpej rp->er_flags = ER_ALLOC;
1155 1.6 thorpej LIST_INSERT_HEAD(&fex->fex_freelist, rp,
1156 1.6 thorpej er_link);
1157 1.6 thorpej goto wake_em_up;
1158 1.6 thorpej } else {
1159 1.26 pk s = splhigh();
1160 1.41 thorpej pool_put(&expool, rp);
1161 1.26 pk splx(s);
1162 1.6 thorpej }
1163 1.6 thorpej } else {
1164 1.6 thorpej /* Clear all flags. */
1165 1.6 thorpej rp->er_flags = 0;
1166 1.6 thorpej LIST_INSERT_HEAD(&fex->fex_freelist, rp, er_link);
1167 1.6 thorpej }
1168 1.6 thorpej
1169 1.1 thorpej if (ex->ex_flags & EXF_FLWANTED) {
1170 1.6 thorpej wake_em_up:
1171 1.1 thorpej ex->ex_flags &= ~EXF_FLWANTED;
1172 1.1 thorpej wakeup(&fex->fex_freelist);
1173 1.1 thorpej }
1174 1.6 thorpej return;
1175 1.6 thorpej }
1176 1.6 thorpej
1177 1.6 thorpej /*
1178 1.6 thorpej * We know it's dynamically allocated if we get here.
1179 1.6 thorpej */
1180 1.26 pk s = splhigh();
1181 1.41 thorpej pool_put(&expool, rp);
1182 1.26 pk splx(s);
1183 1.1 thorpej }
1184 1.1 thorpej
1185 1.1 thorpej void
1186 1.1 thorpej extent_print(ex)
1187 1.1 thorpej struct extent *ex;
1188 1.1 thorpej {
1189 1.1 thorpej struct extent_region *rp;
1190 1.1 thorpej
1191 1.1 thorpej if (ex == NULL)
1192 1.1 thorpej panic("extent_print: NULL extent");
1193 1.1 thorpej
1194 1.12 thorpej simple_lock(&ex->ex_slock);
1195 1.12 thorpej
1196 1.5 christos printf("extent `%s' (0x%lx - 0x%lx), flags = 0x%x\n", ex->ex_name,
1197 1.1 thorpej ex->ex_start, ex->ex_end, ex->ex_flags);
1198 1.1 thorpej
1199 1.47 matt LIST_FOREACH(rp, &ex->ex_regions, er_link)
1200 1.5 christos printf(" 0x%lx - 0x%lx\n", rp->er_start, rp->er_end);
1201 1.12 thorpej
1202 1.12 thorpej simple_unlock(&ex->ex_slock);
1203 1.1 thorpej }
1204