Ubuntu集群下利用Shell脚本进行SSH免密码登陆
如果我们有一个服务器的集群,在这个集群下面需要为集群的每一个节点实现SSH的免密码,其实是一件非常繁琐的工作。此时,我们可以借助Shell脚本方便的实现。
在开始看具体实现之前,我们先来看一下过程中的要点。分别为ssh-copy-id命令和expect命令
ssh-copy-id
ssh-copy-id命令可以把本地主机的公钥复制到远程主机的authorized_keys
文件上,ssh-copy-id命令也会给远程主机的用户主目录(home)和~/.ssh
, 和~/.ssh/authorized_keys
设置合适的权限。
expect
expect是一个免费的编程工具语言,用来实现自动和交互式任务进行通信,而无需人的干预。expect是不断发展的,随着时间的流逝,其功能越来越强大,已经成为系统管理员的的一个强大助手。expect需要Tcl编程语言的支持,要在系统上运行expect必须首先安装Tcl。
我们通过Shell可以实现简单的控制流功能,如:循环、判断等。但是对于需要交互的场合则必须通过人工来干预,有时候我们可能会需要实现和交互程序如telnet服务器等进行交互的功能。而expect就使用来实现这种功能的工具。
具体的实现
首先,我们需要在Ubuntu下面安装expect。
expect在ubuntu下的安装
使用以下代码检测expect是否已经安装
ls /usr/bin | grep expect
如果显示为空,则使用以下命令安装
sudo apt-get install tcl tk expect
脚本代码
下面给出脚本代码,然后稍作说明。在执行下面的脚本之前必须在本机通过ssh-keygen -t rsa
指令生成秘钥。
#!/bin/sh
SERVERS="localhost anode1 anode2 anode3"
PASSWORD=123456
auto_ssh_copy_id() {
expect -c "set timeout -1;
spawn ssh-copy-id $1;
expect {
*(yes/no)* {send -- yes\r;exp_continue;}
*assword* {send -- $2\r;exp_continue;}
eof {exit 0;}
}";
}
ssh_copy_id_to_all() {
for SERVER in $SERVERS
do
auto_ssh_copy_id $SERVER $PASSWORD
done
}
ssh_copy_id_to_all
$SERVERS里面存放了所有需要进行ssh免密码登录授权的机器列表,用空格分开。
另外的一个重点就是auto_ssh_copy_id
函数这个函数负责对一台机器进行ssh免密码登录授权。它通过expect -c
指令在命令行里面执行了一串命令。
spawn相当于一个程序壳,通过它,我们在expect中执行了ssh-copy-id指令。后面的参数$1则是调用方传入的需要进行ssh免密码登录的机器名。
后续的expect块中则是对ssh-copy-id指令可能产生的响应进行匹配,并决定后续的动作。如果响应中包含”(yes/no)”则输出yes+回车。后面的exp_continue
指令表示继续进行下一个结果的匹配。如果响应中包含”assword”则输出密码+回车。
结论
使用上面的代码,我们方便的实现了本机到集群各节点间的ssh登录,同时也体会到了Shell脚本的强大之处。有时间好好学习研究一番。