diff --git a/.templates/ha_entrypoint.sh b/.templates/ha_entrypoint.sh index ccc3cd712..57bf9773e 100644 --- a/.templates/ha_entrypoint.sh +++ b/.templates/ha_entrypoint.sh @@ -7,24 +7,22 @@ echo "Starting..." # Starting scripts # #################### -run_script() { - runfile="$1" - script_kind="$2" - echo "$runfile: executing" +# Loop through /etc/cont-init.d/* +for SCRIPTS in /etc/cont-init.d/*; do + [ -e "$SCRIPTS" ] || continue + echo "$SCRIPTS: executing" + # Check if run as root if [ "$(id -u)" -eq 0 ]; then - chown "$(id -u)":"$(id -g)" "$runfile" - chmod a+x "$runfile" + chown "$(id -u)":"$(id -g)" "$SCRIPTS" + chmod a+x "$SCRIPTS" else echo -e "\e[38;5;214m$(date) WARNING: Script executed with user $(id -u):$(id -g), things can break and chown won't work\e[0m" # Disable chown and chmod in scripts - sed -i -E 's/^([[:space:]]*)chown /\1true # chown /' "$runfile" - sed -i -E 's/^([[:space:]]*)chmod /\1true # chmod /' "$runfile" - fi - - # Replace s6-setuidgid with su-based equivalent - if ! command -v s6-setuidgid >/dev/null 2>&1; then - sed -i -E 's|s6-setuidgid[[:space:]]+([a-zA-Z0-9._-]+)[[:space:]]+(.*)$|su -s /bin/bash \1 -c "\2"|g' "$runfile" + sed -i "s/^chown /true # chown /g" "$SCRIPTS" + sed -i "s/ chown / true # chown /g" "$SCRIPTS" + sed -i "s/^chmod /true # chmod /g" "$SCRIPTS" + sed -i "s/ chmod / true # chmod /g" "$SCRIPTS" fi # Get current shebang, if not available use another @@ -32,7 +30,7 @@ run_script() { if [ ! -f "${currentshebang%% *}" ]; then for shebang in "/command/with-contenv bashio" "/usr/bin/with-contenv bashio" "/usr/bin/env bashio" "/usr/bin/bashio" "/usr/bin/bash" "/usr/bin/sh" "/bin/bash" "/bin/sh"; do command_path="${shebang%% *}" - if [ -x "$command_path" ] && "$command_path" echo "yes" > /dev/null 2>&1; then + if [ -x "$command_path" ] && "$command_path" echo "yes" >/dev/null 2>&1; then echo "Valid shebang: $shebang" break fi @@ -41,37 +39,64 @@ run_script() { fi # Use source to share env variables when requested - if [[ "$script_kind" == service ]]; then - (exec "$runfile") & true + if [ "${ha_entry_source:-null}" = true ] && command -v "source" &>/dev/null; then + sed -i "s/(.*\s|^)exit \([0-9]\+\)/ \1 return \2 || exit \2/g" "$SCRIPTS" + sed -i "s/bashio::exit.nok/return 1/g" "$SCRIPTS" + sed -i "s/bashio::exit.ok/return 0/g" "$SCRIPTS" + # shellcheck disable=SC1090 + source "$SCRIPTS" || echo -e "\033[0;31mError\033[0m : $SCRIPTS exiting $?" else - if [ "${ha_entry_source:-null}" = true ]; then - sed -Ei 's/(^|[[:space:]])exit ([0-9]+)/\1return \2 || exit \2/g' "$runfile" - sed -i "s/bashio::exit.nok/return 1/g" "$runfile" - sed -i "s/bashio::exit.ok/return 0/g" "$runfile" - # shellcheck disable=SC1090 - source "$runfile" || echo -e "\033[0;31mError\033[0m : $runfile exiting $?" - else - "$runfile" || echo -e "\033[0;31mError\033[0m : $runfile exiting $?" - fi + "$SCRIPTS" || echo -e "\033[0;31mError\033[0m : $SCRIPTS exiting $?" fi # Cleanup - if [[ "$script_kind" != service ]]; then - rm "$runfile" - fi -} - -# Loop through /etc/cont-init.d/* -for SCRIPTS in /etc/cont-init.d/*; do - [ -e "$SCRIPTS" ] || continue - run_script "$SCRIPTS" script + rm "$SCRIPTS" done -# Start services.d -for service_dir in /etc/services.d/*; do - SCRIPTS="${service_dir}/run" - [ -e "$SCRIPTS" ] || continue - run_script "$SCRIPTS" service +# Loop through /etc/services.d/*/run +[[ -d /etc/services.d ]] && for SERVICE in /etc/services.d/*/run; do + [ -e "$SERVICE" ] || continue + echo "$SERVICE: executing" + + # Check if run as root + if [ "$(id -u)" -eq 0 ]; then + chown "$(id -u)":"$(id -g)" "$SERVICE" + chmod a+x "$SERVICE" + else + echo -e "\e[38;5;214m$(date) WARNING: Script executed with user $(id -u):$(id -g), things can break and chown won't work\e[0m" + # Disable chown and chmod in scripts + sed -i "s/^chown /true # chown /g" "$SERVICE" + sed -i "s/ chown / true # chown /g" "$SERVICE" + sed -i "s/^chmod /true # chmod /g" "$SERVICE" + sed -i "s/ chmod / true # chmod /g" "$SERVICE" + fi + + # Get current shebang, if not available use another + currentshebang="$(sed -n '1{s/^#![[:blank:]]*//p;q}' "$SERVICE")" + if [ ! -f "${currentshebang%% *}" ]; then + for shebang in "/command/with-contenv bashio" "/usr/bin/with-contenv bashio" "/usr/bin/env bashio" "/usr/bin/bashio" "/usr/bin/bash" "/usr/bin/sh" "/bin/bash" "/bin/sh"; do + command_path="${shebang%% *}" + if [ -x "$command_path" ] && "$command_path" echo "yes" >/dev/null 2>&1; then + echo "Valid shebang: $shebang" + break + fi + done + sed -i "s|$currentshebang|$shebang|g" "$SERVICE" + fi + + # Use source to share env variables when requested + if [ "${ha_entry_source:-null}" = true ] && command -v "source" &>/dev/null; then + sed -i "s/(.*\s|^)exit \([0-9]\+\)/ \1 return \2 || exit \2/g" "$SERVICE" + sed -i "s/bashio::exit.nok/return 1/g" "$SERVICE" + sed -i "s/bashio::exit.ok/return 0/g" "$SERVICE" + # shellcheck disable=SC1090 + source "$SERVICE" || echo -e "\033[0;31mError\033[0m : $SERVICE exiting $?" + else + "$SERVICE" || echo -e "\033[0;31mError\033[0m : $SERVICE exiting $?" + fi + + # Cleanup + rm "$SERVICE" done ###################### @@ -85,32 +110,32 @@ if [ "$$" -eq 1 ]; then terminate() { echo "Termination signal received, forwarding to subprocesses..." # Terminate all subprocesses - if command -v pgrep &> /dev/null; then + if command -v pgrep &>/dev/null; then for pid in $(pgrep -P $$); do echo "Terminating child PID $pid" - kill -TERM "$pid" 2> /dev/null || echo "Failed to terminate PID $pid" + kill -TERM "$pid" 2>/dev/null || echo "Failed to terminate PID $pid" done else # Fallback to iterating through /proc if pgrep is not available for pid in /proc/[0-9]*/; do pid=${pid#/proc/} pid=${pid%/} - if [[ "$pid" -ne 1 ]] && grep -q "^PPid:\s*$$" "/proc/$pid/status" 2> /dev/null; then + if [[ "$pid" -ne 1 ]] && grep -q "^PPid:\s*$$" "/proc/$pid/status" 2>/dev/null; then echo "Terminating child PID $pid" - kill -TERM "$pid" 2> /dev/null || echo "Failed to terminate PID $pid" + kill -TERM "$pid" 2>/dev/null || echo "Failed to terminate PID $pid" fi done fi - kill -TERM -$$ 2>/dev/null || true - sleep 5 - kill -KILL -$$ 2>/dev/null || true - + wait echo "All subprocesses terminated. Exiting." exit 0 } trap terminate SIGTERM SIGINT - wait -n + while :; do + sleep infinity & + wait $! + done else echo " " echo -e "\033[0;32mStarting the upstream container\033[0m"