欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python远程登录代码

程序员文章站 2022-10-10 08:30:37
在 python 中有一个 telnetlib,它的作用就是建立一个通到主机的 telnet连线实体, 然后向主机传送命令...

在 python 中有一个 telnetlib,它的作用就是建立一个通到主机的 telnet连线实体, 然后向主机传送命令 (就像用键盘输入一样 )并从该连线接收数据。利用它, 我们可以把示范 1的所有内容从 "人 -机 '交流变成'机 -机 '交流,这样也可以做到处理 pop3 邮箱的工作。不过既然我们已经试过了 pop3,这一次可以试用真的 telnet 埠 23 做些好玩的东西。 

 # telnetdo.py   
 #!/usr/bin/env python   
   
 def telnetdo(host=none, user=none, pass=none, command=none): #定义一个函数, 这将要用它会很容易   
  import telnetlib, sys   
   
  if  not  host:              #如果没有给出所要的资料,则要求用户输入           
   try:   
     host  = sys.argv[1]         #记得吧, 序列是从 0开始数的,而sys.argv[0]会是你程式的名字   
     user  = sys.argv[2]  
     pass  = sys.argv[3]  
     command  = sys.argv[4]  
   except:  
     print  "usage: telnetdo.py host user pass 'command'"  
     return  
   
  msg  = ['debug  mesages:\n']       #这个用来存起所有从主机传回的讯息, 作除错时很有用   
   
  tn  = telnetlib.telnet()         #准备一个 telnet 连线的实体   
  try:  
   tn.open(host)              #连接端绑定到主机 host 去   
  except:  
   print  "cannot open host"  
   return  
   
  msg.append(tn.expect(['login:'],5))    #等待主机传回含有 'login:'字符的讯息,等候时限为 5秒   
  tn.write(user+'\n')            #向主机送出字串 user  + '\n',如 user 是  
                       # 'pcheung' 则等于 'pcheung\n'  
  if  pass:                 #就像是在键盘打入一样。   
   msg.append(tn.expect(['password:'],5)) #如果有 password 要打的话就送出密码字串,   
   tn.write(pass+'\n')           #但首先要等主机传回含有 'password:'字样的讯息   
   
  msg.append(tn.expect([user],5))      #因为通常登入后主机会显示出登入者名称,我们在主机回应中找这   
                       #样的字符,如有的话则代表登入成功了   
  tn.write(command+'\n')           #向主机发出指令   
  msg.append(tn.expect(['%'],5))       #等 5秒,如果程式完成了一般我们会收到   
                       # shell prompt 吧,假设为 '%'  
  tn.close()                 #关闭连线   
  del  tn  
  return  msg[len(msg)-1][2]        #把收到的讯息通通传回去。   
                       #(注意 msg 中第 2个元素才是真的讯息,   
                       #其他是附加资讯。   
   44 if __name__  == '__main__'         #这是 python 常用的技巧:如果 telnetdo.py 程式  
                       #是从 command prompt  
                       #引发的话则 __name__ 的内容为 __main__,相反  
                       #如果是从别的程式用 import telnetdo 的话则  
                       # __name__ 会变成 'telnetdo'  
  print  telnetdo()             #这样写的好处是从此 telnetdo 会成为你的扩展  
                       #模组,你可以在别的程式中   
                        #用telnetdo.telnetdo(host,user,pass,command)来调用它! 

这个程式用法如下:

> chmod +x telnetdo.py
> telnetdo.py 'somehost' 'glace' 'xxxxxx' 'ls   -lf'

(0, <sre_match object   at   200f75a8>, '\015\012\015\012linux
(somehost)\015\012\015\015\012\015login: ')
(0, <sre_match object   at   20124848>, 'password:')
(0, <sre_match object   at   20103e08>, '\015\012yup release 2.6 
somehost\015\012last login: wed   mar    6 18:21:01 gmt 2002 by
unknown@xxx.xxx.xxx.xxxyou have   mail\015\012somehost:glace%')

total 320
-rw-r--r--    1 glace    user        139788 feb    8 17:54 pqr2.1.txt
drwxr-xr-x    3 glace    user          4096 feb   10 16:45 mytts/
drwxr-xr-x    3 glace    user          4096 jan   29 19:03 sample/
drwxr-xr-x    2 glace    user          4096 jan    6 16:38 tex/
drwxr-xr-x    2 glace    user          4096 sep    5  2001 tmp/
drwxr-xr-x    2 glace    user            29 feb   23  2001 tools/
drwxr-xr-x    2 glace    user            26 feb    6 18:43 trash/
somehost:glace%

可以看到执行的结果和一些附加的资讯。这就是远端执行程式了。就算没有 rsh,照用可也。 哈,很方便吧。不过你应该留意到了程式执行时只等候了 5秒,就是说如果你要向主机发出像'find . -name xxx   -print'这样的命令应该等不到执行完这个 telnet session 就会被关闭了。不过仔细想一下,这要紧吗?我们现在所能做到的和真正人手 telnet 的差别并不大,想一想你会怎样解决长时间执行的问题?没错,就是 'nohup'和背景作业了。 就是说只要把程式呼叫改成: telnetdo.py 'apocal' 'pcheung' 'xxxxxx' 'nohup   myprogram_or_scrīpt&' 就行了。如此一来,就算对方主机的 shell prompt 是 '>'或是 '>>>'都没有关系了。

(注意安全性并非是这类范例程式的着重点, 因此并不建议在实际工作中用它.) 

python中telnetlib模块的使用

python下能支持telnet的模块telnetlib是内置模块,直接import就可以了,其基本的使用方法也是比较简单的。

#encoding=utf-8 
 
def do_telnet(host, username, password, finish, commands): 
  import telnetlib 
  '''''telnet远程登录:windows客户端连接linux服务器''' 
  
  # 连接telnet服务器 
  tn = telnetlib.telnet(host, port=23, timeout=10) 
  tn.set_debuglevel(2) 
    
  # 输入登录用户名 
  tn.read_until('login: ') 
  tn.write(username + '\n') 
   
  # 输入登录密码 
  tn.read_until('password: ') 
  tn.write(password + '\n') 
    
  # 登录完毕后执行命令 
  tn.read_until(finish) 
  for command in commands: 
    tn.write('%s\n' % command) 
   
  #执行完毕后,终止telnet连接(或输入exit退出) 
  tn.read_until(finish) 
  tn.close() # tn.write('exit\n') 
 
if __name__=='__main__': 
   # 配置选项 
  host = '10.255.254.205' # telnet服务器ip 
  username = 'administrator'  # 登录用户名 
  password = 'dell1950' # 登录密码 
  finish = ':~$ '   # 命令提示符 
  commands = ['echo "test"'] 
  do_telnet(host, username, password, finish, commands) 

其中port和timeout是可选的参数,而timeout的只是在初始化socket连接时起作用,而一旦连接成功后如果出现等待那就不会起作用了,比如使用read_until方式获取内容时返回的内容与指定的内容没有吻合,那么就会造成提示等待的情况,这时timeout是不会起作用的,而这个socket连接会一直保持着,永生不死。

那么如何解决这个问题呢,其实还有一种比较原始的方法,就是使用sleep方法来代替read_until方法,这样就不会出现种情况,因为到点就会自己输入,最多也就是最后得不到想要的结果,但是这个方式很不稳定,兼容性也不好;另一种方法是使用线程来启动这个函数,然后对子线程进行超时设置,这样就可以达到间接控制这个telnet连接的目的了。

import threading 
pars = replace_db_keyworlds(vars_dict, pars) 
configs = pars.split(r'@') 
host = configs[0].encode() 
user = configs[1] 
passwd = configs[2] 
finish = configs[3] 
commands = configs[4].split(r'\n') 
th1 = threading.thread(target=do_telnet, args=(host.encode('utf-8'), user.encode('utf-8'), passwd.encode('utf-8'), finish.encode('utf-8'), commands)) 
th1.start() 
th1.join(20) ##20秒超时时间 

还有一个需要注意的是,传递给telnet方法的字符串都会被解一次码,所以如果你传递过去需要write的字符串是已经解码的unicode的话,那么就会报错的,所以在传递发送的字符串之前还是先编成utf-8为妥,其它字符不知道支持不,我只试了utf-8,也没看源码。

此外,貌似还有一个pexpect的第三方模块可以支持telnet等一系列的协议连接,并支持交互式的通信,只是这个模块够用就没学习了,这里先备注一下。