1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
|
#!/bin/sh
# Copyright (C) 2011 Canonical Ltd.
#
# Authors: Scott Moser <smoser@canonical.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
# the fudge factor. if its within this many 512 byte sectors, dont bother
FUDGE=${GROWPART_FUDGE:-$((20*1024))}
TEMP_D=""
RESTORE_FROM=""
VERBOSITY=0
DISK=""
error() { echo "$@" 1>&2; }
fail() { [ $# -eq 0 ] || echo "FAILED:" "$@"; exit 1; }
nochange() { echo "NOCHANGE:" "$@"; exit 0; }
changed() { echo "CHANGED:" "$@"; exit 0; }
cleanup() {
if [ -n "${RESTORE_FROM}" ]; then
error "***** WARNING: Resize failed, attempting to revert ******"
if sfdisk "${DISK}" ${CHS} -I "${RESTORE_FROM}"; then
error "***** Appears to have gone OK ****"
else
error "***** FAILED! or original partition table looked like: ****"
cat "${RESTORE_HUMAN}" 1>&2
fi
fi
[ -z "${TEMP_D}" -o ! -d "${TEMP_D}" ] || rm -Rf "${TEMP_D}"
}
debug() {
local level=${1}; shift;
[ "${level}" -gt "${VERBOSITY}" ] && return
error "${@}"
}
mktemp_d() {
# just a mktemp -d that doens't need mktemp if its not there.
_RET=$(mktemp -d "${TMPDIR:-/tmp}/${0##*/}.XXXXXX" 2>/dev/null) && return
_RET=$(umask 077 && t="${TMPDIR:-/tmp}/${0##*/}.$$" && mkdir "${t}" &&
echo "${t}")
return
}
Usage() {
cat <<EOF
${0##*/} partition
rewrite partition table so that partition takes up all the space it can
options:
-h | --help print Usage an exit
--fudge F if part could be resized, but change would be
less than 'F', do not resize (default: ${FUDGE})
-N | --dry-run only report what would be done, show new 'sfdisk -d'
-v | --verbose increase verbosity / debug
EOF
}
bad_Usage() { Usage 1>&2; error "$@"; exit 1; }
#short_opts="hNv"
#long_opts="help,dry-run,fudge:,verbose"
#getopt_out=$(getopt --name "${0##*/}" \
# --options "${short_opts}" --long "${long_opts}" -- "$@") &&
# eval set -- "${getopt_out}" ||
# bad_Usage
dry_run=0
fudge=${FUDGE}
disk=""
part=""
while [ $# -ne 0 ]; do
cur=${1}; next=${2};
case "$cur" in
-h|--help) Usage; exit 0;;
--fudge) fudge=${next}; shift;;
--dry-run) dry_run=1;;
-v|--verbose) VERBOSITY=$(($VERBOSITY+1));;
--) shift; break;;
-*) fail "unknown option ${cur}";;
*)
if [ -z "${disk}" ]; then
disk=${cur};
else
[ -z "${part}" ] || fail "confused by arg ${cur}"
part=${cur};
fi
;;
esac
shift;
done
[ -n "${disk}" ] || bad_Usage "must supply disk and partition-number"
[ -n "${part}" ] || bad_Usage "must supply partition-number"
[ -e "${disk}" ] || { fail "${disk}: does not exist"; }
[ "${part#*[!0-9]}" = "${part}" ] || fail "partition-number must be a number"
mktemp_d && TEMP_D="${_RET}" || fail "failed to make temp dir"
trap cleanup EXIT
change_out=${TEMP_D}/change.out
change_err=${TEMP_D}/change.err
dump_out=${TEMP_D}/dump.out
new_out=${TEMP_D}/new.out
dump_mod=${TEMP_D}/dump.mod
tmp="${TEMP_D}/tmp.out"
err="${TEMP_D}/err.out"
orig_bin="${TEMP_D}/orig.save"
RESTORE_HUMAN="${TEMP_D}/recovery"
# --show-pt-geometry outputs something like
# /dev/sda: 164352 cylinders, 4 heads, 32 sectors/track
sfdisk "${disk}" --show-pt-geometry > "${tmp}" 2>"${err}" &&
read _devc cyl _word1 heads _word2 sectors _word3 < "${tmp}" &&
CHS="-C $cyl -H $heads -S $sectors" ||
fail "failed to get CHS from ${disk}"
tot=$((${cyl}*${heads}*${sectors}))
debug 1 "geometry is $CHS. total size=${tot}"
sfdisk ${CHS} -uS -d "${disk}" > "${dump_out}" 2>"${err}" ||
fail "failed to dump sfdisk info for ${disk}"
{
echo "## sfdisk ${CHS} -uS -d ${disk}";
cat "${dump_out}"
} > "${RESTORE_HUMAN}"
[ $? -eq 0 ] || fail "failed to save sfdisk -d output"
sed -e 's/,//g; s/start=/start /; s/size=/size /' "${dump_out}" > "${dump_mod}"
dpart="${disk}${part}" # disk and partition number
pt_start=$(awk '$1 == pt { print $4 }' "pt=${dpart}" < "${dump_mod}") &&
pt_size=$(awk '$1 == pt { print $6 }' "pt=${dpart}" < "${dump_mod}") &&
[ -n "${pt_start}" -a -n "${pt_size}" ] &&
pt_end=$((${pt_size}+${pt_start})) ||
fail "failed to get start and end for ${dpart} in ${disk}"
# find the minimal starting location that is >= pt_end
max_end=$(awk '
$3 == "start" { if($4 >= pt_end && $4 < min) { min = $4 } }
END { printf("%s\n",min); }' \
min=${tot} pt_end=${pt_end} "${dump_mod}") &&
[ -n "${max_end}" ] ||
fail "failed to get max_end for partition ${part}"
debug 1 "max_end=${max_end} tot=${tot} pt_end=${pt_end} pt_start=${pt_start} pt_size=${pt_size}"
[ $((${pt_end})) -eq ${max_end} ] &&
nochange "partition ${part} is size ${pt_size}. it cannot be grown"
[ $((${pt_end}+${fudge})) -gt ${max_end} ] &&
nochange "partition ${part} could only be grown by $((${max_end}-${pt_end})) [fudge=${fudge}]"
# now, change the size for this partition in ${dump_out} to be the
# new size
new_size=$((${max_end}-${pt_start}))
sed "\|^${dpart} |s/${pt_size},/${new_size},/" "${dump_out}" > "${new_out}" ||
fail "failed to change size in output"
change_info="partition=${part} start=${pt_start} old: size=${pt_size} end=${pt_end} new: size=${new_size},end=${max_end}"
if [ $dry_run -ne 0 ]; then
echo "CHANGE: ${change_info}"
{
echo "# === old sfdisk -d ==="
cat "${dump_out}"
echo "# === new sfdisk -d ==="
cat "${new_out}"
} 1>&2
exit 0
fi
sfdisk "${disk}" ${CHS} --force -O "${orig_bin}" \
< "${new_out}" > "${change_out}" 2> "${change_err}" || {
DISK=${disk}; RESTORE_FROM="${orig_bin}";
fail "failed to resize"
}
changed "${change_info}"
# dump_out looks something like:
## partition table of /tmp/out.img
#unit: sectors
#
#/tmp/out.img1 : start= 1, size= 48194, Id=83
#/tmp/out.img2 : start= 48195, size= 963900, Id=83
#/tmp/out.img3 : start= 1012095, size= 305235, Id=82
#/tmp/out.img4 : start= 1317330, size= 771120, Id= 5
#/tmp/out.img5 : start= 1317331, size= 642599, Id=83
#/tmp/out.img6 : start= 1959931, size= 48194, Id=83
#/tmp/out.img7 : start= 2008126, size= 80324, Id=83
# vi: ts=4 noexpandtab
|