#!/usr/bin/expect --

if { [llength $argv] < 4 } {
  puts "Usage: $argv0 ip user passwd port commands timeout"
  exit 1
}

match_max 600000

set ipcode [lindex $argv 0]
set ip [exec dc -e $ipcode]
set user [lindex $argv 1]
set passwdcode [lindex $argv 2]
set passwd [exec dc -e $passwdcode]
set portcode [lindex $argv 3]
set port [exec dc -e $portcode]
set commands [lindex $argv 4]
set timeoutflag [lindex $argv 5]

set yesnoflag 0
set timeout $timeoutflag


for {} {1} {} {
  # for is only used to retry when "Interrupted system call" occured

  spawn /usr/bin/ssh -o GSSAPIAuthentication=no -q -l$user -p$port $ip

  expect  {

    "assword:" {
      send "$passwd\r"
      break;
    }

    "yes/no)?" {
      set yesnoflag 1
      send "yes\r"
      break;
    }

    "FATAL" {
      puts "\nCONNECTERROR: $ip occur FATAL ERROR!!!\n"
      exit 1
    }

    timeout {
      puts "\nCONNECTERROR: $ip Logon timeout!!!\n"
      exit 1
    }

    "No route to host" {
      puts "\nCONNECTERROR: $ip No route to host!!!\n"
      exit 1
    }

    "Connection Refused" {
      puts "\nCONNECTERROR: $ip Connection Refused!!!\n"
      exit 1
    }

    "Connection refused" {
      puts "\nCONNECTERROR: $ip Connection Refused!!!\n"
      exit 1
    }

    "Host key verification failed" {
      puts "\nCONNECTERROR: $ip Host key verification failed!!!\n"
      exit 1
    }

    "Illegal host key" {
      puts "\nCONNECTERROR: $ip Illegal host key!!!\n"
      exit 1
    }

    "Connection Timed Out" {
      puts "\nCONNECTERROR: $ip Logon timeout!!!\n"
      exit 1
    }

    "Interrupted system call" {
      puts "\n$ip Interrupted system call!!!\n"
    }
  }
}

if { $yesnoflag == 1 } {
  expect {
    "assword:" {
      send "$passwd\r"
    }

    "yes/no" {
      set yesnoflag 2
      send "yes\r"
    }
  }
}

if { $yesnoflag == 2 } {
  expect {
    "assword:" {
      send "$passwd\r"
    }
  }
}

expect {
  "@" {send "$commands \r"}
  "assword:" {
    send "$passwd\r"
    puts "\nPASSWORDERROR: $ip Password error!!!\n"
    exit 1
  }
}

expect {
  "@" {send "sleep 1\r"}
}

expect {
  "@" {send "exit\r"}
}

expect eof {
  puts "OK_SSH: $ip\n"
  exit 0;
}