debian-config-functions 21.4 KB
Newer Older
1
2
#!/bin/bash
#
3
# Copyright (c) Authors: http://www.armbian.com/authors, info@armbian.com
4
5
6
7
8
#
# This file is licensed under the terms of the GNU General Public
# License version 2. This program is licensed "as is" without any
# warranty of any kind, whether express or implied.

9
10
# Functions:
# main
11
# check_desktop
12
# exceptions
13
14
15
16
17
18
# check_if_installed
# is_package_manager_running
# display_qr_code
# beta_disclaimer
# show_box
# description
19
# generic_select
20
# reload_bsp
21
# other_kernel_version
22
23
24
25
# aval_dtbs
# get_a20modes
# get_h3modes
# add_choose_user
26
# google_token_allusers
27
# configure_desktop
28

29

30
#
31
32
# gather info about the board and start with loading menu
#
33
function main(){
34

35
36
	DIALOG_CANCEL=1
	DIALOG_ESC=255
37

38
39
40
41
42
	[[ -f /etc/armbian-release ]] && source /etc/armbian-release && ARMBIAN="Armbian $VERSION $IMAGE_TYPE";
	DISTRO=$(lsb_release -is)
	DISTROID=$(lsb_release -sc)
	KERNELID=$(uname -r)
	[[ -z "${ARMBIAN// }" ]] && ARMBIAN="$DISTRO $DISTROID"
Igor Pečovnik's avatar
bugfix    
Igor Pečovnik committed
43
	DEFAULT_ADAPTER=$(ip -4 route ls | grep default | tail -1 | grep -Po '(?<=dev )(\S+)')
Igor Pecovnik's avatar
Igor Pecovnik committed
44
	LOCALIPADD=$(ip -4 addr show dev $DEFAULT_ADAPTER | awk '/inet/ {print $2}' | cut -d'/' -f1)
45
46
	BACKTITLE="Configuration utility, $ARMBIAN"
	[[ -n "$LOCALIPADD" ]] && BACKTITLE=$BACKTITLE", "$LOCALIPADD
Igor Pecovnik's avatar
Igor Pecovnik committed
47
	TITLE="$BOARD_NAME "
48
49
50
	[[ -z "${DEFAULT_ADAPTER// }" ]] && DEFAULT_ADAPTER="lo"
	OVERLAYDIR="/boot/dtb/overlay";
	[[ "$LINUXFAMILY" == "sunxi64" ]] && OVERLAYDIR="/boot/dtb/allwinner/overlay";
51
	[[ "$LINUXFAMILY" == "meson64" ]] && OVERLAYDIR="/boot/dtb/amlogic/overlay";
52
	# detect desktop
53
	check_desktop
54
55
	dialog --backtitle "$BACKTITLE" --title "Please wait" --infobox "\nLoading Armbian configuration utility ... " 5 45
	sleep 1
56

57
58
59
60
}



Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
61

62
63
64
65
#
# compare two strings in dot separated version format
#
vercomp () {
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
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

	if [[ $1 == $2 ]]
	then
		return 0
	fi
	local IFS=.
	local i ver1=($1) ver2=($2)
	# fill empty fields in ver1 with zeros
	for ((i=${#ver1[@]}; i<${#ver2[@]}; i++))
	do
		ver1[i]=0
	done
	for ((i=0; i<${#ver1[@]}; i++))
	do
		if [[ -z ${ver2[i]} ]]
		then
			# fill empty fields in ver2 with zeros
			ver2[i]=0
		fi
		if ((10#${ver1[i]} > 10#${ver2[i]}))
		then
			return 1
		fi
		if ((10#${ver1[i]} < 10#${ver2[i]}))
		then
			return 2
		fi
	done
	return 0

96
97
98
99
100
101
102
103
104
}




#
# test compare two strings $1="3.4.12" $2="5.4.2" $3="<" returns 0 if relation is correct
#
testvercomp () {
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
105
106
107
108
109
110
111
112
113

	vercomp $1 $2
	case $? in
		0) op='=';;
		1) op='>';;
		2) op='<';;
	esac
	if [[ $op != $3 ]]
	then
114
		return 1
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
115
	else
116
		return 0
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
117
118
	fi

119
120
121
122
}



123

124
125
126
127
128
#
# read desktop parameters
#
function check_desktop()
{
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
129
130
131
132
133
134
135
136
137

	DISPLAY_MANAGER=""; DESKTOP_INSTALLED=""
	check_if_installed nodm && DESKTOP_INSTALLED="nodm";
	check_if_installed lightdm && DESKTOP_INSTALLED="lightdm";
	check_if_installed lightdm && DESKTOP_INSTALLED="gnome";
	[[ -n $(service lightdm status 2> /dev/null | grep -w active) ]] && DISPLAY_MANAGER="lightdm"
	[[ -n $(service nodm status 2> /dev/null | grep -w active) ]] && DISPLAY_MANAGER="nodm"
	[[ -n $(service gdm status 2> /dev/null | grep -w active) ]] && DISPLAY_MANAGER="gdm"

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
#
# naming exceptions for packages
#
function exceptions ()
{

	TARGET_FAMILY=$LINUXFAMILY
	UBOOT_BRANCH=$TARGET_BRANCH # uboot naming is different

	if [[ $TARGET_BRANCH == "default" ]]; then TARGET_BRANCH=""; else TARGET_BRANCH="-"$TARGET_BRANCH; fi
	# pine64
	if [[ $TARGET_FAMILY == pine64 ]]; then
		TARGET_FAMILY="sunxi64"
	fi
	# allwinner legacy kernels
	if [[ $TARGET_FAMILY == sun*i ]]; then
		TARGET_FAMILY="sunxi"
		if [[ $UBOOT_BRANCH == "default" ]]; then
			TARGET_FAMILY=$(cat /proc/cpuinfo | grep "Hardware" | sed 's/^.*Allwinner //' | awk '{print $1;}')
		fi
	fi

}




170
#
JC Staudt's avatar
JC Staudt committed
171
# check dpkg status of $1 -- currently only 'not installed at all' case caught
172
173
174
175
176
177
178
179
180
#
check_if_installed (){

	local DPKG_Status="$(dpkg -s "$1" 2>/dev/null | awk -F": " '/^Status/ {print $2}')"
	if [[ "X${DPKG_Status}" = "X" || "${DPKG_Status}" = *deinstall* ]]; then
		return 1
	else
		return 0
	fi
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
181

182
183
184
185
186
187
}




#
188
189
190
# check if package manager is doing something
#
function is_package_manager_running() {
191

192
193
194
195
  fuser -s /var/lib/dpkg/lock
  if [[ $? = 0 ]]; then
    # 0 = true
	dialog --colors --title " \Z1Error\Z0 " --backtitle "$BACKTITLE" --no-collapse --msgbox \
root's avatar
root committed
196
	"\n\Z0Package manager is running in the background. \n\nCan't install dependencies. Try again later." 9 53
197
198
199
200
201
    return 0
  else
    # 1 = false
    return 1
  fi
202
203
204
205
206

}



Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
207

Igor Pecovnik's avatar
Igor Pecovnik committed
208
209
210
211
212
#
# wget with dialog progress bar $1=URL $2=parameters
#
function fancy_wget()
{
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
213
214
215
216
217

	LANG=C wget $2 --progress=bar:force:noscroll $1 2>&1 | stdbuf -i0 -o0 -e0 tr '>' '\n' | \
	stdbuf -i0 -o0 -e0 sed -rn 's/^.*\<([0-9]+)%\[.*$/\1/p' | dialog --backtitle "$BACKTITLE" --title " Downloading " \
	--gauge "Please wait" 7 70 0

Igor Pecovnik's avatar
Igor Pecovnik committed
218
219
220
}


221

Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
222

223
#
JC Staudt's avatar
JC Staudt committed
224
# display qr code for authentication method
225
226
227
#
function display_qr_code()
{
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
228

229
230
		clear
		SECRET=$(head -1 /root/.google_authenticator)
Igor Pečovnik's avatar
Igor Pečovnik committed
231
		qrencode -m 2 -d 9 -8 -t ANSI256 "otpauth://totp/test?secret=$SECRET"
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
232
233
234
235
236
		echo -e "\nHow to setup your one type password generator?\n"
		echo -e "\nInstall a one-time password authenticator on your mobile device (e.g. FreeOTP) from the Android \
market or F-Droid.	\n\nIn the application menu, click the corresponding button to create a new account and either \
scan the QR code or enter the secret key manually:\\n\n$SECRET \n\nYou should now see a new passcode token being \
generated every 60 seconds on your phone.\n"  | fold -sw 60
237
		read -n 1 -s -r -p "Press any key to continue"
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
238

239
240
241
242
243
}




244
245
246
247
248
#
# show disclaimer
#
function beta_disclaimer ()
{
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
249
250
251
252
253
254

	exec 3>&1
	ACKNOWLEDGEMENT=$(dialog --colors --nocancel --backtitle "$BACKTITLE" --no-collapse --title " Warning " \
	--clear \--radiolist "\n$1\n \n" 0 56 5 "Yes, I understand" "" off	 2>&1 1>&3)
	exec 3>&-

255
256
}

257
258
259
260
261
262
263
264



#
# show box
#
function show_box ()
{
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
265

266
	dialog --colors --backtitle "$BACKTITLE" --no-collapse --title " $1 " --clear --msgbox "\n$2\n \n" $3 56
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
267

268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
}




#
# show description for MOTD files
#
function description
{
	case $1 in
		*header*)
			echo "Big board logo and kernel info"
		;;
		*sysinfo*)
			echo "Sysinfo - load, ip, memory, uptime, ..."
		;;
		*tips*)
			echo "Shows tip of the day"
		;;
		*updates*)
Chris Rohlfs's avatar
Chris Rohlfs committed
289
			echo "Display number of available updates"
290
291
292
293
294
295
296
297
298
299
		;;
		*armbian-config*)
			echo "Show command for system configuration"
		;;
		*autoreboot-warn*)
			echo "Show warning when reboot is needed"
		;;
		*uk.armbian.com*)
			echo "United Kingdom"
		;;
Igor Pečovnik's avatar
Igor Pečovnik committed
300
301
302
		*mirrors.tuna.tsinghua.edu.cn/armbian/*)
			echo "China"
		;;
Igor Pečovnik's avatar
bugfix    
Igor Pečovnik committed
303
		*mirrors.netix.net/armbian/apt/*)
304
			echo "Bulgarija"
Igor Pečovnik's avatar
Igor Pečovnik committed
305
306
307
308
		;;		
		*mirrors.dotsrc.org/armbian-apt/*)
			echo "Denmark"
		;;		
309
310
311
312
313
314
315
316
317
318
		*.armbian.com*)
			echo "Estonia"
		;;
		*)
		echo ""
		;;
	esac
}


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
#
# Generic select box
#
function generic_select()
{
        IFS=$' '
        PARAMETER=($1)
        local LIST=()
        for i in "${PARAMETER[@]}"
        do
                        if [[ -n $3 ]]; then
                                [[ ${i[0]} -ge $3 ]] && \
                                LIST+=( "${i[0]//[[:blank:]]/}" "" )
                        else
                                LIST+=( "${i[0]//[[:blank:]]/}" "" )
                        fi
        done
        LIST_LENGTH=$((${#LIST[@]}/2));
        if [ "$LIST_LENGTH" -eq 1 ]; then
                        PARAMETER=${PARAMETER[0]}
        else
                        exec 3>&1
                        PARAMETER=$(dialog --nocancel --backtitle "$BACKTITLE" --no-collapse \
                        --title "$2" --clear --menu "" $((6+${LIST_LENGTH})) 0 $((1+${LIST_LENGTH})) "${LIST[@]}" 2>&1 1>&3)
                        exec 3>&-
        fi
}


348
349


350
351
352
353
#
# reload kernel, bsp and armbian-config
#
function reload_bsp(){
354
	clear
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
355
	debconf-apt-progress -- apt-get update
356
357
358
359
360
361
362
363
364
365

	# new branch names introduced with 12/2019
	if [[ -n $(apt-cache search --names-only "^linux-image-current") ]]; then
		[[ BRANCH == next ]] && TARGET_BRANCH="legacy" && TARGET_UBOOT_BRANCH="legacy"
		TARGET_BRANCH="current"
		TARGET_UBOOT_BRANCH="current"
	else
		TARGET_BRANCH="${BRANCH}"
		TARGET_UBOOT_BRANCH="${UBOOT_BRANCH}"
	fi
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
366
	exceptions "$INSTALL_KERNEL"
367
	unset PACKAGE_LIST
368
369
370
371
372
	PACKAGE_LIST="linux-${DISTROID}-root$TARGET_BRANCH-$BOARD"
	[[ -z ${TARGET_UBOOT_BRANCH} ]] && TARGET_UBOOT_BRANCH="current"
	PACKAGE_LIST=$PACKAGE_LIST" linux-u-boot-$BOARD-$TARGET_UBOOT_BRANCH"
	check_if_installed armbian-${DISTROID}-desktop && PACKAGE_LIST=$PACKAGE_LIST" armbian-${DISTROID}-desktop"
	check_if_installed linux-headers${TARGET_BRANCH}-${TARGET_FAMILY} && PACKAGE_LIST=$PACKAGE_LIST" linux-headers${TARGET_BRANCH}-${TARGET_FAMILY}"
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
373

374
	IFS=" "
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
375
376
	[[ -n $(apt-cache search --names-only "^linux-dtb$TARGET_BRANCH-$TARGET_FAMILY") ]] && \
	PACKAGE_LIST=$PACKAGE_LIST" linux-dtb$TARGET_BRANCH-$TARGET_FAMILY"
377
	PACKAGE_LIST=$PACKAGE_LIST" linux-image${TARGET_BRANCH}-${TARGET_FAMILY}"
378
	debconf-apt-progress --	apt --download-only --allow-downgrades -y --no-install-recommends install $PACKAGE_LIST armbian-config
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
379
380
	# if download is ok, remove old packages
	if [[ $? = 0 ]]; then
381
382
383
384
385
386

		debconf-apt-progress -- apt -y -qq purge linux-u-boot* linux-image* linux-dtb* linux-headers* linux-${DISTROID}-root* armbian-${DISTROID}-desktop* armbian-config
		find "/boot/" -name "System.map*" -type f -delete
		find "/boot/" -name "config*" -type f -delete
		find "/boot/" -name "vmlinuz*" -type f -delete
		find "/boot/" -name "*nitrd*" -type f -delete
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
387
		# install packages
388
		echo $PACKAGE_LIST >> /var/log/upgrade.log
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
389
390
		debconf-apt-progress -- apt -y -qq --allow-downgrades --no-install-recommends --reinstall \
		-o Dpkg::Options::="--force-confdef" -o Dpkg::Options::="--force-confold" install $PACKAGE_LIST armbian-config
391

Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
392
		if [[ $? = 1 ]]; then
393
			echo "Something went wrong ... check logs."; exit;
Igor Pečovnik's avatar
Igor Pečovnik committed
394
		else
395
			apt clean
396
		fi
397

Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
398
399
	fi

400
401
402
}


Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
403
404


405
406
407
function other_kernel_version ()
{

Igor Pečovnik's avatar
Igor Pečovnik committed
408
409
410
411
412
413
	IFS=$'\r\n'
	GLOBIGNORE='*'

	# get current kernel information
	CURRENT_VERSION_TEMP=$(dpkg -l | grep '^ii' | grep linux-image)
	CURRENT_VERSION=$(echo $CURRENT_VERSION_TEMP | awk '{print $2}')"="$(echo $CURRENT_VERSION_TEMP | awk '{print $3}')
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
414

Igor Pečovnik's avatar
Igor Pečovnik committed
415
	# check what is available from the repository
Igor Pecovnik's avatar
Bugfix    
Igor Pecovnik committed
416
	debconf-apt-progress -- apt-get update
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
417
418
	LIST=($(apt-cache show linux-image*${LINUXFAMILY} | grep -E  "Package:|Version:|version:" \
	| grep -v "Config-Version" | sed -n -e 's/^.*: //p' | sed 's/\.$//g'))
419
	new_list=()
Igor Pečovnik's avatar
Igor Pečovnik committed
420
421

	# create a human readable menu
422
423
	for ((n=0;n<$((${#LIST[@]}));n++));
	do
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
424
425
426
427
428
		m=$(( $n + 1 ))
		prvi=$((3*$m - 3))
		drugi=$((3*$m - 2))
		tretji=$((3*$m - 1))
		[[ -z ${LIST[$prvi]} ]] && break
Igor Pečovnik's avatar
Igor Pečovnik committed
429
430
431
432
		if [[ $CURRENT_VERSION != "${LIST[$prvi]}=${LIST[$drugi]}" ]]; then
			new_list+=( "${LIST[$prvi]}=${LIST[$drugi]}" )
			new_list+=( ${LIST[$tretji]} )
		fi
433
434
435
436
	done

	# copy back to main array
	LIST=("${new_list[@]}")
JC Staudt's avatar
JC Staudt committed
437
	LIST_LENGTH=$((${#LIST[@]}/2));
438

JC Staudt's avatar
JC Staudt committed
439
	if [ "$LIST_LENGTH" -eq 0 ]; then
Igor Pečovnik's avatar
Igor Pečovnik committed
440
		dialog --backtitle "$BACKTITLE" --title " Warning " --msgbox  "\nNo other kernels available!" 7 32
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
441
442
443
	else
		beta_disclaimer "Switching between kernels might change functionality of your board or it might fail to boot. \
		\n\n\Z1In case of troubles expect no help!\Z0"
Igor Pečovnik's avatar
Igor Pečovnik committed
444
445
446
		if [[ -n $ACKNOWLEDGEMENT ]]; then
			exec 3>&1
            TARGET_VERSION=$(dialog --cancel-label "Cancel" --backtitle "$BACKTITLE" --no-collapse \
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
447
448
			--title "Switch from and reboot" --clear --menu "\n${CURRENT_VERSION} $(uname -r) \n \n" \
			$((9+${LIST_LENGTH})) 62 25 "${LIST[@]}" 2>&1 1>&3)
Igor Pečovnik's avatar
Igor Pečovnik committed
449
450
451
452
			exitstatus=$?;
			exec 3>&-
			if [[ $exitstatus = 0 ]]; then
				IFS=" "
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
453

454
455
456
457
458
459
460
				# determine upgrade packages
				UPGRADE_ROOT="default"
				UPGRADE_UBOOT=""
				[[ $TARGET_VERSION == *legacy* ]] && UPGRADE_ROOT="-legacy" && UPGRADE_UBOOT="legacy"
				[[ $TARGET_VERSION == *next* ]] && UPGRADE_ROOT="-next"  && UPGRADE_UBOOT="next"
				[[ $TARGET_VERSION == *current* ]] && UPGRADE_ROOT="-current" && UPGRADE_UBOOT="current"
				[[ $TARGET_VERSION == *dev* ]] && UPGRADE_ROOT="-dev" && UPGRADE_UBOOT="dev"
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
461

Igor Pečovnik's avatar
Igor Pečovnik committed
462
463
464
465
466
				# install packages
				PACKAGE_LIST="$TARGET_VERSION"
				TARGET_VERSION_DTB=${TARGET_VERSION/image/dtb}
				TARGET_VERSION_PRE=$(echo $TARGET_VERSION_DTB | cut -f1 -d"=")
				TARGET_VERSION_SUB=$(echo $TARGET_VERSION_DTB | cut -f2 -d"=")
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
467
				[[ -n $(apt-cache madison "$TARGET_VERSION_PRE" | grep $TARGET_VERSION_SUB ) ]] && \
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
468
469

				PACKAGE_LIST=$PACKAGE_LIST" $TARGET_VERSION_DTB linux-u-boot-${BOARD}-${UPGRADE_UBOOT} linux-${DISTROID}-root${UPGRADE_ROOT}-${BOARD}"
Igor Pečovnik's avatar
Igor Pečovnik committed
470
				echo $PACKAGE_LIST > /tmp/switch_kernel.log 2>&1
471
				debconf-apt-progress -- apt --download-only --allow-downgrades -y --no-install-recommends install $PACKAGE_LIST
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
472

Igor Pečovnik's avatar
Igor Pečovnik committed
473
474
				if [[ $? = 0 ]]; then
					dialog --backtitle "$BACKTITLE" --title "Please wait" --infobox "\nRemoving current kernel ..." 5 36
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
475

476
					# remove old kernel
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
477
478
					debconf-apt-progress -- apt -y -qq purge linux-u-boot* linux-image* linux-dtb* linux-headers* linux-${DISTROID}-root*

479
					# cleanup
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
480
481
					find "/boot/" -name "System.map*" -type f -delete;	find "/boot/" -name "config*" -type f -delete
					find "/boot/" -name "vmlinuz*" -type f -delete;	find "/boot/" -name "*nitrd*" -type f -delete
Igor Pecovnik's avatar
Igor Pecovnik committed
482
					apt clean
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
483
					debconf-apt-progress -- apt -y -qq --allow-downgrades --no-install-recommends install $PACKAGE_LIST
Igor Pečovnik's avatar
Igor Pečovnik committed
484
485
					if [[ $? = 0 ]]; then reboot; fi
				else
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
486
487
					dialog --backtitle "$BACKTITLE" --title "Warning" --msgbox \
					"\nTest install failed. Can't change firmware \n\nCheck /tmp/switch_kernel.log" 9 48
488
				fi
489
			fi
Igor Pečovnik's avatar
Igor Pečovnik committed
490
		fi
491
492
493
494
495
	fi
}



Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
496

497
498
499
#
# check if board has alternative kernels
#
500
function aval_dtbs ()
501
502
{

503
504
	if [[ $LINUXFAMILY == cubox ]]; then
	local width=80
505
506
507
508
509
510
	LIST=("imx6dl-hummingboard.dtb" "HB Solo/DualLite" "imx6dl-hummingboard-emmc-som-v15.dtb" "HB Solo/DualLite v1.5 with eMMC" "imx6dl-hummingboard-som-v15.dtb" "HB Solo/DualLite v1.5" \
	"imx6dl-hummingboard2.dtb" "HB2 Solo/DualLite" "imx6dl-hummingboard2-emmc-som-v15.dtb" "HB2 Solo/DualLite v1.5 with eMMC" "imx6dl-hummingboard2-som-v15.dtb" "HB2 Solo/DualLite v1.5" \
	"imx6q-hummingboard.dtb" "HB Dual/Quad" "imx6q-hummingboard-emmc-som-v15.dtb" "HB Dual/Quad v1.5 with eMMC" "imx6q-hummingboard-som-v15.dtb" "HB Dual/Quad v1.5" \
	"imx6q-hummingboard2.dtb" "HB2 Dual/Quad" "imx6q-hummingboard2-emmc-som-v15.dtb" "HB2 Dual/Quad v1.5 with eMMC" "imx6q-hummingboard2-som-v15.dtb" "HB2 Dual/Quad v1.5" \
	"imx6dl-cubox-i.dtb" "Cubox-i Solo/DualLite" "imx6dl-cubox-i-emmc-som-v15.dtb" "Cubox-i Solo/DualLite v1.5 with eMMC" "imx6dl-cubox-i-som-v15.dtb" "Cubox-i Solo/DualLite v1.5" \
	"imx6q-cubox-i.dtb" "Cubox-i Dual/Quad" "imx6q-cubox-i-emmc-som-v15.dtb" "Cubox-i Dual/Quad v1.5 with eMMC" "imx6q-cubox-i-som-v15.dtb" "Cubox-i Dual/Quad v1.5")
511
512
513
514
	else
	local width=52
	LIST=("xu4" "Odroid XU4" "xu3" "Odroid XU3" "xu3l" "Odroid XU3 Lite" "hc1" "Odroid HC1/HC2")
	fi
515

JC Staudt's avatar
JC Staudt committed
516
517
	LIST_LENGTH=$((${#LIST[@]}/2));
	if [ "$LIST_LENGTH" -eq 1 ]; then
518
519
520
521
			TARGET_BOARD=${AVAL_KERNEL[0]}
	else
			exec 3>&1
			TARGET_BOARD=$(dialog --cancel-label "Cancel" --backtitle "$BACKTITLE" --no-collapse \
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
522
523
			--title "Select optimised board configuration" --clear --menu \
			"" $((6+${LIST_LENGTH})) ${width} 25 "${LIST[@]}" 2>&1 1>&3)
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
			exitstatus=$?;
			exec 3>&-
	fi

}




#
# select video modes for a10 and a20
#
function get_a20modes ()
{

	IFS=$'\r'
	GLOBIGNORE='*'
	SCREEN_RESOLUTION=("1920x1080p60" "1280x720p60" "1920x1080p50" "1280x1024p60" "1024x768p60" "800x600p60" "640x480p60" "1360x768p60" "1440x900p60" "1680x1050p60")
	local LIST=()
	for i in "${SCREEN_RESOLUTION[@]}"
	do
			LIST+=( "${i[0]//[[:blank:]]/}" "" )
	done
JC Staudt's avatar
JC Staudt committed
547
548
	LIST_LENGTH=$((${#LIST[@]}/2));
	#echo $LIST_LENGTH
549
	#exit
JC Staudt's avatar
JC Staudt committed
550
	if [ "$LIST_LENGTH" -eq 1 ]; then
551
552
553
554
			SCREEN_RESOLUTION=${SCREEN_RESOLUTION[0]}
	else
			exec 3>&1
			SCREEN_RESOLUTION=$(dialog --nocancel --backtitle "$BACKTITLE" --no-collapse \
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
555
556
			--title "Select video mode" --clear --menu \
			"" $((6+${LIST_LENGTH})) 25 $((1+${LIST_LENGTH})) "${LIST[@]}" 2>&1 1>&3)
557
558
559
560
561
562
563
564
			exec 3>&-
	fi

}




565
566
567
568
569
570
571
572
573
#
# select video modes for odroid c1/c2
#
function get_odroidmodes ()
{

        IFS=$'\r\n'
        GLOBIGNORE='*'
        SCREEN_RESOLUTION=($(cat /boot/boot.ini | grep -w "# setenv" | grep "hz" | cut -d'"' -f 2))
574
	SCREEN_RESOLUTION=($(cat /boot/boot.ini | grep "Progressive" | grep -v "setenv" | cut -d'"' -f 2))
575
576
577
578
579
        local LIST=()
        for i in "${SCREEN_RESOLUTION[@]}"
        do
                        LIST+=( "${i[0]//[[:blank:]]/}" "" )
        done
JC Staudt's avatar
JC Staudt committed
580
581
        LIST_LENGTH=$((${#LIST[@]}/2));
        #echo $LIST_LENGTH
582
        #exit
JC Staudt's avatar
JC Staudt committed
583
        if [ "$LIST_LENGTH" -eq 1 ]; then
584
585
586
587
                        SCREEN_RESOLUTION=${SCREEN_RESOLUTION[0]}
        else
                        exec 3>&1
                        SCREEN_RESOLUTION=$(dialog --nocancel --backtitle "$BACKTITLE" --no-collapse \
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
588
589
                        --title "Select video mode" --clear --menu \
                        "" $((6+${LIST_LENGTH})) 25 $((1+${LIST_LENGTH})) "${LIST[@]}" 2>&1 1>&3)
590
591
592
593
594
595
596
597
                        exec 3>&-
        fi

}




598
599
600
601
602
603
604
605
#
# select video modes for h3
#
function get_h3modes ()
{

	IFS=$'\r\n'
	GLOBIGNORE='*'
Igor Pečovnik's avatar
Igor Pečovnik committed
606
	SCREEN_RESOLUTION=($(h3disp -i clean))
607
608
609
610
611
	local LIST=()
	for i in "${SCREEN_RESOLUTION[@]}"
	do
			LIST+=( "${i[0]//[[:blank:]]/}" "" )
	done
JC Staudt's avatar
JC Staudt committed
612
613
	LIST_LENGTH=$((${#LIST[@]}/2));
	#echo $LIST_LENGTH
614
	#exit
JC Staudt's avatar
JC Staudt committed
615
	if [ "$LIST_LENGTH" -eq 1 ]; then
616
617
618
619
			SCREEN_RESOLUTION=${SCREEN_RESOLUTION[0]}
	else
			exec 3>&1
			SCREEN_RESOLUTION=$(dialog --nocancel --backtitle "$BACKTITLE" --no-collapse \
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
620
621
			--title "Select video mode" --clear --menu \
			"" $((6+${LIST_LENGTH})) 25 $((1+${LIST_LENGTH})) "${LIST[@]}" 2>&1 1>&3)
622
623
624
625
626
627
628
			exec 3>&-
	fi

}



629

630
631
632
633
634
635
636
637
638
639
640
641
642
#
# create or pick unprivileged user
#
function add_choose_user ()
{

	IFS=$'\r\n'
	GLOBIGNORE='*'

	local USERS=($(awk -F'[/:]' '{if ($3 >= 1000 && $3 != 65534) print $1}' /etc/passwd))
	local LIST=()
	for i in "${USERS[@]}"
	do
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
643
		LIST+=( "${i[0]//[[:blank:]]/}" "" )
644
	done
JC Staudt's avatar
JC Staudt committed
645
	LIST_LENGTH=$((${#LIST[@]}/2));
646

JC Staudt's avatar
JC Staudt committed
647
	if [ "$LIST_LENGTH" -eq 0 ]; then
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
648
649
		dialog --backtitle "$BACKTITLE" --title " Notice " --msgbox \
		"\nWe didn't find any unprivileged user with sudo rights which is required to run this service.\
650
651
652
653
654
655
656
657
		\n\nPress enter to create one!" 10 48
		read -t 0 temp
		echo -e "\nPlease provide a username (eg. your forename) or leave blank for canceling user creation: \c"
		read -e username
		CHOSEN_USER="$(echo "$username" | tr '[:upper:]' '[:lower:]' | tr -d -c '[:alnum:]')"
		[ -z "$CHOSEN_USER" ] && return
		echo "Trying to add user $CHOSEN_USER"
		adduser $CHOSEN_USER || return
JC Staudt's avatar
JC Staudt committed
658
	elif [ "$LIST_LENGTH" -eq 1 ]; then
659
660
661
662
			CHOSEN_USER=${USERS[0]}
	else
			exec 3>&1
			CHOSEN_USER=$(dialog --nocancel --backtitle "$BACKTITLE" --no-collapse \
JC Staudt's avatar
JC Staudt committed
663
			--title "Select unprivileged user" --clear --menu "" $((6+${LIST_LENGTH})) 40 15 "${LIST[@]}" 2>&1 1>&3)
664
665
666
667
668
669
670
671
			exec 3>&-
	fi

}




672
673
674
675
676
677
#
# Copy Google token to all local users.
#
function google_token_allusers ()
{

Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
678
679
680
681
682
683
684
685
686
687
688
689
	if [[ -f /root/.google_authenticator ]]; then
		local USERS=($(awk -F'[/:]' '{if ($3 >= 1000 && $3 != 65534) print $1}' /etc/passwd))
		for i in "${USERS[@]}"
		do
			USER=${i[0]//[[:blank:]]/}
			if [[ -d /home/$USER ]]; then
				cp /root/.google_authenticator /home/$USER/
				chown ${USER}:${USER} /home/${USER}/.google_authenticator
			fi
		done
	fi

690
691
692
693
694
}




695
696
697
698
699
700
701
702
703
#
# configure armbian desktop
#
function configure_desktop ()
{

	add_choose_user

	if [ -n "$CHOSEN_USER" ]; then
704
705

		# update packages
706
		debconf-apt-progress -- apt-get update
707

Igor Pečovnik's avatar
Igor Pečovnik committed
708
709
710
711
		# install new package if exists
		unset PACKAGE_SUFIX
		[[ -n $(apt-cache search --names-only "^armbian-${DISTROID}-desktop-xfce$") ]] && PACKAGE_SUFIX="-xfce"

712
713
		# remove desktop package to secure proper install
		if check_if_installed armbian-${DISTROID}-desktop ; then
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
714
715
			debconf-apt-progress -- apt-get -y \
			remove armbian-${DISTROID}-desktop${PACKAGE_SUFIX} lightdm lightdm-gtk-greeter
716
717
718
		fi

		# install desktop package
Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
719
720
721
		debconf-apt-progress -- apt-get --reinstall -o Dpkg::Options::="--force-confdef" \
		-o Dpkg::Options::="--force-confold" -y \
		install $1 armbian-${DISTROID}-desktop${PACKAGE_SUFIX} lightdm lightdm-gtk-greeter
root's avatar
root committed
722

Igor Pečovnik's avatar
Igor Pečovnik committed
723
724
		# in case previous install was interrupted
		[[ $? -eq 130 ]] && dpkg --configure -a
root's avatar
root committed
725

726
		# clean apt cache
Igor Pecovnik's avatar
Igor Pecovnik committed
727
		apt clean
728

729
		# add user to groups
730
		for additionalgroup in sudo netdev audio video dialout plugdev input bluetooth systemd-journal ssh; do
731
732
733
				usermod -aG ${additionalgroup} ${CHOSEN_USER} 2>/dev/null
		done

Igor Pecovnik's avatar
Cleanup    
Igor Pecovnik committed
734
735
		# Prevent loading paralel printer port drivers which we don't need here.
		# suppress boot error if kernel modules are absent
736
		if [[ -f /etc/modules-load.d/cups-filters.conf ]]; then
Igor Pečovnik's avatar
Igor Pečovnik committed
737
738
739
			sed "s/^lp/#lp/" -i /etc/modules-load.d/cups-filters.conf
			sed "s/^ppdev/#ppdev/" -i /etc/modules-load.d/cups-filters.conf
			sed "s/^parport_pc/#parport_pc/" -i /etc/modules-load.d/cups-filters.conf
740
741
		fi

742
743
744
745
746
747
748
		# enable show windows content on stronger boards
		cpu_cores=$(grep -c '^processor' /proc/cpuinfo | sed 's/^0$/1/')
		if [[ ${cpu_cores} -gt 2 && -f /etc/skel/.config/xfce4/xfconf/xfce-perchannel-xml/xfwm4.xml ]]; then
			sed -i 's/<property name="box_move" type="bool" value=".*/<property name="box_move" type="bool" value="false"\/>/g' \
			/etc/skel/.config/xfce4/xfconf/xfce-perchannel-xml/xfwm4.xml
		fi

749
750
751
752
753
754
		# fix for gksu in Xenial
		touch /home/${CHOSEN_USER}/.Xauthority
		cp -R /etc/skel/. /home/${CHOSEN_USER}

		# set up profile sync daemon on desktop systems
		which psd >/dev/null 2>&1
Igor Pečovnik's avatar
Igor Pečovnik committed
755
		if [[ $? -eq 0 && -z $(grep overlay-helper /etc/sudoers) ]]; then
756
757
758
759
			echo "${CHOSEN_USER} ALL=(ALL) NOPASSWD: /usr/bin/psd-overlay-helper" >> /etc/sudoers
			touch /home/${CHOSEN_USER}/.activate_psd
		fi

Igor Pečovnik's avatar
Igor Pečovnik committed
760
761
762
763
764
765
		mkdir -p /etc/lightdm/lightdm.conf.d
		echo "[Seat:*]" > /etc/lightdm/lightdm.conf.d/22-armbian-autologin.conf
		echo "autologin-user=$CHOSEN_USER" >> /etc/lightdm/lightdm.conf.d/22-armbian-autologin.conf
		echo "autologin-user-timeout=0" >> /etc/lightdm/lightdm.conf.d/22-armbian-autologin.conf
		echo "user-session=xfce" >> /etc/lightdm/lightdm.conf.d/22-armbian-autologin.conf
		ln -s /lib/systemd/system/lightdm.service /etc/systemd/system/display-manager.service >/dev/null 2>&1
Igor Pečovnik's avatar
bugfix    
Igor Pečovnik committed
766
767
		# fix permissions
		chown -R ${CHOSEN_USER}:${CHOSEN_USER} /home/${CHOSEN_USER}/.
Igor Pečovnik's avatar
Igor Pečovnik committed
768
		service lightdm start >/dev/null 2>&1
769
770
	fi

771
}