plan9/rc/bin/fshalt

158 lines
2.9 KiB
Bash
Executable File

#!/bin/rc
# fshalt [-r] [new-kernel] - sync (flush) and, if possible, halt all
# file servers and optionally reboot
rfork nes
kern=()
kerncopy=()
reboot=no
fn usage {
echo usage: $1 '[-r] [new-kernel]' >[1=2]
exit usage
}
if (! ~ $#* 0)
switch ($1) {
case -r
reboot=yes
shift
case -*
usage $0
}
switch ($#*) {
case 0
case 1
kern = $1
case *
usage $0
}
path=(/bin)
builtin cd /
9fs boot # in case of odd namespace
ramfs # stash files here that we'll need after the fs halts
setrtc
# turn off graphics, if any
if (test -e '#v/vgactl' && test -e '#P/realmode') {
vgatype=`{read '#v/vgactl'}
if (! ~ $vgatype(2) cga) {
prompt=''
kill rio | rc -i &
sleep 2
aux/vga -l text
}
}
fn isbootable {
test -e $1 && ~ `{file $1} *' plan 9 boot image' \
*' plan 9 executable'* *ELF*' executable'
}
fn usekernel {
kerncopy=kern
cp $1 /tmp/$kerncopy && echo -n using kernel $1...
}
# make a copy of the right kernel
if (~ $reboot yes) {
if (~ $#kern 0)
kern=`{echo $terminal |
sed 's;^([^ ]+) .*/([^/ ]+).*$;/n/boot/'$cputype'/9\2;'}
if (isbootable $kern)
usekernel $kern
if not {
# $kern is probably a raw binary, try s^name as boot image
kern=`{ echo $kern | sed 's;.*/;&s;' }
if (isbootable $kern)
usekernel $kern
if not
echo -n using default kernel...
}
}
# start venti flushing
venti/sync -h localhost >[2]/dev/null &
venti/sync >[2]/dev/null &
unmount /mnt/consoles >[2]/dev/null
kill consolefs | rc # don't compete with /mnt/consoles
kill cron | rc # unlock /cron/lock
kill watchdog | rc
sleep 1
f=`{ls /srv/fscons*>[2]/dev/null}
k=`{ls /srv/kfs*cmd >[2]/dev/null|sort -r}
echo -n syncing...
for(i in $f) @ {
echo -n $i...
{
echo
dial/drain &
sleep 2
echo fsys all sync
if(! dial/expect -t 120 ': ')
echo -n 'not synced...' > /dev/cons
} < $i >> $i
}
# flush the last bit of possible fossil traffic
if (ls /srv | grep -s fossil) {
echo -n venti...
venti/sync -h localhost >[2]/dev/null &
venti/sync >[2]/dev/null &
sleep 5
}
for (i in $k){
echo -n $i...
switch($i){
case /srv/kfs.cmd
disk/kfscmd sync
case *
disk/kfscmd -n `{echo $i | sed -n 's%/srv/kfs.(.*).cmd%\1%p'} sync
}
sleep 2
}
# halting (binaries we run can't be on the fs we're halting)
builtin cd /bin
cp dial/expect echo disk/kfscmd rc sed /tmp
builtin cd /tmp
mkdir /tmp/lib
cp /rc/lib/rcmain /tmp/lib
bind /tmp /rc
bind /tmp /bin
# put this in a shell function so this rc script doesn't get read
# when it's no longer accessible
fn x {
rfork s
echo
echo -n halting...
for(i in $f) @ {
echo -n $i...
{
echo fsys all halt
if(! expect -t 60 ': ')
echo -n 'not halted...' > /dev/cons
} < $i >> $i
}
for (i in $k){
echo -n $i...
switch($i){
case /srv/kfs.cmd
kfscmd halt
case *
kfscmd -n `{echo $i | sed -n 's%/srv/kfs.(.*).cmd%\1%p'} halt
}
}
echo
echo done halting
if (~ $reboot yes)
echo reboot $kerncopy >'#c/reboot'
}
x