当前位置:首页 >> 脚本专栏

python远程登录代码

在 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等一系列的协议连接,并支持交互式的通信,只是这个模块够用就没学习了,这里先备注一下。