#!/bin/sh

# Copyright (C) 2015, 2017 Red Hat Inc.
# Author(s): Dennis Gilmore <dennis@ausil.us>
#            Merlin Mathesius <mmathesi@redhat.com>
#
# 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; either version 2 of the License, or (at your
# option) any later version.  See http://www.gnu.org/copyleft/gpl.html for
# the full text of the license.

# is the system a calxeda one
is_calxeda() {
    grep -Eq '^Hardware[^ ]* (Highbank|Midway)' /proc/cpuinfo
}

# update /etc/sysconfig/uboot to reflect if the platform provides its own dtb and
# we do not need to pass anything.
update_sysconfig() {
    sed -i -e 's/#SHIPSDTB=no/SHIPSDTB=yes/g' /etc/sysconfig/uboot
}

# insert into /boot/extlinux/extlinux.conf a fdtdir line
insert_fdtdir() {
    # for every 'kernel' directive found, map it to an 'fdtdir' directive
    # referencing the dtb directory and add it after the 'append' directive
    sed -i \
        -e '/	fdtdir/d' \
        -e '/	kernel/{h;s/kernel/fdtdir/g;s/vmlinuz/dtb/g;x}' \
        -e '/	append/{p;x}' \
        /boot/extlinux/extlinux.conf
}

# check if there is a rescue image and copy the dtb files if there is
copy_rescue_dtbs() {
    has_rescue=0
    for target in /boot/vmlinuz-0-rescue* ; do
        # if target exists, the glob matched a file
        if [ -e "$target" ] ; then
            has_rescue=1
            dtbdest=$(echo "$target" | sed -e 's/vmlinuz/dtb/g')
        fi
	break
    done
    for target in /boot/dtb* ; do
        # if target exists, the glob matched a file
        if [ -e "$target" ] ; then
            dtbsource="$target"
        fi
	break
    done
    if [ $has_rescue = 1 ] && [ ! -d "$dtbdest" ] ; then
       mkdir "$dtbdest"
       for dtb in $(ls "$dtbsource") ; do
           cp "$dtbsource/$dtb" "$dtbdest/$dtb"
       done
    fi
}

# platform ships its own dtb
if is_calxeda
then
    update_sysconfig
else
    # need to add fdtdir to extlinux.conf
    insert_fdtdir
    copy_rescue_dtbs
fi