#!/bin/bash
# --- T2-COPYRIGHT-NOTE-BEGIN ---
# This copyright note is auto-generated by ./scripts/Create-CopyPatch.
# 
# T2 SDE: package/.../mkinitrd/mkinitrd.sh
# Copyright (C) 2005 - 2012 The T2 SDE Project
# 
# More information can be found in the files COPYING and README.
# 
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 2 of the License. A copy of the
# GNU General Public License can be found in the file COPYING.
# --- T2-COPYRIGHT-NOTE-END ---

set -e

if [ $UID != 0 ]; then
	echo "Non root - exiting ..."
	exit 1
fi

while [ "$1" ]; do
  case $1 in
	[0-9]*) kernelver="$1" ;;
	-R) root="$2" ; shift ;;
	*) echo "Usage: mkinitrd [ -R root ] [ kernelver ]"
	   exit 1 ;;
  esac
  shift
done

[ "$root" ] || root=""
[ "$kernelver" ] || kernelver=`uname -r`
[ "$moddir" ] || moddir="${root}/lib/modules/$kernelver"

echo "Kernel: $kernelver, module dir: $moddir"

if [ ! -d $moddir ]; then
	echo "Module dir $moddir does not exist!"
	exit 2
fi

sysmap=""
[ -f "${root}/boot/System.map_$kernelver" ] && sysmap="${root}/boot/System.map_$kernelver"

if [ -z "$sysmap" ]; then
	echo "System.map_$kernelver not found!"
	exit 2
fi

echo "System.map: $sysmap"

# check needed tools
for x in cpio gzip ; do
	if ! type -p $x >/dev/null ; then
		echo "$x not found!"
		exit 2
	fi
done

tmpdir=`mktemp`

# create basic structure
#
rm -rf $tmpdir >/dev/null

echo "Create dirtree ..."

mkdir -p $tmpdir/{dev,bin,sbin,proc,sys,lib/modules,lib/udev,etc/hotplug.d/default}
mknod $tmpdir/dev/console c 5 1

# copy the basic / rootfs kernel modules
#
echo "Copying kernel modules ..."

(
  find $moddir/kernel -type f |
  grep -v -e /wireless/ -e netfilter |
  grep  -e reiserfs -e reiser4 -e ext2 -e ext3 -e ext4 -e btrfs -e /jfs -e /xfs \
	-e isofs -e udf -e /unionfs -e ntfs -e fat -e dm-mod \
	-e /ide/ -e /ata/ -e /scsi/ -e /message/ \
	-e cciss -e ips -e virtio \
	-e hci -e usb-common -e usb-storage -e sbp2 \
	-e /net/ -e drivers/md/ -e '/ipv6\.' -e usbhid -e hid-apple |
  while read fn ; do

	for x in $fn `modinfo $fn | grep depends |
	         cut -d : -f 2- | sed -e 's/ //g' -e 's/,/ /g' `
	do
		# expand to full name if it was a depend
		[ $x = ${x##*/} ] &&
		x=`find $moddir/kernel -name "$x.*o"`

		echo -n "${x##*/} "

		# strip $root prefix
		xt=${x##$root}

		mkdir -p `dirname $tmpdir/$xt`
		cp $x $tmpdir/$xt 2>/dev/null
	done
  done
) | fold -s ; echo

# generate map files
#
depmod -ae -b $tmpdir -F $sysmap $kernelver

echo "Injecting programs and configuration ..."

# copying config
#
cp -ar ${root}/etc/udev $tmpdir/etc/
[ -e ${root}/lib/udev/rules.d ] && cp -ar ${root}/lib/udev/rules.d $tmpdir/lib/udev
[ -e ${root}/etc/mdadm.conf ] && cp -ar ${root}/etc/mdadm.conf $tmpdir/etc/
cp -ar ${root}/etc/modprobe.* $tmpdir/etc/ 2>/dev/null || true
# in theory all, but fat and currently only cdrom_id is needed ...
#cp -ar ${root}/lib/udev/cdrom_id $tmpdir/lib/udev/

# copy dynamic libraries, if any.
#
copy_dyn_libs () {
	# we can not use ldd(1) as it loads the object, which does not work on cross builds
	for lib in `readelf -a $1 | sed -n 's/.*Shared library.*\[\([^]\]*\)\]/\1/p'`; do
		[[ $1 = *bin/* ]] && echo "Warning: $1 is dynamically linked!"
		for libdir in $root/lib*/ $root/usr/lib*/; do
			if [ -e $libdir$lib ]; then
				xlibdir=${libdir#$root}
				echo "$1 NEEDS $xlibdir$lib"

				mkdir -p $tmpdir$xlibdir

				while local x=`readlink $libdir$lib`; [ "$x" ]; do
					echo " $libdir$lib SYMLINKS to $x"
					ln -sfv $x $tmpdir$xlibdir$lib
					lib=$x
				done
				cp -fv $libdir$lib $tmpdir$xlibdir$lib

				copy_dyn_libs $libdir$lib
			fi
		done
	done
}

# setup programs
#
for x in ${root}/sbin/{udevd,udevadm,modprobe,insmod} \
         ${root}/usr/sbin/disktype
do
	cp $x $tmpdir/sbin/
	copy_dyn_libs $x
done

# setup optional programs
#
for x in ${root}/sbin/{hotplug++,insmod.old,mdadm}
do
  if [ ! -e $x ]; then
	echo "Warning: Skipped optional file $x!"
  else
	cp $x $tmpdir/sbin/
	ln -sf insmod.old $tmpdir/sbin/modprobe.old
	copy_dyn_libs $x
  fi
done

ln -s /sbin/udev $tmpdir/etc/hotplug.d/default/10-udev.hotplug
cp ${root}/bin/pdksh $tmpdir/bin/sh

# static, tiny embutils and friends
#
cp ${root}/usr/embutils/{mount,umount,rm,mv,mkdir,ln,ls,switch_root,chroot,sleep,losetup,chmod,cat,sed,mknod,dmesg} \
   $tmpdir/bin/
ln -s mv $tmpdir/bin/cp

cp ${root}/sbin/initrdinit $tmpdir/init

# Custom ACPI DSDT table
if test -f "${root}/boot/DSDT.aml"; then
	echo "Adding local DSDT file: $dsdt"
	cp ${root}/boot/DSDT.aml $tmpdir/DSDT.aml
fi

# create the cpio image
#
echo "Archiving ..."
( cd $tmpdir
  find . | cpio -o -H newc | gzip -c9 > ${root}/boot/initrd-$kernelver.img
)

# display the resulting image
#
du -sh ${root}/boot/initrd-$kernelver.img
rm -rf $tmpdir