python通过nmap扫描在线设备并尝试AAA登录(实例代码)


Posted in Python onDecember 30, 2019

如果管理网络设备很多,不可能靠人力每天去登录设备去查看是否在线。所以,可以利用python脚本通过每天扫描网络中的在线设备。可以部署在服务器上做成定时任务,每天发送AAA巡检报告。

下面是我写的一个python练手小程序。用来扫描一个网段中的在线主机,并尝试AAA去登录。统计一个大网段内可以成功aaa登录的主机。

注意:

该程序只是测试小程序,还有些小bug需要解决。不是通用的程序。主要提供一个大致思路。

主要用到了python-nmap, paramiko库。

程序大概思路:

  • 利用nmap扫描一个指定网段,只做ping扫描,所以前提所管理的设备中ping必须开启。获取存活设备IP列表。
  • 利用paramiko库模拟ssh去登录个IP,如果登录成功,返回设备名称,并及将设备名称和对应ip写入文件。

代码示例:

# -*- coding: utf-8 -*-

import nmap 
import datetime
import paramiko
import re

def get_name(host, user, password, port=22):
  client = paramiko.SSHClient()
  client.load_system_host_keys()
  client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  #client.connect(host, port, user, password, allow_agent=False, look_for_keys=False, timeout=5)
  try:
    client.connect(ip, port, user, password, allow_agent=False, look_for_keys=False, timeout=3)
  except Exception as err:
    return 0, str(err)
  #get shell
  ssh_shell = client.invoke_shell()

  dev_name = ''
  while True:
    line = ssh_shell.recv(1024)
    if line.endswith(b'>'):#华为 华三
      dev_name = re.findall(r'<(.*)>', str(line))[0]
      #dev_name = str(line)[3:-2]
      break
    if line.endswith(b'# ') | line.endswith(b'#'): #思科
      dev_name = re.findall(r'[\\r\\n|\\r]+(.*)#', str(line))[0]
      break
    if line.endswith(b'> '): 
      if 'ConnetOS' in str(line):#分流器
        dev_name =re.findall(r'[\\r\\n|\\r]+(.*)>', str(line))[0].strip()
      if '@' in str(line): #junpier防火墙
        dev_name =re.findall(r'@(.*)>', str(line))[0].strip()
      break
  #怎么跳出recv阻塞
  ssh_shell.close()
  return 1, dev_name


#print('扫描时间:'+res['nmap']['scanstats']['timestr']+'\n命令参数:'+res['nmap']['command_line'])

def get_ip_list(hosts):
  nm = nmap.PortScanner()
  #nmap填入参数列表可以填很多
  res = nm.scan(hosts=hosts, arguments='-sn -PE')
  #count = res['nmap']['scanstats']['uphosts'] #存活的主机数
  return list(res['scan'].keys()) #存活主机IP地址

    
if __name__ == '__main__':
  start = datetime.datetime.now()
  user = 'user'
  password = 'password'
  hosts = '10.0.0.0/24'
  dev = {} #存放AAA登录成功的主机
  f = open('ip_list.txt', 'w') #存放能ping通的IP
  ip_list = get_ip_list(hosts) 
  end = datetime.datetime.now()
  #f.write("存活的IP地址有:" + str(len(ip_list)) + "\n")
  #f.write("程序运行时间:" + str(end-start) + '\n')
  for ip in ip_list:
    f.write(ip + '\n')
  f.close()
  #print(ip_list)
  login_failed_count = 0
  f1 = open('login_succeed.txt', 'w', encoding='utf-8')
  f2 = open('login_failed.txt', 'w', encoding='utf-8')
  f3 = open('mtil_add.txt', 'w', encoding='utf-8')
  #ip_list = ip_list.split('\n')
  for ip in ip_list:
    ok, dev_name = get_name(ip, user, password)
    if ok == 1:
      if dev_name not in dev.keys():
        vendor = '' 
        print(dev_name + "\t\t" + ip)
        if 'h' in dev_name[-12:]:
          vendor = 'h3c'
        elif 'c' in dev_name[-12:]:
          vendor = 'cisco'
        elif 'w' in dev_name[-12:]:
          vendor = 'huawei'
        else:
          vendor = 'unknow'
        f1.write(dev_name + '\t\t' + ip + '\t' + vendor + '\n')
        f1.flush()
        dev.update({dev_name : ip})
      else:
        f3.write(dev_name + '\t\t' + str(dev[dev_name]) + ' ' + ip +'\n')
        print(dev_name + '\t\t' + str(dev[dev_name]) + ' ' + ip +'\n')
        dev.update({dev_name: [dev[dev_name] , ip]})
        f3.flush()
    else:
      login_failed_count += 1
      print(dev_name)
      f2.write(dev_name + '\t\t' + ip + '\n')
      f2.flush()
  end = datetime.datetime.now()
  f1.write('AAA登录成功' + str(len(dev)) +'台\n' )
  f1.write('AAA登录失败' + str(login_failed_count) +'台\n' )
  f1.write("程序运行时间:" + str(end-start) +'\n')
  f1.close()
  f2.close()
  f3.close()
  
  print("程序运行时间:" + str(end-start) +'\n')
  print("存活的IP地址有:" + str(len(ip_list)) + "\n")
  print("AAA登录成功:" + str(len(dev)) + "\n")
  print('AAA登录失败' + str(login_failed_count) +'台\n')

这个小程序例子,只是一个大概思路。

可以添加或则改善的思路:

  • 比想要获取设备名,可以通过snmp,知道ip地址和snmp读团体名就可以直接获取。
  • 可以将获取到的数据存入数据库中,从而可以做更的事情。
  • 通过类似代码,也可以实现每天去设备上备份网络配置等功能。
  • 可以将利用扫描结果,添加更多处理逻辑,生成每日巡检日报,通过邮件或者短信发送。

nmap库使用:

nmap工具使用可参考:nmap扫描工具学习笔记)

如果在windows上写nmap库,有两个事要解决。

第一步:安装nmap软件

因为在python程序中,nmap包所调用的是nmap可执行程序,所以必须先安装nmap软件。nmap下载地址: https://nmap.org/download.html

第二步: 需要在nmap库中文件的init方法中添加的nmap.exe的路径。

不然会报错,提示找不到nmap。

在nmap.py的class PortScanner()中的__init__()中更改:

def __init__(self, nmap_search_path=('nmap', '/usr/bin/nmap', '/usr/local/bin/nmap', '/sw/bin/nmap', '/opt/local/bin/nmap',r"D:\software\nmap-7.80\nmap.exe")):

主要添加了‘r”D:\software\nmap-7.80\nmap.exe”, nmap.exe可执行文件路径。

import nmap
nm = nmap.PortScanner()
#nmap填入参数列表可以填很多
res = nm.scan(hosts=hosts, arguments='-sn -PE')

其他使用示例:

#!/usr/bin/env python
import nmap # import nmap.py module
nm = nmap.PortScanner() # instantiate nmap.PortScanner object
nm.scan('127.0.0.1', '22-443') # scan host 127.0.0.1, ports from 22 to 443
nm.command_line() # get command line used for the scan : nmap -oX - -p 22-443 127.0.0.1
nm.scaninfo() # get nmap scan informations {'tcp': {'services': '22-443', 'method': 'connect'}}
nm.all_hosts() # get all hosts that were scanned
nm['127.0.0.1'].hostname() # get one hostname for host 127.0.0.1, usualy the user record
nm['127.0.0.1'].hostnames() # get list of hostnames for host 127.0.0.1 as a list of dict
# [{'name':'hostname1', 'type':'PTR'}, {'name':'hostname2', 'type':'user'}]
nm['127.0.0.1'].hostname() # get hostname for host 127.0.0.1
nm['127.0.0.1'].state() # get state of host 127.0.0.1 (up|down|unknown|skipped)
nm['127.0.0.1'].all_protocols() # get all scanned protocols ['tcp', 'udp'] in (ip|tcp|udp|sctp)
nm['127.0.0.1']['tcp'].keys() # get all ports for tcp protocol
nm['127.0.0.1'].all_tcp() # get all ports for tcp protocol (sorted version)
nm['127.0.0.1'].all_udp() # get all ports for udp protocol (sorted version)
nm['127.0.0.1'].all_ip() # get all ports for ip protocol (sorted version)
nm['127.0.0.1'].all_sctp() # get all ports for sctp protocol (sorted version)
nm['127.0.0.1'].has_tcp(22) # is there any information for port 22/tcp on host 127.0.0.1
nm['127.0.0.1']['tcp'][22] # get infos about port 22 in tcp on host 127.0.0.1
nm['127.0.0.1'].tcp(22) # get infos about port 22 in tcp on host 127.0.0.1
nm['127.0.0.1']['tcp'][22]['state'] # get state of port 22/tcp on host 127.0.0.1 (open

参考文档:

https://pypi.org/project/python-nmap/

总结

以上所述是小编给大家介绍的python通过nmap扫描在线设备并尝试AAA登录,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对三水点靠木网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!

Python 相关文章推荐
Windows下安装python2.7及科学计算套装
Mar 05 Python
Python多线程编程(五):死锁的形成
Apr 05 Python
Python通过OpenCV的findContours获取轮廓并切割实例
Jan 05 Python
python使用numpy读取、保存txt数据的实例
Oct 14 Python
python使用Plotly绘图工具绘制气泡图
Apr 01 Python
基于python实现百度翻译功能
May 09 Python
Python线程条件变量Condition原理解析
Jan 20 Python
对tensorflow中tf.nn.conv1d和layers.conv1d的区别详解
Feb 11 Python
150行python代码实现贪吃蛇游戏
Apr 24 Python
Python字典dict常用方法函数实例
Nov 09 Python
Django filter动态过滤与排序实现过程解析
Nov 26 Python
Python利用Turtle绘制哆啦A梦和小猪佩奇
Apr 04 Python
pytorch中获取模型input/output shape实例
Dec 30 #Python
Python读取csv文件实例解析
Dec 30 #Python
Pytorch Tensor的统计属性实例讲解
Dec 30 #Python
PyTorch中permute的用法详解
Dec 30 #Python
python实现多进程按序号批量修改文件名的方法示例
Dec 30 #Python
Pytorch Tensor基本数学运算详解
Dec 30 #Python
python垃圾回收机制(GC)原理解析
Dec 30 #Python
You might like
虹吸式咖啡壶操作
2021/03/03 冲泡冲煮
20个PHP常用类库小结
2011/09/11 PHP
Smarty模板类内部原理实例分析
2019/07/03 PHP
PHP获取当前时间不准确问题解决方案
2020/08/14 PHP
通过JS 获取Mouse Position(鼠标坐标)的代码
2009/09/21 Javascript
jQuery chili图片远处放大插件
2009/11/30 Javascript
poshytip 基于jquery的 插件 主要用于显示微博人的图像和鼠标提示等
2012/10/12 Javascript
jQuery实现TAB风格的全国省份城市滑动切换效果代码
2015/08/24 Javascript
JS基础随笔(菜鸟必看篇)
2016/07/13 Javascript
AnjularJS中$scope和$rootScope的区别小结
2016/09/18 Javascript
浅谈jquery的html方法里包含特殊字符的处理
2016/11/30 Javascript
JS实现中文汉字按拼音排序的方法
2017/10/09 Javascript
clipboard.js在移动端复制失败的解决方法
2018/06/13 Javascript
vue路由前进后退动画效果的实现代码
2018/12/10 Javascript
uniapp与webview之间的相互传值的实现
2020/06/29 Javascript
JavaScript实现缓动动画
2020/11/25 Javascript
python 多线程应用介绍
2012/12/19 Python
Python使用urllib模块的urlopen超时问题解决方法
2014/11/08 Python
浅析Python中的join()方法的使用
2015/05/19 Python
python tensorflow基于cnn实现手写数字识别
2018/01/01 Python
Python科学计算包numpy用法实例详解
2018/02/08 Python
对Python3.x版本print函数左右对齐详解
2018/12/22 Python
Python常用的json标准库
2019/02/19 Python
python 实现绘制整齐的表格
2019/11/18 Python
PyCharm中配置PySide2的图文教程
2020/06/18 Python
Pytorch实现将模型的所有参数的梯度清0
2020/06/24 Python
用CSS禁用输入法(CSS3 UI规范)实例解析
2012/12/04 HTML / CSS
Strawberrynet草莓网新加坡站:护肤、彩妆、香水及美发产品
2018/08/31 全球购物
空指针到底是什么
2012/08/07 面试题
大学信息公开实施方案
2014/03/09 职场文书
施工单位安全责任书
2014/07/24 职场文书
幼儿园八一建军节活动方案
2014/08/27 职场文书
民警个人对照检查剖析材料
2014/09/17 职场文书
悬崖上的金鱼姬观后感
2015/06/15 职场文书
解决Pytorch中关于model.eval的问题
2021/05/22 Python
dubbo服务整合zipkin详解
2021/07/26 Java/Android