1#!/bin/sh
2#
3# Copyright 2014 John-Mark Gurney
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 AUTHOR 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 AUTHOR 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: stable/10/etc/rc.d/growfs 332731 2018-04-18 16:22:23Z gjb $
28#
29
30# PROVIDE: growfs
31# BEFORE: sysctl
32# KEYWORD: firstboot
33
34# This allows us to distribute a image
35# and have it work on essentially any size drive.
36#
37# TODO: Figure out where this should really be ordered.
38# I suspect it should go just after fsck but before mountcritlocal.
39# 
40
41. /etc/rc.subr
42
43name="growfs"
44start_cmd="growfs_start"
45stop_cmd=":"
46rcvar="growfs_enable"
47
48growfs_start ()
49{
50	echo "Growing root partition to fill device"
51	rootdev=$(df / | tail -n 1 | awk '{ sub("/dev/", "", $1); print $1 }')
52	if [ x"$rootdev" = x"${rootdev%/*}" ]; then
53		# raw device
54		rawdev="$rootdev"
55	else
56		rawdev=$(glabel status | awk '$1 == "'"$rootdev"'" { print $3 }')
57		if [ x"$rawdev" = x"" ]; then
58			echo "Can't figure out device for: $rootdev"
59			return
60		fi
61	fi
62
63	sysctl -b kern.geom.conftxt | awk '
64{
65	lvl=$1
66	device[lvl] = $3
67	type[lvl] = $2
68	idx[lvl] = $7
69	parttype[lvl] = $13
70	if (dev == $3) {
71		for (i = 1; i <= lvl; i++) {
72			# resize
73			if (type[i] == "PART") {
74				pdev = device[i - 1]
75				cmd[i] = "gpart resize -i " idx[i] " " pdev
76				if (parttype[i] == "GPT")
77					cmd[i] = "gpart recover " pdev " ; " cmd[i]
78			} else if (type[i] == "LABEL") {
79				continue
80			} else {
81				print "unhandled type: " type[i]
82				exit 1
83			}
84		}
85		for (i = 1; i <= lvl; i++) {
86			if (cmd[i])
87				system(cmd[i])
88		}
89		exit 0
90	}
91}' dev="$rawdev"
92	gpart commit "$rootdev"
93	growfs -y /dev/"$rootdev"
94}
95
96load_rc_config $name
97run_rc_command "$1"
98