vge: Switch from device_polling to ifpoll
[dragonfly.git] / sys / boot / pc32 / loader / main.c
1 /*
2  * Copyright (c) 2003,2004 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  * Copyright (c) 1998 Michael Smith <msmith@freebsd.org>
35  * All rights reserved.
36  *
37  * Redistribution and use in source and binary forms, with or without
38  * modification, are permitted provided that the following conditions
39  * are met:
40  * 1. Redistributions of source code must retain the above copyright
41  *    notice, this list of conditions and the following disclaimer.
42  * 2. Redistributions in binary form must reproduce the above copyright
43  *    notice, this list of conditions and the following disclaimer in the
44  *    documentation and/or other materials provided with the distribution.
45  *
46  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
47  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
48  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
49  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
50  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
51  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
52  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
53  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
54  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
55  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
56  * SUCH DAMAGE.
57  *
58  * $FreeBSD: src/sys/boot/i386/loader/main.c,v 1.28 2003/08/25 23:28:32 obrien Exp $
59  */
60
61 /*
62  * MD bootstrap main() and assorted miscellaneous
63  * commands.
64  */
65
66 #include <stand.h>
67 #include <string.h>
68 #include <machine/bootinfo.h>
69 #include <machine/psl.h>
70 #include <sys/reboot.h>
71
72 #include "bootstrap.h"
73 #include "libi386/libi386.h"
74 #include "btxv86.h"
75
76 #define KARGS_FLAGS_CD          0x1
77 #define KARGS_FLAGS_PXE         0x2
78
79 /* Arguments passed in from the boot1/boot2 loader */
80 static struct 
81 {
82     u_int32_t   howto;
83     u_int32_t   bootdev;
84     u_int32_t   bootflags;
85     u_int32_t   pxeinfo;
86     u_int32_t   res2;
87     u_int32_t   bootinfo;
88 } *kargs;
89
90 static u_int32_t        initial_howto;
91 static u_int32_t        initial_bootdev;
92 static struct bootinfo  *initial_bootinfo;
93
94 struct arch_switch      archsw;         /* MI/MD interface boundary */
95
96 static void             extract_currdev(void);
97 static int              isa_inb(int port);
98 static void             isa_outb(int port, int value);
99 void                    exit(int code);
100
101 /* from vers.c */
102 extern  char bootprog_name[], bootprog_rev[], bootprog_date[], bootprog_maker[];
103
104 /* XXX debugging */
105 extern char _end[];
106
107 #define COMCONSOLE_DEBUG
108 #ifdef COMCONSOLE_DEBUG
109
110 static void
111 WDEBUG_INIT(void)
112 {
113     isa_outb(0x3f8+3, 0x83);    /* DLAB + 8N1 */
114     isa_outb(0x3f8+0, (115200 / 9600) & 0xFF);
115     isa_outb(0x3f8+1, (115200 / 9600) >> 8);
116     isa_outb(0x3f8+3, 0x03);    /* 8N1 */
117     isa_outb(0x3f8+4, 0x03);    /* RTS+DTR */
118     isa_outb(0x3f8+2, 0x01);    /* FIFO_ENABLE */
119 }
120
121 static void
122 WDEBUG(char c)
123 {
124     isa_outb(0x3f8, c);
125 }
126
127 #else
128
129 #define WDEBUG(x)
130 #define WDEBUG_INIT()
131
132 #endif
133
134 int
135 main(void)
136 {
137     char *memend;
138     int i;
139
140     WDEBUG_INIT();
141     WDEBUG('X');
142
143     /* Pick up arguments */
144     kargs = (void *)__args;
145     initial_howto = kargs->howto;
146     initial_bootdev = kargs->bootdev;
147     initial_bootinfo = kargs->bootinfo ? (struct bootinfo *)PTOV(kargs->bootinfo) : NULL;
148
149 #ifdef COMCONSOLE_DEBUG
150     printf("args at %p initial_howto = %08x bootdev = %08x bootinfo = %p\n", 
151         kargs, initial_howto, initial_bootdev, initial_bootinfo);
152 #endif
153
154     /* Initialize the v86 register set to a known-good state. */
155     bzero(&v86, sizeof(v86));
156     v86.efl = PSL_RESERVED_DEFAULT | PSL_I;
157
158
159     /* 
160      * Initialize the heap as early as possible.  Once this is done, 
161      * malloc() is usable. 
162      *
163      * Don't include our stack in the heap.  If the stack is in low
164      * user memory use {end,bios_basemem}.  If the stack is in high
165      * user memory but not extended memory then don't let the heap
166      * overlap the stack.  If the stack is in extended memory limit
167      * the heap to bios_basemem.
168      *
169      * Be sure to use the virtual bios_basemem address rather then
170      * the physical bios_basemem address or we may overwrite BIOS 
171      * data.
172      */
173     bios_getmem();
174     memend = (char *)&memend - 0x8000;  /* space for stack */
175     memend = (char *)((uintptr_t)memend & ~(uintptr_t)(0x1000 - 1));
176     if (memend < (char *)_end) {
177         setheap((void *)_end, PTOV(bios_basemem));
178     } else {
179         if (memend > (char *)PTOV(bios_basemem))
180             memend = (char *)PTOV(bios_basemem);
181         setheap((void *)_end, memend);
182     }
183
184     /* 
185      * XXX Chicken-and-egg problem; we want to have console output early, 
186      * but some console attributes may depend on reading from eg. the boot
187      * device, which we can't do yet.
188      *
189      * We can use printf() etc. once this is done.   The previous boot stage
190      * might have requested a video or serial preference, in which case we
191      * set it.  If neither is specified or both are specified we leave the
192      * console environment variable alone, defaulting to dual boot.
193      */
194     if (initial_howto & RB_MUTE) {
195         setenv("console", "nullconsole", 1);
196     } else if ((initial_howto & (RB_VIDEO|RB_SERIAL)) != (RB_VIDEO|RB_SERIAL)) {
197         if (initial_howto & RB_VIDEO)
198             setenv("console", "vidconsole", 1);
199         if (initial_howto & RB_SERIAL)
200             setenv("console", "comconsole", 1);
201     }
202     cons_probe();
203
204     /*
205      * Initialise the block cache
206      */
207     bcache_init(32, 512);       /* 16k cache XXX tune this */
208
209     /*
210      * Special handling for PXE and CD booting.
211      */
212     if (kargs->bootinfo == 0) {
213         /*
214          * We only want the PXE disk to try to init itself in the below
215          * walk through devsw if we actually booted off of PXE.
216          */
217         if (kargs->bootflags & KARGS_FLAGS_PXE)
218             pxe_enable(kargs->pxeinfo ? PTOV(kargs->pxeinfo) : NULL);
219         else if (kargs->bootflags & KARGS_FLAGS_CD)
220             bc_add(initial_bootdev);
221     }
222
223     /*
224      * March through the device switch probing for things.
225      */
226     for (i = 0; devsw[i] != NULL; i++) {
227         WDEBUG('M' + i);
228         if (devsw[i]->dv_init != NULL)
229             (devsw[i]->dv_init)();
230         WDEBUG('M' + i);
231     }
232     printf("BIOS %dkB/%dkB available memory\n", 
233             bios_basemem / 1024, bios_extmem / 1024);
234     if (initial_bootinfo != NULL) {
235         initial_bootinfo->bi_basemem = bios_basemem / 1024;
236         initial_bootinfo->bi_extmem = bios_extmem / 1024;
237     }
238
239     /* detect ACPI for future reference */
240     biosacpi_detect();
241
242     /* enable EHCI */
243     setenv("ehci_load", "YES", 1);
244
245     printf("\n");
246     printf("%s, Revision %s\n", bootprog_name, bootprog_rev);
247     printf("(%s, %s)\n", bootprog_maker, bootprog_date);
248
249     extract_currdev();                          /* set $currdev and $loaddev */
250     setenv("LINES", "24", 1);                   /* optional */
251     
252     bios_getsmap();
253
254     archsw.arch_autoload = i386_autoload;
255     archsw.arch_getdev = i386_getdev;
256     archsw.arch_copyin = i386_copyin;
257     archsw.arch_copyout = i386_copyout;
258     archsw.arch_readin = i386_readin;
259     archsw.arch_isainb = isa_inb;
260     archsw.arch_isaoutb = isa_outb;
261
262     interact();                 /* doesn't return */
263
264     /* if we ever get here, it is an error */
265     return (1);
266 }
267
268 /*
269  * Set the 'current device' by (if possible) recovering the boot device as 
270  * supplied by the initial bootstrap.
271  *
272  * XXX should be extended for netbooting.
273  */
274 static void
275 extract_currdev(void)
276 {
277     struct i386_devdesc new_currdev;
278     int                 biosdev = -1;
279
280     /* Assume we are booting from a BIOS disk by default */
281     new_currdev.d_dev = &biosdisk;
282
283     /* new-style boot loaders such as pxeldr and cdldr */
284     if (kargs->bootinfo == 0) {
285         if ((kargs->bootflags & KARGS_FLAGS_CD) != 0) {
286             /* we are booting from a CD with cdboot */
287             new_currdev.d_dev = &bioscd;
288             new_currdev.d_kind.bioscd.unit = bc_bios2unit(initial_bootdev);
289         } else if ((kargs->bootflags & KARGS_FLAGS_PXE) != 0) {
290             /* we are booting from pxeldr */
291             new_currdev.d_dev = &pxedisk;
292             new_currdev.d_kind.netif.unit = 0;
293         } else {
294             /* we don't know what our boot device is */
295             new_currdev.d_kind.biosdisk.slice = -1;
296             new_currdev.d_kind.biosdisk.partition = 0;
297             biosdev = -1;
298         }
299     } else if ((initial_bootdev & B_MAGICMASK) != B_DEVMAGIC) {
300         /* The passed-in boot device is bad */
301         new_currdev.d_kind.biosdisk.slice = -1;
302         new_currdev.d_kind.biosdisk.partition = 0;
303         biosdev = -1;
304     } else {
305         new_currdev.d_kind.biosdisk.slice = (B_ADAPTOR(initial_bootdev) << 4) +
306                                              B_CONTROLLER(initial_bootdev) - 1;
307         new_currdev.d_kind.biosdisk.partition = B_PARTITION(initial_bootdev);
308         biosdev = initial_bootinfo->bi_bios_dev;
309
310         /*
311          * If we are booted by an old bootstrap, we have to guess at the BIOS
312          * unit number.  We will loose if there is more than one disk type
313          * and we are not booting from the lowest-numbered disk type 
314          * (ie. SCSI when IDE also exists).
315          */
316         if ((biosdev == 0) && (B_TYPE(initial_bootdev) != 2))   /* biosdev doesn't match major */
317             biosdev = 0x80 + B_UNIT(initial_bootdev);           /* assume harddisk */
318     }
319     new_currdev.d_type = new_currdev.d_dev->dv_type;
320     
321     /*
322      * If we are booting off of a BIOS disk and we didn't succeed in determining
323      * which one we booted off of, just use disk0: as a reasonable default.
324      */
325     if ((new_currdev.d_type == biosdisk.dv_type) &&
326         ((new_currdev.d_kind.biosdisk.unit = bd_bios2unit(biosdev)) == -1)) {
327         printf("Can't work out which disk we are booting from.\n"
328                "Guessed BIOS device 0x%x not found by probes, defaulting to disk0:\n", biosdev);
329         new_currdev.d_kind.biosdisk.unit = 0;
330     }
331     env_setenv("currdev", EV_VOLATILE, i386_fmtdev(&new_currdev),
332                i386_setcurrdev, env_nounset);
333     env_setenv("loaddev", EV_VOLATILE, i386_fmtdev(&new_currdev), env_noset,
334                env_nounset);
335 }
336
337 COMMAND_SET(reboot, "reboot", "reboot the system", command_reboot);
338
339 static int
340 command_reboot(int argc, char *argv[])
341 {
342     int i;
343
344     for (i = 0; devsw[i] != NULL; ++i)
345         if (devsw[i]->dv_cleanup != NULL)
346             (devsw[i]->dv_cleanup)();
347
348     printf("Rebooting...\n");
349     delay(1000000);
350     __exit(0);
351 }
352
353 /* provide this for panic, as it's not in the startup code */
354 void
355 exit(int code)
356 {
357     __exit(code);
358 }
359
360 COMMAND_SET(heap, "heap", "show heap usage", command_heap);
361
362 static int
363 command_heap(int argc, char *argv[])
364 {
365     mallocstats();
366     printf("heap base at %p, top at %p\n", _end, sbrk(0));
367     return(CMD_OK);
368 }
369
370 /* ISA bus access functions for PnP, derived from <machine/cpufunc.h> */
371 static int              
372 isa_inb(int port)
373 {
374     u_char      data;
375     
376     if (__builtin_constant_p(port) && 
377         (((port) & 0xffff) < 0x100) && 
378         ((port) < 0x10000)) {
379         __asm __volatile("inb %1,%0" : "=a" (data) : "id" ((u_short)(port)));
380     } else {
381         __asm __volatile("inb %%dx,%0" : "=a" (data) : "d" (port));
382     }
383     return(data);
384 }
385
386 static void
387 isa_outb(int port, int value)
388 {
389     u_char      al = value;
390     
391     if (__builtin_constant_p(port) && 
392         (((port) & 0xffff) < 0x100) && 
393         ((port) < 0x10000)) {
394         __asm __volatile("outb %0,%1" : : "a" (al), "id" ((u_short)(port)));
395     } else {
396         __asm __volatile("outb %0,%%dx" : : "a" (al), "d" (port));
397     }
398 }
399