#!/bin/bash

# The IP address shared by all USB network interfaces created by this script.
net_ip=192.168.42.120
# The associated netmask.
net_mask=255.255.0.0

enable_vcom=0
enable_rndis=0
enable_bulk=1

startup_bulk_dir=$(dirname "$0")
startup_bulk_exe=${startup_bulk_dir}/build/startup_bulk
#
#if [ ! -d /sys/kernel/config/usb_gadget ]; then
#    if ! $(grep -q dtoverlay=dwc2 /boot/config.txt) ; then
#        echo -e "\n\n\n## customized\n# for op-sdk enable USB device RNDIS\n[all]\ndtoverlay=dwc2" >> /boot/config.txt
#        init 6
#    fi
#
#    if ! $(grep -q modules-load=dwc2 /boot/cmdline.txt) ; then
#        sed -i 's/rootwait /&modules-load=dwc2 /' /boot/cmdline.txt
#        init 6
#    fi
#
#    if ! $(grep -q libcomposite /etc/modules) ; then
#        echo -e "\n\n\n## customized\n# for op-sdk enable USB device RNDIS\nlibcomposite" >> /etc/modules
#        init 6
#    fi
#
#    exit 1
#fi
# 卸载 g_mass_storage
if lsmod | grep -q g_mass_storage; then
    echo "Unloading g_mass_storage module..."
    modprobe -r g_mass_storage || { echo "WARNING: Failed to unload g_mass_storage. This might cause issues."; }
    sleep 1
fi

# 卸载 g_ether
if lsmod | grep -q g_ether; then
    echo "Unloading g_ether module..."
    modprobe -r g_ether || { echo "WARNING: Failed to unload g_ether. This might cause issues."; }
    sleep 1
fi

rmmod g_mass_storage
sleep 1
GADGET_NAME="pi4"
GADGET_ROOT="/sys/kernel/config/usb_gadget/${GADGET_NAME}"
# 2. 卸载并清理旧的 gadget 配置 (可选，但推荐在测试时使用以确保干净状态)
if [ -d "${GADGET_ROOT}" ]; then
    echo "Disabling existing gadget '${GADGET_NAME}'..."
    # 尝试解绑当前的 UDC (如果已经绑定)
    if [ -f "${GADGET_ROOT}/UDC" ]; then
        current_udc=$(cat "${GADGET_ROOT}/UDC")
        if [ -n "$current_udc" ]; then
            echo "Unbinding ${current_udc} from previous gadget..."
            echo "" > "${GADGET_ROOT}/UDC" # 解绑 UDC
            sleep 1
        fi
    fi
    rm -rf "${GADGET_ROOT}" # 删除所有旧的配置
    echo "Old gadget configuration removed."
    sleep 1
fi

if [ 1 ]; then
  echo "Creating gadget..."
    mkdir -p /sys/kernel/config/usb_gadget/pi4
    cd /sys/kernel/config/usb_gadget/pi4
        echo 0x2ca3 > idVendor # Linux Foundation
        echo 0xF001 > idProduct # Multifunction Composite Gadget

        if [ ${enable_vcom} -eq 1 ]; then
            echo 0xF002 > idProduct # Multifunction Composite Gadget
        fi

        if [ ${enable_rndis} -eq 1 ]; then
            echo 0xF003 > idProduct # Multifunction Composite Gadget
        fi

        if [ ${enable_bulk} -eq 1 ]; then
            echo 0xF001 > idProduct # Multifunction Composite Gadget
        fi

        echo 0x0001 > bcdDevice # v1.0.0
        echo 0x0200 > bcdUSB # USB2

        echo 0xEF > bDeviceClass
        echo 0x02 > bDeviceSubClass
        echo 0x01 > bDeviceProtocol

        mkdir -p strings/0x409
            echo "psdk-rpi4b" > strings/0x409/serialnumber
            echo "raspberry" > strings/0x409/manufacturer
            echo "PI4" > strings/0x409/product

        cfg=configs/c.1

        mkdir -p "${cfg}"
            echo 0x80 > ${cfg}/bmAttributes
            echo 250 > ${cfg}/MaxPower

        cfg_str=""

        if [ ${enable_rndis} -eq 1 ]; then
            cfg_str="${cfg_str}+RNDIS"
            func=functions/rndis.usb0
            mkdir -p "${func}"
                ln -sf "${func}" "${cfg}"

            # Informs Windows that this device is compatible with the built-in RNDIS
            # driver. This allows automatic driver installation without any need for
            # a .inf file or manual driver selection.
            echo 1 > os_desc/use
            echo 0xcd > os_desc/b_vendor_code
            echo MSFT100 > os_desc/qw_sign
                echo RNDIS > "${func}/os_desc/interface.rndis/compatible_id"
                echo 5162001 > "${func}/os_desc/interface.rndis/sub_compatible_id"

            ln -sf "${cfg}" os_desc
        fi
        mkdir -p "${cfg}/strings/0x409"
        echo "${cfg_str:1}" > "${cfg}/strings/0x409/configuration"
    # cd -

fi
    udevadm settle -t 5 || :
    ls /sys/class/udc > UDC

if [ ${enable_rndis} -eq 1 ]; then
    /sbin/brctl addbr pi4br0
    /sbin/ifconfig pi4br0 ${net_ip} netmask ${net_mask} up

    /sbin/brctl addif pi4br0 usb0
    /sbin/ifconfig usb0 down
    /sbin/ifconfig usb0 up
fi
exit 0