1#!/bin/bash
2#
3# This file is part of util-linux.
4#
5# Copyright (C) 2013 Karel Zak <kzak@redhat.com>
6#
7# This file is free software; you can redistribute it and/or modify
8# it under the terms of the GNU General Public License as published by
9# the Free Software Foundation; either version 2 of the License, or
10# (at your option) any later version.
11#
12# This file is distributed in the hope that it will be useful,
13# but WITHOUT ANY WARRANTY; without even the implied warranty of
14# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15# GNU General Public License for more details.
16#
17#
18
19TS_TOPDIR="${0%/*}/../.."
20TS_DESC="GPT"
21
22. $TS_TOPDIR/functions.sh
23ts_init "$*"
24
25ts_check_test_command "$TS_CMD_FDISK"
26
27FDISK_CMD_CREATE_GPTLABEL="g\n"              # create GPT label
28FDISK_CMD_WRITE_CLOSE="w\nq\n"               # write to image
29
30FDISK_CMD_CHANGE_TYPE_BY_NUM="t\n1\n1\n"     # set 1st partition to EFI System
31                                             # set 5th partition to Linux swap
32FDISK_CMD_CHANGE_TYPE_BY_UUID="t\n5\n0657FD6D-A4AB-43C4-84E5-0933C84B4F4F\n"
33
34FDISK_CMD_DELETE_PART="d\n2\n"               # delete 2nd partition
35FDISK_CMD_CREATE_PART="n\n2\n\n\n"           # create 2nd partition
36
37function print_layout {
38	echo -ne "\n---layout----------" >> $TS_OUTPUT
39	$TS_CMD_FDISK -l ${TEST_IMAGE_NAME} &>> $TS_OUTPUT
40	echo -ne   "-------------------\n\n" >> $TS_OUTPUT
41}
42
43ts_log "Initialize empty image"
44TEST_IMAGE_NAME=$(ts_image_init 10)
45
46ts_log "Create new GPT partition table"
47echo -e "${FDISK_CMD_CREATE_GPTLABEL}${FDISK_CMD_WRITE_CLOSE}" \
48	 | $TS_CMD_FDISK ${TEST_IMAGE_NAME} &> /dev/null
49
50print_layout
51
52ts_log "Create partitions"
53for i in {1..8}; do
54	echo -e "n\n${i}\n\n+1M\n${FDISK_CMD_WRITE_CLOSE}" | \
55	$TS_CMD_FDISK ${TEST_IMAGE_NAME} &> /dev/null
56done
57
58ts_log "Change partition type by number"
59echo -e "${FDISK_CMD_CHANGE_TYPE_BY_NUM}${FDISK_CMD_WRITE_CLOSE}" \
60	 | $TS_CMD_FDISK ${TEST_IMAGE_NAME} &> /dev/null
61
62print_layout
63
64ts_log "Change partition type by UUID"
65echo -e "${FDISK_CMD_CHANGE_TYPE_BY_UUID}${FDISK_CMD_WRITE_CLOSE}" \
66	 | $TS_CMD_FDISK ${TEST_IMAGE_NAME} &> /dev/null
67
68print_layout
69
70ts_log "Delete partition"
71echo -e "${FDISK_CMD_DELETE_PART}${FDISK_CMD_WRITE_CLOSE}" \
72	 | $TS_CMD_FDISK ${TEST_IMAGE_NAME} &> /dev/null
73
74print_layout
75
76ts_log "Create again partition"
77echo -e "${FDISK_CMD_CREATE_PART}${FDISK_CMD_WRITE_CLOSE}" \
78	 | $TS_CMD_FDISK ${TEST_IMAGE_NAME} &> /dev/null
79
80print_layout
81
82ts_log "Corrupt primary table"
83dd if=/dev/zero of=${TEST_IMAGE_NAME} bs=512 count=1 seek=1 conv=notrunc &> /dev/null
84echo -e "w\n" | $TS_CMD_FDISK ${TEST_IMAGE_NAME} &> /dev/null
85
86print_layout
87
88ts_log "Corrupt backup table"
89dd if=/dev/zero of=${TEST_IMAGE_NAME} bs=512 count=1 \
90	seek=$(( 1024 * 1024 * 10 / 512 - 1)) conv=notrunc &> /dev/null
91echo -e "w\n" | $TS_CMD_FDISK ${TEST_IMAGE_NAME} &> /dev/null
92
93print_layout
94
95ts_fdisk_clean ${TEST_IMAGE_NAME}
96ts_finalize
97