459436f8d95b29513a6e641ce532bf1dad0806c0
[dragonfly.git] / sys / boot / pc32 / boot2 / boot2.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 Robert Nordier
35  * All rights reserved.
36  *
37  * Redistribution and use in source and binary forms are freely
38  * permitted provided that the above copyright notice and this
39  * paragraph and the following disclaimer are duplicated in all
40  * such forms.
41  *
42  * This software is provided "AS IS" and without any express or
43  * implied warranties, including, without limitation, the implied
44  * warranties of merchantability and fitness for a particular
45  * purpose.
46  *
47  * $FreeBSD: src/sys/boot/i386/boot2/boot2.c,v 1.64 2003/08/25 23:28:31 obrien Exp $
48  * $DragonFly: src/sys/boot/pc32/boot2/boot2.c,v 1.18 2008/09/13 11:45:45 corecode Exp $
49  */
50
51 #define AOUT_H_FORCE32
52 #include <sys/param.h>
53 #ifdef DISKLABEL64
54 #include <sys/disklabel64.h>
55 #else
56 #include <sys/disklabel32.h>
57 #endif
58 #include <sys/diskslice.h>
59 #include <sys/diskmbr.h>
60 #include <sys/dtype.h>
61 #include <sys/dirent.h>
62 #include <machine/bootinfo.h>
63 #include <machine/elf.h>
64 #include <machine/psl.h>
65
66 #include <stdarg.h>
67
68 #include <a.out.h>
69
70 #include <btxv86.h>
71
72 #ifdef DISKLABEL64
73 #include "boot2_64.h"
74 #else
75 #include "boot2_32.h"
76 #endif
77 #include "boot2.h"
78 #include "lib.h"
79 #include "../bootasm.h"
80
81 #define SECOND          18      /* Circa that many ticks in a second. */
82
83 #define RBX_ASKNAME     0x0     /* -a */
84 #define RBX_SINGLE      0x1     /* -s */
85 #define RBX_DFLTROOT    0x5     /* -r */
86 #define RBX_KDB         0x6     /* -d */
87 #define RBX_CONFIG      0xa     /* -c */
88 #define RBX_VERBOSE     0xb     /* -v */
89 #define RBX_SERIAL      0xc     /* -h */
90 #define RBX_CDROM       0xd     /* -C */
91 #define RBX_GDB         0xf     /* -g */
92 #define RBX_MUTE        0x10    /* -m */
93 #define RBX_PAUSE       0x12    /* -p */
94 #define RBX_NOINTR      0x1c    /* -n */
95 #define RBX_VIDEO       0x1d    /* -V */
96 #define RBX_PROBEKBD    0x1e    /* -P */
97 /* 0x1f is reserved for the historical RB_BOOTINFO option */
98
99 #define RBF_MUTE        (1 << RBX_MUTE)
100 #define RBF_SERIAL      (1 << RBX_SERIAL)
101 #define RBF_VIDEO       (1 << RBX_VIDEO)
102
103 /* pass: -a, -s, -r, -d, -c, -v, -h, -C, -g, -m, -p, -V */
104 #define RBX_MASK        0x2005ffff
105
106 #define PATH_CONFIG     "/boot.config"
107 #define PATH_BOOT3      "/loader"               /* /boot is dedicated */
108 #define PATH_BOOT3_ALT  "/boot/loader"          /* /boot in root */
109 #define PATH_KERNEL     "/kernel"
110
111 #define NDEV            3
112 #define MEM_BASE        0x12
113 #define MEM_EXT         0x15
114 #define V86_CY(x)       ((x) & PSL_C)
115 #define V86_ZR(x)       ((x) & PSL_Z)
116
117 #define DRV_HARD        0x80
118 #define DRV_MASK        0x7f
119
120 #define TYPE_AD         0
121 #define TYPE_DA         1
122 #define TYPE_MAXHARD    TYPE_DA
123 #define TYPE_FD         2
124
125 #define NOPT            12
126
127 #define INVALID_S       "Bad %s\n"
128
129 extern uint32_t _end;
130
131 static const char optstr[NOPT] = { "VhaCgmnPprsv" };
132 static const unsigned char flags[NOPT] = {
133     RBX_VIDEO,
134     RBX_SERIAL,
135     RBX_ASKNAME,
136     RBX_CDROM,
137     RBX_GDB,
138     RBX_MUTE,
139     RBX_NOINTR,
140     RBX_PROBEKBD,
141     RBX_PAUSE,
142     RBX_DFLTROOT,
143     RBX_SINGLE,
144     RBX_VERBOSE
145 };
146
147 static const char *const dev_nm[NDEV] = {"ad", "da", "fd"};
148 static const unsigned char dev_maj[NDEV] = {30, 4, 2};
149
150 static struct dsk {
151     unsigned drive;
152     unsigned type;
153     unsigned unit;
154     unsigned slice;
155     unsigned part;
156     unsigned start;
157     int init;
158 } dsk;
159
160 static char cmd[512];
161 static char kname[1024];
162 static uint32_t opts = RBF_VIDEO;
163 static struct bootinfo bootinfo;
164
165 /*
166  * boot2 encapsulated ABI elements provided to *fsread.c
167  *
168  * NOTE: boot2_dmadat is extended by per-filesystem APIs
169  */
170 uint32_t fs_off;
171 int     ls;
172 struct boot2_dmadat *boot2_dmadat;
173
174 void exit(int);
175 static void load(void);
176 static int parse(void);
177 static int dskprobe(void);
178 static int xfsread(boot2_ino_t, void *, size_t);
179 static uint32_t memsize(void);
180 static int drvread(void *, unsigned, unsigned);
181 static int keyhit(unsigned);
182 static void xputc(int);
183 static int xgetc(int);
184 static int getc(int);
185
186 void
187 memcpy(void *d, const void *s, int len)
188 {
189     char *dd = d;
190     const char *ss = s;
191
192 #if 0
193     if (dd < ss) {
194         while (--len >= 0)
195             *dd++ = *ss++;
196     } else {
197 #endif
198         while (--len >= 0)
199             dd[len] = ss[len];
200 #if 0
201     }
202 #endif
203 }
204
205 int
206 strcmp(const char *s1, const char *s2)
207 {
208     for (; *s1 == *s2 && *s1; s1++, s2++)
209         ;
210     return ((int)((unsigned char)*s1 - (unsigned char)*s2));
211 }
212
213 #if defined(UFS) && defined(HAMMERFS)
214
215 const struct boot2_fsapi *fsapi;
216
217 #elif defined(UFS)
218
219 #define fsapi   (&boot2_ufs_api)
220
221 #elif defined(HAMMERFS)
222
223 #define fsapi   (&boot2_hammer_api)
224
225 #endif
226
227 static int
228 xfsread(boot2_ino_t inode, void *buf, size_t nbyte)
229 {
230     if ((size_t)fsapi->fsread(inode, buf, nbyte) != nbyte) {
231         printf(INVALID_S, "format");
232         return -1;
233     }
234     return 0;
235 }
236
237 static inline uint32_t
238 memsize(void)
239 {
240     v86.addr = MEM_EXT;
241     v86.eax = 0x8800;
242     v86int();
243     return v86.eax;
244 }
245
246 static inline void
247 getstr(void)
248 {
249     char *s;
250     int c;
251
252     s = cmd;
253     for (;;) {
254         switch (c = xgetc(0)) {
255         case 0:
256             break;
257         case '\177':
258         case '\b':
259             if (s > cmd) {
260                 s--;
261                 printf("\b \b");
262             }
263             break;
264         case '\n':
265         case '\r':
266             *s = 0;
267             return;
268         default:
269             if (s - cmd < sizeof(cmd) - 1)
270                 *s++ = c;
271             putchar(c);
272         }
273     }
274 }
275
276 static inline void
277 putc(int c)
278 {
279     v86.addr = 0x10;
280     v86.eax = 0xe00 | (c & 0xff);
281     v86.ebx = 0x7;
282     v86int();
283 }
284
285 int
286 main(void)
287 {
288     int autoboot;
289     boot2_ino_t ino;
290
291     boot2_dmadat =
292                 (void *)(roundup2(__base + (int32_t)&_end, 0x10000) - __base);
293     v86.ctl = V86_FLAGS;
294     v86.efl = PSL_RESERVED_DEFAULT | PSL_I;
295     dsk.drive = *(uint8_t *)PTOV(MEM_BTX_USR_ARG);
296     dsk.type = dsk.drive & DRV_HARD ? TYPE_AD : TYPE_FD;
297     dsk.unit = dsk.drive & DRV_MASK;
298     dsk.slice = *(uint8_t *)PTOV(MEM_BTX_USR_ARG + 1) + 1;
299     bootinfo.bi_version = BOOTINFO_VERSION;
300     bootinfo.bi_size = sizeof(bootinfo);
301     bootinfo.bi_basemem = 0;    /* XXX will be filled by loader or kernel */
302     bootinfo.bi_extmem = memsize();
303     bootinfo.bi_memsizes_valid++;
304
305     autoboot = 1;
306
307     /*
308      * Probe the default disk and process the configuration file if
309      * successful.
310      */
311     if (dskprobe() == 0) {
312         if ((ino = fsapi->fslookup(PATH_CONFIG)))
313             fsapi->fsread(ino, cmd, sizeof(cmd));
314     }
315
316     /*
317      * Parse config file if present.  parse() will re-probe if necessary.
318      */
319     if (cmd[0]) {
320         printf("%s: %s", PATH_CONFIG, cmd);
321         if (parse())
322             autoboot = 0;
323         /* Do not process this command twice */
324         *cmd = 0;
325     }
326
327     /*
328      * Setup our (serial) console after processing the config file.  If
329      * the initialization fails, don't try to use the serial port.  This
330      * can happen if the serial port is unmaped (happens on new laptops a lot).
331      */
332     if ((opts & (RBF_MUTE|RBF_SERIAL|RBF_VIDEO)) == 0)
333         opts |= RBF_SERIAL|RBF_VIDEO;
334     if (opts & RBF_SERIAL) {
335         if (sio_init())
336             opts = RBF_VIDEO;
337     }
338
339
340     /*
341      * Try to exec stage 3 boot loader. If interrupted by a keypress,
342      * or in case of failure, try to load a kernel directly instead.
343      *
344      * We have to try boot /boot/loader and /loader to support booting
345      * from a /boot partition instead of a root partition.
346      */
347     if (autoboot && !*kname) {
348         memcpy(kname, PATH_BOOT3, sizeof(PATH_BOOT3));
349         if (!keyhit(3*SECOND)) {
350             load();
351             memcpy(kname, PATH_BOOT3_ALT, sizeof(PATH_BOOT3_ALT));
352             load();
353             memcpy(kname, PATH_KERNEL, sizeof(PATH_KERNEL));
354         }
355     }
356
357     /* Present the user with the boot2 prompt. */
358
359     for (;;) {
360         printf("\nDragonFly boot\n"
361                "%u:%s(%u,%c)%s: ",
362                dsk.drive & DRV_MASK, dev_nm[dsk.type], dsk.unit,
363                'a' + dsk.part, kname);
364         if (!autoboot || keyhit(5*SECOND))
365             getstr();
366         else
367             putchar('\n');
368         autoboot = 0;
369         if (parse())
370             putchar('\a');
371         else
372             load();
373     }
374 }
375
376 /* XXX - Needed for btxld to link the boot2 binary; do not remove. */
377 void
378 exit(int x)
379 {
380 }
381
382 static void
383 load(void)
384 {
385     union {
386         struct exec ex;
387         Elf32_Ehdr eh;
388     } hdr;
389     Elf32_Phdr ep[2];
390     Elf32_Shdr es[2];
391     caddr_t p;
392     boot2_ino_t ino;
393     uint32_t addr, x;
394     int fmt, i, j;
395
396     if (!(ino = fsapi->fslookup(kname))) {
397         if (!ls)
398             printf("No %s\n", kname);
399         return;
400     }
401     if (xfsread(ino, &hdr, sizeof(hdr)))
402         return;
403     if (N_GETMAGIC(hdr.ex) == ZMAGIC)
404         fmt = 0;
405     else if (IS_ELF(hdr.eh))
406         fmt = 1;
407     else {
408         printf(INVALID_S, "format");
409         return;
410     }
411     if (fmt == 0) {
412         addr = hdr.ex.a_entry & 0xffffff;
413         p = PTOV(addr);
414         fs_off = PAGE_SIZE;
415         if (xfsread(ino, p, hdr.ex.a_text))
416             return;
417         p += roundup2(hdr.ex.a_text, PAGE_SIZE);
418         if (xfsread(ino, p, hdr.ex.a_data))
419             return;
420         p += hdr.ex.a_data + roundup2(hdr.ex.a_bss, PAGE_SIZE);
421         bootinfo.bi_symtab = VTOP(p);
422         memcpy(p, &hdr.ex.a_syms, sizeof(hdr.ex.a_syms));
423         p += sizeof(hdr.ex.a_syms);
424         if (hdr.ex.a_syms) {
425             if (xfsread(ino, p, hdr.ex.a_syms))
426                 return;
427             p += hdr.ex.a_syms;
428             if (xfsread(ino, p, sizeof(int)))
429                 return;
430             x = *(uint32_t *)p;
431             p += sizeof(int);
432             x -= sizeof(int);
433             if (xfsread(ino, p, x))
434                 return;
435             p += x;
436         }
437     } else {
438         fs_off = hdr.eh.e_phoff;
439         for (j = i = 0; i < hdr.eh.e_phnum && j < 2; i++) {
440             if (xfsread(ino, ep + j, sizeof(ep[0])))
441                 return;
442             if (ep[j].p_type == PT_LOAD)
443                 j++;
444         }
445         for (i = 0; i < 2; i++) {
446             p = PTOV(ep[i].p_paddr & 0xffffff);
447             fs_off = ep[i].p_offset;
448             if (xfsread(ino, p, ep[i].p_filesz))
449                 return;
450         }
451         p += roundup2(ep[1].p_memsz, PAGE_SIZE);
452         bootinfo.bi_symtab = VTOP(p);
453         if (hdr.eh.e_shnum == hdr.eh.e_shstrndx + 3) {
454             fs_off = hdr.eh.e_shoff + sizeof(es[0]) *
455                 (hdr.eh.e_shstrndx + 1);
456             if (xfsread(ino, &es, sizeof(es)))
457                 return;
458             for (i = 0; i < 2; i++) {
459                 memcpy(p, &es[i].sh_size, sizeof(es[i].sh_size));
460                 p += sizeof(es[i].sh_size);
461                 fs_off = es[i].sh_offset;
462                 if (xfsread(ino, p, es[i].sh_size))
463                     return;
464                 p += es[i].sh_size;
465             }
466         }
467         addr = hdr.eh.e_entry & 0xffffff;
468     }
469     bootinfo.bi_esymtab = VTOP(p);
470     bootinfo.bi_kernelname = VTOP(kname);
471     bootinfo.bi_bios_dev = dsk.drive;
472     __exec((caddr_t)addr, opts & RBX_MASK,
473            MAKEBOOTDEV(dev_maj[dsk.type], 0, dsk.slice, dsk.unit, dsk.part),
474            0, 0, 0, VTOP(&bootinfo));
475 }
476
477 static int
478 parse(void)
479 {
480     char *arg = cmd;
481     char *p, *q;
482     unsigned int drv;
483     int c, i;
484
485     while ((c = *arg++)) {
486         if (c == ' ' || c == '\t' || c == '\n')
487             continue;
488         for (p = arg; *p && *p != '\n' && *p != ' ' && *p != '\t'; p++)
489             ;
490         if (*p)
491             *p++ = 0;
492         if (c == '-') {
493             while ((c = *arg++)) {
494                 for (i = NOPT - 1; i >= 0; --i) {
495                     if (optstr[i] == c) {
496                         opts ^= 1 << flags[i];
497                         goto ok;
498                     }
499                 }
500                 return(-1);
501                 ok: ;   /* ugly but save space */
502             }
503             if (opts & (1 << RBX_PROBEKBD)) {
504                 i = *(uint8_t *)PTOV(0x496) & 0x10;
505                 if (!i) {
506                     printf("NO KB\n");
507                     opts |= RBF_VIDEO | RBF_SERIAL;
508                 }
509                 opts &= ~(1 << RBX_PROBEKBD);
510             }
511         } else {
512             for (q = arg--; *q && *q != '('; q++);
513             if (*q) {
514                 drv = -1;
515                 if (arg[1] == ':') {
516                     drv = *arg - '0';
517                     if (drv > 9)
518                         return (-1);
519                     arg += 2;
520                 }
521                 if (q - arg != 2)
522                     return -1;
523                 for (i = 0; arg[0] != dev_nm[i][0] ||
524                             arg[1] != dev_nm[i][1]; i++)
525                     if (i == NDEV - 1)
526                         return -1;
527                 dsk.type = i;
528                 arg += 3;
529                 dsk.unit = *arg - '0';
530                 if (arg[1] != ',' || dsk.unit > 9)
531                     return -1;
532                 arg += 2;
533                 dsk.slice = WHOLE_DISK_SLICE;
534                 if (arg[1] == ',') {
535                     dsk.slice = *arg - '0' + 1;
536                     if (dsk.slice > NDOSPART)
537                         return -1;
538                     arg += 2;
539                 }
540                 if (arg[1] != ')')
541                     return -1;
542                 dsk.part = *arg - 'a';
543                 if (dsk.part > 7)
544                     return (-1);
545                 arg += 2;
546                 if (drv == -1)
547                     drv = dsk.unit;
548                 dsk.drive = (dsk.type <= TYPE_MAXHARD
549                              ? DRV_HARD : 0) + drv;
550             }
551             if ((i = p - arg - !*(p - 1))) {
552                 if ((size_t)i >= sizeof(kname))
553                     return -1;
554                 memcpy(kname, arg, i + 1);
555             }
556         }
557         arg = p;
558     }
559     return dskprobe();
560 }
561
562 static int
563 dskprobe(void)
564 {
565     struct dos_partition *dp;
566 #ifdef DISKLABEL64
567     struct disklabel64 *d;
568 #else
569     struct disklabel32 *d;
570 #endif
571     char *sec;
572     unsigned sl, i;
573
574     /*
575      * Probe slice table
576      */
577     sec = boot2_dmadat->secbuf;
578     dsk.start = 0;
579     if (drvread(sec, DOSBBSECTOR, 1))
580         return -1;
581     dp = (void *)(sec + DOSPARTOFF);
582     sl = dsk.slice;
583     if (sl < BASE_SLICE) {
584         for (i = 0; i < NDOSPART; i++)
585             if (dp[i].dp_typ == DOSPTYP_386BSD &&
586                 (dp[i].dp_flag & 0x80 || sl < BASE_SLICE)) {
587                 sl = BASE_SLICE + i;
588                 if (dp[i].dp_flag & 0x80 ||
589                     dsk.slice == COMPATIBILITY_SLICE)
590                     break;
591             }
592         if (dsk.slice == WHOLE_DISK_SLICE)
593             dsk.slice = sl;
594     }
595     if (sl != WHOLE_DISK_SLICE) {
596         if (sl != COMPATIBILITY_SLICE)
597             dp += sl - BASE_SLICE;
598         if (dp->dp_typ != DOSPTYP_386BSD) {
599             printf(INVALID_S, "slice");
600             return -1;
601         }
602         dsk.start = dp->dp_start;
603     }
604
605     /*
606      * Probe label and partition table
607      */
608 #ifdef DISKLABEL64
609     if (drvread(sec, dsk.start, (sizeof(struct disklabel64) + 511) / 512))
610             return -1;
611     d = (void *)sec;
612     if (d->d_magic != DISKMAGIC64) {
613         printf(INVALID_S, "label");
614         return -1;
615     } else {
616         if (dsk.part >= d->d_npartitions || d->d_partitions[dsk.part].p_bsize == 0) {
617             printf(INVALID_S, "partition");
618             return -1;
619         }
620         dsk.start += d->d_partitions[dsk.part].p_boffset / 512;
621     }
622 #else
623     if (drvread(sec, dsk.start + LABELSECTOR32, 1))
624             return -1;
625     d = (void *)(sec + LABELOFFSET32);
626     if (d->d_magic != DISKMAGIC32 || d->d_magic2 != DISKMAGIC32) {
627         if (dsk.part != RAW_PART) {
628             printf(INVALID_S, "label");
629             return -1;
630         }
631     } else {
632         if (!dsk.init) {
633             if (d->d_type == DTYPE_SCSI)
634                 dsk.type = TYPE_DA;
635             dsk.init++;
636         }
637         if (dsk.part >= d->d_npartitions ||
638             !d->d_partitions[dsk.part].p_size) {
639             printf(INVALID_S, "partition");
640             return -1;
641         }
642         dsk.start += d->d_partitions[dsk.part].p_offset;
643         dsk.start -= d->d_partitions[RAW_PART].p_offset;
644     }
645 #endif
646     /*
647      * Probe filesystem
648      */
649 #if defined(UFS) && defined(HAMMERFS)
650     if (boot2_hammer_api.fsinit() == 0) {
651         fsapi = &boot2_hammer_api;
652     } else if (boot2_ufs_api.fsinit() == 0) {
653         fsapi = &boot2_ufs_api;
654     } else {
655         printf("fs probe failed\n");
656         fsapi = &boot2_ufs_api;
657         return -1;
658     }
659     return 0;
660 #else
661     return fsapi->fsinit();
662 #endif
663 }
664
665
666 /*
667  * Read from the probed disk.  We have established the slice and partition
668  * base sector.
669  */
670 int
671 dskread(void *buf, unsigned lba, unsigned nblk)
672 {
673     return drvread(buf, dsk.start + lba, nblk);
674 }
675
676 /*
677  * boot encapsulated ABI
678  */
679 void
680 printf(const char *fmt,...)
681 {
682     va_list ap;
683     char buf[10];
684     char *s;
685     unsigned u;
686     int c;
687
688     va_start(ap, fmt);
689     while ((c = *fmt++)) {
690         if (c == '%') {
691             c = *fmt++;
692             switch (c) {
693             case 'c':
694                 putchar(va_arg(ap, int));
695                 continue;
696             case 's':
697                 for (s = va_arg(ap, char *); *s; s++)
698                     putchar(*s);
699                 continue;
700             case 'u':
701                 u = va_arg(ap, unsigned);
702                 s = buf;
703                 do
704                     *s++ = '0' + u % 10U;
705                 while (u /= 10U);
706                 while (--s >= buf)
707                     putchar(*s);
708                 continue;
709             }
710         }
711         putchar(c);
712     }
713     va_end(ap);
714 }
715
716 /*
717  * boot encapsulated ABI
718  */
719 void
720 putchar(int c)
721 {
722     if (c == '\n')
723         xputc('\r');
724     xputc(c);
725 }
726
727 /*
728  * boot encapsulated ABI
729  */
730 int
731 drvread(void *buf, unsigned lba, unsigned nblk)
732 {
733     static unsigned c = 0x2d5c7c2f;     /* twiddle */
734
735     c = (c << 8) | (c >> 24);
736     xputc(c);
737     xputc('\b');
738     v86.ctl = V86_ADDR | V86_CALLF | V86_FLAGS;
739     v86.addr = XREADORG;                /* call to xread in boot1 */
740     v86.es = VTOPSEG(buf);
741     v86.eax = lba;
742     v86.ebx = VTOPOFF(buf);
743     v86.ecx = lba >> 16;
744     v86.edx = nblk << 8 | dsk.drive;
745     v86int();
746     v86.ctl = V86_FLAGS;
747     if (V86_CY(v86.efl)) {
748         printf("error %u lba %u\n", v86.eax >> 8 & 0xff, lba);
749         return -1;
750     }
751     return 0;
752 }
753
754 static int
755 keyhit(unsigned ticks)
756 {
757     uint32_t t0, t1;
758
759     if (opts & 1 << RBX_NOINTR)
760         return 0;
761     t0 = 0;
762     for (;;) {
763         if (xgetc(1))
764             return 1;
765         t1 = *(uint32_t *)PTOV(0x46c);
766         if (!t0)
767             t0 = t1;
768         if (t1 < t0 || t1 >= t0 + ticks)
769             return 0;
770     }
771 }
772
773 static void
774 xputc(int c)
775 {
776     if (opts & RBF_VIDEO)
777         putc(c);
778     if (opts & RBF_SERIAL)
779         sio_putc(c);
780 }
781
782 static int
783 xgetc(int fn)
784 {
785     if (opts & 1 << RBX_NOINTR)
786         return 0;
787     for (;;) {
788         if ((opts & RBF_VIDEO) && getc(1))
789             return fn ? 1 : getc(0);
790         if ((opts & RBF_SERIAL) && sio_ischar())
791             return fn ? 1 : sio_getc();
792         if (fn)
793             return 0;
794     }
795 }
796
797 static int
798 getc(int fn)
799 {
800     v86.addr = 0x16;
801     v86.eax = fn << 8;
802     v86int();
803     return fn == 0 ? v86.eax & 0xff : !V86_ZR(v86.efl);
804 }