Clean up yabootconfig kernel/initrd code - Joseph Jezak <josejx@gentoo.org>
[yaboot.git] / ybin / yabootconfig
1 #! /bin/sh
2
3 ###############################################################################
4 ##
5 ## yabootconfig generates a simple /etc/yaboot.conf
6 ## Copyright (C) 2001, 2002, 2003 Ethan Benson
7 ##
8 ## This program is free software; you can redistribute it and/or
9 ## modify it under the terms of the GNU General Public License
10 ## as published by the Free Software Foundation; either version 2
11 ## of the License, or (at your option) any later version.
12 ##
13 ## This program is distributed in the hope that it will be useful,
14 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
15 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 ## GNU General Public License for more details.
17 ##
18 ## You should have received a copy of the GNU General Public License
19 ## along with this program; if not, write to the Free Software
20 ## Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
21 ##
22 ###############################################################################
23
24 PATH="/sbin:/bin:/usr/sbin:/usr/bin:/usr/local/sbin:/usr/local/bin"
25 ## allow to run out of /target in boot-floppies
26 if [ -n "$PATH_PREFIX" ] ; then
27     PATH="${PATH}:${PATH_PREFIX}/sbin:${PATH_PREFIX}/bin:${PATH_PREFIX}/usr/sbin:${PATH_PREFIX}/usr/bin:${PATH_PREFIX}/usr/local/sbin:${PATH_PREFIX}/usr/local/bin"
28 fi
29 PRG="${0##*/}"
30 VERSION=1.0.8
31 CHROOT=/
32 ## $CONFIG is relative to $CHROOT
33 CONFIG=etc/yaboot.conf
34 NOINSTALL=0
35 QUIET=0
36 DEBUG=0
37 SIGINT="$PRG: Interrupt caught ... exiting"
38 export LC_COLLATE=C
39
40 ## avoid older versions of ofpath shipped in debian boot-floppies etc.
41 if [ -x "${PATH_PREFIX}/usr/sbin/ofpath" ] ; then
42     OFPATH="${PATH_PREFIX}/usr/sbin/ofpath"
43 else
44     OFPATH=ofpath
45 fi
46
47 ## catch signals, clean up temporary file
48 trap "cleanup" 0
49 trap "exit 129" 1
50 trap "echo 1>&2 $SIGINT ; exit 130" 2
51 trap "exit 131" 3
52 trap "exit 143" 15
53
54 ## check for printf, use it if possible otherwise fall back on
55 ## unreliable echo -e -n ("SUS" says echo shall support no switches)
56 if [ "$(printf printf_test 2>/dev/null)" = printf_test ] ; then
57     PRINTF=printf
58 else
59     PRINTF="echo -e -n"
60 fi
61
62 ## make sure echo is not lame if we must use it.
63 if [ "$PRINTF" != printf ] ; then
64     if [ "$(echo -e -n echo_test)" != "echo_test" ] ; then
65         echo 1>&2 "$PRG: printf unavailable and echo is broken, sorry."
66         exit 1
67     fi
68 fi
69
70 ## make fake `id' if its missing, outputs 0 since if its missing we
71 ## are probably running on boot floppies and thus are root.
72 if (command -v id > /dev/null 2>&1) ; then
73     true
74 else
75     id()
76     {
77     echo 0
78     }
79 fi
80
81 ## --version output
82 version()
83 {
84 echo \
85 "$PRG $VERSION
86 Written by Ethan Benson
87
88 Copyright (C) 2001, 2002, 2003 Ethan Benson
89 This is free software; see the source for copying conditions.  There is NO
90 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE."
91 }
92
93 ## --help output.
94 usage()
95 {
96 echo \
97 "Usage: $PRG [OPTION]...
98 Generate a working /etc/yaboot.conf.
99
100   -t, --chroot               set root directory $PRG should work from
101   -r, --root                 set root partition, Example: /dev/hda3
102                                default: determined from {chroot}/etc/fstab
103   -b, --boot                 set bootstrap partition, Example: /dev/hda2
104                                default: first type: Apple_Bootstrap partition
105       --kernel-args          add an append= line with specified arguments
106   -q, --quiet                don't ask any questions/confirmation
107       --noinstall            don't automatically run mkofboot
108   -h, --help                 display this help and exit
109   -V, --version              output version information and exit"
110 }
111
112 debug()
113 {
114     [ "$DEBUG" = 0 ] && return 0
115     $PRINTF 1>&2 "$PRG: DEBUG: $1"
116 }
117
118 confirm()
119 {
120     $PRINTF \
121 "yaboot is the Linux Loader for PowerPC.  $PRG sets up your system to boot directly
122 from your hard disk, without the need for a boot CD, floppy or a network boot.\n"
123 [ "$NOINSTALL" = 0 ] && $PRINTF "Install yaboot bootstrap on $BOOT to boot Linux from $ROOT? [Yes] "
124 [ "$NOINSTALL" = 1 ] && $PRINTF "Create simple ${CHROOT}${CONFIG} to boot Linux from $ROOT? [Yes] "
125     read ans
126     case "$ans" in
127         Y|y|Yes|yes|YES|"")
128         echo "Creating a simple ${CHROOT}${CONFIG}..."
129         return 0
130         ;;
131         *)
132         if [ "$NOINSTALL" = 0 ] ; then
133             $PRINTF "Create simple ${CHROOT}${CONFIG} without installing the bootstrap? [Yes] "
134             read ans
135             case "$ans" in
136                 Y|y|Yes|yes|YES|"")
137                 NOINSTALL=1
138                 echo 1>&2 "Creating a simple ${CHROOT}${CONFIG}..."
139                 return 0
140                 ;;
141                 *)
142                 echo "OK, quitting"
143                 return 1
144                 ;;
145             esac
146         else
147             echo "OK, quitting"
148             return 1
149         fi
150         ;;
151     esac
152 }
153
154 ## find out whether we have mac-fdisk or pdisk (they work the same)
155 ckmacfdisk()
156 {
157     if (command -v mac-fdisk > /dev/null 2>&1) ; then
158         FDISK=mac-fdisk
159     elif (command -v pdisk > /dev/null 2>&1) ; then
160         FDISK=pdisk
161     else
162         echo 1>&2 "$PRG: Unable to locate mac-fdisk"
163         return 1
164     fi
165
166     if [ ! -x `command -v $FDISK` 2> /dev/null ] ; then
167         echo 1>&2 "$PRG: `command -v $FDISK`: Permission denied"
168         return 1
169     fi
170     debug "mac-fdisk is: $FDISK\n"
171     return 0
172 }
173
174 ## find out if we have ddisk or fdisk (fdisk for dos labels) debian
175 ## uses both names
176 ckfdisk()
177 {
178     if (command -v ddisk > /dev/null 2>&1) ; then
179         FDISK=ddisk
180     elif (command -v fdisk > /dev/null 2>&1) ; then
181         FDISK=fdisk
182     else
183         echo 1>&2 "$PRG: Unable to locate fdisk"
184         return 1
185     fi
186
187     if [ ! -x `command -v $FDISK` 2> /dev/null ] ; then
188         echo 1>&2 "$PRG: `command -v $FDISK`: Permission denied"
189         return 1
190     fi
191     debug "fdisk is: $FDISK\n"
192     return 0
193 }
194
195 ## find bootstrap partition, supports IBM CHRP with msdos disklabels
196 findbootblock()
197 {
198     ## mac partition table magic == ER
199     if [ "$(dd if="$DISK" bs=2 count=1 2> /dev/null)" = ER ] ; then
200         ckmacfdisk || return 1
201         if [ "$FDISK" = pdisk ] ; then
202             ## handle braindamaged pdisk
203             debug "dealing with pdisk deficiency...\n"
204             BOOT="$(v=`$FDISK -l "$DISK" 2>/dev/null | grep '\<Apple_Bootstrap\>'` ; echo ${v%%:*})"
205             debug "BOOT before fixup: $BOOT\n"
206             if [ -n "$BOOT" ] ; then
207                 BOOT="${DISK}${BOOT}"
208             fi
209             debug "BOOT after fixup: $BOOT\n"
210         else
211             BOOT="$(v=`$FDISK -l "$DISK" 2>/dev/null | grep '\<Apple_Bootstrap\>'` ; echo ${v%%[ ]*})"
212             debug "BOOT=$BOOT\n"
213         fi
214         if [ -z "$BOOT" ] ; then
215             echo 1>&2 "$PRG: Unable to locate bootstrap partition on $DISK..."
216             echo 1>&2 "$PRG: You must create an 800K type: Apple_Bootstrap partition to make the disk bootable"
217             return 1
218         fi
219     else
220         ckfdisk || return 1
221         BOOT="$(v=`$FDISK -l "$DISK" 2>/dev/null | grep '\<PPC PReP Boot\>'` ; echo ${v%%[ ]*})"
222         debug "BOOT=$BOOT\n"
223         if [ -z "$BOOT" ] ; then
224             echo 1>&2 "$PRG: Unable to locate bootstrap partition on $DISK..."
225             echo 1>&2 "$PRG: You must create an 800K type: 0x41 PPC PReP Boot partition to make the disk bootable"
226             return 1
227         fi
228     fi
229     return 0
230 }
231
232 ## if readlink is missing use a kludge
233 if (command -v readlink > /dev/null 2>&1) ; then
234     true
235 else
236     readlink()
237     {
238         SYMTARGET="$(v=`ls -l "$2" 2>/dev/null` ; echo ${v##*> })"
239         if [ -n "$SYMTARGET" ] ; then
240             echo "$SYMTARGET"
241             return 0
242         else
243             return 1
244         fi
245     }
246 fi
247
248 ## we have to do some things differently with a retarded devfs name.
249 ckdevfs()
250 {
251     case "$1" in
252         /dev/ide/*|/dev/scsi/*|/dev/discs/*)
253         return 0
254         ;;
255         *)
256         return 1
257         ;;
258     esac
259 }
260
261 cleanup()
262 {
263     if [ -n "$TMPCONF" ] ; then rm -f "$TMPCONF" ; fi
264     return 0
265 }
266
267 ##########
268 ## Main ##
269 ##########
270
271 if [ $# != 0 ] ; then
272     while true ; do
273         case "$1" in
274             -V|--version)
275                 version
276                 exit 0
277                 ;;
278             -h|--help)
279                 usage
280                 exit 0
281                 ;;
282             -t|--chroot)
283                 if [ -n "$2" ] ; then
284                     CHROOT="$2"
285                     shift 2
286                 else
287                     echo 1>&2 "$PRG: option requires an argument $1"
288                     echo 1>&2 "Try \`$PRG --help' for more information."
289                     exit 1
290                 fi
291                 ;;
292             -b|--boot)
293                 if [ -n "$2" ] ; then
294                     BOOT="$2"
295                     shift 2
296                 else
297                     echo 1>&2 "$PRG: option requires an argument $1"
298                     echo 1>&2 "Try \`$PRG --help' for more information."
299                     exit 1
300                 fi
301                 ;;
302             -r|--root)
303                 if [ -n "$2" ] ; then
304                     ROOT="$2"
305                     shift 2
306                 else
307                     echo 1>&2 "$PRG: option requires an argument $1"
308                     echo 1>&2 "Try \`$PRG --help' for more information."
309                     exit 1
310                 fi
311                 ;;
312             --kernel-args)
313                 if [ -n "$2" ] ; then
314                     KERNARGS="$2"
315                     shift 2
316                 else
317                     echo 1>&2 "$PRG: option requires an argument $1"
318                     echo 1>&2 "Try \`$PRG --help' for more information."
319                     exit 1
320                 fi
321                 ;;
322             -q|--quiet)
323                 QUIET=1
324                 shift 1
325                 ;;
326             --noinstall)
327                 NOINSTALL=1
328                 shift 1
329                 ;;
330             --debug)
331                 DEBUG=1
332                 shift 1
333                 ;;
334             "")
335                 break
336                 ;;
337             *)
338                 echo 1>&2 "$PRG: unrecognized option \`$1'"
339                 echo 1>&2 "Try \`$PRG --help' for more information."
340                 exit 1
341                 ;;
342         esac
343     done
344 fi
345
346 if [ `id -u` != 0 ] ; then
347     echo 1>&2 "$PRG: You are not root, go away"
348     exit 1
349 fi
350
351 ## we need /proc because df does
352 if [ ! -f /proc/uptime ] ; then
353     echo 1>&2 "$PRG: This utility requires the /proc filesystem"
354     exit 1
355 fi
356
357 ## check that chroot exists
358 if [ -d "$CHROOT" ] ; then
359     ## HACK: add trailing / to chroot, otherwise are paths later get b0rked.
360     case "$CHROOT" in
361         */)
362         true
363         ;;
364         *)
365         CHROOT="${CHROOT}/"
366         ;;
367     esac
368 elif [ ! -e "$CHROOT" ] ; then
369     echo 1>&2 "$PRG: $CHROOT: No such file or directory"
370     exit 1
371 elif [ ! -d "$CHROOT" ] ; then
372     echo 1>&2 "$PRG: $CHROOT: Not a directory"
373     exit 1
374 fi
375
376 ## make sure the chroot is an actual root filesystem
377 if [ ! -f "${CHROOT}etc/fstab" ] ; then
378     echo 1>&2 "$PRG: $CHROOT does not appear to be a valid root filesystem"
379     exit 1
380 fi
381
382 ## find / device
383 if [ -z "$ROOT" ] ; then
384     ## IMPORTANT! that last substitution is [<space><tab>] thats all ash will grok
385     ROOT="$(v=`grep '^[^#].*[[:blank:]]/[[:blank:]]' ${CHROOT}etc/fstab` ; echo ${v%%[  ]*})"
386     debug "ROOT=$ROOT\n"
387     if [ -z "$ROOT" ] ; then
388         echo 1>&2 "$PRG: Could not determine root partition, aborting..."
389         exit 1
390     fi
391 fi
392
393 ## dereference label or uuid if necessary
394 case "$ROOT" in
395     LABEL=*|UUID=*)
396         if ! (command -v findfs > /dev/null 2>&1) ; then
397             echo 1>&2 "$PRG: Unable to locate findfs, aborting..."
398             exit 1
399         fi
400         ROOT="$(findfs "$ROOT")"
401         if [ -z "$ROOT" -o $? != 0 ] ; then
402             echo 1>&2 "$PRG: Could not determine root partition, aborting..."
403             exit 1
404         fi
405         ;;
406 esac
407
408 ## make sure root device exists
409 if [ ! -e "$ROOT" ] ; then
410     echo 1>&2 "$PRG: $ROOT: No such file or directory"
411     exit 1
412 fi
413
414 ## find root disk.
415 if ckdevfs "$ROOT" ; then
416     DISK="${ROOT%/*}/disc"
417 else
418     DISK="${ROOT%%[0-9]*}"
419 fi
420 if [ -z "$DISK" ] ; then
421     echo 1>&2 "$PRG: Could not determine root disk, aborting..."
422     exit 1
423 fi
424
425 ## make sure main disk exists
426 if [ ! -e "$DISK" ] ; then
427     echo 1>&2 "$PRG: $DISK: No such file or directory"
428     exit 1
429 fi
430
431 ## find bootstrap partition
432 if [ -z "$BOOT" ] ; then
433     findbootblock || exit 1
434 fi
435
436 ## make sure bootstrap device exists
437 if [ ! -e "$BOOT" ] ; then
438     echo 1>&2 "$PRG: $BOOT: No such file or directory"
439     exit 1
440 fi
441
442 ## sanity check
443 for i in "$DISK" "$ROOT" "$BOOT" ; do
444     if [ ! -b "$i" ] ; then
445         echo 1>&2 "$PRG: $i: Not a block device"
446         exit 1
447     fi
448 done
449
450 ## unless --quiet ask permission to proceed
451 if [ "$QUIET" = 0 ] ; then
452     confirm || exit 2
453 fi
454
455 READLINKKV=`readlink /usr/src/linux`
456 ## find the kernel in the usual places and (if not --quiet) ask the
457 ## user if we cannot find one.
458 for k in "vmlinux" "vmlinux-`uname -r`" "vmlinux-`uname -r`" "$READLINKKV" ; do
459         if [ -f "${CHROOT}${k}" ] ; then
460             KERNEL="${CHROOT}${k}"
461             break;
462         elif [ -f "${CHROOT}boot/${k}" ] ; then
463             KERNEL="${CHROOT}boot/${k}"
464             break;
465         fi
466 done
467
468 for i in "initrd" "initrd-`uname -r`" "`echo $READLINKKV | cut -f1 -d- --complement`" ; do
469     for b in "" "boot/" ; do
470             if [ -f "${CHROOT}${b}${i}" ] ; then
471                 INITRD="${CHROOT}${b}${i}"
472             elif [ -f "${CHROOT}${b}${i}.img" ] ; then
473                 INITRD="${CHROOT}${b}${i}.img"
474             elif [ -f "${CHROOT}${b}${i}.gz" ] ; then
475                 INITRD="${CHROOT}${b}${i}.gz"
476             fi
477     done
478 done
479
480 if [ ! -f "$KERNEL" ] && [ ${QUIET} == 0 ] ; then
481     echo 1>&2 "$PRG: Cannot find a kernel, please provide one"
482     while true ; do
483         if [ "$CHROOT" == "/" ] ; then
484                 $PRINTF 1>&2 "Enter path to a kernel image: "
485         else
486                 $PRINTF 1>&2 "Enter path to a kernel image (not including chroot): "
487         fi
488         read KERN
489
490         ### Remove leading /
491         case "${KERN}" in
492         /*)     KERN=${CHROOT}${KERN:1} ;;
493         *)      KERN=${CHROOT}${KERN} ;;
494         esac
495
496         if [ -f "$KERN" ] ; then
497             KERNEL="$KERN"
498             break
499         elif [ ${KERN} == ${CHROOT} ] ; then
500             echo 1>&2 "$PRG: You must provide a valid kernel"
501         elif [ ! -e "$KERN" ] ; then
502             echo 1>&2 "$PRG: $KERN: No such file or directory"
503         elif [ -d "$KERN" ] ; then
504             echo 1>&2 "$PRG: $KERN: Is a directory"
505         else
506             echo 1>&2 "$PRG: $KERN: Is not a regular file"
507         fi
508     done
509     while true ; do
510         if [ "$CHROOT" == "/" ] ; then
511                 $PRINTF 1>&2 "Enter path to an initrd image (hit Enter for none): "
512         else
513                 $PRINTF 1>&2 "Enter path to an initrd image (hit Enter for none, don't include the chroot): "
514         fi
515         read IRD
516
517         if [ "$IRD" == "" ]; then
518             break
519         fi
520         
521         ### Remove leading /
522         case "${IRD}" in
523         /*)     IRD=${CHROOT}${IRD:1} ;;
524         *)      IRD=${CHROOT}${IRD} ;;
525         esac
526
527         if [ -f "$IRD" ] ; then
528             INITRD="$IRD"
529             break
530         elif [ ! -e "$IRD" ] ; then
531             echo 1>&2 "$PRG: $IRD: No such file or directory"
532         elif [ -d "$IRD" ] ; then
533             echo 1>&2 "$PRG: $IRD: Is a directory"
534         else
535             echo 1>&2 "$PRG: $IRD: Is not a regular file"
536         fi
537     done
538 elif [ ! -f "$KERNEL" ] ; then
539     echo 1>&2 "$PRG: Cannot find a kernel, aborting..."
540     exit 1
541 fi
542
543 debug "KERNEL=$KERNEL\n"
544 debug "INITRD=$INITRD\n"
545
546 ## get partition number the kernel lives on, and the OF device= name
547 ## of the whole disk.
548 KERNDEV="$(v=`df "$KERNEL" 2> /dev/null | grep ^/dev/` ; echo ${v%%[ ]*})"
549 KERNDIR="$(v=`df "$KERNEL" 2> /dev/null | grep ^/dev/` ; echo ${v##*[ ]})"
550 LINKDEV="$(v=`df "${KERNEL%/*}/" 2>/dev/null | grep ^/dev/` ; echo ${v%%[ ]*})"
551 PARTITION="${KERNDEV##*[a-z]}"
552
553 if ckdevfs "$KERNDEV" ; then
554     KERNELDISK="${KERNDEV%/*}/disc"
555 else
556     KERNELDISK="${KERNDEV%%[0-9]*}"
557 fi
558
559 if [ -n "$INITRD" ] ; then
560     ## get partition number the initrd lives on, and the OF device= name
561     ## of the whole disk.
562     IRDDEV="$(v=`df "$INITRD" 2> /dev/null | grep ^/dev/` ; echo ${v%%[ ]*})"
563     IRDDIR="$(v=`df "$INITRD" 2> /dev/null | grep ^/dev/` ; echo ${v##*[ ]})"
564     IRDLINKDEV="$(v=`df "${INITRD%/*}/" 2>/dev/null | grep ^/dev/` ; echo ${v%%[ ]*})"
565     IRDPARTITION="${KERNDEV##*[a-z]}"
566
567     if ckdevfs "$IRDDEV" ; then
568         INITRDDISK="${IRDDEV%/*}/disc"
569     else
570         INITRDDISK="${IRDDEV%%[0-9]*}"
571     fi
572 fi
573
574 debug "KERNEL=$KERNEL\nKERNDEV=$KERNDEV\nKERNDIR=$KERNDIR\nLINKDEV=$LINKDEV\nPARTITION=$PARTITION\nKERNELDISK=$KERNELDISK\n"
575
576 if [ -n "$INITRD" ] ; then
577     debug "INITRD=$INITRD\nIRDDEV=$IRDDEV\nIRDDIR=$IRDDIR\nIRDLINKDEV=$IRDLINKDEV\nIRDPARTITION=$IRDPARTITION\nINITRDDISK=$INITRDDISK\n"
578 fi
579
580 ## sanity check
581 for i in "$KERNDEV" "$KERNDIR" "$LINKDEV" "$PARTITION" "$KERNELDISK" ; do
582     if [ -z "$i" ] ; then
583         echo 1>&2 "$PRG: Could not determine necessary information, aborting..."
584         echo 1>&2 "$PRG: Are you using chroot $PRG instead of $PRG --chroot ?"
585         exit 1
586     fi
587 done
588
589 if [ -n "$INITRD" ] ; then
590     ## sanity check
591     for i in "$IRDDEV" "$IRDDIR" "$IRDLINKDEV" "$IRDPARTITION" "$INITRDDISK" ; do
592         if [ -z "$i" ] ; then
593             echo 1>&2 "$PRG: Could not determine necessary information, aborting..."
594             echo 1>&2 "$PRG: Are you using chroot $PRG instead of $PRG --chroot ?"
595             exit 1
596         fi
597     done
598 fi
599
600 ## check for cross device symlink
601 if [ -L "$KERNEL" ] ; then
602     if [ "$KERNDEV" != "$LINKDEV" ] ; then
603         echo 1>&2 "$PRG: Warning: Cross device symlink $KERNEL, using it's target instead"
604         KERNEL="$(readlink -f "$KERNEL" 2>/dev/null)"
605         if [ ! -f "$KERNEL" ] ; then
606             echo 1>&2 "$PRG: Unable to canonicalize symlink's target.  Do not create cross device symlinks."
607             exit 1
608         fi
609     fi
610 fi
611
612 if [ -n "$INITRD" ] ; then
613     ## initrd must be on same device as kernel.
614     if [ "$IRDDEV" != "$KERNDEV" -o "$IRDPARTITION" != "$IRDPARTITION" -o "$INITRDDISK" != "$KERNELDISK" ] ; then
615         echo 1>&2 "$PRG: Initrd image must be on same device as kernel image."
616         exit 1
617     fi
618
619     ## check for cross device symlink
620     if [ -L "$INITRD" ] ; then
621         if [ "$IRDDEV" != "$IRDLINKDEV" ] ; then
622             echo 1>&2 "$PRG: Warning: Cross device symlink $INITRD, using it's target instead"
623             INITRD="$(readlink -f "$INITRD" 2>/dev/null)"
624             if [ ! -f "$INITRD" ] ; then
625                 echo 1>&2 "$PRG: Unable to canonicalize symlink's target.  Do not create cross device symlinks."
626                 exit 1
627             fi
628         fi
629     fi
630 fi
631
632 ## only powermacs appear to need device=
633 if (cat /proc/cpuinfo 2>/dev/null | grep -q pmac-generation 2> /dev/null) ; then
634     DEVICE="\ndevice=$($OFPATH $KERNELDISK)"
635     if [ $? != 0 ] ; then
636         echo 1>&2 "$PRG: Unable to determine OpenFirmware device name to $KERNELDISK, aborting..."
637         exit 1
638     fi
639 fi
640
641 ## if there is a separate /boot partition we must strip off the /boot
642 ## mountpoint or else yaboot will not find the kernel.
643 if [ "$KERNDIR" != "$CHROOT" ] ; then
644     IMAGE="${KERNEL##*$KERNDIR}"
645 else
646     IMAGE="$KERNEL"
647 fi
648
649 ## fix chrooted path
650 if [ "$CHROOT" != / ] ; then
651     IMAGE="${IMAGE##*$CHROOT}"
652 fi
653
654 ## fix relative path (caused by chroot path fix)
655 case "$IMAGE" in
656     /*)
657     true
658     ;;
659     *)
660     IMAGE="/${IMAGE}"
661     ;;
662 esac
663
664 if [ -n "$INITRD" ] ; then
665     ## if there is a separate /boot partition we must strip off the /boot
666     ## mountpoint or else yaboot will not find the kernel.
667     if [ "$IRDDIR" != "$CHROOT" ] ; then
668         INITRDIMG="${INITRD##*$IRDDIR}"
669     else
670         INITRDIMG="$INITRD"
671     fi
672
673     ## fix chrooted path
674     if [ "$CHROOT" != / ] ; then
675         INITRDIMG="${INITRDIMG##*$CHROOT}"
676     fi
677
678     ## fix relative path (caused by chroot path fix)
679     case "$INITRDIMG" in
680         /*)
681             true
682             ;;
683         *)
684             INITRDIMG="/${INITRDIMG}"
685             ;;
686     esac
687 fi
688
689 ## figure out if yaboot is installed in /usr/local or not
690 if [ -f /usr/local/lib/yaboot/yaboot ] ; then
691     INSTALL=/usr/local/lib/yaboot/yaboot
692 elif [ -f /usr/lib/yaboot/yaboot ] ; then
693     INSTALL=/usr/lib/yaboot/yaboot
694 else
695     echo 1>&2 "$PRG: yaboot is not installed correctly"
696     exit 1
697 fi
698
699 ## newworld powermacs need the ofboot first stage loader
700 if [ "$(v=`cat /proc/cpuinfo 2>/dev/null | grep pmac-generation` ; echo ${v##*:})" = NewWorld ] ; then
701     if [ -f /usr/local/lib/yaboot/ofboot ] ; then
702         OFBOOT="\nmagicboot=/usr/local/lib/yaboot/ofboot"
703     elif [ -f /usr/lib/yaboot/ofboot ] ; then
704         OFBOOT="\nmagicboot=/usr/lib/yaboot/ofboot"
705     else
706         echo 1>&2 "$PRG: yaboot is not installed correctly"
707         exit 1
708     fi
709 fi
710
711 ## check for properly (read debian) packaged yaboot.
712 if [ -d ${CHROOT}usr/share/doc/yaboot/examples ] ; then
713     HEADER="## see also: /usr/share/doc/yaboot/examples for example configurations.\n"
714 fi
715
716 ## setup append line
717 if [ -n "$KERNARGS" ] ; then
718     APPEND="\tappend=\"${KERNARGS}\"\n"
719 fi
720
721 ## avoid user confusion when they boot an installer with video=ofonly
722 ## (usually via a install-safe label) and then reboot and have the box
723 ## not boot properly again.
724 if [ -z "$APPEND" ] ; then
725     if (grep -q '\<video=ofonly\>' /proc/cmdline 2> /dev/null) ; then
726         APPEND="\tappend=\"video=ofonly\"\n"
727     fi
728 fi
729
730 ## generate initrd= lines
731 if [ -n "$INITRDIMG" ] ; then
732     INITRDIMGS="\tinitrd=$INITRDIMG\n\tinitrd-size=8192\n"
733 fi
734
735 ## generate global section of yaboot.conf
736 GLOBAL="## yaboot.conf generated by $PRG $VERSION
737 ##
738 ## run: \"man yaboot.conf\" for details. Do not make changes until you have!!
739 ${HEADER}##
740 ## For a dual-boot menu, add one or more of:
741 ## bsd=/dev/hdaX, macos=/dev/hdaY, macosx=/dev/hdaZ\n
742 boot=${BOOT}${DEVICE:-}
743 partition=$PARTITION
744 root=$ROOT
745 timeout=30
746 install=${INSTALL}${OFBOOT:-}\n"
747
748 ## generate image= section
749 IMAGES="
750 image=$IMAGE
751 \tlabel=Linux
752 \tread-only\n${APPEND:-}${INITRDIMGS:-}"
753
754 ## safely create a tmp file then move it into place after we are sure
755 ## it was written.
756 TMPCONF=`mktemp -q "${CHROOT}${CONFIG}.XXXXXX"`
757 if [ $? != 0 ] ; then
758     echo 1>&2 "$PRG: Unable to write to ${CHROOT}${CONFIG%/*}"
759     exit 1
760 fi
761
762 $PRINTF "${GLOBAL}${IMAGES}" > "$TMPCONF"
763 if [ $? != 0 ] ; then
764     echo 1>&2 "$PRG: Unable to write temporary file ${TMPCONF}, aborting..."
765     exit 1
766 fi
767
768 ## rotate backups of /etc/yaboot.conf, 3 backups are kept
769 if [ -f "${CHROOT}${CONFIG}.old" ] ; then
770     for i in 1 0 ; do
771         if [ -f "${CHROOT}${CONFIG}.old.${i}" ] ; then
772             mv -f "${CHROOT}${CONFIG}.old.$i" "${CHROOT}${CONFIG}.old.$(($i + 1))"
773             if [ $? != 0 ] ; then
774                 echo 1>&2 "$PRG: Unable to make backup of existing ${CHROOT}${CONFIG}.old.$i, aborting..."
775                 exit 1
776             fi
777         fi
778     done
779
780     mv -f "${CHROOT}${CONFIG}.old" "${CHROOT}${CONFIG}.old.0"
781     if [ $? != 0 ] ; then
782         echo 1>&2 "$PRG: Unable to make backup of existing ${CHROOT}${CONFIG}.old, aborting..."
783         exit 1
784     fi
785 fi
786
787 ## backup /etc/yaboot.conf
788 if [ -f "${CHROOT}${CONFIG}" ] ; then
789     mv -f "${CHROOT}${CONFIG}" "${CHROOT}${CONFIG}.old"
790     if [ $? != 0 ] ; then
791         echo 1>&2 "$PRG: Unable to make backup of existing ${CHROOT}${CONFIG}, aborting..."
792         exit 1
793     fi
794 fi
795
796 ## move new config into place
797 mv -f "${TMPCONF}" "${CHROOT}${CONFIG}"
798 if [ $? != 0 ] ; then
799     echo "$PRG: Unable to write file ${CHROOT}${CONFIG}"
800     exit 1
801 else
802     ## nothing sensitive in generated config, comply with debian policy
803     chmod 644 "${CHROOT}${CONFIG}"
804 fi
805
806 ## tell mkofboot where to find the config file if necessary
807 if [ "${CHROOT}${CONFIG}" != /etc/yaboot.conf ] ; then
808     YBINARGS="-C ${CHROOT}${CONFIG}"
809 fi
810
811 ## run mkofboot to install the bootstrap, unless --noinstall
812 if [ "$NOINSTALL" = 0 ] ; then
813     if (command -v mkofboot 2>&1 > /dev/null) ; then
814         [ "$QUIET" = 0 ] && echo "Running mkofboot to make the disk bootable..."
815         mkofboot -f $YBINARGS || exit 1
816         [ "$QUIET" = 0 ] && echo "Done"
817     else
818         echo 1>&2 "$PRG: yaboot is not installed correctly, not running mkofboot"
819         exit 1
820     fi
821 fi
822
823 exit 0