← Back
Editing: testing.cpython-38.pyc
U W[�C � @ s d Z ddlZddlZddlmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZm Z m Z m Z dd l m Z d Z!dd� Z"dZ#dd� Z$dd� Z%dd� Z&G dd� de'�Z(dd� Z)ee �G dd � d e'��Z*G d d � d e'�Z+dS ) zD Tools for automated testing of L{twisted.pair}-based applications. � N)�EPERM�EAGAIN� EWOULDBLOCK�ENOSYS�EBADF�EINVAL�EINTR�ENOBUFS)�deque��wraps)� implementer)�DatagramProtocol)�EthernetProtocol)�RawUDPProtocol)� IPProtocol)� _IFNAMSIZ� _TUNSETIFF�_IInputOutputSystem� TunnelFlags)� nativeString� c C s t �d| �S )z� Pack an integer into a network-order two-byte string. @param n: The integer to pack. Only values that fit into 16 bits are supported. @return: The packed representation of the integer. @rtype: L{bytes} z>H)�struct�pack)�n� r �6/usr/lib/python3/dist-packages/twisted/pair/testing.py�_H! s r � c C s || t |� | S )a� Construct an ethernet frame. @param src: The source ethernet address, encoded. @type src: L{bytes} @param dst: The destination ethernet address, encoded. @type dst: L{bytes} @param protocol: The protocol number of the payload of this datagram. @type protocol: L{int} @param payload: The content of the ethernet frame (such as an IP datagram). @type payload: L{bytes} @return: The full ethernet frame. @rtype: L{bytes} )r ��src�dstZprotocol�payloadr r r � _ethernet1 s r# c C s� dt dt|� � d t d� t�tjt| �� t�tjt|�� }tt�d|��}|d? }|d@ | }|dA }|dd � t� d |� |d d� }|| S ) a� Construct an IP datagram with the given source, destination, and application payload. @param src: The source IPv4 address as a dotted-quad string. @type src: L{bytes} @param dst: The destination IPv4 address as a dotted-quad string. @type dst: L{bytes} @param payload: The content of the IP datagram (such as a UDP datagram). @type payload: L{bytes} @return: An IP datagram header and payload. @rtype: L{bytes} s E � s @r z!10H� i�� N� z!H� ) r �len�socketZ inet_ptonZAF_INETr �sumr �unpackr )r r! r" ZipHeaderZ checksumStep1ZcarryZ checksumStep2Z checksumStep3r r r �_ipH s. ��� � �� � ��r, c C s0 t | �t |� t t|�d � t d� }|| S )a~ Construct a UDP datagram with the given source, destination, and application payload. @param src: The source port number. @type src: L{int} @param dst: The destination port number. @type dst: L{int} @param payload: The content of the UDP datagram. @type payload: L{bytes} @return: A UDP datagram header and payload. @rtype: L{bytes} � r )r r( )r r! r" Z udpHeaderr r r �_udp} s ���� r. c @ sr e Zd ZdZdZeed�Zee d�Z ee d�Z eZ dZdd� Zed d � �Zed d � �Zd d� Zdd� Zdd� ZdS )�Tunnelz� An in-memory implementation of a tun or tap device. @cvar _DEVICE_NAME: A string representing the conventional filesystem entry for the tunnel factory character special device. @type _DEVICE_NAME: C{bytes} s /dev/net/tunz Resource temporarily unavailablezOperation would blockzInterrupted function calli c C s: || _ || _d| _d| _d| _t� | _t� | _t� | _dS )a @param system: An L{_IInputOutputSystem} provider to use to perform I/O. @param openFlags: Any flags to apply when opening the tunnel device. See C{os.O_*}. @type openFlags: L{int} @param fileMode: ignored N) �system� openFlags� tunnelMode� requestedName�namer � readBuffer� writeBuffer�pendingSignals)�selfr0 r1 ZfileModer r r �__init__� s zTunnel.__init__c C s | j | jj@ S )zx If the file descriptor for this tunnel is open in blocking mode, C{True}. C{False} otherwise. )r1 r0 � O_NONBLOCK�r8 r r r �blocking� s zTunnel.blockingc C s t | j| jj@ �S )zz If the file descriptor for this tunnel is marked as close-on-exec, C{True}. C{False} otherwise. )�boolr1 r0 � O_CLOEXECr; r r r � closeOnExec� s zTunnel.closeOnExecc C s. | j tjj@ r tddt|d�}| j�|� dS )aI Deliver a datagram to this tunnel's read buffer. This makes it available to be read later using the C{read} method. @param datagram: The IPv4 datagram to deliver. If the mode of this tunnel is TAP then ethernet framing will be added automatically. @type datagram: L{bytes} s s ������r N)r2 r �IFF_TAP�valuer# �_IPv4r5 �append�r8 �datagramr r r �addToReadBuffer� s �zTunnel.addToReadBufferc C sX | j r@| jtjj@ rd}ndt }|d8 }|| j �� d|� S | jrNt� �n| j �dS )a Read a datagram out of this tunnel. @param limit: The maximum number of bytes from the datagram to return. If the next datagram is larger than this, extra bytes are dropped and lost forever. @type limit: L{int} @raise OSError: Any of the usual I/O problems can result in this exception being raised with some particular error number set. @raise IOError: Any of the usual I/O problems can result in this exception being raised with some particular error number set. @return: The datagram which was read from the tunnel. If the tunnel mode does not include L{TunnelFlags.IFF_NO_PI} then the datagram is prefixed with a 4 byte PI header. @rtype: L{bytes} � � r N) r5 r2 r � IFF_NO_PIrA �_PI_SIZE�popleftr< �NotImplementedError�nonBlockingExceptionStyle)r8 �limit�headerr r r �read� s z Tunnel.readc C sF | j r| j �� ttd��t|�| jkr2ttd��| j�|� t|�S )a{ Write a datagram into this tunnel. @param datagram: The datagram to write. @type datagram: L{bytes} @raise IOError: Any of the usual I/O problems can result in this exception being raised with some particular error number set. @return: The number of bytes of the datagram which were written. @rtype: L{int} zInterrupted system callzNo buffer space available) r7 rK �IOErrorr r( �SEND_BUFFER_SIZEr r6 rC rD r r r �write s z Tunnel.writeN)�__name__� __module__� __qualname__�__doc__Z _DEVICE_NAMErQ r Z EAGAIN_STYLE�OSErrorr ZEWOULDBLOCK_STYLEr Z EINTR_STYLErM rR r9 �propertyr< r? rF rP rS r r r r r/ � s $r/ c s t � �� fdd��}|S )a| Wrap a L{MemoryIOSystem} method with permission-checking logic. The returned function will check C{self.permissions} and raise L{IOError} with L{errno.EPERM} if the function name is not listed as an available permission. @param original: The L{MemoryIOSystem} instance to wrap. @return: A wrapper around C{original} that applies permission checks. c s& � j | jkrttd��� | f|�|�S )NzOperation not permitted)rT � permissionsrQ r )r8 �args�kwargs��originalr r �permissionChecker6 s z&_privileged.<locals>.permissionCheckerr )r^ r_ r r] r � _privileged+ s r` c @ sz e Zd ZdZdZdZdZdZdd� Zdd � Z d d � Z e dd d��Z dd� Z dd� Zdd� Ze dd� �Zdd� Zdd� Zd S ) �MemoryIOSystemz� An in-memory implementation of basic I/O primitives, useful in the context of unit testing as a drop-in replacement for parts of the C{os} module. @ivar _devices: @ivar _openFiles: @ivar permissions: @ivar _counter: i � � r c C s i | _ i | _tddg�| _d S )N�open�ioctl)�_devices� _openFiles�setrZ r; r r r r9 Q s zMemoryIOSystem.__init__c C s | j |�� S )aX Get the L{Tunnel} object associated with the given L{TuntapPort}. @param port: A L{TuntapPort} previously initialized using this L{MemoryIOSystem}. @return: The tunnel object created by a prior use of C{open} on this object on the tunnel special device file. @rtype: L{Tunnel} )rg �fileno)r8 �portr r r � getTunnelW s zMemoryIOSystem.getTunnelc C s || j |< dS )a1 Specify a class which will be used to handle I/O to a device of a particular name. @param name: The filesystem path name of the device. @type name: L{bytes} @param cls: A class (like L{Tunnel}) to instantiated whenever this device is opened. N)rf )r8 r4 �clsr r r �registerSpecialDevicee s z$MemoryIOSystem.registerSpecialDeviceNc C sH || j kr:| j}| jd7 _| j | | ||�| j|< |S ttd��dS )a� A replacement for C{os.open}. This initializes state in this L{MemoryIOSystem} which will be reflected in the behavior of the other file descriptor-related methods (eg L{MemoryIOSystem.read}, L{MemoryIOSystem.write}, etc). @param name: A string giving the name of the file to open. @type name: C{bytes} @param flags: The flags with which to open the file. @type flags: C{int} @param mode: The mode with which to open the file. @type mode: C{int} @raise OSError: With C{ENOSYS} if the file is not a recognized special device file. @return: A file descriptor associated with the newly opened file description. @rtype: L{int} rb zFunction not implementedN)rf �_counterrg rX r )r8 r4 �flags�mode�fdr r r rd s s zMemoryIOSystem.openc C s6 z| j | �|�W S tk r0 ttd��Y nX dS )z� Try to read some bytes out of one of the in-memory buffers which may previously have been populated by C{write}. @see: L{os.read} �Bad file descriptorN)rg rP �KeyErrorrX r )r8 rq rN r r r rP � s zMemoryIOSystem.readc C s6 z| j | �|�W S tk r0 ttd��Y nX dS )z� Try to add some bytes to one of the in-memory buffers to be accessed by a later C{read} call. @see: L{os.write} rr N)rg rS rs rX r )r8 rq �datar r r rS � s zMemoryIOSystem.writec C s0 z | j |= W n tk r* ttd��Y nX dS )z� Discard the in-memory buffer and other in-memory state for the given file descriptor. @see: L{os.close} rr N)rg rs rX r )r8 rq r r r �close� s zMemoryIOSystem.closec C s� z| j | }W n tk r, ttd��Y nX |tkr@ttd��t�dtf |�\}}||_ ||_ |dtd � d |_ t� dtf |j |�S )z� Perform some configuration change to the in-memory state for the given file descriptor. @see: L{fcntl.ioctl} rr z Request or args is not valid.z%dsHN� s 123) rg rs rQ r r r r r+ r r2 r3 r4 r )r8 rq Zrequestr[ Ztunnelr4 rp r r r re � s zMemoryIOSystem.ioctlc C sL d}d}t ||d t||d |d�d�}t| j�� �}|d �|� ||fS )ah Write an ethernet frame containing an ip datagram containing a udp datagram containing the given payload, addressed to the given address, to a tunnel device previously opened on this I/O system. @param datagram: A UDP datagram payload to send. @type datagram: L{bytes} @param address: The destination to which to send the datagram. @type address: L{tuple} of (L{bytes}, L{int}) @return: A two-tuple giving the address from which gives the address from which the datagram was sent. @rtype: L{tuple} of (L{bytes}, L{int}) z10.1.2.3iaS r rb )r r! r" )r, r. �listrg �valuesrF )r8 rE �addressZsrcIPZsrcPortZ serializedZ openFilesr r r �sendUDP� s ��zMemoryIOSystem.sendUDPc C s t | |�S )aa Get a socket-like object which can be used to receive a datagram sent from the given address. @param fileno: A file descriptor representing a tunnel device which the datagram will be received via. @type fileno: L{int} @param host: The IPv4 address to which the datagram was sent. @type host: L{bytes} @param port: The UDP port number to which the datagram was sent. received. @type port: L{int} @return: A L{socket.socket}-like object which can be used to receive the specified datagram. )� _FakePort)r8 ri Zhostrj r r r � receiveUDP� s zMemoryIOSystem.receiveUDP)N)rT rU rV rW rn �O_RDWRr: r> r9 rk rm r` rd rP rS ru re rz r| r r r r ra ? s" ra c @ s e Zd ZdZdd� Zdd� ZdS )r{ z� A socket-like object which can be used to read UDP datagrams from tunnel-like file descriptors managed by a L{MemoryIOSystem}. c C s || _ || _d S �N)�_system�_fileno)r8 r0 ri r r r r9 s z_FakePort.__init__c s� | j j| j j�� }g � t� }� fdd�}||_t� }|�d|� t � ���d|� | j j| j j }|t j j @ r�t� }|�d�� |j}n �fdd�}|t jj @ } | r�|td� }||� � d d|� S ) a_ Receive a datagram sent to this port using the L{MemoryIOSystem} which created this object. This behaves like L{socket.socket.recv} but the data being I{sent} and I{received} only passes through various memory buffers managed by this object and L{MemoryIOSystem}. @see: L{socket.socket.recv} c s � � | � d S r~ )rC )rE ry )� datagramsr r �capture! s z _FakePort.recv.<locals>.capturei90 � r c s � � | d d d d �S r~ )�datagramReceived)rt )�ipr r �<lambda>2 s �z _FakePort.recv.<locals>.<lambda>Nr )r rg r� r6 rK r r� r ZaddProtor r2 r r@ rA r rI rJ ) r8 �nbytesrt Zreceiverr� Zudprp Zetherr� Z dataHasPIr )r� r� r �recv s( z_FakePort.recvN)rT rU rV rW r9 r� r r r r r{ s r{ ),rW r r) �errnor r r r r r r r � collectionsr � functoolsr Zzope.interfacer Ztwisted.internet.protocolr Ztwisted.pair.ethernetr Ztwisted.pair.rawudpr Ztwisted.pair.ipr Ztwisted.pair.tuntapr r r r Ztwisted.python.compatr rJ r rB r# r, r. �objectr/ r` ra r{ r r r r �<module> s2 ( 5 H
Save File
Cancel