Linux/Unix shell內嵌expect自動配置多臺主機SSH互信

deepalley發表於2014-03-22
Linux
在自動化測試，或者Oracle RAC安裝過程中，需要配置主機之間SSH 無密碼訪問，當有多臺主機時，配置起來很麻煩，網上的很多類似指令碼拿到本機有問題，所以自己寫了一個：
   

  1. #!/bin/bash

  3. NUM_OF_NODES= 
  4. NODE1="" 
  5. NODE2=""
  6. NODE3=""
  7. NODE4=""
  8. NODE5=""

  10. USER=root
  11. EXPECT=/usr/bin/expect
  12. PASSWD=""
  13. #USER_PROMPT="*$ "
  14. USER_PROMPT="*# "

  16. if [ "x${NODE1}" == "x" -o "x${USER}" == "x" -o "x${PASSWD}" == "x" ]; then
  17.     echo ""
  18.     echo "Please set the NODE INFO, USER and PASSWD"
  19.     echo "then $0 to start..."
  20.     exit 1
  21. fi

  23. declare -i l_i=1
  24. while [ $l_i -le $NUM_OF_NODES ]
  25. do
  26.     eval l_current_node=\$NODE$l_i
  27.     
  28.     $EXPECT <<EOF
  29.     spawn ssh $USER@$l_current_node 
  30.     expect "*(yes/no)?*" {
  31.         send -- "yes\r"
  32.         expect "*?assword:*"
  33.         send -- "$PASSWD\r"
  34.     } "*?assword:*" {send -- "$PASSWD\r"}
  35.     expect "$USER_PROMPT"
  36.     send -- "ssh-keygen -t rsa -q -f ~/.ssh/id_rsa -P '' \r"
  37.     expect "*Overwrite (yes/no)? " {
  38.         send -- "yes\r"
  39.     } "$USER_PROMPT" {send -- "\r"}
  40.     expect "$USER_PROMPT"
  41.     send -- "cat ~/.ssh/id_rsa.pub | ssh $USER@$NODE1 'cat - >> ~/.ssh/authorized_keys' \r"
  42.     expect "*(yes/no)?*" {
  43.         send -- "yes\r"
  44.         expect "*?assword:*" 
  45.         send -- "$PASSWD\r"
  46.     } "*?assword:*" {send -- "$PASSWD\r"}
  47.     expect "$USER_PROMPT"
  48.     send -- "exit\r"
  49. EOF
  50.     ((l_i++))
  51. done

  53. declare -i l_n=1
  54. while [ $l_n -le $NUM_OF_NODES ]
  55. do
  56.     eval l_current_node=\$NODE$l_n
  57.     $EXPECT <<EOF

  59.     spawn ssh $USER@$NODE1
  60.     expect "*?assword:*" {
  61.         send -- "$PASSWD\r"
  62.         expect "$USER_PROMPT"
  63.     } "$USER_PROMPT" {send -- "scp ~/.ssh/authorized_keys $l_current_node:~/.ssh/ \r"}
  64.     expect "*?assword:*"
  65.     send -- "$PASSWD\r"
  66.     expect "$USER_PROMPT"
  67.     send -- "exit\r"
  68. EOF
  69.     ((l_n++))
  70. done

來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/20482996/viewspace-1127298/，如需轉載，請註明出處，否則將追究法律責任。

相關文章