File: //opt/alt/python37/lib/python3.7/site-packages/pyroute2/ndb/__pycache__/source.cpython-37.pyc
B
��f?B � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl m
Z
ddlmZ ddlmZ ddlmZ ddlmZ d d
lmZmZ d d lmZmZm Z ej � d �r�dd
l m Z ddl!m"Z" ddl#m$Z$ ndZ$dZ"d Z%dZ&G dd� de'�Z(G dd� de)�Z*dS )aa
Local RTNL
----------
Local RTNL source is a simple `IPRoute` instance. By default NDB
starts with one local RTNL source names `localhost`::
>>> ndb = NDB()
>>> ndb.sources.summary().format("json")
[
{
"name": "localhost",
"spec": "{'target': 'localhost', 'nlm_generator': 1}",
"state": "running"
},
{
"name": "localhost/nsmanager",
"spec": "{'target': 'localhost/nsmanager'}",
"state": "running"
}
]
>>> ndb.sources['localhost']
[running] <IPRoute {'target: 'localhost', 'nlm_generator': 1}>
The `localhost` RTNL source starts an additional async cache thread.
The `nlm_generator` option means that instead of collections the
`IPRoute` object returns generators, so `IPRoute` responses will not
consume memory regardless of the RTNL objects number::
>>> ndb.sources['localhost'].nl.link('dump')
<generator object RTNL_API.filter_messages at 0x7f61a99a34a0>
See also: :ref:`iproute`
Network namespaces
------------------
There are two ways to connect additional sources to an NDB instance.
One is to specify sources when creating an NDB object::
ndb = NDB(sources=[{'target': 'localhost'}, {'netns': 'test01'}])
Another way is to call `ndb.sources.add()` method::
ndb.sources.add(netns='test01')
This syntax: `{target': 'localhost'}` and `{'netns': 'test01'}` is the
short form. The full form would be::
{'target': 'localhost', # the label for the DB
'kind': 'local', # use IPRoute class to start the source
'nlm_generator': 1} #
{'target': 'test01', # the label
'kind': 'netns', # use NetNS class
'netns': 'test01'} #
See also: :ref:`netns`
Remote systems
--------------
It is possible also to connect to remote systems using SSH. In order to
use this kind of sources it is required to install the
`mitogen <https://github.com/dw/mitogen>`_ module. The `remote` kind
of sources uses the `RemoteIPRoute` class. The short form::
ndb.sources.add(hostname='worker1.example.com')
In some more extended form::
ndb.sources.add(**{'target': 'worker1.example.com',
'kind': 'remote',
'hostname': 'worker1.example.com',
'username': 'jenkins',
'check_host_keys': False})
See also: :ref:`remote`
� N)�
basestring)�IPRoute)� NetlinkError)�NetlinkSocketBase)�ifinfmsg)�
RemoteIPRoute� )�ShutdownException�State)�
cmsg_event� cmsg_failed� cmsg_sstart�linux)�netns)� NetNSManager)�NetNS� c @ s e Zd Zdd� Zdd� ZdS )� SourceProxyc C s || _ t�� | _|| _d S )N)�ndb�queue�Queue�events�target)�selfr r � r �D/opt/alt/python37/lib/python3.7/site-packages/pyroute2/ndb/source.py�__init__t s
zSourceProxy.__init__c O s� t t�� j�}t�� | jj|< }|�� | jj �
d| j ||||d�� |� � | jj�
|�}d|krn|d S d|kr~|d �d S )N�api)�typer �call_id�name�argv�kwarg�return� exception)�str�uuid�uuid4�hex� threading�Eventr Z_call_registry�clear� messenger�emitr �wait�pop)r r r! r" r �event�responser r r r y s
zSourceProxy.apiN)�__name__�
__module__� __qualname__r r r r r r r s s r c @ s e Zd ZdZdZdZdZdZdZe e
e e d�Z
dd� Zedd � �Zed
d � �Zed d
� �Zdd� Zedd� �Zdd� Zedd� �Zedd� �Zedd� �Zedd� �Zed d � �Zd d � Z d d!� Z d"d#� Z d$d%� Z d&d'� Z d(d)� Z!e"j#d*fd+d,�Z$d6d.d/�Z%d0d1� Z&d2d3� Z'd4d5� Z(dS )7�Sourcez�
The RNTL source. The source that is used to init the object
must comply to IPRoute API, must support the async_cache. If
the source starts additional threads, they must be joined
in the source.close()
�srcN�sources)�localr �remote� nsmanagerc K s d | _ d | _|| _| jj| _|d | _|�dd�| _|�dt�| _ |�d�| _
| � | j�| _ || _
| jjd k r�| jjj�| j� d| _t�� | _t�� | _t�� | _t�� | _| j�� |j�d| j �| _t | jdgd �| _ | j � d
� | jj � | j| j|� | �!� d S ) Nr �kindr8 �
max_errorsr0 r z
sources.%s�running)�logZ wait_list�init)"�th�nlr Z _event_queue�evqr r/ r; �SOURCE_MAX_ERRORSr< r0 � get_prime�nl_prime�nl_kwargr, �targets�add�errors_counterr) r* �shutdown�started�RLock�lock�
shutdown_lockr+ r> Zchannelr
�state�set� task_managerZdb_add_nl_source�load_sql)r r �specr r r r � s.
zSource.__init__c C s |j j�d|j �S )NzSELECT count(*) FROM %s)r rQ � db_fetchone�table)�cls�viewr r r �_count� s z
Source._countc C s | j dk s| j| j krdS dS )Nr TF)r< rI )r r r r � must_restart� s zSource.must_restartc C s$ t tdd� ddd| j�d�ff��S )Nc S s | d d k S )Nr r )�xr r r �<lambda>� � z'Source.bind_arguments.<locals>.<lambda>)Z async_cacheT)Z clone_socketT�groups)�dict�filterrF �get)r r r r �bind_arguments� s zSource.bind_argumentsc C sn yH| j d k r,| jjt| j| j �f| jd� n| jjt| j�f| jd� W n tk
rh | j�d� dS X dS )N)�source�stopFT) r0 rB �putr r r
r rO rP )r r r r � set_ready� s
zSource.set_readyc C s� t |�}i }d|kr0d|d< d|d< |d |d< d|kr^d|d< |d |d< t�|d �|d< x |D ]}||krd|| ||< qdW |S )N�hostnamer9 r; �ssh�protocolr r )r^ r Z_get_netnspath)rV rS �ret�defaults�keyr r r rj � s
zSource.defaultsc C sB t | jt�r| jjj}nt | jt�r,| jj}d| j�� || jf S )Nz [%s] <%s %s>) �
isinstancerE r � __class__r2 r rO r` rF )r r r r r �__repr__� s
zSource.__repr__c C s |S )Nr )rV r r r r �nla2name s zSource.nla2namec C s |S )Nr )rV r r r r �name2nla s zSource.name2nlac c s> dV x2|� � D ]&}|| j�� |d|| jf fV qW d S )N)rO r rS z%s)�keysrO r` rF )rV rW rk r r r �summary
s zSource.summaryc C s
| � |�S )N)rr )rV rW r r r �dump s z Source.dumpc C s t |t�r||d kS d S )Nr )rl r )r �left�rightr r r �compare_record s
zSource.compare_recordc C s" | j �| jd �p tt�d�| j�S )N�pyroute2)�vmapr` r; �getattr� importlib�
import_module)r r r r r rD s zSource.get_primec O s� x�t d�D ]�}| j�� y0| j�d|� d|� d|� �� t| j|�||�S tttt t
t j t
j fk
rn � Y nN tk
r� } z0| jd7 _| j�d|� d�� t�d� W d d }~X Y nX W d Q R X q
W td��d S )N�d zsource api run � r zsource api error: <�>zapi call failed)�rangerM r> �debugry rA r �AttributeError�
ValueError�KeyError� TypeError�socket�error�struct� ExceptionrI �time�sleep� RuntimeError)r r r! r" �_�er r r r s& *z
Source.apic C s� d}t � }d|d< d|d< d|d< d|d d< d
|d d < | j|d d < d
|d< d|fdg|d< |�� | jj|g| jd� d S )Nz.https://github.com/svinota/pyroute2/issues/737r �indexZuprO r �flags� �header� r r Z RTM_NEWLINKr0 Z IFLA_IFNAME)Z IFLA_ADDRESSz00:00:00:00:00:00�attrs)rb )r r �encoderB rd )r �urlZzero_ifr r r � fake_zero_if6 s
zSource.fake_zero_ifc C s2 �x�| j �� dk�r�| j�� r P | j��( | jd k rzy| jjdd� W n2 tk
rx } z| j� d| � W d d }~X Y nX y�| j �
d� t | j t
�r�i }|�| j� | jdkr�| jj|d< | j f |�| _ntd��| j �
d � | jjf | j� z@| jj�| j� | jd
k�r| �� | jj| j� � | jd � W d X d| _ W �n tk
�rJ } z�| j d 7 _ | j �
� | j �
d
| j � �� | j� dt
|�� d|� �� y | jjt | j�f| jd � W n$ t!k
�r� | j �
d� P Y nX | j"�r0| j�#d� | j �
d� | j�$t%� | j�� �r8| j�#d� | j �
d� P n| �&� S wW d d }~X Y nX W d Q R X | j�B | j �� d k�r�| �&� �szP | j �
� | j�'� | j �
d� W d Q R X �x*| j �� dk�r�yt(| j�� �}W nr tk
�r< } zR| j d 7 _ | j� dt
|�|f � d }| j"�r | j �
d� n | j �
d� P W d d }~X Y nX d}|�rl|d d d �rl|d d d j)}|d k�s�|t*j+k�r�| j �
d� P y| jj|| jd � W n$ t!k
�r� | j �
d� P Y nX �q�W qW y(| �,� | j�#d� | jj�| j� W n" t!k
�r | j�#d� Y nX | j �
d� d S )Nrc r )�codezsource restart: %sZ
connecting)r: �libcz source channel not supportedZloading)r8 r r9 )rb r zfailed, counter zsource error: r} zsleeping before restart�restartzsource shutdownr= )rc r� zsource error: %s %sr� r� zflush DB for the targetz#shutdown handled by the main thread�stopped)-rO r` rJ �is_setrM rA �closer� r> �warningrP rl rE r �updaterF r; r r� r� �bindra rQ Zdb_flushr r� rB rd rs rI rK r� r r rY r� r. �SOURCE_FAIL_PAUSEre r+ �tupler� �errno�
ECONNRESET�sync)r r� rS �msgr� r r r �receiverG s�
"
zSource.receiverc C s< | j �d� t�� }| jjt| j|�f| jd� |�� d S )Nr� )rb ) r> r� r) r* rB rd r r r. )r r� r r r r� � s z Source.syncc C s` | j �P | j�d� | jd k r0| j�� r0td��tj| jd| j d�| _| j�
� | S Q R X d S )Nzstarting the sourcezsource is runningzNDB event source: %s)r r ) rM r> r� r@ �is_aliver� r) �Threadr� r �start)r r r r r� � s
z Source.startTc C s� | j �� | j�� r"| j�d� d S | j�d� | j�� | jd k r�y| jj|d� W n2 tk
r� } z| j� d| � W d d }~X Y nX W d Q R X |r�| j
d k r�| j
� � d | _
n | j�d� d S )Nzalready stoppedzsource shutdown)r� zsource close: %szreceiver thread missing) rN rJ r� r> r� rP rA r� r� r� r@ �join)r r� r� r� r r r r� � s
,
z Source.close�unknownc
C s~ | j �d | j�R | j�d| � | j�� z.| �� | jrD| j�� | j �� | �
� W d X W d Q R X W d Q R X | j� � d S )Nz"restarting the source, reason <%s>) rM rN r> r� rK r+ r� r@ r� rJ r� r. )r �reasonr r r r� � s
zSource.restartc C s | S )Nr )r r r r � __enter__� s zSource.__enter__c C s | � � d S )N)r� )r �exc_type� exc_value� tracebackr r r �__exit__� s zSource.__exit__c C s~ | j j�d| j jj | jf�}|\| d< | d< xJ| j j�d| j jj | jf�D ](}|\}}}}|dkrnt|�n|| |< qNW d S )Nz�
SELECT * FROM sources
WHERE f_target = %s
r r; z�
SELECT * FROM sources_options
WHERE f_target = %s
�int)r rQ rT ZschemaZplchr Zdb_fetchr� )r rS Zf_target�f_nameZf_typeZf_valuer r r rR � s
zSource.load_sql)r� ))r2 r3 r4 �__doc__Z table_aliasZ dump_headerZsummary_headerrW rU r r r r rx r � classmethodrX �propertyrY ra re rj rn ro rp rr rs rv rD r r� r� r� r� r� r� r� r� r� r� rR r r r r r5 � sB
p
r5 )+r� r� rz r r� r� �sysr) r� r& Zpyroute2.commonr Zpyroute2.iprouter Zpyroute2.netlink.exceptionsr Zpyroute2.netlink.nlsocketr Z pyroute2.netlink.rtnl.ifinfmsgr Zpyroute2.remoter r r r
�messagesr r r
�platform�
startswithrw r Zpyroute2.netns.managerr Zpyroute2.nslink.nslinkr r� rC �objectr r^ r5 r r r r �<module>Q s6