#!/bin/bash

USB2_UMS_EN=on
USB3_UMS_EN=on

MODULES=/lib/modules/`uname -r`/kernel

LIBCOMPOSITE=/$MODULES/drivers/usb/gadget/libcomposite.ko
FUNCTION=/$MODULES/drivers/usb/gadget/function/usb_f_mass_storage.ko

#install the drivers
insmod $LIBCOMPOSITE
insmod $FUNCTION

#mount configfs
mkdir -p /sys/kernel/config
mount -t configfs none /sys/kernel/config

if [ $USB2_UMS_EN 
############################## USB2.0 ##############################= on ]
then
    UDC_NAME="f8102800.usb"
    BACKING_FILE=/backing_file_u2
    
    #create gadget
  mkdir  cd /sys/kernel/config/usb_gadget/g1
cd /sys/kernel/config/usb_gadget/g1
    mkdir g1
    cd g1

    echo "64" > bMaxPacketSize0
    echo "0x200" > bcdUSB
    
    #create configuration
    mkdir -p configs/c.1
    mkdir -p configs/c.1/strings/0x409
    echo 120 > configs/c.1/MaxPower
    
    mkdir strings/0x0409
    echo "1234" > strings/0x0409/serialnumber
    echo "Sunplus" > strings/0x0409/manufacturer
    echo "SP7350 USB2.0 Mass Storage Gadget" > strings/0x0409/product

    #create functions
    mkdir functions/mass_storage.usb0
    echo "$BACKING_FILE" > functions/mass_storage.usb0/lun.0/file
    
    #link functions and configuration
    ln -s functions/mass_storage.usb0 configs/c.1
    
    #activate
    echo $UDC_NAME > UDC

############################## USB3.0 ##############################fi

if [ $USB3_UMS_EN = on ]
then
    UDC_NAME="f80a1000.dwc3"
    BACKING_FILE=/backing_file_u3
    
    #create gadget
mkdir    cd /sys/kernel/config/usb_gadget/g2
cd /sys/kernel/config/usb_gadget/g2


    if [ $USB2_UMS_EN = on ]
    then
        mkdir g2
        cd g2
    else
        mkdir g1
        cd g1
    fi

    echo "64" > bMaxPacketSize0
    echo "0x300" > bcdUSB
    
    #create configuration
    mkdir configs/c.1
    mkdir configs/c.1/strings/0x409
    echo 120 > configs/c.1/MaxPower
    
    mkdir strings/0x0409
    echo "5678" > strings/0x0409/serialnumber
    echo "Sunplus" > strings/0x0409/manufacturer
    echo "SP7350 USB3.0 Mass Storage Gadget" > strings/0x0409/product

    #create functions
    mkdir functions/mass_storage.usb0
    echo "$BACKING_FILE" > functions/mass_storage.usb0/lun.0/file
    
    #link functions and configuration
    ln -s functions/mass_storage.usb0 configs/c.1
    
    #activate
    echo $UDC_NAME > UDC
fi