用於集群中管理機器上的密鑰對分發

版本：Python2.7

模塊：paramiko、multiprocessing

備註：註意在腳本70行左右獲取遠程服務器秘鑰指紋信息的時候不同的操作系統可能會有出入，如果使用本腳本務必在本地測試通過

# -*- coding:utf-8 -*-

import sys
reload(sys), sys.setdefaultencoding(‘utf-8‘)
import os
import time
import paramiko
import multiprocessing


# 此腳本用於進行秘鑰對分發
# 需要配合nmap使用，所以需要在linux環境下

# 密碼，不同服務器的不同密碼可寫在這一個列表中
password = [‘19820909‘]

# ip地址，多個寫在一個列表中，具體列表請自行循環生成
ip_list = [‘10.10.10.139‘,‘10.10.10.140‘]

# 端口,不能有多個端口,需要統一
port = 22

# 本地用戶的家目錄和用戶名 保持默認即執行腳本的用戶
load_username = os.environ["USER"]
load_userhome = os.environ["HOME"]

# 遠程服務器的用戶名，可以指定用戶名。默認和 load_username 一致
long_username = load_username
#long_username = "zouri"



try:
    # 獲取本地公鑰字符串
    keys_str = open("%s/.ssh/id_rsa.pub" % load_userhome).read().split("\n")[0]
except:
    print ‘Attempt command: \033[1;31mssh-keygen -t rsa -P "" -f ~/.ssh/id_rsa \033[0m ‘
    sys.exit(0)

def network_test(ip):
    # 使用 nmap 測試網絡的連通性
    sss = os.popen("nmap -sP %s|grep Host|grep up|awk ‘{print $3}‘" % ip)
    sun = sss.read()
    if ‘up‘ in sun:
        return True
    else:
        return False



def ssh_test(ip, password):
    sun = False

    s = paramiko.SSHClient()
    s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
        # 建立ssh鏈接
        s.connect(ip, port, long_username, password)

        a, b, c = s.exec_command("echo $HOME")
        long_range_home = b.read().replace("\n", "")

        s.exec_command("mkdir %s/.ssh && chmod 700 %s/.ssh" % (long_range_home, long_range_home))
        s.exec_command("echo %s >> %s/.ssh/authorized_keys" % (keys_str, long_range_home))
        s.exec_command("chmod 600 %s/.ssh/authorized_keys" % (long_range_home))
        s.close()
        sun = True

        # 獲取遠程服務器的 秘鑰指紋信息 寫入本地
        key_1 = os.popen("ssh-keyscan -p %d %s" % (port,ip)).read().split("\n")[0]
        know_file = open(load_userhome + "/.ssh/known_hosts","a+")
        know_file.write(key_1 + "\n")
        know_file.close()

    except "Authentication":
        pass
    finally:
        return sun

def ssh_run(ip):  # 執行函數
    sun = 1
    if not network_test(ip):
        print "\033[1;31m%s\033[0m is no,Maybe the Network" % ip
    else:
        for i_in in password:
            if ssh_test(ip, i_in):
                print "%s is ok,Keys has been added" % ip
                sun = 0
                break
        if sun == 1:
            print "\033[1;31m%s\033[0m is no,Maybe the Password" % ip


def sun_run(x):
    # 開啟多線程運行 x代表最多開多少線程
    sun_pool = multiprocessing.Pool(processes=x)
    for i in ip_list:
        sun_pool.apply_async(ssh_run, (i,))
    sun_pool.close()
    sun_pool.join()


if "__main__" == __name__:
    sun_time_01 = int(time.time())
    sun_run(30)
    sun_time_02 = int(time.time())
    print "共耗時%s秒" % str(sun_time_02 - sun_time_01)

本文出自 “馬走日” 博客，謝絕轉載！

SSH密鑰對批量認證Python腳本