#!/bin/sh
#
# YAFFS: Yet another FFS. A NAND-flash specific file system.
#
# Copyright (C) 2002-2010 Aleph One Ltd.
#
# Created by Charles Manning <charles@aleph1.co.uk>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
#
# Patch YAFFS into the kernel
#
#  args:  l/c : link or copy
#	  kpath  : Full path to kernel sources to be patched
#
#  Somewhat "inspired by" the mtd patchin script
#

VERSION=0
PATCHLEVEL=0
SUBLEVEL=0
COPYORLINK=$1
MULTIORSINGLE=$2
LINUXDIR=$3

# To be a Linux directory, it must have a Makefile


# Display usage of this script
usage () {
	echo "usage:  $0  c/l m/s kernelpath"
	echo " if c/l is c, then copy. If l then link"
	echo " if m/s is m, then use multi version code. If s then use single version code"
	exit 1
}



if [ -z $LINUXDIR ]
then
    usage;
fi

if [ $COPYORLINK = l ]; then
   CPY="ln -s"
elif [ $COPYORLINK = c ]; then
   CPY="cp"
else
   echo "unknown copy or link type"
   usage;
fi

if [ $MULTIORSINGLE = m ]; then
   VFS_CODE="yaffs_vfs_multi.c"
   MTD1_CODE="yaffs_mtdif1_multi.c"
   MTD2_CODE="yaffs_mtdif2_multi.c"
   YPORTENV="yportenv_multi.h"
   KCONFIG_SRC="Kconfig_multi"
elif [ $MULTIORSINGLE = s ]; then
   VFS_CODE="yaffs_vfs_single.c"
   MTD1_CODE="yaffs_mtdif1_single.c"
   MTD2_CODE="yaffs_mtdif2_single.c"
   YPORTENV="yportenv_single.h"
   KCONFIG_SRC="Kconfig_single"

   echo ""
   echo "*** Warning ***"
   echo "You have chosen to use the single kernel variant of the yaffs VFS glue code"
   echo "that only works with the latest Linux kernel tree. If you are using an older"
   echo "version of Linux then you probably wanted to use the multi-version variant by"
   echo "re-running the patch-ker.sh script using m as a the second argument."
   echo " ie $0 $COPYORLINK m $LINUXDIR"
   echo ""
else
   echo "unknown multi/single version selection"
   usage;
fi


# Check if kerneldir contains a Makefile
if [ ! -f $LINUXDIR/Makefile ]
then
	echo "Directory $LINUXDIR does not exist or is not a kernel source directory";
	exit 1;
fi

# Get kernel version
VERSION=`grep -s VERSION <$LINUXDIR/Makefile | head -n 1 | sed s/'VERSION = '//`
PATCHLEVEL=`grep -s PATCHLEVEL <$LINUXDIR/Makefile | head -n 1 | sed s/'PATCHLEVEL = '//`
SUBLEVEL=`grep -s SUBLEVEL <$LINUXDIR/Makefile | head -n 1 | sed s/'SUBLEVEL = '//`

# Can we handle this version?
if [ $VERSION -ne 2  -o $PATCHLEVEL -lt 6  ]
then
	echo "Cannot patch kernel version $VERSION.$PATCHLEVEL.$SUBLEVEL, must be 2.6.x or higher"
	exit 1;
fi


KCONFIG=$LINUXDIR/fs/Kconfig
KCONFIGOLD=$LINUXDIR/fs/Kconfig.pre.yaffs
YAFFS_PATCHED_STRING=`grep -s yaffs <$KCONFIG | head -n 1`

MAKEFILE=$LINUXDIR/fs/Makefile
MAKEFILEOLD=$LINUXDIR/fs/Makefile.pre.yaffs

if [ ! -z "$YAFFS_PATCHED_STRING" ]
then
    YAFFS_PATCHED=0
    echo "$KCONFIG already mentions YAFFS, so we will not change it"
else
   # Change the fs/Kconfig file
   # Save the old Kconfig
   # Copy all stuff up to JFFS
   # Insert some YAFFS stuff
   # Copy all the rest of the stuff

    YAFFS_PATCHED=1
    echo "Updating $KCONFIG"
    mv -f $KCONFIG  $KCONFIGOLD
    sed -n -e "/[Jj][Ff][Ff][Ss]/,99999 ! p" $KCONFIGOLD >$KCONFIG
    # echo "">>$KCONFIG
    # echo "# Patched by YAFFS" >>$KCONFIG
    echo "source \"fs/yaffs2/Kconfig\"">>$KCONFIG
    # echo "">>$KCONFIG
    sed -n -e "/[Jj][Ff][Ff][Ss]/,99999 p" $KCONFIGOLD >>$KCONFIG

   # now do fs/Makefile -- simply add the target at the end
    echo "Updating $MAKEFILE"
    cp -f $MAKEFILE $MAKEFILEOLD
    # echo "">>$MAKEFILE
    # echo "# Patched by YAFFS" >>$MAKEFILE
    echo "obj-\$(CONFIG_YAFFS_FS)		+= yaffs2/" >>$MAKEFILE

fi

YAFFSDIR=$LINUXDIR/fs/yaffs2

if [ -e $YAFFSDIR ]
then
   echo "$YAFFSDIR exists, so not patching. If you want to replace what is"
   echo "already there then delete $YAFFSDIR and re-run this script"
   echo " eg.  \"rm -rf $YAFFSDIR\" "
else
   rm yaffs*.mod.c 2> /dev/null
   mkdir $LINUXDIR/fs/yaffs2
   $CPY  $PWD/Makefile.kernel $LINUXDIR/fs/yaffs2/Makefile
   $CPY $PWD/$KCONFIG_SRC $LINUXDIR/fs/yaffs2/Kconfig
   $CPY $PWD/*.c $PWD/*.h  $LINUXDIR/fs/yaffs2
   rm $LINUXDIR/fs/yaffs2/yaffs_vfs*.c $LINUXDIR/fs/yaffs2/yaffs_mtdif[12]*.c
   rm $LINUXDIR/fs/yaffs2/yportenv*.h
   rm $LINUXDIR/fs/yaffs2/moduleconfig.h
   $CPY $PWD/$VFS_CODE $LINUXDIR/fs/yaffs2/yaffs_vfs.c
   $CPY $PWD/$MTD1_CODE $LINUXDIR/fs/yaffs2/yaffs_mtdif1.c
   $CPY $PWD/$MTD2_CODE $LINUXDIR/fs/yaffs2/yaffs_mtdif2.c
   $CPY $PWD/$YPORTENV $LINUXDIR/fs/yaffs2/yportenv.h
fi