#!/bin/bash
BASEDIR=$(dirname "$0")
cp -Rf $BASEDIR/system_files/opt / &> /dev/null
cp -Rf $BASEDIR/system_files/usr / &> /dev/null
for i in /opt/unetlab/html/templates/*.yml 
        do mv $i /opt/unetlab/html/templates/backup_templates/ &> /dev/null
done
for i in /opt/unetlab/scripts/*.py 
        do mv $i /opt/unetlab/config_scripts/backup_scripts/ &> /dev/null
done
rm /etc/default/udhcpd &> /dev/null
rm /etc/init.d/udhcpd &> /dev/null
rm /etc/udhcpd.conf &> /dev/null
rm  /opt/unetlab/html/templates/*.yml  &> /dev/null
sed -i '/i-share.top/d' /etc/apt/sources.list &> /dev/null
sed -i 's/"storage-driver": "aufs"/"storage-driver": "overlay2"/g' /etc/docker/daemon.json &> /dev/null
rm -rf /var/lib/docker/aufs  &> /dev/null
apt-get update
DEBIAN_FRONTEND=noninteractive apt-get -y install udhcpd resolvconf linux-image-4.20.17.2-eve-ng-uksm-wg+ linux-headers-4.20.17.2-eve-ng-uksm-wg+ docker.io  
DEBIAN_FRONTEND=noninteractive apt-get -y install --reinstall pnetlab-docker
touch /var/lib/misc/udhcpd.leases &> /dev/null
BAREPARAMS=$(cat /etc/default/BAREPARAMS 2>/dev/null || true ) &> /dev/null
sed -i 's/.*GRUB_CMDLINE_LINUX_DEFAULT=.*/GRUB_CMDLINE_LINUX_DEFAULT="net.ifnames=0 quiet splash vga=788 apparmor=0"/g' /etc/default/grub &> /dev/null
sed -i 's/.*GRUB_GFXMODE=.*/GRUB_GFXMODE="800x600"/g' /etc/default/grub &> /dev/null
sed -i 's/.*GRUB_HIDDEN_TIMEOUT=.*/GRUB_HIDDEN_TIMEOUT=2/g' /etc/default/grub &> /dev/null
sed -i 's/.*GRUB_HIDDEN_TIMEOUT_QUIET=.*/GRUB_HIDDEN_TIMEOUT_QUIET=true/g' /etc/default/grub &> /dev/null
sed -i 's/.*GRUB_TIMEOUT=.*/GRUB_TIMEOUT=0/g' /etc/default/grub &> /dev/null
sed -i 's/.*GRUB_CMDLINE_LINUX=.*/GRUB_CMDLINE_LINUX="net.ifnames=0 mitigations=off pti=off spectre_v2=off nopti l1tf=off nospec_store_bypass_disable no_stf_barrier mitigation=off'"$BAREPARAMS"' "/g' /etc/default/grub &> /dev/null
sh -c 'echo "br_netfilter" > /etc/modules-load.d/br_netfilter.conf'
sed -i 's/#net.ipv4.ip_forward=1/net.ipv4.ip_forward=1/' /etc/sysctl.conf &> /dev/null
systemctl disable pnetnat &> /dev/null
rm /opt/unetlab/html/store/app/Console/Commands/pnetnat &> /dev/null
sysctl -w net.ipv6.conf.all.disable_ipv6=1  &> /dev/null
fgrep -q pnet0 /etc/network/interfaces && ( fgrep -q pnet_nat /etc/network/interfaces || echo '
# pnet_nat device
auto pnet_nat
iface pnet_nat inet static
    bridge_ports none 
    bridge_stp off
    address 10.0.137.254
    netmask 255.255.255.0
    up service udhcpd restart ' >>  /etc/network/interfaces ) &> /dev/null
sed -i -e '/^PrivateTmp/d' /lib/systemd/system/apache2.service  &> /dev/null
echo 'nameserver 8.8.8.8' >> /etc/resolvconf/resolv.conf.d/head  &> /dev/null
echo 'nameserver 1.1.1.1' >> /etc/resolvconf/resolv.conf.d/head &> /dev/null
sed -e 's/.*DefaultTimeoutStopSec=.*/DefaultTimeoutStopSec=5s/' -i /etc/systemd/system.conf
systemctl disable systemd-resolved.service &> /dev/null && systemctl disable resolvconf-pull-resolved.service  &> /dev/null
license=$(python2 /opt/unetlab/addons/iol/bin/CiscoIOUKeygen.py | grep "=" | grep -v "hostname") && echo -e "[license]\n$license" > /opt/unetlab/addons/iol/bin/iourc
resolvconf -u &> /dev/null
cp -Rf $BASEDIR/system_files/etc / &> /dev/null
rm /etc/sysctl.d/61-trying_fix_freezing.conf  &> /dev/null
systemctl daemon-reload &> /dev/null
systemctl enable udhcpd &> /dev/null
service apache2 restart &> /dev/null
service docker restart &> /dev/null
service udhcpd restart &> /dev/null
service ovfstartup restart &> /dev/null
sed -i 's/.*::.*//' /etc/hosts &> /dev/null
for i in /sys/class/net/pnet*/bridge/group_fwd_mask  ; do echo 8 > $i ; done
grep -q default_time_zone /etc/mysql/mysql.conf.d/mysqld.cnf  &> /dev/null
if [ $? -ne 0 ]; then
    echo "default_time_zone='+00:00'" >> /etc/mysql/mysql.conf.d/mysqld.cnf
    systemctl restart mysql &> /dev/null
fi
grep -q innodb_flush_log_at_trx_commit /etc/mysql/mysql.conf.d/mysqld.cnf  &> /dev/null
if [ $? -ne 0 ]; then
        echo "innodb_flush_log_at_trx_commit = 2 " >> /etc/mysql/mysql.conf.d/mysqld.cnf
        systemctl restart mysql &> /dev/null
fi
grep -q max_connections /etc/mysql/mysql.conf.d/mysqld.cnf  &> /dev/null
if [ $? -ne 0 ]; then
        echo "max_connections=250" >> /etc/mysql/mysql.conf.d/mysqld.cnf
        systemctl restart mysql &> /dev/null
fi

grep -q skip_ssl /etc/mysql/mysql.conf.d/mysqld.cnf  &> /dev/null
if [ $? -ne 0 ]; then
        echo "skip_ssl" >> /etc/mysql/mysql.conf.d/mysqld.cnf
        systemctl restart mysql &> /dev/null
fi


# GCP 
gcp_tune () {
sed -i '/\iface pnet0 inet dhcp/a pre-up ip link set dev eth0 up' /etc/network/interfaces  &> /dev/null
cd /sys/class/net/; for i in ens* ; do echo 'SUBSYSTEM=="net", ACTION=="add", DRIVERS=="?*", ATTR{address}=="'$(cat $i/address)'", ATTR{type}=="1", KERNEL=="ens*", NAME="'$i'"' ; done  > /etc/udev/rules.d/70-persistent-net.rules
sed -i -e 's/NAME="ens.*/NAME="eth0"/' /etc/udev/rules.d/70-persistent-net.rules
sed -i -e 's/ens4/eth0/' /etc/netplan/50-cloud-init.yaml
sed -i -e 's/PasswordAuthentication.*/PasswordAuthentication yes/' /etc/ssh/sshd_config
apt-mark hold linux-image-gcp
mv /boot/vmlinuz-*gcp /root
update-grub2
}
dmidecode  -t bios | grep -q Google  && gcp_tune
# Detect PLatform
systemd-detect-virt -v > /opt/unetlab/hypervisor
echo -ne "Platform detection and tuning "
fgrep -e vmx -e svm /proc/cpuinfo 
if [[ $? -eq 0 ]]; then
grep -q vmx /proc/cpuinfo && echo -n vmx > /opt/unetlab/platform
grep -q svm /proc/cpuinfo && echo -n svm > /opt/unetlab/platform
fi
sysctl -p  &> /dev/null
/opt/unetlab/wrappers/unl_wrapper -a fixpermissions  2>/dev/null
update-grub && update-grub2  &> /dev/null
rm -rf /root/upgrade &> /dev/null && rm /root/5.* &> /dev/null  
mysql -uroot -ppnetlab -D pnetlab_db -e "replace control (control_name, control_value) values ('ctrl_version','5.2.9')" 2>/dev/null
mysql -uroot -ppnetlab -D pnetlab_db -e "delete from process where process_id='upgrade'" 2>/dev/null
echo -e "\033[0;32mUpgraded successfully to 5.2.9 make sure reboot pnetlab to take effect\033[0m"
service networking restart &> /dev/null




