下面是编程之家 jb51.cc 通过网络收集整理的代码片段。
编程之家小编现在分享给大家，也给大家做个参考。
#!/bin/bash
#Author: Castle Liu
#Description: upload a file or excute a script on remote servers
#Usage: excute_cmd.sh -l {hosts_list} -f {script_file} -d /tmp
####functions###
#help
#function used to print help info.
help() {
cat <<EOF
-l preferred,the list file of hosts,1st host,2nd user name,and,optional,3st password for the user.
-f preferred,the file to upload.
-d preferred,directory to place the file uploaded.
-p optional,remote sshd listen port,if not provided,use 22 as default.
-L optional,log file path to save excute log,use "excute_cmd.log" as default.
-S optional,selected if chosed to use sudo
-e optional,if selected,will excute the script file
-h print help info.
EOF
}
#help
#parse_args
#function to parse paremeters.
#do pass in the args from shell,for example:parse_args $*
parse_args(){
while getopts "hl:f:d:p:L:Se" arg
do
case ${arg} in
h)
help
exit 0
;;
l)
export _HOST_F=${OPTARG}
;;
f)
export _SCRIPT_F=${OPTARG}
;;
d)
export _DIR_R=${OPTARG}
;;
p)
export _PORT_R=${OPTARG}
;;
L)
export _LOG_F=${OPTARG}
;;
S)
export _USE_SUDO=true
;;
e)
export _EXC_F=true
;;
?)
echo "Unknown argument,exit..."
exit 1
;;
esac
done
#echo ${_HOST_F},${_SCRIPT_F}
}
#parse_args $*
#scp_file
#function to scp a file to a remote server.
#$1,Remote host ip addresss
#$2,Remote user name
#$3,Remote user password
#$4,The file to be copied local
#$5,Remote loacation of the script to be copied,can be a directory ora absulote file name
#$6,The remote sshd port in use
scp_file() {
_host_r=$1
_user_r=$2
_passwd_r=$3
_script_l=$4
_script_r=$5
_port_r=$6
sed -i "/${_host_r}/d" ~/.ssh/known_hosts
expect << EXPECT_EOF >> .runninglog
spawn scp -P ${_port_r} ${_script_l} ${_user_r}@${_host_r}:${_script_r}
expect "continue connecting (yes/no)?" {send "yes\n"}
expect "password:" {send "${_passwd_r}\n"}
expect off
EXPECT_EOF
}
#excute_script_with_sudo
#function to excute a script on remote server and scp back the running log,sudo will be used.
#$1,The script to be excuted in remote server,must be a absulote file name
#$5,The remote sshd port in use
excute_script_with_sudo () {
_host_r=$1
_user_r=$2
_passwd_r=$3
_script_r=$4
_port_r=$5
_log_r=`dirname ${_script_r}`"/.tmpexc.log"
expect << EXPECT_EOF >> .runninglog
spawn ssh -p ${_port_r} ${_user_r}@${_host_r} -C \""sudo su - -c \"bash ${_script_r} > ${_log_r}\""\"
expect "password:" {send "${_passwd_r}\n"}
expect "*password*" {send "${_passwd_r}\n"}
expect off
spawn ssh -p ${_port_r} ${_user_r}@${_host_r} -C \""sudo su - -c \"chown ${_user_r} ${_log_r}\""\"
expect "password:" {send "${_passwd_r}\n"}
expect "*password*" {send "${_passwd_r}\n"}
expect off
spawn scp -P ${_port_r} ${_user_r}@${_host_r}:${_log_r} ./${_host_r}.log.tmp
expect "password:" {send "${_passwd_r}\n"}
expect off
spawn ssh -p ${_port_r} ${_user_r}@${_host_r} -C "rm -rf ${_log_r};rm -rf ${_script_r}"
expect "password:" {send "${_passwd_r}\n"}
expect off
EXPECT_EOF
}
#excute_script
#function to excute a script on remote server and scp back the running log,The remote sshd port in use
excute_script () {
_host_r=$1
_user_r=$2
_passwd_r=$3
_script_r=$4
_port_r=$5
_log_r=`dirname ${_script_r}`"/.tmpexc.log"
expect << EXPECT_EOF >> .runninglog
spawn ssh -p ${_port_r} ${_user_r}@${_host_r} -C "bash ${_script_r} > ${_log_r}"
expect "password:" {send "${_passwd_r}\n"}
expect off
spawn scp -P ${_port_r} ${_user_r}@${_host_r}:${_log_r} ./${_host_r}.log.tmp
expect "password:" {send "${_passwd_r}\n"}
expect off
spawn ssh -p ${_port_r} ${_user_r}@${_host_r} -C "rm -rf ${_log_r};rm -rf ${_script_r}"
expect "password:" {send "${_passwd_r}\n"}
expect off
EXPECT_EOF
}
#handle_excute_log
#function to uniq the log file excute remotely and scp back.
#$1,the final log file to be make
#$2,the host list file,using awk -F [,],so the file must be seperated by ',',and the field 1 will be used.
handle_excute_log() {
_log_l=$1
_host_l=$2
echo "###Handling Log###"
if [[ -f ${_log_l} ]];then
echo "log file exist,will be move as .bak..."
mv ${_log_l} ${_log_l}.bak
fi
cat ${_host_l}|awk -F [,] '{print $1}'|while read _host
do
if [[ "${host}" != "EOF" ]];then
_tmp_l=${_host}.log.tmp
echo "###################" >> ${_log_l}
echo "###${_host} excute log begin...." >> ${_log_l}
cat ${_tmp_l} >> ${_log_l}
echo "###${_host}'s excute log ended." >> ${_log_l}
echo "###################" >> ${_log_l}
echo "" >> ${_log_l}
rm -rf ${_tmp_l}
fi
done
}
#excute_script 192.168.56.5 shell shell /home/shell/test.sh
#handle_excute_log
##$$main$$##
#parsing the args
parse_args $*
export _LOG_F=${_LOG_F:-'excute_cmd.log'}
export _PORT_R=${_PORT_R:-'22'}
#check the required file valid or not
if [[ ! -f ${_HOST_F} ]];then
echo 'The host_list file not valid,please retry!!!'
exit 1
fi
if [[ ! -f ${_SCRIPT_F} ]];then
echo 'The script file not valid,please retry!!!'
exit 1
fi
#read in password
read -t 20 -s -p "Enter password for ${_HOST},just [enter] if provided by file:" _PASSWD
#excute script file one by one according to _HOST_F
cat ${_HOST_F}|while read _LINE
do
if [[ ${_LINE} != "EOF" ]];then
_HOST=`echo ${_LINE}|awk -F [,] '{print $1}'`
_USER=`echo ${_LINE}|awk -F [,] '{print $2}'`
#check if password is provided.
if [[ "${_PASSWD}"x == ""x ]];then
_PASSWD=`echo ${_LINE}|awk -F [,] '{print $3}'`
if [[ "${_PASSWD}"x == ""x ]];then
echo ""
echo "Password needed,please retry."
exit 1
fi
fi
echo "Handling ${_HOST}......"
#upload file to remote server.
scp_file ${_HOST} ${_USER} ${_PASSWD} ${_SCRIPT_F} ${_DIR_R} ${_PORT_R}
#excute the script if "-e" is selected.
if [[ ${_EXC_F} ]];then
echo "${_SCRIPT_F} will be excuted on ${_HOST}."
_NAME_F=`basename ${_SCRIPT_F}`
_SCRIPT_R="${_DIR_R}/${_NAME_F}"
#use excute_script_with_sudo if "-S" is selected
if [[ ${_USE_SUDO} ]];then
echo "Script will be excuted with sudo."
excute_script_with_sudo ${_HOST} ${_USER} ${_PASSWD} ${_SCRIPT_R} ${_PORT_R}
else
excute_script ${_HOST} ${_USER} ${_PASSWD} ${_SCRIPT_R} ${_PORT_R}
fi
fi
echo "${_HOST} done."
fi
done
#collect the log file
if [[ ${_EXC_F} ]];then
handle_excute_log ${_LOG_F} ${_HOST_F}
fi
以上是编程之家(jb51.cc)为你收集整理的全部代码内容，希望文章能够帮你解决所遇到的程序开发问题。
如果觉得编程之家网站内容还不错，欢迎将编程之家网站推荐给程序员好友。
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。