Ubuntu集群下利用Shell脚本进行SSH免密码登陆

集群下利用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脚本的强大之处。有时间好好学习研究一番。

下面关于SSH相关的文章您也可能喜欢，不妨参考下：