File: outbound-ssh-ncclient.py

package info (click to toggle)
python-ncclient 0.6.17-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,448 kB
  • sloc: python: 9,548; xml: 476; makefile: 77; sh: 5
file content (98 lines) | stat: -rw-r--r-- 2,726 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#!/usr/bin/python3

"""
 Listen on TCP port 2200 for incoming SSH session from Junos devices with
 the following ssh outbound configuration and collect host-name and junos-version
 upon connect, then terminate

 lab@router> show configuration system services outbound-ssh
 client outbound-ssh-ncclient {
     device-id vRR;
     services netconf;
     10.0.2.2 port 2200;
  }

 Example:

 $ ./outbound-ssh-ncclient.py
 Listening on port 2200 for incoming sessions ...
 Got a connection from 172.17.0.1:48038!
 MSG DEVICE-CONN-INFO V1 vRR
 Logging in ...
 requesting info...
   Hostname: vRR
    Version: 16.1R3.10
 $
"""


import logging
import socket
import sys
import time

from ncclient import manager
from ncclient.xml_ import *


def listener(port, user, password):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    s.bind(('', port))
    s.listen(5)
    logging.info('Listening on port %d for incoming sessions ...', port)
    while True:
        client, addr = s.accept()
        logging.info('Got a connection from %s:%d!', addr[0], addr[1])
        launch_junos_proxy(client, addr, user, password)


def launch_junos_proxy(client, addr, user, password):
    val = {
        'MSG-ID': None,
        'MSG-VER': None,
        'DEVICE-ID': None
    }
    msg = ''
    count = 3
    while len(msg) < 100 and count > 0:
        c = client.recv(1)
        c = c.decode()
        if c == '\r':
            continue

        if c == '\n':
            count -= 1
            if msg.find(':'):
                (key, value) = msg.split(': ')
                val[key] = value
                msg = ''
        else:
            msg += c

    logging.info('MSG %s %s %s', val['MSG-ID'], val['MSG-VER'], val['DEVICE-ID'])
    logging.info('Logging in ...')

    sock_fd = client.fileno()
    conn = manager.connect(host=None,
                           sock_fd=sock_fd,
                           username=user,
                           password=password,
                           timeout=60,
                           device_params={'name': 'junos'},
                           hostkey_verify=False)

    rpc = new_ele('get-software-information')

    logging.info('requesting info...')
    result = conn.rpc(rpc)
    logging.info('   Hostname: ' + result.xpath('//software-information/host-name')[0].text)
    logging.info('    Version: ' + result.xpath('//software-information/junos-version')[0].text)
    sys.exit(0)


if __name__ == '__main__':
    LOG_FORMAT = '%(asctime)s %(levelname)s %(filename)s:%(lineno)d %(message)s'
    logging.basicConfig(stream=sys.stdout, level=logging.INFO, format=LOG_FORMAT)

    listener(2200, 'netconf', 'juniper!')