多台计算机之间的ssh无密钥登录

小程序：扫一扫查出行
【扫一扫了解最新限行尾号】
复制小程序

在很多分布式系统中，我们最常遇到的一个问题是，需要在服务器集群上保证多台机器之间的SSH无密钥登录。以Hadoop为例，为了方便，我们需要在master和slaves之间配置密钥登录，这样我们启动Hadoop时就不用一个个输入密码了。问题是在配置无密钥登录过程中，如果集群机器很多，我们要将master机器上产生的公钥分发给所有的slaves, 分发一次就要输入一次密码，这种超费体力的活我们何不交给计算机自己来做呢？于是就有了这篇文章。

1. 准备工作

集群机器信息

本次实验使用Openstack云平台来创建测试集群，所有节点采用的配置为CPU: 1Core MEM: 512M Disk: 10G OS: CentOS 6.4. 各个节点的详细信息如下表:

主机名	IP地址
master	172.16.0.200
slave1	172.16.0.201
slave2	172.16.0.202
slave3	172.16.0.203
slave4	172.16.0.204

注意: 由Openstack创建的所有节点的初始登录用户名密码均相同(本实验中使用root用户)

maste机器配置

1. 登录到master机器, 打开文件ssh_config(/etc/ssh_config)，在ssh_config文件中找到Host *这一行，在下面行中加入 StrictHostKeyChecking no 如下图:

2. 生成ssh密钥，命令: ssh-keygen -t rsa。不用管提示，一路回车即可。最后会生成两个密钥文件id_rsa, id_rsa.pub, 分别为私钥和公钥。公钥是我们待会要发送给各个slave节点的。

安装pexpect(Python 模块)

在配置无密钥登录的过程中，最重要的一步是将master生成id_rsa.pub文件拷贝的各个slave节点上，并追加到各个slave节点的authorized_keys文件中(上面两步操作也可以使用ssh提供的ssh-copy-id命令，本文用的就是这种方法)。但在拷贝过程中会让我们输入slave节点的密码。这也是阻碍我们自动化配置无密钥登录的最大障碍。为了解决这个问题，本文采用python的pexpect模块，该模块能够让程序与shell命令进行交互。 pexpect下载地址: https://pypi.python.org/pypi/pexpect/ , 下载完成后，解压。解压后在目录下运行：python ./setup.py install(root权限) 安装完成以后，就可以进行我们下一步的工作了。

另外我们也可以直接安装expect工具。具体的请百度 expect。

2. 无密钥登录程序

下面两个脚本就是本文中实现SSH批量无密钥登录的程序，free_login.sh配置slave的用户名密码，并调用free_login.py, free_login.py负责将master的公钥拷贝到各个slave机器上。代码如下:

free_login.sh

#!/bin/bash#slave节点的user_name和password
user_name="root"
user_pass=""#命令参数为保存所有slave节点ip地址文件,一个ip一行
if [ $# != ]; then
echo "need ip list file"
exit
fi
ip_list_file=$# first invoke free_python.py to setup passphraseless ssh to
# all hosts in ip list file `./free_login.py ${ip_list_file} ${user_name} ${user_pass}`

free_login.py

#!/usr/bin/pythonimport os
import pexpect
import sysfile_path = sys.argv[]
user_name = sys.argv[]
pwd = sys.argv[]
file = open(file_path,"r")
lines = file.readlines()
for line in lines:
line = line.strip('\n')
print "set free login on host:%s" % line
child = pexpect.spawn('ssh-copy-id -i /root/.ssh/id_rsa.pub ' + user_name + '@' + line)
message = ''
try:
i = child.expect(['[Pp]assword:','continue connecting (yes/no?)'])
if i == :
child.sendline(pwd)
elif i == :
child.sendline('yes')
child.expect('[Pp]assword')
child.sendline(pwd)
else:
pass
except pexpect.EOF:
message = child.read()
child.close()
else:
message = child.read()
child.expect(pexpect.EOF)
child.close()