背景
机房多，机器多，而且没有完善的运维工具的时候，经常要登录不通机房不同网段的服务器进行问题的排查。而且密码一多也不是很方便管理。
脚本
python 的pexpect 真的是很好用啊，相比shell 有个非常大的好处在于，用shell 的pexpect 登录服务器，在跳板机上ps 一下会发现里面的密码竟然是明文，比如这样的:
mysql 22506 22502 0 00:21 ? 00:00:00 sshd: mysql@pts/476
mysql 23695 23693 0 00:42 pts/476 00:00:00 expect -c ?spawn ssh -q barney@10.30.1.1?expect {?"yes/no" {send "yes\r";exp_continue}?"assword:" {send "passwd\r"}?}?expect "$*"?send "su -\n"?expect "assword:"?send "passwd\n"?interact
所以不是太安全，还是用现成的模块吧。
#! /usr/bin/python
# -*- coding: utf-8 -*-
#auther barney add by 20151110
import pexpect
import struct,fcntl,sys
import termios
input_ip=sys.argv[1]
length=len(input_ip.split('.'))
##shangha password
sh_passwd='shpasswd'
sh_root_pass='shpasswd'
sh_location='\033[5;32;40m######上海IDC...######\033[0m'
##tianjin password
tj_passwd='tjpasswd'
tj_root_pass='tjpasswd'
tj_location='\033[5;32;40m######天津IDC...######\033[0m'
##suzhou password
sz_passwd='sjpasswd'
sz_root_pass='sjpasswd'
sz_location='\033[5;32;40m######苏州IDC...######\033[0m'
def getwinsize(): #这个模块主要用于记录登录前的session的窗口大小，不然会发生登录后窗口大小变小的问题
if 'TIOCGWINSZ' in dir(termios):
TIOCGWINSZ = termios.TIOCGWINSZ
else:
TIOCGWINSZ = 1074295912L
s = struct.pack('HHHH', 0, 0, 0, 0)
x = fcntl.ioctl(sys.stdout.fileno(), TIOCGWINSZ, s)
return struct.unpack('HHHH', x)[0:2]
def ipinfo():
if network in ('1','2','6','7','8'):
return (ip,sh_passwd,sh_root_pass,sh_location)
elif network in ('30','31','32','33'):
return (ip,tj_passwd,tj_root_pass,tj_location)
elif network in ('4','11','71','72'):
return (ip,sz_passwd,sz_root_pass,sz_location)
else:
print '请输入正确的2位 IP'
def ssh_host(): #因为都是192.168 开头的私有地址，所以就输入后2位就行了
try:
print"%s"%location
child=pexpect.spawn('ssh -p 56000 barney@%s'%ip)
child.expect(['密码','(?i)password:'])
child.sendline('%s'%passwd)
child.sendline('su -')
child.expect(['密码','(?i)password:'])
child.sendline('%s'%root_pass)
child.sendline('clear')
winsize = getwinsize();
child.setwinsize(winsize[0], winsize[1])
child.interact()
child.close()
except OSError:
print 'Login Ended'
if length == 2:
ip='192.168.'+input_ip
network=input_ip.split('.')[0]
ip,passwd,root_pass,location=ipinfo()
ssh_host()
else:
print "IP地址错误"
