#!/bin/sh

PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin
DAEMON=/usr/bin/cgminer
NAME=cgminer
DESC="Cgminer daemon"

set -e

test -x "$DAEMON" || exit 0

do_start() {
        # Stop SPI poller
        spi_ena=0
        i2cset -y 2 0x71 2 $spi_ena

        good_ports=""
        bad_ports=""

        # CLear faults in megadlynx's
        for b in 3 4 5 6 7 8 ; do
                for d in 0 1 2 3 4 5 6 7 ; do
                        i2cset -y $b 0x1$d 3 >/dev/null 2>&1 || true
                done
        done

        for p in 0 1 2 3 4 5 ; do
                i2cset -y 2 0x71 1 $((p+1))
                good_flag=0
                ar="$(spi-test -s 50000 -OHC -D /dev/spidev1.0 0x80,3,0,0,0,0,0,
                if [ "x$ar" = "x00 30 A0 01" ] ; then
                        good_flag=1
                fi
                ar="$(spi-test -s 50000 -OHC -D /dev/spidev1.0 0x80,2,0,0,0,0,0,
                if [ "x$ar" = "x00 30 A0 01" ] ; then
                        good_flag=1
                fi
                ar="$(spi-test -s 50000 -OHC -D /dev/spidev1.0 0x80,1,0,0,0,0,0,
                if [ "x$ar" = "x00 30 A0 01" ] ; then
                        good_flag=1
                fi
                ar="$(spi-test -s 50000 -OHC -D /dev/spidev1.0 0x80,0,0,0,0,0,0,
                if [ "x$ar" = "x00 30 A0 01" ] ; then
                        good_flag=1
                fi

                if [ "$good_flag" = "1" ] ; then
                        good_ports=$good_ports" $p"
                else
                        bad_ports=$bad_ports" $p"
                fi
        done

        if [ -n "$good_ports" ] ; then
                for p in $good_ports ; do
                        # Re-enable PLL
                        i2cset -y 2 0x71 1 $((p+1))
                        for c in 0 1 2 3 ; do
                                cmd=$(printf "0x84,0x%02X,0,0" $c)
                                spi-test -s 50000 -OHC -D /dev/spidev1.0 $cmd >/
                                cmd=$(printf "0x86,0x%02X,0x02,0x71" $c)
if [ $p -eq 0 ]
then
       if [ $c -eq 0 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x71" $c)
       fi
       if [ $c -eq 1 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x71" $c)
       fi
       if [ $c -eq 2 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x71" $c)
       fi
       if [ $c -eq 3 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x71" $c)
       fi
fi

if [ $p -eq 1 ]
then
       if [ $c -eq 0 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x41" $c)
       fi
       if [ $c -eq 1 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x41" $c)
       fi
       if [ $c -eq 2 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x41" $c)
       fi
       if [ $c -eq 3 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x41" $c)
       fi
fi

if [ $p -eq 2 ]
then
       if [ $c -eq 0 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x71" $c)
       fi
       if [ $c -eq 1 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x71" $c)
       fi
       if [ $c -eq 2 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x71" $c)
       fi
       if [ $c -eq 3 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x71" $c)
       fi
fi

if [ $p -eq 3 ]
then
       if [ $c -eq 0 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x41" $c)
       fi
       if [ $c -eq 1 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x41" $c)
       fi
       if [ $c -eq 2 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x41" $c)
       fi
       if [ $c -eq 3 ]
       then
               cmd=$(printf "0x86,0x%02X,0x02,0x41" $c)
       fi
fi
                                spi-test -s 50000 -OHC -D /dev/spidev1.0 $cmd >/
                                cmd=$(printf "0x85,0x%02X,0,0" $c)
                                spi-test -s 50000 -OHC -D /dev/spidev1.0 $cmd >/
                        done
                        # re-enable all cores
                        i=0
                        while [[ $i -lt 192 ]] ; do
                                i2cset -y 2 0x2$p $i 1
                                i=$((i+1))
                        done
                        spi_ena=$(( spi_ena | (1 << $p) ))
                done
        fi
        if [ -n "$bad_ports" ] ; then
                for p in $bad_ports ; do
                        # Disable PLL
                        i2cset -y 2 0x71 1 $((p+1))
                        for c in 0 1 2 3 ; do
                                cmd=$(printf "0x84,0x%02X,0,0" $c)
                                spi-test -s 50000 -OHC -D /dev/spidev1.0 $cmd >/
                        done

                        # disable all cores
                        i=0
                        while [[ $i -lt 192 ]] ; do
                                i2cset -y 2 0x2$p $i 0
                                i=$((i+1))
                        done
                        spi_ena=$(( spi_ena & ~(1 << $p) ))
                done
        fi

        # Disable direct SPI
        i2cset -y 2 0x71 1 0

        # Enable SPI poller
        i2cset -y 2 0x71 2 $spi_ena

        MINING_SW=`ls -l /usr/bin/cgminer`
        if [ "`echo $MINING_SW | grep bfgminer`" != "" ] ; then
                export LD_LIBRARY_PATH=/usr/bfgminer/
                start-stop-daemon -b -S -x screen -- -S cgminer -t cgminer -m -d
        else
                start-stop-daemon -b -S -x screen -- -S cgminer -t cgminer -m -d
        fi
}

do_stop() {
        killall -9 cgminer || true
}
case "$1" in
  start)
        echo -n "Starting $DESC: "
        do_start
        echo "$NAME."
        ;;
  stop)
        echo -n "Stopping $DESC: "
        do_stop
        echo "$NAME."
        ;;
  restart|force-reload)
        echo -n "Restarting $DESC: "
        do_stop
        do_start
        echo "$NAME."
        ;;
  *)
        N=/etc/init.d/$NAME
        echo "Usage: $N {start|stop|restart|force-reload}" >&2
        exit 1
        ;;
esac

exit 0