v2

.templates/00-smb_mounts.sh

@@ -0,0 +1,276 @@
+#!/usr/bin/with-contenv bashio
+# shellcheck shell=bash
+# shellcheck disable=
+set -e
+
+####################
+# DEFINE FUNCTIONS #
+####################
+
+test_mount () {
+
+    # Set initial test
+    MOUNTED=false
+    ERROR_MOUNT=false
+
+    # Exit if not mounted
+    if ! mountpoint -q /mnt/"$diskname"; then
+        return 0
+    fi
+
+    # Exit if can't write
+    # shellcheck disable=SC2015
+    mkdir "/mnt/$diskname/testaze" && touch "/mnt/$diskname/testaze/testaze" && rm -r "/mnt/$diskname/testaze" || ERROR_MOUNT=true
+    if [[ "$ERROR_MOUNT" == "true" ]]; then
+        # Test write permissions
+        if [[ "$MOUNTOPTIONS" == *"noserverino"* ]]; then
+            bashio::log.fatal "Disk is mounted, however unable to write in the shared disk. Please check UID/GID for permissions, and if the share is rw"
+        else
+            MOUNTOPTIONS="$MOUNTOPTIONS,noserverino"
+            echo "... testing with noserverino"
+            mount_drive "$MOUNTOPTIONS"
+            return 0
+        fi
+    fi
+
+    # Set correctly mounted bit
+    MOUNTED=true
+    return 0
+
+}
+
+mount_drive () {
+
+    # Define options
+    MOUNTED=true
+    MOUNTOPTIONS="$1"
+
+    # Try mounting
+    mount -t cifs -o "$MOUNTOPTIONS" "$disk" /mnt/"$diskname" 2>ERRORCODE || MOUNTED=false
+
+    # Test if succesful
+    if [[ "$MOUNTED" == "true" ]]; then
+        # shellcheck disable=SC2015
+        test_mount
+    fi
+
+}
+
+####################
+# MOUNT SMB SHARES #
+####################
+
+if bashio::config.has_value 'networkdisks'; then
+
+    # Alert message that it is a new code
+    if [[ "$(date +"%Y%m%d")" -lt "20240201" ]]; then
+        bashio::log.warning "------------------------"
+        bashio::log.warning "This is a new code, please report any issues on https://github.com/Mesteriis/hassio-addons-avm"
+        bashio::log.warning "------------------------"
+    fi
+
+    echo 'Mounting smb share(s)...'
+
+    ####################
+    # Define variables #
+    ####################
+
+    # Set variables
+    MOREDISKS=$(bashio::config 'networkdisks')
+    USERNAME=$(bashio::config 'cifsusername')
+    PASSWORD=$(bashio::config 'cifspassword')
+    SMBVERS=""
+    SECVERS=""
+    CHARSET=",iocharset=utf8"
+
+    # Clean data
+    MOREDISKS=${MOREDISKS// \/\//,\/\/}
+    MOREDISKS=${MOREDISKS//, /,}
+    MOREDISKS=${MOREDISKS// /"\040"}
+
+    # Is domain set
+    DOMAIN=""
+    DOMAINCLIENT=""
+    if bashio::config.has_value 'cifsdomain'; then
+        echo "... using domain $(bashio::config 'cifsdomain')"
+        DOMAIN=",domain=$(bashio::config 'cifsdomain')"
+        DOMAINCLIENT="--workgroup=$(bashio::config 'cifsdomain')"
+    fi
+
+    # Is  UID/GID set
+    PUID=",uid=$(id -u)"
+    PGID=",gid=$(id -g)"
+    if bashio::config.has_value 'PUID' && bashio::config.has_value 'PGID'; then
+        echo "... using PUID $(bashio::config 'PUID') and PGID $(bashio::config 'PGID')"
+        PUID=",uid=$(bashio::config 'PUID')"
+        PGID=",gid=$(bashio::config 'PGID')"
+    fi
+
+    ##################
+    # Mounting disks #
+    ##################
+
+    # shellcheck disable=SC2086
+    for disk in ${MOREDISKS//,/ }; do # Separate comma separated values
+
+        # Clean name of network share
+        # shellcheck disable=SC2116,SC2001
+        disk=$(echo $disk | sed "s,/$,,") # Remove / at end of name
+        disk="${disk//"\040"/ }"            #replace \040 with
+        diskname="${disk//\\//}"            #replace \ with /
+        diskname="${diskname##*/}"          # Get only last part of the name
+        MOUNTED=false
+
+        # Start
+        echo "... mounting $disk"
+
+        # Data validation
+        if [[ ! "$disk" =~ ^.*+[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+[/]+.*+$ ]]; then
+            bashio::log.fatal "...... the structure of your \"networkdisks\" option : \"$disk\" doesn't seem correct, please use a structure like //123.12.12.12/sharedfolder,//123.12.12.12/sharedfolder2. If you don't use it, you can simply remove the text, this will avoid this error message in the future."
+            touch ERRORCODE
+            continue
+        fi
+
+        # Prepare mount point
+        mkdir -p /mnt/"$diskname"
+        chown root:root /mnt/"$diskname"
+
+        # Quickly try to mount with defaults
+        mount_drive "rw,file_mode=0775,dir_mode=0775,username=${USERNAME},password=${PASSWORD},nobrl${SMBVERS}${SECVERS}${PUID}${PGID}${CHARSET}${DOMAIN}"
+
+        # Deeper analysis if failed
+        if [ "$MOUNTED" = false ]; then
+
+            # Extract ip part of server for further manipulation
+            server="$(echo "$disk" | grep -E -o "[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+")"
+
+            # Does server exists
+            output="$(nmap -F $server -T5 -oG -)"
+            if ! echo "$output" | grep 445/open &>/dev/null; then
+                if echo "$output" | grep /open &>/dev/null; then
+                    bashio::log.fatal "...... $server is reachable but SMB port not opened, stopping script"
+                    touch ERRORCODE
+                    continue
+                else
+                    bashio::log.fatal "...... fatal : $server not reachable, is it correct"
+                    touch ERRORCODE
+                    continue
+                fi
+            else
+                echo "...... $server is confirmed reachable"
+            fi
+
+            # Are credentials correct
+            OUTPUT="$(smbclient -t 2 -L "$disk" -U "$USERNAME"%"$PASSWORD" -c "exit" $DOMAINCLIENT 2>&1 || true)"
+            if echo "$OUTPUT" | grep -q "LOGON_FAILURE"; then
+                bashio::log.fatal "...... incorrect Username, Password, or Domain! Script will stop."
+                touch ERRORCODE
+                # Should there be a workgroup
+                if ! smbclient -t 2 -L $disk -N $DOMAINCLIENT -c "exit" &>/dev/null; then
+                    bashio::log.fatal "...... perhaps a workgroup must be specified"
+                    touch ERRORCODE
+                fi
+                continue
+            elif echo "$OUTPUT" | grep -q "tree connect failed" || echo "$OUTPUT" | grep -q "NT_STATUS_CONNECTION_DISCONNECTED"; then
+                echo "... testing path"
+                bashio::log.fatal "...... invalid or inaccessible SMB path. Script will stop."
+                touch ERRORCODE
+                continue
+            elif ! echo "$OUTPUT" | grep -q "Disk"; then
+                echo "... testing path"
+                bashio::log.fatal "...... no shares found. Invalid or inaccessible SMB path?"
+            else
+                echo "...... credentials are valid"
+            fi
+
+            # Extracting SMB versions and normalize output
+            # shellcheck disable=SC2210,SC2094
+            SMBVERS="$(nmap --script smb-protocols "$server" -p 445 2>1 | awk '/  [0-9]/' | awk '{print $NF}'  | cut -c -3 | sort -V | tail -n 1  || true)"
+            # Avoid :
+            SMBVERS="${SMBVERS/:/.}"
+            # Manage output
+            if [ -n "$SMBVERS" ]; then
+                case $SMBVERS in
+                    "202"|"200"|"20")
+                        SMBVERS="2.0"
+                        ;;
+                    21)
+                        SMBVERS="2.1"
+                        ;;
+                    302)
+                        SMBVERS="3.02"
+                        ;;
+                    311)
+                        SMBVERS="3.1.1"
+                        ;;
+                    "3.1")
+                        echo "SMB 3.1 detected, converting to 3.0"
+                        SMBVERS="3.0"
+                        ;;          
+                esac
+                echo "...... SMB version detected : $SMBVERS"
+                SMBVERS=",vers=$SMBVERS"
+            elif smbclient -t 2 -L "$server" -m NT1 -N $DOMAINCLIENT &>/dev/null; then
+                echo "...... SMB version : only SMBv1 is supported, this can lead to issues"
+                SECVERS=",sec=ntlm"
+                SMBVERS=",vers=1.0"
+            else
+                echo "...... SMB version : couldn't detect, default used"
+                SMBVERS=""
+            fi
+
+            # Test with different security versions
+            #######################################
+            for SECVERS in "$SECVERS" ",sec=ntlmv2" ",sec=ntlmssp" ",sec=ntlmsspi" ",sec=krb5i" ",sec=krb5" ",sec=ntlm" ",sec=ntlmv2i"; do
+                if [ "$MOUNTED" = false ]; then
+                    mount_drive "rw,file_mode=0775,dir_mode=0775,username=${USERNAME},password=${PASSWORD},nobrl${SMBVERS}${SECVERS}${PUID}${PGID}${CHARSET}${DOMAIN}"
+                fi
+            done
+
+        fi
+
+        # Messages
+        if [ "$MOUNTED" = true ]; then
+
+            bashio::log.info "...... $disk successfully mounted to /mnt/$diskname with options ${MOUNTOPTIONS/$PASSWORD/XXXXXXXXXX}"
+            # Remove errorcode
+            if [ -f ERRORCODE ]; then
+                rm ERRORCODE
+            fi
+
+            # Alert if smbv1
+            if [[ "$MOUNTOPTIONS" == *"1.0"* ]]; then
+                bashio::log.warning ""
+                bashio::log.warning "Your smb system requires smbv1. This is an obsolete protocol. Please correct this to prevent issues."
+                bashio::log.warning ""
+            fi
+
+        else
+            # Mounting failed messages
+            bashio::log.fatal "Error, unable to mount $disk to /mnt/$diskname with username $USERNAME, $PASSWORD. Please check your remote share path, username, password, domain, try putting 0 in UID and GID"
+            bashio::log.fatal "Here is some debugging info :"
+
+            # Provide debugging info
+            smbclient -t 2 -L $disk -U "$USERNAME%$PASSWORD" -c "exit"
+
+            # Error code
+            SMBVERS=""
+            SECVERS=""
+            PUID=""
+            PGID=""
+            CHARSET=""
+            mount_drive "rw,file_mode=0775,dir_mode=0775,username=${USERNAME},password=${PASSWORD},nobrl${SMBVERS}${SECVERS}${PUID}${PGID}${CHARSET}${DOMAIN}"
+            bashio::log.fatal "Error read : $(<ERRORCODE), addon will stop in 1 min"
+
+            # clean folder
+            umount "/mnt/$diskname" 2>/dev/null || true
+            rmdir "/mnt/$diskname" || true
+
+            # Stop addon
+            bashio::addon.stop
+
+        fi
+
+    done
+
+fi