使用Python登录网关设备执行命令

古城微笑少年丶 2023-02-16 06:43 232阅读 0赞

一、概述

最近发现办公室有一个感测器网关，偶尔会掉线，重启之后，就正常了。

手动操作过程如下：

  1. # telnet 10.212.82.90
  2. Trying 10.212.82.90...
  3. Connected to 10.212.82.90.
  4. Escape character is '^]'.
  5. (none) login: root
  6. Password: 
  7. Please make sure that what you're doing!  If you aren't administrator, please Don't do anything! :)
  8. # ps
  9.   PID USER       VSZ STAT COMMAND
  10.     1 root      1124 S    init       
  11.     2 root         0 SW   [kthreadd]
  12.     3 root         0 SW   [ksoftirqd/0]
  13.     4 root      3888 S    /usr/app/bin/app

说明：先telnet进去，输入用户名和密码之后，执行ps命令。如果有/usr/app/bin/app进程，说明设备正常，否则就需要重启。

但是手动操作比较麻烦，需要使用python登录网关设备，检测是否正常。

二、Telnet对象

Telnet.read_until(expected, timeout=None) ： #读取连接服务器后显示的内容，直到遇到同expected相同的字节串。或者等待时间大于timeout时直接向下运行。

Telnet.read_very_eager() ： 读取从上次IO阻断到现在所有的内容，返回的是字节串，需要进行decode()编码。如果连接关闭或者没有可用数据时会抛出EOFError，如果没有其他可用的数据，返回的是b””，除非在IAC中间，否则不会阻碍。

Telnet.open(host, port=23[, timeout]) ： 连接到主机，端口号为第二个可选参数，默认为标准的Telnet端口(23)，可选的timeout参数指定连接的超时时间，如果未指定，将使用全局默认超时设置。不要尝试去重新打开一个已经连接的实例对象。

Telnet.close() ： 关闭连接。

Telnet.write(buffer) ： # 将一个字节串(byte string)写进socket，如果连接被阻塞，这也会被阻塞，如果连接关闭，会抛出OSError。

Telnet.interact() ： telnet的交互功能，下面用了一个死循环保证用户能够一直输入命令进行某些操作，也可以使用Telnet.interact()这个方法来使所连接终端持久化，不过官网说 (emulates a very dumb Telnet client)直译是一个非常愚蠢的客户端。

官方文档，请参考：

https://docs.python.org/zh-cn/3/library/telnetlib.html

三、检测脚本

check_gateway.py

aHR0cHM6Ly9pbWFnZXMuY25ibG9ncy5jb20vT3V0bGluaW5nSW5kaWNhdG9ycy9Db250cmFjdGVkQmxvY2suZ2lm aHR0cHM6Ly9pbWFnZXMuY25ibG9ncy5jb20vT3V0bGluaW5nSW5kaWNhdG9ycy9FeHBhbmRlZEJsb2NrU3RhcnQuZ2lm

  1. #!/usr/bin/env python3
  2. # coding: utf-8
  3. import telnetlib
  4. import time
  6. class TelnetClient(object):
  7.     def __init__(self, ip, user, pswd):
  8.         self.tn = telnetlib.Telnet()
  9.         self.host_ip = ip
  10.         self.username = user
  11.         self.password = pswd
  12.         self.last_res = ''  # 记录上次命令执行结果
  14.     # 此函数实现telnet登录主机
  15.     def login_host(self):
  16.         try:
  17.             # self.tn = telnetlib.Telnet(host_ip,port=23)
  18.             self.tn.open(self.host_ip)
  19.         except:
  20.             text = '{} 网络连接失败'.format(self.host_ip)
  21.             print(text)
  22.             return False
  23.         # 等待login出现后输入用户名，最多等待0.3秒
  24.         self.tn.read_until(b'Username: ', timeout=0.3)
  26.         self.tn.write(self.username.encode('ascii') + b'\n')
  27.         # 等待Password出现后输入用户名，最多等待0.3秒
  28.         self.tn.read_until(b'Password: ', timeout=0.3)
  30.         self.tn.write(self.password.encode('ascii') + b'\n')
  31.         # 延时5秒再收取返回结果，给服务端足够响应时间
  32.         # time.sleep(5)
  34.         # 获取登录结果
  35.         # read_very_eager()获取到的是的是上次获取之后本次获取之前的所有输出
  36.         command_result = self.tn.read_very_eager().decode('utf-8')
  38.         if 'Login invalid' in command_result:  # Cisco交换登录失败提示语
  39.             text = '{} 登录失败，用户名或密码错误'.format(self.host_ip)
  40.             print(text)
  41.             return False
  42.         else:
  43.             text = '{} 登录成功'.format(self.host_ip)
  44.             print(text)
  45.             return True
  47.     # 执行某一条命令
  48.     def execute_command(self, command, show_res=False):
  49.         self.tn.write(command.encode() + b'\n')
  50.         time.sleep(3)
  52.         # 获取命令结果
  53.         command_result = self.tn.read_very_eager().decode('utf-8')
  54.         # if show_res:
  55.         #     print('命令执行结果：%s' % command_result)
  56.         return command_result
  58.     # 退出telnet
  59.     def logout_host(self):
  60.         self.tn.write(b"exit\n")
  61.         # print('本次操作结束，连接断开\n')
  63. def check_gateway(ip):
  64.     # 指定连接的交换机管理IP
  65.     telnet_client = TelnetClient(ip=ip, user='root', pswd='123456')
  67.     # 如果登录结果返加True，则执行命令，然后退出
  68.     if not telnet_client.login_host():
  69.         print("错误，ip: %s 登录失败"%ip)
  70.         return False
  72.     # 执行指定命令,查看进程
  73.     res1 = telnet_client.execute_command('ps')
  74.     # print(res1, type(res1))
  75.     if not res1:
  76.         print("错误，ip: %s 执行命令ps失败" % ip)
  77.         return False
  79.     if "/usr/app/bin/app" in res1:
  80.         print("正常，ip: %s 设备正常" % ip)
  81.         telnet_client.execute_command('exit')  # 退出配置模式
  82.     else:
  83.         print("错误，ip: %s 设备掉线" % ip)
  84.         # 执行重启
  85.         res2 = telnet_client.execute_command('reboot')
  87.     # 退出登录
  88.     telnet_client.logout_host()
  89.     return True
  91. if __name__ == '__main__':
  92.     ip = "10.212.82.90"
  93.     check_gateway(ip)

注意：请根据实际情况，修改ip地址，用户名和密码。

执行脚本

  1. # python3 check_gateway.py 
  2. 10.212.82.90 登录成功
  3. 正常，ip: 10.212.82.90 设备正常

本文参考链接：

https://blog.csdn.net/XYliurui/article/details/103082512

发表评论

表情：
评论列表 （有 0 条评论，232人围观）

还没有评论，来说两句吧...

相关阅读

    相关 activity网关使用

    什么是网关 网关用来控制流程的流向，简而言之，当流程实例在运转过程中，从一个节点跳转到下一个节点时，需要人为的控制某些条件，即只有满足这些条件的时候才会触发节点的跳转时，

    野性酷女野性酷女/ 0 赞/ 94 阅读

    相关 使用 API 网关

    简介 我们假设您正在为一个购物应用开发一个原生移动客户端。您可能需要实现一个产品详细信息页面，用于展示给定商品的信息。 例如，图 2-1 展示了在 Amazon 的 A

    绝地灬酷狼绝地灬酷狼/ 0 赞/ 279 阅读

    相关 网关及网关作用

    什么是网关 网关（Gateway）又称网间连接器、协议转换器。网关在传输层上以实现网络互连，是最复杂的网络互连设备，仅用于两个高层协议不同的网络互连。网关的结构也和路由器类似

    我就是我我就是我/ 0 赞/ 403 阅读