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