summaryrefslogtreecommitdiff
path: root/lx-trainer-vm/dd-multi.sh
blob: 50f87868deecc7ffc49d84115911ee4f269d93ce (plain)
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
#!/bin/sh
set -e

# This script only supports GPT partitions for the lx-trainer image.
ROOTPT=3
HOMEPT=4

usage_exit()
{
	echo
	echo "usage: sudo $0 src [--layer=tarball]... dest [dest]..."
	echo
	echo "example: copying image to devices: sde, sdf, sdg"
	echo "$0 lx-trainer.img /dev/sde /dev/sdf /dev/sdg"
	echo
	exit 1
}

# minimum 2 arguments
if [ $# -lt 2 ]; then
	echo "error: invalid arguments"
	usage_exit
fi

if [ `id -u` -ne 0 ]; then
	echo "error: sorry, this script must be run as root"
	exit 1
fi

# gather arguments
SRC=""
LAYERS=""
DESTS=""

for arg in $@; do
	case $arg in
	--layer=*)
		LAYERS="$LAYERS `echo $arg | sed -e 's/^--layer=//'`"
		;;
	--*)
		echo "error: unknown option: $arg"
		usage_exit
		;;
	*)
		if [ -z "$SRC" ]; then
			SRC="$arg"
		else
			DESTS="$DESTS $arg"
		fi
		;;
	esac
done

# source must be file or block device
if [ ! -f "$SRC" -a ! -b "$SRC" ]; then
	echo "error: invalid src"
	usage_exit
fi

# destination must be block device
for dest in $DESTS; do
	if [ ! -b $dest ]; then
		echo "error: invalid dest: $dest"
		usage_exit
	fi
done

# detect partion type of source
if [ "`env - /sbin/parted --script $SRC print | \
	grep 'Partition Table' | awk '{print $3}'`" != "gpt" ]; then
	echo "error: only gpt partition tables supported"
	usage_exit
fi

# delete mbr's
dd if=/dev/zero bs=10M count=1 2> /dev/null | tee $DESTS > /dev/null

# efficiently copy source to all destinations and rescan devices
echo 'monitor copy status with: sudo kill -s USR1 `pidof dd`'
dd if=$SRC bs=64M | tee $DESTS > /dev/null

echo 'setting up /home partition'

# fix partition table and add extra /home partition
for dest in $DESTS; do
	parted $dest print fix > /dev/null 2>&1
	parted --script --align optimal -- $dest mkpart /home 17GB -1MB
done

# rescan devices
while ! partprobe $DESTS 2> /dev/null; do
	echo "waiting for partprobe..."
	sleep 1
done

# wait for partitions to appear
for dest in $DESTS; do
	while [ ! -b ${dest}${ROOTPT} ]; do
		echo "waiting for ${dest}${ROOTPT}..."
		sleep 1
	done
	while [ ! -b ${dest}${HOMEPT} ]; do
		echo "waiting for ${dest}${HOMEPT}..."
		sleep 1
	done
done

# create ext4 on home partition
for dest in $DESTS; do
	mkfs.ext4 -L lxhome ${dest}${HOMEPT} > /dev/null 2>&1 &
done
wait

# setup temp directory for mountpoints
TMP_ROOT="/tmp/dd-multi-`date +%s`.$$"
rm -rf $TMP_ROOT
mkdir -p $TMP_ROOT
echo "using directory $TMP_ROOT for mount points"

for dest in $DESTS; do
	# create mountpoints
	mkdir -p ${TMP_ROOT}${dest}${ROOTPT}
	mkdir -p ${TMP_ROOT}${dest}${HOMEPT}

	# mount partitions
	mount ${dest}${ROOTPT} ${TMP_ROOT}${dest}${ROOTPT}
	mount ${dest}${HOMEPT} ${TMP_ROOT}${dest}${HOMEPT}

	# move home directories to home partition
	echo "moving /home/* to ${dest}${HOMEPT}"
	mv ${TMP_ROOT}${dest}${ROOTPT}/home/* ${TMP_ROOT}${dest}${HOMEPT}/

	# setup home partition to be mounted as /home
	echo 'LABEL=lxhome /home ext4 defaults 0 0' | \
		tee -a ${TMP_ROOT}${dest}${ROOTPT}/etc/fstab > /dev/null

	# copy layer tarballs to /home
	for tarball in $LAYERS; do
		destfile="${TMP_ROOT}${dest}${HOMEPT}/`basename $tarball`"
		if [ -f $destfile ]; then
			echo
			echo "WARNING: OVERWRITING $destfile" 1>&2
			echo
			sleep 1
		fi
		cp $tarball $destfile
	done

	# remount home to /home
	umount ${TMP_ROOT}${dest}${HOMEPT}
	mount ${dest}${HOMEPT} ${TMP_ROOT}${dest}${ROOTPT}/home

	# unpack layers
	for tarball in $LAYERS; do
		destfile=`basename $tarball`
		echo "unpacking $destfile to ${dest}"
		chroot ${TMP_ROOT}${dest}${ROOTPT} \
			tar -x -f /home/$destfile --numeric-owner -C /
		rm ${TMP_ROOT}${dest}${ROOTPT}/home/$destfile
	done

	# unmount partitions
	umount ${TMP_ROOT}${dest}${ROOTPT}/home
	umount ${TMP_ROOT}${dest}${ROOTPT}
done

# cleanup temp directory
rm -rf $TMP_ROOT

echo "done, no errors"