python之telnetlib模块实现远程登录代码

在 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
[email protected] 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，也没看源码。

python之telnetlib模块实现远程登录代码

猜你喜欢