#!/bin/sh
# This script loads/unloads the loadable kernel modules for Wireless feature for non-HND images
# usage:
# bcm-wlan-drivers.sh operation [module]
#
# operation:  start, stop, restart
# [module]:   WLAN kernel module/s
#
# Script uses below order to find the WLAN modules to operate on
# argument $2, nvram (kernel_modules), default list based on image
# check existance of driver in the system before loading

trap "" 3
#The following will be populated by buildFS during the make process:
KERNELVER=4.1.52
HNDROUTER=
CPEROUTER=
WLAN_BTEST=
PROD_FW_PATH=""
CONFIG_WLDPDCTL=
BRCM_CHIP=63138
if [ ! -z $PROD_FW_PATH ]; then
    MFG_FW_PATH=$PROD_FW_PATH"/mfg"
else
    MFG_FW_PATH="/etc/wlan/dhd/mfg"
fi

WLDPDCTL=0
#unit (radio) index list - max (3)
UNITLIST="0 1 2"

# PCIe Wireless Card Status - bad card detection
pwlcs_preload()
{
  # Check if PWLCS is enabled or not
  # 0/not present - Disabled
  # N             - Enabled for N cards (N is generally 2 or 3)
  # Also need to compile DHD with DSLCPE_PWLCS flag
  MAXPWLCS=`nvram kget pwlcsmaxcrd`
  if [ "$MAXPWLCS" == "" ];  then
    MAXPWLCS=0
  fi

  if [ $MAXPWLCS != 0 ];  then
    echo "PWLCS:: DHD pre-load processing ..."
    needcommit=0
    card=0
    panic_on_oops=`cat /proc/sys/kernel/panic_on_oops`

    #disable panic_on_oops to stop auto rebooting on dhd fail
    echo 0 > /proc/sys/kernel/panic_on_oops

    # Initialize card status to GOOD, if not present
    while [ $card -lt $MAXPWLCS ]
    do
      status=`nvram kget pwlcspcie$card`
      if [ "$status" == "" ];  then
        echo "PWLCS:: PCIe[$card] status intializing to GOOD"
        nvram kset pwlcspcie$card="GOOD"
        needcommit=1
      fi
      echo "PWLCS:: PCIe[$card] pre-load status = $status"
      card=$(($card+1))
    done

    # Commit changes if needed
    if [ $needcommit == 1 ];  then
      nvram kcommit
	  sync
    fi
  fi
}

pwlcs_postload()
{
  if [ $MAXPWLCS != 0 ];  then
    echo "PWLCS:: DHD post-load processing ..."
    needcommit=0
    card=0

    # Get the updated card status changed by DHD
    while [ $card -lt $MAXPWLCS ]
    do
      status=`nvram kget pwlcspcie$card`
      echo "PWLCS:: PCIe[$card] post-load status = $status"

      # Save and commit driver changes to the nvram
      nvram kset pwlcspcie$card="$status"
      nvram kcommit
	  sync

      #restore the panic_on_oops for the rest of the system
      echo $panic_on_oops > /proc/sys/kernel/panic_on_oops

      # Check if card is a bad card
      case "$status" in
        *BOOT*) 
          ifexists=`cat /proc/net/dev |grep wl$card`
          set -- $ifexists
          if [ "$1" == "" ];  then 
            echo "==============================================================================="
            echo "Detected bad WLAN card on PCIe$card, rebooting system"
            echo "==============================================================================="
            # Trigger magic Sys Request reboot for fast reboot
            echo b > /proc/sysrq-trigger
          fi 
        ;; 
      esac
      card=$(($card+1))
    done
  fi
}

# to get instance base mixed with dhd and wl
get_unit()
{
#   selection method - first available unit

    for unit in $UNITLIST; do
        wlx=`cat /proc/net/dev |grep wl$unit`
        if [ ${#wlx} == 0 ]; then
            break
        fi
    done

     echo "$unit"
}

# create dummy network interfaces for power down units
wldpdctl_init()
{
  if [ ! -z $CPEROUTER ]; then
    if [ ! -z $CONFIG_WLDPDCTL ]; then
      WLDPDCTL=$(nvram kget wl_dpdctl_enable)
    fi
  fi

  if [ -z $WLDPDCTL ]; then
    WLDPDCTL=0
  fi

  if [ $WLDPDCTL = 1 ]; then
    for unit in $UNITLIST; do
      pwrdn=`nvram kget wl${unit}_dpd`
      if [[ -z "$pwrdn" ]]; then
        pwrdn="0"
      fi
      if [ "$pwrdn" = "1" ]; then
        ip link add wl$unit type dummy
        echo "Created dummy network interface wl$unit"
      fi
    done
  fi
}

# remove dummy network interfaces for power down units
wldpdctl_deinit()
{
  if [ ! -z $CPEROUTER ]; then
    if [ ! -z $CONFIG_WLDPDCTL ]; then
      WLDPDCTL=$(nvram kget wl_dpdctl_enable)
    fi
  fi

  if [ -z $WLDPDCTL ]; then
    WLDPDCTL=0
  fi

  if [ $WLDPDCTL = 1 ]; then
    intfl=`cat /proc/net/dev | grep " wl" | cut -d ":" -f1`
    for intf in $intfl; do
      noarp=`ifconfig $intf | grep NOARP`
      if [[ -z "$noarp" ]]; then
        dummy="0"
      else
        dummy="1"
      fi
      if [ "$dummy" = "1" ]; then
        ip link delete $intf
        echo "Deleted dummy network interface $intf"
      fi
    done
  fi
}

# Load given WLAN drivers
# modules_list contains the list of drivers to be loaded
load_modules()
{

        # set the default module list and parameters
        # check whether the module exists or not before adding
        # set the module parameters based on hnd (cperouter) or wlemf (legacy)
        if [ ! -z $CPEROUTER ]; then
            #CPE Router
            dhd_module_params="iface_name=wl dhd_console_ms=0"

            # For 47189 ,it only has single core CPU. Use kthread for dhd_dpc task.
            if [ "$BRCM_CHIP" == "47189" ]; then
                dhd_module_params="${dhd_module_params} dhd_dpc_prio=5"
            fi

            wl_module_params="intf_name=wl%d"
            wl_mfgtest_module="wl_mfgtest"
            hnd_mfgtest_module="hnd_mfgtest"
            if [ -f /proc/nvram/wl_nand_manufacturer ]; then
                is_mfg=`cat /proc/nvram/wl_nand_manufacturer`
                case $is_mfg in
                    2|3|6|7)
                        dhd_module_params=$dhd_module_params" firmware_path=$MFG_FW_PATH"
                        if [ -f /lib/modules/$KERNELVER/extra/$wl_mfgtest_module.ko ]; then
                            modules_list=${modules_list/wl/$wl_mfgtest_module}
                        fi
                        if [ -f /lib/modules/$KERNELVER/extra/$hnd_mfgtest_module.ko ]; then
                            modules_list=${modules_list/hnd/$hnd_mfgtest_module}
                        fi
                        ;;
                    *)
                        ;;
                esac
            fi
        else
            #Legacy
            dhd_module_params="iface_name=wl dhd_console_ms=0 firmware_path=/etc/wlan/dhd mfg_firmware_path=/etc/wlan/dhd/mfg dhd_dpc_prio=5"
            wl_module_params="intf_name=wl%d"
        fi

        wldpdctl_init

        echo "loading WLAN kernel modules ... $modules_list"

        for module in $modules_list
        do
            case "$module" in
                firmware_class)
                    if [ -f /lib/modules/$KERNELVER/kernel/drivers/base/firmware_loader/firmware_class.ko ]; then
                       insmod /lib/modules/$KERNELVER/kernel/drivers/base/firmware_loader/firmware_class.ko 
                    fi
                    ;;
                cfg80211)
                    insmod /lib/modules/$KERNELVER/$module.ko
                    ;;
                wlemf|hnd|hnd_mfgtest|emf|igs)
                    #no module parameters
                    module_params=""
                    ;;
                wl|wl_mfgtest)
                    module_params=$wl_module_params
                    ;;
                dhd)
                    module_params=$dhd_module_params
                    ;;
                *)
                    echo "wlan-drivers: unrecognized module [$module] in the load module list"
                    module_params=""
                    ;;
            esac

            if [ "$module" == "dhd" ] || [ "$module" == "wl" ] || [ "$module" == "wl_mfgtest" ]; then
                idx=`get_unit`
                instance_base="instance_base=$idx"
                module_params=$module_params" "$instance_base
            fi

            if [ -e /lib/modules/$KERNELVER/extra/$module.ko ]; then
                if [ "$module" == "dhd" ]; then
                    pwlcs_preload
                    insmod /lib/modules/$KERNELVER/extra/$module.ko $module_params
                    pwlcs_postload
                else
                    nic_nvram=`nvram kget wl_path`
                    if [ "$module" == "wl" ] && [ -f "$nic_nvram" ] ; then
                        echo "*** wl$idx Using alternate nic driver from $nic_nvram"
                        insmod $nic_nvram $module_params
                    else
                        insmod /lib/modules/$KERNELVER/extra/$module.ko $module_params
                    fi
                fi
            fi
        done

        # Configuing wlan related threads
        wlaffinity auto
}

# Unload given WLAN drivers
# modules_list contains the list of drivers to be loaded
unload_modules()
{
    echo "unloading WLAN kernel modules ... $unload_modules_list"

    for module in $unload_modules_list
    do
	echo unload $module
        rmmod $module.ko
    done

    wldpdctl_deinit
}


# Check for arguments.
# User should give atleast one argument to the program.
if [ $# -lt 1 ]; then
    echo "Usage: $0 operation [module] ..."
    echo "       operation:  start, stop, restart"
    echo "       [module]:   WLAN kernel module"
    exit 64
fi

# Do nothing for hnd only builds (HNDROUTER=y and CPEROUTER=)
if [ ! -z $HNDROUTER ]; then
    if [ -z $CPEROUTER ]; then
        echo "Skipping wlan-drivers.sh for HND images ..."
        exit 0
    fi
fi

# Do nothing for WLAN BTEST builds
WLAN_BTEST_FILEPATH=/proc/brcm/blxparms/wlan_btest
if  [ -e $WLAN_BTEST_FILEPATH ]; then
    btest_mode=$(cat $WLAN_BTEST_FILEPATH)
    btest_mode=${btest_mode:1}
    #nvram kset btest_mode=$btest_mode
    #nvram kcommit
else
    btest_mode=`nvram kget wlan_btest`
fi
if [ "$btest_mode" == "1" ];  then
    echo "Skipping wlan-drivers.sh since the kernel nvram wlan_btest=1"
    exit 0
fi
if [ "$WLAN_BTEST" == "y" ]; then
    if [ -f /usr/sbin/utelnetd ] ; then
        /usr/sbin/utelnetd -d 
    fi
    echo "Skipping wlan-drivers.sh for BTEST images ..."
    exit 0
fi

# For 47189 Host CPU, Check nvram "forcegen1rc" and set default vaule if not exist.
# the part must before dhd.ko insmod.
if [ "$BRCM_CHIP" == "47189" ]; then
    forcegen1rc_val=`nvram kget forcegen1rc`
    if [ -z "$forcegen1rc_val" ]; then
        `nvram kset forcegen1rc=1 ; nvram kcommit`
    fi
fi

# Sanity check for drivers directory
if [ ! -d /lib/modules/$KERNELVER/extra ]; then
    echo "ERROR: wlan-drivers.sh: /lib/modules/$KERNELVER/extra does not exist" 1>&2
    exit 1
fi

# Get the modules list
# Check argument2 first, then nvram, then default based on image
if [ ! -z "$2" ]; then
    modules_list=$2
    unload_modules_list=$modules_list
else
    if [ ! -z $CPEROUTER ]; then
        if [ -f /lib/modules/$KERNELVER/cfg80211.ko ]; then
            all_wlan_modules="firmware_class cfg80211 hnd emf igs dhd wl"
        else
            all_wlan_modules="hnd emf igs dhd wl"
        fi
    else
        all_wlan_modules="wlemf dhd wl"
    fi

    # Update the wlan module list from nvram if exists
    is_nvmodules_list=`nvram show 2>&1 |grep kernel_mods | grep -c '^'`
    if [ $is_nvmodules_list -eq '0' ]; then
        echo "no modules list in nvram, using defaults..."
        modules_list=$all_wlan_modules
    else
        modules_list=`nvram kget kernel_mods`
    fi
    for module in $modules_list
    do
        unload_modules_list="$module $unload_modules_list";
    done
fi

# Parse the first argument
case "$1" in
    start)
        load_modules
        exit 0
        ;;

    stop)
        unload_modules
        exit 0
        ;;

    restart)
        unload_modules
        sleep 3
        load_modules
        exit 0
        ;;

    *)
        echo "wlan-drivers: unrecognized option [$1]"
        exit 1
        ;;
esac