#!/bin/sh

# Default PATH differs between shells, and is not automatically exported
# by klibc dash.  Make it consistent.
export PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

[ -d /dev ] || mkdir -m 0755 /dev
[ -d /root ] || mkdir -m 0700 /root
[ -d /sys ] || mkdir /sys
[ -d /proc ] || mkdir /proc
[ -d /tmp ] || mkdir /tmp
mkdir -p /var/lock
mount -t sysfs -o nodev,noexec,nosuid sysfs /sys
mount -t proc -o nodev,noexec,nosuid proc /proc

# shellcheck disable=SC2013
for x in $(cat /proc/cmdline); do
	case $x in
	initramfs.clear)
		clear
		;;
	quiet)
		quiet=y
		;;
	esac
done

if [ "$quiet" != "y" ]; then
	quiet=n
	echo "Loading, please wait..."
fi
export quiet

# Note that this only becomes /dev on the real filesystem if udev's scripts
# are used; which they will be, but it's worth pointing out
mount -t devtmpfs -o nosuid,mode=0755 udev /dev

mkdir /dev/pts
mount -t devpts -o noexec,nosuid,gid=5,mode=0620 devpts /dev/pts || true

# Export the dpkg architecture
export DPKG_ARCH=
. /conf/arch.conf

# Set modprobe env
export MODPROBE_OPTIONS="-qb"

# Export relevant variables
export ROOT=
export ROOTDELAY=
export ROOTFLAGS=
export ROOTFSTYPE=
export IP=
export DEVICE=
export BOOT=
export BOOTIF=
export UBIMTD=
export break=
export init=/sbin/init
export readonly=y
export rootmnt=/root
export debug=
export panic=
export blacklist=
export btrfs_blacklist=
export resume=
export resume_offset=
export noresume=
export drop_caps=
export fastboot=n
export forcefsck=n
export fsckfix=
export IP6=
export VLAN=
export NETWORK_SKIP_ENSLAVED=
export LIVE_BOOT_CMDLINE=
export origin_root=
export origin_rootmnt=

# mdadm needs hostname to be set. This has to be done before the udev rules are called!
if [ -f "/etc/hostname" ]; then
        /bin/hostname -F /etc/hostname >/dev/null 2>&1
fi


# Bring in the main config
. /conf/initramfs.conf
for conf in conf/conf.d/*; do
	[ -f "${conf}" ] && . "${conf}"
done
. /scripts/functions

# Parse command line options
# shellcheck disable=SC2013
for x in $(cat /proc/cmdline); do
	case $x in
	# The following lines blocked for security reason
	init=*)
		init=${x#init=}
		;;
	root=*)
		ROOT=${x#root=}
		if [ -z "${BOOT}" ] && [ "$ROOT" = "/dev/nfs" ]; then
			BOOT=nfs
		fi
		;;
	rootflags=*)
		ROOTFLAGS="-o ${x#rootflags=}"
		;;
	rootfstype=*)
		ROOTFSTYPE="${x#rootfstype=}"
		;;
	rootdelay=*)
		ROOTDELAY="${x#rootdelay=}"
		case ${ROOTDELAY} in
		*[![:digit:].]*)
			ROOTDELAY=
			;;
		esac
		;;
	roottimeout=*)
		ROOTDELAY="${x#roottimeout=}"
		case ${ROOTDELAY} in
		*[![:digit:].]*)
			ROOTDELAY=
			;;
		esac
		;;
	loop=*)
		# shellcheck disable=SC2034
		LOOP="${x#loop=}"
		;;
	loopflags=*)
		# shellcheck disable=SC2034
		LOOPFLAGS="-o ${x#loopflags=}"
		;;
	loopfstype=*)
		# shellcheck disable=SC2034
		LOOPFSTYPE="${x#loopfstype=}"
		;;
	ip6=*)
		IP6="${x#ip6=}"
		;;
	vlan=*)
		VLAN="${x#vlan=}"
		;;
	nfsroot=*)
		# shellcheck disable=SC2034
		NFSROOT="${x#nfsroot=}"
		;;
	initramfs.runsize=*)
		RUNSIZE="${x#initramfs.runsize=}"
		;;
	ip=*)
		IP="${x#ip=}"
		;;
	boot=*)
		BOOT=${x#boot=}
		;;
	ubi.mtd=*)
		UBIMTD=${x#ubi.mtd=}
		;;
	resume=*)
		RESUME="${x#resume=}"
		case $RESUME in
		UUID=*)
			RESUME="/dev/disk/by-uuid/${RESUME#UUID=}"
		esac
		;;
	resume_offset=*)
		resume_offset="${x#resume_offset=}"
		;;
	noresume)
		noresume=y
		;;
	drop_capabilities=*)
		drop_caps="-d ${x#drop_capabilities=}"
		;;
	panic=*)
		panic="${x#panic=}"
		case ${panic} in
        -1) ;;
		*[![:digit:].]*)
			panic=
			;;
		esac
		;;
	ro)
		readonly=y
		;;
	rw)
		readonly=n
		;;
	debug)
		debug=y
		quiet=n
		if [ -n "${netconsole}" ]; then
			log_output=/dev/kmsg
		else
			log_output=/run/initramfs/initramfs.debug
		fi
		set -x
		;;
	debug=*)
		debug=y
		quiet=n
		set -x
		;;
	break=*)
		break=${x#break=}
		;;
	break)
		break=premount
		;;
	blacklist=*)
		blacklist=${x#blacklist=}
		;;
	netconsole=*)
		netconsole=${x#netconsole=}
		[ "x$debug" = "xy" ] && log_output=/dev/kmsg
		;;
	BOOTIF=*)
		BOOTIF=${x#BOOTIF=}
		;;
	hwaddr=*)
		BOOTIF=${x#hwaddr=}
		;;
	fastboot|fsck.mode=skip)
		fastboot=y
		;;
	forcefsck|fsck.mode=force)
		forcefsck=y
		;;
	fsckfix|fsck.repair=yes)
		fsckfix=y
		;;
	fsck.repair=no)
		fsckfix=n
		;;
	esac
done

# Default to BOOT=local if no boot script defined.
if [ -z "${BOOT}" ]; then
	BOOT=local
fi

if [ -n "${noresume}" ] || [ "$RESUME" = none ]; then
	noresume=y
else
	resume=${RESUME:-}
fi

mount -t tmpfs -o "nodev,noexec,nosuid,size=${RUNSIZE:-10%},mode=0755" tmpfs /run
mkdir -m 0700 /run/initramfs


if [ -n "$log_output" ]; then
	exec >$log_output 2>&1
	unset log_output
fi

maybe_break top

# For hisi sas dirver take 3 to 4 seconds to init, so dont init hisi sas driver here when root disk dont use sas controller.
# And at initrd phase,dont init btrfs.
root_uuid=${ROOT#UUID=}
#echo "root_uuid:${root_uuid}"
kernel_type="$(uname -r | grep desktop)"
#echo "kernel_type:${kernel_type}"
if [ -n "${root_uuid}" ];then


	btrfs_blacklist="raid6_pq"

	if [ -f "/dev/disk/by-uuid/${root_uuid}" ]; then

		if [ -n "${kernel_type}" ];then
#			echo "-->desktop"
			if [ -z "${blacklist}" ];then
				blacklist="scsi_transport_sas,libsas,hisi_sas_main,hisi_sas_v3_hw,${btrfs_blacklist}"
			else
				blacklist=${blacklist}",scsi_transport_sas,libsas,hisi_sas_main,hisi_sas_v3_hw,${btrfs_blacklist}"
			fi
		fi
	else
		if [ -n "${kernel_type}" ];then
#			echo "-->desktop sas"
			if [ -z "${blacklist}" ];then
				blacklist="${btrfs_blacklist}"
			else
				blacklist=${blacklist}",${btrfs_blacklist}"
			fi
		fi
	fi
fi

# export BOOT variable value for compcache,
# so we know if we run from casper
export BOOT

# Don't do log messages here to avoid confusing graphical boots
run_scripts /scripts/init-top

maybe_break modules
[ "$quiet" != "y" ] && log_begin_msg "Loading essential drivers"
[ -n "${netconsole}" ] && modprobe netconsole netconsole="${netconsole}"
load_modules
[ "$quiet" != "y" ] && log_end_msg

if [ "$ROOTDELAY" ]; then
	sleep $ROOTDELAY
fi

maybe_break premount
[ "$quiet" != "y" ] && log_begin_msg "Running /scripts/init-premount"
run_scripts /scripts/init-premount
[ "$quiet" != "y" ] && log_end_msg

maybe_break mount
log_begin_msg "Mounting root file system"
# Always load local and nfs (since these might be needed for /etc or
# /usr, irrespective of the boot script used to mount the rootfs).
LIVE_BOOT_CMDLINE="${LIVE_BOOT_CMDLINE:-$(cat /proc/cmdline)}"
real_boot=${BOOT}
result=$(echo "${ROOT}" | grep "UUID")
[ "${result}" != "" ] && [ "${BOOT}" = "live" ] && BOOT=local
. /scripts/local
. /scripts/nfs
. /scripts/${BOOT}
parse_numeric "${ROOT}"
maybe_break mountroot
mount_top
mount_premount
mountroot

if [ -e "${rootmnt}/live.flag" ]; then
        mount -o rw,remount ${rootmnt}
        rm -f ${rootmnt}/live.flag
        real_boot=live
        LIVE_BOOT_CMDLINE="$LIVE_BOOT_CMDLINE  boot=live  toram=filesystem.squashfs locales=zh_CN.UTF-8 live-media-path=usr/doppel"
fi

if [ -e "${rootmnt}/SystemdLocalfsDependencyFailed.flag" ]; then
        if echo "${LIVE_BOOT_CMDLINE}" | grep -q "boot=live"; then
                mount -o rw,remount ${rootmnt}
                rm ${rootmnt}/SystemdLocalfsDependencyFailed.flag
        else
                mount -o rw,remount ${rootmnt}
                echo "Loading file repair system..."
		#移除sleep 15，再panic函数中加上，以免每次都调用panic时都要加sleep 15
                rm ${rootmnt}/SystemdLocalfsDependencyFailed.flag
                panic
        fi
fi

#清理系统盘中的“/tmp/*”目录，以确保“/”目录可用空间>50M
clean_system_disk() {

    rootDisk_is_full=0
    rootDiskStandard=500
    RootIsFullFile="RootIsFull.flag"
    RootIsFullFilePath="${rootmnt}/${RootIsFullFile}"

    #判断是否存在RootIsFull.flag，若存在则继续，否则直接返回
    if [ -e "${RootIsFullFilePath}" ]; then
        rootDisk_is_full=1
    fi

    if [ $rootDisk_is_full -eq 1 ]; then
        root_disk_cleaned_space=$(df -BK "${rootmnt}" | awk 'NR==2 {print $4}')
        if [ "$root_disk_cleaned_space" -gt ${rootDiskStandard} ];then
	   rootDisk_is_full=0
	   mount -o rw,remount "${rootmnt}"
	   rm "${RootIsFullFilePath}" -rf
        else
	   echo "we must sure / availe size > 500K"
	   panic
    	fi
    fi
}

#清理数据盘中的"/data/var/cache/apt/archives"和"/data/var/log"，以确保"/data"目录可用空间>100M
clean_data_disk() {
    dataDisk_is_full=0
    log_disk_threshold=20
    data_disk_threshold=100

    DataIsFullFile="DataIsFull.flag"
    DataIsFullFilePath="${rootmnt}/${DataIsFullFile}"

    if [ -e "${DataIsFullFilePath}" ]; then
        get_system_disk_uuid=$(head -n 1 "${DataIsFullFilePath}")
        get_data_disk_uuid=$(tail -n 1 "${DataIsFullFilePath}")
        dataDisk_is_full=1
    fi

    if [ $dataDisk_is_full -eq 1 ]; then
        tmpdata="data"
        data_mount="${rootmnt}/${tmpdata}"
        data_disk_name=$(blkid -t UUID="${get_data_disk_uuid}" | awk -F ":" '{ print $1 }')
        mount -o rw "$data_disk_name" "$data_mount"

        rm "$data_mount/var/cache/apt/archives" -rf
        log_used_dir_space=$(du -sm "$data_mount/var/log" | awk '{print $1}')
        #maybe_break clean_data_archives

        if [ "$log_used_dir_space" -gt "$log_disk_threshold" ]; then
            rm "$data_mount/var/log" -rf
            #maybe_break clean_data_var_log
        fi

        data_disk_cleaned_space=$(df -BM "${data_mount}" | awk 'NR==2 {print $4}')
        if [ "$data_disk_cleaned_space" -ge "$data_disk_threshold" ]; then
            dataDisk_is_full=0
            system_disk_name=$(blkid -t UUID="${get_system_disk_uuid}" | awk -F ":" '{print $1}')
            mount -o rw,remount "$system_disk_name" "$rootmnt"
            rm  "${DataIsFullFilePath}" -rf
            #maybe_break varlogainif
        else
            echo "we must sure /data available size > 100M"
            panic
       fi
    fi
}

if ! echo "${LIVE_BOOT_CMDLINE}" | grep -q "boot=live"; then
	clean_system_disk
	clean_data_disk
fi
if [ "${real_boot}" = "live" ] && [ "${BOOT}" = "local" ]; then
        origin_root=${ROOT}
        umount ${rootmnt}
        LIVE_BOOT_CMDLINE="$LIVE_BOOT_CMDLINE bootfrom=${ROOT}"
        BOOT=live
        . /scripts/local
        . /scripts/nfs
        . /scripts/${BOOT}
        parse_numeric "${ROOT}"
        maybe_break mountroot
        mount_top
        mount_premount
        mountroot

        origin_rootmnt="/tmp"
        mount -r "${origin_root}" "${origin_rootmnt}"
        kernel_version="$(uname -r)"
        src_dir="${origin_rootmnt}/lib/modules/${kernel_version}/"
        dst_dir="${rootmnt}/lib/modules/"
        rm -rf ${dst_dir}*
        mkdir -p "${dst_dir}${kernel_version}"
        substance_list=$(ls "${src_dir}")
        for substance in ${substance_list}; do
            cp -rf "${src_dir}${substance}" "${dst_dir}${kernel_version}"
        done
        cp -rf "${origin_rootmnt}/etc/modules-load.d"     "${rootmnt}/etc/"
        cp -rf "${origin_rootmnt}/usr/lib/modules-load.d"  "${rootmnt}/usr/lib/"
        cp -a "${origin_rootmnt}/usr/share/fonts" "${rootmnt}/usr/share/"
        cp -a "${origin_rootmnt}/etc/crypttab" "${rootmnt}/etc/crypttab"
        if [ "$(grep -w LOCAL ${origin_rootmnt}/etc/adjtime)" = "LOCAL" ]; then
            cp -a "${origin_rootmnt}/etc/adjtime" "${rootmnt}/etc/adjtime"
            rm -rf "${rootmnt}/etc/localtime"
            cp -a "${origin_rootmnt}/etc/localtime" "${rootmnt}/etc/localtime"
        fi
        if [ -e ${origin_rootmnt}/etc/nvidia/current/nvidia-blacklists-nouveau.conf ]; then
            rm -rf "${dst_dir}${kernel_version}"/updates/dkms/nvidia*.ko
            echo "blacklist nvidiafb" > ${rootmnt}/etc/modprobe.d/nvidia-blacklists-nouveau.conf
        fi

        if [ -e ${origin_rootmnt}/usr/share/X11/xorg.conf.d/10-innogpu.conf ]; then
            cp -rf "${origin_rootmnt}/usr/share/X11/xorg.conf.d/10-innogpu.conf"  "${rootmnt}/usr/share/X11/xorg.conf.d/"
        fi
        umount ${origin_rootmnt}
fi
log_end_msg

if read_fstab_entry /usr; then
	log_begin_msg "Mounting /usr file system"
	mountfs /usr
	log_end_msg
fi

uos_config_init() {
	#Only specified parameters are allowed
	checkflag=y
	#Allow deepinisocheck.sh
	for inittest in /usr/bin/deepinisocheck.sh
	do
		if [ "$init" = "$inittest" ];then
			checkflag=n
			break
		fi
	done

	# Whether to enable grub menu encryption.
	# Mount /boot temporary, in order to read /boot/grub/grub.cfg.
	if read_fstab_entry /boot; then
		log_begin_msg "Mounting /boot file system"
		mountfs /boot
		log_end_msg
	fi
	#if init is not in the above list and grub menu encryption is disabled, user-defined parameter is invalid.
	if [ "$checkflag" = "y" ] && [ "$(grep -c ^password_pbkdf2 ${rootmnt}/boot/grub/grub.cfg)" -eq "0" ];then
		#no enable, default parameter is used.
		init=/sbin/init
	fi
	if read_fstab_entry /boot; then
		umount ${rootmnt}/boot
	fi
}

uos_config_init

# Mount cleanup
mount_bottom
nfs_bottom
local_bottom

maybe_break bottom
[ "$quiet" != "y" ] && log_begin_msg "Running /scripts/init-bottom"
# We expect udev's init-bottom script to move /dev to ${rootmnt}/dev
run_scripts /scripts/init-bottom
[ "$quiet" != "y" ] && log_end_msg

# Move /run to the root
mount -n -o move /run ${rootmnt}/run

validate_init() {
	run-init -n "${rootmnt}" "${1}"
}


# Check init is really there
if ! validate_init "$init"; then
	echo "Target filesystem doesn't have requested ${init}."
	init=
	for inittest in /sbin/init /etc/init /bin/init /bin/sh; do
		if validate_init "${inittest}"; then
			init="$inittest"
			break
		fi
	done
fi

# No init on rootmount
if ! validate_init "${init}" ; then
	panic "No init found. Try passing init= bootarg."
fi

maybe_break init

# don't leak too much of env - some init(8) don't clear it
# (keep init, rootmnt, drop_caps)
unset debug
unset MODPROBE_OPTIONS
unset DPKG_ARCH
unset ROOTFLAGS
unset ROOTFSTYPE
unset ROOTDELAY
unset ROOT
unset IP
unset BOOT
unset BOOTIF
unset DEVICE
unset UBIMTD
unset blacklist
unset break
unset noresume
unset panic
unset quiet
unset readonly
unset resume
unset resume_offset
unset noresume
unset fastboot
unset forcefsck
unset fsckfix
unset IP6
unset VLAN

# Move virtual filesystems over to the real filesystem
mount -n -o move /sys ${rootmnt}/sys
mount -n -o move /proc ${rootmnt}/proc

# Chain to real filesystem
# shellcheck disable=SC2086,SC2094
exec run-init ${drop_caps} "${rootmnt}" "${init}" "$@" <"${rootmnt}/dev/console" >"${rootmnt}/dev/console" 2>&1
echo "Something went badly wrong in the initramfs."
panic "Please file a bug on initramfs-tools."
