forked from guysoft/FullPageOS
-
Notifications
You must be signed in to change notification settings - Fork 0
/
common.sh
executable file
·357 lines (297 loc) · 9.62 KB
/
common.sh
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
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
#!/usr/bin/env bash
function die () {
echo >&2 "$@"
exit 1
}
function fixLd(){
sed -i 's@/usr/lib/arm-linux-gnueabihf/libcofi_rpi.so@\#/usr/lib/arm-linux-gnueabihf/libcofi_rpi.so@' etc/ld.so.preload
sed -i 's@/usr/lib/arm-linux-gnueabihf/libarmmem.so@\#/usr/lib/arm-linux-gnueabihf/libarmmem.so@' etc/ld.so.preload
}
function restoreLd(){
sed -i 's@\#/usr/lib/arm-linux-gnueabihf/libcofi_rpi.so@/usr/lib/arm-linux-gnueabihf/libcofi_rpi.so@' etc/ld.so.preload
sed -i 's@\#/usr/lib/arm-linux-gnueabihf/libarmmem.so@/usr/lib/arm-linux-gnueabihf/libarmmem.so@' etc/ld.so.preload
}
function pause() {
# little debug helper, will pause until enter is pressed and display provided
# message
read -p "$*"
}
function gitclone(){
# call like this: gitclone FULLPAGEOS_REPO someDirectory -- this will do:
#
# sudo -u pi git clone -b $FULLPAGEOS_REPO_BRANCH --depth $FULLPAGEOS_REPO_DEPTH $FULLPAGEOS_REPO_BUILD someDirectory
#
# and if $FULLPAGEOS_REPO_BUILD != $FULLPAGEOS_REPO_SHIP also:
#
# pushd someDirectory
# sudo -u pi git remote set-url origin $FULLPAGEOS_REPO_SHIP
# popd
#
# if second parameter is not provided last URL segment of the BUILD repo URL
# minus the optional .git postfix will be used
repo_build_var=$1_BUILD
repo_ship_var=$1_SHIP
repo_branch_var=$1_BRANCH
repo_depth_var=$1_DEPTH
repo_dir=$2
if [ ! -n "$repo_dir" ]
then
repo_dir=$(echo ${REPO} | sed 's%^.*/\([^/]*\)\(\.git\)?$%\1%g')
fi
repo_depth=${!repo_depth_var}
if [ -n "$repo_depth" ]
then
depth=$repo_depth
else
if [ "$#" -gt 2 ]
then
depth=$3
fi
fi
build_repo=${!repo_build_var}
ship_repo=${!repo_ship_var}
branch=${!repo_branch_var}
if [ ! -n "$build_repo" ]
then
build_repo=$ship_repo
fi
clone_params=
if [ -n "$branch" ]
then
clone_params="-b $branch"
fi
if [ -n "$depth" ]
then
clone_params="$clone_params --depth $depth"
fi
sudo -u pi git clone $clone_params "$build_repo" "$repo_dir"
if [ "$build_repo" != "$ship_repo" ]
then
pushd "$repo_dir"
sudo -u pi git remote set-url origin "$ship_repo"
popd
fi
}
function unpack() {
# call like this: unpack /path/to/source /target user -- this will copy
# all files & folders from source to target, preserving mode and timestamps
# and chown to user. If user is not provided, no chown will be performed
from=$1
to=$2
owner=
if [ "$#" -gt 2 ]
then
owner=$3
fi
# $from/. may look funny, but does exactly what we want, copy _contents_
# from $from to $to, but not $from itself, without the need to glob -- see
# https://stackoverflow.com/a/4645159/2028598
cp -v -r --preserve=mode,timestamps $from/. $to
if [ -n "$owner" ]
then
chown -hR $owner:$owner $to
fi
}
function mount_image() {
image_path=$1
mount_path=$2
# dump the partition table, locate boot partition and root partition
boot_partition=1
root_partition=2
fdisk_output=$(sfdisk -d $image_path)
boot_offset=$(($(echo "$fdisk_output" | grep "$image_path$boot_partition" | awk '{print $4-0}') * 512))
root_offset=$(($(echo "$fdisk_output" | grep "$image_path$root_partition" | awk '{print $4-0}') * 512))
echo "Mounting image $image_path on $mount_path, offset for boot partition is $boot_offset, offset for root partition is $root_offset"
# mount root and boot partition
sudo mount -o loop,offset=$root_offset $image_path $mount_path/
sudo mount -o loop,offset=$boot_offset $image_path $mount_path/boot
sudo mount -o bind /dev/pts $mount_path/dev/pts
}
function unmount_image() {
mount_path=$1
force=
if [ "$#" -gt 1 ]
then
force=$2
fi
if [ -n "$force" ]
then
for process in $(sudo lsof $mount_path | awk '{print $2}')
do
echo "Killing process id $process..."
sudo kill -9 $process
done
fi
# Unmount everything that is mounted
#
# We might have "broken" mounts in the mix that point at a deleted image (in case of some odd
# build errors). So our "sudo mount" output can look like this:
#
# /path/to/our/image.img (deleted) on /path/to/our/mount type ext4 (rw)
# /path/to/our/image.img on /path/to/our/mount type ext4 (rw)
# /path/to/our/image.img on /path/to/our/mount/boot type vfat (rw)
#
# so we split on "on" first, then do a whitespace split to get the actual mounted directory.
# Also we sort in reverse to get the deepest mounts first.
for m in $(sudo mount | grep $mount_path | awk -F "on" '{print $2}' | awk '{print $1}' | sort -r)
do
echo "Unmounting $m..."
sudo umount $m
done
}
function cleanup() {
# make sure that all child processed die when we die
local pids=$(jobs -pr)
[ -n "$pids" ] && kill $pids && sleep 5 && kill -9 $pids
exit 0
}
function cleanup() {
# make sure that all child processed die when we die
local pids=$(jobs -pr)
[ -n "$pids" ] && kill $pids
exit 0
}
function install_fail_on_error_trap() {
set -e
trap 'previous_command=$this_command; this_command=$BASH_COMMAND' DEBUG
trap 'if [ $? -ne 0 ]; then echo -e "\nexit $? due to $previous_command \nBUILD FAILED!" && echo "unmounting image..." && ( unmount_image $FULLPAGEOS_MOUNT_PATH force || true ); fi;' EXIT
}
function install_chroot_fail_on_error_trap() {
set -e
trap 'previous_command=$this_command; this_command=$BASH_COMMAND' DEBUG
trap 'if [ $? -ne 0 ]; then echo -e "\nexit $? due to $previous_command \nBUILD FAILED!"; fi;' EXIT
}
function install_cleanup_trap() {
set -e
trap "cleanup" SIGINT SIGTERM
}
function enlarge_ext() {
# call like this: enlarge_ext /path/to/image partition size
#
# will enlarge partition number <partition> on /path/to/image by <size> MB
image=$1
partition=$2
size=$3
echo "Adding $size MB to partition $partition of $image"
start=$(sfdisk -d $image | grep "$image$partition" | awk '{print $4-0}')
offset=$(($start*512))
dd if=/dev/zero bs=1M count=$size >> $image
fdisk $image <<FDISK
p
d
$partition
n
p
$partition
$start
p
w
FDISK
LODEV=$(losetup -f --show -o $offset $image)
trap 'losetup -d $LODEV' EXIT
e2fsck -fy $LODEV
resize2fs -p $LODEV
losetup -d $LODEV
trap - EXIT
echo "Resized parition $partition of $image to +$size MB"
}
function shrink_ext() {
# call like this: shrink_ext /path/to/image partition size
#
# will shrink partition number <partition> on /path/to/image to <size> MB
image=$1
partition=$2
size=$3
echo "Resizing file system to $size MB..."
start=$(sfdisk -d $image | grep "$image$partition" | awk '{print $4-0}')
offset=$(($start*512))
LODEV=$(losetup -f --show -o $offset $image)
trap 'losetup -d $LODEV' EXIT
e2fsck -fy $LODEV
e2ftarget_bytes=$(($size * 1024 * 1024))
e2ftarget_blocks=$(($e2ftarget_bytes / 512 + 1))
echo "Resizing file system to $e2ftarget_blocks blocks..."
resize2fs $LODEV ${e2ftarget_blocks}s
losetup -d $LODEV
trap - EXIT
new_end=$(($start + $e2ftarget_blocks))
echo "Resizing partition to end at $start + $e2ftarget_blocks = $new_end blocks..."
fdisk $image <<FDISK
p
d
$partition
n
p
$partition
$start
$new_end
p
w
FDISK
new_size=$((($new_end + 1) * 512))
echo "Truncating image to $new_size bytes..."
truncate --size=$new_size $image
fdisk -l $image
echo "Resizing filesystem ..."
LODEV=$(losetup -f --show -o $offset $image)
trap 'losetup -d $LODEV' EXIT
e2fsck -fy $LODEV
resize2fs -p $LODEV
losetup -d $LODEV
trap - EXIT
}
function minimize_ext() {
image=$1
partition=$2
buffer=$3
echo "Resizing partition $partition on $image to minimal size + $buffer MB"
partitioninfo=$(sfdisk -d $image | grep "$image$partition")
start=$(echo $partitioninfo | awk '{print $4-0}')
e2fsize_blocks=$(echo $partitioninfo | awk '{print $6-0}')
offset=$(($start*512))
LODEV=$(losetup -f --show -o $offset $image)
trap 'losetup -d $LODEV' EXIT
e2fsck -fy $LODEV
e2fblocksize=$(tune2fs -l $LODEV | grep -i "block size" | awk -F: '{print $2-0}')
e2fminsize=$(resize2fs -P $LODEV 2>/dev/null | grep -i "minimum size" | awk -F: '{print $2-0}')
e2fminsize_bytes=$(($e2fminsize * $e2fblocksize))
e2ftarget_bytes=$(($buffer * 1024 * 1024 + $e2fminsize_bytes))
e2fsize_bytes=$((($e2fsize_blocks - 1) * 512))
e2fminsize_mb=$(($e2fminsize_bytes / 1024 / 1024))
e2fminsize_blocks=$(($e2fminsize_bytes / 512 + 1))
e2ftarget_mb=$(($e2ftarget_bytes / 1024 / 1024))
e2ftarget_blocks=$(($e2ftarget_bytes / 512 + 1))
e2fsize_mb=$(($e2fsize_bytes / 1024 / 1024))
size_offset_mb=$(($e2fsize_mb - $e2ftarget_mb))
losetup -d $LODEV
echo "Actual size is $e2fsize_mb MB ($e2fsize_blocks blocks), Minimum size is $e2fminsize_mb MB ($e2fminsize file system blocks, $e2fminsize_blocks blocks)"
echo "Resizing to $e2ftarget_mb MB ($e2ftarget_blocks blocks)"
if [ $size_offset_mb -gt 0 ]; then
echo "Partition size is bigger then the desired size, shrinking"
shrink_ext $image 2 $(($e2ftarget_mb - 1)) # -1 to compensat rounding mistakes
elif [ $size_offset_mb -lt 0 ]; then
echo "Partition size is lower then the desired size, enlarging"
enlarge_ext $image 2 $((-$size_offset_mb + 1)) # +1 to compensat rounding mistakes
fi
}
function is_installed(){
# checks if a package is installed, returns 1 if installed and 0 if not.
# usage: is_installed <package_name>
dpkg-query -W -f='${Status}' $1 2>/dev/null | grep -c "ok installed"
}
function is_in_apt(){
#checks if a package is in the apt repo, returns 1 if exists and 0 if not
#usage is_in_apt <package_name>
if [ $(apt-cache policy $1 | wc | awk '{print $1}') -gt 0 ]; then
echo 1
else
echo 0
fi
}
systemctl_if_exists() {
if hash systemctl 2>/dev/null; then
systemctl "$@"
else
echo "no systemctl, not running"
fi
}