Merge from vendor branch NTPD:
[dragonfly.git] / sys / boot / i386 / 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  * $DragonFly: src/sys/boot/i386/loader/Attic/main.c,v 1.8 2004/09/09 03:47:08 joerg Exp $
60  */
61
62 /*
63  * MD bootstrap main() and assorted miscellaneous
64  * commands.
65  */
66
67 #include <stand.h>
68 #include <string.h>
69 #include <machine/bootinfo.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     /* 
155      * Initialize the heap as early as possible.  Once this is done, 
156      * malloc() is usable. 
157      *
158      * Don't include our stack in the heap.  If the stack is in low
159      * user memory use {end,bios_basemem}.  If the stack is in high
160      * user memory but not extended memory then don't let the heap
161      * overlap the stack.  If the stack is in extended memory limit
162      * the heap to bios_basemem.
163      *
164      * Be sure to use the virtual bios_basemem address rather then
165      * the physical bios_basemem address or we may overwrite BIOS 
166      * data.
167      */
168     bios_getmem();
169     memend = (char *)&memend - 0x8000;  /* space for stack */
170     memend = (char *)((uintptr_t)memend & ~(uintptr_t)(0x1000 - 1));
171     if (memend < (char *)end) {
172         setheap((void *)end, PTOV(bios_basemem));
173     } else {
174         if (memend > (char *)PTOV(bios_basemem))
175             memend = (char *)PTOV(bios_basemem);
176         setheap((void *)end, memend);
177     }
178
179     /* 
180      * XXX Chicken-and-egg problem; we want to have console output early, 
181      * but some console attributes may depend on reading from eg. the boot
182      * device, which we can't do yet.
183      *
184      * We can use printf() etc. once this is done.   The previous boot stage
185      * might have requested a video or serial preference, in which case we
186      * set it.  If neither is specified or both are specified we leave the
187      * console environment variable alone, defaulting to dual boot.
188      */
189     if (initial_howto & RB_MUTE) {
190         setenv("console", "nullconsole", 1);
191     } else if ((initial_howto & (RB_VIDEO|RB_SERIAL)) != (RB_VIDEO|RB_SERIAL)) {
192         if (initial_howto & RB_VIDEO)
193             setenv("console", "vidconsole", 1);
194         if (initial_howto & RB_SERIAL)
195             setenv("console", "comconsole", 1);
196     }
197     cons_probe();
198
199     /*
200      * Initialise the block cache
201      */
202     bcache_init(32, 512);       /* 16k cache XXX tune this */
203
204     /*
205      * Special handling for PXE and CD booting.
206      */
207     if (kargs->bootinfo == NULL) {
208         /*
209          * We only want the PXE disk to try to init itself in the below
210          * walk through devsw if we actually booted off of PXE.
211          */
212         if (kargs->bootflags & KARGS_FLAGS_PXE)
213             pxe_enable(kargs->pxeinfo ? PTOV(kargs->pxeinfo) : NULL);
214         else if (kargs->bootflags & KARGS_FLAGS_CD)
215             bc_add(initial_bootdev);
216     }
217
218     /*
219      * March through the device switch probing for things.
220      */
221     for (i = 0; devsw[i] != NULL; i++) {
222         WDEBUG('M' + i);
223         if (devsw[i]->dv_init != NULL)
224             (devsw[i]->dv_init)();
225         WDEBUG('M' + i);
226     }
227     printf("BIOS %dkB/%dkB available memory\n", 
228             bios_basemem / 1024, bios_extmem / 1024);
229     if (initial_bootinfo != NULL) {
230         initial_bootinfo->bi_basemem = bios_basemem / 1024;
231         initial_bootinfo->bi_extmem = bios_extmem / 1024;
232     }
233
234     /* detect ACPI for future reference */
235     biosacpi_detect();
236
237     printf("\n");
238     printf("%s, Revision %s\n", bootprog_name, bootprog_rev);
239     printf("(%s, %s)\n", bootprog_maker, bootprog_date);
240
241     extract_currdev();                          /* set $currdev and $loaddev */
242     setenv("LINES", "24", 1);                   /* optional */
243     
244     bios_getsmap();
245
246     archsw.arch_autoload = i386_autoload;
247     archsw.arch_getdev = i386_getdev;
248     archsw.arch_copyin = i386_copyin;
249     archsw.arch_copyout = i386_copyout;
250     archsw.arch_readin = i386_readin;
251     archsw.arch_isainb = isa_inb;
252     archsw.arch_isaoutb = isa_outb;
253
254     interact();                 /* doesn't return */
255
256     /* if we ever get here, it is an error */
257     return (1);
258 }
259
260 /*
261  * Set the 'current device' by (if possible) recovering the boot device as 
262  * supplied by the initial bootstrap.
263  *
264  * XXX should be extended for netbooting.
265  */
266 static void
267 extract_currdev(void)
268 {
269     struct i386_devdesc new_currdev;
270     int                 major, biosdev = -1;
271
272     /* Assume we are booting from a BIOS disk by default */
273     new_currdev.d_dev = &biosdisk;
274
275     /* new-style boot loaders such as pxeldr and cdldr */
276     if (kargs->bootinfo == NULL) {
277         if ((kargs->bootflags & KARGS_FLAGS_CD) != 0) {
278             /* we are booting from a CD with cdboot */
279             new_currdev.d_dev = &bioscd;
280             new_currdev.d_kind.bioscd.unit = bc_bios2unit(initial_bootdev);
281         } else if ((kargs->bootflags & KARGS_FLAGS_PXE) != 0) {
282             /* we are booting from pxeldr */
283             new_currdev.d_dev = &pxedisk;
284             new_currdev.d_kind.netif.unit = 0;
285         } else {
286             /* we don't know what our boot device is */
287             new_currdev.d_kind.biosdisk.slice = -1;
288             new_currdev.d_kind.biosdisk.partition = 0;
289             biosdev = -1;
290         }
291     } else if ((initial_bootdev & B_MAGICMASK) != B_DEVMAGIC) {
292         /* The passed-in boot device is bad */
293         new_currdev.d_kind.biosdisk.slice = -1;
294         new_currdev.d_kind.biosdisk.partition = 0;
295         biosdev = -1;
296     } else {
297         new_currdev.d_kind.biosdisk.slice = (B_ADAPTOR(initial_bootdev) << 4) +
298                                              B_CONTROLLER(initial_bootdev) - 1;
299         new_currdev.d_kind.biosdisk.partition = B_PARTITION(initial_bootdev);
300         biosdev = initial_bootinfo->bi_bios_dev;
301         major = B_TYPE(initial_bootdev);
302
303         /*
304          * If we are booted by an old bootstrap, we have to guess at the BIOS
305          * unit number.  We will loose if there is more than one disk type
306          * and we are not booting from the lowest-numbered disk type 
307          * (ie. SCSI when IDE also exists).
308          */
309         if ((biosdev == 0) && (B_TYPE(initial_bootdev) != 2))   /* biosdev doesn't match major */
310             biosdev = 0x80 + B_UNIT(initial_bootdev);           /* assume harddisk */
311     }
312     new_currdev.d_type = new_currdev.d_dev->dv_type;
313     
314     /*
315      * If we are booting off of a BIOS disk and we didn't succeed in determining
316      * which one we booted off of, just use disk0: as a reasonable default.
317      */
318     if ((new_currdev.d_type == biosdisk.dv_type) &&
319         ((new_currdev.d_kind.biosdisk.unit = bd_bios2unit(biosdev)) == -1)) {
320         printf("Can't work out which disk we are booting from.\n"
321                "Guessed BIOS device 0x%x not found by probes, defaulting to disk0:\n", biosdev);
322         new_currdev.d_kind.biosdisk.unit = 0;
323     }
324     env_setenv("currdev", EV_VOLATILE, i386_fmtdev(&new_currdev),
325                i386_setcurrdev, env_nounset);
326     env_setenv("loaddev", EV_VOLATILE, i386_fmtdev(&new_currdev), env_noset,
327                env_nounset);
328 }
329
330 COMMAND_SET(reboot, "reboot", "reboot the system", command_reboot);
331
332 static int
333 command_reboot(int argc, char *argv[])
334 {
335     int i;
336
337     for (i = 0; devsw[i] != NULL; ++i)
338         if (devsw[i]->dv_cleanup != NULL)
339             (devsw[i]->dv_cleanup)();
340
341     printf("Rebooting...\n");
342     delay(1000000);
343     __exit(0);
344 }
345
346 /* provide this for panic, as it's not in the startup code */
347 void
348 exit(int code)
349 {
350     __exit(code);
351 }
352
353 COMMAND_SET(heap, "heap", "show heap usage", command_heap);
354
355 static int
356 command_heap(int argc, char *argv[])
357 {
358     mallocstats();
359     printf("heap base at %p, top at %p\n", end, sbrk(0));
360     return(CMD_OK);
361 }
362
363 /* ISA bus access functions for PnP, derived from <machine/cpufunc.h> */
364 static int              
365 isa_inb(int port)
366 {
367     u_char      data;
368     
369     if (__builtin_constant_p(port) && 
370         (((port) & 0xffff) < 0x100) && 
371         ((port) < 0x10000)) {
372         __asm __volatile("inb %1,%0" : "=a" (data) : "id" ((u_short)(port)));
373     } else {
374         __asm __volatile("inb %%dx,%0" : "=a" (data) : "d" (port));
375     }
376     return(data);
377 }
378
379 static void
380 isa_outb(int port, int value)
381 {
382     u_char      al = value;
383     
384     if (__builtin_constant_p(port) && 
385         (((port) & 0xffff) < 0x100) && 
386         ((port) < 0x10000)) {
387         __asm __volatile("outb %0,%1" : : "a" (al), "id" ((u_short)(port)));
388     } else {
389         __asm __volatile("outb %0,%%dx" : : "a" (al), "d" (port));
390     }
391 }
392