1 files changed, 166 insertions, 0 deletions
diff --git a/ctdb/config/events/legacy/50.samba.script b/ctdb/config/events/legacy/50.samba.script
new file mode 100755
index 0000000..84600e2
--- /dev/null
+++ b/ctdb/config/events/legacy/50.samba.script
@@ -0,0 +1,166 @@
+#!/bin/sh
+# ctdb event script for Samba
+
+[ -n "$CTDB_BASE" ] ||
+	CTDB_BASE=$(d=$(dirname "$0") && cd -P "$d" && dirname "$PWD")
+
+. "${CTDB_BASE}/functions"
+
+detect_init_style
+
+case $CTDB_INIT_STYLE in
+suse)
+	CTDB_SERVICE_SMB=${CTDB_SERVICE_SMB:-smb}
+	;;
+debian)
+	CTDB_SERVICE_SMB=${CTDB_SERVICE_SMB:-smbd}
+	;;
+*)
+	# Use redhat style as default:
+	CTDB_SERVICE_SMB=${CTDB_SERVICE_SMB:-smb}
+	;;
+esac
+
+service_name="samba"
+
+load_script_options
+
+ctdb_setup_state_dir "service" "$service_name"
+
+service_start()
+{
+	# make sure samba is not already started
+	service "$CTDB_SERVICE_SMB" stop >/dev/null 2>&1
+	killall -0 -q smbd && {
+		sleep 1
+		# make absolutely sure samba is dead
+		killall -q -9 smbd
+	}
+	# start Samba service. Start it reniced, as under very heavy load
+	# the number of smbd processes will mean that it leaves few cycles
+	# for anything else
+	nice_service "$CTDB_SERVICE_SMB" start || die "Failed to start samba"
+}
+
+service_stop()
+{
+	service "$CTDB_SERVICE_SMB" stop
+	program_stack_traces "smbd" 5
+}
+
+######################################################################
+# Show the testparm output using a cached smb.conf to avoid delays due
+# to registry access.
+
+# script_state_dir set by ctdb_setup_state_dir()
+# shellcheck disable=SC2154
+smbconf_cache="$script_state_dir/smb.conf.cache"
+
+testparm_foreground_update()
+{
+	_timeout="$1"
+
+	# No need to remove these temporary files, since there are only 2
+	# of them.
+	_out="${smbconf_cache}.out"
+	_err="${smbconf_cache}.err"
+
+	timeout "$_timeout" testparm -v -s >"$_out" 2>"$_err"
+	case $? in
+	0) : ;;
+	124)
+		if [ -f "$smbconf_cache" ]; then
+			echo "WARNING: smb.conf cache update timed out - using old cache file"
+			return 1
+		else
+			echo "ERROR: smb.conf cache create failed - testparm command timed out"
+			exit 1
+		fi
+		;;
+	*)
+		if [ -f "$smbconf_cache" ]; then
+			echo "WARNING: smb.conf cache update failed - using old cache file"
+			cat "$_err"
+			return 1
+		else
+			echo "ERROR: smb.conf cache create failed - testparm failed with:"
+			cat "$_err"
+			exit 1
+		fi
+		;;
+	esac
+
+	# Only using $$ here to avoid a collision.  This is written into
+	# CTDB's own state directory so there is no real need for a secure
+	# temporary file.
+	_tmpfile="${smbconf_cache}.$$"
+	# Patterns to exclude...
+	_pat='^[[:space:]]+(registry[[:space:]]+shares|include|copy|winbind[[:space:]]+separator)[[:space:]]+='
+	grep -Ev "$_pat" <"$_out" >"$_tmpfile"
+	mv "$_tmpfile" "$smbconf_cache" # atomic
+
+	return 0
+}
+
+testparm_background_update()
+{
+	_timeout="$1"
+
+	testparm_foreground_update "$_timeout" >/dev/null 2>&1 </dev/null &
+}
+
+testparm_get ()
+{
+	_param="$1"
+
+	sed -n \
+	    -e "s|^[[:space:]]*${_param}[[:space:]]*=[[:space:]]\(..*\)|\1|p" \
+	    "$smbconf_cache"
+
+}
+
+list_samba_shares()
+{
+	testparm_get "path" | sed -e 's/"//g'
+}
+
+list_samba_ports()
+{
+	testparm_get "smb ports"
+}
+
+###########################
+
+case "$1" in
+startup)
+	service_start
+	;;
+
+shutdown)
+	service_stop
+	;;
+
+monitor)
+	testparm_foreground_update 10
+	ret=$?
+
+	smb_ports="$CTDB_SAMBA_CHECK_PORTS"
+	if [ -z "$smb_ports" ]; then
+		smb_ports=$(list_samba_ports)
+		[ -n "$smb_ports" ] || die "Failed to set smb ports"
+	fi
+	# Intentionally unquoted multi-word value here
+	# shellcheck disable=SC2086
+	ctdb_check_tcp_ports $smb_ports || exit $?
+
+	if [ "$CTDB_SAMBA_SKIP_SHARE_CHECK" != "yes" ]; then
+		list_samba_shares | ctdb_check_directories || exit $?
+	fi
+
+	if [ $ret -ne 0 ]; then
+		testparm_background_update 10
+	fi
+	;;
+esac
+
+exit 0