xref: /netbsd-src/sys/kern/subr_bufq.c (revision e5548b402ae4c44fb816de42c7bba9581ce23ef5)
1 /*	$NetBSD: subr_bufq.c,v 1.10 2005/12/11 12:24:30 christos Exp $	*/
2 /*	NetBSD: subr_disk.c,v 1.70 2005/08/20 12:00:01 yamt Exp $	*/
3 
4 /*-
5  * Copyright (c) 1996, 1997, 1999, 2000 The NetBSD Foundation, Inc.
6  * All rights reserved.
7  *
8  * This code is derived from software contributed to The NetBSD Foundation
9  * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
10  * NASA Ames Research Center.
11  *
12  * Redistribution and use in source and binary forms, with or without
13  * modification, are permitted provided that the following conditions
14  * are met:
15  * 1. Redistributions of source code must retain the above copyright
16  *    notice, this list of conditions and the following disclaimer.
17  * 2. Redistributions in binary form must reproduce the above copyright
18  *    notice, this list of conditions and the following disclaimer in the
19  *    documentation and/or other materials provided with the distribution.
20  * 3. All advertising materials mentioning features or use of this software
21  *    must display the following acknowledgement:
22  *	This product includes software developed by the NetBSD
23  *	Foundation, Inc. and its contributors.
24  * 4. Neither the name of The NetBSD Foundation nor the names of its
25  *    contributors may be used to endorse or promote products derived
26  *    from this software without specific prior written permission.
27  *
28  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
29  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
30  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
31  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
32  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
33  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
34  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
35  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
36  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
37  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
38  * POSSIBILITY OF SUCH DAMAGE.
39  */
40 
41 /*
42  * Copyright (c) 1982, 1986, 1988, 1993
43  *	The Regents of the University of California.  All rights reserved.
44  * (c) UNIX System Laboratories, Inc.
45  * All or some portions of this file are derived from material licensed
46  * to the University of California by American Telephone and Telegraph
47  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
48  * the permission of UNIX System Laboratories, Inc.
49  *
50  * Redistribution and use in source and binary forms, with or without
51  * modification, are permitted provided that the following conditions
52  * are met:
53  * 1. Redistributions of source code must retain the above copyright
54  *    notice, this list of conditions and the following disclaimer.
55  * 2. Redistributions in binary form must reproduce the above copyright
56  *    notice, this list of conditions and the following disclaimer in the
57  *    documentation and/or other materials provided with the distribution.
58  * 3. Neither the name of the University nor the names of its contributors
59  *    may be used to endorse or promote products derived from this software
60  *    without specific prior written permission.
61  *
62  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
63  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
64  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
65  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
66  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
67  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
68  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
69  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
70  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
71  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
72  * SUCH DAMAGE.
73  *
74  *	@(#)ufs_disksubr.c	8.5 (Berkeley) 1/21/94
75  */
76 
77 #include <sys/cdefs.h>
78 __KERNEL_RCSID(0, "$NetBSD: subr_bufq.c,v 1.10 2005/12/11 12:24:30 christos Exp $");
79 
80 #include <sys/param.h>
81 #include <sys/systm.h>
82 #include <sys/buf.h>
83 #include <sys/bufq.h>
84 #include <sys/bufq_impl.h>
85 #include <sys/malloc.h>
86 #include <sys/sysctl.h>
87 
88 BUFQ_DEFINE(dummy, 0, NULL); /* so that bufq_strats won't be empty */
89 
90 #define	STRAT_MATCH(id, bs)	(strcmp((id), (bs)->bs_name) == 0)
91 
92 /*
93  * Create a device buffer queue.
94  */
95 int
96 bufq_alloc(struct bufq_state **bufqp, const char *strategy, int flags)
97 {
98 	__link_set_decl(bufq_strats, const struct bufq_strat);
99 	const struct bufq_strat *bsp;
100 	const struct bufq_strat * const *it;
101 	struct bufq_state *bufq;
102 	int error = 0;
103 
104 	KASSERT((flags & BUFQ_EXACT) == 0 || strategy != BUFQ_STRAT_ANY);
105 
106 	switch (flags & BUFQ_SORT_MASK) {
107 	case BUFQ_SORT_RAWBLOCK:
108 	case BUFQ_SORT_CYLINDER:
109 		break;
110 	case 0:
111 		/*
112 		 * for strategies which don't care about block numbers.
113 		 * eg. fcfs
114 		 */
115 		flags |= BUFQ_SORT_RAWBLOCK;
116 		break;
117 	default:
118 		panic("bufq_alloc: sort out of range");
119 	}
120 
121 	/*
122 	 * select strategy.
123 	 * if a strategy specified by flags is found, use it.
124 	 * otherwise, select one with the largest bs_prio.
125 	 */
126 	bsp = NULL;
127 	__link_set_foreach(it, bufq_strats) {
128 		if ((*it) == &bufq_strat_dummy)
129 			continue;
130 		if (strategy != BUFQ_STRAT_ANY &&
131 		    STRAT_MATCH(strategy, (*it))) {
132 			bsp = *it;
133 			break;
134 		}
135 		if (bsp == NULL || (*it)->bs_prio > bsp->bs_prio)
136 			bsp = *it;
137 	}
138 
139 	if (bsp == NULL) {
140 		panic("bufq_alloc: no strategy");
141 	}
142 	if (strategy != BUFQ_STRAT_ANY && !STRAT_MATCH(strategy, bsp)) {
143 		if ((flags & BUFQ_EXACT)) {
144 			error = ENOENT;
145 			goto out;
146 		}
147 #if defined(DEBUG)
148 		printf("bufq_alloc: '%s' is not available. using '%s'.\n",
149 		    strategy, bsp->bs_name);
150 #endif
151 	}
152 #if defined(BUFQ_DEBUG)
153 	/* XXX aprint? */
154 	printf("bufq_alloc: using '%s'\n", bsp->bs_name);
155 #endif
156 
157 	*bufqp = bufq = malloc(sizeof(*bufq), M_DEVBUF, M_WAITOK | M_ZERO);
158 	bufq->bq_flags = flags;
159 	bufq->bq_strat = bsp;
160 	(*bsp->bs_initfn)(bufq);
161 
162 out:
163 	return 0;
164 }
165 
166 void
167 bufq_put(struct bufq_state *bufq, struct buf *bp)
168 {
169 
170 	(*bufq->bq_put)(bufq, bp);
171 }
172 
173 struct buf *
174 bufq_get(struct bufq_state *bufq)
175 {
176 
177 	return (*bufq->bq_get)(bufq, 1);
178 }
179 
180 struct buf *
181 bufq_peek(struct bufq_state *bufq)
182 {
183 
184 	return (*bufq->bq_get)(bufq, 0);
185 }
186 
187 /*
188  * Drain a device buffer queue.
189  */
190 void
191 bufq_drain(struct bufq_state *bufq)
192 {
193 	struct buf *bp;
194 
195 	while ((bp = BUFQ_GET(bufq)) != NULL) {
196 		bp->b_error = EIO;
197 		bp->b_flags |= B_ERROR;
198 		bp->b_resid = bp->b_bcount;
199 		biodone(bp);
200 	}
201 }
202 
203 /*
204  * Destroy a device buffer queue.
205  */
206 void
207 bufq_free(struct bufq_state *bufq)
208 {
209 
210 	KASSERT(bufq->bq_private != NULL);
211 	KASSERT(BUFQ_PEEK(bufq) == NULL);
212 
213 	free(bufq->bq_private, M_DEVBUF);
214 	free(bufq, M_DEVBUF);
215 }
216 
217 #if 0
218 /*
219  * get a strategy identifier of a buffer queue.
220  */
221 const char *
222 bufq_getstrategyname(struct bufq_state *bufq)
223 {
224 
225 	return bufq->bq_strat->bs_name;
226 }
227 
228 /*
229  * move all requests on a buffer queue to another.
230  */
231 void
232 bufq_move(struct bufq_state *dst, struct bufq_state *src)
233 {
234 	struct buf *bp;
235 
236 	while ((bp = BUFQ_GET(src)) != NULL) {
237 		BUFQ_PUT(dst, bp);
238 	}
239 }
240 #endif
241 
242 static int
243 docopy(char *buf, size_t *bufoffp, size_t buflen,
244     const char *datap, size_t datalen)
245 {
246 	int error = 0;
247 
248 	if (buf != NULL && datalen > 0) {
249 
250 		if (*bufoffp + datalen > buflen) {
251 			goto out;
252 		}
253 		error = copyout(datap, buf + *bufoffp, datalen);
254 		if (error) {
255 			goto out;
256 		}
257 	}
258 out:
259 	if (error == 0) {
260 		*bufoffp += datalen;
261 	}
262 
263 	return error;
264 }
265 
266 static int
267 docopystr(char *buf, size_t *bufoffp, size_t buflen, const char *datap)
268 {
269 
270 	return docopy(buf, bufoffp, buflen, datap, strlen(datap));
271 }
272 
273 static int
274 docopynul(char *buf, size_t *bufoffp, size_t buflen)
275 {
276 
277 	return docopy(buf, bufoffp, buflen, "", 1);
278 }
279 
280 /*
281  * sysctl function that will print all bufq strategies
282  * built in the kernel.
283  */
284 static int
285 sysctl_kern_bufq_strategies(SYSCTLFN_ARGS)
286 {
287 	__link_set_decl(bufq_strats, const struct bufq_strat);
288 	const struct bufq_strat * const *bq_strat;
289 	const char *delim = "";
290 	size_t off = 0;
291 	size_t buflen = *oldlenp;
292 	int error;
293 
294 	__link_set_foreach(bq_strat, bufq_strats) {
295 		if ((*bq_strat) == &bufq_strat_dummy) {
296 			continue;
297 		}
298 		error = docopystr(oldp, &off, buflen, delim);
299 		if (error) {
300 			goto out;
301 		}
302 		error = docopystr(oldp, &off, buflen, (*bq_strat)->bs_name);
303 		if (error) {
304 			goto out;
305 		}
306 		delim = " ";
307 	}
308 
309 	/* NUL terminate */
310 	error = docopynul(oldp, &off, buflen);
311 out:
312 	*oldlenp = off;
313 	return error;
314 }
315 
316 SYSCTL_SETUP(sysctl_kern_bufq_strategies_setup, "sysctl kern.bufq tree setup")
317 {
318 	const struct sysctlnode *node;
319 
320 	sysctl_createv(clog, 0, NULL, NULL,
321 			CTLFLAG_PERMANENT,
322 			CTLTYPE_NODE, "kern", NULL,
323 			NULL, 0, NULL, 0,
324 			CTL_KERN, CTL_EOL);
325 	node = NULL;
326 	sysctl_createv(clog, 0, NULL, &node,
327 			CTLFLAG_PERMANENT,
328 			CTLTYPE_NODE, "bufq",
329 			SYSCTL_DESCR("buffer queue subtree"),
330 			NULL, 0, NULL, 0,
331 			CTL_KERN, CTL_CREATE, CTL_EOL);
332 	if (node != NULL) {
333 		sysctl_createv(clog, 0, NULL, NULL,
334 			CTLFLAG_PERMANENT,
335 			CTLTYPE_STRING, "strategies",
336 			SYSCTL_DESCR("List of bufq strategies present"),
337 			sysctl_kern_bufq_strategies, 0, NULL, 0,
338 			CTL_KERN, node->sysctl_num, CTL_CREATE, CTL_EOL);
339 	}
340 }
341