【ssh】通过跳板机自动登录目标服务器

【ssh】

 

整理自：

 

1.使用环境
需要通过跳板机登录服务器.
将脚本放在跳板机的"~/bin/ssh.py"目录

2.脚本内容:

1.      #!/usr/bin/python

2.      #-*- coding: UTF-8 -*-

3.       

4.      import pexpect

5.      import os,os.path

6.      import stat

7.      import getpass

8.      import sys

9.      import getopt

10.   import socket

11.   from ConfigParser import ConfigParser

12.   #import subprocess

13.    

14.   def usage():

15.       print '\nUsage: ssh.py hostname'

16.       sys.exit(0)

17.    

18.   # 定义用户名和密码文件

19.   def load_config():

20.       cfg_path = os.path.join(os.environ["HOME"], ".ssh.cfg")

21.       cfg_temp = '''[connects]

22.   user   =

23.   id_rsa =

24.   '''

25.       if not os.path.exists(cfg_path):

26.           open(cfg_path, 'w').write(cfg_temp)

27.   #    os.chmod(cfg_path, stat.S_IREAD)

28.       cfg = ConfigParser()

29.       cfg.read(cfg_path)

30.       user   = cfg.get('connects', 'user')

31.       id_rsa = cfg.get('connects', 'id_rsa', 'True')

32.       return (user,id_rsa)

33.    

34.   def ssh_login (user, host, id_rsa):

35.       ssh_newkey = 'Are you sure you want to continue connecting'

36.       try:

37.           child = pexpect.spawn('ssh -l %s %s '%(user, host))

38.           child.expect([pexpect.TIMEOUT, ssh_newkey, 'password: ', "id_rsa': "])

39.           child.sendline(id_rsa)

40.           # 预知chr(26)的含义,请查看ASCII码对照表,字符一列.

41.           child.interact(chr(26))

42.           print "\n"

43.       except:

44.           print "interact error"

45.           sys.exit(0)

46.    

47.   def main ():

48.       if len(sys.argv) == 1:

49.           load_config()

50.           usage()

51.    

52.       # load file

53.       user,id_rsa = load_config()

54.    

55.       if user == '':

56.           usage()

57.       elif id_rsa == '':

58.           usage()

59.       else:

60.           pass

61.       #host    = raw_input('hostname: ')

62.       host = sys.argv[1]

63.   #    print host

64.       # use `ping` to check if the host is available now

65.       # 用ping的方式测试主机是否存在

66.       try:

67.           sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

68.           sock.settimeout(3)

69.           sock.connect((host, 22))

70.           sock.close()

71.       except:

72.           print 'ping: unknown host %s' % (host)

73.           sys.exit(0)

74.       # ssh

75.       ssh_login(user, host, id_rsa)

76.    

77.   if __name__ == '__main__':

78.       try:

79.           main()

80.       except Exception, e:

81.           print str(e)

82.           os._exit(1)

 

 

使用方法
1.直接执行ssh.py不跟主机名,则生成~/.ssh.cfg配置文件
2.修改.ssh.py添加用户名和密码
3.执行ssh.py hostname,则自动登录主机
4.如果要退出主机,不需要输入exit,直接按CTRL+Z即可