#!/bin/bash

shell_user="root"
shell_pass="1233"
shell_port="22"
shell_list="/root/ip_list"
shell_row=`cat $shell_list |wc -l`

comad[0]=$1
comad[1]=$2
comad[2]=$3
comad[3]=$4
comad[4]=$5
comad[5]=$6
comad[6]=$7
comad[7]=$8
comad[8]=$9
comad[9]=${10}
comad[10]=${11}
comad[11]=${12}
comad[12]=${13}
comad[13]=${14}
comad[14]=${15}
comad[15]=${16}
comad[16]=${17}
comad[17]=${18}
comad[18]=${19}
comad[19]=${20}
comad[20]=${21}

function shell_exp(){

date_start=`date +"%T"`
rm -fr ./shell_log
for temp in `seq 1 $shell_row`
do
        Ip_Addr=`cat $shell_list |head -n $temp |tail -n 1`
    
    ping -c 1 $Ip_Addr &>/dev/null
    if [ $? -eq 0 ]
    then
        echo -e "\e[34;1m====================================================>$Ip_Addr<=======================================================\e[;m\n"
            ssh -p $shell_port $shell_user@$Ip_Addr ${comad[*]}
        if [ $? -eq 0 ]
        then
            echo -e "$Ip_Addr\t\t$shell_port\t\t${comad[*]} \t\t成功\n" >>./shell_log
        fi
    else
        echo -e "$Ip_Addr\t\t$shell_port\t\t${comad[*]} \t\t失败\n" >>./shell_log
        continue 
    fi
done
date_end=`date +"%T"`
    clear 
    echo -e "\e[34;1m===========================================================================================================================\e[;m"
    echo -e "\e[34;1m主机IP           端口      命令                      状态  \e[;m\n"
    cat ./shell_log
    echo -e "\e[34;1m===========================================================================================================================\e[;m\n"
    echo -e "\e[34;1m 开始时间:$date_start \t\t\t 结束时间:$date_end  \t\t\t\t  \e[;m\n"
    echo -e "\e[34;1m 执行主机数:$shell_row \t\t\t 成功主机:`cat ./shell_log |grep "成功" |wc -l`  \t\t\t 失败主机:`cat ./shell_log |grep "失败" |wc -l`\t\t\t\e[;m\n"
    echo -e "\e[34;1m 失败主机记录 ↓\e[;m\n"
    cat ./shell_log |grep "失败"
    echo -e "\e[34;1m===========================================================================================================================\e[;m\n"

}

function shell_upload(){

date_start=`date +"%T"`
rm -fr ./shell_log

for temp in `seq 1 $shell_row`
do

        take_ip=`cat $shell_list |head -n $temp |tail -n 1`
        Ip_Addr="$take_ip"
    ping -c 1 $Ip_Addr &>/dev/null
    if [ $? -eq 0 ]
    then
        scp ${comad[1]} $shell_user@$Ip_Addr:${comad[2]} &>/dev/null
        if [ $? -eq 0 ]
        then
            echo -e "\e[34;1m$Ip_Addr\t\t 传输完成....\e[;m\n"
            echo -e "$Ip_Addr\t\t$shell_port\t\t${comad[1]}\t\t${comad[2]}\t\t\t\t成功\n" >>./shell_log
        fi
    else
            echo -e "\e[34;1m$Ip_Addr\t\t 传输失败....\e[;m\n"
            echo -e "$Ip_Addr\t\t$shell_port\t\t${comad[1]}\t\t${comad[2]}\t\t\t\t失败\n" >>./shell_log
            continue
    fi

done

    date_end=`date +"%T"`
    clear 
    echo -e "\e[34;1m===========================================================================================================================\e[;m"
    echo -e "\e[34;1m主机IP           端口      传输源         传输到         状态  \e[;m\n"
    cat ./shell_log
    echo -e "\e[34;1m===========================================================================================================================\e[;m\n"
    echo -e "\e[34;1m 开始时间:$date_start \t\t\t 结束时间:$date_end  \t\t\t\t  \e[;m\n"
    echo -e "\e[34;1m 执行主机数:$shell_row \t\t\t 成功主机:`cat ./shell_log |grep "成功" |wc -l`  \t\t\t 失败主机:`cat ./shell_log |grep "失败" |wc -l`\t\t\t\e[;m\n"
    echo -e "\e[34;1m 失败主机记录 ↓\e[;m\n"
    cat ./shell_log |grep "失败"
    echo -e "\e[34;1m===========================================================================================================================\e[;m\n"
}


function shell_meminfo(){

        echo -e "\e[34;1m===========================================================================================================================\e[;m"
        echo -e "\e[34;1m IP地址\t\t\t总量\t\t已使用\t\t剩余\t\t共享\t\t缓冲\t\t缓存\t\t\n \e[;m"
for temp in `seq 1 $shell_row`
do
        take_ip=`cat $shell_list |head -n $temp |tail -n 1`
        Ip_Addr="$take_ip"
        echo -ne "\e[34;1m $take_ip\t\t \e[;m"
#       sleep 0.5
        ping -c 1 $Ip_Addr &>/dev/null
        if [ $? -eq 0 ]
        then
                ssh $shell_user@$take_ip free -h |grep "Mem:" |awk '{print $2 "\t\t" $3 "\t\t" $4 "\t\t" $5 "\t\t" $6 "\t\t" $7}'
        else
                echo -e "0M\t\t0M\t\t0M\t\t0M\t\t0M\t\t0M"
        fi
done
        echo -e "\e[34;1m===========================================================================================================================\e[;m"

}


#==============================================批量分发SSH密钥函数体-开始=======================================================================
function shell_ssh_keygen(){
      /usr/bin/expect << EOF
set timeout 5
spawn ssh-keygen -t rsa
expect {
"*save the key*" {send "\n";exp_continue}
"Enter passphrase*" {send "\n";exp_continue}
"*passphrase again:" {send "\n"}
}
expect eof
EOF
}

function shell_push_sshkey(){
local ssh_user=$1
local ssh_host=$2
local ssh_pass=$3
/usr/bin/expect << EOF
set timeout 10
spawn ssh-copy-id $ssh_user@$ssh_host
expect {
"(yes/no)" {send "yes\n"; exp_continue}
"password:" {send "$ssh_pass\n"}
"id_rsa.pub" {puts "(^_^)\n";exit 2\n}
}
expect eof 
EOF
}



function shell_expect(){

for temp in `seq 1 $shell_row`
do
        Ip_Addr=`cat $shell_list |head -n $temp |tail -n 1`
        shell_push_sshkey $shell_user $Ip_Addr $shell_pass
done
}
#==============================================批量分发SSH密钥函数体-结束=======================================================================


function shell_add_ip(){

        [ -e $shell_list ] || touch $shell_list
        echo "${comad[1]}" >>$shell_list
        if [ $? -ne 0 ]
        then
                echo -e "\e[34;1m 添加IP:${comad[1]}失败 \e[;m\n\n"
        exit 1
    fi
}

function shell_show_ip(){

        [ -e $shell_list ] || touch $shell_list
    echo -e "\e[34;1m====================================================================\n\n\e[;m"
    echo -e "\e[34;1m`cat $shell_list` \n\n\e[;m"
    echo -e "\e[34;1m总计:`cat $shell_list |wc -l `\t\t\t 操作列表:$shell_list\n\n\e[;m"
    echo -e "\e[34;1m====================================================================\n\n\e[;m"
}


function shell_drop_ip(){
    
#sed -i '/echo "${comad[1]}"/d' $shell_list
#   for i in `seq 1 $shell_row`
#   do
#       Ip_Addr=`cat $shell_list |head -n $temp |tail -n 1`
#       if [ "$Ip_Addr" == "$del_ip" ]
#       then
#   donei
#   cat $shell_list |grep -v ${comad[1]}  >/root/ip_list.tmp
#   rm -fr /root/ip_list
#   mv /root/ip_list.tmp /root/ip_list

    rm -fr $shell_list
    touch $shell_list

}


function shell_init(){

    touch $shell_list
    yum install -y expect
    [ $? -eq 0 ]
    echo -e "\e[34;1m 初始化成功... \e[m"
    
}

function shell_help(){

        echo -e "\e[34;1m====================================================================\n"

#        echo -e "注:如需要多线程执行,请自行在for语句下加入{} 并写入 wait 即可支持多线程,加快传输速度\n\n"

#   echo -e "\t\t\t[使用时请自行在/root/目录下创建ip_list文件并写入你要控制的主机]\n\n"

        echo -e "\t-shell [任意命令]            批量远程执行命令\n"
    
    echo -e "\t-init                    初始化\n"

        echo -e "\t-show                    显示控制列表\n"

        echo -e "\t-add                 添加一个被管理主机\n"
        
    echo -e "\t-del                 删除一个被管理主机\n"
    
    echo -e "\t-drop                    清空一个主机列表\n"

        echo -e "\t-keys                    本地生成密钥对\n"

        echo -e "\t-scpkeys             批量分发密钥对\n"

        echo -e "\t-upload [本地文件]  [传输到]        批量传输文件\n"

        echo -e "\t-mem                 批量统计主机内存使用\n"


        echo -e "====================================================================\n\n"
    echo -e " Powered by LyShark        瑞王保留所有权利\n"
        echo -e "====================================================================\e[;m"


}





case ${comad[0]} in

    "")
        shell_help
        exit 1
        ;;
    "-upload")  
        shell_upload
        exit 1
        ;;
    "-mem")
        shell_meminfo
        exit 1
        ;;
    "-keys")
        shell_ssh_keygen
        exit 1
        ;;
    "-scpkeys")
        shell_expect
        exit 1
        ;;
    "-add")
        shell_add_ip
        exit 1
        ;;
    "-show")
        shell_show_ip
        exit 1
        ;;
    "-drop")
        shell_drop_ip
        exit 1
        ;;
    "-init")
        shell_init
        exit 1
        ;;
    *)  shell_exp
        exit 1
        ;;

esac
扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄