#!/usr/bin/expect
if { $argc<2 } {
    send_user "usage: $argv0 <host file> <cmd file> /n"
    exit
}
 
# 機器列表數據格式:  IP  端口  舊密碼  新密碼
set hostfile    [ open [lindex $argv 0] ]
# 命令列表數據格式:  一條命令一行
set cmdfile    [ open [lindex $argv 1] ]
 
# 數據文件分割符,默認為空格
set part "/ "
 
# 過濾關鍵字
set key_password "password:/ "
set key_init "/(yes/no/)/?/ "
set key_confirm "'yes'/ or/ 'no':/ "
set key_ps "*]#/ "
set key_newpassword "UNIX password:/ "
set timeout 30
 
log_file ./exprct.log
match_max 20480
 
while {[gets $hostfile _hosts_] >= 0} {
    set hosts [string trim $_hosts_]
    set str_index [string first $part $hosts]
    set host [string trim [string range $hosts 0 $str_index]]
    set temp [string trim [string range $hosts [expr $str_index + 1] [string length $hosts]]]
    set str_index [string first $part $temp]
 
    if { $str_index == -1 } {
        set port 22
        set pass $temp
        set newpass $temp
    } else {
        set port [string trim [string range $temp 0 $str_index]]
        set temp_pass [string trim [string range $temp [expr $str_index + 1] [string length $temp]]]
        set str_index [string first $part $temp_pass]
        set pass [string trim [string range $temp_pass 0 $str_index]]
        set newpass [string trim [string range $temp_pass [expr $str_index + 1] [string length $temp_pass]]]
    }
 
    spawn ssh -p $port $host
    while {1} {
        expect {
            "$key_password" {
                send "$pass/r"
            }
            "$key_init" {
                send "yes/r"
            }
            "$key_confirm" {
                send "yes/r"
            }
            "$key_ps" {
                while {[gets $cmdfile cmd] >= 0} {
                    send "$cmd/r"
                    expect {
                        "$key_ps" {
                            continue
                        }
                        "$key_newpassword" {
                            send "$newpass/r"
                            expect "$key_newpassword" {
                                send "$newpass/r"
                                expect "$key_ps"
                                continue
                            }
                        }
                    }
                }
                seek $cmdfile 0 start
                send_user "/r"
                break
            }
            timeout {
                puts "$host timeout/n"
                break
            }
        }
    }
    send "exit/r"
    close
    wait
}
 
close $hostfile
close $cmdfile
 
exit