1#!/bin/bash -eu 2 3# Copyright (c) 2010 The Chromium OS Authors. All rights reserved. 4# Use of this source code is governed by a BSD-style license that can be 5# found in the LICENSE file. 6# 7# Run tests for cgpt utility. 8 9# Load common constants and variables. 10. "$(dirname "$0")/common.sh" 11 12CGPT=$(readlink -f "$1") 13[ -x "$CGPT" ] || error "Can't execute $CGPT" 14 15MTD="${@:2}" 16 17# Run tests in a dedicated directory for easy cleanup or debugging. 18DIR="${TEST_DIR}/cgpt_test_dir" 19[ -d "$DIR" ] || mkdir -p "$DIR" 20warning "testing $CGPT in $DIR" 21cd "$DIR" 22 23assert_fail() { 24 set +e 25 "$@" 2>/dev/null 26 if [ $? == 0 ]; then 27 error "$*" " should have failed but did not" 28 fi 29 set -e 30} 31 32# Test failure on non existing file. 33assert_fail ${CGPT} show $MTD blah_404_haha 34 35echo "Create an empty file to use as the device..." 36NUM_SECTORS=1000 37DEV=fake_dev.bin 38rm -f ${DEV} 39dd if=/dev/zero of=${DEV} conv=notrunc bs=512 count=${NUM_SECTORS} 2>/dev/null 40 41 42echo "Create a bunch of partitions, using the real GUID types..." 43DATA_START=100 44DATA_SIZE=20 45DATA_LABEL="data stuff" 46DATA_GUID='ebd0a0a2-b9e5-4433-87c0-68b6b72699c7' 47DATA_NUM=1 48 49KERN_START=200 50KERN_SIZE=30 51KERN_LABEL="kernel stuff" 52KERN_GUID='fe3a2a5d-4f32-41a7-b725-accc3285a309' 53KERN_NUM=2 54 55ROOTFS_START=300 56ROOTFS_SIZE=40 57ROOTFS_LABEL="rootfs stuff" 58ROOTFS_GUID='3cb8e202-3b7e-47dd-8a3c-7ff2a13cfcec' 59ROOTFS_NUM=3 60 61ESP_START=400 62ESP_SIZE=50 63ESP_LABEL="ESP stuff" 64ESP_GUID='c12a7328-f81f-11d2-ba4b-00a0c93ec93b' 65ESP_NUM=4 66 67FUTURE_START=500 68FUTURE_SIZE=60 69FUTURE_LABEL="future stuff" 70FUTURE_GUID='2e0a753d-9e48-43b0-8337-b15192cb1b5e' 71FUTURE_NUM=5 72 73RANDOM_START=600 74RANDOM_SIZE=70 75RANDOM_LABEL="random stuff" 76RANDOM_GUID='2364a860-bf63-42fb-a83d-9ad3e057fcf5' 77RANDOM_NUM=6 78 79$CGPT create $MTD ${DEV} 80 81$CGPT add $MTD -b ${DATA_START} -s ${DATA_SIZE} -t ${DATA_GUID} \ 82 -l "${DATA_LABEL}" ${DEV} 83$CGPT add $MTD -b ${KERN_START} -s ${KERN_SIZE} -t ${KERN_GUID} \ 84 -l "${KERN_LABEL}" ${DEV} 85$CGPT add $MTD -b ${ROOTFS_START} -s ${ROOTFS_SIZE} -t ${ROOTFS_GUID} \ 86 -l "${ROOTFS_LABEL}" ${DEV} 87$CGPT add $MTD -b ${ESP_START} -s ${ESP_SIZE} -t ${ESP_GUID} \ 88 -l "${ESP_LABEL}" ${DEV} 89$CGPT add $MTD -b ${FUTURE_START} -s ${FUTURE_SIZE} -t ${FUTURE_GUID} \ 90 -l "${FUTURE_LABEL}" ${DEV} 91$CGPT add $MTD -b ${RANDOM_START} -s ${RANDOM_SIZE} -t ${RANDOM_GUID} \ 92 -l "${RANDOM_LABEL}" ${DEV} 93 94 95echo "Extract the start and size of given partitions..." 96 97X=$($CGPT show $MTD -b -i $DATA_NUM ${DEV}) 98Y=$($CGPT show $MTD -s -i $DATA_NUM ${DEV}) 99[ "$X $Y" = "$DATA_START $DATA_SIZE" ] || error 100 101X=$($CGPT show $MTD -b -i $KERN_NUM ${DEV}) 102Y=$($CGPT show $MTD -s -i $KERN_NUM ${DEV}) 103[ "$X $Y" = "$KERN_START $KERN_SIZE" ] || error 104 105X=$($CGPT show $MTD -b -i $ROOTFS_NUM ${DEV}) 106Y=$($CGPT show $MTD -s -i $ROOTFS_NUM ${DEV}) 107[ "$X $Y" = "$ROOTFS_START $ROOTFS_SIZE" ] || error 108 109X=$($CGPT show $MTD -b -i $ESP_NUM ${DEV}) 110Y=$($CGPT show $MTD -s -i $ESP_NUM ${DEV}) 111[ "$X $Y" = "$ESP_START $ESP_SIZE" ] || error 112 113X=$($CGPT show $MTD -b -i $FUTURE_NUM ${DEV}) 114Y=$($CGPT show $MTD -s -i $FUTURE_NUM ${DEV}) 115[ "$X $Y" = "$FUTURE_START $FUTURE_SIZE" ] || error 116 117X=$($CGPT show $MTD -b -i $RANDOM_NUM ${DEV}) 118Y=$($CGPT show $MTD -s -i $RANDOM_NUM ${DEV}) 119[ "$X $Y" = "$RANDOM_START $RANDOM_SIZE" ] || error 120 121 122echo "Change the beginning..." 123DATA_START=$((DATA_START + 10)) 124$CGPT add $MTD -i 1 -b ${DATA_START} ${DEV} || error 125X=$($CGPT show $MTD -b -i 1 ${DEV}) 126[ "$X" = "$DATA_START" ] || error 127 128echo "Change the size..." 129DATA_SIZE=$((DATA_SIZE + 10)) 130$CGPT add $MTD -i 1 -s ${DATA_SIZE} ${DEV} || error 131X=$($CGPT show $MTD -s -i 1 ${DEV}) 132[ "$X" = "$DATA_SIZE" ] || error 133 134echo "Change the type..." 135$CGPT add $MTD -i 1 -t reserved ${DEV} || error 136X=$($CGPT show $MTD -t -i 1 ${DEV} | tr 'A-Z' 'a-z') 137[ "$X" = "$FUTURE_GUID" ] || error 138# arbitrary value 139$CGPT add $MTD -i 1 -t 610a563a-a55c-4ae0-ab07-86e5bb9db67f ${DEV} || error 140X=$($CGPT show $MTD -t -i 1 ${DEV}) 141[ "$X" = "610A563A-A55C-4AE0-AB07-86E5BB9DB67F" ] || error 142 143$CGPT add $MTD -i 1 -t data ${DEV} || error 144X=$($CGPT show $MTD -t -i 1 ${DEV} | tr 'A-Z' 'a-z') 145[ "$X" = "$DATA_GUID" ] || error 146 147 148echo "Set the boot partition.." 149$CGPT boot $MTD -i ${KERN_NUM} ${DEV} >/dev/null 150 151echo "Check the PMBR's idea of the boot partition..." 152X=$($CGPT boot $MTD ${DEV}) 153Y=$($CGPT show $MTD -u -i $KERN_NUM $DEV) 154[ "$X" = "$Y" ] || error 155 156echo "Test the cgpt prioritize command..." 157 158# Input: sequence of priorities 159# Output: ${DEV} has kernel partitions with the given priorities 160make_pri() { 161 local idx=0 162 $CGPT create $MTD ${DEV} 163 for pri in "$@"; do 164 idx=$((idx+1)) 165 $CGPT add $MTD -t kernel -l "kern$idx" -b $((100 + 2 * $idx)) -s 1 -P $pri ${DEV} 166 done 167} 168 169# Output: returns string containing priorities of all kernels 170get_pri() { 171 echo $( 172 for idx in $($CGPT find $MTD -t kernel ${DEV} | sed -e s@${DEV}@@); do 173 $CGPT show $MTD -i $idx -P ${DEV} 174 done 175 ) 176} 177 178# Input: list of priorities 179# Operation: expects ${DEV} to contain those kernel priorities 180assert_pri() { 181 local expected="$*" 182 local actual=$(get_pri) 183 [ "$actual" = "$expected" ] || \ 184 error 1 "expected priority \"$expected\", actual priority \"$actual\"" 185} 186 187# no kernels at all. This should do nothing. 188$CGPT create $MTD ${DEV} 189$CGPT add $MTD -t rootfs -b 100 -s 1 ${DEV} 190$CGPT prioritize $MTD ${DEV} 191assert_pri "" 192 193# common install/upgrade sequence 194make_pri 2 0 0 195$CGPT prioritize $MTD -i 1 ${DEV} 196assert_pri 1 0 0 197$CGPT prioritize $MTD -i 2 ${DEV} 198assert_pri 1 2 0 199$CGPT prioritize $MTD -i 1 ${DEV} 200assert_pri 2 1 0 201$CGPT prioritize $MTD -i 2 ${DEV} 202assert_pri 1 2 0 203# lots of kernels, all same starting priority, should go to priority 1 204make_pri 8 8 8 8 8 8 8 8 8 8 8 0 0 8 205$CGPT prioritize $MTD ${DEV} 206assert_pri 1 1 1 1 1 1 1 1 1 1 1 0 0 1 207 208# now raise them all up again 209$CGPT prioritize $MTD -P 4 ${DEV} 210assert_pri 4 4 4 4 4 4 4 4 4 4 4 0 0 4 211 212# set one of them higher, should leave the rest alone 213$CGPT prioritize $MTD -P 5 -i 3 ${DEV} 214assert_pri 4 4 5 4 4 4 4 4 4 4 4 0 0 4 215 216# set one of them lower, should bring the rest down 217$CGPT prioritize $MTD -P 3 -i 4 ${DEV} 218assert_pri 1 1 2 3 1 1 1 1 1 1 1 0 0 1 219 220# raise a group by including the friends of one partition 221$CGPT prioritize $MTD -P 6 -i 1 -f ${DEV} 222assert_pri 6 6 4 5 6 6 6 6 6 6 6 0 0 6 223 224# resurrect one, should not affect the others 225make_pri 0 0 0 0 0 0 0 0 0 0 0 0 0 0 226$CGPT prioritize $MTD -i 2 ${DEV} 227assert_pri 0 1 0 0 0 0 0 0 0 0 0 0 0 0 228 229# resurrect one and all its friends 230make_pri 0 0 0 0 0 0 0 0 1 2 0 0 0 0 231$CGPT prioritize $MTD -P 5 -i 2 -f ${DEV} 232assert_pri 5 5 5 5 5 5 5 5 3 4 5 5 5 5 233 234# no options should maintain the same order 235$CGPT prioritize $MTD ${DEV} 236assert_pri 3 3 3 3 3 3 3 3 1 2 3 3 3 3 237 238# squish all the ranks 239make_pri 1 1 2 2 3 3 4 4 5 5 0 6 7 7 240$CGPT prioritize $MTD -P 6 ${DEV} 241assert_pri 1 1 1 1 2 2 3 3 4 4 0 5 6 6 242 243# squish the ranks by not leaving room 244make_pri 1 1 2 2 3 3 4 4 5 5 0 6 7 7 245$CGPT prioritize $MTD -P 7 -i 3 ${DEV} 246assert_pri 1 1 7 1 2 2 3 3 4 4 0 5 6 6 247 248# squish the ranks while bringing the friends along 249make_pri 1 1 2 2 3 3 4 4 5 5 0 6 7 7 250$CGPT prioritize $MTD -P 6 -i 3 -f ${DEV} 251assert_pri 1 1 6 6 1 1 2 2 3 3 0 4 5 5 252 253# squish them pretty hard 254make_pri 1 1 2 2 3 3 4 4 5 5 0 6 7 7 255$CGPT prioritize $MTD -P 2 ${DEV} 256assert_pri 1 1 1 1 1 1 1 1 1 1 0 1 2 2 257 258# squish them really really hard (nobody gets reduced to zero, though) 259make_pri 1 1 2 2 3 3 4 4 5 5 0 6 7 7 260$CGPT prioritize $MTD -P 1 -i 3 ${DEV} 261assert_pri 1 1 1 1 1 1 1 1 1 1 0 1 1 1 262 263make_pri 15 15 14 14 13 13 12 12 11 11 10 10 9 9 8 8 7 7 6 6 5 5 4 4 3 3 2 2 1 1 0 264$CGPT prioritize $MTD -i 3 ${DEV} 265assert_pri 14 14 15 13 12 12 11 11 10 10 9 9 8 8 7 7 6 6 5 5 4 4 3 3 2 2 1 1 1 1 0 266$CGPT prioritize $MTD -i 5 ${DEV} 267assert_pri 13 13 14 12 15 11 10 10 9 9 8 8 7 7 6 6 5 5 4 4 3 3 2 2 1 1 1 1 1 1 0 268# but if I bring friends I don't have to squish 269$CGPT prioritize $MTD -i 1 -f ${DEV} 270assert_pri 15 15 13 12 14 11 10 10 9 9 8 8 7 7 6 6 5 5 4 4 3 3 2 2 1 1 1 1 1 1 0 271 272# Now make sure that we don't need write access if we're just looking. 273echo "Test read vs read-write access..." 274chmod 0444 ${DEV} 275 276# These should fail 277$CGPT create $MTD -z ${DEV} 2>/dev/null && error 278$CGPT add $MTD -i 2 -P 3 ${DEV} 2>/dev/null && error 279$CGPT repair $MTD ${DEV} 2>/dev/null && error 280$CGPT prioritize $MTD -i 3 ${DEV} 2>/dev/null && error 281 282# Most 'boot' usage should fail too. 283$CGPT boot $MTD -p ${DEV} 2>/dev/null && error 284dd if=/dev/zero of=fake_mbr.bin bs=100 count=1 2>/dev/null 285$CGPT boot $MTD -b fake_mbr.bin ${DEV} 2>/dev/null && error 286$CGPT boot $MTD -i 2 ${DEV} 2>/dev/null && error 287 288$CGPT boot $MTD ${DEV} >/dev/null 289$CGPT show $MTD ${DEV} >/dev/null 290$CGPT find $MTD -t kernel ${DEV} >/dev/null 291 292# Enable write access again to test boundary in off device storage 293chmod 600 ${DEV} 294# GPT too small 295dd if=/dev/zero of=${DEV} bs=5632 count=1 296assert_fail $CGPT create -D 1024 ${DEV} 297# GPT is just right for 16 entries (512 + 512 + 16 * 128) * 2 = 6144 298dd if=/dev/zero of=${DEV} bs=6144 count=1 299$CGPT create -D 1024 ${DEV} 300# Create a small 8K file to simulate Flash NOR section 301dd if=/dev/zero of=${DEV} bs=8K count=1 302# Drive size is not multiple of 512 303assert_fail $CGPT create -D 511 ${DEV} 304assert_fail $CGPT create -D 513 ${DEV} 305MTD="-D 1024" 306# Create a GPT table for a device of 1024 bytes (2 sectors) 307$CGPT create $MTD ${DEV} 308# Make sure number of entries is reasonable for 8KiB GPT 309X=$($CGPT show -D 1024 -d ${DEV} | grep -c "Number of entries: 24") 310[ "$X" = "2" ] || error 311# This fails because header verification is off due to different drive size 312assert_fail $CGPT show ${DEV} 313# But this passes because we pass in correct drive size 314$CGPT show $MTD ${DEV} 315# This fails because beginning sector is over the size of the device 316assert_fail $CGPT add $MTD -b 2 -s 1 -t data ${DEV} 317# This fails because partition size is over the size of the device 318assert_fail $CGPT add $MTD -b 0 -s 3 -t data ${DEV} 319 320 321echo "Done." 322 323happy "All tests passed." 324