1 files changed, 5 insertions, 4 deletions
diff --git a/bin/slave-build-connect b/bin/slave-build-connect
index 2ca72a3..f8ee9ad 100755
--- a/bin/slave-build-connect
+++ b/bin/slave-build-connect
@@ -3,11 +3,12 @@
 # shellcheck source=conf/default.conf
 . "${0%/*}/../conf/default.conf"
 
-if [ "${SSH_ORIGINAL_COMMAND%% *}" = "get-assignment" ] || \
-  [ "${SSH_ORIGINAL_COMMAND%% *}" = "return-assignment" ]; then
+if [ "x${SSH_ORIGINAL_COMMAND%% *}" = 'xget-assignment' ] || \
+  [ "x${SSH_ORIGINAL_COMMAND%% *}" = 'xreturn-assignment' ] || \
+  [ "x${SSH_ORIGINAL_COMMAND%% *}" = 'xping-from-slave' ]; then
 
   # small check to prevent some shell-injections
-  if echo "${SSH_ORIGINAL_COMMAND}" | \
+  if printf '%s\n' "${SSH_ORIGINAL_COMMAND}" | \
     grep -q '[^-a-zA-Z0-9.+_ ]'; then
 
     >&2 printf 'Invalid command: "%s".\n' "${SSH_ORIGINAL_COMMAND}"
@@ -28,7 +29,7 @@ if [ "${SSH_ORIGINAL_COMMAND%% *}" = "get-assignment" ] || \
           base64 -w0
       )"
   } | \
-    ${mysql_command}
+    mysql_run_query
 
   slave="$1" /bin/sh -c "${base_dir}/bin/${SSH_ORIGINAL_COMMAND}"