oneinstack/tools/mabs.sh

#!/bin/bash
# Author:  yeho <lj2007331 AT gmail.com>
# Blog:  http://blog.linuxeye.com

######################  proc defination  ########################
# ignore rule
ignore_init()
{
    # ignore password
    array_ignore_pwd_length=0
    if [ -f ./ignore_pwd ]; then
        while read IGNORE_PWD
        do
            array_ignore_pwd[$array_ignore_pwd_length]=$IGNORE_PWD
            let array_ignore_pwd_length=$array_ignore_pwd_length+1
        done < ./ignore_pwd
    fi

    # ignore ip address
    array_ignore_ip_length=0
    if [ -f ./ignore_ip ]; then
        while read IGNORE_IP
        do
            array_ignore_ip[$array_ignore_ip_length]=$IGNORE_IP
            let array_ignore_ip_length=$array_ignore_ip_length+1
        done < ./ignore_ip
    fi
}

show_version()
{
    echo "version: 1.0"
    echo "updated date: 2014-06-08"
}

show_usage()
{
    echo -e "`printf %-16s "Usage: $0"` [-h|--help]"
        echo -e "`printf %-16s ` [-v|-V|--version]"
        echo -e "`printf %-16s ` [-l|--iplist ... ]"
        echo -e "`printf %-16s ` [-c|--config ... ]"
        echo -e "`printf %-16s ` [-t|--sshtimeout ... ]"
        echo -e "`printf %-16s ` [-T|--fttimeout ... ]"
        echo -e "`printf %-16s ` [-L|--bwlimit ... ]"
        echo -e "`printf %-16s ` [-n|--ignore]"
}

IPLIST="iplist.txt"
CONFIG_FILE="config.txt"
IGNRFLAG="noignr"
SSHTIMEOUT=100
SCPTIMEOUT=2000
BWLIMIT=1024000
[ ! -e 'logs' ] && mkdir logs

TEMP=`getopt -o hvVl:c:t:T:L:n --long help,version,iplist:,config:,sshtimeout:,fttimeout:,bwlimit:,log:,ignore -- "$@" 2>/dev/null`

[ $? != 0 ] && echo -e "\033[31mERROR: unknown argument! \033[0m\n" && show_usage && exit 1

eval set -- "$TEMP"

while :; do
    [ -z "$1" ] && break;
    case "$1" in
        -h|--help)
            show_usage; exit 0
            ;;
        -v|-V|--version)
            show_version; exit 0
            ;;
        -l|--iplist)
            IPLIST=$2; shift 2
            ;;
        -c|--config)
            CONFIG_FILE=$2; shift 2
            ;;
        -t|--sshtimeout)
            SSHTIMEOUT=$2; shift 2
            ;;
        -T|--fttimeout)
            SCPTIMEOUT=$2; shift 2
            ;;
        -L|--bwlimit)
            BWLIMIT=$2; shift 2
            ;;
        --log)
            LOG_FILE=$2; shift 2
            ;;
        -n|--ignore)
            IGNRFLAG="ignr"; shift
            ;;
        --)
            shift
            ;;
        *)
            echo -e "\033[31mERROR: unknown argument! \033[0m\n" && show_usage && exit 1
            ;;
    esac
done

################  main  #######################
BEGINDATETIME=`date "+%F %T"`
[ ! -f $IPLIST ] && echo -e "\033[31mERROR: iplist \"$IPLIST\" not exists, please check! \033[0m\n" && exit 1

[ ! -f $CONFIG_FILE ] && echo -e "\033[31mERROR: config \"$CONFIG_FILE\" not exists, please check! \033[0m\n" && exit 1

IP_count=$(egrep -v '^#|^$' $IPLIST|wc -l)
IP_init=1
while [[ $IP_init -le $IP_count ]]
do
        egrep -v '^#|^$' $IPLIST | sed -n "$IP_init,$(expr $IP_init + 50)p" > $IPLIST.tmp

    IPSEQ=0

    while read IP PORT USER PASSWD PASSWD_2ND PASSWD_3RD PASSWD_4TH OTHERS
       # while read Line
        do
        [ -z "`echo $IP | grep -E '^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}|CNS'`" ] && continue
            if [ "`python ./ckssh.py $IP $PORT`" == 'no' ];then
            [ ! -e ipnologin.txt ] && > ipnologin.txt
            [ -z "`grep $IP ipnologin.txt | grep $(date +%F)`" ] && echo "`date +%F_%H%M` $IP" >> ipnologin.txt
            continue
            fi

        #[ -e "~/.ssh/known_hosts" ] && grep $IP ~/.ssh/known_hosts | sed -i "/$IP/d" ~/.ssh/known_hosts

        let IPSEQ=$IPSEQ+1

            if [ $IGNRFLAG == "ignr" ]; then
                    ignore_init
                    ignored_flag=0

                    i=0
                    while [ $i -lt $array_ignore_pwd_length ]
                    do
                            [ ${PASSWD}x == ${array_ignore_pwd[$i]}x ] && ignored_flag=1 && break
                            let i=$i+1
                    done

                    [ $ignored_flag -eq 1 ] && continue

                    j=0
                    while [ $j -lt $array_ignore_ip_length ]
                    do
                            [ ${IP}x == ${array_ignore_ip[$j]}x ] && ignored_flag=1 && break
                            let j=$j+1
                    done

                    [ $ignored_flag -eq 1 ] && continue
            fi

        PASSWD_USE=$PASSWD

                IPcode=$(echo "ibase=16;$(echo "$IP" | xxd -ps -u)"|bc|tr -d '\\'|tr -d '\n')
                Portcode=$(echo "ibase=16;$(echo "$PORT" | xxd -ps -u)"|bc|tr -d '\\'|tr -d '\n')
                #USER=$USER
                PWcode=$(echo "ibase=16;$(echo "$PASSWD_USE" | xxd -ps -u)"|bc|tr -d '\\'|tr -d '\n')
                Othercode=$(echo "ibase=16;$(echo "$OTHERS" | xxd -ps -u)"|bc|tr -d '\\'|tr -d '\n')
                #echo $IPcode $Portcode $USER $PWcode $CONFIG_FILE $SSHTIMEOUT $SCPTIMEOUT $BWLIMIT $Othercode
        ./thread.sh $IPcode $Portcode $USER $PWcode $CONFIG_FILE $SSHTIMEOUT $SCPTIMEOUT $BWLIMIT $Othercode | tee logs/$IP.log &
        done < $IPLIST.tmp
        sleep 3
        IP_init=$(expr $IP_init + 50)
done

ENDDATETIME=`date "+%F %T"`

echo "$BEGINDATETIME -- $ENDDATETIME"
echo "$0 $* --excutes over!"

exit 0