1#!/bin/sh
2#
3# Copyright (c) 2010 Pawel Jakub Dawidek <pjd@FreeBSD.org>
4# All rights reserved.
5#
6# Redistribution and use in source and binary forms, with or without
7# modification, are permitted provided that the following conditions
8# are met:
9# 1. Redistributions of source code must retain the above copyright
10#    notice, this list of conditions and the following disclaimer.
11# 2. Redistributions in binary form must reproduce the above copyright
12#    notice, this list of conditions and the following disclaimer in the
13#    documentation and/or other materials provided with the distribution.
14#
15# THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
16# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
19# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25# SUCH DAMAGE.
26#
27# $FreeBSD$
28#
29
30# PROVIDE: gptboot
31# REQUIRE: mountcritremote
32# KEYWORD: nojail
33
34. /etc/rc.subr
35
36name="gptboot"
37rcvar="gptboot_enable"
38start_cmd="gptboot_report"
39
40gptboot_report()
41{
42	gpart show | \
43		egrep '(^=>| freebsd-ufs .*(\[|,)(bootfailed|bootonce)(,|\]))' | \
44		sed 's/^=>//' | \
45		egrep -v '(\[|,)bootme(,|\])' | \
46	while read start size pos type attrs rest; do
47		case "${pos}" in
48		[0-9]*)
49			if [ -n "${disk}" ]; then
50				part="${disk}p${pos}"
51				echo "${attrs}" | egrep -q '(\[|,)bootfailed(,|\])'
52				bootfailed=$?
53				echo "${attrs}" | egrep -q '(\[|,)bootonce(,|\])'
54				bootonce=$?
55				if [ ${bootfailed} -eq 0 ]; then
56					logger -t gptboot -p local0.notice "Boot from ${part} failed."
57					gpart unset -a bootfailed -i ${pos} ${disk} >/dev/null
58				elif [ ${bootonce} -eq 0 ]; then
59					# We want to log success after all failures.
60					echo -n "Boot from ${part} succeeded."
61					gpart unset -a bootonce -i ${pos} ${disk} >/dev/null
62				fi
63			fi
64			;;
65		*)
66			if [ "${type}" = "GPT" ]; then
67				disk="${pos}"
68			else
69				disk=""
70			fi
71			;;
72		esac
73	done | logger -t gptboot -p local0.notice
74}
75
76load_rc_config $name
77run_rc_command "$1"
78