core/scripts/splash_manager.in

   1 #!/bin/bash
   2 #
   3 # splash_manager -- a convenience utility to perform various splash-related
   4 #                   tasks in an easy and user-friendly way
   5 #
   6 # Copyright (C) 2005-2007, Michal Januszewski <spock@gentoo.org>
   7 #
   8 # This file is a part of the splashutils package.
   9 #
  10 # This file is subject to the terms and conditions of the GNU General Public
  11 # License v2.  See the file COPYING in the main directory of this archive for
  12 # more details.
  13 #
  14
  15 # TODO:
  16 # - support for icons and scripts in themes
  17 # - support for '-v'
  18 # - support for 8bpp modes
  19
  20 # BUGS:
  21 # - replay won't work correctly when started from a tty with a video mode
  22 #   different from the one used for the silent tty
  23
  24 # System configuration variables
  25 spl_util=splash_util.static
  26 spl_daemon=fbsplashd.static
  27 spl_decor=fbcondecor_ctl
  28 spl_dir="@libdir@/splash"
  29 spl_fifo=${spl_dir}/cache/.splash
  30 spl_pidfile=${spl_dir}/cache/daemon.pid
  31 themedir="@themedir@"
  32
  33 spl_daemon_exename="$(basename ${spl_daemon})"
  34 spl_daemon_short="${spl_daemon:0:9}"
  35
  36 cleanup() {
  37         killall ${spl_daemon_exename} 2>/dev/null
  38         chvt ${ctty}
  39 }
  40
  41 usage() {
  42         cat <<EOTB
  43 splash_geninitramfs/splashutils-@PACKAGE_VERSION@
  44 Usage: splash_manager [options] -c <command>
  45
  46 Commands:
  47  benchmark
  48  demo    test a splash theme
  49  list    list available splash themes
  50  set     set a splash theme on a specific tty
  51  shot    take a screenshot of a selected splash theme
  52  switch  switch a splash theme by changing GRUB/LILO config files
  53  replay  replay a complete boot sequence using splash profile data
  54
  55 Options:
  56  -c, --cmd=CMD      execute CMD
  57  -h, --help         display this message
  58  -t, --theme=THEME  [bdehsr] theme name
  59      --steps=N      [ d    ] use N steps when testing the silent splash
  60      --delay=N      [ d    ] delay each step by N seconds
  61  -m, --mode=(v|s)   [ d h  ] specify which mode to use (verbose/silent)
  62      --tty=N        [bdeh  ] set the theme on the Nth tty
  63  -e, --exec=CMD     [   h  ] run CMD on the tty where splash is being tested
  64      --progress=N   [   h  ] set progress to N (0-65535)
  65      --msg=MSG      [bd h  ] set the main system message to MSG
  66      --profile=FILE [     r] read splash profile data from FILE
  67
  68 The letters in the square brackets indicate the commands with which
  69 specific options can be used: b (bechmark) d (demo), e (set), h (shot),
  70 s (switch), r (replay).
  71
  72 Examples:
  73  splash_manager -c switch -t livecd-2005.0
  74    (switch to the 'livecd-2005.0' theme; the new theme will be used
  75     after reboot)
  76
  77  splash_manager -c set -t linux --tty=8
  78    (set the 'linux' theme on the 8th tty)
  79
  80  splash_manager -c demo -t livecd-2005.0 -m s --steps=100
  81    (run a demonstration of the 'livecd-2005.0' theme, silent mode;
  82     use 100 steps for the progress bar)
  83
  84  splash_manager -c shot -t foobar -m s --progress=32767 --msg='Smile!'
  85    (take a screenshot of the 'foobar' theme, silent mode; set progress
  86     to 50% and use 'Smile!' as the main system message)
  87 EOTB
  88 }
  89
  90 CGOOD=$'\e[32;01m'
  91 CWARN=$'\e[33;01m'
  92 CBAD=$'\e[31;01m'
  93 CNORM=$'\e[0m'
  94 CHILITE=$'\e[36;01m'
  95 CBRACKET=$'\e[34;01m'
  96
  97 # void err(char *msg)
  98 #   Displays an error message.
  99 err() {
 100         echo -e "${CBAD}Error:${CNORM} $*" >&2
 101 }
 102
 103 # void einfo(char *msg)
 104 #   Displays an informational message.
 105 einfo() {
 106         echo -e "${CGOOD}*${CNORM} $*"
 107 }
 108
 109 # void ewarn(char *msg0
 110 #   Display a warning message.
 111 ewarn() {
 112         echo -e "${CWARN}*${CNORM} $*"
 113 }
 114
 115 # void spl_comm(char *msg)
 116 #   Sends 'msg' to the splash fifo.
 117 spl_comm() {
 118         if [ ! -e ${spl_pidfile} ]; then
 119                 return 1
 120         fi
 121
 122         if [ "$(pgrep ${spl_daemon_short})" == "$(<${spl_pidfile})" ]; then
 123                 echo "$*" > ${spl_fifo} &
 124         else
 125                 err "Splash daemon not running!"
 126                 rm -f "${spl_pidfile}"
 127         fi
 128 }
 129
 130 # void chvt(int n)
 131 #   Switches to the n-th tty.
 132 chvt() {
 133         local ntty=$1
 134
 135         if [ -x /usr/bin/chvt ] ; then
 136                 /usr/bin/chvt ${ntty}
 137         else
 138                 echo -en "\e[12;${ntty}]"
 139         fi
 140 }
 141
 142
 143 # void theme_sanity_check(char *theme)
 144 #   Performs a theme sanity check -- checks whether a theme exists.
 145 theme_sanity_check() {
 146         local theme=$1
 147         local t=${res}
 148
 149         # For verbose mode, we care about the real resolution, not the one
 150         # fbsplash would be using for the silent mode.
 151         if [ "${mode}" == "v" ]; then
 152                 t=${rres}
 153         fi
 154
 155         if [ -z "${theme}" ]; then
 156                 err "You have to specify a theme name."
 157                 exit 1
 158         fi
 159
 160         if [ ! -e "${themedir}/${theme}/${t}.cfg" ]; then
 161                 err "Theme '${theme}' doesn't seem to provide a config file for the current resolution (${t})."
 162                 exit 1
 163         fi
 164 }
 165
 166 # void set_theme(char *theme, int tty)
 167 #   Sets a theme on a specific tty.
 168 set_theme() {
 169         local theme=$1
 170         local tty=$2
 171
 172         theme_sanity_check "${theme}"
 173         [ -z "${tty}" ] && tty="${ctty}"
 174         ${spl_decor} -c setcfg -t "${theme}" --tty="${tty}"
 175         [ "${tty}" == "${ctty}" ] && ${spl_decor} -c setpic -t "${theme}" --tty="${tty}"
 176         ${spl_decor} -c on --tty=${tty}
 177 }
 178
 179 # void get_theme(int tty)
 180 #   Prints the name of the theme that is used on a tty.
 181 get_theme() {
 182         local ctty=$1
 183         echo $(${spl_decor} --tty=${ctty} -c getcfg | grep theme | awk '{ print $2 }')
 184 }
 185
 186 # void get_state(int tty)
 187 #   Prints 'on' or 'off', indicating fbsplash state on a tty.
 188 get_state() {
 189         local ctty=$1
 190         echo $(${spl_decor} --tty=${ctty} -c getstate | cut -f2 -d: | tr -d ' ')
 191 }
 192
 193 # void check_silent(char *theme, char *res, char *action)
 194 #   Checks whether a theme supports the silent splash mode.
 195 check_silent() {
 196         local ctheme=$1
 197         local cres=$2
 198         local action=$3
 199
 200         if [ -z $(cat "${themedir}/${ctheme}/${cres}.cfg" 2>/dev/null | egrep "^silent(pic|jpeg)=") ]; then
 201                 [ "${action}" == "test" ] && return 1
 202                 err "Theme '${ctheme}' doesn't support the silent splash mode."
 203                 exit 1
 204         fi
 205 }
 206
 207 # void check_verbose(char *theme, char *res, char *action)
 208 #   Checks whether a theme supports the verbose splash mode.
 209 check_verbose() {
 210         local ctheme=$1
 211         local cres=$2
 212         local action=$3
 213
 214         if [ -z $(cat "${themedir}/${ctheme}/${cres}.cfg" 2>/dev/null | egrep "^(pic|jpeg)=") ]; then
 215                 [ "${action}" == "test" ] && return 1
 216                 err "Theme '${ctheme}' doesn't support the silent splash mode."
 217                 exit 1
 218         fi
 219
 220         return 0
 221 }
 222
 223 set_silent() {
 224         trap "cleanup" EXIT
 225
 226         # Make sure the splash daemon is not running
 227         killall -9 ${spl_daemon_exename} 2>/dev/null
 228         ${spl_daemon} -t "${theme}" --pidfile ${spl_pidfile} ${util_params}
 229         pid=$(cat ${spl_pidfile})
 230
 231         [ -z "${msg}" ] && msg="Testing the '${theme}' theme (\$progress%)..."
 232         spl_comm "set message ${msg}"
 233         [ -n "${tty}" ] && spl_comm "set tty silent ${tty}"
 234         spl_comm "set mode silent"
 235         spl_comm "repaint"
 236 }
 237
 238 exit_silent() {
 239         spl_comm "exit"
 240
 241         while [[ "$(head -n 1 /proc/${pid}/status 2>/dev/null | cut -f2)" == "${spl_daemon_short}" ]]; do
 242                 sleep 0.5
 243         done
 244
 245         while [ "`fgconsole`" != ${ctty} ] ; do
 246                 chvt ${ctty}
 247         done
 248 }
 249
 250 benchmark() {
 251         check_silent "${theme}" "${res}"
 252         set_silent
 253
 254         step=16
 255         timestart=$(date +"%s.%N")
 256         for ((i=0;i<65536;i=$i+$step)) ; do
 257                 echo "progress $i" > ${spl_fifo}
 258                 echo "paint" > ${spl_fifo}
 259         done
 260         timeend=$(date +"%s.%N")
 261         exit_silent
 262
 263         echo "Test took $(echo "$timeend - $timestart" | bc) seconds."
 264 }
 265
 266 demo_theme_silent() {
 267         check_silent "${theme}" "${res}"
 268         if [ -z "$(echo ${util_params} | grep -- '--type=')" ]; then
 269                 export util_params="${util_params} --type=bootup"
 270         fi
 271         set_silent
 272
 273         # Set the input device if it exists. This will make it possible to use F2 to
 274         # switch from verbose to silent.
 275         local t=$(grep -Hsi keyboard /sys/class/input/input*/name | sed -e 's#.*input\([0-9]*\)/name.*#event\1#')
 276         if [[ -z "${t}" ]]; then
 277                 t=$(grep -Hsi keyboard /sys/class/input/event*/device/driver/description | grep -o 'event[0-9]\+')
 278                 if [[ -z "${t}" ]]; then
 279                         # Try an alternative method of finding the event device. The idea comes
 280                         # from Bombadil <bombadil(at)h3c.de>. We're couting on the keyboard controller
 281                         # being the first device handled by kbd listed in input/devices.
 282                         t=$(/bin/grep -s -m 1 '^H: Handlers=kbd' /proc/bus/input/devices | grep -o 'event[0-9]*')
 283                 fi
 284         fi
 285
 286         if [[ -n "${t}" ]]; then
 287                 spl_comm "set event dev /dev/input/${t}"
 288         fi
 289
 290         if [[ -n "$(pgrep gpm)" ]]; then
 291                 spl_comm "set gpm"
 292         fi
 293
 294         [ -z "${delay}" ] && delay=0.05
 295         step=$((65535/$steps))
 296
 297         for ((i=0;i<65536;i=$i+$step)) ; do
 298                 spl_comm "progress $i"
 299                 spl_comm "paint"
 300                 sleep ${delay}
 301         done
 302         spl_comm "progress 65535"
 303         spl_comm "paint"
 304         exit_silent
 305 }
 306
 307 replay() {
 308         check_silent "${theme}" "${res}"
 309
 310         if [[ -z "${profile}" ]]; then
 311                 err "You have to specify a splash profile file."
 312                 exit 1
 313         fi
 314
 315         if [[ ! -r "${profile}" ]]; then
 316                 err "Profile file '${profile}' not found."
 317                 exit 1
 318         fi
 319
 320         trap "cleanup" EXIT
 321
 322         ctime=0
 323
 324         if [[ -e /etc/gentoo-release ]]; then
 325                 export RUNLEVEL="S"
 326                 export SOFTLEVEL=""
 327         fi
 328
 329         source @sbindir@/splash-functions.sh
 330         splash_setup
 331
 332         export SPLASH_XRES="$xres"
 333         export SPLASH_YRES="$yres"
 334         export SPLASH_THEME=${theme}
 335
 336         while read line ; do
 337                 arr=($line)
 338                 t=${arr[0]:0:$((${#arr[0]}-1))}
 339                 tdec=${t##*.}
 340                 utime=$((${t%.*}*1000 + ${tdec#0*}*10))
 341
 342                 [[ "$ctime" == "0" ]] && ctime=${utime}
 343
 344                 delay=$((${utime} - ${ctime}))
 345                 sleep "$(($delay/1000)).$(printf '%03d' $(($delay%1000)))"
 346                 ctime=${utime}
 347
 348                 # Start the splash daemon after rc_init/pre.
 349                 if [[ -z "${pid}" ]]; then
 350                         if [[ "${arr[1]}" != "pre" || "${arr[2]}" != "rc_init" ]]; then
 351                                 BOOT_MSG="$(splash_get_boot_message)" ${spl_daemon} -t ${theme} --pidfile ${spl_pidfile} --type=bootup
 352                                 pid=$(cat ${spl_pidfile})
 353                         fi
 354                 fi
 355
 356                 case ${arr[1]} in
 357                         pre)
 358                                 if [[ -x ${themedir}/${theme}/scripts/${arr[2]}-pre ]]; then
 359                                         ${themedir}/${theme}/scripts/${arr[2]}-pre ${arr[*]:3}
 360                                 fi
 361                                 ;;
 362
 363                         post)
 364                                 if [[ -x ${themedir}/${theme}/scripts/${arr[2]}-post ]]; then
 365                                         ${themedir}/${theme}/scripts/${arr[2]}-post ${arr[*]:3}
 366                                 fi
 367
 368                                 if [[ -e /etc/gentoo-release && ${arr[3]} == 'boot' ]]; then
 369                                         export RUNLEVEL="3"
 370                                 fi
 371                                 ;;
 372
 373                         comm)
 374                                 spl_comm "${arr[*]:2}"
 375                                 ;;
 376                 esac
 377         done < ${profile}
 378 }
 379
 380 set_verbose() {
 381         [ -z "${tty}" ]         && tty=${ctty}
 382
 383         export cstate=$(get_state ${tty})
 384         [ "${cstate}" == "on" ] && export ctheme=$(get_theme ${tty})
 385
 386         set_theme "${theme}" "${tty}"
 387         chvt "${tty}"
 388 }
 389
 390 exit_verbose() {
 391         chvt "${ctty}"
 392         if [ "${cstate}" == "on" ]; then
 393                 set_theme "${ctheme}" "${tty}"
 394         else
 395                 ${spl_decor} --tty=${tty} -c off
 396         fi
 397 }
 398
 399 demo_theme_verbose() {
 400         [ -z "${delay}" ]       && delay=5
 401
 402         check_verbose "${theme}" "${res}"
 403         set_verbose
 404
 405         einfo "Testing the '${theme}' theme..." >/dev/tty${tty}
 406         for (( i = 0 ; i < ${delay} ; i = i+1 )) ; do
 407                 echo -n "$i " >/dev/tty${tty}
 408                 sleep 1
 409         done
 410         echo ""
 411
 412         exit_verbose
 413 }
 414
 415 take_shot_silent() {
 416         check_silent "${theme}" "${res}"
 417         set_silent
 418
 419         [ -z "${progress}" ] && progress=16384
 420         spl_comm "progress ${progress}"
 421         spl_comm "repaint"
 422
 423         sleep 1
 424         fbgrab ${theme}-${res}-silent.png
 425
 426         exit_silent
 427 }
 428
 429 take_shot_verbose() {
 430         check_verbose "${theme}" "${res}"
 431         set_verbose
 432
 433         if [ -n "${cexec}" ]; then
 434                 ps -A -o pid,comm,tty > /tmp/spl-shot-$$.pre
 435                 openvt -f -c ${tty} -- ${cexec}
 436                 sleep 2
 437                 ps -A -o pid,comm,tty > /tmp/spl-shot-$$.post
 438                 local ce=$(basename "${cexec}")
 439                 pid=$(diff /tmp/spl-shot-$$.pre /tmp/spl-shot-$$.post | grep "${ce:0:15}" | grep "tty${tty}" | grep '^>' | awk '{ print $2 }')
 440                 rm -f /tmp/spl-shot-$$.pre /tmp/spl-shot-$$.post
 441         fi
 442
 443         fbgrab ${theme}-${res}-verbose.png
 444
 445         if [[ -n "${ce}" && -n "${pid}" &&
 446                   "$(cat /proc/${pid}/status 2>/dev/null | grep 'Name:' | cut -f2)" == "${ce:0:15}" ]]; then
 447                 kill "${pid}"
 448         fi
 449         exit_verbose
 450 }
 451
 452 list_themes() {
 453         for i in "${themedir}"/* ; do
 454                 [ ! -d "${i}" ] && continue
 455                 [ -z "$(ls "${i}"/*.cfg 2>/dev/null)" ] && continue
 456
 457                 tn="${i/$themedir/}"
 458                 tn="${tn//\//}"
 459
 460                 echo -n "$tn:"
 461                 local first=true
 462
 463                 for res in "${i}"/*.cfg ; do
 464                         res=${res##*/}
 465                         res=${res/.cfg/}
 466
 467                         [ "$first" != "true" ] && echo -n ","
 468                         first=false
 469                         echo -n " ${res}"
 470
 471                         local sil ver
 472
 473                         if check_silent "$tn" "$res" "test"; then
 474                                 sil=true
 475                         else
 476                                 sil=false
 477                         fi
 478
 479                         if check_verbose "$tn" "$res" "test"; then
 480                                 ver=true
 481                         else
 482                                 ver=false
 483                         fi
 484
 485                         if [[ "$ver" == "true" && "$sil" == "true" ]]; then
 486                                 echo -n " (sv)"
 487                         elif [[ "$sil" == "true" ]]; then
 488                                 echo -n " (s)"
 489                         else
 490                                 echo -n " (v)"
 491                         fi
 492                 done
 493                 echo ""
 494         done
 495 }
 496
 497 bootloader=""
 498
 499 detect_bootloader() {
 500         if [[ -e /boot/grub/grub.conf && -e /etc/lilo.conf ]]; then
 501                 ewarn "Both GRUB and LILO seem to be installed and configured. Please"
 502                 ewarn "select which one should be configured by this script:"
 503                 ewarn "  1) GRUB"
 504                 ewarn "  2) LILO"
 505                 while true; do
 506                         echo -n "Your choice [1]: "
 507                         read choice
 508                         if [ -z "${choice}" ]; then
 509                                 bootloader="grub"
 510                                 return
 511                         elif [ "${choice}" == 1 ]; then
 512                                 bootloader="grub"
 513                                 return
 514                         elif [ "${choice}" == 2 ]; then
 515                                 bootloader="lilo"
 516                                 return
 517                         else
 518                                 err "You have chosen an invalid number."
 519                         fi
 520                 done
 521         elif [[ -e /boot/grub/grub.conf ]]; then
 522                 bootloader="grub"
 523         elif [[ -e /etc/lilo.conf ]]; then
 524                 bootloader="lilo"
 525         else
 526                 err "It appears that none of the supported bootloaders (GRUB, LILO) is installed"
 527                 err "on your system. Please install one of these or configure fbsplash/splashutils"
 528                 err "manually."
 529                 exit 1
 530         fi
 531 }
 532
 533 # The code of mount_boot_partition() is stolen from Gentoo's mount-boot.eclass.
 534 mount_boot_partition() {
 535         # note that /dev/BOOT is in the Gentoo default /etc/fstab file
 536         local fstabstate="$(cat /etc/fstab | awk '!/^#|^[[:blank:]]+#|^\/dev\/BOOT/ {print $2}' | egrep "^/boot$" )"
 537         local procstate="$(cat /proc/mounts | awk '{print $2}' | egrep "^/boot$" )"
 538         local proc_ro="$(cat /proc/mounts | awk '{ print $2, $4 }' | sed -n '/\/boot/{ /[ ,]\?ro[ ,]\?/p }' )"
 539
 540         if [ -n "${fstabstate}" ] && [ -n "${procstate}" ]; then
 541                 if [ -n "${proc_ro}" ]; then
 542                         einfo "Your boot partition, detected as being mounted as /boot, is read-only."
 543                         einfo "Remounting it in read-write mode ..."
 544                         mount -o remount,rw /boot &>/dev/null
 545                         if [ "$?" -ne 0 ]; then
 546                                 err "Unable to remount in rw mode. Please do it manually!"
 547                                 exit 1
 548                         fi
 549                 else
 550                         einfo "Your boot partition was detected as being mounted as /boot."
 551                 fi
 552         elif [ -n "${fstabstate}" ] && [ -z "${procstate}" ]; then
 553                 mount /boot -o rw &>/dev/null
 554                 if [ "$?" -eq 0 ]; then
 555                         einfo "Your boot partition was not mounted as /boot, but this"
 556                         einfo "script was able to mount it without additional intervention."
 557                 else
 558                         err "Cannot automatically mount your /boot partition."
 559                         err "Your boot partition has to be mounted rw before the splash configuration"
 560                         err "can be continued."
 561                         exit 1
 562                 fi
 563         else
 564                 einfo "Assuming you do not have a separate /boot partition."
 565         fi
 566 }
 567
 568 # int input_number(int min, int max, int default)
 569 #   Asks the user to enter a number in the range [min;max].
 570 input_number() {
 571         local min=$1
 572         local max=$2
 573         local default=$3
 574         local done=false
 575
 576         echo -n "Your choice: "
 577
 578         while [[ ${done} != "true" ]]; do
 579                 read n
 580                 n=${n%% *}
 581
 582                 if [[ -z "${n}" || "${n}" -lt ${min} || "${n}" -gt ${max} ]]; then
 583                         err "Invalid number. Try again. Enter a number in the range: ${min}-${max}}."
 584                         echo -n "Your choice: "
 585                 else
 586                         done="true"
 587                 fi
 588         done
 589
 590         return ${n}
 591 }
 592
 593 # bool input_bool(bool default)
 594 #   Asks the user to enter 'y' or 'n'.
 595 input_bool() {
 596         local default=$1
 597         local choice
 598
 599         echo -n "Your choice "
 600
 601         if [[ "${default}" == "y" ]]; then
 602                 echo -n "[Y/n]: "
 603         else
 604                 echo -n "[y/N]: "
 605         fi
 606
 607         read choice
 608
 609         if [[ "${default}" == "y" ]]; then
 610                 if [[ "${choice}" != "n" && "${choice}" != "N" ]]; then
 611                         return 0
 612                 else
 613                         return 1
 614                 fi
 615         else
 616                 if [[ "${choice}" != "y" && "${choice}" != "Y" ]]; then
 617                         return 1
 618                 else
 619                         return 0
 620                 fi
 621         fi
 622 }
 623
 624 # void theme_switch_initrd(char *initrd)
 625 #   Asks the user whether (s)he wants to generate and use a new initrd.
 626 theme_switch_initrd() {
 627         local initrd=$1
 628         newinitrd=""
 629         want_initrd=false
 630
 631         if [ -z "${initrd}" ]; then
 632                 einfo "It appears that you are not using an initrd. Do you want this script"
 633                 einfo "to automatically generate one for use with the '${theme}' theme?"
 634                 if input_bool 'y'; then
 635                         want_initrd=true
 636                 fi
 637         elif [ "${initrd:0:13}" == "/boot/splash/" ]; then
 638                 want_initrd=true
 639         else
 640                 echo ""
 641                 einfo "It appears that you are already using an initrd:"
 642                 einfo "  '${initrd}'"
 643                 einfo "Do you want this script to automatically generate another initrd for"
 644                 einfo "use with the '${theme}' theme?"
 645                 einfo "The new initrd will be used instead of the current one."
 646                 if input_bool 'n'; then
 647                         want_initrd=true
 648                 fi
 649         fi
 650
 651         if [ ${want_initrd} == "true" ]; then
 652                 [ ! -d /boot/splash ] && mkdir /boot/splash
 653                 newinitrd="/boot/splash/initrd-spl-${theme}-all"
 654                 splash_geninitramfs -g "${newinitrd}" -v "${theme}"
 655         fi
 656 }
 657
 658 theme_switch_lilo() {
 659         local cnt=0
 660         local entries
 661         local lines
 662         local want_name=false
 663         local linecnt=1
 664
 665         # Display a list of entries from lilo.conf and ask the user to select one of them.
 666         einfo "Please select which kernel should be configured for use with splashutils:"
 667         while read i ; do
 668                 line=$(echo ${i})
 669                 if [[ ${i:0:5} == "image" ]]; then
 670                         want_name=true
 671                         lines[$cnt]="$linecnt"
 672                 fi
 673
 674                 if [[ ${want_name} == "true" && ${i:0:5} == "label" ]]; then
 675                         i=$(echo ${i} | cut -d= -f2)
 676                         i=$(echo ${i})
 677                         entries[$cnt]="$i"
 678                         cnt=$(($cnt+1))
 679                         printf "%3d) %s\n" "$cnt" "$i"
 680                         want_name=false
 681                 fi
 682                 linecnt=$(($linecnt+1))
 683         done < /etc/lilo.conf
 684
 685         input_number 1 ${cnt} 1
 686         n=$(($?-1))
 687         einfo "Configuring '${entries[$n]}'.."
 688
 689         line=${lines[$n]}
 690         cp /etc/lilo.conf /etc/lilo.conf.work.$$
 691
 692         initrd="$(awk "FNR == $line {
 693                                 print ;
 694                                 while(getline > 0 && \$1 != \"image\") {
 695                                                 print \$0
 696                                 }
 697                         }" /etc/lilo.conf.work.$$ | grep "^[[:space:]]*initrd")"
 698         initrd=$(echo "$initrd" | cut -f2 -d'=')
 699
 700         theme_switch_initrd "${initrd}"
 701         cp /etc/lilo.conf /etc/lilo.conf.backup
 702
 703         awk "FNR == $(($line+1)) {
 704                 initrd=0
 705                 if (\"${newinitrd}\" != \"\") {
 706                         initrd=1
 707                 }
 708
 709                 do {
 710                         if (\$1 ~ /^append/) {
 711                                 break;
 712                         } else if (\$1 ~ /^image/ || \$1 ~ /^other/) {
 713                                 print \"\\tappend = \\\"splash=silent,fadein,theme:${theme} console=tty1\\\"\"
 714                                 if (initrd == 1) {
 715                                         print \"\\\tinitrd = ${newinitrd}\"
 716                                 }
 717                                 print \$0
 718                                 next
 719                         } else {
 720                                 if (initrd == 1 && \$1 ~ /^initrd/) {
 721                                         sub(/initrd.*/, \"initrd = ${newinitrd}\", \$0)
 722                                         initrd=0
 723                                 }
 724                                 print \$0
 725                         }
 726                 } while (getline > 0)
 727
 728                 splash=0
 729                 cons=0
 730                 quiet=0
 731                 full=\$0
 732                 \$0=gensub(/.*\"([^\"]+)\"/,\"\\\\1\",\"g\",\$0)
 733                 sub(/\".*/,\"\",\$0)
 734                 for(i=1; i<=NF+1; i++) {
 735                         if (substr(\$i,0,6) == \"splash\") {
 736                                 splash=1
 737                         } else if (\$i == \"console=tty1\") {
 738                                 cons=1
 739                         }
 740                 }
 741                 if (splash == 1) {
 742                         sub(/splash=[^ ]+/, \"splash=silent,fadein,theme:${theme}\", \$0)
 743                 } else {
 744                         \$0=\$0 \" splash=silent,fadein,theme:${theme}\"
 745                 }
 746                 if (cons == 0) {  \$0 = \$0 \" console=tty1\"   }
 747
 748                 sub(/\".*\"/, \"\\\"\" \$0 \"\\\"\", full)
 749                 print full
 750
 751                 if (initrd) {
 752                         getline
 753                         while (!(\$1 ~ /^image/) && !(\$1 ~ /^other/)) {
 754                                 if (\$1 ~ /^initrd/) {
 755                                         sub(/initrd.*/, \"initrd = ${newinitrd}\", \$0)
 756                                         initrd=0
 757                                 }
 758                                 print \$0
 759                                 getline
 760                         }
 761                         if (initrd) {
 762                                 print \"\\tinitrd = ${newinitrd}\"
 763                         }
 764                         print \$0
 765                 }
 766                 next
 767         }
 768
 769         FNR != $(($line+1)) { print \$0 }" /etc/lilo.conf.work.$$ > /etc/lilo.conf.tmp.$$
 770
 771         rm /etc/lilo.conf.work.$$
 772         mv /etc/lilo.conf.tmp.$$ /etc/lilo.conf
 773
 774         einfo "LILO config file has been updated."
 775 }
 776
 777 theme_switch_grub() {
 778         local cnt="0"
 779         local entries
 780         local lines
 781         local done=false
 782
 783         # Display a list of entries from grub.conf and ask the user to select one of them.
 784         einfo "Please select which kernel should be configured for use with splashutils:"
 785         grep -n '^[[:space:]]*title' /boot/grub/grub.conf > /tmp/splash_mngr.$$
 786         while read i ; do
 787                 line=$(echo "${i}" | cut -d: -f1)
 788                 i=$(echo "${i}" | cut -d: -f2)
 789                 i=${i:6}
 790                 entries[$cnt]="$i"
 791                 lines[$cnt]="$line"
 792                 cnt=$(($cnt+1))
 793                 printf "%3d) %s\n" "$cnt" "$i"
 794         done < /tmp/splash_mngr.$$
 795         rm /tmp/splash_mngr.$$
 796
 797         input_number 1 ${cnt} 1
 798         n=$(($?-1))
 799         einfo "Configuring '${entries[$n]}'.."
 800
 801         line=${lines[$n]}
 802         t="$(sed -e "${line}p" -e 'd' /boot/grub/grub.conf)"
 803
 804         if [[ "$(echo ${t})" != "title ${entries[$n]}" ]]; then
 805                 err "The GRUB config file appears to have been modified while this script was"
 806                 err "working. Please run it again after making sure no one is modifying"
 807                 err "/boot/grub/grub.conf."
 808                 exit 1
 809         fi
 810
 811         cp /boot/grub/grub.conf /boot/grub/grub.conf.work.$$
 812
 813         initrd="$(awk "FNR == $line {
 814                                 print ;
 815                                 while(getline > 0 && \$1 != \"title\") {
 816                                                 print \$0
 817                                 }
 818                         }" /boot/grub/grub.conf.work.$$ | grep "^[[:space:]]*initrd")"
 819         initrd=$(echo "$initrd" | cut -f2 -d' ')
 820
 821         theme_switch_initrd "${initrd}"
 822         cp /boot/grub/grub.conf /boot/grub/grub.conf.backup
 823
 824         awk "FNR == $line {
 825                 initrd=0
 826                 if (\"${newinitrd}\" != \"\") {
 827                         initrd=1
 828                 }
 829
 830                 while(\$1 != \"kernel\") {
 831                         if (initrd == 1 && \$1 == \"initrd\") {
 832                                 sub(/initrd .*/, \"initrd ${newinitrd}\", \$0)
 833                                 initrd=0
 834                         }
 835                         print \$0
 836                         if (getline <= 0) {
 837                                 exit 0
 838                         }
 839                 }
 840                 splash=0
 841                 cons=0
 842                 quiet=0
 843                 for(i=1; i<=NF+1; i++) {
 844                         if (substr(\$i,0,6) == \"splash\") {
 845                                 splash=1
 846                         } else if (\$i == \"console=tty1\") {
 847                                 cons=1
 848                         }
 849                 }
 850                 if (splash == 1) {
 851                         sub(/splash=[^ ]+/, \"splash=silent,fadein,theme:${theme}\", \$0)
 852                         printf \"%s\",\$0
 853                 } else {
 854                         printf \"%s splash=silent,fadein,theme:${theme}\", \$0
 855                 }
 856                 if (cons == 0) {  printf \" %s\", \"console=tty1\"      }
 857                 printf \"\n\"
 858
 859                 if (initrd) {
 860                         getline
 861                         while (\$1 != \"title\") {
 862                                 if (\$1 == \"initrd\") {
 863                                         sub(/initrd .*/, \"initrd ${newinitrd}\", \$0)
 864                                         initrd=0
 865                                 }
 866                                 print \$0
 867                                 if (getline <= 0) {
 868                                         break
 869                                 }
 870                         }
 871                         if (initrd) {
 872                                 print \"\\tinitrd ${newinitrd}\"
 873                         }
 874                         print \$0
 875                 }
 876                 next
 877         }
 878
 879         FNR != $line { print \$0 }" /boot/grub/grub.conf.work.$$
 880         #> /boot/grub/grub.conf.tmp.$$
 881
 882         rm /boot/grub/grub.conf.work.$$
 883 #       mv /boot/grub/grub.conf.tmp.$$ /boot/grub/grub.conf
 884
 885         einfo "GRUB config file has been updated."
 886 }
 887
 888 theme_switch() {
 889         mount_boot_partition
 890         detect_bootloader
 891
 892         if [[ ${bootloader} == "grub" ]]; then
 893                 theme_switch_grub
 894         else
 895                 theme_switch_lilo
 896         fi
 897 }
 898
 899 # Set some useful variables that we'll be using in various places
 900 ctty=$(${spl_dir}/bin/fgconsole)
 901
 902 # Default settings
 903 steps=100
 904 mode='s'
 905 cexec=''
 906 progress=''
 907 msg=''
 908 pid=''
 909 profile="${spl_dir}/cache/profile"
 910
 911 args="$@"
 912 temp=`getopt -l help,cmd:,theme:,steps:,delay:,mode:,msg:,tty:,exec:,progress:,profile: c:m:t:he: "$@"`
 913
 914 if [ $? != 0 ]; then
 915         usage; exit 2
 916 fi
 917
 918 eval set -- "$temp"
 919
 920 for i ; do
 921         case "$i" in
 922                 -c|--cmd)       op="$2"; shift; shift;;
 923                 -h|--help)      usage; exit 0;;
 924                 -m|--mode)      mode="$2"; shift; shift;;
 925                 -t|--theme)     theme="$2"; shift; shift;;
 926                 -e|--exec)      cexec="$2"; shift; shift;;
 927                 --progress)     progress="$2"; shift; shift;;
 928                 --profile)  profile="$2"; shift; shift;;
 929                 --steps)        steps="$2"; shift; shift;;
 930                 --delay)        delay="$2"; shift; shift;;
 931                 --tty)          tty="$2"; shift; shift;;
 932                 --msg)          msg="$2"; shift; shift;;
 933                 --)                     shift; break;;
 934         esac
 935 done
 936
 937 util_params="$@"
 938
 939 if [ -z "${op}" ]; then
 940         usage
 941         exit 0
 942 fi
 943
 944 rres=$(${spl_dir}/bin/fbres)
 945 res=$(${spl_util} -c getres -t ${theme})
 946 yres=${res#*x}
 947 xres=${res%x*}
 948
 949 case "${op}" in
 950         'set')
 951                 mode='v'
 952                 set_theme "${theme}" "${tty}" ;;
 953         'demo')
 954                 theme_sanity_check "${theme}"
 955                 if [ ${mode} == 's' ]; then
 956                         demo_theme_silent
 957                 else
 958                         demo_theme_verbose
 959                 fi
 960                 ;;
 961
 962         'benchmark')
 963                 mode='s'
 964                 if [ -z "$(which bc)" ]; then
 965                         err "Benchmarking requires the 'bc' command-line calculator."
 966                         exit 1
 967                 fi
 968                 theme_sanity_check "${theme}"
 969                 benchmark
 970                 ;;
 971
 972         'shot')
 973                 theme_sanity_check "${theme}"
 974                 if [ ! -x /usr/bin/fbgrab ]; then
 975                         err "Cannot find /usr/bin/fbgrab"
 976                         exit 1
 977                 fi
 978
 979                 if [ ${mode} == 's' ]; then
 980                         take_shot_silent
 981                 else
 982                         take_shot_verbose
 983                 fi
 984                 ;;
 985
 986         'switch')
 987                 theme_sanity_check "${theme}"
 988                 theme_switch
 989                 ;;
 990
 991         'list')
 992                 list_themes
 993                 ;;
 994
 995         'replay')
 996                 theme_sanity_check "${theme}"
 997                 replay "${theme}"
 998                 ;;
 999
1000         *)      err "Unrecognized command"
1001                 usage
1002                 ;;
1003 esac
1004
1005 exit 0
1006
1007 # vim: set ts=4 sts=4: