mirror of
https://github.com/GNS3/gns3-server
synced 2024-12-28 01:38:08 +00:00
141 lines
4.4 KiB
Python
141 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2013 GNS3 Technologies Inc.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
Interface for automatic UDP NIOs.
|
|
"""
|
|
|
|
import asyncio
|
|
from .nio import NIO
|
|
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class NIOUDPAuto(NIO):
|
|
|
|
"""
|
|
Dynamips auto UDP NIO.
|
|
|
|
:param hypervisor: Dynamips hypervisor instance
|
|
:param laddr: local address
|
|
:param lport_start: start local port range
|
|
:param lport_end: end local port range
|
|
"""
|
|
|
|
_instance_count = 0
|
|
|
|
def __init__(self, hypervisor, laddr, lport_start, lport_end):
|
|
|
|
# create an unique ID and name
|
|
nio_id = NIOUDPAuto._instance_count
|
|
NIOUDPAuto._instance_count += 1
|
|
name = 'nio_udp_auto' + str(nio_id)
|
|
self._laddr = laddr
|
|
self._lport = None
|
|
self._raddr = None
|
|
self._rport = None
|
|
NIO.__init__(self, name, hypervisor)
|
|
|
|
@classmethod
|
|
def reset(cls):
|
|
"""
|
|
Reset the instance count.
|
|
"""
|
|
|
|
cls._instance_count = 0
|
|
|
|
@asyncio.coroutine
|
|
def create(self):
|
|
|
|
port = yield from self._hypervisor.send("nio create_udp_auto {name} {laddr} {lport_start} {lport_end}".format(name=self._name,
|
|
laddr=self._laddr,
|
|
lport_start=self._lport_start,
|
|
lport_end=self._lport_end))
|
|
self._lport = int(port[0])
|
|
|
|
log.info("NIO UDP AUTO {name} created with laddr={laddr}, lport_start={start}, lport_end={end}".format(name=self._name,
|
|
laddr=self._laddr,
|
|
start=self._lport_start,
|
|
end=self._lport_end))
|
|
|
|
@property
|
|
def laddr(self):
|
|
"""
|
|
Returns the local address
|
|
|
|
:returns: local address
|
|
"""
|
|
|
|
return self._laddr
|
|
|
|
@property
|
|
def lport(self):
|
|
"""
|
|
Returns the local port
|
|
|
|
:returns: local port number
|
|
"""
|
|
|
|
return self._lport
|
|
|
|
@property
|
|
def raddr(self):
|
|
"""
|
|
Returns the remote address
|
|
|
|
:returns: remote address
|
|
"""
|
|
|
|
return self._raddr
|
|
|
|
@property
|
|
def rport(self):
|
|
"""
|
|
Returns the remote port
|
|
|
|
:returns: remote port number
|
|
"""
|
|
|
|
return self._rport
|
|
|
|
@asyncio.coroutine
|
|
def connect(self, raddr, rport):
|
|
"""
|
|
Connects this NIO to a remote socket
|
|
|
|
:param raddr: remote address
|
|
:param rport: remote port number
|
|
"""
|
|
|
|
yield from self._hypervisor.send("nio connect_udp_auto {name} {raddr} {rport}".format(name=self._name,
|
|
raddr=raddr,
|
|
rport=rport))
|
|
self._raddr = raddr
|
|
self._rport = rport
|
|
|
|
log.info("NIO UDP AUTO {name} connected to {raddr}:{rport}".format(name=self._name,
|
|
raddr=raddr,
|
|
rport=rport))
|
|
|
|
def __json__(self):
|
|
|
|
return {"type": "nio_udp_auto",
|
|
"lport": self._lport,
|
|
"rport": self._rport,
|
|
"raddr": self._raddr}
|