Junos PyEZ, използвайки примера на задачата за търсене на безплатни ipv4 подмрежи

Статия за работата с Junos PyEZ - „Микрорамка на Python, която ви позволява да управлявате и автоматизирате устройства, работещи с Junos OS“ автоматизация и управление, всичко, което обичаме. Писането на скрипта, описан в тази статия, имаше няколко цели - изучаване на Python и автоматизиране на задачи за събиране на информация или промяна на конфигурации на оборудване, работещо с Junos OS. Изборът на тази специфична комбинация от Python + Junos PyEZ беше направен поради ниската бариера за навлизане в езика за програмиране Python и лекотата на използване на библиотеката Junos PyEZ, която не изисква експертни познания за Junos OS.

Задача

Одит на безплатни ipv4 подмрежи, принадлежащи на компанията. Критерият, че подмрежата е свободна, е липсата на запис за нея в маршрутите на комутатора, действащ като рутер, работещ с Junos OS.

Изпълнение

Python + Junos PyEZ, въпреки че имаше изкушение да се направи чрез paramiko и ssh.exec_command, В резултат на това ще трябва да конфигурирате протокола за мрежово управление на устройството netconf на запитваното оборудване. Netconf работи с хардуер чрез извикване на отдалечена процедура RPC и използва XML, в този пример, за да предостави информацията, която получава.

Инсталирането на текущата версия на Junos PyEZ от PyPI се извършва със следната команда:

$ pip install junos-eznc

Можете също да инсталирате от главния клон на проекта в GitHub със следната команда:

$ pip install git+https://github.com/Juniper/py-junos-eznc.git

И още една опция чрез

$ pip install -r requirements.txt 

Тази команда ще инсталира библиотеки, които липсват в системата и са необходими за работа. В моя вариант изисквания.txt Има само две от тях, най-новите версии са посочени към момента на писане на скрипта:

junos-eznc
netaddr

По подразбиране скриптът приема името на текущия потребител в системата; можете да влезете под името на друг потребител, като използвате ключа show_route.py -u getpass.getpass взема паролата от stdin, така че паролата няма да остане в системата. За да се свържете с оборудването, ще трябва също да въведете неговото име на хост или IP адрес, когато бъдете подканени. Получени са всички данни, необходими за авторизация на устройството.

Junos PyEZ поддържа свързване към оборудване, работещо с Junos OS, използвайки конзолата, telnet или netconf чрез ssh. Статията разглежда последния вариант.

За да се свържете с оборудване, използвайте класа Device на модула jnpr.junos

with jnpr.junos.Device(host=router,
                           user=args.name,
                           passwd=password) as dev:

Прави се заявка за всички маршрути, известни на рутера чрез извикване на отдалечена процедура или извикване на отдалечена процедура, което от двете е по-удобно.

data = dev.rpc.get_route_information()

Подобна команда на Junos OS

user@router> show route | display xml

Като добавим rpc в края на командата, получаваме таг за заявка и можем да го съпоставим с името на RPC метода, като по този начин можем да открием други интересни имена. Струва си да се отбележи, че синтаксисът за писане на етикета на заявката е различен от името на метода, а именно трябва да замените тиретата с долна черта.

user@router> show route | display xml rpc
<rpc-reply >route_list = data.xpath("//rt-destination/text()")

Останалата част беше обвита в цикъл while, за да не се повтаря заявката към рутера, ако е необходимо да се провери друга подмрежа от тези, за които рутерът вече знае. Струва си да се спомене, че рутерът, на който правя заявката, познава маршрути само през OSPF, така че за периферен рутер е по-добре да промените малко заявката, за да намалите времето за изпълнение на скрипта

data = dev.rpc.get_ospf_route_information()

Сега нека да разгледаме съдържанието на цикъла while

В началото потребителят ще бъде помолен да въведе подмрежа с маска и не повече от три октета от мрежата на същата подмрежа, това е необходимо за задаване на обхвата на търсене. Не ми харесва много тази реализация на определяне на критериите и обхвата на търсене, но досега не съм намерил по-добро решение. След това от получения списък с подмрежи route_list, използвайки променлива, съдържаща не повече от три октета, избирам подмрежите, които ме интересуват

tmp = re.search(r'^%sS*' % subnet_search, route_list[i])

Чрез IPNetwork, модула netaddr, получавам подмрежи под формата на списък с ipv4 адреси

range_subnet = netaddr.IPNetwork(tmp.group(0))

Използвайки IPNetwork, получавам диапазон от адреси от въведена от потребител мрежа с маска и генерирам списък на всички адреси от този диапазон за сравнение със списъка на заетите адреси.

for i in set(net_list).difference(set(busyip)):
        freeip.append(i)

Показвам получения списък със свободни адреси под формата на подмрежи

print(netaddr.IPSet(freeip))

По-долу е пълният скрипт, тестван на комутатори, използвани като рутер, модели ex4550, ex4600

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import argparse
import getpass
import netaddr
import re
import sys

import jnpr.junos

parser = argparse.ArgumentParser()
parser.add_argument('-u', '--user',
                    action='store',
                    dest='name',
                    help='Enter login from tacacs if it differs from the '
                         'username in the system.')
args = parser.parse_args()

if not args.name:
    args.name = getpass.getuser()    # Return the “login name” of the user.
router = input("Full routers name: ")
password = getpass.getpass("Password: ")

try:
    # Authenticates to a device running Junos, for get information about routs
    # into xml format and selects by tag.
    route_list = []
    with jnpr.junos.Device(host=router,
                           user=args.name,
                           passwd=password) as dev:
        data = dev.rpc.get_route_information()
    route_list = data.xpath("//rt-destination/text()")
except (jnpr.junos.exception.ConnectRefusedError,
        jnpr.junos.exception.ConnectUnknownHostError) as err:
    print("Equipment name or password wrong.")
    sys.exit(1)

while True:
    subnet = input("Net with mask: ")
    subnet_search = input("Input no more three octet: ")
    # Gets a list of busy IP addresses from the received subnets.
    busyip = []
    for i in range(len(route_list)):
        tmp = re.search(r'^%sS*' % subnet_search, route_list[i])
        if tmp:
            range_subnet = netaddr.IPNetwork(tmp.group(0))
            for ip in range_subnet:
                busyip.append("%s" % ip)
    range_subnet = netaddr.IPNetwork(subnet)
    # Gets list ip adresses from subnetworks lists.
    net_list = []
    for ip in range_subnet:
        net_list.append("%s" % ip)
    # Сomparing lists.
    freeip = []
    for i in set(net_list).difference(set(busyip)):
        freeip.append(i)
    print(netaddr.IPSet(freeip))

    request = input("To run request again enter yes or y, "
                    "press 'enter', complete request: ")
    if request in ("yes", "y"):
        continue
    else:
        print('Bye')
        break

Източник: www.habr.com