python远程登录代码
作者:mdxy-dxy 时间:2022-09-12 15:48:15
在 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等一系列的协议连接,并支持交互式的通信,只是这个模块够用就没学习了,这里先备注一下。