Merge branch 'vendor/LIBARCHIVE'
[dragonfly.git] / sys / kern / kern_iosched.c
1 /*
2  * Copyright (c) 2008 The DragonFly Project.  All rights reserved.
3  * 
4  * This code is derived from software contributed to The DragonFly Project
5  * by Matthew Dillon <dillon@backplane.com>
6  * 
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in
15  *    the documentation and/or other materials provided with the
16  *    distribution.
17  * 3. Neither the name of The DragonFly Project nor the names of its
18  *    contributors may be used to endorse or promote products derived
19  *    from this software without specific, prior written permission.
20  * 
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  * 
34  * $DragonFly: src/sys/kern/kern_iosched.c,v 1.1 2008/06/28 17:59:49 dillon Exp $
35  */
36
37 #include <sys/param.h>
38 #include <sys/systm.h>
39 #include <sys/kernel.h>
40 #include <sys/proc.h>
41 #include <sys/rtprio.h>
42 #include <sys/queue.h>
43 #include <machine/cpu.h>
44 #include <sys/spinlock.h>
45 #include <sys/iosched.h>
46 #include <sys/sysctl.h>
47 #include <sys/buf.h>
48 #include <sys/limits.h>
49
50 #include <sys/thread2.h>
51 #include <sys/spinlock2.h>
52
53 #include <vm/vm.h>
54 #include <vm/vm_param.h>
55 #include <vm/vm_kern.h>
56 #include <vm/vm_extern.h>
57
58 SYSCTL_NODE(, OID_AUTO, iosched, CTLFLAG_RW, 0, "I/O Scheduler");
59
60 static int iosched_debug = 0;
61 SYSCTL_INT(_iosched, OID_AUTO, debug, CTLFLAG_RW, &iosched_debug, 0, "");
62
63 static struct iosched_data      ioscpu[SMP_MAXCPU];
64
65 static int
66 badjiosched(thread_t td, size_t bytes)
67 {
68         globaldata_t gd = mycpu;
69         size_t iostotal;
70         int factor;
71         int i;
72         int delta;
73
74         iostotal = 0;
75         for (i = 0; i < ncpus; ++i)
76                 iostotal += ioscpu[i].iowbytes;
77         if (SIZE_T_MAX / SMP_MAXCPU - td->td_iosdata.iowbytes < bytes)
78                 bytes = SIZE_T_MAX / SMP_MAXCPU - td->td_iosdata.iowbytes;
79         td->td_iosdata.iowbytes += bytes;
80         ioscpu[gd->gd_cpuid].iowbytes += bytes;
81         iostotal += bytes;
82         delta = ticks - td->td_iosdata.lastticks;
83         if (delta) {
84                 td->td_iosdata.lastticks = ticks;
85                 if (delta < 0 || delta > hz * 10)
86                         delta = hz * 10;
87                 bytes = (int64_t)td->td_iosdata.iowbytes * delta / (hz * 10);
88                 td->td_iosdata.iowbytes -= bytes;
89                 ioscpu[gd->gd_cpuid].iowbytes -= bytes;
90                 iostotal -= bytes;
91         }
92         if (iostotal > 0)
93                 factor = td->td_iosdata.iowbytes * 100 / iostotal;
94         else
95                 factor = 50;
96         if (delta && (iosched_debug & 1)) {
97                 kprintf("proc %12s (%-5d) factor %3d (%zd/%zd)\n",
98                         td->td_comm,
99                         (td->td_lwp ? (int)td->td_lwp->lwp_proc->p_pid : -1),
100                         factor, td->td_iosdata.iowbytes, iostotal);
101         }
102         return (factor);
103 }
104
105 void
106 biosched_done(thread_t td)
107 {
108         globaldata_t gd = mycpu;
109         size_t bytes;
110
111         if ((bytes = td->td_iosdata.iowbytes) != 0) {
112                 td->td_iosdata.iowbytes = 0;
113                 ioscpu[gd->gd_cpuid].iowbytes -= bytes;
114         }
115 }
116
117 /*
118  * Caller intends to write (bytes)
119  */
120 void
121 bwillwrite(int bytes)
122 {
123         int count;
124         int factor;
125
126         count = bd_heatup();
127         if (count > 0) {
128                 factor = badjiosched(curthread, (size_t)bytes);
129                 count = hidirtybufspace * factor / 100;
130                 bd_wait(count);
131         }
132 }
133
134 /*
135  * Caller intends to read (bytes)
136  */
137 void
138 bwillread(int bytes)
139 {
140 }
141
142 /*
143  * Call intends to do an inode-modifying operation of some sort.
144  */
145 void
146 bwillinode(int n)
147 {
148         int count;
149         int factor;
150
151         count = bd_heatup();
152         if (count > 0) {
153                 factor = badjiosched(curthread, PAGE_SIZE);
154                 count = count * factor / 100;
155                 bd_wait(count);
156         }
157 }
158