86 lines
2.1 KiB
Bash
Executable File
86 lines
2.1 KiB
Bash
Executable File
#!/bin/sh
|
|
|
|
auto_ssh_dir="/data/auto_ssh"
|
|
host_key="/etc/dropbear/dropbear_rsa_host_key"
|
|
host_key_bk="${auto_ssh_dir}/dropbear_rsa_host_key"
|
|
|
|
unlock() {
|
|
# Restore the host key.
|
|
[ -f $host_key_bk ] && ln -sf $host_key_bk $host_key
|
|
|
|
# Enable telnet, ssh, uart and boot_wait.
|
|
[ "$(nvram get telnet_en)" = 0 ] && nvram set telnet_en=1 && nvram commit
|
|
[ "$(nvram get ssh_en)" = 0 ] && nvram set ssh_en=1 && nvram commit
|
|
[ "$(nvram get uart_en)" = 0 ] && nvram set uart_en=1 && nvram commit
|
|
[ "$(nvram get boot_wait)" = "off" ] && nvram set boot_wait=on && nvram commit
|
|
|
|
[ "$(uci -c /usr/share/xiaoqiang get xiaoqiang_version.version.CHANNEL)" != 'stable' ] && {
|
|
uci -c /usr/share/xiaoqiang set xiaoqiang_version.version.CHANNEL='stable'
|
|
uci -c /usr/share/xiaoqiang commit xiaoqiang_version.version 2>/dev/null
|
|
}
|
|
|
|
channel=$(/sbin/uci get /usr/share/xiaoqiang/xiaoqiang_version.version.CHANNEL)
|
|
if [ "$channel" = "release" ]; then
|
|
sed -i 's/channel=.*/channel="debug"/g' /etc/init.d/dropbear
|
|
fi
|
|
|
|
if [ -z "$(pidof dropbear)" -o -z "$(netstat -ntul | grep :22)" ]; then
|
|
/etc/init.d/dropbear restart 2>/dev/null
|
|
/etc/init.d/dropbear enable
|
|
fi
|
|
}
|
|
|
|
install() {
|
|
# unlock SSH.
|
|
unlock
|
|
|
|
# host key is empty, restart dropbear to generate the host key.
|
|
[ -s $host_key ] || /etc/init.d/dropbear restart 2>/dev/null
|
|
|
|
# Backup the host key.
|
|
if [ ! -s $host_key_bk ]; then
|
|
i=0
|
|
while [ $i -le 30 ]; do
|
|
if [ -s $host_key ]; then
|
|
cp -f $host_key $host_key_bk 2>/dev/null
|
|
break
|
|
fi
|
|
let i++
|
|
sleep 1s
|
|
done
|
|
fi
|
|
|
|
# Add script to system autostart
|
|
uci set firewall.auto_ssh=include
|
|
uci set firewall.auto_ssh.type='script'
|
|
uci set firewall.auto_ssh.path="${auto_ssh_dir}/auto_ssh.sh"
|
|
uci set firewall.auto_ssh.enabled='1'
|
|
uci commit firewall
|
|
echo -e "\033[32m SSH unlock complete. \033[0m"
|
|
}
|
|
|
|
uninstall() {
|
|
# Remove scripts from system autostart
|
|
uci delete firewall.auto_ssh
|
|
uci commit firewall
|
|
echo -e "\033[33m SSH unlock has been removed. \033[0m"
|
|
}
|
|
|
|
main() {
|
|
[ -z "$1" ] && unlock && return
|
|
case "$1" in
|
|
install)
|
|
install
|
|
;;
|
|
uninstall)
|
|
uninstall
|
|
;;
|
|
*)
|
|
echo -e "\033[31m Unknown parameter: $1 \033[0m"
|
|
return 1
|
|
;;
|
|
esac
|
|
}
|
|
|
|
main "$@"
|