File manager - Edit - /home/newsbmcs.com/public_html/static/img/logo/test.zip
Back
PK ə�Z][�K� � test_scripts.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for the command-line interfaces to conch. """ from unittest import skipIf from twisted.python.reflect import requireModule from twisted.python.test.test_shellcomp import ZshScriptTestMixin from twisted.scripts.test.test_scripts import ScriptTestsMixin from twisted.trial.unittest import TestCase doSkip = False skipReason = "" if not requireModule("pyasn1"): doSkip = True skipReason = "Cannot run without PyASN1" if not requireModule("cryptography"): doSkip = True cryptoSkip = "can't run w/o cryptography" if not requireModule("tty"): doSkip = True ttySkip = "can't run w/o tty" try: import tkinter except ImportError: doSkip = True skipReason = "can't run w/o tkinter" else: try: tkinter.Tk().destroy() except tkinter.TclError as e: doSkip = True skipReason = "Can't test Tkinter: " + str(e) @skipIf(doSkip, skipReason) class ScriptTests(TestCase, ScriptTestsMixin): """ Tests for the Conch scripts. """ def test_conch(self): self.scriptTest("conch/conch") def test_cftp(self): self.scriptTest("conch/cftp") def test_ckeygen(self): self.scriptTest("conch/ckeygen") def test_tkconch(self): self.scriptTest("conch/tkconch") class ZshIntegrationTests(TestCase, ZshScriptTestMixin): """ Test that zsh completion functions are generated without error """ generateFor = [ ("conch", "twisted.conch.scripts.conch.ClientOptions"), ("cftp", "twisted.conch.scripts.cftp.ClientOptions"), ("ckeygen", "twisted.conch.scripts.ckeygen.GeneralOptions"), ("tkconch", "twisted.conch.scripts.tkconch.GeneralOptions"), ] PK ə�Z� ��� � $ __pycache__/__init__.cpython-310.pycnu �[��� o �b � @ s d Z dS )z conch testsN)�__doc__� r r �=/usr/lib/python3/dist-packages/twisted/conch/test/__init__.py�<module> s PK ə�Z� �� � ( __pycache__/test_scripts.cpython-310.pycnu �[��� o �b� � @ s d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z dZ dZ ed �s,d Z d Z ed �s4d Z d Z ed�s<d Z dZzddlZW n eyO d Z dZ Y n%w ze�� �� W n ejys Z zd Z dee� Z W Y dZ[ndZ[ww ee e �G dd� de e��ZG dd� de e�ZdS )z1 Tests for the command-line interfaces to conch. � )�skipIf)� requireModule)�ZshScriptTestMixin)�ScriptTestsMixin)�TestCaseF� �pyasn1TzCannot run without PyASN1� cryptographyzcan't run w/o cryptography�ttyzcan't run w/o ttyNzcan't run w/o tkinterzCan't test Tkinter: c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � ScriptTestsz& Tests for the Conch scripts. c C � | � d� d S )Nz conch/conch�� scriptTest��self� r �A/usr/lib/python3/dist-packages/twisted/conch/test/test_scripts.py� test_conch0 � zScriptTests.test_conchc C r )Nz conch/cftpr r r r r � test_cftp3 r zScriptTests.test_cftpc C r )Nz conch/ckeygenr r r r r � test_ckeygen6 r zScriptTests.test_ckeygenc C r )Nz conch/tkconchr r r r r � test_tkconch9 r zScriptTests.test_tkconchN)�__name__� __module__� __qualname__�__doc__r r r r r r r r r * s r c @ s e Zd ZdZg d�ZdS )�ZshIntegrationTestszH Test that zsh completion functions are generated without error ))�conchz)twisted.conch.scripts.conch.ClientOptions)�cftpz(twisted.conch.scripts.cftp.ClientOptions)�ckeygenz,twisted.conch.scripts.ckeygen.GeneralOptions)�tkconchz,twisted.conch.scripts.tkconch.GeneralOptionsN)r r r r � generateForr r r r r = s r )r �unittestr �twisted.python.reflectr �"twisted.python.test.test_shellcompr �!twisted.scripts.test.test_scriptsr �twisted.trial.unittestr �doSkip� skipReason� cryptoSkip�ttySkip�tkinter� ImportError�Tk�destroy�TclError�e�strr r r r r r �<module> sB ���PK ə�Z�}9 __init__.pynu �[��� "conch tests" PK ���Z���^� � ) __pycache__/test_sentence.cpython-310.pycnu �[��� o �bZ � @ s� d Z ddlZddlmZ ddlmZ dZdZG dd� d�ZG d d � d ej �Z G d d � d ej �Z G d d� dej �Z G dd� d�ZG dd� dee�ZdS )z" Tests for positioning sentences. � N)� _sentence)�TestCase�someStringValue�someOtherStringValuec @ s e Zd ZdZedd� �ZdS )� DummyProtocolz" A simple, fake protocol. c C s dt tgS )N�type)�sentinelValueOne�sentinelValueTwo� r r �H/usr/lib/python3/dist-packages/twisted/positioning/test/test_sentence.py�getSentenceAttributes s z#DummyProtocol.getSentenceAttributesN)�__name__� __module__� __qualname__�__doc__� staticmethodr r r r r r s r c @ � e Zd ZdZe�� ZdS )� DummySentencez* A sentence for L{DummyProtocol}. N)r r r r r r �ALLOWED_ATTRIBUTESr r r r r � r c @ s e Zd ZdZdeedgiZdS )� MixinProtocolz� A simple, fake protocol that declaratively tells you the sentences it produces using L{base.PositioningSentenceProducerMixin}. N)r r r r r r �_SENTENCE_CONTENTSr r r r r # s ��r c @ r )� MixinSentencez* A sentence for L{MixinProtocol}. N)r r r r r r r r r r r r 2 r r c @ sJ e Zd ZdZdd� Zdd� Zdd� ZdZd Zdd d �Z dd� Z dd� Z dS )�SentenceTestsMixinzI Tests for positioning protocols and their respective sentences. c C s: t � }| �t|i�}| �t|t�|� | �t|t�� dS )z� A sentence attribute gets the correct value, and accessing an unset attribute (which is specified as being a valid sentence attribute) gets L{None}. N)�object� sentenceClassr � assertEqual�getattr� assertIsNoner )�self� thisSentinel�sentencer r r �test_attributeAccess? s z'SentenceTestsMixin.test_attributeAccessc C s | � i �}| �tt|d� dS )zM Accessing a nonexistent attribute raises C{AttributeError}. �BOGUSN�r � assertRaises�AttributeErrorr �r r! r r r �"test_raiseOnMissingAttributeAccessJ s z5SentenceTestsMixin.test_raiseOnMissingAttributeAccessc C s" | � ddi�}| �tt|d� dS )z� Accessing bogus attributes raises C{AttributeError}, *even* when that attribute actually is in the sentence data. r# Nr$ r' r r r � test_raiseOnBadAttributeAccessQ s z1SentenceTestsMixin.test_raiseOnBadAttributeAccess�tummiesz<%s (%s) {%s}>� unknown type� c C s | j j}| j|||f S )aQ Builds the expected repr for a sentence. @param sentenceType: The name of the sentence type (e.g "GPGGA"). @type sentenceType: C{str} @param dataRepr: The repr of the data in the sentence. @type dataRepr: C{str} @return: The expected repr of the sentence. @rtype: C{str} )r r � reprTemplate)r � sentenceType�dataRepr�clsNamer r r � _expectedRepr\ s z SentenceTestsMixin._expectedReprc C s& | � i �}| �� }| �t|�|� dS )zE Test the repr of an empty sentence of unknown type. N)r r1 r �repr�r r! � expectedReprr r r �test_unknownTypeReprj s z'SentenceTestsMixin.test_unknownTypeReprc C s0 | � d| ji�}| �| j�}| �t|�|� dS )zC Test the repr of an empty sentence of known type. r N)r r. r1 r r2 r3 r r r �test_knownTypeReprr s z%SentenceTestsMixin.test_knownTypeReprN)r+ r, ) r r r r r" r( r) r. r- r1 r5 r6 r r r r r : s r c @ s e Zd ZdZdd� Zdd� ZdS )� MixinTestszm Tests for protocols deriving from L{base.PositioningSentenceProducerMixin} and their sentences. c C s t � | _t| _d S )N)r �protocolr r )r r r r �setUp� s zMixinTests.setUpc C s< | j �� }| �d|� | j j}tj|�� � }| �d|� dS )a� L{None} does not appear in the sentence attributes of the protocol, even though it's in the specification. This is because L{None} is a placeholder for parts of the sentence you don't really need or want, but there are some bits later on in the sentence that you do want. The alternative would be to have to specify things like "_UNUSED0", "_UNUSED1"... which would end up cluttering the sentence data and eventually adapter state. N)r8 r � assertNotInr � itertools�chain�values�assertIn)r �sentenceAttributes�sentenceContents�sentenceSpecAttributesr r r � test_noNoneInSentenceAttributes� s z*MixinTests.test_noNoneInSentenceAttributesN)r r r r r9 rB r r r r r7 { s r7 )r r; �twisted.positioningr �twisted.trial.unittestr r r r � _BaseSentencer �!_PositioningSentenceProducerMixinr r r r7 r r r r �<module> s APK ���Z�U5{ % __pycache__/test_base.cpython-310.pycnu �[��� o �b�p � @ s� d Z ddlmZ ddlmZ ddlmZmZ ddlm Z ddl m Z G dd� de �Z G d d � d e �Z G d d � d e �Zd d� Zdd� ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd � d e �ZG d d � d e �Zd S ) z( Test cases for positioning primitives. � )�verify)�base)�Angles� Directions)�IPositioningBeacon)�TestCasec @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � AngleTestsz@ Tests for the L{twisted.positioning.base.Angle} class. c C s t �� }| �dt|�� dS )zd The repr of an empty angle says that is of unknown type and unknown value. z'<Angle of unknown type (unknown value)>N�r �Angle� assertEqual�repr��self�a� r �D/usr/lib/python3/dist-packages/twisted/positioning/test/test_base.py� test_empty s zAngleTests.test_emptyc C s" t jtjd�}| �dt|�� dS )zf The repr of an empty variation says that it is a variation of unknown value. �� angleTypez<Variation (unknown value)>N)r r r � VARIATIONr r r r r r �test_variation s zAngleTests.test_variationc C s t �d�}| �dt|�� dS )zz The repr of an angle of unknown type but a given value displays that type and value in its repr. � �?z%<Angle of unknown type (1.0 degrees)>Nr r r r r �test_unknownType# s zAngleTests.test_unknownTypec C s | j ttjdd� dS )zS Trying to create an angle with a bogus type raises C{ValueError}. �BOGUSr N)� assertRaises� ValueErrorr r �r r r r �test_bogusType+ � zAngleTests.test_bogusTypeN)�__name__� __module__� __qualname__�__doc__r r r r r r r r r s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.S )/� HeadingTestszB Tests for the L{twisted.positioning.base.Heading} class. c C sD t �d�}| �|jd� | �t|�d� | �|j� | �|j� dS )z� Tests that a simple heading has a value in decimal degrees, which is also its value when converted to a float. Its variation, and by consequence its corrected heading, is L{None}. r N)r �Headingr �inDecimalDegrees�float� assertIsNone� variation�correctedHeading�r �hr r r � test_simple7 s zHeadingTests.test_simplec C s" t �d�}d}| �t|�|� dS )zt A repr of a heading with no variation reports its value and that the variation is unknown. r z*<Heading (1.0 degrees, unknown variation)>N)r r% r r )r �heading� expectedReprr r r � test_headingWithoutVariationReprC s z-HeadingTests.test_headingWithoutVariationReprc C s8 d\}}t jj||d�}d}| �t|�|�||�� dS )zu A repr of a heading with known variation reports its value and the value of that variation. )r � $���variationValuez2<Heading ({0} degrees, <Variation ({1} degrees)>)>N)r r% � fromFloatsr r �format)r �angler) r. � reprTemplater r r � test_headingWithVariationReprL s z*HeadingTests.test_headingWithVariationReprc C s | � t�d�t�d�� dS )z> Headings with the same values compare equal. r N)r r r% r r r r �test_valueEqualityV � z HeadingTests.test_valueEqualityc C s | � t�d�t�d�� dS )zA Headings with different values compare unequal. r � @N)�assertNotEqualr r% r r r r �test_valueInequality\ r: z!HeadingTests.test_valueInequalityc C s t �d� dS )zR Headings can be instantiated with a value of 0 and no variation. r N�r r% r r r r �test_zeroHeadingEdgeCaseb � z%HeadingTests.test_zeroHeadingEdgeCasec C s t �dd� dS )zh Headings can be instantiated with a value of 0 and a variation of 180 degrees. r � Nr> r r r r �*test_zeroHeading180DegreeVariationEdgeCaseh s z7HeadingTests.test_zeroHeading180DegreeVariationEdgeCasec K s | j ttjjfi |�� dS )z� Helper function for verifying that bad values raise C{ValueError}. @param kw: The keyword arguments passed to L{base.Heading.fromFloats}. N)r r r r% r4 )r �kwr r r � _badValueTesto s zHeadingTests._badValueTestc C � | j dd� dS )zO Headings can not be instantiated with a value of 360 degrees. g �v@�� angleValueN�rD r r r r �test_badAngleValueEdgeCasew � z'HeadingTests.test_badAngleValueEdgeCasec C rE )zT Headings can not be instantiated with a variation of -180 degrees. � �f�r2 NrH r r r r �test_badVariationEdgeCase} rJ z&HeadingTests.test_badVariationEdgeCasec C rE )z> Negative heading values raise C{ValueError}. r1 rF NrH r r r r �test_negativeHeading� rJ z!HeadingTests.test_negativeHeadingc C rE )zK Heading values greater than C{360.0} raise C{ValueError}. g w@rF NrH r r r r �test_headingTooLarge� rJ z!HeadingTests.test_headingTooLargec C rE )zK Variation values less than C{-180.0} raise C{ValueError}. g �g�r2 NrH r r r r �test_variationTooNegative� rJ z&HeadingTests.test_variationTooNegativec C rE )zM Variation values greater than C{180.0} raise C{ValueError}. g �g@r2 NrH r r r r �test_variationTooPositive� rJ z&HeadingTests.test_variationTooPositivec C �, t jjddd�}| �|jt �dtj�� dS )zQ A heading with a value and a variation has a corrected heading. r r1 r2 g &@N�r r% r4 r r* r r �HEADINGr+ r r r �test_correctedHeading� s z"HeadingTests.test_correctedHeadingc C rQ )z� A heading with a value and a variation has the appropriate corrected heading value, even when the variation puts it across the 360 degree boundary. � pv@g �r2 r NrR r+ r r r � test_correctedHeadingOverflow� � z*HeadingTests.test_correctedHeadingOverflowc C rQ )z� A heading with a value and a variation has the appropriate corrected heading value, even when the variation puts it exactly at the 360 degree boundary. rU � �r2 � NrR r+ r r r �%test_correctedHeadingOverflowEdgeCase� rW z2HeadingTests.test_correctedHeadingOverflowEdgeCasec C rQ )z� A heading with a value and a variation has the appropriate corrected heading value, even when the variation puts it under the 0 degree boundary. r r; r2 rU NrR r+ r r r � test_correctedHeadingUnderflow� rW z+HeadingTests.test_correctedHeadingUnderflowc C s, t jjddd�}| �|jt �dtj�� dS )z� A heading with a value and a variation has the appropriate corrected heading value, even when the variation puts it exactly at the 0 degree boundary. r r2 rY NrR r+ r r r �&test_correctedHeadingUnderflowEdgeCase� rW z3HeadingTests.test_correctedHeadingUnderflowEdgeCasec C sH t jjddd�}|�d� | �|jjd� |�d� | �|jjd� dS )zK Setting the sign of a heading changes the variation sign. r r2 � ���rX N)r r% r4 �setSignr r) r& r+ r r r �test_setVariationSign� s z"HeadingTests.test_setVariationSignc C st t jjddd�}| �t|jd� | �|jjd� | �t|jd� | �|jjd� | �t|jd� | �|jjd� dS )z� Setting the sign of a heading to values that aren't C{-1} or C{1} raises C{ValueError} and does not affect the heading. r r2 ����r �2 N) r r% r4 r r r_ r r) r& r+ r r r �test_setBadVariationSign� s z%HeadingTests.test_setBadVariationSignc C s. t j�d�}| �|jj� | �t|jd� dS )zd Setting the sign on a heading with unknown variation raises C{ValueError}. r r] N) r r% r4 r( r) r& r r r_ r+ r r r � test_setUnknownVariationSign� s z)HeadingTests.test_setUnknownVariationSignN)r r! r" r# r- r0 r8 r9 r= r? rB rD rI rL rM rN rO rP rT rV rZ r[ r\ r` rc rd r r r r r$ 2 s0 r$ c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1S )2�CoordinateTestsc C s t �d�}| �t|�d� dS )z9 Coordinates can be converted to floats. � $@N)r � Coordinater r' �r � coordinater r r � test_float� � zCoordinateTests.test_floatc C s* t �d�}dd� d�}| �t|�|� dS )zn Coordinates that aren't explicitly latitudes or longitudes have an appropriate repr. rf z<Angle of unknown type (� degrees)>N)r rg r r �r ri r/ r r r � test_repr� s zCoordinateTests.test_reprc C �. t �dtj�}dd� d�}| �t|�|� dS )zU Positive latitudes have a repr that specifies their type and value. rf � <Latitude (rl N�r rg r �LATITUDEr r rm r r r �test_positiveLatitude� � z%CoordinateTests.test_positiveLatitudec C ro )zU Negative latitudes have a repr that specifies their type and value. � I�rp rl Nrq rm r r r �test_negativeLatitude rt z%CoordinateTests.test_negativeLatitudec C ro )zV Positive longitudes have a repr that specifies their type and value. � I@� <Longitude (rl N�r rg r � LONGITUDEr r �r � longituder/ r r r �test_positiveLongitude rt z&CoordinateTests.test_positiveLongitudec C ro )zV Negative longitudes have a repr that specifies their type and value. ru rx rl Nry r{ r r r �test_negativeLongitude rt z&CoordinateTests.test_negativeLongitudec C s | � ttjdd� dS )zM Creating coordinates with bogus types rasies C{ValueError}. � �b@r N)r r r rg r r r r �test_bogusCoordinateType r z(CoordinateTests.test_bogusCoordinateTypec C s | � ttjdtj� dS )zm Creating coordinates with angle types that aren't coordinates raises C{ValueError}. r N)r r r rg r rS r r r r �test_angleTypeNotCoordinate s z+CoordinateTests.test_angleTypeNotCoordinatec C s dd� }| � |� |� � dS )zE Coordinates with the same value and type are equal. c S s t �dtj�S )Nr �r rg r rz r r r r �makeCoordinate, s z5CoordinateTests.test_equality.<locals>.makeCoordinateN)r )r r� r r r � test_equality' s z CoordinateTests.test_equalityc C �$ t �d�}t �d�}| �||� dS )zA Coordinates with different values aren't equal. r rX N)r rg r< �r �c1�c2r r r � test_differentAnglesInequality1 � z.CoordinateTests.test_differentAnglesInequalityc C s, t �dtj�}t �dtj�}| �||� dS )zT Coordinates with the same values but different types aren't equal. r N)r rg r rr rz r< r� r r r � test_differentTypesInequality9 s z-CoordinateTests.test_differentTypesInequalityc C sB t �dtj�}|�d� | �|jd� |�d� | �|jd� dS )zh Setting the sign on a coordinate sets the sign of the value of the coordinate. rw r] r^ ru N)r rg r rr r_ r r& �r �cr r r � test_signA s zCoordinateTests.test_signc C sp d}t �|tj�}| �t|jd� | �|jd� | �t|jd� | �|jd� | �t|jd� | �|jd� dS )z� Setting a bogus sign value (not -1 or 1) on a coordinate raises C{ValueError} and doesn't affect the coordinate. rw ra r rb N) r rg r rr r r r_ r r& )r �valuer� r r r �test_badVariationSignL s z%CoordinateTests.test_badVariationSignc C �" t �dtj�}| �|jtj� dS )zD Positive latitudes are in the northern hemisphere. r N)r rg r rr r � hemispherer �NORTHrh r r r �test_northernHemisphere] � z'CoordinateTests.test_northernHemispherec C r� )zD Positive longitudes are in the eastern hemisphere. r N)r rg r rz r r� r �EASTrh r r r �test_easternHemisphered r� z&CoordinateTests.test_easternHemispherec C r� )zD Negative latitudes are in the southern hemisphere. rX N)r rg r rr r r� r �SOUTHrh r r r �test_southernHemispherek r� z'CoordinateTests.test_southernHemispherec C r� )zD Negative longitudes are in the western hemisphere. rX N)r rg r rz r r� r �WESTrh r r r �test_westernHemispherer r� z&CoordinateTests.test_westernHemispherec s$ t �dd�� | �t� fdd�� dS )zo Accessing the hemisphere for a coordinate that can't compute it raises C{ValueError}. r Nc s � j S )N)r� r �ri r r �<lambda> s z4CoordinateTests.test_badHemisphere.<locals>.<lambda>)r rg r r r r r� r �test_badHemispherey s z"CoordinateTests.test_badHemispherec C � | � ttd� | � ttd� dS )zt Creating a latitude with a value greater than or equal to 90 degrees raises C{ValueError}. r g �V@N�r r � _makeLatituder r r r �test_latitudeTooLarge� � z%CoordinateTests.test_latitudeTooLargec C r� )zr Creating a latitude with a value less than or equal to -90 degrees raises C{ValueError}. g �b�g �V�Nr� r r r r �test_latitudeTooSmall� r� z%CoordinateTests.test_latitudeTooSmallc C r� )zv Creating a longitude with a value greater than or equal to 180 degrees raises C{ValueError}. g @o@g �f@N�r r �_makeLongituder r r r �test_longitudeTooLarge� r� z&CoordinateTests.test_longitudeTooLargec C r� )zt Creating a longitude with a value less than or equal to -180 degrees raises C{ValueError}. g @o�rK Nr� r r r r �test_longitudeTooSmall� r� z&CoordinateTests.test_longitudeTooSmallc C s< t �dtj�}| �|jd� t �dtj�}| �|jd� dS )zQ Coordinate values can be accessed in degrees, minutes, seconds. g @I@)rb � r g%��CI@)rb � �. N)r rg r rr r �inDegreesMinutesSecondsr� r r r � test_inDegreesMinutesSeconds� s z,CoordinateTests.test_inDegreesMinutesSecondsc C s t �dd�}| �|j� dS )zz If the vaue of a coordinate is L{None}, its values in degrees, minutes, seconds is also L{None}. N)r rg r( r� r� r r r �(test_unknownAngleInDegreesMinutesSeconds� s z8CoordinateTests.test_unknownAngleInDegreesMinutesSecondsN)r r! r" rj rn rs rv r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r re � s2 re c C � t �| tj�S )z7 Builds and returns a latitude of given value. )r rg r rr �r� r r r r� � r@ r� c C r� )z8 Builds and returns a longitude of given value. r� r� r r r r� � r@ r� c @ r ) � AltitudeTestszC Tests for the L{twisted.positioning.base.Altitude} class. c C s@ t �d�}| �t|�d� | �|jd� | �|jdt j � dS )z� Altitudes can be instantiated and reports the correct value in meters and feet, as well as when converted to float. r N)r �Altituder r' �inMeters�inFeet�METERS_PER_FOOT�r �altituder r r � test_value� s zAltitudeTests.test_valuec C � t �d�}| �t|�d� dS )zF Altitudes report their type and value in their repr. r z<Altitude (1.0 m)>N)r r� r r r� r r r rn � rk zAltitudeTests.test_reprc C s$ t �d�}t �d�}| �||� dS )z< Altitudes with equal values compare equal. r N)r r� r �r � firstAltitude�secondAltituder r r r� � r� zAltitudeTests.test_equalityc C r� )zF Altitudes with different values don't compare equal. r rX N)r r� r< r� r r r �test_inequality� r� z AltitudeTests.test_inequalityN)r r! r" r# r� rn r� r� r r r r r� � s r� c @ �8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � SpeedTestsz@ Tests for the L{twisted.positioning.base.Speed} class. c C �, t �d�}| �|jd� | �t|�d� dS )�� Speeds can be instantiated, and report their value in meters per second, and can be converted to floats. rw N)r �Speedr �inMetersPerSecondr' �r �speedr r r r� � � zSpeedTests.test_valuec C r� )zC Speeds report their type and value in their repr. rw z<Speed (50.0 m/s)>N)r r� r r r� r r r rn � rk zSpeedTests.test_reprc C s | � ttjd� dS )zA Creating a negative speed raises C{ValueError}. rX N)r r r r� r r r r �test_negativeSpeeds� s z SpeedTests.test_negativeSpeedsc C �" t �d�}| �dt j |j� dS )zC A speed can be converted into its value in knots. r r] N)r r� r � MPS_PER_KNOT�inKnotsr� r r r � test_inKnots � zSpeedTests.test_inKnotsc C � | � dtt�d��� dS )z; A speed can be converted into a C{float}. r N)r r' r r� r r r r � test_asFloat � zSpeedTests.test_asFloatN) r r! r" r# r� rn r� r� r� r r r r r� � s r� c @ r� ) � ClimbTestsz6 Tests for L{twisted.positioning.base.Climb}. c C r� )r� � E@N�r �Climbr r� r' �r �climbr r r r- r� zClimbTests.test_simplec C r� )zC Climbs report their type and value in their repr. r� z<Climb (42.0 m/s)>N)r r� r r r� r r r rn rk zClimbTests.test_reprc C r� )z� Climbs can have negative values, and still report that value in meters per second and when converted to floats. g E�Nr� r� r r r �test_negativeClimbs' r� z ClimbTests.test_negativeClimbsc C r� )zC A climb can be converted into its value in knots. r r] N)r r� r r� r� r� r r r �test_speedInKnots0 r� z ClimbTests.test_speedInKnotsc C r� )z; A climb can be converted into a C{float}. r N)r r' r r� r r r r r� 7 r� zClimbTests.test_asFloatN) r r! r" r# r- rn r� r� r� r r r r r� s r� c @ st e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dZ dd� Z dd� Zdd� Zdd� ZdS ) �PositionErrorTestsz> Tests for L{twisted.positioning.base.PositionError}. c C s0 t �� }| �|j� | �|j� | �|j� dS )z~ In an empty L{base.PositionError} with no invariant testing, all dilutions of positions are L{None}. N�r � PositionErrorr( �pdop�hdop�vdop�r � positionErrorr r r � test_allUnsetC s z PositionErrorTests.test_allUnsetc C s4 t jdd�}| �|j� | �|j� | �|j� dS )z{ In an empty L{base.PositionError} with invariant testing, all dilutions of positions are L{None}. T)� testInvariantNr� r� r r r �test_allUnsetWithInvariantM s z-PositionErrorTests.test_allUnsetWithInvariantc C s t jdd�}| �|jd� dS )zN L{base.PositionError}s can be instantiated with just a HDOP. r )r� N�r r� r r� r� r r r �test_withoutInvariantW s z(PositionErrorTests.test_withoutInvariantc C s t jddd�}| �|jd� dS )zv Creating a simple L{base.PositionError} with just a HDOP while checking the invariant works. r T)r� r� Nr� r� r r r �test_withInvariant^ r� z%PositionErrorTests.test_withInvariantc C s> t jdddd�}| �|jd� | �|jd� | �|jd� dS )z� Creating a L{base.PositionError} with values set to an impossible combination works if the invariant is not checked. r )r� r� r� N)r r� r r� r� r� )r �errorr r r � test_invalidWithoutInvariantf s z/PositionErrorTests.test_invalidWithoutInvariantc C s | j ttjddddd� dS )z� Creating a L{base.PositionError} with values set to an impossible combination raises C{ValueError} if the invariant is being tested. r T)r� r� r� r� N)r r r r� r r r r �test_invalidWithInvariantp s �z,PositionErrorTests.test_invalidWithInvariantc C s& t jddd�}d|_| �|jd� dS )z} You can set the PDOP value to value inconsisted with HDOP and VDOP when not checking the invariant. r �r� r� � Y@N)r r� r� r )r �per r r �test_setDOPWithoutInvariant~ s z.PositionErrorTests.test_setDOPWithoutInvariantc C s> t jdddd�}|j}dd� }| �t||� | �|j|� dS )z� Attempting to set the PDOP value to value inconsisted with HDOP and VDOP when checking the invariant raises C{ValueError}. r T)r� r� r� c S s d| _ d S )Nr� )r� )r� r r r �setPDOP� s z<PositionErrorTests.test_setDOPWithInvariant.<locals>.setPDOPN)r r� r� r r r )r r� r� r� r r r �test_setDOPWithInvariant� s z+PositionErrorTests.test_setDOPWithInvariantz.<PositionError (pdop: %s, hdop: %s, vdop: %s)>c C sJ | � |j|� | � |j|� | � |j|� | � t|�| j|||f � dS )a� Tests the DOP values in a position error, and the repr of that position error. @param pe: The position error under test. @type pe: C{PositionError} @param pdop: The expected position dilution of precision. @type pdop: C{float} or L{None} @param hdop: The expected horizontal dilution of precision. @type hdop: C{float} or L{None} @param vdop: The expected vertical dilution of precision. @type vdop: C{float} or L{None} N)r r� r� r� r � REPR_TEMPLATE)r r� r� r� r� r r r �_testDOP� s zPositionErrorTests._testDOPc C s> d\}}|d |d d }t j||d�}| �||||� dS )zF The VDOP is correctly determined from PDOP and HDOP. �r; r � � �?)r� r� N�r r� r� )r r� r� r� r� r r r � test_positionAndHorizontalSet� � z0PositionErrorTests.test_positionAndHorizontalSetc C s> d\}}|d |d d }t j||d�}| �||||� dS )zF The HDOP is correctly determined from PDOP and VDOP. r� r� r� )r� r� Nr� )r r� r� r� r� r r r �test_positionAndVerticalSet� r� z.PositionErrorTests.test_positionAndVerticalSetc C s> d\}}|d |d d }t j||d�}| �||||� dS )zF The PDOP is correctly determined from HDOP and VDOP. )r r r� r� r� Nr� )r r� r� r� r� r r r � test_horizontalAndVerticalSet� r� z0PositionErrorTests.test_horizontalAndVerticalSetN)r r! r" r# r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� > s r� c @ s4 e Zd ZdZdd� Zdddd�Zdd� Zd d � Zd S ) �BeaconInformationTestszB Tests for L{twisted.positioning.base.BeaconInformation}. c C s2 t �� }| �t|j�d� d}| �t|�|� dS )z� For an empty beacon information object, the number of used beacons is zero, the number of seen beacons is zero, and the repr of the object reflects that. r z><BeaconInformation (used beacons (0): [], unused beacons: [])>N)r �BeaconInformationr �len� usedBeaconsr )r �bir/ r r r � test_minimal� s �z#BeaconInformationTests.test_minimalr] r )�azimuth� elevation�signalToNoiseRatioc st � fdd�}t � }tdd�D ] }|�||d�� qt�|�}� �t|j�d� � �t|j�d� � �t |�d� d S ) zw Tests a beacon information with a bunch of satellites, none of which used in computing a fix. c s$ t � j�}|�| � tjdi |��S )Nr )�dict�satelliteKwargs�updater � Satellite)rC �kwargsr r r �_buildSatellite� s z;BeaconInformationTests.test_simple.<locals>._buildSatelliter] � )� identifier� r a <BeaconInformation (used beacons (0): [], unused beacons: [<Satellite (1), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (2), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (3), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (4), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (5), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (6), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (7), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (8), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (9), azimuth: 1, elevation: 1, snr: 1.0>])>N) �set�range�addr r� r r� � seenBeaconsr� r )r r �beacons�prnr� r r r r- � s �z"BeaconInformationTests.test_simplec C s� t �� }tdd�D ] }t jd d|i| j��}|j�|� |d r&|j�|� q | �t |j�d� | �t |j�d� | �t |�d� dS ) zv Tests a beacon information with a bunch of satellites, some of them used in computing a fix. r] r r r� r � a <BeaconInformation (used beacons (5): [<Satellite (1), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (3), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (5), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (7), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (9), azimuth: 1, elevation: 1, snr: 1.0>], unused beacons: [<Satellite (2), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (4), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (6), azimuth: 1, elevation: 1, snr: 1.0>, <Satellite (8), azimuth: 1, elevation: 1, snr: 1.0>])>Nr ) r r� r r r r r r� r r� r )r r� r � satelliter r r �test_someSatellitesUsed� s ��z.BeaconInformationTests.test_someSatellitesUsedN)r r! r" r# r� r r- r r r r r r� � s #r� c @ � e Zd ZdZdd� Zdd� ZdS )�PositioningBeaconTestsz. Tests for L{base.PositioningBeacon}. c C s, t �tj�}| �|� t�t t�d�� dS )zX Tests that L{base.PositioningBeacon} implements L{IPositioningBeacon}. r] N)r � implementedByr �PositioningBeacon� assertTruer � verifyObject)r � implementsr r r �test_interface" s z%PositioningBeaconTests.test_interfacec C s | � tt�d��d� dS )z9 Tests the repr of a positioning beacon. �Az <Beacon (A)>N)r r r r r r r r rn * r� z PositioningBeaconTests.test_reprN)r r! r" r# r rn r r r r r s r c @ r )�SatelliteTestsz: Tests for L{twisted.positioning.base.Satellite}. c C sP t �d�}| �|jd� | �|j� | �|j� | �|j� | �t|�d� dS )z� Tests a minimal satellite that only has a known PRN. Tests that the azimuth, elevation and signal to noise ratios are L{None} and verifies the repr. r] z:<Satellite (1), azimuth: None, elevation: None, snr: None>N) r r r r r( r� r r r �r �sr r r r� 6 s �zSatelliteTests.test_minimalc C s^ t jddddd�}| �|jd� | �|jd� | �|jd� | �|jd� | �t|�d� dS )z� Tests a minimal satellite that only has a known PRN. Tests that the azimuth, elevation and signal to noise ratios are correct and verifies the repr. r] g �p@g >@g 9@)r r� r r z;<Satellite (1), azimuth: 270.0, elevation: 30.0, snr: 25.0>N)r r r r r� r r r r r r r r- F s ��zSatelliteTests.test_simpleN)r r! r" r# r� r- r r r r r 1 s r N)r# �zope.interfacer �twisted.positioningr �twisted.positioning.baser r � twisted.positioning.ipositioningr �twisted.trial.unittestr r r$ re r� r� r� r� r� r� r� r r r r r r �<module> s* $ 8 L'), WPK ���Z�"b}� � $ __pycache__/receiver.cpython-310.pycnu �[��� o �b] � @ s* d Z ddlmZmZ G dd� dej�ZdS )z Receivers for use in tests. � )�base� ipositioningc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �MockPositioningReceivera A mock positioning receiver. Mocks all the L{IPositioningReceiver} methods with stubs that don't do anything but register that they were called. @ivar called: A mapping of names of callbacks that have been called to C{True}. @type called: C{dict} c C s" | � � tjD ]}| �|� qd S )N)�clearr �IPositioningReceiver� _addCallback)�self� methodName� r �C/usr/lib/python3/dist-packages/twisted/positioning/test/receiver.py�__init__ s �z MockPositioningReceiver.__init__c C s i | _ dS )zt Forget all the methods that have been called on this receiver, by emptying C{self.called}. N��called)r r r r r s z MockPositioningReceiver.clearc s � �fdd�}t �� |� dS )zq Adds a callback of the given name, setting C{self.called[name]} to C{True} when called. c s d�j � < d S )NTr )�a�kw��namer r r �callback* s z6MockPositioningReceiver._addCallback.<locals>.callbackN)�setattr)r r r r r r r $ s z$MockPositioningReceiver._addCallbackN)�__name__� __module__� __qualname__�__doc__r r r r r r r r s r N)r �twisted.positioningr r �BasePositioningReceiverr r r r r �<module> s PK ���Z��t� � % __pycache__/test_nmea.cpython-310.pycnu �[��� o �b`� � @ s( d Z ddlZddlmZ ddlmZ ddlmZmZm Z ddl m Z ddl m Z ddlmZ d Zd Zd Zd Zd ZdZdZdZd�� Z\ZZZ eej �G dd� d��Z G dd� de�Z G dd� de�Z G dd� de�Z!G dd� de�Z"G d d � d �Z#G d d � d e#e�Z$G d d!� d!e#e�Z%G d"d#� d#e#e�Z&G d$d%� d%e#e�Z'G d&d'� d'e�Z(G d(d)� d)�Z)G d*d+� d+e)e�Z*G d,d-� d-e)e�Z+d.d/� Z,d0d1� Z-d2d3� Z.G d4d5� d5e)e�Z/G d6d7� d7e)e�Z0G d8d9� d9e)e�Z1G d:d;� d;e)e�Z2G d<d=� d=e)e�Z3G d>d?� d?e)e�Z4G d@dA� dAe)e�Z5G dBdC� dCe�Z6dS )Dz& Test cases for using NMEA sentences. � N)� attrgetter)� implementer)�base� ipositioning�nmea)�Angles)�MockPositioningReceiver)�TestCasesA $GPGGA,123519,4807.038,N,01131.000,E,1,08,0.9,545.4,M,46.9,M,,*47sD $GPRMC,123519,A,4807.038,N,01131.000,E,022.4,084.4,230394,003.1,W*6As5 $GPGSA,A,3,19,28,14,18,27,22,31,39,,,,,1.7,1.0,1.3*34s $GPHDT,038.005,T*3Bs' $GPGLL,4916.45,N,12311.12,W,225444,A*31s $GPGLL,3751.65,S,14507.36,E*77s; $GPGSV,1,1,11,03,03,111,00,04,15,270,00,06,01,010,00,,,,*4bs2 $GPGSV,1,1,11,03,03,111,00,,,,,,,,,13,06,292,00*75s� $GPGSV,3,1,11,03,03,111,00,04,15,270,00,06,01,010,00,13,06,292,00*74 $GPGSV,3,2,11,14,25,170,00,16,57,208,39,18,67,296,40,19,40,246,00*74 $GPGSV,3,3,11,22,42,067,42,24,14,311,43,27,05,244,00,,,,*4D c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) �NMEATestReceiverzY An NMEA receiver for testing. Remembers the last sentence it has received. c C s | � � d S �N)�clear��self� r �D/usr/lib/python3/dist-packages/twisted/positioning/test/test_nmea.py�__init__/ s zNMEATestReceiver.__init__c C s d| _ dS )zq Forgets the received sentence (if any), by setting C{self.receivedSentence} to L{None}. N��receivedSentencer r r r r 2 s zNMEATestReceiver.clearc C s || _ d S r r �r �sentencer r r �sentenceReceived9 � z!NMEATestReceiver.sentenceReceivedN)�__name__� __module__� __qualname__�__doc__r r r r r r r r ' s r c @ r ) � CallbackTestsaO Tests if the NMEA protocol correctly calls its sentence callback. @ivar protocol: The NMEA protocol under test. @type protocol: L{nmea.NMEAProtocol} @ivar sentenceTypes: The set of sentence types of all sentences the test's sentence callback function has been called with. @type sentenceTypes: C{set} c C s" t � }t�|| j�| _t� | _d S r )r r � NMEAProtocol�_sentenceCallback�protocol�set� sentenceTypes�r �receiverr r r �setUpH s zCallbackTests.setUpc C s | j �|j� dS )zC Remembers that a sentence of this type was fired. N)r" �add�typer r r r r M � z CallbackTests._sentenceCallbackc C sd dgdgdgdgdgdgd�}|� � D ] \}}|D ]}| j�|� | �| j|h� | j�� qqdS ) zI The correct callbacks fire, and that *only* those fire. � $GPGGA*56s $GPGLL*50s $GPGSA*42s $GPGSV*55s $GPHDT*4fs $GPRMC*4b)�GPGGA�GPGLL�GPGSA�GPGSV�GPHDT�GPRMCN)�itemsr � lineReceived� assertEqualr" r )r �sentencesByType� sentenceType� sentencesr r r r �test_callbacksCalledS s � ��z"CallbackTests.test_callbacksCalledN)r r r r r% r r6 r r r r r = s r c @ r ) �BrokenSentenceCallbackTestsz3 Tests for broken NMEA sentence callbacks. c C s t � }t�|| j�| _d S r )r r r r r r# r r r r% l s z!BrokenSentenceCallbackTests.setUpc C s t d��)z+ Raises C{AttributeError}. zERROR!!!)�AttributeErrorr r r r r p s z-BrokenSentenceCallbackTests._sentenceCallbackc C s | j j}| �t|d� dS )zt An C{AttributeError} in the sentence callback of an C{NMEAProtocol} doesn't get swallowed. r) N)r r1 � assertRaisesr8 )r r1 r r r �"test_dontSwallowCallbackExceptionsv � z>BrokenSentenceCallbackTests.test_dontSwallowCallbackExceptionsN)r r r r r% r r: r r r r r7 g s r7 c @ � e Zd ZdZdd� Zdd� ZdS )� SplitTestsz- Checks splitting of NMEA sentences. c C � t �d�}| �|g d�� dS )zH An NMEA sentence with a checksum gets split correctly. s $GPGGA,spam,eggs*00�s GPGGAs spams eggsN�r �_splitr2 �r � splitSentencer r r �test_withChecksum� � z SplitTests.test_withChecksumc C r> )zK An NMEA sentence without a checksum gets split correctly. s $GPGGA,spam,eggs*r? Nr@ rB r r r �test_noCheckum� rE zSplitTests.test_noCheckumN)r r r r rD rF r r r r r= s r= c @ r ) � ChecksumTestsz4 NMEA sentence checksum verification tests. c C s t �t� dS )z? Sentences with valid checksums get validated. N�r �_validateChecksumr* r r r r � test_valid� s zChecksumTests.test_validc C s t �tdd� � dS )zA Sentences with missing checksums get validated. N���rH r r r r � test_missing� � zChecksumTests.test_missingc C sT t j}t�d�\}}dt|d�d f }|d | g}|D ] }| �tj||� q dS )zw Sentences with a bad checksum raise L{base.InvalidChecksum} when attempting to validate them. � *s %d� � N)r rI r* �split�intr9 r �InvalidChecksum)r �validate� bareSentence�checksum� badChecksumr5 �sr r r � test_invalid� s �zChecksumTests.test_invalidN)r r r r rJ rL rY r r r r rG � s rG c @ � e Zd ZdZdd� ZdS )�NMEAReceiverSetupaG A mixin for tests that need an NMEA receiver (and a protocol attached to it). @ivar receiver: An NMEA receiver that remembers the last sentence. @type receiver: L{NMEATestReceiver} @ivar protocol: An NMEA protocol attached to the receiver. @type protocol: L{twisted.positioning.nmea.NMEAProtocol} c C s t � | _t�| j�| _dS )z+ Sets up an NMEA receiver. N)r r$ r r r r r r r r% � � zNMEAReceiverSetup.setUpN)r r r r r% r r r r r[ � s r[ c @ r ) �GSVSequenceTestsz8 Tests for the interpretation of GSV sequences. c C �4 | j �t� | jj}| �|�� � | �|�� � dS )zO The first sentence in a GSV sequence is correctly identified. N) r r1 � GPGSV_FIRSTr$ r � assertTrue�_isFirstGSVSentence� assertFalse�_isLastGSVSentencer r r r �test_firstSentence� � z#GSVSequenceTests.test_firstSentencec C s4 | j �t� | jj}| �|�� � | �|�� � dS )z� A sentence in the middle of a GSV sequence is correctly identified (as being neither the last nor the first). N)r r1 � GPGSV_MIDDLEr$ r rb ra rc r r r r �test_middleSentence� s z$GSVSequenceTests.test_middleSentencec C r^ )zN The last sentence in a GSV sequence is correctly identified. N) r r1 � GPGSV_LASTr$ r rb ra r` rc r r r r �test_lastSentence� re z"GSVSequenceTests.test_lastSentenceN)r r r r rd rg ri r r r r r] � s r] c @ r ) �BogusSentenceTestszK Tests for verifying predictable failure for bogus NMEA sentences. c C s | � || jj|� dS )a= Asserts that the protocol raises C{exceptionClass} when it receives C{sentence}. @param exceptionClass: The exception class expected to be raised. @type exceptionClass: C{Exception} subclass @param sentence: The (bogus) NMEA sentence. @type sentence: C{str} N)r9 r r1 )r �exceptionClassr r r r �assertRaisesOnSentence� s z)BogusSentenceTests.assertRaisesOnSentencec C s | � td� dS )z` Receiving a well-formed sentence of unknown type raises C{ValueError}. s $GPBOGUS*5bN)rl � ValueErrorr r r r � test_raiseOnUnknownSentenceType� � z2BogusSentenceTests.test_raiseOnUnknownSentenceTypec C s | � tjd� dS )zP Receiving a malformed sentence raises L{base.InvalidSentence}. �GPBOGUSN)rl r �InvalidSentencer r r r � test_raiseOnMalformedSentences r( z1BogusSentenceTests.test_raiseOnMalformedSentencesN)r r r r rl rn rr r r r r rj � s rj c @ rZ )�NMEASentenceTestsz1 Tests for L{nmea.NMEASentence} objects. c C s@ t dfg}|D ]\}}| j�|� | jj}| �t|�|� qdS )zI The C{repr} of L{nmea.NMEASentence} objects is correct. aX <NMEASentence (GPGSA) {dataMode: A, fixType: 3, horizontalDilutionOfPrecision: 1.0, positionDilutionOfPrecision: 1.7, usedSatellitePRN_0: 19, usedSatellitePRN_1: 28, usedSatellitePRN_2: 14, usedSatellitePRN_3: 18, usedSatellitePRN_4: 27, usedSatellitePRN_5: 22, usedSatellitePRN_6: 31, usedSatellitePRN_7: 39, verticalDilutionOfPrecision: 1.3}>N)r, r r1 r$ r r2 �repr)r �sentencesWithExpectedReprr � expectedRepr�receivedr r r � test_repr s �� �zNMEASentenceTests.test_reprN)r r r r rx r r r r rs � rs c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )� ParsingTestsz� Tests if raw NMEA sentences get parsed correctly. This doesn't really involve any interpretation, just turning ugly raw NMEA representations into objects that are more pleasant to work with. c C s& | j �|� | jj}| �||j� dS )z� Passes a sentence to the protocol and gets the parsed sentence from the receiver. Then verifies that the parsed sentence contains the expected data. N)r r1 r$ r r2 � _sentenceData)r r �expectedrw r r r � _parserTest7 s zParsingTests._parserTestc C s. ddddddddd d d d d � }| � t|� dS )z: A full RMC sentence is correctly parsed. r/ �4807.038�N� 01131.000�Ez003.1�Wz022.4�123519�230394z084.4�A) r' � latitudeFloat�latitudeHemisphere�longitudeFloat�longitudeHemisphere�magneticVariation�magneticVariationDirection� speedInKnots� timestamp� datestamp� trueHeading�dataModeN)r} r/ �r r| r r r � test_fullRMCA s �zParsingTests.test_fullRMCc C s0 dddddddddd d d d d � }| � t|� dS )z: A full GGA sentence is correctly parsed. r* �545.4�M�46.9z0.9r~ r r� r� �08r� �1) r' �altitude� altitudeUnits�heightOfGeoidAboveWGS84� heightOfGeoidAboveWGS84Units� horizontalDilutionOfPrecisionr� r� r� r� �numberOfSatellitesSeenr� � fixQualityN)r} r* r� r r r � test_fullGGAU s �zParsingTests.test_fullGGAc C s$ dddddddd�}| � t|� d S ) z: A full GLL sentence is correctly parsed. r+ z4916.45r z12311.12r� �225444r� )r' r� r� r� r� r� r� N)r} r+ r� r r r � test_fullGLLj s � zParsingTests.test_fullGLLc C s dddddd�}| � t|� dS )z= A partial GLL sentence is correctly parsed. r+ z3751.65�Sz14507.36r� )r' r� r� r� r� N)r} � GPGLL_PARTIALr� r r r �test_partialGLLy s �z ParsingTests.test_partialGLLc C s� i dd�dd�dd�dd�d d �d d �d d�dd�dd�dd�dd�dd�dd�dd�d d�d d �d d �d d d d!��}| � t|� d"S )#z: A full GSV sentence is correctly parsed. r' r- �GSVSentenceIndexr� �numberOfGSVSentences�3r� �11� azimuth_0�111� azimuth_1�270� azimuth_2�010� azimuth_3�292� elevation_0�03� elevation_1�15� elevation_2�01� elevation_3�06�satellitePRN_0�satellitePRN_1�04�satellitePRN_2�satellitePRN_3�13�signalToNoiseRatio_0�00)�signalToNoiseRatio_1�signalToNoiseRatio_2�signalToNoiseRatio_3N)r} r_ r� r r r � test_fullGSV� sP �������� � � � � ������zParsingTests.test_fullGSVc C st i dd�dd�dd�dd�dd �d d �d d �dd�dd�dd�dd�dd�dd�dd�dd �d d �}| � t|� d S ) z= A partial GSV sentence is correctly parsed. r' r- r� r� r� r� r� r� �067r� �311r� �244r� �42r� �14r� �05r� �22r� �24r� �27r� r� �43r� r� N)r} rh r� r r r �test_partialGSV� sD �������� � � � � ����z ParsingTests.test_partialGSVc C s ddd�}| � t|� dS )z: A full HDT sentence is correctly parsed. r. z038.005)r' r� N)r} r. r� r r r � test_fullHDT� s �zParsingTests.test_fullHDTc C s2 ddddddddd d d d d dd�}| � t|� dS )z= A typical GSA sentence is correctly parsed. r, r� r� �19�28r� �18r� r� �31�39z1.7z1.0z1.3)r' r� �fixType�usedSatellitePRN_0�usedSatellitePRN_1�usedSatellitePRN_2�usedSatellitePRN_3�usedSatellitePRN_4�usedSatellitePRN_5�usedSatellitePRN_6�usedSatellitePRN_7�positionDilutionOfPrecisionr� �verticalDilutionOfPrecisionN)r} r, r� r r r �test_typicalGSA� s �z ParsingTests.test_typicalGSAN) r r r r r} r� r� r� r� r� r� r� r� r r r r rz / s rz c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � FixUnitsTestsz� Tests for the generic unit fixing method, L{nmea.NMEAAdapter._fixUnits}. @ivar adapter: The NMEA adapter. @type adapter: L{nmea.NMEAAdapter} c C � t �t�� �| _d S r �r � NMEAAdapterr �BasePositioningReceiver�adapterr r r r r% � � zFixUnitsTests.setUpc C s@ G dd� d�}|� | j _| j jddd� | �| j jd d� dS ) zv Tests that when no C{valueKey} is provided, C{unitKey} is used, minus C{"Units"} at the end. c @ rZ )z3FixUnitsTests.test_noValueKey.<locals>.FakeSentencezJ A fake sentence that just has a "foo" attribute. c S s d| _ d S )NrP )�foor r r r r � r z<FixUnitsTests.test_noValueKey.<locals>.FakeSentence.__init__N�r r r r r r r r r � FakeSentence� ry r� �fooUnitsr )�unitKey�unitr� rP N�r� �currentSentence� _fixUnits�assertNotEqualr{ �r r� r r r �test_noValueKey� s z FixUnitsTests.test_noValueKeyc C s> G dd� d�}|� | j _| j jdd� | �| j jd d� dS )z� Tests that if a unit key is provided but the unit isn't, the unit is automatically determined from the unit key. c @ rZ )z9FixUnitsTests.test_unitKeyButNoUnit.<locals>.FakeSentencezX A fake sentence that just has "foo" and "fooUnits" attributes. c S s d| _ d| _d S )NrP r )r� r� r r r r r s zBFixUnitsTests.test_unitKeyButNoUnit.<locals>.FakeSentence.__init__Nr� r r r r r� � ry r� r� )r� r� rP Nr� r� r r r �test_unitKeyButNoUnit� s z#FixUnitsTests.test_unitKeyButNoUnitc C s | j t| jjdd� dS )z� Tests that when a unit is specified but neither C{valueKey} nor C{unitKey} is provided, C{ValueError} is raised. �K)r� N)r9 rm r� r� r r r r �test_noValueKeyAndNoUnitKey s z)FixUnitsTests.test_noValueKeyAndNoUnitKeyN)r r r r r% r� r� r� r r r r r� � s r� c @ s" e Zd ZdZdd� Zddd�ZdS )�FixerTestMixinz� Mixin for tests for the fixers on L{nmea.NMEAAdapter} that adapt from NMEA-specific notations to generic Python objects. @ivar adapter: The NMEA adapter. @type adapter: L{nmea.NMEAAdapter} c C r� r r� r r r r r% r� zFixerTestMixin.setUpNc sR t �|��� �fdd�}|du r |� � �� jj|� n� �||� � j�� dS )a� A generic adapter fixer test. Creates a sentence from the C{sentenceData} and sends that to the adapter. If C{exceptionClass} is not passed, this is assumed to work, and C{expected} is compared with the adapter's internal state. Otherwise, passing the sentence to the adapter is checked to raise C{exceptionClass}. @param sentenceData: Raw sentence content. @type sentenceData: C{dict} mapping C{str} to C{str} @param expected: The expected state of the adapter. @type expected: C{dict} or L{None} @param exceptionClass: The exception to be raised by the adapter. @type exceptionClass: subclass of C{Exception} c s � j ��� d S r )r� r r r r r �receiveSentence4 s z2FixerTestMixin._fixerTest.<locals>.receiveSentenceN)r � NMEASentencer2 r� �_stater9 r )r � sentenceDatar| rk r� r r r � _fixerTest s zFixerTestMixin._fixerTest)NN)r r r r r% r� r r r r r� s r� c @ r< )�TimestampFixerTestszL Tests conversion from NMEA timestamps to C{datetime.time} objects. c C s* ddi}dt �ddd�i}| �||� dS )z< A simple timestamp is converted correctly. r� �123456�_time� �" �8 N)�datetime�timer� )r �datar| r r r � test_simpleE s z TimestampFixerTests.test_simplec C s$ d}|D ] }| j d|itd� qdS )z: A broken timestamp raises C{ValueError}. )�993456�129956�123499r� �rk N�r� rm )r � badTimestamps�tr r r � test_brokenM s �z TimestampFixerTests.test_brokenN)r r r r r r r r r r r� @ s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �DatestampFixerTestsc C s | � | jjd� dS )z5 The default year threshold is 1980. i� N)r2 r� � yearThresholdr r r r �test_defaultYearThresholdX s z-DatestampFixerTests.test_defaultYearThresholdc C �, dt �ddd�}}| �d|id|i� dS )z� Dates before the threshold are interpreted as being in the century after the threshold. (Since the threshold is the earliest possible date.) �010115i� rP r� �_dateN�r �dater� �r � datestringr r r r �test_beforeThreshold^ s z(DatestampFixerTests.test_beforeThresholdc C r )zr Dates after the threshold are interpreted as being in the same century as the threshold. �010195i� rP r� r Nr r r r r �test_afterThresholdg s z'DatestampFixerTests.test_afterThresholdc C s | j dditd� dS )zQ A datestring with an invalid month (> 12) raises C{ValueError}. r� �011301r Nr r r r r �test_invalidMontho rM z%DatestampFixerTests.test_invalidMonthc C s( | j dditd� | j dditd� dS )zy A datestring with an invalid day (more days than there are in that month) raises C{ValueError}. r� �320101r �300201Nr r r r r �test_invalidDayu s z#DatestampFixerTests.test_invalidDayN)r r r r r r r r! r r r r r W s r c C s d| |f S )a^ Builds an NMEA float representation for a given angle in degrees and decimal minutes. @param degrees: The integer degrees for this angle. @type degrees: C{int} @param minutes: The decimal minutes value for this angle. @type minutes: C{float} @return: The NMEA float representation for this angle. @rtype: C{str} z%i%0.3fr )�degrees�minutesr r r � _nmeaFloat~ s r$ c C s | dv rdS dS )a8 Return the sign of a coordinate. This is C{1} if the coordinate is in the northern or eastern hemispheres, C{-1} otherwise. @param hemisphere: NMEA shorthand for the hemisphere. One of "NESW". @type hemisphere: C{str} @return: The sign of the coordinate value. @rtype: C{int} �NErP ���r �� hemispherer r r �_coordinateSign� s r) c C s | dv rt jS t jS )ao Return the type of a coordinate. This is L{Angles.LATITUDE} if the coordinate is in the northern or southern hemispheres, L{Angles.LONGITUDE} otherwise. @param hemisphere: NMEA shorthand for the hemisphere. One of "NESW". @type hemisphere: C{str} @return: The type of the coordinate (L{Angles.LATITUDE} or L{Angles.LONGITUDE}) �NS)r �LATITUDE� LONGITUDEr' r r r �_coordinateType� s r- c @ �@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�CoordinateFixerTestszO Tests turning NMEA coordinate notations into something more pleasant. c C �, ddd�}dt �dtj�i}| �||� dS )zg NMEA coordinate representations in the northern hemisphere convert correctly. �1030.000r �r� r� �latitude� %@N�r � Coordinater r+ r� �r r� �stater r r � test_north� � z CoordinateFixerTests.test_northc C r0 )zg NMEA coordinate representations in the southern hemisphere convert correctly. r1 r� r2 r3 � %�Nr5 r7 r r r � test_south� r: z CoordinateFixerTests.test_southc C r0 )zf NMEA coordinate representations in the eastern hemisphere convert correctly. r1 r� �r� r� � longituder4 N�r r6 r r, r� r7 r r r � test_east� r: z CoordinateFixerTests.test_eastc C r0 )zf NMEA coordinate representations in the western hemisphere convert correctly. r1 r� r= r> r; Nr? r7 r r r � test_west� r: z CoordinateFixerTests.test_westc C s ddi}| j |td� dS )z� NMEA coordinate representations for nonexistent hemispheres raise C{ValueError} when you attempt to parse them. r� �Qr Nr �r r� r r r �test_badHemisphere� r; z'CoordinateFixerTests.test_badHemispherec s � fdd�}� � t|� dS )z� NMEA coordinate repesentation parsing fails predictably when you pass nonexistent coordinate types (not latitude or longitude). c s � j �d�S )N� BOGUS_VALUE)r� �_getHemisphereSignr r r r �<lambda>� s z=CoordinateFixerTests.test_badHemisphereSign.<locals>.<lambda>N)r9 rm )r �getSignr r r �test_badHemisphereSign� s z+CoordinateFixerTests.test_badHemisphereSignN) r r r r r9 r< r@ rA rD rI r r r r r/ � s r/ c @ r< )�AltitudeFixerTestszO Tests that NMEA representations of altitudes are correctly converted. c C �. d\}}t �t|��}| �||i||i� dS )zo The NMEA representation of an altitude (above mean sea level) is correctly converted. )r� r� N�r �Altitude�floatr� �r �key�valuer� r r r �test_fixAltitude� � z#AltitudeFixerTests.test_fixAltitudec C rK )z� The NMEA representation of an altitude of the geoid (above the WGS84 reference level) is correctly converted. )r� r� NrL rO r r r � test_heightOfGeoidAboveWGS84� rS z/AltitudeFixerTests.test_heightOfGeoidAboveWGS84N)r r r r rR rT r r r r rJ � s rJ c @ rZ )�SpeedFixerTestszL Tests that NMEA representations of speeds are correctly converted. c C s6 d\}}}t �t|�t j �}| �||i||i� dS )z` Speeds reported in knots correctly get converted to meters per second. )r� �10�speedN)r �SpeedrN � MPS_PER_KNOTr� )r rP rQ � targetKeyrW r r r �test_speedInKnots s z!SpeedFixerTests.test_speedInKnotsN)r r r r r[ r r r r rU ry rU c @ r ) �VariationFixerTestsz� Tests if the absolute values of magnetic variations on the heading and their sign get combined correctly, and if that value gets combined with a heading correctly. c C s< d\}}t jjdt|� d�}||d�}| �|d|i� dS )z? Tests westward (negative) magnetic variation. )�1.34r� r& ��variationValue�r� r� �headingN�r �Heading� fromFloatsrN r� �r � variation� directionra r� r r r rA s �z VariationFixerTests.test_westc C s8 d\}}t jjt|�d�}||d�}| �|d|i� dS )z? Tests eastward (positive) magnetic variation. )r] r� r^ r` ra Nrb re r r r r@ # s �z VariationFixerTests.test_eastc C sB d\}}}|||d�}t jjt|�t|�d�}| �|d|i� dS )zH Variation values get combined with headings correctly. )z123.12r] r� )r� r� r� r^ ra Nrb )r r� rf rg r� ra r r r �test_withHeading0 s � �z$VariationFixerTests.test_withHeadingN)r r r r rA r@ rh r r r r r\ s r\ c @ r< )�PositionErrorFixerTestsa� Position errors in NMEA are passed as dilutions of precision (DOP). This is a measure relative to some specified value of the GPS device as its "reference" precision. Unfortunately, there are very few ways of figuring this out from just the device (sans manual). There are two basic DOP values: vertical and horizontal. HDOP tells you how precise your location is on the face of the earth (pretending it's flat, at least locally). VDOP tells you how precise your altitude is known. PDOP (position DOP) is a dependent value defined as the Euclidean norm of those two, and gives you a more generic "goodness of fix" value. c C s | � ddidtjdd�i� d S )Nr� r� � positionErrorg &@)�hdop)r� r � PositionErrorr r r r r N s �z#PositionErrorFixerTests.test_simplec C sF d\}}}t jt|�t|�t|�d�}|||d�}| �|d|i� d S )N)r� r� r� )�pdoprk �vdop)r� r� r� rj )r rl rN r� )r rm rk rn rj r� r r r � test_mixingT s ��z#PositionErrorFixerTests.test_mixingN)r r r r r ro r r r r ri @ s ri c @ r< )� ValidFixTestsz< Tests that data reported from a valid fix is used. c C �. ddt jjd�}dt�d�i}| �||� dS )z4 GGA data with a valid fix is used. r* r� �r' r� r� r� �33333 �@N)r �GPGGAFixQualities�GPS_FIXr rM r� �r r� � expectedStater r r �test_GGAf � �zValidFixTests.test_GGAc C rq )z: GLL data with a valid data mode is used. r+ r� �r' r� r� r� rs N)r �GPGLLGPRMCFixQualities�ACTIVEr rM r� rv r r r �test_GLLs ry zValidFixTests.test_GLLN)r r r r rx r} r r r r rp a s rp c @ r. )�InvalidFixTestsa: Tests that data being reported from a bad or incomplete fix isn't used. Although the specification dictates that GPSes shouldn't produce NMEA sentences with real-looking values for altitude or position in them unless they have at least some semblance of a GPS fix, this is widely ignored. c C s | � |i � dS )zu Sentences with an invalid fix or data mode result in empty state (ie, the data isn't used). N)r� rC r r r �_invalidFixTest� ro z InvalidFixTests._invalidFixTestc C � ddt jjd�}| �|� dS )zC GGA sentence data is unused when there is no fix. r* r� rr N)r rt � INVALID_FIXr rC r r r rx � � �zInvalidFixTests.test_GGAc C r� )zO GLL sentence data is unused when the data is flagged as void. r+ r� rz N)r r{ �VOIDr rC r r r r} � r� zInvalidFixTests.test_GLLc C �$ ddt jjt jjd�}| �|� dS )z� GSA sentence data is not used when there is no GPS fix, but the data mode claims the data is "active". Some GPSes do this, unfortunately, and that means you shouldn't use the data. r, r� �r' r� r� r� N)r r{ r| � GPGSAFixTypes� GSA_NO_FIXr rC r r r �test_badGSADataMode� � �z#InvalidFixTests.test_badGSADataModec C r� )z� GSA sentence data is not used when the fix claims to be valid (albeit only 2D), but the data mode says the data is void. Some GPSes do this, unfortunately, and that means you shouldn't use the data. r, r� r� N)r r{ r� r� � GSA_2D_FIXr rC r r r �test_badGSAFixType� r� z"InvalidFixTests.test_badGSAFixTypec C r� )zj GSA sentence data is not use when neither the fix nor the data mode is any good. r, r� r� N)r r{ r� r� r� r rC r r r � test_badGSADataModeAndFixType� s �z-InvalidFixTests.test_badGSADataModeAndFixTypeN) r r r r r rx r} r� r� r� r r r r r~ � s r~ c @ s� e Zd ZdZdd� Zdd� Zd$dd �Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� ZdS )%�NMEAReceiverTestsz& Tests for the NMEA receiver. c C s( t � | _t�| j�| _t�| j�| _d S r )r r$ r r� r� r r r r r r r% � s zNMEAReceiverTests.setUpc C st | j �t� h d�}| �t| jj�� �|� | j�� | � | j j i � | j �t � dh}| �t| jj�� �|� dS )z� If the current sentence does not contain any new fields for a particular callback, that callback is not called; even if all necessary information is still in the state from one or more previous messages. > �altitudeReceived�positionReceived�positionErrorReceived�headingReceivedN) r r1 r* r2 r! r$ �called�keysr r� r� r� r. )r �gpggaCallbacks�gphdtCallbacksr r r �1test_onlyFireWhenCurrentSentenceHasNewInformation� s zCNMEAReceiverTests.test_onlyFireWhenCurrentSentenceHasNewInformationr Nc C s\ |D ]}| j �|� q| jj�� }| �t|�t|�� |dur"|� | j�� | j�� dS )a� A generic test for NMEA receiver behavior. @param sentences: The sequence of sentences to simulate receiving. @type sentences: iterable of C{str} @param expectedFired: The names of the callbacks expected to fire. @type expectedFired: iterable of C{str} @param extraTest: An optional extra test hook. @type extraTest: nullary callable N) r r1 r$ r� r� r2 r! r r� )r r5 � expectedFired� extraTestr � actuallyFiredr r r � _receiverTest� s z NMEAReceiverTests._receiverTestc s: t gt }ddg}dd� � � �fdd�}��|||� dS )zJ The positioning error is updated across multiple states. r� �beaconInformationReceivedc S s t ttd�| ��S )N� identifier)�sorted�mapr )�beaconsr r r �_getIdentifiers s zONMEAReceiverTests.test_positionErrorUpdateAcrossStates.<locals>._getIdentifiersc sH �j jd } � | j�}g d�}��||� � | j�}��|g d�� d S )N�beaconInformation) � � � � � rO � � � � � )r� r� r� r� r� )r� r� � seenBeaconsr2 � usedBeacons)r� �seenIdentifiersr| �usedIdentifiers�r� r r r �checkBeaconInformation s zVNMEAReceiverTests.test_positionErrorUpdateAcrossStates.<locals>.checkBeaconInformationN)r, � GPGSV_SEQr� �r r5 �callbacksFiredr� r r� r �$test_positionErrorUpdateAcrossStates s z6NMEAReceiverTests.test_positionErrorUpdateAcrossStatesc s* t g}dg}� fdd�}� �|||� dS )z� A GSV sentence with empty entries in any position does not mean that entries in subsequent positions of the same GSV sentence are ignored. r� c s< � j jd } | j}� �t|�d� � �ddd� |D �� d S )Nr� � r� c S s g | ]}|j �qS r )r� )�.0�br r r � <listcomp>7 s zYNMEAReceiverTests.test_emptyMiddleGSV.<locals>.checkBeaconInformation.<locals>.<listcomp>)r� r� r� r2 �len�assertIn)r� r� r r r r� 2 s zENMEAReceiverTests.test_emptyMiddleGSV.<locals>.checkBeaconInformationN��GPGSV_EMPTY_MIDDLEr� r� r r r �test_emptyMiddleGSV* s z%NMEAReceiverTests.test_emptyMiddleGSVc C � t g}g d�}| �||� dS )z� A sequence of GGA sentences fires C{positionReceived}, C{positionErrorReceived} and C{altitudeReceived}. )r� r� r� N)r* r� �r r5 r� r r r �test_GGASentences; s z#NMEAReceiverTests.test_GGASentencesc C s4 t �ddd�| jjd< tg}g d�}| �||� dS )z� When receiving a GPGGA sentence and a date was already in the state, the new time (from the GPGGA sentence) is combined with that date. i� rP r )r� r� r� � timeReceivedN)r r r� r� r* r� r� r r r �test_GGAWithDateInStateI s z)NMEAReceiverTests.test_GGAWithDateInStatec C r� )z� A sequence of RMC sentences fires C{positionReceived}, C{speedReceived}, C{headingReceived} and C{timeReceived}. )r� � speedReceivedr� r� N)r/ r� r� r r r �test_RMCSentences[ s z#NMEAReceiverTests.test_RMCSentencesc s. t ttg}dg}� fdd�}� �|||� dS )zb A complete sequence of GSV sentences fires C{beaconInformationReceived}. r� c s � � d� jj� d S )N�_partialBeaconInformation)� assertNotInr� r� r r r r �checkPartialInformationr r� zDNMEAReceiverTests.test_GSVSentences.<locals>.checkPartialInformationN)r_ rf rh r� )r r5 r� r� r r r �test_GSVSentencesj s z#NMEAReceiverTests.test_GSVSentencesc C � t g}| �|dg� dS )z� A complete sequence of GSV sentences with empty entries in the middle still fires C{beaconInformationReceived}. r� Nr� �r r5 r r r �"test_emptyMiddleEntriesGSVSequencew � z4NMEAReceiverTests.test_emptyMiddleEntriesGSVSequencec C s t g}| �|� dS )zV An incomplete sequence of GSV sentences does not fire any callbacks. N)r_ r� r� r r r �test_incompleteGSVSequence s z,NMEAReceiverTests.test_incompleteGSVSequencec C r� )z� The parser does not fail badly when the sequence consists of only one sentence (but is otherwise complete). r� N)� GPGSV_SINGLEr� r� r r r � test_singleSentenceGSVSequence� r� z0NMEAReceiverTests.test_singleSentenceGSVSequencec C s t tg}| �|dg� dS )z9 GLL sentences fire C{positionReceived}. r� N)r� r+ r� r� r r r �test_GLLSentences� r\ z#NMEAReceiverTests.test_GLLSentencesc C r� )z8 HDT sentences fire C{headingReceived}. r� N)r. r� r� r r r �test_HDTSentences� s z#NMEAReceiverTests.test_HDTSentencesc s. t tg}g d�}� fdd�}� �|||� dS )zA A mix of sentences fires the correct callbacks. )r� r� r� r� r� r� c s, t � dddddd�} � �� jjd | � d S )Ni� r� � r �# r� r )r r2 r� r� )�expectedDateTimer r r � checkTime� s z8NMEAReceiverTests.test_mixedSentences.<locals>.checkTimeN)r/ r* r� )r r5 r� r� r r r �test_mixedSentences� s z%NMEAReceiverTests.test_mixedSentencesc C s, t gt tttg }g d�}| �||� dS )a# Sends an entire gamut of sentences and verifies the appropriate callbacks fire. These are more than you'd expect from your average consumer GPS device. They have most of the important information, including beacon information and visibility. )r� r� r� r� r� r� r� N)r, r� r/ r* r+ r� r� r r r �test_lotsOfMixedSentences� s z+NMEAReceiverTests.test_lotsOfMixedSentences)r N)r r r r r% r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s$ r� )7r r �operatorr �zope.interfacer �twisted.positioningr r r �twisted.positioning.baser �!twisted.positioning.test.receiverr �twisted.trial.unittestr r* r/ r, r. r+ r� r� r� rQ r� r_ rf rh � INMEAReceiverr r r7 r= rG r[ r] rj rs rz r� r� r� r r$ r) r- r/ rJ rU r\ ri rp r~ r� r r r r �<module> sb �* % $ -8-';1! TPK ���Z��ç�p �p test_base.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for positioning primitives. """ from zope.interface import verify from twisted.positioning import base from twisted.positioning.base import Angles, Directions from twisted.positioning.ipositioning import IPositioningBeacon from twisted.trial.unittest import TestCase class AngleTests(TestCase): """ Tests for the L{twisted.positioning.base.Angle} class. """ def test_empty(self): """ The repr of an empty angle says that is of unknown type and unknown value. """ a = base.Angle() self.assertEqual("<Angle of unknown type (unknown value)>", repr(a)) def test_variation(self): """ The repr of an empty variation says that it is a variation of unknown value. """ a = base.Angle(angleType=Angles.VARIATION) self.assertEqual("<Variation (unknown value)>", repr(a)) def test_unknownType(self): """ The repr of an angle of unknown type but a given value displays that type and value in its repr. """ a = base.Angle(1.0) self.assertEqual("<Angle of unknown type (1.0 degrees)>", repr(a)) def test_bogusType(self): """ Trying to create an angle with a bogus type raises C{ValueError}. """ self.assertRaises(ValueError, base.Angle, angleType="BOGUS") class HeadingTests(TestCase): """ Tests for the L{twisted.positioning.base.Heading} class. """ def test_simple(self): """ Tests that a simple heading has a value in decimal degrees, which is also its value when converted to a float. Its variation, and by consequence its corrected heading, is L{None}. """ h = base.Heading(1.0) self.assertEqual(h.inDecimalDegrees, 1.0) self.assertEqual(float(h), 1.0) self.assertIsNone(h.variation) self.assertIsNone(h.correctedHeading) def test_headingWithoutVariationRepr(self): """ A repr of a heading with no variation reports its value and that the variation is unknown. """ heading = base.Heading(1.0) expectedRepr = "<Heading (1.0 degrees, unknown variation)>" self.assertEqual(repr(heading), expectedRepr) def test_headingWithVariationRepr(self): """ A repr of a heading with known variation reports its value and the value of that variation. """ angle, variation = 1.0, -10.0 heading = base.Heading.fromFloats(angle, variationValue=variation) reprTemplate = "<Heading ({0} degrees, <Variation ({1} degrees)>)>" self.assertEqual(repr(heading), reprTemplate.format(angle, variation)) def test_valueEquality(self): """ Headings with the same values compare equal. """ self.assertEqual(base.Heading(1.0), base.Heading(1.0)) def test_valueInequality(self): """ Headings with different values compare unequal. """ self.assertNotEqual(base.Heading(1.0), base.Heading(2.0)) def test_zeroHeadingEdgeCase(self): """ Headings can be instantiated with a value of 0 and no variation. """ base.Heading(0) def test_zeroHeading180DegreeVariationEdgeCase(self): """ Headings can be instantiated with a value of 0 and a variation of 180 degrees. """ base.Heading(0, 180) def _badValueTest(self, **kw): """ Helper function for verifying that bad values raise C{ValueError}. @param kw: The keyword arguments passed to L{base.Heading.fromFloats}. """ self.assertRaises(ValueError, base.Heading.fromFloats, **kw) def test_badAngleValueEdgeCase(self): """ Headings can not be instantiated with a value of 360 degrees. """ self._badValueTest(angleValue=360.0) def test_badVariationEdgeCase(self): """ Headings can not be instantiated with a variation of -180 degrees. """ self._badValueTest(variationValue=-180.0) def test_negativeHeading(self): """ Negative heading values raise C{ValueError}. """ self._badValueTest(angleValue=-10.0) def test_headingTooLarge(self): """ Heading values greater than C{360.0} raise C{ValueError}. """ self._badValueTest(angleValue=370.0) def test_variationTooNegative(self): """ Variation values less than C{-180.0} raise C{ValueError}. """ self._badValueTest(variationValue=-190.0) def test_variationTooPositive(self): """ Variation values greater than C{180.0} raise C{ValueError}. """ self._badValueTest(variationValue=190.0) def test_correctedHeading(self): """ A heading with a value and a variation has a corrected heading. """ h = base.Heading.fromFloats(1.0, variationValue=-10.0) self.assertEqual(h.correctedHeading, base.Angle(11.0, Angles.HEADING)) def test_correctedHeadingOverflow(self): """ A heading with a value and a variation has the appropriate corrected heading value, even when the variation puts it across the 360 degree boundary. """ h = base.Heading.fromFloats(359.0, variationValue=-2.0) self.assertEqual(h.correctedHeading, base.Angle(1.0, Angles.HEADING)) def test_correctedHeadingOverflowEdgeCase(self): """ A heading with a value and a variation has the appropriate corrected heading value, even when the variation puts it exactly at the 360 degree boundary. """ h = base.Heading.fromFloats(359.0, variationValue=-1.0) self.assertEqual(h.correctedHeading, base.Angle(0.0, Angles.HEADING)) def test_correctedHeadingUnderflow(self): """ A heading with a value and a variation has the appropriate corrected heading value, even when the variation puts it under the 0 degree boundary. """ h = base.Heading.fromFloats(1.0, variationValue=2.0) self.assertEqual(h.correctedHeading, base.Angle(359.0, Angles.HEADING)) def test_correctedHeadingUnderflowEdgeCase(self): """ A heading with a value and a variation has the appropriate corrected heading value, even when the variation puts it exactly at the 0 degree boundary. """ h = base.Heading.fromFloats(1.0, variationValue=1.0) self.assertEqual(h.correctedHeading, base.Angle(0.0, Angles.HEADING)) def test_setVariationSign(self): """ Setting the sign of a heading changes the variation sign. """ h = base.Heading.fromFloats(1.0, variationValue=1.0) h.setSign(1) self.assertEqual(h.variation.inDecimalDegrees, 1.0) h.setSign(-1) self.assertEqual(h.variation.inDecimalDegrees, -1.0) def test_setBadVariationSign(self): """ Setting the sign of a heading to values that aren't C{-1} or C{1} raises C{ValueError} and does not affect the heading. """ h = base.Heading.fromFloats(1.0, variationValue=1.0) self.assertRaises(ValueError, h.setSign, -50) self.assertEqual(h.variation.inDecimalDegrees, 1.0) self.assertRaises(ValueError, h.setSign, 0) self.assertEqual(h.variation.inDecimalDegrees, 1.0) self.assertRaises(ValueError, h.setSign, 50) self.assertEqual(h.variation.inDecimalDegrees, 1.0) def test_setUnknownVariationSign(self): """ Setting the sign on a heading with unknown variation raises C{ValueError}. """ h = base.Heading.fromFloats(1.0) self.assertIsNone(h.variation.inDecimalDegrees) self.assertRaises(ValueError, h.setSign, 1) class CoordinateTests(TestCase): def test_float(self): """ Coordinates can be converted to floats. """ coordinate = base.Coordinate(10.0) self.assertEqual(float(coordinate), 10.0) def test_repr(self): """ Coordinates that aren't explicitly latitudes or longitudes have an appropriate repr. """ coordinate = base.Coordinate(10.0) expectedRepr = f"<Angle of unknown type ({10.0} degrees)>" self.assertEqual(repr(coordinate), expectedRepr) def test_positiveLatitude(self): """ Positive latitudes have a repr that specifies their type and value. """ coordinate = base.Coordinate(10.0, Angles.LATITUDE) expectedRepr = f"<Latitude ({10.0} degrees)>" self.assertEqual(repr(coordinate), expectedRepr) def test_negativeLatitude(self): """ Negative latitudes have a repr that specifies their type and value. """ coordinate = base.Coordinate(-50.0, Angles.LATITUDE) expectedRepr = f"<Latitude ({-50.0} degrees)>" self.assertEqual(repr(coordinate), expectedRepr) def test_positiveLongitude(self): """ Positive longitudes have a repr that specifies their type and value. """ longitude = base.Coordinate(50.0, Angles.LONGITUDE) expectedRepr = f"<Longitude ({50.0} degrees)>" self.assertEqual(repr(longitude), expectedRepr) def test_negativeLongitude(self): """ Negative longitudes have a repr that specifies their type and value. """ longitude = base.Coordinate(-50.0, Angles.LONGITUDE) expectedRepr = f"<Longitude ({-50.0} degrees)>" self.assertEqual(repr(longitude), expectedRepr) def test_bogusCoordinateType(self): """ Creating coordinates with bogus types rasies C{ValueError}. """ self.assertRaises(ValueError, base.Coordinate, 150.0, "BOGUS") def test_angleTypeNotCoordinate(self): """ Creating coordinates with angle types that aren't coordinates raises C{ValueError}. """ self.assertRaises(ValueError, base.Coordinate, 150.0, Angles.HEADING) def test_equality(self): """ Coordinates with the same value and type are equal. """ def makeCoordinate(): return base.Coordinate(1.0, Angles.LONGITUDE) self.assertEqual(makeCoordinate(), makeCoordinate()) def test_differentAnglesInequality(self): """ Coordinates with different values aren't equal. """ c1 = base.Coordinate(1.0) c2 = base.Coordinate(-1.0) self.assertNotEqual(c1, c2) def test_differentTypesInequality(self): """ Coordinates with the same values but different types aren't equal. """ c1 = base.Coordinate(1.0, Angles.LATITUDE) c2 = base.Coordinate(1.0, Angles.LONGITUDE) self.assertNotEqual(c1, c2) def test_sign(self): """ Setting the sign on a coordinate sets the sign of the value of the coordinate. """ c = base.Coordinate(50.0, Angles.LATITUDE) c.setSign(1) self.assertEqual(c.inDecimalDegrees, 50.0) c.setSign(-1) self.assertEqual(c.inDecimalDegrees, -50.0) def test_badVariationSign(self): """ Setting a bogus sign value (not -1 or 1) on a coordinate raises C{ValueError} and doesn't affect the coordinate. """ value = 50.0 c = base.Coordinate(value, Angles.LATITUDE) self.assertRaises(ValueError, c.setSign, -50) self.assertEqual(c.inDecimalDegrees, 50.0) self.assertRaises(ValueError, c.setSign, 0) self.assertEqual(c.inDecimalDegrees, 50.0) self.assertRaises(ValueError, c.setSign, 50) self.assertEqual(c.inDecimalDegrees, 50.0) def test_northernHemisphere(self): """ Positive latitudes are in the northern hemisphere. """ coordinate = base.Coordinate(1.0, Angles.LATITUDE) self.assertEqual(coordinate.hemisphere, Directions.NORTH) def test_easternHemisphere(self): """ Positive longitudes are in the eastern hemisphere. """ coordinate = base.Coordinate(1.0, Angles.LONGITUDE) self.assertEqual(coordinate.hemisphere, Directions.EAST) def test_southernHemisphere(self): """ Negative latitudes are in the southern hemisphere. """ coordinate = base.Coordinate(-1.0, Angles.LATITUDE) self.assertEqual(coordinate.hemisphere, Directions.SOUTH) def test_westernHemisphere(self): """ Negative longitudes are in the western hemisphere. """ coordinate = base.Coordinate(-1.0, Angles.LONGITUDE) self.assertEqual(coordinate.hemisphere, Directions.WEST) def test_badHemisphere(self): """ Accessing the hemisphere for a coordinate that can't compute it raises C{ValueError}. """ coordinate = base.Coordinate(1.0, None) self.assertRaises(ValueError, lambda: coordinate.hemisphere) def test_latitudeTooLarge(self): """ Creating a latitude with a value greater than or equal to 90 degrees raises C{ValueError}. """ self.assertRaises(ValueError, _makeLatitude, 150.0) self.assertRaises(ValueError, _makeLatitude, 90.0) def test_latitudeTooSmall(self): """ Creating a latitude with a value less than or equal to -90 degrees raises C{ValueError}. """ self.assertRaises(ValueError, _makeLatitude, -150.0) self.assertRaises(ValueError, _makeLatitude, -90.0) def test_longitudeTooLarge(self): """ Creating a longitude with a value greater than or equal to 180 degrees raises C{ValueError}. """ self.assertRaises(ValueError, _makeLongitude, 250.0) self.assertRaises(ValueError, _makeLongitude, 180.0) def test_longitudeTooSmall(self): """ Creating a longitude with a value less than or equal to -180 degrees raises C{ValueError}. """ self.assertRaises(ValueError, _makeLongitude, -250.0) self.assertRaises(ValueError, _makeLongitude, -180.0) def test_inDegreesMinutesSeconds(self): """ Coordinate values can be accessed in degrees, minutes, seconds. """ c = base.Coordinate(50.5, Angles.LATITUDE) self.assertEqual(c.inDegreesMinutesSeconds, (50, 30, 0)) c = base.Coordinate(50.213, Angles.LATITUDE) self.assertEqual(c.inDegreesMinutesSeconds, (50, 12, 46)) def test_unknownAngleInDegreesMinutesSeconds(self): """ If the vaue of a coordinate is L{None}, its values in degrees, minutes, seconds is also L{None}. """ c = base.Coordinate(None, None) self.assertIsNone(c.inDegreesMinutesSeconds) def _makeLatitude(value): """ Builds and returns a latitude of given value. """ return base.Coordinate(value, Angles.LATITUDE) def _makeLongitude(value): """ Builds and returns a longitude of given value. """ return base.Coordinate(value, Angles.LONGITUDE) class AltitudeTests(TestCase): """ Tests for the L{twisted.positioning.base.Altitude} class. """ def test_value(self): """ Altitudes can be instantiated and reports the correct value in meters and feet, as well as when converted to float. """ altitude = base.Altitude(1.0) self.assertEqual(float(altitude), 1.0) self.assertEqual(altitude.inMeters, 1.0) self.assertEqual(altitude.inFeet, 1.0 / base.METERS_PER_FOOT) def test_repr(self): """ Altitudes report their type and value in their repr. """ altitude = base.Altitude(1.0) self.assertEqual(repr(altitude), "<Altitude (1.0 m)>") def test_equality(self): """ Altitudes with equal values compare equal. """ firstAltitude = base.Altitude(1.0) secondAltitude = base.Altitude(1.0) self.assertEqual(firstAltitude, secondAltitude) def test_inequality(self): """ Altitudes with different values don't compare equal. """ firstAltitude = base.Altitude(1.0) secondAltitude = base.Altitude(-1.0) self.assertNotEqual(firstAltitude, secondAltitude) class SpeedTests(TestCase): """ Tests for the L{twisted.positioning.base.Speed} class. """ def test_value(self): """ Speeds can be instantiated, and report their value in meters per second, and can be converted to floats. """ speed = base.Speed(50.0) self.assertEqual(speed.inMetersPerSecond, 50.0) self.assertEqual(float(speed), 50.0) def test_repr(self): """ Speeds report their type and value in their repr. """ speed = base.Speed(50.0) self.assertEqual(repr(speed), "<Speed (50.0 m/s)>") def test_negativeSpeeds(self): """ Creating a negative speed raises C{ValueError}. """ self.assertRaises(ValueError, base.Speed, -1.0) def test_inKnots(self): """ A speed can be converted into its value in knots. """ speed = base.Speed(1.0) self.assertEqual(1 / base.MPS_PER_KNOT, speed.inKnots) def test_asFloat(self): """ A speed can be converted into a C{float}. """ self.assertEqual(1.0, float(base.Speed(1.0))) class ClimbTests(TestCase): """ Tests for L{twisted.positioning.base.Climb}. """ def test_simple(self): """ Speeds can be instantiated, and report their value in meters per second, and can be converted to floats. """ climb = base.Climb(42.0) self.assertEqual(climb.inMetersPerSecond, 42.0) self.assertEqual(float(climb), 42.0) def test_repr(self): """ Climbs report their type and value in their repr. """ climb = base.Climb(42.0) self.assertEqual(repr(climb), "<Climb (42.0 m/s)>") def test_negativeClimbs(self): """ Climbs can have negative values, and still report that value in meters per second and when converted to floats. """ climb = base.Climb(-42.0) self.assertEqual(climb.inMetersPerSecond, -42.0) self.assertEqual(float(climb), -42.0) def test_speedInKnots(self): """ A climb can be converted into its value in knots. """ climb = base.Climb(1.0) self.assertEqual(1 / base.MPS_PER_KNOT, climb.inKnots) def test_asFloat(self): """ A climb can be converted into a C{float}. """ self.assertEqual(1.0, float(base.Climb(1.0))) class PositionErrorTests(TestCase): """ Tests for L{twisted.positioning.base.PositionError}. """ def test_allUnset(self): """ In an empty L{base.PositionError} with no invariant testing, all dilutions of positions are L{None}. """ positionError = base.PositionError() self.assertIsNone(positionError.pdop) self.assertIsNone(positionError.hdop) self.assertIsNone(positionError.vdop) def test_allUnsetWithInvariant(self): """ In an empty L{base.PositionError} with invariant testing, all dilutions of positions are L{None}. """ positionError = base.PositionError(testInvariant=True) self.assertIsNone(positionError.pdop) self.assertIsNone(positionError.hdop) self.assertIsNone(positionError.vdop) def test_withoutInvariant(self): """ L{base.PositionError}s can be instantiated with just a HDOP. """ positionError = base.PositionError(hdop=1.0) self.assertEqual(positionError.hdop, 1.0) def test_withInvariant(self): """ Creating a simple L{base.PositionError} with just a HDOP while checking the invariant works. """ positionError = base.PositionError(hdop=1.0, testInvariant=True) self.assertEqual(positionError.hdop, 1.0) def test_invalidWithoutInvariant(self): """ Creating a L{base.PositionError} with values set to an impossible combination works if the invariant is not checked. """ error = base.PositionError(pdop=1.0, vdop=1.0, hdop=1.0) self.assertEqual(error.pdop, 1.0) self.assertEqual(error.hdop, 1.0) self.assertEqual(error.vdop, 1.0) def test_invalidWithInvariant(self): """ Creating a L{base.PositionError} with values set to an impossible combination raises C{ValueError} if the invariant is being tested. """ self.assertRaises( ValueError, base.PositionError, pdop=1.0, vdop=1.0, hdop=1.0, testInvariant=True, ) def test_setDOPWithoutInvariant(self): """ You can set the PDOP value to value inconsisted with HDOP and VDOP when not checking the invariant. """ pe = base.PositionError(hdop=1.0, vdop=1.0) pe.pdop = 100.0 self.assertEqual(pe.pdop, 100.0) def test_setDOPWithInvariant(self): """ Attempting to set the PDOP value to value inconsisted with HDOP and VDOP when checking the invariant raises C{ValueError}. """ pe = base.PositionError(hdop=1.0, vdop=1.0, testInvariant=True) pdop = pe.pdop def setPDOP(pe): pe.pdop = 100.0 self.assertRaises(ValueError, setPDOP, pe) self.assertEqual(pe.pdop, pdop) REPR_TEMPLATE = "<PositionError (pdop: %s, hdop: %s, vdop: %s)>" def _testDOP(self, pe, pdop, hdop, vdop): """ Tests the DOP values in a position error, and the repr of that position error. @param pe: The position error under test. @type pe: C{PositionError} @param pdop: The expected position dilution of precision. @type pdop: C{float} or L{None} @param hdop: The expected horizontal dilution of precision. @type hdop: C{float} or L{None} @param vdop: The expected vertical dilution of precision. @type vdop: C{float} or L{None} """ self.assertEqual(pe.pdop, pdop) self.assertEqual(pe.hdop, hdop) self.assertEqual(pe.vdop, vdop) self.assertEqual(repr(pe), self.REPR_TEMPLATE % (pdop, hdop, vdop)) def test_positionAndHorizontalSet(self): """ The VDOP is correctly determined from PDOP and HDOP. """ pdop, hdop = 2.0, 1.0 vdop = (pdop ** 2 - hdop ** 2) ** 0.5 pe = base.PositionError(pdop=pdop, hdop=hdop) self._testDOP(pe, pdop, hdop, vdop) def test_positionAndVerticalSet(self): """ The HDOP is correctly determined from PDOP and VDOP. """ pdop, vdop = 2.0, 1.0 hdop = (pdop ** 2 - vdop ** 2) ** 0.5 pe = base.PositionError(pdop=pdop, vdop=vdop) self._testDOP(pe, pdop, hdop, vdop) def test_horizontalAndVerticalSet(self): """ The PDOP is correctly determined from HDOP and VDOP. """ hdop, vdop = 1.0, 1.0 pdop = (hdop ** 2 + vdop ** 2) ** 0.5 pe = base.PositionError(hdop=hdop, vdop=vdop) self._testDOP(pe, pdop, hdop, vdop) class BeaconInformationTests(TestCase): """ Tests for L{twisted.positioning.base.BeaconInformation}. """ def test_minimal(self): """ For an empty beacon information object, the number of used beacons is zero, the number of seen beacons is zero, and the repr of the object reflects that. """ bi = base.BeaconInformation() self.assertEqual(len(bi.usedBeacons), 0) expectedRepr = ( "<BeaconInformation (" "used beacons (0): [], " "unused beacons: [])>" ) self.assertEqual(repr(bi), expectedRepr) satelliteKwargs = {"azimuth": 1, "elevation": 1, "signalToNoiseRatio": 1.0} def test_simple(self): """ Tests a beacon information with a bunch of satellites, none of which used in computing a fix. """ def _buildSatellite(**kw): kwargs = dict(self.satelliteKwargs) kwargs.update(kw) return base.Satellite(**kwargs) beacons = set() for prn in range(1, 10): beacons.add(_buildSatellite(identifier=prn)) bi = base.BeaconInformation(beacons) self.assertEqual(len(bi.seenBeacons), 9) self.assertEqual(len(bi.usedBeacons), 0) self.assertEqual( repr(bi), "<BeaconInformation (used beacons (0): [], " "unused beacons: [" "<Satellite (1), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (2), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (3), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (4), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (5), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (6), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (7), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (8), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (9), azimuth: 1, elevation: 1, snr: 1.0>" "])>", ) def test_someSatellitesUsed(self): """ Tests a beacon information with a bunch of satellites, some of them used in computing a fix. """ bi = base.BeaconInformation() for prn in range(1, 10): satellite = base.Satellite(identifier=prn, **self.satelliteKwargs) bi.seenBeacons.add(satellite) if prn % 2: bi.usedBeacons.add(satellite) self.assertEqual(len(bi.seenBeacons), 9) self.assertEqual(len(bi.usedBeacons), 5) self.assertEqual( repr(bi), "<BeaconInformation (used beacons (5): [" "<Satellite (1), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (3), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (5), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (7), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (9), azimuth: 1, elevation: 1, snr: 1.0>], " "unused beacons: [" "<Satellite (2), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (4), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (6), azimuth: 1, elevation: 1, snr: 1.0>, " "<Satellite (8), azimuth: 1, elevation: 1, snr: 1.0>])>", ) class PositioningBeaconTests(TestCase): """ Tests for L{base.PositioningBeacon}. """ def test_interface(self): """ Tests that L{base.PositioningBeacon} implements L{IPositioningBeacon}. """ implements = IPositioningBeacon.implementedBy(base.PositioningBeacon) self.assertTrue(implements) verify.verifyObject(IPositioningBeacon, base.PositioningBeacon(1)) def test_repr(self): """ Tests the repr of a positioning beacon. """ self.assertEqual(repr(base.PositioningBeacon("A")), "<Beacon (A)>") class SatelliteTests(TestCase): """ Tests for L{twisted.positioning.base.Satellite}. """ def test_minimal(self): """ Tests a minimal satellite that only has a known PRN. Tests that the azimuth, elevation and signal to noise ratios are L{None} and verifies the repr. """ s = base.Satellite(1) self.assertEqual(s.identifier, 1) self.assertIsNone(s.azimuth) self.assertIsNone(s.elevation) self.assertIsNone(s.signalToNoiseRatio) self.assertEqual( repr(s), "<Satellite (1), azimuth: None, " "elevation: None, snr: None>" ) def test_simple(self): """ Tests a minimal satellite that only has a known PRN. Tests that the azimuth, elevation and signal to noise ratios are correct and verifies the repr. """ s = base.Satellite( identifier=1, azimuth=270.0, elevation=30.0, signalToNoiseRatio=25.0 ) self.assertEqual(s.identifier, 1) self.assertEqual(s.azimuth, 270.0) self.assertEqual(s.elevation, 30.0) self.assertEqual(s.signalToNoiseRatio, 25.0) self.assertEqual( repr(s), "<Satellite (1), azimuth: 270.0, " "elevation: 30.0, snr: 25.0>" ) PK ���Z�}_�`� `� test_nmea.pynu �[��� # Copyright (c) 2009-2011 Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for using NMEA sentences. """ import datetime from operator import attrgetter from zope.interface import implementer from twisted.positioning import base, ipositioning, nmea from twisted.positioning.base import Angles from twisted.positioning.test.receiver import MockPositioningReceiver from twisted.trial.unittest import TestCase # Sample sentences GPGGA = b"$GPGGA,123519,4807.038,N,01131.000,E,1,08,0.9,545.4,M,46.9,M,,*47" GPRMC = b"$GPRMC,123519,A,4807.038,N,01131.000,E,022.4,084.4,230394,003.1,W*6A" GPGSA = b"$GPGSA,A,3,19,28,14,18,27,22,31,39,,,,,1.7,1.0,1.3*34" GPHDT = b"$GPHDT,038.005,T*3B" GPGLL = b"$GPGLL,4916.45,N,12311.12,W,225444,A*31" GPGLL_PARTIAL = b"$GPGLL,3751.65,S,14507.36,E*77" GPGSV_SINGLE = b"$GPGSV,1,1,11,03,03,111,00,04,15,270,00,06,01,010,00,,,,*4b" GPGSV_EMPTY_MIDDLE = b"$GPGSV,1,1,11,03,03,111,00,,,,,,,,,13,06,292,00*75" GPGSV_SEQ = ( GPGSV_FIRST, GPGSV_MIDDLE, GPGSV_LAST, ) = b""" $GPGSV,3,1,11,03,03,111,00,04,15,270,00,06,01,010,00,13,06,292,00*74 $GPGSV,3,2,11,14,25,170,00,16,57,208,39,18,67,296,40,19,40,246,00*74 $GPGSV,3,3,11,22,42,067,42,24,14,311,43,27,05,244,00,,,,*4D """.split() @implementer(ipositioning.INMEAReceiver) class NMEATestReceiver: """ An NMEA receiver for testing. Remembers the last sentence it has received. """ def __init__(self): self.clear() def clear(self): """ Forgets the received sentence (if any), by setting C{self.receivedSentence} to L{None}. """ self.receivedSentence = None def sentenceReceived(self, sentence): self.receivedSentence = sentence class CallbackTests(TestCase): """ Tests if the NMEA protocol correctly calls its sentence callback. @ivar protocol: The NMEA protocol under test. @type protocol: L{nmea.NMEAProtocol} @ivar sentenceTypes: The set of sentence types of all sentences the test's sentence callback function has been called with. @type sentenceTypes: C{set} """ def setUp(self): receiver = NMEATestReceiver() self.protocol = nmea.NMEAProtocol(receiver, self._sentenceCallback) self.sentenceTypes = set() def _sentenceCallback(self, sentence): """ Remembers that a sentence of this type was fired. """ self.sentenceTypes.add(sentence.type) def test_callbacksCalled(self): """ The correct callbacks fire, and that *only* those fire. """ sentencesByType = { "GPGGA": [b"$GPGGA*56"], "GPGLL": [b"$GPGLL*50"], "GPGSA": [b"$GPGSA*42"], "GPGSV": [b"$GPGSV*55"], "GPHDT": [b"$GPHDT*4f"], "GPRMC": [b"$GPRMC*4b"], } for sentenceType, sentences in sentencesByType.items(): for sentence in sentences: self.protocol.lineReceived(sentence) self.assertEqual(self.sentenceTypes, {sentenceType}) self.sentenceTypes.clear() class BrokenSentenceCallbackTests(TestCase): """ Tests for broken NMEA sentence callbacks. """ def setUp(self): receiver = NMEATestReceiver() self.protocol = nmea.NMEAProtocol(receiver, self._sentenceCallback) def _sentenceCallback(self, sentence): """ Raises C{AttributeError}. """ raise AttributeError("ERROR!!!") def test_dontSwallowCallbackExceptions(self): """ An C{AttributeError} in the sentence callback of an C{NMEAProtocol} doesn't get swallowed. """ lineReceived = self.protocol.lineReceived self.assertRaises(AttributeError, lineReceived, b"$GPGGA*56") class SplitTests(TestCase): """ Checks splitting of NMEA sentences. """ def test_withChecksum(self): """ An NMEA sentence with a checksum gets split correctly. """ splitSentence = nmea._split(b"$GPGGA,spam,eggs*00") self.assertEqual(splitSentence, [b"GPGGA", b"spam", b"eggs"]) def test_noCheckum(self): """ An NMEA sentence without a checksum gets split correctly. """ splitSentence = nmea._split(b"$GPGGA,spam,eggs*") self.assertEqual(splitSentence, [b"GPGGA", b"spam", b"eggs"]) class ChecksumTests(TestCase): """ NMEA sentence checksum verification tests. """ def test_valid(self): """ Sentences with valid checksums get validated. """ nmea._validateChecksum(GPGGA) def test_missing(self): """ Sentences with missing checksums get validated. """ nmea._validateChecksum(GPGGA[:-2]) def test_invalid(self): """ Sentences with a bad checksum raise L{base.InvalidChecksum} when attempting to validate them. """ validate = nmea._validateChecksum bareSentence, checksum = GPGGA.split(b"*") badChecksum = b"%d" % (int(checksum, 16) + 1,) sentences = [bareSentence + b"*" + badChecksum] for s in sentences: self.assertRaises(base.InvalidChecksum, validate, s) class NMEAReceiverSetup: """ A mixin for tests that need an NMEA receiver (and a protocol attached to it). @ivar receiver: An NMEA receiver that remembers the last sentence. @type receiver: L{NMEATestReceiver} @ivar protocol: An NMEA protocol attached to the receiver. @type protocol: L{twisted.positioning.nmea.NMEAProtocol} """ def setUp(self): """ Sets up an NMEA receiver. """ self.receiver = NMEATestReceiver() self.protocol = nmea.NMEAProtocol(self.receiver) class GSVSequenceTests(NMEAReceiverSetup, TestCase): """ Tests for the interpretation of GSV sequences. """ def test_firstSentence(self): """ The first sentence in a GSV sequence is correctly identified. """ self.protocol.lineReceived(GPGSV_FIRST) sentence = self.receiver.receivedSentence self.assertTrue(sentence._isFirstGSVSentence()) self.assertFalse(sentence._isLastGSVSentence()) def test_middleSentence(self): """ A sentence in the middle of a GSV sequence is correctly identified (as being neither the last nor the first). """ self.protocol.lineReceived(GPGSV_MIDDLE) sentence = self.receiver.receivedSentence self.assertFalse(sentence._isFirstGSVSentence()) self.assertFalse(sentence._isLastGSVSentence()) def test_lastSentence(self): """ The last sentence in a GSV sequence is correctly identified. """ self.protocol.lineReceived(GPGSV_LAST) sentence = self.receiver.receivedSentence self.assertFalse(sentence._isFirstGSVSentence()) self.assertTrue(sentence._isLastGSVSentence()) class BogusSentenceTests(NMEAReceiverSetup, TestCase): """ Tests for verifying predictable failure for bogus NMEA sentences. """ def assertRaisesOnSentence(self, exceptionClass, sentence): """ Asserts that the protocol raises C{exceptionClass} when it receives C{sentence}. @param exceptionClass: The exception class expected to be raised. @type exceptionClass: C{Exception} subclass @param sentence: The (bogus) NMEA sentence. @type sentence: C{str} """ self.assertRaises(exceptionClass, self.protocol.lineReceived, sentence) def test_raiseOnUnknownSentenceType(self): """ Receiving a well-formed sentence of unknown type raises C{ValueError}. """ self.assertRaisesOnSentence(ValueError, b"$GPBOGUS*5b") def test_raiseOnMalformedSentences(self): """ Receiving a malformed sentence raises L{base.InvalidSentence}. """ self.assertRaisesOnSentence(base.InvalidSentence, "GPBOGUS") class NMEASentenceTests(NMEAReceiverSetup, TestCase): """ Tests for L{nmea.NMEASentence} objects. """ def test_repr(self): """ The C{repr} of L{nmea.NMEASentence} objects is correct. """ sentencesWithExpectedRepr = [ ( GPGSA, "<NMEASentence (GPGSA) {" "dataMode: A, " "fixType: 3, " "horizontalDilutionOfPrecision: 1.0, " "positionDilutionOfPrecision: 1.7, " "usedSatellitePRN_0: 19, " "usedSatellitePRN_1: 28, " "usedSatellitePRN_2: 14, " "usedSatellitePRN_3: 18, " "usedSatellitePRN_4: 27, " "usedSatellitePRN_5: 22, " "usedSatellitePRN_6: 31, " "usedSatellitePRN_7: 39, " "verticalDilutionOfPrecision: 1.3" "}>", ), ] for sentence, expectedRepr in sentencesWithExpectedRepr: self.protocol.lineReceived(sentence) received = self.receiver.receivedSentence self.assertEqual(repr(received), expectedRepr) class ParsingTests(NMEAReceiverSetup, TestCase): """ Tests if raw NMEA sentences get parsed correctly. This doesn't really involve any interpretation, just turning ugly raw NMEA representations into objects that are more pleasant to work with. """ def _parserTest(self, sentence, expected): """ Passes a sentence to the protocol and gets the parsed sentence from the receiver. Then verifies that the parsed sentence contains the expected data. """ self.protocol.lineReceived(sentence) received = self.receiver.receivedSentence self.assertEqual(expected, received._sentenceData) def test_fullRMC(self): """ A full RMC sentence is correctly parsed. """ expected = { "type": "GPRMC", "latitudeFloat": "4807.038", "latitudeHemisphere": "N", "longitudeFloat": "01131.000", "longitudeHemisphere": "E", "magneticVariation": "003.1", "magneticVariationDirection": "W", "speedInKnots": "022.4", "timestamp": "123519", "datestamp": "230394", "trueHeading": "084.4", "dataMode": "A", } self._parserTest(GPRMC, expected) def test_fullGGA(self): """ A full GGA sentence is correctly parsed. """ expected = { "type": "GPGGA", "altitude": "545.4", "altitudeUnits": "M", "heightOfGeoidAboveWGS84": "46.9", "heightOfGeoidAboveWGS84Units": "M", "horizontalDilutionOfPrecision": "0.9", "latitudeFloat": "4807.038", "latitudeHemisphere": "N", "longitudeFloat": "01131.000", "longitudeHemisphere": "E", "numberOfSatellitesSeen": "08", "timestamp": "123519", "fixQuality": "1", } self._parserTest(GPGGA, expected) def test_fullGLL(self): """ A full GLL sentence is correctly parsed. """ expected = { "type": "GPGLL", "latitudeFloat": "4916.45", "latitudeHemisphere": "N", "longitudeFloat": "12311.12", "longitudeHemisphere": "W", "timestamp": "225444", "dataMode": "A", } self._parserTest(GPGLL, expected) def test_partialGLL(self): """ A partial GLL sentence is correctly parsed. """ expected = { "type": "GPGLL", "latitudeFloat": "3751.65", "latitudeHemisphere": "S", "longitudeFloat": "14507.36", "longitudeHemisphere": "E", } self._parserTest(GPGLL_PARTIAL, expected) def test_fullGSV(self): """ A full GSV sentence is correctly parsed. """ expected = { "type": "GPGSV", "GSVSentenceIndex": "1", "numberOfGSVSentences": "3", "numberOfSatellitesSeen": "11", "azimuth_0": "111", "azimuth_1": "270", "azimuth_2": "010", "azimuth_3": "292", "elevation_0": "03", "elevation_1": "15", "elevation_2": "01", "elevation_3": "06", "satellitePRN_0": "03", "satellitePRN_1": "04", "satellitePRN_2": "06", "satellitePRN_3": "13", "signalToNoiseRatio_0": "00", "signalToNoiseRatio_1": "00", "signalToNoiseRatio_2": "00", "signalToNoiseRatio_3": "00", } self._parserTest(GPGSV_FIRST, expected) def test_partialGSV(self): """ A partial GSV sentence is correctly parsed. """ expected = { "type": "GPGSV", "GSVSentenceIndex": "3", "numberOfGSVSentences": "3", "numberOfSatellitesSeen": "11", "azimuth_0": "067", "azimuth_1": "311", "azimuth_2": "244", "elevation_0": "42", "elevation_1": "14", "elevation_2": "05", "satellitePRN_0": "22", "satellitePRN_1": "24", "satellitePRN_2": "27", "signalToNoiseRatio_0": "42", "signalToNoiseRatio_1": "43", "signalToNoiseRatio_2": "00", } self._parserTest(GPGSV_LAST, expected) def test_fullHDT(self): """ A full HDT sentence is correctly parsed. """ expected = { "type": "GPHDT", "trueHeading": "038.005", } self._parserTest(GPHDT, expected) def test_typicalGSA(self): """ A typical GSA sentence is correctly parsed. """ expected = { "type": "GPGSA", "dataMode": "A", "fixType": "3", "usedSatellitePRN_0": "19", "usedSatellitePRN_1": "28", "usedSatellitePRN_2": "14", "usedSatellitePRN_3": "18", "usedSatellitePRN_4": "27", "usedSatellitePRN_5": "22", "usedSatellitePRN_6": "31", "usedSatellitePRN_7": "39", "positionDilutionOfPrecision": "1.7", "horizontalDilutionOfPrecision": "1.0", "verticalDilutionOfPrecision": "1.3", } self._parserTest(GPGSA, expected) class FixUnitsTests(TestCase): """ Tests for the generic unit fixing method, L{nmea.NMEAAdapter._fixUnits}. @ivar adapter: The NMEA adapter. @type adapter: L{nmea.NMEAAdapter} """ def setUp(self): self.adapter = nmea.NMEAAdapter(base.BasePositioningReceiver()) def test_noValueKey(self): """ Tests that when no C{valueKey} is provided, C{unitKey} is used, minus C{"Units"} at the end. """ class FakeSentence: """ A fake sentence that just has a "foo" attribute. """ def __init__(self): self.foo = 1 self.adapter.currentSentence = FakeSentence() self.adapter._fixUnits(unitKey="fooUnits", unit="N") self.assertNotEqual(self.adapter._sentenceData["foo"], 1) def test_unitKeyButNoUnit(self): """ Tests that if a unit key is provided but the unit isn't, the unit is automatically determined from the unit key. """ class FakeSentence: """ A fake sentence that just has "foo" and "fooUnits" attributes. """ def __init__(self): self.foo = 1 self.fooUnits = "N" self.adapter.currentSentence = FakeSentence() self.adapter._fixUnits(unitKey="fooUnits") self.assertNotEqual(self.adapter._sentenceData["foo"], 1) def test_noValueKeyAndNoUnitKey(self): """ Tests that when a unit is specified but neither C{valueKey} nor C{unitKey} is provided, C{ValueError} is raised. """ self.assertRaises(ValueError, self.adapter._fixUnits, unit="K") class FixerTestMixin: """ Mixin for tests for the fixers on L{nmea.NMEAAdapter} that adapt from NMEA-specific notations to generic Python objects. @ivar adapter: The NMEA adapter. @type adapter: L{nmea.NMEAAdapter} """ def setUp(self): self.adapter = nmea.NMEAAdapter(base.BasePositioningReceiver()) def _fixerTest(self, sentenceData, expected=None, exceptionClass=None): """ A generic adapter fixer test. Creates a sentence from the C{sentenceData} and sends that to the adapter. If C{exceptionClass} is not passed, this is assumed to work, and C{expected} is compared with the adapter's internal state. Otherwise, passing the sentence to the adapter is checked to raise C{exceptionClass}. @param sentenceData: Raw sentence content. @type sentenceData: C{dict} mapping C{str} to C{str} @param expected: The expected state of the adapter. @type expected: C{dict} or L{None} @param exceptionClass: The exception to be raised by the adapter. @type exceptionClass: subclass of C{Exception} """ sentence = nmea.NMEASentence(sentenceData) def receiveSentence(): self.adapter.sentenceReceived(sentence) if exceptionClass is None: receiveSentence() self.assertEqual(self.adapter._state, expected) else: self.assertRaises(exceptionClass, receiveSentence) self.adapter.clear() class TimestampFixerTests(FixerTestMixin, TestCase): """ Tests conversion from NMEA timestamps to C{datetime.time} objects. """ def test_simple(self): """ A simple timestamp is converted correctly. """ data = {"timestamp": "123456"} # 12:34:56Z expected = {"_time": datetime.time(12, 34, 56)} self._fixerTest(data, expected) def test_broken(self): """ A broken timestamp raises C{ValueError}. """ badTimestamps = "993456", "129956", "123499" for t in badTimestamps: self._fixerTest({"timestamp": t}, exceptionClass=ValueError) class DatestampFixerTests(FixerTestMixin, TestCase): def test_defaultYearThreshold(self): """ The default year threshold is 1980. """ self.assertEqual(self.adapter.yearThreshold, 1980) def test_beforeThreshold(self): """ Dates before the threshold are interpreted as being in the century after the threshold. (Since the threshold is the earliest possible date.) """ datestring, date = "010115", datetime.date(2015, 1, 1) self._fixerTest({"datestamp": datestring}, {"_date": date}) def test_afterThreshold(self): """ Dates after the threshold are interpreted as being in the same century as the threshold. """ datestring, date = "010195", datetime.date(1995, 1, 1) self._fixerTest({"datestamp": datestring}, {"_date": date}) def test_invalidMonth(self): """ A datestring with an invalid month (> 12) raises C{ValueError}. """ self._fixerTest({"datestamp": "011301"}, exceptionClass=ValueError) def test_invalidDay(self): """ A datestring with an invalid day (more days than there are in that month) raises C{ValueError}. """ self._fixerTest({"datestamp": "320101"}, exceptionClass=ValueError) self._fixerTest({"datestamp": "300201"}, exceptionClass=ValueError) def _nmeaFloat(degrees, minutes): """ Builds an NMEA float representation for a given angle in degrees and decimal minutes. @param degrees: The integer degrees for this angle. @type degrees: C{int} @param minutes: The decimal minutes value for this angle. @type minutes: C{float} @return: The NMEA float representation for this angle. @rtype: C{str} """ return "%i%0.3f" % (degrees, minutes) def _coordinateSign(hemisphere): """ Return the sign of a coordinate. This is C{1} if the coordinate is in the northern or eastern hemispheres, C{-1} otherwise. @param hemisphere: NMEA shorthand for the hemisphere. One of "NESW". @type hemisphere: C{str} @return: The sign of the coordinate value. @rtype: C{int} """ return 1 if hemisphere in "NE" else -1 def _coordinateType(hemisphere): """ Return the type of a coordinate. This is L{Angles.LATITUDE} if the coordinate is in the northern or southern hemispheres, L{Angles.LONGITUDE} otherwise. @param hemisphere: NMEA shorthand for the hemisphere. One of "NESW". @type hemisphere: C{str} @return: The type of the coordinate (L{Angles.LATITUDE} or L{Angles.LONGITUDE}) """ return Angles.LATITUDE if hemisphere in "NS" else Angles.LONGITUDE class CoordinateFixerTests(FixerTestMixin, TestCase): """ Tests turning NMEA coordinate notations into something more pleasant. """ def test_north(self): """ NMEA coordinate representations in the northern hemisphere convert correctly. """ sentenceData = {"latitudeFloat": "1030.000", "latitudeHemisphere": "N"} state = {"latitude": base.Coordinate(10.5, Angles.LATITUDE)} self._fixerTest(sentenceData, state) def test_south(self): """ NMEA coordinate representations in the southern hemisphere convert correctly. """ sentenceData = {"latitudeFloat": "1030.000", "latitudeHemisphere": "S"} state = {"latitude": base.Coordinate(-10.5, Angles.LATITUDE)} self._fixerTest(sentenceData, state) def test_east(self): """ NMEA coordinate representations in the eastern hemisphere convert correctly. """ sentenceData = {"longitudeFloat": "1030.000", "longitudeHemisphere": "E"} state = {"longitude": base.Coordinate(10.5, Angles.LONGITUDE)} self._fixerTest(sentenceData, state) def test_west(self): """ NMEA coordinate representations in the western hemisphere convert correctly. """ sentenceData = {"longitudeFloat": "1030.000", "longitudeHemisphere": "W"} state = {"longitude": base.Coordinate(-10.5, Angles.LONGITUDE)} self._fixerTest(sentenceData, state) def test_badHemisphere(self): """ NMEA coordinate representations for nonexistent hemispheres raise C{ValueError} when you attempt to parse them. """ sentenceData = {"longitudeHemisphere": "Q"} self._fixerTest(sentenceData, exceptionClass=ValueError) def test_badHemisphereSign(self): """ NMEA coordinate repesentation parsing fails predictably when you pass nonexistent coordinate types (not latitude or longitude). """ getSign = lambda: self.adapter._getHemisphereSign("BOGUS_VALUE") self.assertRaises(ValueError, getSign) class AltitudeFixerTests(FixerTestMixin, TestCase): """ Tests that NMEA representations of altitudes are correctly converted. """ def test_fixAltitude(self): """ The NMEA representation of an altitude (above mean sea level) is correctly converted. """ key, value = "altitude", "545.4" altitude = base.Altitude(float(value)) self._fixerTest({key: value}, {key: altitude}) def test_heightOfGeoidAboveWGS84(self): """ The NMEA representation of an altitude of the geoid (above the WGS84 reference level) is correctly converted. """ key, value = "heightOfGeoidAboveWGS84", "46.9" altitude = base.Altitude(float(value)) self._fixerTest({key: value}, {key: altitude}) class SpeedFixerTests(FixerTestMixin, TestCase): """ Tests that NMEA representations of speeds are correctly converted. """ def test_speedInKnots(self): """ Speeds reported in knots correctly get converted to meters per second. """ key, value, targetKey = "speedInKnots", "10", "speed" speed = base.Speed(float(value) * base.MPS_PER_KNOT) self._fixerTest({key: value}, {targetKey: speed}) class VariationFixerTests(FixerTestMixin, TestCase): """ Tests if the absolute values of magnetic variations on the heading and their sign get combined correctly, and if that value gets combined with a heading correctly. """ def test_west(self): """ Tests westward (negative) magnetic variation. """ variation, direction = "1.34", "W" heading = base.Heading.fromFloats(variationValue=-1 * float(variation)) sentenceData = { "magneticVariation": variation, "magneticVariationDirection": direction, } self._fixerTest(sentenceData, {"heading": heading}) def test_east(self): """ Tests eastward (positive) magnetic variation. """ variation, direction = "1.34", "E" heading = base.Heading.fromFloats(variationValue=float(variation)) sentenceData = { "magneticVariation": variation, "magneticVariationDirection": direction, } self._fixerTest(sentenceData, {"heading": heading}) def test_withHeading(self): """ Variation values get combined with headings correctly. """ trueHeading, variation, direction = "123.12", "1.34", "E" sentenceData = { "trueHeading": trueHeading, "magneticVariation": variation, "magneticVariationDirection": direction, } heading = base.Heading.fromFloats( float(trueHeading), variationValue=float(variation) ) self._fixerTest(sentenceData, {"heading": heading}) class PositionErrorFixerTests(FixerTestMixin, TestCase): """ Position errors in NMEA are passed as dilutions of precision (DOP). This is a measure relative to some specified value of the GPS device as its "reference" precision. Unfortunately, there are very few ways of figuring this out from just the device (sans manual). There are two basic DOP values: vertical and horizontal. HDOP tells you how precise your location is on the face of the earth (pretending it's flat, at least locally). VDOP tells you how precise your altitude is known. PDOP (position DOP) is a dependent value defined as the Euclidean norm of those two, and gives you a more generic "goodness of fix" value. """ def test_simple(self): self._fixerTest( {"horizontalDilutionOfPrecision": "11"}, {"positionError": base.PositionError(hdop=11.0)}, ) def test_mixing(self): pdop, hdop, vdop = "1", "1", "1" positionError = base.PositionError( pdop=float(pdop), hdop=float(hdop), vdop=float(vdop) ) sentenceData = { "positionDilutionOfPrecision": pdop, "horizontalDilutionOfPrecision": hdop, "verticalDilutionOfPrecision": vdop, } self._fixerTest(sentenceData, {"positionError": positionError}) class ValidFixTests(FixerTestMixin, TestCase): """ Tests that data reported from a valid fix is used. """ def test_GGA(self): """ GGA data with a valid fix is used. """ sentenceData = { "type": "GPGGA", "altitude": "545.4", "fixQuality": nmea.GPGGAFixQualities.GPS_FIX, } expectedState = {"altitude": base.Altitude(545.4)} self._fixerTest(sentenceData, expectedState) def test_GLL(self): """ GLL data with a valid data mode is used. """ sentenceData = { "type": "GPGLL", "altitude": "545.4", "dataMode": nmea.GPGLLGPRMCFixQualities.ACTIVE, } expectedState = {"altitude": base.Altitude(545.4)} self._fixerTest(sentenceData, expectedState) class InvalidFixTests(FixerTestMixin, TestCase): """ Tests that data being reported from a bad or incomplete fix isn't used. Although the specification dictates that GPSes shouldn't produce NMEA sentences with real-looking values for altitude or position in them unless they have at least some semblance of a GPS fix, this is widely ignored. """ def _invalidFixTest(self, sentenceData): """ Sentences with an invalid fix or data mode result in empty state (ie, the data isn't used). """ self._fixerTest(sentenceData, {}) def test_GGA(self): """ GGA sentence data is unused when there is no fix. """ sentenceData = { "type": "GPGGA", "altitude": "545.4", "fixQuality": nmea.GPGGAFixQualities.INVALID_FIX, } self._invalidFixTest(sentenceData) def test_GLL(self): """ GLL sentence data is unused when the data is flagged as void. """ sentenceData = { "type": "GPGLL", "altitude": "545.4", "dataMode": nmea.GPGLLGPRMCFixQualities.VOID, } self._invalidFixTest(sentenceData) def test_badGSADataMode(self): """ GSA sentence data is not used when there is no GPS fix, but the data mode claims the data is "active". Some GPSes do this, unfortunately, and that means you shouldn't use the data. """ sentenceData = { "type": "GPGSA", "altitude": "545.4", "dataMode": nmea.GPGLLGPRMCFixQualities.ACTIVE, "fixType": nmea.GPGSAFixTypes.GSA_NO_FIX, } self._invalidFixTest(sentenceData) def test_badGSAFixType(self): """ GSA sentence data is not used when the fix claims to be valid (albeit only 2D), but the data mode says the data is void. Some GPSes do this, unfortunately, and that means you shouldn't use the data. """ sentenceData = { "type": "GPGSA", "altitude": "545.4", "dataMode": nmea.GPGLLGPRMCFixQualities.VOID, "fixType": nmea.GPGSAFixTypes.GSA_2D_FIX, } self._invalidFixTest(sentenceData) def test_badGSADataModeAndFixType(self): """ GSA sentence data is not use when neither the fix nor the data mode is any good. """ sentenceData = { "type": "GPGSA", "altitude": "545.4", "dataMode": nmea.GPGLLGPRMCFixQualities.VOID, "fixType": nmea.GPGSAFixTypes.GSA_NO_FIX, } self._invalidFixTest(sentenceData) class NMEAReceiverTests(TestCase): """ Tests for the NMEA receiver. """ def setUp(self): self.receiver = MockPositioningReceiver() self.adapter = nmea.NMEAAdapter(self.receiver) self.protocol = nmea.NMEAProtocol(self.adapter) def test_onlyFireWhenCurrentSentenceHasNewInformation(self): """ If the current sentence does not contain any new fields for a particular callback, that callback is not called; even if all necessary information is still in the state from one or more previous messages. """ self.protocol.lineReceived(GPGGA) gpggaCallbacks = { "positionReceived", "positionErrorReceived", "altitudeReceived", } self.assertEqual(set(self.receiver.called.keys()), gpggaCallbacks) self.receiver.clear() self.assertNotEqual(self.adapter._state, {}) # GPHDT contains heading information but not position, # altitude or anything like that; but that information is # still in the state. self.protocol.lineReceived(GPHDT) gphdtCallbacks = {"headingReceived"} self.assertEqual(set(self.receiver.called.keys()), gphdtCallbacks) def _receiverTest(self, sentences, expectedFired=(), extraTest=None): """ A generic test for NMEA receiver behavior. @param sentences: The sequence of sentences to simulate receiving. @type sentences: iterable of C{str} @param expectedFired: The names of the callbacks expected to fire. @type expectedFired: iterable of C{str} @param extraTest: An optional extra test hook. @type extraTest: nullary callable """ for sentence in sentences: self.protocol.lineReceived(sentence) actuallyFired = self.receiver.called.keys() self.assertEqual(set(actuallyFired), set(expectedFired)) if extraTest is not None: extraTest() self.receiver.clear() self.adapter.clear() def test_positionErrorUpdateAcrossStates(self): """ The positioning error is updated across multiple states. """ sentences = [GPGSA] + GPGSV_SEQ callbacksFired = ["positionErrorReceived", "beaconInformationReceived"] def _getIdentifiers(beacons): return sorted(map(attrgetter("identifier"), beacons)) def checkBeaconInformation(): beaconInformation = self.adapter._state["beaconInformation"] seenIdentifiers = _getIdentifiers(beaconInformation.seenBeacons) expected = [3, 4, 6, 13, 14, 16, 18, 19, 22, 24, 27] self.assertEqual(seenIdentifiers, expected) usedIdentifiers = _getIdentifiers(beaconInformation.usedBeacons) # These are not actually all the PRNs in the sample GPGSA: # only the ones also reported by the GPGSV sequence. This # is just because the sample data doesn't come from the # same reporting cycle of a GPS device. self.assertEqual(usedIdentifiers, [14, 18, 19, 22, 27]) self._receiverTest(sentences, callbacksFired, checkBeaconInformation) def test_emptyMiddleGSV(self): """ A GSV sentence with empty entries in any position does not mean that entries in subsequent positions of the same GSV sentence are ignored. """ sentences = [GPGSV_EMPTY_MIDDLE] callbacksFired = ["beaconInformationReceived"] def checkBeaconInformation(): beaconInformation = self.adapter._state["beaconInformation"] seenBeacons = beaconInformation.seenBeacons self.assertEqual(len(seenBeacons), 2) self.assertIn(13, [b.identifier for b in seenBeacons]) self._receiverTest(sentences, callbacksFired, checkBeaconInformation) def test_GGASentences(self): """ A sequence of GGA sentences fires C{positionReceived}, C{positionErrorReceived} and C{altitudeReceived}. """ sentences = [GPGGA] callbacksFired = [ "positionReceived", "positionErrorReceived", "altitudeReceived", ] self._receiverTest(sentences, callbacksFired) def test_GGAWithDateInState(self): """ When receiving a GPGGA sentence and a date was already in the state, the new time (from the GPGGA sentence) is combined with that date. """ self.adapter._state["_date"] = datetime.date(2014, 1, 1) sentences = [GPGGA] callbacksFired = [ "positionReceived", "positionErrorReceived", "altitudeReceived", "timeReceived", ] self._receiverTest(sentences, callbacksFired) def test_RMCSentences(self): """ A sequence of RMC sentences fires C{positionReceived}, C{speedReceived}, C{headingReceived} and C{timeReceived}. """ sentences = [GPRMC] callbacksFired = [ "headingReceived", "speedReceived", "positionReceived", "timeReceived", ] self._receiverTest(sentences, callbacksFired) def test_GSVSentences(self): """ A complete sequence of GSV sentences fires C{beaconInformationReceived}. """ sentences = [GPGSV_FIRST, GPGSV_MIDDLE, GPGSV_LAST] callbacksFired = ["beaconInformationReceived"] def checkPartialInformation(): self.assertNotIn("_partialBeaconInformation", self.adapter._state) self._receiverTest(sentences, callbacksFired, checkPartialInformation) def test_emptyMiddleEntriesGSVSequence(self): """ A complete sequence of GSV sentences with empty entries in the middle still fires C{beaconInformationReceived}. """ sentences = [GPGSV_EMPTY_MIDDLE] self._receiverTest(sentences, ["beaconInformationReceived"]) def test_incompleteGSVSequence(self): """ An incomplete sequence of GSV sentences does not fire any callbacks. """ sentences = [GPGSV_FIRST] self._receiverTest(sentences) def test_singleSentenceGSVSequence(self): """ The parser does not fail badly when the sequence consists of only one sentence (but is otherwise complete). """ sentences = [GPGSV_SINGLE] self._receiverTest(sentences, ["beaconInformationReceived"]) def test_GLLSentences(self): """ GLL sentences fire C{positionReceived}. """ sentences = [GPGLL_PARTIAL, GPGLL] self._receiverTest(sentences, ["positionReceived"]) def test_HDTSentences(self): """ HDT sentences fire C{headingReceived}. """ sentences = [GPHDT] self._receiverTest(sentences, ["headingReceived"]) def test_mixedSentences(self): """ A mix of sentences fires the correct callbacks. """ sentences = [GPRMC, GPGGA] callbacksFired = [ "altitudeReceived", "speedReceived", "positionReceived", "positionErrorReceived", "timeReceived", "headingReceived", ] def checkTime(): expectedDateTime = datetime.datetime(1994, 3, 23, 12, 35, 19) self.assertEqual(self.adapter._state["time"], expectedDateTime) self._receiverTest(sentences, callbacksFired, checkTime) def test_lotsOfMixedSentences(self): """ Sends an entire gamut of sentences and verifies the appropriate callbacks fire. These are more than you'd expect from your average consumer GPS device. They have most of the important information, including beacon information and visibility. """ sentences = [GPGSA] + GPGSV_SEQ + [GPRMC, GPGGA, GPGLL] callbacksFired = [ "headingReceived", "beaconInformationReceived", "speedReceived", "positionReceived", "timeReceived", "altitudeReceived", "positionErrorReceived", ] self._receiverTest(sentences, callbacksFired) PK ���Z�v^] ] receiver.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Receivers for use in tests. """ from twisted.positioning import base, ipositioning class MockPositioningReceiver(base.BasePositioningReceiver): """ A mock positioning receiver. Mocks all the L{IPositioningReceiver} methods with stubs that don't do anything but register that they were called. @ivar called: A mapping of names of callbacks that have been called to C{True}. @type called: C{dict} """ def __init__(self): self.clear() for methodName in ipositioning.IPositioningReceiver: self._addCallback(methodName) def clear(self): """ Forget all the methods that have been called on this receiver, by emptying C{self.called}. """ self.called = {} def _addCallback(self, name): """ Adds a callback of the given name, setting C{self.called[name]} to C{True} when called. """ def callback(*a, **kw): self.called[name] = True setattr(self, name, callback) PK ���Z�`� Z Z test_sentence.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for positioning sentences. """ import itertools from twisted.positioning import _sentence from twisted.trial.unittest import TestCase sentinelValueOne = "someStringValue" sentinelValueTwo = "someOtherStringValue" class DummyProtocol: """ A simple, fake protocol. """ @staticmethod def getSentenceAttributes(): return ["type", sentinelValueOne, sentinelValueTwo] class DummySentence(_sentence._BaseSentence): """ A sentence for L{DummyProtocol}. """ ALLOWED_ATTRIBUTES = DummyProtocol.getSentenceAttributes() class MixinProtocol(_sentence._PositioningSentenceProducerMixin): """ A simple, fake protocol that declaratively tells you the sentences it produces using L{base.PositioningSentenceProducerMixin}. """ _SENTENCE_CONTENTS = { None: [ sentinelValueOne, sentinelValueTwo, None, # See MixinTests.test_noNoneInSentenceAttributes ], } class MixinSentence(_sentence._BaseSentence): """ A sentence for L{MixinProtocol}. """ ALLOWED_ATTRIBUTES = MixinProtocol.getSentenceAttributes() class SentenceTestsMixin: """ Tests for positioning protocols and their respective sentences. """ def test_attributeAccess(self): """ A sentence attribute gets the correct value, and accessing an unset attribute (which is specified as being a valid sentence attribute) gets L{None}. """ thisSentinel = object() sentence = self.sentenceClass({sentinelValueOne: thisSentinel}) self.assertEqual(getattr(sentence, sentinelValueOne), thisSentinel) self.assertIsNone(getattr(sentence, sentinelValueTwo)) def test_raiseOnMissingAttributeAccess(self): """ Accessing a nonexistent attribute raises C{AttributeError}. """ sentence = self.sentenceClass({}) self.assertRaises(AttributeError, getattr, sentence, "BOGUS") def test_raiseOnBadAttributeAccess(self): """ Accessing bogus attributes raises C{AttributeError}, *even* when that attribute actually is in the sentence data. """ sentence = self.sentenceClass({"BOGUS": None}) self.assertRaises(AttributeError, getattr, sentence, "BOGUS") sentenceType = "tummies" reprTemplate = "<%s (%s) {%s}>" def _expectedRepr(self, sentenceType="unknown type", dataRepr=""): """ Builds the expected repr for a sentence. @param sentenceType: The name of the sentence type (e.g "GPGGA"). @type sentenceType: C{str} @param dataRepr: The repr of the data in the sentence. @type dataRepr: C{str} @return: The expected repr of the sentence. @rtype: C{str} """ clsName = self.sentenceClass.__name__ return self.reprTemplate % (clsName, sentenceType, dataRepr) def test_unknownTypeRepr(self): """ Test the repr of an empty sentence of unknown type. """ sentence = self.sentenceClass({}) expectedRepr = self._expectedRepr() self.assertEqual(repr(sentence), expectedRepr) def test_knownTypeRepr(self): """ Test the repr of an empty sentence of known type. """ sentence = self.sentenceClass({"type": self.sentenceType}) expectedRepr = self._expectedRepr(self.sentenceType) self.assertEqual(repr(sentence), expectedRepr) class MixinTests(TestCase, SentenceTestsMixin): """ Tests for protocols deriving from L{base.PositioningSentenceProducerMixin} and their sentences. """ def setUp(self): self.protocol = MixinProtocol() self.sentenceClass = MixinSentence def test_noNoneInSentenceAttributes(self): """ L{None} does not appear in the sentence attributes of the protocol, even though it's in the specification. This is because L{None} is a placeholder for parts of the sentence you don't really need or want, but there are some bits later on in the sentence that you do want. The alternative would be to have to specify things like "_UNUSED0", "_UNUSED1"... which would end up cluttering the sentence data and eventually adapter state. """ sentenceAttributes = self.protocol.getSentenceAttributes() self.assertNotIn(None, sentenceAttributes) sentenceContents = self.protocol._SENTENCE_CONTENTS sentenceSpecAttributes = itertools.chain(*sentenceContents.values()) self.assertIn(None, sentenceSpecAttributes) PK ͪ�Z��|� test_logger.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._logger}. """ from typing import List, Optional, Type, cast from zope.interface import implementer from constantly import NamedConstant # type: ignore[import] from twisted.trial import unittest from .._format import formatEvent from .._global import globalLogPublisher from .._interfaces import ILogObserver, LogEvent from .._levels import InvalidLogLevelError, LogLevel from .._logger import Logger class TestLogger(Logger): """ L{Logger} with an overridden C{emit} method that keeps track of received events. """ def emit( self, level: NamedConstant, format: Optional[str] = None, **kwargs: object ) -> None: @implementer(ILogObserver) def observer(event: LogEvent) -> None: self.event = event globalLogPublisher.addObserver(observer) try: Logger.emit(self, level, format, **kwargs) finally: globalLogPublisher.removeObserver(observer) self.emitted = { "level": level, "format": format, "kwargs": kwargs, } class LogComposedObject: """ A regular object, with a logger attached. """ log = TestLogger() def __init__(self, state: Optional[str] = None) -> None: self.state = state def __str__(self) -> str: return f"<LogComposedObject {self.state}>" class LoggerTests(unittest.TestCase): """ Tests for L{Logger}. """ def test_repr(self) -> None: """ repr() on Logger """ namespace = "bleargh" log = Logger(namespace) self.assertEqual(repr(log), f"<Logger {repr(namespace)}>") def test_namespaceDefault(self) -> None: """ Default namespace is module name. """ log = Logger() self.assertEqual(log.namespace, __name__) def test_namespaceOMGItsTooHard(self) -> None: """ Default namespace is C{"<unknown>"} when a logger is created from a context in which is can't be determined automatically and no namespace was specified. """ result: List[Logger] = [] exec( "result.append(Logger())", dict(Logger=Logger), locals(), ) self.assertEqual(result[0].namespace, "<unknown>") def test_namespaceAttribute(self) -> None: """ Default namespace for classes using L{Logger} as a descriptor is the class name they were retrieved from. """ obj = LogComposedObject() expectedNamespace = "{}.{}".format( obj.__module__, obj.__class__.__name__, ) self.assertEqual(cast(TestLogger, obj.log).namespace, expectedNamespace) self.assertEqual( cast(Type[TestLogger], LogComposedObject.log).namespace, expectedNamespace ) self.assertIs( cast(Type[TestLogger], LogComposedObject.log).source, LogComposedObject ) self.assertIs(cast(TestLogger, obj.log).source, obj) self.assertIsNone(Logger().source) def test_descriptorObserver(self) -> None: """ When used as a descriptor, the observer is propagated. """ observed: List[LogEvent] = [] class MyObject: log = Logger(observer=cast(ILogObserver, observed.append)) MyObject.log.info("hello") self.assertEqual(len(observed), 1) self.assertEqual(observed[0]["log_format"], "hello") def test_sourceAvailableForFormatting(self) -> None: """ On instances that have a L{Logger} class attribute, the C{log_source} key is available to format strings. """ obj = LogComposedObject("hello") log = cast(TestLogger, obj.log) log.error("Hello, {log_source}.") self.assertIn("log_source", log.event) self.assertEqual(log.event["log_source"], obj) stuff = formatEvent(log.event) self.assertIn("Hello, <LogComposedObject hello>.", stuff) def test_basicLogger(self) -> None: """ Test that log levels and messages are emitted correctly for Logger. """ log = TestLogger() for level in LogLevel.iterconstants(): format = "This is a {level_name} message" message = format.format(level_name=level.name) logMethod = getattr(log, level.name) logMethod(format, junk=message, level_name=level.name) # Ensure that test_emit got called with expected arguments self.assertEqual(log.emitted["level"], level) self.assertEqual(log.emitted["format"], format) self.assertEqual(log.emitted["kwargs"]["junk"], message) self.assertTrue(hasattr(log, "event"), "No event observed.") self.assertEqual(log.event["log_format"], format) self.assertEqual(log.event["log_level"], level) self.assertEqual(log.event["log_namespace"], __name__) self.assertIsNone(log.event["log_source"]) self.assertEqual(log.event["junk"], message) self.assertEqual(formatEvent(log.event), message) def test_sourceOnClass(self) -> None: """ C{log_source} event key refers to the class. """ @implementer(ILogObserver) def observer(event: LogEvent) -> None: self.assertEqual(event["log_source"], Thingo) class Thingo: log = TestLogger(observer=observer) cast(TestLogger, Thingo.log).info() def test_sourceOnInstance(self) -> None: """ C{log_source} event key refers to the instance. """ @implementer(ILogObserver) def observer(event: LogEvent) -> None: self.assertEqual(event["log_source"], thingo) class Thingo: log = TestLogger(observer=observer) thingo = Thingo() cast(TestLogger, thingo.log).info() def test_sourceUnbound(self) -> None: """ C{log_source} event key is L{None}. """ @implementer(ILogObserver) def observer(event: LogEvent) -> None: self.assertIsNone(event["log_source"]) log = TestLogger(observer=observer) log.info() def test_defaultFailure(self) -> None: """ Test that log.failure() emits the right data. """ log = TestLogger() try: raise RuntimeError("baloney!") except RuntimeError: log.failure("Whoops") errors = self.flushLoggedErrors(RuntimeError) self.assertEqual(len(errors), 1) self.assertEqual(log.emitted["level"], LogLevel.critical) self.assertEqual(log.emitted["format"], "Whoops") def test_conflictingKwargs(self) -> None: """ Make sure that kwargs conflicting with args don't pass through. """ log = TestLogger() log.warn( "*", log_format="#", log_level=LogLevel.error, log_namespace="*namespace*", log_source="*source*", ) self.assertEqual(log.event["log_format"], "*") self.assertEqual(log.event["log_level"], LogLevel.warn) self.assertEqual(log.event["log_namespace"], log.namespace) self.assertIsNone(log.event["log_source"]) def test_logInvalidLogLevel(self) -> None: """ Test passing in a bogus log level to C{emit()}. """ log = TestLogger() log.emit("*bogus*") errors = self.flushLoggedErrors(InvalidLogLevelError) self.assertEqual(len(errors), 1) def test_trace(self) -> None: """ Tracing keeps track of forwarding to the publisher. """ @implementer(ILogObserver) def publisher(event: LogEvent) -> None: observer(event) @implementer(ILogObserver) def observer(event: LogEvent) -> None: self.assertEqual(event["log_trace"], [(log, publisher)]) log = TestLogger(observer=publisher) log.info("Hello.", log_trace=[]) PK ͪ�Z*�8�3 �3 test_filter.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._filter}. """ from typing import Iterable, List, Tuple, Union, cast from zope.interface import implementer from zope.interface.exceptions import BrokenMethodImplementation from zope.interface.verify import verifyObject from constantly import NamedConstant # type: ignore[import] from twisted.trial import unittest from .._filter import ( FilteringLogObserver, ILogFilterPredicate, LogLevelFilterPredicate, PredicateResult, ) from .._interfaces import ILogObserver, LogEvent from .._levels import InvalidLogLevelError, LogLevel from .._observer import LogPublisher, bitbucketLogObserver class FilteringLogObserverTests(unittest.TestCase): """ Tests for L{FilteringLogObserver}. """ def test_interface(self) -> None: """ L{FilteringLogObserver} is an L{ILogObserver}. """ observer = FilteringLogObserver(cast(ILogObserver, lambda e: None), ()) try: verifyObject(ILogObserver, observer) except BrokenMethodImplementation as e: self.fail(e) def filterWith( self, filters: Iterable[str], other: bool = False ) -> Union[List[int], Tuple[List[int], List[int]]]: """ Apply a set of pre-defined filters on a known set of events and return the filtered list of event numbers. The pre-defined events are four events with a C{count} attribute set to C{0}, C{1}, C{2}, and C{3}. @param filters: names of the filters to apply. Options are: - C{"twoMinus"} (count <=2), - C{"twoPlus"} (count >= 2), - C{"notTwo"} (count != 2), - C{"no"} (False). @param other: Whether to return a list of filtered events as well. @return: event numbers or 2-tuple of lists of event numbers. """ events: List[LogEvent] = [ dict(count=0), dict(count=1), dict(count=2), dict(count=3), ] class Filters: @staticmethod def twoMinus(event: LogEvent) -> NamedConstant: """ count <= 2 @param event: an event @return: L{PredicateResult.yes} if C{event["count"] <= 2}, otherwise L{PredicateResult.maybe}. """ if event["count"] <= 2: return PredicateResult.yes return PredicateResult.maybe @staticmethod def twoPlus(event: LogEvent) -> NamedConstant: """ count >= 2 @param event: an event @return: L{PredicateResult.yes} if C{event["count"] >= 2}, otherwise L{PredicateResult.maybe}. """ if event["count"] >= 2: return PredicateResult.yes return PredicateResult.maybe @staticmethod def notTwo(event: LogEvent) -> NamedConstant: """ count != 2 @param event: an event @return: L{PredicateResult.yes} if C{event["count"] != 2}, otherwise L{PredicateResult.maybe}. """ if event["count"] == 2: return PredicateResult.no return PredicateResult.maybe @staticmethod def no(event: LogEvent) -> NamedConstant: """ No way, man. @param event: an event @return: L{PredicateResult.no} """ return PredicateResult.no @staticmethod def bogus(event: LogEvent) -> NamedConstant: """ Bogus result. @param event: an event @return: something other than a valid predicate result. """ return None predicates = (getattr(Filters, f) for f in filters) eventsSeen: List[LogEvent] = [] eventsNotSeen: List[LogEvent] = [] trackingObserver = cast(ILogObserver, eventsSeen.append) if other: negativeObserver = cast(ILogObserver, eventsNotSeen.append) else: negativeObserver = bitbucketLogObserver filteringObserver = FilteringLogObserver( trackingObserver, predicates, negativeObserver ) for e in events: filteringObserver(e) if other: return ( [cast(int, e["count"]) for e in eventsSeen], [cast(int, e["count"]) for e in eventsNotSeen], ) else: return [cast(int, e["count"]) for e in eventsSeen] def test_shouldLogEventNoFilters(self) -> None: """ No filters: all events come through. """ self.assertEqual(self.filterWith([]), [0, 1, 2, 3]) def test_shouldLogEventNoFilter(self) -> None: """ Filter with negative predicate result. """ self.assertEqual(self.filterWith(["notTwo"]), [0, 1, 3]) def test_shouldLogEventOtherObserver(self) -> None: """ Filtered results get sent to the other observer, if passed. """ self.assertEqual(self.filterWith(["notTwo"], True), ([0, 1, 3], [2])) def test_shouldLogEventYesFilter(self) -> None: """ Filter with positive predicate result. """ self.assertEqual(self.filterWith(["twoPlus"]), [0, 1, 2, 3]) def test_shouldLogEventYesNoFilter(self) -> None: """ Series of filters with positive and negative predicate results. """ self.assertEqual(self.filterWith(["twoPlus", "no"]), [2, 3]) def test_shouldLogEventYesYesNoFilter(self) -> None: """ Series of filters with positive, positive and negative predicate results. """ self.assertEqual(self.filterWith(["twoPlus", "twoMinus", "no"]), [0, 1, 2, 3]) def test_shouldLogEventBadPredicateResult(self) -> None: """ Filter with invalid predicate result. """ self.assertRaises(TypeError, self.filterWith, ["bogus"]) def test_call(self) -> None: """ Test filtering results from each predicate type. """ e: LogEvent = dict(obj=object()) def callWithPredicateResult(result: NamedConstant) -> List[LogEvent]: seen: List[LogEvent] = [] observer = FilteringLogObserver( cast(ILogObserver, lambda e: seen.append(e)), (cast(ILogFilterPredicate, lambda e: result),), ) observer(e) return seen self.assertIn(e, callWithPredicateResult(PredicateResult.yes)) self.assertIn(e, callWithPredicateResult(PredicateResult.maybe)) self.assertNotIn(e, callWithPredicateResult(PredicateResult.no)) def test_trace(self) -> None: """ Tracing keeps track of forwarding through the filtering observer. """ event: LogEvent = dict(log_trace=[]) oYes = cast(ILogObserver, lambda e: None) oNo = cast(ILogObserver, lambda e: None) @implementer(ILogObserver) def testObserver(e: LogEvent) -> None: self.assertIs(e, event) self.assertEqual( event["log_trace"], [ (publisher, yesFilter), (yesFilter, oYes), (publisher, noFilter), # ... noFilter doesn't call oNo (publisher, oTest), ], ) oTest = testObserver yesFilter = FilteringLogObserver( oYes, (cast(ILogFilterPredicate, lambda e: PredicateResult.yes),) ) noFilter = FilteringLogObserver( oNo, (cast(ILogFilterPredicate, lambda e: PredicateResult.no),) ) publisher = LogPublisher(yesFilter, noFilter, testObserver) publisher(event) class LogLevelFilterPredicateTests(unittest.TestCase): """ Tests for L{LogLevelFilterPredicate}. """ def test_defaultLogLevel(self) -> None: """ Default log level is used. """ predicate = LogLevelFilterPredicate() # Test using both "" and None as default namespace, because None was the # documented default value in the past. for default in ("", cast(str, None)): self.assertEqual( predicate.logLevelForNamespace(default), predicate.defaultLogLevel ) self.assertEqual( predicate.logLevelForNamespace("rocker.cool.namespace"), predicate.defaultLogLevel, ) def test_setLogLevel(self) -> None: """ Setting and retrieving log levels. """ predicate = LogLevelFilterPredicate() # Test using both "" and None as default namespace, because None was the # documented default value in the past. for default in ("", cast(str, None)): predicate.setLogLevelForNamespace(default, LogLevel.error) predicate.setLogLevelForNamespace("twext.web2", LogLevel.debug) predicate.setLogLevelForNamespace("twext.web2.dav", LogLevel.warn) self.assertEqual(predicate.logLevelForNamespace(""), LogLevel.error) self.assertEqual( predicate.logLevelForNamespace(cast(str, None)), LogLevel.error ) self.assertEqual(predicate.logLevelForNamespace("twisted"), LogLevel.error) self.assertEqual( predicate.logLevelForNamespace("twext.web2"), LogLevel.debug ) self.assertEqual( predicate.logLevelForNamespace("twext.web2.dav"), LogLevel.warn ) self.assertEqual( predicate.logLevelForNamespace("twext.web2.dav.test"), LogLevel.warn ) self.assertEqual( predicate.logLevelForNamespace("twext.web2.dav.test1.test2"), LogLevel.warn, ) def test_setInvalidLogLevel(self) -> None: """ Can't pass invalid log levels to C{setLogLevelForNamespace()}. """ predicate = LogLevelFilterPredicate() self.assertRaises( InvalidLogLevelError, predicate.setLogLevelForNamespace, "twext.web2", object(), ) # Level must be a constant, not the name of a constant self.assertRaises( InvalidLogLevelError, predicate.setLogLevelForNamespace, "twext.web2", "debug", ) def test_clearLogLevels(self) -> None: """ Clearing log levels. """ predicate = LogLevelFilterPredicate() predicate.setLogLevelForNamespace("twext.web2", LogLevel.debug) predicate.setLogLevelForNamespace("twext.web2.dav", LogLevel.error) predicate.clearLogLevels() self.assertEqual( predicate.logLevelForNamespace("twisted"), predicate.defaultLogLevel ) self.assertEqual( predicate.logLevelForNamespace("twext.web2"), predicate.defaultLogLevel ) self.assertEqual( predicate.logLevelForNamespace("twext.web2.dav"), predicate.defaultLogLevel ) self.assertEqual( predicate.logLevelForNamespace("twext.web2.dav.test"), predicate.defaultLogLevel, ) self.assertEqual( predicate.logLevelForNamespace("twext.web2.dav.test1.test2"), predicate.defaultLogLevel, ) def test_filtering(self) -> None: """ Events are filtered based on log level/namespace. """ predicate = LogLevelFilterPredicate() predicate.setLogLevelForNamespace("", LogLevel.error) predicate.setLogLevelForNamespace("twext.web2", LogLevel.debug) predicate.setLogLevelForNamespace("twext.web2.dav", LogLevel.warn) def checkPredicate( namespace: str, level: NamedConstant, expectedResult: NamedConstant ) -> None: event: LogEvent = dict(log_namespace=namespace, log_level=level) self.assertEqual(expectedResult, predicate(event)) checkPredicate("", LogLevel.debug, PredicateResult.no) checkPredicate(cast(str, None), LogLevel.debug, PredicateResult.no) checkPredicate("", LogLevel.error, PredicateResult.no) checkPredicate(cast(str, None), LogLevel.error, PredicateResult.no) checkPredicate("twext.web2", LogLevel.debug, PredicateResult.maybe) checkPredicate("twext.web2", LogLevel.error, PredicateResult.maybe) checkPredicate("twext.web2.dav", LogLevel.debug, PredicateResult.no) checkPredicate("twext.web2.dav", LogLevel.error, PredicateResult.maybe) checkPredicate("", LogLevel.critical, PredicateResult.no) checkPredicate(cast(str, None), LogLevel.critical, PredicateResult.no) checkPredicate("twext.web2", None, PredicateResult.no) PK ͪ�Z��.� * * ( __pycache__/test_flatten.cpython-310.pycnu �[��� o �b�% � @ s� d Z ddlZddlmZ ddlmZmZmZ zddlm Z W n e y) dZ Y nw ddl m Z ddl mZmZmZmZ dd lmZ dd lmZ G d d � d e j�ZdS ) z+ Test cases for L{twisted.logger._format}. � N)�count)�Any�Callable�Optional)�tzset)�unittest� )� KeyFlattener� aFormatter� extractField� flattenEvent)� formatEvent)�LogEventc @ s� e Zd ZdZd"dd�Zd"dd�Zd"dd �Zd"d d �Z d#d ee de fd d�Z d"dd�Z d"dd�Z d"dd�Z dd� fdee ge f ddfdd�Zd"dd�Zd"d d �Zd"d d �Zd"d d!�ZdS )$�FlatFormattingTestsz9 Tests for flattened event formatting functions. �returnNc sr t � � G dd� d�}td� fdd�|� dddd �}t|� t|�}|d = |d = t�t�|��}| �t|�d � d S )a) L{flattenEvent} will "flatten" an event so that, if scrubbed of all but serializable objects, it will preserve all necessary data to be formatted once serialized. When presented with an event thusly flattened, L{formatEvent} will produce the same output. c @ s e Zd ZdZdS )z;FlatFormattingTests.test_formatFlatEvent.<locals>.Ephemeral�valueN)�__name__� __module__� __qualname__� attribute� r r �B/usr/lib/python3/dist-packages/twisted/logger/test/test_flatten.py� Ephemeral% s r z�callable: {callme()} attribute: {object.attribute} numrepr: {number!r} numstr: {number!s} strrepr: {string!r} unistr: {unistr!s}c s t � �S �N)�nextr ��counterr r �<lambda>1 s z:FlatFormattingTests.test_formatFlatEvent.<locals>.<lambda>� �hello� ö)� log_format�callme�object�number�string�unistrr" r# uM callable: 0 attribute: value numrepr: 7 numstr: 7 strrepr: 'hello' unistr: öN)r �dictr �json�loads�dumps� assertEqualr )�selfr �event1�event2�event3r r r �test_formatFlatEvent s& ��z(FlatFormattingTests.test_formatFlatEventc C s< t ddd�}t|� t�t�|��}| �t|��d�� dS )zH If the format string is invalid, an error is produced. zstrrepr: {string!X}r )r! r% zUnable to format eventN)r' r r( r) r* � assertTruer � startswith)r, r- r. r r r � test_formatFlatEventBadFormatJ s �z1FlatFormattingTests.test_formatFlatEventBadFormatc C sD G dd� d�}|� }t d|d�}t|� |�� | �t|�d� dS )z� L{formatEvent} will prefer the stored C{str()} or C{repr()} value for an object, in case the other version. c @ s, e Zd ZdZdZd dd�Zdefdd�ZdS ) zPFlatFormattingTests.test_formatFlatEventWithMutatedFields.<locals>.Unpersistablez. Unpersitable object. Fr Nc S s d| _ dS )z0 Self destruct. TN�� destructed�r, r r r � selfDestructe s z]FlatFormattingTests.test_formatFlatEventWithMutatedFields.<locals>.Unpersistable.selfDestructc S s | j rdS dS )Nzpost-serialization garbagezun-persistabler4 r6 r r r �__repr__k s zYFlatFormattingTests.test_formatFlatEventWithMutatedFields.<locals>.Unpersistable.__repr__�r N)r r r �__doc__r5 r7 �strr8 r r r r � Unpersistable^ s r< z unpersistable: {unpersistable})r! � unpersistablez unpersistable: un-persistableN)r' r r7 r+ r )r, r< �upr- r r r �%test_formatFlatEventWithMutatedFieldsX s z9FlatFormattingTests.test_formatFlatEventWithMutatedFieldsc C s dt dt fdd�}z | �|d�d� W n ty � w | �|d�d� | �|d �d � | �|d �d � | �|d �d� | �|d�d� | �|d�d� | �|d�d� | �|d�d� | �|d�d� t� }t�d�\\}}}}|duswJ �| �|�|||�d� | �|�|||�d � dS ) zh Test that L{KeyFlattener.flatKey} returns the expected keys for format fields. �formatr c S s> t �| �D ]\}}}}|d usJ �t� �|||� S J d��)NFz*Unable to derive key from format: {format})r �parser �flatKey)r@ � literalText� fieldName� formatSpec� conversionr r r � keyFromFormat s � z=FlatFormattingTests.test_keyFlattening.<locals>.keyFromFormatz{}z!:z{foo}zfoo!:z{foo!s}zfoo!s:z{foo!r}zfoo!r:z{foo:%s}zfoo!:%sz{foo:!}zfoo!:!z{foo::}zfoo!::z {foo!s:%s}zfoo!s:%sz {foo!s:!}zfoo!s:!z {foo!s::}zfoo!s::z{x}Nzx!:zx!:/2)r; r+ � ValueErrorr r rA rB )r, rG � sameFlattenerrC rD rE rF r r r �test_keyFlatteningy s< �� ��z&FlatFormattingTests.test_keyFlattening�eventc sJ |du rt � � G � fdd�d�}td|� d�}t|� | �t|�d� |S )z� The same format field used twice in one event is rendered twice. @param event: An event to flatten. If L{None}, create a new event. @return: C{event} or the event created. Nc s" e Zd ZdZdef� fdd� ZdS )zJFlatFormattingTests._test_formatFlatEvent_fieldNamesSame.<locals>.CountStrz& Hack r c s t t� ��S r )r; r r6 r r r �__str__� s zRFlatFormattingTests._test_formatFlatEvent_fieldNamesSame.<locals>.CountStr.__str__N)r r r r: r; rL r r r r �CountStr� s rM z{x} {x}�r! �xz0 1)r r' r r+ r )r, rK rM r r r �$_test_formatFlatEvent_fieldNamesSame� s �z8FlatFormattingTests._test_formatFlatEvent_fieldNamesSamec C s | � � dS )zR The same format field used twice in one event is rendered twice. N�rP r6 r r r �"test_formatFlatEventFieldNamesSame� s z6FlatFormattingTests.test_formatFlatEventFieldNamesSamec C s | � � }| � |� dS )zb The same event flattened twice gives the same (already rendered) result. NrQ )r, rK r r r �'test_formatFlatEventFieldNamesSameAgain� s z;FlatFormattingTests.test_formatFlatEventFieldNamesSameAgainc C s, t ddd�}t|� t|�}| �|d� dS )zs L{formatEvent} will handle a flattened event with tailing text after a replacement field. ztest {x} trailingr rN ztest value trailingN)r' r r r+ )r, rK �resultr r r � test_formatEventFlatTrailingText� s �z4FlatFormattingTests.test_formatEventFlatTrailingTextc C s | S r r )rO r r r r � s z FlatFormattingTests.<lambda>� flattenFirstc s� G dd� d�� G � fdd�d�}t d|� d�}||��dtdtf�fd d � }| �|d �d � | �|d �d� | �|d�d� dS )z� L{extractField} will extract a field used in the format string. @param flattenFirst: callable to flatten an event c @ s e Zd Zdefdd�ZdS )z=FlatFormattingTests.test_extractField.<locals>.ObjectWithReprr c S s dS )N�reprr r6 r r r r8 � s zFFlatFormattingTests.test_extractField.<locals>.ObjectWithRepr.__repr__N)r r r r; r8 r r r r �ObjectWithRepr� s rX c s$ e Zd Zd� fdd� Zddd�ZdS )z8FlatFormattingTests.test_extractField.<locals>.Somethingr Nc s d| _ � � | _d S )Nr )r$ r# r6 �rX r r �__init__� s zAFlatFormattingTests.test_extractField.<locals>.Something.__init__c S s t d��)Nz Just in case.)�NotImplementedErrorr6 r r r � __getstate__ s zEFlatFormattingTests.test_extractField.<locals>.Something.__getstate__r9 )r r r rZ r\ r rY r r � Something� s r] z%{something.number} {something.object})r! � something�fieldr c s t | � �S r )r )r_ )� flattenedr r �extract s z6FlatFormattingTests.test_extractField.<locals>.extractzsomething.numberr zsomething.number!s�7zsomething.object!srW N)r' r; r r+ )r, rV r] rK ra r )rX r` r �test_extractField� s �z%FlatFormattingTests.test_extractFieldc C s dt dt fdd�}| �|� dS )zi L{extractField} behaves identically if the event is explicitly flattened first. rK r c S s t | � | S r )r )rK r r r r` s zDFlatFormattingTests.test_extractFieldFlattenFirst.<locals>.flattenedN)r rc )r, r` r r r � test_extractFieldFlattenFirst s z1FlatFormattingTests.test_extractFieldFlattenFirstc C s( ddd�}t |� | �|ddd�� dS )zT L{flattenEvent} will do nothing to an event with no format string. �b� )�a�cN�r r+ �r, � inputEventr r r � test_flattenEventWithoutFormat s z2FlatFormattingTests.test_flattenEventWithoutFormatc C s, dddd�}t |� | �|dddd�� dS )zz L{flattenEvent} will do nothing to an event with a format string that contains no format fields. re rf zsimple message�rg rh r! Nri rj r r r � test_flattenEventWithInertFormat( � ��z4FlatFormattingTests.test_flattenEventWithInertFormatc C s, dddd�}t |� | �|dddd�� dS )zb L{flattenEvent} will do nothing to an event with log_format set to None. re rf Nrm ri rj r r r � test_flattenEventWithNoneFormat8 ro z3FlatFormattingTests.test_flattenEventWithNoneFormatr9 r )r r r r: r0 r3 r? rJ r r rP rR rS rU r rc rd rl rn rp r r r r r s2 . !>�� � � � � # r )r: r( � itertoolsr �typingr r r �timer � ImportError� twisted.trialr �_flattenr r r r �_formatr � _interfacesr �TestCaser r r r r �<module> s � PK ͪ�ZHf/� � ' __pycache__/test_levels.cpython-310.pycnu �[��� o �bc � @ s6 d Z ddlmZ ddlmZmZ G dd� dej�ZdS )z+ Test cases for L{twisted.logger._levels}. � )�unittest� )�InvalidLogLevelError�LogLevelc @ s$ e Zd ZdZddd�Zddd�ZdS ) � LogLevelTestsz Tests for L{LogLevel}. �returnNc C s& t �� D ] }| �t �|j�|� qdS )z, Look up log level by name. N)r � iterconstants�assertIs� levelWithName�name)�self�level� r �A/usr/lib/python3/dist-packages/twisted/logger/test/test_levels.py�test_levelWithName s �z LogLevelTests.test_levelWithNamec C sV d}zt �|� W n ty# } z| �|j|� W Y d}~dS d}~ww | �d� dS )z4 You can't make up log level names. z*bogus*Nz Expected InvalidLogLevelError.)r r r r r �fail)r �bogus�er r r �test_levelWithInvalidName s ��z'LogLevelTests.test_levelWithInvalidName)r N)�__name__� __module__� __qualname__�__doc__r r r r r r r s r N)r � twisted.trialr �_levelsr r �TestCaser r r r r �<module> s PK ͪ�Z��>��1 �1 ' __pycache__/test_filter.cpython-310.pycnu �[��� o �b�3 � @ s� d Z ddlmZmZmZmZmZ ddlmZ ddl m Z ddl m Z ddl mZ ddlmZ dd lmZmZmZmZ dd lmZmZ dd lmZmZ dd l m Z m Z G d d� dej �Z G dd� dej �Z!dS )z+ Test cases for L{twisted.logger._filter}. � )�Iterable�List�Tuple�Union�cast)� implementer)�BrokenMethodImplementation)� verifyObject)� NamedConstant)�unittest� )�FilteringLogObserver�ILogFilterPredicate�LogLevelFilterPredicate�PredicateResult)� ILogObserver�LogEvent)�InvalidLogLevelError�LogLevel)� LogPublisher�bitbucketLogObserverc @ s� e Zd ZdZd dd�Z d dee dedee e e e e e e f f fd d �Z d d d �Z d d d�Zd dd�Zd dd�Zd dd�Zd dd�Zd dd�Zd dd�Zd dd �ZdS ) �FilteringLogObserverTestsz, Tests for L{FilteringLogObserver}. �returnNc C sV t ttdd� �d�}ztt|� W dS ty* } z | �|� W Y d}~dS d}~ww )z@ L{FilteringLogObserver} is an L{ILogObserver}. c S � d S �N� ��er r �A/usr/lib/python3/dist-packages/twisted/logger/test/test_filter.py�<lambda>% � z:FilteringLogObserverTests.test_interface.<locals>.<lambda>r N)r r r r r �fail)�self�observerr r r r �test_interface! s ��z(FilteringLogObserverTests.test_interfaceF�filters�otherc s� t dd�t dd�t dd�t dd�g}G dd� d�� � fdd �|D �}g }g }tt|j�}|r5tt|j�}nt}t|||�} |D ]} | | � q?|rVd d � |D �d d � |D �fS d d � |D �S )a� Apply a set of pre-defined filters on a known set of events and return the filtered list of event numbers. The pre-defined events are four events with a C{count} attribute set to C{0}, C{1}, C{2}, and C{3}. @param filters: names of the filters to apply. Options are: - C{"twoMinus"} (count <=2), - C{"twoPlus"} (count >= 2), - C{"notTwo"} (count != 2), - C{"no"} (False). @param other: Whether to return a list of filtered events as well. @return: event numbers or 2-tuple of lists of event numbers. r ��count� r � c @ sz e Zd Zededefdd��Zededefdd��Zededefdd��Zededefd d ��Z ededefd d ��Z d S )z5FilteringLogObserverTests.filterWith.<locals>.Filters�eventr c S s | d dkr t jS t jS )z� count <= 2 @param event: an event @return: L{PredicateResult.yes} if C{event["count"] <= 2}, otherwise L{PredicateResult.maybe}. r( r �r �yes�maybe�r+ r r r �twoMinusG � z>FilteringLogObserverTests.filterWith.<locals>.Filters.twoMinusc S s | d dkr t jS t jS )z� count >= 2 @param event: an event @return: L{PredicateResult.yes} if C{event["count"] >= 2}, otherwise L{PredicateResult.maybe}. r( r r, r/ r r r �twoPlusU r1 z=FilteringLogObserverTests.filterWith.<locals>.Filters.twoPlusc S s | d dkr t jS t jS )z� count != 2 @param event: an event @return: L{PredicateResult.yes} if C{event["count"] != 2}, otherwise L{PredicateResult.maybe}. r( r )r �nor. r/ r r r �notTwoc r1 z<FilteringLogObserverTests.filterWith.<locals>.Filters.notTwoc S � t jS )z� No way, man. @param event: an event @return: L{PredicateResult.no} �r r3 r/ r r r r3 q s z8FilteringLogObserverTests.filterWith.<locals>.Filters.noc S s dS )z� Bogus result. @param event: an event @return: something other than a valid predicate result. Nr r/ r r r �bogus| s z;FilteringLogObserverTests.filterWith.<locals>.Filters.bogusN) �__name__� __module__� __qualname__� staticmethodr r r0 r2 r4 r3 r7 r r r r �FiltersF s r< c 3 s � | ]}t � |�V qd S r )�getattr)�.0�f�r< r r � <genexpr>� s � z7FilteringLogObserverTests.filterWith.<locals>.<genexpr>c S � g | ] }t t|d ��qS r' �r �int�r> r r r r � <listcomp>� � z8FilteringLogObserverTests.filterWith.<locals>.<listcomp>c S rB r' rC rE r r r rF � rG c S rB r' rC rE r r r rF � rG )�dictr r �appendr r ) r" r% r& �events� predicates� eventsSeen� eventsNotSeen�trackingObserver�negativeObserver�filteringObserverr r r@ r � filterWith+ s. �A � �z$FilteringLogObserverTests.filterWithc C s | � | �g �g d�� dS )z6 No filters: all events come through. �r r) r r* N�� assertEqualrQ �r" r r r � test_shouldLogEventNoFilters� s z6FilteringLogObserverTests.test_shouldLogEventNoFiltersc C � | � | �dg�g d�� dS )z8 Filter with negative predicate result. r4 �r r) r* NrS rU r r r �test_shouldLogEventNoFilter� � z5FilteringLogObserverTests.test_shouldLogEventNoFilterc C s$ | � | �dgd�g d�dgf� dS )zM Filtered results get sent to the other observer, if passed. r4 TrX r NrS rU r r r � test_shouldLogEventOtherObserver� s $z:FilteringLogObserverTests.test_shouldLogEventOtherObserverc C rW )z8 Filter with positive predicate result. r2 rR NrS rU r r r � test_shouldLogEventYesFilter� rZ z6FilteringLogObserverTests.test_shouldLogEventYesFilterc C s | � | �ddg�ddg� dS )zQ Series of filters with positive and negative predicate results. r2 r3 r r* NrS rU r r r � test_shouldLogEventYesNoFilter� s z8FilteringLogObserverTests.test_shouldLogEventYesNoFilterc C s | � | �g d��g d�� dS )zc Series of filters with positive, positive and negative predicate results. )r2 r0 r3 rR NrS rU r r r �!test_shouldLogEventYesYesNoFilter� s z;FilteringLogObserverTests.test_shouldLogEventYesYesNoFilterc C s | � t| jdg� dS )z7 Filter with invalid predicate result. r7 N)� assertRaises� TypeErrorrQ rU r r r �%test_shouldLogEventBadPredicateResult� s z?FilteringLogObserverTests.test_shouldLogEventBadPredicateResultc s` t t� d�� dtdtt f� fdd� }| �� |tj�� | �� |tj�� | � � |tj �� dS )zB Test filtering results from each predicate type. )�obj�resultr c s8 g �t tt�fdd��tt� fdd��f�}|�� �S )Nc s � � | �S r )rI r )�seenr r r � s zVFilteringLogObserverTests.test_call.<locals>.callWithPredicateResult.<locals>.<lambda>c s � S r r r )rc r r r � r )r r r r )rc r# r )rc rd r �callWithPredicateResult� s �zDFilteringLogObserverTests.test_call.<locals>.callWithPredicateResultN) rH �objectr r r �assertInr r- r. � assertNotInr3 )r" re r r r � test_call� s z#FilteringLogObserverTests.test_callc s� t g d�� ttdd� ��ttdd� �}tt�dtddf� ������fdd � �}|�t�ttd d� �f��t|ttd d� �f��t��|���� � dS ) zS Tracing keeps track of forwarding through the filtering observer. )� log_tracec S r r r r r r r r � r z6FilteringLogObserverTests.test_trace.<locals>.<lambda>c S r r r r r r r r � r r r Nc s8 �� | � � ��� d ��f��f��f��fg� d S )Nrj )�assertIsrT r �r+ �noFilter�oTest�oYes� publisherr" � yesFilterr r � testObserver� s ��z:FilteringLogObserverTests.test_trace.<locals>.testObserverc S r5 r )r r- r r r r r � � c S r5 r r6 r r r r r � rs )rH r r r r r r r )r" �oNorr r rl r � test_trace� s $ �� z$FilteringLogObserverTests.test_trace�r N)F)r8 r9 r: �__doc__r$ r �str�boolr r rD r rQ rV rY r[ r\ r] r^ ra ri ru r r r r r s( ��� � u r c @ sB e Zd ZdZddd�Zddd�Zddd �Zdd d �Zdd d �ZdS )� LogLevelFilterPredicateTestsz/ Tests for L{LogLevelFilterPredicate}. r Nc C sF t � }dttd�fD ]}| �|�|�|j� | �|�d�|j� q dS )z, Default log level is used. � Nzrocker.cool.namespace)r r rx rT �logLevelForNamespace�defaultLogLevel�r" � predicate�defaultr r r �test_defaultLogLevel s ���z1LogLevelFilterPredicateTests.test_defaultLogLevelc C s� t � }dttd�fD ]`}|�|tj� |�dtj� |�dtj� | �|� d�tj� | �|� ttd��tj� | �|� d�tj� | �|� d�tj� | �|� d�tj� | �|� d�tj� | �|� d�tj� q dS )z4 Setting and retrieving log levels. r{ N� twext.web2�twext.web2.dav�twisted�twext.web2.dav.test�twext.web2.dav.test1.test2) r r rx �setLogLevelForNamespacer �error�debug�warnrT r| r~ r r r �test_setLogLevel s0 � � � ���z-LogLevelFilterPredicateTests.test_setLogLevelc C s0 t � }| �t|jdt� � | �t|jdd� dS )zP Can't pass invalid log levels to C{setLogLevelForNamespace()}. r� r� N)r r_ r r� rf �r" r r r r �test_setInvalidLogLevel: s ��z4LogLevelFilterPredicateTests.test_setInvalidLogLevelc C s� t � }|�dtj� |�dtj� |�� | �|�d�|j� | �|�d�|j� | �|�d�|j� | �|�d�|j� | �|�d�|j� dS )z& Clearing log levels. r� r� r� r� r� N) r r� r r� r� �clearLogLevelsrT r| r} r� r r r �test_clearLogLevelsO s* � � ���z0LogLevelFilterPredicateTests.test_clearLogLevelsc s t � � � �dtj� � �dtj� � �dtj� dtdtdtddf� �fd d � }|dtjtj � |t td�tjtj � |dtjtj � |t td�tjtj � |dtjtj � |dtjtj � |dtjtj � |dtjtj � |dtj tj � |t td�tj tj � |ddtj � dS ) zC Events are filtered based on log level/namespace. r{ r� r� � namespace�level�expectedResultr Nc s t | |d�}��|� |�� d S )N)� log_namespace� log_level)rH rT )r� r� r� r+ �r r" r r �checkPredicatev s zCLogLevelFilterPredicateTests.test_filtering.<locals>.checkPredicate) r r� r r� r� r� rx r r r3 r r. �critical)r" r� r r� r �test_filteringl s0 ����z+LogLevelFilterPredicateTests.test_filteringrv ) r8 r9 r: rw r� r� r� r� r� r r r r rz s ! rz N)"rw �typingr r r r r �zope.interfacer �zope.interface.exceptionsr �zope.interface.verifyr � constantlyr � twisted.trialr �_filterr r r r � _interfacesr r �_levelsr r � _observerr r �TestCaser rz r r r r �<module> s gPK ͪ�Z��ک� � ( __pycache__/test_capture.cpython-310.pycnu �[��� o �b@ � @ s@ d Z ddlmZmZ ddlmZ ddlmZ G dd� de�ZdS ) z, Test cases for L{twisted.logger._capture}. � )�Logger�LogLevel)�TestCase� )� capturedLogsc @ s e Zd ZdZe� Zddd�ZdS )�LogCaptureTestsz' Tests for L{LogCaptureTests}. �returnNc C s� t � }t� �}| jjd|d� | jjd|d� W d � n1 s!w Y | �t|�dk� | �|d d d� | �|d d tj� | �|d d |� | �|d d d� | �|d d tj� | �|d d |� dS ) z< Events logged within context are captured. zCapture this, please)�foozCapture this too, pleaseNr r � log_format� log_levelr � ) �objectr �log�debug�info� assertTrue�len� assertEqualr )�selfr �captured� r �B/usr/lib/python3/dist-packages/twisted/logger/test/test_capture.py� test_capture s �z LogCaptureTests.test_capture)r N)�__name__� __module__� __qualname__�__doc__r r r r r r r r s r N) r �twisted.loggerr r �twisted.trial.unittestr �_capturer r r r r r �<module> s PK ͪ�Z��{�$ �$ ' __pycache__/test_logger.cpython-310.pycnu �[��� o �b � @ s� d Z ddlmZmZmZmZ ddlmZ ddlm Z ddl m Z ddl m Z ddlmZ dd lmZmZ dd lmZmZ dd lmZ G d d � d e�ZG dd� d�ZG dd� de j�ZdS )z+ Test cases for L{twisted.logger._logger}. � )�List�Optional�Type�cast)� implementer)� NamedConstant)�unittest� )� formatEvent)�globalLogPublisher)� ILogObserver�LogEvent)�InvalidLogLevelError�LogLevel��Loggerc @ s2 e Zd ZdZ d dedee deddfdd�ZdS ) � TestLoggerz^ L{Logger} with an overridden C{emit} method that keeps track of received events. N�level�format�kwargs�returnc sl t t�dtdd f� fdd� �}t�|� ztj� ||fi |�� W t�|� nt�|� w |||d�� _d S )N�eventr c s | � _ d S �N�r r ��self� �A/usr/lib/python3/dist-packages/twisted/logger/test/test_logger.py�observer s z!TestLogger.emit.<locals>.observer)r r r ) r r r r � addObserverr �emit�removeObserver�emitted)r r r r r r r r r s �zTestLogger.emitr ) �__name__� __module__� __qualname__�__doc__r r �str�objectr r r r r r s �����r c @ s< e Zd ZdZe� Zd dee ddfdd�Zdefdd�Z dS ) �LogComposedObjectz3 A regular object, with a logger attached. N�stater c C s || _ d S r �r* )r r* r r r �__init__7 s zLogComposedObject.__init__c C s d| j � d�S )Nz<LogComposedObject �>r+ r r r r �__str__: s zLogComposedObject.__str__r ) r# r$ r% r& r �logr r' r, r. r r r r r) 0 s r) c @ s� e Zd ZdZd dd�Zd dd�Zd dd �Zd d d �Zd d d �Zd dd�Z d dd�Z d dd�Z d dd�Z d dd�Z d dd�Zd dd�Zd d d �Zd d d �ZdS )!� LoggerTestsz Tests for L{Logger}. r Nc C s, d}t |�}| �t|�dt|�� d�� dS )z" repr() on Logger �blearghz<Logger r- N)r � assertEqual�repr)r � namespacer/ r r r � test_reprC s zLoggerTests.test_reprc C s t � }| �|jt� dS )z3 Default namespace is module name. N)r r2 r4 r# �r r/ r r r �test_namespaceDefaultK s z!LoggerTests.test_namespaceDefaultc C s. g }t dttd�t� � | �|d jd� dS )z� Default namespace is C{"<unknown>"} when a logger is created from a context in which is can't be determined automatically and no namespace was specified. zresult.append(Logger())r r z <unknown>N)�exec�dictr �localsr2 r4 )r �resultr r r �test_namespaceOMGItsTooHardR s �z'LoggerTests.test_namespaceOMGItsTooHardc C s� t � }d�|j|jj�}| �tt|j�j |� | �tt t t j�j |� | � tt t t j�j t � | � tt|j�j |� | � t� j � dS )z� Default namespace for classes using L{Logger} as a descriptor is the class name they were retrieved from. z{}.{}N)r) r r$ � __class__r# r2 r r r/ r4 r �assertIs�source� assertIsNoner )r �obj�expectedNamespacer r r �test_namespaceAttribute` s ���z#LoggerTests.test_namespaceAttributec sJ g � G � fdd�d�}|j �d� | �t� �d� | �� d d d� dS )zH When used as a descriptor, the observer is propagated. c s e Zd Zeee� j�d�ZdS )z5LoggerTests.test_descriptorObserver.<locals>.MyObject�r N)r# r$ r% r r r �appendr/ r ��observedr r �MyObject| s rH �hello� r � log_formatN)r/ �infor2 �len)r rH r rF r �test_descriptorObserverv s z#LoggerTests.test_descriptorObserverc C sX t d�}tt|j�}|�d� | �d|j� | �|jd |� t|j�}| �d|� dS )z� On instances that have a L{Logger} class attribute, the C{log_source} key is available to format strings. rI zHello, {log_source}.� log_sourcez!Hello, <LogComposedObject hello>.N) r) r r r/ �error�assertInr r2 r )r rA r/ �stuffr r r �!test_sourceAvailableForFormatting� s z-LoggerTests.test_sourceAvailableForFormattingc C s� t � }t�� D ]t}d}|j|jd�}t||j�}||||jd� | �|jd |� | �|jd |� | �|jd d |� | �t |d�d � | �|j d |� | �|j d |� | �|j d t � | � |j d � | �|j d |� | �t |j �|� qdS )z] Test that log levels and messages are emitted correctly for Logger. z This is a {level_name} message)� level_name)�junkrT r r r rU r zNo event observed.rK � log_level� log_namespacerO N)r r � iterconstantsr �name�getattrr2 r" � assertTrue�hasattrr r# r@ r )r r/ r r �message� logMethodr r r �test_basicLogger� s" �z LoggerTests.test_basicLoggerc sF t t�dtddf� �fdd� ��G �fdd�d�� tt� j��� dS )z> C{log_source} event key refers to the class. r r Nc s �� | d � � d S �NrO �r2 r )�Thingor r r r � � z0LoggerTests.test_sourceOnClass.<locals>.observerc � e Zd Ze� d�ZdS )z.LoggerTests.test_sourceOnClass.<locals>.ThingorD N�r# r$ r% r r/ r rD r r rb � � rb �r r r r r r/ rL r r )rb r r r �test_sourceOnClass� s z LoggerTests.test_sourceOnClassc sL t t�dtddf��fdd� �� G � fdd�d�}|� �tt�j��� dS )zA C{log_source} event key refers to the instance. r r Nc s � � | d �� d S r` ra r )r �thingor r r � rc z3LoggerTests.test_sourceOnInstance.<locals>.observerc rd )z1LoggerTests.test_sourceOnInstance.<locals>.ThingorD Nre r rD r r rb � rf rb rg )r rb r )r r ri r �test_sourceOnInstance� s z!LoggerTests.test_sourceOnInstancec s4 t t�dtddf� fdd� �}t|d�}|�� dS )z5 C{log_source} event key is L{None}. r r Nc s � � | d � d S r` )r@ r r r r r � s z0LoggerTests.test_sourceUnbound.<locals>.observerrD �r r r r rL )r r r/ r r r �test_sourceUnbound� s z LoggerTests.test_sourceUnboundc C sp t � }ztd�� ty |�d� Y nw | �t�}| �t|�d� | �|jd tj� | �|jd d� dS )z? Test that log.failure() emits the right data. zbaloney!�WhoopsrJ r r N) r � RuntimeError�failure�flushLoggedErrorsr2 rM r" r �critical�r r/ �errorsr r r �test_defaultFailure� s � z LoggerTests.test_defaultFailurec C sj t � }|jddtjddd� | �|jd d� | �|jd tj� | �|jd |j� | �|jd � d S ) zQ Make sure that kwargs conflicting with args don't pass through. �*�#z *namespace*z*source*)rK rV rW rO rK rV rW rO N)r �warnr rP r2 r r4 r@ r6 r r r �test_conflictingKwargs� s �z"LoggerTests.test_conflictingKwargsc C s. t � }|�d� | �t�}| �t|�d� dS )zA Test passing in a bogus log level to C{emit()}. z*bogus*rJ N)r r rp r r2 rM rr r r r �test_logInvalidLogLevel� s z#LoggerTests.test_logInvalidLogLevelc s\ t t�dtddf�fdd� ��t t�dtddf� ��fdd� ��t�d�� � jd g d � dS ) zE Tracing keeps track of forwarding to the publisher. r r Nc s � | � d S r r r rD r r � publisher s z)LoggerTests.test_trace.<locals>.publisherc s �� | d � �fg� d S )N� log_tracera r )r/ rz r r r r s z(LoggerTests.test_trace.<locals>.observerrD zHello.)r{ rk r r )r/ r rz r r � test_trace s zLoggerTests.test_trace)r N)r# r$ r% r& r5 r7 r< rC rN rS r_ rh rj rl rt rx ry r| r r r r r0 > s r0 N) r& �typingr r r r �zope.interfacer � constantlyr � twisted.trialr �_formatr �_globalr � _interfacesr r �_levelsr r �_loggerr r r) �TestCaser0 r r r r �<module> s PK ͪ�ZrIxFwS wS ' __pycache__/test_format.cpython-310.pycnu �[��� o �bU � @ s0 d Z ddlmZmZmZ z ddlmZ e W n ey! dZY nw ddlm Z ddl m Z m Z m Z ddlmZ ddlmZ d d lmZmZmZmZmZmZ d d lmZ d d lm Z G d d� dej �Z G dd� dej �Z G dd� dej �Z G dd� dej �Z!G dd� d�Z"G dd� de#�Z$G dd� dej �Z%dS )z+ Test cases for L{twisted.logger._format}. � )�AnyStr�Optional�cast)�tzsetN)�Failure)� addTZCleanup�mktime�setTZ)�unittest)�SkipTest� )� eventAsText� formatEvent�formatEventAsClassicLogText� formatTime�formatUnformattableEvent�formatWithCall)�LogEvent)�LogLevelc @ sV e Zd ZdZddd�Zddd�Zddd �Zdd d �Zdd d �Zddd�Z ddd�Z dS )�FormattingTestsz5 Tests for basic event formatting functions. �returnNc s� dt dtdtf� fdd� }� �d|d�� � �d|d�� � �d|d dd �� � �d |d d dd� d�� � �d|d�� � �d|d�� |ddd�}� �d|� ttd��}� �d| d |ddd�� d S ) ac L{formatEvent} will format an event according to several rules: - A string with no formatting instructions will be passed straight through. - PEP 3101 strings will be formatted using the keys and values of the event as named fields. - PEP 3101 keys ending with C{()} will be treated as instructions to call that key (which ought to be a callable) before formatting. L{formatEvent} will always return L{str}, and if given bytes, will always treat its format string as UTF-8 encoded. � logFormat�eventr c s$ | |d< t |�}� �t|�t� |S )N� log_format)r �assertIs�type�str)r r �result��self� �A/usr/lib/python3/dist-packages/twisted/logger/test/test_format.py�format9 s z0FormattingTests.test_formatEvent.<locals>.format� � �abcz{x})�xzno, yes.z{not_called}, {called()}.�noc S � dS )N�yesr r r r r! �<lambda>D � z2FormattingTests.test_formatEvent.<locals>.<lambda>)� not_called�calledu Sánchezs Sánchez�Unable to format events S�nchezs S{a!s}nchez� �)�az Sb'\xe1'nchez�S�nchezs S{a!r}nchezN)r �objectr � assertEqual�assertIn�repr)r r" � maybeResult�xe1r r r! �test_formatEvent' s � z FormattingTests.test_formatEventc C s$ t ddd�}t|�}| �d|� dS )z5 Formatting an event with no format. � r )�foo�barr# N)�dictr r4 �r r r r r r! �test_formatEventNoFormatN s z(FormattingTests.test_formatEventNoFormatc C s8 t t� ddd�}t|�}| �d|� | �t|�|� dS )z: Formatting an event with a bogus format. r: r )r r; r< zLog format must be strN)r= r3 r r5 r6 r> r r r! �test_formatEventWeirdFormatW � z+FormattingTests.test_formatEventWeirdFormatc C s8 t ddd� d�}t|�}| �d|� | �t|�|� dS )zF Formatting an event that's just plain out to get us. �{evil()}c S � dd S �Nr: r r r r r r! r* e � z?FormattingTests.test_formatUnformattableEvent.<locals>.<lambda>�r �evilr. N)r= r r5 r6 r> r r r! � test_formatUnformattableEventa rA z-FormattingTests.test_formatUnformattableEventc C sL ddddd� t tt� �di}t|�}| �d|� | �d|� | �d |� d S ) zR Formatting an unformattable event that has an unformattable key. r rB rG c S rC rD r r r r r! r* q rE zSFormattingTests.test_formatUnformattableEventWithUnformattableKey.<locals>.<lambda>�gurk�*MESSAGE LOST: unformattable object logged:�Recoverable data:� Exception during formatting:N)r r � Unformattabler r5 r> r r r! �1test_formatUnformattableEventWithUnformattableKeyk s � zAFormattingTests.test_formatUnformattableEventWithUnformattableKeyc C sD t ddd� t� d�}t|�}| �d|� | �d|� | �d|� dS ) �T Formatting an unformattable event that has an unformattable value. rB c S rC rD r r r r r! r* rE zUFormattingTests.test_formatUnformattableEventWithUnformattableValue.<locals>.<lambda>)r rG rI rJ rK rL N)r= rM r r5 r> r r r! �3test_formatUnformattableEventWithUnformattableValuey s � zCFormattingTests.test_formatUnformattableEventWithUnformattableValuec C sP t ddd� dd�}t|ttt� ��}| �d|� | �td�d td� |� d S ) rO rB c S rC rD r r r r r! r* � rE zbFormattingTests.test_formatUnformattableEventWithUnformattableErrorOMGWillItStop.<locals>.<lambda>�okay)r rG � recoverablerJ rR z = N)r= r r � BaseExceptionrM r5 r6 r> r r r! �@test_formatUnformattableEventWithUnformattableErrorOMGWillItStop� s � zPFormattingTests.test_formatUnformattableEventWithUnformattableErrorOMGWillItStop�r N) �__name__� __module__� __qualname__�__doc__r9 r? r@ rH rN rP rT r r r r! r " s ' r c @ sL e Zd ZdZddd�Zddd�Zddd �Zdd d �Zdd d �Zddd�Z dS )�TimeFormattingTestsz. Tests for time formatting functions. r Nc C s t | � d S )N)r r r r r! �setUp� � zTimeFormattingTests.setUpc sb t du rtd��dtdtdtddf� fdd� }|d d d � |d d d� |ddd� |ddd� dS )z� Default time stamp format is RFC 3339 and offset respects the timezone as set by the standard C{TZ} environment variable and L{tzset} API. N�:Platform cannot change timezone; unable to verify offsets.�name� expectedDST� expectedSTDr c s< t | � td�}td�}� �t|�|� � �t|�|� d S )N) i� � � r r r � � r: ) i� r: � r r r r re r )r r r4 r )r^ r_ r` �localDST�localSTDr r r! �testForTimeZone� s zMTimeFormattingTests.test_formatTimeWithDefaultFormat.<locals>.testForTimeZone�UTC+00z2006-06-30T00:00:00+0000z2007-01-31T00:00:00+0000zEST+05EDT,M4.1.0,M10.5.0z2006-06-30T00:00:00-0400z2007-01-31T00:00:00-0500zCEST-01CEDT,M4.1.0,M10.5.0z2006-06-30T00:00:00+0200z2007-01-31T00:00:00+0100zCST+06z2006-06-30T00:00:00-0600z2007-01-31T00:00:00-0600)r r r )r rh r r r! � test_formatTimeWithDefaultFormat� s. ����z4TimeFormattingTests.test_formatTimeWithDefaultFormatc C s( | � td�d� | � tddd�d� dS )zL If C{when} argument is L{None}, we get the default output. N�-�!)�default�r4 r r r r r! �test_formatTimeWithNoTime� s z-TimeFormattingTests.test_formatTimeWithNoTimec C s6 t d�}| �t|dd�d� | �t|ddd�d� dS )zR If C{timeFormat} argument is L{None}, we get the default output. � i� � � � �( �/ r: i r: N�� timeFormatrk rl )rw rm �r r4 r �r �tr r r! �test_formatTimeWithNoFormat� s z/TimeFormattingTests.test_formatTimeWithNoFormatc C s t d�}| �t|dd�d� dS )z2 Alternate time format in output. rp z%Y/%Wrv z2013/38Nrx ry r r r! �&test_formatTimeWithAlternateTimeFormat� s z:TimeFormattingTests.test_formatTimeWithAlternateTimeFormatc C s | � tddd�d� dS )z0 "%f" supported in time format. g�?x��.Az%frv �234560Nrn r r r r! �test_formatTimePercentF� s z+TimeFormattingTests.test_formatTimePercentFrU ) rV rW rX rY r[ rj ro r{ r| r~ r r r r! rZ � s - rZ c @ s~ e Zd ZdZddd�Zddd�Zddd �Zdd d �Zdd d �Zddd�Z ddd�Z ddd�Z ddd�Z ddd�Z ddd�ZdS )�ClassicLogFormattingTestsz@ Tests for classic text log event formatting functions. r Nc C sH t du rtd��t| � td� td�}td|d�}| �t|�d� dS )z� Time is first field. Default time stamp format is RFC 3339 and offset respects the timezone as set by the standard C{TZ} environment variable and L{tzset} API. Nr] ri rp �XYZZY�r �log_timez%2013-09-24T11:40:47+0000 [-#-] XYZZY )r r r r r r= r4 r )r rz r r r r! �test_formatTimeDefault� s �z0ClassicLogFormattingTests.test_formatTimeDefaultc C s: dt t dtfdd�}tddd�}| �t||d�d � d S ) zc Time is first field. Custom formatting function is an optional argument. rz r c S s d| � d�S )N�__r )rz r r r! r r\ zCClassicLogFormattingTests.test_formatTimeCustom.<locals>.formatTimer� i90 r� )r z__12345__ [-#-] XYZZY N)r �floatr r= r4 r )r r r r r r! �test_formatTimeCustom s �z/ClassicLogFormattingTests.test_formatTimeCustomc C � t ddd�}| �t|�d� dS )z: Namespace is first part of second field. r� � my.namespace)r � log_namespacez- [my.namespace#-] XYZZY N�r= r4 r �r r r r r! �test_formatNamespace � �z.ClassicLogFormattingTests.test_formatNamespacec C s" t dtjd�}| �t|�d� dS )z7 Level is second part of second field. r� )r � log_levelz- [-#warn] XYZZY N�r= r �warnr4 r r� r r r! �test_formatLevel � �z*ClassicLogFormattingTests.test_formatLevelc C r� )z) System is second field. r� � S.Y.S.T.E.M.�r � log_system�- [S.Y.S.T.E.M.] XYZZY Nr� r� r r r! �test_formatSystem# r� z+ClassicLogFormattingTests.test_formatSystemc C s& t ddtjdd�}| �t|�d� dS )�B System is not supplanted by namespace and level. r� r� r� )r r� r� r� r� Nr� r� r r r! �test_formatSystemRulz- s ��z/ClassicLogFormattingTests.test_formatSystemRulzc C s" t dt� d�}| �t|�d� dS )r� r� r� z- [UNFORMATTABLE] XYZZY N)r= rM r4 r r� r r r! � test_formatSystemUnformattable< r� z8ClassicLogFormattingTests.test_formatSystemUnformattablec C r� )z0 Formatted event is last field. zid:{id}�123�r �idz- [-#-] id:123 Nr� r� r r r! �test_formatFormatF r� z+ClassicLogFormattingTests.test_formatFormatc C s t dd�}| �t|�d� dS )z# No format string. r� )r� N�r= r r r� r r r! �test_formatNoFormatP s z-ClassicLogFormattingTests.test_formatNoFormatc C s t ddd�}| �t|�d� dS )z& Empty format string. r# r� r� Nr� r� r r r! �test_formatEmptyFormatW s z0ClassicLogFormattingTests.test_formatEmptyFormatc C s t dd�}| �t|�d� dS )zO If the formatted event has newlines, indent additional lines. z"XYZZY A hollow voice says: "Plugh")r z-- [-#-] XYZZY A hollow voice says: "Plugh" Nr� r� r r r! �test_formatFormatMultiLine^ s �z4ClassicLogFormattingTests.test_formatFormatMultiLinerU )rV rW rX rY r� r� r� r� r� r� r� r� r� r� r� r r r r! r � s r c @ s e Zd ZdZddd�ZdS )�FormatFieldTestsz+ Tests for format field functions. r Nc C s> | � tdtddd� d��d� | � tdtdd� d ��d � d S ) z� L{formatWithCall} is an extended version of L{str.format} that will interpret a set of parentheses "C{()}" at the end of a format key to mean that the format key ought to be I{called} rather than stringified. zHello, {world}. {callme()}.�earthc S r( )N�mayber r r r r! r* x r+ z6FormatFieldTests.test_formatWithCall.<locals>.<lambda>)�world�callmezHello, earth. maybe.zHello, {repr()!r}.c S r( )Nr6 r r r r r! r* } r+ )r6 zHello, 'repr'.N)r4 r r= r r r r! �test_formatWithCalln s ���z$FormatFieldTests.test_formatWithCallrU )rV rW rX rY r� r r r r! r� i s r� c @ s e Zd ZdZdefdd�ZdS )rM z> An object that raises an exception from C{__repr__}. r c C s t dd �S rD )r r r r r! �__repr__� r\ zUnformattable.__repr__N)rV rW rX rY r r� r r r r! rM � s rM c @ s e Zd ZdZdS )� CapturedErrorz3 A captured error for use in format tests. N)rV rW rX rY r r r r! r� � s r� c @ s� e Zd ZdZd dd�Zd dd�Zd dd �Zd d d �Zd d d �Zd dd�Z d dd�Z d dd�Z d dd�Z d dd�Z d dd�Zd dd�Zd d d �ZdS ) �EventAsTextTestszi Tests for L{eventAsText}, all of which ensure that the returned type is UTF-8 decoded text. r Nc C sd zt d�� t y t� }Y nw ddi}||d< t|ddd�}| �t|�� �|� | �d|� dS ) zT An event with a C{log_failure} key will have a traceback appended. �This is a fake errorr �This is a test log message� log_failureTF��includeTimestamp� includeSystemN�r� r r r5 r � getTraceback�r �fr � eventTextr r r! �test_eventWithTraceback� s �z(EventAsTextTests.test_eventWithTracebackc C sd zt d�� t y t� }Y nw ddi}||d< t|ddd�}| �t|�� �|� | �d|� dS ) zu An event with an empty C{log_format} key appends a traceback from the accompanying failure. r� r r# r� TFr� Nr� r� r r r! �"test_formatEmptyEventWithTraceback� s �z3EventAsTextTests.test_formatEmptyEventWithTracebackc C sv zt d�� t y t� }Y nw ddd� d�}||d< t|ddd �}| �|t� | �t|�� �|� | �d|� d S ) zw An event with an unformattable value in the C{log_format} key still has a traceback appended. r� rB c S rC rD r r r r r! r* � rE zHEventAsTextTests.test_formatUnformattableWithTraceback.<locals>.<lambda>rF r� TFr� N)r� r r �assertIsInstancer r5 r� r� r r r! �%test_formatUnformattableWithTraceback� s �� z6EventAsTextTests.test_formatUnformattableWithTracebackc C s� zt d�� t y t� }Y nw ddddd� ttt� �di}||d< t|d d d �}| �|t� | �d |� | �t|�� �|� | �d|� d S )z� An event with an unformattable value in the C{log_format} key, that throws an exception when __repr__ is invoked still has a traceback appended. r� r rB rG c S rC rD r r r r r! r* � rE zMEventAsTextTests.test_formatUnformattableErrorWithTraceback.<locals>.<lambda>rI r� TFr� z MESSAGE LOSTN) r� r r r rM r r� r5 r� r� r r r! �*test_formatUnformattableErrorWithTraceback� s � � z;EventAsTextTests.test_formatUnformattableErrorWithTracebackc C s< ddi}t � |d< t|ddd�}| �|t� | �d|� dS ) zk If a traceback cannot be appended, a message indicating this is true is appended. r r# r� TFr� z'(UNABLE TO OBTAIN TRACEBACK FROM EVENT)N)r3 r r� r r5 �r r r� r r r! �&test_formatEventUnformattableTraceback� s z7EventAsTextTests.test_formatEventUnformattableTracebackc C s2 ddi}t |ddd�}| �|t� | �d|� dS )zY An event with no C{log_failure} key will not have a traceback appended. r r� TFr� N)r r� r r5 r� r r r! �test_formatEventNonCritical� s z,EventAsTextTests.test_formatEventNonCriticalc C s\ zt d�� t y t� }Y nw ddi}||d< t|ddd�}| �d|� | �d|� d S ) zU An exception message with multibyte characters is properly handled. u €r r� r� TFr� � TracebackN�r� r r r5 r� r r r! � test_formatTracebackMultibyte� s � z.EventAsTextTests.test_formatTracebackMultibytec C s\ zt d�� t y t� }Y nw ddi}||d< t|ddd�}| �d|� | �d |� d S ) zj An error raised attempting to decode the UTF still produces a valid log message. s ��t e s t r r� r� TFr� r� z.CapturedError(b"\xff\xfet\x00e\x00s\x00t\x00")Nr� r� r r r! �,test_formatTracebackHandlesUTF8DecodeFailure s � z=EventAsTextTests.test_formatTracebackHandlesUTF8DecodeFailurec C s^ zt d�� t y t� }Y nw td�}dd|d�}||d< t|dddd �}| �|d � d S ) zq If includeSystem is specified as the only option no timestamp or traceback are printed. r� rp �ABCD� fake_system�r r� r� r� FT�r� �includeTracebackr� z[fake_system] ABCDN�r� r r r r4 �r r� rz r r� r r r! �test_eventAsTextSystemOnly s* ����z+EventAsTextTests.test_eventAsTextSystemOnlyc C s~ t du rtd��t| � td� ztd�� ty t� }Y nw td�}dd|d�}||d < t|d d d d �}| �|d � dS )zq If includeTimestamp is specified as the only option no system or traceback are printed. Nr] ri r� rp r� r� r� r� TFr� z 2013-09-24T11:40:47+0000 ABCD) r r r r r� r r r r4 r� r r r! � test_eventAsTextTimestampOnly0 s2 ����z.EventAsTextTests.test_eventAsTextTimestampOnlyc C s\ zt d�� t y t� }Y nw td�}d|d�}||d< t|dddd�}| �|d � d S ) z\ If includeSystem is specified with a missing system [-#-] is used. r� rp r� r� r� FTr� z [-#-] ABCDNr� r� r r r! � test_eventAsTextSystemMissingR s( ����z.EventAsTextTests.test_eventAsTextSystemMissingc C sb zt d�� t y t� }Y nw td�}d|tjdd�}||d< t|dddd �}| �|d � d S ) z� If includeSystem is specified with a missing system but namespace and level are present they are used. r� rp r� �test)r r� r� r� r� FTr� z[test#info] ABCDN�r� r r r �infor r4 r� r r r! �.test_eventAsTextSystemMissingNamespaceAndLevelm s, ����z?EventAsTextTests.test_eventAsTextSystemMissingNamespaceAndLevelc C s` zt d�� t y t� }Y nw td�}d|tjd�}||d< t|dddd�}| �|d � d S ) zv If includeSystem is specified with a missing system but level is present, level is included. r� rp r� )r r� r� r� FTr� z [-#info] ABCDNr� r� r r r! �&test_eventAsTextSystemMissingLevelOnly� s* ����z7EventAsTextTests.test_eventAsTextSystemMissingLevelOnlyrU )rV rW rX rY r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! r� � s " r� )&rY �typingr r r �timer � ImportError�twisted.python.failurer �!twisted.python.test.test_tzhelperr r r � twisted.trialr �twisted.trial.unittestr �_formatr r r r r r � _interfacesr �_levelsr �TestCaser rZ r r� rM � Exceptionr� r� r r r r! �<module> s. � tR PK ͪ�Z�1. # # ' __pycache__/test_stdlib.cpython-310.pycnu �[��� o �b�! � @ s& d Z ddlZddlZddlmZ ddlmZmZ ddlm Z m Z m Z m Z ddl mZmZmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ d d lm Z m Z d dl m Z d dl m!Z! deee" e#f fdd�Z$G dd� d�Z%G dd� dej&�Z'dee ef fdd�Z(G dd� dej)�Z*dS )z+ Test cases for L{twisted.logger._format}. � N)� getsourcefile)�BytesIO� TextIOWrapper)� Formatter� LogRecord� StreamHandler� getLogger)�List�Optional�Tuple)�BrokenMethodImplementation)� verifyObject)� currentframe)�Failure)�unittest� )� ILogObserver�LogEvent)�LogLevel)�STDLibLogObserver�returnc C s& t d�} ttj| jd �| jd fS )z� Retrive the file name and line number immediately after where this function is called. @return: the file name and line number � �__name__)r r �sys�modules� f_globals�f_lineno)�caller� r �A/usr/lib/python3/dist-packages/twisted/logger/test/test_stdlib.py�nextLine s �r c @ s2 e Zd ZdZd dd�Zd dd�Zdefdd �ZdS ) �StdlibLoggingContainerzF Continer for a test configuration of stdlib logging objects. r Nc C sZ t d�| _| j�� | _| j�tj� t� | _| j� | j� t � \| _ | _ | j� | j � d S )N� ) r � rootLogger�getEffectiveLevel� originalLevel�setLevel� py_logging�DEBUG�BufferedHandler�bufferedHandler� addHandler�handlerAndBytesIO� streamHandler�output��selfr r r �__init__- s z StdlibLoggingContainer.__init__c C sB | j �| j� | j �| j� | j �| j� | j�� | j�� dS )z# Close the logger. N)r# r& r% � removeHandlerr* r- �closer. r/ r r r r3 9 s z StdlibLoggingContainer.closec C s | j �� �d�S )zd Get the output to the underlying stream as text. @return: the output text �utf-8)r. �getvalue�decoder/ r r r � outputAsTextC s z#StdlibLoggingContainer.outputAsText�r N)r � __module__� __qualname__�__doc__r1 r3 �strr7 r r r r r! ( s r! c @ s� e Zd ZdZddd�Zdefdd�Zdedee e e f fd d �Z dd d �Z dd d�Zddd�Zddd�Zddd�Zddd�Zddd�Zddd�ZdS ) �STDLibLogObserverTestsz) Tests for L{STDLibLogObserver}. r Nc C sH t � }ztt|� W dS ty# } z | �|� W Y d}~dS d}~ww )z= L{STDLibLogObserver} is an L{ILogObserver}. N)r r r r �fail)r0 �observer�er r r �test_interfaceQ s ��z%STDLibLogObserverTests.test_interfacec C s t � }| �|j� |S )zj Create a logging object we can use to test with. @return: a stdlib-style logger )r! � addCleanupr3 )r0 �loggerr r r � py_logger[ s z STDLibLogObserverTests.py_logger�eventsc G s: | � � }ttjd d�}|D ]}||� q|jj|�� fS )z� Send one or more events to Python's logging module, and capture the emitted L{LogRecord}s and output stream as a string. @param events: events @return: a tuple: (records, output) r )� stackDepth)rD r �defaultStackDepthr* �recordsr7 )r0 rE �plr? �eventr r r �logEvente s �� z STDLibLogObserverTests.logEventc C s4 | � i �\}}| �t|�d� | �|d jd� dS )z Logger name. r r �twistedN)rK � assertEqual�len�name�r0 rH r. r r r � test_name{ s z STDLibLogObserverTests.test_namec C s� dt jtjt jtjt jtjt jtjt j tj t j i}g }|� � D ]\}}i }|dur-||d< t |�|d< |�|� q | j|� \}}| �t|�t|�� tt|��D ]}| �|| j|| d � qPdS )z Log levels. N� log_level� py_levelno)r' �INFOr �debugr( �info�warn�WARNING�error�ERROR�critical�CRITICAL�items�int�appendrK rM rN �range�levelno) r0 � levelMappingrE �level�pyLevelrJ rH r. �ir r r � test_levels� s& � �z"STDLibLogObserverTests.test_levelsc C s` t � \}}| �i �\}}| �t|�d� | �|d j|� | �|d j|� | �|d j� dS )zb C{pathname}, C{lineno}, C{exc_info}, C{func} is set properly on records. r r N)r rK rM rN �pathname�lineno� assertIsNone�exc_info)r0 �filename�logLinerH r. r r r �test_callerInfo� s z&STDLibLogObserverTests.test_callerInfoc C sV t ddd�}| �|�\}}| �t|�d� | �t|d j�d� | �|d jd� dS ) zL Basic formattable event passes the format along correctly. � Hello, {who}!�dude�� log_format�whor r z Hello, dude!r N)�dictrK rM rN r< �msg�args�r0 rJ rH r. r r r �test_basicFormat� s z'STDLibLogObserverTests.test_basicFormatc C sD t ddd�}| �|�\}}| �t|�d� | �|�d�t|�� dS )z< Basic formattable event renders correctly. rn ro rp r z:Hello, dude! N)rs rK rM rN � assertTrue�endswith�reprrv r r r �test_basicFormatRendered� s z/STDLibLogObserverTests.test_basicFormatRenderedc C s8 | � i �\}}| �t|�d� | �t|d j�d� dS )z' Event with no format. r r r" N)rK rM rN r< rt rP r r r � test_noFormat� s z$STDLibLogObserverTests.test_noFormatc C s� d dd�}z|� W n t y t� }Y nw tdd|d�}| �|�\}}| �t|�d� | �d|� | �d |� | �d |� dS ) zK An event with a failure logs the failure details as well. r Nc S � dd d S �Nr r r r r r r � failing_func� � z9STDLibLogObserverTests.test_failure.<locals>.failing_func�Hi mom�me�rq rr � log_failurer �in failing_func�ZeroDivisionErrorr8 )r� r rs rK rM rN �assertIn�r0 r �failurerJ rH r. r r r � test_failure� s � z#STDLibLogObserverTests.test_failurec C s� d dd�}z|� W n t y t� }|�� Y nw tdd|d�}| �|�\}}| �t|�d� | �d|� | �d |� | �d |� dS ) z� A cleaned Failure object has a fake traceback object; make sure that logging such a failure still results in the exception details being logged. r Nc S r} r~ r r r r r r � r� z@STDLibLogObserverTests.test_cleanedFailure.<locals>.failing_funcr� r� r� r r� r� r8 )r� r � cleanFailurers rK rM rN r� r� r r r �test_cleanedFailure� s � z*STDLibLogObserverTests.test_cleanedFailurer8 )r r9 r: r; rA r! rD r r r r r< rK rQ rf rm rw r{ r| r� r� r r r r r= L s # r= c C s< t � } tj}t| ddd�}t|�}t|�}|�|� || fS )z� Construct a 2-tuple of C{(StreamHandler, BytesIO)} for testing interaction with the 'logging' module. @return: handler and io object r4 � )�encoding�newline)r r' � BASIC_FORMATr r r � setFormatter)r. �template�stream� formatter�handlerr r r r, s r, c @ s, e Zd ZdZd dd�Zdeddfdd�ZdS ) r) zN A L{py_logging.Handler} that remembers all logged records in a list. r Nc C s t j�| � g | _dS )z5 Initialize this L{BufferedHandler}. N)r' �Handlerr1 rH r/ r r r r1 s zBufferedHandler.__init__�recordc C s | j �|� dS )z& Remember the record. N)rH r_ )r0 r� r r r �emit s zBufferedHandler.emitr8 )r r9 r: r; r1 r r� r r r r r) s r) )+r; �loggingr' r �inspectr �ior r r r r r �typingr r r �zope.interface.exceptionsr �zope.interface.verifyr �twisted.python.compatr �twisted.python.failurer � twisted.trialr � _interfacesr r �_levelsr �_stdlibr r< r^ r r! �TestCaser= r, r� r) r r r r �<module> s* $ 9PK ͪ�Z����t t % __pycache__/test_util.cpython-310.pycnu �[��� o �b� � @ sZ d Z ddlmZ ddlmZ ddlmZmZ ddlm Z ddl m Z G dd � d ej �Z d S ) z) Test cases for L{twisted.logger._util}. � )� implementer)�unittest� )� ILogObserver�LogEvent)� LogPublisher)� formatTracec @ s$ e Zd ZdZddd�Zddd�ZdS ) � UtilTestsz Utility tests. �returnNc s� t � �tg d�� tt�dtddfdd���tt�dtddf� ����fdd� ��tt�dtddf� �����fd d � ������ ���� ���� �� � dS ) zJ Tracing keeps track of forwarding done by the publisher. �� log_trace�er Nc S � d S �N� �r r r �?/usr/lib/python3/dist-packages/twisted/logger/test/test_util.py�o1 � z UtilTests.test_trace.<locals>.o1c s, �� | � � ��| d ��f��fg� d S �Nr ��assertIs� assertEqualr )�eventr �o2� publisher�selfr r r ! s ��z UtilTests.test_trace.<locals>.o2c s2 �� | � � ��| d ��f��f��fg� d S r r r �r r r �o3r r r r r - s ��z UtilTests.test_trace.<locals>.o3)r �dictr r r � addObserver)r r r r � test_trace s " zUtilTests.test_tracec s t g d�� tt�dtddfdd���tt�dtddfdd���tt�dtddfd d ���tt�dtddfd d ���tt�dtddfd d���d�_d�_d�_d�_d�_tt�dtddf� ��������� � f dd� �}|�t���t�����d�_d�_t������ d� _� � � dS )z) Format trace as string. r r r Nc S r r r r r r r r D r z&UtilTests.test_formatTrace.<locals>.o1c S r r r r r r r r H r z&UtilTests.test_formatTrace.<locals>.o2c S r r r r r r r r L r z&UtilTests.test_formatTrace.<locals>.o3c S r r r r r r r �o4P r z&UtilTests.test_formatTrace.<locals>.o4c S r r r r r r r �o5T r z&UtilTests.test_formatTrace.<locals>.o5zroot/o1z root/p1/o2z root/p1/o3z root/p1/p2/o4zroot/o5c s@ � � | � � t| d �}� �|dj� ��������d� � d S )Nr z�{root} ({root.name}) -> {o1} ({o1.name}) -> {p1} ({p1.name}) -> {o2} ({o2.name}) -> {o3} ({o3.name}) -> {p2} ({p2.name}) -> {o4} ({o4.name}) -> {o5} ({o5.name}) -> {oTest} ) �rootr r r r" r# �p1�p2�oTest)r r r �format)r �trace� r r r r r" r# r' r% r& r$ r r r � testObserver^ s" ��z0UtilTests.test_formatTrace.<locals>.testObserverz root/p1/p2/zroot/p1/zroot/)r r r r �namer )r r+ r r* r �test_formatTrace> s4 , zUtilTests.test_formatTrace)r N)�__name__� __module__� __qualname__�__doc__r! r- r r r r r s )r N)r1 �zope.interfacer � twisted.trialr � _interfacesr r � _observerr �_utilr �TestCaser r r r r �<module> s PK ͪ�Z�Dq�/# /# # __pycache__/test_io.cpython-310.pycnu �[��� o �b�" � @ s� d Z ddlZddlmZmZ ddlmZ ddlmZ ddl m Z ddl m Z m Z dd lmZ dd lmZ dd lmZ dd lmZ ee �G d d� de��ZG dd� de j�ZdS )z' Test cases for L{twisted.logger._io}. � N)�List�Optional)� implementer)� NamedConstant)�unittest� )� ILogObserver�LogEvent)� LoggingFile)�LogLevel)�Logger)� LogPublisherc sP e Zd ZdZejdfdededee ddf� fdd� Z d e ddfd d �Z � Z S ) �TestLoggingFilezU L{LoggingFile} that is also an observer which captures events and messages. N�logger�level�encoding�returnc s" t � j|||d� g | _g | _d S )N�r r r )�super�__init__�events�messages)�selfr r r �� __class__� �=/usr/lib/python3/dist-packages/twisted/logger/test/test_io.pyr s zTestLoggingFile.__init__�eventc C s, | j �|� d|v r| j�|d � d S d S )N�log_io)r �appendr )r r r r r �__call__' s �zTestLoggingFile.__call__)�__name__� __module__� __qualname__�__doc__r �infor r r �strr r r � __classcell__r r r r r s ����� r c @ s� e Zd ZdZd0dd�Zd0dd�Zd0dd �Zd0d d �Zd0d d �Zd0dd�Z d0dd�Z d0dd�Z d0dd�Z d0dd�Z d0dd�Zd0dd�Zd0d d �Zd0d d �Zd0d d!�Zd0d"d#�Zd0d$d%�Zd0d&d'�Zd0d(d)�Zd0d*d+�Zejdfd,ed-ee de fd.d/�Z dS )1�LoggingFileTestsz# Tests for L{LoggingFile}. r Nc C s t � | _t| jd�| _dS )zK Create a logger for test L{LoggingFile} instances to use. ��observerN)r � publisherr r )r r r r �setUp2 s zLoggingFileTests.setUpc C sf | � t| j�jd� | �| jg�}| � t|�d� | � |d d t� d}| � |d d d| � dS )z0 L{LoggingFile.softspace} is 0. r � �categoryz(twisted.logger._io.LoggingFile.softspace�messagez#%s was deprecated in Twisted 21.2.0N)� assertEqualr r � softspace� flushWarnings�test_softspace�len�DeprecationWarning)r � warningsShown�deprecatedClassr r r r3 9 s �z LoggingFileTests.test_softspacec C sj t | j�}| �tt|dd� | �tt|dd� | �tt|dd� | �tt|ddg� | �tt|d d � d S ) z? Some L{LoggingFile} attributes are read-only. �closedTr �utf-8�mode�r�newlines� �name�fooN)r r � assertRaises�AttributeError�setattr�r �fr r r �test_readOnlyAttributesH s z(LoggingFileTests.test_readOnlyAttributesc C s~ t | j�}| �t|j� | �t|j� | �t|j� | �t|j� | �t|j� | �t|j � | �t|j � | �t|j � dS )z> Some L{LoggingFile} methods are unsupported. N) r r r@ �IOError�read�next�readline� readlines� xreadlines�seek�tell�truncaterC r r r �test_unsupportedMethodsT s z(LoggingFileTests.test_unsupportedMethodsc C s> t | j�}| �|jtj� t | jtjd�}| �|jtj� dS )z? Default level is L{LogLevel.info} if not set. �r N)r r r0 r r r% �errorrC r r r � test_levelc s zLoggingFileTests.test_levelc C s< t | j�}| �|jt�� � t | jdd�}| �|jd� dS )zM Default encoding is C{sys.getdefaultencoding()} if not set. r9 �r N)r r r0 r �sys�getdefaultencodingrC r r r � test_encodingm s z LoggingFileTests.test_encodingc C � t | j�}| �|jd� dS )z* Reported mode is C{"w"}. �wN)r r r0 r: rC r r r � test_modew � zLoggingFileTests.test_modec C s t | j�}| �|j� dS )z7 The C{newlines} attribute is L{None}. N)r r � assertIsNoner< rC r r r � test_newlines~ s z LoggingFileTests.test_newlinesc C rW )z1 The C{name} attribute is fixed. z.<LoggingFile twisted.logger.test.test_io#info>N)r r r0 r> rC r r r � test_name� rZ zLoggingFileTests.test_namec C s2 t | j�}|�� | �|j� | �t|jd� dS )z7 L{LoggingFile.close} closes the file. �HelloN)r r �close� assertTruer8 r@ � ValueError�writerC r r r � test_close� s zLoggingFileTests.test_closec C s t | j�}|�� dS )z4 L{LoggingFile.flush} does nothing. N)r r �flushrC r r r � test_flush� s zLoggingFileTests.test_flushc C s t | j�}| �|�� d� dS )z6 L{LoggingFile.fileno} returns C{-1}. ���N)r r r0 �filenorC r r r � test_fileno� s z LoggingFileTests.test_filenoc C s t | j�}| �|�� � dS )z9 L{LoggingFile.isatty} returns C{False}. N)r r � assertFalse�isattyrC r r r � test_isatty� rZ z LoggingFileTests.test_isattyc C �Z | � � }|�d� | �|jg � |�d� | �|jdg� |�d� | �|jg d�� dS )z, Writing buffers correctly. r^ z , world! � Hello, world!z It's nice to meet you. Indeed.�rm zIt's nice to meet you.� N�� observedFilerb r0 r rC r r r �test_writeBuffering� s �z$LoggingFileTests.test_writeBufferingc C �* | j dd�}|�d� | �|jdg� dS )z, Bytes are decoded to text. r9 rS s Hello, Mr. Sánchez � Hello, Mr. SánchezNrp rC r r r �test_writeBytesDecoded� � z'LoggingFileTests.test_writeBytesDecodedc C rs )z( Unicode is unmodified. r9 rS u Hello, Mr. Sánchez rt Nrp rC r r r �test_writeUnicode� rv z"LoggingFileTests.test_writeUnicodec C s� | � � }|�d� | �t|j�d� | �|jd d tj� | j tjd�}|�d� | �t|j�d� | �|jd d tj� dS )z0 Log level is emitted properly. �Hello r- r � log_levelrP N)rq rb r0 r4 r r r% rQ rC r r r �test_writeLevel� s z LoggingFileTests.test_writeLevelc C s> | � � }|�d� | �t|j�d� | �|jd d d� dS )z/ Log format is C{"{message}"}. rx r- r � log_formatz{log_io}N)rq rb r0 r4 r rC r r r �test_writeFormat� s z!LoggingFileTests.test_writeFormatc C rl )z6 C{writelines} does not add newlines. )r^ z, ro )zworld! rm )zIt's nice to meet you. zIndeed.rn N)rq � writelinesr0 r rC r r r �test_writelinesBuffering� s �z)LoggingFileTests.test_writelinesBufferingc C s> | � � }| �td|� tddd� td� | �|jdg� dS )z; L{LoggingFile} can replace L{sys.stdout}. �stdoutzHello,� )�endzworld.z Hello, world.N)rq �patchrT �printr0 r rC r r r � test_print� s zLoggingFileTests.test_printr r c sH g � t t�dtddf� fdd� �}t|d�}� �t|||d�� � d S ) a� Construct a L{LoggingFile} with a built-in observer. @param level: C{level} argument to L{LoggingFile} @param encoding: C{encoding} argument to L{LoggingFile} @return: a L{TestLoggingFile} with an observer that appends received events into the file's C{events} attribute (a L{list}) and event messages into the file's C{messages} attribute (a L{list}). r r Nc s � d | � d S )Nr r )r �� loggingFilesr r r* s z/LoggingFileTests.observedFile.<locals>.observerr) r r )r r r r r r )r r r r* �logr r� r rq s z LoggingFileTests.observedFile)r N) r! r" r# r$ r, r3 rE rO rR rV rY r\ r] rc re rh rk rr ru rw rz r| r~ r� r r% r r r& r rq r r r r r( - s> ����r( )r$ rT �typingr r �zope.interfacer � constantlyr � twisted.trialr � _interfacesr r �_ior �_levelsr �_loggerr � _observerr r �TestCaser( r r r r �<module> s PK ͪ�ZV1�-0 -0 ' __pycache__/test_global.cpython-310.pycnu �[��� o �b�1 � @ s� d Z ddlZddlmZmZmZmZmZmZm Z m Z ddl m Z ddl mZ ddlmZ ddlmZmZ dd lmZmZ dd lmZ dd lmZ dd lm Z dd l m Z dej dee dee ddfdd�Z G dd� dej �Z!dS )z+ Test cases for L{twisted.logger._global}. � N)�IO�Any�List�Optional�TextIO�Tuple�Type�cast)�Failure)�unittest� )�textFileLogObserver)�MORE_THAN_ONCE_WARNING� LogBeginner)� ILogObserver�LogEvent)�LogLevel)�Logger)� LogPublisher)�nextLine�test� actualEvents�expectedEvents�returnc st t |�t |�kr| �||� t� � |D ] }� t|�� �O � qdtdtf� fdd� ��fdd�|D �}| �||� dS )a7 Compare two sequences of log events, examining only the the keys which are present in both. @param test: a test case doing the comparison @param actualEvents: A list of log events that were emitted by a logger. @param expectedEvents: A list of log events that were expected by a test. �eventr c s, | � � }| �� D ] }|� vr|�|� q|S �N)�copy�keys�pop)r r �key)� allMergedKeys� �A/usr/lib/python3/dist-packages/twisted/logger/test/test_global.py�simplify* s �z compareEvents.<locals>.simplifyc s g | ]}� |��qS r! r! ��.0r )r# r! r" � <listcomp>1 s z!compareEvents.<locals>.<listcomp>N)�len� assertEqual�setr r )r r r r �simplifiedActualr! )r r# r" � compareEvents s r+ c @ s� e Zd ZdZd"dd�Zd"dd�Zd"dd �Zd ed eddfd d �Z d"dd�Z d"dd�Z d"dd�Z d"dd�Z d"dd�Zd"dd�Zd"dd�Zd"d d �Zd"d d �Zd"d d!�ZdS )#�LogBeginnerTestsz# Tests for L{LogBeginner}. r Nc C sZ t � | _t�� | _G dd� d�}G dd� d�}|� | _|� | _t| j| j| j| j�| _d S )Nc @ s e Zd Ze� Ze� ZdS )z&LogBeginnerTests.setUp.<locals>.NotSysN)�__name__� __module__� __qualname__�object�stdout�stderrr! r! r! r" �NotSys> s r3 c @ sR e Zd Zd dd�Z ddedee deded ee e d ee ddfd d �Z dS )z+LogBeginnerTests.setUp.<locals>.NotWarningsr Nc S s g | _ d S r )�warnings)�selfr! r! r" �__init__C s �z4LogBeginnerTests.setUp.<locals>.NotWarnings.__init__�message�category�filename�lineno�file�linec S s | j �||||||f� dS )a Emulate warnings.showwarning. @param message: A warning message to emit. @param category: A warning category to associate with C{message}. @param filename: A file name for the source code file issuing the warning. @param lineno: A line number in the source file where the warning was issued. @param file: A file to write the warning message to. If L{None}, write to L{sys.stderr}. @param line: A line of source code to include with the warning message. If L{None}, attempt to read the line from C{filename} and C{lineno}. N)r4 �append)r5 r7 r8 r9 r: r; r< r! r! r" � showwarningJ s z7LogBeginnerTests.setUp.<locals>.NotWarnings.showwarning�r N)NN) r- r. r/ r6 �strr �Warning�intr r r r> r! r! r! r" � NotWarningsB s& ����� ���rC ) r � publisher�io�StringIO� errorStream� sysModule�warningsModuler �beginner)r5 r3 rC r! r! r" �setUp: s # �zLogBeginnerTests.setUpc sr t ddd�}g � g �tt� fdd��}tt�fdd��}| j�||f� | �|� | �|g� � | �|g�� dS )z? Test that C{beginLoggingTo()} adds observers. � r ��foo�barc � � � | �S r �r= ��e��events1r! r" �<lambda>t � zBLogBeginnerTests.test_beginLoggingToAddObservers.<locals>.<lambda>c rP r rQ rR ��events2r! r" rV u rW N)�dictr r rJ �beginLoggingTorD r( �r5 r �o1�o2r! �rU rY r" � test_beginLoggingToAddObserversk s z0LogBeginnerTests.test_beginLoggingToAddObserversc sr t ddd�}g � g �tt� fdd��}tt�fdd��}| �|� | j�||f� | �|g� � | �|g�� dS )z\ Test that events are buffered until C{beginLoggingTo()} is called. rL r rM c rP r rQ rR rT r! r" rV � rW zDLogBeginnerTests.test_beginLoggingToBufferedEvents.<locals>.<lambda>c rP r rQ rR rX r! r" rV � rW N)rZ r r rD rJ r[ r( r\ r! r_ r" �!test_beginLoggingToBufferedEvents} s z2LogBeginnerTests.test_beginLoggingToBufferedEvents�limitrJ c C sf t |d �D ] }| �t|d�� qg }|�tt|j�g� | �tt d|d ��tdd� |D ��� dS )a� Verify that when more than C{limit} events are logged to L{LogBeginner}, only the last C{limit} are replayed by L{LogBeginner.beginLoggingTo}. @param limit: The maximum number of events the log beginner should buffer. @param beginner: The L{LogBeginner} against which to verify. @raise: C{self.failureException} if the wrong events are replayed by C{beginner}. rL )�countc s s � | ]}|d V qdS )rc Nr! r$ r! r! r" � <genexpr>� s � z4LogBeginnerTests._bufferLimitTest.<locals>.<genexpr>N) �rangerD rZ r[ r r r= r( �list)r5 rb rJ rc �eventsr! r! r" �_bufferLimitTest� s �z!LogBeginnerTests._bufferLimitTestc C s t j}| �|| j� dS )z� Up to C{LogBeginner._DEFAULT_BUFFER_SIZE} log events are buffered for replay by L{LogBeginner.beginLoggingTo}. N)r �_DEFAULT_BUFFER_SIZErh rJ )r5 rb r! r! r" �test_defaultBufferLimit� s z(LogBeginnerTests.test_defaultBufferLimitc C s. d}t | j| j| j| j|d�}| �||� dS )z� The size of the L{LogBeginner} event buffer can be overridden with the C{initialBufferSize} initilizer argument. � )�initialBufferSizeN)r rD rG rH rI rh )r5 rb rJ r! r! r" �test_overrideBufferLimit� s �z)LogBeginnerTests.test_overrideBufferLimitc C s$ g }g }t �� }t|�}| �tdd�� t� \}}| j�tt |j �|g� | �tdd�� t� \}}| j�tt |j �|g� | �tdd�� tt t j ||||d�} d| _t| |tdd�tdd�| tdd�g� t| || tdd�g� |�� } | �d|� d|� d �| � | �d|� d|� d �| � dS ) z� When invoked twice, L{LogBeginner.beginLoggingTo} will emit a log message warning the user that they previously began logging, and add the new log observers. � prebuffer)r � postbuffer�postwarn)� log_format� log_level�fileNow�lineNow�fileThen�lineThenN�<�:�>)rE rF r rD rZ r rJ r[ r r r= r r �warn�maxDiffr+ �getvalue�assertIn) r5 rU rY � fileHandle� textObserver� firstFilename� firstLine�secondFilename� secondLine�warning�outputr! r! r" �test_beginLoggingToTwice� sB � �� z)LogBeginnerTests.test_beginLoggingToTwicec C s: t | jd�}|�d� |jddd� | �| j�� d� dS )zP Critical messages will be written as text to the error stream. ��observerz ignore thisza critical {message}r7 )r7 za critical message N)r rD �info�criticalr( rG r| �r5 �logr! r! r" �test_criticalLogging� s z%LogBeginnerTests.test_criticalLoggingc C s8 t | jd�}| j�d� |�d� | �| j�� d� dS )z� Once logging has begun with C{beginLoggingTo}, critical messages are no longer written to the output stream. r� r! zanother critical message� N)r rD rJ r[ r� r( rG r| r� r! r! r" �test_criticalLoggingStops� s z*LogBeginnerTests.test_criticalLoggingStopsc C s� g }| j �tt|j�g� tdtt| jj�d� t | |t ddd�g� |dd�= tdtt| jj �d� t | |t ddd�g� dS )z� L{LogBeginner.beginLoggingTo} will re-direct the standard output and error streams by setting the C{stdio} and C{stderr} attributes on its sys module object. z Hello, world.�r; r1 )� log_namespace�log_ioNz Error, world.r2 ) rJ r[ r r r= �printr rH r1 r+ rZ r2 )r5 rg r! r! r" �%test_beginLoggingToRedirectStandardIO� s � �z6LogBeginnerTests.test_beginLoggingToRedirectStandardIOc C sD | j j}| j j}| jjddd� | �| j j|� | �| j j|� dS )z� L{LogBeginner.beginLoggingTo} will leave the existing stdout/stderr in place if it has been told not to replace them. r! F)�redirectStandardION)rH r1 r2 rJ r[ �assertIs)r5 �oldOut�oldErrr! r! r" � test_beginLoggingToDontRedirect s z0LogBeginnerTests.test_beginLoggingToDontRedirectc C s� t �t �� d�}t �t �� d�}|| j_|| j_g }| j�tt |j �g� tt | jj�}tt | jj�}| � |j d� | � |j d� |�d� |�d� t| |tdd�tdd�g� dS ) z� When L{LogBeginner.beginLoggingTo} redirects stdout/stderr streams, the replacement streams will preserve the encoding of the replaced streams, to minimally disrupt any application relying on a specific encoding. z shift-JIS�big5s �� s �� u 李)r� u 瑩N)rE � TextIOWrapper�BytesIOrH r1 r2 rJ r[ r r r= r r( �encoding�writer+ rZ )r5 �weird�weirderrrg r1 r2 r! r! r" �$test_beginLoggingToPreservesEncoding s z5LogBeginnerTests.test_beginLoggingToPreservesEncodingc C s� | j �dttd� g }| j�tt|j�g� | j �dttd� t � � }| j jdttd|d� | � | j j dttdddfdttd|dfg� t | |tdtjd tj tdd �g� dS ) z� L{LogBeginner.beginLoggingTo} will redirect the warnings of its warnings module into the logging system. z a messagerL zanother messager z yet anotherrk r� N�.)r� r8 r9 r: )rI r> �DeprecationWarning�__file__rJ r[ r r r= rE rF r( r4 r+ rZ r. r- )r5 rg �fr! r! r" �test_warningsModule/ s> � ��������z$LogBeginnerTests.test_warningsModulec C sX t td��}t| jd�}|jd|d� | j�� }| �d|� | �d|� | �d|� dS )zR The string resulting from a logged failure contains a traceback. z,this is not the behavior you are looking forr� z a failure)�failure� TracebackN)r � Exceptionr rD r� rG r| r} )r5 r� r� �msgr! r! r" � test_failuresAppendTracebacksV s z.LogBeginnerTests.test_failuresAppendTracebacksr? )r- r. r/ �__doc__rK r` ra rB r rh rj rm r� r� r� r� r� r� r� r� r! r! r! r" r, 5 s 1 + 'r, )"r� rE �typingr r r r r r r r �twisted.python.failurer � twisted.trialr �_filer �_globalr r � _interfacesr r �_levelsr �_loggerr � _observerr �test.test_stdlibr �TestCaser+ r, r! r! r! r" �<module> s, ( ��� � PK ͪ�Z��^� % __pycache__/test_file.cpython-310.pycnu �[��� o �b� � @ s� d Z ddlmZ ddlmZ ddlmZmZmZm Z m Z m Z ddl m Z ddlmZ ddlmZ ddlmZ d d lmZmZ d d lmZ G d d � d e�ZG dd� de�ZG dd� d�ZdS )z) Test cases for L{twisted.logger._file}. � )�StringIO)� TracebackType)�IO�Any�AnyStr�Optional�Type�cast)�BrokenMethodImplementation)� verifyObject)�Failure)�TestCase� )�FileLogObserver�textFileLogObserver)� ILogObserverc @ s\ e Zd ZdZddd�Zddd�Zdee d eddfd d �Z dd d �Z ddd�Z ddd�Z dS )�FileLogObserverTestsz' Tests for L{FileLogObserver}. �returnNc C s� t � �7}t|dd� �}ztt|� W n ty) } z | �|� W Y d}~n d}~ww W d � dS W d � dS 1 s=w Y dS )z; L{FileLogObserver} is an L{ILogObserver}. c S � t | �S �N��str��e� r �?/usr/lib/python3/dist-packages/twisted/logger/test/test_file.py�<lambda> � z5FileLogObserverTests.test_interface.<locals>.<lambda>N)r r r r r �fail)�self� fileHandle�observerr r r r �test_interface s ����"�z#FileLogObserverTests.test_interfacec C s` t � �#}t|dd� �}tdd�}||� | �|�� t|�� W d � dS 1 s)w Y dS )zV L{FileLogObserver} writes to the given file when it observes events. c S r r r r r r r r * r z9FileLogObserverTests.test_observeWrites.<locals>.<lambda>� ��xN)r r �dict� assertEqual�getvaluer �r r r! �eventr r r �test_observeWrites% s "�z'FileLogObserverTests.test_observeWrites�what�countc sh t � �'}tttt |�� fdd��}tdd�}||� | �|j|� W d � dS 1 s-w Y dS )z� Verify that observer performs an expected number of writes when the formatter returns a given value. @param what: the value for the formatter to return. @param count: the expected number of writes. c s � S r r r �r, r r r 8 s z:FileLogObserverTests._test_observeWrites.<locals>.<lambda>r# r$ N)� DummyFiler r r r r& r' �writes)r r, r- r r! r* r r. r �_test_observeWrites/ s "�z(FileLogObserverTests._test_observeWritesc C s | � dd� dS )z� L{FileLogObserver} does not write to the given file when it observes events and C{formatEvent} returns L{None}. Nr �r1 �r r r r �test_observeWritesNone= � z+FileLogObserverTests.test_observeWritesNonec C s | � dd� dS )z� L{FileLogObserver} does not write to the given file when it observes events and C{formatEvent} returns C{""}. � r Nr2 r3 r r r �test_observeWritesEmptyD r5 z,FileLogObserverTests.test_observeWritesEmptyc C sd t � �%}tttt |�dd� �}tdd�}||� | �|jd� W d � dS 1 s+w Y dS )zl L{FileLogObserver} calles C{flush()} on the output file when it observes an event. c S r r r r r r r r Q r z:FileLogObserverTests.test_observeFlushes.<locals>.<lambda>r# r$ N)r/ r r r r r& r' �flushesr) r r r �test_observeFlushesK s "�z(FileLogObserverTests.test_observeFlushes�r N) �__name__� __module__� __qualname__�__doc__r" r+ r r �intr1 r4 r7 r9 r r r r r s r c @ sB e Zd ZdZddd�Zddd�Zddd �Zdd d �Zdd d �ZdS )�TextFileLogObserverTestsz+ Tests for L{textFileLogObserver}. r Nc C s@ t � �}t|�}| �|t� W d � dS 1 sw Y dS )zF L{textFileLogObserver} returns a L{FileLogObserver}. N)r r �assertIsInstancer �r r r! r r r �test_returnsFileLogObserver\ s "�z4TextFileLogObserverTests.test_returnsFileLogObserverc C sB t � �}t|�}| �|j|� W d � dS 1 sw Y dS )�F Returned L{FileLogObserver} has the correct outFile. N)r r �assertIs�_outFilerB r r r � test_outFiled s "�z%TextFileLogObserverTests.test_outFilec C sX t � � }t|dd�}|tddd�� | �|�� d� W d � dS 1 s%w Y dS )rD z%f)� timeFormat�XYZZYg�����m�@)� log_format�log_timez600000 [-#-] XYZZY N)r r r& r' r( rB r r r �test_timeFormatl s "�z(TextFileLogObserverTests.test_timeFormatc C s� t � �>}t|�}zdd W n ty t� }Y nw t|d�}||� |�� }| j|�d�d �d�t |�d� W d � dS 1 sDw Y dS )z� If the C{"log_failure"} key exists in an event, the observer appends the failure's traceback to the output. r# r �� log_failure� z Traceback )�msgN) r r �ZeroDivisionErrorr r& r( � assertTrue�split� startswith�repr)r r r! �failurer* �outputr r r �test_observeFailureu s � �"�z,TextFileLogObserverTests.test_observeFailurec C s` t � �#}t|�}tt� d�}||� |�� }d}| �||� W d � dS 1 s)w Y dS )z� If the C{"log_failure"} key exists in an event, and contains an object that raises when you call its C{getTraceback()}, then the observer appends a message noting the problem, instead of raising. rM z'(UNABLE TO OBTAIN TRACEBACK FROM EVENT)N)r r r& �objectr( �assertIn)r r r! r* rW �expectedr r r �+test_observeFailureThatRaisesInGetTraceback� s "�zDTextFileLogObserverTests.test_observeFailureThatRaisesInGetTracebackr: ) r; r< r= r> rC rG rL rX r\ r r r r r@ W s r@ c @ sn e Zd ZdZddd�Zdeddfdd�Zdd d �Zdd d �Zd e e e de e de e de e fdd�ZdS )r/ z. File that counts writes and flushes. r Nc C s d| _ d| _d S )Nr )r0 r8 r3 r r r �__init__� s zDummyFile.__init__�datac C � | j d7 _ dS )z8 Write data. @param data: data r# N)r0 )r r^ r r r �write� s zDummyFile.writec C r_ )z Flush buffers. r# N)r8 r3 r r r �flush� s zDummyFile.flushc C s | S r r r3 r r r � __enter__� s zDummyFile.__enter__�exc_type� exc_value� tracebackc C s d S r r )r rc rd re r r r �__exit__� s zDummyFile.__exit__r: )r r/ )r; r< r= r> r] r r` ra rb r r � BaseExceptionr �boolrf r r r r r/ � s ����r/ N) r> �ior �typesr �typingr r r r r r �zope.interface.exceptionsr �zope.interface.verifyr �twisted.python.failurer �twisted.trial.unittestr �_filer r � _interfacesr r r@ r/ r r r r �<module> s BAPK ͪ�Z��2�2 2 ' __pycache__/test_buffer.cpython-310.pycnu �[��� o �b � @ sj d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z mZ G d d � d e j�Zd S ) z+ Test cases for L{twisted.logger._buffer}. � )�List�cast)�BrokenMethodImplementation)� verifyObject)�unittest� )�LimitedHistoryLogObserver)� ILogObserver�LogEventc @ s. e Zd ZdZd dd�Zd dd�Zd dd �ZdS ) � LimitedHistoryLogObserverTestsz1 Tests for L{LimitedHistoryLogObserver}. �returnNc C sJ t d�}ztt|� W dS ty$ } z | �|� W Y d}~dS d}~ww )zH L{LimitedHistoryLogObserver} provides L{ILogObserver}. r N)r r r r �fail)�self�observer�e� r �A/usr/lib/python3/dist-packages/twisted/logger/test/test_buffer.py�test_interface s ��z-LimitedHistoryLogObserverTests.test_interfacec C sZ d}dd� t |d �D �}t|�}|D ]}||� qg }|�tt|j�� | �||� dS )zY L{LimitedHistoryLogObserver} saves history in the order it is received. � c S � g | ]}t |d ��qS �)�n��dict��.0r r r r � <listcomp>& � z=LimitedHistoryLogObserverTests.test_order.<locals>.<listcomp>r N��ranger �replayTor r �append� assertEqual�r �size�eventsr �event� outEventsr r r � test_order! s z)LimitedHistoryLogObserverTests.test_orderc C sd d}dd� t |d �D �}t|�}|D ]}||� qg }|�tt|j�� | �|| d� |� dS )z� When more events than a L{LimitedHistoryLogObserver}'s maximum size are buffered, older events will be dropped. r c S r r r r r r r r 6 r z=LimitedHistoryLogObserverTests.test_limit.<locals>.<listcomp>r Nr r# r r r � test_limit0 s z)LimitedHistoryLogObserverTests.test_limit)r N)�__name__� __module__� __qualname__�__doc__r r( r) r r r r r s r N)r- �typingr r �zope.interface.exceptionsr �zope.interface.verifyr � twisted.trialr �_bufferr � _interfacesr r �TestCaser r r r r �<module> s PK ͪ�Z[h�H H ) __pycache__/test_observer.cpython-310.pycnu �[��� o �b* � @ s� d Z ddlmZmZmZmZ ddlmZ ddlm Z ddl m Z ddl m Z ddlmZmZ dd lmZ dd lmZ G d d � d e j�Zd S )z- Test cases for L{twisted.logger._observer}. � )�Dict�List�Tuple�cast)� implementer)�BrokenMethodImplementation)� verifyObject)�unittest� )� ILogObserver�LogEvent)�Logger)� LogPublisherc @ st e Zd ZdZddd�Zddd�Zddd �Zdd d �Zdd d �Zddd�Z ddd�Z ddd�Z ddd�Z ddd�Z dS )�LogPublisherTestsz$ Tests for L{LogPublisher}. �returnNc C sH t � }ztt|� W dS ty# } z | �|� W Y d}~dS d}~ww )z8 L{LogPublisher} is an L{ILogObserver}. N)r r r r �fail)�self� publisher�e� r �C/usr/lib/python3/dist-packages/twisted/logger/test/test_observer.py�test_interface s ��z LogPublisherTests.test_interfacec C s@ t tdd� �}t tdd� �}t||�}| �||ht|j�� dS )zB L{LogPublisher.observers} returns the observers. c S � d S �Nr �r r r r �<lambda>' � z2LogPublisherTests.test_observers.<locals>.<lambda>c S r r r r r r r r ( r N)r r r � assertEqual�set� _observers)r �o1�o2r r r r �test_observers# s z LogPublisherTests.test_observersc C sZ t tdd� �}t tdd� �}t tdd� �}t||�}|�|� | �|||ht|j�� dS )z? L{LogPublisher.addObserver} adds an observer. c S r r r r r r r r 1 r z4LogPublisherTests.test_addObserver.<locals>.<lambda>c S r r r r r r r r 2 r c S r r r r r r r r 3 r N)r r r � addObserverr r r �r r r! �o3r r r r �test_addObserver- s z"LogPublisherTests.test_addObserverc C s t � }| �t|jt� � dS )ze L{LogPublisher.addObserver} refuses to add an observer that's not callable. N)r � assertRaises� TypeErrorr# �object)r r r r r �test_addObserverNotCallable9 s z-LogPublisherTests.test_addObserverNotCallablec C sZ t tdd� �}t tdd� �}t tdd� �}t|||�}|�|� | �||ht|j�� dS )zE L{LogPublisher.removeObserver} removes an observer. c S r r r r r r r r E r z7LogPublisherTests.test_removeObserver.<locals>.<lambda>c S r r r r r r r r F r c S r r r r r r r r G r N�r r r �removeObserverr r r r$ r r r �test_removeObserverA s z%LogPublisherTests.test_removeObserverc C sX t tdd� �}t tdd� �}t tdd� �}t||�}|�|� | �||ht|j�� dS )zd L{LogPublisher.removeObserver} removes an observer that is not registered. c S r r r r r r r r R r zDLogPublisherTests.test_removeObserverNotRegistered.<locals>.<lambda>c S r r r r r r r r S r c S r r r r r r r r T r Nr+ r$ r r r � test_removeObserverNotRegisteredM s z2LogPublisherTests.test_removeObserverNotRegisteredc C sx t ddd�}g }g }g }tt|j�}tt|j�}tt|j�}t|||�}||� | �||� | �||� | �||� dS )z6 L{LogPublisher} calls its observers. � r ��foo�barN)�dictr r �appendr �assertIn) r �event�events1�events2�events3r r! r% r r r r � test_fanOutZ s z LogPublisherTests.test_fanOutc s� t ddd�}td��g � tt�dtddf� �fdd � �}g }t|tt|j��}||� | �|� � d d � |D �}| � t |�d� | � |d j �� | � t � �d� dS ) z} Observer raises an exception during fan out: a failure is logged, but not re-raised. Life goes on. r/ r r0 �ARGH! EVIL DEATH!r6 r Nc s � }� � | � |r ��d S r )r4 )r6 � shouldRaise��events� exceptionr r �observerx s �z7LogPublisherTests.test_observerRaises.<locals>.observerc S s g | ] }d |v r|d �qS )� log_failurer )�.0r r r r � <listcomp>� s z9LogPublisherTests.test_observerRaises.<locals>.<listcomp>r ) r3 � RuntimeErrorr r r r r r4 r5 r �len�assertIs�value)r r6 r@ � collectorr �errorsr r= r �test_observerRaisesn s z%LogPublisherTests.test_observerRaisesc s^ t ddd�}td�� tt�dtddfdd ��}G � fd d �d t�}t|�}|� |_||� dS ) z� Observer raises an exception during fan out and the publisher's Logger pukes when the failure is reported. The exception does not propagate back to the caller. r/ r r0 r; r6 r Nc S s t d��)Nz Sad panda)rD )r6 r r r r@ � s zGLogPublisherTests.test_observerRaisesAndLoggerHatesMe.<locals>.observerc s& e Zd Zdededdf� fdd� ZdS )zILogPublisherTests.test_observerRaisesAndLoggerHatesMe.<locals>.GurkLogger�args�kwargsr Nc s � �r r )r rK rL �r? r r �failure� s zQLogPublisherTests.test_observerRaisesAndLoggerHatesMe.<locals>.GurkLogger.failure)�__name__� __module__� __qualname__r) rN r rM r r � GurkLogger� s rR )r3 rD r r r r r �log)r r6 r@ rR r r rM r �#test_observerRaisesAndLoggerHatesMe� s z5LogPublisherTests.test_observerRaisesAndLoggerHatesMec s� t ddg d�}i � tt�dtddf� fdd� �}tt�dtddf� fd d � �}t||�}||� | �� d ||ff� | �� d ||f||ff� dS ) zA Tracing keeps track of forwarding to observers. r/ r )r1 r2 � log_tracer r Nc �* � � dtttttf t| d ��� d S )Nr/ rU �� setdefaultr r r r �tupler ��tracesr r r � � �z(LogPublisherTests.test_trace.<locals>.o1c rV )Nr rU rW r rZ r r r! � r\ z(LogPublisherTests.test_trace.<locals>.o2)r3 r r r r r )r r6 r r! r r rZ r � test_trace� s z LogPublisherTests.test_trace)r N)rO rP rQ �__doc__r r" r& r* r- r. r: rJ rT r] r r r r r s r N)r^ �typingr r r r �zope.interfacer �zope.interface.exceptionsr �zope.interface.verifyr � twisted.trialr � _interfacesr r �_loggerr � _observerr �TestCaser r r r r �<module> s PK ͪ�Z��ʕ�> �> ' __pycache__/test_legacy.cpython-310.pycnu �[��� o �b 9 � @ s� d Z ddlZddlmZ ddlmZmZ ddlmZ ddl m Z ddl m Z ddl mZmZ dd lmZ dd lmZ d d lmZ d d lmZmZ d dlmZm Z d dl m Z G dd� dej �Z G dd� dej �Z!dS )z+ Test cases for L{twisted.logger._legacy}. � N)�time)�List�cast)� implementer)�BrokenMethodImplementation)� verifyObject)�context�log)�Failure)�unittest� )� formatEvent)� ILogObserver�LogEvent)�LegacyLogObserverWrapper�publishToNewObserver)�LogLevelc @ s e Zd ZdZd4dd�Zd4dd�Zdedefd d �Zdedefd d �Zd4d d�Z d4dd�Z d4dd�Z d4dd�Z d4dd�Z d4dd�Zd4dd�Zd4dd �Zd4d d �Zd4d d �Zd4d!d"�Zd4d#d$�Zd4d%d&�Zd4d'd(�Zd)edefd*d+�Zd4d,d-�Zd4d.d/�Zd4d0d1�Zd4d2d3�Z dS )5� LegacyLogObserverWrapperTestsz0 Tests for L{LegacyLogObserverWrapper}. �returnNc C sZ t tjdd� �}t|�}ztt|� W dS ty, } z | �|� W Y d}~dS d}~ww )zD L{LegacyLogObserverWrapper} is an L{ILogObserver}. c S � d S �N� ��er r �A/usr/lib/python3/dist-packages/twisted/logger/test/test_legacy.py�<lambda>" s z>LegacyLogObserverWrapperTests.test_interface.<locals>.<lambda>N)r � legacyLogr r r r �fail)�self�legacyObserver�observerr r r r �test_interface s ��z,LegacyLogObserverWrapperTests.test_interfacec C s6 t tj�G dd� d��}t|� �}| �t|�d� dS )zJ L{LegacyLogObserverWrapper} returns the expected string. c @ s. e Zd Zdefdd�Zdejddfdd�ZdS )z?LegacyLogObserverWrapperTests.test_repr.<locals>.LegacyObserverr c S s dS )Nz<Legacy Observer>r �r r r r �__repr__0 � zHLegacyLogObserverWrapperTests.test_repr.<locals>.LegacyObserver.__repr__� eventDictNc S r r r )r r% r r r �__call__3 r$ zHLegacyLogObserverWrapperTests.test_repr.<locals>.LegacyObserver.__call__)�__name__� __module__� __qualname__�strr# r � EventDictr& r r r r �LegacyObserver. s r, z+LegacyLogObserverWrapper(<Legacy Observer>)N)r r r r � assertEqual�repr)r r, r r r r � test_repr) s z'LegacyLogObserverWrapperTests.test_repr�eventc s@ g � t tj� fdd��}t|�}||� | �t� �d� � d S )z� Send an event to a wrapped legacy observer and capture the event as seen by that observer. @param event: an event @return: the event as observed by the legacy wrapper c s � � | �S r )�appendr ��eventsr r r E s z7LegacyLogObserverWrapperTests.observe.<locals>.<lambda>� r )r r r r r- �len)r r0 r r r r2 r �observe: s z%LegacyLogObserverWrapperTests.observec C sX |� dt� � |� dd� |� dtj� | �t|��}|�� D ] \}}| �||� q |S )z� Send an event to a wrapped legacy observer and verify that its data is preserved. @param event: an event @return: the event as observed by the legacy wrapper �log_time� log_system�-� log_level)� setdefaultr r �infor6 �dict�items�assertIn)r r0 �observed�key�valuer r r �forwardAndVerifyL s z.LegacyLogObserverWrapperTests.forwardAndVerifyc C s | � tddd�� dS )ze Basic forwarding: event keys as observed by a legacy observer are the same. r4 r ��foo�barN)rC r= r" r r r � test_forwardc s z*LegacyLogObserverWrapperTests.test_forwardc C s* t � }| �t|d��}| �|d |� dS )zc The new-style C{"log_time"} key is copied to the old-style C{"time"} key. )r7 r N�r rC r= r- �r �stampr0 r r r � test_timej s z'LegacyLogObserverWrapperTests.test_timec C s0 t � }| �t|d |d��}| �|d |� dS )zr The new-style C{"log_time"} key does not step on a pre-existing old-style C{"time"} key. r4 )r7 r r NrH rI r r r �test_timeAlreadySets s z1LegacyLogObserverWrapperTests.test_timeAlreadySetc C �$ | � tdd��}| �|d d� dS )zg The new-style C{"log_system"} key is copied to the old-style C{"system"} key. rE )r8 �systemN�rC r= r- �r r0 r r r � test_system| s z)LegacyLogObserverWrapperTests.test_systemc C s& | � tddd��}| �|d d� dS )zv The new-style C{"log_system"} key does not step on a pre-existing old-style C{"system"} key. rE rF )r8 rN rN NrO rP r r r �test_systemAlreadySet� s z3LegacyLogObserverWrapperTests.test_systemAlreadySetc C s2 t t� tjd�}| �t |��}| �|d d� dS )zy If the new-style C{"log_system"} key is absent, the old-style C{"system"} key is set to C{"-"}. )r7 r: rN r9 N)r= r r r< r6 r- )r r0 r@ r r r � test_noSystem� s z+LegacyLogObserverWrapperTests.test_noSystemc C sT | � ttjdd�� | � ttjdd�� | � ttjdd�� | � ttjdd�� dS )z� If explicitly set, the C{isError} key will be preserved when forwarding from a new-style logging emitter to a legacy logging observer, regardless of log level. r4 )r: �isErrorr N)rC r= r r< �warn�error�criticalr" r r r �test_levelNotChange� s z1LegacyLogObserverWrapperTests.test_levelNotChangec C s" | � ttjd��}| �d|� dS )a) The new-style C{"log_level"} key is not translated to the old-style C{"logLevel"} key. Events are forwarded from the old module from to new module and are then seen by old-style observers. We don't want to add unexpected keys to old-style events. )r: �logLevelN)rC r= r r< � assertNotInrP r r r �test_pythonLogLevelNotSet� s z7LegacyLogObserverWrapperTests.test_pythonLogLevelNotSetc C rM )a If a stdlib log level was provided as a string (eg. C{"WARNING"}) in the legacy "logLevel" key, it does not get converted to a number. The documentation suggested that numerical values should be used but this was not a requirement. �WARNING�rY rY NrO rP r r r �test_stringPythonLogLevel� s ��z7LegacyLogObserverWrapperTests.test_stringPythonLogLevelc C s | � t� �}| �|d d� dS )z� The old-style C{"message"} key is added, even if no new-style C{"log_format"} is given, as it is required, but may be empty. �messager NrO rP r r r � test_message� � z*LegacyLogObserverWrapperTests.test_messagec C rM )zV The old-style C{"message"} key is not modified if it already exists. rD )r_ r_ NrO rP r r r �test_messageAlreadySet� s z4LegacyLogObserverWrapperTests.test_messageAlreadySetc C �( | � tddd��}| �t�|�d� dS )z� Formatting is translated such that text is rendered correctly, even though old-style logging doesn't use PEP 3101 formatting. z Hello, {who}!�world)� log_format�who� Hello, world!N�rC r= r- r �textFromEventDictrP r r r � test_format� � z)LegacyLogObserverWrapperTests.test_formatc C rc )zo Using the message key, which is special in old-style, works for new-style formatting. zHello, {message}!rd )re r_ rg Nrh rP r r r �test_formatMessage� s �z0LegacyLogObserverWrapperTests.test_formatMessagec C s( | � tddd��}| �t�|�d� dS )zd Formatting is not altered if the old-style C{"format"} key already exists. zHello!zHowdy!)re �formatNrh rP r r r �test_formatAlreadySet� rk z3LegacyLogObserverWrapperTests.test_formatAlreadySet�valuesc K s& t td��}| �td|dd�|���S )z� Create a new-style event with a captured failure. @param values: Additional values to include in the event. @return: the new event znyargh!� oopsie...)� log_failurere Nr )r � RuntimeErrorrC r= )r ro �failurer r r �eventWithFailure� s �z.LegacyLogObserverWrapperTests.eventWithFailurec C s> | � � }| �|d |d � | �|d � | �|d d� dS )z} Captured failures in the new style set the old-style C{"failure"}, C{"isError"}, and C{"why"} keys. rs rq rT �whyrp N)rt �assertIs� assertTruer- rP r r r � test_failure� s z*LegacyLogObserverWrapperTests.test_failurec C s, t td��}| j|d�}| �|d |� dS )�v Captured failures in the new style do not step on a pre-existing old-style C{"failure"} key. z Weak salsa!)rs rs N)r rr rt rv )r rs r0 r r r �test_failureAlreadySet� s z4LegacyLogObserverWrapperTests.test_failureAlreadySetc C � | j dd�}| �|d d� dS )zv Captured failures in the new style do not step on a pre-existing old-style C{"isError"} key. r �rT rT N�rt r- rP r r r �test_isErrorAlreadySet ra z4LegacyLogObserverWrapperTests.test_isErrorAlreadySetc C r{ )ry �blah)ru ru Nr} rP r r r �test_whyAlreadySet ra z0LegacyLogObserverWrapperTests.test_whyAlreadySet�r N) r' r( r) �__doc__r! r/ r r6 rC rG rK rL rQ rR rS rX r[ r^ r` rb rj rl rn �objectrt rx rz r~ r� r r r r r s2 r c @ s� e Zd ZdZd dd�Zdededejfdd �Z d d d �Z d d d �Z d dd�Z d dd�Z d dd�Zd dd�Zd dd�Zd dd�Zd dd�Zd d d �ZdS ) �PublishToNewObserverTestsz, Tests for L{publishToNewObserver}. r Nc C s g | _ tt| j j�| _d S r )r3 r r r1 r r" r r r �setUp s z PublishToNewObserverTests.setUpr_ ro c O sD t �tj�pi �� }|�|� ||d< t� |d< d|vr d|d< |S )a Return a basic old-style event as would be created by L{legacyLog.msg}. @param message: a message event value in the legacy event format @param values: additional event values in the legacy event format @return: a legacy event r_ r rT r )r �getr � ILogContext�copy�updater )r r_ ro r0 r r r � legacyEvent s z%PublishToNewObserverTests.legacyEventc C s* t | j| �� tj� | �t| j�d� dS )z6 The observer is called exactly once. r4 N)r r r� r ri r- r5 r3 r" r r r � test_observed/ s �z'PublishToNewObserverTests.test_observedc C �8 t | j| �� tj� | �| jd d | jd d � dS )zc The old-style C{"time"} key is copied to the new-style C{"log_time"} key. r r7 r N�r r r� r ri r- r3 r" r r r rK 8 � �$z#PublishToNewObserverTests.test_timec C sN dt dtfdd�}| �dd�}||�}t| j||� | �t| jd �|� dS ) z� A published old-style event should format as text in the same way as the given C{textFromEventDict} callable would format it. r0 r c S s d� td� | d ���S )N� � r_ )�join�reversed)r0 r r r ri H s zAPublishToNewObserverTests.test_message.<locals>.textFromEventDictzHello,zworld!r N)r r* r� r r r- r r3 )r ri r0 �textr r r r` B s z&PublishToNewObserverTests.test_messagec C s0 t | j| �� tj� | �| jd d tj� dS )zL Published event should have log level of L{LogLevel.info}. r r: N) r r r� r ri r- r3 r r< r" r r r �test_defaultLogLevelQ s � z.PublishToNewObserverTests.test_defaultLogLevelc C �4 t | j| jdd�tj� | �| jd d tj� dS )z� If C{"isError"} is set to C{1} (true) on the legacy event, the C{"log_level"} key should get set to L{LogLevel.critical}. r4 r| r r: N) r r r� r ri r- r3 r rW r" r r r � test_isErrorZ s � z&PublishToNewObserverTests.test_isErrorc C s6 t | j| jtjd�tj� | �| jd d t j � dS )z� If the old-style C{"logLevel"} key is set to a standard library logging level, using a predefined (L{int}) constant, the new-style C{"log_level"} key should get set to the corresponding log level. r] r r: N) r r r� � py_loggingr\ r ri r- r3 r rU r" r r r �test_stdlibLogLeveld s � z-PublishToNewObserverTests.test_stdlibLogLevelc C r� )z� If the old-style C{"logLevel"} key is set to a standard library logging level, using a string value, the new-style C{"log_level"} key should get set to the corresponding log level. r\ r] r r: N) r r r� r ri r- r3 r rU r" r r r � test_stdlibLogLevelWithStringq s � z7PublishToNewObserverTests.test_stdlibLogLevelWithStringc C s. t | j| jdd�tj� | �d| jd � dS )z� If the old-style C{"logLevel"} key is set to a standard library logging level, using an unknown value, the new-style C{"log_level"} key should not get set. zFoo!!!!!r] r: r N)r r r� r ri rZ r3 r" r r r � test_stdlibLogLevelWithGarbage~ s �z8PublishToNewObserverTests.test_stdlibLogLevelWithGarbagec C s. t | j| �� tj� | �| jd d d� dS )z� Published event should have a namespace of C{"log_legacy"} to indicate that it was forwarded from legacy logging. r � log_namespace� log_legacyNr� r" r r r �test_defaultNamespace� s �z/PublishToNewObserverTests.test_defaultNamespacec C r� )zg The old-style C{"system"} key is copied to the new-style C{"log_system"} key. r r8 rN Nr� r" r r r rQ � r� z%PublishToNewObserverTests.test_systemr� )r' r( r) r� r� r* r� r r+ r� r� rK r` r� r� r� r� r� r� rQ r r r r r� s r� )"r� �loggingr� r �typingr r �zope.interfacer �zope.interface.exceptionsr �zope.interface.verifyr �twisted.pythonr r r �twisted.python.failurer � twisted.trialr �_formatr � _interfacesr r �_legacyr r �_levelsr �TestCaser r� r r r r �<module> s" }PK ͪ�ZDZ�J E E % __pycache__/test_json.cpython-310.pycnu �[��� o �bbG � @ s. d Z ddlmZmZ ddlmZmZmZmZm Z m Z ddl m Z ddl mZ ddlmZ ddlmZ ddlmZ d d lmZ d d lmZ d d lmZ d d lm Z m Z d dl m Z m Z m!Z!m"Z"m#Z$ d dl%m&Z& d dl'm(Z( d dl)m*Z* dede+de+fdd�Z,G dd� de�Z-G dd� de�Z.G dd � d e�Z/d S ) z$ Tests for L{twisted.logger._json}. � )�BytesIO�StringIO)�IO�Any�List�Optional�Sequence�cast)� implementer)�BrokenMethodImplementation)� verifyObject)�Failure)�TestCase� )� extractField)� formatEvent)�globalLogPublisher)� ILogObserver�LogEvent)� eventAsJSON� eventFromJSON�eventsFromJSONLogFile�jsonFileLogObserver�log)�LogLevel)�Logger)� LogPublisher�testCase� savedJSON�returnc C s" | � |t� | �|�d�d� |S )a* Assert a few things about the result of L{eventAsJSON}, then return it. @param testCase: The L{TestCase} with which to perform the assertions. @param savedJSON: The result of L{eventAsJSON}. @return: C{savedJSON} @raise AssertionError: If any of the preconditions fail. � r )�assertIsInstance�str� assertEqual�count)r r � r% �?/usr/lib/python3/dist-packages/twisted/logger/test/test_json.py�savedJSONInvariants! s r' c @ s� e Zd ZdZdedefdd�Zddd�Zdd d �Zdd d �Z dd d�Z ddd�Z ddd�Z ddd�Z ddd�Zddd�Zddd�ZdS ) � SaveLoadTestsz2 Tests for loading and saving log events. �eventr c C s t | t|��S )z� Serialize some an events, assert some things about it, and return the JSON. @param event: An event. @return: JSON. )r' r )�selfr) r% r% r& �savedEventJSON6 s z SaveLoadTests.savedEventJSONNc C s | � t| �i ��i � dS )zX Saving and loading an empty dictionary results in an empty dictionary. N�r# r r+ �r* r% r% r& �test_simpleSaveLoadA s z!SaveLoadTests.test_simpleSaveLoadc C s& | � t| �ddd���ddd�� dS )a Saving and loading a dictionary with some simple values in it results in those same simple values in the output; according to JSON's rules, though, all dictionary keys must be L{str} and any non-L{str} keys will be converted. r �4)� �3��1r1 Nr, r- r% r% r& � test_saveLoadG s �zSaveLoadTests.test_saveLoadc C �, | � t| �dt� d���dddid�� dS )zx Saving and loading an object which cannot be represented in JSON will result in a placeholder. r r2 � unpersistableTN�r# r r+ �objectr- r% r% r& �test_saveUnPersistableS s �z$SaveLoadTests.test_saveUnPersistablec C r5 )zD Non-ASCII keys and values can be saved and loaded. u 䌡)u ሴr1 r6 TNr7 r- r% r% r& �test_saveNonASCII] s �z SaveLoadTests.test_saveNonASCIIc C sJ dt td��i}|�ddi� | �t| �|��dt td���d�i� dS )zx Any L{bytes} objects will be saved as if they are latin-1 so they can be faithfully re-loaded. �hello� s skipped�okay�charmapN)�bytes�range�updater# r r+ �decode)r* � inputEventr% r% r& �test_saveBytesf s �z SaveLoadTests.test_saveBytesc C s> G dd� d�}d|d�d�}t | �|��}| �t|�d� dS )a Saving and loading an object which cannot be represented in JSON, but has a string representation which I{can} be saved as JSON, will result in the same string formatting; any extractable fields will retain their data types. c @ s, e Zd Zdeddfdd�Zdefdd�ZdS )z@SaveLoadTests.test_saveUnPersistableThenFormat.<locals>.Reprable�valuer Nc S s || _ d S �N�rE )r* rE r% r% r& �__init__~ � zISaveLoadTests.test_saveUnPersistableThenFormat.<locals>.Reprable.__init__c S s dS )N�reprabler% r- r% r% r& �__repr__� s zISaveLoadTests.test_saveUnPersistableThenFormat.<locals>.Reprable.__repr__)�__name__� __module__� __qualname__r8 rH r" rK r% r% r% r& �Reprable} s rO z{object} {object.value}� �� log_formatr8 z reprable 7N)r r+ r# r )r* rO rC � outputEventr% r% r& � test_saveUnPersistableThenFormatu s z.SaveLoadTests.test_saveUnPersistableThenFormatc C s` G dd� d�}t d|� d�}t| �|��}| �td|�d� | �ttd|� | �ttd|� dS ) zs L{extractField} can extract fields from an object that's been saved and loaded from JSON. c @ s e Zd Zddd�ZdS )z8SaveLoadTests.test_extractingFieldsPostLoad.<locals>.Objr Nc S s d| _ d S )N�Y rG r- r% r% r& rH � rI zASaveLoadTests.test_extractingFieldsPostLoad.<locals>.Obj.__init__�r N)rL rM rN rH r% r% r% r& �Obj� s rW z{object.value}rQ z object.valuerU r8 N)�dictr r+ r# r � assertRaises�KeyError)r* rW rC � loadedEventr% r% r& � test_extractingFieldsPostLoad� s z+SaveLoadTests.test_extractingFieldsPostLoadc C s� g }t tt|j�d�}zdd W n ty# t� }|�d|� Y nw | �t|�d� t | � |d ��d }| � |t� | � |� t�� | � |�� t� dS )zl Round-tripping a failure through L{eventAsJSON} preserves its class and structure. ��observerr0 r za message about failure� log_failureN)r r r �append�ZeroDivisionErrorr �failurer# �lenr r+ r! � assertTrue�check� getTracebackr" )r* �eventsr �f�loadedr% r% r& � test_failureStructurePreserved� s � z,SaveLoadTests.test_failureStructurePreservedc C s0 t tjd�}t| �|��}| �|d tj� dS )zo It's important that the C{log_level} key remain a L{constantly.NamedConstant} object. �� log_levelrl N)rX r �warnr r+ �assertIs)r* rC r[ r% r% r& �test_saveLoadLevel� s z SaveLoadTests.test_saveLoadLevelc C s t d�}| �|tdd�� dS )z� If a saved bit of JSON (let's say, from a future version of Twisted) were to persist a different log_level, it will resolve as None. zZ{"log_level": {"name": "other", "__class_uuid__": "02E59486-F24D-46AD-8224-3ACDF2A5732A"}}Nrk )r r# rX )r* r[ r% r% r& �test_saveLoadUnknownLevel� s �z'SaveLoadTests.test_saveLoadUnknownLevelrV )rL rM rN �__doc__r r" r+ r. r4 r9 r: rD rT r\ rj ro rp r% r% r% r& r( 1 s r( c @ sL e Zd ZdZddd�Zddeddfdd �Zdd d �Zdd d �Zddd�Z dS )�FileLogObserverTestsz+ Tests for L{jsonFileLogObserver}. r Nc C s� t � �4}t|�}ztt|� W n ty& } z | �|� W Y d}~n d}~ww W d � dS W d � dS 1 s:w Y dS )zh A L{FileLogObserver} returned by L{jsonFileLogObserver} is an L{ILogObserver}. N)r r r r r �fail)r* � fileHandler^ �er% r% r& �test_interface� s ����"�z#FileLogObserverTests.test_interface� �recordSeparatorc C s^ t � �"}t||�}tdd�}||� | �|�� |� d�� W d � dS 1 s(w Y dS )a! Asserts that an observer created by L{jsonFileLogObserver} with the given arguments writes events serialized as JSON text, using the given record separator. @param recordSeparator: C{recordSeparator} argument to L{jsonFileLogObserver} r0 ��xz {"x": 1} N)r r rX r# �getvalue)r* rx rt r^ r) r% r% r& �assertObserverWritesJSON� s "�z-FileLogObserverTests.assertObserverWritesJSONc C s | � � dS )z� A L{FileLogObserver} created by L{jsonFileLogObserver} writes events serialzed as JSON text to a file when it observes events. By default, the record separator is C{"\x1e"}. N�r| r- r% r% r& �(test_observeWritesDefaultRecordSeparator� s z=FileLogObserverTests.test_observeWritesDefaultRecordSeparatorc C s | j dd� dS )z� A L{FileLogObserver} created by L{jsonFileLogObserver} writes events serialzed as JSON text to a file when it observes events. This test sets the record separator to C{""}. � �rx Nr} r- r% r% r& �&test_observeWritesEmptyRecordSeparator� s z;FileLogObserverTests.test_observeWritesEmptyRecordSeparatorc s� t � }t� }g }|�tt|j�� |�t|�� t|d�}zdd W n ty1 |� d� Y nw t |� � �}t t |��}dt t ddf� fdd � }||� ||� dS ) z� A L{FileLogObserver} created by L{jsonFileLogObserver} writes failures serialized as JSON text to a file when it observes events. r] r0 r zfailed as expected� logEventsr Nc s` � � t| �d� | \}� �d|� |d }� �|t� |�� }� � |jjj� d�t � d�� d S )Nr0 r_ �co) r# rc �assertInr! r �getTracebackObject�tb_frame�f_code� co_filename�rstrip�__file__)r� � failureEvent� failureObject�tracebackObjectr- r% r& � checkEvents s �z@FileLogObserverTests.test_failureFormatting.<locals>.checkEvents)r r � addObserverr r r` r r � BaseExceptionrb r{ �listr r r )r* �io� publisher�logged�logger�reader� deserializedr� r% r- r& �test_failureFormatting� s � z+FileLogObserverTests.test_failureFormattingrV )rw ) rL rM rN rq rv r" r| r~ r� r� r% r% r% r& rr � s rr c @ s� e Zd ZdZd&dd�Zd&dd�Z d'd ee d ee d e ddfd d �Z d&dd�Z d&dd�Z d&dd�Zd&dd�Zd&dd�Zd&dd�Zd&dd�Zd&d d �Zd&d d �Zd&d d!�Zd&d"d#�Zd&d$d%�ZdS )(�LogFileReaderTestsz- Tests for L{eventsFromJSONLogFile}. r Nc s8 g � _ tt�dtdd f� fdd� �}|� _t�|� d S )Nr) r c s. | d t jkrd| v r� j�| � d S d S d S )N� log_namespace�record)�jsonLog� namespace� errorEventsr` )r) r- r% r& r^ s �z*LogFileReaderTests.setUp.<locals>.observer)r� r r r � logObserverr r� )r* r^ r% r- r& �setUp s zLogFileReaderTests.setUpc C s t �| j� d S rF )r �removeObserverr� r- r% r% r& �tearDown& s zLogFileReaderTests.tearDown� �inFilerx � bufferSizec C sJ t t|||��}| �t|�ddi� | �t|�ddi� | �tt|� dS )a| Test that L{eventsFromJSONLogFile} reads two pre-defined events from a file: C{{"x": 1}} and C{{"y": 2}}. @param inFile: C{inFile} argument to L{eventsFromJSONLogFile} @param recordSeparator: C{recordSeparator} argument to L{eventsFromJSONLogFile} @param bufferSize: C{bufferSize} argument to L{eventsFromJSONLogFile} rz r0 �yr N)�iterr r# �nextrY � StopIteration)r* r� rx r� rg r% r% r& � _readEvents) s z LogFileReaderTests._readEventsc C �J t d��}| �|� | �t| j�d� W d � dS 1 s w Y dS )z� L{eventsFromJSONLogFile} reads events from a file and automatically detects use of C{"\x1e"} as the record separator. � {"x": 1} {"y": 2} r N�r r� r# rc r� �r* rt r% r% r& �&test_readEventsAutoWithRecordSeparator> � "�z9LogFileReaderTests.test_readEventsAutoWithRecordSeparatorc C r� )z� L{eventsFromJSONLogFile} reads events from a file and automatically detects use of C{""} as the record separator. z{"x": 1} {"y": 2} r Nr� r� r% r% r& �'test_readEventsAutoEmptyRecordSeparatorG r� z:LogFileReaderTests.test_readEventsAutoEmptyRecordSeparatorc C �N t d��}| j|dd� | �t| j�d� W d � dS 1 s w Y dS )z{ L{eventsFromJSONLogFile} reads events from a file and is told to use a specific record separator. z{"x": 1} {"y": 2} �r� r Nr� r� r% r% r& �&test_readEventsExplicitRecordSeparatorP s "�z9LogFileReaderTests.test_readEventsExplicitRecordSeparatorc C r� )zM L{eventsFromJSONLogFile} handles buffering a partial event. r� r0 �r� r Nr� r� r% r% r& � test_readEventsPartialBufferZ s "�z/LogFileReaderTests.test_readEventsPartialBufferc C � t d��?}tt|��}| �t|�ddi� | �tt|� | �t| j�d� | �| jd d d� | �| jd d d � W d � d S 1 sFw Y d S ) zF If the JSON text for a record is truncated, skip it. z {"x": 1 {"y": 2} r� r r0 r rR z0Unable to read truncated JSON record: {record!r}r� s {"x": 1N� r r� r r# r� rY r� rc r� �r* rt rg r% r% r& �test_readTruncatedc � �"�z%LogFileReaderTests.test_readTruncatedc C �n t d��)}tt|��}| �t|�ddi� | �tt|� | �t| j�d� W d � dS 1 s0w Y dS )zc If the file being read from vends L{str}, strings decode from JSON as-is. u {"currency": "€"} �currency� €r Nr� r� r% r% r& �test_readUnicodeu s "�z#LogFileReaderTests.test_readUnicodec C r� )zh If the file being read from vends L{bytes}, strings decode from JSON as UTF-8. s {"currency": "€"} r� r� r N� r r� r r# r� rY r� rc r� r� r% r% r& �test_readUTF8Bytes� s "�z%LogFileReaderTests.test_readUTF8Bytesc C sr t d��+}tt|dd��}| �t|�ddi� | �tt|� | �t| j�d� W d � dS 1 s2w Y dS )z� If the JSON text for a record is truncated in the middle of a two-byte Unicode codepoint, we don't want to see a codec exception and the stream is read properly when the additional data arrives. s {"x": "€"} � r� rz r� r Nr� r� r% r% r& �test_readTruncatedUTF8Bytes� s "�z.LogFileReaderTests.test_readTruncatedUTF8Bytesc C r� ) zh If the JSON text for a record contains invalid UTF-8 text, ignore that record. s {"x": "�"} {"y": 2} r� r r0 r rR z2Unable to decode UTF-8 for JSON record: {record!r}r� s {"x": "�"} Nr� r� r% r% r& �test_readInvalidUTF8Bytes� s �"�z,LogFileReaderTests.test_readInvalidUTF8Bytesc C r� ) zD If the JSON text for a record is invalid, skip it. z {"x": } {"y": 2} r� r r0 r rR �&Unable to read JSON record: {record!r}r� s {"x": } Nr� r� r% r% r& �test_readInvalidJSON� r� z'LogFileReaderTests.test_readInvalidJSONc C s� t d��3}t|�}| �tt|� | �t| j�d� | �| jd d d� | �| jd d d� W d � dS 1 s:w Y dS ) zI Multiple events without a record separator are skipped. z {"x": 1} {"y": 2} r0 r rR r� r� s {"x": 1} {"y": 2} N)r r rY r� r� r# rc r� r� r% r% r& �test_readUnseparated� s �"�z'LogFileReaderTests.test_readUnseparatedc C s| t dd�}t� �,}t|�}||� |�d� t|�}| �t|�|f� | �t| j�d� W d � dS 1 s7w Y dS )z� Data written by a L{FileLogObserver} returned by L{jsonFileLogObserver} and read by L{eventsFromJSONLogFile} is reconstructed properly. r0 ry r N) rX r r �seekr r# �tuplerc r� )r* r) rt r^ rg r% r% r& �test_roundTrip� s "�z!LogFileReaderTests.test_roundTriprV )Nr� )rL rM rN rq r� r� r r r r" �intr� r� r� r� r� r� r� r� r� r� r� r� r� r% r% r% r& r� s6 ���� � r� N)0rq r� r r �typingr r r r r r �zope.interfacer �zope.interface.exceptionsr �zope.interface.verifyr �twisted.python.failurer �twisted.trial.unittestr �_flattenr �_formatr �_globalr � _interfacesr r �_jsonr r r r r r� �_levelsr �_loggerr � _observerr r" r' r( rr r� r% r% r% r& �<module> s* RPK ͪ�Z! XPU U test_format.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._format}. """ from typing import AnyStr, Optional, cast try: from time import tzset # We should upgrade to a version of pyflakes that does not require this. tzset except ImportError: tzset = None # type: ignore[assignment] from twisted.python.failure import Failure from twisted.python.test.test_tzhelper import addTZCleanup, mktime, setTZ from twisted.trial import unittest from twisted.trial.unittest import SkipTest from .._format import ( eventAsText, formatEvent, formatEventAsClassicLogText, formatTime, formatUnformattableEvent, formatWithCall, ) from .._interfaces import LogEvent from .._levels import LogLevel class FormattingTests(unittest.TestCase): """ Tests for basic event formatting functions. """ def test_formatEvent(self) -> None: """ L{formatEvent} will format an event according to several rules: - A string with no formatting instructions will be passed straight through. - PEP 3101 strings will be formatted using the keys and values of the event as named fields. - PEP 3101 keys ending with C{()} will be treated as instructions to call that key (which ought to be a callable) before formatting. L{formatEvent} will always return L{str}, and if given bytes, will always treat its format string as UTF-8 encoded. """ def format(logFormat: AnyStr, **event: object) -> str: event["log_format"] = logFormat result = formatEvent(event) self.assertIs(type(result), str) return result self.assertEqual("", format(b"")) self.assertEqual("", format("")) self.assertEqual("abc", format("{x}", x="abc")) self.assertEqual( "no, yes.", format("{not_called}, {called()}.", not_called="no", called=lambda: "yes"), ) self.assertEqual("S\xe1nchez", format(b"S\xc3\xa1nchez")) self.assertIn("Unable to format event", format(b"S\xe1nchez")) maybeResult = format(b"S{a!s}nchez", a=b"\xe1") self.assertIn("Sb'\\xe1'nchez", maybeResult) xe1 = str(repr(b"\xe1")) self.assertIn("S" + xe1 + "nchez", format(b"S{a!r}nchez", a=b"\xe1")) def test_formatEventNoFormat(self) -> None: """ Formatting an event with no format. """ event = dict(foo=1, bar=2) result = formatEvent(event) self.assertEqual("", result) def test_formatEventWeirdFormat(self) -> None: """ Formatting an event with a bogus format. """ event = dict(log_format=object(), foo=1, bar=2) result = formatEvent(event) self.assertIn("Log format must be str", result) self.assertIn(repr(event), result) def test_formatUnformattableEvent(self) -> None: """ Formatting an event that's just plain out to get us. """ event = dict(log_format="{evil()}", evil=lambda: 1 / 0) result = formatEvent(event) self.assertIn("Unable to format event", result) self.assertIn(repr(event), result) def test_formatUnformattableEventWithUnformattableKey(self) -> None: """ Formatting an unformattable event that has an unformattable key. """ event: LogEvent = { "log_format": "{evil()}", "evil": lambda: 1 / 0, cast(str, Unformattable()): "gurk", } result = formatEvent(event) self.assertIn("MESSAGE LOST: unformattable object logged:", result) self.assertIn("Recoverable data:", result) self.assertIn("Exception during formatting:", result) def test_formatUnformattableEventWithUnformattableValue(self) -> None: """ Formatting an unformattable event that has an unformattable value. """ event = dict( log_format="{evil()}", evil=lambda: 1 / 0, gurk=Unformattable(), ) result = formatEvent(event) self.assertIn("MESSAGE LOST: unformattable object logged:", result) self.assertIn("Recoverable data:", result) self.assertIn("Exception during formatting:", result) def test_formatUnformattableEventWithUnformattableErrorOMGWillItStop(self) -> None: """ Formatting an unformattable event that has an unformattable value. """ event = dict( log_format="{evil()}", evil=lambda: 1 / 0, recoverable="okay", ) # Call formatUnformattableEvent() directly with a bogus exception. result = formatUnformattableEvent(event, cast(BaseException, Unformattable())) self.assertIn("MESSAGE LOST: unformattable object logged:", result) self.assertIn(repr("recoverable") + " = " + repr("okay"), result) class TimeFormattingTests(unittest.TestCase): """ Tests for time formatting functions. """ def setUp(self) -> None: addTZCleanup(self) def test_formatTimeWithDefaultFormat(self) -> None: """ Default time stamp format is RFC 3339 and offset respects the timezone as set by the standard C{TZ} environment variable and L{tzset} API. """ if tzset is None: raise SkipTest("Platform cannot change timezone; unable to verify offsets.") def testForTimeZone(name: str, expectedDST: str, expectedSTD: str) -> None: setTZ(name) localDST = mktime((2006, 6, 30, 0, 0, 0, 4, 181, 1)) localSTD = mktime((2007, 1, 31, 0, 0, 0, 2, 31, 0)) self.assertEqual(formatTime(localDST), expectedDST) self.assertEqual(formatTime(localSTD), expectedSTD) # UTC testForTimeZone( "UTC+00", "2006-06-30T00:00:00+0000", "2007-01-31T00:00:00+0000", ) # West of UTC testForTimeZone( "EST+05EDT,M4.1.0,M10.5.0", "2006-06-30T00:00:00-0400", "2007-01-31T00:00:00-0500", ) # East of UTC testForTimeZone( "CEST-01CEDT,M4.1.0,M10.5.0", "2006-06-30T00:00:00+0200", "2007-01-31T00:00:00+0100", ) # No DST testForTimeZone( "CST+06", "2006-06-30T00:00:00-0600", "2007-01-31T00:00:00-0600", ) def test_formatTimeWithNoTime(self) -> None: """ If C{when} argument is L{None}, we get the default output. """ self.assertEqual(formatTime(None), "-") self.assertEqual(formatTime(None, default="!"), "!") def test_formatTimeWithNoFormat(self) -> None: """ If C{timeFormat} argument is L{None}, we get the default output. """ t = mktime((2013, 9, 24, 11, 40, 47, 1, 267, 1)) self.assertEqual(formatTime(t, timeFormat=None), "-") self.assertEqual(formatTime(t, timeFormat=None, default="!"), "!") def test_formatTimeWithAlternateTimeFormat(self) -> None: """ Alternate time format in output. """ t = mktime((2013, 9, 24, 11, 40, 47, 1, 267, 1)) self.assertEqual(formatTime(t, timeFormat="%Y/%W"), "2013/38") def test_formatTimePercentF(self) -> None: """ "%f" supported in time format. """ self.assertEqual(formatTime(1000000.23456, timeFormat="%f"), "234560") class ClassicLogFormattingTests(unittest.TestCase): """ Tests for classic text log event formatting functions. """ def test_formatTimeDefault(self) -> None: """ Time is first field. Default time stamp format is RFC 3339 and offset respects the timezone as set by the standard C{TZ} environment variable and L{tzset} API. """ if tzset is None: raise SkipTest("Platform cannot change timezone; unable to verify offsets.") addTZCleanup(self) setTZ("UTC+00") t = mktime((2013, 9, 24, 11, 40, 47, 1, 267, 1)) event = dict(log_format="XYZZY", log_time=t) self.assertEqual( formatEventAsClassicLogText(event), "2013-09-24T11:40:47+0000 [-\x23-] XYZZY\n", ) def test_formatTimeCustom(self) -> None: """ Time is first field. Custom formatting function is an optional argument. """ def formatTime(t: Optional[float]) -> str: return f"__{t}__" event = dict(log_format="XYZZY", log_time=12345) self.assertEqual( formatEventAsClassicLogText(event, formatTime=formatTime), "__12345__ [-\x23-] XYZZY\n", ) def test_formatNamespace(self) -> None: """ Namespace is first part of second field. """ event = dict(log_format="XYZZY", log_namespace="my.namespace") self.assertEqual( formatEventAsClassicLogText(event), "- [my.namespace\x23-] XYZZY\n", ) def test_formatLevel(self) -> None: """ Level is second part of second field. """ event = dict(log_format="XYZZY", log_level=LogLevel.warn) self.assertEqual( formatEventAsClassicLogText(event), "- [-\x23warn] XYZZY\n", ) def test_formatSystem(self) -> None: """ System is second field. """ event = dict(log_format="XYZZY", log_system="S.Y.S.T.E.M.") self.assertEqual( formatEventAsClassicLogText(event), "- [S.Y.S.T.E.M.] XYZZY\n", ) def test_formatSystemRulz(self) -> None: """ System is not supplanted by namespace and level. """ event = dict( log_format="XYZZY", log_namespace="my.namespace", log_level=LogLevel.warn, log_system="S.Y.S.T.E.M.", ) self.assertEqual( formatEventAsClassicLogText(event), "- [S.Y.S.T.E.M.] XYZZY\n", ) def test_formatSystemUnformattable(self) -> None: """ System is not supplanted by namespace and level. """ event = dict(log_format="XYZZY", log_system=Unformattable()) self.assertEqual( formatEventAsClassicLogText(event), "- [UNFORMATTABLE] XYZZY\n", ) def test_formatFormat(self) -> None: """ Formatted event is last field. """ event = dict(log_format="id:{id}", id="123") self.assertEqual( formatEventAsClassicLogText(event), "- [-\x23-] id:123\n", ) def test_formatNoFormat(self) -> None: """ No format string. """ event = dict(id="123") self.assertIs(formatEventAsClassicLogText(event), None) def test_formatEmptyFormat(self) -> None: """ Empty format string. """ event = dict(log_format="", id="123") self.assertIs(formatEventAsClassicLogText(event), None) def test_formatFormatMultiLine(self) -> None: """ If the formatted event has newlines, indent additional lines. """ event = dict(log_format='XYZZY\nA hollow voice says:\n"Plugh"') self.assertEqual( formatEventAsClassicLogText(event), '- [-\x23-] XYZZY\n\tA hollow voice says:\n\t"Plugh"\n', ) class FormatFieldTests(unittest.TestCase): """ Tests for format field functions. """ def test_formatWithCall(self) -> None: """ L{formatWithCall} is an extended version of L{str.format} that will interpret a set of parentheses "C{()}" at the end of a format key to mean that the format key ought to be I{called} rather than stringified. """ self.assertEqual( formatWithCall( "Hello, {world}. {callme()}.", dict(world="earth", callme=lambda: "maybe"), ), "Hello, earth. maybe.", ) self.assertEqual( formatWithCall("Hello, {repr()!r}.", dict(repr=lambda: "repr")), "Hello, 'repr'.", ) class Unformattable: """ An object that raises an exception from C{__repr__}. """ def __repr__(self) -> str: return str(1 / 0) class CapturedError(Exception): """ A captured error for use in format tests. """ class EventAsTextTests(unittest.TestCase): """ Tests for L{eventAsText}, all of which ensure that the returned type is UTF-8 decoded text. """ def test_eventWithTraceback(self) -> None: """ An event with a C{log_failure} key will have a traceback appended. """ try: raise CapturedError("This is a fake error") except CapturedError: f = Failure() event: LogEvent = {"log_format": "This is a test log message"} event["log_failure"] = f eventText = eventAsText(event, includeTimestamp=True, includeSystem=False) self.assertIn(str(f.getTraceback()), eventText) self.assertIn("This is a test log message", eventText) def test_formatEmptyEventWithTraceback(self) -> None: """ An event with an empty C{log_format} key appends a traceback from the accompanying failure. """ try: raise CapturedError("This is a fake error") except CapturedError: f = Failure() event: LogEvent = {"log_format": ""} event["log_failure"] = f eventText = eventAsText(event, includeTimestamp=True, includeSystem=False) self.assertIn(str(f.getTraceback()), eventText) self.assertIn("This is a fake error", eventText) def test_formatUnformattableWithTraceback(self) -> None: """ An event with an unformattable value in the C{log_format} key still has a traceback appended. """ try: raise CapturedError("This is a fake error") except CapturedError: f = Failure() event = { "log_format": "{evil()}", "evil": lambda: 1 / 0, } event["log_failure"] = f eventText = eventAsText(event, includeTimestamp=True, includeSystem=False) self.assertIsInstance(eventText, str) self.assertIn(str(f.getTraceback()), eventText) self.assertIn("This is a fake error", eventText) def test_formatUnformattableErrorWithTraceback(self) -> None: """ An event with an unformattable value in the C{log_format} key, that throws an exception when __repr__ is invoked still has a traceback appended. """ try: raise CapturedError("This is a fake error") except CapturedError: f = Failure() event: LogEvent = { "log_format": "{evil()}", "evil": lambda: 1 / 0, cast(str, Unformattable()): "gurk", } event["log_failure"] = f eventText = eventAsText(event, includeTimestamp=True, includeSystem=False) self.assertIsInstance(eventText, str) self.assertIn("MESSAGE LOST", eventText) self.assertIn(str(f.getTraceback()), eventText) self.assertIn("This is a fake error", eventText) def test_formatEventUnformattableTraceback(self) -> None: """ If a traceback cannot be appended, a message indicating this is true is appended. """ event: LogEvent = {"log_format": ""} event["log_failure"] = object() eventText = eventAsText(event, includeTimestamp=True, includeSystem=False) self.assertIsInstance(eventText, str) self.assertIn("(UNABLE TO OBTAIN TRACEBACK FROM EVENT)", eventText) def test_formatEventNonCritical(self) -> None: """ An event with no C{log_failure} key will not have a traceback appended. """ event: LogEvent = {"log_format": "This is a test log message"} eventText = eventAsText(event, includeTimestamp=True, includeSystem=False) self.assertIsInstance(eventText, str) self.assertIn("This is a test log message", eventText) def test_formatTracebackMultibyte(self) -> None: """ An exception message with multibyte characters is properly handled. """ try: raise CapturedError("€") except CapturedError: f = Failure() event: LogEvent = {"log_format": "This is a test log message"} event["log_failure"] = f eventText = eventAsText(event, includeTimestamp=True, includeSystem=False) self.assertIn("€", eventText) self.assertIn("Traceback", eventText) def test_formatTracebackHandlesUTF8DecodeFailure(self) -> None: """ An error raised attempting to decode the UTF still produces a valid log message. """ try: # 'test' in utf-16 raise CapturedError(b"\xff\xfet\x00e\x00s\x00t\x00") except CapturedError: f = Failure() event: LogEvent = {"log_format": "This is a test log message"} event["log_failure"] = f eventText = eventAsText(event, includeTimestamp=True, includeSystem=False) self.assertIn("Traceback", eventText) self.assertIn(r'CapturedError(b"\xff\xfet\x00e\x00s\x00t\x00")', eventText) def test_eventAsTextSystemOnly(self) -> None: """ If includeSystem is specified as the only option no timestamp or traceback are printed. """ try: raise CapturedError("This is a fake error") except CapturedError: f = Failure() t = mktime((2013, 9, 24, 11, 40, 47, 1, 267, 1)) event: LogEvent = { "log_format": "ABCD", "log_system": "fake_system", "log_time": t, } event["log_failure"] = f eventText = eventAsText( event, includeTimestamp=False, includeTraceback=False, includeSystem=True, ) self.assertEqual( eventText, "[fake_system] ABCD", ) def test_eventAsTextTimestampOnly(self) -> None: """ If includeTimestamp is specified as the only option no system or traceback are printed. """ if tzset is None: raise SkipTest("Platform cannot change timezone; unable to verify offsets.") addTZCleanup(self) setTZ("UTC+00") try: raise CapturedError("This is a fake error") except CapturedError: f = Failure() t = mktime((2013, 9, 24, 11, 40, 47, 1, 267, 1)) event: LogEvent = { "log_format": "ABCD", "log_system": "fake_system", "log_time": t, } event["log_failure"] = f eventText = eventAsText( event, includeTimestamp=True, includeTraceback=False, includeSystem=False, ) self.assertEqual( eventText, "2013-09-24T11:40:47+0000 ABCD", ) def test_eventAsTextSystemMissing(self) -> None: """ If includeSystem is specified with a missing system [-#-] is used. """ try: raise CapturedError("This is a fake error") except CapturedError: f = Failure() t = mktime((2013, 9, 24, 11, 40, 47, 1, 267, 1)) event: LogEvent = { "log_format": "ABCD", "log_time": t, } event["log_failure"] = f eventText = eventAsText( event, includeTimestamp=False, includeTraceback=False, includeSystem=True, ) self.assertEqual( eventText, "[-\x23-] ABCD", ) def test_eventAsTextSystemMissingNamespaceAndLevel(self) -> None: """ If includeSystem is specified with a missing system but namespace and level are present they are used. """ try: raise CapturedError("This is a fake error") except CapturedError: f = Failure() t = mktime((2013, 9, 24, 11, 40, 47, 1, 267, 1)) event: LogEvent = { "log_format": "ABCD", "log_time": t, "log_level": LogLevel.info, "log_namespace": "test", } event["log_failure"] = f eventText = eventAsText( event, includeTimestamp=False, includeTraceback=False, includeSystem=True, ) self.assertEqual( eventText, "[test\x23info] ABCD", ) def test_eventAsTextSystemMissingLevelOnly(self) -> None: """ If includeSystem is specified with a missing system but level is present, level is included. """ try: raise CapturedError("This is a fake error") except CapturedError: f = Failure() t = mktime((2013, 9, 24, 11, 40, 47, 1, 267, 1)) event: LogEvent = { "log_format": "ABCD", "log_time": t, "log_level": LogLevel.info, } event["log_failure"] = f eventText = eventAsText( event, includeTimestamp=False, includeTraceback=False, includeSystem=True, ) self.assertEqual( eventText, "[-\x23info] ABCD", ) PK ͪ�Z���� 9 9 test_legacy.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._legacy}. """ import logging as py_logging from time import time from typing import List, cast from zope.interface import implementer from zope.interface.exceptions import BrokenMethodImplementation from zope.interface.verify import verifyObject from twisted.python import context, log as legacyLog from twisted.python.failure import Failure from twisted.trial import unittest from .._format import formatEvent from .._interfaces import ILogObserver, LogEvent from .._legacy import LegacyLogObserverWrapper, publishToNewObserver from .._levels import LogLevel class LegacyLogObserverWrapperTests(unittest.TestCase): """ Tests for L{LegacyLogObserverWrapper}. """ def test_interface(self) -> None: """ L{LegacyLogObserverWrapper} is an L{ILogObserver}. """ legacyObserver = cast(legacyLog.ILogObserver, lambda e: None) observer = LegacyLogObserverWrapper(legacyObserver) try: verifyObject(ILogObserver, observer) except BrokenMethodImplementation as e: self.fail(e) def test_repr(self) -> None: """ L{LegacyLogObserverWrapper} returns the expected string. """ @implementer(legacyLog.ILogObserver) class LegacyObserver: def __repr__(self) -> str: return "<Legacy Observer>" def __call__(self, eventDict: legacyLog.EventDict) -> None: return observer = LegacyLogObserverWrapper(LegacyObserver()) self.assertEqual(repr(observer), "LegacyLogObserverWrapper(<Legacy Observer>)") def observe(self, event: LogEvent) -> LogEvent: """ Send an event to a wrapped legacy observer and capture the event as seen by that observer. @param event: an event @return: the event as observed by the legacy wrapper """ events: List[LogEvent] = [] legacyObserver = cast(legacyLog.ILogObserver, lambda e: events.append(e)) observer = LegacyLogObserverWrapper(legacyObserver) observer(event) self.assertEqual(len(events), 1) return events[0] def forwardAndVerify(self, event: LogEvent) -> LogEvent: """ Send an event to a wrapped legacy observer and verify that its data is preserved. @param event: an event @return: the event as observed by the legacy wrapper """ # Make sure keys that are expected by the logging system are present event.setdefault("log_time", time()) event.setdefault("log_system", "-") event.setdefault("log_level", LogLevel.info) # Send a copy: don't mutate me, bro observed = self.observe(dict(event)) # Don't expect modifications for key, value in event.items(): self.assertIn(key, observed) return observed def test_forward(self) -> None: """ Basic forwarding: event keys as observed by a legacy observer are the same. """ self.forwardAndVerify(dict(foo=1, bar=2)) def test_time(self) -> None: """ The new-style C{"log_time"} key is copied to the old-style C{"time"} key. """ stamp = time() event = self.forwardAndVerify(dict(log_time=stamp)) self.assertEqual(event["time"], stamp) def test_timeAlreadySet(self) -> None: """ The new-style C{"log_time"} key does not step on a pre-existing old-style C{"time"} key. """ stamp = time() event = self.forwardAndVerify(dict(log_time=stamp + 1, time=stamp)) self.assertEqual(event["time"], stamp) def test_system(self) -> None: """ The new-style C{"log_system"} key is copied to the old-style C{"system"} key. """ event = self.forwardAndVerify(dict(log_system="foo")) self.assertEqual(event["system"], "foo") def test_systemAlreadySet(self) -> None: """ The new-style C{"log_system"} key does not step on a pre-existing old-style C{"system"} key. """ event = self.forwardAndVerify(dict(log_system="foo", system="bar")) self.assertEqual(event["system"], "bar") def test_noSystem(self) -> None: """ If the new-style C{"log_system"} key is absent, the old-style C{"system"} key is set to C{"-"}. """ # Don't use forwardAndVerify(), since that sets log_system. event = dict(log_time=time(), log_level=LogLevel.info) observed = self.observe(dict(event)) self.assertEqual(observed["system"], "-") def test_levelNotChange(self) -> None: """ If explicitly set, the C{isError} key will be preserved when forwarding from a new-style logging emitter to a legacy logging observer, regardless of log level. """ self.forwardAndVerify(dict(log_level=LogLevel.info, isError=1)) self.forwardAndVerify(dict(log_level=LogLevel.warn, isError=1)) self.forwardAndVerify(dict(log_level=LogLevel.error, isError=0)) self.forwardAndVerify(dict(log_level=LogLevel.critical, isError=0)) def test_pythonLogLevelNotSet(self) -> None: """ The new-style C{"log_level"} key is not translated to the old-style C{"logLevel"} key. Events are forwarded from the old module from to new module and are then seen by old-style observers. We don't want to add unexpected keys to old-style events. """ event = self.forwardAndVerify(dict(log_level=LogLevel.info)) self.assertNotIn("logLevel", event) def test_stringPythonLogLevel(self) -> None: """ If a stdlib log level was provided as a string (eg. C{"WARNING"}) in the legacy "logLevel" key, it does not get converted to a number. The documentation suggested that numerical values should be used but this was not a requirement. """ event = self.forwardAndVerify( dict( logLevel="WARNING", # py_logging.WARNING is 30 ) ) self.assertEqual(event["logLevel"], "WARNING") def test_message(self) -> None: """ The old-style C{"message"} key is added, even if no new-style C{"log_format"} is given, as it is required, but may be empty. """ event = self.forwardAndVerify(dict()) self.assertEqual(event["message"], ()) # "message" is a tuple def test_messageAlreadySet(self) -> None: """ The old-style C{"message"} key is not modified if it already exists. """ event = self.forwardAndVerify(dict(message=("foo", "bar"))) self.assertEqual(event["message"], ("foo", "bar")) def test_format(self) -> None: """ Formatting is translated such that text is rendered correctly, even though old-style logging doesn't use PEP 3101 formatting. """ event = self.forwardAndVerify(dict(log_format="Hello, {who}!", who="world")) self.assertEqual(legacyLog.textFromEventDict(event), "Hello, world!") def test_formatMessage(self) -> None: """ Using the message key, which is special in old-style, works for new-style formatting. """ event = self.forwardAndVerify( dict(log_format="Hello, {message}!", message="world") ) self.assertEqual(legacyLog.textFromEventDict(event), "Hello, world!") def test_formatAlreadySet(self) -> None: """ Formatting is not altered if the old-style C{"format"} key already exists. """ event = self.forwardAndVerify(dict(log_format="Hello!", format="Howdy!")) self.assertEqual(legacyLog.textFromEventDict(event), "Howdy!") def eventWithFailure(self, **values: object) -> LogEvent: """ Create a new-style event with a captured failure. @param values: Additional values to include in the event. @return: the new event """ failure = Failure(RuntimeError("nyargh!")) return self.forwardAndVerify( dict(log_failure=failure, log_format="oopsie...", **values) ) def test_failure(self) -> None: """ Captured failures in the new style set the old-style C{"failure"}, C{"isError"}, and C{"why"} keys. """ event = self.eventWithFailure() self.assertIs(event["failure"], event["log_failure"]) self.assertTrue(event["isError"]) self.assertEqual(event["why"], "oopsie...") def test_failureAlreadySet(self) -> None: """ Captured failures in the new style do not step on a pre-existing old-style C{"failure"} key. """ failure = Failure(RuntimeError("Weak salsa!")) event = self.eventWithFailure(failure=failure) self.assertIs(event["failure"], failure) def test_isErrorAlreadySet(self) -> None: """ Captured failures in the new style do not step on a pre-existing old-style C{"isError"} key. """ event = self.eventWithFailure(isError=0) self.assertEqual(event["isError"], 0) def test_whyAlreadySet(self) -> None: """ Captured failures in the new style do not step on a pre-existing old-style C{"failure"} key. """ event = self.eventWithFailure(why="blah") self.assertEqual(event["why"], "blah") class PublishToNewObserverTests(unittest.TestCase): """ Tests for L{publishToNewObserver}. """ def setUp(self) -> None: self.events: List[LogEvent] = [] self.observer = cast(ILogObserver, self.events.append) def legacyEvent(self, *message: str, **values: object) -> legacyLog.EventDict: """ Return a basic old-style event as would be created by L{legacyLog.msg}. @param message: a message event value in the legacy event format @param values: additional event values in the legacy event format @return: a legacy event """ event = (context.get(legacyLog.ILogContext) or {}).copy() event.update(values) event["message"] = message event["time"] = time() if "isError" not in event: event["isError"] = 0 return event def test_observed(self) -> None: """ The observer is called exactly once. """ publishToNewObserver( self.observer, self.legacyEvent(), legacyLog.textFromEventDict ) self.assertEqual(len(self.events), 1) def test_time(self) -> None: """ The old-style C{"time"} key is copied to the new-style C{"log_time"} key. """ publishToNewObserver( self.observer, self.legacyEvent(), legacyLog.textFromEventDict ) self.assertEqual(self.events[0]["log_time"], self.events[0]["time"]) def test_message(self) -> None: """ A published old-style event should format as text in the same way as the given C{textFromEventDict} callable would format it. """ def textFromEventDict(event: LogEvent) -> str: return "".join(reversed(" ".join(event["message"]))) event = self.legacyEvent("Hello,", "world!") text = textFromEventDict(event) publishToNewObserver(self.observer, event, textFromEventDict) self.assertEqual(formatEvent(self.events[0]), text) def test_defaultLogLevel(self) -> None: """ Published event should have log level of L{LogLevel.info}. """ publishToNewObserver( self.observer, self.legacyEvent(), legacyLog.textFromEventDict ) self.assertEqual(self.events[0]["log_level"], LogLevel.info) def test_isError(self) -> None: """ If C{"isError"} is set to C{1} (true) on the legacy event, the C{"log_level"} key should get set to L{LogLevel.critical}. """ publishToNewObserver( self.observer, self.legacyEvent(isError=1), legacyLog.textFromEventDict ) self.assertEqual(self.events[0]["log_level"], LogLevel.critical) def test_stdlibLogLevel(self) -> None: """ If the old-style C{"logLevel"} key is set to a standard library logging level, using a predefined (L{int}) constant, the new-style C{"log_level"} key should get set to the corresponding log level. """ publishToNewObserver( self.observer, self.legacyEvent(logLevel=py_logging.WARNING), legacyLog.textFromEventDict, ) self.assertEqual(self.events[0]["log_level"], LogLevel.warn) def test_stdlibLogLevelWithString(self) -> None: """ If the old-style C{"logLevel"} key is set to a standard library logging level, using a string value, the new-style C{"log_level"} key should get set to the corresponding log level. """ publishToNewObserver( self.observer, self.legacyEvent(logLevel="WARNING"), legacyLog.textFromEventDict, ) self.assertEqual(self.events[0]["log_level"], LogLevel.warn) def test_stdlibLogLevelWithGarbage(self) -> None: """ If the old-style C{"logLevel"} key is set to a standard library logging level, using an unknown value, the new-style C{"log_level"} key should not get set. """ publishToNewObserver( self.observer, self.legacyEvent(logLevel="Foo!!!!!"), legacyLog.textFromEventDict, ) self.assertNotIn("log_level", self.events[0]) def test_defaultNamespace(self) -> None: """ Published event should have a namespace of C{"log_legacy"} to indicate that it was forwarded from legacy logging. """ publishToNewObserver( self.observer, self.legacyEvent(), legacyLog.textFromEventDict ) self.assertEqual(self.events[0]["log_namespace"], "log_legacy") def test_system(self) -> None: """ The old-style C{"system"} key is copied to the new-style C{"log_system"} key. """ publishToNewObserver( self.observer, self.legacyEvent(), legacyLog.textFromEventDict ) self.assertEqual(self.events[0]["log_system"], self.events[0]["system"]) PK ͪ�Z��Is�! �! test_stdlib.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._format}. """ import logging as py_logging import sys from inspect import getsourcefile from io import BytesIO, TextIOWrapper from logging import Formatter, LogRecord, StreamHandler, getLogger from typing import List, Optional, Tuple from zope.interface.exceptions import BrokenMethodImplementation from zope.interface.verify import verifyObject from twisted.python.compat import currentframe from twisted.python.failure import Failure from twisted.trial import unittest from .._interfaces import ILogObserver, LogEvent from .._levels import LogLevel from .._stdlib import STDLibLogObserver def nextLine() -> Tuple[Optional[str], int]: """ Retrive the file name and line number immediately after where this function is called. @return: the file name and line number """ caller = currentframe(1) return ( getsourcefile(sys.modules[caller.f_globals["__name__"]]), caller.f_lineno + 1, ) class StdlibLoggingContainer: """ Continer for a test configuration of stdlib logging objects. """ def __init__(self) -> None: self.rootLogger = getLogger("") self.originalLevel = self.rootLogger.getEffectiveLevel() self.rootLogger.setLevel(py_logging.DEBUG) self.bufferedHandler = BufferedHandler() self.rootLogger.addHandler(self.bufferedHandler) self.streamHandler, self.output = handlerAndBytesIO() self.rootLogger.addHandler(self.streamHandler) def close(self) -> None: """ Close the logger. """ self.rootLogger.setLevel(self.originalLevel) self.rootLogger.removeHandler(self.bufferedHandler) self.rootLogger.removeHandler(self.streamHandler) self.streamHandler.close() self.output.close() def outputAsText(self) -> str: """ Get the output to the underlying stream as text. @return: the output text """ return self.output.getvalue().decode("utf-8") class STDLibLogObserverTests(unittest.TestCase): """ Tests for L{STDLibLogObserver}. """ def test_interface(self) -> None: """ L{STDLibLogObserver} is an L{ILogObserver}. """ observer = STDLibLogObserver() try: verifyObject(ILogObserver, observer) except BrokenMethodImplementation as e: self.fail(e) def py_logger(self) -> StdlibLoggingContainer: """ Create a logging object we can use to test with. @return: a stdlib-style logger """ logger = StdlibLoggingContainer() self.addCleanup(logger.close) return logger def logEvent(self, *events: LogEvent) -> Tuple[List[LogRecord], str]: """ Send one or more events to Python's logging module, and capture the emitted L{LogRecord}s and output stream as a string. @param events: events @return: a tuple: (records, output) """ pl = self.py_logger() observer = STDLibLogObserver( # Add 1 to default stack depth to skip *this* frame, since # tests will want to know about their own frames. stackDepth=STDLibLogObserver.defaultStackDepth + 1 ) for event in events: observer(event) return pl.bufferedHandler.records, pl.outputAsText() def test_name(self) -> None: """ Logger name. """ records, output = self.logEvent({}) self.assertEqual(len(records), 1) self.assertEqual(records[0].name, "twisted") def test_levels(self) -> None: """ Log levels. """ levelMapping = { None: py_logging.INFO, # Default LogLevel.debug: py_logging.DEBUG, LogLevel.info: py_logging.INFO, LogLevel.warn: py_logging.WARNING, LogLevel.error: py_logging.ERROR, LogLevel.critical: py_logging.CRITICAL, } # Build a set of events for each log level events = [] for level, pyLevel in levelMapping.items(): event = {} # Set the log level on the event, except for default if level is not None: event["log_level"] = level # Remember the Python log level we expect to see for this # event (as an int) event["py_levelno"] = int(pyLevel) events.append(event) records, output = self.logEvent(*events) self.assertEqual(len(records), len(levelMapping)) # Check that each event has the correct level for i in range(len(records)): self.assertEqual(records[i].levelno, events[i]["py_levelno"]) def test_callerInfo(self) -> None: """ C{pathname}, C{lineno}, C{exc_info}, C{func} is set properly on records. """ filename, logLine = nextLine() records, output = self.logEvent({}) self.assertEqual(len(records), 1) self.assertEqual(records[0].pathname, filename) self.assertEqual(records[0].lineno, logLine) self.assertIsNone(records[0].exc_info) # Attribute "func" is missing from record, which is weird because it's # documented. # self.assertEqual(records[0].func, "test_callerInfo") def test_basicFormat(self) -> None: """ Basic formattable event passes the format along correctly. """ event = dict(log_format="Hello, {who}!", who="dude") records, output = self.logEvent(event) self.assertEqual(len(records), 1) self.assertEqual(str(records[0].msg), "Hello, dude!") self.assertEqual(records[0].args, ()) def test_basicFormatRendered(self) -> None: """ Basic formattable event renders correctly. """ event = dict(log_format="Hello, {who}!", who="dude") records, output = self.logEvent(event) self.assertEqual(len(records), 1) self.assertTrue(output.endswith(":Hello, dude!\n"), repr(output)) def test_noFormat(self) -> None: """ Event with no format. """ records, output = self.logEvent({}) self.assertEqual(len(records), 1) self.assertEqual(str(records[0].msg), "") def test_failure(self) -> None: """ An event with a failure logs the failure details as well. """ def failing_func() -> None: 1 / 0 try: failing_func() except ZeroDivisionError: failure = Failure() event = dict(log_format="Hi mom", who="me", log_failure=failure) records, output = self.logEvent(event) self.assertEqual(len(records), 1) self.assertIn("Hi mom", output) self.assertIn("in failing_func", output) self.assertIn("ZeroDivisionError", output) def test_cleanedFailure(self) -> None: """ A cleaned Failure object has a fake traceback object; make sure that logging such a failure still results in the exception details being logged. """ def failing_func() -> None: 1 / 0 try: failing_func() except ZeroDivisionError: failure = Failure() failure.cleanFailure() event = dict(log_format="Hi mom", who="me", log_failure=failure) records, output = self.logEvent(event) self.assertEqual(len(records), 1) self.assertIn("Hi mom", output) self.assertIn("in failing_func", output) self.assertIn("ZeroDivisionError", output) def handlerAndBytesIO() -> Tuple[StreamHandler, BytesIO]: """ Construct a 2-tuple of C{(StreamHandler, BytesIO)} for testing interaction with the 'logging' module. @return: handler and io object """ output = BytesIO() template = py_logging.BASIC_FORMAT stream = TextIOWrapper(output, encoding="utf-8", newline="\n") formatter = Formatter(template) handler = StreamHandler(stream) handler.setFormatter(formatter) return handler, output class BufferedHandler(py_logging.Handler): """ A L{py_logging.Handler} that remembers all logged records in a list. """ def __init__(self) -> None: """ Initialize this L{BufferedHandler}. """ py_logging.Handler.__init__(self) self.records: List[LogRecord] = [] def emit(self, record: LogRecord) -> None: """ Remember the record. """ self.records.append(record) PK ͪ�Z�g�bG bG test_json.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.logger._json}. """ from io import BytesIO, StringIO from typing import IO, Any, List, Optional, Sequence, cast from zope.interface import implementer from zope.interface.exceptions import BrokenMethodImplementation from zope.interface.verify import verifyObject from twisted.python.failure import Failure from twisted.trial.unittest import TestCase from .._flatten import extractField from .._format import formatEvent from .._global import globalLogPublisher from .._interfaces import ILogObserver, LogEvent from .._json import ( eventAsJSON, eventFromJSON, eventsFromJSONLogFile, jsonFileLogObserver, log as jsonLog, ) from .._levels import LogLevel from .._logger import Logger from .._observer import LogPublisher def savedJSONInvariants(testCase: TestCase, savedJSON: str) -> str: """ Assert a few things about the result of L{eventAsJSON}, then return it. @param testCase: The L{TestCase} with which to perform the assertions. @param savedJSON: The result of L{eventAsJSON}. @return: C{savedJSON} @raise AssertionError: If any of the preconditions fail. """ testCase.assertIsInstance(savedJSON, str) testCase.assertEqual(savedJSON.count("\n"), 0) return savedJSON class SaveLoadTests(TestCase): """ Tests for loading and saving log events. """ def savedEventJSON(self, event: LogEvent) -> str: """ Serialize some an events, assert some things about it, and return the JSON. @param event: An event. @return: JSON. """ return savedJSONInvariants(self, eventAsJSON(event)) def test_simpleSaveLoad(self) -> None: """ Saving and loading an empty dictionary results in an empty dictionary. """ self.assertEqual(eventFromJSON(self.savedEventJSON({})), {}) def test_saveLoad(self) -> None: """ Saving and loading a dictionary with some simple values in it results in those same simple values in the output; according to JSON's rules, though, all dictionary keys must be L{str} and any non-L{str} keys will be converted. """ self.assertEqual( eventFromJSON(self.savedEventJSON({1: 2, "3": "4"})), # type: ignore[dict-item] {"1": 2, "3": "4"}, ) def test_saveUnPersistable(self) -> None: """ Saving and loading an object which cannot be represented in JSON will result in a placeholder. """ self.assertEqual( eventFromJSON(self.savedEventJSON({"1": 2, "3": object()})), {"1": 2, "3": {"unpersistable": True}}, ) def test_saveNonASCII(self) -> None: """ Non-ASCII keys and values can be saved and loaded. """ self.assertEqual( eventFromJSON(self.savedEventJSON({"\u1234": "\u4321", "3": object()})), {"\u1234": "\u4321", "3": {"unpersistable": True}}, ) def test_saveBytes(self) -> None: """ Any L{bytes} objects will be saved as if they are latin-1 so they can be faithfully re-loaded. """ inputEvent = {"hello": bytes(range(255))} # On Python 3, bytes keys will be skipped by the JSON encoder. Not # much we can do about that. Let's make sure that we don't get an # error, though. inputEvent.update({b"skipped": "okay"}) # type: ignore[dict-item] self.assertEqual( eventFromJSON(self.savedEventJSON(inputEvent)), {"hello": bytes(range(255)).decode("charmap")}, ) def test_saveUnPersistableThenFormat(self) -> None: """ Saving and loading an object which cannot be represented in JSON, but has a string representation which I{can} be saved as JSON, will result in the same string formatting; any extractable fields will retain their data types. """ class Reprable: def __init__(self, value: object) -> None: self.value = value def __repr__(self) -> str: return "reprable" inputEvent = {"log_format": "{object} {object.value}", "object": Reprable(7)} outputEvent = eventFromJSON(self.savedEventJSON(inputEvent)) self.assertEqual(formatEvent(outputEvent), "reprable 7") def test_extractingFieldsPostLoad(self) -> None: """ L{extractField} can extract fields from an object that's been saved and loaded from JSON. """ class Obj: def __init__(self) -> None: self.value = 345 inputEvent = dict(log_format="{object.value}", object=Obj()) loadedEvent = eventFromJSON(self.savedEventJSON(inputEvent)) self.assertEqual(extractField("object.value", loadedEvent), 345) # The behavior of extractField is consistent between pre-persistence # and post-persistence events, although looking up the key directly # won't be: self.assertRaises(KeyError, extractField, "object", loadedEvent) self.assertRaises(KeyError, extractField, "object", inputEvent) def test_failureStructurePreserved(self) -> None: """ Round-tripping a failure through L{eventAsJSON} preserves its class and structure. """ events: List[LogEvent] = [] log = Logger(observer=cast(ILogObserver, events.append)) try: 1 / 0 except ZeroDivisionError: f = Failure() log.failure("a message about failure", f) self.assertEqual(len(events), 1) loaded = eventFromJSON(self.savedEventJSON(events[0]))["log_failure"] self.assertIsInstance(loaded, Failure) self.assertTrue(loaded.check(ZeroDivisionError)) self.assertIsInstance(loaded.getTraceback(), str) def test_saveLoadLevel(self) -> None: """ It's important that the C{log_level} key remain a L{constantly.NamedConstant} object. """ inputEvent = dict(log_level=LogLevel.warn) loadedEvent = eventFromJSON(self.savedEventJSON(inputEvent)) self.assertIs(loadedEvent["log_level"], LogLevel.warn) def test_saveLoadUnknownLevel(self) -> None: """ If a saved bit of JSON (let's say, from a future version of Twisted) were to persist a different log_level, it will resolve as None. """ loadedEvent = eventFromJSON( '{"log_level": {"name": "other", ' '"__class_uuid__": "02E59486-F24D-46AD-8224-3ACDF2A5732A"}}' ) self.assertEqual(loadedEvent, dict(log_level=None)) class FileLogObserverTests(TestCase): """ Tests for L{jsonFileLogObserver}. """ def test_interface(self) -> None: """ A L{FileLogObserver} returned by L{jsonFileLogObserver} is an L{ILogObserver}. """ with StringIO() as fileHandle: observer = jsonFileLogObserver(fileHandle) try: verifyObject(ILogObserver, observer) except BrokenMethodImplementation as e: self.fail(e) def assertObserverWritesJSON(self, recordSeparator: str = "\x1e") -> None: """ Asserts that an observer created by L{jsonFileLogObserver} with the given arguments writes events serialized as JSON text, using the given record separator. @param recordSeparator: C{recordSeparator} argument to L{jsonFileLogObserver} """ with StringIO() as fileHandle: observer = jsonFileLogObserver(fileHandle, recordSeparator) event = dict(x=1) observer(event) self.assertEqual(fileHandle.getvalue(), f'{recordSeparator}{{"x": 1}}\n') def test_observeWritesDefaultRecordSeparator(self) -> None: """ A L{FileLogObserver} created by L{jsonFileLogObserver} writes events serialzed as JSON text to a file when it observes events. By default, the record separator is C{"\\x1e"}. """ self.assertObserverWritesJSON() def test_observeWritesEmptyRecordSeparator(self) -> None: """ A L{FileLogObserver} created by L{jsonFileLogObserver} writes events serialzed as JSON text to a file when it observes events. This test sets the record separator to C{""}. """ self.assertObserverWritesJSON(recordSeparator="") def test_failureFormatting(self) -> None: """ A L{FileLogObserver} created by L{jsonFileLogObserver} writes failures serialized as JSON text to a file when it observes events. """ io = StringIO() publisher = LogPublisher() logged: List[LogEvent] = [] publisher.addObserver(cast(ILogObserver, logged.append)) publisher.addObserver(jsonFileLogObserver(io)) logger = Logger(observer=publisher) try: 1 / 0 except BaseException: logger.failure("failed as expected") reader = StringIO(io.getvalue()) deserialized = list(eventsFromJSONLogFile(reader)) def checkEvents(logEvents: Sequence[LogEvent]) -> None: self.assertEqual(len(logEvents), 1) [failureEvent] = logEvents self.assertIn("log_failure", failureEvent) failureObject = failureEvent["log_failure"] self.assertIsInstance(failureObject, Failure) tracebackObject = failureObject.getTracebackObject() self.assertEqual( tracebackObject.tb_frame.f_code.co_filename.rstrip("co"), __file__.rstrip("co"), ) checkEvents(logged) checkEvents(deserialized) class LogFileReaderTests(TestCase): """ Tests for L{eventsFromJSONLogFile}. """ def setUp(self) -> None: self.errorEvents: List[LogEvent] = [] @implementer(ILogObserver) def observer(event: LogEvent) -> None: if event["log_namespace"] == jsonLog.namespace and "record" in event: self.errorEvents.append(event) self.logObserver = observer globalLogPublisher.addObserver(observer) def tearDown(self) -> None: globalLogPublisher.removeObserver(self.logObserver) def _readEvents( self, inFile: IO[Any], recordSeparator: Optional[str] = None, bufferSize: int = 4096, ) -> None: """ Test that L{eventsFromJSONLogFile} reads two pre-defined events from a file: C{{"x": 1}} and C{{"y": 2}}. @param inFile: C{inFile} argument to L{eventsFromJSONLogFile} @param recordSeparator: C{recordSeparator} argument to L{eventsFromJSONLogFile} @param bufferSize: C{bufferSize} argument to L{eventsFromJSONLogFile} """ events = iter(eventsFromJSONLogFile(inFile, recordSeparator, bufferSize)) self.assertEqual(next(events), {"x": 1}) self.assertEqual(next(events), {"y": 2}) self.assertRaises(StopIteration, next, events) # No more events def test_readEventsAutoWithRecordSeparator(self) -> None: """ L{eventsFromJSONLogFile} reads events from a file and automatically detects use of C{"\\x1e"} as the record separator. """ with StringIO('\x1e{"x": 1}\n' '\x1e{"y": 2}\n') as fileHandle: self._readEvents(fileHandle) self.assertEqual(len(self.errorEvents), 0) def test_readEventsAutoEmptyRecordSeparator(self) -> None: """ L{eventsFromJSONLogFile} reads events from a file and automatically detects use of C{""} as the record separator. """ with StringIO('{"x": 1}\n' '{"y": 2}\n') as fileHandle: self._readEvents(fileHandle) self.assertEqual(len(self.errorEvents), 0) def test_readEventsExplicitRecordSeparator(self) -> None: """ L{eventsFromJSONLogFile} reads events from a file and is told to use a specific record separator. """ # Use "\x08" (backspace)... because that seems weird enough. with StringIO('\x08{"x": 1}\n' '\x08{"y": 2}\n') as fileHandle: self._readEvents(fileHandle, recordSeparator="\x08") self.assertEqual(len(self.errorEvents), 0) def test_readEventsPartialBuffer(self) -> None: """ L{eventsFromJSONLogFile} handles buffering a partial event. """ with StringIO('\x1e{"x": 1}\n' '\x1e{"y": 2}\n') as fileHandle: # Use a buffer size smaller than the event text. self._readEvents(fileHandle, bufferSize=1) self.assertEqual(len(self.errorEvents), 0) def test_readTruncated(self) -> None: """ If the JSON text for a record is truncated, skip it. """ with StringIO('\x1e{"x": 1' '\x1e{"y": 2}\n') as fileHandle: events = iter(eventsFromJSONLogFile(fileHandle)) self.assertEqual(next(events), {"y": 2}) self.assertRaises(StopIteration, next, events) # No more events # We should have logged the lost record self.assertEqual(len(self.errorEvents), 1) self.assertEqual( self.errorEvents[0]["log_format"], "Unable to read truncated JSON record: {record!r}", ) self.assertEqual(self.errorEvents[0]["record"], b'{"x": 1') def test_readUnicode(self) -> None: """ If the file being read from vends L{str}, strings decode from JSON as-is. """ # The Euro currency sign is "\u20ac" with StringIO('\x1e{"currency": "\u20ac"}\n') as fileHandle: events = iter(eventsFromJSONLogFile(fileHandle)) self.assertEqual(next(events), {"currency": "\u20ac"}) self.assertRaises(StopIteration, next, events) # No more events self.assertEqual(len(self.errorEvents), 0) def test_readUTF8Bytes(self) -> None: """ If the file being read from vends L{bytes}, strings decode from JSON as UTF-8. """ # The Euro currency sign is b"\xe2\x82\xac" in UTF-8 with BytesIO(b'\x1e{"currency": "\xe2\x82\xac"}\n') as fileHandle: events = iter(eventsFromJSONLogFile(fileHandle)) # The Euro currency sign is "\u20ac" self.assertEqual(next(events), {"currency": "\u20ac"}) self.assertRaises(StopIteration, next, events) # No more events self.assertEqual(len(self.errorEvents), 0) def test_readTruncatedUTF8Bytes(self) -> None: """ If the JSON text for a record is truncated in the middle of a two-byte Unicode codepoint, we don't want to see a codec exception and the stream is read properly when the additional data arrives. """ # The Euro currency sign is "\u20ac" and encodes in UTF-8 as three # bytes: b"\xe2\x82\xac". with BytesIO(b'\x1e{"x": "\xe2\x82\xac"}\n') as fileHandle: events = iter(eventsFromJSONLogFile(fileHandle, bufferSize=8)) self.assertEqual(next(events), {"x": "\u20ac"}) # Got text self.assertRaises(StopIteration, next, events) # No more events self.assertEqual(len(self.errorEvents), 0) def test_readInvalidUTF8Bytes(self) -> None: """ If the JSON text for a record contains invalid UTF-8 text, ignore that record. """ # The string b"\xe2\xac" is bogus with BytesIO(b'\x1e{"x": "\xe2\xac"}\n' b'\x1e{"y": 2}\n') as fileHandle: events = iter(eventsFromJSONLogFile(fileHandle)) self.assertEqual(next(events), {"y": 2}) self.assertRaises(StopIteration, next, events) # No more events # We should have logged the lost record self.assertEqual(len(self.errorEvents), 1) self.assertEqual( self.errorEvents[0]["log_format"], "Unable to decode UTF-8 for JSON record: {record!r}", ) self.assertEqual(self.errorEvents[0]["record"], b'{"x": "\xe2\xac"}\n') def test_readInvalidJSON(self) -> None: """ If the JSON text for a record is invalid, skip it. """ with StringIO('\x1e{"x": }\n' '\x1e{"y": 2}\n') as fileHandle: events = iter(eventsFromJSONLogFile(fileHandle)) self.assertEqual(next(events), {"y": 2}) self.assertRaises(StopIteration, next, events) # No more events # We should have logged the lost record self.assertEqual(len(self.errorEvents), 1) self.assertEqual( self.errorEvents[0]["log_format"], "Unable to read JSON record: {record!r}", ) self.assertEqual(self.errorEvents[0]["record"], b'{"x": }\n') def test_readUnseparated(self) -> None: """ Multiple events without a record separator are skipped. """ with StringIO('\x1e{"x": 1}\n' '{"y": 2}\n') as fileHandle: events = eventsFromJSONLogFile(fileHandle) self.assertRaises(StopIteration, next, events) # No more events # We should have logged the lost record self.assertEqual(len(self.errorEvents), 1) self.assertEqual( self.errorEvents[0]["log_format"], "Unable to read JSON record: {record!r}", ) self.assertEqual(self.errorEvents[0]["record"], b'{"x": 1}\n{"y": 2}\n') def test_roundTrip(self) -> None: """ Data written by a L{FileLogObserver} returned by L{jsonFileLogObserver} and read by L{eventsFromJSONLogFile} is reconstructed properly. """ event = dict(x=1) with StringIO() as fileHandle: observer = jsonFileLogObserver(fileHandle) observer(event) fileHandle.seek(0) events = eventsFromJSONLogFile(fileHandle) self.assertEqual(tuple(events), (event,)) self.assertEqual(len(self.errorEvents), 0) PK ͪ�Z&����1 �1 test_global.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._global}. """ import io from typing import IO, Any, List, Optional, TextIO, Tuple, Type, cast from twisted.python.failure import Failure from twisted.trial import unittest from .._file import textFileLogObserver from .._global import MORE_THAN_ONCE_WARNING, LogBeginner from .._interfaces import ILogObserver, LogEvent from .._levels import LogLevel from .._logger import Logger from .._observer import LogPublisher from ..test.test_stdlib import nextLine def compareEvents( test: unittest.TestCase, actualEvents: List[LogEvent], expectedEvents: List[LogEvent], ) -> None: """ Compare two sequences of log events, examining only the the keys which are present in both. @param test: a test case doing the comparison @param actualEvents: A list of log events that were emitted by a logger. @param expectedEvents: A list of log events that were expected by a test. """ if len(actualEvents) != len(expectedEvents): test.assertEqual(actualEvents, expectedEvents) allMergedKeys = set() for event in expectedEvents: allMergedKeys |= set(event.keys()) def simplify(event: LogEvent) -> LogEvent: copy = event.copy() for key in event.keys(): if key not in allMergedKeys: copy.pop(key) return copy simplifiedActual = [simplify(event) for event in actualEvents] test.assertEqual(simplifiedActual, expectedEvents) class LogBeginnerTests(unittest.TestCase): """ Tests for L{LogBeginner}. """ def setUp(self) -> None: self.publisher = LogPublisher() self.errorStream = io.StringIO() class NotSys: stdout = object() stderr = object() class NotWarnings: def __init__(self) -> None: self.warnings: List[ Tuple[ str, Type[Warning], str, int, Optional[IO[Any]], Optional[int] ] ] = [] def showwarning( self, message: str, category: Type[Warning], filename: str, lineno: int, file: Optional[IO[Any]] = None, line: Optional[int] = None, ) -> None: """ Emulate warnings.showwarning. @param message: A warning message to emit. @param category: A warning category to associate with C{message}. @param filename: A file name for the source code file issuing the warning. @param lineno: A line number in the source file where the warning was issued. @param file: A file to write the warning message to. If L{None}, write to L{sys.stderr}. @param line: A line of source code to include with the warning message. If L{None}, attempt to read the line from C{filename} and C{lineno}. """ self.warnings.append((message, category, filename, lineno, file, line)) self.sysModule = NotSys() self.warningsModule = NotWarnings() self.beginner = LogBeginner( self.publisher, self.errorStream, self.sysModule, self.warningsModule ) def test_beginLoggingToAddObservers(self) -> None: """ Test that C{beginLoggingTo()} adds observers. """ event = dict(foo=1, bar=2) events1: List[LogEvent] = [] events2: List[LogEvent] = [] o1 = cast(ILogObserver, lambda e: events1.append(e)) o2 = cast(ILogObserver, lambda e: events2.append(e)) self.beginner.beginLoggingTo((o1, o2)) self.publisher(event) self.assertEqual([event], events1) self.assertEqual([event], events2) def test_beginLoggingToBufferedEvents(self) -> None: """ Test that events are buffered until C{beginLoggingTo()} is called. """ event = dict(foo=1, bar=2) events1: List[LogEvent] = [] events2: List[LogEvent] = [] o1 = cast(ILogObserver, lambda e: events1.append(e)) o2 = cast(ILogObserver, lambda e: events2.append(e)) self.publisher(event) # Before beginLoggingTo; this is buffered self.beginner.beginLoggingTo((o1, o2)) self.assertEqual([event], events1) self.assertEqual([event], events2) def _bufferLimitTest(self, limit: int, beginner: LogBeginner) -> None: """ Verify that when more than C{limit} events are logged to L{LogBeginner}, only the last C{limit} are replayed by L{LogBeginner.beginLoggingTo}. @param limit: The maximum number of events the log beginner should buffer. @param beginner: The L{LogBeginner} against which to verify. @raise: C{self.failureException} if the wrong events are replayed by C{beginner}. """ for count in range(limit + 1): self.publisher(dict(count=count)) events: List[LogEvent] = [] beginner.beginLoggingTo([cast(ILogObserver, events.append)]) self.assertEqual( list(range(1, limit + 1)), list(event["count"] for event in events), ) def test_defaultBufferLimit(self) -> None: """ Up to C{LogBeginner._DEFAULT_BUFFER_SIZE} log events are buffered for replay by L{LogBeginner.beginLoggingTo}. """ limit = LogBeginner._DEFAULT_BUFFER_SIZE self._bufferLimitTest(limit, self.beginner) def test_overrideBufferLimit(self) -> None: """ The size of the L{LogBeginner} event buffer can be overridden with the C{initialBufferSize} initilizer argument. """ limit = 3 beginner = LogBeginner( self.publisher, self.errorStream, self.sysModule, self.warningsModule, initialBufferSize=limit, ) self._bufferLimitTest(limit, beginner) def test_beginLoggingToTwice(self) -> None: """ When invoked twice, L{LogBeginner.beginLoggingTo} will emit a log message warning the user that they previously began logging, and add the new log observers. """ events1: List[LogEvent] = [] events2: List[LogEvent] = [] fileHandle = io.StringIO() textObserver = textFileLogObserver(fileHandle) self.publisher(dict(event="prebuffer")) firstFilename, firstLine = nextLine() self.beginner.beginLoggingTo([cast(ILogObserver, events1.append), textObserver]) self.publisher(dict(event="postbuffer")) secondFilename, secondLine = nextLine() self.beginner.beginLoggingTo([cast(ILogObserver, events2.append), textObserver]) self.publisher(dict(event="postwarn")) warning = dict( log_format=MORE_THAN_ONCE_WARNING, log_level=LogLevel.warn, fileNow=secondFilename, lineNow=secondLine, fileThen=firstFilename, lineThen=firstLine, ) self.maxDiff = None compareEvents( self, events1, [ dict(event="prebuffer"), dict(event="postbuffer"), warning, dict(event="postwarn"), ], ) compareEvents(self, events2, [warning, dict(event="postwarn")]) output = fileHandle.getvalue() self.assertIn(f"<{firstFilename}:{firstLine}>", output) self.assertIn(f"<{secondFilename}:{secondLine}>", output) def test_criticalLogging(self) -> None: """ Critical messages will be written as text to the error stream. """ log = Logger(observer=self.publisher) log.info("ignore this") log.critical("a critical {message}", message="message") self.assertEqual(self.errorStream.getvalue(), "a critical message\n") def test_criticalLoggingStops(self) -> None: """ Once logging has begun with C{beginLoggingTo}, critical messages are no longer written to the output stream. """ log = Logger(observer=self.publisher) self.beginner.beginLoggingTo(()) log.critical("another critical message") self.assertEqual(self.errorStream.getvalue(), "") def test_beginLoggingToRedirectStandardIO(self) -> None: """ L{LogBeginner.beginLoggingTo} will re-direct the standard output and error streams by setting the C{stdio} and C{stderr} attributes on its sys module object. """ events: List[LogEvent] = [] self.beginner.beginLoggingTo([cast(ILogObserver, events.append)]) print("Hello, world.", file=cast(TextIO, self.sysModule.stdout)) compareEvents( self, events, [dict(log_namespace="stdout", log_io="Hello, world.")] ) del events[:] print("Error, world.", file=cast(TextIO, self.sysModule.stderr)) compareEvents( self, events, [dict(log_namespace="stderr", log_io="Error, world.")] ) def test_beginLoggingToDontRedirect(self) -> None: """ L{LogBeginner.beginLoggingTo} will leave the existing stdout/stderr in place if it has been told not to replace them. """ oldOut = self.sysModule.stdout oldErr = self.sysModule.stderr self.beginner.beginLoggingTo((), redirectStandardIO=False) self.assertIs(self.sysModule.stdout, oldOut) self.assertIs(self.sysModule.stderr, oldErr) def test_beginLoggingToPreservesEncoding(self) -> None: """ When L{LogBeginner.beginLoggingTo} redirects stdout/stderr streams, the replacement streams will preserve the encoding of the replaced streams, to minimally disrupt any application relying on a specific encoding. """ weird = io.TextIOWrapper(io.BytesIO(), "shift-JIS") weirderr = io.TextIOWrapper(io.BytesIO(), "big5") self.sysModule.stdout = weird self.sysModule.stderr = weirderr events: List[LogEvent] = [] self.beginner.beginLoggingTo([cast(ILogObserver, events.append)]) stdout = cast(TextIO, self.sysModule.stdout) stderr = cast(TextIO, self.sysModule.stderr) self.assertEqual(stdout.encoding, "shift-JIS") self.assertEqual(stderr.encoding, "big5") stdout.write(b"\x97\x9B\n") # type: ignore[arg-type] stderr.write(b"\xBC\xFC\n") # type: ignore[arg-type] compareEvents(self, events, [dict(log_io="\u674e"), dict(log_io="\u7469")]) def test_warningsModule(self) -> None: """ L{LogBeginner.beginLoggingTo} will redirect the warnings of its warnings module into the logging system. """ self.warningsModule.showwarning("a message", DeprecationWarning, __file__, 1) events: List[LogEvent] = [] self.beginner.beginLoggingTo([cast(ILogObserver, events.append)]) self.warningsModule.showwarning( "another message", DeprecationWarning, __file__, 2 ) f = io.StringIO() self.warningsModule.showwarning( "yet another", DeprecationWarning, __file__, 3, file=f ) self.assertEqual( self.warningsModule.warnings, [ ("a message", DeprecationWarning, __file__, 1, None, None), ("yet another", DeprecationWarning, __file__, 3, f, None), ], ) compareEvents( self, events, [ dict( warning="another message", category=( DeprecationWarning.__module__ + "." + DeprecationWarning.__name__ ), filename=__file__, lineno=2, ) ], ) def test_failuresAppendTracebacks(self) -> None: """ The string resulting from a logged failure contains a traceback. """ f = Failure(Exception("this is not the behavior you are looking for")) log = Logger(observer=self.publisher) log.failure("a failure", failure=f) msg = self.errorStream.getvalue() self.assertIn("a failure", msg) self.assertIn("this is not the behavior you are looking for", msg) self.assertIn("Traceback", msg) PK ͪ�Z����@ @ test_capture.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._capture}. """ from twisted.logger import Logger, LogLevel from twisted.trial.unittest import TestCase from .._capture import capturedLogs class LogCaptureTests(TestCase): """ Tests for L{LogCaptureTests}. """ log = Logger() def test_capture(self) -> None: """ Events logged within context are captured. """ foo = object() with capturedLogs() as captured: self.log.debug("Capture this, please", foo=foo) self.log.info("Capture this too, please", foo=foo) self.assertTrue(len(captured) == 2) self.assertEqual(captured[0]["log_format"], "Capture this, please") self.assertEqual(captured[0]["log_level"], LogLevel.debug) self.assertEqual(captured[0]["foo"], foo) self.assertEqual(captured[1]["log_format"], "Capture this too, please") self.assertEqual(captured[1]["log_level"], LogLevel.info) self.assertEqual(captured[1]["foo"], foo) PK ͪ�Zj �� � test_util.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._util}. """ from zope.interface import implementer from twisted.trial import unittest from .._interfaces import ILogObserver, LogEvent from .._observer import LogPublisher from .._util import formatTrace class UtilTests(unittest.TestCase): """ Utility tests. """ def test_trace(self) -> None: """ Tracing keeps track of forwarding done by the publisher. """ publisher = LogPublisher() event: LogEvent = dict(log_trace=[]) @implementer(ILogObserver) def o1(e: LogEvent) -> None: pass @implementer(ILogObserver) def o2(e: LogEvent) -> None: self.assertIs(e, event) self.assertEqual( e["log_trace"], [ (publisher, o1), (publisher, o2), # Event hasn't been sent to o3 yet ], ) @implementer(ILogObserver) def o3(e: LogEvent) -> None: self.assertIs(e, event) self.assertEqual( e["log_trace"], [ (publisher, o1), (publisher, o2), (publisher, o3), ], ) publisher.addObserver(o1) publisher.addObserver(o2) publisher.addObserver(o3) publisher(event) def test_formatTrace(self) -> None: """ Format trace as string. """ event: LogEvent = dict(log_trace=[]) @implementer(ILogObserver) def o1(e: LogEvent) -> None: pass @implementer(ILogObserver) def o2(e: LogEvent) -> None: pass @implementer(ILogObserver) def o3(e: LogEvent) -> None: pass @implementer(ILogObserver) def o4(e: LogEvent) -> None: pass @implementer(ILogObserver) def o5(e: LogEvent) -> None: pass o1.name = "root/o1" # type: ignore[attr-defined] o2.name = "root/p1/o2" o3.name = "root/p1/o3" o4.name = "root/p1/p2/o4" o5.name = "root/o5" @implementer(ILogObserver) def testObserver(e: LogEvent) -> None: self.assertIs(e, event) trace = formatTrace(e["log_trace"]) self.assertEqual( trace, ( "{root} ({root.name})\n" " -> {o1} ({o1.name})\n" " -> {p1} ({p1.name})\n" " -> {o2} ({o2.name})\n" " -> {o3} ({o3.name})\n" " -> {p2} ({p2.name})\n" " -> {o4} ({o4.name})\n" " -> {o5} ({o5.name})\n" " -> {oTest}\n" ).format( root=root, o1=o1, o2=o2, o3=o3, o4=o4, o5=o5, p1=p1, p2=p2, oTest=oTest, ), ) oTest = testObserver p2 = LogPublisher(o4) p1 = LogPublisher(o2, o3, p2) p2.name = "root/p1/p2/" # type: ignore[attr-defined] p1.name = "root/p1/" # type: ignore[attr-defined] root = LogPublisher(o1, p1, o5, oTest) root.name = "root/" # type: ignore[attr-defined] root(event) PK ͪ�Zo6K=�% �% test_flatten.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._format}. """ import json from itertools import count from typing import Any, Callable, Optional try: from time import tzset except ImportError: tzset = None # type: ignore[assignment] from twisted.trial import unittest from .._flatten import KeyFlattener, aFormatter, extractField, flattenEvent from .._format import formatEvent from .._interfaces import LogEvent class FlatFormattingTests(unittest.TestCase): """ Tests for flattened event formatting functions. """ def test_formatFlatEvent(self) -> None: """ L{flattenEvent} will "flatten" an event so that, if scrubbed of all but serializable objects, it will preserve all necessary data to be formatted once serialized. When presented with an event thusly flattened, L{formatEvent} will produce the same output. """ counter = count() class Ephemeral: attribute = "value" event1 = dict( log_format=( "callable: {callme()} " "attribute: {object.attribute} " "numrepr: {number!r} " "numstr: {number!s} " "strrepr: {string!r} " "unistr: {unistr!s}" ), callme=lambda: next(counter), object=Ephemeral(), number=7, string="hello", unistr="ö", ) flattenEvent(event1) event2 = dict(event1) del event2["callme"] del event2["object"] event3 = json.loads(json.dumps(event2)) self.assertEqual( formatEvent(event3), ( "callable: 0 " "attribute: value " "numrepr: 7 " "numstr: 7 " "strrepr: 'hello' " "unistr: ö" ), ) def test_formatFlatEventBadFormat(self) -> None: """ If the format string is invalid, an error is produced. """ event1 = dict( log_format=("strrepr: {string!X}"), string="hello", ) flattenEvent(event1) event2 = json.loads(json.dumps(event1)) self.assertTrue(formatEvent(event2).startswith("Unable to format event")) def test_formatFlatEventWithMutatedFields(self) -> None: """ L{formatEvent} will prefer the stored C{str()} or C{repr()} value for an object, in case the other version. """ class Unpersistable: """ Unpersitable object. """ destructed = False def selfDestruct(self) -> None: """ Self destruct. """ self.destructed = True def __repr__(self) -> str: if self.destructed: return "post-serialization garbage" else: return "un-persistable" up = Unpersistable() event1 = dict(log_format="unpersistable: {unpersistable}", unpersistable=up) flattenEvent(event1) up.selfDestruct() self.assertEqual(formatEvent(event1), "unpersistable: un-persistable") def test_keyFlattening(self) -> None: """ Test that L{KeyFlattener.flatKey} returns the expected keys for format fields. """ def keyFromFormat(format: str) -> str: for ( literalText, fieldName, formatSpec, conversion, ) in aFormatter.parse(format): assert fieldName is not None return KeyFlattener().flatKey(fieldName, formatSpec, conversion) assert False, "Unable to derive key from format: {format}" # No name try: self.assertEqual(keyFromFormat("{}"), "!:") except ValueError: # In python 2.6, an empty field name causes Formatter.parse to # raise ValueError. # In Python 2.7, it's allowed, so this exception is unexpected. raise # Just a name self.assertEqual(keyFromFormat("{foo}"), "foo!:") # Add conversion self.assertEqual(keyFromFormat("{foo!s}"), "foo!s:") self.assertEqual(keyFromFormat("{foo!r}"), "foo!r:") # Add format spec self.assertEqual(keyFromFormat("{foo:%s}"), "foo!:%s") self.assertEqual(keyFromFormat("{foo:!}"), "foo!:!") self.assertEqual(keyFromFormat("{foo::}"), "foo!::") # Both self.assertEqual(keyFromFormat("{foo!s:%s}"), "foo!s:%s") self.assertEqual(keyFromFormat("{foo!s:!}"), "foo!s:!") self.assertEqual(keyFromFormat("{foo!s::}"), "foo!s::") sameFlattener = KeyFlattener() ( ( literalText, fieldName, formatSpec, conversion, ), ) = aFormatter.parse("{x}") assert fieldName is not None self.assertEqual( sameFlattener.flatKey(fieldName, formatSpec, conversion), "x!:" ) self.assertEqual( sameFlattener.flatKey(fieldName, formatSpec, conversion), "x!:/2" ) def _test_formatFlatEvent_fieldNamesSame( self, event: Optional[LogEvent] = None ) -> LogEvent: """ The same format field used twice in one event is rendered twice. @param event: An event to flatten. If L{None}, create a new event. @return: C{event} or the event created. """ if event is None: counter = count() class CountStr: """ Hack """ def __str__(self) -> str: return str(next(counter)) event = dict( log_format="{x} {x}", x=CountStr(), ) flattenEvent(event) self.assertEqual(formatEvent(event), "0 1") return event def test_formatFlatEventFieldNamesSame(self) -> None: """ The same format field used twice in one event is rendered twice. """ self._test_formatFlatEvent_fieldNamesSame() def test_formatFlatEventFieldNamesSameAgain(self) -> None: """ The same event flattened twice gives the same (already rendered) result. """ event = self._test_formatFlatEvent_fieldNamesSame() self._test_formatFlatEvent_fieldNamesSame(event) def test_formatEventFlatTrailingText(self) -> None: """ L{formatEvent} will handle a flattened event with tailing text after a replacement field. """ event = dict( log_format="test {x} trailing", x="value", ) flattenEvent(event) result = formatEvent(event) self.assertEqual(result, "test value trailing") def test_extractField( self, flattenFirst: Callable[[LogEvent], LogEvent] = lambda x: x ) -> None: """ L{extractField} will extract a field used in the format string. @param flattenFirst: callable to flatten an event """ class ObjectWithRepr: def __repr__(self) -> str: return "repr" class Something: def __init__(self) -> None: self.number = 7 self.object = ObjectWithRepr() def __getstate__(self) -> None: raise NotImplementedError("Just in case.") event = dict( log_format="{something.number} {something.object}", something=Something(), ) flattened = flattenFirst(event) def extract(field: str) -> Any: return extractField(field, flattened) self.assertEqual(extract("something.number"), 7) self.assertEqual(extract("something.number!s"), "7") self.assertEqual(extract("something.object!s"), "repr") def test_extractFieldFlattenFirst(self) -> None: """ L{extractField} behaves identically if the event is explicitly flattened first. """ def flattened(event: LogEvent) -> LogEvent: flattenEvent(event) return event self.test_extractField(flattened) def test_flattenEventWithoutFormat(self) -> None: """ L{flattenEvent} will do nothing to an event with no format string. """ inputEvent = {"a": "b", "c": 1} flattenEvent(inputEvent) self.assertEqual(inputEvent, {"a": "b", "c": 1}) def test_flattenEventWithInertFormat(self) -> None: """ L{flattenEvent} will do nothing to an event with a format string that contains no format fields. """ inputEvent = {"a": "b", "c": 1, "log_format": "simple message"} flattenEvent(inputEvent) self.assertEqual( inputEvent, { "a": "b", "c": 1, "log_format": "simple message", }, ) def test_flattenEventWithNoneFormat(self) -> None: """ L{flattenEvent} will do nothing to an event with log_format set to None. """ inputEvent = {"a": "b", "c": 1, "log_format": None} flattenEvent(inputEvent) self.assertEqual( inputEvent, { "a": "b", "c": 1, "log_format": None, }, ) PK ͪ�Zu�:��" �" test_io.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._io}. """ import sys from typing import List, Optional from zope.interface import implementer from constantly import NamedConstant # type: ignore[import] from twisted.trial import unittest from .._interfaces import ILogObserver, LogEvent from .._io import LoggingFile from .._levels import LogLevel from .._logger import Logger from .._observer import LogPublisher @implementer(ILogObserver) class TestLoggingFile(LoggingFile): """ L{LoggingFile} that is also an observer which captures events and messages. """ def __init__( self, logger: Logger, level: NamedConstant = LogLevel.info, encoding: Optional[str] = None, ) -> None: super().__init__(logger=logger, level=level, encoding=encoding) self.events: List[LogEvent] = [] self.messages: List[str] = [] def __call__(self, event: LogEvent) -> None: self.events.append(event) if "log_io" in event: self.messages.append(event["log_io"]) class LoggingFileTests(unittest.TestCase): """ Tests for L{LoggingFile}. """ def setUp(self) -> None: """ Create a logger for test L{LoggingFile} instances to use. """ self.publisher = LogPublisher() self.logger = Logger(observer=self.publisher) def test_softspace(self) -> None: """ L{LoggingFile.softspace} is 0. """ self.assertEqual(LoggingFile(self.logger).softspace, 0) warningsShown = self.flushWarnings([self.test_softspace]) self.assertEqual(len(warningsShown), 1) self.assertEqual(warningsShown[0]["category"], DeprecationWarning) deprecatedClass = "twisted.logger._io.LoggingFile.softspace" self.assertEqual( warningsShown[0]["message"], "%s was deprecated in Twisted 21.2.0" % (deprecatedClass), ) def test_readOnlyAttributes(self) -> None: """ Some L{LoggingFile} attributes are read-only. """ f = LoggingFile(self.logger) self.assertRaises(AttributeError, setattr, f, "closed", True) self.assertRaises(AttributeError, setattr, f, "encoding", "utf-8") self.assertRaises(AttributeError, setattr, f, "mode", "r") self.assertRaises(AttributeError, setattr, f, "newlines", ["\n"]) self.assertRaises(AttributeError, setattr, f, "name", "foo") def test_unsupportedMethods(self) -> None: """ Some L{LoggingFile} methods are unsupported. """ f = LoggingFile(self.logger) self.assertRaises(IOError, f.read) self.assertRaises(IOError, f.next) self.assertRaises(IOError, f.readline) self.assertRaises(IOError, f.readlines) self.assertRaises(IOError, f.xreadlines) self.assertRaises(IOError, f.seek) self.assertRaises(IOError, f.tell) self.assertRaises(IOError, f.truncate) def test_level(self) -> None: """ Default level is L{LogLevel.info} if not set. """ f = LoggingFile(self.logger) self.assertEqual(f.level, LogLevel.info) f = LoggingFile(self.logger, level=LogLevel.error) self.assertEqual(f.level, LogLevel.error) def test_encoding(self) -> None: """ Default encoding is C{sys.getdefaultencoding()} if not set. """ f = LoggingFile(self.logger) self.assertEqual(f.encoding, sys.getdefaultencoding()) f = LoggingFile(self.logger, encoding="utf-8") self.assertEqual(f.encoding, "utf-8") def test_mode(self) -> None: """ Reported mode is C{"w"}. """ f = LoggingFile(self.logger) self.assertEqual(f.mode, "w") def test_newlines(self) -> None: """ The C{newlines} attribute is L{None}. """ f = LoggingFile(self.logger) self.assertIsNone(f.newlines) def test_name(self) -> None: """ The C{name} attribute is fixed. """ f = LoggingFile(self.logger) self.assertEqual(f.name, "<LoggingFile twisted.logger.test.test_io#info>") def test_close(self) -> None: """ L{LoggingFile.close} closes the file. """ f = LoggingFile(self.logger) f.close() self.assertTrue(f.closed) self.assertRaises(ValueError, f.write, "Hello") def test_flush(self) -> None: """ L{LoggingFile.flush} does nothing. """ f = LoggingFile(self.logger) f.flush() def test_fileno(self) -> None: """ L{LoggingFile.fileno} returns C{-1}. """ f = LoggingFile(self.logger) self.assertEqual(f.fileno(), -1) def test_isatty(self) -> None: """ L{LoggingFile.isatty} returns C{False}. """ f = LoggingFile(self.logger) self.assertFalse(f.isatty()) def test_writeBuffering(self) -> None: """ Writing buffers correctly. """ f = self.observedFile() f.write("Hello") self.assertEqual(f.messages, []) f.write(", world!\n") self.assertEqual(f.messages, ["Hello, world!"]) f.write("It's nice to meet you.\n\nIndeed.") self.assertEqual( f.messages, [ "Hello, world!", "It's nice to meet you.", "", ], ) def test_writeBytesDecoded(self) -> None: """ Bytes are decoded to text. """ f = self.observedFile(encoding="utf-8") f.write(b"Hello, Mr. S\xc3\xa1nchez\n") self.assertEqual(f.messages, ["Hello, Mr. S\xe1nchez"]) def test_writeUnicode(self) -> None: """ Unicode is unmodified. """ f = self.observedFile(encoding="utf-8") f.write("Hello, Mr. S\xe1nchez\n") self.assertEqual(f.messages, ["Hello, Mr. S\xe1nchez"]) def test_writeLevel(self) -> None: """ Log level is emitted properly. """ f = self.observedFile() f.write("Hello\n") self.assertEqual(len(f.events), 1) self.assertEqual(f.events[0]["log_level"], LogLevel.info) f = self.observedFile(level=LogLevel.error) f.write("Hello\n") self.assertEqual(len(f.events), 1) self.assertEqual(f.events[0]["log_level"], LogLevel.error) def test_writeFormat(self) -> None: """ Log format is C{"{message}"}. """ f = self.observedFile() f.write("Hello\n") self.assertEqual(len(f.events), 1) self.assertEqual(f.events[0]["log_format"], "{log_io}") def test_writelinesBuffering(self) -> None: """ C{writelines} does not add newlines. """ # Note this is different behavior than t.p.log.StdioOnnaStick. f = self.observedFile() f.writelines(("Hello", ", ", "")) self.assertEqual(f.messages, []) f.writelines(("world!\n",)) self.assertEqual(f.messages, ["Hello, world!"]) f.writelines(("It's nice to meet you.\n\n", "Indeed.")) self.assertEqual( f.messages, [ "Hello, world!", "It's nice to meet you.", "", ], ) def test_print(self) -> None: """ L{LoggingFile} can replace L{sys.stdout}. """ f = self.observedFile() self.patch(sys, "stdout", f) print("Hello,", end=" ") print("world.") self.assertEqual(f.messages, ["Hello, world."]) def observedFile( self, level: NamedConstant = LogLevel.info, encoding: Optional[str] = None, ) -> TestLoggingFile: """ Construct a L{LoggingFile} with a built-in observer. @param level: C{level} argument to L{LoggingFile} @param encoding: C{encoding} argument to L{LoggingFile} @return: a L{TestLoggingFile} with an observer that appends received events into the file's C{events} attribute (a L{list}) and event messages into the file's C{messages} attribute (a L{list}). """ # Logger takes an observer argument, for which we want to use the # TestLoggingFile we will create, but that takes the Logger as an # argument, so we'll use an array to indirectly reference the # TestLoggingFile. loggingFiles: List[TestLoggingFile] = [] @implementer(ILogObserver) def observer(event: LogEvent) -> None: loggingFiles[0](event) log = Logger(observer=observer) loggingFiles.append(TestLoggingFile(logger=log, level=level, encoding=encoding)) return loggingFiles[0] PK ͪ�Z��c c test_levels.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._levels}. """ from twisted.trial import unittest from .._levels import InvalidLogLevelError, LogLevel class LogLevelTests(unittest.TestCase): """ Tests for L{LogLevel}. """ def test_levelWithName(self) -> None: """ Look up log level by name. """ for level in LogLevel.iterconstants(): self.assertIs(LogLevel.levelWithName(level.name), level) def test_levelWithInvalidName(self) -> None: """ You can't make up log level names. """ bogus = "*bogus*" try: LogLevel.levelWithName(bogus) except InvalidLogLevelError as e: self.assertIs(e.level, bogus) else: self.fail("Expected InvalidLogLevelError.") PK ͪ�Z3=� test_buffer.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._buffer}. """ from typing import List, cast from zope.interface.exceptions import BrokenMethodImplementation from zope.interface.verify import verifyObject from twisted.trial import unittest from .._buffer import LimitedHistoryLogObserver from .._interfaces import ILogObserver, LogEvent class LimitedHistoryLogObserverTests(unittest.TestCase): """ Tests for L{LimitedHistoryLogObserver}. """ def test_interface(self) -> None: """ L{LimitedHistoryLogObserver} provides L{ILogObserver}. """ observer = LimitedHistoryLogObserver(0) try: verifyObject(ILogObserver, observer) except BrokenMethodImplementation as e: self.fail(e) def test_order(self) -> None: """ L{LimitedHistoryLogObserver} saves history in the order it is received. """ size = 4 events = [dict(n=n) for n in range(size // 2)] observer = LimitedHistoryLogObserver(size) for event in events: observer(event) outEvents: List[LogEvent] = [] observer.replayTo(cast(ILogObserver, outEvents.append)) self.assertEqual(events, outEvents) def test_limit(self) -> None: """ When more events than a L{LimitedHistoryLogObserver}'s maximum size are buffered, older events will be dropped. """ size = 4 events = [dict(n=n) for n in range(size * 2)] observer = LimitedHistoryLogObserver(size) for event in events: observer(event) outEvents: List[LogEvent] = [] observer.replayTo(cast(ILogObserver, outEvents.append)) self.assertEqual(events[-size:], outEvents) PK ͪ�Z�x�� � test_file.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._file}. """ from io import StringIO from types import TracebackType from typing import IO, Any, AnyStr, Optional, Type, cast from zope.interface.exceptions import BrokenMethodImplementation from zope.interface.verify import verifyObject from twisted.python.failure import Failure from twisted.trial.unittest import TestCase from .._file import FileLogObserver, textFileLogObserver from .._interfaces import ILogObserver class FileLogObserverTests(TestCase): """ Tests for L{FileLogObserver}. """ def test_interface(self) -> None: """ L{FileLogObserver} is an L{ILogObserver}. """ with StringIO() as fileHandle: observer = FileLogObserver(fileHandle, lambda e: str(e)) try: verifyObject(ILogObserver, observer) except BrokenMethodImplementation as e: self.fail(e) def test_observeWrites(self) -> None: """ L{FileLogObserver} writes to the given file when it observes events. """ with StringIO() as fileHandle: observer = FileLogObserver(fileHandle, lambda e: str(e)) event = dict(x=1) observer(event) self.assertEqual(fileHandle.getvalue(), str(event)) def _test_observeWrites(self, what: Optional[str], count: int) -> None: """ Verify that observer performs an expected number of writes when the formatter returns a given value. @param what: the value for the formatter to return. @param count: the expected number of writes. """ with DummyFile() as fileHandle: observer = FileLogObserver(cast(IO[Any], fileHandle), lambda e: what) event = dict(x=1) observer(event) self.assertEqual(fileHandle.writes, count) def test_observeWritesNone(self) -> None: """ L{FileLogObserver} does not write to the given file when it observes events and C{formatEvent} returns L{None}. """ self._test_observeWrites(None, 0) def test_observeWritesEmpty(self) -> None: """ L{FileLogObserver} does not write to the given file when it observes events and C{formatEvent} returns C{""}. """ self._test_observeWrites("", 0) def test_observeFlushes(self) -> None: """ L{FileLogObserver} calles C{flush()} on the output file when it observes an event. """ with DummyFile() as fileHandle: observer = FileLogObserver(cast(IO[Any], fileHandle), lambda e: str(e)) event = dict(x=1) observer(event) self.assertEqual(fileHandle.flushes, 1) class TextFileLogObserverTests(TestCase): """ Tests for L{textFileLogObserver}. """ def test_returnsFileLogObserver(self) -> None: """ L{textFileLogObserver} returns a L{FileLogObserver}. """ with StringIO() as fileHandle: observer = textFileLogObserver(fileHandle) self.assertIsInstance(observer, FileLogObserver) def test_outFile(self) -> None: """ Returned L{FileLogObserver} has the correct outFile. """ with StringIO() as fileHandle: observer = textFileLogObserver(fileHandle) self.assertIs(observer._outFile, fileHandle) def test_timeFormat(self) -> None: """ Returned L{FileLogObserver} has the correct outFile. """ with StringIO() as fileHandle: observer = textFileLogObserver(fileHandle, timeFormat="%f") observer(dict(log_format="XYZZY", log_time=112345.6)) self.assertEqual(fileHandle.getvalue(), "600000 [-#-] XYZZY\n") def test_observeFailure(self) -> None: """ If the C{"log_failure"} key exists in an event, the observer appends the failure's traceback to the output. """ with StringIO() as fileHandle: observer = textFileLogObserver(fileHandle) try: 1 / 0 except ZeroDivisionError: failure = Failure() event = dict(log_failure=failure) observer(event) output = fileHandle.getvalue() self.assertTrue( output.split("\n")[1].startswith("\tTraceback "), msg=repr(output) ) def test_observeFailureThatRaisesInGetTraceback(self) -> None: """ If the C{"log_failure"} key exists in an event, and contains an object that raises when you call its C{getTraceback()}, then the observer appends a message noting the problem, instead of raising. """ with StringIO() as fileHandle: observer = textFileLogObserver(fileHandle) event = dict(log_failure=object()) # object has no getTraceback() observer(event) output = fileHandle.getvalue() expected = "(UNABLE TO OBTAIN TRACEBACK FROM EVENT)" self.assertIn(expected, output) class DummyFile: """ File that counts writes and flushes. """ def __init__(self) -> None: self.writes = 0 self.flushes = 0 def write(self, data: AnyStr) -> None: """ Write data. @param data: data """ self.writes += 1 def flush(self) -> None: """ Flush buffers. """ self.flushes += 1 def __enter__(self) -> "DummyFile": return self def __exit__( self, exc_type: Optional[Type[BaseException]], exc_value: Optional[BaseException], traceback: Optional[TracebackType], ) -> Optional[bool]: pass PK ͪ�Z8»�* * test_observer.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Test cases for L{twisted.logger._observer}. """ from typing import Dict, List, Tuple, cast from zope.interface import implementer from zope.interface.exceptions import BrokenMethodImplementation from zope.interface.verify import verifyObject from twisted.trial import unittest from .._interfaces import ILogObserver, LogEvent from .._logger import Logger from .._observer import LogPublisher class LogPublisherTests(unittest.TestCase): """ Tests for L{LogPublisher}. """ def test_interface(self) -> None: """ L{LogPublisher} is an L{ILogObserver}. """ publisher = LogPublisher() try: verifyObject(ILogObserver, publisher) except BrokenMethodImplementation as e: self.fail(e) def test_observers(self) -> None: """ L{LogPublisher.observers} returns the observers. """ o1 = cast(ILogObserver, lambda e: None) o2 = cast(ILogObserver, lambda e: None) publisher = LogPublisher(o1, o2) self.assertEqual({o1, o2}, set(publisher._observers)) def test_addObserver(self) -> None: """ L{LogPublisher.addObserver} adds an observer. """ o1 = cast(ILogObserver, lambda e: None) o2 = cast(ILogObserver, lambda e: None) o3 = cast(ILogObserver, lambda e: None) publisher = LogPublisher(o1, o2) publisher.addObserver(o3) self.assertEqual({o1, o2, o3}, set(publisher._observers)) def test_addObserverNotCallable(self) -> None: """ L{LogPublisher.addObserver} refuses to add an observer that's not callable. """ publisher = LogPublisher() self.assertRaises(TypeError, publisher.addObserver, object()) def test_removeObserver(self) -> None: """ L{LogPublisher.removeObserver} removes an observer. """ o1 = cast(ILogObserver, lambda e: None) o2 = cast(ILogObserver, lambda e: None) o3 = cast(ILogObserver, lambda e: None) publisher = LogPublisher(o1, o2, o3) publisher.removeObserver(o2) self.assertEqual({o1, o3}, set(publisher._observers)) def test_removeObserverNotRegistered(self) -> None: """ L{LogPublisher.removeObserver} removes an observer that is not registered. """ o1 = cast(ILogObserver, lambda e: None) o2 = cast(ILogObserver, lambda e: None) o3 = cast(ILogObserver, lambda e: None) publisher = LogPublisher(o1, o2) publisher.removeObserver(o3) self.assertEqual({o1, o2}, set(publisher._observers)) def test_fanOut(self) -> None: """ L{LogPublisher} calls its observers. """ event = dict(foo=1, bar=2) events1: List[LogEvent] = [] events2: List[LogEvent] = [] events3: List[LogEvent] = [] o1 = cast(ILogObserver, events1.append) o2 = cast(ILogObserver, events2.append) o3 = cast(ILogObserver, events3.append) publisher = LogPublisher(o1, o2, o3) publisher(event) self.assertIn(event, events1) self.assertIn(event, events2) self.assertIn(event, events3) def test_observerRaises(self) -> None: """ Observer raises an exception during fan out: a failure is logged, but not re-raised. Life goes on. """ event = dict(foo=1, bar=2) exception = RuntimeError("ARGH! EVIL DEATH!") events: List[LogEvent] = [] @implementer(ILogObserver) def observer(event: LogEvent) -> None: shouldRaise = not events events.append(event) if shouldRaise: raise exception collector: List[LogEvent] = [] publisher = LogPublisher(observer, cast(ILogObserver, collector.append)) publisher(event) # Verify that the observer saw my event self.assertIn(event, events) # Verify that the observer raised my exception errors = [e["log_failure"] for e in collector if "log_failure" in e] self.assertEqual(len(errors), 1) self.assertIs(errors[0].value, exception) # Make sure the exceptional observer does not receive its own error. self.assertEqual(len(events), 1) def test_observerRaisesAndLoggerHatesMe(self) -> None: """ Observer raises an exception during fan out and the publisher's Logger pukes when the failure is reported. The exception does not propagate back to the caller. """ event = dict(foo=1, bar=2) exception = RuntimeError("ARGH! EVIL DEATH!") @implementer(ILogObserver) def observer(event: LogEvent) -> None: raise RuntimeError("Sad panda") class GurkLogger(Logger): def failure(self, *args: object, **kwargs: object) -> None: raise exception publisher = LogPublisher(observer) publisher.log = GurkLogger() publisher(event) # Here, the lack of an exception thus far is a success, of sorts def test_trace(self) -> None: """ Tracing keeps track of forwarding to observers. """ event = dict(foo=1, bar=2, log_trace=[]) traces: Dict[int, Tuple[Tuple[Logger, ILogObserver]]] = {} # Copy trace to a tuple; otherwise, both observers will store the same # mutable list, and we won't be able to see o1's view distinctly. @implementer(ILogObserver) def o1(e: LogEvent) -> None: traces.setdefault( 1, cast(Tuple[Tuple[Logger, ILogObserver]], tuple(e["log_trace"])) ) @implementer(ILogObserver) def o2(e: LogEvent) -> None: traces.setdefault( 2, cast(Tuple[Tuple[Logger, ILogObserver]], tuple(e["log_trace"])) ) publisher = LogPublisher(o1, o2) publisher(event) self.assertEqual(traces[1], ((publisher, o1),)) self.assertEqual(traces[2], ((publisher, o1), (publisher, o2))) PK F��Z�!��} �} test_insults.pynu �[��� # -*- test-case-name: twisted.conch.test.test_insults -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. import textwrap from typing import Optional, Type from twisted.conch.insults.insults import ( BLINK, CS_ALTERNATE, CS_ALTERNATE_SPECIAL, CS_DRAWING, CS_UK, CS_US, G0, G1, UNDERLINE, ClientProtocol, ServerProtocol, modes, privateModes, ) from twisted.internet.protocol import Protocol from twisted.python.compat import iterbytes from twisted.python.constants import ValueConstant, Values from twisted.test.proto_helpers import StringTransport from twisted.trial import unittest def _getattr(mock, name): return super(Mock, mock).__getattribute__(name) def occurrences(mock): return _getattr(mock, "occurrences") def methods(mock): return _getattr(mock, "methods") def _append(mock, obj): occurrences(mock).append(obj) default = object() def _ecmaCodeTableCoordinate(column, row): """ Return the byte in 7- or 8-bit code table identified by C{column} and C{row}. "An 8-bit code table consists of 256 positions arranged in 16 columns and 16 rows. The columns and rows are numbered 00 to 15." "A 7-bit code table consists of 128 positions arranged in 8 columns and 16 rows. The columns are numbered 00 to 07 and the rows 00 to 15 (see figure 1)." p.5 of "Standard ECMA-35: Character Code Structure and Extension Techniques", 6th Edition (December 1994). """ # 8 and 15 both happen to take up 4 bits, so the first number # should be shifted by 4 for both the 7- and 8-bit tables. return bytes(bytearray([(column << 4) | row])) def _makeControlFunctionSymbols(name, colOffset, names, doc): # the value for each name is the concatenation of the bit values # of its x, y locations, with an offset of 4 added to its x value. # so CUP is (0 + 4, 8) = (4, 8) = 4||8 = 1001000 = 72 = b"H" # this is how it's defined in the standard! attrs = { name: ValueConstant(_ecmaCodeTableCoordinate(i + colOffset, j)) for j, row in enumerate(names) for i, name in enumerate(row) if name } attrs["__doc__"] = doc return type(name, (Values,), attrs) CSFinalByte = _makeControlFunctionSymbols( "CSFinalByte", colOffset=4, names=[ # 4, 5, 6 ["ICH", "DCH", "HPA"], ["CUU", "SSE", "HPR"], ["CUD", "CPR", "REP"], ["CUF", "SU", "DA"], ["CUB", "SD", "VPA"], ["CNL", "NP", "VPR"], ["CPL", "PP", "HVP"], ["CHA", "CTC", "TBC"], ["CUP", "ECH", "SM"], ["CHT", "CVT", "MC"], ["ED", "CBT", "HPB"], ["EL", "SRS", "VPB"], ["IL", "PTX", "RM"], ["DL", "SDS", "SGR"], ["EF", "SIMD", "DSR"], ["EA", None, "DAQ"], ], doc=textwrap.dedent( """ Symbolic constants for all control sequence final bytes that do not imply intermediate bytes. This happens to cover movement control sequences. See page 11 of "Standard ECMA 48: Control Functions for Coded Character Sets", 5th Edition (June 1991). Each L{ValueConstant} maps a control sequence name to L{bytes} """ ), ) C1SevenBit = _makeControlFunctionSymbols( "C1SevenBit", colOffset=4, names=[ [None, "DCS"], [None, "PU1"], ["BPH", "PU2"], ["NBH", "STS"], [None, "CCH"], ["NEL", "MW"], ["SSA", "SPA"], ["ESA", "EPA"], ["HTS", "SOS"], ["HTJ", None], ["VTS", "SCI"], ["PLD", "CSI"], ["PLU", "ST"], ["RI", "OSC"], ["SS2", "PM"], ["SS3", "APC"], ], doc=textwrap.dedent( """ Symbolic constants for all 7 bit versions of the C1 control functions See page 9 "Standard ECMA 48: Control Functions for Coded Character Sets", 5th Edition (June 1991). Each L{ValueConstant} maps a control sequence name to L{bytes} """ ), ) class Mock: callReturnValue = default def __init__(self, methods=None, callReturnValue=default): """ @param methods: Mapping of names to return values @param callReturnValue: object __call__ should return """ self.occurrences = [] if methods is None: methods = {} self.methods = methods if callReturnValue is not default: self.callReturnValue = callReturnValue def __call__(self, *a, **kw): returnValue = _getattr(self, "callReturnValue") if returnValue is default: returnValue = Mock() # _getattr(self, 'occurrences').append(('__call__', returnValue, a, kw)) _append(self, ("__call__", returnValue, a, kw)) return returnValue def __getattribute__(self, name): methods = _getattr(self, "methods") if name in methods: attrValue = Mock(callReturnValue=methods[name]) else: attrValue = Mock() # _getattr(self, 'occurrences').append((name, attrValue)) _append(self, (name, attrValue)) return attrValue class MockMixin: def assertCall( self, occurrence, methodName, expectedPositionalArgs=(), expectedKeywordArgs={} ): attr, mock = occurrence self.assertEqual(attr, methodName) self.assertEqual(len(occurrences(mock)), 1) [(call, result, args, kw)] = occurrences(mock) self.assertEqual(call, "__call__") self.assertEqual(args, expectedPositionalArgs) self.assertEqual(kw, expectedKeywordArgs) return result _byteGroupingTestTemplate = """\ def testByte%(groupName)s(self): transport = StringTransport() proto = Mock() parser = self.protocolFactory(lambda: proto) parser.factory = self parser.makeConnection(transport) bytes = self.TEST_BYTES while bytes: chunk = bytes[:%(bytesPer)d] bytes = bytes[%(bytesPer)d:] parser.dataReceived(chunk) self.verifyResults(transport, proto, parser) """ class ByteGroupingsMixin(MockMixin): protocolFactory: Optional[Type[Protocol]] = None for word, n in [ ("Pairs", 2), ("Triples", 3), ("Quads", 4), ("Quints", 5), ("Sexes", 6), ]: exec(_byteGroupingTestTemplate % {"groupName": word, "bytesPer": n}) del word, n def verifyResults(self, transport, proto, parser): result = self.assertCall(occurrences(proto).pop(0), "makeConnection", (parser,)) self.assertEqual(occurrences(result), []) del _byteGroupingTestTemplate class ServerArrowKeysTests(ByteGroupingsMixin, unittest.TestCase): protocolFactory = ServerProtocol # All the arrow keys once TEST_BYTES = b"\x1b[A\x1b[B\x1b[C\x1b[D" def verifyResults(self, transport, proto, parser): ByteGroupingsMixin.verifyResults(self, transport, proto, parser) for arrow in ( parser.UP_ARROW, parser.DOWN_ARROW, parser.RIGHT_ARROW, parser.LEFT_ARROW, ): result = self.assertCall( occurrences(proto).pop(0), "keystrokeReceived", (arrow, None) ) self.assertEqual(occurrences(result), []) self.assertFalse(occurrences(proto)) class PrintableCharactersTests(ByteGroupingsMixin, unittest.TestCase): protocolFactory = ServerProtocol # Some letters and digits, first on their own, then capitalized, # then modified with alt TEST_BYTES = b"abc123ABC!@#\x1ba\x1bb\x1bc\x1b1\x1b2\x1b3" def verifyResults(self, transport, proto, parser): ByteGroupingsMixin.verifyResults(self, transport, proto, parser) for char in iterbytes(b"abc123ABC!@#"): result = self.assertCall( occurrences(proto).pop(0), "keystrokeReceived", (char, None) ) self.assertEqual(occurrences(result), []) for char in iterbytes(b"abc123"): result = self.assertCall( occurrences(proto).pop(0), "keystrokeReceived", (char, parser.ALT) ) self.assertEqual(occurrences(result), []) occs = occurrences(proto) self.assertFalse(occs, f"{occs!r} should have been []") class ServerFunctionKeysTests(ByteGroupingsMixin, unittest.TestCase): """Test for parsing and dispatching function keys (F1 - F12)""" protocolFactory = ServerProtocol byteList = [] for byteCodes in ( b"OP", b"OQ", b"OR", b"OS", # F1 - F4 b"15~", b"17~", b"18~", b"19~", # F5 - F8 b"20~", b"21~", b"23~", b"24~", ): # F9 - F12 byteList.append(b"\x1b[" + byteCodes) TEST_BYTES = b"".join(byteList) del byteList, byteCodes def verifyResults(self, transport, proto, parser): ByteGroupingsMixin.verifyResults(self, transport, proto, parser) for funcNum in range(1, 13): funcArg = getattr(parser, "F%d" % (funcNum,)) result = self.assertCall( occurrences(proto).pop(0), "keystrokeReceived", (funcArg, None) ) self.assertEqual(occurrences(result), []) self.assertFalse(occurrences(proto)) class ClientCursorMovementTests(ByteGroupingsMixin, unittest.TestCase): protocolFactory = ClientProtocol d2 = b"\x1b[2B" r4 = b"\x1b[4C" u1 = b"\x1b[A" l2 = b"\x1b[2D" # Move the cursor down two, right four, up one, left two, up one, left two TEST_BYTES = d2 + r4 + u1 + l2 + u1 + l2 del d2, r4, u1, l2 def verifyResults(self, transport, proto, parser): ByteGroupingsMixin.verifyResults(self, transport, proto, parser) for (method, count) in [ ("Down", 2), ("Forward", 4), ("Up", 1), ("Backward", 2), ("Up", 1), ("Backward", 2), ]: result = self.assertCall( occurrences(proto).pop(0), "cursor" + method, (count,) ) self.assertEqual(occurrences(result), []) self.assertFalse(occurrences(proto)) class ClientControlSequencesTests(unittest.TestCase, MockMixin): def setUp(self): self.transport = StringTransport() self.proto = Mock() self.parser = ClientProtocol(lambda: self.proto) self.parser.factory = self self.parser.makeConnection(self.transport) result = self.assertCall( occurrences(self.proto).pop(0), "makeConnection", (self.parser,) ) self.assertFalse(occurrences(result)) def testSimpleCardinals(self): self.parser.dataReceived( b"".join( b"\x1b[" + n + ch for ch in iterbytes(b"BACD") for n in (b"", b"2", b"20", b"200") ) ) occs = occurrences(self.proto) for meth in ("Down", "Up", "Forward", "Backward"): for count in (1, 2, 20, 200): result = self.assertCall(occs.pop(0), "cursor" + meth, (count,)) self.assertFalse(occurrences(result)) self.assertFalse(occs) def testScrollRegion(self): self.parser.dataReceived(b"\x1b[5;22r\x1b[r") occs = occurrences(self.proto) result = self.assertCall(occs.pop(0), "setScrollRegion", (5, 22)) self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "setScrollRegion", (None, None)) self.assertFalse(occurrences(result)) self.assertFalse(occs) def testHeightAndWidth(self): self.parser.dataReceived(b"\x1b#3\x1b#4\x1b#5\x1b#6") occs = occurrences(self.proto) result = self.assertCall(occs.pop(0), "doubleHeightLine", (True,)) self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "doubleHeightLine", (False,)) self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "singleWidthLine") self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "doubleWidthLine") self.assertFalse(occurrences(result)) self.assertFalse(occs) def testCharacterSet(self): self.parser.dataReceived( b"".join( [ b"".join([b"\x1b" + g + n for n in iterbytes(b"AB012")]) for g in iterbytes(b"()") ] ) ) occs = occurrences(self.proto) for which in (G0, G1): for charset in ( CS_UK, CS_US, CS_DRAWING, CS_ALTERNATE, CS_ALTERNATE_SPECIAL, ): result = self.assertCall( occs.pop(0), "selectCharacterSet", (charset, which) ) self.assertFalse(occurrences(result)) self.assertFalse(occs) def testShifting(self): self.parser.dataReceived(b"\x15\x14") occs = occurrences(self.proto) result = self.assertCall(occs.pop(0), "shiftIn") self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "shiftOut") self.assertFalse(occurrences(result)) self.assertFalse(occs) def testSingleShifts(self): self.parser.dataReceived(b"\x1bN\x1bO") occs = occurrences(self.proto) result = self.assertCall(occs.pop(0), "singleShift2") self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "singleShift3") self.assertFalse(occurrences(result)) self.assertFalse(occs) def testKeypadMode(self): self.parser.dataReceived(b"\x1b=\x1b>") occs = occurrences(self.proto) result = self.assertCall(occs.pop(0), "applicationKeypadMode") self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "numericKeypadMode") self.assertFalse(occurrences(result)) self.assertFalse(occs) def testCursor(self): self.parser.dataReceived(b"\x1b7\x1b8") occs = occurrences(self.proto) result = self.assertCall(occs.pop(0), "saveCursor") self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "restoreCursor") self.assertFalse(occurrences(result)) self.assertFalse(occs) def testReset(self): self.parser.dataReceived(b"\x1bc") occs = occurrences(self.proto) result = self.assertCall(occs.pop(0), "reset") self.assertFalse(occurrences(result)) self.assertFalse(occs) def testIndex(self): self.parser.dataReceived(b"\x1bD\x1bM\x1bE") occs = occurrences(self.proto) result = self.assertCall(occs.pop(0), "index") self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "reverseIndex") self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "nextLine") self.assertFalse(occurrences(result)) self.assertFalse(occs) def testModes(self): self.parser.dataReceived( b"\x1b[" + b";".join(b"%d" % (m,) for m in [modes.KAM, modes.IRM, modes.LNM]) + b"h" ) self.parser.dataReceived( b"\x1b[" + b";".join(b"%d" % (m,) for m in [modes.KAM, modes.IRM, modes.LNM]) + b"l" ) occs = occurrences(self.proto) result = self.assertCall( occs.pop(0), "setModes", ([modes.KAM, modes.IRM, modes.LNM],) ) self.assertFalse(occurrences(result)) result = self.assertCall( occs.pop(0), "resetModes", ([modes.KAM, modes.IRM, modes.LNM],) ) self.assertFalse(occurrences(result)) self.assertFalse(occs) def testErasure(self): self.parser.dataReceived(b"\x1b[K\x1b[1K\x1b[2K\x1b[J\x1b[1J\x1b[2J\x1b[3P") occs = occurrences(self.proto) for meth in ( "eraseToLineEnd", "eraseToLineBeginning", "eraseLine", "eraseToDisplayEnd", "eraseToDisplayBeginning", "eraseDisplay", ): result = self.assertCall(occs.pop(0), meth) self.assertFalse(occurrences(result)) result = self.assertCall(occs.pop(0), "deleteCharacter", (3,)) self.assertFalse(occurrences(result)) self.assertFalse(occs) def testLineDeletion(self): self.parser.dataReceived(b"\x1b[M\x1b[3M") occs = occurrences(self.proto) for arg in (1, 3): result = self.assertCall(occs.pop(0), "deleteLine", (arg,)) self.assertFalse(occurrences(result)) self.assertFalse(occs) def testLineInsertion(self): self.parser.dataReceived(b"\x1b[L\x1b[3L") occs = occurrences(self.proto) for arg in (1, 3): result = self.assertCall(occs.pop(0), "insertLine", (arg,)) self.assertFalse(occurrences(result)) self.assertFalse(occs) def testCursorPosition(self): methods(self.proto)["reportCursorPosition"] = (6, 7) self.parser.dataReceived(b"\x1b[6n") self.assertEqual(self.transport.value(), b"\x1b[7;8R") occs = occurrences(self.proto) result = self.assertCall(occs.pop(0), "reportCursorPosition") # This isn't really an interesting assert, since it only tests that # our mock setup is working right, but I'll include it anyway. self.assertEqual(result, (6, 7)) def test_applicationDataBytes(self): """ Contiguous non-control bytes are passed to a single call to the C{write} method of the terminal to which the L{ClientProtocol} is connected. """ occs = occurrences(self.proto) self.parser.dataReceived(b"a") self.assertCall(occs.pop(0), "write", (b"a",)) self.parser.dataReceived(b"bc") self.assertCall(occs.pop(0), "write", (b"bc",)) def _applicationDataTest(self, data, calls): occs = occurrences(self.proto) self.parser.dataReceived(data) while calls: self.assertCall(occs.pop(0), *calls.pop(0)) self.assertFalse(occs, f"No other calls should happen: {occs!r}") def test_shiftInAfterApplicationData(self): """ Application data bytes followed by a shift-in command are passed to a call to C{write} before the terminal's C{shiftIn} method is called. """ self._applicationDataTest(b"ab\x15", [("write", (b"ab",)), ("shiftIn",)]) def test_shiftOutAfterApplicationData(self): """ Application data bytes followed by a shift-out command are passed to a call to C{write} before the terminal's C{shiftOut} method is called. """ self._applicationDataTest(b"ab\x14", [("write", (b"ab",)), ("shiftOut",)]) def test_cursorBackwardAfterApplicationData(self): """ Application data bytes followed by a cursor-backward command are passed to a call to C{write} before the terminal's C{cursorBackward} method is called. """ self._applicationDataTest(b"ab\x08", [("write", (b"ab",)), ("cursorBackward",)]) def test_escapeAfterApplicationData(self): """ Application data bytes followed by an escape character are passed to a call to C{write} before the terminal's handler method for the escape is called. """ # Test a short escape self._applicationDataTest(b"ab\x1bD", [("write", (b"ab",)), ("index",)]) # And a long escape self._applicationDataTest( b"ab\x1b[4h", [("write", (b"ab",)), ("setModes", ([4],))] ) # There's some other cases too, but they're all handled by the same # codepaths as above. class ServerProtocolOutputTests(unittest.TestCase): """ Tests for the bytes L{ServerProtocol} writes to its transport when its methods are called. """ # From ECMA 48: CSI is represented by bit combinations 01/11 # (representing ESC) and 05/11 in a 7-bit code or by bit # combination 09/11 in an 8-bit code ESC = _ecmaCodeTableCoordinate(1, 11) CSI = ESC + _ecmaCodeTableCoordinate(5, 11) def setUp(self): self.protocol = ServerProtocol() self.transport = StringTransport() self.protocol.makeConnection(self.transport) def test_cursorUp(self): """ L{ServerProtocol.cursorUp} writes the control sequence ending with L{CSFinalByte.CUU} to its transport. """ self.protocol.cursorUp(1) self.assertEqual( self.transport.value(), self.CSI + b"1" + CSFinalByte.CUU.value ) def test_cursorDown(self): """ L{ServerProtocol.cursorDown} writes the control sequence ending with L{CSFinalByte.CUD} to its transport. """ self.protocol.cursorDown(1) self.assertEqual( self.transport.value(), self.CSI + b"1" + CSFinalByte.CUD.value ) def test_cursorForward(self): """ L{ServerProtocol.cursorForward} writes the control sequence ending with L{CSFinalByte.CUF} to its transport. """ self.protocol.cursorForward(1) self.assertEqual( self.transport.value(), self.CSI + b"1" + CSFinalByte.CUF.value ) def test_cursorBackward(self): """ L{ServerProtocol.cursorBackward} writes the control sequence ending with L{CSFinalByte.CUB} to its transport. """ self.protocol.cursorBackward(1) self.assertEqual( self.transport.value(), self.CSI + b"1" + CSFinalByte.CUB.value ) def test_cursorPosition(self): """ L{ServerProtocol.cursorPosition} writes a control sequence ending with L{CSFinalByte.CUP} and containing the expected coordinates to its transport. """ self.protocol.cursorPosition(0, 0) self.assertEqual( self.transport.value(), self.CSI + b"1;1" + CSFinalByte.CUP.value ) def test_cursorHome(self): """ L{ServerProtocol.cursorHome} writes a control sequence ending with L{CSFinalByte.CUP} and no parameters, so that the client defaults to (1, 1). """ self.protocol.cursorHome() self.assertEqual(self.transport.value(), self.CSI + CSFinalByte.CUP.value) def test_index(self): """ L{ServerProtocol.index} writes the control sequence ending in the 8-bit code table coordinates 4, 4. Note that ECMA48 5th Edition removes C{IND}. """ self.protocol.index() self.assertEqual( self.transport.value(), self.ESC + _ecmaCodeTableCoordinate(4, 4) ) def test_reverseIndex(self): """ L{ServerProtocol.reverseIndex} writes the control sequence ending in the L{C1SevenBit.RI}. """ self.protocol.reverseIndex() self.assertEqual(self.transport.value(), self.ESC + C1SevenBit.RI.value) def test_nextLine(self): """ L{ServerProtocol.nextLine} writes C{"\r\n"} to its transport. """ # Why doesn't it write ESC E? Because ESC E is poorly supported. For # example, gnome-terminal (many different versions) fails to scroll if # it receives ESC E and the cursor is already on the last row. self.protocol.nextLine() self.assertEqual(self.transport.value(), b"\r\n") def test_setModes(self): """ L{ServerProtocol.setModes} writes a control sequence containing the requested modes and ending in the L{CSFinalByte.SM}. """ modesToSet = [modes.KAM, modes.IRM, modes.LNM] self.protocol.setModes(modesToSet) self.assertEqual( self.transport.value(), self.CSI + b";".join(b"%d" % (m,) for m in modesToSet) + CSFinalByte.SM.value, ) def test_setPrivateModes(self): """ L{ServerProtocol.setPrivatesModes} writes a control sequence containing the requested private modes and ending in the L{CSFinalByte.SM}. """ privateModesToSet = [ privateModes.ERROR, privateModes.COLUMN, privateModes.ORIGIN, ] self.protocol.setModes(privateModesToSet) self.assertEqual( self.transport.value(), self.CSI + b";".join(b"%d" % (m,) for m in privateModesToSet) + CSFinalByte.SM.value, ) def test_resetModes(self): """ L{ServerProtocol.resetModes} writes the control sequence ending in the L{CSFinalByte.RM}. """ modesToSet = [modes.KAM, modes.IRM, modes.LNM] self.protocol.resetModes(modesToSet) self.assertEqual( self.transport.value(), self.CSI + b";".join(b"%d" % (m,) for m in modesToSet) + CSFinalByte.RM.value, ) def test_singleShift2(self): """ L{ServerProtocol.singleShift2} writes an escape sequence followed by L{C1SevenBit.SS2} """ self.protocol.singleShift2() self.assertEqual(self.transport.value(), self.ESC + C1SevenBit.SS2.value) def test_singleShift3(self): """ L{ServerProtocol.singleShift3} writes an escape sequence followed by L{C1SevenBit.SS3} """ self.protocol.singleShift3() self.assertEqual(self.transport.value(), self.ESC + C1SevenBit.SS3.value) def test_selectGraphicRendition(self): """ L{ServerProtocol.selectGraphicRendition} writes a control sequence containing the requested attributes and ending with L{CSFinalByte.SGR} """ self.protocol.selectGraphicRendition(str(BLINK), str(UNDERLINE)) self.assertEqual( self.transport.value(), self.CSI + b"%d;%d" % (BLINK, UNDERLINE) + CSFinalByte.SGR.value, ) def test_horizontalTabulationSet(self): """ L{ServerProtocol.horizontalTabulationSet} writes the escape sequence ending in L{C1SevenBit.HTS} """ self.protocol.horizontalTabulationSet() self.assertEqual(self.transport.value(), self.ESC + C1SevenBit.HTS.value) def test_eraseToLineEnd(self): """ L{ServerProtocol.eraseToLineEnd} writes the control sequence sequence ending in L{CSFinalByte.EL} and no parameters, forcing the client to default to 0 (from the active present position's current location to the end of the line.) """ self.protocol.eraseToLineEnd() self.assertEqual(self.transport.value(), self.CSI + CSFinalByte.EL.value) def test_eraseToLineBeginning(self): """ L{ServerProtocol.eraseToLineBeginning} writes the control sequence sequence ending in L{CSFinalByte.EL} and a parameter of 1 (from the beginning of the line up to and include the active present position's current location.) """ self.protocol.eraseToLineBeginning() self.assertEqual(self.transport.value(), self.CSI + b"1" + CSFinalByte.EL.value) def test_eraseLine(self): """ L{ServerProtocol.eraseLine} writes the control sequence sequence ending in L{CSFinalByte.EL} and a parameter of 2 (the entire line.) """ self.protocol.eraseLine() self.assertEqual(self.transport.value(), self.CSI + b"2" + CSFinalByte.EL.value) def test_eraseToDisplayEnd(self): """ L{ServerProtocol.eraseToDisplayEnd} writes the control sequence sequence ending in L{CSFinalByte.ED} and no parameters, forcing the client to default to 0 (from the active present position's current location to the end of the page.) """ self.protocol.eraseToDisplayEnd() self.assertEqual(self.transport.value(), self.CSI + CSFinalByte.ED.value) def test_eraseToDisplayBeginning(self): """ L{ServerProtocol.eraseToDisplayBeginning} writes the control sequence sequence ending in L{CSFinalByte.ED} a parameter of 1 (from the beginning of the page up to and include the active present position's current location.) """ self.protocol.eraseToDisplayBeginning() self.assertEqual(self.transport.value(), self.CSI + b"1" + CSFinalByte.ED.value) def test_eraseToDisplay(self): """ L{ServerProtocol.eraseDisplay} writes the control sequence sequence ending in L{CSFinalByte.ED} a parameter of 2 (the entire page) """ self.protocol.eraseDisplay() self.assertEqual(self.transport.value(), self.CSI + b"2" + CSFinalByte.ED.value) def test_deleteCharacter(self): """ L{ServerProtocol.deleteCharacter} writes the control sequence containing the number of characters to delete and ending in L{CSFinalByte.DCH} """ self.protocol.deleteCharacter(4) self.assertEqual( self.transport.value(), self.CSI + b"4" + CSFinalByte.DCH.value ) def test_insertLine(self): """ L{ServerProtocol.insertLine} writes the control sequence containing the number of lines to insert and ending in L{CSFinalByte.IL} """ self.protocol.insertLine(5) self.assertEqual(self.transport.value(), self.CSI + b"5" + CSFinalByte.IL.value) def test_deleteLine(self): """ L{ServerProtocol.deleteLine} writes the control sequence containing the number of lines to delete and ending in L{CSFinalByte.DL} """ self.protocol.deleteLine(6) self.assertEqual(self.transport.value(), self.CSI + b"6" + CSFinalByte.DL.value) def test_setScrollRegionNoArgs(self): """ With no arguments, L{ServerProtocol.setScrollRegion} writes a control sequence with no parameters, but a parameter separator, and ending in C{b'r'}. """ self.protocol.setScrollRegion() self.assertEqual(self.transport.value(), self.CSI + b";" + b"r") def test_setScrollRegionJustFirst(self): """ With just a value for its C{first} argument, L{ServerProtocol.setScrollRegion} writes a control sequence with that parameter, a parameter separator, and finally a C{b'r'}. """ self.protocol.setScrollRegion(first=1) self.assertEqual(self.transport.value(), self.CSI + b"1;" + b"r") def test_setScrollRegionJustLast(self): """ With just a value for its C{last} argument, L{ServerProtocol.setScrollRegion} writes a control sequence with a parameter separator, that parameter, and finally a C{b'r'}. """ self.protocol.setScrollRegion(last=1) self.assertEqual(self.transport.value(), self.CSI + b";1" + b"r") def test_setScrollRegionFirstAndLast(self): """ When given both C{first} and C{last} L{ServerProtocol.setScrollRegion} writes a control sequence with the first parameter, a parameter separator, the last parameter, and finally a C{b'r'}. """ self.protocol.setScrollRegion(first=1, last=2) self.assertEqual(self.transport.value(), self.CSI + b"1;2" + b"r") def test_reportCursorPosition(self): """ L{ServerProtocol.reportCursorPosition} writes a control sequence ending in L{CSFinalByte.DSR} with a parameter of 6 (the Device Status Report returns the current active position.) """ self.protocol.reportCursorPosition() self.assertEqual( self.transport.value(), self.CSI + b"6" + CSFinalByte.DSR.value ) PK F��Z`l�_ _ test_ckeygen.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.scripts.ckeygen}. """ import getpass import subprocess import sys import os from io import StringIO from twisted.conch.test.keydata import ( privateECDSA_openssh, privateEd25519_openssh_new, privateRSA_openssh, privateRSA_openssh_encrypted, publicRSA_openssh, ) from twisted.python.filepath import FilePath from twisted.python.reflect import requireModule from twisted.trial.unittest import TestCase if requireModule("cryptography") and requireModule("pyasn1"): from twisted.conch.scripts.ckeygen import ( _saveKey, changePassPhrase, displayPublicKey, enumrepresentation, printFingerprint, ) from twisted.conch.ssh.keys import ( BadFingerPrintFormat, BadKeyError, FingerprintFormats, Key, ) else: skip = "cryptography and pyasn1 required for twisted.conch.scripts.ckeygen" def makeGetpass(*passphrases): """ Return a callable to patch C{getpass.getpass}. Yields a passphrase each time called. Use case is to provide an old, then new passphrase(s) as if requested interactively. @param passphrases: The list of passphrases returned, one per each call. @return: A callable to patch C{getpass.getpass}. """ passphrases = iter(passphrases) def fakeGetpass(_): return next(passphrases) return fakeGetpass class KeyGenTests(TestCase): """ Tests for various functions used to implement the I{ckeygen} script. """ def setUp(self): """ Patch C{sys.stdout} so tests can make assertions about what's printed. """ self.stdout = StringIO() self.patch(sys, "stdout", self.stdout) def _testrun(self, keyType, keySize=None, privateKeySubtype=None): filename = self.mktemp() args = ["ckeygen", "-t", keyType, "-f", filename, "--no-passphrase"] if keySize is not None: args.extend(["-b", keySize]) if privateKeySubtype is not None: args.extend(["--private-key-subtype", privateKeySubtype]) try: subprocess.call(args) except FileNotFoundError: args[0] = 'ckeygen3' subprocess.call(args) privKey = Key.fromFile(filename) pubKey = Key.fromFile(filename + ".pub") if keyType == "ecdsa": self.assertEqual(privKey.type(), "EC") elif keyType == "ed25519": self.assertEqual(privKey.type(), "Ed25519") else: self.assertEqual(privKey.type(), keyType.upper()) self.assertTrue(pubKey.isPublic()) def test_keygeneration(self): self._testrun("ecdsa", "384") self._testrun("ecdsa", "384", privateKeySubtype="v1") self._testrun("ecdsa") self._testrun("ecdsa", privateKeySubtype="v1") self._testrun("ed25519") self._testrun("dsa", "2048") self._testrun("dsa", "2048", privateKeySubtype="v1") self._testrun("dsa") self._testrun("dsa", privateKeySubtype="v1") self._testrun("rsa", "2048") self._testrun("rsa", "2048", privateKeySubtype="v1") self._testrun("rsa") self._testrun("rsa", privateKeySubtype="v1") def test_runBadKeytype(self): filename = self.mktemp() with self.assertRaises(subprocess.CalledProcessError): with open(os.devnull, "rb") as devnull: try: subprocess.check_call( ['ckeygen', '-t', 'foo', '-f', filename], stderr=devnull) except FileNotFoundError: subprocess.check_call( ['ckeygen3', '-t', 'foo', '-f', filename], stderr=devnull) def test_enumrepresentation(self): """ L{enumrepresentation} takes a dictionary as input and returns a dictionary with its attributes changed to enum representation. """ options = enumrepresentation({"format": "md5-hex"}) self.assertIs(options["format"], FingerprintFormats.MD5_HEX) def test_enumrepresentationsha256(self): """ Test for format L{FingerprintFormats.SHA256-BASE64}. """ options = enumrepresentation({"format": "sha256-base64"}) self.assertIs(options["format"], FingerprintFormats.SHA256_BASE64) def test_enumrepresentationBadFormat(self): """ Test for unsupported fingerprint format """ with self.assertRaises(BadFingerPrintFormat) as em: enumrepresentation({"format": "sha-base64"}) self.assertEqual( "Unsupported fingerprint format: sha-base64", em.exception.args[0] ) def test_printFingerprint(self): """ L{printFingerprint} writes a line to standard out giving the number of bits of the key, its fingerprint, and the basename of the file from it was read. """ filename = self.mktemp() FilePath(filename).setContent(publicRSA_openssh) printFingerprint({"filename": filename, "format": "md5-hex"}) self.assertEqual( self.stdout.getvalue(), "2048 85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:da temp\n", ) def test_printFingerprintsha256(self): """ L{printFigerprint} will print key fingerprint in L{FingerprintFormats.SHA256-BASE64} format if explicitly specified. """ filename = self.mktemp() FilePath(filename).setContent(publicRSA_openssh) printFingerprint({"filename": filename, "format": "sha256-base64"}) self.assertEqual( self.stdout.getvalue(), "2048 FBTCOoknq0mHy+kpfnY9tDdcAJuWtCpuQMaV3EsvbUI= temp\n", ) def test_printFingerprintBadFingerPrintFormat(self): """ L{printFigerprint} raises C{keys.BadFingerprintFormat} when unsupported formats are requested. """ filename = self.mktemp() FilePath(filename).setContent(publicRSA_openssh) with self.assertRaises(BadFingerPrintFormat) as em: printFingerprint({"filename": filename, "format": "sha-base64"}) self.assertEqual( "Unsupported fingerprint format: sha-base64", em.exception.args[0] ) def test_saveKey(self): """ L{_saveKey} writes the private and public parts of a key to two different files and writes a report of this to standard out. """ base = FilePath(self.mktemp()) base.makedirs() filename = base.child("id_rsa").path key = Key.fromString(privateRSA_openssh) _saveKey(key, {"filename": filename, "pass": "passphrase", "format": "md5-hex"}) self.assertEqual( self.stdout.getvalue(), "Your identification has been saved in %s\n" "Your public key has been saved in %s.pub\n" "The key fingerprint in <FingerprintFormats=MD5_HEX> is:\n" "85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:da\n" % (filename, filename), ) self.assertEqual( key.fromString(base.child("id_rsa").getContent(), None, "passphrase"), key ) self.assertEqual( Key.fromString(base.child("id_rsa.pub").getContent()), key.public() ) def test_saveKeyECDSA(self): """ L{_saveKey} writes the private and public parts of a key to two different files and writes a report of this to standard out. Test with ECDSA key. """ base = FilePath(self.mktemp()) base.makedirs() filename = base.child("id_ecdsa").path key = Key.fromString(privateECDSA_openssh) _saveKey(key, {"filename": filename, "pass": "passphrase", "format": "md5-hex"}) self.assertEqual( self.stdout.getvalue(), "Your identification has been saved in %s\n" "Your public key has been saved in %s.pub\n" "The key fingerprint in <FingerprintFormats=MD5_HEX> is:\n" "1e:ab:83:a6:f2:04:22:99:7c:64:14:d2:ab:fa:f5:16\n" % (filename, filename), ) self.assertEqual( key.fromString(base.child("id_ecdsa").getContent(), None, "passphrase"), key ) self.assertEqual( Key.fromString(base.child("id_ecdsa.pub").getContent()), key.public() ) def test_saveKeyEd25519(self): """ L{_saveKey} writes the private and public parts of a key to two different files and writes a report of this to standard out. Test with Ed25519 key. """ base = FilePath(self.mktemp()) base.makedirs() filename = base.child("id_ed25519").path key = Key.fromString(privateEd25519_openssh_new) _saveKey(key, {"filename": filename, "pass": "passphrase", "format": "md5-hex"}) self.assertEqual( self.stdout.getvalue(), "Your identification has been saved in %s\n" "Your public key has been saved in %s.pub\n" "The key fingerprint in <FingerprintFormats=MD5_HEX> is:\n" "ab:ee:c8:ed:e5:01:1b:45:b7:8d:b2:f0:8f:61:1c:14\n" % (filename, filename), ) self.assertEqual( key.fromString(base.child("id_ed25519").getContent(), None, "passphrase"), key, ) self.assertEqual( Key.fromString(base.child("id_ed25519.pub").getContent()), key.public() ) def test_saveKeysha256(self): """ L{_saveKey} will generate key fingerprint in L{FingerprintFormats.SHA256-BASE64} format if explicitly specified. """ base = FilePath(self.mktemp()) base.makedirs() filename = base.child("id_rsa").path key = Key.fromString(privateRSA_openssh) _saveKey( key, {"filename": filename, "pass": "passphrase", "format": "sha256-base64"} ) self.assertEqual( self.stdout.getvalue(), "Your identification has been saved in %s\n" "Your public key has been saved in %s.pub\n" "The key fingerprint in <FingerprintFormats=SHA256_BASE64> is:\n" "FBTCOoknq0mHy+kpfnY9tDdcAJuWtCpuQMaV3EsvbUI=\n" % (filename, filename), ) self.assertEqual( key.fromString(base.child("id_rsa").getContent(), None, "passphrase"), key ) self.assertEqual( Key.fromString(base.child("id_rsa.pub").getContent()), key.public() ) def test_saveKeyBadFingerPrintformat(self): """ L{_saveKey} raises C{keys.BadFingerprintFormat} when unsupported formats are requested. """ base = FilePath(self.mktemp()) base.makedirs() filename = base.child("id_rsa").path key = Key.fromString(privateRSA_openssh) with self.assertRaises(BadFingerPrintFormat) as em: _saveKey( key, {"filename": filename, "pass": "passphrase", "format": "sha-base64"}, ) self.assertEqual( "Unsupported fingerprint format: sha-base64", em.exception.args[0] ) def test_saveKeyEmptyPassphrase(self): """ L{_saveKey} will choose an empty string for the passphrase if no-passphrase is C{True}. """ base = FilePath(self.mktemp()) base.makedirs() filename = base.child("id_rsa").path key = Key.fromString(privateRSA_openssh) _saveKey( key, {"filename": filename, "no-passphrase": True, "format": "md5-hex"} ) self.assertEqual( key.fromString(base.child("id_rsa").getContent(), None, b""), key ) def test_saveKeyECDSAEmptyPassphrase(self): """ L{_saveKey} will choose an empty string for the passphrase if no-passphrase is C{True}. """ base = FilePath(self.mktemp()) base.makedirs() filename = base.child("id_ecdsa").path key = Key.fromString(privateECDSA_openssh) _saveKey( key, {"filename": filename, "no-passphrase": True, "format": "md5-hex"} ) self.assertEqual(key.fromString(base.child("id_ecdsa").getContent(), None), key) def test_saveKeyEd25519EmptyPassphrase(self): """ L{_saveKey} will choose an empty string for the passphrase if no-passphrase is C{True}. """ base = FilePath(self.mktemp()) base.makedirs() filename = base.child("id_ed25519").path key = Key.fromString(privateEd25519_openssh_new) _saveKey( key, {"filename": filename, "no-passphrase": True, "format": "md5-hex"} ) self.assertEqual( key.fromString(base.child("id_ed25519").getContent(), None), key ) def test_saveKeyNoFilename(self): """ When no path is specified, it will ask for the path used to store the key. """ base = FilePath(self.mktemp()) base.makedirs() keyPath = base.child("custom_key").path import twisted.conch.scripts.ckeygen self.patch(twisted.conch.scripts.ckeygen, "_inputSaveFile", lambda _: keyPath) key = Key.fromString(privateRSA_openssh) _saveKey(key, {"filename": None, "no-passphrase": True, "format": "md5-hex"}) persistedKeyContent = base.child("custom_key").getContent() persistedKey = key.fromString(persistedKeyContent, None, b"") self.assertEqual(key, persistedKey) def test_saveKeySubtypeV1(self): """ L{_saveKey} can be told to write the new private key file in OpenSSH v1 format. """ base = FilePath(self.mktemp()) base.makedirs() filename = base.child("id_rsa").path key = Key.fromString(privateRSA_openssh) _saveKey( key, { "filename": filename, "pass": "passphrase", "format": "md5-hex", "private-key-subtype": "v1", }, ) self.assertEqual( self.stdout.getvalue(), "Your identification has been saved in %s\n" "Your public key has been saved in %s.pub\n" "The key fingerprint in <FingerprintFormats=MD5_HEX> is:\n" "85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:da\n" % (filename, filename), ) privateKeyContent = base.child("id_rsa").getContent() self.assertEqual(key.fromString(privateKeyContent, None, "passphrase"), key) self.assertTrue( privateKeyContent.startswith(b"-----BEGIN OPENSSH PRIVATE KEY-----\n") ) self.assertEqual( Key.fromString(base.child("id_rsa.pub").getContent()), key.public() ) def test_displayPublicKey(self): """ L{displayPublicKey} prints out the public key associated with a given private key. """ filename = self.mktemp() pubKey = Key.fromString(publicRSA_openssh) FilePath(filename).setContent(privateRSA_openssh) displayPublicKey({"filename": filename}) displayed = self.stdout.getvalue().strip("\n") if isinstance(displayed, str): displayed = displayed.encode("ascii") self.assertEqual(displayed, pubKey.toString("openssh")) def test_displayPublicKeyEncrypted(self): """ L{displayPublicKey} prints out the public key associated with a given private key using the given passphrase when it's encrypted. """ filename = self.mktemp() pubKey = Key.fromString(publicRSA_openssh) FilePath(filename).setContent(privateRSA_openssh_encrypted) displayPublicKey({"filename": filename, "pass": "encrypted"}) displayed = self.stdout.getvalue().strip("\n") if isinstance(displayed, str): displayed = displayed.encode("ascii") self.assertEqual(displayed, pubKey.toString("openssh")) def test_displayPublicKeyEncryptedPassphrasePrompt(self): """ L{displayPublicKey} prints out the public key associated with a given private key, asking for the passphrase when it's encrypted. """ filename = self.mktemp() pubKey = Key.fromString(publicRSA_openssh) FilePath(filename).setContent(privateRSA_openssh_encrypted) self.patch(getpass, "getpass", lambda x: "encrypted") displayPublicKey({"filename": filename}) displayed = self.stdout.getvalue().strip("\n") if isinstance(displayed, str): displayed = displayed.encode("ascii") self.assertEqual(displayed, pubKey.toString("openssh")) def test_displayPublicKeyWrongPassphrase(self): """ L{displayPublicKey} fails with a L{BadKeyError} when trying to decrypt an encrypted key with the wrong password. """ filename = self.mktemp() FilePath(filename).setContent(privateRSA_openssh_encrypted) self.assertRaises( BadKeyError, displayPublicKey, {"filename": filename, "pass": "wrong"} ) def test_changePassphrase(self): """ L{changePassPhrase} allows a user to change the passphrase of a private key interactively. """ oldNewConfirm = makeGetpass("encrypted", "newpass", "newpass") self.patch(getpass, "getpass", oldNewConfirm) filename = self.mktemp() FilePath(filename).setContent(privateRSA_openssh_encrypted) changePassPhrase({"filename": filename}) self.assertEqual( self.stdout.getvalue().strip("\n"), "Your identification has been saved with the new passphrase.", ) self.assertNotEqual( privateRSA_openssh_encrypted, FilePath(filename).getContent() ) def test_changePassphraseWithOld(self): """ L{changePassPhrase} allows a user to change the passphrase of a private key, providing the old passphrase and prompting for new one. """ newConfirm = makeGetpass("newpass", "newpass") self.patch(getpass, "getpass", newConfirm) filename = self.mktemp() FilePath(filename).setContent(privateRSA_openssh_encrypted) changePassPhrase({"filename": filename, "pass": "encrypted"}) self.assertEqual( self.stdout.getvalue().strip("\n"), "Your identification has been saved with the new passphrase.", ) self.assertNotEqual( privateRSA_openssh_encrypted, FilePath(filename).getContent() ) def test_changePassphraseWithBoth(self): """ L{changePassPhrase} allows a user to change the passphrase of a private key by providing both old and new passphrases without prompting. """ filename = self.mktemp() FilePath(filename).setContent(privateRSA_openssh_encrypted) changePassPhrase( {"filename": filename, "pass": "encrypted", "newpass": "newencrypt"} ) self.assertEqual( self.stdout.getvalue().strip("\n"), "Your identification has been saved with the new passphrase.", ) self.assertNotEqual( privateRSA_openssh_encrypted, FilePath(filename).getContent() ) def test_changePassphraseWrongPassphrase(self): """ L{changePassPhrase} exits if passed an invalid old passphrase when trying to change the passphrase of a private key. """ filename = self.mktemp() FilePath(filename).setContent(privateRSA_openssh_encrypted) error = self.assertRaises( SystemExit, changePassPhrase, {"filename": filename, "pass": "wrong"} ) self.assertEqual( "Could not change passphrase: old passphrase error", str(error) ) self.assertEqual(privateRSA_openssh_encrypted, FilePath(filename).getContent()) def test_changePassphraseEmptyGetPass(self): """ L{changePassPhrase} exits if no passphrase is specified for the C{getpass} call and the key is encrypted. """ self.patch(getpass, "getpass", makeGetpass("")) filename = self.mktemp() FilePath(filename).setContent(privateRSA_openssh_encrypted) error = self.assertRaises(SystemExit, changePassPhrase, {"filename": filename}) self.assertEqual( "Could not change passphrase: Passphrase must be provided " "for an encrypted key", str(error), ) self.assertEqual(privateRSA_openssh_encrypted, FilePath(filename).getContent()) def test_changePassphraseBadKey(self): """ L{changePassPhrase} exits if the file specified points to an invalid key. """ filename = self.mktemp() FilePath(filename).setContent(b"foobar") error = self.assertRaises(SystemExit, changePassPhrase, {"filename": filename}) expected = "Could not change passphrase: cannot " "guess the type of b'foobar'" self.assertEqual(expected, str(error)) self.assertEqual(b"foobar", FilePath(filename).getContent()) def test_changePassphraseCreateError(self): """ L{changePassPhrase} doesn't modify the key file if an unexpected error happens when trying to create the key with the new passphrase. """ filename = self.mktemp() FilePath(filename).setContent(privateRSA_openssh) def toString(*args, **kwargs): raise RuntimeError("oops") self.patch(Key, "toString", toString) error = self.assertRaises( SystemExit, changePassPhrase, {"filename": filename, "newpass": "newencrypt"}, ) self.assertEqual("Could not change passphrase: oops", str(error)) self.assertEqual(privateRSA_openssh, FilePath(filename).getContent()) def test_changePassphraseEmptyStringError(self): """ L{changePassPhrase} doesn't modify the key file if C{toString} returns an empty string. """ filename = self.mktemp() FilePath(filename).setContent(privateRSA_openssh) def toString(*args, **kwargs): return "" self.patch(Key, "toString", toString) error = self.assertRaises( SystemExit, changePassPhrase, {"filename": filename, "newpass": "newencrypt"}, ) expected = "Could not change passphrase: cannot guess the type of b''" self.assertEqual(expected, str(error)) self.assertEqual(privateRSA_openssh, FilePath(filename).getContent()) def test_changePassphrasePublicKey(self): """ L{changePassPhrase} exits when trying to change the passphrase on a public key, and doesn't change the file. """ filename = self.mktemp() FilePath(filename).setContent(publicRSA_openssh) error = self.assertRaises( SystemExit, changePassPhrase, {"filename": filename, "newpass": "pass"} ) self.assertEqual("Could not change passphrase: key not encrypted", str(error)) self.assertEqual(publicRSA_openssh, FilePath(filename).getContent()) def test_changePassphraseSubtypeV1(self): """ L{changePassPhrase} can be told to write the new private key file in OpenSSH v1 format. """ oldNewConfirm = makeGetpass("encrypted", "newpass", "newpass") self.patch(getpass, "getpass", oldNewConfirm) filename = self.mktemp() FilePath(filename).setContent(privateRSA_openssh_encrypted) changePassPhrase({"filename": filename, "private-key-subtype": "v1"}) self.assertEqual( self.stdout.getvalue().strip("\n"), "Your identification has been saved with the new passphrase.", ) privateKeyContent = FilePath(filename).getContent() self.assertNotEqual(privateRSA_openssh_encrypted, privateKeyContent) self.assertTrue( privateKeyContent.startswith(b"-----BEGIN OPENSSH PRIVATE KEY-----\n") ) PK F��ZCXF@�d �d test_conch.pynu �[��� # -*- test-case-name: twisted.conch.test.test_conch -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. import os import socket import subprocess import sys from itertools import count from unittest import skipIf from zope.interface import implementer from twisted.conch.error import ConchError from twisted.conch.test.keydata import ( privateDSA_openssh, privateRSA_openssh, publicDSA_openssh, publicRSA_openssh, ) from twisted.conch.test.test_ssh import ConchTestRealm from twisted.cred import portal from twisted.internet import defer, protocol, reactor from twisted.internet.error import ProcessExitedAlready from twisted.internet.task import LoopingCall from twisted.internet.utils import getProcessValue from twisted.python import filepath, log, runtime from twisted.python.filepath import FilePath from twisted.python.procutils import which from twisted.python.reflect import requireModule from twisted.trial.unittest import SkipTest, TestCase try: from twisted.conch.test.test_ssh import ( ConchTestServerFactory, conchTestPublicKeyChecker, ) except ImportError: pass pyasn1 = requireModule("pyasn1") cryptography = requireModule("cryptography") if cryptography: from twisted.conch.avatar import ConchUser from twisted.conch.ssh.session import ISession, SSHSession, wrapProtocol else: from twisted.conch.interfaces import ISession class ConchUser: # type: ignore[no-redef] pass try: from twisted.conch.scripts.conch import SSHSession as _StdioInteractingSession except ImportError as e: StdioInteractingSession = None _reason = str(e) del e else: StdioInteractingSession = _StdioInteractingSession def _has_ipv6(): """Returns True if the system can bind an IPv6 address.""" sock = None has_ipv6 = False try: sock = socket.socket(socket.AF_INET6) sock.bind(("::1", 0)) has_ipv6 = True except OSError: pass if sock: sock.close() return has_ipv6 HAS_IPV6 = _has_ipv6() class FakeStdio: """ A fake for testing L{twisted.conch.scripts.conch.SSHSession.eofReceived} and L{twisted.conch.scripts.cftp.SSHSession.eofReceived}. @ivar writeConnLost: A flag which records whether L{loserWriteConnection} has been called. """ writeConnLost = False def loseWriteConnection(self): """ Record the call to loseWriteConnection. """ self.writeConnLost = True class StdioInteractingSessionTests(TestCase): """ Tests for L{twisted.conch.scripts.conch.SSHSession}. """ if StdioInteractingSession is None: skip = _reason def test_eofReceived(self): """ L{twisted.conch.scripts.conch.SSHSession.eofReceived} loses the write half of its stdio connection. """ stdio = FakeStdio() channel = StdioInteractingSession() channel.stdio = stdio channel.eofReceived() self.assertTrue(stdio.writeConnLost) class Echo(protocol.Protocol): def connectionMade(self): log.msg("ECHO CONNECTION MADE") def connectionLost(self, reason): log.msg("ECHO CONNECTION DONE") def dataReceived(self, data): self.transport.write(data) if b"\n" in data: self.transport.loseConnection() class EchoFactory(protocol.Factory): protocol = Echo class ConchTestOpenSSHProcess(protocol.ProcessProtocol): """ Test protocol for launching an OpenSSH client process. @ivar deferred: Set by whatever uses this object. Accessed using L{_getDeferred}, which destroys the value so the Deferred is not fired twice. Fires when the process is terminated. """ deferred = None buf = b"" problems = b"" def _getDeferred(self): d, self.deferred = self.deferred, None return d def outReceived(self, data): self.buf += data def errReceived(self, data): self.problems += data def processEnded(self, reason): """ Called when the process has ended. @param reason: a Failure giving the reason for the process' end. """ if reason.value.exitCode != 0: self._getDeferred().errback( ConchError( "exit code was not 0: {} ({})".format( reason.value.exitCode, self.problems.decode("charmap"), ) ) ) else: buf = self.buf.replace(b"\r\n", b"\n") self._getDeferred().callback(buf) class ConchTestForwardingProcess(protocol.ProcessProtocol): """ Manages a third-party process which launches a server. Uses L{ConchTestForwardingPort} to connect to the third-party server. Once L{ConchTestForwardingPort} has disconnected, kill the process and fire a Deferred with the data received by the L{ConchTestForwardingPort}. @ivar deferred: Set by whatever uses this object. Accessed using L{_getDeferred}, which destroys the value so the Deferred is not fired twice. Fires when the process is terminated. """ deferred = None def __init__(self, port, data): """ @type port: L{int} @param port: The port on which the third-party server is listening. (it is assumed that the server is running on localhost). @type data: L{str} @param data: This is sent to the third-party server. Must end with '\n' in order to trigger a disconnect. """ self.port = port self.buffer = None self.data = data def _getDeferred(self): d, self.deferred = self.deferred, None return d def connectionMade(self): self._connect() def _connect(self): """ Connect to the server, which is often a third-party process. Tries to reconnect if it fails because we have no way of determining exactly when the port becomes available for listening -- we can only know when the process starts. """ cc = protocol.ClientCreator(reactor, ConchTestForwardingPort, self, self.data) d = cc.connectTCP("127.0.0.1", self.port) d.addErrback(self._ebConnect) return d def _ebConnect(self, f): reactor.callLater(0.1, self._connect) def forwardingPortDisconnected(self, buffer): """ The network connection has died; save the buffer of output from the network and attempt to quit the process gracefully, and then (after the reactor has spun) send it a KILL signal. """ self.buffer = buffer self.transport.write(b"\x03") self.transport.loseConnection() reactor.callLater(0, self._reallyDie) def _reallyDie(self): try: self.transport.signalProcess("KILL") except ProcessExitedAlready: pass def processEnded(self, reason): """ Fire the Deferred at self.deferred with the data collected from the L{ConchTestForwardingPort} connection, if any. """ self._getDeferred().callback(self.buffer) class ConchTestForwardingPort(protocol.Protocol): """ Connects to server launched by a third-party process (managed by L{ConchTestForwardingProcess}) sends data, then reports whatever it received back to the L{ConchTestForwardingProcess} once the connection is ended. """ def __init__(self, protocol, data): """ @type protocol: L{ConchTestForwardingProcess} @param protocol: The L{ProcessProtocol} which made this connection. @type data: str @param data: The data to be sent to the third-party server. """ self.protocol = protocol self.data = data def connectionMade(self): self.buffer = b"" self.transport.write(self.data) def dataReceived(self, data): self.buffer += data def connectionLost(self, reason): self.protocol.forwardingPortDisconnected(self.buffer) def _makeArgs(args, mod="conch"): start = [ sys.executable, "-c" """ ### Twisted Preamble import sys, os path = os.path.abspath(sys.argv[0]) while os.path.dirname(path) != path: if os.path.basename(path).startswith('Twisted'): sys.path.insert(0, path) break path = os.path.dirname(path) from twisted.conch.scripts.%s import run run()""" % mod, ] madeArgs = [] for arg in start + list(args): if isinstance(arg, str): arg = arg.encode("utf-8") madeArgs.append(arg) return madeArgs class ConchServerSetupMixin: if not cryptography: skip = "can't run without cryptography" if not pyasn1: skip = "Cannot run without PyASN1" @staticmethod def realmFactory(): return ConchTestRealm(b"testuser") def _createFiles(self): for f in ["rsa_test", "rsa_test.pub", "dsa_test", "dsa_test.pub", "kh_test"]: if os.path.exists(f): os.remove(f) with open("rsa_test", "wb") as f: f.write(privateRSA_openssh) with open("rsa_test.pub", "wb") as f: f.write(publicRSA_openssh) with open("dsa_test.pub", "wb") as f: f.write(publicDSA_openssh) with open("dsa_test", "wb") as f: f.write(privateDSA_openssh) os.chmod("dsa_test", 0o600) os.chmod("rsa_test", 0o600) permissions = FilePath("dsa_test").getPermissions() if permissions.group.read or permissions.other.read: raise SkipTest( "private key readable by others despite chmod;" " possible windows permission issue?" " see https://tm.tl/9767" ) with open("kh_test", "wb") as f: f.write(b"127.0.0.1 " + publicRSA_openssh) def _getFreePort(self): s = socket.socket() s.bind(("", 0)) port = s.getsockname()[1] s.close() return port def _makeConchFactory(self): """ Make a L{ConchTestServerFactory}, which allows us to start a L{ConchTestServer} -- i.e. an actually listening conch. """ realm = self.realmFactory() p = portal.Portal(realm) p.registerChecker(conchTestPublicKeyChecker()) factory = ConchTestServerFactory() factory.portal = p return factory def setUp(self): self._createFiles() self.conchFactory = self._makeConchFactory() self.conchFactory.expectedLoseConnection = 1 self.conchServer = reactor.listenTCP( 0, self.conchFactory, interface="127.0.0.1" ) self.echoServer = reactor.listenTCP(0, EchoFactory()) self.echoPort = self.echoServer.getHost().port if HAS_IPV6: self.echoServerV6 = reactor.listenTCP(0, EchoFactory(), interface="::1") self.echoPortV6 = self.echoServerV6.getHost().port def tearDown(self): try: self.conchFactory.proto.done = 1 except AttributeError: pass else: self.conchFactory.proto.transport.loseConnection() deferreds = [ defer.maybeDeferred(self.conchServer.stopListening), defer.maybeDeferred(self.echoServer.stopListening), ] if HAS_IPV6: deferreds.append(defer.maybeDeferred(self.echoServerV6.stopListening)) return defer.gatherResults(deferreds) class ForwardingMixin(ConchServerSetupMixin): """ Template class for tests of the Conch server's ability to forward arbitrary protocols over SSH. These tests are integration tests, not unit tests. They launch a Conch server, a custom TCP server (just an L{EchoProtocol}) and then call L{execute}. L{execute} is implemented by subclasses of L{ForwardingMixin}. It should cause an SSH client to connect to the Conch server, asking it to forward data to the custom TCP server. """ def test_exec(self): """ Test that we can use whatever client to send the command "echo goodbye" to the Conch server. Make sure we receive "goodbye" back from the server. """ d = self.execute("echo goodbye", ConchTestOpenSSHProcess()) return d.addCallback(self.assertEqual, b"goodbye\n") def test_localToRemoteForwarding(self): """ Test that we can use whatever client to forward a local port to a specified port on the server. """ localPort = self._getFreePort() process = ConchTestForwardingProcess(localPort, b"test\n") d = self.execute( "", process, sshArgs="-N -L%i:127.0.0.1:%i" % (localPort, self.echoPort) ) d.addCallback(self.assertEqual, b"test\n") return d def test_remoteToLocalForwarding(self): """ Test that we can use whatever client to forward a port from the server to a port locally. """ localPort = self._getFreePort() process = ConchTestForwardingProcess(localPort, b"test\n") d = self.execute( "", process, sshArgs="-N -R %i:127.0.0.1:%i" % (localPort, self.echoPort) ) d.addCallback(self.assertEqual, b"test\n") return d # Conventionally there is a separate adapter object which provides ISession for # the user, but making the user provide ISession directly works too. This isn't # a full implementation of ISession though, just enough to make these tests # pass. @implementer(ISession) class RekeyAvatar(ConchUser): """ This avatar implements a shell which sends 60 numbered lines to whatever connects to it, then closes the session with a 0 exit status. 60 lines is selected as being enough to send more than 2kB of traffic, the amount the client is configured to initiate a rekey after. """ def __init__(self): ConchUser.__init__(self) self.channelLookup[b"session"] = SSHSession def openShell(self, transport): """ Write 60 lines of data to the transport, then exit. """ proto = protocol.Protocol() proto.makeConnection(transport) transport.makeConnection(wrapProtocol(proto)) # Send enough bytes to the connection so that a rekey is triggered in # the client. def write(counter): i = next(counter) if i == 60: call.stop() transport.session.conn.sendRequest( transport.session, b"exit-status", b"\x00\x00\x00\x00" ) transport.loseConnection() else: line = "line #%02d\n" % (i,) line = line.encode("utf-8") transport.write(line) # The timing for this loop is an educated guess (and/or the result of # experimentation) to exercise the case where a packet is generated # mid-rekey. Since the other side of the connection is (so far) the # OpenSSH command line client, there's no easy way to determine when the # rekey has been initiated. If there were, then generating a packet # immediately at that time would be a better way to test the # functionality being tested here. call = LoopingCall(write, count()) call.start(0.01) def closed(self): """ Ignore the close of the session. """ def eofReceived(self): # ISession.eofReceived pass def execCommand(self, proto, command): # ISession.execCommand pass def getPty(self, term, windowSize, modes): # ISession.getPty pass def windowChanged(self, newWindowSize): # ISession.windowChanged pass class RekeyRealm: """ This realm gives out new L{RekeyAvatar} instances for any avatar request. """ def requestAvatar(self, avatarID, mind, *interfaces): return interfaces[0], RekeyAvatar(), lambda: None class RekeyTestsMixin(ConchServerSetupMixin): """ TestCase mixin which defines tests exercising L{SSHTransportBase}'s handling of rekeying messages. """ realmFactory = RekeyRealm def test_clientRekey(self): """ After a client-initiated rekey is completed, application data continues to be passed over the SSH connection. """ process = ConchTestOpenSSHProcess() d = self.execute("", process, "-o RekeyLimit=2K") def finished(result): expectedResult = "\n".join(["line #%02d" % (i,) for i in range(60)]) + "\n" expectedResult = expectedResult.encode("utf-8") self.assertEqual(result, expectedResult) d.addCallback(finished) return d class OpenSSHClientMixin: if not which("ssh"): skip = "no ssh command-line client available" def execute(self, remoteCommand, process, sshArgs=""): """ Connects to the SSH server started in L{ConchServerSetupMixin.setUp} by running the 'ssh' command line tool. @type remoteCommand: str @param remoteCommand: The command (with arguments) to run on the remote end. @type process: L{ConchTestOpenSSHProcess} @type sshArgs: str @param sshArgs: Arguments to pass to the 'ssh' process. @return: L{defer.Deferred} """ # PubkeyAcceptedKeyTypes does not exist prior to OpenSSH 7.0 so we # first need to check if we can set it. If we can, -V will just print # the version without doing anything else; if we can't, we will get a # configuration error. d = getProcessValue( which("ssh")[0], ("-o", "PubkeyAcceptedKeyTypes=ssh-dss", "-V") ) def hasPAKT(status): if status == 0: opts = "-oPubkeyAcceptedKeyTypes=ssh-dss " else: opts = "" process.deferred = defer.Deferred() # Pass -F /dev/null to avoid the user's configuration file from # being loaded, as it may contain settings that cause our tests to # fail or hang. cmdline = ( ( "ssh -2 -l testuser -p %i " "-F /dev/null " "-oUserKnownHostsFile=kh_test " "-oPasswordAuthentication=no " # Always use the RSA key, since that's the one in kh_test. "-oHostKeyAlgorithms=ssh-rsa " "-a " "-i dsa_test " ) + opts + sshArgs + " 127.0.0.1 " + remoteCommand ) port = self.conchServer.getHost().port cmds = (cmdline % port).split() encodedCmds = [] for cmd in cmds: if isinstance(cmd, str): cmd = cmd.encode("utf-8") encodedCmds.append(cmd) reactor.spawnProcess(process, which("ssh")[0], encodedCmds) return process.deferred return d.addCallback(hasPAKT) class OpenSSHKeyExchangeTests(ConchServerSetupMixin, OpenSSHClientMixin, TestCase): """ Tests L{SSHTransportBase}'s key exchange algorithm compatibility with OpenSSH. """ def assertExecuteWithKexAlgorithm(self, keyExchangeAlgo): """ Call execute() method of L{OpenSSHClientMixin} with an ssh option that forces the exclusive use of the key exchange algorithm specified by keyExchangeAlgo @type keyExchangeAlgo: L{str} @param keyExchangeAlgo: The key exchange algorithm to use @return: L{defer.Deferred} """ kexAlgorithms = [] try: output = subprocess.check_output( [which("ssh")[0], "-Q", "kex"], stderr=subprocess.STDOUT ) if not isinstance(output, str): output = output.decode("utf-8") kexAlgorithms = output.split() except BaseException: pass if keyExchangeAlgo not in kexAlgorithms: raise SkipTest(f"{keyExchangeAlgo} not supported by ssh client") d = self.execute( "echo hello", ConchTestOpenSSHProcess(), "-oKexAlgorithms=" + keyExchangeAlgo, ) return d.addCallback(self.assertEqual, b"hello\n") def test_ECDHSHA256(self): """ The ecdh-sha2-nistp256 key exchange algorithm is compatible with OpenSSH """ return self.assertExecuteWithKexAlgorithm("ecdh-sha2-nistp256") def test_ECDHSHA384(self): """ The ecdh-sha2-nistp384 key exchange algorithm is compatible with OpenSSH """ return self.assertExecuteWithKexAlgorithm("ecdh-sha2-nistp384") def test_ECDHSHA521(self): """ The ecdh-sha2-nistp521 key exchange algorithm is compatible with OpenSSH """ return self.assertExecuteWithKexAlgorithm("ecdh-sha2-nistp521") def test_DH_GROUP14(self): """ The diffie-hellman-group14-sha1 key exchange algorithm is compatible with OpenSSH. """ return self.assertExecuteWithKexAlgorithm("diffie-hellman-group14-sha1") def test_DH_GROUP_EXCHANGE_SHA1(self): """ The diffie-hellman-group-exchange-sha1 key exchange algorithm is compatible with OpenSSH. """ return self.assertExecuteWithKexAlgorithm("diffie-hellman-group-exchange-sha1") def test_DH_GROUP_EXCHANGE_SHA256(self): """ The diffie-hellman-group-exchange-sha256 key exchange algorithm is compatible with OpenSSH. """ return self.assertExecuteWithKexAlgorithm( "diffie-hellman-group-exchange-sha256" ) def test_unsupported_algorithm(self): """ The list of key exchange algorithms supported by OpenSSH client is obtained with C{ssh -Q kex}. """ self.assertRaises( SkipTest, self.assertExecuteWithKexAlgorithm, "unsupported-algorithm" ) class OpenSSHClientForwardingTests(ForwardingMixin, OpenSSHClientMixin, TestCase): """ Connection forwarding tests run against the OpenSSL command line client. """ @skipIf(not HAS_IPV6, "Requires IPv6 support") def test_localToRemoteForwardingV6(self): """ Forwarding of arbitrary IPv6 TCP connections via SSH. """ localPort = self._getFreePort() process = ConchTestForwardingProcess(localPort, b"test\n") d = self.execute( "", process, sshArgs="-N -L%i:[::1]:%i" % (localPort, self.echoPortV6) ) d.addCallback(self.assertEqual, b"test\n") return d class OpenSSHClientRekeyTests(RekeyTestsMixin, OpenSSHClientMixin, TestCase): """ Rekeying tests run against the OpenSSL command line client. """ class CmdLineClientTests(ForwardingMixin, TestCase): """ Connection forwarding tests run against the Conch command line client. """ if runtime.platformType == "win32": skip = "can't run cmdline client on win32" def execute(self, remoteCommand, process, sshArgs="", conchArgs=None): """ As for L{OpenSSHClientTestCase.execute}, except it runs the 'conch' command line tool, not 'ssh'. """ if conchArgs is None: conchArgs = [] process.deferred = defer.Deferred() port = self.conchServer.getHost().port cmd = ( "-p {} -l testuser " "--known-hosts kh_test " "--user-authentications publickey " "-a " "-i dsa_test " "-v ".format(port) + sshArgs + " 127.0.0.1 " + remoteCommand ) cmds = _makeArgs(conchArgs + cmd.split()) env = os.environ.copy() env["PYTHONPATH"] = os.pathsep.join(sys.path) encodedCmds = [] encodedEnv = {} for cmd in cmds: if isinstance(cmd, str): cmd = cmd.encode("utf-8") encodedCmds.append(cmd) for var in env: val = env[var] if isinstance(var, str): var = var.encode("utf-8") if isinstance(val, str): val = val.encode("utf-8") encodedEnv[var] = val reactor.spawnProcess(process, sys.executable, encodedCmds, env=encodedEnv) return process.deferred def test_runWithLogFile(self): """ It can store logs to a local file. """ def cb_check_log(result): logContent = logPath.getContent() self.assertIn(b"Log opened.", logContent) logPath = filepath.FilePath(self.mktemp()) d = self.execute( remoteCommand="echo goodbye", process=ConchTestOpenSSHProcess(), conchArgs=[ "--log", "--logfile", logPath.path, "--host-key-algorithms", "ssh-rsa", ], ) d.addCallback(self.assertEqual, b"goodbye\n") d.addCallback(cb_check_log) return d def test_runWithNoHostAlgorithmsSpecified(self): """ Do not use --host-key-algorithms flag on command line. """ d = self.execute( remoteCommand="echo goodbye", process=ConchTestOpenSSHProcess() ) d.addCallback(self.assertEqual, b"goodbye\n") return d PK F��Zή�� �� test_session.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for the 'session' channel implementation in twisted.conch.ssh.session. See also RFC 4254. """ import os import signal import struct import sys from unittest import skipIf from zope.interface import implementer from twisted.internet import defer, error, protocol from twisted.internet.address import IPv4Address from twisted.internet.error import ProcessDone, ProcessTerminated from twisted.python import components, failure from twisted.python.failure import Failure from twisted.python.reflect import requireModule from twisted.python.test.test_components import RegistryUsingMixin from twisted.trial.unittest import TestCase cryptography = requireModule("cryptography") if cryptography: from twisted.conch.ssh import common, connection, session else: class session: # type: ignore[no-redef] from twisted.conch.interfaces import ( EnvironmentVariableNotPermitted, ISession, ISessionSetEnv, ) class SubsystemOnlyAvatar: """ A stub class representing an avatar that is only useful for getting a subsystem. """ def lookupSubsystem(self, name, data): """ If the other side requests the 'subsystem' subsystem, allow it by returning a MockProtocol to implement it. Otherwise raise an assertion. """ assert name == b"subsystem" return MockProtocol() class StubAvatar: """ A stub class representing the avatar representing the authenticated user. It implements the I{ISession} interface. """ def lookupSubsystem(self, name, data): """ If the user requests the TestSubsystem subsystem, connect them to a MockProtocol. If they request neither, then None is returned which is interpreted by SSHSession as a failure. """ if name == b"TestSubsystem": self.subsystem = MockProtocol() self.subsystem.packetData = data return self.subsystem @implementer(session.ISession) class StubSessionForStubAvatar: """ A stub ISession implementation for our StubAvatar. The instance variables generally keep track of method invocations so that we can test that the methods were called. @ivar avatar: the L{StubAvatar} we are adapting. @ivar ptyRequest: if present, the terminal, window size, and modes passed to the getPty method. @ivar windowChange: if present, the window size passed to the windowChangned method. @ivar shellProtocol: if present, the L{SSHSessionProcessProtocol} passed to the openShell method. @ivar shellTransport: if present, the L{EchoTransport} connected to shellProtocol. @ivar execProtocol: if present, the L{SSHSessionProcessProtocol} passed to the execCommand method. @ivar execTransport: if present, the L{EchoTransport} connected to execProtocol. @ivar execCommandLine: if present, the command line passed to the execCommand method. @ivar gotEOF: if present, an EOF message was received. @ivar gotClosed: if present, a closed message was received. """ def __init__(self, avatar): """ Store the avatar we're adapting. """ self.avatar = avatar self.shellProtocol = None def getPty(self, terminal, window, modes): """ If the terminal is 'bad', fail. Otherwise, store the information in the ptyRequest variable. """ if terminal != b"bad": self.ptyRequest = (terminal, window, modes) else: raise RuntimeError("not getting a pty") def windowChanged(self, window): """ If all the window sizes are 0, fail. Otherwise, store the size in the windowChange variable. """ if window == (0, 0, 0, 0): raise RuntimeError("not changing the window size") else: self.windowChange = window def openShell(self, pp): """ If we have gotten a shell request before, fail. Otherwise, store the process protocol in the shellProtocol variable, connect it to the EchoTransport and store that as shellTransport. """ if self.shellProtocol is not None: raise RuntimeError("not getting a shell this time") else: self.shellProtocol = pp self.shellTransport = EchoTransport(pp) def execCommand(self, pp, command): """ If the command is 'true', store the command, the process protocol, and the transport we connect to the process protocol. Otherwise, just store the command and raise an error. """ self.execCommandLine = command if command == b"success": self.execProtocol = pp elif command[:6] == b"repeat": self.execProtocol = pp self.execTransport = EchoTransport(pp) pp.outReceived(command[7:]) else: raise RuntimeError("not getting a command") def eofReceived(self): """ Note that EOF has been received. """ self.gotEOF = True def closed(self): """ Note that close has been received. """ self.gotClosed = True @implementer(session.ISessionSetEnv) class StubSessionForStubAvatarWithEnv(StubSessionForStubAvatar): """ Same as StubSessionForStubAvatar, but supporting environment variables setting. End users would want to have the same class annotated with C{@implementer(session.ISession, session.ISessionSetEnv)}. The interfaces are split for backwards compatibility, so we split it here to test this compatibility too. @ivar environ: a L{dict} of environment variables passed to the setEnv method. """ def __init__(self, avatar): super().__init__(avatar) # The representation of the environment as updated by remote requests. self.environ = {} # A snapshot of the environment when PTY request is received. self.environAtPty = {} def setEnv(self, name, value): """ If the requested environment variable is 'FAIL', fail. If it is 'IGNORED', raise EnvironmentVariableNotPermitted, which should cause it to be silently ignored. Otherwise, store the requested environment variable. (Real applications should normally implement an allowed list rather than a blocked list.) """ if name == b"FAIL": raise RuntimeError("disallowed environment variable name") elif name == b"IGNORED": raise session.EnvironmentVariableNotPermitted( "ignored environment variable name" ) else: self.environ[name] = value def getPty(self, term, windowSize, modes): """ Just a simple implementation which records the current environment when PTY is requested. """ self.environAtPty = self.environ.copy() class EchoTransport: """ A transport for a ProcessProtocol which echos data that is sent to it with a Window newline (CR LF) appended to it. If a null byte is in the data, disconnect. When we are asked to disconnect, disconnect the C{ProcessProtocol} with a 0 exit code. @ivar proto: the C{ProcessProtocol} connected to us. @ivar data: a L{bytes} of data written to us. """ def __init__(self, processProtocol): """ Initialize our instance variables. @param processProtocol: a C{ProcessProtocol} to connect to ourself. """ self.proto = processProtocol self.closed = False self.data = b"" processProtocol.makeConnection(self) def write(self, data): """ We got some data. Give it back to our C{ProcessProtocol} with a newline attached. Disconnect if there's a null byte. """ self.data += data self.proto.outReceived(data) self.proto.outReceived(b"\r\n") if b"\x00" in data: # mimic 'exit' for the shell test self.loseConnection() def loseConnection(self): """ If we're asked to disconnect (and we haven't already) shut down the C{ProcessProtocol} with a 0 exit code. """ if self.closed: return self.closed = 1 self.proto.inConnectionLost() self.proto.outConnectionLost() self.proto.errConnectionLost() self.proto.processEnded(failure.Failure(error.ProcessTerminated(0, None, None))) class MockProtocol(protocol.Protocol): """ A sample Protocol which stores the data passed to it. @ivar packetData: a L{bytes} of data to be sent when the connection is made. @ivar data: a L{bytes} of the data passed to us. @ivar open: True if the channel is open. @ivar reason: if not None, the reason the protocol was closed. """ packetData = b"" def connectionMade(self): """ Set up the instance variables. If we have any packetData, send it along. """ self.data = b"" self.open = True self.reason = None if self.packetData: self.dataReceived(self.packetData) def dataReceived(self, data): """ Store the received data and write it back with a tilde appended. The tilde is appended so that the tests can verify that we processed the data. """ self.data += data self.transport.write(data + b"~") def connectionLost(self, reason): """ Close the protocol and store the reason. """ self.open = False self.reason = reason class StubConnection: """ A stub for twisted.conch.ssh.connection.SSHConnection. Record the data that channels send, and when they try to close the connection. @ivar data: a L{dict} mapping C{SSHChannel}s to a C{list} of L{bytes} of data they sent. @ivar extData: a L{dict} mapping L{SSHChannel}s to a C{list} of L{tuple} of (L{int}, L{bytes}) of extended data they sent. @ivar requests: a L{dict} mapping L{SSHChannel}s to a C{list} of L{tuple} of (L{str}, L{bytes}) of channel requests they made. @ivar eofs: a L{dict} mapping L{SSHChannel}s to C{true} if they have sent an EOF. @ivar closes: a L{dict} mapping L{SSHChannel}s to C{true} if they have sent a close. """ def __init__(self, transport=None): """ Initialize our instance variables. """ self.data = {} self.extData = {} self.requests = {} self.eofs = {} self.closes = {} self.transport = transport def logPrefix(self): """ Return our logging prefix. """ return "MockConnection" def sendData(self, channel, data): """ Record the sent data. """ self.data.setdefault(channel, []).append(data) def sendExtendedData(self, channel, type, data): """ Record the sent extended data. """ self.extData.setdefault(channel, []).append((type, data)) def sendRequest(self, channel, request, data, wantReply=False): """ Record the sent channel request. """ self.requests.setdefault(channel, []).append((request, data, wantReply)) if wantReply: return defer.succeed(None) def sendEOF(self, channel): """ Record the sent EOF. """ self.eofs[channel] = True def sendClose(self, channel): """ Record the sent close. """ self.closes[channel] = True class StubTransport: """ A stub transport which records the data written. @ivar buf: the data sent to the transport. @type buf: L{bytes} @ivar close: flags indicating if the transport has been closed. @type close: L{bool} """ buf = b"" close = False def getPeer(self): """ Return an arbitrary L{IAddress}. """ return IPv4Address("TCP", "remotehost", 8888) def getHost(self): """ Return an arbitrary L{IAddress}. """ return IPv4Address("TCP", "localhost", 9999) def write(self, data): """ Record data in the buffer. """ self.buf += data def loseConnection(self): """ Note that the connection was closed. """ self.close = True def setTcpNoDelay(self, enabled): """ Pretend to set C{TCP_NODELAY}. """ # Required for testing SSHSessionForUnixConchUser. class StubTransportWithWriteErr(StubTransport): """ A version of StubTransport which records the error data sent to it. @ivar err: the extended data sent to the transport. @type err: L{bytes} """ err = b"" def writeErr(self, data): """ Record the extended data in the buffer. This was an old interface that allowed the Transports from ISession.openShell() or ISession.execCommand() to receive extended data from the client. """ self.err += data class StubClient: """ A stub class representing the client to a SSHSession. @ivar transport: A L{StubTransport} object which keeps track of the data passed to it. """ def __init__(self): self.transport = StubTransportWithWriteErr() class SessionInterfaceTests(RegistryUsingMixin, TestCase): """ Tests for the SSHSession class interface. This interface is not ideal, but it is tested in order to maintain backwards compatibility. """ if not cryptography: skip = "cannot run without cryptography" def setUp(self, register_adapters=True): """ Make an SSHSession object to test. Give the channel some window so that it's allowed to send packets. 500 and 100 are arbitrary values. """ RegistryUsingMixin.setUp(self) self.session = self.getSSHSession() if register_adapters: components.registerAdapter( StubSessionForStubAvatarWithEnv, StubAvatar, session.ISession ) self.session = self.getSSHSession() def getSSHSession(self, register_adapters=True): """ Return a new SSH session. """ return session.SSHSession( remoteWindow=500, remoteMaxPacket=100, conn=StubConnection(), avatar=StubAvatar(), ) def assertSessionIsStubSession(self): """ Asserts that self.session.session is an instance of StubSessionForStubOldAvatar. """ self.assertIsInstance(self.session.session, StubSessionForStubAvatar) def test_init(self): """ SSHSession initializes its buffer (buf), client, and ISession adapter. The avatar should not need to be adaptable to an ISession immediately. """ s = session.SSHSession(avatar=object) # use object because it doesn't # have an adapter self.assertEqual(s.buf, b"") self.assertIsNone(s.client) self.assertIsNone(s.session) def test_client_dataReceived(self): """ SSHSession.dataReceived() passes data along to a client. If the data comes before there is a client, the data should be discarded. """ self.session.dataReceived(b"1") self.session.client = StubClient() self.session.dataReceived(b"2") self.assertEqual(self.session.client.transport.buf, b"2") def test_client_extReceived(self): """ SSHSession.extReceived() passed data of type EXTENDED_DATA_STDERR along to the client. If the data comes before there is a client, or if the data is not of type EXTENDED_DATA_STDERR, it is discared. """ self.session.extReceived(connection.EXTENDED_DATA_STDERR, b"1") self.session.extReceived(255, b"2") # 255 is arbitrary self.session.client = StubClient() self.session.extReceived(connection.EXTENDED_DATA_STDERR, b"3") self.assertEqual(self.session.client.transport.err, b"3") def test_client_extReceivedWithoutWriteErr(self): """ SSHSession.extReceived() should handle the case where the transport on the client doesn't have a writeErr method. """ client = self.session.client = StubClient() client.transport = StubTransport() # doesn't have writeErr # should not raise an error self.session.extReceived(connection.EXTENDED_DATA_STDERR, b"ignored") def test_client_closed(self): """ SSHSession.closed() should tell the transport connected to the client that the connection was lost. """ self.session.client = StubClient() self.session.closed() self.assertTrue(self.session.client.transport.close) self.session.client.transport.close = False def test_badSubsystemDoesNotCreateClient(self): """ When a subsystem request fails, SSHSession.client should not be set. """ ret = self.session.requestReceived(b"subsystem", common.NS(b"BadSubsystem")) self.assertFalse(ret) self.assertIsNone(self.session.client) def test_lookupSubsystem(self): """ When a client requests a subsystem, the SSHSession object should get the subsystem by calling avatar.lookupSubsystem, and attach it as the client. """ ret = self.session.requestReceived( b"subsystem", common.NS(b"TestSubsystem") + b"data" ) self.assertTrue(ret) self.assertIsInstance(self.session.client, protocol.ProcessProtocol) self.assertIs( self.session.client.transport.proto, self.session.avatar.subsystem ) def test_lookupSubsystemDoesNotNeedISession(self): """ Previously, if one only wanted to implement a subsystem, an ISession adapter wasn't needed because subsystems were looked up using the lookupSubsystem method on the avatar. """ s = session.SSHSession(avatar=SubsystemOnlyAvatar(), conn=StubConnection()) ret = s.request_subsystem(common.NS(b"subsystem") + b"data") self.assertTrue(ret) self.assertIsNotNone(s.client) self.assertIsNone(s.conn.closes.get(s)) s.eofReceived() self.assertTrue(s.conn.closes.get(s)) # these should not raise errors s.loseConnection() s.closed() def test_lookupSubsystem_data(self): """ After having looked up a subsystem, data should be passed along to the client. Additionally, subsystems were passed the entire request packet as data, instead of just the additional data. We check for the additional tidle to verify that the data passed through the client. """ # self.session.dataReceived('1') # subsystems didn't get extended data # self.session.extReceived(connection.EXTENDED_DATA_STDERR, '2') self.session.requestReceived( b"subsystem", common.NS(b"TestSubsystem") + b"data" ) self.assertEqual( self.session.conn.data[self.session], [b"\x00\x00\x00\x0dTestSubsystemdata~"], ) self.session.dataReceived(b"more data") self.assertEqual(self.session.conn.data[self.session][-1], b"more data~") def test_lookupSubsystem_closeReceived(self): """ SSHSession.closeReceived() should sent a close message to the remote side. """ self.session.requestReceived( b"subsystem", common.NS(b"TestSubsystem") + b"data" ) self.session.closeReceived() self.assertTrue(self.session.conn.closes[self.session]) def assertRequestRaisedRuntimeError(self): """ Assert that the request we just made raised a RuntimeError (and only a RuntimeError). """ errors = self.flushLoggedErrors(RuntimeError) self.assertEqual( len(errors), 1, "Multiple RuntimeErrors raised: %s" % "\n".join([repr(error) for error in errors]), ) errors[0].trap(RuntimeError) def test_requestShell(self): """ When a client requests a shell, the SSHSession object should get the shell by getting an ISession adapter for the avatar, then calling openShell() with a ProcessProtocol to attach. """ # gets a shell the first time ret = self.session.requestReceived(b"shell", b"") self.assertTrue(ret) self.assertSessionIsStubSession() self.assertIsInstance(self.session.client, session.SSHSessionProcessProtocol) self.assertIs(self.session.session.shellProtocol, self.session.client) # doesn't get a shell the second time self.assertFalse(self.session.requestReceived(b"shell", b"")) self.assertRequestRaisedRuntimeError() def test_requestShellWithData(self): """ When a client executes a shell, it should be able to give pass data back and forth between the local and the remote side. """ ret = self.session.requestReceived(b"shell", b"") self.assertTrue(ret) self.assertSessionIsStubSession() self.session.dataReceived(b"some data\x00") self.assertEqual(self.session.session.shellTransport.data, b"some data\x00") self.assertEqual( self.session.conn.data[self.session], [b"some data\x00", b"\r\n"] ) self.assertTrue(self.session.session.shellTransport.closed) self.assertEqual( self.session.conn.requests[self.session], [(b"exit-status", b"\x00\x00\x00\x00", False)], ) def test_requestExec(self): """ When a client requests a command, the SSHSession object should get the command by getting an ISession adapter for the avatar, then calling execCommand with a ProcessProtocol to attach and the command line. """ ret = self.session.requestReceived(b"exec", common.NS(b"failure")) self.assertFalse(ret) self.assertRequestRaisedRuntimeError() self.assertIsNone(self.session.client) self.assertTrue(self.session.requestReceived(b"exec", common.NS(b"success"))) self.assertSessionIsStubSession() self.assertIsInstance(self.session.client, session.SSHSessionProcessProtocol) self.assertIs(self.session.session.execProtocol, self.session.client) self.assertEqual(self.session.session.execCommandLine, b"success") def test_requestExecWithData(self): """ When a client executes a command, it should be able to give pass data back and forth. """ ret = self.session.requestReceived(b"exec", common.NS(b"repeat hello")) self.assertTrue(ret) self.assertSessionIsStubSession() self.session.dataReceived(b"some data") self.assertEqual(self.session.session.execTransport.data, b"some data") self.assertEqual( self.session.conn.data[self.session], [b"hello", b"some data", b"\r\n"] ) self.session.eofReceived() self.session.closeReceived() self.session.closed() self.assertTrue(self.session.session.execTransport.closed) self.assertEqual( self.session.conn.requests[self.session], [(b"exit-status", b"\x00\x00\x00\x00", False)], ) def test_requestPty(self): """ When a client requests a PTY, the SSHSession object should make the request by getting an ISession adapter for the avatar, then calling getPty with the terminal type, the window size, and any modes the client gave us. """ # Cleanup the registered adapters from setUp. self.doCleanups() self.setUp(register_adapters=False) components.registerAdapter( StubSessionForStubAvatar, StubAvatar, session.ISession ) test_session = self.getSSHSession() # 'bad' terminal type fails ret = test_session.requestReceived( b"pty_req", session.packRequest_pty_req(b"bad", (1, 2, 3, 4), b"") ) self.assertFalse(ret) self.assertIsInstance(test_session.session, StubSessionForStubAvatar) self.assertRequestRaisedRuntimeError() # 'good' terminal type succeeds self.assertTrue( test_session.requestReceived( b"pty_req", session.packRequest_pty_req(b"good", (1, 2, 3, 4), b"") ) ) self.assertEqual(test_session.session.ptyRequest, (b"good", (1, 2, 3, 4), [])) def test_setEnv(self): """ When a client requests passing an environment variable, the SSHSession object should make the request by getting an ISessionSetEnv adapter for the avatar, then calling setEnv with the environment variable name and value. """ # Blocked environment variable name fails. self.assertFalse( self.session.requestReceived(b"env", common.NS(b"FAIL") + common.NS(b"bad")) ) self.assertIsInstance(self.session.session, StubSessionForStubAvatarWithEnv) self.assertRequestRaisedRuntimeError() # An environment variable name for which setEnv raises # EnvironmentVariableNotPermitted is silently ignored. self.assertFalse( self.session.requestReceived( b"env", common.NS(b"IGNORED") + common.NS(b"ignored") ) ) self.assertEqual(self.flushLoggedErrors(), []) # Allowed environment variable name succeeds. self.assertTrue( self.session.requestReceived( b"env", common.NS(b"NAME") + common.NS(b"value") ) ) self.assertEqual(self.session.session.environ, {b"NAME": b"value"}) def test_setEnvSessionShare(self): """ Multiple setenv requests will share the same session. """ test_session = self.getSSHSession() self.assertTrue( test_session.requestReceived( b"env", common.NS(b"Key1") + common.NS(b"Value 1") ) ) self.assertTrue( test_session.requestReceived( b"env", common.NS(b"Key2") + common.NS(b"Value2") ) ) self.assertIsInstance(test_session.session, StubSessionForStubAvatarWithEnv) self.assertEqual( {b"Key1": b"Value 1", b"Key2": b"Value2"}, test_session.session.environ ) def test_setEnvMultiplexShare(self): """ Calling another session service after setenv will provide the previous session with the environment variables. """ test_session = self.getSSHSession() test_session.requestReceived(b"env", common.NS(b"Key1") + common.NS(b"Value 1")) test_session.requestReceived(b"env", common.NS(b"Key2") + common.NS(b"Value2")) test_session.requestReceived( b"pty_req", session.packRequest_pty_req(b"term", (0, 0, 0, 0), b"") ) self.assertIsInstance(test_session.session, StubSessionForStubAvatarWithEnv) self.assertEqual( {b"Key1": b"Value 1", b"Key2": b"Value2"}, test_session.session.environAtPty ) def test_setEnvNotProvidingISessionSetEnv(self): """ If the avatar does not have an ISessionSetEnv adapter, then a request to pass an environment variable fails gracefully. """ # Cleanup the registered adapters. self.doCleanups() self.setUp(register_adapters=False) # Register a ISession adapter that does not support ISessionSetEnv. components.registerAdapter( StubSessionForStubAvatar, StubAvatar, session.ISession ) self.assertFalse( self.session.requestReceived( b"env", common.NS(b"NAME") + common.NS(b"value") ) ) def test_requestWindowChange(self): """ When the client requests to change the window size, the SSHSession object should make the request by getting an ISession adapter for the avatar, then calling windowChanged with the new window size. """ ret = self.session.requestReceived( b"window_change", session.packRequest_window_change((0, 0, 0, 0)) ) self.assertFalse(ret) self.assertRequestRaisedRuntimeError() self.assertSessionIsStubSession() self.assertTrue( self.session.requestReceived( b"window_change", session.packRequest_window_change((1, 2, 3, 4)) ) ) self.assertEqual(self.session.session.windowChange, (1, 2, 3, 4)) def test_eofReceived(self): """ When an EOF is received and an ISession adapter is present, it should be notified of the EOF message. """ self.session.session = session.ISession(self.session.avatar) self.session.eofReceived() self.assertTrue(self.session.session.gotEOF) def test_closeReceived(self): """ When a close is received, the session should send a close message. """ ret = self.session.closeReceived() self.assertIsNone(ret) self.assertTrue(self.session.conn.closes[self.session]) def test_closed(self): """ When a close is received and an ISession adapter is present, it should be notified of the close message. """ self.session.session = session.ISession(self.session.avatar) self.session.closed() self.assertTrue(self.session.session.gotClosed) class SessionWithNoAvatarTests(RegistryUsingMixin, TestCase): """ Test for the SSHSession interface. Several of the methods (request_shell, request_exec, request_pty_req, request_env, request_window_change) would create a 'session' instance variable from the avatar if one didn't exist when they were called. """ if not cryptography: skip = "cannot run without cryptography" def setUp(self): RegistryUsingMixin.setUp(self) components.registerAdapter( StubSessionForStubAvatar, StubAvatar, session.ISession ) self.session = session.SSHSession() self.session.avatar = StubAvatar() self.assertIsNone(self.session.session) def assertSessionProvidesISession(self): """ self.session.session should provide I{ISession}. """ self.assertTrue( session.ISession.providedBy(self.session.session), "ISession not provided by %r" % self.session.session, ) def test_requestShellGetsSession(self): """ If an ISession adapter isn't already present, request_shell should get one. """ self.session.requestReceived(b"shell", b"") self.assertSessionProvidesISession() def test_requestExecGetsSession(self): """ If an ISession adapter isn't already present, request_exec should get one. """ self.session.requestReceived(b"exec", common.NS(b"success")) self.assertSessionProvidesISession() def test_requestPtyReqGetsSession(self): """ If an ISession adapter isn't already present, request_pty_req should get one. """ self.session.requestReceived( b"pty_req", session.packRequest_pty_req(b"term", (0, 0, 0, 0), b"") ) self.assertSessionProvidesISession() def test_requestWindowChangeGetsSession(self): """ If an ISession adapter isn't already present, request_window_change should get one. """ self.session.requestReceived( b"window_change", session.packRequest_window_change((1, 1, 1, 1)) ) self.assertSessionProvidesISession() class WrappersTests(TestCase): """ A test for the wrapProtocol and wrapProcessProtocol functions. """ if not cryptography: skip = "cannot run without cryptography" def test_wrapProtocol(self): """ L{wrapProtocol}, when passed a L{Protocol} should return something that has write(), writeSequence(), loseConnection() methods which call the Protocol's dataReceived() and connectionLost() methods, respectively. """ protocol = MockProtocol() protocol.transport = StubTransport() protocol.connectionMade() wrapped = session.wrapProtocol(protocol) wrapped.dataReceived(b"dataReceived") self.assertEqual(protocol.transport.buf, b"dataReceived") wrapped.write(b"data") wrapped.writeSequence([b"1", b"2"]) wrapped.loseConnection() self.assertEqual(protocol.data, b"data12") protocol.reason.trap(error.ConnectionDone) def test_wrapProcessProtocol_Protocol(self): """ L{wrapPRocessProtocol}, when passed a L{Protocol} should return something that follows the L{IProcessProtocol} interface, with connectionMade() mapping to connectionMade(), outReceived() mapping to dataReceived() and processEnded() mapping to connectionLost(). """ protocol = MockProtocol() protocol.transport = StubTransport() process_protocol = session.wrapProcessProtocol(protocol) process_protocol.connectionMade() process_protocol.outReceived(b"data") self.assertEqual(protocol.transport.buf, b"data~") process_protocol.processEnded( failure.Failure(error.ProcessTerminated(0, None, None)) ) protocol.reason.trap(error.ProcessTerminated) class HelpersTests(TestCase): """ Tests for the 4 helper functions: parseRequest_* and packRequest_*. """ if not cryptography: skip = "cannot run without cryptography" def test_parseRequest_pty_req(self): """ The payload of a pty-req message is:: string terminal uint32 columns uint32 rows uint32 x pixels uint32 y pixels string modes Modes are:: byte mode number uint32 mode value """ self.assertEqual( session.parseRequest_pty_req( common.NS(b"xterm") + struct.pack(">4L", 1, 2, 3, 4) + common.NS(struct.pack(">BL", 5, 6)) ), (b"xterm", (2, 1, 3, 4), [(5, 6)]), ) def test_packRequest_pty_req_old(self): """ See test_parseRequest_pty_req for the payload format. """ packed = session.packRequest_pty_req( b"xterm", (2, 1, 3, 4), b"\x05\x00\x00\x00\x06" ) self.assertEqual( packed, common.NS(b"xterm") + struct.pack(">4L", 1, 2, 3, 4) + common.NS(struct.pack(">BL", 5, 6)), ) def test_packRequest_pty_req(self): """ See test_parseRequest_pty_req for the payload format. """ packed = session.packRequest_pty_req( b"xterm", (2, 1, 3, 4), b"\x05\x00\x00\x00\x06" ) self.assertEqual( packed, common.NS(b"xterm") + struct.pack(">4L", 1, 2, 3, 4) + common.NS(struct.pack(">BL", 5, 6)), ) def test_parseRequest_window_change(self): """ The payload of a window_change request is:: uint32 columns uint32 rows uint32 x pixels uint32 y pixels parseRequest_window_change() returns (rows, columns, x pixels, y pixels). """ self.assertEqual( session.parseRequest_window_change(struct.pack(">4L", 1, 2, 3, 4)), (2, 1, 3, 4), ) def test_packRequest_window_change(self): """ See test_parseRequest_window_change for the payload format. """ self.assertEqual( session.packRequest_window_change((2, 1, 3, 4)), struct.pack(">4L", 1, 2, 3, 4), ) class SSHSessionProcessProtocolTests(TestCase): """ Tests for L{SSHSessionProcessProtocol}. """ if not cryptography: skip = "cannot run without cryptography" def setUp(self): self.transport = StubTransport() self.session = session.SSHSession( conn=StubConnection(self.transport), remoteWindow=500, remoteMaxPacket=100 ) self.pp = session.SSHSessionProcessProtocol(self.session) self.pp.makeConnection(self.transport) def assertSessionClosed(self): """ Assert that C{self.session} is closed. """ self.assertTrue(self.session.conn.closes[self.session]) def assertRequestsEqual(self, expectedRequests): """ Assert that C{self.session} has sent the C{expectedRequests}. """ self.assertEqual(self.session.conn.requests[self.session], expectedRequests) def test_init(self): """ SSHSessionProcessProtocol should set self.session to the session passed to the __init__ method. """ self.assertEqual(self.pp.session, self.session) def test_getHost(self): """ SSHSessionProcessProtocol.getHost() just delegates to its session.conn.transport.getHost(). """ self.assertEqual(self.session.conn.transport.getHost(), self.pp.getHost()) def test_getPeer(self): """ SSHSessionProcessProtocol.getPeer() just delegates to its session.conn.transport.getPeer(). """ self.assertEqual(self.session.conn.transport.getPeer(), self.pp.getPeer()) def test_connectionMade(self): """ SSHSessionProcessProtocol.connectionMade() should check if there's a 'buf' attribute on its session and write it to the transport if so. """ self.session.buf = b"buffer" self.pp.connectionMade() self.assertEqual(self.transport.buf, b"buffer") @skipIf(not hasattr(signal, "SIGALRM"), "Not all signals available") def test_getSignalName(self): """ _getSignalName should return the name of a signal when given the signal number. """ for signalName in session.SUPPORTED_SIGNALS: signalName = "SIG" + signalName signalValue = getattr(signal, signalName) sshName = self.pp._getSignalName(signalValue) self.assertEqual( sshName, signalName, "%i: %s != %s" % (signalValue, sshName, signalName) ) @skipIf(not hasattr(signal, "SIGALRM"), "Not all signals available") def test_getSignalNameWithLocalSignal(self): """ If there are signals in the signal module which aren't in the SSH RFC, we map their name to [signal name]@[platform]. """ signal.SIGTwistedTest = signal.NSIG + 1 # value can't exist normally # Force reinitialization of signals self.pp._signalValuesToNames = None self.assertEqual( self.pp._getSignalName(signal.SIGTwistedTest), "SIGTwistedTest@" + sys.platform, ) def test_outReceived(self): """ When data is passed to the outReceived method, it should be sent to the session's write method. """ self.pp.outReceived(b"test data") self.assertEqual(self.session.conn.data[self.session], [b"test data"]) def test_write(self): """ When data is passed to the write method, it should be sent to the session channel's write method. """ self.pp.write(b"test data") self.assertEqual(self.session.conn.data[self.session], [b"test data"]) def test_writeSequence(self): """ When a sequence is passed to the writeSequence method, it should be joined together and sent to the session channel's write method. """ self.pp.writeSequence([b"test ", b"data"]) self.assertEqual(self.session.conn.data[self.session], [b"test data"]) def test_errReceived(self): """ When data is passed to the errReceived method, it should be sent to the session's writeExtended method. """ self.pp.errReceived(b"test data") self.assertEqual(self.session.conn.extData[self.session], [(1, b"test data")]) def test_outConnectionLost(self): """ When outConnectionLost and errConnectionLost are both called, we should send an EOF message. """ self.pp.outConnectionLost() self.assertFalse(self.session in self.session.conn.eofs) self.pp.errConnectionLost() self.assertTrue(self.session.conn.eofs[self.session]) def test_errConnectionLost(self): """ Make sure reverse ordering of events in test_outConnectionLost also sends EOF. """ self.pp.errConnectionLost() self.assertFalse(self.session in self.session.conn.eofs) self.pp.outConnectionLost() self.assertTrue(self.session.conn.eofs[self.session]) def test_loseConnection(self): """ When loseConnection() is called, it should call loseConnection on the session channel. """ self.pp.loseConnection() self.assertTrue(self.session.conn.closes[self.session]) def test_connectionLost(self): """ When connectionLost() is called, it should call loseConnection() on the session channel. """ self.pp.connectionLost(failure.Failure(ProcessDone(0))) def test_processEndedWithExitCode(self): """ When processEnded is called, if there is an exit code in the reason it should be sent in an exit-status method. The connection should be closed. """ self.pp.processEnded(Failure(ProcessDone(None))) self.assertRequestsEqual([(b"exit-status", struct.pack(">I", 0), False)]) self.assertSessionClosed() @skipIf(not hasattr(os, "WCOREDUMP"), "can't run this w/o os.WCOREDUMP") def test_processEndedWithExitSignalCoreDump(self): """ When processEnded is called, if there is an exit signal in the reason it should be sent in an exit-signal message. The connection should be closed. """ self.pp.processEnded( Failure(ProcessTerminated(1, signal.SIGTERM, 1 << 7)) ) # 7th bit means core dumped self.assertRequestsEqual( [ ( b"exit-signal", common.NS(b"TERM") # signal name + b"\x01" # core dumped is true + common.NS(b"") # error message + common.NS(b""), # language tag False, ) ] ) self.assertSessionClosed() @skipIf(not hasattr(os, "WCOREDUMP"), "can't run this w/o os.WCOREDUMP") def test_processEndedWithExitSignalNoCoreDump(self): """ When processEnded is called, if there is an exit signal in the reason it should be sent in an exit-signal message. If no core was dumped, don't set the core-dump bit. """ self.pp.processEnded(Failure(ProcessTerminated(1, signal.SIGTERM, 0))) # see comments in test_processEndedWithExitSignalCoreDump for the # meaning of the parts in the request self.assertRequestsEqual( [ ( b"exit-signal", common.NS(b"TERM") + b"\x00" + common.NS(b"") + common.NS(b""), False, ) ] ) self.assertSessionClosed() class SSHSessionClientTests(TestCase): """ SSHSessionClient is an obsolete class used to connect standard IO to an SSHSession. """ if not cryptography: skip = "cannot run without cryptography" def test_dataReceived(self): """ When data is received, it should be sent to the transport. """ client = session.SSHSessionClient() client.transport = StubTransport() client.dataReceived(b"test data") self.assertEqual(client.transport.buf, b"test data") PK F��Zgv[ test_mixin.pynu �[��� # -*- twisted.conch.test.test_mixin -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. from twisted.conch import mixin from twisted.test.proto_helpers import StringTransport from twisted.trial import unittest class TestBufferingProto(mixin.BufferingMixin): scheduled = False rescheduled = 0 def schedule(self): self.scheduled = True return object() def reschedule(self, token): self.rescheduled += 1 class BufferingTests(unittest.TestCase): def testBuffering(self): p = TestBufferingProto() t = p.transport = StringTransport() self.assertFalse(p.scheduled) L = [b"foo", b"bar", b"baz", b"quux"] p.write(b"foo") self.assertTrue(p.scheduled) self.assertFalse(p.rescheduled) for s in L: n = p.rescheduled p.write(s) self.assertEqual(p.rescheduled, n + 1) self.assertEqual(t.value(), b"") p.flush() self.assertEqual(t.value(), b"foo" + b"".join(L)) PK F��ZG�~�� � test_manhole_tap.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.manhole_tap}. """ from twisted.application.internet import StreamServerEndpointService from twisted.application.service import MultiService from twisted.conch import telnet from twisted.cred import error from twisted.cred.credentials import UsernamePassword from twisted.python import usage from twisted.python.reflect import requireModule from twisted.trial.unittest import TestCase cryptography = requireModule("cryptography") pyasn1 = requireModule("pyasn1") if cryptography and pyasn1: from twisted.conch import manhole_ssh, manhole_tap class MakeServiceTests(TestCase): """ Tests for L{manhole_tap.makeService}. """ if not cryptography: skip = "can't run without cryptography" if not pyasn1: skip = "Cannot run without PyASN1" usernamePassword = (b"iamuser", b"thisispassword") def setUp(self): """ Create a passwd-like file with a user. """ self.filename = self.mktemp() with open(self.filename, "wb") as f: f.write(b":".join(self.usernamePassword)) self.options = manhole_tap.Options() def test_requiresPort(self): """ L{manhole_tap.makeService} requires either 'telnetPort' or 'sshPort' to be given. """ with self.assertRaises(usage.UsageError) as e: manhole_tap.Options().parseOptions([]) self.assertEqual( e.exception.args[0], ("At least one of --telnetPort " "and --sshPort must be specified"), ) def test_telnetPort(self): """ L{manhole_tap.makeService} will make a telnet service on the port defined by C{--telnetPort}. It will not make a SSH service. """ self.options.parseOptions(["--telnetPort", "tcp:222"]) service = manhole_tap.makeService(self.options) self.assertIsInstance(service, MultiService) self.assertEqual(len(service.services), 1) self.assertIsInstance(service.services[0], StreamServerEndpointService) self.assertIsInstance( service.services[0].factory.protocol, manhole_tap.makeTelnetProtocol ) self.assertEqual(service.services[0].endpoint._port, 222) def test_sshPort(self): """ L{manhole_tap.makeService} will make a SSH service on the port defined by C{--sshPort}. It will not make a telnet service. """ # Why the sshKeyDir and sshKeySize params? To prevent it stomping over # (or using!) the user's private key, we just make a super small one # which will never be used in a temp directory. self.options.parseOptions( [ "--sshKeyDir", self.mktemp(), "--sshKeySize", "512", "--sshPort", "tcp:223", ] ) service = manhole_tap.makeService(self.options) self.assertIsInstance(service, MultiService) self.assertEqual(len(service.services), 1) self.assertIsInstance(service.services[0], StreamServerEndpointService) self.assertIsInstance(service.services[0].factory, manhole_ssh.ConchFactory) self.assertEqual(service.services[0].endpoint._port, 223) def test_passwd(self): """ The C{--passwd} command-line option will load a passwd-like file. """ self.options.parseOptions(["--telnetPort", "tcp:22", "--passwd", self.filename]) service = manhole_tap.makeService(self.options) portal = service.services[0].factory.protocol.portal self.assertEqual(len(portal.checkers.keys()), 2) # Ensure it's the passwd file we wanted by trying to authenticate self.assertTrue( self.successResultOf( portal.login( UsernamePassword(*self.usernamePassword), None, telnet.ITelnetProtocol, ) ) ) self.assertIsInstance( self.failureResultOf( portal.login( UsernamePassword(b"wrong", b"user"), None, telnet.ITelnetProtocol ) ).value, error.UnauthorizedLogin, ) PK F��ZL2� � test_text.pynu �[��� # -*- test-case-name: twisted.conch.test.test_text -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. from twisted.conch.insults import text from twisted.conch.insults.text import attributes as A from twisted.trial import unittest class FormattedTextTests(unittest.TestCase): """ Tests for assembling formatted text. """ def test_trivial(self): """ Using no formatting attributes produces no VT102 control sequences in the flattened output. """ self.assertEqual( text.assembleFormattedText(A.normal["Hello, world."]), "Hello, world." ) def test_bold(self): """ The bold formatting attribute, L{A.bold}, emits the VT102 control sequence to enable bold when flattened. """ self.assertEqual( text.assembleFormattedText(A.bold["Hello, world."]), "\x1b[1mHello, world." ) def test_underline(self): """ The underline formatting attribute, L{A.underline}, emits the VT102 control sequence to enable underlining when flattened. """ self.assertEqual( text.assembleFormattedText(A.underline["Hello, world."]), "\x1b[4mHello, world.", ) def test_blink(self): """ The blink formatting attribute, L{A.blink}, emits the VT102 control sequence to enable blinking when flattened. """ self.assertEqual( text.assembleFormattedText(A.blink["Hello, world."]), "\x1b[5mHello, world." ) def test_reverseVideo(self): """ The reverse-video formatting attribute, L{A.reverseVideo}, emits the VT102 control sequence to enable reversed video when flattened. """ self.assertEqual( text.assembleFormattedText(A.reverseVideo["Hello, world."]), "\x1b[7mHello, world.", ) def test_minus(self): """ Formatting attributes prefixed with a minus (C{-}) temporarily disable the prefixed attribute, emitting no VT102 control sequence to enable it in the flattened output. """ self.assertEqual( text.assembleFormattedText( A.bold[A.blink["Hello", -A.bold[" world"], "."]] ), "\x1b[1;5mHello\x1b[0;5m world\x1b[1;5m.", ) def test_foreground(self): """ The foreground color formatting attribute, L{A.fg}, emits the VT102 control sequence to set the selected foreground color when flattened. """ self.assertEqual( text.assembleFormattedText( A.normal[A.fg.red["Hello, "], A.fg.green["world!"]] ), "\x1b[31mHello, \x1b[32mworld!", ) def test_background(self): """ The background color formatting attribute, L{A.bg}, emits the VT102 control sequence to set the selected background color when flattened. """ self.assertEqual( text.assembleFormattedText( A.normal[A.bg.red["Hello, "], A.bg.green["world!"]] ), "\x1b[41mHello, \x1b[42mworld!", ) def test_flattenDeprecated(self): """ L{twisted.conch.insults.text.flatten} emits a deprecation warning when imported or accessed. """ warningsShown = self.flushWarnings([self.test_flattenDeprecated]) self.assertEqual(len(warningsShown), 0) # Trigger the deprecation warning. text.flatten warningsShown = self.flushWarnings([self.test_flattenDeprecated]) self.assertEqual(len(warningsShown), 1) self.assertEqual(warningsShown[0]["category"], DeprecationWarning) self.assertEqual( warningsShown[0]["message"], "twisted.conch.insults.text.flatten was deprecated in Twisted " "13.1.0: Use twisted.conch.insults.text.assembleFormattedText " "instead.", ) PK F��Z<���' ' / __pycache__/test_openssh_compat.cpython-310.pycnu �[��� o �b � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z dd lmZ d Zd Ze d �rCe d �rCddlmZ ndZdZeed�sPdZdZeee�G dd� de��ZdS )z, Tests for L{twisted.conch.openssh_compat}. � N)�skipIf)�getDHGeneratorAndPrime)�keydata)�FilePath)� requireModule)�MockOS)�TestCaseF� � cryptography�pyasn1)�OpenSSHFactoryTz)Cannot run without cryptography or PyASN1�geteuidz geteuid/seteuid not availablec @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �OpenSSHFactoryTestsz& Tests for L{OpenSSHFactory}. c C s t � | _t| �� �| _| j�� | jj| j_t| �� �| _| j�� | jj| j_ | j� d�� d� | j� d�� d� | j� d�� t j � | j� d�� t j� | j� d�� d� | j� d�� t j� | j� d �� d � t� | _| �td | jj� | �td | jj� d S ) N� ssh_host_foos foo�bar_key�ssh_host_one_key�ssh_host_two_key�ssh_host_three_keys not a key contentzssh_host_one_key.pub�modulis� # $OpenBSD: moduli,v 1.xx 2016/07/26 12:34:56 jhacker Exp $i # Time Type Tests Tries Size Generator Modulus 20030501000000 2 6 100 2047 2 FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995497CEA956AE515D2261898FA051015728E5A8AACAA68FFFFFFFFFFFFFFFF �seteuid�setegid)r �factoryr �mktemp�keysDir�makedirs�path�dataRoot� moduliDir� moduliRoot�child� setContentr �privateRSA_openssh�privateDSA_openssh�publicRSA_opensshr �mockos�patch�osr r )�self� r( �H/usr/lib/python3/dist-packages/twisted/conch/test/test_openssh_compat.py�setUp% s& �zOpenSSHFactoryTests.setUpc C s8 | j �� }| �t|�d� |�� }| �t|�dg� dS )zw L{OpenSSHFactory.getPublicKeys} should return the available public keys in the data directory � � ssh-rsaN)r � getPublicKeys� assertEqual�len�keys�list�r' r0 �keyTypesr( r( r) �test_getPublicKeysJ s z&OpenSSHFactoryTests.test_getPublicKeysc C sZ | j �� }| �t|�d� |�� }| �t|�ddh� | �| jjg � | �| jjg � dS )z� Will return the available private keys in the data directory, ignoring key files which failed to be loaded. � r, � ssh-dssN) r �getPrivateKeysr. r/ r0 �setr$ � seteuidCalls� setegidCallsr2 r( r( r) �test_getPrivateKeysT s z'OpenSSHFactoryTests.test_getPrivateKeysc s� | j �d�� � �d� | �� jd� tj�� �fdd�}| �td|� | j�� }| � t |�d� |� � }| � t |�dd h� | � | j jdt�� g� | � | j jdt�� g� d S ) z� L{OpenSSHFactory.getPrivateKeys} should switch to root if the keys aren't readable by the current user. r r � c s � � d� �| �S )Nr< )�chmod)�euid��keyFile� savedSeteuidr( r) r l s z>OpenSSHFactoryTests.test_getPrivateKeysAsRoot.<locals>.seteuidr r5 r, r6 N)r r r= � addCleanupr& r r% r r7 r. r/ r0 r8 r$ r9 r r: �getegid)r' r r0 r3 r( r? r) �test_getPrivateKeysAsRoot` s z-OpenSSHFactoryTests.test_getPrivateKeysAsRootc C s$ | j �� }| �|dtd�gi� dS )zq L{OpenSSHFactory.getPrimes} should return the available primes in the moduli directory. i s diffie-hellman-group14-sha1N)r � getPrimesr. r )r' �primesr( r( r) �test_getPrimesx s ��z"OpenSSHFactoryTests.test_getPrimesN) �__name__� __module__� __qualname__�__doc__r* r4 r; rD rG r( r( r( r) r s % r )rK r& �unittestr �twisted.conch.ssh._kexr �twisted.conch.testr �twisted.python.filepathr �twisted.python.reflectr �twisted.test.test_processr �twisted.trial.unittestr �doSkip� skipReason�$twisted.conch.openssh_compat.factoryr �hasattrr r( r( r( r) �<module> s( PK F��Z ��m % __pycache__/test_unix.cpython-310.pycnu �[��� o �b( � @ sz d dl mZ d dlmZ d dlmZ d dlmZ ed�Zed�Z ee�G dd� d��Z G d d � d �Z G d d � d ej �Z d S )� )� implementer)�IReactorProcess)� requireModule)�unittest� cryptographyztwisted.conch.unixc @ s0 e Zd ZdZdd� Zdi dddddfdd�ZdS ) �MockProcessSpawnerzC An L{IReactorProcess} that logs calls to C{spawnProcess}. c C s g | _ d S �N)�_spawnProcessCalls��self� r �>/usr/lib/python3/dist-packages/twisted/conch/test/test_unix.py�__init__ s zMockProcessSpawner.__init__r Nr c C s$ | j �||||||||| d� � dS )zQ Log a call to C{spawnProcess}. Do not actually spawn a process. ) �processProtocol� executable�args�env�path�uid�gid�usePTY�childFDsN)r �append) r r r r r r r r r r r r r � spawnProcess s ��z MockProcessSpawner.spawnProcess)�__name__� __module__� __qualname__�__doc__r r r r r r r s �r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �StubUnixConchUserz` Enough of UnixConchUser to exercise SSHSessionForUnixConchUser in the tests below. c C s( ddl m}m} || _||� d�| _d S )N� )� StubClient�StubConnection)� transport)� test_sessionr r! �_homeDirectory�conn)r � homeDirectoryr r! r r r r = s zStubUnixConchUser.__init__c C s dS )N)NNr r r r r �getUserGroupIdC � z StubUnixConchUser.getUserGroupIdc C s | j S r )r$ r r r r � getHomeDirF s z StubUnixConchUser.getHomeDirc C s d S r r r r r r �getShellI r( zStubUnixConchUser.getShellN)r r r r r r' r) r* r r r r r 7 s r c @ s. e Zd Zedu r dZnedu rdZdd� ZdS )� TestSSHSessionForUnixConchUserNz Cannot run without cryptographyzUnix system requiredc C sV t � }d}t|�}tj||d�}d}dg}|�||� |j\}| �||d d � dS )zn C{execCommand} sets the C{HOME} environment variable to the avatar's home directory. z/made/up/path/)�reactorNznot-actually-executedr �HOME)r r �unix�SSHSessionForUnixConchUser� execCommandr � assertEqual)r � mockReactorr& �avatar�session�protocol�command�callr r r �testExecCommandEnvironmentT s z9TestSSHSessionForUnixConchUser.testExecCommandEnvironment)r r r r �skipr. r8 r r r r r+ M s r+ N)�zope.interfacer �twisted.internet.interfacesr �twisted.python.reflectr � twisted.trialr r r. r r �TestCaser+ r r r r �<module> s &PK F��Z��F�A A $ __pycache__/loopback.cpython-310.pycnu �[��� o �b� � @ s& d Z ddlmZ G dd� dej�ZdS )z4 Loopback helper used in test_ssh and test_recvline � )�loopbackc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS )� LoopbackRelayNc C s d| j jj�d�S )NzLoopbackRelay(�))�target� __class__�__name__��self� r �=/usr/lib/python3/dist-packages/twisted/conch/test/loopback.py� logPrefix s zLoopbackRelay.logPrefixc C sB t j�| |� | jd ur| j�� ddlm} |�d| j�| _d S )Nr )�reactor) r r �write� clearCall�cancel�twisted.internetr � callLater� _clearBuffer)r �datar r r r r s zLoopbackRelay.writec C s d | _ tj�| � d S )N)r r r � clearBufferr r r r r s zLoopbackRelay._clearBuffer)r � __module__� __qualname__r r r r r r r r r s r N)�__doc__�twisted.protocolsr r r r r r �<module> s PK F��Z�H�FE� E� $ __pycache__/test_ssh.cpython-310.pycnu �[��� o �b�� � @ s� d Z ddlZddlmZmZmZmZ ddlmZ ddl m Z ddl m Z ddl mZmZmZ ddlmZ dd lmZmZ dd lmZ dd lmZ ed �Zed �Zergddl m Z m Z ddl m Z m!Z!m"Z"m#Z# nG dd� d�Z G dd� d�Z$G dd� de j%�Z&G dd� d�Z'ddlm(Z( er�e(�)e'e&e#j*� G dd� dej+�Z,G dd � d �Z-G d d � d �Z.G d d � d �Z/G d!d"� d"�Z0edu�r-edu�r-dd#l m1Z1 dd$l m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 G d%d&� d&�Z8G d'd(� d(e1j9�Z:G d)d*� d*e4j;�Z<G d+d,� d,�Z=G d-d.� d.e=e6j>�Z?G d/d0� d0e=e6j@�ZAG d1d2� d2e7jB�ZCG d3d4� d4e3jD�ZEG d5d6� d6e2jF�ZGd7d8� ZHG d9d:� d:ejI�ZJG d;d<� d<ejI�ZKG d=d>� d>ejI�ZLG d?d@� d@ejI�ZMdS )Az! Tests for L{twisted.conch.ssh}. � N)�privateDSA_openssh�privateRSA_openssh�publicDSA_openssh�publicRSA_openssh)� LoopbackRelay)�portal)�UnauthorizedLogin)�defer�protocol�reactor)�ProcessTerminated)�failure�log)� requireModule)�unittest� cryptography�pyasn1)�avatar�error)�_kex�common� forwarding�sessionc @ s e Zd ZG dd� d�ZdS )r c @ s e Zd ZdS )zavatar.ConchUserN)�__name__� __module__� __qualname__� r r �=/usr/lib/python3/dist-packages/twisted/conch/test/test_ssh.py� ConchUser# s r N)r r r r r r r r r " s r c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�ConchTestRealma" A realm which expects a particular avatarId to log in once and creates a L{ConchTestAvatar} for that request. @ivar expectedAvatarID: The only avatarID that this realm will produce an avatar for. @ivar avatar: A reference to the avatar after it is requested. Nc C � || _ d S �N)�expectedAvatarID)�selfr" r r r �__init__4 � zConchTestRealm.__init__c G sN || j kr | jdurtd��t� | _|d | j| jjfS td| j �d|����)z� Return a new L{ConchTestAvatar} if the avatarID matches the expected one and this is the first avatar request. NzOnly one login allowedr zOnly z may log in, not )r" r r �ConchTestAvatar�logout)r# �avatarID�mind� interfacesr r r � requestAvatar7 s �z ConchTestRealm.requestAvatar)r r r �__doc__r r$ r+ r r r r r ' s r c @ sL e Zd ZdZes dZdZdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dS )r& z� An avatar against which various SSH features can be tested. @ivar loggedOut: A flag indicating whether the avatar logout method has been called. � cannot run without cryptographyFc C sB t j�| � i | _i | _| j�tjt j d�� | j �dt i� d S )N)� sessions direct-tcpip� crazy) r r r$ � listeners�globalRequests� channelLookup�updater � SSHSessionr �openConnectForwardingClient�subsystemLookup�CrazySubsystem�r# r r r r$ S s ��zConchTestAvatar.__init__c C � || j d< dS )N�foo� �r1 �r# �datar r r � global_foo_ � zConchTestAvatar.global_fooc C r9 )N�foo_2)r; � datar< r= r r r � global_foo_2c r@ z ConchTestAvatar.global_foo_2c C sh t �|�\}}ztj|t �| j||ft j�|d�}W n ty* t� d d� Y dS w || j ||f< dS )N)� interfacez2something went wrong with remote->local forwardingr r; ) r �unpackGlobal_tcpip_forwardr � listenTCP�SSHListenForwardingFactory�conn� SSHListenServerForwardingChannel� BaseExceptionr �errr0 �r# r> �host�port�listenerr r r �global_tcpip_forwardg s � � �z$ConchTestAvatar.global_tcpip_forwardc C s@ t �|�\}}| j�||fd �}|sdS | j||f= |�� dS )Nr r; )r rE r0 �get� stopListeningrL r r r �global_cancel_tcpip_forwardx s z+ConchTestAvatar.global_cancel_tcpip_forwardc C s0 d| _ | j�� D ] }t�d| � |�� qd S )NTzstopListening %s)� loggedOutr0 �valuesr �msgrR )r# rO r r r r' � s �zConchTestAvatar.logoutN) r r r r, r �skiprT r$ r? rC rP rS r' r r r r r& F s r& c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�ConchSessionForTestAvatarz2 An ISession adapter for ConchTestAvatar. c C s4 || _ | | j _d| _d| _d| _d| _t�� | _dS )zq Initialize the session and create a reference to it on the avatar for later inspection. NFr ) r � _testSession�cmd�proto�ptyReq�eofr �Deferred�onClose)r# r r r r r$ � s z"ConchSessionForTestAvatar.__init__c C s t �d� || _|| _d| _d S )Nzpty reqT)r rV � _terminalType� _windowSizer\ )r# �term� windowSize�attrsr r r �getPty� s z ConchSessionForTestAvatar.getPtyc C s" t �d� || _t|� d| _d S )Nz opening shell� shell)r rV r[ � EchoTransportrZ )r# r[ r r r � openShell� s z#ConchSessionForTestAvatar.openShellc C s� || _ || _|�� d }|dkr t|�}t�d|j� nG|dkr2t|�}|�|dd � � |�� n1|dkrHt |�}|�|dd � � |�� n|dkr^t |�}|�|dd � � |�� nt � d��d | j jj_d S ) Nr � falses echo� s secho� s eecho�bad execr; )rZ r[ �split�FalseTransportr � callLater�loseConnectionrg �write�SuperEchoTransport�ErrEchoTransportr � ConchErrorr rH � transport�expectedLoseConnection)r# r[ rZ �f�tr r r � execCommand� s( z%ConchSessionForTestAvatar.execCommandc C � d| _ d S �Nr; )r] r8 r r r � eofReceived� r% z%ConchSessionForTestAvatar.eofReceivedc C s, t �d| j � | jjj| _| j�d � d S )Nzclosed cmd "%s") r rV rZ r[ r �remoteWindowLeft�remoteWindowLeftAtCloser_ �callbackr8 r r r �closed� s z ConchSessionForTestAvatar.closedN) r r r r, r$ re rh ry r| r� r r r r rX � s rX )� componentsc @ s e Zd Zdd� Zdd� ZdS )r7 c O s d S r! r )r# �args�kwr r r r$ � s zCrazySubsystem.__init__c C s dS )z good ... good Nr r8 r r r �connectionMade� � z CrazySubsystem.connectionMadeN)r r r r$ r� r r r r r7 � s r7 c @ s e Zd ZdZdd� Zdd� ZdS )rn a False transport should act like a /bin/false execution, i.e. just exit with nonzero status, writing nothing to the terminal. @ivar proto: The protocol associated with this transport. @ivar closed: A flag tracking whether C{loseConnection} has been called yet. c C s || _ |�| � d| _dS )zY @type p L{twisted.conch.ssh.session.SSHSessionProcessProtocol} instance r N�r[ �makeConnectionr� �r# �pr r r r$ � s zFalseTransport.__init__c C sL | j rdS d| _ | j�� | j�� | j�� | j�t�tddd��� dS )zI Disconnect the protocol associated with this transport. Nr; � � r� r[ �inConnectionLost�outConnectionLost�errConnectionLost� processEndedr �Failurer r8 r r r rp � s z FalseTransport.loseConnectionN)r r r r, r$ rp r r r r rn � s rn c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )rg c C � || _ |�| � d| _d S �Nr r� r� r r r r$ � � zEchoTransport.__init__c C s> t �t|�� | j�|� | j�d� d|v r | �� d S d S )N� � )r rV �reprr[ � outReceivedrp r= r r r rq � s �zEchoTransport.writec C �L | j rd S d| _ | j�� | j�� | j�� | j�t�tdd d ��� d S �Nr; r r� r8 r r r rp � z EchoTransport.loseConnectionN�r r r r$ rq rp r r r r rg � s rg c @ r� )rs c C r� r� r� r� r r r r$ r� zErrEchoTransport.__init__c C s | j �|� | j �d� d S �Nr� )r[ � errReceivedr= r r r rq s zErrEchoTransport.writec C r� r� r� r8 r r r rp r� z ErrEchoTransport.loseConnectionNr� r r r r rs s rs c @ r� )rr c C r� r� r� r� r r r r$ $ r� zSuperEchoTransport.__init__c C s4 | j �|� | j �d� | j �|� | j �d� d S r� )r[ r� r� r= r r r rq ) s zSuperEchoTransport.writec C r� r� r� r8 r r r rp / r� z!SuperEchoTransport.loseConnectionNr� r r r r rr # s rr )�checkers)�channel� connection�factory�keysru �userauthc @ s e Zd ZejfZdd� ZdS )�ConchTestPasswordCheckerc C s. |j dkr|jdkrt�|j �S t�td��S )N� testuser� testpasszBad credentials)�username�passwordr �succeed�fail� Exception)r# � credentialsr r r �requestAvatarIdG s z(ConchTestPasswordChecker.requestAvatarIdN)r r r r� �IUsernamePassword�credentialInterfacesr� r r r r r� D s r� c @ � e Zd Zdd� ZdS )�ConchTestSSHCheckerc C s" |dks t | j| �dk rdS dS )Nr� � FT)�len�successfulCredentials)r# �avatarIdr r r �areDoneP s zConchTestSSHChecker.areDoneN)r r r r� r r r r r� O � r� c @ sF e Zd ZdZejejd�Zdd� Z dd� Z dd� Z d d � Z d d � Z d S )�ConchTestServerFactoryF)s ssh-userauth� ssh-connectionc C s4 t � }| j�� |_| |_t| d�r| j|_|| _|S )Nrv )�ConchTestServer� privateKeysr� �supportedPublicKeysr� �hasattrrv r[ )r# �addrr[ r r r � buildProtocol] s z$ConchTestServerFactory.buildProtocolc C � t j�t�t j�t�d�S �N)s ssh-rsas ssh-dss)r� �Key� fromStringr r r8 r r r � getPublicKeysh � �z$ConchTestServerFactory.getPublicKeysc C r� r� )r� r� r� r r r8 r r r �getPrivateKeysn r� z%ConchTestServerFactory.getPrivateKeysc C s dt �d�giS )a% Diffie-Hellman primes that can be used for the diffie-hellman-group-exchange-sha1 key exchange. @return: The primes and generators. @rtype: L{dict} mapping the key size to a C{list} of C{(generator, prime)} tupple. i s diffie-hellman-group14-sha1)r �getDHGeneratorAndPrimer8 r r r � getPrimest s z ConchTestServerFactory.getPrimesc C s t j�| ||�S r! )r� � SSHFactory� getService)r# �trans�namer r r r� � � z!ConchTestServerFactory.getServiceN)r r r �noisyr� �SSHUserAuthServerr� � SSHConnection�servicesr� r� r� r� r� r r r r r� U s � r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � ConchTestBaser c C s4 | j rd S t| d�st�d| � d|� ���d| _ d S )Nrv z unexpectedly lost connection � r; )�doner� r �FailTest�r# �reasonr r r �connectionLost� s � z ConchTestBase.connectionLostc C s4 d| _ |tjkrt�td| ||f �� | �� d S )Nr; z*got disconnect for %s: reason %s, desc: %s)rv ru �DISCONNECT_BY_APPLICATIONr rK r� rp )r# � reasonCode�descr r r � receiveError� s ��� zConchTestBase.receiveErrorc C s t �d|� ���)Nzgot unimplemented: seqid )r r� )r# �seqIDr r r �receiveUnimplemented� r� z"ConchTestBase.receiveUnimplementedN)r r r r� r� r� r� r r r r r� � s r� c @ r� )r� c C � t �| |� tj�| |� d S r! )r� r� ru �SSHServerTransportr� r r r r� � � z ConchTestServer.connectionLostN)r r r r� r r r r r� � r� r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �ConchTestClientz� @ivar _channelFactory: A callable which accepts an SSH connection and returns a channel which will be attached to a new channel on that connection. c C r r! )�_channelFactory�r# �channelFactoryr r r r$ � r% zConchTestClient.__init__c C r� r! )r� r� ru �SSHClientTransportr� r r r r� � r� z ConchTestClient.connectionLostc C s< |t j�t��� k}|dk}|r|rt�d�S t�td��S )Ns/ 85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:dar; zKey or fingerprint mismatch) r� r� r� r �blobr r� r� r� )r# �key�fp�keyMatch�fingerprintMatchr r r � verifyHostKey� s z ConchTestClient.verifyHostKeyc C s | � tdt| j��� d S )Nr� )�requestService�ConchTestClientAuth�ConchTestClientConnectionr� r8 r r r �connectionSecure� s ��z ConchTestClient.connectionSecureN)r r r r, r$ r� r� r� r r r r r� � s r� c @ s8 e Zd ZdZdZdZdd� Zdd� Zdd� Zdd � Z d S ) r� r c C s( | j s | jr t�d��tj�| |� d S )Nz2got USERAUTH_SUCCESS before password and publickey)�canSucceedPassword�canSucceedPublicKeyr r� r� �SSHUserAuthClient�ssh_USERAUTH_SUCCESS)r# �packetr r r r� � s �z(ConchTestClientAuth.ssh_USERAUTH_SUCCESSc C s d| _ t�d�S )Nr; r� )r� r r� r8 r r r � getPassword� s z ConchTestClientAuth.getPasswordc C s d| _ t�tj�t��S r{ )r� r r� r� r� r� r r8 r r r � getPrivateKey� � z!ConchTestClientAuth.getPrivateKeyc C s t j�t�S r! )r� r� r� r r8 r r r � getPublicKey� s z ConchTestClientAuth.getPublicKeyN) r r r � hasTriedNoner� r� r� r� r� r� r r r r r� � s r� c @ s, e Zd ZdZdZdZdZdd� Zdd� Zd S ) r� z� @ivar _completed: A L{Deferred} which will be fired when the number of results collected reaches C{totalResults}. r� r � c C s t j�| � || _d S r! )r� r� r$ r� r� r r r r$ � s z"ConchTestClientConnection.__init__c C s | � | j| d�� d S )N�rH )� openChannelr� r8 r r r �serviceStarted� s z(ConchTestClientConnection.serviceStartedN) r r r r, r� �results� totalResultsr$ r� r r r r r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�SSHTestChannelc O s@ || _ || _g | _g | _t�� | _tjj | g|�R i |�� d S r! ) r� �_opened�received� receivedExtr r^ r_ r� � SSHChannelr$ )r# r� �openedr� �kwargsr r r r$ � s zSSHTestChannel.__init__c C � | j �|� d S r! )r �errbackr� r r r � openFailed� r� zSSHTestChannel.openFailedc C s | j �| � d S r! )r r )r# �ignorer r r � channelOpen� r� zSSHTestChannel.channelOpenc C r r! )r �appendr= r r r � dataReceived� r� zSSHTestChannel.dataReceivedc C s. |t jkr | j�|� d S t�d|��� d S )Nz Unrecognized extended data: )r� �EXTENDED_DATA_STDERRr r r rV )r# �dataTyper> r r r � extReceived s zSSHTestChannel.extReceivedc C s t �d|�\| _d S )Nz>L)�struct�unpack�status)r# r r r r �request_exit_status s z"SSHTestChannel.request_exit_statusc C rz )NT)� eofCalledr8 r r r r| r% zSSHTestChannel.eofReceivedc C s | j �d � d S r! )r_ r r8 r r r r� r� zSSHTestChannel.closedN) r r r r$ r r r r r r| r� r r r r r � s r c C s" t �dtj�t�gi�} t �| �S )z� Produces a SSHPublicKeyChecker with an in-memory key mapping with a single use: 'testuser' @return: L{twisted.conch.checkers.SSHPublicKeyChecker} r� )r� �InMemorySSHKeyDBr� r� r� r �SSHPublicKeyChecker)�conchTestPublicKeyDBr r r �conchTestPublicKeyChecker s � r c @ sj e Zd ZdZes dZesdZddd�Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� ZdS )�SSHProtocolTestsz^ Tests for communication between L{SSHServerTransport} and L{SSHClientTransport}. � can't run without cryptography�Cannot run without PyASN1r. c s� t �� �td�| _t�| j�}t� }|�t� � |�t � � |�|� t � }||_|� � |� d�| _ t| j �| _t� ��fdd��| _t| j�| _| j �| j� | j�| j� �S )z� Create a connected SSH client and server protocol pair and return a L{Deferred} which fires with an L{SSHTestChannel} instance connected to a channel on that SSH connection. r� Nc s t ��fd| i� ��S )NrH )r r� �r r� �resultr r �<lambda>= s z:SSHProtocolTests._ourServerOurClientTest.<locals>.<lambda>)r r^ r �realmr �Portalr� �registerCheckerr� r r� � startFactoryr� �serverr �clientTransportr� �client�serverTransportr� )r# r� r r� �sshpc�facr r r �_ourServerOurClientTest* s& � z(SSHProtocolTests._ourServerOurClientTestc sd � � � }� fdd�}|�|� � fdd�}|�|� � fdd�}|�|� � fdd�}|�|� |S ) z� Run the Conch server against the Conch client. Set up several different channels which exercise different behaviors and wait for them to complete. Verify that the channels with errors log them. c �& | � _ � �| j�| dt�d�d�t�S )N� subsystems not-crazyr; �r� � assertFailurerH � sendRequestr �NSr� �r� r8 r r � cbSubsystemN s ��zFSSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.cbSubsystemc s � j }|j�|dt�d�d�S )Nr. r/ r; �r� rH r1 r r2 ��ignoredr� r8 r r �cbNotCrazyFailedY � �zKSSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.cbNotCrazyFailedc s\ � j }|j�ddd�}|j�ddd�}|�� jd� � �|j�ddd�t�}t�|||g�S )N� foo� barr; s foo-2� bar2rB ) r� rH �sendGlobalRequest� addCallback� assertEqualr0 r� r � gatherResults)r7 r� �d1�d2�d3r8 r r �cbGlobalRequestsa s �zKSSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.cbGlobalRequestsc s> � � � jjjddd�� � j}d|jj_|j�� |� � d S )Nr; r< )r: rA T) r? r" r r1 r� rH ru rv �serviceStoppedrp r6 r8 r r � disconnectp s � zESSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.disconnect�r, r> )r# r� r4 r8 rD rF r r8 r � test_subsystemsAndGlobalRequestsF s z1SSHProtocolTests.test_subsystemsAndGlobalRequestsc st �� � }t�ddd�� � �fdd�}|�|� �fdd�}|�|� �fdd �}|�|� �fd d �}|�|� |S ) z� L{SSHChannel.sendRequest} can open a shell with a I{pty-req} request, specifying a terminal type and window size. � conch-test-term�� �P r r � c s | �_ | j�| d� d�S )Ns pty-reqr; )r� rH r1 r3 �r> r# r r � cbChannel� r� z.SSHProtocolTests.test_shell.<locals>.cbChannelc sd � j jjjd j}� �|j� j j� � �|jd� � �|jd� � � |j � � j }|j� |ddd�S )Nr rI rJ rf rM r; ) r" r rH �channelsr �assertIsr? r` ra � assertTruer\ r� r1 )r7 r r� r8 r r �cbPty� s z*SSHProtocolTests.test_shell.<locals>.cbPtyc s6 � j �d� � j j�� j � t�� j j� jjj jg�S )Ns testing the shell! ) r� rq rH �sendEOFr r@ r_ r" r rY �r7 r8 r r �cbShell� s �z,SSHProtocolTests.test_shell.<locals>.cbShellc sZ � j jdkrt�d� j jf � � �d�� j j�d� � �� j j� � �� j j j j � d S )Nr z shell exit status was not 0: %irM s testing the shell! ) r� r r rV r? �joinr rR r r" r rY r] rU r8 r r �cbExited� s �z-SSHProtocolTests.test_shell.<locals>.cbExited)r, r �packRequest_pty_reqr> )r# r� rO rS rV rX r rN r � test_shell} s zSSHProtocolTests.test_shellc s8 � � � }� fdd�}|�|� � fdd�}|�|� |S )z� If L{SSHChannel.sendRequest} issues an exec which the server responds to with an error, the L{Deferred} it returns fires its errback. c r- )N� execs jumboliahr; r/ r3 r8 r r rO � s �z3SSHProtocolTests.test_failedExec.<locals>.cbChannelc s$ � � tj�}� �|d jjd� d S )Nr )rl N)�flushLoggedErrorsr rt r? �valuer� �r7 �errorsr8 r r �cbFailed� s z2SSHProtocolTests.test_failedExec.<locals>.cbFailedrG )r# r� rO r` r r8 r �test_failedExec� s z SSHProtocolTests.test_failedExecc sN � � � }� fdd�}|�|� � fdd�}|�|� � fdd�}|�|� |S )z� When the process started by a L{SSHChannel.sendRequest} exec request exits, the exit status is reported to the channel. c � | � _ | j�| dt�d�d�S )Nr[ ri r; r5 r3 r8 r r rO � s z5SSHProtocolTests.test_falseChannel.<locals>.cbChannelc � � j jS r! �r� r_ rU r8 r r �cbExec� � z2SSHProtocolTests.test_falseChannel.<locals>.cbExecc s$ � � � jjg � � �� jjd� d S r� )r? r� r �assertNotEqualr rU r8 r r �cbClosed� s z4SSHProtocolTests.test_falseChannel.<locals>.cbClosedrG �r# r� rO re rh r r8 r �test_falseChannel� s z"SSHProtocolTests.test_falseChannelc �T � j ddd�}� fdd�}|�|� � fdd�}|�|� � fdd �}|�|� |S ) z� Bytes sent over the extended channel for stderr data are delivered to the channel's C{extReceived} method. � rj �� localWindow�localMaxPacketc rb )Nr[ s eecho hellor; r5 r3 r8 r r rO � r9 z5SSHProtocolTests.test_errorChannel.<locals>.cbChannelc � t �� jj� jjjjg�S r! �r r@ r� r_ r" r rY rU r8 r r re � � �z2SSHProtocolTests.test_errorChannel.<locals>.cbExecc sp � � � jjg � � � d�� jj�d� � � � jjd� � �� jj� � � � jjd� � � � jj� j j j j � d S )NrM � hello r rl ) r? r� r rW r r rR r �localWindowLeftr" r rY r~ rU r8 r r rh � s �z4SSHProtocolTests.test_errorChannel.<locals>.cbClosedrG ri r r8 r �test_errorChannel� s z"SSHProtocolTests.test_errorChannelc s, � � � �d�t�}� fdd�}|�|� |S )z� When an attempt is made to open an unknown channel type, the L{Deferred} returned by L{SSHChannel.sendRequest} fires its errback. s crazy-unknown-channelc s4 � � tj�}� �|d jjd� � �t|�d� d S )Nr )� zunknown channelr; )r\ r rt r? r] r� r� r^ r8 r r r` s z6SSHProtocolTests.test_unknownChannel.<locals>.cbFailed)r0 r, r� r> )r# �dr` r r8 r �test_unknownChannel s � z$SSHProtocolTests.test_unknownChannelc rk ) zc An L{SSHChannel} can be configured with a maximum packet size to receive. � r; rm c rb )Nr[ s secho hellor; r5 r3 r8 r r rO r9 z2SSHProtocolTests.test_maxPacket.<locals>.cbChannelc rc r! rd rU r8 r r re & rf z/SSHProtocolTests.test_maxPacket.<locals>.cbExecc s^ � � � jjd� � � d�� jj�d� � � d�� jj�d� � � � jjd� � �� jj� d S )Nr rM rs ry ) r? r� r rW r r rt rR r rU r8 r r rh + s z1SSHProtocolTests.test_maxPacket.<locals>.cbClosedrG ri r r8 r �test_maxPacket s z SSHProtocolTests.test_maxPacketc rk ) ze Normal standard out bytes are sent to the channel's C{dataReceived} method. rl rj rm c rb )Nr[ s echo hellor; r5 r3 r8 r r rO < r9 z-SSHProtocolTests.test_echo.<locals>.cbChannelc rp r! rq rU r8 r r �cbEchoD rr z*SSHProtocolTests.test_echo.<locals>.cbEchoc s` � � � jjd� � � d�� jj�d� � � � jjd� � �� jj� � � � jj� jj j j � d S )Nr rM rs rl ) r? r� r rW r rt rR r r" r rY r~ rU r8 r r rh K s �z,SSHProtocolTests.test_echo.<locals>.cbClosedrG )r# r� rO r{ rh r r8 r � test_echo5 s zSSHProtocolTests.test_echoN)r. )r r r r, r rW r r, rH rZ ra rj ru rx rz r| r r r r r s 70$ r c @ sN e Zd ZesdZes dZddd�Zdd� Zdd � Zd d � Z d d � Z dd� Z dS )�SSHFactoryTestsr r Nc s6 t �� }dd� }� fdd�|_| |_|_|�� |S )Nc S s dt �d �iS )Nzssh-rsa)r� r� r r r r r! c s z0SSHFactoryTests.makeSSHFactory.<locals>.<lambda>c s � S r! r r ��primesr r r! d r� )r� r� r� r� r� r% )r# r � sshFactory�gpkr r~ r �makeSSHFactorya s z SSHFactoryTests.makeSSHFactoryc C s$ | � � }|�d�}| �|tj� dS )zc By default, buildProtocol() constructs an instance of SSHServerTransport. N)r� r� �assertIsInstanceru r� )r# r� r r r r �test_buildProtocoli s z"SSHFactoryTests.test_buildProtocolc s: g � � fdd�}| � � }||_|�d� | �dg� � dS )zc buildProtocol() calls 'self.protocol()' to construct a protocol instance. c s � � | � t�� S r! )r ru r� )r� ��callsr r � makeProtocoly s zHSSHFactoryTests.test_buildProtocolRespectsProtocol.<locals>.makeProtocolNr )r� r r� r? )r# r� r� r r� r �"test_buildProtocolRespectsProtocolr s z2SSHFactoryTests.test_buildProtocolRespectsProtocolc C s6 | j dd�}|�d�}| �d|j� | �d|j� dS )zg Group key exchanges are not supported when we don't have the primes database. Nr~ �" diffie-hellman-group-exchange-sha1�$ diffie-hellman-group-exchange-sha256)r� r� � assertNotIn�supportedKeyExchanges)r# �f1�p1r r r �test_buildProtocolNoPrimes� s ��z*SSHFactoryTests.test_buildProtocolNoPrimesc C s: | j ddid�}|�d�}| �d|j� | �d|j� dS )zU Group key exchanges are supported when we have the primes database. r; )r� rv r~ Nr� r� )r� r� �assertInr� �r# �f2�p2r r r � test_buildProtocolWithPrimes� s z,SSHFactoryTests.test_buildProtocolWithPrimesc C s* | � � }|�d�}| �dd�|j�� dS )z_ ECDSA key exchanges are listed with 256 having a higher priority among ECDSA. Ns8 ecdh-sha2-nistp256,ecdh-sha2-nistp384,ecdh-sha2-nistp521� ,)r� r� r� rW r� r� r r r �test_buildProtocolKexECDSA� s �z*SSHFactoryTests.test_buildProtocolKexECDSAr! ) r r r r rW r r� r� r� r� r� r� r r r r r} Y s r} c @ sV e Zd ZdZes dZesdZereej �Z dd� Z dd� Z dd � Z d d � Z d d � ZdS )�MPTestszs Tests for L{common.getMP}. @cvar getMP: a method providing a MP parser. @type getMP: C{callable} r r c C � | � | �d�d� dS )z� L{common.getMP} should parse the a multiple precision integer from a string: a 4-byte length followed by length bytes of the integer. s )r; rM N�r? �getMPr8 r r r � test_getMP� � zMPTests.test_getMPc C r� )zv L{common.getMP} should be able to parse a big enough integer (that doesn't fit on one byte). s )irM Nr� r8 r r r �test_getMPBigInteger� � �z MPTests.test_getMPBigIntegerc C s | � | �dd�d� dS )zg L{common.getMP} has the ability to parse multiple integer in the same string. s r� )r; r� rM Nr� r8 r r r �test_multipleGetMP� s ��zMPTests.test_multipleGetMPc C r� )zu When more data than needed is sent to L{common.getMP}, it should return the remaining data. s foo)r; r: Nr� r8 r r r �test_getMPRemainingData� r� z MPTests.test_getMPRemainingDatac C s | � tj| jd� dS )z� When the string passed to L{common.getMP} doesn't even make 5 bytes, it should raise a L{struct.error}. s N)� assertRaisesr r r� r8 r r r �test_notEnoughData� r� zMPTests.test_notEnoughDataN)r r r r, r rW r � staticmethodr r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd ZdZes dZdd� ZdS )�GMPYInstallDeprecationTestszI Tests for the deprecation of former GMPY accidental public API. r- c C s> t �� | �| jg�}| �t|�d� | �|d d d� dS )zD L{twisted.conch.ssh.common.install} is deprecated. r; r �messagezAtwisted.conch.ssh.common.install was deprecated in Twisted 16.5.0N)r �install� flushWarnings�test_deprecatedr? r� )r# �warningsr r r r� � s �z+GMPYInstallDeprecationTests.test_deprecatedN)r r r r, r rW r� r r r r r� � s r� )Nr, r �twisted.conch.test.keydatar r r r �twisted.conch.test.loopbackr � twisted.credr �twisted.cred.errorr �twisted.internetr r r �twisted.internet.errorr �twisted.pythonr r �twisted.python.reflectr � twisted.trialr r r � twisted.conchr r �twisted.conch.sshr r r r r r r& rX r� �registerAdapter�ISession�Protocolr7 rn rg rs rr r� r� r� r� r� ru r� r� �SSHProtocolCheckerr� r� r� r� r� r� r� r� r� r� r� r� r r r �TestCaser r} r� r� r r r r �<module> sb B @� 1 ! =T?PK F��Z��k�i �i & __pycache__/test_conch.cpython-310.pycnu �[��� o �b�d � @ s d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dl m Z d dl m Z mZmZmZ d dlmZ d dlmZ d d lmZmZmZ d d lmZ d d lm Z d d l m Z d d l m Z m!Z!m"Z" d dl#m$Z$ d dl%m&Z& d dl'm(Z( d dl)m*Z*m+Z+ z d dlm,Z,m-Z- W n e.y� Y nw e(d�Z/e(d�Z0e0r�d dl1m2Z2 d dl3m4Z4m5Z5m6Z6 n d dl7m4Z4 G dd� d�Z2zd dl8m5Z9 W n e.y� Z: z dZ;e<e:�Z=[:W Y dZ:[:ndZ:[:ww e9Z;dd � Z>e>� Z?G d d � d �Z@G d d � d e+�ZAG d!d"� d"ejB�ZCG d#d$� d$ejD�ZEG d%d&� d&ejF�ZGG d'd(� d(ejF�ZHG d)d*� d*ejB�ZIdBd,d-�ZJG d.d/� d/�ZKG d0d1� d1eK�ZLe e4�G d2d3� d3e2��ZMG d4d5� d5�ZNG d6d7� d7eK�ZOG d8d9� d9�ZPG d:d;� d;eKePe+�ZQG d<d=� d=eLePe+�ZRG d>d?� d?eOePe+�ZSG d@dA� dAeLe+�ZTdS )C� N)�count)�skipIf)� implementer)� ConchError)�privateDSA_openssh�privateRSA_openssh�publicDSA_openssh�publicRSA_openssh��ConchTestRealm)�portal)�defer�protocol�reactor)�ProcessExitedAlready)� LoopingCall)�getProcessValue)�filepath�log�runtime)�FilePath)�which)� requireModule)�SkipTest�TestCase)�ConchTestServerFactory�conchTestPublicKeyChecker�pyasn1� cryptography)� ConchUser)�ISession� SSHSession� wrapProtocol)r c @ s e Zd ZdS )r N)�__name__� __module__� __qualname__� r& r& �?/usr/lib/python3/dist-packages/twisted/conch/test/test_conch.pyr 2 s r )r! c C sJ d} d}zt � t j�} | �d� d}W n ty Y nw | r#| �� |S )z4Returns True if the system can bind an IPv6 address.NF)�::1r T)�socket�AF_INET6�bind�OSError�close)�sock�has_ipv6r& r& r' � _has_ipv6@ s �r0 c @ s e Zd ZdZdZdd� ZdS )� FakeStdioz� A fake for testing L{twisted.conch.scripts.conch.SSHSession.eofReceived} and L{twisted.conch.scripts.cftp.SSHSession.eofReceived}. @ivar writeConnLost: A flag which records whether L{loserWriteConnection} has been called. Fc C s d| _ dS )z9 Record the call to loseWriteConnection. TN)� writeConnLost��selfr& r& r' �loseWriteConnection_ s z FakeStdio.loseWriteConnectionN)r# r$ r% �__doc__r2 r5 r& r& r& r' r1 T s r1 c @ s$ e Zd ZdZedu r eZdd� ZdS )� StdioInteractingSessionTestsz> Tests for L{twisted.conch.scripts.conch.SSHSession}. Nc C s* t � }t� }||_|�� | �|j� dS )z} L{twisted.conch.scripts.conch.SSHSession.eofReceived} loses the write half of its stdio connection. N)r1 �StdioInteractingSession�stdio� eofReceived� assertTruer2 )r4 r9 �channelr& r& r' �test_eofReceivedn s z-StdioInteractingSessionTests.test_eofReceived)r# r$ r% r6 r8 �_reason�skipr= r& r& r& r' r7 f s r7 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Echoc C � t �d� d S )NzECHO CONNECTION MADE�r �msgr3 r& r& r' �connectionMade{ � zEcho.connectionMadec C rA )NzECHO CONNECTION DONErB �r4 �reasonr& r& r' �connectionLost~ rE zEcho.connectionLostc C s& | j �|� d|v r| j �� d S d S )N� )� transport�write�loseConnection�r4 �datar& r& r' � dataReceived� s �zEcho.dataReceivedN)r# r$ r% rD rH rO r& r& r& r' r@ z s r@ c @ s e Zd ZeZdS )� EchoFactoryN)r# r$ r% r@ r r& r& r& r' rP � � rP c @ s< e Zd ZdZdZdZdZdd� Zdd� Zdd � Z d d � Z dS ) �ConchTestOpenSSHProcessa Test protocol for launching an OpenSSH client process. @ivar deferred: Set by whatever uses this object. Accessed using L{_getDeferred}, which destroys the value so the Deferred is not fired twice. Fires when the process is terminated. N� c C � | j d }| _ |S �N��deferred�r4 �dr& r& r' � _getDeferred� � z$ConchTestOpenSSHProcess._getDeferredc C � | j |7 _ d S rU )�bufrM r& r& r' � outReceived� � z#ConchTestOpenSSHProcess.outReceivedc C r\ rU )�problemsrM r& r& r' � errReceived� r_ z#ConchTestOpenSSHProcess.errReceivedc C sV |j jdkr| �� �td�|j j| j�d���� dS | j� dd�}| �� � |� dS )z~ Called when the process has ended. @param reason: a Failure giving the reason for the process' end. r z exit code was not 0: {} ({})�charmaps rI N) �value�exitCoderZ �errbackr �formatr` �decoder] �replace�callback)r4 rG r] r& r& r' � processEnded� s ��� z$ConchTestOpenSSHProcess.processEnded) r# r$ r% r6 rW r] r` rZ r^ ra rj r& r& r& r' rR � s rR c @ sT e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dS )�ConchTestForwardingProcessa� Manages a third-party process which launches a server. Uses L{ConchTestForwardingPort} to connect to the third-party server. Once L{ConchTestForwardingPort} has disconnected, kill the process and fire a Deferred with the data received by the L{ConchTestForwardingPort}. @ivar deferred: Set by whatever uses this object. Accessed using L{_getDeferred}, which destroys the value so the Deferred is not fired twice. Fires when the process is terminated. Nc C s || _ d| _|| _dS )aF @type port: L{int} @param port: The port on which the third-party server is listening. (it is assumed that the server is running on localhost). @type data: L{str} @param data: This is sent to the third-party server. Must end with ' ' in order to trigger a disconnect. N)�port�bufferrN )r4 rl rN r& r& r' �__init__� s z#ConchTestForwardingProcess.__init__c C rT rU rV rX r& r& r' rZ � r[ z'ConchTestForwardingProcess._getDeferredc C s | � � d S rU )�_connectr3 r& r& r' rD � s z)ConchTestForwardingProcess.connectionMadec C s0 t �tt| | j�}|�d| j�}|�| j� |S )a Connect to the server, which is often a third-party process. Tries to reconnect if it fails because we have no way of determining exactly when the port becomes available for listening -- we can only know when the process starts. � 127.0.0.1) r � ClientCreatorr �ConchTestForwardingPortrN � connectTCPrl � addErrback� _ebConnect)r4 �ccrY r& r& r' ro � s z#ConchTestForwardingProcess._connectc C s t �d| j� d S )Ng�������?)r � callLaterro )r4 �fr& r& r' ru � r_ z%ConchTestForwardingProcess._ebConnectc C s. || _ | j�d� | j�� t�d| j� dS )z� The network connection has died; save the buffer of output from the network and attempt to quit the process gracefully, and then (after the reactor has spun) send it a KILL signal. � r N)rm rJ rK rL r rw � _reallyDie)r4 rm r& r& r' �forwardingPortDisconnected� s z5ConchTestForwardingProcess.forwardingPortDisconnectedc C s( z | j �d� W d S ty Y d S w )N�KILL)rJ � signalProcessr r3 r& r& r' rz � s �z%ConchTestForwardingProcess._reallyDiec C s | � � �| j� dS )z� Fire the Deferred at self.deferred with the data collected from the L{ConchTestForwardingPort} connection, if any. N)rZ ri rm rF r& r& r' rj � s z'ConchTestForwardingProcess.processEnded) r# r$ r% r6 rW rn rZ rD ro ru r{ rz rj r& r& r& r' rk � s rk c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) rr z� Connects to server launched by a third-party process (managed by L{ConchTestForwardingProcess}) sends data, then reports whatever it received back to the L{ConchTestForwardingProcess} once the connection is ended. c C s || _ || _dS )z� @type protocol: L{ConchTestForwardingProcess} @param protocol: The L{ProcessProtocol} which made this connection. @type data: str @param data: The data to be sent to the third-party server. N)r rN )r4 r rN r& r& r' rn s z ConchTestForwardingPort.__init__c C s d| _ | j�| j� d S )NrS )rm rJ rK rN r3 r& r& r' rD s z&ConchTestForwardingPort.connectionMadec C r\ rU )rm rM r& r& r' rO r_ z$ConchTestForwardingPort.dataReceivedc C s | j �| j� d S rU )r r{ rm rF r& r& r' rH r_ z&ConchTestForwardingPort.connectionLostN)r# r$ r% r6 rn rD rO rH r& r& r& r' rr s rr �conchc C sF t jd| g}g }|t| � D ]}t|t�r|�d�}|�|� q|S )Na$ -c ### Twisted Preamble import sys, os path = os.path.abspath(sys.argv[0]) while os.path.dirname(path) != path: if os.path.basename(path).startswith('Twisted'): sys.path.insert(0, path) break path = os.path.dirname(path) from twisted.conch.scripts.%s import run run()�utf-8)�sys� executable�list� isinstance�str�encode�append)�args�mod�start�madeArgs�argr& r& r' � _makeArgs s �� r� c @ sP e Zd ZesdZes dZedd� �Zdd� Zdd� Z d d � Z d d � Z d d� Z dS )�ConchServerSetupMixinz can't run without cryptographyzCannot run without PyASN1c C s t d�S )Ns testuserr r& r& r& r' � realmFactoryA s z"ConchServerSetupMixin.realmFactoryc C sj dD ] }t j�|�rt �|� qtdd�� }|�t� W d � n1 s%w Y tdd�� }|�t� W d � n1 s?w Y tdd�� }|�t� W d � n1 sYw Y tdd�� }|�t � W d � n1 ssw Y t � dd� t � dd� t d�� � }|j js�|jjr�td��td d��}|�d t � W d � d S 1 s�w Y d S ) N)�rsa_test� rsa_test.pub�dsa_test� dsa_test.pub�kh_testr� �wbr� r� r� i� zgprivate key readable by others despite chmod; possible windows permission issue? see https://tm.tl/9767r� s 127.0.0.1 )�os�path�exists�remove�openrK r r r r �chmodr �getPermissions�group�read�otherr )r4 rx � permissionsr& r& r' � _createFilesE s4 � � � � � � "�z"ConchServerSetupMixin._createFilesc C s* t � � }|�d� |�� d }|�� |S )N)� r � )r) r+ � getsocknamer- )r4 �srl r& r& r' � _getFreePort] s z"ConchServerSetupMixin._getFreePortc C s. | � � }t�|�}|�t� � t� }||_|S )z� Make a L{ConchTestServerFactory}, which allows us to start a L{ConchTestServer} -- i.e. an actually listening conch. )r� r �Portal�registerCheckerr r )r4 �realm�p�factoryr& r& r' �_makeConchFactoryd s z'ConchServerSetupMixin._makeConchFactoryc C sz | � � | �� | _d| j_tjd| jdd�| _t�dt� �| _| j� � j | _ t r;tjdt� dd�| _ | j � � j | _d S d S )Nr� r rp )� interfacer( )r� r� � conchFactory�expectedLoseConnectionr � listenTCP� conchServerrP � echoServer�getHostrl �echoPort�HAS_IPV6� echoServerV6� echoPortV6r3 r& r& r' �setUpp s ��zConchServerSetupMixin.setUpc C sn zd| j j_W n ty Y nw | j jj�� t�| jj �t�| j j �g}t r2|� t�| j j �� t�|�S )Nr� )r� �proto�done�AttributeErrorrJ rL r � maybeDeferredr� � stopListeningr� r� r� r� � gatherResults)r4 � deferredsr& r& r' �tearDown} s � � z ConchServerSetupMixin.tearDownN) r# r$ r% r r? r � staticmethodr� r� r� r� r� r� r& r& r& r' r� : s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ForwardingMixina� Template class for tests of the Conch server's ability to forward arbitrary protocols over SSH. These tests are integration tests, not unit tests. They launch a Conch server, a custom TCP server (just an L{EchoProtocol}) and then call L{execute}. L{execute} is implemented by subclasses of L{ForwardingMixin}. It should cause an SSH client to connect to the Conch server, asking it to forward data to the custom TCP server. c C s | � dt� �}|�| jd�S )z� Test that we can use whatever client to send the command "echo goodbye" to the Conch server. Make sure we receive "goodbye" back from the server. � echo goodbye� goodbye ��executerR � addCallback� assertEqualrX r& r& r' � test_exec� s zForwardingMixin.test_execc C �> | � � }t|d�}| jd|d|| jf d�}|�| jd� |S )zy Test that we can use whatever client to forward a local port to a specified port on the server. � test r� z-N -L%i:127.0.0.1:%i��sshArgs�r� rk r� r� r� r� �r4 � localPort�processrY r& r& r' � test_localToRemoteForwarding� � �z,ForwardingMixin.test_localToRemoteForwardingc C r� )zs Test that we can use whatever client to forward a port from the server to a port locally. r� r� z-N -R %i:127.0.0.1:%ir� r� r� r& r& r' � test_remoteToLocalForwarding� r� z,ForwardingMixin.test_remoteToLocalForwardingN)r# r$ r% r6 r� r� r� r& r& r& r' r� � s r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )� RekeyAvatara# This avatar implements a shell which sends 60 numbered lines to whatever connects to it, then closes the session with a 0 exit status. 60 lines is selected as being enough to send more than 2kB of traffic, the amount the client is configured to initiate a rekey after. c C s t �| � t| jd< d S )Ns session)r rn r! � channelLookupr3 r& r& r' rn � s zRekeyAvatar.__init__c sH t �� }|��� ��t|�� � �fdd�}t|t� �� � �d� dS )zE Write 60 lines of data to the transport, then exit. c sZ t | �}|dkr � �� �jj��jdd� ��� d S d|f }|�d�}��|� d S )N�< s exit-statuss z line #%02d r )�next�stop�session�conn� sendRequestrL r� rK )�counter�i�line��callrJ r& r' rK � s � z$RekeyAvatar.openShell.<locals>.writeg{�G�z�?N)r �Protocol�makeConnectionr" r r r� )r4 rJ r� rK r& r� r' � openShell� s zRekeyAvatar.openShellc C s dS )z2 Ignore the close of the session. Nr& r3 r& r& r' �closed� � zRekeyAvatar.closedc C � d S rU r& r3 r& r& r' r: � � zRekeyAvatar.eofReceivedc C r� rU r& )r4 r� �commandr& r& r' � execCommand� r� zRekeyAvatar.execCommandc C r� rU r& )r4 �term� windowSize�modesr& r& r' �getPty� r� zRekeyAvatar.getPtyc C r� rU r& )r4 � newWindowSizer& r& r' � windowChanged r� zRekeyAvatar.windowChangedN) r# r$ r% r6 rn r� r� r: r� r� r� r& r& r& r' r� � s ! r� c @ s e Zd ZdZdd� ZdS )� RekeyRealmzS This realm gives out new L{RekeyAvatar} instances for any avatar request. c G s |d t � dd� fS )Nr c S r� rU r& r& r& r& r' �<lambda> r� z*RekeyRealm.requestAvatar.<locals>.<lambda>)r� )r4 �avatarID�mind� interfacesr& r& r' � requestAvatar s zRekeyRealm.requestAvatarN)r# r$ r% r6 r� r& r& r& r' r� s r� c @ s e Zd ZdZeZdd� ZdS )�RekeyTestsMixinzp TestCase mixin which defines tests exercising L{SSHTransportBase}'s handling of rekeying messages. c s. t � }� �d|d�}� fdd�}|�|� |S )z� After a client-initiated rekey is completed, application data continues to be passed over the SSH connection. r� z-o RekeyLimit=2Kc s6 d� dd� td�D ��d }|�d�}� �| |� d S )N� c S s g | ]}d |f �qS )z line #%02dr& )�.0r� r& r& r' � <listcomp>" s zFRekeyTestsMixin.test_clientRekey.<locals>.finished.<locals>.<listcomp>r� r )�join�ranger� r� )�result�expectedResultr3 r& r' �finished! s z2RekeyTestsMixin.test_clientRekey.<locals>.finished)rR r� r� )r4 r� rY r� r& r3 r' �test_clientRekey s z RekeyTestsMixin.test_clientRekeyN)r# r$ r% r6 r� r� r� r& r& r& r' r� s r� c @ s" e Zd Zed�s dZddd�ZdS )�OpenSSHClientMixin�sshz$no ssh command-line client availabler� c s. t td�d d�}� ���fdd�}|�|�S )a� Connects to the SSH server started in L{ConchServerSetupMixin.setUp} by running the 'ssh' command line tool. @type remoteCommand: str @param remoteCommand: The command (with arguments) to run on the remote end. @type process: L{ConchTestOpenSSHProcess} @type sshArgs: str @param sshArgs: Arguments to pass to the 'ssh' process. @return: L{defer.Deferred} r r )z-oz PubkeyAcceptedKeyTypes=ssh-dssz-Vc s� | dkrd}nd}t �� � _d| � d � }�j�� j}|| �� }g }|D ]}t|t�r4|� d�}|� |� q(t � � t d�d |� � jS )Nr z!-oPubkeyAcceptedKeyTypes=ssh-dss r� z�ssh -2 -l testuser -p %i -F /dev/null -oUserKnownHostsFile=kh_test -oPasswordAuthentication=no -oHostKeyAlgorithms=ssh-rsa -a -i dsa_test � 127.0.0.1 r r )r �DeferredrW r� r� rl �splitr� r� r� r� r � spawnProcessr )�status�opts�cmdlinerl �cmds� encodedCmds�cmd�r� � remoteCommandr4 r� r& r' �hasPAKTF s. � � � �� z+OpenSSHClientMixin.execute.<locals>.hasPAKT)r r r� )r4 r r� r� rY r r& r r' r� . s � $zOpenSSHClientMixin.executeN)r� )r# r$ r% r r? r� r& r& r& r' r * s r c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�OpenSSHKeyExchangeTestsz\ Tests L{SSHTransportBase}'s key exchange algorithm compatibility with OpenSSH. c C s� g }z t jtd�d ddgt jd�}t|t�s |�d�}|�� }W n ty* Y nw ||vr6t |� d���| � dt � d | �}|� | j d �S ) aI Call execute() method of L{OpenSSHClientMixin} with an ssh option that forces the exclusive use of the key exchange algorithm specified by keyExchangeAlgo @type keyExchangeAlgo: L{str} @param keyExchangeAlgo: The key exchange algorithm to use @return: L{defer.Deferred} r r z-Q�kex)�stderrr z not supported by ssh clientz echo helloz-oKexAlgorithms=s hello )� subprocess� check_outputr �STDOUTr� r� rg r � BaseExceptionr r� rR r� r� )r4 �keyExchangeAlgo� kexAlgorithms�outputrY r& r& r' � assertExecuteWithKexAlgorithms s& � ��z5OpenSSHKeyExchangeTests.assertExecuteWithKexAlgorithmc C � | � d�S )zb The ecdh-sha2-nistp256 key exchange algorithm is compatible with OpenSSH zecdh-sha2-nistp256�r r3 r& r& r' �test_ECDHSHA256� � z'OpenSSHKeyExchangeTests.test_ECDHSHA256c C r )zb The ecdh-sha2-nistp384 key exchange algorithm is compatible with OpenSSH zecdh-sha2-nistp384r r3 r& r& r' �test_ECDHSHA384� r z'OpenSSHKeyExchangeTests.test_ECDHSHA384c C r )zb The ecdh-sha2-nistp521 key exchange algorithm is compatible with OpenSSH zecdh-sha2-nistp521r r3 r& r& r' �test_ECDHSHA521� r z'OpenSSHKeyExchangeTests.test_ECDHSHA521c C r )zl The diffie-hellman-group14-sha1 key exchange algorithm is compatible with OpenSSH. zdiffie-hellman-group14-sha1r r3 r& r& r' �test_DH_GROUP14� r z'OpenSSHKeyExchangeTests.test_DH_GROUP14c C r )zs The diffie-hellman-group-exchange-sha1 key exchange algorithm is compatible with OpenSSH. z"diffie-hellman-group-exchange-sha1r r3 r& r& r' �test_DH_GROUP_EXCHANGE_SHA1� r z3OpenSSHKeyExchangeTests.test_DH_GROUP_EXCHANGE_SHA1c C r )zu The diffie-hellman-group-exchange-sha256 key exchange algorithm is compatible with OpenSSH. z$diffie-hellman-group-exchange-sha256r r3 r& r& r' � test_DH_GROUP_EXCHANGE_SHA256� s �z5OpenSSHKeyExchangeTests.test_DH_GROUP_EXCHANGE_SHA256c C s | � t| jd� dS )zy The list of key exchange algorithms supported by OpenSSH client is obtained with C{ssh -Q kex}. zunsupported-algorithmN)� assertRaisesr r r3 r& r& r' �test_unsupported_algorithm� s �z2OpenSSHKeyExchangeTests.test_unsupported_algorithmN) r# r$ r% r6 r r r r r r! r" r$ r& r& r& r' r m s r c @ s$ e Zd ZdZee d�dd� �ZdS )� OpenSSHClientForwardingTestszR Connection forwarding tests run against the OpenSSL command line client. zRequires IPv6 supportc C r� )zG Forwarding of arbitrary IPv6 TCP connections via SSH. r� r� z-N -L%i:[::1]:%ir� )r� rk r� r� r� r� r� r& r& r' � test_localToRemoteForwardingV6� r� z;OpenSSHClientForwardingTests.test_localToRemoteForwardingV6N)r# r$ r% r6 r r� r& r& r& r& r' r% � s r% c @ s e Zd ZdZdS )�OpenSSHClientRekeyTestszE Rekeying tests run against the OpenSSL command line client. N)r# r$ r% r6 r& r& r& r' r' � rQ r' c @ s8 e Zd ZdZejdkr dZd dd�Zdd � Zd d � Z dS ) �CmdLineClientTestszP Connection forwarding tests run against the Conch command line client. �win32z!can't run cmdline client on win32r� Nc C s� |du rg }t �� |_| j�� j}d�|�| d | }t||�� �}t j � � }t j � tj�|d< g } i } |D ]}t|t�rD|�d�}| �|� q8|D ] } || } t| t�r\| �d�} t| t�rf| �d�} | | | < qLtj|tj| | d� |jS )z{ As for L{OpenSSHClientTestCase.execute}, except it runs the 'conch' command line tool, not 'ssh'. Nz[-p {} -l testuser --known-hosts kh_test --user-authentications publickey -a -i dsa_test -v r � PYTHONPATHr )�env)r r rW r� r� rl rf r� r r� �environ�copy�pathsepr� r� r� r� r� r� r� r r r� ) r4 r r� r� � conchArgsrl r r r+ r � encodedEnv�var�valr& r& r' r� � s< ���� zCmdLineClientTests.executec sV � �fdd�}t ���� �� �jdt� dd� jddgd�}|��jd � |�|� |S ) z4 It can store logs to a local file. c s � � � }��d|� d S )Ns Log opened.)� getContent�assertIn)r� � logContent��logPathr4 r& r' � cb_check_log s z<CmdLineClientTests.test_runWithLogFile.<locals>.cb_check_logr� z--logz --logfilez--host-key-algorithmszssh-rsa)r r� r/ r� )r r �mktempr� rR r� r� r� )r4 r8 rY r& r6 r' �test_runWithLogFile s �� z&CmdLineClientTests.test_runWithLogFilec C s" | j dt� d�}|�| jd� |S )zH Do not use --host-key-algorithms flag on command line. r� )r r� r� r� rX r& r& r' �%test_runWithNoHostAlgorithmsSpecified* s �z8CmdLineClientTests.test_runWithNoHostAlgorithmsSpecified)r� N) r# r$ r% r6 r � platformTyper? r� r: r; r& r& r& r' r( � s % r( )r~ )Ur� r) r r� � itertoolsr �unittestr �zope.interfacer �twisted.conch.errorr �twisted.conch.test.keydatar r r r �twisted.conch.test.test_sshr � twisted.credr �twisted.internetr r r �twisted.internet.errorr �twisted.internet.taskr �twisted.internet.utilsr �twisted.pythonr r r �twisted.python.filepathr �twisted.python.procutilsr �twisted.python.reflectr �twisted.trial.unittestr r r r � ImportErrorr r �twisted.conch.avatarr �twisted.conch.ssh.sessionr r! r" �twisted.conch.interfaces�twisted.conch.scripts.conch�_StdioInteractingSession�er8 r� r> r0 r� r1 r7 r� r@ �FactoryrP �ProcessProtocolrR rk rr r� r� r� r� r� r� r r r% r' r( r& r& r& r' �<module> s| � �� +L S6D C\PK F��Z�c�ngM gM ( __pycache__/test_ckeygen.cpython-310.pycnu �[��� o 5�@g_ � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZm Z m Z m Z m Z ddl mZ ddlmZ ddlmZ ed�r[ed �r[dd lmZmZmZmZmZ dd lmZmZm Z m Z nd Z d d� Z G dd� de�Z dS )z- Tests for L{twisted.conch.scripts.ckeygen}. � N)�StringIO)�privateECDSA_openssh�privateEd25519_openssh_new�privateRSA_openssh� privateRSA_openssh_encrypted�publicRSA_openssh)�FilePath)� requireModule)�TestCase� cryptography�pyasn1)�_saveKey�changePassPhrase�displayPublicKey�enumrepresentation�printFingerprint)�BadFingerPrintFormat� BadKeyError�FingerprintFormats�KeyzBcryptography and pyasn1 required for twisted.conch.scripts.ckeygenc s t � �� � fdd�}|S )a@ Return a callable to patch C{getpass.getpass}. Yields a passphrase each time called. Use case is to provide an old, then new passphrase(s) as if requested interactively. @param passphrases: The list of passphrases returned, one per each call. @return: A callable to patch C{getpass.getpass}. c s t � �S �N)�next��_�� passphrases� �A/usr/lib/python3/dist-packages/twisted/conch/test/test_ckeygen.py� fakeGetpass7 � z makeGetpass.<locals>.fakeGetpass)�iter)r r r r r � makeGetpass+ s r! c @ s" e Zd ZdZdd� ZdGdd�Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dS )H� KeyGenTestszN Tests for various functions used to implement the I{ckeygen} script. c C s t � | _| �td| j� dS )zX Patch C{sys.stdout} so tests can make assertions about what's printed. �stdoutN)r r# �patch�sys��selfr r r �setUpB s zKeyGenTests.setUpNc C s� | � � }dd|d|dg}|d ur|�d|g� |d ur"|�d|g� zt�|� W n ty; d|d< t�|� Y nw t�|�}t�|d �}|d krU| �|�� d � n|d krb| �|�� d � n | �|�� |� � � | � |� � � d S )N�ckeygen�-t�-fz--no-passphrasez-bz--private-key-subtype�ckeygen3r z.pub�ecdsa�EC�ed25519�Ed25519) �mktemp�extend� subprocess�call�FileNotFoundErrorr �fromFile� assertEqual�type�upper� assertTrue�isPublic)r' �keyType�keySize�privateKeySubtype�filename�args�privKey�pubKeyr r r �_testrunI s( � zKeyGenTests._testrunc C s� | � dd� | j dddd� | � d� | j ddd� | � d� | � dd� | j dddd� | � d� | j ddd� | � dd� | j dddd� | � d� | j ddd� d S ) Nr- �384�v1)r> r/ �dsa�2048�rsa)rC r& r r r �test_keygeneration_ s z KeyGenTests.test_keygenerationc C s� | � � }| �tj��K ttjd��,}ztjdddd|g|d� W n ty5 tjdddd|g|d� Y nw W d � n1 s@w Y W d � d S W d � d S 1 sXw Y d S )N�rbr) r* �foor+ )�stderrr, ) r1 � assertRaisesr3 �CalledProcessError�open�os�devnull� check_callr5 )r' r? rQ r r r �test_runBadKeytypen s& � ��� ��"�z KeyGenTests.test_runBadKeytypec C �" t ddi�}| �|d tj� dS )z� L{enumrepresentation} takes a dictionary as input and returns a dictionary with its attributes changed to enum representation. �format�md5-hexN)r �assertIsr �MD5_HEX�r' �optionsr r r �test_enumrepresentation{ s z#KeyGenTests.test_enumrepresentationc C rT )zF Test for format L{FingerprintFormats.SHA256-BASE64}. rU � sha256-base64N)r rW r � SHA256_BASE64rY r r r � test_enumrepresentationsha256� s z)KeyGenTests.test_enumrepresentationsha256c C sN | � t��}tddi� W d � n1 sw Y | �d|jjd � dS )z9 Test for unsupported fingerprint format rU � sha-base64N�*Unsupported fingerprint format: sha-base64r )rM r r r7 � exceptionr@ )r' �emr r r � test_enumrepresentationBadFormat� s � �z,KeyGenTests.test_enumrepresentationBadFormatc C �: | � � }t|��t� t|dd�� | �| j�� d� dS )z� L{printFingerprint} writes a line to standard out giving the number of bits of the key, its fingerprint, and the basename of the file from it was read. rV �r? rU z:2048 85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:da temp N�r1 r � setContentr r r7 r# �getvalue�r' r? r r r �test_printFingerprint� s �z!KeyGenTests.test_printFingerprintc C rd )z� L{printFigerprint} will print key fingerprint in L{FingerprintFormats.SHA256-BASE64} format if explicitly specified. r\ re z72048 FBTCOoknq0mHy+kpfnY9tDdcAJuWtCpuQMaV3EsvbUI= temp Nrf ri r r r �test_printFingerprintsha256� s �z'KeyGenTests.test_printFingerprintsha256c C sf | � � }t|��t� | �t��}t|dd�� W d � n1 s"w Y | �d|jj d � dS )zx L{printFigerprint} raises C{keys.BadFingerprintFormat} when unsupported formats are requested. r_ re Nr` r ) r1 r rg r rM r r r7 ra r@ )r' r? rb r r r �)test_printFingerprintBadFingerPrintFormat� s � �z5KeyGenTests.test_printFingerprintBadFingerPrintFormatc C � t | �� �}|�� |�d�j}t�t�}t||ddd�� | � | j � � d||f � | � |�|�d�� � dd�|� | � t�|�d�� � �|� � � dS )z� L{_saveKey} writes the private and public parts of a key to two different files and writes a report of this to standard out. �id_rsa� passphraserV �r? �passrU ��Your identification has been saved in %s Your public key has been saved in %s.pub The key fingerprint in <FingerprintFormats=MD5_HEX> is: 85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:da N� id_rsa.pub�r r1 �makedirs�child�pathr � fromStringr r r7 r# rh � getContent�public�r' �baser? �keyr r r � test_saveKey� s" ����zKeyGenTests.test_saveKeyc C rm )z� L{_saveKey} writes the private and public parts of a key to two different files and writes a report of this to standard out. Test with ECDSA key. �id_ecdsaro rV rp z�Your identification has been saved in %s Your public key has been saved in %s.pub The key fingerprint in <FingerprintFormats=MD5_HEX> is: 1e:ab:83:a6:f2:04:22:99:7c:64:14:d2:ab:fa:f5:16 Nz id_ecdsa.pub)r r1 ru rv rw r rx r r r7 r# rh ry rz r{ r r r �test_saveKeyECDSA� s" ����z KeyGenTests.test_saveKeyECDSAc C rm )z� L{_saveKey} writes the private and public parts of a key to two different files and writes a report of this to standard out. Test with Ed25519 key. � id_ed25519ro rV rp z�Your identification has been saved in %s Your public key has been saved in %s.pub The key fingerprint in <FingerprintFormats=MD5_HEX> is: ab:ee:c8:ed:e5:01:1b:45:b7:8d:b2:f0:8f:61:1c:14 Nzid_ed25519.pub)r r1 ru rv rw r rx r r r7 r# rh ry rz r{ r r r �test_saveKeyEd25519� s$ ����z KeyGenTests.test_saveKeyEd25519c C rm )z� L{_saveKey} will generate key fingerprint in L{FingerprintFormats.SHA256-BASE64} format if explicitly specified. rn ro r\ rp z�Your identification has been saved in %s Your public key has been saved in %s.pub The key fingerprint in <FingerprintFormats=SHA256_BASE64> is: FBTCOoknq0mHy+kpfnY9tDdcAJuWtCpuQMaV3EsvbUI= Nrs rt r{ r r r �test_saveKeysha256 s& �����z KeyGenTests.test_saveKeysha256c C s~ t | �� �}|�� |�d�j}t�t�}| �t ��}t ||ddd�� W d � n1 s.w Y | � d|j j d � dS )zq L{_saveKey} raises C{keys.BadFingerprintFormat} when unsupported formats are requested. rn ro r_ rp Nr` r )r r1 ru rv rw r rx r rM r r r7 ra r@ )r' r| r? r} rb r r r � test_saveKeyBadFingerPrintformat! s � � �z,KeyGenTests.test_saveKeyBadFingerPrintformatc C s` t | �� �}|�� |�d�j}t�t�}t||ddd�� | � |�|�d�� � dd�|� dS )�q L{_saveKey} will choose an empty string for the passphrase if no-passphrase is C{True}. rn TrV �r? z no-passphraserU N� ) r r1 ru rv rw r rx r r r7 ry r{ r r r �test_saveKeyEmptyPassphrase3 s ��z'KeyGenTests.test_saveKeyEmptyPassphrasec C �^ t | �� �}|�� |�d�j}t�t�}t||ddd�� | � |�|�d�� � d�|� dS )r� r TrV r� N) r r1 ru rv rw r rx r r r7 ry r{ r r r � test_saveKeyECDSAEmptyPassphraseC s �"z,KeyGenTests.test_saveKeyECDSAEmptyPassphrasec C r� )r� r� TrV r� N) r r1 ru rv rw r rx r r r7 ry r{ r r r �"test_saveKeyEd25519EmptyPassphraseQ s ��z.KeyGenTests.test_saveKeyEd25519EmptyPassphrasec s� t | �� �}|�� |�d�j� ddl}| �|jjj d� fdd�� t � t �}t |dddd �� |�d��� }|� |dd �}| �||� dS ) zd When no path is specified, it will ask for the path used to store the key. � custom_keyr N�_inputSaveFilec s � S r r r ��keyPathr r �<lambda>l � z4KeyGenTests.test_saveKeyNoFilename.<locals>.<lambda>TrV r� r� )r r1 ru rv rw � twisted.conch.scripts.ckeygenr$ �conch�scriptsr) r rx r r ry r7 )r' r| �twistedr} �persistedKeyContent� persistedKeyr r� r �test_saveKeyNoFilenamea s z"KeyGenTests.test_saveKeyNoFilenamec C s� t | �� �}|�� |�d�j}t�t�}t||dddd�� | � | j � � d||f � |�d�� � }| � |�|dd�|� | � |�d�� | � t�|�d �� � �|�� � dS ) zi L{_saveKey} can be told to write the new private key file in OpenSSH v1 format. rn ro rV rE )r? rq rU �private-key-subtyperr N�$ -----BEGIN OPENSSH PRIVATE KEY----- rs )r r1 ru rv rw r rx r r r7 r# rh ry r: � startswithrz )r' r| r? r} �privateKeyContentr r r �test_saveKeySubtypeV1t s4 �� ����z!KeyGenTests.test_saveKeySubtypeV1c C sf | � � }t�t�}t|��t� td|i� | j� � � d�}t |t �r(|� d�}| �||�d�� dS )zl L{displayPublicKey} prints out the public key associated with a given private key. r? � �ascii�opensshN)r1 r rx r r rg r r r# rh �strip� isinstance�str�encoder7 �toString�r' r? rB � displayedr r r �test_displayPublicKey� s z!KeyGenTests.test_displayPublicKeyc C sh | � � }t�t�}t|��t� t|dd�� | j� � � d�}t |t �r)|� d�}| �||�d�� dS )z� L{displayPublicKey} prints out the public key associated with a given private key using the given passphrase when it's encrypted. � encrypted�r? rq r� r� r� N)r1 r rx r r rg r r r# rh r� r� r� r� r7 r� r� r r r � test_displayPublicKeyEncrypted� s z*KeyGenTests.test_displayPublicKeyEncryptedc C sx | � � }t�t�}t|��t� | �tddd� � t d|i� | j � � � d�}t |t�r1|�d�}| �||�d�� dS ) z� L{displayPublicKey} prints out the public key associated with a given private key, asking for the passphrase when it's encrypted. �getpassc S � dS )Nr� r )�xr r r r� � r� zLKeyGenTests.test_displayPublicKeyEncryptedPassphrasePrompt.<locals>.<lambda>r? r� r� r� N)r1 r rx r r rg r r$ r� r r# rh r� r� r� r� r7 r� r� r r r �.test_displayPublicKeyEncryptedPassphrasePrompt� s z:KeyGenTests.test_displayPublicKeyEncryptedPassphrasePromptc C s. | � � }t|��t� | �tt|dd�� dS )z� L{displayPublicKey} fails with a L{BadKeyError} when trying to decrypt an encrypted key with the wrong password. �wrongr� N)r1 r rg r rM r r ri r r r �$test_displayPublicKeyWrongPassphrase� s �z0KeyGenTests.test_displayPublicKeyWrongPassphrasec C sl t ddd�}| �td|� | �� }t|��t� td|i� | �| j � � � d�d� | � tt|�� � � dS )zt L{changePassPhrase} allows a user to change the passphrase of a private key interactively. r� �newpassr� r? r� �;Your identification has been saved with the new passphrase.N�r! r$ r� r1 r rg r r r7 r# rh r� �assertNotEqualry )r' � oldNewConfirmr? r r r �test_changePassphrase� s � �z!KeyGenTests.test_changePassphrasec C sl t dd�}| �td|� | �� }t|��t� t|dd�� | �| j � � � d�d� | � tt|�� � � dS )z� L{changePassPhrase} allows a user to change the passphrase of a private key, providing the old passphrase and prompting for new one. r� r� r� r� r� r� Nr� )r' � newConfirmr? r r r � test_changePassphraseWithOld� s � �z(KeyGenTests.test_changePassphraseWithOldc C sV | � � }t|��t� t|ddd�� | �| j�� �d�d� | � tt|�� � � dS )z� L{changePassPhrase} allows a user to change the passphrase of a private key by providing both old and new passphrases without prompting. r� � newencrypt)r? rq r� r� r� N) r1 r rg r r r7 r# rh r� r� ry ri r r r � test_changePassphraseWithBoth� s �� �z)KeyGenTests.test_changePassphraseWithBothc C �R | � � }t|��t� | �tt|dd��}| �dt|�� | �tt|�� � � dS )z� L{changePassPhrase} exits if passed an invalid old passphrase when trying to change the passphrase of a private key. r� r� z1Could not change passphrase: old passphrase errorN) r1 r rg r rM � SystemExitr r7 r� ry �r' r? �errorr r r �$test_changePassphraseWrongPassphrase s ��z0KeyGenTests.test_changePassphraseWrongPassphrasec C sb | � tdtd�� | �� }t|��t� | �tt d|i�}| � dt |�� | � tt|�� � � dS )z� L{changePassPhrase} exits if no passphrase is specified for the C{getpass} call and the key is encrypted. r� � r? zMCould not change passphrase: Passphrase must be provided for an encrypted keyN) r$ r� r! r1 r rg r rM r� r r7 r� ry r� r r r �!test_changePassphraseEmptyGetPass s �z-KeyGenTests.test_changePassphraseEmptyGetPassc C sT | � � }t|��d� | �ttd|i�}d}| �|t|�� | �dt|��� � dS )zc L{changePassPhrase} exits if the file specified points to an invalid key. s foobarr? z?Could not change passphrase: cannot guess the type of b'foobar'N) r1 r rg rM r� r r7 r� ry )r' r? r� �expectedr r r �test_changePassphraseBadKey& s z'KeyGenTests.test_changePassphraseBadKeyc C sh | � � }t|��t� dd� }| �td|� | �tt|dd��}| � dt |�� | � tt|�� � � dS )z� L{changePassPhrase} doesn't modify the key file if an unexpected error happens when trying to create the key with the new passphrase. c _ s t d��)N�oops)� RuntimeError�r@ �kwargsr r r r� ; r z>KeyGenTests.test_changePassphraseCreateError.<locals>.toStringr� r� �r? r� z!Could not change passphrase: oopsN� r1 r rg r r$ r rM r� r r7 r� ry )r' r? r� r� r r r � test_changePassphraseCreateError3 s �z,KeyGenTests.test_changePassphraseCreateErrorc C sl | � � }t|��t� dd� }| �td|� | �tt|dd��}d}| � |t |�� | � tt|�� � � dS )zq L{changePassPhrase} doesn't modify the key file if C{toString} returns an empty string. c _ r� )Nr� r r� r r r r� R s zCKeyGenTests.test_changePassphraseEmptyStringError.<locals>.toStringr� r� r� z9Could not change passphrase: cannot guess the type of b''Nr� )r' r? r� r� r� r r r �%test_changePassphraseEmptyStringErrorJ s �z1KeyGenTests.test_changePassphraseEmptyStringErrorc C r� )z� L{changePassPhrase} exits when trying to change the passphrase on a public key, and doesn't change the file. rq r� z.Could not change passphrase: key not encryptedN) r1 r rg r rM r� r r7 r� ry r� r r r � test_changePassphrasePublicKeyb s �z*KeyGenTests.test_changePassphrasePublicKeyc C s� t ddd�}| �td|� | �� }t|��t� t|dd�� | �| j � � � d�d� t|�� � }| � t|� | �|�d�� d S ) zq L{changePassPhrase} can be told to write the new private key file in OpenSSH v1 format. r� r� r� rE )r? r� r� r� r� N)r! r$ r� r1 r rg r r r7 r# rh r� ry r� r: r� )r' r� r? r� r r r � test_changePassphraseSubtypeV1o s � �z*KeyGenTests.test_changePassphraseSubtypeV1)NN)&�__name__� __module__� __qualname__�__doc__r( rC rI rS r[ r^ rc rj rk rl r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r" = sH " r" )!r� r� r3 r% rP �ior �twisted.conch.test.keydatar r r r r �twisted.python.filepathr �twisted.python.reflectr �twisted.trial.unittestr r� r r r r r �twisted.conch.ssh.keysr r r r �skipr! r" r r r r �<module> s PK F��ZE8w�& & , __pycache__/test_manhole_tap.cpython-310.pycnu �[��� o �b� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddl mZ dd lmZ ed �Zed �ZerFerFdd lmZmZ G d d� de�ZdS )z) Tests for L{twisted.conch.manhole_tap}. � )�StreamServerEndpointService)� MultiService)�telnet)�error)�UsernamePassword)�usage)� requireModule)�TestCase� cryptography�pyasn1)� manhole_ssh� manhole_tapc @ sL e Zd ZdZes dZesdZdZdd� Zdd� Z d d � Z d d � Z d d� Z dS )�MakeServiceTestsz/ Tests for L{manhole_tap.makeService}. z can't run without cryptographyzCannot run without PyASN1)s iamusers thisispasswordc C sV | � � | _t| jd��}|�d�| j�� W d � n1 s w Y t�� | _dS )z8 Create a passwd-like file with a user. �wb� :N) �mktemp�filename�open�write�join�usernamePasswordr �Options�options)�self�f� r �E/usr/lib/python3/dist-packages/twisted/conch/test/test_manhole_tap.py�setUp% s �zMakeServiceTests.setUpc C sR | � tj��}t�� �g � W d � n1 sw Y | �|jjd d� dS )zk L{manhole_tap.makeService} requires either 'telnetPort' or 'sshPort' to be given. Nr z<At least one of --telnetPort and --sshPort must be specified) � assertRaisesr � UsageErrorr r � parseOptions� assertEqual� exception�args)r �er r r �test_requiresPort. s � �z"MakeServiceTests.test_requiresPortc C s~ | j �ddg� t�| j �}| �|t� | �t|j�d� | �|jd t � | �|jd j j tj � | �|jd j jd� dS )z� L{manhole_tap.makeService} will make a telnet service on the port defined by C{--telnetPort}. It will not make a SSH service. � --telnetPortztcp:222� r �� N)r r r � makeService�assertIsInstancer r! �len�servicesr �factory�protocol�makeTelnetProtocol�endpoint�_port�r �servicer r r �test_telnetPort; s �z MakeServiceTests.test_telnetPortc C s� | j �d| �� ddddg� t�| j �}| �|t� | �t|j �d� | �|j d t � | �|j d j t j � | �|j d jjd� d S ) z� L{manhole_tap.makeService} will make a SSH service on the port defined by C{--sshPort}. It will not make a telnet service. z --sshKeyDirz --sshKeySize�512z --sshPortztcp:223r' r �� N)r r r r r) r* r r! r+ r, r r- r � ConchFactoryr0 r1 r2 r r r � test_sshPortJ s �� z MakeServiceTests.test_sshPortc C s� | j �ddd| jg� t�| j �}|jd jjj}| � t |j � � �d� | � | �|�t| j� dtj��� | �| �|�tdd�dtj��jtj� dS ) zS The C{--passwd} command-line option will load a passwd-like file. r& ztcp:22z--passwdr � Ns wrongs user)r r r r r) r, r- r. �portalr! r+ �checkers�keys� assertTrue�successResultOf�loginr r r �ITelnetProtocolr* �failureResultOf�valuer �UnauthorizedLogin)r r3 r: r r r � test_passwdc s, ��� ���z MakeServiceTests.test_passwdN) �__name__� __module__� __qualname__�__doc__r �skipr r r r% r4 r8 rD r r r r r s r N)rH � twisted.application.internetr �twisted.application.servicer � twisted.conchr � twisted.credr �twisted.cred.credentialsr �twisted.pythonr �twisted.python.reflectr �twisted.trial.unittestr r r r r r r r r r �<module> s PK F��Zd�O^�6 �6 & __pycache__/test_agent.cpython-310.pycnu �[��� o �b�3 � @ sx d Z ddlZddlmZ ddlmZ zddlZW n ey# dZY nw eZzddl Z W n ey7 dZ Y nw e Z z ddl m Z mZ W n eyQ d ZZ Y nw ee ZZ ddlmZmZ ddlmZ G dd � d �ZG d d � d ej�ZG d d � d e�ZG dd� de�Ze dur�G dd� de j�ZG dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z dS ) z' Tests for L{twisted.conch.ssh.agent}. � N)�iosim)�unittest)�agent�keys)� ConchError�MissingKeyStoreError)�keydatac @ � e Zd ZdZdd� ZdS )� StubFactoryzb Mock factory that provides the keys attribute required by the SSHAgentServerProtocol c C s i | _ d S �N)r ��self� r �?/usr/lib/python3/dist-packages/twisted/conch/test/test_agent.py�__init__, s zStubFactory.__init__N)�__name__� __module__� __qualname__�__doc__r r r r r r & s r c @ s, e Zd ZdZedu sedu rdZdd� ZdS )� AgentTestBasez* Tests for SSHAgentServer/Client. Nz)Cannot run without cryptography or PyASN1c C sj t �tjtj�\| _| _| _t� | j_ t j � t j�| _t j � t j�| _t j � t j�| _t j � t j�| _d S r )r �connectedServerAndClientr �SSHAgentServer�SSHAgentClient�client�server�pumpr �factoryr �Key� fromStringr �privateRSA_openssh� rsaPrivate�privateDSA_openssh� dsaPrivate�publicRSA_openssh� rsaPublic�publicDSA_openssh� dsaPublicr r r r �setUp8 s � zAgentTestBase.setUp)r r r r r r �skipr' r r r r r 0 s r c @ r )�&ServerProtocolContractWithFactoryTestsz� The server protocol is stateful and so uses its factory to track state across requests. This test asserts that the protocol raises if its factory doesn't provide the necessary storage for that state. c C s2 t �ddtj�}| jjjd= | �t| jj |� d S )Nz!LB� r ) �struct�packr �AGENTC_REQUEST_IDENTITIESr r �__dict__� assertRaisesr � dataReceived)r �msgr r r �/test_factorySuppliesKeyStorageForServerProtocolQ s zVServerProtocolContractWithFactoryTests.test_factorySuppliesKeyStorageForServerProtocolN)r r r r r2 r r r r r) J s r) c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) �"UnimplementedVersionOneServerTestsa! Tests for methods with no-op implementations on the server. We need these for clients, such as openssh, that try v1 methods before going to v2. Because the client doesn't expose these operations with nice method names, we invoke sendRequest directly with an op code. c s0 � j �tjd�}� j�� � fdd�}|�|�S )zV assert that we get the correct op code for an RSA identities request � c s � � tjt| dd� �� d S )Nr r* )� assertEqualr �AGENT_RSA_IDENTITIES_ANSWER�ord)�packetr r r �_cbh s zRUnimplementedVersionOneServerTests.test_agentc_REQUEST_RSA_IDENTITIES.<locals>._cb)r � sendRequestr � AGENTC_REQUEST_RSA_IDENTITIESr �flush� addCallback)r �dr: r r r �"test_agentc_REQUEST_RSA_IDENTITIESa s zEUnimplementedVersionOneServerTests.test_agentc_REQUEST_RSA_IDENTITIESc C �( | j �tjd�}| j�� |�| jd�S )z[ assert that we get the correct op code for an RSA remove identity request r5 )r r; r �AGENTC_REMOVE_RSA_IDENTITYr r= r> r6 �r r? r r r � test_agentc_REMOVE_RSA_IDENTITYm s zBUnimplementedVersionOneServerTests.test_agentc_REMOVE_RSA_IDENTITYc C rA )zj assert that we get the correct op code for an RSA remove all identities request. r5 )r r; r � AGENTC_REMOVE_ALL_RSA_IDENTITIESr r= r> r6 rC r r r �%test_agentc_REMOVE_ALL_RSA_IDENTITIESu s zHUnimplementedVersionOneServerTests.test_agentc_REMOVE_ALL_RSA_IDENTITIESN)r r r r r@ rD rF r r r r r4 X s r4 c @ s e Zd ZdZdd� Zdd� ZdS )� CorruptServerz� A misbehaving server that returns bogus response op codes so that we can verify that our callbacks that deal with these op codes handle such miscreants. c C � | � dd� d S �N� r5 �� sendResponse�r �datar r r �agentc_REQUEST_IDENTITIES� � z'CorruptServer.agentc_REQUEST_IDENTITIESc C rH rI rK rM r r r �agentc_SIGN_REQUEST� rP z!CorruptServer.agentc_SIGN_REQUESTN)r r r r rO rQ r r r r rG � s rG c @ r3 ) �ClientWithBrokenServerTestszM verify error handling code in the client using a misbehaving server c C s2 t �| � t�ttj�\| _| _| _ t � | j_ d S r ) r r' r r rG r r r r r r r r r r r r' � s �z!ClientWithBrokenServerTests.setUpc C s* | j �| j�� d�}| j�� | �|t�S )z� Assert that L{SSHAgentClient.signData} raises a ConchError if we get a response from the server whose opcode doesn't match the protocol for data signing requests. � John Hancock)r �signDatar$ �blobr r= � assertFailurer rC r r r �"test_signDataCallbackErrorHandling� s z>ClientWithBrokenServerTests.test_signDataCallbackErrorHandlingc C s | j �� }| j�� | �|t�S )z� Assert that L{SSHAgentClient.requestIdentities} raises a ConchError if we get a response from the server whose opcode doesn't match the protocol for identity requests. )r �requestIdentitiesr r= rV r rC r r r �+test_requestIdentitiesCallbackErrorHandling� s zGClientWithBrokenServerTests.test_requestIdentitiesCallbackErrorHandlingN)r r r r r' rW rY r r r r rR � s rR c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �AgentKeyAdditionTestsz< Test adding different flavors of keys to an agent. c �2 � j �� j�� �}� j�� � fdd�}|�|�S )�@ L{SSHAgentClient.addIdentity} adds the private key it is called with to the SSH agent server to which it is connected, associating it with the comment it is called with. This test asserts that omitting the comment produces an empty string for the comment on the server. c �: � j jj� j�� }� �� j|d � � �d|d � d S �Nr r5 r* �r r r r rU r6 ��ignored� serverKeyr r r �_check� � zBAgentKeyAdditionTests.test_addRSAIdentityNoComment.<locals>._check�r � addIdentityr � privateBlobr r= r> �r r? rd r r r � test_addRSAIdentityNoComment� � z2AgentKeyAdditionTests.test_addRSAIdentityNoCommentc r\ )r] c r^ r_ �r r r r" rU r6 ra r r r rd � re zBAgentKeyAdditionTests.test_addDSAIdentityNoComment.<locals>._check�r rg r" rh r r= r> ri r r r � test_addDSAIdentityNoComment� rk z2AgentKeyAdditionTests.test_addDSAIdentityNoCommentc �6 � j j� j�� dd�}� j�� � fdd�}|�|�S )�1 L{SSHAgentClient.addIdentity} adds the private key it is called with to the SSH agent server to which it is connected, associating it with the comment it is called with. This test asserts that the server receives/stores the comment as sent by the client. � My special key��commentc r^ �Nr rq r* r` ra r r r rd � re zDAgentKeyAdditionTests.test_addRSAIdentityWithComment.<locals>._checkrf ri r r r � test_addRSAIdentityWithComment� � � z4AgentKeyAdditionTests.test_addRSAIdentityWithCommentc ro )rp rq rr c r^ rt rl ra r r r rd re zDAgentKeyAdditionTests.test_addDSAIdentityWithComment.<locals>._checkrm ri r r r � test_addDSAIdentityWithComment� rv z4AgentKeyAdditionTests.test_addDSAIdentityWithCommentN)r r r r rj rn ru rw r r r r r[ � s r[ c @ s e Zd Zdd� ZdS )�AgentClientFailureTestsc C s$ | j �dd�}| j�� | �|t�S )zK verify that the client raises ConchError on AGENT_FAILURE rJ r5 )r r; r r= rV r rC r r r �test_agentFailure s z)AgentClientFailureTests.test_agentFailureN)r r r ry r r r r rx s rx c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �AgentIdentityRequestsTestszJ Test operations against a server with identities already loaded. c C �B t �| � | jdf| jjj| j�� < | jdf| jjj| j�� < d S �N� a comment� another comment�r r' r" r r r rU r r r r r r' � ��z AgentIdentityRequestsTests.setUpc C sX | j �| j�� d�}| j�� | �|�}| j�d�}| � ||� | � | j� |d�� dS )zc Sign data with an RSA private key and then verify it with the public key. rS N) r rT r$ rU r r= �successResultOfr �signr6 � assertTrue�verify)r r? � signature�expectedr r r �test_signDataRSA" s z+AgentIdentityRequestsTests.test_signDataRSAc s4 � j �� j�� d�}� j�� � fdd�}|�|�S )zb Sign data with a DSA private key and then verify it with the public key. rS c s � � � j�| d�� d S )NrS )r� r& r� )�sigr r r rd 7 s z;AgentIdentityRequestsTests.test_signDataDSA.<locals>._check)r rT r& rU r r= r> ri r r r �test_signDataDSA/ s z+AgentIdentityRequestsTests.test_signDataDSAc C s< | j jj| j�� = | j�| j�� d�}| j�� | � |t �S )zm Assert that we get an errback if we try to sign data using a key that wasn't added. rS ) r r r r$ rU r rT r r= rV r rC r r r �$test_signDataRSAErrbackOnUnknownBlob? s z?AgentIdentityRequestsTests.test_signDataRSAErrbackOnUnknownBlobc �* � j �� }� j�� � fdd�}|�|�S )z} Assert that we get all of the keys/comments that we add when we issue a request for all identities. c s^ i }d|� j �� < d|� j�� < i }| D ]}|d |tjj|d dd��� < q� �||� d S )Nr} r~ r* r rU )�type)r& rU r$ r r r r6 )�keytr� �received�kr r r rd Q s "zAAgentIdentityRequestsTests.test_requestIdentities.<locals>._check)r rX r r= r> ri r r r �test_requestIdentitiesI s z1AgentIdentityRequestsTests.test_requestIdentitiesN) r r r r r' r� r� r� r� r r r r rz s rz c @ rZ ) �AgentKeyRemovalTestsz< Test support for removing keys in a remote server. c C r{ r| r r r r r r' c r� zAgentKeyRemovalTests.setUpc r\ )z< Assert that we can remove an RSA identity. c sJ � � dt� jjj�� � �� j�� � jjj� � �� j �� � jjj� d S �Nr* ) r6 �lenr r r �assertInr" rU � assertNotInr �rb r r r rd v s z;AgentKeyRemovalTests.test_removeRSAIdentity.<locals>._check)r �removeIdentityr rU r r= r> ri r r r �test_removeRSAIdentityn s z+AgentKeyRemovalTests.test_removeRSAIdentityc r\ )z; Assert that we can remove a DSA identity. c s2 � � dt� jjj�� � �� j�� � jjj� d S r� )r6 r� r r r r� r rU r� r r r rd � s z;AgentKeyRemovalTests.test_removeDSAIdentity.<locals>._check)r r� r" rU r r= r> ri r r r �test_removeDSAIdentity} s z+AgentKeyRemovalTests.test_removeDSAIdentityc r� )z; Assert that we can remove all identities. c s � � dt� jjj�� d S )Nr )r6 r� r r r r� r r r rd � s z=AgentKeyRemovalTests.test_removeAllIdentities.<locals>._check)r �removeAllIdentitiesr r= r> ri r r r �test_removeAllIdentities� s z-AgentKeyRemovalTests.test_removeAllIdentitiesN)r r r r r' r� r� r� r r r r r� ^ s r� )!r r+ � twisted.testr � twisted.trialr � cryptography� _cryptography� ImportError�pyasn1�_pyasn1�twisted.conch.sshr �_agentr �_keys�twisted.conch.errorr r �twisted.conch.testr r �TestCaser r) r4 r rG rR r[ rx rz r� r r r r �<module> sF � � � '#V LPK F��Z�?�� � & __pycache__/test_mixin.cpython-310.pycnu �[��� o �b � @ sL d dl mZ d dlmZ d dlmZ G dd� dej�ZG dd� dej�Z dS ) � )�mixin)�StringTransport)�unittestc @ s$ e Zd ZdZdZdd� Zdd� ZdS )�TestBufferingProtoFr c C s d| _ t� S )NT)� scheduled�object)�self� r �?/usr/lib/python3/dist-packages/twisted/conch/test/test_mixin.py�schedule s zTestBufferingProto.schedulec C s | j d7 _ d S )N� )� rescheduled)r �tokenr r r � reschedule s z TestBufferingProto.rescheduleN)�__name__� __module__� __qualname__r r r r r r r r r s r c @ s e Zd Zdd� ZdS )�BufferingTestsc C s� t � }t� }|_| �|j� g d�}|�d� | �|j� | �|j� |D ]}|j}|�|� | �|j|d � | �|� � d� q&|� � | �|� � dd� |� � d S )N)� foos bars bazs quuxr r � ) r r � transport� assertFalser �write� assertTruer � assertEqual�value�flush�join)r �p�t�L�s�nr r r � testBuffering s z BufferingTests.testBufferingN)r r r r# r r r r r s r N) � twisted.conchr �twisted.test.proto_helpersr � twisted.trialr �BufferingMixinr �TestCaser r r r r �<module> s PK F��Z"�� ̹ ̹ * __pycache__/test_endpoints.cpython-310.pycnu �[��� o �bP� � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddl m Z m Z ddl Z ddl mZmZmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZmZm Z m Z dd l m Z m Z m!Z!m"Z" ddl#m$Z$m%Z% ddl&m'Z'm(Z( ddl)m*Z*m+Z+ ddl,m-Z- ddl.m/Z/ ddl0m1Z1 ddl2m3Z3 ddl4m5Z5 ddl6m7Z7m8Z8 ddl9m:Z: e5d��re5d��rddl;m<Z< ddl=m>Z>m?Z? dd l@mAZAmBZB dd lCmDZDmEZEmFZFmGZGmHZHmIZImJZJ dd lKmLZL dd lMmNZN dd lOmPZP dd!lQmRZR dd"lSmTZT dd#lUmVZV dd$lWmXZX dd%lYmZZZ dd&l[m\Z\m]Z]m^Z^m_Z_ nd'Z`eaZTeaZZeaZReaZVeaZPeaZNeaZBeaZ?eaZ<dd(lbmcZcmdZd dd)l6meZe G d*d+� d+ec�ZfG d,d-� d-eP�ZgG d.d/� d/eP�ZhG d0d1� d1eP�ZiG d2d3� d3�ZjG d4d5� d5e'�ZkG d6d7� d7�ZlG d8d9� d9eZ�ZmG d:d;� d;eT�Znee$�G d<d=� d=��Zoee%�G d>d?� d?��ZpG d@dA� dA�ZqG dBdC� dCe:eq�ZrG dDdE� dEe:eq�ZsG dFdG� dGe:�ZtG dHdI� dI�ZuG dJdK� dKe:�ZvdS )Lz' Tests for L{twisted.conch.endpoints}. � N)�ENOSYS)�pack)� implementer)� verifyClass� verifyObject)� ConchError�HostKeyChanged�UserRejectedKey)� IConchUser)�'InMemoryUsernamePasswordDatabaseDontUse)�Portal)� IPv4Address)�CancelledError�Deferred�fail�succeed)�ConnectingCancelledError�ConnectionDone�ConnectionRefusedError�ProcessTerminated)�IAddress�IStreamClientEndpoint)�Factory�Protocol)�LogLevel�globalLogPublisher)� networkString)�Failure)�FilePath)�msg)� requireModule)�EventLoggingObserver�MemoryReactorClock)�TestCase� cryptographyz pyasn1.type)� ConchUser)�InMemorySSHKeyDB�SSHPublicKeyChecker)� ConsoleUI�KnownHostsFile)�AuthenticationFailed�SSHCommandAddress�SSHCommandClientEndpoint�_ExistingConnectionHelper�_ISSHConnectionCreator�_NewConnectionHelper� _ReadFile)�common)�SSHAgentServer)� SSHChannel)� SSHConnection)� SSHFactory)�Key)�SSHClientTransport)�SSHUserAuthServer)�privateDSA_openssh�privateRSA_openssh� privateRSA_openssh_encrypted_aes�publicRSA_opensshz%can't run w/o cryptography and pyasn1)� FakeTransport�connect)�StringTransportc @ s e Zd ZdZdZdd� ZdS )�AbortableFakeTransportzC A L{FakeTransport} with added C{abortConnection} support. Fc C � d| _ dS )z} Abort the connection in a fake manner. This should really be implemented in the underlying module. TN��aborted��self� rF �C/usr/lib/python3/dist-packages/twisted/conch/test/test_endpoints.py�abortConnectionZ � z&AbortableFakeTransport.abortConnectionN)�__name__� __module__� __qualname__�__doc__rC rH rF rF rF rG r@ S s r@ c @ � e Zd ZdZdd� ZdS )�BrokenExecSessionzO L{BrokenExecSession} is a session on which exec requests always fail. c C � dS )z� Fail all exec requests. @param data: Information about what is being executed. @type data: L{bytes} @return: C{0} to indicate failure @rtype: L{int} r rF �rE �datarF rF rG � request_exech � z BrokenExecSession.request_execN�rJ rK rL rM rS rF rF rF rG rO c � rO c @ rN )�WorkingExecSessionzS L{WorkingExecSession} is a session on which exec requests always succeed. c C rP )z� Succeed all exec requests. @param data: Information about what is being executed. @type data: L{bytes} @return: C{1} to indicate success @rtype: L{int} � rF rQ rF rF rG rS z rT z WorkingExecSession.request_execNrU rF rF rF rG rW u rV rW c @ rN )�UnsatisfiedExecSessionz� L{UnsatisfiedExecSession} is a session on which exec requests are always delayed indefinitely, never succeeding or failing. c C s t � S )z� Delay all exec requests indefinitely. @param data: Information about what is being executed. @type data: L{bytes} @return: A L{Deferred} which will never fire. @rtype: L{Deferred} )r rQ rF rF rG rS � s z#UnsatisfiedExecSession.request_execNrU rF rF rF rG rY � s rY c @ s e Zd Zdd� Zdd� ZdS )� TrivialRealmc C s i | _ d S �N)� channelLookuprD rF rF rG �__init__� � zTrivialRealm.__init__c G s t � }| j|_t|dd� fS )Nc S � d S r[ rF rF rF rF rG �<lambda>� � z,TrivialRealm.requestAvatar.<locals>.<lambda>)r% r\ r )rE �avatarId�mind� interfaces�avatarrF rF rG � requestAvatar� s zTrivialRealm.requestAvatarN)rJ rK rL r] rf rF rF rF rG rZ � s rZ c @ s e Zd ZdZdd� ZdS )�AddressSpyFactoryNc C s || _ t�| |�S r[ )�addressr � buildProtocol)rE rh rF rF rG ri � s z AddressSpyFactory.buildProtocol)rJ rK rL rh ri rF rF rF rG rg � � rg c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�FixedResponseUIc C s || _ d S r[ ��result)rE rm rF rF rG r] � r^ zFixedResponseUI.__init__c C s t | j�S r[ )r rm �rE �textrF rF rG �prompt� r^ zFixedResponseUI.promptc C r_ r[ rF rn rF rF rG �warn� s zFixedResponseUI.warnN)rJ rK rL r] rp rq rF rF rF rG rk � s rk c @ s$ e Zd Zedd� �Zedd� �ZdS )�FakeClockSSHUserAuthServerc C � | j jjS )zy Use the C{attemptsBeforeDisconnect} value defined by the factory to make it easier to override. )� transport�factory�attemptsBeforeDisconnectrD rF rF rG rv � rI z3FakeClockSSHUserAuthServer.attemptsBeforeDisconnectc C rs )z� Use the reactor defined by the factory, rather than the default global reactor, to simplify testing (by allowing an alternate implementation to be supplied by tests). )rt ru �reactorrD rF rF rG �clock� s z FakeClockSSHUserAuthServer.clockN)rJ rK rL �propertyrv rx rF rF rF rG rr � s rr c @ s2 e Zd Zedd� �Zedd� �Zeed�ZdZ dS )�CommandFactoryc C � dt jtd�iS �N� ssh-rsa)rR )r6 � fromStringr< rD rF rF rG � publicKeys� � zCommandFactory.publicKeysc C r{ r| )r6 r~ r: rD rF rF rG � privateKeys� r� zCommandFactory.privateKeys)s ssh-userauths ssh-connectionr N) rJ rK rL ry r r� rr r4 �servicesrv rF rF rF rG rz � s � rz c @ s e Zd ZdS )� MemoryAddressN)rJ rK rL rF rF rF rG r� � s r� c @ � e Zd ZdZdd� Zdd� ZdS )�SingleUseMemoryEndpointa] L{SingleUseMemoryEndpoint} is a client endpoint which allows one connection to be set up and then exposes an API for moving around bytes related to that connection. @ivar pump: L{None} until a connection is attempted, then a L{IOPump} instance associated with the protocol which is connected. @type pump: L{IOPump} c C s d| _ || _dS )z� @param server: An L{IProtocol} provider to which the client will be connected. @type server: L{IProtocol} provider N)�pump�_server)rE �serverrF rF rG r] � s z SingleUseMemoryEndpoint.__init__c C sh | j d ur td��z|�t� �}W n ty t� Y S w t| jt| jdd�|t|dd��| _ t |�S )Nz(SingleUseMemoryEndpoint was already usedT��isServerF) r� � Exceptionri r� � BaseExceptionr r> r� r@ r )rE ru �protocolrF rF rG r> � s � �z SingleUseMemoryEndpoint.connectN)rJ rK rL rM r] r> rF rF rF rG r� � s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd*d#d$�Zd%d&� Zd'd(� Zd)S )+�"SSHCommandClientEndpointTestsMixina� Tests for L{SSHCommandClientEndpoint}, an L{IStreamClientEndpoint} implementations which connects a protocol with the stdin and stdout of a command running in an SSH session. These tests apply to L{SSHCommandClientEndpoint} whether it is constructed using L{SSHCommandClientEndpoint.existingConnection} or L{SSHCommandClientEndpoint.newConnection}. Subclasses must override L{create}, L{assertClientTransportState}, and L{finishConnection}. c C s� d| _ d| _d| _d| _t� | _t� | _t| j�| _ t � | _ | j � | j| j� | j � | j � t� | _| j| j_| j | j_ | j�� | �| jj� tddd�| _tddd �| _d S ) Ns ssh.example.comi&� s users password�TCPz10.0.0.1i90 z192.168.100.200i1� )�hostname�port�user�passwordr" rw rZ �realmr �portalr �passwdDB�addUser�registerCheckerrz ru �doStart� addCleanup�doStopr � clientAddress� serverAddressrD rF rF rG �setUp s" z(SSHCommandClientEndpointTestsMixin.setUpc C � t | jj�d���)z� Create and return a new L{SSHCommandClientEndpoint} to be tested. Override this to implement creation in an interesting way the endpoint. z did not implement create��NotImplementedError� __class__rJ rD rF rF rG �create/ � �z)SSHCommandClientEndpointTestsMixin.createc C s t d| jjf ��)a� Make an assertion about the connectedness of the given protocol's transport. Override this to implement either a check for the connection still being open or having been closed as appropriate. @param client: The client whose state is being checked. @param immediateClose: Boolean indicating whether the connection was closed immediately or not. z/%r did not implement assertClientTransportStater� �rE �client�immediateCloserF rF rG �assertClientTransportState8 s ��z=SSHCommandClientEndpointTestsMixin.assertClientTransportStatec C r� )z� Do any remaining work necessary to complete an in-memory connection attempted initiated using C{self.reactor}. z# did not implement finishConnectionr� rD rF rF rG �finishConnectionH r� z3SSHCommandClientEndpointTestsMixin.finishConnectionc C sT |� d�}|� d�}t|d| j| jd�}t|d| j| jd�}t||||�}|||fS )aw Set up an in-memory connection between protocols created by C{serverFactory} and C{clientFactory}. @return: A three-tuple. The first element is the protocol created by C{serverFactory}. The second element is the protocol created by C{clientFactory}. The third element is the L{IOPump} connecting them. NF)r� � hostAddress� peerAddressT)ri r@ r� r� r> )rE � serverFactory� clientFactory�clientProtocol�serverProtocol�clientTransport�serverTransportr� rF rF rG �connectedServerAndClientQ s �� z;SSHCommandClientEndpointTestsMixin.connectedServerAndClientc C s� | � � }t� }t|_|�|�}| �� \}}}| �t�}| �d|d j j |d j j f� | � dt |�� | � |�}|�t� | � d|j j � | �|d� dS )z� If a channel cannot be opened on the authenticated SSH connection, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping the reason given by the server. �unknown channelr rX � unknown channelFN)r� r r r� r> r� �flushLoggedErrorsr �assertIn�valuerR � assertEqual�len�failureResultOf�trapr� ) rE �endpointru � connectedr� r� r� �errors�frF rF rG �test_channelOpenFailuren s z:SSHCommandClientEndpointTestsMixin.test_channelOpenFailurec C sl t | jjd< | �� }t� }t|_|�|�}| �� \}}}| � |�}|� t � | � d|j j � | �|d� dS )z� If execution of the command fails, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping the reason given by the server. � sessionzchannel request failedFN)rO r� r\ r� r r r� r> r� r� r� r r� r� r� �rE r� ru r� r� r� r� r� rF rF rG �test_execFailure� s z3SSHCommandClientEndpointTestsMixin.test_execFailurec C sd t | jjd< | �� }t� }t|_|�|�}| �� \}}}|� � | � |�}|� t � | � |d� dS )z� If execution of the command is cancelled via the L{Deferred} returned by L{SSHCommandClientEndpoint.connect}, the connection is closed immediately. r� TN)rY r� r\ r� r r r� r> r� �cancelr� r� r r� r� rF rF rG �test_execCancelled� s z5SSHCommandClientEndpointTestsMixin.test_execCancelledc C s� t | jjd< | �� }t� }t|_|�|� | �� \}}}| � |j t � | � |j �� |j j� | � | j|j j� | � d|j j� dS )aD Once the necessary SSH actions have completed successfully, L{SSHCommandClientEndpoint.connect} uses the factory passed to it to construct a protocol instance by calling its C{buildProtocol} method with an address object representing the SSH connection and command executed. r� � /bin/ls -lN)rW r� r\ r� rg r r� r> r� �assertIsInstancerh r+ r� rt �getHostr� r� �username�command)rE r� ru r� r� r� rF rF rG �test_buildProtocol� s z5SSHCommandClientEndpointTestsMixin.test_buildProtocolc C sR t | jjd< | �� }t� }t|_|�|�}| �� \}}}| � |�}| � |j � dS )a L{SSHCommandClientEndpoint} establishes an SSH connection, creates a channel in it, runs a command in that channel, and uses the protocol's C{makeConnection} to associate it with a protocol representing that command's stdin and stdout. r� N) rW r� r\ r� r r r� r> r� �successResultOf�assertIsNotNonert �rE r� ru r� r� r� r� r� rF rF rG �test_makeConnection� s z6SSHCommandClientEndpointTestsMixin.test_makeConnectionc C s� t | jjd< | �� }t� }t|_|�|�}| �� \}}}| � |�}g }|j |_ |j j } |jj| �d� |�� | �dd�|�� dS )z� After establishing the connection, when the command on the SSH server produces output, it is delivered to the protocol's C{dataReceived} method. r� � hello, world� N)rW r� r\ r� r r r� r> r� r� �append� dataReceivedrt �id�service�channels�writer� r� �join) rE r� ru r� r� r� r� r� r� � channelIdrF rF rG �test_dataReceived� s z4SSHCommandClientEndpointTestsMixin.test_dataReceivedc C s� t | jjd< | �� }t� }t|_|�|�}| �� \}}}| � |�}g }|j |_ |j j } |jj| �� |�� |d �t� | �|d� dS )zq When the command closes the channel, the protocol's C{connectionLost} method is called. r� r FN)rW r� r\ r� r r r� r> r� r� r� �connectionLostrt r� r� r� �loseConnectionr� r� r r� ) rE r� ru r� r� r� r� r� r� r� rF rF rG �test_connectionLost� s z6SSHCommandClientEndpointTestsMixin.test_connectionLostc C s� t | jjd< | �� }t� }t|_|�|�}| �� \}}}| � |�} g } | j | _ | j j } |jj| } |j�| ||� | �� |�� | �|d� | d S )zJ Test handling of non-zero exit statuses or exit signals. r� Fr )rW r� r\ r� r r r� r> r� r� r� r� rt r� r� r� � sendRequestr� r� r� ) rE �request� requestArgr� ru r� r� r� r� r� r� r� �channelrF rF rG �_exitStatusTest s z2SSHCommandClientEndpointTestsMixin._exitStatusTestc C s$ d}| � dtd|��}|�t� dS )�� When the command exits with a non-zero status, the protocol's C{connectionLost} method is called with a L{Failure} wrapping an exception which encapsulates that status. r � exit-status�>LN)r� r r� r )rE �exitCode�excrF rF rG �test_zeroExitCode0 s z4SSHCommandClientEndpointTestsMixin.test_zeroExitCodec C sH d}d}| � dtd|��}|�t� | �||jj� | �||jj� dS )r� �{ Nr� r� )r� r r� r r� r� r� �signal)rE r� r� r� rF rF rG �test_nonZeroExitStatus: s z9SSHCommandClientEndpointTestsMixin.test_nonZeroExitStatusc C s� t � }t�|� | �tj|� d}d}d�t�d�dt�d�t�d�g�}| �d|�}|� t � | � ||j j � | � ||j j� d }t�|t�t�t�tj�|dd d dd ���� dS ) a When the command exits with a non-zero signal, the protocol's C{connectionLost} method is called with a L{Failure} wrapping an exception which encapsulates that status. Additional packet contents are logged at the C{info} level. N� r� s TERM� s messages en-USs exit-signalz'twisted.conch.endpoints._CommandChannelT�message)� log_level� log_namespace�shortSignalName� coreDumped� errorMessage� languageTag)r! r � addObserverr� �removeObserverr� r1 �NSr� r� r r� r� r� r� �hamcrest� assert_that�has_item� has_entries�equal_tor �info)rE � logObserverr� r� �packetr� � logNamespacerF rF rG �test_nonZeroExitSignalG s> �� ����z9SSHCommandClientEndpointTestsMixin.test_nonZeroExitSignalFc s< |j j}g � |r� fdd�}n� j}t|jj| ||� � S )a Hook into and record events which happen to C{protocol}. @param server: The SSH server protocol over which C{protocol} is running. @type server: L{IProtocol} provider @param protocol: @param event: @param noArgs: c s � � d �S r[ )r� rF ��recorderrF rG r` � s z;SSHCommandClientEndpointTestsMixin.record.<locals>.<lambda>)rt r� r� �setattrr� r� )rE r� r� �event�noArgsr� r� rF r rG �recordu s z)SSHCommandClientEndpointTestsMixin.recordc C sz t | jjd< | �� }t� }t|_|�|�}| �� \}}}| � |�}| � ||d�}|j � d� |� � | �dd�|�� dS )z� The transport connected to the protocol has a C{write} method which sends bytes to the input of the command executing on the SSH server. r� r� r� r� N)rW r� r\ r� r r r� r> r� r� r rt r� r� r� r� � rE r� ru r� r� r� r� r� r� rF rF rG � test_write� s z-SSHCommandClientEndpointTestsMixin.test_writec C s| t | jjd< | �� }t� }t|_|�|�}| �� \}}}| � |�}| � ||d�}|j � dg� |� � | �dd�|�� dS )z� The transport connected to the protocol has a C{writeSequence} method which sends bytes to the input of the command executing on the SSH server. r� r� r� r� N)rW r� r\ r� r r r� r> r� r� r rt � writeSequencer� r� r� r rF rF rG �test_writeSequence� s z5SSHCommandClientEndpointTestsMixin.test_writeSequenceN)F)rJ rK rL rM r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rF rF rF rG r� s* . r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0S )1�NewConnectionTestsz` Tests for L{SSHCommandClientEndpoint} when using the C{newConnection} constructor. c C sh t �| � t| �� �| _t| j�| _| j�| j| j j d � | j�t | j j �| j j d � | j�� dS )� Configure an SSH server with password authentication enabled for a well-known (to the tests) account. r} N)r� r� r �mktemp� hostKeyPathr) � knownHosts� addHostKeyr� ru r r r� �host�saverD rF rF rG r� � s �zNewConnectionTests.setUpc C s* t j| jd| j| j| j| j| jtd�d�S )zu Create and return a new L{SSHCommandClientEndpoint} using the C{newConnection} constructor. r� F�r� r �ui) r, � newConnectionrw r� r� r� r� r rk rD rF rF rG r� � s �zNewConnectionTests.createc C s | � | j| jjd d �S )z} Establish the first attempted TCP connection using the SSH server which C{self.factory} can create. r � )r� ru rw � tcpClientsrD rF rF rG r� � s �z#NewConnectionTests.finishConnectionc C sH | j ||ddd�}|j�� |�� | �dg|� |�� |j�� dS )a� Lose the connection to a server and pump the L{IOPump} sufficiently for the client to handle the lost connection. Asserts that the client disconnects its transport. @param server: The SSH server protocol over which C{protocol} is running. @type server: L{IProtocol} provider @param client: The SSH client protocol over which C{protocol} is running. @type client: L{IProtocol} provider @param protocol: The protocol created by calling connect on the ssh endpoint under test. @type protocol: L{IProtocol} provider @param pump: The L{IOPump} connecting client to server. @type pump: L{IOPump} �closedT)r N)r rt r� r� r� �reportDisconnect)rE r� r� r� r� r rF rF rG �loseConnectionToServer� s z)NewConnectionTests.loseConnectionToServerc C s( |r | � |jj� dS | � |jj� dS )z� Assert that the transport for the given protocol has been disconnected. L{SSHCommandClientEndpoint.newConnection} creates a new dedicated SSH connection and cleans it up after the command exits. N)� assertTruert rC � disconnectingr� rF rF rG r� s z-NewConnectionTests.assertClientTransportStatec C s, t �| jdd| j| j�}| �tt|�� dS )zY L{SSHCommandClientEndpoint} instances provide L{IStreamClientEndpoint}. � dummy command� dummy userN)r, r rw r� r� r r r �rE r� rF rF rG �test_interface s �z!NewConnectionTests.test_interfacec C s( t �| jdd| j�}| �d|jj� dS )z� L{SSHCommandClientEndpoint} uses the default port number for SSH when the C{port} argument is not specified. r r � N�r, r rw r� r� �_creatorr� r rF rF rG �test_defaultPort" s �z#NewConnectionTests.test_defaultPortc C s, t j| jdd| jdd�}| �d|jj� dS )zU L{SSHCommandClientEndpoint} uses the C{port} argument if specified. r r i� )r� Nr# r rF rF rG �test_specifiedPort, s �z%NewConnectionTests.test_specifiedPortc C s� t j| jd| j| j| j| j| jtd�d�}t � }t |_ |� |� | jj d \}}}}}| �| jt|�� | �| j|� | �dt| jj �� dS )z� L{SSHCommandClientEndpoint} uses the L{IReactorTCP} passed to it to attempt a connection to the host/port address also passed to it. r� Fr r rX N)r, r rw r� r� r� r� r rk r r r� r> r r� r r� )rE r� ru r r� �timeout� bindAddressrF rF rG �test_destination5 s" � z#NewConnectionTests.test_destinationc C sp t j| jdd| j| j| jtd�d�}t� }t|_ |� |�}| jj d d }|� dt t� �� | �|��t� dS )z� If a connection cannot be established, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} representing the reason for the connection setup failure. r� r F�r r r r N)r, r rw r� r� r rk r r r� r> r �clientConnectionFailedr r r� r� �rE r� ru �drF rF rG �test_connectionFailedM s � z(NewConnectionTests.test_connectionFailedc C sx t j| jdd| j| jt| �� �td�d�}t� }t |_ |� |�}| � | j | jjd d �\}}}| �|�}|�t� dS )a" If the L{KnownHostsFile} instance used to construct L{SSHCommandClientEndpoint} rejects the SSH public key presented by the server, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{UserRejectedKey}. r� r Fr* r r N)r, r rw r� r� r) r rk r r r� r> r� ru r r� r� r r� rF rF rG �test_userRejectedHostKeye s" � � z+NewConnectionTests.test_userRejectedHostKeyc C s� t �t��� }tt| �� ��}|�t| j j �|� t jt dd��� }|�| j |� t d�}tj| jdd| j | jd||d�}t� }t|_|�|�}| �| j| jjd d �\}} } | �|�} | �t� d S ) ac If the SSH public key presented by the SSH server does not match the previously remembered key, as reported by the L{KnownHostsFile} instance use to construct the endpoint, for that server, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{HostKeyChanged}. s testxp)� passphraseTr� r � dummy passwordr r r N) r6 r~ r: �publicr) r r r r r� r r; r� rk r, r rw r� r r r� r> r� ru r r� r� r ) rE �firstKeyr � differentKeyr r� ru r� r� r� r� r� rF rF rG �test_mismatchedHostKey� s8 ��� � z)NewConnectionTests.test_mismatchedHostKeyc C s� t j| jdd| j| j| jtd�d�}t� }t|_ |� |�}t � }| jj d d }|� d�}|�|� |�tt� �� | �|��t� dS )aU If the connection closes at any point before the SSH transport layer has finished key exchange (ie, gotten to the point where we may attempt to authenticate), the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping the reason for the lost connection. r� r Fr* r r N)r, r rw r� r� r rk r r r� r> r? r ri �makeConnectionr� r r r� r� �rE r� ru r- rt r� rF rF rG �!test_connectionClosedBeforeSecure� s$ � z4NewConnectionTests.test_connectionClosedBeforeSecurec C s� t j| jdd| j| j| jtd�d�}t� }t|_ |� |�}t ddd�}| jj d d }|� d�}|�|� |�� | �|��t� | �|j� |�tt� �� dS ) a[ If the connection is cancelled before the SSH transport layer has finished key exchange (ie, gotten to the point where we may attempt to authenticate), the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{CancelledError} and the connection is aborted. r� r Fr* Nr� r r )r, r rw r� r� r rk r r r� r> r@ r ri r6 r� r� r� r r rC r� r r r7 rF rF rG �$test_connectionCancelledBeforeSecure� s( � z7NewConnectionTests.test_connectionCancelledBeforeSecurec C sj t j| jdd| j| j| jtd�d�}t� }t|_ |� |�}|� � | � |�� t� | �| jjd j� dS )zz If the connection is cancelled before it finishes connecting, the connection attempt is stopped. r� r Fr* r N)r, r rw r� r� r rk r r r� r> r� r� r� r r � connectors�stoppedConnectingr, rF rF rG �'test_connectionCancelledBeforeConnected� s � z:NewConnectionTests.test_connectionCancelledBeforeConnectedc C s� t j| jdd| j| jd| jtd�d�}t� }t|_ |� |�}| � | j | jj d d �\}}}| j�|jj� |�� | �|�}|�t� | �|d� dS ) z� If the SSH server rejects the password presented during authentication, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{AuthenticationFailed}. r� r r1 Fr r r N)r, r rw r� r� r rk r r r� r> r� ru r �advancer� � passwordDelay�flushr� r� r* r� r� rF rF rG �"test_passwordAuthenticationFailure s* � � z5NewConnectionTests.test_passwordAuthenticationFailurec C s, dd� |� � D �}tt|��}|�|� dS )a� Create an L{ISSHPrivateKey} checker which recognizes C{users} and add it to C{portal}. @param portal: A L{Portal} to which to add the checker. @type portal: L{Portal} @param users: The users and their keys the checker will recognize. Keys are byte strings giving user names. Values are byte strings giving OpenSSH-formatted private keys. @type users: L{dict} c S s" i | ] \}}|t �|��� g�qS rF )r6 r~ r2 )�.0�k�vrF rF rG � <dictcomp>8 s " z6NewConnectionTests.setupKeyChecker.<locals>.<dictcomp>N)�itemsr' r& r� )rE r� �users�mapping�checkerrF rF rG �setupKeyChecker+ s z"NewConnectionTests.setupKeyCheckerc C s� t �t�}| �| j| jti� tj| j d| j| j | j |g| j t d�d�}t� }t|_|�|�}| �| j| j jd d �\}}}| �|�}|�t� | �|jj� dS )z� If the SSH server rejects the key pair presented during authentication, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{AuthenticationFailed}. r� F��keysr r r r N)r6 r~ r: rI r� r� r9 r, r rw r� r� r rk r r r� r> r� ru r r� r� r* r rt r ) rE �badKeyr� ru r� r� r� r� r� rF rF rG �#test_publicKeyAuthenticationFailure< s* � � z6NewConnectionTests.test_publicKeyAuthenticationFailurec C s t �t�}| �| j| jti� tj| j d| j| j | j |g| j | j td�d� }t� }t|_|�|�}| j jd7 _| �| j| j jd d �\}}}|�� | �t�}| �d|d jj |d jjf� | � dt |�� | � |�} | � t� | � d| jj� | �!|j"j#� d S ) z{ If the SSH server does not accept any of the specified SSH keys, the specified password is tried. r� F)rK r� r r rX r r r� r� N)$r6 r~ r: rI r� r� r9 r, r rw r� r� r� r rk r r r� r> ru rv r� r r� r� r r� r� rR r� r� r� r� r rt r ) rE rL r� ru r� r� r� r� r� r� rF rF rG �test_authenticationFallbacka s8 � � z.NewConnectionTests.test_authenticationFallbackc C s� t �t�}| �| j| jti� t| jjd< t j | j d| j| j | j |g| jtd�d�}t� }t|_|�|�}| �| j| j jd d �\}}}| �|�}| �|j� dS )z� If L{SSHCommandClientEndpoint} is initialized with any private keys, it will try to use them to authenticate with the SSH server. r� r� FrJ r r N)r6 r~ r9 rI r� r� rW r� r\ r, r rw r� r� r rk r r r� r> r� ru r r� r� rt ) rE �keyr� ru r� r� r� r� r� rF rF rG � test_publicKeyAuthentication� s* � � z/NewConnectionTests.test_publicKeyAuthenticationc C s� t j| jd| j| j| j| jtd�d�}t� }t |_ |� |�}| � | j | jjd d �\}}}|�� | �|�}|�t� | �|jj� dS )z� If the password is not specified, L{SSHCommandClientEndpoint} doesn't try it as an authentication mechanism. r� Fr* r r N)r, r rw r� r� r� r rk r r r� r> r� ru r r� r� r� r* r rt r r� rF rF rG � test_skipPasswordAuthentication� s& � � z2NewConnectionTests.test_skipPasswordAuthenticationc C s t �t�}t� }t� |_|�� |dfi|j_| �| j | j ti� t |�}t j | jd| j | j| j| jtd�|d�}t| jjd< t� }t|_|�|�}| �| j| jjd d �\}}} td�D ] } |j � � | � � q]| � |�} | � | j � | � ||| | � | �!|j j"� | �!|j j#j"� d S ) a If L{SSHCommandClientEndpoint} is initialized with an L{SSHAgentClient}, the agent is used to authenticate with the SSH server. Once the connection with the SSH server has concluded, the connection to the agent is disconnected. r� r� F)r r � agentEndpointr� r r � N)$r6 r~ r: r2 r ru �blobrK rI r� r� r� r, r rw r� r� r rk rW r� r\ r r� r> r� r �ranger� r� r� rt r r r �clientIO) rE rO � agentServerrR r� ru r� r� r� r� �ir� rF rF rG �test_agentAuthentication� s> � � z+NewConnectionTests.test_agentAuthenticationc C sd t | jjd< | �� }t� }t|_|�|�}| �� \}}}| � |�}| � ||||� | � |j j � dS )z� The transport connected to the protocol has a C{loseConnection} method which causes the channel in which the command is running to close and the overall connection to be closed. r� N)rW r� r\ r� r r r� r> r� r� r r rt r r� rF rF rG �test_loseConnection s z&NewConnectionTests.test_loseConnectionN)rJ rK rL rM r� r� r� r r� r! r% r& r) r. r/ r5 r8 r9 r< r@ rI rM rN rP rQ rY rZ rF rF rF rG r � s2 # , "(%0 ! 3r c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �ExistingConnectionTestsze Tests for L{SSHCommandClientEndpoint} when using the C{existingConnection} constructor. c C sz t �| � tt| �� ��}|�| j| jjd � |�t | j j �| jjd � t j | jd| j| j| j| j|td�d�| _dS )r r} r� Fr N)r� r� r) r r r r� ru r r r� r r, r rw r� r� r� rk r� )rE r rF rF rG r� ! s � �z ExistingConnectionTests.setUpc C s� t � }t|_| j�|�}| jj�� }z%t| jjd< | � | j | j j d d �\}}}W | jj� � | jj�|� n| jj� � | jj�|� w || _|| _|| _| �|�}|jj}t�|d�S )zz Create and return a new L{SSHCommandClientEndpoint} using the C{existingConnection} constructor. r� r r r� )r r r� r� r> r� r\ �copyrW r� ru rw r �clear�updater� �_client�_pumpr� rt �connr, �existingConnection) rE ru r� r\ r� r� r� r� � connectionrF rF rG r� 9 s& � � z ExistingConnectionTests.createc C s8 | j �� | j �� | j �� | j �� | j| j| j fS )z� Give back the connection established in L{create} over which the new command channel being tested will exchange data. )ra r� r� r` rD rF rF rG r� W s z(ExistingConnectionTests.finishConnectionc C s | � |jj� | � |jj� dS )a Assert that the transport for the given protocol is still connected. L{SSHCommandClientEndpoint.existingConnection} re-uses an SSH connected created by some other code, so other code is responsible for cleaning it up. N)� assertFalsert r rC r� rF rF rG r� e s z2ExistingConnectionTests.assertClientTransportStateN)rJ rK rL rM r� r� r� r� rF rF rF rG r\ s r\ c @ r[ ) � ExistingConnectionHelperTestsz1 Tests for L{_ExistingConnectionHelper}. c C � | � ttt�� dS )zT L{_ExistingConnectionHelper} implements L{_ISSHConnectionCreator}. N)r r r. r- rD rF rF rG r! u � z,ExistingConnectionHelperTests.test_interfacec C s( t � }t|�}| �|| �|�� �� dS )z� L{_ExistingConnectionHelper.secureConnection} returns a L{Deferred} which fires with whatever object was fed to L{_ExistingConnectionHelper.__init__}. N)�objectr- �assertIsr� �secureConnection)rE rm �helperrF rF rG �test_secureConnection{ s z3ExistingConnectionHelperTests.test_secureConnectionc C � t t� �}|�t� d� dS )z� L{_ExistingConnectionHelper.cleanupConnection} does nothing to the existing connection if called with C{immediate} set to C{False}. FN�r- ri �cleanupConnection�rE rl rF rF rG �$test_cleanupConnectionNotImmediately� � zBExistingConnectionHelperTests.test_cleanupConnectionNotImmediatelyc C rn )z� L{_ExistingConnectionHelper.cleanupConnection} does nothing to the existing connection if called with C{immediate} set to C{True}. TNro rq rF rF rG �!test_cleanupConnectionImmediately� rs z?ExistingConnectionHelperTests.test_cleanupConnectionImmediatelyN)rJ rK rL rM r! rm rr rt rF rF rF rG rf p s rf c @ r� )�_PTYPathzk A L{FilePath}-like object which can be opened to create a L{_ReadFile} with certain contents. c C s || _ dS )zz @param contents: L{bytes} which will be the contents of the L{_ReadFile} this path can open. N)�contents)rE rv rF rF rG r] � s z_PTYPath.__init__c C s |dkr t | j�S ttd��)z� If the mode is r+, return a L{_ReadFile} with the contents given to this path's initializer. @raise OSError: If the mode is unsupported. @return: A L{_ReadFile} instance zrb+zFunction not implemented)r0 rv �OSErrorr )rE �moderF rF rG �open� s z _PTYPath.openN)rJ rK rL rM r] ry rF rF rF rG ru � s ru c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dS )�NewConnectionHelperTestsz, Tests for L{_NewConnectionHelper}. c C rg )zO L{_NewConnectionHelper} implements L{_ISSHConnectionCreator}. N)r r r. r/ rD rF rF rG r! � rh z'NewConnectionHelperTests.test_interfacec C s | � dtj� dS )zK The default I{known_hosts} path is I{~/.ssh/known_hosts}. z~/.ssh/known_hostsN)r� r/ � _KNOWN_HOSTSrD rF rF rG �test_defaultPath� s z)NewConnectionHelperTests.test_defaultPathc sH t � � | �td� fdd�� tdddddddddd� }| �� |j� dS )z� L{_NewConnectionHelper._knownHosts} is used to create a L{KnownHostsFile} if one is not passed to the initializer. � _knownHostsc s � S r[ rF )�clsrl rF rG r` � ra zANewConnectionHelperTests.test_defaultKnownHosts.<locals>.<lambda>N)ri �patchr/ rj r rq rF rl rG �test_defaultKnownHosts� s �z/NewConnectionHelperTests.test_defaultKnownHostsc C s� t � jd }t| �� �}t|�}|�d|� |�� td|j��� t j� d�}|j� |d�}| � t d|� td|��� t �� }| �|�d|�� dS )z� Existing entries in the I{known_hosts} file are reflected by the L{KnownHostsFile} created by L{_NewConnectionHelper} when none is supplied to it. r} s 127.0.0.1z Created known_hosts file at z~/r{ zPatched _KNOWN_HOSTS with N)rz r r r r) r r r �path�os� expanduser�replacer r/ r} r � hasHostKey)rE rO r� r �home�default�loadedrF rF rG �test_readExisting� s z*NewConnectionHelperTests.test_readExistingc C s, t dddddddddd� }| �|jt� dS )zz If L{None} is passed for the C{ui} parameter to L{_NewConnectionHelper}, a L{ConsoleUI} is used. N)r/ r� r r( rq rF rF rG �test_defaultConsoleUI� s �z.NewConnectionHelperTests.test_defaultConsoleUIc C sD t d�}tdddddddddd|� }| �|j�d��}| �|� dS )z� If L{None} is passed for the C{ui} parameter to L{_NewConnectionHelper} and /dev/tty is available, the L{ConsoleUI} used is associated with /dev/tty. s yesNs does this work?)ru r/ r� r rp r �rE �ttyrl rm rF rF rG �test_ttyConsoleUI� s �z*NewConnectionHelperTests.test_ttyConsoleUIc C sH t | �� �}tdddddddddd|� }| �|j�d��}| �|� dS )z� If L{None} is passed for the C{ui} parameter to L{_NewConnectionHelper} and /dev/tty is not available, the L{ConsoleUI} used is associated with some file which always produces a C{b"no"} response. Ns did this break?)r r r/ r� r rp re r� rF rF rG � test_nottyUI s �z%NewConnectionHelperTests.test_nottyUIc C s0 t dddddddddd� }| �td�|j� dS )zy If not passed the name of a tty in the filesystem, L{_NewConnectionHelper} uses C{b"/dev/tty"}. Ns /dev/tty)r/ r� r r� rq rF rF rG �test_defaultTTYFilename s �z0NewConnectionHelperTests.test_defaultTTYFilenamec C sF t dddddddddd� }t� }t� |_|�|d� | �|jj� dS )z� L{_NewConnectionHelper.cleanupConnection} closes the transport cleanly if called with C{immediate} set to C{False}. NF)r/ r4 r? rt rp r r )rE rl rd rF rF rG rr s � z=NewConnectionHelperTests.test_cleanupConnectionNotImmediatelyc C s` G dd� d�}t dddddddddd� }t� }t� |_|� |j_|�|d� | �|jjj� dS )z� L{_NewConnectionHelper.cleanupConnection} closes the transport with C{abortConnection} if called with C{immediate} set to C{True}. c @ rN )zMNewConnectionHelperTests.test_cleanupConnectionImmediately.<locals>.AbortableFc S rA )z7 Abort the connection. TNrB rD rF rF rG rH 1 s z]NewConnectionHelperTests.test_cleanupConnectionImmediately.<locals>.Abortable.abortConnectionN)rJ rK rL rC rH rF rF rF rG � Abortable. rj r� NT)r/ r4 r7 rt rp r rC )rE r� rl rd rF rF rG rt ( s � z:NewConnectionHelperTests.test_cleanupConnectionImmediatelyN)rJ rK rL rM r! r| r� r� r� r� r� r� rr rt rF rF rF rG rz � s rz )wrM �os.pathr� �errnor �structr �zope.interfacer �zope.interface.verifyr r r� �twisted.conch.errorr r r �twisted.conch.interfacesr �twisted.cred.checkersr �twisted.cred.portalr �twisted.internet.addressr �twisted.internet.deferr r r r �twisted.internet.errorr r r r �twisted.internet.interfacesr r �twisted.internet.protocolr r �twisted.loggerr r �twisted.python.compatr �twisted.python.failurer �twisted.python.filepathr �twisted.python.logr �twisted.python.reflectr �twisted.test.proto_helpersr! r" �twisted.trial.unittestr# �twisted.conch.avatarr% �twisted.conch.checkersr& r' � twisted.conch.client.knownhostsr( r) �twisted.conch.endpointsr* r+ r, r- r. r/ r0 �twisted.conch.sshr1 �twisted.conch.ssh.agentr2 �twisted.conch.ssh.channelr3 � twisted.conch.ssh.connectionr4 �twisted.conch.ssh.factoryr5 �twisted.conch.ssh.keysr6 �twisted.conch.ssh.transportr7 �twisted.conch.ssh.userauthr8 �twisted.conch.test.keydatar9 r: r; r< �skipri �twisted.test.iosimr= r> r? r@ rO rW rY rZ rg rk rr rz r� r� r� r r\ rf ru rz rF rF rF rG �<module> s� $ & 1 dU,PK F��Z�x�Xn Xn # __pycache__/keydata.cpython-310.pycnu �[��� o �b�� � @ s. d Z ddlmZ ed�ded�ed�ed�ed�d �Zed �ed �d d ed�d�Zed�ed�ed�dd�Zed�ed�ed�dd�Zed�ed�ed �d d�Zd d d �Z d!Z d"Z d#Z d$Z d%Zd&Zd'Zd(Zd)Zd*Zd+Zd,Zd-Zd.Zd/Zd0Zd1Zd2Zd3Z d4Z d5Z d6Z d7Z d8Z!ed9�Z"ed:�Z#ed;�Z$g d<�Z%d=S )>z+ Data used by test_keys as well as others. � )� decodebytes�i 26941361723811343819866101037675839921988027796838212268786269729694247120995560307112039197577328384456023037188438995206797878968413594751534120947806520945542732736910235620425910680704796413952531053913307374311617582141751307970630110060002581550978672180871930267106805241446648367682198750572038464556170842579437938319127485694162851261635543719756071289200110782824779256185832708552199140780701504775021850897161159085057587032100799190904325247073013454703884183936776407437943984310855088870943095814327141704475031474288054200294805383574542944648501531660749404403945254975473896534482849256068133525751i �h 4203357242869996956805024384854898198000024172950710597804898118408283516367542062349826827520762053970472184495045374765239609876131483075734873227204810666771052111553888020795198692497467740858822192444932906638025692012136764331594257829371597667863297420532149579339412600421013771755656838497323547005256289752390005486513466208261362009527404465627516909243353659408106589312384106125214417397021705035470250180168681160370530139354514779304260137038861930164164532159500721474403446561377189590538972686639694286801448419876249629285768083527396272629416756177724661940425316604626522633351193810751757014073�5 152689878451107675391723141129365667732639179427453246378763774448531436802867910180261906924087589684175595016060014593521649964959248408388984465569934780790357826811592229318702991401054226302790395714901636384511513449977061729214247279176398290513085108930550446985490864812445551198848562639933888780317�5 176444974592327996338888725079951900172097062203378367409936859072670162290963119826394224277287608693818012745872307600855894647300295516866118620024751601329775653542084052616260193174546400544176890518564317596334518015173606460860373958663673307503231977779632583864454001476729233959405710696795574874403�4 93601800238809584296951849856100709096513640338471561343936480322938679350640222284741501977205308045825703424183279521046061292444508537267852417684200791227665453277330154626999702097081815595682855341826611032986722267304009888565134822567329894852993885224775891490070400861134282266967852120152546563278)�n�e�d�p�q�u�5 102532613268641171576406907617235869673823343194357786952917153381541139247781992153835073240035039544621198205496512489289702949127531056893725702005035043292195216541115250589114284140427928363951954324455112005663182517891057569583666939618174684114192449854549414999828295140718645344764026044855941864175�5 102920317262317564432088500821911987877929665167903819917750207689976375116629109208566602236252561412937470263326262930887668422949051881895212412718444016917144560705456752517757471564532371459197940894961685025172028691607867489309937144494080086589760710215938634531338471675218590012064772045092956919481l I'Bfa'�/� |�P5#�l -QKku[�!|)�u q�sK 4W�3 1460442306266194757979024072033757031500854998345220801539426429789435409684914513123700756086453120500041882809102836102771941880716191917395123794084436959467635544938639859431446862982376796470255970943061826392752976576910270265745700231533660131769648708944711006508965764877684264272082256183140297951)�gr r �x�y�M76282513020392096317118503144964731774299773481750550543382904345687059013883�M81543197864602852632265664769441647534344375894314319681061137159310646683104�M34638743477210341700964008455655698253555655678826059678074967909361042656500s ecdsa-sha2-nistp256)r r � privateValue�curve�t28081410713485847059875391639480752139823963353428163398257609908335787109896602102090002196616273211495718603965098�t10036914308591746758780165503819213553101287571902957054148542504671046744460374996612408381962208627004841444205030�t17337335659928075994560513699823544906448896792102247714689323575406618073069185107088229463828921069465902299522926s ecdsa-sha2-nistp384)r r r r ڝ1294474282625742084665952775268376319340138427139151328602291729910013082920512632908350502247952686156279140016049549948975670668730618745449113644014505462ڝ1078410881027197618673758774943629578298556364036868908105288616296815984553198866894145509329328086635278430266482551941240591605833440825557820439734509311ڝ6627512352154608862902939026581288474953476911992147066970891407696722739507679613314422655305240639435488467243480486142397914984425997823106818915698960565s ecdsa-sha2-nistp521s ��J ^F��D �R��*k#sE�@�W{�s 7/%ڍԨ�x|a���^mgi17Li U��DX)�a�ksl -----BEGIN EC PRIVATE KEY----- MIHcAgEBBEIAjn0lSVF6QweS4bjOGP9RHwqxUiTastSE0MVuLtFvkxygZqQ712oZ ewMvqKkxthMQgxzSpGtRBcmkL7RqZ94+18qgBwYFK4EEACOhgYkDgYYABAFpX/6B mxxglwD+VpEvw0hcyxVzLxNnMGzxZGF7xmNj8nlF7M+TQctdlR2Xv/J+AgIeVGmB j2p84bkV9jBzrUNJEACsJjttZw8NbUrhxjkLT/3rMNtuwjE4vLja0P7DMTE0EV8X f09ETdku/z/1tOSSrSvRwmUcM9nQUJtHHAZlr5Q0fw== -----END EC PRIVATE KEY-----s� -----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAArAAAABNlY2RzYS 1zaGEyLW5pc3RwNTIxAAAACG5pc3RwNTIxAAAAhQQBaV/+gZscYJcA/laRL8NIXMsVcy8T ZzBs8WRhe8ZjY/J5RezPk0HLXZUdl7/yfgICHlRpgY9qfOG5FfYwc61DSRAArCY7bWcPDW 1K4cY5C0/96zDbbsIxOLy42tD+wzExNBFfF39PRE3ZLv8/9bTkkq0r0cJlHDPZ0FCbRxwG Za+UNH8AAAEAeRISlnkSEpYAAAATZWNkc2Etc2hhMi1uaXN0cDUyMQAAAAhuaXN0cDUyMQ AAAIUEAWlf/oGbHGCXAP5WkS/DSFzLFXMvE2cwbPFkYXvGY2PyeUXsz5NBy12VHZe/8n4C Ah5UaYGPanzhuRX2MHOtQ0kQAKwmO21nDw1tSuHGOQtP/esw227CMTi8uNrQ/sMxMTQRXx d/T0RN2S7/P/W05JKtK9HCZRwz2dBQm0ccBmWvlDR/AAAAQgCOfSVJUXpDB5LhuM4Y/1Ef CrFSJNqy1ITQxW4u0W+THKBmpDvXahl7Ay+oqTG2ExCDHNKka1EFyaQvtGpn3j7XygAAAA ABAg== -----END OPENSSH PRIVATE KEY----- s ecdsa-sha2-nistp521 AAAAE2VjZHNhLXNoYTItbmlzdHA1MjEAAAAIbmlzdHA1MjEAAACFBAFpX/6BmxxglwD+VpEvw0hcyxVzLxNnMGzxZGF7xmNj8nlF7M+TQctdlR2Xv/J+AgIeVGmBj2p84bkV9jBzrUNJEACsJjttZw8NbUrhxjkLT/3rMNtuwjE4vLja0P7DMTE0EV8Xf09ETdku/z/1tOSSrSvRwmUcM9nQUJtHHAZlr5Q0fw== comments -----BEGIN EC PRIVATE KEY----- MIGkAgEBBDAtAi7I8j73WCX20qUM5hhHwHuFzYWYYILs2Sh8UZ+awNkARZ/Fu2LU LLl5RtOQpbWgBwYFK4EEACKhZANiAATU17sA9P5FRwSknKcFsjjsk0+E3CeXPYX0 Tk/M0HK3PpWQWgrO8JdRHP9eFE9O/23P8BumwFt7F/AvPlCzVd35VfraFT0o4cCW G0RqpQ+np31aKmeJshkcYALEchnU+tQ= -----END EC PRIVATE KEY-----sN -----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAiAAAABNlY2RzYS 1zaGEyLW5pc3RwMzg0AAAACG5pc3RwMzg0AAAAYQTU17sA9P5FRwSknKcFsjjsk0+E3CeX PYX0Tk/M0HK3PpWQWgrO8JdRHP9eFE9O/23P8BumwFt7F/AvPlCzVd35VfraFT0o4cCWG0 RqpQ+np31aKmeJshkcYALEchnU+tQAAADIiktpWIpLaVgAAAATZWNkc2Etc2hhMi1uaXN0 cDM4NAAAAAhuaXN0cDM4NAAAAGEE1Ne7APT+RUcEpJynBbI47JNPhNwnlz2F9E5PzNBytz 6VkFoKzvCXURz/XhRPTv9tz/AbpsBbexfwLz5Qs1Xd+VX62hU9KOHAlhtEaqUPp6d9Wipn ibIZHGACxHIZ1PrUAAAAMC0CLsjyPvdYJfbSpQzmGEfAe4XNhZhgguzZKHxRn5rA2QBFn8 W7YtQsuXlG05CltQAAAAA= -----END OPENSSH PRIVATE KEY----- s� ecdsa-sha2-nistp384 AAAAE2VjZHNhLXNoYTItbmlzdHAzODQAAAAIbmlzdHAzODQAAABhBNTXuwD0/kVHBKScpwWyOOyTT4TcJ5c9hfROT8zQcrc+lZBaCs7wl1Ec/14UT07/bc/wG6bAW3sX8C8+ULNV3flV+toVPSjhwJYbRGqlD6enfVoqZ4myGRxgAsRyGdT61A== comments� ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBKimX1DZ7+Qj0SpfePMbo1pb6yGkAb5l7duC1l855yD7tEfQfqk7bc7v46We1hLMyz6ObUBYgkN/34n42F4vpeA= comments� -----BEGIN EC PRIVATE KEY----- MHcCAQEEIEyU1YOT2JxxofwbJXIjGftdNcJK55aQdNrhIt2xYQz0oAoGCCqGSM49 AwEHoUQDQgAEqKZfUNnv5CPRKl948xujWlvrIaQBvmXt24LWXznnIPu0R9B+qTtt zu/jpZ7WEszLPo5tQFiCQ3/fifjYXi+l4A== -----END EC PRIVATE KEY-----s� -----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAaAAAABNlY2RzYS 1zaGEyLW5pc3RwMjU2AAAACG5pc3RwMjU2AAAAQQSopl9Q2e/kI9EqX3jzG6NaW+shpAG+ Ze3bgtZfOecg+7RH0H6pO23O7+OlntYSzMs+jm1AWIJDf9+J+NheL6XgAAAAmCKU4hcilO IXAAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBKimX1DZ7+Qj0Spf ePMbo1pb6yGkAb5l7duC1l855yD7tEfQfqk7bc7v46We1hLMyz6ObUBYgkN/34n42F4vpe AAAAAgTJTVg5PYnHGh/BslciMZ+101wkrnlpB02uEi3bFhDPQAAAAA -----END OPENSSH PRIVATE KEY----- sX ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIPEW0RVKHhUOGV4ZRrXyRA2yUqCuKmsjE3NF/UDZV3uL comments� -----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW QyNTUxOQAAACDxFtEVSh4VDhleGUa18kQNslKgriprIxNzRf1A2Vd7iwAAAJA61eMLOtXj CwAAAAtzc2gtZWQyNTUxOQAAACDxFtEVSh4VDhleGUa18kQNslKgriprIxNzRf1A2Vd7iw AAAEA3LyXajdSomnh8YfCYAcb0Xm1nBWkxN0xpDQVVu8lEWPEW0RVKHhUOGV4ZRrXyRA2y UqCuKmsjE3NF/UDZV3uLAAAAB2NvbW1lbnQBAgMEBQY= -----END OPENSSH PRIVATE KEY-----s� ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDVaqx4I9bWG+wloVDEd2NQhEUBVUIUKirg0GDu1OmjrUr6OQZehFV1XwA2v2+qKj+DJjfBaS5b/fDz0n3WmM06QHjVyqgYwBGTJAkMgUyP95ztExZqpATpSXfD5FVks3loniwI66zoBC0hdwWnju9TMA2l5bs9auIJNm/9NNN9b0b/h9qpKSeq/631heY+Grh6HUqx6sBa9zDfH8Kk5O8/kUmWQNUZdy03w17snaY6RKXCpCnd1bqcPUWzxiwYZNW6Pd+rf81CrKfxGAugWBViC6QqbkPD5ASfNaNHjkbtM6Vlvbw7KW4CC1ffdOgTtDc1foNfICZgptyti8ZseZj3 comments� -----BEGIN RSA PRIVATE KEY----- MIIEogIBAAKCAQEA1WqseCPW1hvsJaFQxHdjUIRFAVVCFCoq4NBg7tTpo61K+jkG XoRVdV8ANr9vqio/gyY3wWkuW/3w89J91pjNOkB41cqoGMARkyQJDIFMj/ec7RMW aqQE6Ul3w+RVZLN5aJ4sCOus6AQtIXcFp47vUzANpeW7PWriCTZv/TTTfW9G/4fa qSknqv+t9YXmPhq4eh1KserAWvcw3x/CpOTvP5FJlkDVGXctN8Ne7J2mOkSlwqQp 3dW6nD1Fs8YsGGTVuj3fq3/NQqyn8RgLoFgVYgukKm5Dw+QEnzWjR45G7TOlZb28 OyluAgtX33ToE7Q3NX6DXyAmYKbcrYvGbHmY9wIDAQABAoIBACFMCGaiKNW0+44P chuFCQC58k438BxXS+NRf54jp+Q6mFUb6ot6mB682Lqx+YkSGGCs6MwLTglaQGq6 L5n4syRghLnOaZWa+eL8H1FNJxXbKyet77RprL59EOuGR3BztACHlRU7N/nnFOeA u2geG+bdu3NjuWfmsid/z88wm8KY/dkYNi82LvE9gXqf4QMtR9s0UWI53U/prKiL 2dbzhMQXuXGdBghCeE27xSr0w1jNVSvtvjNfBOp75gQkY/It1z0bbNWcY0MvkoiN Pm7aGDfYDyVniR25RjReyc7Ei+2SWjMHD9+GCPmS6dvrOAg2yc3NCgFIWzk+esrG gKnc1DkCgYEA2XAG2OK81HiRUJTUwRuJOGxGZFpRoJoHPUiPA1HMaxKOfRqxZedx dTngMgV1jRhMr5OxSbFmX3hietEMyuZNQ7Oc9Gt95gyY3M8hYo7VLhLeBK7XJG6D MaIVokQ9IqliJiK5su1UCp0Ig6cHDf8ZGI7Yqx3aSJwxaBGhZm3j2B0CgYEA+0QX i6Q2vh43Haf2YWwExKrdeD4HjB4zAq4DFIeDeuWefQhnqPKqvxJwz3Kpp8cLHYjV IP2cY8pHMFVOi8TP9H8WpJISdKEJwsRunIwz76Xl9+ArrU9cEaoahDdb/Xrqw818 sMjkH1Rjtcev3/QJp/zHJfxc6ZHXksWYHlbTsSMCgYBRr+mSn5QLSoRlPpSzO5IQ tXS4jMnvyQ4BMvovaBKhAyauz1FoFEwmmyikAjMIX+GncJgBNHleUo7Ezza8H0tV rOvBU4TH4WGoStSi/0ANgB8SqVDAKhh1lAwGmxZQqEvsQc177/dLyXUCaMSYuIaI GFpD5wIzlyJkk4MMRSp87QKBgGlmN8ZA3SHFBPOwuD5HlHx2/C3rPzk8lcNDAVHE Qpfz6Bakxu7s1EkQUDgE7jvN19DMzDJpkAegG1qf/jHNHjp+cR4ZlBpOTwzfX1LV 0Rdu7NectlWd244hX7wkiLb8r6vw76QssNyfhrADEriL4t0PwO4jPUpQ/i+4KUZY v7YnAoGAZhb5IDTQVCW8YTGsgvvvnDUefkpVAmiVDQqTvh6/4UD6kKdUcDHpePzg Zrcid5rr3dXSMEbK4tdeQZvPtUg1Uaol3N7bNClIIdvWdPx+5S9T95wJcLnkoHam rXp0IjScTxfLP+Cq5V6lJ94/pX8Ppoj1FdZfNxeS4NYFSRA7kvY= -----END RSA PRIVATE KEY-----s� -----BEGIN RSA PRIVATE KEY----- MIIEqDCCBKICAQACggEBANVqrHgj1tYb7CWhUMR3Y1CERQFVQhQqKuDQYO7U6aOt Svo5Bl6EVXVfADa/b6oqP4MmN8FpLlv98PPSfdaYzTpAeNXKqBjAEZMkCQyBTI/3 nO0TFmqkBOlJd8PkVWSzeWieLAjrrOgELSF3BaeO71MwDaXluz1q4gk2b/00031v Rv+H2qkpJ6r/rfWF5j4auHodSrHqwFr3MN8fwqTk7z+RSZZA1Rl3LTfDXuydpjpE pcKkKd3Vupw9RbPGLBhk1bo936t/zUKsp/EYC6BYFWILpCpuQ8PkBJ81o0eORu0z pWW9vDspbgILV9906BO0NzV+g18gJmCm3K2Lxmx5mPcCAwEAAQKCAQAhTAhmoijV tPuOD3IbhQkAufJON/AcV0vjUX+eI6fkOphVG+qLepgevNi6sfmJEhhgrOjMC04J WkBqui+Z+LMkYIS5zmmVmvni/B9RTScV2ysnre+0aay+fRDrhkdwc7QAh5UVOzf5 5xTngLtoHhvm3btzY7ln5rInf8/PMJvCmP3ZGDYvNi7xPYF6n+EDLUfbNFFiOd1P 6ayoi9nW84TEF7lxnQYIQnhNu8Uq9MNYzVUr7b4zXwTqe+YEJGPyLdc9G2zVnGND L5KIjT5u2hg32A8lZ4kduUY0XsnOxIvtklozBw/fhgj5kunb6zgINsnNzQoBSFs5 PnrKxoCp3NQ5AoGBANlwBtjivNR4kVCU1MEbiThsRmRaUaCaBz1IjwNRzGsSjn0a sWXncXU54DIFdY0YTK+TsUmxZl94YnrRDMrmTUOznPRrfeYMmNzPIWKO1S4S3gSu 1yRugzGiFaJEPSKpYiYiubLtVAqdCIOnBw3/GRiO2Ksd2kicMWgRoWZt49gdAoGB APtEF4ukNr4eNx2n9mFsBMSq3Xg+B4weMwKuAxSHg3rlnn0IZ6jyqr8ScM9yqafH Cx2I1SD9nGPKRzBVTovEz/R/FqSSEnShCcLEbpyMM++l5ffgK61PXBGqGoQ3W/16 6sPNfLDI5B9UY7XHr9/0Caf8xyX8XOmR15LFmB5W07EjAoGAUa/pkp+UC0qEZT6U szuSELV0uIzJ78kOATL6L2gSoQMmrs9RaBRMJpsopAIzCF/hp3CYATR5XlKOxM82 vB9LVazrwVOEx+FhqErUov9ADYAfEqlQwCoYdZQMBpsWUKhL7EHNe+/3S8l1AmjE mLiGiBhaQ+cCM5ciZJODDEUqfO0CgYBpZjfGQN0hxQTzsLg+R5R8dvwt6z85PJXD QwFRxEKX8+gWpMbu7NRJEFA4BO47zdfQzMwyaZAHoBtan/4xzR46fnEeGZQaTk8M 319S1dEXbuzXnLZVnduOIV+8JIi2/K+r8O+kLLDcn4awAxK4i+LdD8DuIz1KUP4v uClGWL+2JwKBgGYW+SA00FQlvGExrIL775w1Hn5KVQJolQ0Kk74ev+FA+pCnVHAx 6Xj84Ga3Inea693V0jBGyuLXXkGbz7VINVGqJdze2zQpSCHb1nT8fuUvU/ecCXC5 5KB2pq16dCI0nE8Xyz/gquVepSfeP6V/D6aI9RXWXzcXkuDWBUkQO5L2MAA= -----END RSA PRIVATE KEY-----s -----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAABFwAAAAdzc2gtcn NhAAAAAwEAAQAAAQEA1WqseCPW1hvsJaFQxHdjUIRFAVVCFCoq4NBg7tTpo61K+jkGXoRV dV8ANr9vqio/gyY3wWkuW/3w89J91pjNOkB41cqoGMARkyQJDIFMj/ec7RMWaqQE6Ul3w+ RVZLN5aJ4sCOus6AQtIXcFp47vUzANpeW7PWriCTZv/TTTfW9G/4faqSknqv+t9YXmPhq4 eh1KserAWvcw3x/CpOTvP5FJlkDVGXctN8Ne7J2mOkSlwqQp3dW6nD1Fs8YsGGTVuj3fq3 /NQqyn8RgLoFgVYgukKm5Dw+QEnzWjR45G7TOlZb28OyluAgtX33ToE7Q3NX6DXyAmYKbc rYvGbHmY9wAAA7gXkBoMF5AaDAAAAAdzc2gtcnNhAAABAQDVaqx4I9bWG+wloVDEd2NQhE UBVUIUKirg0GDu1OmjrUr6OQZehFV1XwA2v2+qKj+DJjfBaS5b/fDz0n3WmM06QHjVyqgY wBGTJAkMgUyP95ztExZqpATpSXfD5FVks3loniwI66zoBC0hdwWnju9TMA2l5bs9auIJNm /9NNN9b0b/h9qpKSeq/631heY+Grh6HUqx6sBa9zDfH8Kk5O8/kUmWQNUZdy03w17snaY6 RKXCpCnd1bqcPUWzxiwYZNW6Pd+rf81CrKfxGAugWBViC6QqbkPD5ASfNaNHjkbtM6Vlvb w7KW4CC1ffdOgTtDc1foNfICZgptyti8ZseZj3AAAAAwEAAQAAAQAhTAhmoijVtPuOD3Ib hQkAufJON/AcV0vjUX+eI6fkOphVG+qLepgevNi6sfmJEhhgrOjMC04JWkBqui+Z+LMkYI S5zmmVmvni/B9RTScV2ysnre+0aay+fRDrhkdwc7QAh5UVOzf55xTngLtoHhvm3btzY7ln 5rInf8/PMJvCmP3ZGDYvNi7xPYF6n+EDLUfbNFFiOd1P6ayoi9nW84TEF7lxnQYIQnhNu8 Uq9MNYzVUr7b4zXwTqe+YEJGPyLdc9G2zVnGNDL5KIjT5u2hg32A8lZ4kduUY0XsnOxIvt klozBw/fhgj5kunb6zgINsnNzQoBSFs5PnrKxoCp3NQ5AAAAgQCFSxt6mxIQN54frV7a/s aW/t81a7k04haXkiYJvb1wIAOnNb0tG6DSB0cr1N6oqAcHG7gEIKcnQTxsOTnpQc7nFx3R TFy8PdImJv5q1v1Icq5G+nvD0xlgRB2lE6eA9WMp1HpdBgcWXfaLPctkOuKEWk2MBi0tnR zrg0x4PXlUzgAAAIEA2XAG2OK81HiRUJTUwRuJOGxGZFpRoJoHPUiPA1HMaxKOfRqxZedx dTngMgV1jRhMr5OxSbFmX3hietEMyuZNQ7Oc9Gt95gyY3M8hYo7VLhLeBK7XJG6DMaIVok Q9IqliJiK5su1UCp0Ig6cHDf8ZGI7Yqx3aSJwxaBGhZm3j2B0AAACBAPtEF4ukNr4eNx2n 9mFsBMSq3Xg+B4weMwKuAxSHg3rlnn0IZ6jyqr8ScM9yqafHCx2I1SD9nGPKRzBVTovEz/ R/FqSSEnShCcLEbpyMM++l5ffgK61PXBGqGoQ3W/166sPNfLDI5B9UY7XHr9/0Caf8xyX8 XOmR15LFmB5W07EjAAAAAAEC -----END OPENSSH PRIVATE KEY----- s� -----BEGIN RSA PRIVATE KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: DES-EDE3-CBC,FFFFFFFFFFFFFFFF p2A1YsHLXkpMVcsEqhh/nCYb5AqL0uMzfEIqc8hpZ/Ub8PtLsypilMkqzYTnZIGS ouyPjU/WgtR4VaDnutPWdgYaKdixSEmGhKghCtXFySZqCTJ4O8NCczsktYjUK3D4 Jtl90zL6O81WBY6xP76PBQo9lrI/heAetATeyqutc18bwQIGU+gKk32qvfo15DfS VYiY0Ds4D7F7fd9pz+f5+UbFUCgU+tfDvBrqodYrUgmH7jKoW/CRDCHHyeEIZDbF mcMwdcKOyw1sRLaPdihRSVx3kOMvIotHKVTkIDMp+0RTNeXzQnp5U2qzsxzTcG/M UyJN38XXkuvq5VMj2zmmjHzx34w3NK3ZxpZcoaFUqUBlNp2C8hkCLrAa/DWobKqN 5xA1ElrQvli9XXkT/RIuy4Gc10bbGEoJjuxNRibtSxxWd5Bd1E40ocOd4l1ebI8+ w69XvMTnsmHvkBEADGF2zfRszKnMelg+W5NER1UDuNT03i+1cuhp+2AZg8z7niTO M17XP3ScGVxrQAEYgtxPrPeIpFJvOx2j5Yt78U9Y2WlaAG6DrubbYv2RsMIibhOG yk139vMdD8FwCey6yMkkhFAJwnBtC22MAWgjmC5c6AF3SRQSjjQXepPsJcLgpOjy YwjhnL8w56x9kVDUNPw9A9Cqgxo2sty34ATnKrh4h59PsP83LOL6OC5WjbASgZRd OIBD8RloQPISo+RUF7X0i4kdaHVNPlR0KyapR+3M5BwhQuvEO99IArDV2LNKGzfc W4ssugm8iyAJlmwmb2yRXIDHXabInWY7XCdGk8J2qPFbDTvnPbiagJBimjVjgpWw tV3sVlJYqmOqmCDP78J6he04l0vaHtiOWTDEmNCrK7oFMXIIp3XWjOZGPSOJFdPs 6Go3YB+EGWfOQxqkFM28gcqmYfVPF2sa1FbZLz0ffO11Ma/rliZxZu7WdrAXe/tc BgIQ8etp2PwAK4jCwwVwjIO8FzqQGpS23Y9NY3rfi97ckgYXKESFtXPsMMA+drZd ThbXvccfh4EPmaqQXKf4WghHiVJ+/yuY1kUIDEl/O0jRZWT7STgBim/Aha1m6qRs zl1H7hkDbU4solb1GM5oPzbgGTzyBc+z0XxM9iFRM+fMzPB8+yYHTr4kPbVmKBjy SCovjQQVsHE4YeUGTq6k/NF5cVIRKTW/RlHvzxsky1Zj31MC736jrxGw4KG7VSLZ fP6F5jj+mXwS7m0v5to42JBZmRJdKUD88QaGE3ncyQ4yleW5bn9Lf9SuzQg1Dhao 3rSA1RuexsHlIAHvGxx/17X+pyygl8DJbt6TBfbLQk9wc707DJTfh5M/bnk9wwIX l/Hsa1WtylAMW/2MzgiVy83MbYz4+Ss6GQ5W66okWji+NxrnrYEy6q+WgVQanp7X D+D7oKykqE1Cdvvulvtfl5fh8wlAs8mrUnKPBBUru348u++2lfacLkxRXyT1ooqY uSNE5nlwFt08N2Ou/bl7yq6QNRMYrRkn+UEfHWCNYDoGMHln2/i6Z1RapQzNarik tJf7radBz5nBwBjP08YAEACNSQvpsUgdqiuYjLwX7efFXQva2RzqaQ== -----END RSA PRIVATE KEY-----s@ -----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAACmFlczI1Ni1jdHIAAAAGYmNyeXB0AAAAGAAAABD0f9WAof DTbmwztb8pdrSeAAAAEAAAAAEAAAEXAAAAB3NzaC1yc2EAAAADAQABAAABAQDVaqx4I9bW G+wloVDEd2NQhEUBVUIUKirg0GDu1OmjrUr6OQZehFV1XwA2v2+qKj+DJjfBaS5b/fDz0n 3WmM06QHjVyqgYwBGTJAkMgUyP95ztExZqpATpSXfD5FVks3loniwI66zoBC0hdwWnju9T MA2l5bs9auIJNm/9NNN9b0b/h9qpKSeq/631heY+Grh6HUqx6sBa9zDfH8Kk5O8/kUmWQN UZdy03w17snaY6RKXCpCnd1bqcPUWzxiwYZNW6Pd+rf81CrKfxGAugWBViC6QqbkPD5ASf NaNHjkbtM6Vlvbw7KW4CC1ffdOgTtDc1foNfICZgptyti8ZseZj3AAADwPQaac8s1xX3af hQTQexj0vEAWDQsLYzDHN9G7W+UP5WHUu7igeu2GqAC/TOnjUXDP73I+EN3n7T3JFeDRfs U1Z6Zqb0NKHSRVYwDIdIi8qVohFv85g6+xQ01OpaoOzz+vI34OUvCRHQGTgR6L9fQShZyC McopYMYfbIse6KcqkfxX3KSdG1Pao6Njx/ShFRbgvmALpR/z0EaGCzHCDxpfUyAdnxm621 Jzaf+LverWdN7sfrfMptaS9//9iJb70sL67K+YIB64qhDnA/w9UOQfXGQFL+AEtdM0BPv8 thP1bs7T0yucBl+ZXdrDKVLZfaS3S/w85Jlgfu+a1DG73pOBOuag435iEJ9EnspjXiiydx GrfSRk2C+/c4fBDZVGFscK5bfQuUUZyU1qOagekxX7WLHFKk9xajnud+nrAN070SeNwlX8 FZ2CI4KGlQfDvVUpKanYn8Kkj3fZ+YBGyx4M+19clF65FKSM0x1Rrh5tAmNT/SNDbSc28m ASxrBhztzxUFTrIn3tp+uqkJniFLmFsUtiAUmj8fNyE9blykU7dqq+CqpLA872nQ9bOHHA JsS1oBYmQ0n6AJz8WrYMdcepqWVld6Q8QSD1zdrY/sAWUovuBA1s4oIEXZhpXSS4ZJiMfh PVktKBwj5bmoG/mmwYLbo0JHntK8N3TGTzTGLq5TpSBBdVvWSWo7tnfEkrFObmhi1uJSrQ 3zfPVP6BguboxBv+oxhaUBK8UOANe6ZwM4vfiu+QN+sZqWymHIfAktz7eWzwlToe4cKpdG Uv+e3/7Lo2dyMl3nke5HsSUrlsMGPREuGkBih8+o85ii6D+cuCiVtus3f5c78Cir80zLIr Z0wWvEAjciEvml00DWaA+JIaOrWwvXySaOzFGpCqC9SQjao379bvn9P3b7kVZsy6zBfHqm bNEJUOuhBZaY8Okz36chh1xqh4sz7m3nsZ3GYGcvM+3mvRY72QnqsQEG0Sp1XYIn2bHa29 tqp7CG9X8J6dqMcPeoPRDWIX9gw7EPl/M0LP6xgewGJ9bgxwle6Mnr9kNITIswjAJqrLec zx7dfixjAPc42ADqrw/tEdFQcSqxigcfJNKO1LbDBjh+Hk/cSBou2PoxbIcl0qfQfbGcqI Dbpd695IEuiW9pYR22txNoIi+7cbMsuFHxQ/OqbrX/jCsprGNNJLAjgGsVEI1JnHWDH0db 3UbqbOHAeY3ufoYXNY1utVOIACpW3r9wBw3FjRi04d70VcKr16OXvOAHGN2G++Y+kMya84 Hl/Kt/gA== -----END OPENSSH PRIVATE KEY----- s� -----BEGIN RSA PRIVATE KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: AES-128-CBC,0673309A6ACCAB4B77DEE1C1E536AC26 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----s� {KDEwOnB1YmxpYy1rZXkoMTQ6cnNhLXBrY3MxLXNoYTEoMTpuMjU3OgDVaqx4I9bWG+wloVDEd2NQhEUBVUIUKirg0GDu1OmjrUr6OQZehFV1XwA2v2+qKj+DJjfBaS5b/fDz0n3WmM06QHjVyqgYwBGTJAkMgUyP95ztExZqpATpSXfD5FVks3loniwI66zoBC0hdwWnju9TMA2l5bs9auIJNm/9NNN9b0b/h9qpKSeq/631heY+Grh6HUqx6sBa9zDfH8Kk5O8/kUmWQNUZdy03w17snaY6RKXCpCnd1bqcPUWzxiwYZNW6Pd+rf81CrKfxGAugWBViC6QqbkPD5ASfNaNHjkbtM6Vlvbw7KW4CC1ffdOgTtDc1foNfICZgptyti8ZseZj3KSgxOmUzOgEAASkpKQ==}s� (11:private-key(9:rsa-pkcs1(1:n257: �j�x#���%�P�wcP�EUB**��`��飭J�9^�Uu_ 6�o�*?�&7�i.[����}֘�:@x�ʨ��$ �L����j��Iw��Ud�yh�,��-!w���S0 ��=j� 6o�4�}oF��ک)'������>�z J��Z�0� ¤��?�I�@�w-7�^읦:D�¤)�պ�=E��,dպ=߫�B��� �Xb �*nC���5�G�F�3�e��;)n W�t��75~�_ &`�ܭ��ly��)(1:e3: )(1:d256:!Lf�(մ��r� ��N7� WK�Q�#��:�U�z� �غ���`��� N Z@j�/���$`���i���� QM'�+'��i��}�Gps� ��;7��瀻h �ݻsc�g�'��0���6/6.�=�z��-G�4Qb9�O鬨������q�BxM��*��X�U+�3_�{�$c�-�=l՜cC/���>n�7�%g� �F4^��ċ�Z3߆�����86��� H[9>z�ƀ���9)(1:p129: �D��6� 7 ��alĪ�x>� 3���z�}g��p�r��� �� ��c�G0UN������t� ��n��3���+�O\��7[�z���|��� Tc�ǯ�� ���%�\�גŘ Vӱ#)(1:q129: �p���x�P���8lFdZQ��=H�Q�k�}�e�qu9�2u�L���I�f_xbz� ��MC���k}� ���!b��.���$n�1��D="�b&"���T ��� ��ث �H�1h�fm�� )(1:a128:if7�@�!��>G�|v�-�?9<��CQ�B��������IP8�;�����2i��Z��1� :~q �NO �_R��n�ל�U�ێ!_�$�������,�ܟ��������#=JP�/�)FX��')(1:b128:Q�钟� J�e>��;��t�����2�/h�&��QhL&�(�3_�p�4y^R���6� KU��S���a�JԢ�@ � �P�*u� �P�K�A�{�K�uhĘ���ZC�3�"d�� E*|�)(1:c128:f� 4�T%�a1����5 ~JUh� �� ��@���Tp1�x��f�"w�����0F���^A�ϵH5Q�%���4)H!��t�~�/S�� p��v��zt"4�O�?��^�'�?�����_7���I;��)))s� ssh-rsa !Lf�(մ��r� ��N7� WK�Q�#��:�U�z� �غ���`��� N Z@j�/���$`���i���� QM'�+'��i��}�Gps� ��;7��瀻h �ݻsc�g�'��0���6/6.�=�z��-G�4Qb9�O鬨������q�BxM��*��X�U+�3_�{�$c�-�=l՜cC/���>n�7�%g� �F4^��ċ�Z3߆�����86��� H[9>z�ƀ���9 �j�x#���%�P�wcP�EUB**��`��飭J�9^�Uu_ 6�o�*?�&7�i.[����}֘�:@x�ʨ��$ �L����j��Iw��Ud�yh�,��-!w���S0 ��=j� 6o�4�}oF��ک)'������>�z J��Z�0� ¤��?�I�@�w-7�^읦:D�¤)�պ�=E��,dպ=߫�B��� �Xb �*nC���5�G�F�3�e��;)n W�t��75~�_ &`�ܭ��ly�� � �Kz�7� �^�Ɩ��5k�4���& ��p �5�-��G+�ި�� �'A<l99�A�� �L\�=�&&�j�Hr�F�{��`D ����c)�z]]��=�d:�ZM�--� �Lx=yT� � �p���x�P���8lFdZQ��=H�Q�k�}�e�qu9�2u�L���I�f_xbz� ��MC���k}� ���!b��.���$n�1��D="�b&"���T ��� ��ث �H�1h�fm�� � �D��6� 7 ��alĪ�x>� 3���z�}g��p�r��� �� ��c�G0UN������t� ��n��3���+�O\��7[�z���|��� Tc�ǯ�� ���%�\�גŘ Vӱ#sT ssh-dss AAAAB3NzaC1kc3MAAACBAJKQOsVERVDQIpANHH+JAAylo9LvFYmFFVMIuHFGlZpIL7sh3IMkqy+cssINM/lnHD3fmsAyLlUXZtt6PD9LgZRazsPOgptuH+Gu48G+yFuE8l0fVVUivos/MmYVJ66qT99htcZKatrTWZnpVW7gFABoqw+he2LZ0gkeU0+Sx9a5AAAAFQD0EYmTNaFJ8CS0+vFSF4nYcyEnSQAAAIEAkgLjxHJAE7qFWdTqf7EZngu7jAGmdB9k3YzMHe1ldMxEB7zNw5aOnxjhoYLtiHeoEcOk2XOyvnE+VfhIWwWAdOiKRTEZlmizkvhGbq0DCe2EPMXirjqWACI5nDioQX1oEMonR8N3AEO5v9SfBqS2Q9R6OBr6lf04RvwpHZ0UGu8AAACAAhRpxGMIWEyaEh8YnjiazQTNEpklRZqeBGo1gotJggNmVaIQNIClGlLyCi359efEUuQcZ9SXxM59P+hecc/GU/GHakW5YWE4dP2GgdgMQWC7S6WFIXePGGXqNQDdWxlX8umhenvQqa1PnKrFRhDrJw8Z7GjdHxflsxCEmXPoLN8= comments� -----BEGIN DSA PRIVATE KEY----- MIIBvAIBAAKBgQCSkDrFREVQ0CKQDRx/iQAMpaPS7xWJhRVTCLhxRpWaSC+7IdyD JKsvnLLCDTP5Zxw935rAMi5VF2bbejw/S4GUWs7DzoKbbh/hruPBvshbhPJdH1VV Ir6LPzJmFSeuqk/fYbXGSmra01mZ6VVu4BQAaKsPoXti2dIJHlNPksfWuQIVAPQR iZM1oUnwJLT68VIXidhzISdJAoGBAJIC48RyQBO6hVnU6n+xGZ4Lu4wBpnQfZN2M zB3tZXTMRAe8zcOWjp8Y4aGC7Yh3qBHDpNlzsr5xPlX4SFsFgHToikUxGZZos5L4 Rm6tAwnthDzF4q46lgAiOZw4qEF9aBDKJ0fDdwBDub/UnwaktkPUejga+pX9OEb8 KR2dFBrvAoGAAhRpxGMIWEyaEh8YnjiazQTNEpklRZqeBGo1gotJggNmVaIQNICl GlLyCi359efEUuQcZ9SXxM59P+hecc/GU/GHakW5YWE4dP2GgdgMQWC7S6WFIXeP GGXqNQDdWxlX8umhenvQqa1PnKrFRhDrJw8Z7GjdHxflsxCEmXPoLN8CFQDV2gbL czUdxCus0pfEP1bddaXRLQ== -----END DSA PRIVATE KEY-----sQ -----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAABsgAAAAdzc2gtZH NzAAAAgQCSkDrFREVQ0CKQDRx/iQAMpaPS7xWJhRVTCLhxRpWaSC+7IdyDJKsvnLLCDTP5 Zxw935rAMi5VF2bbejw/S4GUWs7DzoKbbh/hruPBvshbhPJdH1VVIr6LPzJmFSeuqk/fYb XGSmra01mZ6VVu4BQAaKsPoXti2dIJHlNPksfWuQAAABUA9BGJkzWhSfAktPrxUheJ2HMh J0kAAACBAJIC48RyQBO6hVnU6n+xGZ4Lu4wBpnQfZN2MzB3tZXTMRAe8zcOWjp8Y4aGC7Y h3qBHDpNlzsr5xPlX4SFsFgHToikUxGZZos5L4Rm6tAwnthDzF4q46lgAiOZw4qEF9aBDK J0fDdwBDub/UnwaktkPUejga+pX9OEb8KR2dFBrvAAAAgAIUacRjCFhMmhIfGJ44ms0EzR KZJUWangRqNYKLSYIDZlWiEDSApRpS8got+fXnxFLkHGfUl8TOfT/oXnHPxlPxh2pFuWFh OHT9hoHYDEFgu0ulhSF3jxhl6jUA3VsZV/LpoXp70KmtT5yqxUYQ6ycPGexo3R8X5bMQhJ lz6CzfAAAB2MVcBjzFXAY8AAAAB3NzaC1kc3MAAACBAJKQOsVERVDQIpANHH+JAAylo9Lv FYmFFVMIuHFGlZpIL7sh3IMkqy+cssINM/lnHD3fmsAyLlUXZtt6PD9LgZRazsPOgptuH+ Gu48G+yFuE8l0fVVUivos/MmYVJ66qT99htcZKatrTWZnpVW7gFABoqw+he2LZ0gkeU0+S x9a5AAAAFQD0EYmTNaFJ8CS0+vFSF4nYcyEnSQAAAIEAkgLjxHJAE7qFWdTqf7EZngu7jA GmdB9k3YzMHe1ldMxEB7zNw5aOnxjhoYLtiHeoEcOk2XOyvnE+VfhIWwWAdOiKRTEZlmiz kvhGbq0DCe2EPMXirjqWACI5nDioQX1oEMonR8N3AEO5v9SfBqS2Q9R6OBr6lf04RvwpHZ 0UGu8AAACAAhRpxGMIWEyaEh8YnjiazQTNEpklRZqeBGo1gotJggNmVaIQNIClGlLyCi35 9efEUuQcZ9SXxM59P+hecc/GU/GHakW5YWE4dP2GgdgMQWC7S6WFIXePGGXqNQDdWxlX8u mhenvQqa1PnKrFRhDrJw8Z7GjdHxflsxCEmXPoLN8AAAAVANXaBstzNR3EK6zSl8Q/Vt11 pdEtAAAAAAE= -----END OPENSSH PRIVATE KEY----- sK e0tERXdPbkIxWW14cFl5MXJaWGtvTXpwa2MyRW9NVHB3TVRJNU9nQ1NrRHJGUkVWUTBDS1FEUngv aVFBTXBhUFM3eFdKaFJWVENMaHhScFdhU0MrN0lkeURKS3N2bkxMQ0RUUDVaeHc5MzVyQU1pNVZG MmJiZWp3L1M0R1VXczdEem9LYmJoL2hydVBCdnNoYmhQSmRIMVZWSXI2TFB6Sm1GU2V1cWsvZlli WEdTbXJhMDFtWjZWVnU0QlFBYUtzUG9YdGkyZElKSGxOUGtzZld1U2tvTVRweE1qRTZBUFFSaVpN MW9VbndKTFQ2OFZJWGlkaHpJU2RKS1NneE9tY3hNams2QUpJQzQ4UnlRQk82aFZuVTZuK3hHWjRM dTR3QnBuUWZaTjJNekIzdFpYVE1SQWU4emNPV2pwOFk0YUdDN1loM3FCSERwTmx6c3I1eFBsWDRT RnNGZ0hUb2lrVXhHWlpvczVMNFJtNnRBd250aER6RjRxNDZsZ0FpT1p3NHFFRjlhQkRLSjBmRGR3 QkR1Yi9Vbndha3RrUFVlamdhK3BYOU9FYjhLUjJkRkJydktTZ3hPbmt4TWpnNkFoUnB4R01JV0V5 YUVoOFluamlhelFUTkVwa2xSWnFlQkdvMWdvdEpnZ05tVmFJUU5JQ2xHbEx5Q2kzNTllZkVVdVFj WjlTWHhNNTlQK2hlY2MvR1UvR0hha1c1WVdFNGRQMkdnZGdNUVdDN1M2V0ZJWGVQR0dYcU5RRGRX eGxYOHVtaGVudlFxYTFQbktyRlJoRHJKdzhaN0dqZEh4ZmxzeENFbVhQb0xOOHBLU2s9fQ== s� KDExOnByaXZhdGUta2V5KDM6ZHNhKDE6cDEyOToAkpA6xURFUNAikA0cf4kADKWj0u8ViYUVUwi4 cUaVmkgvuyHcgySrL5yywg0z+WccPd+awDIuVRdm23o8P0uBlFrOw86Cm24f4a7jwb7IW4TyXR9V VSK+iz8yZhUnrqpP32G1xkpq2tNZmelVbuAUAGirD6F7YtnSCR5TT5LH1rkpKDE6cTIxOgD0EYmT NaFJ8CS0+vFSF4nYcyEnSSkoMTpnMTI5OgCSAuPEckATuoVZ1Op/sRmeC7uMAaZ0H2TdjMwd7WV0 zEQHvM3Dlo6fGOGhgu2Id6gRw6TZc7K+cT5V+EhbBYB06IpFMRmWaLOS+EZurQMJ7YQ8xeKuOpYA IjmcOKhBfWgQyidHw3cAQ7m/1J8GpLZD1Ho4GvqV/ThG/CkdnRQa7ykoMTp5MTI4OgIUacRjCFhM mhIfGJ44ms0EzRKZJUWangRqNYKLSYIDZlWiEDSApRpS8got+fXnxFLkHGfUl8TOfT/oXnHPxlPx h2pFuWFhOHT9hoHYDEFgu0ulhSF3jxhl6jUA3VsZV/LpoXp70KmtT5yqxUYQ6ycPGexo3R8X5bMQ hJlz6CzfKSgxOngyMToA1doGy3M1HcQrrNKXxD9W3XWl0S0pKSk= sm AAAAB3NzaC1kc3MAAACBAJKQOsVERVDQIpANHH+JAAylo9LvFYmFFVMIuHFGlZpIL7sh3IMkqy+c ssINM/lnHD3fmsAyLlUXZtt6PD9LgZRazsPOgptuH+Gu48G+yFuE8l0fVVUivos/MmYVJ66qT99h tcZKatrTWZnpVW7gFABoqw+he2LZ0gkeU0+Sx9a5AAAAFQD0EYmTNaFJ8CS0+vFSF4nYcyEnSQAA AIEAkgLjxHJAE7qFWdTqf7EZngu7jAGmdB9k3YzMHe1ldMxEB7zNw5aOnxjhoYLtiHeoEcOk2XOy vnE+VfhIWwWAdOiKRTEZlmizkvhGbq0DCe2EPMXirjqWACI5nDioQX1oEMonR8N3AEO5v9SfBqS2 Q9R6OBr6lf04RvwpHZ0UGu8AAACAAhRpxGMIWEyaEh8YnjiazQTNEpklRZqeBGo1gotJggNmVaIQ NIClGlLyCi359efEUuQcZ9SXxM59P+hecc/GU/GHakW5YWE4dP2GgdgMQWC7S6WFIXePGGXqNQDd WxlX8umhenvQqa1PnKrFRhDrJw8Z7GjdHxflsxCEmXPoLN8AAAAVANXaBstzNR3EK6zSl8Q/Vt11 pdEt ) �DSAData�RSAData�privateDSA_agentv3�privateDSA_lsh�privateDSA_openssh�privateRSA_agentv3�privateRSA_lsh�privateRSA_openssh� publicDSA_lsh�publicDSA_openssh� publicRSA_lsh�publicRSA_openssh� privateRSA_openssh_alternateN)&�__doc__�base64r �intr" r! �ECDatanistp256�ECDatanistp384�ECDatanistp521� Ed25519Data�privateECDSA_openssh521�privateECDSA_openssh521_new�publicECDSA_openssh521�privateECDSA_openssh384�privateECDSA_openssh384_new�publicECDSA_openssh384�publicECDSA_openssh�privateECDSA_openssh�privateECDSA_openssh_new�publicEd25519_openssh�privateEd25519_openssh_newr, r( r- �privateRSA_openssh_new� privateRSA_openssh_encrypted� privateRSA_openssh_encrypted_new� privateRSA_openssh_encrypted_aesr+ r' r& r* r% �privateDSA_openssh_newr) r$ r# �__all__� rF rF �</usr/lib/python3/dist-packages/twisted/conch/test/keydata.py�<module> s� � � ����2���� ������������� � �� � � ! � �5�* ��� PK F��Z�%i�. �. ( __pycache__/test_channel.cpython-310.pycnu �[��� o �b�/ � @ s� d Z ddlmZ ddlmZ z.ddlmZ ddlmZ ddl m Z ddl m Z ddl mZ dd lmZ dd lmZ d ZW n eyI d ZeZ Y nw dd lmZ G dd� de �Zddd�Zeee�G dd� de��ZdS )z Test ssh/channel.py. � )�skipIf)� verifyObject)�channel)�SSHTransportAddress)� SSHService)�SSHServerTransport)� interfaces)� IPv4Address)�StringTransport� zConch SSH not supported.)�TestCasec @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �MockConnectiona� A mock for twisted.conch.ssh.connection.SSHConnection. Record the data that channels send, and when they try to close the connection. @ivar data: a L{dict} mapping channel id #s to lists of data sent by that channel. @ivar extData: a L{dict} mapping channel id #s to lists of 2-tuples (extended data type, data) sent by that channel. @ivar closes: a L{dict} mapping channel id #s to True if that channel sent a close message. c C s i | _ i | _i | _d S )N)�data�extData�closes��self� r �A/usr/lib/python3/dist-packages/twisted/conch/test/test_channel.py�__init__* s zMockConnection.__init__c C s dS )z, Return our logging prefix. r r r r r r � logPrefix/ s zMockConnection.logPrefixc C s | j �|g ��|� dS )z' Record the sent data. N)r � setdefault�append)r r r r r r �sendData5 � zMockConnection.sendDatac C s | j �|g ��||f� dS )z0 Record the sent extended data. N)r r r )r r �typer r r r �sendExtendedData; s z MockConnection.sendExtendedDatac C s d| j |< dS )z? Record that the channel sent a close message. TN)r )r r r r r � sendCloseA s zMockConnection.sendCloseN) �__name__� __module__� __qualname__�__doc__r r r r r r r r r r s r Nc C s& t � }|�t||d�� |�| � dS )a� Connect a SSHTransport which is already connected to a remote peer to the channel under test. @param service: Service used over the connected transport. @type service: L{SSHService} @param hostAddress: Local address of the connected transport. @type hostAddress: L{interfaces.IAddress} @param peerAddress: Remote address of the connected transport. @type peerAddress: L{interfaces.IAddress} )� hostAddress� peerAddressN)r �makeConnectionr � setService)�servicer" r# � transportr r r �connectSSHTransportH s �r( c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd S )!� ChannelTestsz" Tests for L{SSHChannel}. c C s& t � | _tj| jdd�| _d| j_dS )z� Initialize the channel. remoteMaxPacket is 10 so that data is able to be sent (the default of 0 means no data is sent because no packets are made). � )�conn�remoteMaxPackets channelN)r r+ r � SSHChannel�namer r r r �setUpc s zChannelTests.setUpc C s | � ttj| j�� dS )zK L{SSHChannel} instances provide L{interfaces.ITransport}. N)� assertTruer r � ITransportr r r r r �test_interfacem r zChannelTests.test_interfacec C s t j| jd�}| �|jd� | �|jd� | �|jd� | �|jd� | �|jd� | �|j| j� | � |j � | � |j � t �ddddd d d �}| �|jd� | �|jd� | �|jd� | �|jd� | �|jd� | �|jd � | �|j d � | �|j d � d S ) a� Test that SSHChannel initializes correctly. localWindowSize defaults to 131072 (2**17) and localMaxPacket to 32768 (2**15) as reasonable defaults (what OpenSSH uses for those variables). The values in the second set of assertions are meaningless; they serve only to verify that the instance variables are assigned in the correct order. )r+ i i � r � � � � � � � N) r r- r+ � assertEqual�localWindowSize�localWindowLeft�localMaxPacket�remoteWindowLeftr, � assertIsNoner �avatar)r �c�c2r r r � test_inits s$ zChannelTests.test_initc C s. | � t| j�d� | � ttjdd��d� dS )z{ Test that str(SSHChannel) works gives the channel name and local and remote windows at a glance.. z%<SSHChannel channel (lw 131072 rw 0)>r3 �� localWindowz <SSHChannel None (lw 1 rw 0)>N)r: �strr r- r r r r �test_str� s �zChannelTests.test_strc C s. | � | j�� d� | � tjdd��� d� dS )z Test that bytes(SSHChannel) works, gives the channel name and local and remote windows at a glance.. s% <SSHChannel channel (lw 131072 rw 0)>r3 rD s <SSHChannel None (lw 1 rw 0)>N)r: r � __bytes__r- r r r r � test_bytes� s ��zChannelTests.test_bytesc C s | � | j�� d� dS )z� Test that SSHChannel.logPrefix gives the name of the channel, the local channel ID and the underlying connection. z.SSHChannel channel (unknown) on MockConnectionN)r: r r r r r r �test_logPrefix� s �zChannelTests.test_logPrefixc s dg� � fdd�}|| j _| j �d� | j �dd� | j �d� | �| j jd� | �| j j� | �� d � | �| j j d � | �| j j | j dg� | �| j j g � | �| j j | j d g� d� d< | j �d � | �� d � | j �d � | j �� | j �d � | �� d � d S )zq Test that addWindowBytes adds bytes to the window and resumes writing if it was paused. Fc � d� d< d S �NTr r r ��cbr r �stubStartWriting� � z:ChannelTests.test_addWindowBytes.<locals>.stubStartWriting� testr3 �2 �* r � )r3 rQ � sP aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaN)r � startWriting�write� writeExtended�addWindowBytesr: r> r0 � areWriting�bufr+ r �extBufr � assertFalse�loseConnection)r rO r rM r �test_addWindowBytes� s( z ChannelTests.test_addWindowBytesc C sL dd� | j _| �| j �dd�� | �| j �dd�� | �| j �dd�� dS )zn Test that requestReceived handles requests by dispatching them to request_* methods. c S s | dkS )NrT r )r r r r �<lambda>� s z3ChannelTests.test_requestReceived.<locals>.<lambda>s test-methodrT � as bad-methodN)r �request_test_methodr0 �requestReceivedr] r r r r �test_requestReceived� s z!ChannelTests.test_requestReceivedc C s* | � | jj� | j�� | �| jj� dS )zM Test that the default closeReceieved closes the connection. N)r] r �closing� closeReceivedr0 r r r r �test_closeReceieved� s z ChannelTests.test_closeReceievedc s$ dg� � fdd�}|| j _| j �d� | j �d� | �| j j� | �� d � | j �d� | j �d� | jj| j }| � |d dg� | � | j j d � | j �d � | � |g d �� | � | j j d � d� d< | j �d� | �| j j� | �� d � | � |g d�� | � | j j d� | � | j j d� dS )z� Test that write handles data correctly. Send data up to the size of the remote window, splitting the data into packets of length remoteMaxPacket. Fc rK rL r r rM r r �stubStopWriting� rP z0ChannelTests.test_write.<locals>.stubStopWriting� dra r rU � ta� da� � 12345678901)rk rj � 1234567890� 1r7 � 123456)rk rj rn ro � 12345� 6N) r � stopWritingrW r] rZ r0 rY r+ r r: r> r[ �r rh r r rM r � test_write� s, zChannelTests.test_writec sB dg� � fdd�}|| j _| j �dd� | j �dd� | j �dd� | �| j j� | �� d � | j �d � | j �dd� | jj| j }| � |g d �� | � | j j d � | j �d d� | � |g d�� | � | j j d� d� d < | j �dd� | �| j j� | �� d � | � |g d�� | � | j j ddgg� | � | j j d � dS )z� Test that writeExtended handles data correctly. Send extended data up to the size of the window, splitting the extended data into packets of length remoteMaxPacket. Fc rK rL r r rM r r rh rP z8ChannelTests.test_writeExtended.<locals>.stubStopWritingr3 ri ra r4 � tr rU )�r3 rk �r4 rv �r4 ra rl r5 rm )rw rx ry �r5 rn �r5 ro r7 r6 rp )rw rx ry rz r{ )r6 rq rr N) r rs rX r] rZ r0 rY r+ r r: r> r\ rt r rM r �test_writeExtended s8 �� z ChannelTests.test_writeExtendedc C sB | j �d� | j �dd� td�D �� | �| jj| j dg� dS )zS Test that writeSequence is equivalent to write(''.join(sequece)). rU c s s � | ]}d |f V qdS )s %dNr )�.0�ir r r � <genexpr>= s � z2ChannelTests.test_writeSequence.<locals>.<genexpr>r* s 0123456789N)r rY � writeSequence�ranger: r+ r r r r r �test_writeSequence8 s z ChannelTests.test_writeSequencec C s� | j �d� | j �dd� | j �� | �| jj�| j �� | j �d� | �| jj�| j �� | j �d� | � | jj�| j �� dS )zk Tesyt that loseConnection() doesn't close the channel until all the data is sent. s datar3 s datadatar6 � N) r rW rX r^ r? r+ r �getrY r0 r r r r �test_loseConnection@ s z ChannelTests.test_loseConnectionc C �6 t ddd�}t| jj|d� | �t|�| j�� � dS )z� L{SSHChannel.getPeer} returns the same object as the underlying transport's C{getPeer} method returns. �TCPz 192.168.0.1i1� )r& r# N)r r( r r+ r: r �getPeer)r �peerr r r � test_getPeerN � zChannelTests.test_getPeerc C r� )z� L{SSHChannel.getHost} returns the same object as the underlying transport's C{getHost} method returns. r� z 127.0.0.1i90 )r& r" N)r r( r r+ r: r �getHost)r �hostr r r � test_getHostX r� zChannelTests.test_getHostN)r r r r! r/ r2 rC rG rI rJ r_ rd rg ru r| r� r� r� r� r r r r r) ] s" $1 r) )NN)r! �unittestr �zope.interface.verifyr �twisted.conch.sshr �twisted.conch.ssh.addressr �twisted.conch.ssh.servicer �twisted.conch.ssh.transportr �twisted.internetr �twisted.internet.addressr �twisted.test.proto_helpersr �skipTest� ImportError�object�twisted.trial.unittestr r r( r) r r r r �<module> s* � +PK F��Z x��J �J ' __pycache__/test_helper.cpython-310.pycnu �[��� o �bAN � @ s� d dl mZ d dlmZmZmZmZmZmZm Z m Z m Z m Z m Z d dlmZ dZdZG dd� dej�ZG dd � d �ZG d d � d �ZG d d � d ej�ZG dd� dej�ZdS )� )�helper) �BLINK�BOLD�G0�G1�G2�G3�NORMAL� REVERSE_VIDEO� UNDERLINE�modes� privateModes)�unittest�P � c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dES )F� BufferTestsc C s t �� | _| j�� d S �N)r �TerminalBuffer�term�connectionMade��self� r �@/usr/lib/python3/dist-packages/twisted/conch/test/test_helper.py�setUp s zBufferTests.setUpc C sP | � | jjt� | � | jjt� | � | j�� dtd � | � | j�� d� d S )N� � �r r )� assertEqualr �width�WIDTH�height�HEIGHT� __bytes__�reportCursorPositionr r r r �testInitialState s z BufferTests.testInitialStatec C s | � tjdtjdi| jj� dS )z� Verify that only DEC Auto Wrap Mode (DECAWM) and DEC Text Cursor Enable Mode (DECTCEM) are initially in the Set Mode (SM) state. TN)r r � AUTO_WRAP� CURSOR_MODEr r r r r �test_initialPrivateModes$ s �z$BufferTests.test_initialPrivateModesc C �L | j �d� | j �d� | �| j �� d� | j �d� | �| j �� d� dS )zQ C{" "} moves the cursor to the first column in the current row. � � �r* r+ � �r r+ N)r � cursorForward� cursorDownr r$ �insertAtCursorr r r r �test_carriageReturn. � z BufferTests.test_carriageReturnc C s@ | j �d� | �| j �� d� | j �d� | �| j �� d� dS )zV C{" "} moves the cursor to the next row without changing the column. r* )r* r r �r* r N)r r/ r r$ r1 r r r r � test_linefeed8 s zBufferTests.test_linefeedc C r) )z: C{write} transforms C{" "} into C{" "}. r* r+ r, r �r � N)r r/ r0 r r$ �writer r r r � test_newlineA r3 zBufferTests.test_newlinec C sH | j j�� }| j �tjtjg� d|tj< d|tj< | �|| j j� dS )z� Verify that L{helper.TerminalBuffer.setPrivateModes} changes the Set Mode (SM) state to "set" for the private modes it is passed. TN)r r �copy�setPrivateModes�SCROLL�SCREENr �r �expectedr r r �test_setPrivateModesK s z BufferTests.test_setPrivateModesc C sD | j j�� }| j �tjtjg� |tj= |tj= | �|| j j� dS )z� Verify that L{helper.TerminalBuffer.resetPrivateModes} changes the Set Mode (SM) state to "reset" for the private modes it is passed. N)r r r: �resetPrivateModesr& r' r r> r r r �test_resetPrivateModesV s z"BufferTests.test_resetPrivateModesc C sd | j �d� | �| j �� d� | j �� | �| j �� d� | j �t� | �| j �� dtd f� d S )Nr+ r. r6 r r )r r0 r r$ r" r r r r �testCursorDowna s zBufferTests.testCursorDownc C sj | j �d� | �| j �� d� | j �d� | j �d� | �| j �� d� | j �d� | �| j �� d� d S )Nr* r � r )r � rE )r �cursorUpr r$ r0 r r r r � testCursorUpi s zBufferTests.testCursorUpc C sb | j �d� | �| j �� d� | j �d� | �| j �� d� | j �t� | �| j �� tdf� d S )N� )rH r )r7 r r )r r/ r r$ r r r r r �testCursorForwardt s z BufferTests.testCursorForwardc C s� | j �d� | j �d� | �| j �� d� | j �d� | �| j �� d� | j �d� | �| j �� d� | j �d� | �| j �� d� d S )N� rH )� r � )r r r r )r r/ �cursorBackwardr r$ r r r r �testCursorBackward| s z BufferTests.testCursorBackwardc C s$ | j �dd� | �| j �� d� d S )Nr+ � )r+ rO )r �cursorPositionr r$ r r r r �testCursorPositioning� s z!BufferTests.testCursorPositioningc C s6 d}| j �|� | �| j �� |d dtd � d S )N� Hello, world.r rH )r r8 r r# r" �r �sr r r �testSimpleWriting� s &z BufferTests.testSimpleWritingc C sn d}| j �|� | j �t|�� | j �tjg� | j �d� | �| j �� d|dd � d dt d � d S )Ns hello, world.� Hr r rH ) r r8 rM �len� resetModesr �IRMr r# r" rS r r r � testOvertype� s &�zBufferTests.testOvertypec C sf d}| j �|� | j �t|�� | j �tjg� | j �d� | �| j �� d| d dt d � d S )Ns ello, world.rV r rH ) r r8 rM rW �setModesr rY r r# r" rS r r r � testInsert� s �zBufferTests.testInsertc C s^ d}| j �d� | j �d� | j �|� | �| j �� d| j jd | d dtd � d S )NrR r* s r rL )r r0 r/ r8 r r# �fillr" rS r r r �testWritingInTheMiddle� s "�z"BufferTests.testWritingInTheMiddlec C sd d}| j �td � | j �|� | �| j �� |d d� �t�d |dd � d dtd � d S )NrR r* r r+ )r r/ r r8 r r# �rjustr" rS r r r � testWritingWrappedAtEndOfLine� s 0�z)BufferTests.testWritingWrappedAtEndOfLinec C sj | j �� | �| j �� d� | j �t� | �| j �� dtd f� | j �� | �| j �� dtd f� d S )N�r r r r )r �indexr r$ r0 r" r r r r � testIndex� s zBufferTests.testIndexc C sZ | j �� | �| j �� d� | j �d� | �| j �� d� | j �� | �| j �� d� d S )Nr rH �r rH ra )r � reverseIndexr r$ r0 r r r r �testReverseIndex� s z BufferTests.testReverseIndexc C sZ | j �� | �| j �� d� | j �d� | �| j �� d� | j �� | �| j �� d� dS )zm C{nextLine} positions the cursor at the beginning of the row below the current row. ra r* r4 rd N)r �nextLiner r$ r/ r r r r � test_nextLine� s zBufferTests.test_nextLinec C s~ | j �d� | j �d� | �| j �� d� | j �� | j �d� | j �d� | �| j �� d� | j �� | �| j �� d� d S )Nr* rL )rL r* r+ )r7 � )r r0 r/ r r$ � saveCursorrM � restoreCursorr r r r �testSaveCursor� s zBufferTests.testSaveCursorc C s� | j �� | j �d� | j �dd�}| �|d d� | �|d jt� | j �dd�}| �|d d� | �|d jt� | j �� | j �d� | j �dd�}| �|d d� | �|d jt � | j �d d�}| �|d d� | �|d jt� d S ) Ns Hir rV r � is !!rH � !r+ ) r � singleShift2r8 � getCharacterr �charsetr r � singleShift3r �r �chr r r �testSingleShifts� s z BufferTests.testSingleShiftsc C s� d}d}d}| j �d� | j �� | j �d� | j �� | j �d� t}d}|||fD ]5}tt|��D ] }| j �||�}| �|d |||d � � | �|d j |� q3|tkrZt p[t}|d7 }q+d S ) Ns Hellos Worlds Bye!s Hello s World s Bye! r r ) r r8 �shiftOut�shiftInr �rangerW rp r rq r ) r �s1�s2�s3�g�hrT �irt r r r � testShifting� s$ �zBufferTests.testShiftingc C s� | j �tttt� | j �d� | j �t� | j �d� | j �t� | j �d� | j �t� | j �d� | j �dd�}| � |d d� | � |d j � | � |d j � | � |d j � | � |d j� | j �dd�}| � |d d� | �|d j � | �|d j � | �|d j � | �|d j� | j �dd�}| � |d d� | � |d j � | �|d j � | �|d j � | �|d j� | j �dd�}| � |d d� | � |d j � | � |d j � | �|d j � | �|d j� d S ) N� W� X� Y� Zr r rH r+ )r �selectGraphicRenditionr r r r r8 r rp r � assertTrue�bold� underline�blink� reverseVideo� assertFalsers r r r �testGraphicRendition s@ z BufferTests.testGraphicRenditionc C s� d}d}| j �tjtj tjtj � | j �|d � | j �t� | j �|d � t t |��D ]T}| j � |d�}| � |d |||d � � | � |d j t� | �|d j� | �|d j� | �|d j� | �|d j� | � |d jtj� | � |d jtj� q/t t |��D ]T}| j � |d�}| � |d |||d � � | � |d j t� | �|d j� | �|d j� | �|d j� | �|d j� | � |d jtj� | � |d jtj� q�d S )Ns Merry xmass Just kiddingr r r )r r� r � FOREGROUND�RED� BACKGROUND�GREENr8 r rx rW rp r rq r r� r� r� r� r� � foreground� background�WHITE�BLACK)r ry rz r~ rt r r r �testColorAttributes/ s: � �z BufferTests.testColorAttributesc C sr d}d}d}| j �d�|||f�d � | j �dd� | j �� | �| j �� |d d | d dtd � d S )Ns line 1s line 2s line 3r r r7 )r r8 �joinrP � eraseLiner r# r" �r ry rz r{ r r r � testEraseLineO s �zBufferTests.testEraseLinec C sT d}| j �|� | j �d� | j �� | �| j �� |d d� d dtd � d S )NrR r* ���r rH )r r8 rM �eraseToLineEndr r# r" rS r r r �testEraseToLineEnd\ s .z BufferTests.testEraseToLineEndc C s^ d}| j �|� | j �d� | j �� | �| j �� |dd � �t|��d dtd � d S )NrR r* ���r rH ) r r8 rM �eraseToLineBeginningr r# r_ rW r" rS r r r �testEraseToLineBeginningc s ,�z$BufferTests.testEraseToLineBeginningc C s@ | j �d� | j �d� | j �� | �| j �� dtd � d S )Ns Hello world s Goodbye world r r )r r8 � eraseDisplayr r# r" r r r r �testEraseDisplayl s z BufferTests.testEraseDisplayc C sn d}d}| j �d�||df�� | j �dd� | j �� | �| j �� |d |d d� d dtd � d S )N� Hello world� Goodbye worldr � r* r r+ )r r8 r� rP �eraseToDisplayEndr r# r" �r ry rz r r r �testEraseToDisplayEnds s *�z!BufferTests.testEraseToDisplayEndc C sr d}d}| j �d�||f�� | j �dd� | j �� | �| j �� d|dd � �t|�� d dt d � d S )Nr� r� r r* r � r+ ) r r8 r� rP �eraseToDisplayBeginningr r# r_ rW r" r� r r r �testEraseToDisplayBeginning~ s (�z'BufferTests.testEraseToDisplayBeginningc C sh d}d}| j �d�||f�� | j �dd� | j �� | �| j �� |d d | d dtd � d S )Nr� r� r rL r r7 )r r8 r� rP � insertLiner r# r" r� r r r �testLineInsertion� s �z BufferTests.testLineInsertionc C sj d}d}d}| j �d�|||f�� | j �dd� | j �� | �| j �� |d | d dtd � d S )Nr� s Middle wordsr� r rO r r+ )r r8 r� rP � deleteLiner r# r" r� r r r �testLineDeletion� s "�z BufferTests.testLineDeletionN)%�__name__� __module__� __qualname__r r% r( r2 r5 r9 r@ rB rC rG rI rN rQ rU rZ r\ r^ r` rc rf rh rl ru r r� r� r� r� r� r� r� r� r� r� r r r r r sF & r c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zdd � Zd S ) �FakeDelayedCallFc C s" || _ || _|| _|| _|| _d S r )�fs�timeout�f�a�kw)r r� r� r� r� r� r r r �__init__� s zFakeDelayedCall.__init__c C s | j p| j S r )� cancelled�calledr r r r �active� s zFakeDelayedCall.activec C s d| _ d S �NT)r� r r r r �cancel� � zFakeDelayedCall.cancelc C s d| _ | j| ji | j�� d S r� )r� r� r� r� r r r r �call� s zFakeDelayedCall.callN) r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� FakeSchedulerc C s g | _ d S r )�callsr r r r r� � r� zFakeScheduler.__init__c O s" | j �t| ||||�� | j d S )N���)r� �appendr� )r r� r� r� r� r r r � callLater� s zFakeScheduler.callLaterN)r� r� r� r� r� r r r r r� � s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� ExpectTestsc C s t �� | _| j�� t� | _d S r )r �ExpectableBufferr r r� r� r r r r r � s zExpectTests.setUpc C s� g }| j jdd| jd�}|�|j� | j �d� | �|� | j �d� | �|� | �|d � � d� | �t | jj �d� | �| jj d � � � d S )N� hello worldr �r� � schedulers greeting puny earthlings s hello world r ) r �expectr� � addCallbackr� r8 r� r� r �grouprW r� r� �r �result�dr r r �testSimpleString� s z ExpectTests.testSimpleStringc C s� g }| j �d�}|�|j� | �|� | j �d� | �|� | j �d� | �|� | j �d� | �|� | �|d �� d� d S )Nr� � hello s worl� dr ) r r� r� r� r� r8 r� r r� r� r r r �testBrokenUpString� s z ExpectTests.testBrokenUpStringc C s� g }| j �d�}|�|j� | j �d�}|�|j� | �|� | j �d� | �|� | j �d� | �t|�d� | j �d� | �t|�d� | �|d �� d� | �|d �� d� d S )Nr� � worlds hello� r rH r ) r r� r� r� r� r8 r rW r� �r r� �d1�d2r r r � testMultiple� s zExpectTests.testMultiplec C sJ | j �d� g }| j �d�}|�|j� | �|� | �|d �� d� d S )Nr� r )r r8 r� r� r� r� r r� r� r r r �testSynchronous� s zExpectTests.testSynchronousc C s| | j �d� g }| j �d�}|�|j� | j �d�}|�|j� | �t|�d� | �|d �� d� | �|d �� d� d S )Ns goodbye worlds byer� rH r r )r r8 r� r� r� r rW r� r� r r r �testMultipleSynchronous� s z#ExpectTests.testMultipleSynchronousc C s$ | � t|d�� | �|jtj� d S )N�type)r� �hasattrr r� r �ExpectationTimeout)r �resr r r �_cbTestTimeoutFailure s z!ExpectTests._cbTestTimeoutFailurec C s4 | j jdd| jd�}|�| j� | jjd �� d S )Nr� r r� r )r r� r� �addBothr� r� r� )r r� r r r �testTimeoutFailure s z ExpectTests.testTimeoutFailurec C s� | j �d� g }| j jdd| jd�}|�| j� | j �d�}|�|j� | jjd � � | � t |�d� | � |d � � d� d S )Ns not zoomtasticr� r r� s zoomr ) r r8 r� r� r� r� r� r� r� r� r rW r� r� r r r �testOverlappingTimeout s z"ExpectTests.testOverlappingTimeoutN) r� r� r� r r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )�CharacterAttributeTestszG Tests for L{twisted.conch.insults.helper.CharacterAttribute}. c C s� | � t�� t�� � | � t�� tjtd�� | � tjddddtjd�tjddddtjd�� | �t�� tjtd�� | �tjdd�tjdd�� dS )z� L{CharacterAttribute}s must have matching character attribute values (bold, blink, underline, etc) with the same values to be considered equal. )rq TF)r� r� r� r� r� �r� N)r r �CharacterAttributer �BLUE�assertNotEqualr r r r r � test_equality) s4 ������z%CharacterAttributeTests.test_equalityc C sb t �� jdd� | �| jg�}| �t|�d� | �|d d t� d}| �|d d d| � d S ) z~ L{twisted.conch.insults.helper.CharacterAttribute.wantOne} emits a deprecation warning when invoked. Tr� r r �categoryz5twisted.conch.insults.helper._FormattingState.wantOne�messagez#%s was deprecated in Twisted 13.1.0N)r �_FormattingState�wantOne� flushWarnings�test_wantOneDeprecatedr rW �DeprecationWarning)r � warningsShown�deprecatedClassr r r r� N s �z.CharacterAttributeTests.test_wantOneDeprecatedN)r� r� r� �__doc__r� r� r r r r r� $ s %r� N)�twisted.conch.insultsr � twisted.conch.insults.insultsr r r r r r r r r r r � twisted.trialr r r" �TestCaser r� r� r� r� r r r r �<module> s 4 `PK F��Z� ��B/ B/ ( __pycache__/test_default.cpython-310.pycnu �[��� o �bD- � @ s2 d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z dd lmZ dd lmZ dd lmZ d Zd Ze d�rge d�rgddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ndZdZeZ e�!� rudZdZej"�#� s~dZdZej$�#� s�dZdZG dd� de�Z%G d d � d e�Z&dS ) z, Tests for L{twisted.conch.client.default}. � N)�skipIf)� ConchError)�keydata)� nativeString)�FilePath)� requireModule)�platform)�StringTransport)�TestCaseF� � cryptography�pyasn1)�default)�SSHAgentClient)�SSHUserAuthClient)� ConchOptions)�KeyTzBcryptography and PyASN1 required for twisted.conch.client.default.zlgenericAnswers and getPassword does not work on Windows. Should be fixed as part of fixing bug 6409 and 6410z#sys.stdin is not an interactive ttyz$sys.stdout is not an interactive ttyc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z e e e�dd� �Ze e e�dd� �Ze e e�dd� �Ze e e�dd� �ZdS )�SSHUserAuthClientTestszm Tests for L{SSHUserAuthClient}. @type rsaPublic: L{Key} @ivar rsaPublic: A public RSA key. c C sZ t �tj�| _t| �� �| _| j�� | j� d�| _ | j � tj � | j� d�� tj� d S )N�id_rsa� id_rsa.pub) r � fromStringr �publicRSA_openssh� rsaPublicr �mktemp�tmpdir�makedirs�child�rsaFile� setContent�privateRSA_openssh��self� r" �A/usr/lib/python3/dist-packages/twisted/conch/test/test_default.py�setUp: s z SSHUserAuthClientTests.setUpc C sf t dt� d�}t� }t� }|�|� ||_d}|�| j|� | �|� � d| j� � d | d � dS )z� When connected to an agent, L{SSHUserAuthClient} can use it to request signatures of particular data with a particular L{Key}. � userNs Sign heres - s s ) r r r r �makeConnection�keyAgent�signDatar � assertEqual�value�blob)r! �client�agent� transport� cleartextr" r" r# �test_signDataWithAgentB s& �����z-SSHUserAuthClientTests.test_signDataWithAgentc C sJ t � }| j�� g|_|�� }| �|�� � | �|| j� | �|�� � dS )a L{SSHUserAuthClient} looks up public keys from the agent using the L{SSHAgentClient} class. That L{SSHAgentClient.getPublicKey} returns a L{Key} object with one of the public keys in the agent. If no more keys are present, it returns L{None}. N) r r r+ �blobs� getPublicKey� assertTrue�isPublicr) � assertIsNone)r! r- �keyr" r" r# �test_agentGetPublicKeyW s z-SSHUserAuthClientTests.test_agentGetPublicKeyc C sF t � }| jjg|_td|d�}|�� }| �|�� � | �|| j � dS )z� L{SSHUserAuthClient.getPublicKey()} is able to get a public key from the first file described by its options' C{identitys} list, and return the corresponding public L{Key} object. r% N) r r �path� identitysr r2 r3 r4 r) r )r! �optionsr, r6 r" r" r# �test_getPublicKeyFromFilee s z0SSHUserAuthClientTests.test_getPublicKeyFromFilec C sR t � }| jjg|_t� }td|d�}||_|�� }| �|� � � | � || j � dS )z� If an agent is present, but doesn't return a key, L{SSHUserAuthClient.getPublicKey} continue with the normal key lookup. r% N) r r r8 r9 r r r' r2 r3 r4 r) r )r! r: r- r, r6 r" r" r# � test_getPublicKeyAgentFallbackr s z5SSHUserAuthClientTests.test_getPublicKeyAgentFallbackc C s� t � }| j�d��tj� | j�d�}|�tj� | jj|jg|_ | j�d��d� t d|d�}|� � }| � |� � � | �|t�tj�� | �|j| jj|jg� dS )z� If L{keys.Key.fromFile} raises a L{keys.BadKeyError}, the L{SSHUserAuthClient.getPublicKey} tries again to get a public key by calling itself recursively. z id_dsa.pub�id_dsar s not a key!r% N)r r r r r �publicDSA_openssh�privateDSA_opensshr r8 r9 r r2 r3 r4 r) r r � usedFiles)r! r: �dsaFiler, r6 r" r" r# � test_getPublicKeyBadKeyError� s z3SSHUserAuthClientTests.test_getPublicKeyBadKeyErrorc sN t �tj�� t� }�jjg|_td|d�}|� � � �fdd�}|� � � |�S )z� L{SSHUserAuthClient.getPrivateKey} will load a private key from the last used file populated by L{SSHUserAuthClient.getPublicKey}, and return a L{Deferred} which fires with the corresponding private L{Key}. r% Nc � �� | �� � ��| � � d S �N�� assertFalser4 r) �r6 �� rsaPrivater! r" r# �_cbGetPrivateKey� � zCSSHUserAuthClientTests.test_getPrivateKey.<locals>._cbGetPrivateKey) r r r r r r r8 r9 r r2 � getPrivateKey� addCallback)r! r: r, rJ r" rH r# �test_getPrivateKey� s z)SSHUserAuthClientTests.test_getPrivateKeyc s� t �tj��d� �j��jd� d�� t� }�jjg|_ t d|d�}|� � � �fdd�}��fdd �}�� |d |� |� � �|�S ) z� L{SSHUserAuthClient} can get a private key from a file, and return a Deferred called back with a private L{Key} object, even if the key is encrypted. s this is the passphrase�openssh)� passphraser% Nc s �� | d�jj� d�� t� �S )NzEnter passphrase for key 'z': )r) r r8 r ��prompt)rP r! r" r# � _getPassword� s �zISSHUserAuthClientTests.test_getPrivateKeyPassphrase.<locals>._getPasswordc rC rD rE rG rH r" r# rJ � rK zMSSHUserAuthClientTests.test_getPrivateKeyPassphrase.<locals>._cbGetPrivateKeyrS )r r r r r r �toStringr r8 r9 r r2 �patchrL rM )r! r: r, rS rJ r" )rP rI r! r# � test_getPrivateKeyPassphrase� s z3SSHUserAuthClientTests.test_getPrivateKeyPassphrasec s` G dd� d�}t � }td|d�}|d�|_� fdd�}� �tjd|� |�� }|�� jd � |S ) zn Get the password using L{twisted.conch.client.default.SSHUserAuthClient.getPassword} c @ s e Zd Zdd� Zdd� ZdS )z>SSHUserAuthClientTests.test_getPassword.<locals>.FakeTransportc S s | | _ || _d S rD )r. �host)r! rW r" r" r# �__init__� s zGSSHUserAuthClientTests.test_getPassword.<locals>.FakeTransport.__init__c S s | S rD r" r r" r" r# �getPeer� s zFSSHUserAuthClientTests.test_getPassword.<locals>.FakeTransport.getPeerN)�__name__� __module__� __qualname__rX rY r" r" r" r# � FakeTransport� s r] r% Nz 127.0.0.1c � � � | d� dS )Nzuser@127.0.0.1's password: � bad password�r) rQ r r" r# �getpass� � z8SSHUserAuthClientTests.test_getPassword.<locals>.getpassra � bad password) r r r. rU r ra � getPasswordrM r) )r! r] r: r, ra �dr" r r# �test_getPassword� s z'SSHUserAuthClientTests.test_getPasswordc sP t � }td|d�}d� � �fdd�}��tjd|� |�� �}|��jd� |S )z� Get the password using L{twisted.conch.client.default.SSHUserAuthClient.getPassword} using a different prompt. r% Ns Give up your passwordc s �� | t� �� dS )Nr_ )r) r )�p�rR r! r" r# ra � s z>SSHUserAuthClientTests.test_getPasswordPrompt.<locals>.getpassra rc )r r rU r ra rd rM r) )r! r: r, ra re r" rh r# �test_getPasswordPrompt� s z-SSHUserAuthClientTests.test_getPasswordPromptc sh t � }td|d�}dd� }� �tjd|� tjtj��|�d�}|j � ��fdd��}� � |t � dS ) z� Get the password using L{twisted.conch.client.default.SSHUserAuthClient.getPassword} and trigger a {twisted.conch.error import ConchError}. r% Nc S s t d��)NzUser pressed CTRL-C)�KeyboardInterruptrQ r" r" r# ra � s zBSSHUserAuthClientTests.test_getPasswordConchError.<locals>.getpassra � ?c s � � ��gtjtjg� | S rD )r) �sys�stdout�stdin)�fail�r! rn rm r" r# � check_sys s zDSSHUserAuthClientTests.test_getPasswordConchError.<locals>.check_sys) r r rU r ra rl rm rn rd � addErrback� assertFailurer )r! r: r, ra re rq r" rp r# �test_getPasswordConchError� s z1SSHUserAuthClientTests.test_getPasswordConchErrorc sp t � }td|d�}� fdd�}� �tjd|� � fdd�}� �td|� |�d d d d g�}|�� jdd g� |S )zU L{twisted.conch.client.default.SSHUserAuthClient.getGenericAnswers} r% Nc r^ )Nz pass promptra r` rQ r r" r# ra rb z>SSHUserAuthClientTests.test_getGenericAnswers.<locals>.getpassra c r^ )Nzraw_input prompt� raw_inputr` rQ r r" r# ru rb z@SSHUserAuthClientTests.test_getGenericAnswers.<locals>.raw_input�_inputs Names Instruction)s pass promptF)s raw_input promptTru )r r rU r ra �getGenericAnswersrM �assertListEqual)r! r: r, ra ru re r" r r# �test_getGenericAnswers s �z-SSHUserAuthClientTests.test_getGenericAnswersN)rZ r[ r\ �__doc__r$ r0 r7 r; r< rB rN rV r �doSkip� skipReasonrf ri rt ry r" r" r" r# r 2 s$ r c @ s e Zd ZdZdd� Zdd� ZdS )�ConchOptionsParsingz Options parsing. c C s� t � }| �t|jd�}| �d|j� t � }|�d� | �|d dg� |�d� | �|d dg� |�d� | �|d g d�� dS ) z) Specify MAC algorithms. z invalid-maczUnknown mac typez hmac-sha2-512�macss hmac-sha2-512z hmac-sha2-256,hmac-sha1,hmac-md5)s hmac-sha2-256s hmac-sha1s hmac-md5N)r � assertRaises� SystemExit�opt_macs�assertIn�coder) �r! �opts�er" r" r# � test_macs* � z ConchOptionsParsing.test_macsc C s� t � }| �t|jd�}| �d|j� t � }|�d� | �|d dg� |�d� | �|d dg� |�d� | �|d ddg� dS ) z. Specify host key algorithms. z invalid-keyzUnknown host key typezssh-rsazhost-key-algorithmss ssh-rsas ssh-dsszssh-rsa,ssh-dssN)r r r� �opt_host_key_algorithmsr� r� r) r� r" r" r# �test_host_key_algorithms9 r� z,ConchOptionsParsing.test_host_key_algorithmsN)rZ r[ r\ rz r� r� r" r" r" r# r} % s r} )'rz rl �unittestr �twisted.conch.errorr �twisted.conch.testr �twisted.python.compatr �twisted.python.filepathr �twisted.python.reflectr �twisted.python.runtimer �twisted.test.proto_helpersr �twisted.trial.unittestr r{ r| �twisted.conch.clientr �twisted.conch.client.agentr � twisted.conch.client.defaultr � twisted.conch.client.optionsr �twisted.conch.ssh.keysr �skip� isWindowsrn �isattyrm r r} r" r" r" r# �<module> sF � tPK F��Z��mB B % __pycache__/test_text.cpython-310.pycnu �[��� o �b� � @ s: d dl mZ d dlmZ d dlmZ G dd� dej�ZdS )� )�text)� attributes)�unittestc @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )�FormattedTextTestsz. Tests for assembling formatted text. c C s | � t�tjd �d� dS )zu Using no formatting attributes produces no VT102 control sequences in the flattened output. � Hello, world.N)� assertEqualr �assembleFormattedText�A�normal��self� r �>/usr/lib/python3/dist-packages/twisted/conch/test/test_text.py� test_trivial � �z FormattedTextTests.test_trivialc C � | � t�tjd �d� dS )z� The bold formatting attribute, L{A.bold}, emits the VT102 control sequence to enable bold when flattened. r z[1mHello, world.N)r r r r �boldr r r r � test_bold r z FormattedTextTests.test_boldc C r )z� The underline formatting attribute, L{A.underline}, emits the VT102 control sequence to enable underlining when flattened. r z[4mHello, world.N)r r r r � underliner r r r �test_underline! � �z!FormattedTextTests.test_underlinec C r )z� The blink formatting attribute, L{A.blink}, emits the VT102 control sequence to enable blinking when flattened. r z[5mHello, world.N)r r r r �blinkr r r r � test_blink+ r z FormattedTextTests.test_blinkc C r )z� The reverse-video formatting attribute, L{A.reverseVideo}, emits the VT102 control sequence to enable reversed video when flattened. r z[7mHello, world.N)r r r r � reverseVideor r r r �test_reverseVideo4 r z$FormattedTextTests.test_reverseVideoc C s0 | � t�tjtjdtjd df �d� dS )z� Formatting attributes prefixed with a minus (C{-}) temporarily disable the prefixed attribute, emitting no VT102 control sequence to enable it in the flattened output. �Helloz world�.z [1;5mHello[0;5m world[1;5m.N)r r r r r r r r r r � test_minus> s ��z FormattedTextTests.test_minusc C �0 | � t�tjtjjd tjjd f �d� dS )z� The foreground color formatting attribute, L{A.fg}, emits the VT102 control sequence to set the selected foreground color when flattened. �Hello, �world!z[31mHello, [32mworld!N)r r r r r �fg�red�greenr r r r �test_foregroundK � ��z"FormattedTextTests.test_foregroundc C r )z� The background color formatting attribute, L{A.bg}, emits the VT102 control sequence to set the selected background color when flattened. r r z[41mHello, [42mworld!N)r r r r r �bgr" r# r r r r �test_backgroundW r% z"FormattedTextTests.test_backgroundc C sn | � | jg�}| �t|�d� tj | � | jg�}| �t|�d� | �|d d t� | �|d d d� dS )zv L{twisted.conch.insults.text.flatten} emits a deprecation warning when imported or accessed. r � �category�messagez�twisted.conch.insults.text.flatten was deprecated in Twisted 13.1.0: Use twisted.conch.insults.text.assembleFormattedText instead.N)� flushWarnings�test_flattenDeprecatedr �lenr �flatten�DeprecationWarning)r � warningsShownr r r r, c s �z)FormattedTextTests.test_flattenDeprecatedN) �__name__� __module__� __qualname__�__doc__r r r r r r r$ r' r, r r r r r s r N) �twisted.conch.insultsr �twisted.conch.insults.textr r � twisted.trialr �TestCaser r r r r �<module> s PK F��Z�� l l $ __pycache__/test_tap.cpython-310.pycnu �[��� o �b� � @ s� d Z ddlmZ ddlmZ ddlmZmZmZ ddl m Z ddl m Z e d�Z e d�Ze d �Ze rBerBerBdd lmZ dd lmZ G d d � d e �ZdS )z! Tests for L{twisted.conch.tap}. � )�StreamServerEndpointService)�error)�ISSHPrivateKey�IUsernamePassword�UsernamePassword)� requireModule)�TestCase� cryptography�pyasn1ztwisted.conch.unix)�tap)�OpenSSHFactoryc @ sl e Zd ZdZes dZesdZesdZdZdd� Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� ZdS )�MakeServiceTestsz' Tests for L{tap.makeService}. z can't run without cryptographyzCannot run without PyASN1z can't run on non-posix computers)s iamusers thisispasswordc C sV | � � | _t| jd��}|�d�| j�� W d � n1 s w Y t�� | _dS )z/ Create a file with two users. zwb+� :N) �mktemp�filename�open�write�join�usernamePasswordr �Options�options)�self�f� r �=/usr/lib/python3/dist-packages/twisted/conch/test/test_tap.py�setUp( s �zMakeServiceTests.setUpc C s@ t �� }t �|�}| �|t� | �|jjd� | �|jt � dS )z� L{tap.makeService} returns a L{StreamServerEndpointService} instance running on TCP port 22, and the linked protocol factory is an instance of L{OpenSSHFactory}. � N) r r � makeService�assertIsInstancer � assertEqual�endpoint�_port�factoryr )r �config�servicer r r � test_basic1 s zMakeServiceTests.test_basicc C sN d}| � t| jd d� | � t| jd d� | �|t| jd �d|f � dS )z� Make sure that if the C{--auth} command-line option is not passed, the default checkers are (for backwards compatibility): SSH and UNIX � �credInterfacesz)SSH should be one of the default checkersz*UNIX should be one of the default checkers� credCheckersz&There should be %d checkers by defaultN)�assertInr r r r �len)r � numCheckersr r r �test_defaultAuths= s �� �z"MakeServiceTests.test_defaultAuthsc C s0 | j �dd| j g� | �t| j d �d� dS )z� The C{--auth} command-line option will add a checker to the list of checkers, and it should be the only auth checker �--auth�file:r( � N�r � parseOptionsr r r* �r r r r �test_authAddedT s z MakeServiceTests.test_authAddedc C s4 | j �dd| j ddg� | �t| j d �d� dS )z� Multiple C{--auth} command-line options will add all checkers specified to the list ofcheckers, and there should only be the specified auth checkers (no default checkers). r- r. z memory:testuser:testpasswordr( r& Nr0 r2 r r r �test_multipleAuthAdded\ s ��z'MakeServiceTests.test_multipleAuthAddedc C sH | j �dd| j g� | j d d }t| jd d�}| �|�|�tj�S )z� The checker created by the C{--auth} command-line option returns a L{Deferred} that fails with L{UnauthorizedLogin} when presented with credentials that are unknown to that checker. r- r. r( ���r �fake) r r1 r r r � assertFailure�requestAvatarIdr �UnauthorizedLogin)r �checker�invalidr r r �test_authFailurel s �z!MakeServiceTests.test_authFailurec sP �j �dd�j g� �j d d }t�j� � |�� �}� �fdd�}|�|�S )z� The checker created by the C{--auth} command-line option returns a L{Deferred} that returns the avatar id when presented with credentials that are known to that checker. r- r. r( r5 c s �� | � j� d S )N)r �username)r= ��correctr r r � checkSuccess� s z7MakeServiceTests.test_authSuccess.<locals>.checkSuccess)r r1 r r r r8 � addCallback)r r: �dr@ r r> r �test_authSuccessz s z!MakeServiceTests.test_authSuccessc C s8 t �� }t �|�}|jj}| �t|j�� �t t h� dS )z� The L{OpenSSHFactory} built by L{tap.makeService} has a portal with L{ISSHPrivateKey} and L{IUsernamePassword} interfaces registered as checkers. N) r r r r" �portalr �set�checkers�keysr r )r r# r$ rD r r r � test_checkers� s �z MakeServiceTests.test_checkersN)�__name__� __module__� __qualname__�__doc__r �skipr �unixr r r% r, r3 r4 r< rC rH r r r r r s" r N)rL � twisted.application.internetr � twisted.credr �twisted.cred.credentialsr r r �twisted.python.reflectr �twisted.trial.unittestr r r rN � twisted.conchr �$twisted.conch.openssh_compat.factoryr r r r r r �<module> s PK F��ZSx�6� 6� % __pycache__/test_keys.cpython-310.pycnu �[��� o �b7� � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddl mZ e d �Zedu r8d Ze d �ZerVerVdd lmZ dd lmZmZmZ e� �� ZndZdd� ZG dd� dej�Z G dd� dej�Z dS )z& Tests for L{twisted.conch.ssh.keys}. � N)�dedent)�keydata)� randbytes)�FilePath)� requireModule)�unittest� cryptographyz Cannot run without cryptography.�pyasn1)�default_backend)�common�keys�sexpyFc C s t sd| _| S )Nz$ed25519 not supported on this system)�ED25519_SUPPORTED�skip)�f� r �>/usr/lib/python3/dist-packages/twisted/conch/test/test_keys.py�skipWithoutEd25519$ s r c @ s� e Zd Zedu r eZedu rdZdd� Zdd� Zdd� Z d d � Z d d � Z d d� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%dAdB� Z&dCdD� Z'dEdF� Z(dGdH� Z)dIdJ� Z*e+dKdL� �Z,dMdN� Z-dOdP� Z.dQdR� Z/dSdT� Z0e+dUdV� �Z1dWdX� Z2dYdZ� Z3d[d\� Z4e+d]d^� �Z5d_d`� Z6dadb� Z7dcdd� Z8dedf� Z9e+dgdh� �Z:didj� Z;dkdl� Z<dmdn� Z=dodp� Z>dqdr� Z?dsdt� Z@dudv� ZAe+dwdx� �ZBe+dydz� �ZCd{d|� ZDd}d~� ZEdd�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLe+d�d�� �ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTe+d�d�� �ZUe+d�d�� �ZVdS )��KeyTestsNzCannot run without PyASN1c C s� t jjtjd tjd tjd tjd tjd tjd d�j| _t jjtjd tjd tjd tjd tjd d �j| _ t jj tj d tj d tj d tj d d�j| _ t jj tj d tj d tj d tj d d�j| _t jj tjd tjd tjd tjd d�j| _tr�t jjtjd tjd d�j| _d| _d| _| �tddd� � | �� | _t| jd��}|� tj � W d � d S 1 s�w Y d S )N�n�e�d�p�q�u)r r r r r r �y�g�x)r r r r r � privateValue�curve)r r r r �a�k)r r! s ssh-rsa ~Y��W�pu@�1�S�O�E�/ x ��G3��U �������� d� ��,'= Qg�x�j�I�ޯb�@�r ΣzM1q5� {wjKN��C��i����}�RT� DJH �l�[�������|�^n���ъ�aY�B��Ǿ7� 9j�.������ 9p�?�N��/ ) >Aµj�UWԮ�B�t�y����fˆ�k���u �N�ɓY�.V�L44`CǦ6��Y���%� 4^�rY� �.�%H� �A�s7 ssh-dss (?��;�TFA�� �E@4,d� ٮ�[�-�nQ�f���jrg�p<� secureRandomc S s d| S )N� �r )r r r r �<lambda>k s z KeyTests.setUp.<locals>.<lambda>�wb) r �Key�_fromRSAComponentsr �RSAData� _keyObject�rsaObj�_fromDSAComponents�DSAData�dsaObj�_fromECComponents�ECDatanistp256�ecObj�ECDatanistp384�ecObj384�ECDatanistp521�ecObj521r �_fromEd25519Components� Ed25519Data� ed25519Obj� rsaSignature� dsaSignature�patchr �mktemp�keyFile�open�write�privateRSA_lsh)�selfr r r r �setUp1 st �������������� "�zKeyTests.setUpc C s t �| j� d S �N)�os�unlinkr< �r@ r r r �tearDownp s zKeyTests.tearDownc C s� | � t�| j��� d� | � t�| j��� d� | � t�| j��� d� | � t�| j��� d� | � t�| j��� d� t rL| � t�| j ��� d� dS dS )zU The L{keys.Key.size} method returns the size of key object in bits. i � � i� i N) � assertEqualr r& r* �sizer- r0 r2 r4 r r7 rE r r r � test_sizes s �zKeyTests.test_sizec C s� | � tj�tj�d� | � tj�tj�d� | � tj�tj�d� tr.| � tj�tj �d� | � tj�tj �d� | � tj�tj �d� | � tj�tj �d� | � tj�tj �d� | � tj�tj�d� | � tj�tj�d� tr}| � tj�tj�d� | � tj�tj�d� | � tj�tj�d� | � tj�tj�d� | � tj�tj�d� | � tj�tj�d� | � tj�tj�d� | � tj�d�d� | � tj�d�d� | � tj�d �d � d S ) z_ Test that the _guessStringType method guesses string types correctly. �public_openssh�private_openssh� public_lsh� private_lsh�agentv3s ssh-rsa �blobs ssh-dss s not a keyN)rI r r& �_guessStringTyper �publicRSA_openssh�publicDSA_openssh�publicECDSA_opensshr �publicEd25519_openssh�privateRSA_openssh�privateRSA_openssh_new�privateDSA_openssh�privateDSA_openssh_new�privateECDSA_openssh�privateECDSA_openssh_new�privateEd25519_openssh_new� publicRSA_lsh� publicDSA_lshr? �privateDSA_lsh�privateRSA_agentv3�privateDSA_agentv3rE r r r �test__guessStringType sz ��� ������ � ����� � �z KeyTests.test__guessStringTypec C s� t j�tj�}t j�tj�}| �|�� |�� � t j�tj�}t j�tj �}| �|�� |�� � t j�tj �}t j�tj �}| �|�� |�� � t rdt j�tj �}t j�tj�}| �|�� |�� � dS dS )zn The L{keys.Key.public} method returns a public key for both public and private keys. N)r r& � fromStringr rW rS rI �publicrY rT r[ rU r r] rV ) r@ � privateRSAKey� publicRSAKey� privateDSAKey� publicDSAKey�privateECDSAKey�publicECDSAKey�privateEd25519Key�publicEd25519Keyr r r � test_public� s �zKeyTests.test_publicc C s� t j�tj�}t j�tj�}t j�tj�}| �|�� � � � | � |� � � | �|�� � � � | � |� � � | �|�� � � � | � |� � � t r`t j�tj �}| �|�� � � � | � |� � � dS dS )zg The L{keys.Key.isPublic} method returns True for public keys otherwise False. N) r r& rd r rW rY r[ � assertTruere �isPublic� assertFalser r] )r@ �rsaKey�dsaKey�ecdsaKey� ed25519Keyr r r � test_isPublic� s �zKeyTests.test_isPublicc C s | � |||� | �|||� d S rB )�_testPublicFromString�_testPrivateFromString)r@ re �private�type�datar r r � _testPublicPrivateFromString� s z%KeyTests._testPublicPrivateFromStringc C sT t j�|�}| �|�� � | �|�� |� |�� �� D ] \}}| �|| |� qd S rB ) r r& rd ro rp rI rz r{ �items)r@ re rz r{ � publicKeyr! �vr r r rw � s �z KeyTests._testPublicFromStringc C sT t j�|�}| �|�� � | �|�� |� |�� D ]\}}| �|�� | |� qd S rB ) r r& rd rq rp rI rz r} r{ )r@ ry rz r{ � privateKeyr! r r r r rx � s �z KeyTests._testPrivateFromStringc C s� | � tjtjdtj� | � tjtjdtj� | �t j j tj dd�t j � tj�� | �t j � tj �t j � tj�� | � tjtjdtj� trR| � tjtjdtj� dS dS )zN Test that keys are correctly generated from OpenSSH strings. �EC�RSA� encrypted�� passphrase�DSA�Ed25519N)r| r rU r[ r/ rS rW r( rI r r& rd � privateRSA_openssh_encrypted� privateRSA_openssh_alternaterT rY r, r rV r] r6 rE r r r �test_fromOpenSSH sH ��� � ����zKeyTests.test_fromOpenSSHc C s d}| � tjtjj|d� dS )z. Tests for invalid key types. s -----BEGIN FOO PRIVATE KEY----- MIGkAgEBBDAtAi7I8j73WCX20qUM5hhHwHuFzYWYYILs2Sh8UZ+awNkARZ/Fu2LU LLl5RtOQpbWgBwYFK4EEACKhZANiAATU17sA9P5FRwSknKcFsjjsk0+E3CeXPYX0 Tk/M0HK3PpWQWgrO8JdRHP9eFE9O/23P8BumwFt7F/AvPlCzVd35VfraFT0o4cCW G0RqpQ+np31aKmeJshkcYALEchnU+tQ= -----END EC PRIVATE KEY-----N)� assertRaisesr � BadKeyErrorr& �_fromString_PRIVATE_OPENSSH�r@ �badKeyr r r �test_fromOpenSSHErrors) s �z KeyTests.test_fromOpenSSHErrorsc C s( d}| � tj�|�tj�|d �� dS )zP If key strings have trailing whitespace, it should be ignored. � -----BEGIN DSA PRIVATE KEY----- MIIBuwIBAAKBgQDylESNuc61jq2yatCzZbenlr9llG+p9LhIpOLUbXhhHcwC6hrh EZIdCKqTO0USLrGoP5uS9UHAUoeN62Z0KXXWTwOWGEQn/syyPzNJtnBorHpNUT9D Qzwl1yUa53NNgEctpo4NoEFOx8PuU6iFLyvgHCjNn2MsuGuzkZm7sI9ZpQIVAJiR 9dPc08KLdpJyRxz8T74b4FQRAoGAGBc4Z5Y6R/HZi7AYM/iNOM8su6hrk8ypkBwR a3Dbhzk97fuV3SF1SDrcQu4zF7c4CtH609N5nfZs2SUjLLGPWln83Ysb8qhh55Em AcHXuROrHS/sDsnqu8FQp86MaudrqMExCOYyVPE7jaBWW+/JWFbKCxmgOCSdViUJ esJpBFsCgYEA7+jtVvSt9yrwsS/YU1QGP5wRAiDYB+T5cK4HytzAqJKRdC5qS4zf C7R0eKcDHHLMYO39aPnCwXjscisnInEhYGNblTDyPyiyNxAOXuC8x7luTmwzMbNJ /ow0IqSj0VF72VJN9uSoPpFd4lLT0zN8v42RWja0M8ohWNf+YNJluPgCFE0PT4Vm SUrCyZXsNh6VXwjs3gKQ -----END DSA PRIVATE KEY-----� N)rI r r& rd �r@ �privateDSADatar r r � test_fromOpenSSH_with_whitespace7 s �z)KeyTests.test_fromOpenSSH_with_whitespacec C sH t jjtjdd�}| �|�� d� t jjtjd dd�}| �||� dS )z� Newer versions of OpenSSH generate encrypted keys which have a longer IV than the older versions. These newer keys are also loaded. s testxpr� r� r� N)r r& rd r � privateRSA_openssh_encrypted_aesrI rz )r@ �key�key2r r r �test_fromNewerOpenSSHN s � �z KeyTests.test_fromNewerOpenSSHc C s� t jt jft jt jft jt jft jt jft j t j ffD ]\}}| � t j �|�t j �|�� q| � t j jt jdd�t j jt jdd�� dS )z� OpenSSH 6.5 introduced a newer "openssh-key-v1" private key format (made the default in OpenSSH 7.8). Loading keys in this format produces identical results to loading the same keys in the old PEM-based format. r� r� N)r rW rX rY rZ r[ r\ �privateECDSA_openssh384�privateECDSA_openssh384_new�privateECDSA_openssh521�privateECDSA_openssh521_newrI r r& rd � privateRSA_openssh_encrypted_newr� )r@ �old�newr r r �test_fromOpenSSH_v1_format\ s � ���z#KeyTests.test_fromOpenSSH_v1_formatc C s, d}| � tj�|�tj�|�dd��� dS )zp Test that keys are correctly generated from OpenSSH strings with Windows line endings. r� r� s N)rI r r& rd �replacer� r r r �%test_fromOpenSSH_windows_line_endingst s �z.KeyTests.test_fromOpenSSH_windows_line_endingsc C s@ t �ddddgggg�}| jtjtjjdt�|� d d� dS ) za C{BadKeyError} exception is raised when public key has an unknown type. s public-key� bad-key� p� 2� {� })r{ N) r �packr� r r� r& rd �base64� b64encode�r@ �sexpr r r �!test_fromLSHPublicUnsupportedType� s �z*KeyTests.test_fromLSHPublicUnsupportedTypec C s0 t �ddddgggg�}| �tjtjj|� dS )zb C{BadKeyError} exception is raised when private key has an unknown type. s private-keyr� r� r� N)r r� r� r r� r& rd r� r r r �"test_fromLSHPrivateUnsupportedType� s �z+KeyTests.test_fromLSHPrivateUnsupportedTypec C � | � tjtjdtj� dS )zR RSA public and private keys can be generated from a LSH strings. r� N)r| r r^ r? r( rE r r r �test_fromLSHRSA� � �zKeyTests.test_fromLSHRSAc C r� )zH DSA public and private key can be generated from LSHs. r� N)r| r r_ r` r, rE r r r �test_fromLSHDSA� r� zKeyTests.test_fromLSHDSAc C s< | � tjdtj� | � tjdtj� | �tjtj j d� dS )zO Test that keys are correctly generated from Agent v3 strings. r� r� s$ ssh-foo N) rx r ra r( rb r, r� r r� r& rd rE r r r �test_fromAgentv3� s �zKeyTests.test_fromAgentv3c C sR t �| j�}|jdd�� d�}| �t jj|dd�|� | jt jt jj|dd� dS )zf L{keys.Key.fromString} applies Normalization Form KC to Unicode passphrases. �openssh� verschlüsseltr� � verschlüsselt� unassigned N) r r& r* �toString�encoderI rd r� � PassphraseNormalizationError�r@ r� �key_datar r r �*test_fromStringNormalizesUnicodePassphrase� s � �z3KeyTests.test_fromStringNormalizesUnicodePassphrasec C s� | � tjtjjd� | � tjtjjdd� | j tjtjjtjdd� | � tjtjjt�| j�j ddd�� | � tjtjjd� | j tjtjjdd d� | j tjtjjd d d� | j tjtjjd d d� | j tjtjjd d d� d S )zW keys.Key.fromString should raise BadKeyError when the key is invalid. � �bad_types unencryptedr� r� r� s -----BEGIN RSA KEY----- wA== s� -----BEGIN ENCRYPTED RSA KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: weird type 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----� encrypteds� -----BEGIN ENCRYPTED RSA KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: FOO-123-BAR,01234567 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----s� -----BEGIN ENCRYPTED RSA KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: AES-128-CBC,01234 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----s� -----BEGIN ENCRYPTED RSA KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: DES-EDE3-CBC,01234 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----N) r� r r� r& rd r r^ �EncryptedKeyErrorr* r� rE r r r �test_fromStringErrors� sP �� � �$ �$ �$ �z KeyTests.test_fromStringErrorsc C sV | � tj�| j�tj�tj�� | �tj tjj| jd� | jtj tjj| jdd� dS )z5 Test that fromFile works correctly. r� � unencryptedr� N) rI r r& �fromFiler< rd r r? r� r� rE r r r � test_fromFile� s � �zKeyTests.test_fromFilec C s. t jjddd�j}t �|�}| �|j|� dS )zJ Test that the PublicKey object is initialized correctly. � � �r r N)r r& r' r) rI )r@ �objr� r r r � test_init� s zKeyTests.test_initc C s� t �| j�}t �| j�}t �t jjddd�j�}t �| j�}| �||k� | �||k� | �||k� | �|tk� | �|dk� dS )z? Test that Key objects are compared correctly. r� r� r� N) r r& r* r' r) r- ro rq �object�r@ �rsa1�rsa2�rsa3�dsar r r � test_equal� � zKeyTests.test_equalc C s� t �| j�}t �| j�}t �t jjddd�j�}t �| j�}| �||k� | �||k� | �||k� | �|tk� | �|dk� dS )zC Test that Key objects are not-compared correctly. r� r� r� N) r r& r* r' r) r- rq ro r� r� r r r � test_notEqual� r� zKeyTests.test_notEqualc C s t �d�}| �t|j� dS )zO The L{keys.Key.data} method raises RuntimeError for bad keys. r� N)r r& r� � RuntimeErrorr{ r� r r r �test_dataError� s zKeyTests.test_dataErrorc C s4 | � t�| j��� d� | � t�| j��� d� dS )z� Test that the fingerprint method returns fingerprint in L{FingerprintFormats.MD5-HEX} format by default. �/85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:da�/63:15:b3:0e:e6:4f:50:de:91:48:3d:01:6b:b3:13:c1N)rI r r& r* � fingerprintr- rE r r r �test_fingerprintdefault� s ��z KeyTests.test_fingerprintdefaultc C �@ | � t�| j��tjj�d� | � t�| j��tjj�d� dS )z� fingerprint method generates key fingerprint in L{FingerprintFormats.MD5-HEX} format if explicitly specified. r� r� N)rI r r& r* r� �FingerprintFormats�MD5_HEXr- rE r r r �test_fingerprint_md5_hex� � ��z!KeyTests.test_fingerprint_md5_hexc C r� )z� fingerprint method generates key fingerprint in L{FingerprintFormats.SHA256-BASE64} format if explicitly specified. z,FBTCOoknq0mHy+kpfnY9tDdcAJuWtCpuQMaV3EsvbUI=z,Wz5o2YbKyxOEcJn1au/UaALSVruUzfz0vaLI1xiIGyY=N)rI r r& r* r� r� � SHA256_BASE64r- rE r r r �test_fingerprintsha256� r� z KeyTests.test_fingerprintsha256c C sV | � tj��}t�| j��d� W d � n1 sw Y | �d|jjd � dS )zk A C{BadFingerPrintFormat} error is raised when unsupported formats are requested. z sha256-baseNz+Unsupported fingerprint format: sha256-baser ) r� r �BadFingerPrintFormatr& r* r� rI � exception�args)r@ �emr r r �test_fingerprintBadFormat� s � �z"KeyTests.test_fingerprintBadFormatc C s | � t�| j��� d� | � t�| j��� d� | � t�| j��� d� | � t�| j��� d� | � t�| j��� d� | � t�| j��� tj d � t re| � t�| j ��� d� | � t�| j ��� d� | � t t�d �j� | � t t�d �j� | � t t�| �j� | � t t�| �j� d S ) zS Test that the type method returns the correct type for an object. r� � ssh-rsar� � ssh-dssr� r r� � ssh-ed25519N)rI r r& r* rz �sshTyper- r0 r r/ r r7 r� r� rE r r r � test_type� s �zKeyTests.test_typec C �" t �d�}| �tjtjj|� dS )ze A C{BadKeyError} error is raised whey the blob has an unsupported key type. � ssh-badN)r �NSr� r r� r& rd �r@ �badBlobr r r � test_fromBlobUnsupportedType � z%KeyTests.test_fromBlobUnsupportedTypec C sj t jd t jd d�}t�d�t�|d � t�|d � }tj�|�}| �|� � � | � ||� � � dS )zQ A public RSA key is correctly generated from a public key blob. r r r� r� N) r r( r r� �MPr r& rd ro rp rI r{ )r@ � rsaPublicData�rsaBlobrr r r r �test_fromBlobRSA s � � �� zKeyTests.test_fromBlobRSAc C s� t jd t jd t jd t jd d�}t�d�t�|d � t�|d � t�|d � t�|d � }tj�|�}| �|� � � | � ||� � � dS )zQ A public DSA key is correctly generated from a public key blob. r r r r )r r r r r� N) r r, r r� r� r r& rd ro rp rI r{ )r@ � dsaPublicData�dsaBlobrs r r r �test_fromBlobDSA s$ � � � � �� zKeyTests.test_fromBlobDSAc C s� ddl m} tjd tjd tjd d�}t�|d �t�|d dd� � t�d |�|d d � |�|d d � � }tj� |�}| � |� � � | � ||� � � dS ) zA Key.fromString generates ECDSA keys from blobs. r ��utilsr r r �r r r ���N� � )r r� r r/ r r� � int_to_bytesr r& rd ro rp rI r{ )r@ r� � ecPublicData�ecblob�eckeyr r r �test_fromBlobECDSA5 s( � ������ zKeyTests.test_fromBlobECDSAc C sT dt jd i}t�d�t�|d � }tj�|�}| �|�� � | � ||� � � dS )zU A public Ed25519 key is correctly generated from a public key blob. r r� N) r r6 r r� r r& rd ro rp rI r{ )r@ �ed25519PublicData� ed25519Blobru r r r �test_fromBlobEd25519O s � z KeyTests.test_fromBlobEd25519c C r� )zh C{BadKeyError} is raised when loading a private blob with an unsupported type. r� N)r r� r� r r� r& �_fromString_PRIVATE_BLOBr� r r r �#test_fromPrivateBlobUnsupportedType_ r� z,KeyTests.test_fromPrivateBlobUnsupportedTypec C s� t �d�t �tjd � t �tjd � t �tjd � t �tjd � t �tjd � t �tjd � }tj�|�}| �|� � � | � tj|� � � | � |tj�|� � �� dS ) zS A private RSA key is correctly generated from a private key blob. r� r r r r r r N) r r� r� r r( r r& r rq rp rI r{ � privateBlob)r@ r� rr r r r �test_fromPrivateBlobRSAh s( ������� �z KeyTests.test_fromPrivateBlobRSAc C s� t �d�t �tjd � t �tjd � t �tjd � t �tjd � t �tjd � }tj�|�}| �|� � � | � tj|� � � | � |tj�|� � �� dS )zS A private DSA key is correctly generated from a private key blob. r� r r r r r N) r r� r� r r, r r& r rq rp rI r{ r )r@ r� rs r r r �test_fromPrivateBlobDSA~ s$ ������ �z KeyTests.test_fromPrivateBlobDSAc C s� ddl m} ddlm} |jtjd tjd |�� d�}t� tjd �t� tjd dd � � t� |� t � �� |j j|jj�� t�tjd � }tj�|�}| �|�� � | �tj|�� � | �|tj�|�� �� d S ) zR A private EC key is correctly generated from a private key blob. r �� serialization)�ecr r r� r r� Nr )� cryptography.hazmat.primitivesr �)cryptography.hazmat.primitives.asymmetricr �EllipticCurvePublicNumbersr r/ � SECP256R1r r� � public_keyr � public_bytes�Encoding�X962� PublicFormat�UncompressedPointr� r r& r rq rp rI r{ r )r@ r r � publicNumbersr r r r r �test_fromPrivateBlobECDSA� s0 �� ����� z"KeyTests.test_fromPrivateBlobECDSAc C s| t �d�t �tjd � t �tjd tjd � }tj�|�}| �|�� � | � tj|� � � | � |tj�|� � �� dS )zW A private Ed25519 key is correctly generated from a private key blob. r� r r! N) r r� r r6 r r& r rq rp rI r{ r )r@ r ru r r r �test_fromPrivateBlobEd25519� s ��� �z$KeyTests.test_fromPrivateBlobEd25519c C sJ | � t�| j��� t�d�t�| j�� j j � t�| j�� j j � � dS )zL Return the over-the-wire SSH format of the RSA public key. r� N) rI r r& r* rQ r r� r� �private_numbers�public_numbersr r rE r r r � test_blobRSA� s ���zKeyTests.test_blobRSAc C sd | j �� j}| �t�| j ��� t�d�t� |j j � t� |j j � t� |j j � t� |j� � dS )zL Return the over-the-wire SSH format of the DSA public key. r� N)r- r r rI r r& rQ r r� r� �parameter_numbersr r r r �r@ r r r r � test_blobDSA� s � � � ��zKeyTests.test_blobDSAc C s� ddl m} | jjjd d }| �t�| j��� t � t j d �t � t j d dd� � t � d|� | j�� jj|� |� | j�� jj|� � � dS ) zK Return the over-the-wire SSH format of the EC public key. r r� � � r r� Nr� )r r� r0 r �key_sizerI r r& rQ r r� r r/ r r r r r )r@ r� � byteLengthr r r � test_blobEC� s( ��������zKeyTests.test_blobECc C sR ddl m} | j�� �|jj|jj�}| �t � | j�� � t � d�t � |� � dS )zP Return the over-the-wire SSH format of the Ed25519 public key. r r r� N)r r r7 r r r �Rawr rI r r& rQ r r� )r@ r � publicBytesr r r �test_blobEd25519� s ��zKeyTests.test_blobEd25519c C � t �d�}| �t|j� dS )zx C{RuntimeError} is raised when the blob is requested for a Key which is not wrapping anything. N)r r& r� r� rQ r� r r r �test_blobNoKey s zKeyTests.test_blobNoKeyc C sx | j �� }| �t�| j ��� t�d�t�|j j � t�|j j � t�|j � t�|j � t�|j� t�|j� � dS )zn L{keys.Key.privateBlob} returns the SSH protocol-level format of an RSA private key. r� N)r* r rI r r& r r r� r� r r r r �iqmpr r )r@ �numbersr r r �test_privateBlobRSA s" � � � � � ��z KeyTests.test_privateBlobRSAc C sv | j �� j}| �t�| j ��� t�d�t� |j j � t� |j j � t� |j j � t� |j� t� | j �� j� � dS )zm L{keys.Key.privateBlob} returns the SSH protocol-level format of a DSA private key. r� N)r- r r rI r r& r r r� r� r r r r r r r! r r r �test_privateBlobDSA s � � � ���z KeyTests.test_privateBlobDSAc C s~ ddl m} | �t�| j��� t�t j d �t�t j d dd� � t�| j� � � |j j|jj�� t�| j�� j� � dS )zj L{keys.Key.privateBlob} returns the SSH ptotocol-level format of EC private key. r r r r� N)r r rI r r& r0 r r r� r r/ r r r r r r r� r � private_value)r@ r r r r �test_privateBlobEC/ s � �����zKeyTests.test_privateBlobECc C s| ddl m} | j�� �|jj|jj�}| j�|jj|j j|� � �}| � t � | j��� t�d�t�|� t�|| � � dS )zr L{keys.Key.privateBlob} returns the SSH protocol-level format of an Ed25519 private key. r r r� N)r r r7 r r r r( r � private_bytes� PrivateFormat� NoEncryptionrI r r& r r r� )r@ r r) � privateBytesr r r �test_privateBlobEd25519C s" ��� ��z KeyTests.test_privateBlobEd25519c C r+ )zV Raises L{RuntimeError} if the underlying key object does not exists. N)r r& r� r� r r� r r r �test_privateBlobNoKeyObject[ s z$KeyTests.test_privateBlobNoKeyObjectc C sz t j�tj�}| �|�d�tj� | �|jddd�tj� | �|� � �d�tj dd� � | �|� � jddd�tj � dS )zO L{keys.Key.toString} serializes an RSA key in OpenSSH format. r� r� r� Nr� � comment��comment) r r& rd r ra rI r� rW r� re rS �r@ r� r r r �test_toOpenSSHRSAc s ���zKeyTests.test_toOpenSSHRSAc C � t j�tj�}|jddd�}|jdddd�}| �d|�� d � | �d|�� d � | �|t j�|�� | �|t jj|dd�� d S ) zT L{keys.Key.toString} serializes an RSA key in OpenSSH's v1 format. r� �v1��subtyper� �rA r� �# -----BEGIN OPENSSH PRIVATE KEY-----r r� N)r r& rd r rW r� rI � splitlines�r@ r� � new_key_data�new_enc_key_datar r r �test_toOpenSSHRSA_v1_formatu � � ��z$KeyTests.test_toOpenSSHRSA_v1_formatc C sb t j�tj�}| �|�d�tj� | �|�� jddd�tj � | �|�� �d�tj dd� � dS )zN L{keys.Key.toString} serializes a DSA key in OpenSSH format. r� r9 r: Nr� ) r r& rd r r` rI r� rY re rT r< r r r �test_toOpenSSHDSA� s ��zKeyTests.test_toOpenSSHDSAc C r> ) zS L{keys.Key.toString} serializes a DSA key in OpenSSH's v1 format. r� r? r@ r� rB rC r r� N)r r& rd r rY r� rI rD rE r r r �test_toOpenSSHDSA_v1_format� rI z$KeyTests.test_toOpenSSHDSA_v1_formatc C sN t j�tj�}| �|�� jddd�tj� | �|�� �d�tjdd� � dS )zQ L{keys.Key.toString} serializes an ECDSA key in OpenSSH format. r� r9 r: Nr� ) r r& rd r r[ rI re r� rU r< r r r �test_toOpenSSHECDSA� s ��z KeyTests.test_toOpenSSHECDSAc C r> ) zV L{keys.Key.toString} serializes an ECDSA key in OpenSSH's v1 format. r� r? r@ r� rB rC r r� N)r r& rd r r[ r� rI rD rE r r r � test_toOpenSSHECDSA_v1_format� rI z&KeyTests.test_toOpenSSHECDSA_v1_formatc C s� t j�tj�}|�d�}|jddd�}| �d|�� d � | �d|�� d � | �|t j�|�� | �|t jj|dd�� | �||jddd�� dS ) zX L{keys.Key.toString} serializes an Ed25519 key in OpenSSH's v1 format. r� r� r� rC r r? r@ N)r r& rd r r] r� rI rD rE r r r �test_toOpenSSHEd25519� s � ��z KeyTests.test_toOpenSSHEd25519c C s& t j�tj�}| jt|jddd� dS )z� L{keys.Key.toString} refuses to serialize an Ed25519 key in OpenSSH's old PEM format, as no encoding of Ed25519 is defined for that format. r� �PEMr@ N)r r& rd r r] r� � ValueErrorr� r< r r r � test_toOpenSSHEd25519_PEM_format� s z)KeyTests.test_toOpenSSHEd25519_PEM_formatc C �> t j�tj�}| �|�d�tj� | �|�� �d�tj � dS )zK L{keys.Key.toString} serializes an RSA key in LSH format. �lshN) r r& rd r rW rI r� r? re r^ r< r r r � test_toLSHRSA� � zKeyTests.test_toLSHRSAc C rR )zJ L{keys.Key.toString} serializes a DSA key in LSH format. rS N) r r& rd r rY rI r� r` re r_ r< r r r � test_toLSHDSA� rU zKeyTests.test_toLSHDSAc C �& t j�tj�}| �|�d�tj� dS )zP L{keys.Key.toString} serializes an RSA key in Agent v3 format. rP N)r r& rd r rW rI r� ra r< r r r �test_toAgentv3RSA� � zKeyTests.test_toAgentv3RSAc C rW )zO L{keys.Key.toString} serializes a DSA key in Agent v3 format. rP N)r r& rd r rY rI r� rb r< r r r �test_toAgentv3DSA� rY zKeyTests.test_toAgentv3DSAc C sP t �| j�}|jddd�}| �t jj|d�� d�|� | jt j|jddd� dS )zd L{keys.Key.toString} applies Normalization Form KC to Unicode passphrases. r� r� r� r� r� N) r r& r* r� rI rd r� r� r� r� r r r �(test_toStringNormalizesUnicodePassphrase s � �z1KeyTests.test_toStringNormalizesUnicodePassphrasec C s | � tjt�| j�jd� dS )zm L{keys.Key.toString} raises L{keys.BadKeyError} when passed an invalid format type. r� N)r� r r� r& r* r� rE r r r �test_toStringErrors s z KeyTests.test_toStringErrorsc C �H d}t j�tj�}|�|�}| �|�� �||�� | �|�||�� dS )z8 Signed data can be verified using RSA. � some-dataN) r r& rd r rW �signro re �verify�r@ r{ r� � signaturer r r �test_signAndVerifyRSA � z KeyTests.test_signAndVerifyRSAc C r] )z8 Signed data can be verified using DSA. r^ N) r r& rd r rY r_ ro re r` ra r r r �test_signAndVerifyDSA' rd z KeyTests.test_signAndVerifyDSAc C s� d}t j�tj�}|�|�}t j�tj�}|�|�}t j�tj�}|�|�}| �|� � � ||�� | �|� ||�� | �|� � � ||�� | �|� ||�� | �|� � � ||�� | �|� ||�� dS )z7 Signed data can be verified using EC. r^ N) r r& rd r r[ r_ r� r� ro re r` )r@ r{ r� rb �key384� signature384�key521� signature521r r r �test_signAndVerifyEC1 s z KeyTests.test_signAndVerifyECc C r] )z< Signed data can be verified using Ed25519. r^ N) r r& rd r r] r_ ro re r` ra r r r �test_signAndVerifyEd25519F s z"KeyTests.test_signAndVerifyEd25519c C �N t j�tj�}| �|�| jd�� | �|�| jd�� | �|�| j d�� dS )zC A known-good RSA signature verifies successfully. r� � aN) r r& rd r rS ro r` r8 rq r9 r< r r r �test_verifyRSAQ � zKeyTests.test_verifyRSAc C rl )zC A known-good DSA signature verifies successfully. r� rm N) r r& rd r rT ro r` r9 rq r8 r< r r r �test_verifyDSAZ ro zKeyTests.test_verifyDSAc C s. t j�tj�}| �|�| jdd� d�� dS )z� Some commercial SSH servers send DSA keys as 2 20-byte numbers; they are still verified as valid keys. i���Nr� )r r& rd r rT ro r` r9 r< r r r �test_verifyDSANoPrefixc s z KeyTests.test_verifyDSANoPrefixc C s | � tt�| j��d� dS )zn The repr of a L{keys.Key} contains all of the RSA components for an RSA private key. a\ <RSA Private Key (2048 bits) attr d: 21:4c:08:66:a2:28:d5:b4:fb:8e:0f:72:1b:85:09: 00:b9:f2:4e:37:f0:1c:57:4b:e3:51:7f:9e:23:a7: e4:3a:98:55:1b:ea:8b:7a:98:1e:bc:d8:ba:b1:f9: 89:12:18:60:ac:e8:cc:0b:4e:09:5a:40:6a:ba:2f: 99:f8:b3:24:60:84:b9:ce:69:95:9a:f9:e2:fc:1f: 51:4d:27:15:db:2b:27:ad:ef:b4:69:ac:be:7d:10: eb:86:47:70:73:b4:00:87:95:15:3b:37:f9:e7:14: e7:80:bb:68:1e:1b:e6:dd:bb:73:63:b9:67:e6:b2: 27:7f:cf:cf:30:9b:c2:98:fd:d9:18:36:2f:36:2e: f1:3d:81:7a:9f:e1:03:2d:47:db:34:51:62:39:dd: 4f:e9:ac:a8:8b:d9:d6:f3:84:c4:17:b9:71:9d:06: 08:42:78:4d:bb:c5:2a:f4:c3:58:cd:55:2b:ed:be: 33:5f:04:ea:7b:e6:04:24:63:f2:2d:d7:3d:1b:6c: d5:9c:63:43:2f:92:88:8d:3e:6e:da:18:37:d8:0f: 25:67:89:1d:b9:46:34:5e:c9:ce:c4:8b:ed:92:5a: 33:07:0f:df:86:08:f9:92:e9:db:eb:38:08:36:c9: cd:cd:0a:01:48:5b:39:3e:7a:ca:c6:80:a9:dc:d4: 39 attr e: 01:00:01 attr n: 00:d5:6a:ac:78:23:d6:d6:1b:ec:25:a1:50:c4:77: 63:50:84:45:01:55:42:14:2a:2a:e0:d0:60:ee:d4: e9:a3:ad:4a:fa:39:06:5e:84:55:75:5f:00:36:bf: 6f:aa:2a:3f:83:26:37:c1:69:2e:5b:fd:f0:f3:d2: 7d:d6:98:cd:3a:40:78:d5:ca:a8:18:c0:11:93:24: 09:0c:81:4c:8f:f7:9c:ed:13:16:6a:a4:04:e9:49: 77:c3:e4:55:64:b3:79:68:9e:2c:08:eb:ac:e8:04: 2d:21:77:05:a7:8e:ef:53:30:0d:a5:e5:bb:3d:6a: e2:09:36:6f:fd:34:d3:7d:6f:46:ff:87:da:a9:29: 27:aa:ff:ad:f5:85:e6:3e:1a:b8:7a:1d:4a:b1:ea: c0:5a:f7:30:df:1f:c2:a4:e4:ef:3f:91:49:96:40: d5:19:77:2d:37:c3:5e:ec:9d:a6:3a:44:a5:c2:a4: 29:dd:d5:ba:9c:3d:45:b3:c6:2c:18:64:d5:ba:3d: df:ab:7f:cd:42:ac:a7:f1:18:0b:a0:58:15:62:0b: a4:2a:6e:43:c3:e4:04:9f:35:a3:47:8e:46:ed:33: a5:65:bd:bc:3b:29:6e:02:0b:57:df:74:e8:13:b4: 37:35:7e:83:5f:20:26:60:a6:dc:ad:8b:c6:6c:79: 98:f7 attr p: 00:d9:70:06:d8:e2:bc:d4:78:91:50:94:d4:c1:1b: 89:38:6c:46:64:5a:51:a0:9a:07:3d:48:8f:03:51: cc:6b:12:8e:7d:1a:b1:65:e7:71:75:39:e0:32:05: 75:8d:18:4c:af:93:b1:49:b1:66:5f:78:62:7a:d1: 0c:ca:e6:4d:43:b3:9c:f4:6b:7d:e6:0c:98:dc:cf: 21:62:8e:d5:2e:12:de:04:ae:d7:24:6e:83:31:a2: 15:a2:44:3d:22:a9:62:26:22:b9:b2:ed:54:0a:9d: 08:83:a7:07:0d:ff:19:18:8e:d8:ab:1d:da:48:9c: 31:68:11:a1:66:6d:e3:d8:1d attr q: 00:fb:44:17:8b:a4:36:be:1e:37:1d:a7:f6:61:6c: 04:c4:aa:dd:78:3e:07:8c:1e:33:02:ae:03:14:87: 83:7a:e5:9e:7d:08:67:a8:f2:aa:bf:12:70:cf:72: a9:a7:c7:0b:1d:88:d5:20:fd:9c:63:ca:47:30:55: 4e:8b:c4:cf:f4:7f:16:a4:92:12:74:a1:09:c2:c4: 6e:9c:8c:33:ef:a5:e5:f7:e0:2b:ad:4f:5c:11:aa: 1a:84:37:5b:fd:7a:ea:c3:cd:7c:b0:c8:e4:1f:54: 63:b5:c7:af:df:f4:09:a7:fc:c7:25:fc:5c:e9:91: d7:92:c5:98:1e:56:d3:b1:23 attr u: 00:85:4b:1b:7a:9b:12:10:37:9e:1f:ad:5e:da:fe: c6:96:fe:df:35:6b:b9:34:e2:16:97:92:26:09:bd: bd:70:20:03:a7:35:bd:2d:1b:a0:d2:07:47:2b:d4: de:a8:a8:07:07:1b:b8:04:20:a7:27:41:3c:6c:39: 39:e9:41:ce:e7:17:1d:d1:4c:5c:bc:3d:d2:26:26: fe:6a:d6:fd:48:72:ae:46:fa:7b:c3:d3:19:60:44: 1d:a5:13:a7:80:f5:63:29:d4:7a:5d:06:07:16:5d: f6:8b:3d:cb:64:3a:e2:84:5a:4d:8c:06:2d:2d:9d: 1c:eb:83:4c:78:3d:79:54:ce>N)rI �reprr r& r* rE r r r �test_reprPrivateRSAk s �z KeyTests.test_reprPrivateRSAc C s | � tt�| j��� �d� dS )zm The repr of a L{keys.Key} contains all of the RSA components for an RSA public key. a\ <RSA Public Key (2048 bits) attr e: 01:00:01 attr n: 00:d5:6a:ac:78:23:d6:d6:1b:ec:25:a1:50:c4:77: 63:50:84:45:01:55:42:14:2a:2a:e0:d0:60:ee:d4: e9:a3:ad:4a:fa:39:06:5e:84:55:75:5f:00:36:bf: 6f:aa:2a:3f:83:26:37:c1:69:2e:5b:fd:f0:f3:d2: 7d:d6:98:cd:3a:40:78:d5:ca:a8:18:c0:11:93:24: 09:0c:81:4c:8f:f7:9c:ed:13:16:6a:a4:04:e9:49: 77:c3:e4:55:64:b3:79:68:9e:2c:08:eb:ac:e8:04: 2d:21:77:05:a7:8e:ef:53:30:0d:a5:e5:bb:3d:6a: e2:09:36:6f:fd:34:d3:7d:6f:46:ff:87:da:a9:29: 27:aa:ff:ad:f5:85:e6:3e:1a:b8:7a:1d:4a:b1:ea: c0:5a:f7:30:df:1f:c2:a4:e4:ef:3f:91:49:96:40: d5:19:77:2d:37:c3:5e:ec:9d:a6:3a:44:a5:c2:a4: 29:dd:d5:ba:9c:3d:45:b3:c6:2c:18:64:d5:ba:3d: df:ab:7f:cd:42:ac:a7:f1:18:0b:a0:58:15:62:0b: a4:2a:6e:43:c3:e4:04:9f:35:a3:47:8e:46:ed:33: a5:65:bd:bc:3b:29:6e:02:0b:57:df:74:e8:13:b4: 37:35:7e:83:5f:20:26:60:a6:dc:ad:8b:c6:6c:79: 98:f7>N)rI rr r r& r* re rE r r r �test_reprPublicRSA� s �zKeyTests.test_reprPublicRSAc C s2 | � tt�| j��� �td�jdi tj ��� dS )zl The repr of a L{keys.Key} contains all the OpenSSH format for an ECDSA public key. z� <Elliptic Curve Public Key (256 bits) curve: ecdsa-sha2-nistp256 x: {x} y: {y}> Nr ) rI rr r r& r0 re r �formatr r/ rE r r r �test_reprPublicECDSA� s � � ��z KeyTests.test_reprPublicECDSAc C s. | � tt�| j��td�jdi tj��� dS )zm The repr of a L{keys.Key} contains all the OpenSSH format for an ECDSA private key. a <Elliptic Curve Private Key (256 bits) curve: ecdsa-sha2-nistp256 privateValue: {privateValue} x: {x} y: {y}> Nr ) rI rr r r& r0 r ru r r/ rE r r r �test_reprPrivateECDSA� s � � ��z KeyTests.test_reprPrivateECDSAc C s$ | � tt�| j��� �td�� dS )zn The repr of a L{keys.Key} contains all the OpenSSH format for an Ed25519 public key. z� <Ed25519 Public Key (256 bits) attr a: f1:16:d1:15:4a:1e:15:0e:19:5e:19:46:b5:f2:44: 0d:b2:52:a0:ae:2a:6b:23:13:73:45:fd:40:d9:57: 7b:8b>N)rI rr r r& r7 re r rE r r r �test_reprPublicEd25519 s ��z KeyTests.test_reprPublicEd25519c C s | � tt�| j��td�� dS )zo The repr of a L{keys.Key} contains all the OpenSSH format for an Ed25519 private key. a� <Ed25519 Private Key (256 bits) attr a: f1:16:d1:15:4a:1e:15:0e:19:5e:19:46:b5:f2:44: 0d:b2:52:a0:ae:2a:6b:23:13:73:45:fd:40:d9:57: 7b:8b attr k: 37:2f:25:da:8d:d4:a8:9a:78:7c:61:f0:98:01:c6: f4:5e:6d:67:05:69:31:37:4c:69:0d:05:55:bb:c9: 44:58>N)rI rr r r& r7 r rE r r r �test_reprPrivateEd25519 s ��z KeyTests.test_reprPrivateEd25519)W�__name__� __module__� __qualname__r �skipCryptographyr r rA rF rK rc rn rv r| rw rx r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r" r' r* r, r/ r0 r2 r7 r8 r= rH rJ rK rL rM rN rQ rT rV rX rZ r[ r\ rc re rj rk rn rp rq rs rt rv rw rx ry r r r r r * s� ? C) * P r c @ s4 e Zd ZdZedu r eZdd� Zdd� Zdd� Z dS ) �PersistentRSAKeyTestsz1 Tests for L{keys._getPersistentRSAKey}. Nc C sF t | �� �}|�d�}tj|dd�}| �|�� d� | �|�� � dS )z� L{keys._getPersistentRSAKey} will put the key in C{directory}/C{filename}, with the key length of C{keySize}. � mykey.pem� ��keySizeN) r r; �childr �_getPersistentRSAKeyrI rJ ro �exists�r@ �tempDirr< r� r r r �test_providedArguments5 s z,PersistentRSAKeyTests.test_providedArgumentsc C s| t | �� �}|�d�}tj|dd�}| �|�� d� | �|�� � |� � }tj|dd�}| �|�� d� | �|� � |� dS )zm L{keys._getPersistentRSAKey} will not regenerate the key if the key already exists. r r� r� rG N) r r; r� r r� rI rJ ro r� � getContent)r@ r� r< r� � keyContentr r r �test_noRegenerationA s z)PersistentRSAKeyTests.test_noRegenerationc C s> t | �� �}|�d�}tj|dd�}d|_| �|�� d� dS )z If the key generated by L{keys.getPersistentRSAKey} is set to None the key size should then become 0. r r� r� Nr )r r; r� r r� r) rI rJ r� r r r �test_keySizeZeroT s z&PersistentRSAKeyTests.test_keySizeZero) rz r{ r| �__doc__r r} r r� r� r� r r r r r~ - s r~ ) r� r� rC �textwrapr �twisted.conch.testr �twisted.pythonr �twisted.python.filepathr �twisted.python.reflectr � twisted.trialr r r} r � cryptography.hazmat.backendsr �twisted.conch.sshr r r �ed25519_supportedr r �TestCaser r~ r r r r �<module> sB PK F��Z��OS S ' __pycache__/test_window.cpython-310.pycnu �[��� o �bA � @ sH d Z ddlmZmZmZ ddlmZ G dd� de�ZG dd� de�ZdS ) zJ Tests for the insults windowing module, L{twisted.conch.insults.window}. � )� ScrolledArea� TextOutput� TopWindow)�TestCasec @ � e Zd ZdZdd� ZdS )�TopWindowTestszB Tests for L{TopWindow}, the root window container class. c s� g � g }t � fdd�|j�}| �� g � | �|g � |�� | �� g � | �t|�d� |�� | �� g � | �t|�d� |�� � | �t� �d� | �|g � |�� | �t� �d� | �t|�d� dS )z� Verify that L{TopWindow.repaint} schedules an actual paint to occur using the scheduling object passed to its initializer. c s � � d �S )N)�append� ��paintsr �@/usr/lib/python3/dist-packages/twisted/conch/test/test_window.py�<lambda> s z5TopWindowTests.test_paintScheduling.<locals>.<lambda>� N)r r � assertEqual�repaint�len�pop)�self� scheduled�rootr r r �test_paintScheduling s" z#TopWindowTests.test_paintSchedulingN)�__name__� __module__� __qualname__�__doc__r r r r r r s r c @ r )�ScrolledAreaTestsz� Tests for L{ScrolledArea}, a widget which creates a viewport containing another widget and can reposition that viewport using scrollbars. c C s2 t � }t|�}| �|j|j� | �|jj|� dS )z� The parent of the widget passed to L{ScrolledArea} is set to a new L{Viewport} created by the L{ScrolledArea} which itself has the L{ScrolledArea} instance as its parent. N)r r �assertIs�parent� _viewport)r �widget�scrolledr r r � test_parent8 s z ScrolledAreaTests.test_parentN)r r r r r! r r r r r 2 s r N) r � twisted.conch.insults.windowr r r �twisted.trial.unittestr r r r r r r �<module> s )PK F��Z�� �� � % __pycache__/test_cftp.cpython-310.pycnu �[��� o 5�@gG� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZ ddlmZ ddlmZmZ dd lmZ dd lmZmZmZmZm Z dd l m Z dd l m Z m!Z! dd l"m#Z# ddl$m%Z% ddl&m'Z' ddl(m)Z) ddl*m+Z+ ddl,m-Z- ddl.m/Z/ e+d�Z0e+d�Z1e+d�Z2e1r�e0r�z(ddl3m4Z4 ddl5m6Z6 ddl7m8Z8 ddl9m:Z:m;Z; ddl<m=Z= dd lm>Z> W n e?y� Y nw d Z@de2e1e0e�Ae d�fv r�d Z@e e@d �G d d!� d!e/��ZBG d"d#� d#e/�ZCG d$d%� d%e-�ZDG d&d'� d'�ZEG d(d)� d)�ZFe e�G d*d+� d+e ��ZGe e@d �G d,d-� d-e/��ZHG d.d/� d/�ZIG d0d1� d1ejJ�ZKG d2d3� d3e�ZLe e@d �G d4d5� d5eL��ZMe e@d �G d6d7� d7eL��ZNe e@d �e e)d8� d9�G d:d;� d;eL���ZOdS )<z* Tests for L{twisted.conch.scripts.cftp}. � N)�BytesIO)�skipIf)� implementer)�ls)� ISFTPFile)�FileTransferTestAvatar� SFTPTestBase)�portal)�defer�error� interfaces�protocol�reactor)�Clock)�getProcessOutputAndValue�getProcessValue)�log)� UserDatabase)�FilePath)�which)� requireModule)�StringTransport)�TestCase�pyasn1� cryptographyztwisted.conch.unix)�cftp)� SSHSession)� filetransfer)� test_conch�test_ssh)� FakeStdio)�FileTransferForTestAvatarFTz4don't run w/o spawnProcess or cryptography or pyasn1c @ s e Zd ZdZdd� ZdS )�SSHSessionTestsz= Tests for L{twisted.conch.scripts.cftp.SSHSession}. c C s* t � }t� }||_|�� | �|j� dS )z| L{twisted.conch.scripts.cftp.SSHSession.eofReceived} loses the write half of its stdio connection. N)r r �stdio� eofReceived� assertTrue� writeConnLost)�selfr# �channel� r) �>/usr/lib/python3/dist-packages/twisted/conch/test/test_cftp.py�test_eofReceived> s z SSHSessionTests.test_eofReceivedN)�__name__� __module__� __qualname__�__doc__r+ r) r) r) r* r" 8 s r" c @ s� e Zd ZdZeedd�du rdZdd� Zdd� Zd d � Z d d � Z d d� Z e � � Zz z e �e jd� W n e jy> dZY nw e �e je� W n Y eed�dd� �Zdd� ZdS )� ListingTestsz� Tests for L{lsLine}, the function which generates an entry for a file or directory in an SFTP I{ls} command's output. �tzsetNz8Cannot test timestamp formatting code without time.tzsetc sj d� _ � fdd�}� �td|� dtjv r*� �tjtjdtjd � � �tj � dS dd� }� �|� dS ) zo Patch the L{ls} module's time function so the results of L{lsLine} are deterministic. i�[c s � j S �N)�nowr) �r' r) r* �fakeTimeZ s z$ListingTests.setUp.<locals>.fakeTime�time�TZc S s, zt jd= W n ty Y nw t�� d S )Nr7 )�os�environ�KeyErrorr6 r1 r) r) r) r* �cleanupf s � z#ListingTests.setUp.<locals>.cleanupN) r3 �patchr r8 r9 � addCleanup�operator�setitemr6 r1 )r' r5 r; r) r4 r* �setUpS s zListingTests.setUpc C s |t jd< t�� t�d|�S )zl Call L{ls.lsLine} after setting the timezone to C{timezone} and return the result. r7 �foo)r8 r9 r6 r1 r �lsLine)r' �timezone�statr) r) r* � _lsInTimezonep s zListingTests._lsInTimezonec C �T | j d }t�dddddddd|df �}| �| �d|�d� | �| �d|�d� dS )z� A file with an mtime six months (approximately) or more in the past has a listing including a low-resolution timestamp. � r �America/New_Yorkz;!--------- 0 0 0 0 Apr 26 1973 foo�Pacific/Aucklandz;!--------- 0 0 0 0 Apr 27 1973 fooN�r3 r8 � stat_result� assertEqualrE �r' �thenrD r) r) r* � test_oldFile{ � � �zListingTests.test_oldFilec C �X | j d d }t�dddddddd|df �}| �| �d|�d� | �| �d|�d� dS ) �� A file with a high-resolution timestamp which falls on a day of the month which can be represented by one decimal digit is formatted with one padding 0 to preserve the columns which come after it. rG i�� r rH z;!--------- 0 0 0 0 May 01 1973 foorI z;!--------- 0 0 0 0 May 02 1973 fooNrJ rM r) r) r* � test_oldSingleDigitDayOfMonth� � � �z*ListingTests.test_oldSingleDigitDayOfMonthc C rF )z� A file with an mtime fewer than six months (approximately) in the past has a listing including a high-resolution timestamp excluding the year. 逛z r rH �;!--------- 0 0 0 0 Aug 28 17:33 foorI �;!--------- 0 0 0 0 Aug 29 09:33 fooNrJ rM r) r) r* � test_newFile� rP zListingTests.test_newFile� es_AR.UTF8Tz'The es_AR.UTF8 locale is not installed.c C s| | j d }t�dddddddd|df �}t�� }t�tjd� | �tjtj|� | �| � d|�d� | �| � d|�d� dS ) zC The month name in the date is locale independent. rU r rY rH rV rI rW N) r3 r8 rK �locale� getlocale� setlocale�LC_ALLr= rL rE )r' rN rD � currentLocaler) r) r* �test_localeIndependent� s � �z#ListingTests.test_localeIndependentc C rQ ) rR rU i F r rH z;!--------- 0 0 0 0 Sep 01 17:33 foorI z;!--------- 0 0 0 0 Sep 02 09:33 fooNrJ rM r) r) r* � test_newSingleDigitDayOfMonth� rT z*ListingTests.test_newSingleDigitDayOfMonth)r, r- r. r/ �getattrr6 �skipr@ rE rO rS rX rZ r[ r^ r\ r] �Error� localeSkipr r_ r` r) r) r) r* r0 J s. �� r0 c s e Zd ZdZ� fdd�Z� ZS )�InMemorySSHChannelzp Minimal implementation of a L{SSHChannel} like class which only reads and writes data from memory. c s || _ d| _t� �� dS )zt @param conn: The SSH connection associated with this channel. @type conn: L{SSHConnection} r N)�conn� localClosed�super�__init__)r' rf �� __class__r) r* ri � s zInMemorySSHChannel.__init__)r, r- r. r/ ri � __classcell__r) r) rj r* re � s re c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) � FilesystemAccessExpectationszC A test helper used to support expected filesystem access. c C s i | _ d S r2 ��_cacher4 r) r) r* ri � z%FilesystemAccessExpectations.__init__c C s || j ||f< dS )z� @param path: Path at which the stream is requested. @type path: L{str} @param path: Flags with which the stream is requested. @type path: L{str} @param stream: A stream. @type stream: C{File} Nro )r' �path�flags�streamr) r) r* �put s z FilesystemAccessExpectations.putc C s | j �||f�S )a Remove a stream from the memory. @param path: Path at which the stream is requested. @type path: L{str} @param path: Flags with which the stream is requested. @type path: L{str} @return: A stream. @rtype: C{File} )rp �pop)r' rr rs r) r) r* rv s z FilesystemAccessExpectations.popN)r, r- r. r/ ri ru rv r) r) r) r* rn � s rn c @ s e Zd ZdZdd� Zdd� ZdS )�InMemorySFTPClienta' A L{filetransfer.FileTransferClient} which does filesystem operations in memory, without touching the local disc or the network interface. @ivar _availableFiles: File like objects which are available to the SFTP client. @type _availableFiles: L{FilesystemRegister} c C s t | �| _ddd�| _|| _d S )N� � )�requests� buffersize)re � transport�options�_availableFiles)r' �availableFilesr) r) r* ri , s � zInMemorySFTPClient.__init__c C s | j �||�S )z} @see: L{filetransfer.FileTransferClient.openFile}. Retrieve and remove cached file based on flags. )r~ rv )r' �filenamers �attrsr) r) r* �openFile4 s zInMemorySFTPClient.openFileN)r, r- r. r/ ri r� r) r) r) r* rw " s rw c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�InMemoryRemoteFilez; An L{ISFTPFile} which handles all data in memory. c C s || _ t�| � dS )zL @param name: Name of this file. @type name: L{str} N)�namer ri )r' r� r) r) r* ri C s zInMemoryRemoteFile.__init__c C s | � |� | �|� t�| �S )z/ @see: L{ISFTPFile.writeChunk} )�seek�writer �succeed)r' �start�datar) r) r* � writeChunkK s z InMemoryRemoteFile.writeChunkc C s d| _ dS )zo @see: L{ISFTPFile.writeChunk} Keeps data after file was closed to help with testing. TN)�_closedr4 r) r) r* �closeS � zInMemoryRemoteFile.closec C � d S r2 r) r4 r) r) r* �getAttrs[ � zInMemoryRemoteFile.getAttrsc C r� r2 r) )r' �offset�lengthr) r) r* � readChunk_ r� z InMemoryRemoteFile.readChunkc C r� r2 r) )r' r� r) r) r* �setAttrsc r� zInMemoryRemoteFile.setAttrsc C s t �| �S )zb Get current data of file. Allow reading data event when file is closed. )r �getvaluer4 r) r) r* r� g r� zInMemoryRemoteFile.getvalueN) r, r- r. r/ ri r� r� r� r� r� r� r) r) r) r* r� = s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd)d d �Zd*d d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� ZdS )+�StdioClientTestsz( Tests for L{cftp.StdioClient}. c C sT t � | _t| j�}t�|�| _d| j_t� | _| j_ | � dd� | jjj | j_ dS )zm Create a L{cftp.StdioClient} hooked up to dummy transport and a fake user database. �/i� � N) rn �fakeFilesystemrw r � StdioClient�client�currentDirectoryr �database�_pwd�setKnownConsoleSizer| )r' � sftpClientr) r) r* r@ v s zStdioClientTests.setUpc C s@ | j �t�� dt�� dddtj� | j� d�}|� | j d� |S )zs The I{exec} command runs its arguments locally in a child process using the user's shell. �secret�� rA �barzexec print(1 + 2)s 3 ) r� �addUser�getpass�getuserr8 �getuid�sys� executabler� �_dispatchCommand� addCallbackrL �r' �dr) r) r* � test_exec� s � zStdioClientTests.test_execc C �> | j �t�� dt�� dddd� | j�d�}|�| j d� |S )zr If the local user has no shell, the I{exec} command runs its arguments using I{/bin/sh}. r� r� rA r� � zexec echo hello� hello � r� r� r� r� r8 r� r� r� r� rL r� r) r) r* �test_execWithoutShell� s � z&StdioClientTests.test_execWithoutShellc C r� )zO The I{exec} command is run for lines which start with C{"!"}. r� r� rA r� z/bin/shz !echo hellor� r� r� r) r) r* � test_bang� s � zStdioClientTests.test_bangc s2 ddl �G � ��fdd�d�}| �td|� � dS )a For the duration of this test, patch C{cftp}'s C{fcntl} module to return a fixed width and height. @param width: the width in characters @type width: L{int} @param height: the height in characters @type height: L{int} r Nc s e Zd Z� ��fdd�ZdS )z7StdioClientTests.setKnownConsoleSize.<locals>.FakeFcntlc s& |�j kr | �d� t�d� �dd�S )Nz#Only window-size queries supported.�4Hr )� TIOCGWINSZ�fail�struct�pack)r' �fd�opt�mutate��height�tty�widthr) r* �ioctl� s z=StdioClientTests.setKnownConsoleSize.<locals>.FakeFcntl.ioctlN)r, r- r. r� r) r� r) r* � FakeFcntl� s r� �fcntl)r� r< r )r' r� r� r� r) r� r* r� � s z$StdioClientTests.setKnownConsoleSizec C s� | � dd� t� }| j_td�}d|_t�|�}d|_|� � }|� d� | j d7 _ | j� ||� d}| � | jj�� |� d S ) a( L{StdioClient._printProgressBar} prints a progress description, including percent done, amount transferred, transfer rate, and time remaining, all based the given start time, the given L{FileWrapper}'s progress information and the reactor's current time. ry �" � x� samplei ( g @i s# b'sample' 40% 4.0kB 2.0kBps 00:03 N)r� r r� r r r� r � FileWrapper�size�seconds�advance�total�_printProgressBarrL r| �value�r' �clock�wrapped�wrapper� startTime�resultr) r) r* � test_printProgressBarReporting� s z/StdioClientTests.test_printProgressBarReportingc C sd | � dd� t� }| j_td�}d|_t�|�}|�� }| j� ||� d}| � | jj � � |� dS )z� L{StdioClient._printProgressBar} prints a progress description that indicates 0 bytes transferred if no bytes have been transferred and no time has passed. ry r� r� r� s! b'sample' 0% 0.0B 0.0Bps 00:00 N) r� r r� r r r� r r� r� r� rL r| r� r� r) r) r* � test_printProgressBarNoProgress� s z0StdioClientTests.test_printProgressBarNoProgressc C sL | � dd� t� }d|_t�|�}| j�|d� d}| �|| jj� � � dS )z5 Print the progress for empty files. ry r� s empty-filer s% b'empty-file'100% 0.0B 0.0Bps 00:00 N) r� r r� r r� r� r� rL r| r� )r' r� r� r� r) r) r* � test_printProgressBarEmptyFile� s z/StdioClientTests.test_printProgressBarEmptyFilec C � | j �d�}| �d|� dS )zK Returns empty value for both filename and remaining data. z )r� r� N�r� � _getFilenamerL �r' r� r) r) r* �test_getFilenameEmpty� s z&StdioClientTests.test_getFilenameEmptyc C r� )zd Returns empty value for remaining data when line contains only a filename. � only-local)r� r� Nr� r� r) r) r* �test_getFilenameOnlyLocal � z*StdioClientTests.test_getFilenameOnlyLocalc C r� )ze Returns filename and remaining data striped of leading and trailing spaces. z local remote file )�localz remote fileNr� r� r) r) r* �test_getFilenameNotQuoted r� z*StdioClientTests.test_getFilenameNotQuotedc C r� )z� Returns filename and remaining data not striped of leading and trailing spaces when quoted paths are requested. z" " local file " " remote file " )z local file z" remote file "Nr� r� r) r) r* �test_getFilenameQuoted r� z'StdioClientTests.test_getFilenameQuotedN� c C sJ |du r| � � }t|d��}|�|� W d � |S 1 s w Y |S )ab Create a local file and return its path. When `path` is L{None}, it will create a new temporary file. @param path: Optional path for the new file. @type path: L{str} @param content: Content to be written in the new file. @type content: L{bytes} @return: Path to the newly create file. N�wb)�mktemp�openr� )r' rr �content�filer) r) r* �makeFile s ��zStdioClientTests.makeFileFc C sh | j j�� }|�d�}|�d�}g }g }|D ]\}}}g } |D ] } | �|� d| � �� q | �d|� d|� �� |�| � |�d��d��d�} | dd � } | d �d ��d �} | �| � g }| dd � D ]$} | �� � dd �d } | �� �dd �} |�| d � d| d �� � �� qd|�| d � |�|� q|r�| � t |�t |�� n| � ||� | � dt |�d � dS )a Check output of cftp client for a put request. @param transfers: List with tuple of (local, remote, progress). @param randomOrder: When set to C{True}, it will ignore the order in which put reposes are received �utf-8z � z Transferred z to r � N���� � rx z5There are still put responses which were not checked.) r� r| r� �decode�split�appendrv �strip�extend�rsplitrL �sorted�len)r' � transfers� randomOrder�output�expectedOutput� actualOutputr� �remote�expected�expectedTransfer�line� progressParts�actual�last�actualTransferr) r) r* �checkPutMessage3 s< " �z StdioClientTests.checkPutMessagec C s� d}| j |d�}tjtjB tjB }tj�dtj�|��}t |�}| j � ||t � |�� d| jjjd< | j�|�}| �|� | �||�� � | �|j� | �||g d�fg� dS )z� A name based on local path is used when remote path is not provided. The progress is updated while chunks are transferred. s Test Content)r� r� ry r{ )z 76% 10.0B� 100% 13.0Br N)r� r � FXF_WRITE� FXF_CREAT� FXF_TRUNCr8 rr �join�basenamer� r� ru r r� r� r} �cmd_PUT�successResultOfrL r� r% r� r )r' r� � localPathrs � remoteName� remoteFile�deferredr) r) r* � test_cmd_PUTSingleNoRemotePathp s �z/StdioClientTests.test_cmd_PUTSingleNoRemotePathc C s� | � � }tjtjB tjB }d}t|�}| j�||t� |�� | j � |� d|� d��}| � |� | � ||dgfg� | �|j� | �d|�� � dS )z� Remote path is extracted from first filename after local file. Any other data in the line is ignored. z /remote-pathr� z ignored� 100% 0.0Br� N)r� r r r r r� r� ru r r� r� r r r r% r� rL r� )r' r rs r r r r) r) r* � test_cmd_PUTSingleRemotePath� s z-StdioClientTests.test_cmd_PUTSingleRemotePathc C s | � � }tj�|�}d}tj�|�}| j tj�||�d�}tjtjB tj B }d|� �}d|� �}t |�} t |�} | j � ||t �| �� | j � ||t �| �� | j�tj�|d��} | �| � | �| j� | �d| �� � | �| j� | �d| �� � | j||dgf||dgfgdd� d S ) z� When a gobbing expression is used local files are transferred with remote file names based on local names. � second-name�rr r� �*r� r T�r� N)r� r8 rr r �dirnamer r r r r r� r� ru r r� r� r r r% r� rL r� r ) r' �first� firstName� secondName�parent�secondrs �firstRemotePath�secondRemotePath�firstRemoteFile�secondRemoteFiler r) r) r* � test_cmd_PUTMultipleNoRemotePath� s4 � � �z1StdioClientTests.test_cmd_PUTMultipleNoRemotePathc C s | � � }tj�|�}d}tj�|�}| j tj�||�d�}tjtjB tj B }t |�}t |�}d|� �} d|� �} | j � | |t �|�� | j � | |t �|�� | j�d�tj�|d���} | �| � | �|j� | �d|�� � | �|j� | �d|�� � | j||dgf||dgfgdd � d S ) z� When a gobbing expression is used local files are transferred with remote file names based on local names. when a remote folder is requested remote paths are composed from remote path and local filename. r r z/remote/z {} remoter r� r Tr N)r� r8 rr r r r r r r r r� r� ru r r� r� r �formatr r% r� rL r� r ) r' r r r r r rs r r! r r r r) r) r* �"test_cmd_PUTMultipleWithRemotePath� s4 � � �z3StdioClientTests.test_cmd_PUTMultipleWithRemotePath)Nr� )F)r, r- r. r/ r@ r� r� r� r� r� r� r� r� r� r� r� r� r r r r" r$ r) r) r) r* r� p s( = #r� c @ s e Zd Zdd� Zdd� ZdS )�FileTransferTestRealmc C s || _ d S r2 )�testDir)r' r& r) r) r* ri � rq z FileTransferTestRealm.__init__c G s t | j�}|d |dd� fS )Nr c S r� r2 r) r) r) r) r* �<lambda>� s z5FileTransferTestRealm.requestAvatar.<locals>.<lambda>)r r& )r' �avatarID�mindr �ar) r) r* � requestAvatar� s z#FileTransferTestRealm.requestAvatarN)r, r- r. ri r+ r) r) r) r* r% � s r% c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dS )�SFTPTestProcessz� Protocol for testing cftp. Provides an interface between Python (where all the tests are) and the cftp client process (which does the work that is being tested). c C s$ | � � || _d| _d| _d| _dS )zr @param onOutReceived: A L{Deferred} to be fired as soon as data is received from stdout. NF)� clearBuffer� onOutReceived� onProcessEnd�_expectingCommand� _processEnded)r' r. r) r) r* ri � s zSFTPTestProcess.__init__c C s d| _ g | _d| _dS )zR Clear any buffered data received from stdout. Should be private. r� N)�buffer�_linesReceived� _lineBufferr4 r) r) r* r- � s zSFTPTestProcess.clearBufferc C sr t �d| � | j| �d�}|�d�| _| j�|� | jdur,| jd}| _|�|� | j |7 _ | � � dS )zO Called by Twisted when the cftp client prints data to stdout. zgot %r� r� N) r �msgr4 r� rv r3 r� r. �callbackr2 �_checkForCommand)r' r� �linesr� r) r) r* � outReceived s zSFTPTestProcess.outReceivedc C sf d}| j r/| j|kr1d�| j�}|�|�r |t|�d � }| �� | j d }| _ |�|� d S d S d S )Ns cftp> r5 )r0 r4 r r3 � startswithr� r- r7 )r' �prompt�bufr� r) r) r* r8 s �z SFTPTestProcess._checkForCommandc C s t �d| � dS )zO Called by Twisted when the cftp client prints data to stderr. zerr: %sN)r r6 )r' r� r) r) r* � errReceived" s zSFTPTestProcess.errReceivedc C s | j S )zQ Return the contents of the buffer of data received from stdout. )r2 r4 r) r) r* � getBuffer( s zSFTPTestProcess.getBufferc C s< t �� | _| �� t|t�r|�d�}| j�|d � | jS )a� Issue the given command via the cftp client. Return a C{Deferred} that fires when the server returns a result. Note that the C{Deferred} will callback even if the server returns some kind of error. @param command: A string containing an sftp command. @return: A C{Deferred} that fires when the sftp server returns a result. The payload is the server's response string. r� r5 ) r �Deferredr0 r- � isinstance�str�encoder| r� �r' �commandr) r) r* � runCommand. s zSFTPTestProcess.runCommandc s( t �d��� �fdd�|D �}t �|�S )ax Run each command in sequence and return a Deferred that fires when all commands are completed. @param commands: A list of strings containing sftp commands. @return: A C{Deferred} that fires when all commands are completed. The payload is a list of response strings from the server, in the same order as the commands. rx c s g | ] }�� � j|��qS r) )�runrF )�.0rE �r' �semr) r* � <listcomp>L s z-SFTPTestProcess.runScript.<locals>.<listcomp>)r �DeferredSemaphore� gatherResults)r' �commands�dlr) rI r* � runScript@ s zSFTPTestProcess.runScriptc C s, | j rt�d�S t�� | _| j�d� | jS )z� Kill the process if it is still running. If the process is still running, sends a KILL signal to the transport and returns a C{Deferred} which fires when L{processEnded} is called. @return: a C{Deferred}. N�KILL)r1 r r� r@ r/ r| � signalProcessr4 r) r) r* � killProcessO s zSFTPTestProcess.killProcessc C s, d| _ | jr| jd}| _|�d� dS dS )zF Called by Twisted when the cftp client process ends. TN)r1 r/ r7 )r' �reasonr� r) r) r* � processEnded^ s �z SFTPTestProcess.processEndedN)r, r- r. r/ ri r- r: r8 r> r? rF rP rS rU r) r) r) r* r, � s r, c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �CFTPClientTestBasec C s� t dd��}|�tj� W d � n1 sw Y t dd��}|�tj� W d � n1 s1w Y t�dd� t dd��}|�dtj � W d � n1 sTw Y t� | �S )N� dsa_test.pubr� �dsa_testi�� �kh_tests 127.0.0.1 ) r� r� r �publicDSA_openssh�privateDSA_opensshr8 �chmod�publicRSA_opensshr r@ �r' �fr) r) r* r@ i s � � � zCFTPClientTestBase.setUpc C sF t | j�}t�|�}|�t�� � t�� }||_tj d|dd�| _ d S )Nr z 127.0.0.1)� interface) r% r& r �Portal�registerCheckerr �conchTestPublicKeyChecker�ConchTestServerFactoryr � listenTCP�server)r' �realm�p�facr) r) r* � startServers s z CFTPClientTestBase.startServerc C sH t | jjd�s | �d �S d| jjj_t�| jjjjj �}|� | j� |S )N�protorx ) �hasattrrf �factory� _cbStopServerrk �expectedLoseConnectionr � maybeDeferredr| �loseConnectionr� r� r) r) r* � stopServer{ s z CFTPClientTestBase.stopServerc C s t �| jj�S r2 )r rp rf � stopListening�r' �ignoredr) r) r* rn � s z CFTPClientTestBase._cbStopServerc C s4 dD ]}zt �|� W q ty Y qw t�| �S )N)rW rX rY )r8 �remove� BaseExceptionr �tearDownr^ r) r) r* rx � s � zCFTPClientTestBase.tearDownN)r, r- r. r@ rj rr rn rx r) r) r) r* rV h s rV c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z d-dd�Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� ZdS ).�OurServerCmdLineClientTestsz� Functional tests which launch a SFTP server over TCP on localhost and check cftp command line interface using a spawned process. Due to the spawned process you can not add a debugger breakpoint for the client code. c s0 t �� � � �� d}� j�� j}tj|| �� dd�}t � dt j � d|� �� t �� }t|�� _|�� fdd�� tj�� }tj�t j�|d< g }i }|D ]}t|t�rZ|�d �}|�|� qN|D ] }|| } t|t�rr|�d �}t| t�r|| �d �} | ||< qbt � |� t � |� t j � jt j ||d � |S ) Nz�-p %i -l testuser --known-hosts kh_test --user-authentications publickey --host-key-algorithms ssh-rsa -i dsa_test -a -v 127.0.0.1r ��mod�running r� c � � j �� S r2 )�processProtocolr- ��_r4 r) r* r' � � z3OurServerCmdLineClientTests.setUp.<locals>.<lambda>� PYTHONPATHr� ��env) rV r@ rj rf �getHost�portr � _makeArgsr� r r6 r� r� r r@ r, r~ r� r8 r9 �copy�pathsepr rr rA rB rC r� r � spawnProcess) r' �cmdsr� r� r� � encodedCmds� encodedEnv�cmd�var�valr) r4 r* r@ � s> � �z!OurServerCmdLineClientTests.setUpc s � � � }|�� fdd�� |S )Nc r} r2 )r~ rS r r4 r) r* r' � r� z6OurServerCmdLineClientTests.tearDown.<locals>.<lambda>)rr r� r� r) r4 r* rx � s z$OurServerCmdLineClientTests.tearDownc C s, z | j j�d� W d S tjy Y d S w )NrQ )r~ r| rR r �ProcessExitedAlreadyrt r) r) r* � _killProcess� s �z(OurServerCmdLineClientTests._killProcessc C � | j �|�S )z� Run the given command with the cftp client. Return a C{Deferred} that fires when the command is complete. Payload is the server's output for that command. )r~ rF rD r) r) r* rF � � z&OurServerCmdLineClientTests.runCommandc G r� )z� Run the given commands with the cftp client. Returns a C{Deferred} that fires when the commands are all complete. The C{Deferred}'s payload is a list of output for each command. )r~ rP )r' rN r) r) r* rP � r� z%OurServerCmdLineClientTests.runScriptc C sL | j }| �ddddd�}dd� }|�|� |�| j|jt�� d|jg� |S )z� Test that 'pwd' reports the current remote directory, that 'lpwd' reports the current local directory, and that changing to a subdirectory then changing to its parent leaves you in the original remote directory. �pwd�lpwdzcd testDirectory�cd ..c S sD g }| D ]}t |t�r|�d�}|�|� q|dd� |dd� S )zH Callback function for handling command output. r� N� � )rA �bytesr� r� )r� r� r� r) r) r* � cmdOutput� s z8OurServerCmdLineClientTests.testCdPwd.<locals>.cmdOutputr� )r& rP r� rL rr r8 �getcwd)r' �homeDirr� r� r) r) r* � testCdPwd� s z%OurServerCmdLineClientTests.testCdPwdc s& � fdd�}� � dddd�}|�|�S )z� Check that 'ls -l' output includes the access permissions and that this output changes appropriately with 'chmod'. c sZ � � � � �| d �d�� � �| d d� � �| d �d�| d � � �| d d� d S )Nr s -rw-r--r--rx r� r� s ----------r� )�flushLoggedErrorsr% r; rL ��resultsr4 r) r* �_check� s z7OurServerCmdLineClientTests.testChAttrs.<locals>._checkzls -l testfile1zchmod 0 testfile1zchmod 644 testfile1�rP r� �r' r� r� r) r4 r* � testChAttrs� s � z'OurServerCmdLineClientTests.testChAttrsc s@ � fdd�}� � dd� j�� ddd�}|�dd � � |�|�S ) z� Check 'ls' works as expected. Checks for wildcards, hidden files, listing directories and listing empty directories. c sf � � | d g d�� � � | d g d�� � � | d ddg� � � | d g d�� � � | d d g� d S ) Nr �s testDirectory� testRemoveFile� testRenameFiles testfile1rx r� r� r� r� )s .testHiddenFiler� r� r� r� �rL r� r4 r) r* r� s �� �z4OurServerCmdLineClientTests.testList.<locals>._checkr zls ../zls *Filez ls -a *Filezls -l testDirectoryc S s dd� | D �S )Nc S s g | ]}|� d ��qS )r5 )r� )rH �xr) r) r* rK * � zJOurServerCmdLineClientTests.testList.<locals>.<lambda>.<locals>.<listcomp>r) )�xsr) r) r* r' * s z6OurServerCmdLineClientTests.testList.<locals>.<lambda>)rP r& r r� r� r) r4 r* �testList s � z$OurServerCmdLineClientTests.testListc C sD | � d�}t�d��d��� }t|t�r|�d�}|�| j |� |S )zB Check that running the '?' command returns help. �?Nr� r� ) rF r r� �cmd_HELPr� rA rB rC r� rL )r' r� �helpTextr) r) r* �testHelp- s z$OurServerCmdLineClientTests.testHelpNc C s | � |�� |�� |� dS )zg Assert that the files at C{name1} and C{name2} contain exactly the same data. N)rL � getContent)r' �name1�name2r6 r) r) r* �assertFilesEqual9 s z,OurServerCmdLineClientTests.assertFilesEqualc sl d� �jj�jj�� t� t�r� �d�� � �fdd�}��d�jj� d��}|�|� |��fdd�� |S ) z� Test that 'get' saves the remote file to the correct local location, that the output of 'get' is correct and that 'rm' actually removes the file. z)Transferred {}/testfile1 to {}/test file2r� c s8 �� | �� �� ���j�d��j�d�d� ��d�S )N� testfile1� test file2z get failedzrm "test file2")r% �endswithr� r& �childrF �r� �r� r' r) r* � _checkGetN s � z6OurServerCmdLineClientTests.testGet.<locals>._checkGetzget testfile1 "z /test file2"c � � � � j�d��� �S )Nr� �� assertFalser& r� �existsr r4 r) r* r' Z r� z5OurServerCmdLineClientTests.testGet.<locals>.<lambda>)r# r& rr rA rB rC rF r� )r' r� r� r) r� r* �testGet@ s � �z#OurServerCmdLineClientTests.testGetc s � fdd�}� � d�}|�|�S )zQ Test that 'get' works correctly when given wildcard parameters. c s8 � � � j�d�td�d� � � � j�d�td�d� d S )N�testRemoveFile�testRemoveFile get failed�testRenameFile�testRenameFile get failed)r� r& r� r )ru r4 r) r* r� c s � �z;OurServerCmdLineClientTests.testWildcardGet.<locals>._checkz get testR*)rF r� r� r) r4 r* �testWildcardGet^ s z+OurServerCmdLineClientTests.testWildcardGetc sh d�j �� j d �j �� j d � � �fdd�}��d�j j� d��}|�|� |��fdd �� |S ) z� Check that 'put' uploads files correctly and that they can be successfully removed. Also check the output of the put command. s Transferred s /testfile1 to s /test"file2c s6 �� �j�d��j�d�� ��| �� �� ��d�S )Nr� � test"file2zrm "test\"file2")r� r& r� r% r� rF r� r� r) r* � _checkPut� s � z6OurServerCmdLineClientTests.testPut.<locals>._checkPut�put z/testfile1 "test\"file2"c r� )Nr� r� r r4 r) r* r' � r� z5OurServerCmdLineClientTests.testPut.<locals>.<lambda>)r& � asBytesModerr rF r� )r' r� r� r) r� r* �testPutr s" �� ��� �z#OurServerCmdLineClientTests.testPutc s� � j �d�jdd�� }|�d� W d � n1 sw Y � j �d�jdd�� }|�d� W d � n1 s9w Y � fdd �}� �d � j j� d ��}|�|� |S ) zb Check that 'put' uploads files correctly when overwriting a longer file. � shorterFile�w��mode� aN� longerFile� bbc s � � � j�d�� j�d�� d S )Nr� r� )r� r& r� r� r4 r) r* r� � s �zEOurServerCmdLineClientTests.test_putOverLongerFile.<locals>._checkPutr� z/shorterFile longerFile)r& r� r� r� rF rr r� �r' r_ r� r� r) r4 r* �test_putOverLongerFile� s � � z2OurServerCmdLineClientTests.test_putOverLongerFilec s� � j �d����� ��d�jdd�� }|�d� W d � n1 s#w Y � j �d�jdd�� }|�d� W d � n1 sBw Y � �fdd �}� �d � j j� d ��}|�|� |S ) z� Check that 'put' uploads files correctly when overwriting a longer file and you use a wildcard to specify the files to upload. �dirr� r� r� r� Nr� c s � � ��d�� j�d�� d S )Nr� )r� r� r& r� �r' �someDirr) r* r� � s zMOurServerCmdLineClientTests.test_putMultipleOverLongerFile.<locals>._checkPutr� z/dir/*)r& r� �createDirectoryr� r� rF rr r� r� r) r� r* � test_putMultipleOverLongerFile� s � � z:OurServerCmdLineClientTests.test_putMultipleOverLongerFilec s> � fdd�}� � dd� jj� d�d� j�� �}|�|� |S )z� What happens if you issue a 'put' command and include a wildcard (i.e. '*') in parameter? Check that all files matching the wildcard are uploaded to the correct directory. c sh � � | d d� � � | d d� � �� j�d�� j�� �d�d� � �� j�d�� j�� �d�d� d S )Nr r� r� r� r� r� r� )rL r� r& r� r r� r4 r) r* �check� s � �z:OurServerCmdLineClientTests.testWildcardPut.<locals>.checkr� r� z/testR*zcd %s)rP r& rr r r� )r' r� r� r) r4 r* �testWildcardPut� s � z+OurServerCmdLineClientTests.testWildcardPutc �4 � fdd�}� � dd�}|�|� |�� jd� |S )z� Test that 'ln' creates a file which appears as a link in the output of 'ls'. Check that removing the new file succeeds without output. c s8 � � � � �| d d� � �| d �d�d� � �d�S )Nr r� rx � lz link failedz rm testLink)r� rL r% r; rF r� r4 r) r* r� � s z4OurServerCmdLineClientTests.testLink.<locals>._checkzln testLink testfile1zls -l testLinkr� �rP r� rL r� r) r4 r* �testLink� s z$OurServerCmdLineClientTests.testLinkc r� )zV Test that we can create and remove directories with the cftp client. c s. � � | d d� � �| d �d�� � �d�S )Nr r� rx � dzrmdir testMakeDirectory)rL r% r; rF r� r4 r) r* r� � s z?OurServerCmdLineClientTests.testRemoteDirectory.<locals>._check�mkdir testMakeDirectoryzls -l testMakeDirector?r� r� r� r) r4 r* �testRemoteDirectory� � z/OurServerCmdLineClientTests.testRemoteDirectoryc s& � fdd�}� � dd�}|�|� |S )z� Test that a C{mkdir} on an existing directory fails with the appropriate error, and doesn't log an useless error server side. c s$ � � | d d� � � | d d� d S )Nr r� rx s remote error 11: mkdir failedr� r� r4 r) r* r� � s zHOurServerCmdLineClientTests.test_existingRemoteDirectory.<locals>._checkr� r� r� r) r4 r* � test_existingRemoteDirectory� s z8OurServerCmdLineClientTests.test_existingRemoteDirectoryc sH � � d� jj� d��}|�� jd� |�� fdd�� |�� jd� |S )z� Test that we can create a directory locally and remove it with the cftp client. This test works because the 'remote' server is running out of a local directory. zlmkdir z/testLocalDirectoryr� c s � � d�S )Nzrmdir testLocalDirectory)rF r r4 r) r* r' r� z@OurServerCmdLineClientTests.testLocalDirectory.<locals>.<lambda>)rF r& rr r� rL r� r) r4 r* �testLocalDirectory s z.OurServerCmdLineClientTests.testLocalDirectoryc r� )z1 Test that we can rename a file. c s* � � | d d� � � | d d� � �d�S )Nr r� rx s testfile2zrename testfile2 testfile1)rL rF r� r4 r) r* r� s z6OurServerCmdLineClientTests.testRename.<locals>._checkzrename testfile1 testfile2z ls testfile?r� r� r� r) r4 r* � testRename r� z&OurServerCmdLineClientTests.testRenamer2 )r, r- r. r/ r@ rx r� rF rP r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r) r) r) r* ry � s. * ry c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�OurServerBatchFileTestszm Functional tests which launch a SFTP server over localhost and checks csftp in batch interface. c C s t �| � | �� d S r2 �rV r@ rj r4 r) r) r* r@ % s z OurServerBatchFileTests.setUpc C � t �| � | �� S r2 )rV rx rr r4 r) r) r* rx ) � z OurServerBatchFileTests.tearDownc s� | � � � t� d�� }|�|� W d � n1 sw Y | j�� j}d|� f }tj|�� dd�dd � }t � dt j � d|� �� t j�� }t j�t j�|d< d| jj_tt j ||d �}� fd d �}|�d d � � |�|� |S )Nr� z�-p %i -l testuser --known-hosts kh_test --user-authentications publickey --host-key-algorithms ssh-rsa -i dsa_test -a -v -b %s 127.0.0.1r rz rx r| r� r� r� c s t �� � | S r2 )r8 rv ��res��fnr) r* �_cleanupD s z9OurServerBatchFileTests._getBatchOutput.<locals>._cleanupc S s | d S )Nr r) r� r) r) r* r' H s z9OurServerBatchFileTests._getBatchOutput.<locals>.<lambda>)r� r� r� rf r� r� r r� r� r r6 r� r� r8 r9 r� r� r rr rm ro r r� �addBoth)r' r_ �fpr� r� r� r� r� r) r� r* �_getBatchOutput- s$ � � z'OurServerBatchFileTests._getBatchOutputc �( d}� fdd�}� � |�}|�|� |S )z� Test whether batch file function of cftp ('cftp -b batchfile'). This works by treating the file as a list of commands to be run. z pwd ls exit c sP | � d�} t�dt| � � � �� j�� j| d � � �| dd� g d�� d S )Nr5 zRES %srx r� ���r� ) r� r r6 �repr�assertInr& r� rr rL r� r4 r) r* �_cbCheckResultW s �z=OurServerBatchFileTests.testBatchFile.<locals>._cbCheckResult�r� r� �r' r� r� r� r) r4 r* � testBatchFileM s z%OurServerBatchFileTests.testBatchFilec r� )zO Test that an error in the batch file stops running the batch. z chown 0 missingFile pwd exit c � � � � j�� j| � d S r2 )� assertNotInr& r� rr r� r4 r) r* r� m � z9OurServerBatchFileTests.testError.<locals>._cbCheckResultr� r� r) r4 r* � testErrord s z!OurServerBatchFileTests.testErrorc r� )z_ Test that a minus sign '-' at the front of a line ignores any errors. z -chown 0 missingFile pwd exit c r� r2 )r� r& r� rr r� r4 r) r* r� ~ r� z@OurServerBatchFileTests.testIgnoredError.<locals>._cbCheckResultr� r� r) r4 r* �testIgnoredErrort s z(OurServerBatchFileTests.testIgnoredErrorN) r, r- r. r/ r@ rx r� r� r� r� r) r) r) r* r� s r� �sftpz%no sftp command-line client availablec @ rm ) �OurServerSftpClientTestsz@ Test the sftp server against sftp command line client. c C r� r2 r� r4 r) r) r* r@ � r� z OurServerSftpClientTests.setUpc C s | � � S r2 )rr r4 r) r) r* rx � s z!OurServerSftpClientTests.tearDownc s� �� � � t� d�� }|�d� W d � n1 sw Y �j�� j�tj��fdd�}��td|� d�jj _ t dd �}� �fd d �}�fd d �}|� |� |� dd� � |� |�S )a Test the return of extended attributes by the server: the sftp client should ignore them, but still be able to parse the response correctly. This test is mainly here to check that L{filetransfer.FILEXFER_ATTR_EXTENDED} has the correct value. r� z ls . exitNc s � | |�}d|d< |S )Nr� �ext_foor) )r' �sr� )� oldGetAttrr) r* � _getAttrs� s zCOurServerSftpClientTests.test_extendedAttributes.<locals>._getAttrsr T�ssh)�-o� PubkeyAcceptedKeyTypes=ssh-dssz-Vc s> | dkrd}nd}|dddddddd dd �f d � d f 7 }|S ) Nr )r r r) z-Fz /dev/nullr zIdentityFile=dsa_testzUserKnownHostsFile=kh_testzHostKeyAlgorithms=ssh-rsazPort=%iz-bztestuser@127.0.0.1r) )�status�args)r� r� r) r* �hasPAKT� s& �zAOurServerSftpClientTests.test_extendedAttributes.<locals>.hasPAKTc s. � � | d d� dD ] }� �|| d � q d S )Nr� r r� )rL r� )r� �ir4 r) r* r� � s �z?OurServerSftpClientTests.test_extendedAttributes.<locals>.checkc S s t d| �S )Nr )r )r r) r) r* r' � r� zBOurServerSftpClientTests.test_extendedAttributes.<locals>.<lambda>) r� r� r� rf r� r� r! r r< rm ro r r� )r' r_ r r� r r� r) )r� r r� r' r* �test_extendedAttributes� s � z0OurServerSftpClientTests.test_extendedAttributesN)r, r- r. r/ r@ rx r r) r) r) r* r � s r )Pr/ r� rZ r> r8 r� r� r6 �ior �unittestr �zope.interfacer � twisted.conchr �twisted.conch.interfacesr �$twisted.conch.test.test_filetransferr r � twisted.credr �twisted.internetr r r r r �twisted.internet.taskr �twisted.internet.utilsr r �twisted.pythonr �twisted.python.fakepwdr �twisted.python.filepathr �twisted.python.procutilsr �twisted.python.reflectr �twisted.test.proto_helpersr �twisted.trial.unittestr r r �unix�twisted.conch.scriptsr �twisted.conch.scripts.cftpr �twisted.conch.sshr �twisted.conch.testr r � twisted.conch.test.test_conchr r! � ImportError� skipTests�IReactorProcessr" r0 re rn rw r� r� r% �ProcessProtocolr, rV ry r� r r) r) r) r* �<module> s� � #&2 u {' gPK F��Z�n�+�� �� - __pycache__/test_filetransfer.cpython-310.pycnu �[��� o �bm� � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m Z ddl m Z ddl mZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ zdd lmZ W n e y_ dZY nw eZzddl m Z W n e yu dZ Y nw e Z zddl Z!W n e y� dZ Y nw e!Z zddl"m#Z$ W n e y� e%Z#Y nw e$Z#zddl&m'Z'm(Z(m)Z)m*Z* W n e y� Y nw G dd� de#�Z+G dd� de+�Z,G dd� d�Z-e du r�ddl.Z.e.�/def � nG dd� de �Z0e�1e0e+e)j2� G dd� de�Z3ee d �G d d � d e3��Z4G d d � d �Z5ee d �G d!d"� d"e��Z6ee d#�G d$d%� d%e��Z7ee d#�G d&d'� d'e��Z8ee d#�G d(d)� d)e��Z9dS )*z. Tests for L{twisted.conch.ssh.filetransfer}. � N)�skipIf)� assert_that�equal_to)�defer)�ConnectionLost)�loopback)� components)� _PY37PLUS)�FilePath)�StringTransport)�TestCase)�unix)�SFTPServerForUnixConchUser)� ConchUser)�common� connection� filetransfer�sessionc @ � e Zd Zdd� Zdd� ZdS )� TestAvatarc C s& t �| � tj| jd< tj| jd< d S )N� session� sftp)r �__init__r � SSHSession� channelLookupr �FileTransferServer�subsystemLookup��self� r �F/usr/lib/python3/dist-packages/twisted/conch/test/test_filetransfer.pyr ? s zTestAvatar.__init__c O s� zt |�}W n ty |||fg}Y nw |D ]%}|d }t|�dkr(|d p)d}t|�dkr4|d p5i }||i |��}q|S )Nr � r � )�iter� TypeError�len)r �f�args�kw�i�func�rr r r � _runAsUserD s �zTestAvatar._runAsUserN)�__name__� __module__� __qualname__r r, r r r r r > s r c @ r )�FileTransferTestAvatarc C s t �| � || _d S �N)r r �homeDir)r r2 r r r r R s z FileTransferTestAvatar.__init__c C s t t�� ��| jj�S r1 )r �os�getcwd� preauthChildr2 �pathr r r r � getHomeDirV s z!FileTransferTestAvatar.getHomeDirN)r- r. r/ r r7 r r r r r0 Q s r0 c @ � e Zd Zdd� ZdS )�ConchSessionForTestAvatarc C s || _ d S r1 ��avatar)r r; r r r r [ � z"ConchSessionForTestAvatar.__init__N)r- r. r/ r r r r r r9 Z � r9 zNtwisted.conch.unix imported %r, but doesn't define SFTPServerForUnixConchUser'c @ r )�FileTransferForTestAvatarc C s ddiS )N� conchTest� ext datar )r �version�otherExtr r r � gotVersionn s z$FileTransferForTestAvatar.gotVersionc C s |dkrdS t �)N� testExtendedRequest� bar)�NotImplementedError)r �extName�extDatar r r �extendedRequestq s z)FileTransferForTestAvatar.extendedRequestN)r- r. r/ rC rI r r r r r> m s r> c @ r8 )� SFTPTestBasec C sx t | �� �| _| j�d�| _| j�d��d� | j�d�jdd��*}|�d� tdd ��}|�|�d �� W d � n1 s?w Y W d � n1 sNw Y | j�d��d � | j�d �jdd�� }|�d � W d � n1 svw Y | j�d�jdd�� }|�d � W d � n1 s�w Y | j�d�jdd��}|�d � W d � d S 1 s�w Y d S )N�extra� testDirectoryT� testfile1�wb)�mode� aaaaaaaaaabbbbbbbbbbz /dev/urandom�rbi i� �testRemoveFile� a�testRenameFilez.testHiddenFile) r �mktemp�testDir�child�makedirs�open�write�read�chmod)r r& �f2r r r �setUp| s( �� � � � "�zSFTPTestBase.setUpN)r- r. r/ r^ r r r r rJ { r= rJ z can't run on non-posix computersc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zejee d-�d.d/� ��Z ejd0d1� �Z ejd2d3� �Z d4S )5�OurServerOurClientTestsc s� t �� � t� j�� _tj� jd�� _t� � j�}t� � � _ d � _ d � _ � fdd�}|� j _t� � j �}� j �|� � j�|� |� _|� _� �� d S )Nr: c s | � _ |� _d S r1 )�_serverVersion�_extData)� serverVersionrH r r r �_� s z(OurServerOurClientTests.setUp.<locals>._)rJ r^ r0 rV r; r r �serverr � LoopbackRelay�FileTransferClient�clientr` ra �gotServerVersion�makeConnection�clientTransport�serverTransport� _emptyBuffers)r rj rc rk r r r r^ � s z OurServerOurClientTests.setUpc C s< | j js| jjr | j �� | j�� | j js| jjsd S d S r1 )rk �bufferrj � clearBufferr r r r rl � s �z%OurServerOurClientTests._emptyBuffersc C s, | j �� | j�� | j �� | j�� d S r1 )rk �loseConnectionrj rn r r r r �tearDown� s z OurServerOurClientTests.tearDownc C s$ | � | jd� | � | jddi� d S )N� r? r@ )� assertEqualr` ra r r r r �test_serverVersion� s z*OurServerOurClientTests.test_serverVersionc C s& | � tj�| jj�d| jj��� dS )z: It implements the ISFTPServer interface. z ISFTPServer not provided by N)� assertTruer � ISFTPServer� providedByrd rg r r r r � test_interface_implementation� s �z5OurServerOurClientTests.test_interface_implementationc sb �j �dtjtjB i �}��� tj�g � � �fdd�}��td|� � �fdd�}|� |� |S )zW A file opened with C{openFile} is closed when the connection is lost. � testfile1c s � � | � �| � d S r1 ��append)�fd)�closed�oldCloser r �close� � zJOurServerOurClientTests.test_openedFileClosedWithConnection.<locals>.closer~ c s` �j j| jdd � j}�j�� �j�� �j�� �j�� ���j ji � �� |� � d S )N� ) rd � openFiles�handler{ rk ro rj rn rr �assertIn)�openFiler{ )r| r r r � _fileOpened� s zPOurServerOurClientTests.test_openedFileClosedWithConnection.<locals>._fileOpened) rg r� r �FXF_READ� FXF_WRITErl r3 r~ �patch� addCallback)r �dr~ r� r )r| r} r r �#test_openedFileClosedWithConnection� s � z;OurServerOurClientTests.test_openedFileClosedWithConnectionc s. � j �d�}� �� � fdd�}|�|� |S )zh A directory opened with C{openDirectory} is close when the connection is lost. � c s< � j �� � j�� � j �� � j�� � �� jji � d S r1 )rk ro rj rn rr rd �openDirs��openDirr r r � _getFiles� s zSOurServerOurClientTests.test_openedDirectoryClosedWithConnection.<locals>._getFiles�rg � openDirectoryrl r� )r r� r� r r r �(test_openedDirectoryClosedWithConnection� s z@OurServerOurClientTests.test_openedDirectoryClosedWithConnectionc sd �j �dtjtjB i �}��� � ��fdd�}�fdd�� ��fdd���fdd ��|�|� |S ) Nrx c s* �� | t�| �� � | �}|��| � |S r1 )rr r � ISFTPFiler� �r� r� )� _readChunk� _writeChunkr r r r� � s z<OurServerOurClientTests.test_openFileIO.<locals>._fileOpenedc s& | � dd�}� �� |�� jd� |S )Nr � rP �� readChunkrl r� rr r� r r r r� � � z;OurServerOurClientTests.test_openFileIO.<locals>._readChunkc s$ |� dd�}��� |�� |� |S )Nr� s cccccccccc)� writeChunkrl r� �rc r� r� )� _readChunk2r r r r� s z<OurServerOurClientTests.test_openFileIO.<locals>._writeChunkc s& |� dd�}� �� |�� jd� |S )Nr � s aaaaaaaaaabbbbbbbbbbccccccccccr� r� r r r r� r� z<OurServerOurClientTests.test_openFileIO.<locals>._readChunk2�rg r� r r� r� rl r� )r r� r� r )r� r� r� r r �test_openFileIO� s � z'OurServerOurClientTests.test_openFileIOc sV �j �dtjtjB i �}��� �fdd���fdd�� � ��fdd�}|�|� |S )Nrx c � |� � }� �� |S r1 )�getAttrsrl r� r r r � _getAttrs � zBOurServerOurClientTests.test_closedFileGetAttrs.<locals>._getAttrsc s � � � | S r1 )�flushLoggedErrors�r& r r r �_err s z=OurServerOurClientTests.test_closedFileGetAttrs.<locals>._errc s4 | � � }��� |��| � |�� � ��|tj�S r1 )r~ rl r� � addErrback� assertFailurer � SFTPErrorr� �r� r� r r r �_close# s z?OurServerOurClientTests.test_closedFileGetAttrs.<locals>._closer� )r r� r� r r� r �test_closedFileGetAttrs s � z/OurServerOurClientTests.test_closedFileGetAttrsc sD �j �dtjtjB i �}��� � �fdd�}�fdd�� |�|�S )Nrx c � | � � }��� |�� � |S r1 �r� rl r� r� �� _getAttrs2r r r r� 3 � zBOurServerOurClientTests.test_openFileAttributes.<locals>._getAttrsc �& � j �d�}� �� |�� j| � |S �Nrx �rg r� rl r� rr )�attrs1r� r r r r� 9 r� zCOurServerOurClientTests.test_openFileAttributes.<locals>._getAttrs2r� �r r� r� r r� r �test_openFileAttributes- s � z/OurServerOurClientTests.test_openFileAttributesc sV �j �dtjtjB i �}��� ��fdd�}� �fdd���fdd�� |�|� |S )Nrx c r� r1 r� r� )� _setAttrsr r r r� I r� z@OurServerOurClientTests.test_openFileSetAttrs.<locals>._getAttrsc s: d| d< �j �d| �}��� |�� � |��j| � |S )Nr �atimerx )rg �setAttrsrl r� rr ��attrsr� r� r r r� O s z@OurServerOurClientTests.test_openFileSetAttrs.<locals>._setAttrsc � � j �d�}� �� |S r� �rg r� rl �rc r� r r r r� W � zAOurServerOurClientTests.test_openFileSetAttrs.<locals>._getAttrs2r� r� r )r� r� r r �test_openFileSetAttrsA s � z-OurServerOurClientTests.test_openFileSetAttrsc sb i ��j jj� � �fdd�}|�j j_�j�dtjtjB ddi�}��� ��fdd�}|�|�S )z� Check that L{filetransfer.FileTransferClient.openFile} can send extended attributes, that should be extracted server side. By default, they are ignored, so we just verify they are correctly parsed. c s �� |� � | ||�S r1 )�update)�filename�flagsr� )� oldOpenFile�savedAttributesr r r� h r zIOurServerOurClientTests.test_openFileExtendedAttributes.<locals>.openFilerx �ext_foorE c s �� � ddi� d S )Nr� rE �rr )�ign)r� r r r �checku � zFOurServerOurClientTests.test_openFileExtendedAttributes.<locals>.check)rd rg r� r r� r� rl r� )r r� r� r� r )r� r� r r � test_openFileExtendedAttributes_ s � z7OurServerOurClientTests.test_openFileExtendedAttributesc sB � j �d�}� �� � fdd�}|�|� |�|� � �|tj�S )N� testRemoveFilec r� )Nr� )rg � removeFilerl )�ignoredr� r r r � _removeFile~ r� z<OurServerOurClientTests.test_removeFile.<locals>._removeFile)rg r� rl r� r� r r� )r r� r� r r r �test_removeFilez s z'OurServerOurClientTests.test_removeFilec s8 �j �d�}��� � �fdd�}�fdd�� |�|�S )N� testRenameFilec s& �j �dd�}��� |�� | � |S )Nr� � testRenamedFile)rg � renameFilerl r� r� �� _testRenamedr r r �_rename� s z8OurServerOurClientTests.test_renameFile.<locals>._renamec s& � j �d�}� �� |�� j|� d S )Nr� r� )rc r� r� r r r r� � s z=OurServerOurClientTests.test_renameFile.<locals>._testRenamed)rg r� rl r� )r r� r� r r� r �test_renameFile� s z'OurServerOurClientTests.test_renameFilec C �" | j �d�}| �� | �|tj�S �N� testMakeDirectory�rg r� rl r� r r� �r r� r r r �test_directoryBad� � z)OurServerOurClientTests.test_directoryBadc sZ � j �di �}� �� � fdd�}� fdd�}|�|� |�|� |�|� � �|tj�S )Nr� c r� r� r� r� r r r r� � r� zAOurServerOurClientTests.test_directoryCreation.<locals>._getAttrsc r� r� )rg �removeDirectoryrl r� r r r �_removeDirectory� r� zHOurServerOurClientTests.test_directoryCreation.<locals>._removeDirectory)rg � makeDirectoryrl r� r� r r� )r r� r� r� r r r �test_directoryCreation� s z.OurServerOurClientTests.test_directoryCreationc s\ �j �d�}��� g �� ���fdd����fdd�}�fdd�� |��� |�|� |S )N� c sF �� fdd�}t �� j�}��� |�|� |��� |��� � |S )Nc s � � | � �S r1 ry r� )�filesr� r r rz � s zMOurServerOurClientTests.test_openDirectory.<locals>._getFiles.<locals>.append)r � maybeDeferred�nextrl r� r� )r� rz r� �r� r� r� r r� r r� � s z=OurServerOurClientTests.test_openDirectory.<locals>._getFilesc s0 t t t� � �d �}|�� ��|g d�� d S )Nr )� .testHiddenFile� testDirectoryr� r� rx )�list�zip�sortrr )r� �fs)r� r r r � _checkFiles� s �z?OurServerOurClientTests.test_openDirectory.<locals>._checkFilesc r� r1 )r~ rl )rc r� r� r r r r� � r� z:OurServerOurClientTests.test_openDirectory.<locals>._closer� )r r� r� r r� r �test_openDirectory� s z*OurServerOurClientTests.test_openDirectoryc C r� )N� testLinkr� r� r r r �test_linkDoesntExist� r� z,OurServerOurClientTests.test_linkDoesntExistc sB � j �dd�}� �� � fdd�}� fdd�}|�|� |�|�S )Nr� rx c s � j �dd�}� �� |S )Nr� r! r� r� r r r �_getFirstAttrs� s zDOurServerOurClientTests.test_linkSharesAttrs.<locals>._getFirstAttrsc r� r� r� )� firstAttrsr� r r r �_getSecondAttrs� r� zEOurServerOurClientTests.test_linkSharesAttrs.<locals>._getSecondAttrs�rg �makeLinkrl r� )r r� r� r� r r r �test_linkSharesAttrs� s z,OurServerOurClientTests.test_linkSharesAttrsc sF � j �dd�}� �� � fdd�}� fdd�}|�|� |�|� |S )Nr� rx c �H � j �d�}� �� tt�� ��� jj�}|� d�}|� � j |j� |S �Nr� rM ) rg �readLinkrl r r3 r4 r5 rV r6 rW r� rr )rc r� �testFiler r r � _readLink� � z8OurServerOurClientTests.test_linkPath.<locals>._readLinkc r� r� ) rg �realPathrl r r3 r4 r5 rV r6 rW r� rr )rc r� �testLinkr r r � _realPath� r z8OurServerOurClientTests.test_linkPath.<locals>._realPathr� )r r� r� r r r r � test_linkPath� s z%OurServerOurClientTests.test_linkPathc C s4 | j �dd�}| �� |�| jd� |�| j� |S )NrD s foorE )rg rI rl r� rr �_cbTestExtendedRequestr� r r r �test_extendedRequest s z,OurServerOurClientTests.test_extendedRequestc C s" | j �dd�}| �� | �|t�S )Ns testBadRequestr� )rg rI rl r� rF )r r� r� r r r r s z.OurServerOurClientTests._cbTestExtendedRequestz!Broken by PEP 479 and deprecated.c c s� � | j �d�}| �� |V }t� }z!|D ]}| �� |V \}}}|�|� qW |�� }| �� |V n |�� }| �� |V w | �� | �|h d�� dS )z� Check that the object returned by L{filetransfer.FileTransferClient.openDirectory} can be used as an iterator. r� > rx r� r� r� r� N)rg r� rl �set�addr~ rr )r r� r� � filenamesr& r� rc � fileattrsr r r �test_openDirectoryIterator s, � ���z2OurServerOurClientTests.test_openDirectoryIteratorc c sz � | j �d�}| �� |V }|�� }| �� |V | �� }d}| �dt|�� | �t|d d � | �||d d � dS )zJ Using client.openDirectory as an iterator is deprecated. r� zeUsing twisted.conch.ssh.filetransfer.ClientDirectory as an iterator was deprecated in Twisted 18.9.0.r! r �category�messageN)rg r� rl r� � flushWarningsrr r% �DeprecationWarning)r r� r� �oneFile�warningsr r r r �$test_openDirectoryIteratorDeprecatedA s � �z<OurServerOurClientTests.test_openDirectoryIteratorDeprecatedc # s� � �j �dtji �}��� |V }g � � �fdd�}�jj�� \}||_~|�dd�}��� �� t � �d� �� |� �j � � �j �� �j�� ��� ���j j� ��|t� |�� }��|t� dS )z{ If there are requests outstanding when the connection is closed for any reason, they should fail. rx c s" �� � g � t�� }� �| � |S r1 )rr r �Deferredrz )�offset�lengthr� ��gotReadRequestr r r � _slowReadd s zOOurServerOurClientTests.test_closedConnectionCancelsRequests.<locals>._slowRead�d �� r! N)rg r� r r� rl rd r� �valuesr� rr r% �assertNoResultrk ro rn rj � assertFalse� connected�failureResultOfr r� )r r� �fhr � serverSideFhr r r �$test_closedConnectionCancelsRequestsV s* � z<OurServerOurClientTests.test_closedConnectionCancelsRequestsN) r- r. r/ r^ rl rp rs rw r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �inlineCallbacksr r r r r" r r r r r_ � s< ! ( ( r_ c @ r8 )�FakeConnc C � d S r1 r )r �channelr r r � sendClose� � zFakeConn.sendCloseN)r- r. r/ r' r r r r r$ � r= r$ c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�FileTransferCloseTestsc C s t � | _d S r1 )r r; r r r r r^ � s z FileTransferCloseTests.setUpc C s, t �� }G dd� d�}|� |_| j|j_|S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zDFileTransferCloseTests.buildServerConnection.<locals>.DummyTransportc S s | | _ d S r1 )� transportr r r r r � r< zMFileTransferCloseTests.buildServerConnection.<locals>.DummyTransport.__init__c S r% r1 r )r �kind�datar r r � sendPacket� r( zOFileTransferCloseTests.buildServerConnection.<locals>.DummyTransport.sendPacketc S s dS )Nzdummy transportr r r r r � logPrefix� r( zNFileTransferCloseTests.buildServerConnection.<locals>.DummyTransport.logPrefixN)r- r. r/ r r- r. r r r r �DummyTransport� s r/ )r � SSHConnectionr* r; )r �connr/ r r r �buildServerConnection� s z,FileTransferCloseTests.buildServerConnectionc s$ d�_ |j� � �fdd�}||_d S )NFc s d�_ � | � d S )NT)�connectionLostFired)�reason��origConnectionLostr r r �connectionLost� s zFFileTransferCloseTests.interceptConnectionLost.<locals>.connectionLost)r3 r7 )r � sftpServerr7 r r5 r �interceptConnectionLost� s z.FileTransferCloseTests.interceptConnectionLostc C s | � | jd� d S )Nz*sftpServer's connectionLost was not called)rt r3 r r r r �assertSFTPConnectionLost� s �z/FileTransferCloseTests.assertSFTPConnectionLostc C sJ t jt� | jd�}|�t�d�� |jjj }| � |� |� � | � � dS )ze Closing a session should notify an SFTP subsystem launched by that session. )r1 r; r N) r r r$ r; �request_subsystemr �NSrg r* �protor9 � closeReceivedr: )r � testSessionr8 r r r �test_sessionClose� s z(FileTransferCloseTests.test_sessionClosec C s� | � � }t�d�t�dd�d }|�|� |jd }|�t�d�� |jj j }| � |� | � |� |� t�dd�� | � � dS )zy A client sending CHANNEL_CLOSE should trigger closeReceived on the associated channel instance. r �>Lr rq r N)r2 r r< �struct�pack�ssh_CHANNEL_OPEN�channelsr; rg r* r= r9 �ssh_CHANNEL_CLOSEr: �r r1 �packet�sessionChannelr8 r r r �%test_clientClosesChannelOnConnnection� s z<FileTransferCloseTests.test_clientClosesChannelOnConnnectionc C sn | � � }t�d�t�dd�d }|�|� |jd }|�t�d�� |jj j }| � |� |� � | � � dS )zP Closing an SSH connection should close all sessions within it. r rA r rq r N)r2 r r< rB rC rD rE r; rg r* r= r9 �serviceStoppedr: rG r r r �'test_stopConnectionServiceClosesChannel� s z>FileTransferCloseTests.test_stopConnectionServiceClosesChannelN) r- r. r/ r^ r2 r9 r: r@ rJ rL r r r r r) � s r) z Cannot run without cryptographyc @ s e Zd ZdZg d�Zdd� ZdS )�ConstantsTestsag Tests for the constants used by the SFTP protocol implementation. @ivar filexferSpecExcerpts: Excerpts from the draft-ietf-secsh-filexfer-02.txt (draft) specification of the SFTP protocol. There are more recent drafts of the specification, but this one describes version 3, which is what conch (and OpenSSH) implements. )a� The following values are defined for packet types. #define SSH_FXP_INIT 1 #define SSH_FXP_VERSION 2 #define SSH_FXP_OPEN 3 #define SSH_FXP_CLOSE 4 #define SSH_FXP_READ 5 #define SSH_FXP_WRITE 6 #define SSH_FXP_LSTAT 7 #define SSH_FXP_FSTAT 8 #define SSH_FXP_SETSTAT 9 #define SSH_FXP_FSETSTAT 10 #define SSH_FXP_OPENDIR 11 #define SSH_FXP_READDIR 12 #define SSH_FXP_REMOVE 13 #define SSH_FXP_MKDIR 14 #define SSH_FXP_RMDIR 15 #define SSH_FXP_REALPATH 16 #define SSH_FXP_STAT 17 #define SSH_FXP_RENAME 18 #define SSH_FXP_READLINK 19 #define SSH_FXP_SYMLINK 20 #define SSH_FXP_STATUS 101 #define SSH_FXP_HANDLE 102 #define SSH_FXP_DATA 103 #define SSH_FXP_NAME 104 #define SSH_FXP_ATTRS 105 #define SSH_FXP_EXTENDED 200 #define SSH_FXP_EXTENDED_REPLY 201 Additional packet types should only be defined if the protocol version number (see Section ``Protocol Initialization'') is incremented, and their use MUST be negotiated using the version number. However, the SSH_FXP_EXTENDED and SSH_FXP_EXTENDED_REPLY packets can be used to implement vendor-specific extensions. See Section ``Vendor-Specific-Extensions'' for more details. a� The flags bits are defined to have the following values: #define SSH_FILEXFER_ATTR_SIZE 0x00000001 #define SSH_FILEXFER_ATTR_UIDGID 0x00000002 #define SSH_FILEXFER_ATTR_PERMISSIONS 0x00000004 #define SSH_FILEXFER_ATTR_ACMODTIME 0x00000008 #define SSH_FILEXFER_ATTR_EXTENDED 0x80000000 a� The `pflags' field is a bitmask. The following bits have been defined. #define SSH_FXF_READ 0x00000001 #define SSH_FXF_WRITE 0x00000002 #define SSH_FXF_APPEND 0x00000004 #define SSH_FXF_CREAT 0x00000008 #define SSH_FXF_TRUNC 0x00000010 #define SSH_FXF_EXCL 0x00000020 a� Currently, the following values are defined (other values may be defined by future versions of this protocol): #define SSH_FX_OK 0 #define SSH_FX_EOF 1 #define SSH_FX_NO_SUCH_FILE 2 #define SSH_FX_PERMISSION_DENIED 3 #define SSH_FX_FAILURE 4 #define SSH_FX_BAD_MESSAGE 5 #define SSH_FX_NO_CONNECTION 6 #define SSH_FX_CONNECTION_LOST 7 #define SSH_FX_OP_UNSUPPORTED 8 c C s� i }| j D ] }|�� D ]}t�d|�}|r"t|�d�d�||�d�< q q| �t|�dkd� |�� D ] \}}| � |t t |�� q2dS )z� The constants used by the SFTP protocol implementation match those found by searching through the spec. z)^\s*#define SSH_([A-Z_]+)\s+([0-9x]*)\s*$r" r r! z,No constants found (the test must be buggy).N) �filexferSpecExcerpts� splitlines�re�match�int�grouprt r% �itemsrr �getattrr )r � constants�excerpt�line�m�k�vr r r �test_constantsAgainstSpecH s �� ��z(ConstantsTests.test_constantsAgainstSpecN)r- r. r/ �__doc__rN r\ r r r r rM � s LrM c @ s e Zd ZdZdd� Zdd� ZdS )�RawPacketDataServerTestsz� Tests for L{filetransfer.FileTransferServer} which explicitly craft certain less common situations to exercise their handling. c C s t jt� d�| _d S )Nr: )r r r �ftsr r r r r^ a r� z RawPacketDataServerTests.setUpc C s� t � }| j�|� d}d}t�tdg�| t�|� �}| j�|� t�tdg�| tg d�� t�d� t�d� �}t|�� t |�� dS ) zl A close request with an unknown handle receives an FX_NO_SUCH_FILE error response. s 1234s invalid handler� �e )r r r r" s No such file or directoryr� N) r r_ ri r r< �bytes� dataReceivedr �valuer )r r* � requestIdr� r~ �expectedr r r �test_closeInvalidHandled s6 ��� � � � ���z0RawPacketDataServerTests.test_closeInvalidHandleN)r- r. r/ r] r^ rf r r r r r^ Z s r^ c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�RawPacketDataTestsz� Tests for L{filetransfer.FileTransferClient} which explicitly craft certain less common protocol messages to exercise their handling. c C s t �� | _d S r1 )r rf �ftcr r r r r^ � s zRawPacketDataTests.setUpc C sT t �� }|�| j� || jjd< t�ddtj �t � d� t � d� }| j� |� |S )a/ A STATUS packet containing a result code, a message, and a language is parsed to produce the result of an outstanding request L{Deferred}. @see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>} of the SFTP Internet-Draft. r! �!LL� msg� lang) r r r� �_cbTestPacketSTATUSrh � openRequestsrB rC r �FX_OKr r< � packet_STATUS�r r� r, r r r �test_packetSTATUS� s ��� z$RawPacketDataTests.test_packetSTATUSc C �$ | � |d d� | � |d d� dS )z{ Assert that the result is a two-tuple containing the message and language from the STATUS packet. r rj r! rk Nr� �r �resultr r r rl � � z&RawPacketDataTests._cbTestPacketSTATUSc C s@ t �� }|�| j� || jjd< t�ddtj �}| j� |� |S )az A STATUS packet containing only a result code can also be parsed to produce the result of an outstanding request L{Deferred}. Such packets are sent by some SFTP implementations, though not strictly legal. @see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>} of the SFTP Internet-Draft. r! ri ) r r r� �_cbTestPacketSTATUSShortrh rm rB rC r rn ro rp r r r �test_packetSTATUSShort� s z)RawPacketDataTests.test_packetSTATUSShortc C s$ | � |d d� | � |d d� dS )z� Assert that the result is a two-tuple containing empty strings, since the STATUS packet had neither a message nor a language. r r� r! Nr� rs r r r rv � ru z+RawPacketDataTests._cbTestPacketSTATUSShortc C sJ t �� }|�| j� || jjd< t�ddtj �t � d� }| j� |� |S )a� A STATUS packet containing a result code and a message but no language can also be parsed to produce the result of an outstanding request L{Deferred}. Such packets are sent by some SFTP implementations, though not strictly legal. @see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>} of the SFTP Internet-Draft. r! ri rj ) r r r� � _cbTestPacketSTATUSWithoutLangrh rm rB rC r rn r r< ro rp r r r � test_packetSTATUSWithoutLang� s z/RawPacketDataTests.test_packetSTATUSWithoutLangc C rr )z� Assert that the result is a two-tuple containing the message from the STATUS packet and an empty string, since the language was missing. r rj r! r� Nr� rs r r r rx � ru z1RawPacketDataTests._cbTestPacketSTATUSWithoutLangN) r- r. r/ r] r^ rq rl rw rv ry rx r r r r rg � s rg ):r] r3 rP rB �unittestr �hamcrestr r �twisted.internetr �twisted.internet.errorr �twisted.protocolsr �twisted.pythonr �twisted.python.compatr �twisted.python.filepathr �twisted.test.proto_helpersr �twisted.trial.unittestr � twisted.conchr �_unix� ImportError�twisted.conch.unixr �_SFTPServerForUnixConchUser� cryptography� _cryptography�twisted.conch.avatarr � _ConchUser�object�twisted.conch.sshr r r r r r0 r9 r �warnr> �registerAdapterru rJ r_ r$ r) rM r^ rg r r r r �<module> s� � � � � � �� � v g h =PK F��Z?��8 8 ( __pycache__/test_manhole.cpython-310.pycnu �[��� o �b�3 � @ s0 U d Z ddlZddlmZ dZee ed< ddlmZ ddl m Z ddl m Z m Z mZmZmZ ddlmZmZ dd lmZ dd lmZ d d � Ze� ZG d d� dej�ZG dd� dej�ZG dd� dej�Z G dd� d�Z G dd� deeje �Z G dd� de eje �Z G dd� de eje �Z G dd � d ej�Z!dS ) z% Tests for L{twisted.conch.manhole}. � N)�Optional�ssh)�manhole)�insults)� _SSHMixin� _StdioMixin� _TelnetMixinr �stdio)�defer�error)�StringTransport)�unittestc C s2 zdd W dS t y t�� d d Y S w )z� Return the string used by Python as the name for code objects which are compiled from interactive input or at the top-level of modules. � r ���� N)� BaseException� traceback� extract_stack� r r �A/usr/lib/python3/dist-packages/twisted/conch/test/test_manhole.py� determineDefaultFunctionName s �r c @ � e Zd ZdZdd� ZdS )�ManholeInterpreterTestsz2 Tests for L{manhole.ManholeInterpreter}. c C s2 t �d�}|j�ddg� |�� | �|j� dS )zR L{ManholeInterpreter.resetBuffer} should empty the input buffer. N�1�2)r �ManholeInterpreter�buffer�extend� resetBuffer� assertFalse)�self� interpreterr r r �test_resetBuffer4 s z(ManholeInterpreterTests.test_resetBufferN)�__name__� __module__� __qualname__�__doc__r" r r r r r / � r c @ r )�ManholeProtocolTestsz' Tests for L{manhole.Manhole}. c C sP t � }t�tj�}|�|� |j}|j}|j� ddg� |� � | � |j� dS )zn L{manhole.Manhole.handle_INT} should cause the interpreter input buffer to be reset. r r N) r r �ServerProtocolr �Manhole�makeConnection�terminalProtocolr! r r � handle_INTr )r � transport�terminal�protocolr! r r r �%test_interruptResetsInterpreterBufferC s z:ManholeProtocolTests.test_interruptResetsInterpreterBufferN)r# r$ r% r& r1 r r r r r( > r' r( c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )� WriterTestsc C � t �d� dS )z& Colorize an integer. r N�r �lastColorizedLine�r r r r � test_IntegerS � zWriterTests.test_Integerc C r3 )z7 Colorize an integer in double quotes. z"1"Nr4 r6 r r r �test_DoubleQuoteStringY r8 z"WriterTests.test_DoubleQuoteStringc C r3 )z7 Colorize an integer in single quotes. z'1'Nr4 r6 r r r �test_SingleQuoteString_ r8 z"WriterTests.test_SingleQuoteStringc C r3 )z7 Colorize an integer in triple quotes. z'''1'''Nr4 r6 r r r � test_TripleSingleQuotedStringe r8 z)WriterTests.test_TripleSingleQuotedStringc C r3 )zB Colorize an integer in triple and double quotes. z"""1"""Nr4 r6 r r r � test_TripleDoubleQuotedStringk r8 z)WriterTests.test_TripleDoubleQuotedStringc C r3 )z1 Colorize a function definition. z def foo():Nr4 r6 r r r �test_FunctionDefinitionq r8 z#WriterTests.test_FunctionDefinitionc C r3 )z. Colorize a class definition. z class foo:Nr4 r6 r r r �test_ClassDefinitionw r8 z WriterTests.test_ClassDefinitionc C � t �d�}| �t|t�� dS )z, Colorize a Unicode string. � иN�r r5 � assertTrue� isinstance�bytes�r �resr r r � test_unicode} � zWriterTests.test_unicodec C r? )z/ Colorize a UTF-8 byte string. � иNrA rE r r r � test_bytes� rH zWriterTests.test_bytesc C s | � t�d�t�d�� dS )zT The output of UTF-8 bytestrings and Unicode strings are identical. rI r@ N)� assertEqualr r5 r6 r r r �test_identicalOutput� s �z WriterTests.test_identicalOutputN) r# r$ r% r7 r9 r: r; r< r= r> rG rJ rL r r r r r2 R s r2 c @ s� e Zd ZejZdd� Zdd� Zdd� Zdd� Z d d � Z d d � Z d d� Z dd� Z dd� Zejdd� �Zejdd� �Zdd� Zdd� Zejdd � �Zd S ) �ManholeLoopbackMixinc C s t �|�S )N)r �waitForDeferred�r �dr r r �wfd� s zManholeLoopbackMixin.wfdc �, � j �d�}� �d� � fdd�}|�|�S )z- Evaluate simple expression. � dones 1 + 1 donec � � � g d�� d S )N)� >>> 1 + 1� 2� >>> done�� _assertBuffer��ignr6 r r �finished� � z<ManholeLoopbackMixin.test_SimpleExpression.<locals>.finished��recvlineClient�expect� _testwrite� addCallback�r �doner\ r r6 r �test_SimpleExpression� � z*ManholeLoopbackMixin.test_SimpleExpressionc rR )z> Evaluate line continuation in triple quotes. rS s ''' ''' donec rT )N)s >>> '''s ... '''s '\n'rW rX rZ r6 r r r\ � r] zGManholeLoopbackMixin.test_TripleQuoteLineContinuation.<locals>.finishedr^ rc r r6 r � test_TripleQuoteLineContinuation� rf z5ManholeLoopbackMixin.test_TripleQuoteLineContinuationc rR )z/ Evaluate function definition. rS s' def foo(bar): print(bar) foo(42) donec rT )N)s >>> def foo(bar):s ... print(bar)� ... s >>> foo(42)s 42rW rX rZ r6 r r r\ � � �z>ManholeLoopbackMixin.test_FunctionDefinition.<locals>.finishedr^ rc r r6 r r= � � z,ManholeLoopbackMixin.test_FunctionDefinitionc rR )z, Evaluate class definition. rS sE class Foo: def bar(self): print('Hello, world!') Foo().bar() donec rT )N)s >>> class Foo:s ... def bar(self):s" ... print('Hello, world!')rh s >>> Foo().bar()s Hello, world!rW rX rZ r6 r r r\ � ri z;ManholeLoopbackMixin.test_ClassDefinition.<locals>.finishedr^ rc r r6 r r> � s � z)ManholeLoopbackMixin.test_ClassDefinitionc rR )z0 Evaluate raising an exception. rS s# raise Exception('foo bar baz') donec s" � � dddt�d� ddg� d S )Ns" >>> raise Exception('foo bar baz')s" Traceback (most recent call last):s File "<console>", line 1, in zutf-8s Exception: foo bar bazrW )rY �defaultFunctionName�encoderZ r6 r r r\ � s ���z5ManholeLoopbackMixin.test_Exception.<locals>.finishedr^ rc r r6 r �test_Exception� rj z#ManholeLoopbackMixin.test_Exceptionc s6 � j �d�}� �dtj d � � fdd�}|�|�S )z4 Evaluate interrupting with CTRL-C. rS � cancelled linec rT )N)� >>> cancelled line� KeyboardInterruptrW rX rZ r6 r r r\ ri z4ManholeLoopbackMixin.test_ControlC.<locals>.finished)r_ r` ra r �CTRL_Crb rc r r6 r � test_ControlC� s z"ManholeLoopbackMixin.test_ControlCc sF � j �d�}� �d� � fdd�}|�|� � fdd�}|�|� |S )z� Sending ^C to Manhole while in a state where more input is required to complete a statement should discard the entire ongoing statement and reset the input prompt to the non-continuation prompt. s thingss ( thingsc s* � � ddg� � j�d�}� �tj� |S )N� >>> (� ... things� >>> )rY r_ r` ra r rq )�ignored� interruptedr6 r r �gotContinuation s zNManholeLoopbackMixin.test_interruptDuringContinuation.<locals>.gotContinuationc rT )N)rs rt rp ru rX )rv r6 r r �gotInterruption r] zNManholeLoopbackMixin.test_interruptDuringContinuation.<locals>.gotInterruptionr^ )r � continuingrx ry r r6 r � test_interruptDuringContinuation s z5ManholeLoopbackMixin.test_interruptDuringContinuationc s> � � d� � j�d�}� fdd�}� fdd�}|�|��|�S )z2 Evaluate cancelling with CTRL-\. rn c s. � � dg� � �tj� � jj}� �|tj�S )Nro ) rY ra r �CTRL_BACKSLASHr_ �onDisconnection� assertFailurer �ConnectionDone)r[ rP r6 r r �gotPartialLine* s zBManholeLoopbackMixin.test_ControlBackslash.<locals>.gotPartialLinec � � � dg� d S )N� rX rZ r6 r r �gotClearedLine1 � zBManholeLoopbackMixin.test_ControlBackslash.<locals>.gotClearedLine�ra r_ r` rb )r � partialLiner� r� r r6 r �test_ControlBackslash# s z*ManholeLoopbackMixin.test_ControlBackslashc c s� � | � d� | j�d�V | �dg� | � tjd � | j�d�V | �dg� | � d� | j�d�V | � tj� | jj}| �|tj �V dS ) z~ A CTRL+D in the middle of a line doesn't close a connection, but at the beginning of a line it does. s 1 + 1� \+ 1rU � + 1� >>> 1 + 1 + 1� s 3 >>> N) ra r_ r` rY r �CTRL_Dr} r~ r r rO r r r � test_controlD6 s � z"ManholeLoopbackMixin.test_controlDc c sV � | � d� | j�d�V | �ddg� | � tjd � | j�d�V | �dg� dS ) a CTRL+L is generally used as a redraw-screen command in terminal applications. Manhole doesn't currently respect this usage of it, but it should at least do something reasonable in response to this event (rather than, say, eating your face). s 1 + 1r� ru rU r� s 1 \+ 1 \+ 1r� N)ra r_ r` rY r �CTRL_Lr6 r r r � test_ControlLK s � z"ManholeLoopbackMixin.test_ControlLc �, � � d� � j�d�}� fdd�}|�|�S )zl CTRL-A can be used as HOME - returning cursor to beginning of current line buffer. s rint "hello"p� print "hello"c r� �Ns >>> print "hello"rX ��ignorer6 r r �cbe r� z.ManholeLoopbackMixin.test_controlA.<locals>.cbr� �r rP r� r r6 r � test_controlA] � z"ManholeLoopbackMixin.test_controlAc r� )zc CTRL-E can be used as END - setting cursor to end of current line buffer. s rint "hellop"r� c r� r� rX r� r6 r r r� r r� z.ManholeLoopbackMixin.test_controlE.<locals>.cbr� r� r r6 r � test_controlEj r� z"ManholeLoopbackMixin.test_controlEc c sR � | � d� | j�d�V | � d� | j�d�V | j�d�V | �g d�� dS )z� When a deferred is returned to the manhole REPL, it is displayed with a sequence number, and when the deferred fires, the result is printed. sC from twisted.internet import defer, reactor d = defer.Deferred() d � <Deferred #0>s. c = reactor.callLater(0.1, d.callback, 'Hi!') ru s# Deferred #0 called back: 'Hi!' >>> )s/ >>> from twisted.internet import defer, reactors >>> d = defer.Deferred()s >>> dr� s1 >>> c = reactor.callLater(0.1, d.callback, 'Hi!')s Deferred #0 called back: 'Hi!'ru N)ra r_ r` rY r6 r r r � test_deferredw s �� �z"ManholeLoopbackMixin.test_deferredN)r# r$ r% r �ColoredManhole�serverProtocolrQ re rg r= r> rm rr r{ r� r �inlineCallbacksr� r� r� r� r� r r r r rM � s& rM c @ s e Zd ZdZdS )�ManholeLoopbackTelnetTestsz, Test manhole loopback over Telnet. N)r# r$ r% r& r r r r r� � s r� c @ s e Zd ZdZedu rdZdS dS )�ManholeLoopbackSSHTestsz) Test manhole loopback over SSH. Nz!cryptography requirements missing)r# r$ r% r& r �skipr r r r r� � s �r� c @ s& e Zd ZdZedu rdZdS ejZdS )�ManholeLoopbackStdioTestsz1 Test manhole loopback over standard IO. N� Terminal requirements missing)r# r$ r% r& r r� �ConsoleManholer� r r r r r� � s r� c @ s$ e Zd ZdZedu r dZdd� ZdS )�ManholeMainTestsz= Test the I{main} method from the I{manhole} module. Nr� c C s* | j ttjdgd�}| �d|jd � dS )z� Will raise an exception when called with an argument which is a dotted patch which can not be imported.. z no-such-class)�argvzEmpty module namer N)� assertRaises� ValueErrorr �mainrK �args)r � exceptionr r r �test_mainClassNotFound� s �z'ManholeMainTests.test_mainClassNotFound)r# r$ r% r& r r� r� r r r r r� � s r� )"r& r �typingr r �bool�__annotations__� twisted.conchr �twisted.conch.insultsr � twisted.conch.test.test_recvliner r r r �twisted.internetr r �twisted.test.proto_helpersr � twisted.trialr r rk �TestCaser r( r2 rM r� r� r� r� r r r r �<module> s, B PK F��Z�B`-n� n� ) __pycache__/test_userauth.cpython-310.pycnu �[��� o �b� � @ s U d Z ddlmZ ddlmZ ddlmZ ddlmZm Z ddl m Z ddl m Z mZmZ ddlmZ dd lmZmZ dd lmZmZ dd lmZ dd lmZ dd l m Z dZ ee e d< ed�r�ed�r�ddl m!Z! ddl"m Z m#Z#m$Z$ ddl%m&Z& ddl'm(Z( nG dd� d�Z#G dd� d�Z$G dd� de$j)�Z*G d d � d e$j)�Z+G d d � d e$j)�Z,G d d!� d!e#j-�Z.ee�G d"d#� d#��Z/ee �G d$d%� d%��Z0ee �G d&d'� d'��Z1ee �G d(d)� d)��Z2G d*d+� d+e j3�Z4G d,d-� d-e j3�Z5G d.d/� d/e j3�Z6G d0d1� d1e j3�Z7dS )2zT Tests for the implementation of the ssh-userauth service. Maintainer: Paul Swartz � )� ModuleType)�Optional)� implementer)� ConchError�ValidPublicKey)�ICredentialsChecker)� IAnonymous�ISSHPrivateKey�IUsernamePassword)�UnauthorizedLogin)�IRealm�Portal)�defer�task)�loopback)� requireModule)�unittestN�keys� cryptography�pyasn1)�SSHProtocolChecker)r � transport�userauth)�NS)�keydatac @ � e Zd ZG dd� d�ZdS )r c @ � e Zd ZdZdS )ztransport.SSHTransportBase�Q A stub class so that later class definitions won't die. N��__name__� __module__� __qualname__�__doc__� r# r# �B/usr/lib/python3/dist-packages/twisted/conch/test/test_userauth.py�SSHTransportBase" � r% N)r r r! r% r# r# r# r$ r ! � r c @ r )r c @ r )zuserauth.SSHUserAuthClientr Nr r# r# r# r$ �SSHUserAuthClient( r& r( N)r r r! r( r# r# r# r$ r ' r' r c @ s2 e Zd ZdZdd� Zdd� Zd dd�Zd d � ZdS ) �ClientUserAuthz" A mock user auth client. c C s( | j r tj�tj�S t�tj�tj��S )z� If this is the first time we've been called, return a blob for the DSA key. Otherwise, return a blob for the RSA key. ) � lastPublicKeyr �Key� fromStringr �publicRSA_opensshr �succeed�publicDSA_openssh��selfr# r# r$ � getPublicKey3 s zClientUserAuth.getPublicKeyc C s t �tj�tj��S )z@ Return the private key object for the RSA key. )r r. r r+ r, r �privateRSA_opensshr0 r# r# r$ � getPrivateKey> � z ClientUserAuth.getPrivateKeyNc C � t �d�S )z/ Return 'foo' as the password. � foo�r r. )r1 �promptr# r# r$ � getPasswordD � zClientUserAuth.getPasswordc C r6 )z> Return 'foo' as the answer to two questions. )�foor< r8 )r1 �name� information�answersr# r# r$ �getGenericAnswersJ r; z ClientUserAuth.getGenericAnswers�N)r r r! r" r2 r4 r: r@ r# r# r# r$ r) . s r) c @ � e Zd ZdZdd� Zdd� ZdS )� OldClientAuthz~ The old SSHUserAuthClient returned a cryptography key object from getPrivateKey() and a string from getPublicKey c C s t �tj�tj�j�S rA )r r. r r+ r, r r3 � keyObjectr0 r# r# r$ r4 W � zOldClientAuth.getPrivateKeyc C s t j�tj��� S rA )r r+ r, r r- �blobr0 r# r# r$ r2 Z s zOldClientAuth.getPublicKeyN�r r r! r" r4 r2 r# r# r# r$ rC Q s rC c @ rB )�ClientAuthWithoutPrivateKeyzP This client doesn't have a private key, but it does have a public key. c C � d S rA r# r0 r# r# r$ r4 c � z)ClientAuthWithoutPrivateKey.getPrivateKeyc C s t j�tj�S rA )r r+ r, r r- r0 r# r# r$ r2 f � z(ClientAuthWithoutPrivateKey.getPublicKeyNrG r# r# r# r$ rH ^ s rH c @ sL e Zd ZdZG dd� d�ZG dd� d�Zdd� Zdd � Zd d � Zd d � Z dS )� FakeTransporta_ L{userauth.SSHUserAuthServer} expects an SSH transport which has a factory attribute which has a portal attribute. Because the portal is important for testing authentication, we need to be able to provide an interesting portal object to the L{SSHUserAuthServer}. In addition, we want to be able to capture any packets sent over the transport. @ivar packets: a list of 2-tuples: (messageType, data). Each 2-tuple is a sent packet. @type packets: C{list} @param lostConnecion: True if loseConnection has been called on us. @type lostConnection: L{bool} c @ s e Zd ZdZdZdd� ZdS )zFakeTransport.ServicezW A mock service, representing the other service offered by the server. � nancyc C rI rA r# r0 r# r# r$ �serviceStarted� rJ z$FakeTransport.Service.serviceStartedN)r r r! r" r= rN r# r# r# r$ �Service{ s rO c @ � e Zd ZdZdd� ZdS )zFakeTransport.Factoryzg A mock factory, representing the factory that spawned this user auth service. c C s |dkrt jS dS )z2 Return our fake service. � noneN)rL rO )r1 r �servicer# r# r$ � getService� s �z FakeTransport.Factory.getServiceN)r r r! r" rS r# r# r# r$ �Factory� s rT c C s( | � � | _|| j_d| _| | _g | _d S �NF)rT �factory�portal�lostConnectionr �packets)r1 rW r# r# r$ �__init__� s zFakeTransport.__init__c C s | j �||f� dS )z8 Record the packet sent by the service. N)rY �append)r1 � messageType�messager# r# r$ � sendPacket� r5 zFakeTransport.sendPacketc C � dS )z� Pretend that this transport encrypts traffic in both directions. The SSHUserAuthServer disables password authentication if the transport isn't encrypted. Tr# )r1 � directionr# r# r$ � isEncrypted� s zFakeTransport.isEncryptedc C s d| _ d S �NT)rX r0 r# r# r$ �loseConnection� s z FakeTransport.loseConnectionN) r r r! r" rO rT rZ r^ ra rc r# r# r# r$ rL j s rL c @ rP )�Realmz� A mock realm for testing L{userauth.SSHUserAuthServer}. This realm is not actually used in the course of testing, so it returns the simplest thing that could possibly work. c G s t �|d d dd� f�S )Nr c S rI rA r# r# r# r# r$ �<lambda>� � z%Realm.requestAvatar.<locals>.<lambda>r8 )r1 �avatarId�mind� interfacesr# r# r$ � requestAvatar� s zRealm.requestAvatarN)r r r! r" rj r# r# r# r$ rd � s rd c @ � e Zd ZdZefZdd� ZdS )�PasswordCheckerz� A very simple username/password checker which authenticates anyone whose password matches their username and rejects all others. c C s& |j |jkr t�|j �S t�td��S )Nz Invalid username/password pair)�username�passwordr r. �failr )r1 �credsr# r# r$ �requestAvatarId� s z PasswordChecker.requestAvatarIdN)r r r! r" r �credentialInterfacesrq r# r# r# r$ rl � � rl c @ rk )�PrivateKeyCheckerz� A very simple public key checker which authenticates anyone whose public/private keypair is the same keydata.public/privateRSA_openssh. c C sX |j tj�tj�� � kr)|jd ur&tj�|j �}|�|j|j�r#|j S t � �t � �t � �rA ) rF r r+ r, r r- � signature�verify�sigDatarm r r )r1 rp �objr# r# r$ rq � s �z!PrivateKeyChecker.requestAvatarIdN)r r r! r" r rr rq r# r# r# r$ rt � rs rt c @ rk )�AnonymousCheckerzI A simple checker which isn't supported by L{SSHUserAuthServer}. c C rI rA r# )r1 � credentialsr# r# r$ rq � s z AnonymousChecker.requestAvatarIdN)r r r! r" r rr rq r# r# r# r$ ry � s ry c @ s� e Zd ZdZedu r dZdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� ZdS ),�SSHUserAuthServerTestsz& Tests for SSHUserAuthServer. N� cannot run without cryptographyc C sb t � | _t| j�| _| j�t� � | j�t� � t�� | _ t | j�| j _ | j � � | j j �� d S rA )rd �realmr rW �registerCheckerrl rt r �SSHUserAuthServer� authServerrL r rN �supportedAuthentications�sortr0 r# r# r$ �setUp� s z SSHUserAuthServerTests.setUpc C � | j �� d | _ d S rA )r� �serviceStoppedr0 r# r# r$ �tearDown� � z SSHUserAuthServerTests.tearDownc C s( | � | jjjd tjtd�d f� dS )z; Check that the authentication has failed. ���s password,publickey� N)� assertEqualr� r rY r �MSG_USERAUTH_FAILUREr �r1 �ignoredr# r# r$ � _checkFailed� s �z#SSHUserAuthServerTests._checkFailedc C s, | j �td�td� td� �}|�| j�S )z� A client may request a list of authentication 'method name' values that may continue by using the "none" authentication 'method name'. See RFC 4252 Section 5.2. r7 s servicerQ )r� �ssh_USERAUTH_REQUESTr � addCallbackr� )r1 �dr# r# r$ �test_noneAuthentication s � z.SSHUserAuthServerTests.test_noneAuthenticationc sF d� td�td�td�dtd�g�}� j�|�}� fdd�}|�|�S )z� When provided with correct password authentication information, the server should respond by sending a MSG_USERAUTH_SUCCESS message with no other data. See RFC 4252, Section 5.1. � r7 rQ � passwordr� c � � � � jjjtjdfg� d S �Nr� �r� r� r rY r �MSG_USERAUTH_SUCCESS�r� r0 r# r$ �check � �zKSSHUserAuthServerTests.test_successfulPasswordAuthentication.<locals>.check)�joinr r� r� r� �r1 �packetr� r� r# r0 r$ �%test_successfulPasswordAuthentication s $ z<SSHUserAuthServerTests.test_successfulPasswordAuthenticationc C sh d� td�td�td�dtd�g�}t�� | j_| j�|�}| �| jjj g � | jj� d� |� | j �S )a; When provided with invalid authentication details, the server should respond by sending a MSG_USERAUTH_FAILURE message which states whether the authentication was partially successful, and provides other, open options for authentication. See RFC 4252, Section 5.1. r� r7 rQ r� r� � bar� ) r� r r �Clockr� �clockr� r� r rY �advancer� r� �r1 r� r� r# r# r$ �!test_failedPasswordAuthentication' s $ z8SSHUserAuthServerTests.test_failedPasswordAuthenticationc s� t j�tj��� }t j�tj�}td�td� td� d t|�� � t|� }d� j j _ |� td�t tjf� | �}|t|�7 }� j �|�}� fdd�}|�|�S )zN Test that private key authentication completes successfully, r7 rQ � publickey� �� testc r� r� r� r� r0 r# r$ r� M r� zMSSHUserAuthServerTests.test_successfulPrivateKeyAuthentication.<locals>.check)r r+ r, r r- rF r3 r �sshTyper� r � sessionID�sign�bytesr �MSG_USERAUTH_REQUESTr� r� )r1 rF rx r� ru r� r� r# r0 r$ �'test_successfulPrivateKeyAuthentication8 s, ��� ��� � z>SSHUserAuthServerTests.test_successfulPrivateKeyAuthenticationc s� t �� � dd� }dd� }� fdd�}| �| jd|� | �| jd|� | �| jd |� td �td � td � td � }| j�|� | �� t�S )z� ssh_USERAUTH_REQUEST should raise a ConchError if tryAuth returns None. Added to catch a bug noticed by pyflakes. c S s | � d� d S )Nz&request should have raised ConochError)ro r� r# r# r$ �mockCbFinishedAuth\ rK zOSSHUserAuthServerTests.test_requestRaisesConchError.<locals>.mockCbFinishedAuthc S rI rA r# )�kind�user�datar# r# r$ � mockTryAuth_ rJ zHSSHUserAuthServerTests.test_requestRaisesConchError.<locals>.mockTryAuthc s � � | j� d S rA )�errback�value)�reason�r� r# r$ � mockEbBadAuthb s zJSSHUserAuthServerTests.test_requestRaisesConchError.<locals>.mockEbBadAuth�tryAuth�_cbFinishedAuth� _ebBadAuths userrQ s public-keys data)r �Deferred�patchr� r r� � assertFailurer )r1 r� r� r� r� r# r� r$ � test_requestRaisesConchErrorU s z3SSHUserAuthServerTests.test_requestRaisesConchErrorc sb t j�tj��� � td�td� td� d td� t� � }�j�|�}� �fdd�}|� |�S )z@ Test that verifying a valid private key works. r7 rQ r� r� � ssh-rsac s* �� �jjjtjtd�t� � fg� d S )Nr� )r� r� r rY r �MSG_USERAUTH_PK_OKr r� �rF r1 r# r$ r� ~ s �z@SSHUserAuthServerTests.test_verifyValidPrivateKey.<locals>.check) r r+ r, r r- rF r r� r� r� r� r# r� r$ �test_verifyValidPrivateKeyo s ������ z1SSHUserAuthServerTests.test_verifyValidPrivateKeyc C sV t j�tj��� }td�td� td� d td� t|� }| j�|�}|� | j �S )�d Test that private key authentication fails when the public key is invalid. r7 rQ r� r� s ssh-dsa� r r+ r, r r/ rF r r� r� r� r� �r1 rF r� r� r# r# r$ �3test_failedPrivateKeyAuthenticationWithoutSignature� s ������ zJSSHUserAuthServerTests.test_failedPrivateKeyAuthenticationWithoutSignaturec C s| t j�tj��� }t j�tj�}td�td� td� d td� t|� t|�|�� }d| j j _ | j � |�}|� | j�S )r� r7 rQ r� r� r� r� )r r+ r, r r- rF r3 r r� r� r r� r� r� r� )r1 rF rx r� r� r# r# r$ �0test_failedPrivateKeyAuthenticationWithSignature� s& ����� �� zGSSHUserAuthServerTests.test_failedPrivateKeyAuthenticationWithSignaturec C sj t j�tj��� }td�|dd� }td�td� td� d td� t|� }| j�|�}|� | j �S ) z� Private key authentication fails when the public key type is unsupported or the public key is corrupt. s ssh-bad-type� Nr7 rQ r� r� r� r� r� r# r# r$ �test_unsupported_publickey� s ������ z1SSHUserAuthServerTests.test_unsupported_publickeyc C sR t �� }t| j�|_| j�t� � |�� |�� |j � � | � |j ddg� dS )ah L{SSHUserAuthServer} sets up C{SSHUserAuthServer.supportedAuthentications} by checking the portal's credentials interfaces and mapping them to SSH authentication method strings. If the Portal advertises an interface that L{SSHUserAuthServer} can't map, it should be ignored. This is a white box test. r� r� N) r r rL rW r r~ ry rN r� r� r� r� �r1 �serverr# r# r$ � test_ignoreUnknownCredInterfaces� s z7SSHUserAuthServerTests.test_ignoreUnknownCredInterfacesc C s� | � d| jj� t�� }t| j�|_dd� |j_|� � |� � | � d|j� t�� }t| j�|_dd� |j_|� � |� � | � d|j� dS )z� Test that the userauth service does not advertise password authentication if the password would be send in cleartext. r� c S r_ rU r# ��xr# r# r$ re � rf zISSHUserAuthServerTests.test_removePasswordIfUnencrypted.<locals>.<lambda>c S � | dkS �N�inr# r� r# r# r$ re � � N) �assertInr� r� r r rL rW r ra rN r� � assertNotIn)r1 �clearAuthServer�halfAuthServerr# r# r$ � test_removePasswordIfUnencrypted� s z7SSHUserAuthServerTests.test_removePasswordIfUnencryptedc C s� t | j�}|�t� � t�� }t|�|_dd� |j_|� � |� � | � |j dg� t�� }t|�|_dd� |j_|� � |� � | � |j dg� dS )z� If the L{SSHUserAuthServer} is not advertising passwords, then an unencrypted connection should not cause any warnings or exceptions. This is a white box test. c S r_ rU r# r� r# r# r$ re � rf zSSSHUserAuthServerTests.test_unencryptedConnectionWithoutPasswords.<locals>.<lambda>r� c S r� r� r# r� r# r# r$ re � r� N) r r} r~ rt r r rL r ra rN r� r� r� )r1 rW r� r� r# r# r$ �*test_unencryptedConnectionWithoutPasswords� s zASSHUserAuthServerTests.test_unencryptedConnectionWithoutPasswordsc C s� t �� }t�� |_t| j�|_|�� |j� d� |� � | � |jj tj dttjf� td� td� fg� | �|jj� dS )z0 Test that the login times out. 鰚 � s you took too longr� N)r r r r� r� rL rW r rN r� r� r� rY �MSG_DISCONNECTr� �)DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLEr � assertTruerX �r1 �timeoutAuthServerr# r# r$ �test_loginTimeout s( ������ z(SSHUserAuthServerTests.test_loginTimeoutc C s\ t �� }t�� |_t| j�|_|�� |� � |j� d� | � |jj g � | � |jj� dS )zN Test that stopping the service also stops the login timeout. r� N)r r r r� r� rL rW r rN r� r� r� rY � assertFalserX r� r# r# r$ �test_cancelLoginTimeout s z.SSHUserAuthServerTests.test_cancelLoginTimeoutc sn d� td�td�td�dtd�g�}t�� � j_td�D ]}� j�|�}� jj�d� q � fd d �}|� |�S ) zm Test that the server disconnects if the client fails authentication too many times. r� r7 rQ r� r� r� � r� c s< � � � jjjd tjdttjf� td� td� f� d S )Nr� r� s too many bad authsr� )r� r� r rY r� r� r� r r� r0 r# r$ r� 1 s �����z:SSHUserAuthServerTests.test_tooManyAttempts.<locals>.check) r� r r r� r� r� �ranger� r� r� )r1 r� �ir� r� r# r0 r$ �test_tooManyAttempts& s $ z+SSHUserAuthServerTests.test_tooManyAttemptsc C sH t d�t d� t d� d t d� }t�� | j_| j�|�}|�| j�S )zo If the user requests a service that we don't support, the authentication should fail. r7 r� r� r� )r r r� r� r� r� r� r� r� r# r# r$ �test_failIfUnknownService? s $ z0SSHUserAuthServerTests.test_failIfUnknownServicec sV dd� }� � � jd|� � � � jdd� � fdd�}� j�ddd�}� �|t��|�S ) aZ tryAuth() has two edge cases that are difficult to reach. 1) an authentication method auth_* returns None instead of a Deferred. 2) an authentication type that is defined does not have a matching auth_* method. Both these cases should return a Deferred which fails with a ConchError. c S rI rA r# )r� r# r# r$ �mockAuthU rJ z>SSHUserAuthServerTests.test_tryAuthEdgeCases.<locals>.mockAuth�auth_publickey� auth_passwordNc s � j �dd d �}� �|t�S )Nr� )r� r� r� r )r� �d2r0 r# r$ � secondTest[ s z@SSHUserAuthServerTests.test_tryAuthEdgeCases.<locals>.secondTestr� )r� r� r� r� r r� )r1 r� r� �d1r# r0 r$ �test_tryAuthEdgeCasesI s z,SSHUserAuthServerTests.test_tryAuthEdgeCases)r r r! r" r �skipr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r# r# r# r$ r{ � s0 r{ c @ s� e Zd ZdZedu r dZdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� ZdS )"�SSHUserAuthClientTestsz& Tests for SSHUserAuthClient. Nr| c C s4 t dt�� �| _td �| j_d| jj_| j�� d S )Nr7 r� )r) rL rO � authClientr r� rN r0 r# r# r$ r� k s z SSHUserAuthClientTests.setUpc C r� rA )r� r� r0 r# r# r$ r� q r� z SSHUserAuthClientTests.tearDownc C sT | � | jjd� | � | jjjd� | � | jjjtjt d�t d� t d� fg� dS )z; Test that client is initialized properly. r7 rM rQ N) r� r� r� �instancer= r rY r r� r r0 r# r# r$ � test_initu s �z SSHUserAuthClientTests.test_initc s@ dg� � fdd�}|| j j_| j �d� | �� d | j j� dS )z9 Test that the client succeeds properly. Nc s | � d< d S )Nr r# )rR �r� r# r$ �stubSetService� s zDSSHUserAuthClientTests.test_USERAUTH_SUCCESS.<locals>.stubSetServicer� r )r� r � setService�ssh_USERAUTH_SUCCESSr� r� )r1 r� r# r� r$ �test_USERAUTH_SUCCESS� s z,SSHUserAuthClientTests.test_USERAUTH_SUCCESSc C s� | j �td�d � | �| j jjd tjtd�td� td� d td� ttj � t j �� � � f� | j �td�d � ttj � t j�� � �}| �| j jjd tjtd�td� td� d td� | f� | j �td�ttj � t j�� � � � t| j jj�ttjf� td� td� td� d td� | }tj � t j�}| �| j jjd tjtd�td� td� d td� | t|�|�� f� d S ) zJ Test that the client can authenticate with a public key. r� r� r� r7 rM s ssh-dssr� � N)r� �ssh_USERAUTH_FAILUREr r� r rY r r� r r+ r, r r/ rF r- �ssh_USERAUTH_PK_OKr� r� r3 r� )r1 rF rw rx r# r# r$ �test_publickey� s� ������� ������� � �������� ����� ���z%SSHUserAuthClientTests.test_publickeyc C sz t dt�� �}td�|_d|j_|�� |�d� g |j_| �|� d�� | � |jjt j t d�t d� t d� fg� dS )z� If the SSHUserAuthClient doesn't return anything from signData, the client should start the authentication over again by requesting 'none' authentication. r7 Nr� r� r� rM rQ )rH rL rO r r� rN r� rY � assertIsNoner� r� r r� r )r1 r� r# r# r$ �!test_publickey_without_privatekey� s �z8SSHUserAuthClientTests.test_publickey_without_privatekeyc s. dd� � j _� j �d�}� fdd�}|�|�S )z{ If there's no public key, auth_publickey should return a Deferred called back with a False value. c S rI rA r# r� r# r# r$ re � rf z:SSHUserAuthClientTests.test_no_publickey.<locals>.<lambda>r� c s � � | � d S rA )r� ��resultr0 r# r$ r� � rK z7SSHUserAuthClientTests.test_no_publickey.<locals>.check)r� r2 r� r� )r1 r� r� r# r0 r$ �test_no_publickey� s z(SSHUserAuthClientTests.test_no_publickeyc C s� | j �td�d � | �| j jjd tjtd�td� td� d td� f� | j �td�td� � | �| j jjd tjtd�td� td� d td�d f� d S ) zx Test that the client can authentication with a password. This includes changing the password. r� r� r� r7 rM r� r� r� N) r� r� r r� r rY r r� r� r0 r# r# r$ � test_password� s "�� &��z$SSHUserAuthClientTests.test_passwordc C s" dd� | j _| �| j �d�� dS )zK If getPassword returns None, tryAuth should return False. c S rI rA r# r# r# r# r$ re rf z9SSHUserAuthClientTests.test_no_password.<locals>.<lambda>r� N)r� r: r� r� r0 r# r# r$ �test_no_password s z'SSHUserAuthClientTests.test_no_passwordc C s` | j �td�td� td� d td� d � | �| j jjd tjdtd� td� f� dS ) zj Make sure that the client can authenticate with the keyboard interactive method. r� s s Password: r� r� s r7 N)r� �'ssh_USERAUTH_PK_OK_keyboard_interactiver r� r rY r �MSG_USERAUTH_INFO_RESPONSEr0 r# r# r$ �test_keyboardInteractive s& ������ ��z/SSHUserAuthClientTests.test_keyboardInteractivec C sP d| j _g | j j_| j �d� | �| j jjtjtd�td� td� fg� dS )z� If C{SSHUserAuthClient} gets a MSG_USERAUTH_PK_OK packet when it's not expecting it, it should fail the current authentication and move on to the next type. s unknownr� r7 rM rQ N) r� �lastAuthr rY r� r� r r� r r0 r# r# r$ �"test_USERAUTH_PK_OK_unknown_method s �z9SSHUserAuthClientTests.test_USERAUTH_PK_OK_unknown_methodc s� � fdd�}� fdd�}|� j _|� j _� j �td�d � � �� j jjd tj td�td � td � d td� f� � j �td �d � � �� j jjd d� ddg� dS )z� ssh_USERAUTH_FAILURE should sort the methods by their position in SSHUserAuthClient.preferredOrder. Methods that are not in preferredOrder should be sorted at the end of that list. c s � j j�dd� d S )N� � here is data�r� r r^ r# r0 r# r$ �auth_firstmethod2 s zNSSHUserAuthClientTests.test_USERAUTH_FAILURE_sorting.<locals>.auth_firstmethodc s � j j�dd� dS )N� � other dataTr r# r0 r# r$ �auth_anothermethod5 s zPSSHUserAuthClientTests.test_USERAUTH_FAILURE_sorting.<locals>.auth_anothermethods anothermethod,passwordr� r� r7 rM r� s" firstmethod,anothermethod,passwordr� ���N)r r )r r ) r� r r r� r r� r rY r r� )r1 r r r# r0 r$ � test_USERAUTH_FAILURE_sorting+ s$ "�� ��z4SSHUserAuthClientTests.test_USERAUTH_FAILURE_sortingc C sT | j �td�d � | j �td�d � | �| j jjd tjdtd� d f� dS ) z� If there are no more available user authentication messages, the SSHUserAuthClient should disconnect with code DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE. r� r� r� r� s s( no more authentication methods availables N)r� r� r r� r rY r� r0 r# r# r$ �%test_disconnectIfNoMoreAuthenticationO s ����z<SSHUserAuthClientTests.test_disconnectIfNoMoreAuthenticationc C sH g | j j_| j �d� | �| j jjtjtd�td� td� fg� dS )z� _ebAuth (the generic authentication error handler) should send a request for the 'none' authentication method. Nr7 rM rQ )r� r rY �_ebAuthr� r r� r r0 r# r# r$ � test_ebAutha s �z"SSHUserAuthClientTests.test_ebAuthc s` t �dt�� �� ��� �� � � ��fdd�}� ��fdd��dd� �� �� }|��j�� |�S )z� getPublicKey() should return None. getPrivateKey() should return a failed Deferred. getPassword() should return a failed Deferred. getGenericAnswers() should return a failed Deferred. r7 c s$ | � t� � �� }|��j����S rA )�trap�NotImplementedErrorr: r� ro � addErrback�r r� )r� �check2r1 r# r$ r� v s z3SSHUserAuthClientTests.test_defaults.<locals>.checkc s* | � t� � �d d d �}|��j����S rA )r r r@ r� ro r r )r� �check3r1 r# r$ r { s z4SSHUserAuthClientTests.test_defaults.<locals>.check2c S s | � t� d S rA )r r r r# r# r$ r � rK z4SSHUserAuthClientTests.test_defaults.<locals>.check3) r r( rL rO r� r2 r4 r� ro r )r1 r� r� r# )r� r r r1 r$ � test_defaultsm s z$SSHUserAuthClientTests.test_defaults)r r r! r" r r� r� r� r� r� r� r r r r r r r r r r r# r# r# r$ r� c s& > $ r� c @ s. e Zd Zedu r dZG dd� d�Zdd� ZdS )� LoopbackTestsN�)cannot run without cryptography or PyASN1c @ s" e Zd ZG dd� d�Zdd� ZdS )zLoopbackTests.Factoryc @ rB )z LoopbackTests.Factory.Service� TestServicec C s | j �� d S rA )r rc r0 r# r# r$ rN � rK z,LoopbackTests.Factory.Service.serviceStartedc C rI rA r# r0 r# r# r$ r� � rJ z,LoopbackTests.Factory.Service.serviceStoppedN)r r r! r= rN r� r# r# r# r$ rO � s rO c C s | j S rA )rO )r1 �avatarr= r# r# r$ rS � s z LoopbackTests.Factory.getServiceN)r r r! rO rS r# r# r# r$ rT � s rT c s t �� �td�j�� �}t�� �_��j_dd� �j_t�� |_||j_d �j_ |j_ dd� �j_ |j_ ��� �j_ d�_ t � }t|�}t� � � �t� � � �t� � � fdd�� _|�� � |�jj _t��j|j�}dd� �jj_d d� |jj_��� |�� ��fd d �}|�|�S ) zW Test that the userauth server and client play nicely with each other. r7 c S r_ rb r# r� r# r# r$ re � rf z-LoopbackTests.test_loopback.<locals>.<lambda>r� c S rI rA r# r# r# r# r$ re � rf r c s t � j| �dkS )Nr� )�len�successfulCredentials)�aId)�checkerr# r$ re � s c S r_ )N�_ServerLoopbackr# r# r# r# r$ re � rf c S r_ )N�_ClientLoopbackr# r# r# r# r$ re � rf c s � � �jjjd� d S )Nr )r� r rR r= r� r� r# r$ r� � rE z*LoopbackTests.test_loopback.<locals>.check)r r r) rT rO r r% rR ra r� � sendKexInitrV � passwordDelayrd r r r~ rl rt �areDonerW r � loopbackAsync� logPrefixrN r� )r1 �clientr} rW r� r� r# )r% r1 r� r$ � test_loopback� s4 zLoopbackTests.test_loopback)r r r! r r� rT r. r# r# r# r$ r � s r c @ s e Zd Zedu r dZdd� ZdS )�ModuleInitializationTestsNr c C s, | � tjjd d� | � tjjd d� d S )N�< r� )r� r r �protocolMessagesr( r0 r# r# r$ � test_messages� s � �z'ModuleInitializationTests.test_messages)r r r! r r� r2 r# r# r# r$ r/ � s r/ )8r" �typesr �typingr �zope.interfacer �twisted.conch.errorr r �twisted.cred.checkersr �twisted.cred.credentialsr r r �twisted.cred.errorr �twisted.cred.portalr r �twisted.internetr r �twisted.protocolsr �twisted.python.reflectr � twisted.trialr r �__annotations__�twisted.conch.checkersr �twisted.conch.sshr r �twisted.conch.ssh.commonr �twisted.conch.testr r( r) rC rH r% rL rd rl rt ry �TestCaser{ r� r r/ r# r# r# r$ �<module> sR # A } &<PK F��Z_��u� u� * __pycache__/test_transport.cpython-310.pycnu �[��� o 5�@g� � @ sv U d Z ddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z mZ ddlmZ ddlmZ ddlmZmZmZ ddlmZ dd lmZ dd l m Z dd l m Z dd l m!Z! dd l"m#Z# ddl$m%Z% ddl&m'Z' e#d�Z(e#d�Z)e e* e+d< e(r�e)r�dZ,ddl-m.Z. ddl/m0Z0 ddl1m2Z2 ddl3m4Z4m5Z5 ddlm6Z6m7Z7m8Z8m9Z9 ddl:m;Z; e0� �<� Z=n e(s�dZ,ne)s�dZ,dZ=G d d � d �Z9G d d � d �Z7G d d!� d!�Z6d"d#� Z>d$d%� Z?G d&d'� d'e9j@�ZAG d(d)� d)�ZBG d*d+� d+�ZCG d,d-� d-ejD�ZEG d.d/� d/e7jF�ZGG d0d1� d1eG�ZHG d2d3� d3eG�ZId4d5� ZJG d6d7� d7e'�ZKG d8d9� d9�ZLG d:d;� d;�ZMG d<d=� d=�ZNG d>d?� d?�ZOG d@dA� dA�ZPG dBdC� dCePeK�ZQG dDdE� dEeP�ZRG dFdG� dGeReLeK�ZSG dHdI� dIeReMeK�ZTG dJdK� dKeReNeK�ZUe>G dLdM� dMeReOeK��ZVG dNdO� dO�ZWG dPdQ� dQeW�ZXG dRdS� dSeXeK�ZYG dTdU� dUeX�ZZG dVdW� dWeZeLeK�Z[G dXdY� dYeZeMeK�Z\G dZd[� d[eX�Z]G d\d]� d]e]eNeK�Z^e>G d^d_� d_e]eOeK��Z_G d`da� daeW�Z`G dbdc� dce`eK�ZaG ddde� dee`�ZbG dfdg� dgebeLeK�ZcG dhdi� diebeMeK�ZdG djdk� dke`�ZeG dldm� dmeeeNeK�Zfe>G dndo� doeeeOeK��ZgG dpdq� dqe'�ZhG drds� dse'�ZiG dtdu� due'�ZjdS )vz5 Tests for ssh/transport.py and the classes therein. � N)�md5�sha1�sha256�sha384�sha512)�Optional�Type)� __version__)� ConchError)�_kex�address�service)�defer)�loopback)� randbytes)� iterbytes��insecureRandom)� requireModule)� proto_helpers)�TestCase�pyasn1� cryptography�dependencySkip)�UnsupportedAlgorithm)�default_backend)� serialization)�dh�ec)�common�factory�keys� transport)�keydatazCannot run without PyASN1z can't run without cryptographyFc @ s6 e Zd ZG dd� d�ZG dd� d�ZG dd� d�ZdS )r" c @ � e Zd ZdS )ztransport.SSHTransportBaseN��__name__� __module__� __qualname__� r) r) �C/usr/lib/python3/dist-packages/twisted/conch/test/test_transport.py�SSHTransportBase5 � r+ c @ r$ )z transport.SSHServerTransportNr% r) r) r) r* �SSHServerTransport8 r, r- c @ r$ )z transport.SSHClientTransportNr% r) r) r) r* �SSHClientTransport; r, r. N)r& r' r( r+ r- r. r) r) r) r* r" 4 s r" c @ s e Zd ZG dd� d�ZdS )r c @ r$ )zfactory.SSHFactoryNr% r) r) r) r* � SSHFactory? r, r/ N)r& r' r( r/ r) r) r) r* r > s r c @ s e Zd Zedd� �ZdS )r c C � dS )N� r) )�self�argr) r) r* �NSC s z common.NSN)r& r' r( � classmethodr4 r) r) r) r* r B s r c C s t sd| _| S )Nz#x25519 not supported on this system)�X25519_SUPPORTED�skip��fr) r) r* �skipWithoutX25519H s r: c C s t �t| ||��S )z3 Return the MP version of C{(x ** y) % z}. )r �MP�pow)�x�y�zr) r) r* �_MPpowN � r@ c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�MockTransportBasean A base class for the client and server protocols. Stores the messages it receives instead of ignoring them. @ivar errors: a list of tuples: (reasonCode, description) @ivar unimplementeds: a list of integers: sequence number @ivar debugs: a list of tuples: (alwaysDisplay, message, lang) @ivar ignoreds: a list of strings: ignored data c C s. t j�| � g | _g | _g | _g | _d| _dS )z, Set up instance variables. N)r" r+ �connectionMade�errors�unimplementeds�debugs�ignoreds�gotUnsupportedVersion�r2 r) r) r* rC ` s z MockTransportBase.connectionMadec C s || _ tj�| |�S )zZ Intercept unsupported version call. @type remoteVersion: L{str} )rH r" r+ �_unsupportedVersionReceived)r2 � remoteVersionr) r) r* rJ k s �z-MockTransportBase._unsupportedVersionReceivedc C s | j �||f� dS )zp Store any errors received. @type reasonCode: L{int} @type description: L{str} N�rD �append)r2 � reasonCode� descriptionr) r) r* � receiveErrorv � z MockTransportBase.receiveErrorc C � | j �|� dS )zX Store any unimplemented packet messages. @type seqnum: L{int} N)rE rM )r2 �seqnumr) r) r* �receiveUnimplemented � z&MockTransportBase.receiveUnimplementedc C s | j �|||f� dS )z� Store any debug messages. @type alwaysDisplay: L{bool} @type message: L{str} @type lang: L{str} N)rF rM )r2 � alwaysDisplay�message�langr) r) r* � receiveDebug� s z MockTransportBase.receiveDebugc C rR )zG Store any ignored data. @type packet: L{str} N)rG rM �r2 �packetr) r) r* � ssh_IGNORE� rU z MockTransportBase.ssh_IGNOREN) r& r' r( �__doc__rC rJ rP rT rY r\ r) r) r) r* rB U s rB c @ sh e Zd ZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdZdd � Zd d � Zd d � Zdd� Zdd� ZdS )� MockCipherzH A mocked-up version of twisted.conch.ssh.transport.SSHCiphers. � test� � F)Nr1 r1 ra r) c C s> d| _ t|�| j dkr tdt|�| jt|�| j f ��|S )z~ Called to encrypt the packet. Simply record that encryption was used and return the data unchanged. Tr �*length %i modulo blocksize %i is not 0: %i)� usedEncrypt�len� encBlockSize� RuntimeError�r2 r= r) r) r* �encrypt� � ��zMockCipher.encryptc C s> d| _ t|�| j dkr tdt|�| jt|�| j f ��|S )z~ Called to decrypt the packet. Simply record that decryption was used and return the data unchanged. Tr rb )� usedDecryptrd re rf � decBlockSizerg r) r) r* �decrypt� ri zMockCipher.decryptc C s t |f�S )zs Make a Message Authentication Code by sending the character value of the outgoing packet. ��bytes)r2 �outgoingPacketSequence�payloadr) r) r* �makeMAC� s zMockCipher.makeMACc C s t |f�|kS )zy Verify the Message Authentication Code by checking that the packet sequence number is the same. rm )r2 �incomingPacketSequencer[ �macDatar) r) r* �verify� s zMockCipher.verifyc C s ||||||f| _ dS )z" Record the keys. N)r! )r2 �ivOut�keyOut�ivIn�keyIn�macIn�macOutr) r) r* �setKeys� s zMockCipher.setKeysN)r& r' r( r] � outCipTypere � inCipTyperk � inMACType� outMACType�verifyDigestSizerc rj �outMAC�inMACr! rh rl rq rt r{ r) r) r) r* r^ � s& r^ c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �MockCompressionz� A mocked-up compression, based on the zlib interface. Instead of compressing, it reverses the data and adds a 0x66 byte to the end. c C s |d d d� S �N���r) �r2 rp r) r) r* �compress� � zMockCompression.compressc C s |d d� d d d� S r� r) r� r) r) r* � decompress� � zMockCompression.decompressc C r0 )N� fr) �r2 �kindr) r) r* �flush� � zMockCompression.flushN)r& r' r( r] r� r� r� r) r) r) r* r� � s r� c @ sF e Zd ZdZdZdZdZddd�Zdd� Zd d � Z d d � Z d d� Z dS )� MockServicez� A mocked-up service, based on twisted.conch.ssh.service.SSHService. @ivar started: True if this service has been started. @ivar stopped: True if this service has been stopped. � MockServiceF�MSG_TEST� MSG_fiction)� �G c C r0 )Nr� r) rI r) r) r* � logPrefix� r� zMockService.logPrefixc C � d| _ dS )z6 Record that the service was started. TN)�startedrI r) r) r* �serviceStarted� � zMockService.serviceStartedc C r� )z6 Record that the service was stopped. TN)�stoppedrI r) r) r* �serviceStopped r� zMockService.serviceStoppedc C s | j �d|� dS )z: A message that this service responds to. r� N)r" � sendPacketrZ r) r) r* �ssh_TEST rA zMockService.ssh_TESTN) r& r' r( r] �namer� r� �protocolMessagesr� r� r� r� r) r) r) r* r� � s r� c @ s0 e Zd ZdZdeiZdd� Zdd� Zdd� Zd S ) � MockFactoryzL A mocked-up factory based on twisted.conch.ssh.factory.SSHFactory. � ssh-userauthc C � t j�tj�t j�tj�d�S )zG Return the public keys that authenticate this server. �� ssh-rsas ssh-dsa)r! �Key� fromStringr# �publicRSA_openssh�publicDSA_opensshrI r) r) r* � getPublicKeys � �zMockFactory.getPublicKeysc C r� )zH Return the private keys that authenticate this server. r� )r! r� r� r# �privateRSA_openssh�privateDSA_opensshrI r) r) r* �getPrivateKeys r� zMockFactory.getPrivateKeysc C s t �d�}|fdd�S )a/ Diffie-Hellman primes that can be used for key exchange algorithms that use group exchange to establish a prime / generator group. @return: The primes and generators. @rtype: L{dict} mapping the key size to a C{list} of C{(generator, prime)} tuple. � diffie-hellman-group14-sha1))� � )� i )r �getDHGeneratorAndPrime)r2 �group14r) r) r* � getPrimes' s zMockFactory.getPrimesN) r& r' r( r] r� �servicesr� r� r� r) r) r) r* r� s r� c @ � e Zd ZdZdd� ZdS )�MockOldFactoryPublicKeysz� The old SSHFactory returned mappings from key names to strings from getPublicKeys(). We return those here for testing. c C s4 t �| �}|�� dd� D ] \}}|�� ||< q |S )zJ We used to map key types to public key blobs as strings. N)r� r� �items�blob�r2 r! r� �keyr) r) r* r� ? s z&MockOldFactoryPublicKeys.getPublicKeysN)r& r' r( r] r� r) r) r) r* r� 9 � r� c @ r� )�MockOldFactoryPrivateKeysz� The old SSHFactory returned mappings from key names to cryptography key objects from getPrivateKeys(). We return those here for testing. c C s2 t �| �}|�� dd� D ] \}}|j||< q |S )zG We used to map key types to cryptography key objects. N)r� r� r� � keyObjectr� r) r) r* r� O s z(MockOldFactoryPrivateKeys.getPrivateKeysN)r& r' r( r] r� r) r) r) r* r� I r� r� c C s� | j }| j}|�� }tdd� td|d d�D ��}t|||�}zt�|t�|t� ||���� t � �| _ W n t yJ td|� d|� d|� d �� � w t�| j �� �� j�| _d S ) Nc s s � | ]}d |> V qdS )� Nr) ��.0r= r) r) r* � <genexpr>] s � z)generatePredictableKey.<locals>.<genexpr>r � � z p=z g=z x=� )�p�g� bit_length�sum�ranger< r �DHPrivateNumbers�DHPublicNumbers�DHParameterNumbers� private_keyr � dhSecretKey� ValueError�printr r; � public_key�public_numbersr> �dhSecretKeyPublicMP)r" r� r� �bitsr= r> r) r) r* �generatePredictableKeyY s$ �� � �r� c @ sH e Zd ZU dZdZeeej e d< e re Z dd� Z dd� Z dd � ZdS ) �TransportTestCasez. Base class for transport test cases. N�klassc sh t �� � _� �� � _g � _dd� }� �td|� t� t � j�� j_ � fdd�}� j� � j� |� j_ d S )Nc S s d| S )z; Return a consistent entropy value � �r) )rd r) r) r* � secureRandom| s z-TransportTestCase.setUp.<locals>.secureRandomr� c s � j �| |f� d S �N)�packetsrM )� messageTyperp rI r) r* �stubSendPacket� s z/TransportTestCase.setUp.<locals>.stubSendPacket)r �StringTransportr" r� �protor� �patchr �types� MethodTyper� �_startEphemeralDH�makeConnectionr� )r2 r� r� r) rI r* �setUpw s � zTransportTestCase.setUpc C s2 |� d� |�tj| j� |�dd� |j|_dS )z� Deliver enough additional messages to C{proto} so that the key exchange which is started in L{SSHTransportBase.connectionMade} completes and non-key exchange messages can be sent and received. s SSH-2.0-BogoClient-1.2i � foos barN)� dataReceived�dispatchMessager" � MSG_KEXINIT�_A_KEXINIT_MESSAGE� _keySetup�_KEY_EXCHANGE_NONE�_keyExchangeState�r2 r� r) r) r* �finishKeyExchange� s z#TransportTestCase.finishKeyExchangec C s& | j j| j _g | j _| j �||� dS )z� Finish a key exchange by calling C{_keySetup} with the given arguments. Also do extra whitebox stuff to satisfy that method's assumption that some kind of key exchange has actually taken place. N)r� �_KEY_EXCHANGE_REQUESTEDr� �_blockedByKeyExchanger� )r2 � sharedSecret� exchangeHashr) r) r* �simulateKeyExchange� s z%TransportTestCase.simulateKeyExchange)r& r' r( r] r� r r r" r+ �__annotations__r r7 r� r� r� r) r) r) r* r� m s r� c @ � e Zd ZdZdZeZdS )�DHGroupExchangeSHA1Mixinz= Mixin for diffie-hellman-group-exchange-sha1 tests. �" diffie-hellman-group-exchange-sha1N)r& r' r( r] � kexAlgorithmr � hashProcessorr) r) r) r* r� � � r� c @ r� )�DHGroupExchangeSHA256Mixinz? Mixin for diffie-hellman-group-exchange-sha256 tests. �$ diffie-hellman-group-exchange-sha256N�r& r' r( r] r� r r� r) r) r) r* r� � r� r� c @ r� )� ECDHMixinz8 Mixin for elliptic curve diffie-hellman tests. � ecdh-sha2-nistp256Nr r) r) r) r* r � r� r c @ r� )�Curve25519SHA256Mixinz, Mixin for curve25519-sha256 tests. s curve25519-sha256Nr r) r) r) r* r � r� r c @ s( e Zd ZU dZeZeeej e d< dS )�BaseSSHTransportBaseCasez, Base case for TransportBase tests. r� N) r& r' r( r] rB r� r r r" r+ r� r) r) r) r* r � s r c @ s� e Zd ZdZer eZde�d� e�d� e�d� e�d� e�d� e�d� e�d� e�d� e�d� e�d� d d Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z!d=d>� Z"d?d@� Z#dAdB� Z$dCdD� Z%dEdF� Z&dGdH� Z'dIdJ� Z(dKdL� Z)dMdN� Z*dOdP� Z+dQdR� Z,dSdT� Z-dUdV� Z.dWdX� Z/dYdZ� Z0d[d\� Z1d]d^� Z2d_d`� Z3daS )b�BaseSSHTransportTestszs Test TransportBase. It implements the non-server/client specific parts of the SSH transport protocol. � ����������������r� r� � aes256-ctr� hmac-sha1� noner1 � � c C sp | j �� �dd�d }| �|dt�d� � |�d�td�d� }dd �d d � t j D �� d }| � ||� dS ) a Test that the first thing sent over the connection is the version string. The 'softwareversion' part must consist of printable US-ASCII characters, with the exception of whitespace characters and the minus sign. RFC 4253, section 4.2. � ra r s SSH-2.0-Twisted_�asciizSSH-2.0-Nz^(�|c s s* � | ]}|d kr|� � st�|�V qdS )�-N)�isspace�re�escape�r� �cr) r) r* r� � s � �z9BaseSSHTransportTests.test_sendVersion.<locals>.<genexpr>z)*$) r" �value�split� assertEqual�twisted_version�encode�decoderd �join�string� printable� assertRegex)r2 �version�softwareVersion�softwareVersionRegexr) r) r* �test_sendVersion� s ����z&BaseSSHTransportTests.test_sendVersionc C sl t � }|�| j� |�d� |�d� |�d� | �| jj� |�d� | �| jj� | �d| j�� � dS )z� When the peer is not sending its SSH version but keeps sending data, the connection is disconnected after 4KB to prevent buffering too much and running our of memory. s SSH-2-Server-Identifiers� 1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890s 1235678s 1234567s% Preventing a denial of service attackN) rB r� r" r� � assertFalse� disconnecting� assertTrue�assertInr �r2 �sutr) r) r* �'test_dataReceiveVersionNotSentMemoryDOS s z=BaseSSHTransportTests.test_dataReceiveVersionNotSentMemoryDOSc C sX t � }|�| j� | �|� | j�� td�}d}|�||� | j�� }| �|d� dS )a@ Test that plain (unencrypted, uncompressed) packets are sent correctly. The format is:: uint32 length (including type and padding length) byte padding length byte type bytes[length-padding length-2] data bytes[padding length] padding �A� BCDEFGs ABCDEFG����N) rB r� r" r� �clear�ordr� r r )r2 r� rW rp r r) r) r* �test_sendPacketPlain s z*BaseSSHTransportTests.test_sendPacketPlainc C sp t � }|�| j� | �|� t� |_}td�}d}| j�� |�||� | � |j � | j� � }| � |d� dS )z� Test that packets sent while encryption is enabled are sent correctly. The whole packet should be encrypted. r* � BCs ABC����N) rB r� r" r� r^ �currentEncryptionsr- r, r� r% rc r r �r2 r� � testCipherrW rp r r) r) r* �test_sendPacketEncrypted0 s �z.BaseSSHTransportTests.test_sendPacketEncryptedc C sX t � }|�| j� | �|� t� |_| j�� |�td�d� | j� � }| � |d� dS )z� Test that packets sent while compression is enabled are sent correctly. The packet type and data should be encrypted. r* � Bs BAf��������N) rB r� r" r� r� �outgoingCompressionr, r� r- r r )r2 r� r r) r) r* �test_sendPacketCompressedM s �z/BaseSSHTransportTests.test_sendPacketCompressedc C sx t � }|�| j� | �|� t� |_}t� |_td�}d}| j� � |� ||� | � |j � | j� � }| �|d� dS )z� Test that packets sent while compression and encryption are enabled are sent correctly. The packet type and data should be compressed and then the whole packet should be encrypted. r* r/ s CBAf���������N)rB r� r" r� r^ r0 r� r5 r- r, r� r% rc r r r1 r) r) r* �test_sendPacketBoth] s �z)BaseSSHTransportTests.test_sendPacketBothc C sh t � }|�| j� | �|� | j�� |�td�d� | j�� d |_| � |� � d� | � |jd� dS )zt Test that packets are retrieved correctly out of the buffer when no encryption is enabled. r* r/ s extras ABCN) rB r� r" r� r, r� r- r �bufr � getPacketr� r) r) r* �test_getPacketPlain| s z)BaseSSHTransportTests.test_getPacketPlainc C s� t � }dd� |_|�| j� | j�� t� |_}|�td�d� | j� � }|dtj � |_ | � |� � � | �|j� | �|jd� | j |tj d� 7 _ | �|� � d� | �|j d� dS ) zl Test that encrypted packets are retrieved correctly. See test_sendPacketEncrypted. c S � d S r� r) r) r) r) r* �<lambda>� � z?BaseSSHTransportTests.test_getPacketEncrypted.<locals>.<lambda>r* � BCDNs A� ABCDr1 )rB � sendKexInitr� r" r, r^ r0 r� r- r rk r8 � assertIsNoner9 r% rj r �first)r2 r� r2 r r) r) r* �test_getPacketEncrypted� s z-BaseSSHTransportTests.test_getPacketEncryptedc C sf t � }|�| j� | �|� | j�� t� |_|j|_|�t d�d� | j� � |_ | � |� � d� dS )zo Test that compressed packets are retrieved correctly. See test_sendPacketCompressed. r* r> r? N)rB r� r" r� r, r� r5 �incomingCompressionr� r- r r8 r r9 r� r) r) r* �test_getPacketCompressed� s z.BaseSSHTransportTests.test_getPacketCompressedc C sn t � }dd� |_|�| j� | j�� t� |_t� |_|j|_ |� t d�d� | j� � |_ | �|�� d� dS )zv Test that compressed and encrypted packets are retrieved correctly. See test_sendPacketBoth. c S r; r� r) r) r) r) r* r< � r= z:BaseSSHTransportTests.test_getPacketBoth.<locals>.<lambda>r* r+ s ABCDEFGN)rB r@ r� r" r, r^ r0 r� r5 rD r� r- r r8 r r9 r� r) r) r* �test_getPacketBoth� s z(BaseSSHTransportTests.test_getPacketBothc C s> t �dddd�}d }}| jjD ] }| �|�|||�� qdS )z? Test that all the supportedCiphers are valid. � Ar4 � C� D� N)r" � SSHCiphersr� �supportedCiphersr% � _getCipher)r2 �ciphers�ivr� �cipNamer) r) r* �test_ciphersAreValid� s �z*BaseSSHTransportTests.test_ciphersAreValidc C st | j �� �dd�d }|| j_| j�� }| �|dd� tt jf�� | �|dd� d� t � |dd� d�\ }}}}}}} } } } } | �|d� | jj d g �� | �|d� | jj �� | �|d� | jj�� | �|d� | jj�� | �|d� | jj�� | �|d� | jj�� | �| d� | jj�� | �| d� | jj�� | �| d� | jj�� | �| d� | jj�� | �| d � dS ) a� Test that the KEXINIT (key exchange initiation) message is sent correctly. Payload:: bytes[16] cookie string key exchange algorithms string public key algorithms string outgoing ciphers string incoming ciphers string outgoing MACs string incoming MACs string outgoing compressions string incoming compressions bool first packet follows uint32 0 r ra r � s ����������������N� � ,s ext-info-ss )r" r r r� r8 r9 r rn r� r �getNSr �supportedKeyExchanges�supportedPublicKeysrL � supportedMACs�supportedCompressions�supportedLanguages)r2 r r[ � keyExchanges�pubkeys�ciphers1�ciphers2�macs1�macs2� compressions1� compressions2� languages1� languages2r8 r) r) r* �test_sendKexInit� s> ��z&BaseSSHTransportTests.test_sendKexInitc C s. | j �� | j�t j| j� | �| jg � dS )zy Immediately after connecting, the transport expects a KEXINIT message and does not reply to it. N)r" r, r� r� r� r� r r� rI r) r) r* �test_receiveKEXINITReply� s z.BaseSSHTransportTests.test_receiveKEXINITReplyc C sX | � | j� | jdd�= | j�tj| j� | �t| j�d� | �| jd d tj� dS )z� When a KEXINIT message is received which is not a reply to an earlier KEXINIT message which was sent, a KEXINIT reply is sent. Nra r ) r� r� r� r� r" r� r� r rd rI r) r) r* �test_sendKEXINITReply s z+BaseSSHTransportTests.test_sendKEXINITReplyc C s | � t| jj� dS )a( A new key exchange cannot be started while a key exchange is already in progress. If an attempt is made to send a I{KEXINIT} message using L{SSHTransportBase.sendKexInit} while a key exchange is in progress causes that method to raise a L{RuntimeError}. N)� assertRaisesrf r� r@ rI r) r) r* �test_sendKexInitTwiceFails rQ z0BaseSSHTransportTests.test_sendKexInitTwiceFailsc C s� t jt jg}| j �� | j`|D ]}| j�|d� | �| j �� d� q| �| j� t � | j_ | j� � | �| j �� � d�d� dS )z� After L{SSHTransportBase.sendKexInit} has been called, messages types other than the following are queued and not sent until after I{NEWKEYS} is sent by L{SSHTransportBase._keySetup}. RFC 4253, section 7.1. r� r1 � N) r" �MSG_SERVICE_REQUESTr� r, r� r� r r r� r^ �nextEncryptions�_newKeys�count)r2 �disallowedMessageTypesr� r) r) r* � test_sendKexInitBlocksOthers s � z2BaseSSHTransportTests.test_sendKexInitBlocksOthersc C sZ d| j _| j �ddg� | �| jtjdt�d� t�d� t�d� t�d� fg� d S ) ze Test that EXT_INFO messages are sent correctly. See RFC 8308, section 2.3. T�� server-sig-algs� ssh-rsa,rsa-sha2-256)� elevation� d� rr rs rt ru N) r� �_peerSupportsExtensions� sendExtInfor r� r" � MSG_EXT_INFOr r4 rI r) r) r* �test_sendExtInfo9 s* ���������z&BaseSSHTransportTests.test_sendExtInfoc C s | j �dg� | �| jg � dS )z� If the peer has not advertised support for extension negotiation, no EXT_INFO message is sent, since RFC 8308 only guarantees that the peer will be prepared to accept it if it has advertised support. rq N)r� rx r r� rI r) r) r* �test_sendExtInfoUnsupportedS s z1BaseSSHTransportTests.test_sendExtInfoUnsupportedc C sR | j �tjdt�d� t�d� t�d� t�d� � | �| j jddd�� dS )z� When an EXT_INFO message is received, the transport stores a mapping of the peer's advertised extensions. See RFC 8308, section 2.3. rv rr �! ssh-rsa,rsa-sha2-256,rsa-sha2-512� no-flow-control� s)rr r} N)r� r� r" ry r r4 r �peerExtensionsrI r) r) r* � test_EXT_INFO\ s$ �������z#BaseSSHTransportTests.test_EXT_INFOc C s* | j �ddd� | �| jtjdfg� dS )z� Test that debug messages are sent correctly. Payload:: bool always display string debug message string language r_ T� en� test enN)r� � sendDebugr r� r" � MSG_DEBUGrI r) r) r* �test_sendDebugq s �z$BaseSSHTransportTests.test_sendDebugc C s8 | j �tjd� | j �tjd� | �| j jddg� dS )zW Test that debug messages are received correctly. See test_sendDebug. r� s silent en)Tr_ r� )Fs silentr� N)r� r� r" r� r rF rI r) r) r* �test_receiveDebug~ s �� �z'BaseSSHTransportTests.test_receiveDebugc C s& | j �d� | �| jtjdfg� dS )zk Test that ignored messages are sent correctly. Payload:: string ignored data r_ s testN)r� � sendIgnorer r� r" � MSG_IGNORErI r) r) r* �test_sendIgnore� s �z%BaseSSHTransportTests.test_sendIgnorec C s& | j �tjd� | �| j jdg� dS )zb Test that ignored messages are received correctly. See test_sendIgnore. r_ N)r� r� r" r� r rG rI r) r) r* �test_receiveIgnore� � z(BaseSSHTransportTests.test_receiveIgnorec C s$ | j �� | �| jtjdfg� dS )zt Test that unimplemented messages are sent correctly. Payload:: uint32 sequence number r N)r� �sendUnimplementedr r� r" �MSG_UNIMPLEMENTEDrI r) r) r* �test_sendUnimplemented� s �z,BaseSSHTransportTests.test_sendUnimplementedc C s& | j �tjd� | �| j jdg� dS )zo Test that unimplemented messages are received correctly. See test_sendUnimplemented. s �r� N)r� r� r" r� r rE rI r) r) r* �test_receiveUnimplemented� r� z/BaseSSHTransportTests.test_receiveUnimplementedc sP dg� � fdd�}|| j _| j�dd� | �| jt jdfg� | �� d � dS ) z� Test that disconnection messages are sent correctly. Payload:: uint32 reason code string reason description string language Fc � d� d< d S �NTr r) r) �� disconnectedr) r* �stubLoseConnection� � zEBaseSSHTransportTests.test_sendDisconnect.<locals>.stubLoseConnectionr� r_ s � test r N)r" �loseConnectionr� �sendDisconnectr r� �MSG_DISCONNECTr% �r2 r� r) r� r* �test_sendDisconnect� s ��� z)BaseSSHTransportTests.test_sendDisconnectc sN dg� � fdd�}|| j _| j�t jd� | �| jjdg� | �� d � dS )zl Test that disconnection messages are received correctly. See test_sendDisconnect. Fc r� r� r) r) r� r) r* r� � r� zHBaseSSHTransportTests.test_receiveDisconnect.<locals>.stubLoseConnections � test�r� r_ r N)r" r� r� r� r� r rD r% r� r) r� r* �test_receiveDisconnect� s �z,BaseSSHTransportTests.test_receiveDisconnectc s` dg� � fdd�}|| j _| j �| j�� � | �| j j� | �| j j| j j � | �� d � dS )ze Test that dataReceived parses packets and dispatches them to ssh_* methods. Fc r� r� r) )r[ ��kexInitr) r* � stubKEXINIT� r� z<BaseSSHTransportTests.test_dataReceived.<locals>.stubKEXINITr N) r� � ssh_KEXINITr� r" r r% � gotVersionr �ourVersionString�otherVersionString)r2 r� r) r� r* �test_dataReceived� s z'BaseSSHTransportTests.test_dataReceivedc C s� t � }| j�|� | �| jj|� | �|j� | j�dd� | �| jdg� t � }| j�|� | �|j� | �|j � | j� d� | �|j � dS )z� Test that the transport can set the running service and dispatches packets to the service's packetReceived method. r� r_ r� N) r� r� � setServicer r r% r� r� r� r� �connectionLost)r2 r �service2r) r) r* � test_service� s z"BaseSSHTransportTests.test_servicec s@ dg� � fdd�}|| j _d| j _| j �d� | �� d � dS )zP Test that the transport notifies the avatar of disconnections. Fc r� r� r) r) r� r) r* �logout r� z1BaseSSHTransportTests.test_avatar.<locals>.logoutTNr )r� �logoutFunction�avatarr� r% )r2 r� r) r� r* � test_avatar� s z!BaseSSHTransportTests.test_avatarc C �� | � | j�d�� | � | j�d�� | � | j�d�� t� | j_| �| j�d�� | �| j�d�� | �| j�d�� t�dddd�| j_| � | j�d�� | � | j�d�� | � | j�d�� | �t | jjd� dS )zS Test that the transport accurately reflects its encrypted status. �in�out�bothr �badN) r# r� � isEncryptedr^ r0 r% r" rK rh � TypeErrorrI r) r) r* �test_isEncrypted � �z&BaseSSHTransportTests.test_isEncryptedc C r� )zR Test that the transport accurately reflects its verified status. r� r� r� r r� N) r# r� � isVerifiedr^ r0 r% r" rK rh r� rI r) r) r* �test_isVerified! r� z%BaseSSHTransportTests.test_isVerifiedc sf dg� � fdd�}|| j _| j�� | �| jd d t j� | �| jd d dd� tt jf�� dS ) zh Test that loseConnection sends a disconnect message and closes the connection. Fc r� r� r) r) r� r) r* r� < r� zEBaseSSHTransportTests.test_loseConnection.<locals>.stubLoseConnectionr ra r� r� N)r" r� r� r r� r� rn �DISCONNECT_CONNECTION_LOSTr� r) r� r* �test_loseConnection5 s �z)BaseSSHTransportTests.test_loseConnectionc s( � fdd�}|d� |d� |d� dS )zU Test that the transport disconnects when it receives a bad version. c s� g �_ d�j_dg� � fdd�}|�j_t| d �D ]}�j�|� q��� d � ���j d d tj � ���j d d dd� t tj f�� d S ) NFc r� r� r) r) r� r) r* r� P r� zRBaseSSHTransportTests.test_badVersion.<locals>.testBad.<locals>.stubLoseConnectionr r ra r� r� ) r� r� r� r" r� r r� r% r r� rn �)DISCONNECT_PROTOCOL_VERSION_NOT_SUPPORTED)r r� r rI r� r* �testBadK s �z6BaseSSHTransportTests.test_badVersion.<locals>.testBads SSH-1.5-OpenSSHs SSH-3.0-Twisteds GET / HTTP/1.1Nr) )r2 r� r) rI r* �test_badVersionF s z%BaseSSHTransportTests.test_badVersionc sX t � � � �t�� � d� j d }� fdd�t|�D � | �� j� | �� j � j� dS )zV Test that the transport ignores data sent before the version string. s5 here's some stuff beforehand here's some other stuff r c s g | ]}� � |��qS r) )r� r �r� r) r* � <listcomp>n s z@BaseSSHTransportTests.test_dataBeforeVersion.<locals>.<listcomp>N) rB r� r r� r� r r% r� r r� �r2 �datar) r� r* �test_dataBeforeVersiona s ��� z,BaseSSHTransportTests.test_dataBeforeVersionc C �< t � }|�t�� � |�d� | �|j� | �|jd� dS )zw Test that the transport treats the compatibility version (1.99) as equivalent to version 2.0. s SSH-1.99-OpenSSH s SSH-1.99-OpenSSHN� rB r� r r� r� r% r� r r� r� r) r) r* �test_compatabilityVersionr s z/BaseSSHTransportTests.test_compatabilityVersionc C r� )z� It can parse the SSH version string even when it ends only in Unix newlines (CR) and does not follows the RFC 4253 to use network newlines (CR LF). s, SSH-2.0-PoorSSHD Some-comment here more-datas" SSH-2.0-PoorSSHD Some-comment hereNr� r' r) r) r* �&test_dataReceivedSSHVersionUnixNewline} s z<BaseSSHTransportTests.test_dataReceivedSSHVersionUnixNewlinec C r� )a9 The trailing spaces from SSH version comment are not removed. The SSH version string needs to be kept as received (without CR LF end of line) as they are used in the host authentication process. This can happen with a Bitvise SSH server which hides its version. s> SSH-2.0-9.99 FlowSsh: Bitvise SSH Server (WinSSHD) more-datas3 SSH-2.0-9.99 FlowSsh: Bitvise SSH Server (WinSSHD) Nr� r' r) r) r* �)test_dataReceivedSSHVersionTrailingSpaces� s � �z?BaseSSHTransportTests.test_dataReceivedSSHVersionTrailingSpacesc C s4 t � }d|_|�t�� � |�d� | �|j� dS )z� If an unusual SSH version is received and is included in C{supportedVersions}, an unsupported version error is not emitted. )� 9.99� SSH-9.99-OpenSSH N)rB �supportedVersionsr� r r� r� r# rH r� r) r) r* � test_supportedVersionsAreAllowed� s z6BaseSSHTransportTests.test_supportedVersionsAreAllowedc C s6 t � }d|_|�t�� � |�d� | �d|j� dS )z� If an unusual SSH version is received and is not included in C{supportedVersions}, an unsupported version error is emitted. )s 2.0r� r� N)rB r� r� r r� r� r rH r� r) r) r* �6test_unsupportedVersionsCallUnsupportedVersionReceived� s zLBaseSSHTransportTests.test_unsupportedVersionsCallUnsupportedVersionReceivedc s� t jf� fdd� }|d� |d� � jj}t� � j_|dt j� dd� � jj_|d� |� j_t� � j_d d � }|� jj_ |d t j � � � � d S ) zi Test that the transport disconnects with an error when it receives bad packets. c sp g � _ | � j_� �� j�� � � �t� j �d� � �� j d d tj� � �� j d d dd� t |f�� d S )Nra r r� r� ) r� r� r8 rA r9 r rd r" r� rn )r[ �errorrI r) r* r� � s (z6BaseSSHTransportTests.test_badPackets.<locals>.testBads ��������s BCDEs AB123456c S s | d d� S r� r) )r= r) r) r* r< � � z7BaseSSHTransportTests.test_badPackets.<locals>.<lambda>s BCDEFGHIJKc S s t d��)Nzbad compression)� Exception)rp r) r) r* �stubDecompress� s z=BaseSSHTransportTests.test_badPackets.<locals>.stubDecompresss BCDEN) r" �DISCONNECT_PROTOCOL_ERRORr� r0 r^ �DISCONNECT_MAC_ERRORrl r� rD r� � DISCONNECT_COMPRESSION_ERROR�flushLoggedErrors)r2 r� �oldEncryptionsr� r) rI r* �test_badPackets� s& � � z%BaseSSHTransportTests.test_badPacketsc s� � j j}|f� fdd� }� j �dd� |� dtjd< � j �dd� |� � j �dd� |� � j �t� � � j �dd� |� � j �d d� |� d S ) zj Test that unimplemented packet types cause MSG_UNIMPLEMENTED packets to be sent. c sP � � � jd d tj� � � � jd d dd� t| f�� g � j_| d7 } d S )Nr ra r� r� )r r� r" r� rn r� )rS rI r) r* �checkUnimplemented� s $ zKBaseSSHTransportTests.test_unimplementedPackets.<locals>.checkUnimplemented�( r1 s MSG_fiction�) �< �F r� N)r� rr r� r" �messagesr� r� )r2 rS r� r) rI r* �test_unimplementedPackets� s z/BaseSSHTransportTests.test_unimplementedPacketsc C s� | j }|�| j�� � t� |_t� |_t� |_|� t � � t � }|� t �� � |�d� | �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j|j� dS )zD Test that multiple instances have distinct states. r1 N)r� r� r" r r^ r0 r� r5 rD r� r� rB r� r r� r� �assertNotEqualr� ro rr r )r2 r� �proto2r) r) r* �test_multipleClasses� s z*BaseSSHTransportTests.test_multipleClassesN)4r& r' r( r] r r7 r r4 r� r" r) r. r3 r6 r7 r: rC rE rF rQ re rf rg ri rp rz r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r) r) r) r* r � s� �������� � � � �� 1 $ $ r c @ r� )�'BaseSSHTransportDHGroupExchangeBaseCasez@ Diffie-Hellman group exchange tests for TransportBase. c C s� | j | j_d| j_| �d| jj ��� }| �d| ��� }| �d| | ��� }| �d| | | ��� }| �| j�ddd�|| | | � dS )z? Test that _getKey generates the correct keys. � EFs ABCDKr? � K� AB� CDN)r� r� �kexAlg� sessionIDr� �digestr �_getKey)r2 �k1�k2�k3�k4r) r) r* � test_getKey s (z3BaseSSHTransportDHGroupExchangeBaseCase.test_getKeyN)r& r' r( r] r� r) r) r) r* r� � r� c @ � e Zd ZdZdS )�(BaseSSHTransportDHGroupExchangeSHA1TestszE diffie-hellman-group-exchange-sha1 tests for TransportBase. N�r& r' r( r] r) r) r) r* r� � r� c @ r� )�*BaseSSHTransportDHGroupExchangeSHA256TestszG diffie-hellman-group-exchange-sha256 tests for TransportBase. Nr� r) r) r) r* r� & � r� c @ r� )�"BaseSSHTransportEllipticCurveTestsz4 ecdh-sha2-nistp256 tests for TransportBase Nr� r) r) r) r* r� 0 r� r� c @ r� )�%BaseSSHTransportCurve25519SHA256Testsz3 curve25519-sha256 tests for TransportBase Nr� r) r) r) r* r� 8 � r� c @ s\ e Zd ZdZddd�Zddd�Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�#ServerAndClientSSHTransportBaseCasezF Tests that need to be run on both the server and the client. Nc C sN |du rt j}| �| jd d t j� | �| jd d dd� t|f�� dS )zI Helper function to check if the transport disconnected. Nr� r ra r� r� )r" r� r r� r� rn r� r) r) r* �checkDisconnectedF s (z5ServerAndClientSSHTransportBaseCase.checkDisconnectedc C sP |du rt j}| �� }||� |�t�� � | j�|j �� � |r&| � |� |S )zy Helper function to connect a modified protocol to the test protocol and test for disconnection. N) r" � DISCONNECT_KEY_EXCHANGE_FAILEDr� r� r r� r� r� r r� )r2 �protoModificationr� r� r) r) r* �connectModifiedProtocolO s z;ServerAndClientSSHTransportBaseCase.connectModifiedProtocolc C � dd� }| � |� dS )z` Test that the transport disconnects if it can't match the key exchange c S � g | _ d S r� �rV �r� r) r) r* �blankKeyExchangesd � z\ServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchKex.<locals>.blankKeyExchangesN�r� )r2 r r) r) r* � test_disconnectIfCantMatchKex^ s zAServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchKexc C r� )zP Like test_disconnectIfCantMatchKex, but for the key algorithm. c S r� r� )rW r r) r) r* �blankPublicKeysn r z]ServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchKeyAlg.<locals>.blankPublicKeysNr )r2 r r) r) r* � test_disconnectIfCantMatchKeyAlgi � zDServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchKeyAlgc C r� )zN Like test_disconnectIfCantMatchKex, but for the compression. c S r� r� �rY r r) r) r* �blankCompressionsx r zdServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchCompression.<locals>.blankCompressionsNr )r2 r r) r) r* �%test_disconnectIfCantMatchCompressions r zIServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchCompressionc C r� )zM Like test_disconnectIfCantMatchKex, but for the encryption. c S r� r� �rL r r) r) r* � blankCiphers� r zZServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchCipher.<locals>.blankCiphersNr )r2 r r) r) r* � test_disconnectIfCantMatchCipher} r zDServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchCipherc C r� )zF Like test_disconnectIfCantMatchKex, but for the MAC. c S r� r� �rX r r) r) r* � blankMACs� r zTServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchMAC.<locals>.blankMACsNr )r2 r r) r) r* � test_disconnectIfCantMatchMAC� r zAServerAndClientSSHTransportBaseCase.test_disconnectIfCantMatchMACc C �$ | � | j�� t�| jj�� �� dS )z� Test that the transport's L{getPeer} method returns an L{SSHTransportAddress} with the L{IAddress} of the peer. N)r r� �getPeerr �SSHTransportAddressr" rI r) r) r* � test_getPeer� � �z0ServerAndClientSSHTransportBaseCase.test_getPeerc C r )z� Test that the transport's L{getHost} method returns an L{SSHTransportAddress} with the L{IAddress} of the host. N)r r� �getHostr r r" rI r) r) r* � test_getHost� r z0ServerAndClientSSHTransportBaseCase.test_getHostr� ) r& r' r( r] r� r� r r r r r r r r) r) r) r* r� A s r� c @ �: e Zd ZU dZejZeeej e d< dd� Z dd� Z dS )�ServerSSHTransportBaseCasez1 Base case for SSHServerTransport tests. r� c C s$ t �| � t� | j_| jj�� d S r� )r� r� r� r� r � startFactoryrI r) r) r* r� � s z ServerSSHTransportBaseCase.setUpc C s t �| � | jj�� | j`d S r� )r� �tearDownr� r � stopFactoryrI r) r) r* r � s z#ServerSSHTransportBaseCase.tearDownN) r& r' r( r] r" r- r� r r r+ r� r� r r) r) r) r* r � s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd S )!�ServerSSHTransportTestsz' Tests for SSHServerTransport. c C s� | j �d� | �| j jd� | �| j jd� | �| j jd� | �| j jd� | �| j j� | j j }| �|j d� | �|j d� | �|j d� | �|j d� dS )z� Receiving a KEXINIT packet listing multiple supported algorithms will set up the first common algorithm found in the client's preference list. s SSH-2.0-Twisted ����������������� bdiffie-hellman-group1-sha1,diffie-hellman-group-exchange-sha1,diffie-hellman-group-exchange-sha256 ssh-dss,ssh-rsa �aes128-ctr,aes128-cbc,aes192-ctr,aes192-cbc,aes256-ctr,aes256-cbc,cast128-ctr,cast128-cbc,blowfish-ctr,blowfish-cbc,3des-ctr,3des-cbc �aes128-ctr,aes128-cbc,aes192-ctr,aes192-cbc,aes256-ctr,aes256-cbc,cast128-ctr,cast128-cbc,blowfish-ctr,blowfish-cbc,3des-ctr,3des-cbc hmac-md5,hmac-sha1 hmac-md5,hmac-sha1 none,zlib none,zlib ����r� s ssh-dssr s aes128-ctr� hmac-md5N)r� r� r r� �keyAlg�outgoingCompressionType�incomingCompressionTyper# rw rl r| r} r r~ �r2 �ner) r) r* � test_KEXINITMultipleAlgorithms� s �z6ServerSSHTransportTests.test_KEXINITMultipleAlgorithmsc C s� dt �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� d d }| j�|� | �| jjd � | �| jj� d S ) z� If the client sends "ext-info-c" in its key exchange algorithms, then the server notes that the client supports extension negotiation. See RFC 8308, section 2.1. rJ s/ diffie-hellman-group-exchange-sha256,ext-info-cr� r r r r1 r r r� N)r r4 r� r� r r� r% rw �r2 � kexInitPacketr) r) r* � test_KEXINITExtensionNegotiation� s: �������� � � � �� z8ServerSSHTransportTests.test_KEXINITExtensionNegotiationc C s� dd� dd� dd� | jjddd� | jj| jj| jj| jj| jj| jj| jj| jj| jjf D �D �� d }| j�|� | � | jj � | j� d � | � | jj � | j� d � | � | jj � | �| jg � d | j_ | j�d � | � | jj � | �| jg � dS ) a< The client is allowed to send a guessed key exchange packet after it sends the KEXINIT packet. However, if the key exchanges do not match, that guess packet must be ignored. This tests that the packet is ignored in the case of the key exchange method not matching. rJ r1 c S � g | ]}t �|��qS r) �r r4 r� r) r) r* r� � ��zEServerSSHTransportTests.test_ignoreGuessPacketKex.<locals>.<listcomp>c S � g | ]}d � |��qS �rT �r �r� r> r) r) r* r� r* Nr� � � � test � T� �r r� rV rW rL rX rY rZ r� r% �ignoreNextPacket� ssh_DEBUG�ssh_KEX_DH_GEX_REQUEST_OLDr# r r� �ssh_KEX_DH_GEX_REQUESTr% r) r) r* �test_ignoreGuessPacketKex� sB ������� z1ServerSSHTransportTests.test_ignoreGuessPacketKexc C s� dd� dd� dd� | jj| jjddd� | jj| jj| jj| jj| jj| jj| jj| jjf D �D �� d }| j�|� | � | jj � | j� d � | � | jj � | j� d � | � | jj � | �| jg � d | j_ | j�d � | � | jj � | �| jg � dS ) zk Like test_ignoreGuessPacketKex, but for an incorrectly guessed public key format. rJ r1 c S r( r) r) r� r) r) r* r� 2 r* zEServerSSHTransportTests.test_ignoreGuessPacketKey.<locals>.<listcomp>c S r+ r, r- r. r) r) r* r� 4 r* Nr� r/ r0 r1 Tr2 r3 r% r) r) r* �test_ignoreGuessPacketKey) sB ������� z1ServerSSHTransportTests.test_ignoreGuessPacketKeyc C sz |g| j _dg| j _| j �| j�� � t�|�\}}t|d|�}| j � t � |�� t � t � d|d ��d }t| j j|| j j�}| �| j j|� t||| j j�}t� } | �t � | j j�d � | �t � | j j�d � | �t � | j jjd �� �� | �t � |�� | �|� | �|� | �� } | j jjd � | �} | �| j tj t � | j jjd �� �| t � | � ftj dfg� dS ) z� Test that the KEXDH_INIT packet causes the server to send a KEXDH_REPLY with the server's public key and a signature. @param kexAlgorithm: The key exchange algorithm to use. @type kexAlgorithm: L{str} r� i� r� � r rj r1 N) r� rV rW r� r" r r r� r< r6 r r; �getMPr4 r@ r� r� r r� r �updater� �ourKexInitPayloadr � publicKeysr� r� � privateKeys�signr� �MSG_KEXDH_REPLY� MSG_NEWKEYS) r2 r� r� r� r� �er> r9 r� �hr� � signaturer) r) r* �assertKexDHInitResponseV s> �����z/ServerSSHTransportTests.assertKexDHInitResponsec C s, d| j _d| j _| �t| j jt�d�� dS )�� Test that if the server receives a KEX_DH_GEX_REQUEST_OLD message and the key exchange algorithm is not set, we raise a ConchError. s bad-curver� s unused-keyN)r� r� r rh r �_ssh_KEX_ECDH_INITr r4 rI r) r) r* �%test_checkBad_KEX_ECDH_INIT_CurveName� s �z=ServerSSHTransportTests.test_checkBad_KEX_ECDH_INIT_CurveNamec C s� dt �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� d d }| j�|� | �t� | �t� d S ) zy Test that if the server received a bad name for a curve we raise an UnsupportedAlgorithm error. r r r� r r r r1 r r N)r r4 r� r� rh �AttributeErrorr )r2 �kexmsgr) r) r* � test_checkBad_KEX_INIT_CurveName� s: �������� � � � �� z8ServerSSHTransportTests.test_checkBad_KEX_INIT_CurveNamec C � | � dd� dS )z� KEXDH_INIT messages are processed when the diffie-hellman-group14-sha1 key exchange algorithm is requested. r� r� N)rF rI r) r) r* �test_KEXDH_INIT_GROUP14� � z/ServerSSHTransportTests.test_KEXDH_INIT_GROUP14c � d� j _t� � j _� �dd� � �� j jd� � �dd� � �� j jd� � �� jd tj df� � fdd�t d �D �}� �� j jj |d |d |d |d |d |d f� dS )�G Test that _keySetup sets up the next encryption keys. r� r� r� r� r� r1 c � g | ] }� j �|d d��qS �r� r� �r� r� r rI r) r* r� � � z9ServerSSHTransportTests.test_keySetup.<locals>.<listcomp>� ABCDEFra r� r rj r� r� N� r� r� r^ rl r� r r� r� r" rB r r! �r2 �newKeysr) rI r* � test_keySetup� � &�z%ServerSSHTransportTests.test_keySetupc C s� g d�| j _d| j _t� | j _d| j _| �dd� | �| jd t j df� | �| jd t j d t � d � t � d � f� | �dd � | �| jd t j df� d S )a If the client advertised support for extension negotiation, then _keySetup sends SSH_MSG_EXT_INFO with the "server-sig-algs" extension as the next packet following the server's first SSH_MSG_NEWKEYS. See RFC 8308, sections 2.4 and 3.1. )r� s rsa-sha2-256s rsa-sha2-512r� Tr� r� ���r1 r� s rr r| r� N)r� rW r� r^ rl rw r� r r� r" rB ry r r4 rI r) r) r* �test_keySetupWithExtInfo� s$ ���� z0ServerSSHTransportTests.test_keySetupWithExtInfoc rP )rQ r r� r� r� r� r1 c rR rS rT r rI r) r* r� � rU z>ServerSSHTransportTests.test_ECDH_keySetup.<locals>.<listcomp>rV ra r� r rj r� r� NrW rX r) rI r* �test_ECDH_keySetup� r[ z*ServerSSHTransportTests.test_ECDH_keySetupc C s� | � � t�dddd�| j_| j�d� | �| jj| jj� | �| jj � | �| jj � d| j_ | � dd� | j�d� | � | jj � d| j_| � dd� | j�d� | � | jj � dS )zj Test that NEWKEYS transitions the keys in nextEncryptions to currentEncryptions. r r1 � zlibr� r� r� N)r$ r" rK r� rl � ssh_NEWKEYS�assertIsr0 rA r5 rD r r� �assertIsNotNoner! rI r) r) r* � test_NEWKEYS� s � z$ServerSSHTransportTests.test_NEWKEYSc C sD | j �t�d�� | �| jtjt�d�fg� | �| j jj d� dS )z^ Test that the SERVICE_REQUEST message requests and starts a service. r� r� N) r� �ssh_SERVICE_REQUESTr r4 r r� r" �MSG_SERVICE_ACCEPTr r� rI r) r) r* �test_SERVICE_REQUEST s �z,ServerSSHTransportTests.test_SERVICE_REQUESTc C � | j �d� | �� dS �zD Test that NEWKEYS disconnects if it receives data. s bad packetN�r� r` r� rI r) r) r* �test_disconnectNEWKEYSData � z2ServerSSHTransportTests.test_disconnectNEWKEYSDatac C s" | j �t�d�� | �tj� dS )zd Test that SERVICE_REQUESTS disconnects if an unknown service is requested. s no serviceN)r� rd r r4 r� r" � DISCONNECT_SERVICE_NOT_AVAILABLErI r) r) r* �(test_disconnectSERVICE_REQUESTBadService s z@ServerSSHTransportTests.test_disconnectSERVICE_REQUESTBadServiceN)r& r' r( r] r$ r' r8 r9 rF rI rL rN rZ r] r^ rc rf rj rm r) r) r) r* r � s" #0-- r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �)ServerSSHTransportDHGroupExchangeBaseCasezE Diffie-Hellman group exchange tests for SSHServerTransport. c C � | j g| j_dg| j_| j�| j�� � | j�d� | jj� � � d�d \}}| � | j tj t�|�d fg� | � | jjd� | � | jj|� dS )z� Test that the KEX_DH_GEX_REQUEST_OLD message causes the server to reply with a KEX_DH_GEX_GROUP message with the correct Diffie-Hellman group. r� � r� r � rj N)r� r� rV rW r� r" r r6 r r� �getr r� �MSG_KEX_DH_GEX_GROUPr r; r� r� �r2 � dhGenerator�dhPrimer) r) r* �test_KEX_DH_GEX_REQUEST_OLD" s ��� zEServerSSHTransportDHGroupExchangeBaseCase.test_KEX_DH_GEX_REQUEST_OLDc C s d| j _| �t| j jd� dS )rG N)r� r� rh r r6 rI r) r) r* �%test_KEX_DH_GEX_REQUEST_OLD_badKexAlg9 s zOServerSSHTransportDHGroupExchangeBaseCase.test_KEX_DH_GEX_REQUEST_OLD_badKexAlgc C ro )z� Test that the KEX_DH_GEX_REQUEST message causes the server to reply with a KEX_DH_GEX_GROUP message with the correct Diffie-Hellman group. r� � r� r rq rj N)r� r� rV rW r� r" r r7 r r� rr r r� rs r r; r� r� rt r) r) r* �test_KEX_DH_GEX_REQUESTA s � ��� zAServerSSHTransportDHGroupExchangeBaseCase.test_KEX_DH_GEX_REQUESTc C s� | � � t| jjd| jj�}t�d�d }| �| jj� � j |� t | jj|| jj�}| �| jj |� t ||| jj�}| � � }|�t�| jj�d � |�t�| jj�d � |�t�| jjjd �� �� |�d� |�t�| jj�� |�t�| jj�� |�t�|�� |�|� |�|� |�� }| j�t�|�� | �| jdd� tjt�| jjjd �� �| t�| jjjd � |�� ftj d fg� dS ) z� Test that the KEX_DH_GEX_INIT message after the client sends KEX_DH_GEX_REQUEST_OLD causes the server to send a KEX_DH_GEX_INIT message with a public key and signature. r� s ��������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������r rj r� rp ra Nr1 ) rw r< r� r� r� r r; r r� �private_numbersr= r@ r� r� r< r4 r� r= r r> r� r; r� �ssh_KEX_DH_GEX_INITr� r" �MSG_KEX_DH_GEX_REPLYr? r@ rB �r2 rC r> r9 r� rD r� r) r) r* �&test_KEX_DH_GEX_INIT_after_REQUEST_OLDZ sB ������zPServerSSHTransportDHGroupExchangeBaseCase.test_KEX_DH_GEX_INIT_after_REQUEST_OLDc C sb | � � t| jjd| jj�}t�d�d }t| jj|| jj�}t||| jj�}| �� }|� t� | jj �d � |� t� | jj �d � |� t� | jj jd �� �� |� d� |� t�| jj�� |� t�| jj�� |� t�|�� |� |� |� |� |�� }| j�t�|�� | �| jd tjt� | jj jd �� �| t� | jj jd �|�� f� dS ) z� Test that the KEX_DH_GEX_INIT message after the client sends KEX_DH_GEX_REQUEST causes the server to send a KEX_DH_GEX_INIT message with a public key and signature. r� s ����������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������r rj r� ry ra N)rz r< r� r� r� r r; r@ r� r< r4 r� r= r r> r� r; r� r| r r� r" r} r? r@ r~ r) r) r* �"test_KEX_DH_GEX_INIT_after_REQUEST� s: �����zLServerSSHTransportDHGroupExchangeBaseCase.test_KEX_DH_GEX_INIT_after_REQUESTN) r& r' r( r] rw rx rz r r� r) r) r) r* rn s (rn c @ r� )�*ServerSSHTransportDHGroupExchangeSHA1TestszJ diffie-hellman-group-exchange-sha1 tests for SSHServerTransport. Nr� r) r) r) r* r� � r� r� c @ r� )�,ServerSSHTransportDHGroupExchangeSHA256TestszL diffie-hellman-group-exchange-sha256 tests for SSHServerTransport. Nr� r) r) r) r* r� � r� r� c @ r� )� ServerSSHTransportECDHBaseCasezE Elliptic Curve Diffie-Hellman tests for SSHServerTransport. c C s� | j g| j_dg| j_| j�| j�� � | jjjd }| jjj d }| j� � }|� � }| j� |�}| j� t�|�� | j�|| j� | jj��}| �� }|�t�| jj�� |�t�| jj�� |�t�| jj�� |�t�| jj�� |�t�|�� �� |�t�|�� |�t�| j� | jj��� |�|� |�� }|�|�} | �| j tj t�|�� �t�| j� | jj�� t�| � ftj dfg� dS )z� Test that the KEXDH_INIT message causes the server to send a KEXDH_REPLY with the server's public key and a signature. r� r1 N) r� r� rV rW r� r" r r r? r> �_generateECPrivateKeyr� �_encodeECPublicKeyr6 r r4 �_generateECSharedSecret�ecPubr� r< r� r� �otherKexInitPayloadr= r� r� r@ r r� rA rB � r2 �privKey�pubKey�ecPrivr� �encPubr� rD r� rE r) r) r* �test_KEX_ECDH_INIT� sF � �����z1ServerSSHTransportECDHBaseCase.test_KEX_ECDH_INITN)r& r' r( r] r� r) r) r) r* r� � r� r� c @ r� )�ServerSSHTransportECDHTestsz: ecdh-sha2-nistp256 tests for SSHServerTransport. Nr� r) r) r) r* r� � r� r� c @ r� )�'ServerSSHTransportCurve25519SHA256Testsz9 curve25519-sha256 tests for SSHServerTransport. Nr� r) r) r) r* r� � r� r� c @ r )�ClientSSHTransportBaseCasez1 Base case for SSHClientTransport tests. r� c C s@ d| _ | �|| j� | �|�dd�t�t|��� �� t� d�S )zC Mock version of SSHClientTransport.verifyHostKey. T� :r1 ) �calledVerifyHostKeyr r� �replace�binascii�hexlifyr r� r �succeed)r2 r� � fingerprintr) r) r* � verifyHostKey s � z(ClientSSHTransportBaseCase.verifyHostKeyc C sB t �| � tj�tj��� | _tj�tj�| _ d| _ | j | j _ d S )NF) r� r� r! r� r� r# r� r� r� �privObjr� r� r� rI r) r) r* r� s z ClientSSHTransportBaseCase.setUpN) r& r' r( r] r" r. r� r r r+ r� r� r� r) r) r) r* r� s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$S )%�ClientSSHTransportTestsz' Tests for SSHClientTransport. c C s� | j �d� | �| j jd� | �| j jd� | �| j jd� | �| j jd� | j j}| �|jd� | �|j d� | �|j d� | �|j d� dS )z� Receiving a KEXINIT packet listing multiple supported algorithms will set up the first common algorithm, ordered after our preference. s SSH-2.0-Twisted ����������������� bdiffie-hellman-group1-sha1,diffie-hellman-group-exchange-sha1,diffie-hellman-group-exchange-sha256 ssh-dss,ssh-rsa �aes128-ctr,aes128-cbc,aes192-ctr,aes192-cbc,aes256-ctr,aes256-cbc,cast128-ctr,cast128-cbc,blowfish-ctr,blowfish-cbc,3des-ctr,3des-cbc �aes128-ctr,aes128-cbc,aes192-ctr,aes192-cbc,aes256-ctr,aes256-cbc,cast128-ctr,cast128-cbc,blowfish-ctr,blowfish-cbc,3des-ctr,3des-cbc hmac-md5,hmac-sha1 hmac-md5,hmac-sha1 zlib,none zlib,none ����r� r� r r r N) r� r� r r� r r r! rl r| r} r r~ r" r) r) r* r$ ! s �z6ClientSSHTransportTests.test_KEXINITMultipleAlgorithmsc C s< | � t| �� j� dd� }| �� �dd�}|�| j��|�S )z� verifyHostKey() should return a Deferred which fails with a NotImplementedError exception. connectionSecure() should raise NotImplementedError(). c S s | � t� d S r� )�trap�NotImplementedErrorr8 r) r) r* � _checkRaisesK r� zNClientSSHTransportTests.test_notImplementedClientMethods.<locals>._checkRaisesN)rh r� r� �connectionSecurer� � addCallback�fail� addErrback)r2 r� �dr) r) r* � test_notImplementedClientMethodsC s z8ClientSSHTransportTests.test_notImplementedClientMethodsc C sj |g| j _| j �| j�� � | j j�� j}| �t � |�dd� d|d � | �| j tj | j j fg� dS )z� Test that a KEXINIT packet with a group1 or group14 key exchange results in a correct KEXDH_INIT response. @param kexAlgorithm: The key exchange algorithm to use @type kexAlgorithm: L{str} r� Nr� r: )r� rV r� r" r r� r{ r= r r r; r� �MSG_KEXDH_INITr� )r2 r� r� r= r) r) r* �assertKexInitResponseForDHQ s "�z2ClientSSHTransportTests.assertKexInitResponseForDHc C rM )zq KEXINIT messages requesting diffie-hellman-group14-sha1 result in KEXDH_INIT responses. r� r� N)r� rI r) r) r* �test_KEXINIT_group14h rO z,ClientSSHTransportTests.test_KEXINIT_group14c C s2 dg| j _| j�� �dd�}| �t| j j|� dS )z� Test that the client raises a ConchError if it receives a KEXINIT message but doesn't have a key exchange algorithm that we understand. s diffie-hellman-group24-sha1s group14s group24N)r� rV r" r r� rh r r� r� r) r) r* �test_KEXINIT_badKexAlgo s z.ClientSSHTransportTests.test_KEXINIT_badKexAlgc C s� dt �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� d d }| j�|� | �| jj� d S ) z� If the server sends "ext-info-s" in its key exchange algorithms, then the client notes that the server supports extension negotiation. See RFC 8308, section 2.1. rJ s/ diffie-hellman-group-exchange-sha256,ext-info-sr� r r r r1 r r N)r r4 r� r� r% rw r% r) r) r* r' y s8 �������� � � � �� z8ClientSSHTransportTests.test_KEXINITExtensionNegotiationc C s� | � � d}t�|�}| jj�� j}| jj}t|||�}t � }|� t� | jj �d � |� t� | jj �d � |� t� | j�� |� | jj� |� |� |� |� |�� }| j�|�}||t� | j�| fS )a Utility for test_KEXDH_REPLY and test_disconnectKEXDH_REPLYBadSignature. Begins a Diffie-Hellman key exchange in the named group Group-14 and computes information needed to return either a correct or incorrect signature. rj )r� r r; r� r� r{ r= r� r@ r r< r4 r� r= r� r� r� r� r@ ) r2 r9 �fMPr= r� r� rD r� rE r) r) r* �begin_KEXDH_REPLY� s z)ClientSSHTransportTests.begin_KEXDH_REPLYc �@ �� � \� }}� �fdd�}�j�|t�|� �}|�|� |S )zH Test that the KEXDH_REPLY message verifies the server. c �* �� | � ���j� ���jj� � d S r� �rA r% r� r r� r� �r �r� r2 r) r* �_cbTestKEXDH_REPLY� � zDClientSSHTransportTests.test_KEXDH_REPLY.<locals>._cbTestKEXDH_REPLY�r� r� �ssh_KEX_DH_GEX_GROUPr r4 r� �r2 rE � packetStartr� r� r) r� r* �test_KEXDH_REPLY� s z(ClientSSHTransportTests.test_KEXDH_REPLYc rP )rQ r� r� r� r� r� r1 c rR rS rT r rI r) r* r� � rU z9ClientSSHTransportTests.test_keySetup.<locals>.<listcomp>rV r rj ra r� r� r� NrW rX r) rI r* rZ � r[ z%ClientSSHTransportTests.test_keySetupc s | � � dg� � fdd�}|| j_t�dddd�| j_| �dd� | �| jj| jj� t � | j_| j� d� | � | jj � | � | jj � | �| jj| jj� | �� d � d | j_| �dd � | j� d� | �| jj � d | j_| �dd � | j� d� | �| jj � d S ) zl Test that NEWKEYS transitions the keys from nextEncryptions to currentEncryptions. Fc r� r� r) r) ��securer) r* �stubConnectionSecure� r� zBClientSSHTransportTests.test_NEWKEYS.<locals>.stubConnectionSecurer r� r� r1 r r_ s GHs IJN)r$ r� r� r" rK rl r� � assertIsNotr0 r^ r` rA r5 rD ra r% r rb r! )r2 r� r) r� r* rc � s. � z$ClientSSHTransportTests.test_NEWKEYSc C s* t � | j_| j�d� | �| jjj� dS )zS Test that the SERVICE_ACCEPT packet starts the requested service. � MockServiceN)r� r� �instance�ssh_SERVICE_ACCEPTr% r� rI r) r) r* �test_SERVICE_ACCEPT� s z+ClientSSHTransportTests.test_SERVICE_ACCEPTc C s( | j �t� � | �| jtjdfg� dS )zP Test that requesting a service sends a SERVICE_REQUEST packet. r� N)r� �requestServicer� r r� r" rk rI r) r) r* �test_requestService� s �z+ClientSSHTransportTests.test_requestServicec �6 � � � \}}}� j�|t�d� �}|�� fdd��S )zL Test that KEXDH_REPLY disconnects if the signature is bad. � bad signaturec � � � tj�S r� �r� r" r� ��_rI r) r* r< r� zPClientSSHTransportTests.test_disconnectKEXDH_REPLYBadSignature.<locals>.<lambda>r� �r2 r� rE r� r� r) rI r* �&test_disconnectKEXDH_REPLYBadSignature � �z>ClientSSHTransportTests.test_disconnectKEXDH_REPLYBadSignaturec C �4 dt �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� t �d� d d }| j�|� | j�d � t�t�� t� �| j_ | jj � � | j_ t�t�� t� �}|� � }|� t jjt jj�}t�� | j_d| j_| j�t �t� �� d �� �t �|� t �d � � | �tj� d S � �O Test that KEX_ECDH_REPLY disconnects if the signature is bad. r r r� r r r r1 r r s SSH-2.0-OpenSSH s bad-signatureN�r r4 r� r� r� r �generate_private_key� SECP256R1r r� r� r� � public_bytesr �Encoding�X962� PublicFormat�UncompressedPoint�curver� �_ssh_KEX_ECDH_REPLYr� r� r� r� r" r� �r2 rK �thisPriv�thisPubr� r) r) r* �)test_disconnectKEX_ECDH_REPLYBadSignature �Z �������� � � � �� � ���zAClientSSHTransportTests.test_disconnectKEX_ECDH_REPLYBadSignaturec C rg rh ri rI r) r) r* rj = rk z2ClientSSHTransportTests.test_disconnectNEWKEYSDatac C s" t � | j_| j�d� | �� dS )z� Test that SERVICE_ACCEPT disconnects if the accepted protocol is differet from the asked-for protocol. s badN)r� r� r� r� r� rI r) r) r* � test_disconnectSERVICE_ACCEPTD s z5ClientSSHTransportTests.test_disconnectSERVICE_ACCEPTc C s< t � | j_| j�d� | �| jjj� | �t| j�d� dS )z� Some commercial SSH servers don't send a payload with the SERVICE_ACCEPT message. Conch pretends that it got the correct name of the service. r1 r N) r� r� r� r� r% r� r rd r� rI r) r) r* � test_noPayloadSERVICE_ACCEPTM s z4ClientSSHTransportTests.test_noPayloadSERVICE_ACCEPTN)r& r' r( r] r$ r� r� r� r� r' r� r� rZ rc r� r� r� r� rj r� r� r) r) r) r* r� s& " " . r� c @ sL e Zd ZdZ edd�Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dS )�)ClientSSHTransportDHGroupExchangeBaseCasezE Diffie-Hellman group exchange tests for SSHClientTransport. �� FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA237327FFFFFFFFFFFFFFFF� c C s8 | j g| j_| j�| j�� � | �| jtjdfg� dS )zt KEXINIT packet with a group-exchange key exchange results in a KEX_DH_GEX_REQUEST message. � N) r� r� rV r� r" r r r� �MSG_KEX_DH_GEX_REQUESTrI r) r) r* �test_KEXINIT_groupexchangem s ���zDClientSSHTransportDHGroupExchangeBaseCase.test_KEXINIT_groupexchangec C s� | � � | j�t�| j�t�d� � | �| jj| j� | �| jjd� | jj � � j }| �t�|�dd� d� | �| jj t�t d|| j��� | �| jdd� tj| jj fg� dS )z� Test that the KEX_DH_GEX_GROUP message results in a KEX_DH_GEX_INIT message with the client's Diffie-Hellman public key. rj r� Ns� ������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������ra )r� r� r� r r; �P1536r r� r� r� r{ r= r� r< r� r" �MSG_KEX_DH_GEX_INITrg r) r) r* �test_KEX_DH_GEX_GROUP s � �z?ClientSSHTransportDHGroupExchangeBaseCase.test_KEX_DH_GEX_GROUPc C s� | � � | jj}d}t�|�}t|| jj�� j|�}| � � }|� t� | jj �d � |� t� | jj �d � |� t� | j�� |� d� |� t�| j�t�d� � |� | jj� |� |� |� |� |�� }| j�|�}||t� | j�| fS )a Utility for test_KEX_DH_GEX_REPLY and test_disconnectGEX_REPLYBadSignature. Begins a Diffie-Hellman key exchange in an unnamed (server-specified) group and computes information needed to return either a correct or incorrect signature. r� rj r� )r� r� r� r r; r@ r� r{ r= r� r< r4 r� r= r� r� r� r� r� r@ )r2 r� r9 r� r� rD r� rE r) r) r* �begin_KEX_DH_GEX_REPLY� s" z@ClientSSHTransportDHGroupExchangeBaseCase.begin_KEX_DH_GEX_REPLYc r� )z^ Test that the KEX_DH_GEX_REPLY message results in a verified server. c r� r� r� r� r� r) r* �_cbTestKEX_DH_GEX_REPLY� r� z`ClientSSHTransportDHGroupExchangeBaseCase.test_KEX_DH_GEX_REPLY.<locals>._cbTestKEX_DH_GEX_REPLY�r� r� �ssh_KEX_DH_GEX_REPLYr r4 r� )r2 rE r� r� r� r) r� r* �test_KEX_DH_GEX_REPLY� s z?ClientSSHTransportDHGroupExchangeBaseCase.test_KEX_DH_GEX_REPLYc r� )zQ Test that KEX_DH_GEX_REPLY disconnects if the signature is bad. r� c r� r� r� r� rI r) r* r< � r� z`ClientSSHTransportDHGroupExchangeBaseCase.test_disconnectGEX_REPLYBadSignature.<locals>.<lambda>r� r� r) rI r* �$test_disconnectGEX_REPLYBadSignature� r� zNClientSSHTransportDHGroupExchangeBaseCase.test_disconnectGEX_REPLYBadSignaturec C r� r� r� r� r) r) r* r� � r� zSClientSSHTransportDHGroupExchangeBaseCase.test_disconnectKEX_ECDH_REPLYBadSignatureN) r& r' r( r] �intr� r� r� r� r� r� r� r) r) r) r* r� Y s � r� c @ r� )�*ClientSSHTransportDHGroupExchangeSHA1TestszJ diffie-hellman-group-exchange-sha1 tests for SSHClientTransport. Nr� r) r) r) r* r� � r� r� c @ r� )�,ClientSSHTransportDHGroupExchangeSHA256TestszL diffie-hellman-group-exchange-sha256 tests for SSHClientTransport. Nr� r) r) r) r* r� r� r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � ClientSSHTransportECDHBaseCasezE Elliptic Curve Diffie-Hellman tests for SSHClientTransport. c C sJ | j g| j_| j�| j�� � | �| jtjt � | j� | jj ��fg� dS )zm KEXINIT packet with an elliptic curve key exchange results in a KEXDH_INIT message. N) r� r� rV r� r" r r r� r� r r4 r� r� rI r) r) r* � test_KEXINIT s ���z+ClientSSHTransportECDHBaseCase.test_KEXINITc C s. | � � t� �� d }t� �� d }| j�� }|�� }| j�|�}| j�|| j�| jj ��}| � � }|� t � | jj�� |� t � | jj�� |� t � | jj�� |� t � | jj�� |� t � |�� �� |� t � | j�| jj ��� |� t � |�� |� |� |�� }|�|�} || t � |�� �t � |� fS )a Utility for test_KEXDH_REPLY and test_disconnectKEXDH_REPLYBadSignature. Begins an Elliptic Curve Diffie-Hellman key exchange and computes information needed to return either a correct or incorrect signature. r� )r� r� r� r� r� r� r� r� r� r� r� r< r r4 r� r� r= r� r� r� r@ r� r) r) r* r� & s* � z0ClientSSHTransportECDHBaseCase.begin_KEXDH_REPLYc r� )zO Test that the KEXDH_REPLY message completes the key exchange. c r� r� r� r� r� r) r* r� P r� zKClientSSHTransportECDHBaseCase.test_KEXDH_REPLY.<locals>._cbTestKEXDH_REPLYr� r� r) r� r* r� J s z/ClientSSHTransportECDHBaseCase.test_KEXDH_REPLYc r� )r� r� c r� r� r� r� rI r) r* r< a r� zWClientSSHTransportECDHBaseCase.test_disconnectKEXDH_REPLYBadSignature.<locals>.<lambda>r� r� r) rI r* r� Y r� zEClientSSHTransportECDHBaseCase.test_disconnectKEXDH_REPLYBadSignatureN)r& r' r( r] r� r� r� r� r) r) r) r* r� s $ r� c @ r� )�ClientSSHTransportECDHTestsz: ecdh-sha2-nistp256 tests for SSHClientTransport. Nr� r) r) r) r* r� e r� r� c @ r� )�'ClientSSHTransportCurve25519SHA256Testsz9 curve25519-sha256 tests for SSHClientTransport. Nr� r) r) r) r* r� m r� r� c @ s` e Zd ZdZer eZdd� Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� ZdS )� GetMACTestsz* Tests for L{SSHCiphers._getMAC}. c C s t �dddd�| _d S )NrG r4 rH rI )r" rK rN rI r) r) r* r� ~ r� zGetMACTests.setUpc C s t d�S )z� Generate a new shared secret to be used with the tests. @return: A new secret. @rtype: L{bytes} �@ r rI r) r) r* �getSharedSecret� s zGetMACTests.getSharedSecretc C s| | � � }| j�||�}|d|� d| }tdd� t|�D ��}tdd� t|�D ��} | �||| |f|� | �||j� dS )a� Check that when L{SSHCiphers._getMAC} is called with a supportd HMAC algorithm name it returns a tuple of (digest object, inner pad, outer pad, digest size) with a C{key} attribute set to the value of the key supplied. @param hmacName: Identifier of HMAC algorithm. @type hmacName: L{bytes} @param hashProcessor: Callable for the hash algorithm. @type hashProcessor: C{callable} @param digestSize: Size of the digest for algorithm. @type digestSize: L{int} @param blockPadSize: Size of padding applied to the shared secret to match the block size. @type blockPadSize: L{int} Nr c s � � | ] }t |�d A V qdS )�6 N�r- �r� �br) r) r* r� � � � z+GetMACTests.assertGetMAC.<locals>.<genexpr>c s r� )�\ Nr� r� r) r) r* r� � r� )r� rN �_getMACrn r r r� ) r2 �hmacNamer� � digestSize� blockPadSize�secret�paramsr� �innerPad�outerPadr) r) r* � assertGetMAC� s zGetMACTests.assertGetMACc C � | j dtddd� dS )a When L{SSHCiphers._getMAC} is called with the C{b"hmac-sha2-512"} MAC algorithm name it returns a tuple of (sha512 digest object, inner pad, outer pad, sha512 digest size) with a C{key} attribute set to the value of the key supplied. s hmac-sha2-512r� �r r N)r r rI r) r) r* �test_hmacsha2512� � z GetMACTests.test_hmacsha2512c C � | j dtddd� dS )a When L{SSHCiphers._getMAC} is called with the C{b"hmac-sha2-384"} MAC algorithm name it returns a tuple of (sha384 digest object, inner pad, outer pad, sha384 digest size) with a C{key} attribute set to the value of the key supplied. s hmac-sha2-384�0 �P r N)r r rI r) r) r* �test_hmacsha2384� r z GetMACTests.test_hmacsha2384c C r )a When L{SSHCiphers._getMAC} is called with the C{b"hmac-sha2-256"} MAC algorithm name it returns a tuple of (sha256 digest object, inner pad, outer pad, sha256 digest size) with a C{key} attribute set to the value of the key supplied. s hmac-sha2-256� r N)r r rI r) r) r* �test_hmacsha2256� r z GetMACTests.test_hmacsha2256c C r )a When L{SSHCiphers._getMAC} is called with the C{b"hmac-sha1"} MAC algorithm name it returns a tuple of (sha1 digest object, inner pad, outer pad, sha1 digest size) with a C{key} attribute set to the value of the key supplied. r � �, r N)r r rI r) r) r* � test_hmacsha1� r zGetMACTests.test_hmacsha1c C r )a When L{SSHCiphers._getMAC} is called with the C{b"hmac-md5"} MAC algorithm name it returns a tuple of (md5 digest object, inner pad, outer pad, md5 digest size) with a C{key} attribute set to the value of the key supplied. r r� r r N)r r rI r) r) r* � test_hmacmd5� r zGetMACTests.test_hmacmd5c C s& | � � }| j�d|�}| �d|� dS )z� When L{SSHCiphers._getMAC} is called with the C{b"none"} MAC algorithm name it returns a tuple of (None, "", "", 0). r )Nr1 r1 r N)r� rN r� r )r2 r� r r) r) r* � test_none� s zGetMACTests.test_noneN)r& r' r( r] r r7 r� r� r r r r r r r r) r) r) r* r� v s r� c @ �@ e Zd ZdZer eZdd� Zdd� Zdd� Zdd � Z d d � Z d S ) �SSHCiphersTestsz0 Tests for the SSHCiphers helper class. c C sL t �dddd�}| �|jd� | �|jd� | �|jd� | �|jd� dS )zJ Test that the initializer sets up the SSHCiphers object. rG r4 rH rI N)r" rK r r| r} r r~ )r2 rN r) r) r* � test_init� s zSSHCiphersTests.test_initc C sj t �dddd�}d }}|j�� D ]!\}\}}}|�|||�}|dkr+| �|t j� q| �|j|� qdS )zM Test that the _getCipher method returns the correct cipher. rG r4 rH rI rJ r N)r" rK � cipherMapr� rM �assertIsInstance� _DummyCipher� algorithm) r2 rN rO r� rP �algClass�keySize�counter�cipr) r) r* �test_getCipher� s �z SSHCiphersTests.test_getCipherc C s: d}t jjD ]�}t jj| \}}}t �|ddd�}t �d|dd�}|�|||�}|jjd } |�||dddd� |�dd||dd� | � |j | � | � |j | � |� � } | � |d| � �} | � |d| � �} | � |�|d| � �| � | � |�|d| � �| � | � |�| �|d| � � | � |�| �|d| � � qdS )z8 Test that setKeys sets up the ciphers. �@ r r: r1 N)r" r+ rL rK r rM r � block_sizer{ r re rk � encryptorr< rh rl ) r2 r� rP �modNamer r � encCipher� decCipherr �bsr$ �enc�enc2r) r) r* �test_setKeysCiphers s& �z#SSHCiphersTests.test_setKeysCiphersc C s� d}t jj�� D ]s\}}t �dd|d�}t �ddd|�}|�dddd|d� |�ddddd|� |r7|� j}nd}| �|j|� |rL|�||�\}}}}d} |} d| } |re||||| �� � �� � } nd} | �|� | | �| � | � |� | | | �� qdS )z5 Test that setKeys sets up the MACs. r" r r1 r r N) r" rK �macMapr� r{ � digest_sizer r� r� r� rq r% rt ) r2 r� �macName�mod�outMac�inMac�ds�i�o�seqidr� r[ �macr) r) r* �test_setKeysMACs s* �z SSHCiphersTests.test_setKeysMACsc C s� g d�}|D ]9\}}}t �dddd�}|�d|�|_t�d|dd� �\}|dd� }| �|t�|� ||��d|�d|��� qdS ) z� L{SSHCiphers.makeMAC} computes the HMAC of an outgoing SSH message with a particular sequence id and content data. ))s s Hi Theres 9294727a3638bb1c13f48ef8158bfc9d)s Jefes what do ya want for nothing?s 750c783e6ab0b503eaa86e310a5db738)r s2 ��������������������������������������������������s 56be34521d144c88dbb8c733f0e8b3f6r r z>LNr� z Failed HMAC test vector; key=z data=) r" rK r� r� �struct�unpackr r� r� rq )r2 �vectorsr� r� r6 r� r5 � shortenedr) r) r* � test_makeMAC2 s ��z SSHCiphersTests.test_makeMACN) r& r' r( r] r r7 r r! r+ r7 r< r) r) r) r* r � s r c @ r ) �TransportLoopbackTestszL Test the server transport and client transport against each other, c s� t � }t�� �|�_|�� g �_�fdd��_t�� � dd� � _g � _� fdd�� _� fdd�� _ t �j� � � � ��_ |���|� �� �fdd�}t��� �}|�|�� � |S )z� Run an async client and server, modifying each using the mod function provided. Returns a Deferred called back when both Protocols have disconnected. @type mod: C{func} @rtype: C{defer.Deferred} c � � j �| |f�S r� rL ��code�desc)�serverr) r* r< d � z9TransportLoopbackTests._runClientServer.<locals>.<lambda>c S s t �d �S r� )r r� )r= r> r) r) r* r< f s c r> r� rL r? ��clientr) r* r< h rC c s � � � S r� )r� r) rD r) r* r< i s c s� t |jd |jd |jd |jd g�}� �|jg � � �|jtjdfg� |jd dkr>� � |� � |� � � |� � |� n� � |� � |� � � |� � |� |jd dkrg� � |� � |� � � |� � |� d S � � |� � |� � � |� � |� d S )Nr s user closed connectionr ) �reprrL rX rV rY r rD r" r� r# r� r% r� )�ignoredrB rE r� rI r) r* �checkn s, �� �z6TransportLoopbackTests._runClientServer.<locals>.check)r� r" r- r r rD rP r. r� r� �listr� r! rW r � loopbackAsyncr� )r2 r/ r rH r� r) )rE r2 rB r* �_runClientServerV s$ z'TransportLoopbackTests._runClientServerc �B g }t jjdg D ]� � fdd�}|�| �|�� q tj|dd�S )z{ Test that the client and server play nicely together, in all the various combinations of ciphers. r c � � g| _ | S r� r r� ��cipherr) r* � setCipher� � z6TransportLoopbackTests.test_ciphers.<locals>.setCipherT��fireOnOneErrback)r" r+ rL rM rK r � DeferredList)r2 � deferredsrP r) rN r* � test_ciphers� s z#TransportLoopbackTests.test_ciphersc rL )z> Like test_ciphers, but for the various MACs. r c rM r� r r� �r6 r) r* �setMAC� rQ z0TransportLoopbackTests.test_macs.<locals>.setMACTrR )r" r+ rX rM rK r rT )r2 rU rX r) rW r* � test_macs� s z TransportLoopbackTests.test_macsc �< g }t jjD ]� � fdd�}|�| �|�� qtj|dd�S )zG Like test_ciphers, but for the various key exchanges. c rM r� r� r� �r� r) r* �setKeyExchange� rQ z@TransportLoopbackTests.test_keyexchanges.<locals>.setKeyExchangeTrR )r" r+ rV rM rK r rT )r2 rU r\ r) r[ r* �test_keyexchanges� � z(TransportLoopbackTests.test_keyexchangesc rZ )zF Like test_ciphers, but for the various compressions. c rM r� r r� �� compressionr) r* �setCompression� rQ z@TransportLoopbackTests.test_compressions.<locals>.setCompressionTrR )r" r+ rY rM rK r rT )r2 rU ra r) r_ r* �test_compressions� r^ z(TransportLoopbackTests.test_compressionsN) r& r' r( r] r r7 rK rV rY r] rb r) r) r) r* r= N s 7 r= )kr] r� r r r8 r� �hashlibr r r r r �typingr r �twistedr r �twisted.conch.errorr �twisted.conch.sshr r r �twisted.internetr �twisted.protocolsr �twisted.pythonr �twisted.python.compatr �twisted.python.randbytesr �twisted.python.reflectr � twisted.testr �twisted.trial.unittestr r r �strr� r �cryptography.exceptionsr � cryptography.hazmat.backendsr � cryptography.hazmat.primitivesr �)cryptography.hazmat.primitives.asymmetricr r r r r! r" �twisted.conch.testr# �x25519_supportedr6 r: r@ r+ rB r^ r� � SSHServicer� r/ r� r� r� r� r� r� r� r r r r r� r� r� r� r� r� r r rn r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r= r) r) r) r* �<module> s EA#+= ; � � � �e g � � 5� � ? #� � V� �kmPK F��Z��ղ�` �` ' __pycache__/test_telnet.cpython-310.pycnu �[��� o �b�h � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddl mZ eej�G d d � d ��ZG d d � d ej�ZG d d� dej�ZG dd� dej�ZG dd� dej�ZdS )z$ Tests for L{twisted.conch.telnet}. � )� implementer)� verifyObject)�telnet)�defer)� iterbytes)� proto_helpers)�unittestc @ st e Zd ZdZdZdd� Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� ZdS )� TestProtocol� c C s. d| _ g | _g | _g | _g | _g | _g | _d S )N� )�data�subcmd�calls� enabledLocal� enabledRemote� disabledLocal�disabledRemote��selfr r �@/usr/lib/python3/dist-packages/twisted/conch/test/test_telnet.py�__init__ s zTestProtocol.__init__c sL i }|_ � j|d< |j�� }|_dD ]}|f� fdd� |tt|�< qd S )N� ) �EOR�NOP�DM�BRK�IP�AO�AYT�EC�EL�GAc s � j �|�S �N)r �append)�arg�cmdr r r �<lambda>) s z-TestProtocol.makeConnection.<locals>.<lambda>)�negotiationMap�neg_TEST_COMMAND� commandMap�copy�getattrr )r � transport�dr% r r r �makeConnection# s �zTestProtocol.makeConnectionc C s | j |7 _ d S r" )r �r r r r r � dataReceived+ s zTestProtocol.dataReceivedc C � d S r" r )r �reasonr r r �connectionLost. s zTestProtocol.connectionLostc C s || _ d S r" )r )r �payloadr r r r( 1 s z TestProtocol.neg_TEST_COMMANDc C � || j v r | j�|� dS dS �NTF)�localEnableabler r# �r �optionr r r � enableLocal4 � zTestProtocol.enableLocalc C � | j �|� d S r" )r r# r8 r r r � disableLocal: � zTestProtocol.disableLocalc C r5 r6 )�remoteEnableabler r# r8 r r r � enableRemote= r; zTestProtocol.enableRemotec C r< r" )r r# r8 r r r � disableRemoteC r> zTestProtocol.disableRemotec C r1 r" r r r r r �connectionMadeF � zTestProtocol.connectionMadec C r1 r" r )r �command�argumentr r r �unhandledCommandJ rC z TestProtocol.unhandledCommandc C r1 r" r �r rD r r r r �unhandledSubnegotiationN rC z$TestProtocol.unhandledSubnegotiationN)�__name__� __module__� __qualname__r7 r? r r. r0 r3 r( r: r= r@ rA rB rF rH r r r r r s r c @ s e Zd Zdd� ZdS )�InterfacesTestsc C s t �� }tt j|� dS )zO L{telnet.TelnetProtocol} implements L{telnet.ITelnetProtocol} N)r �TelnetProtocolr �ITelnetProtocol)r �pr r r �test_interfaceT s z InterfacesTests.test_interfaceN)rI rJ rK rP r r r r rL S s rL c @ sB e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zg g g g fd&d'�Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLS )M�TelnetTransportTestsz. Tests for L{telnet.TelnetTransport}. c C s( t �t�| _t�� | _| j�| j� d S r" )r �TelnetTransportr rO r �StringTransport�tr. r r r r �setUpa s zTelnetTransportTests.setUpc C s> | j j}g d�}|D ]}| j �|� q | �|jd�|�� d S )N)s here are some bytes la la las some more arrive heres lots of bytes to play withs la la las ta de das dumr )rO �protocolr0 � assertEqualr �join�r �h�L�br r r �testRegularBytesf s z%TelnetTransportTests.testRegularBytesc C s� | j j}g d�}|D ]}| j �|� q | �|j|d d d� d |d d d� d |d d d� d |d d d� d � d S ) N)s here is the first line s here is the second line s here is the third line s here is the last line r ���� � � � � )rO rV r0 rW r rY r r r �testNewlineHandlingy s, ��������z(TelnetTransportTests.testNewlineHandlingc C sF | j j}g d�}|D ]}| j �|� q | �|jd�|��dd�� d S )N)s* here are some bytes�� with an embedded IACs& and here is a test of a border escape�s � did you get that IAC?r s ��� �)rO rV r0 rW r rX �replacerY r r r � testIACEscape� s z"TelnetTransportTests.testIACEscapec C sn | j j}tjtt|� }dd| d g}|D ]}| j �|� q| �|j|g� | �|jd� |�� |d�� d S )Ns Here's some bytes, tra la las But ono!s an interruptr ) rO rV r �IACr+ r0 rW r r rX rf )r �cmdNamerZ r% r[ r\ r r r �_simpleCommandTest� s z'TelnetTransportTests._simpleCommandTestc C � | � d� d S )Nr �rj r r r r � testInterrupt� � z"TelnetTransportTests.testInterruptc C rk )Nr rl r r r r �testEndOfRecord� rn z$TelnetTransportTests.testEndOfRecordc C rk )Nr rl r r r r �testNoOperation� rn z$TelnetTransportTests.testNoOperationc C rk )Nr rl r r r r � testDataMark� rn z!TelnetTransportTests.testDataMarkc C rk )Nr rl r r r r � testBreak� rn z TelnetTransportTests.testBreakc C rk )Nr rl r r r r �testAbortOutput� rn z$TelnetTransportTests.testAbortOutputc C rk )Nr rl r r r r �testAreYouThere� rn z$TelnetTransportTests.testAreYouTherec C rk )Nr rl r r r r �testEraseCharacter� rn z'TelnetTransportTests.testEraseCharacterc C rk )Nr rl r r r r � testEraseLine� rn z"TelnetTransportTests.testEraseLinec C rk )Nr! rl r r r r � testGoAhead� rn z TelnetTransportTests.testGoAheadc C s| | j j}tjtj d tj tj }d| dg}|D ]}| j �|� q| �|jd� |�� |d�� | �|j t t d��� d S )Ns hello worlds These are some bytes but soons there will be some morer s hello world)rO rV r rh �SB�SEr0 rW r rX rf r �listr �r rZ r% r[ r\ r r r �testSubnegotiation� s z'TelnetTransportTests.testSubnegotiationc C s� | j j}tjtj d tj tj tj }d| d dg}|D ]}| j �|� q | �|jd� |�� |d�� | �|j tjg� d S )Nr s Some bytes are heres and herer ) rO rV r rh rx ry r0 rW r rX rf r r{ r r r � testSubnegotiationWithEmbeddedSE� s " z5TelnetTransportTests.testSubnegotiationWithEmbeddedSEc C s� t jt j d t j d t j t j }tt|��D ]L}t� }| j_|� | j� |d |� ||d � }}d| |d g}|D ]}| j� |� q?| � |j d� |��|d�� | � |jt jgttd�� � qd S )Nr s hellos first parts last partr )r rh rx ry �range�lenr rO rV r. r0 rW r rX rf r rz r )r r% �irZ �ar\ r[ r r r r �testBoundarySubnegotiation� s0 ������� �z/TelnetTransportTests.testBoundarySubnegotiationc C s< | � |j|� | � |j|� | � |j|� | � |j|� d S r" )rW r r r r )r �o�eL�eR�dL�dRr r r �_enabledHelper s z#TelnetTransportTests._enabledHelperc C �r t jt j d }d| d }| j�|� | �| jjj|�|d�� | �| j � � t jt j d � | � | jj� d S �Nr s surrounding bytess to spice things upr ) r rh �WILLrO r0 rW rV r rf rT �value�DONTr� �r r% r r r r �testRefuseWill � z#TelnetTransportTests.testRefuseWillc C r� r� ) r rh �DOrO r0 rW rV r rf rT r� �WONTr� r� r r r � testRefuseDo r� z!TelnetTransportTests.testRefuseDoc C �h t jt j d }d| d }| jj}d|_| j�|� | �| j� � t jt j d � | j |dgd� d S )N� � paddings trailer)r� )r� ) r rh r� rO rV r7 r0 rW rT r� r� r� �r r% r rZ r r r � testAcceptDo& s z!TelnetTransportTests.testAcceptDoc C r� )N� �s headerr� )r� �r� ) r rh r� rO rV r? r0 rW rT r� r� r� r� r r r �testAcceptWill3 s z#TelnetTransportTests.testAcceptWillc C � t jt j d }| j�d�}d|j_d| }| j�|� | �| jj j |� |d�� | �| j � � t jt j d � | �|jjd� | j| jj dgd� d S )N� )�yess fiddle deer �no�r� )r rh r� rO �getOptionState�him�stater0 rW rV r rf rT r� r� r� �r r% �sr r r r �testAcceptWont? � z#TelnetTransportTests.testAcceptWontc C r� )Nr� r� s fiddle dum r r� )r� )r rh r� rO r� �usr� r0 rW rV r rf rT r� r� r� r� r r r �testAcceptDontS r� z#TelnetTransportTests.testAcceptDontc C �f t jt j d }d| d }| j�|� | �| jjj|�|d�� | �| j � � d� | � | jj� d S �N� G� dum de dum� tra la lar ) r rh r� rO r0 rW rV r rf rT r� r� r� r r r �testIgnoreWontg s z#TelnetTransportTests.testIgnoreWontc C r� r� ) r rh r� rO r0 rW rV r rf rT r� r� r� r r r �testIgnoreDonts s z#TelnetTransportTests.testIgnoreDontc C �z t jt j d }| j�d�}d|j_d| d }| j�|� | �| jj j |� |d�� | �| j � � d� | �| jj � d S �N� Vr� r� r� r )r rh r� rO r� r� r� r0 rW rV r rf rT r� r� r� r r r �testIgnoreWill� � z#TelnetTransportTests.testIgnoreWillc C r� r� )r rh r� rO r� r� r� r0 rW rV r rf rT r� r� r� r r r � testIgnoreDo� r� z!TelnetTransportTests.testIgnoreDoc sv �j �d�}�j j� d� _���j�� tjtj d � �j � tjtj d � |� �jd� |� � �fdd�� |S )N� B�r� Tc s �j � dgd�S )Nr� r� )r� ��_�rZ r r r r& � s z@TelnetTransportTests.testAcceptedEnableRequest.<locals>.<lambda>) rO �dorV r? rW rT r� r rh r� r0 r� � addCallback�r r- r r� r �testAcceptedEnableRequest� s z.TelnetTransportTests.testAcceptedEnableRequestc �� d�j j_�j �d�}���j�� tjtj d � �j � d�� ��� j j d� ��� j j d� ��� j j� ��� j j� �j �tjtj d � ��|tj�}|��fdd�� |�� �fdd�� |S )z� If the peer refuses to enable an option we request it to enable, the L{Deferred} returned by L{TelnetProtocol.do} fires with an L{OptionRefused} L{Failure}. r� r� r� c � � � � jj�S r" �r� rO rV ��ignoredr r r r& � � z@TelnetTransportTests.test_refusedEnableRequest.<locals>.<lambda>c � �� � jj�S r" )� assertFalser� � negotiatingr� �r� r r r r& � r� )rO rV r? r� rW rT r� r rh r� r� r� r� r� � assertTruer� r� r0 r� � assertFailure� OptionRefusedr� r� r r� r �test_refusedEnableRequest� � z.TelnetTransportTests.test_refusedEnableRequestc r� )z� If the peer refuses to allow us to enable an option, the L{Deferred} returned by L{TelnetProtocol.will} fires with an L{OptionRefused} L{Failure}. r� r� r� c r� r" r� r� r r r r& � r� z>TelnetTransportTests.test_refusedEnableOffer.<locals>.<lambda>c r� r" )r� r� r� r� r� r r r& � r� )rO rV r7 �willrW rT r� r rh r� r� r� r� r� r� r� r� r0 r� r� r� r� r� r r� r �test_refusedEnableOffer� r� z,TelnetTransportTests.test_refusedEnableOfferc sz � j �d�}d|j_� j �d�}� �� j�� tj tj d � � j � tj tj d � |� � jd� |� � fdd�� |S )Nr� r� Tc s � j � jjdgd�S )Nr� r� r� r� r r r r& � s zATelnetTransportTests.testAcceptedDisableRequest.<locals>.<lambda>)rO r� r� r� �dontrW rT r� r rh r� r0 r� r� �r r� r- r r r �testAcceptedDisableRequest� s z/TelnetTransportTests.testAcceptedDisableRequestc sd � j �d�}d|j_� j �d� � fdd�}� fdd�}� fdd�}|d �}|�|� |�|� |S ) N� $r� c � � j �d�}� �|tj�S �Nr� )rO r� r� r �AlreadyNegotiating��xr- r r r �_do � zITelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._doc r� r� )rO r� r� r r� r� r r r �_dont r� zKTelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._dontc s~ � j �tjtj d � � j� j jdgd� d� j j_� j �d�}� j �tjtj d � |� � j d� |� � fdd�� |S )Nr� r� )r� Tc s � j � jjdgdgd�S )Nr� )r� r� r� r� r r r r& s �z^TelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._final.<locals>.<lambda>) rO r0 r rh r� r� rV r? r� r� r� rW r� r r r �_final s �zLTelnetTransportTests.testNegotiationBlocksFurtherNegotiation.<locals>._final)rO r� r� r� r� r� )r r� r� r� r� r- r r r �'testNegotiationBlocksFurtherNegotiation s z<TelnetTransportTests.testNegotiationBlocksFurtherNegotiationc C s | j �d�}| �|tj�S )N� �)rO r� r� r �AlreadyDisabledr� r r r �#testSuperfluousDisableRequestRaises' s z8TelnetTransportTests.testSuperfluousDisableRequestRaisesc C s. | j �d�}d|j_| j �d�}| �|tj�S )Nr� r� )rO r� r� r� r� r� r �AlreadyEnabledr� r r r �"testSuperfluousEnableRequestRaises, s z7TelnetTransportTests.testSuperfluousEnableRequestRaisesc C sx | j �d�}| j �d�}| j �d�}G dd� dt�}| j �|d�� | �||�}| �||�}| �||�}t�|||g�S )Nr � #� 4c @ s e Zd ZdS )zLTelnetTransportTests.testLostConnectionFailsDeferreds.<locals>.TestExceptionN)rI rJ rK r r r r � TestException8 s r� zTotal failure!)rO r� � Exceptionr3 r� r � gatherResults)r �d1�d2�d3r� r r r � testLostConnectionFailsDeferreds3 s z5TelnetTransportTests.testLostConnectionFailsDeferredsN))rI rJ rK �__doc__rU r] rd rg rj rm ro rp rq rr rs rt ru rv rw r| r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rQ \ sN % rQ c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � TestTelnetzP A trivial extension of the telnet protocol class useful to unit tests. c C s t j�| � g | _d S r" )r �Telnetr �eventsr r r r r H s zTestTelnet.__init__c C s | j �d|f� dS )z: Record the given data in C{self.events}. �bytesN�r� r# r/ r r r �applicationDataReceivedL s z"TestTelnet.applicationDataReceivedc C � | j �d||f� dS )z= Record the given command in C{self.events}. rD Nr� rG r r r rF R � zTestTelnet.unhandledCommandc C r� )zL Record the given subnegotiation command in C{self.events}. � negotiateNr� rG r r r rH X r� z"TestTelnet.unhandledSubnegotiationN)rI rJ rK r� r r� rF rH r r r r r� C s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )#� TelnetTestsz� Tests for L{telnet.Telnet}. L{telnet.Telnet} implements the TELNET protocol (RFC 854), including option and suboption negotiation, and option state tracking. c C s t � | _dS )zM Create an unconnected L{telnet.Telnet} to be used by tests. N)r� rV r r r r rU g s zTelnetTests.setUpc C � | � | j�d�� dS )z� L{telnet.Telnet.enableLocal} should reject all options, since L{telnet.Telnet} does not know how to implement any options. � N)r� rV r: r r r r �test_enableLocalm � z TelnetTests.test_enableLocalc C r� )z� L{telnet.Telnet.enableRemote} should reject all options, since L{telnet.Telnet} does not know how to implement any options. r� N)r� rV r@ r r r r �test_enableRemotet r� z TelnetTests.test_enableRemotec C � | � t| jjd� dS )a It is an error for L{telnet.Telnet.disableLocal} to be called, since L{telnet.Telnet.enableLocal} will never allow any options to be enabled locally. If a subclass overrides enableLocal, it must also override disableLocal. r� N)� assertRaises�NotImplementedErrorrV r= r r r r �test_disableLocal{ � z TelnetTests.test_disableLocalc C r� )a It is an error for L{telnet.Telnet.disableRemote} to be called, since L{telnet.Telnet.enableRemote} will never allow any options to be enabled remotely. If a subclass overrides enableRemote, it must also override disableRemote. r� N)r� r� rV rA r r r r �test_disableRemote� r z TelnetTests.test_disableRemotec C �6 t �� }| j�|� | j�dd� | �|�� d� dS )z� L{telnet.Telnet.requestNegotiation} formats the feature byte and the payload bytes into the subnegotiation format and sends them. See RFC 855. � s s ����N�r rS rV r. �requestNegotiationrW r� �r r, r r r �test_requestNegotiation� s �z#TelnetTests.test_requestNegotiationc C r )z� If the payload for a subnegotiation includes I{IAC}, it is escaped by L{telnet.Telnet.requestNegotiation} with another I{IAC}. See RFC 855. r re s ������Nr r r r r �!test_requestNegotiationEscapesIAC� s z-TelnetTests.test_requestNegotiationEscapesIACc G s, g }| j _| j �|� | �|t|�� dS )z Pass the given bytes to the protocol's C{dataReceived} method and assert that the given events occur. N)rV r� r0 rW rz )r r �expected�receivedr r r �_deliver� s zTelnetTests._deliverc C � | � dd� dS )zc One application-data byte in the default state gets delivered right away. � a)r� r N�r r r r r �test_oneApplicationDataByte� � z'TelnetTests.test_oneApplicationDataBytec C r )za Two application-data bytes in the default state get delivered together. � bc)r� r Nr r r r r � test_twoApplicationDataBytes� r z(TelnetTests.test_twoApplicationDataBytesc C s | � dtj d� dS )z~ Three application-data bytes followed by a control byte get delivered, but the control byte doesn't. � def)r� r N�r r rh r r r r � test_threeApplicationDataBytes� r� z*TelnetTests.test_threeApplicationDataBytesc C s, | � tj� | � tjd dtjd f� dS )zy IAC in the escaped state gets delivered and so does another application-data byte following it. � gr� Nr r r r r �test_escapedControl� s z TelnetTests.test_escapedControlc C s� | � d� | � dd� | � dd� | � d� | � dd� | � dd� | � d� | � dd � | � d d � | � d� | � tjtj d d dtj d f� d S )a� A carriage return only puts the protocol into the newline state. A linefeed in the newline state causes just the newline to be delivered. A nul in the newline state causes a carriage return to be delivered. An IAC in the newline state causes a carriage return to be delivered and puts the protocol into the escaped state. Anything else causes a carriage return and that thing to be delivered. ra r_ )r� r_ s r� )r� ra s r )r� � ar � xr� Nr r r r r �test_carriageReturn� s �z TelnetTests.test_carriageReturnc C s& | � dtj tj ddtjdf� dS )zt Application bytes received before a command are delivered before the command is processed. r )r� r rD N)r r rh r r r r r �'test_applicationDataBeforeSimpleCommand� s �z3TelnetTests.test_applicationDataBeforeSimpleCommandc C s2 i | j _| �dtj tj d ddtjdf� dS )z~ Application bytes received before a WILL/WONT/DO/DONT are delivered before the command is processed. � yr� )r� r rD N)rV r) r r rh r� r r r r �!test_applicationDataBeforeCommand� s �z-TelnetTests.test_applicationDataBeforeCommandc C s6 | � dtj tj d tj tj ddddgf� dS )z� Application bytes received before a subnegotiation command are delivered before the negotiation is processed. � zs Qx)r� r r� � Qr N)r r rh rx ry r r r r �(test_applicationDataBeforeSubnegotiation s �z4TelnetTests.test_applicationDataBeforeSubnegotiationN)rI rJ rK r� rU r� r� r r r r r r r r r r r r r! r r r r r� _ s$ r� N)r� �zope.interfacer �zope.interface.verifyr � twisted.conchr �twisted.internetr �twisted.python.compatr � twisted.testr � twisted.trialr rN r �TestCaserL rQ r� r� r� r r r r �<module> s" > j PK F��Z^a� d d ) __pycache__/test_recvline.cpython-310.pycnu �[��� o �bXc � @ sb d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m Z ddl m Z m Z ddlmZmZmZ dd lmZ dd lmZ dd lmZ dd lmZmZ ed �Ze ej �Z ej � ej!�e d< G dd� de�Z"ddlm#Z# ddlm$Z$ ddl%m&Z& G dd� dej'�Z(dZ)dZ*dZ+dZ,dZ-dZ.d Z/d Z0d Z1dd l m2Z2 z dd l3m4Z4m5Z5m6Z6m7Z7m8Z8 dd!l9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z? W n e@y� d"ZAY nGw d#ZAG d$d%� d%e:jB�ZCG d&d'� d'e;jD�ZEG d(d)� d)e?jF�ZGG d*d+� d+e>jH�ZIG d,d-� d-e7�ZJG d.d/� d/e6�ZKG d0d1� d1e8�ZLe�MeKeLe=jN� G d2d3� d3e$jO�ZPG d4d5� d5�ZQG d6d7� d7eQ�ZRdd8lSmTZT G d9d:� d:ejUeTjV�ZWG d;d<� d<ejXeTjV�ZYG d=d>� d>eQ�ZZG d?d@� d@eQ�Z[G dAdB� dB�Z\G dCdD� dDeZee\�Z]G dEdF� dFeRee\�Z^ee dG�G dHdI� dIe[ee\��Z_G dJdK� dK�Z`G dLdM� dMeZee`�ZaG dNdO� dOeRee`�Zbee dP�G dQdR� dRe[ee`��ZcG dSdT� dTe�ZddS )UzU Tests for L{twisted.conch.recvline} and fixtures for testing related functionality. � N)�skipIf)�recvline)�insults)�portal)�defer�error)� components�filepath�reflect)� iterbytes)� requireModule)�StringTransport)�SkipTest�TestCaseztwisted.conch.stdio� PYTHONPATHc @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )� ArrowsTestsc sF t � � _t�� � _t�� � _� fdd�� j_� � j_ � j� � j� d S )Nc s � j S �N)�p� ��selfr �B/usr/lib/python3/dist-packages/twisted/conch/test/test_recvline.py�<lambda>" s z#ArrowsTests.setUp.<locals>.<lambda>) r �underlyingTransportr �ServerProtocol�ptr �HistoricRecvLiner �protocolFactory�factory�makeConnectionr r r r �setUp s zArrowsTests.setUpc C s@ | j �dd� | j �dd� | j �dd� | �| j �� d� dS )zy When L{HistoricRecvLine} receives a printable character, it adds it to the current line buffer. � xN� y� z�� xyz� )r �keystrokeReceived� assertEqual�currentLineBufferr r r r �test_printableCharacters& s z$ArrowsTests.test_printableCharactersc sF � fdd�}t d�D ]}||� q � �� j�� d� |� jj� � �� j�� d� |� jj� � �� j�� d� |� jj� � �� j�� d� |� jj� � �� j�� d� |� jj� � �� j�� d� |� jj� � �� j�� d� |� jj� � �� j�� d� |� jj� � �� j�� d� |� jj� � �� j�� d� dS ) z� When L{HistoricRecvLine} receives a LEFT_ARROW or RIGHT_ARROW keystroke it moves the cursor left or right in the current line buffer, respectively. c � � j �| d �S r �r r' ��chr r r r 7 � z3ArrowsTests.test_horizontalArrows.<locals>.<lambda>r% r$ )� xyr# )r! s yz)r& r% N)r r( r r) r � RIGHT_ARROW� LEFT_ARROW�r �kRr. r r r �test_horizontalArrows1 s, z!ArrowsTests.test_horizontalArrowsc s| � fdd�}t d�D ]}||� q � �� j�� d� |d� |d� |d� � �� j�� d� |d� � �� j�� d � d S ) z� When {HistoricRecvLine} receives a newline, it adds the current line buffer to the end of its history buffer. c r+ r r, r- r r r r ] r/ z*ArrowsTests.test_newline.<locals>.<lambda>� xyz abc 123 ��r% � abc� 123r � c� b� a� ))r% r9 r: s cbar N)r r( r �currentHistoryBufferr3 r r r � test_newlineX s �zArrowsTests.test_newlinec s6 � fdd�}t d�D ]}||� q � �� j�� d� � �� j�� d� |� jj� � �� j�� d� � �� j�� d� |� jj� � �� j�� d� � �� j�� d � |� jj� � �� j�� d � � �� j�� d � |� jj� � �� j�� d � � �� j�� d � td �D ]}|� jj� q�� �� j�� d� d S )a When L{HistoricRecvLine} receives UP_ARROW or DOWN_ARROW keystrokes it move the current index in the current history buffer up or down, and resets the current line buffer to the previous or next line in history, respectively for each. c r+ r r, r- r r r r u r/ z1ArrowsTests.test_verticalArrows.<locals>.<lambda>r6 r7 �r& r& ))r% r9 )r: )r: r& ))r% )r9 r: )r9 r& )r r8 r$ � N) r r( r r? r) r �UP_ARROW�range� DOWN_ARROW)r r4 r. �ir r r �test_verticalArrowsn s( z ArrowsTests.test_verticalArrowsc sV � fdd�}t d�D ]}||� q � �� j�� d� |� jj� � �� j�� d� dS )z� When L{HistoricRecvLine} receives a HOME keystroke it moves the cursor to the beginning of the current line buffer. c r+ r r, r- r r r r � r/ z'ArrowsTests.test_home.<locals>.<lambda>� hello, world�rH r& )r& rH N)r r( r r) r �HOMEr3 r r r � test_home� s zArrowsTests.test_homec sb � fdd�}t d�D ]}||� q � �� j�� d� |� jj� |� jj� � �� j�� d� dS )z� When L{HistoricRecvLine} receives an END keystroke it moves the cursor to the end of the current line buffer. c r+ r r, r- r r r r � r/ z&ArrowsTests.test_end.<locals>.<lambda>rH rI N)r r( r r) r rJ �ENDr3 r r r �test_end� s zArrowsTests.test_endc s� � fdd�}t d�D ]}||� q � �� j�� d� |� jj� � �� j�� d� |� jj� |� jj� � �� j�� d� |� jj� � �� j�� d� dS )z� When L{HistoricRecvLine} receives a BACKSPACE keystroke it deletes the character immediately before the cursor. c r+ r r, r- r r r r � r/ z,ArrowsTests.test_backspace.<locals>.<lambda>r% r$ �r0 r& )r& r" N)r r( r r) r � BACKSPACEr2 r3 r r r �test_backspace� s zArrowsTests.test_backspacec s� � fdd�}t d�D ]}||� q � �� j�� d� |� jj� � �� j�� d� |� jj� |� jj� � �� j�� d� |� jj� |� jj� � �� j�� d� |� jj� |� jj� � �� j�� d� |� jj� � �� j�� d� dS ) z� When L{HistoricRecvLine} receives a DELETE keystroke, it delets the character immediately after the cursor. c r+ r r, r- r r r r � r/ z)ArrowsTests.test_delete.<locals>.<lambda>r% r$ rN )r! r& rA N)r r( r r) r �DELETEr2 r3 r r r � test_delete� s" zArrowsTests.test_deletec sr � fdd�}t d�D ]}||� q |� jj� |d� � �� j�� d� |� jj� |d� � �� j�� d� dS ) z� When not in INSERT mode, L{HistoricRecvLine} inserts the typed character at the cursor before the next character. c r+ r r, r- r r r r � r/ z)ArrowsTests.test_insert.<locals>.<lambda>r% � A)� xyAr# � B)� xyBs AzN)r r r2 r( r r) r3 r r r � test_insert� s zArrowsTests.test_insertc s~ � fdd�}t d�D ]}||� q |� jj� |� jj� |d� � �� j�� d� |� jj� |d� � �� j�� d� dS ) a When in INSERT mode and upon receiving a keystroke with a printable character, L{HistoricRecvLine} replaces the character at the cursor with the typed character rather than inserting before. Ah, the ironies of INSERT mode. c r+ r r, r- r r r r � r/ z+ArrowsTests.test_typeover.<locals>.<lambda>r% rS )rT r& rU )rV r& N)r r �INSERTr2 r( r r) r3 r r r � test_typeover� s zArrowsTests.test_typeoverc sr � fdd�}� j }|j|j|j|j|j|j|j|j|j |j |j |j |j |jfD ]}||� � �� j�� d� q'dS )z� When L{HistoricRecvLine} receives a keystroke for an unprintable function key with no assigned behavior, the line buffer is unmodified. c r+ r r, r- r r r r r/ z8ArrowsTests.test_unprintableCharacters.<locals>.<lambda>rA N)r �F1�F2�F3�F4�F5�F6�F7�F8�F9�F10�F11�F12�PGUP�PGDNr( r r) )r r4 r r. r r r �test_unprintableCharacters s( ��z&ArrowsTests.test_unprintableCharactersN)�__name__� __module__� __qualname__r r* r5 r@ rG rK rM rP rR rW rY rh r r r r r s '# r )�telnet)�helper)� LoopbackRelayc @ � e Zd Zdd� ZdS )� EchoServerc C s | j �|d | j| j � d S )Nr> )�terminal�write�ps�pn)r �liner r r � lineReceived* s zEchoServer.lineReceivedN)ri rj rk rv r r r r rp ) � rp s [Ds [Cs [As [Bs [2~s [1~s [3~s [4~� )�checkers)� ConchFactory� TerminalRealm�TerminalSession�TerminalSessionTransport� TerminalUser)�channel� connection�keys�session� transport�userauthFTc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �SessionChannels sessionc O �< t jj| g|�R i |�� || _|| _|| _|| _|| _d S r )r � SSHChannel�__init__r � protocolArgs�protocolKwArgs�width�height�r r r� r� r� r� �a�kwr r r r� S � zSessionChannel.__init__c C sh t �d| j| jddfd�}| j�| d|� | j�| dd� | j| ji | j��| _ | | j _ | j � | � d S )Ns vt102r r& s pty-reqs shell) r� �packRequest_pty_reqr� r� �conn� sendRequestr r� r� �_protocolInstancer r )r �data�termr r r � channelOpen_ s ���zSessionChannel.channelOpenc C s | j �t�� � d S r )r� �connectionLostr �ConnectionDoner r r r �closedl s zSessionChannel.closedc C � | j �|� d S r )r� � dataReceived�r r� r r r r� o � zSessionChannel.dataReceivedN)ri rj rk �namer� r� r� r� r r r r r� P s r� c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�TestConnectionc O r� r )r� � SSHConnectionr� r r� r� r� r� r� r r r r� s r� zTestConnection.__init__c C s, t | j| j| j| j| j�| _| �| j� d S r )r� r r� r� r� r� �_TestConnection__channel� openChannelr r r r �serviceStarted s �z TestConnection.serviceStartedc C � | j �|�S r )r� rr r� r r r rr � � zTestConnection.writeN)ri rj rk r� r� rr r r r r r� r s r� c @ � e Zd Zdd� Zdd� ZdS )�TestAuthc O s& t jj| |g|�R i |�� || _d S r )r� �SSHUserAuthClientr� �password)r �usernamer� r� r� r r r r� � s zTestAuth.__init__c C s t �| j�S r )r �succeedr� r r r r � getPassword� r� zTestAuth.getPasswordN)ri rj rk r� r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestTransportc O s. || _ || _|| _|| _|| _|| _|| _d S r )r r� r� r� r� r� r� ) r r r� r� r� r� r� r� r� r� r r r r� � s zTestTransport.__init__c C s t �d�S )NT)r r� )r �hostKey� fingerprintr r r � verifyHostKey� s zTestTransport.verifyHostKeyc C s8 t | j| j| j| j| j�| _| �t| j | j | j�� d S r ) r� r r� r� r� r� �_TestTransport__connection�requestServicer� r� r� r r r r �connectionSecure� s ��z TestTransport.connectionSecurec C r� r )r� rr r� r r r rr � r� zTestTransport.writeN)ri rj rk r� r� r� rr r r r r r� � s r� c @ ro )�TestSessionTransportc C s | j jjj�� S r )�avatarr� r� r �serverProtocolr r r r r � r� z$TestSessionTransport.protocolFactoryN)ri rj rk r r r r r r� � rw r� c @ s e Zd ZeZdS )� TestSessionN)ri rj rk r� �transportFactoryr r r r r� � s r� c @ � e Zd ZdS )�TestUserN�ri rj rk r r r r r� � � r� c @ r� )�NotifyingExpectableBufferc C s t �� | _t �� | _d S r )r �Deferred� onConnection�onDisconnectionr r r r r� � s z"NotifyingExpectableBuffer.__init__c C s t j�| � | j�| � d S r )rm �ExpectableBuffer�connectionMader� �callbackr r r r r� � s z(NotifyingExpectableBuffer.connectionMadec C r� r )r� �errback)r �reasonr r r r� � r� z(NotifyingExpectableBuffer.connectionLostN)ri rj rk r� r� r� r r r r r� � s r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )� _BaseMixin�P � c C s� | j �� �� }|dg| jt|� d }| �t|�t|�� tt|��D ]-}| �|| || d�|td|d �|d � �d d�|td|d �|d � � � q%d S )Nr& � r s != ) �recvlineClient� __bytes__� splitlines�HEIGHT�lenr( rD �join�max)r �lines� receivedLines� expectedLinesrF r r r � _assertBuffer� s � ���z_BaseMixin._assertBufferc s. �j �d�}��|� � �fdd�}|�|�S )Ns donec s �� � � d S r )r� )�ign��outputr r r �finished� s z)_BaseMixin._trivialTest.<locals>.finished)r� �expect� _testwrite� addCallback)r � inputLiner� �doner� r r� r � _trivialTest� s z_BaseMixin._trivialTestN)ri rj rk �WIDTHr� r� r� r r r r r� � s r� c @ r� )� _SSHMixinc s> t std��d\}}t� }t|_�fdd�|_t�� }|�||� t � |�}|� |� t |�}t jt�| �� �dd�}||jd< ||jd< | j|_|�� | �� �t��fdd���|�d �}t|�} t� �t��fd d��� t � fd d�d i ||| j | j �} t| �} | � | � |� | � �| _ | | _!|| _"| | _#| | _$�j%S ) NzMcryptography requirements missing, can't run historic recvline tests over ssh)s testusers testpassc � � S r r r �� insultsServerr r r � � z!_SSHMixin.setUp.<locals>.<lambda>i )�keySizes ssh-rsac r� r r r ��recvlineServerr r r r� c r� r r r �r� r r r r� c r� r r r �� insultsClientr r r r� r )&�sshr r{ r� � userFactory�chainedProtocolFactoryry �'InMemoryUsernamePasswordDatabaseDontUse�addUserr �Portal�registerCheckerrz r� �_getPersistentRSAKeyr �FilePath�mktemp� publicKeys� privateKeysr� � startFactoryr r � buildProtocolrn r� �ClientProtocolr� r� r� r r� � sshClient� sshServer�clientTransport�serverTransportr� ) r �ur �rlm�checker�ptl� sshFactory�sshKeyr� r� r� r� r �r� r� r� r� r r � sL � � � z_SSHMixin.setUpc C r� r )r� rr r� r r r r� $ r� z_SSHMixin._testwriteN�ri rj rk r r� r r r r r� � s 2r� )� test_telnetc @ r� )�TestInsultsClientProtocolNr� r r r r r + r� r c @ r� )�TestInsultsServerProtocolNr� r r r r r / r� r c @ r� )� _TelnetMixinc s� | � � �t�fdd���t��fdd��}t|�}t� �t�fdd��� t�� fdd��}t|�}|�|� |�|� |�� |�� �| _ || _ || _ || _ �j S )Nc r� r r r r� r r r 6 r� z$_TelnetMixin.setUp.<locals>.<lambda>c r� r r r r� r r r 7 r� c r� r r r r� r r r ; r� c r� r r r r� r r r < r� )r� r rl �TelnetTransportrn r� r r � clearBufferr� � telnetClientr� r� r� )r � telnetServerr� r r� r r r r 4 s" z_TelnetMixin.setUpc C r� r )r rr r� r r r r� L r� z_TelnetMixin._testwriteNr r r r r r 3 s r c @ r� )� _StdioMixinc s� t � � t�� fdd��}t�|�}tj}tj}|�d�s!|�d�r'|d d� }||t � | j �g}ddl m } |j|||tdd �}� | _| _|| _|| _t�td |j� �d �g��S ) Nc r� r r r �� testTerminalr r r [ r� z#_StdioMixin.setUp.<locals>.<lambda>z.pycz.pyo���r )�reactorT)�env�usePTYs >>> )r� r r� �stdio�TerminalProcessProtocol�sys� executable�__file__�endswithr �qualr� �twisted.internetr � spawnProcess� properEnvr� r � processClientr� r � gatherResults�filterr� r� )r r� r! �exe�module�argsr r� r r r r Q s$ � �z_StdioMixin.setUpc sD z� j �d� W n tjtfy Y nw � fdd�}� jj�|�S )N�KILLc s. | � tj� � �| jj� � �| jjd� d S )N� )�trapr �ProcessTerminated� assertIsNone�value�exitCoder( �status)�failurer r r r) � s z"_StdioMixin.tearDown.<locals>.trap)r� � signalProcessr �ProcessExitedAlready�OSErrorr r� � addErrback)r r) r r r �tearDown| s � z_StdioMixin.tearDownc C r� r )r� rr r� r r r r� � r� z_StdioMixin._testwriteN)ri rj rk r r4 r� r r r r r P s + r c @ sX e Zd ZeZdd� Zdd� Zdd� Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dS )�RecvlineLoopbackMixinc C s | � dg d��S )Ns first line done)� >>> first line� first line� >>> done)r� r r r r � testSimple� s �z RecvlineLoopbackMixin.testSimplec C s | � td td d g d��S )Nr7 rB � xxxx done)s >>> first xxxxs first xxxxr8 )r� �insert�leftr r r r � testLeftArrow� � �z#RecvlineLoopbackMixin.testLeftArrowc C s( | � td td td d g d��S )Ns right linerB � s xx done)s >>> right lixxs right lixxr8 )r� r; r< �rightr r r r �testRightArrow� s �z$RecvlineLoopbackMixin.testRightArrowc C � | � dtd d g d��S )N� second linerB r: )s >>> second xxxxs second xxxxr8 )r� � backspacer r r r � testBackspace� � �z#RecvlineLoopbackMixin.testBackspacec C s$ | � dtd td d g d��S )Ns delete xxxxrB � line done)s >>> delete lines delete liner8 )r� r< �deleter r r r � testDelete� s �z RecvlineLoopbackMixin.testDeletec C rB )Ns third ine� � l done)s >>> third lines third liner8 )r� r< r r r r � testInsert� rF z RecvlineLoopbackMixin.testInsertc C � | � dtd t d g d��S )Ns fourth xinerB rK )s >>> fourth lines fourth liner8 )r� r< r; r r r r � testTypeover� r> z"RecvlineLoopbackMixin.testTypeoverc C s | � td t d g d��S )Ns blah lines home done)s >>> home lines home liner8 )r� r; �homer r r r �testHome� rF z RecvlineLoopbackMixin.testHomec C rM )Ns end rB rG )s >>> end lines end liner8 )r� r< �endr r r r �testEnd� r> z RecvlineLoopbackMixin.testEndN)ri rj rk rp r� r9 r= rA rE rI rL rN rP rR r r r r r5 � s r5 c @ r� )�RecvlineLoopbackTelnetTestsNr� r r r r rS � r� rS c @ r� )�RecvlineLoopbackSSHTestsNr� r r r r rT � r� rT zBTerminal requirements missing, can't run recvline tests over stdioc @ r� )�RecvlineLoopbackStdioTestsNr� r r r r rU � s rU c @ s( e Zd ZeZdd� Zdd� Zdd� ZdS )� HistoricRecvlineLoopbackMixinc C s | � dt d g d��S )Ns first line � done)r6 r7 r6 r7 r8 )r� �upr r r r � testUpArrow� s �z)HistoricRecvlineLoopbackMixin.testUpArrowc C s | � dt t d g d��S )aH Pressing down arrow to visit an entry that was added to the history by pressing the up arrow instead of return does not raise a L{TypeError}. @see: U{http://twistedmatrix.com/trac/ticket/9031} @return: A L{defer.Deferred} that fires when C{b"done"} is echoed back. s first line partial linerW )r6 r7 s >>> partial lines partial liner8 �r� rX �downr r r r �$test_DownArrowToPartialLineInHistory� s �zBHistoricRecvlineLoopbackMixin.test_DownArrowToPartialLineInHistoryc C rM )Ns first line second line r? rW )r6 r7 � >>> second linerC r] rC r8 rZ r r r r � testDownArrow� r> z+HistoricRecvlineLoopbackMixin.testDownArrowN)ri rj rk rp r� rY r\ r^ r r r r rV � s rV c @ r� )�#HistoricRecvlineLoopbackTelnetTestsNr� r r r r r_ � r_ c @ r� )� HistoricRecvlineLoopbackSSHTestsNr� r r r r ra r` ra zKTerminal requirements missing, can't run historic recvline tests over stdioc @ r� )�"HistoricRecvlineLoopbackStdioTestsNr� r r r r rb s rb c @ s e Zd ZdZdd� ZdS )�TransportSequenceTestsz5 L{twisted.conch.recvline.TransportSequence} c C s | � ttj� dS )zh Initializing a L{recvline.TransportSequence} with no args raises an assertion. N)� assertRaises�AssertionErrorr �TransportSequencer r r r �test_invalidSequence$ s z+TransportSequenceTests.test_invalidSequenceN)ri rj rk �__doc__rg r r r r rc s rc )erh �osr �unittestr � twisted.conchr �twisted.conch.insultsr � twisted.credr r r r �twisted.pythonr r r �twisted.python.compatr �twisted.python.reflectr �twisted.test.proto_helpersr �twisted.trial.unittestr r r �dict�environr �pathsepr� �pathr rl rm �twisted.conch.test.loopbackrn r rp r< r@ rX r[ r; rO rH rQ rD ry �twisted.conch.manhole_sshrz r{ r| r} r~ �twisted.conch.sshr r� r� r� r� r� � ImportErrorr� r� r� r� r� r� r� �SSHClientTransportr� r� r� r� �registerAdapter�ISessionr� r� r� r� �twisted.conch.testr r� � TestProtocolr r r r r r5 rS rT rU rV r_ ra rb rc r r r r �<module> s� $ �"' 7 >9 5� �� �PK F��ZJ�t4� � ( __pycache__/test_address.cpython-310.pycnu �[��� o �b � @ sL d Z ddlmZ ddlmZ ddlmZ ddlmZ G dd� dej e�Z dS ) z5 Tests for L{SSHTransportAddrress} in ssh/address.py � )�SSHTransportAddress)� IPv4Address)�AddressTestCaseMixin)�unittestc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �SSHTransportAddressTestsz� L{twisted.conch.ssh.address.SSHTransportAddress} is what Conch transports use to represent the other side of the SSH connection. This tests the basic functionality of that class (string representation, comparison, &c). c C s. | � � }||�}||j�}| �|d| � dS )z� The string representation of C{SSHTransportAddress} should be "SSHTransportAddress(<stringFunction on address>)". zSSHTransportAddress(%s)N)� buildAddress�address� assertEqual)�self�stringFunction�addr� stringValue� addressValue� r �A/usr/lib/python3/dist-packages/twisted/conch/test/test_address.py�_stringRepresentation s z.SSHTransportAddressTests._stringRepresentationc C � t tddd��S )z� Create an arbitrary new C{SSHTransportAddress}. A new instance is created for each call, but always for the same address. �TCPz 127.0.0.1� �r r �r r r r r s z%SSHTransportAddressTests.buildAddressc C r )zK Like C{buildAddress}, but with a different fixed address. r z 127.0.0.2r r r r r r �buildDifferentAddress' s z.SSHTransportAddressTests.buildDifferentAddressN)�__name__� __module__� __qualname__�__doc__r r r r r r r r s r N) r �twisted.conch.ssh.addressr �twisted.internet.addressr �"twisted.internet.test.test_addressr � twisted.trialr �TestCaser r r r r �<module> s PK F��Z�1�|x |x ( __pycache__/test_insults.cpython-310.pycnu �[��� o �b�} � @ sb d dl Z d dlmZmZ d dlmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZ d dlmZ d dlmZ d dlmZmZ d dlmZ d dlm Z d d � Z d d � Z d d� Z dd� Z e!� Z"dd� Z#dd� Z$e$ddg d�g d�g d�g d�g d�g d �g d �g d �g d �g d �g d!�g d"�g d#�g d$�g d%�g d&�ge �%d'�d(�Z&e$d)ddd*gdd+gd,d-gd.d/gdd0gd1d2gd3d4gd5d6gd7d8gd9dgd:d;gd<d=gd>d?gd@dAgdBdCgdDdEgge �%dF�d(�Z'G dGdH� dH�Z(G dIdJ� dJ�Z)dKZ*G dLdM� dMe)�Z+[*G dNdO� dOe+e j,�Z-G dPdQ� dQe+e j,�Z.G dRdS� dSe+e j,�Z/G dTdU� dUe+e j,�Z0G dVdW� dWe j,e)�Z1G dXdY� dYe j,�Z2dS )Z� N)�Optional�Type) �BLINK� CS_ALTERNATE�CS_ALTERNATE_SPECIAL� CS_DRAWING�CS_UK�CS_US�G0�G1� UNDERLINE�ClientProtocol�ServerProtocol�modes� privateModes)�Protocol)� iterbytes)� ValueConstant�Values)�StringTransport)�unittestc C s t t| ��|�S �N)�super�Mock�__getattribute__)�mock�name� r �A/usr/lib/python3/dist-packages/twisted/conch/test/test_insults.py�_getattr s r c C � t | d�S )N� occurrences�r �r r r r r! " � r! c C r )N�methodsr" r# r r r r% & r$ r% c C s t | ��|� d S r )r! �append)r �objr r r �_append* s r( c C s t t| d> |B g��S )a Return the byte in 7- or 8-bit code table identified by C{column} and C{row}. "An 8-bit code table consists of 256 positions arranged in 16 columns and 16 rows. The columns and rows are numbered 00 to 15." "A 7-bit code table consists of 128 positions arranged in 8 columns and 16 rows. The columns are numbered 00 to 07 and the rows 00 to 15 (see figure 1)." p.5 of "Standard ECMA-35: Character Code Structure and Extension Techniques", 6th Edition (December 1994). � )�bytes� bytearray)�column�rowr r r �_ecmaCodeTableCoordinate1 s r. c s, � fdd�t |�D �}||d< t| tf|�S )Nc s: i | ]\}}t |�D ]\}}|r |tt|� |���q qS r )� enumerater r. )�.0�jr- �ir �� colOffsetr r � <dictcomp>J s ���z/_makeControlFunctionSymbols.<locals>.<dictcomp>�__doc__)r/ �typer )r r4 �names�doc�attrsr r3 r �_makeControlFunctionSymbolsE s �r; � CSFinalByter) )�ICH�DCH�HPA)�CUU�SSE�HPR)�CUD�CPR�REP)�CUF�SU�DA)�CUB�SD�VPA)�CNL�NP�VPR)�CPL�PP�HVP)�CHA�CTC�TBC)�CUP�ECH�SM)�CHT�CVT�MC)�ED�CBT�HPB)�EL�SRS�VPB)�IL�PTX�RM)�DL�SDS�SGR)�EF�SIMD�DSR)�EAN�DAQaW Symbolic constants for all control sequence final bytes that do not imply intermediate bytes. This happens to cover movement control sequences. See page 11 of "Standard ECMA 48: Control Functions for Coded Character Sets", 5th Edition (June 1991). Each L{ValueConstant} maps a control sequence name to L{bytes} )r4 r8 r9 � C1SevenBit�DCS�PU1�BPH�PU2�NBH�STS�CCH�NEL�MW�SSA�SPA�ESA�EPA�HTS�SOS�HTJ�VTS�SCI�PLD�CSI�PLU�ST�RI�OSC�SS2�PM�SS3�APCa Symbolic constants for all 7 bit versions of the C1 control functions See page 9 "Standard ECMA 48: Control Functions for Coded Character Sets", 5th Edition (June 1991). Each L{ValueConstant} maps a control sequence name to L{bytes} c @ s. e Zd ZeZdefdd�Zdd� Zdd� ZdS )r Nc C s. g | _ |du r i }|| _|tur|| _dS dS )z� @param methods: Mapping of names to return values @param callReturnValue: object __call__ should return N)r! r% �default�callReturnValue)�selfr% r� r r r �__init__� s �z Mock.__init__c O s. t | d�}|tu r t� }t| d|||f� |S )Nr� �__call__)r r� r r( )r� �a�kw� returnValuer r r r� � s z Mock.__call__c C s: t | d�}||v rt|| d�}nt� }t| ||f� |S )Nr% )r� )r r r( )r� r r% � attrValuer r r r � s zMock.__getattribute__)�__name__� __module__� __qualname__r� r� r� r� r r r r r r � s r c @ s e Zd Zdi fdd�ZdS )� MockMixinr c C sb |\}}| � ||� | � tt|��d� t|�\\}}} } | � |d� | � | |� | � | |� |S )N� r� )� assertEqual�lenr! ) r� � occurrence� methodName�expectedPositionalArgs�expectedKeywordArgs�attrr �call�result�argsr� r r r � assertCall� s zMockMixin.assertCallN)r� r� r� r� r r r r r� � s �r� a� def testByte%(groupName)s(self): transport = StringTransport() proto = Mock() parser = self.protocolFactory(lambda: proto) parser.factory = self parser.makeConnection(transport) bytes = self.TEST_BYTES while bytes: chunk = bytes[:%(bytesPer)d] bytes = bytes[%(bytesPer)d:] parser.dataReceived(chunk) self.verifyResults(transport, proto, parser) c @ sN e Zd ZU dZeee ed< dD ] \ZZ e e ee d� � q[[ dd� Z dS )�ByteGroupingsMixinN�protocolFactory))�Pairs� )�Triples� )�Quadsr) )�Quints� )�Sexes� )� groupName�bytesPerc C s. | � t|��d�d|f�}| �t|�g � d S )Nr �makeConnection)r� r! �popr� )r� � transport�proto�parserr� r r r � verifyResults� s z ByteGroupingsMixin.verifyResults) r� r� r� r� r r r �__annotations__�word�n�exec�_byteGroupingTestTemplater� r r r r r� � s r� c @ � e Zd ZeZdZdd� ZdS )�ServerArrowKeysTestss [A[B[C[Dc C sh t �| |||� |j|j|j|jfD ]}| �t|��d�d|d f�}| � t|�g � q| � t|�� d S )Nr �keystrokeReceived) r� r� �UP_ARROW� DOWN_ARROW� RIGHT_ARROW� LEFT_ARROWr� r! r� r� � assertFalse)r� r� r� r� �arrowr� r r r r� � s ��z"ServerArrowKeysTests.verifyResultsN�r� r� r� r r� � TEST_BYTESr� r r r r r� � s r� c @ r� )�PrintableCharactersTestss abc123ABC!@#abc123c C s� t �| |||� td�D ]}| �t|��d�d|d f�}| �t|�g � q td�D ]}| �t|��d�d||jf�}| �t|�g � q)t|�}| �||�d�� d S )Ns abc123ABC!@#r r� s abc123z should have been []) r� r� r r� r! r� r� �ALTr� )r� r� r� r� �charr� �occsr r r r� s � �z&PrintableCharactersTests.verifyResultsNr� r r r r r� s r� c @ sF e Zd ZdZeZg ZdD ] Ze�de � q d� e�Z [[dd� Z dS )�ServerFunctionKeysTestsz9Test for parsing and dispatching function keys (F1 - F12)) s OPs OQs ORs OSs 15~s 17~s 18~s 19~s 20~s 21~s 23~s 24~� [� c C sn t �| |||� tdd�D ] }t|d|f �}| �t|��d�d|d f�}| �t|�g � q | �t|�� d S )Nr� � zF%dr r� ) r� r� �range�getattrr� r! r� r� r� )r� r� r� r� �funcNum�funcArgr� r r r r� ; s �z%ServerFunctionKeysTests.verifyResultsN) r� r� r� r6 r r� �byteList� byteCodesr& �joinr� r� r r r r r� # s r� c @ sH e Zd ZeZdZdZdZdZee e e e e Z [[[[dd� Z dS )�ClientCursorMovementTestss [2Bs [4Cs [As [2Dc C s^ t �| |||� dD ]\}}| �t|��d�d| |f�}| �t|�g � q | �t|�� d S )N))�Downr� )�Forwardr) ��Upr� ��Backwardr� r� r� r �cursor)r� r� r� r! r� r� r� )r� r� r� r� �method�countr� r r r r� Q s �z'ClientCursorMovementTests.verifyResultsN) r� r� r� r r� �d2�r4�u1�l2r� r� r r r r r� F s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-S ).�ClientControlSequencesTestsc sh t � � _t� � _t� fdd��� _� � j_� j�� j� � �t � j�� d�d� jf�}� � t |�� d S )Nc s � j S r )r� r �r� r r �<lambda>g s z3ClientControlSequencesTests.setUp.<locals>.<lambda>r r� ) r r� r r� r r� �factoryr� r� r! r� r� )r� r� r r� r �setUpd s �z!ClientControlSequencesTests.setUpc C st | j �d�dd� td�D ��� t| j�}dD ]}dD ]}| �|�d�d| |f�}| �t|�� qq| �|� d S ) Nr� c s s& � | ]}d D ] }d| | V qqdS ))r� � 2s 20s 200r� Nr )r0 �chr� r r r � <genexpr>q s � � � �zBClientControlSequencesTests.testSimpleCardinals.<locals>.<genexpr>s BACD)r� r� r� r� )r� r� � �� r r� ) r� � dataReceivedr� r r! r� r� r� r� )r� r� �methr� r� r r r �testSimpleCardinalso s �� �z/ClientControlSequencesTests.testSimpleCardinalsc C sh | j �d� t| j�}| �|�d�dd�}| �t|�� | �|�d�dd�}| �t|�� | �|� d S )Ns [5;22r[rr �setScrollRegion)r� � )NN�r� r� r! r� r� r� r� �r� r� r� r r r �testScrollRegion s z,ClientControlSequencesTests.testScrollRegionc C s� | j �d� t| j�}| �|�d�dd�}| �t|�� | �|�d�dd�}| �t|�� | �|�d�d�}| �t|�� | �|�d�d�}| �t|�� | �|� d S )Ns #3#4#5#6r �doubleHeightLine)T)F�singleWidthLine�doubleWidthLiner� r� r r r �testHeightAndWidth� s z.ClientControlSequencesTests.testHeightAndWidthc C s� | j �d�dd� td�D ��� t| j�}ttfD ] }tt t t t fD ]}| � |�d�d||f�}| �t|�� q"q| �|� d S )Nr� c s( g | ]� d � � fdd�td�D ���qS )r� c s g | ]}d � | �qS )� r )r0 r� ��gr r � <listcomp>� s zKClientControlSequencesTests.testCharacterSet.<locals>.<listcomp>.<listcomp>s AB012)r� r )r0 r r� r r� � s ��z@ClientControlSequencesTests.testCharacterSet.<locals>.<listcomp>s ()r �selectCharacterSet)r� r� r� r r! r� r r r r r r r r� r� r� )r� r� �which�charsetr� r r r �testCharacterSet� s* ��� ��� z,ClientControlSequencesTests.testCharacterSetc C �d | j �d� t| j�}| �|�d�d�}| �t|�� | �|�d�d�}| �t|�� | �|� d S )Ns r �shiftIn�shiftOutr� r� r r r � testShifting� � z(ClientControlSequencesTests.testShiftingc C r )Ns NOr � singleShift2� singleShift3r� r� r r r �testSingleShifts� r z,ClientControlSequencesTests.testSingleShiftsc C r )Ns =>r �applicationKeypadMode�numericKeypadModer� r� r r r �testKeypadMode� r z*ClientControlSequencesTests.testKeypadModec C r )Ns 78r � saveCursor� restoreCursorr� r� r r r � testCursor� r z&ClientControlSequencesTests.testCursorc C sD | j �d� t| j�}| �|�d�d�}| �t|�� | �|� d S )Ns cr �resetr� r� r r r � testReset� s z%ClientControlSequencesTests.testResetc C s� | j �d� t| j�}| �|�d�d�}| �t|�� | �|�d�d�}| �t|�� | �|�d�d�}| �t|�� | �|� d S )Ns DMEr �index� reverseIndex�nextLiner� r� r r r � testIndex� s z%ClientControlSequencesTests.testIndexc C s� | j �dd�dd� tjtjtjfD �� d � | j �dd�dd� tjtjtjfD �� d � t| j�}| � |� d�d tjtjtjgf�}| � t|�� | � |� d�d tjtjtjgf�}| � t|�� | � |� d S ) Nr� � ;c s � � | ]}d |f V qdS �s %dNr �r0 �mr r r r� � � � z8ClientControlSequencesTests.testModes.<locals>.<genexpr>� hc s r r r r r r r r� � r � lr �setModes� resetModes) r� r� r� r �KAM�IRM�LNMr! r� r� r� r� r� r r r � testModes� s0 ��� ��� ��z%ClientControlSequencesTests.testModesc C sp | j �d� t| j�}dD ]}| �|�d�|�}| �t|�� q | �|�d�dd�}| �t|�� | �|� d S )Ns [K[1K[2K[J[1J[2J[3P)�eraseToLineEnd�eraseToLineBeginning� eraseLine�eraseToDisplayEnd�eraseToDisplayBeginning� eraseDisplayr �deleteCharacter)r� r� )r� r� r� r� r r r � testErasure s z'ClientControlSequencesTests.testErasurec C �R | j �d� t| j�}dD ]}| �|�d�d|f�}| �t|�� q | �|� d S )Ns [M[3M�r� r� r � deleteLiner� �r� r� �argr� r r r �testLineDeletion! � z,ClientControlSequencesTests.testLineDeletionc C r* )Ns [L[3Lr+ r � insertLiner� r- r r r �testLineInsertion* r0 z-ClientControlSequencesTests.testLineInsertionc C sX dt | j�d< | j�d� | �| j�� d� t| j�}| �|� d�d�}| �|d� d S )N)r� � �reportCursorPositions [6ns [7;8Rr ) r% r� r� r� r� r� �valuer! r� r� r� r r r �testCursorPosition3 s z.ClientControlSequencesTests.testCursorPositionc C sN t | j�}| j�d� | �|�d�dd� | j�d� | �|�d�dd� dS )z� Contiguous non-control bytes are passed to a single call to the C{write} method of the terminal to which the L{ClientProtocol} is connected. � ar �write)r7 � bc)r9 N)r! r� r� r� r� r� )r� r� r r r �test_applicationDataBytes> s z5ClientControlSequencesTests.test_applicationDataBytesc C sR t | j�}| j�|� |r | j|�d�g|�d��R � |s | �|d|��� d S )Nr z No other calls should happen: )r! r� r� r� r� r� r� )r� �data�callsr� r r r �_applicationDataTestJ s �z0ClientControlSequencesTests._applicationDataTestc C � | � dddg� dS )z� Application data bytes followed by a shift-in command are passed to a call to C{write} before the terminal's C{shiftIn} method is called. s ab�r8 )s ab)r N�r= r� r r r � test_shiftInAfterApplicationDataQ � z<ClientControlSequencesTests.test_shiftInAfterApplicationDatac C r> )z� Application data bytes followed by a shift-out command are passed to a call to C{write} before the terminal's C{shiftOut} method is called. s abr? )r Nr@ r� r r r �!test_shiftOutAfterApplicationDataX rB z=ClientControlSequencesTests.test_shiftOutAfterApplicationDatac C r> )z� Application data bytes followed by a cursor-backward command are passed to a call to C{write} before the terminal's C{cursorBackward} method is called. s abr? )�cursorBackwardNr@ r� r r r �'test_cursorBackwardAfterApplicationData_ s zCClientControlSequencesTests.test_cursorBackwardAfterApplicationDatac C s, | � dddg� | � ddddgffg� dS )z� Application data bytes followed by an escape character are passed to a call to C{write} before the terminal's handler method for the escape is called. s abDr? )r s ab[4hr r) Nr@ r� r r r � test_escapeAfterApplicationDatag s �z;ClientControlSequencesTests.test_escapeAfterApplicationDataN)r� r� r� r� r� r� r� r� r r r r r r r! r) r/ r2 r6 r: r= rA rC rE rF r r r r r� c s. r� c @ s e Zd ZdZedd�Zeedd� Zdd� Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z!d;d<� Z"d=d>� Z#d?d@� Z$dAdB� Z%dCS )D�ServerProtocolOutputTestszh Tests for the bytes L{ServerProtocol} writes to its transport when its methods are called. r� � r� c C s" t � | _t� | _| j�| j� d S r )r �protocolr r� r� r� r r r r� � s z ServerProtocolOutputTests.setUpc C �0 | j �d� | �| j�� | jd tjj � dS )z� L{ServerProtocol.cursorUp} writes the control sequence ending with L{CSFinalByte.CUU} to its transport. r� � 1N)rI �cursorUpr� r� r5 r� r< r@ r� r r r � test_cursorUp� � �z'ServerProtocolOutputTests.test_cursorUpc C rJ )z� L{ServerProtocol.cursorDown} writes the control sequence ending with L{CSFinalByte.CUD} to its transport. r� rK N)rI � cursorDownr� r� r5 r� r< rC r� r r r �test_cursorDown� rN z)ServerProtocolOutputTests.test_cursorDownc C rJ )z� L{ServerProtocol.cursorForward} writes the control sequence ending with L{CSFinalByte.CUF} to its transport. r� rK N)rI � cursorForwardr� r� r5 r� r< rF r� r r r �test_cursorForward� rN z,ServerProtocolOutputTests.test_cursorForwardc C rJ )z� L{ServerProtocol.cursorBackward} writes the control sequence ending with L{CSFinalByte.CUB} to its transport. r� rK N)rI rD r� r� r5 r� r< rI r� r r r �test_cursorBackward� rN z-ServerProtocolOutputTests.test_cursorBackwardc C s2 | j �dd� | �| j�� | jd tjj � dS )z� L{ServerProtocol.cursorPosition} writes a control sequence ending with L{CSFinalByte.CUP} and containing the expected coordinates to its transport. r s 1;1N)rI �cursorPositionr� r� r5 r� r< rU r� r r r �test_cursorPosition� s �z-ServerProtocolOutputTests.test_cursorPositionc C �* | j �� | �| j�� | jtjj � dS )z� L{ServerProtocol.cursorHome} writes a control sequence ending with L{CSFinalByte.CUP} and no parameters, so that the client defaults to (1, 1). N)rI � cursorHomer� r� r5 r� r< rU r� r r r �test_cursorHome� � z)ServerProtocolOutputTests.test_cursorHomec C s, | j �� | �| j�� | jtdd� � dS )z� L{ServerProtocol.index} writes the control sequence ending in the 8-bit code table coordinates 4, 4. Note that ECMA48 5th Edition removes C{IND}. r) N)rI r r� r� r5 �ESCr. r� r r r � test_index� s �z$ServerProtocolOutputTests.test_indexc C rV )zt L{ServerProtocol.reverseIndex} writes the control sequence ending in the L{C1SevenBit.RI}. N)rI r r� r� r5 rZ rl r� r� r r r �test_reverseIndex� � z+ServerProtocolOutputTests.test_reverseIndexc C s | j �� | �| j�� d� dS )zM L{ServerProtocol.nextLine} writes C{" "} to its transport. s N)rI r r� r� r5 r� r r r � test_nextLine� s z'ServerProtocolOutputTests.test_nextLinec C �P t jt jt jg}| j�|� | �| j�� | j d� dd� |D �� t j j � dS )z� L{ServerProtocol.setModes} writes a control sequence containing the requested modes and ending in the L{CSFinalByte.SM}. r c s r r r r r r r r� � r z:ServerProtocolOutputTests.test_setModes.<locals>.<genexpr>N) r r r r rI r r� r� r5 r� r� r< rW �r� � modesToSetr r r � test_setModes� s ���z'ServerProtocolOutputTests.test_setModesc C r_ )z� L{ServerProtocol.setPrivatesModes} writes a control sequence containing the requested private modes and ending in the L{CSFinalByte.SM}. r c s r r r r r r r r� r zAServerProtocolOutputTests.test_setPrivateModes.<locals>.<genexpr>N) r �ERROR�COLUMN�ORIGINrI r r� r� r5 r� r� r< rW )r� �privateModesToSetr r r �test_setPrivateModes� s � ���z.ServerProtocolOutputTests.test_setPrivateModesc C r_ )zs L{ServerProtocol.resetModes} writes the control sequence ending in the L{CSFinalByte.RM}. r c s r r r r r r r r� r z<ServerProtocolOutputTests.test_resetModes.<locals>.<genexpr>N) r r r r rI r r� r� r5 r� r� r< rc r` r r r �test_resetModes s ���z)ServerProtocolOutputTests.test_resetModesc C rV )zp L{ServerProtocol.singleShift2} writes an escape sequence followed by L{C1SevenBit.SS2} N)rI r r� r� r5 rZ rl r� r� r r r �test_singleShift2 r] z+ServerProtocolOutputTests.test_singleShift2c C rV )zp L{ServerProtocol.singleShift3} writes an escape sequence followed by L{C1SevenBit.SS3} N)rI r r� r� r5 rZ rl r� r� r r r �test_singleShift3 r] z+ServerProtocolOutputTests.test_singleShift3c C sB | j �tt�tt�� | �| j�� | jdttf t j j � dS )z� L{ServerProtocol.selectGraphicRendition} writes a control sequence containing the requested attributes and ending with L{CSFinalByte.SGR} s %d;%dN) rI �selectGraphicRendition�strr r r� r� r5 r� r< rf r� r r r �test_selectGraphicRendition$ s �z5ServerProtocolOutputTests.test_selectGraphicRenditionc C rV )zz L{ServerProtocol.horizontalTabulationSet} writes the escape sequence ending in L{C1SevenBit.HTS} N)rI �horizontalTabulationSetr� r� r5 rZ rl rz r� r r r � test_horizontalTabulationSet0 r] z6ServerProtocolOutputTests.test_horizontalTabulationSetc C rV )a L{ServerProtocol.eraseToLineEnd} writes the control sequence sequence ending in L{CSFinalByte.EL} and no parameters, forcing the client to default to 0 (from the active present position's current location to the end of the line.) N)rI r" r� r� r5 r� r< r^ r� r r r �test_eraseToLineEnd8 � z-ServerProtocolOutputTests.test_eraseToLineEndc C �. | j �� | �| j�� | jd tjj � dS )a L{ServerProtocol.eraseToLineBeginning} writes the control sequence sequence ending in L{CSFinalByte.EL} and a parameter of 1 (from the beginning of the line up to and include the active present position's current location.) rK N)rI r# r� r� r5 r� r< r^ r� r r r �test_eraseToLineBeginningB � $z3ServerProtocolOutputTests.test_eraseToLineBeginningc C rr )z� L{ServerProtocol.eraseLine} writes the control sequence sequence ending in L{CSFinalByte.EL} and a parameter of 2 (the entire line.) r� N)rI r$ r� r� r5 r� r< r^ r� r r r �test_eraseLineL � $z(ServerProtocolOutputTests.test_eraseLinec C rV )a L{ServerProtocol.eraseToDisplayEnd} writes the control sequence sequence ending in L{CSFinalByte.ED} and no parameters, forcing the client to default to 0 (from the active present position's current location to the end of the page.) N)rI r% r� r� r5 r� r< r[ r� r r r �test_eraseToDisplayEndU rq z0ServerProtocolOutputTests.test_eraseToDisplayEndc C rr )a L{ServerProtocol.eraseToDisplayBeginning} writes the control sequence sequence ending in L{CSFinalByte.ED} a parameter of 1 (from the beginning of the page up to and include the active present position's current location.) rK N)rI r& r� r� r5 r� r< r[ r� r r r � test_eraseToDisplayBeginning_ rt z6ServerProtocolOutputTests.test_eraseToDisplayBeginningc C rr )z� L{ServerProtocol.eraseDisplay} writes the control sequence sequence ending in L{CSFinalByte.ED} a parameter of 2 (the entire page) r� N)rI r' r� r� r5 r� r< r[ r� r r r �test_eraseToDisplayi rv z-ServerProtocolOutputTests.test_eraseToDisplayc C rJ )z� L{ServerProtocol.deleteCharacter} writes the control sequence containing the number of characters to delete and ending in L{CSFinalByte.DCH} r) � 4N)rI r( r� r� r5 r� r< r> r� r r r �test_deleteCharacterr s �z.ServerProtocolOutputTests.test_deleteCharacterc C rJ )z� L{ServerProtocol.insertLine} writes the control sequence containing the number of lines to insert and ending in L{CSFinalByte.IL} r� � 5N)rI r1 r� r� r5 r� r< ra r� r r r �test_insertLine} � $z)ServerProtocolOutputTests.test_insertLinec C rJ )z� L{ServerProtocol.deleteLine} writes the control sequence containing the number of lines to delete and ending in L{CSFinalByte.DL} r� � 6N)rI r, r� r� r5 r� r< rd r� r r r �test_deleteLine� r~ z)ServerProtocolOutputTests.test_deleteLinec C s* | j �� | �| j�� | jd d � dS )z� With no arguments, L{ServerProtocol.setScrollRegion} writes a control sequence with no parameters, but a parameter separator, and ending in C{b'r'}. r � rN�rI r� r� r� r5 r� r� r r r �test_setScrollRegionNoArgs� rY z4ServerProtocolOutputTests.test_setScrollRegionNoArgsc C �. | j jdd� | �| j�� | jd d � dS )z� With just a value for its C{first} argument, L{ServerProtocol.setScrollRegion} writes a control sequence with that parameter, a parameter separator, and finally a C{b'r'}. r� )�firsts 1;r� Nr� r� r r r � test_setScrollRegionJustFirst� � z7ServerProtocolOutputTests.test_setScrollRegionJustFirstc C r� )z� With just a value for its C{last} argument, L{ServerProtocol.setScrollRegion} writes a control sequence with a parameter separator, that parameter, and finally a C{b'r'}. r� )�lasts ;1r� Nr� r� r r r � test_setScrollRegionJustLast� r� z6ServerProtocolOutputTests.test_setScrollRegionJustLastc C s0 | j jddd� | �| j�� | jd d � dS )z� When given both C{first} and C{last} L{ServerProtocol.setScrollRegion} writes a control sequence with the first parameter, a parameter separator, the last parameter, and finally a C{b'r'}. r� r� )r� r� s 1;2r� Nr� r� r r r � test_setScrollRegionFirstAndLast� s z:ServerProtocolOutputTests.test_setScrollRegionFirstAndLastc C rr )z� L{ServerProtocol.reportCursorPosition} writes a control sequence ending in L{CSFinalByte.DSR} with a parameter of 6 (the Device Status Report returns the current active position.) r N)rI r4 r� r� r5 r� r< ri r� r r r �test_reportCursorPosition� s �z3ServerProtocolOutputTests.test_reportCursorPositionN)&r� r� r� r6 r. rZ r� r� rM rP rR rS rU rX r[ r\ r^ rb rg rh ri rj rm ro rp rs ru rw rx ry r{ r} r� r� r� r� r� r� r r r r rG y sF rG )3�textwrap�typingr r � twisted.conch.insults.insultsr r r r r r r r r r r r r �twisted.internet.protocolr �twisted.python.compatr �twisted.python.constantsr r �twisted.test.proto_helpersr � twisted.trialr r r! r% r( �objectr� r. r; �dedentr< rl r r� r� r� �TestCaser� r� r� r� r� rG r r r r �<module> s� < ���%���""# PK F��Z�� 4� 4� + __pycache__/test_knownhosts.cpython-310.pycnu �[��� o �bY� � @ sB d Z ddlZddlmZmZmZ ddlmZ ddl m Z ddl m Z m Z mZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lm Z ed�r{ed�r{ddl m Z ddl m Z m!Z!m"Z"m#Z#m$Z$ ddl%m&Z&m'Z' ddl(m)Z) ndZ*dZ+dZ,dZ-dZ.ee+�Z/ee,�Z0ee-�Z1ee.�Z2de+ d Z3de, d Z4d e+ d Z5d e+ d Z6G d d � d �Z7G d d!� d!e7e �Z8G d"d#� d#e8�Z9G d$d%� d%e7ee �Z:G d&d'� d'e:�Z;G d(d)� d)e e7�Z<G d*d+� d+e �Z=G d,d-� d-e �Z>G d.d/� d/�Z?G d0d1� d1e �Z@G d2d3� d3�ZAG d4d5� d5�ZBeed6��C� d7�G d8d9� d9e ��ZDdS ):z/ Tests for L{twisted.conch.client.knownhosts}. � N)�Error� a2b_base64� b2a_base64)�skipIf)� verifyObject)�HostKeyChanged� InvalidEntry�UserRejectedKey)�IKnownHostEntry)�Deferred)� networkString)�FilePath)� requireModule)�ComparisonTestsMixin)�TestCase� cryptography�pyasn1)�default)� ConsoleUI� HashedEntry�KnownHostsFile� PlainEntry� UnparsedEntry)� BadKeyError�Key)�keydataz>cryptography and PyASN1 required for twisted.conch.knownhosts.st AAAAB3NzaC1yc2EAAAABIwAAAQEAsV0VMRbGmzhqxxayLRHmvnFvtyNqgbNKV46dU1bVFB+3ytNvue4Riqv/SVkPRNwMb7eWH29SviXaBxUhYyzKkDoNUq3rTNnH1Vnif6d6X4JCrUb5d3W+DmYClyJrZ5HgD/hUpdSkTRqdbQ2TrvSAxRacj+vHHT4F4dm1bJSewm3B2D8HVOoi/CbVh3dsIiCdp8VltdZx4qYVfYe2LwVINCbAa3d3tj9ma7RVfw3OH2Mfb+toLd1N5tBQFb7oqTt2nC6I/6Bd4JwPUld+IEitw/suElq/AIJVQXXujeyiZlea90HE65U2mF1ytr17HTAIT2ySokJWyuBANGACk6iIaw==s� AAAAB3NzaC1yc2EAAAABIwAAAIEAwaeCZd3UCuPXhX39+/p9qO028jTF76DMVd9mPvYVDVXufWckKZauF7+0b7qm+ChT7kan6BzRVo4++gCVNfAlMzLysSt3ylmOR48tFpAfygg9UCX3DjHz0ElOOUKh3iifc9aUShD0OPaK3pR5JJ8jfiBfzSYWt/hDi/iZ4igsSs8=st AAAAB3NzaC1yc2EAAAABIwAAAQEAl/TQakPkePlnwCBRPitIVUTg6Z8VzN1en+DGkyo/evkmLw7o4NWR5qbysk9A9jXW332nxnEuAnbcCam9SHe1su1liVfyIK0+3bdn0YRB0sXIbNEtMs2LtCho/aV3cXPS+Cf1yut3wvIpaRnAzXxuKPCTXQ7/y0IXa8TwkRBH58OJa3RqfQ/NsSp5SAfdsrHyH2aitiVKm2jfbTKzSEqOQG/zq4J9GXTkq61gZugory/Tvl5/yPgSnOR6C9jVOMHf27ZPoRtyj9SY343Hd2QHiIE0KPZJEgCynKeWoKz8v6eTSK8n4rBnaqWdp8MnGZK1WGy05MguXbyCDuTC8AmJXQ==s� AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBIFwh3/zBANyPPIE60SMMfdKMYo3OvfvzGLZphzuKrzSt0q4uF+/iYqtYiHhryAwU/fDWlUQ9kck9f+IlpsNtY4=s www.twistedmatrix.com ssh-rsa � s divmod.com ssh-rsa s- www.twistedmatrix.com,198.49.126.131 ssh-rsa sE |1|gJbSEPBG9ZSBoZpHNtZBD1bHKBA=|bQv+0Xa0dByrwkA1EB0E7Xop/Fo= ssh-rsa c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �EntryTestsMixina< Tests for implementations of L{IKnownHostEntry}. Subclasses must set the 'entry' attribute to a provider of that interface, the implementation of that interface under test. @ivar entry: a provider of L{IKnownHostEntry} with a hostname of www.twistedmatrix.com and an RSA key of sampleKey. c C s t t| j� dS )zA The given entry should provide IKnownHostEntry. N)r r �entry��self� r" �D/usr/lib/python3/dist-packages/twisted/conch/test/test_knownhosts.py�test_providesInterfaceb � z&EntryTestsMixin.test_providesInterfacec C s, | j }| �|jt�t�� | �|jd� dS )a� Constructing a plain text entry from an unhashed known_hosts entry will result in an L{IKnownHostEntry} provider with 'keyString', 'hostname', and 'keyType' attributes. While outside the interface in question, these attributes are held in common by L{PlainEntry} and L{HashedEntry} implementations; other implementations should override this method in subclasses. � ssh-rsaN)r � assertEqual� publicKeyr � fromString� sampleKey�keyType)r! r r" r" r# �test_fromStringh s z EntryTestsMixin.test_fromStringc C s@ t �t�}t �t�}| �d| j�|�� | �d| j�|�� dS )zj L{IKnownHostEntry.matchesKey} checks to see if an entry matches a given SSH key. TFN)r r) r* �otherSampleKeyr' r � matchesKey)r! �twistedmatrixDotCom� divmodDotComr" r" r# �test_matchesKeyu s z EntryTestsMixin.test_matchesKeyc C s( | � | j�d�� | �| j�d�� dS )zl L{IKnownHostEntry.matchesHost} checks to see if an entry matches a given hostname. � www.twistedmatrix.coms www.divmod.comN)� assertTruer � matchesHost� assertFalser r" r" r# �test_matchesHost s z EntryTestsMixin.test_matchesHostN)�__name__� __module__� __qualname__�__doc__r$ r, r1 r6 r" r" r" r# r X s r c @ s0 e Zd ZdZeZeZdd� Zdd� Z dd� Z dS ) �PlainEntryTestsz' Test cases for L{PlainEntry}. c C � t �| j�| _dS )zU Set 'entry' to a sample plain-text entry with sampleKey as its key. N)r r) � plaintextLiner r r" r" r# �setUp� s zPlainEntryTests.setUpc C s, t �| j�| _| �| j�d�� | �� dS )zU A "hostname,ip" formatted line will match both the host and the IP. s 198.49.126.131N)r r) � hostIPLiner r3 r4 r6 r r" r" r# �test_matchesHostIP� s z"PlainEntryTests.test_matchesHostIPc C sB | � | j�� | j�d�� t�| j�}| � |�� | j�d�� dS )z| L{PlainEntry.toString} generates the serialized OpenSSL format string for the entry, sans newline. r N)r' r �toStringr= �rstripr r) r? )r! �multiHostEntryr" r" r# � test_toString� s z PlainEntryTests.test_toStringN) r7 r8 r9 r: �samplePlaintextLiner= �sampleHostIPLiner? r> r@ rD r" r" r" r# r; � s r; c @ s0 e Zd ZdZedd� d Zedd� d ZdS )�PlainTextWithCommentTests�N Test cases for L{PlainEntry} when parsed from a line with a comment. N���� plain text comment. s text following host/IP line )r7 r8 r9 r: rE r= rF r? r" r" r" r# rG � s rG c @ s, e Zd ZdZeZdd� Zdd� Zdd� ZdS ) �HashedEntryTestsaF Tests for L{HashedEntry}. This suite doesn't include any tests for host/IP pairs because hashed entries store IP addresses the same way as hostnames and does not support comma-separated lists. (If you hash the IP and host together you can't tell if you've got the key already for one or the other.) c C r< )zo Set 'entry' to a sample hashed entry for twistedmatrix.com with sampleKey as its key. N)r r) � hashedLiner r r" r" r# r> � s zHashedEntryTests.setUpc C s | � | j�� | j�d�� dS )z� L{HashedEntry.toString} generates the serialized OpenSSL format string for the entry, sans the newline. r N)r' r rA rL rB r r" r" r# rD � s z HashedEntryTests.test_toStringc C s� d}d}t �t�}t|�� �}d}t|||||�}t|||||�}| �||t|ddd� ||||�� | �||t||ddd� |||�� | �||t|||ddd� ||�� | �||t|||t �t�|�� | �||t|||||ddd� �� dS )z� Two L{HashedEntry} instances compare equal if and only if they represent the same host and key in exactly the same way: the host salt, host hash, public key type, public key, and comment fields must all be equal. s gJbSEPBG9ZSBoZpHNtZBD1bHKBAs bQv+0Xa0dByrwkA1EB0E7Xop/Fos hello, worldNrI )r r) r* r �typer �"assertNormalEqualityImplementationr- )r! �hostSalt�hostHashr( r+ �commentr � duplicater" r" r# � test_equality� sD ����� �z HashedEntryTests.test_equalityN) r7 r8 r9 r: �sampleHashedLinerL r> rD rS r" r" r" r# rK � s rK c @ s e Zd ZdZedd� d ZdS )�HashedEntryWithCommentTestsrH NrI rJ )r7 r8 r9 r: rT rL r" r" r" r# rU s rU c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �UnparsedEntryTestsz$ Tests for L{UnparsedEntry} c C s t d�| _dS )zR Set up the 'entry' to be an unparsed entry for some random text. � This is a bogus entry. N)r r r r" r" r# r> s zUnparsedEntryTests.setUpc C s | � d| jj� dS )ze Creating an L{UnparsedEntry} should simply record the string it was passed. rW N)r' r �_stringr r" r" r# r, � z"UnparsedEntryTests.test_fromStringc C s | � | j�d�� dS )z: An unparsed entry can't match any hosts. r2 N)r5 r r4 r r" r" r# r6 s z#UnparsedEntryTests.test_matchesHostc C s | � | j�t�t��� dS )z9 An unparsed entry can't match any keys. N)r5 r r. r r) r* r r" r" r# r1 s z"UnparsedEntryTests.test_matchesKeyc C s | � d| j�� � dS )zd L{UnparsedEntry.toString} returns its input string, sans trailing newline. s This is a bogus entry. N)r' r rA r r" r" r# rD & s z UnparsedEntryTests.test_toStringN) r7 r8 r9 r: r> r, r6 r1 rD r" r" r" r# rV s rV c @ �X e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )�ParseErrorTestsa� L{HashedEntry.fromString} and L{PlainEntry.fromString} can raise a variety of errors depending on misformattings of certain strings. These tests make sure those errors are caught. Since many of the ways that this can go wrong are in the lower-level APIs being invoked by the parsing logic, several of these are integration tests with the C{base64} and L{twisted.conch.ssh.keys} modules. c C � | � t|jd� dS )zm If there are fewer than three elements, C{fromString} should raise L{InvalidEntry}. s invalidN)� assertRaisesr r) �r! �clsr" r" r# �invalidEntryTest8 rY z ParseErrorTests.invalidEntryTestc C r\ )zX If the key is not base64, C{fromString} should raise L{BinasciiError}. s x x xN)r] � BinasciiErrorr) r^ r" r" r# � notBase64Test? s z ParseErrorTests.notBase64Testc C s( | � t|jd�|dtd��� g�� dS )z� If the key portion of the entry is valid base64, but is not actually an SSH key, C{fromString} should raise L{BadKeyError}. � r& s Hey, this isn't an SSH key!N)r] r r) �joinr �strip)r! r_ �prefixr" r" r# � badKeyTestE s ��zParseErrorTests.badKeyTestc C � | � t� dS )z� If there are fewer than three whitespace-separated elements in an entry, L{PlainEntry.fromString} should raise L{InvalidEntry}. N)r` r r r" r" r# �test_invalidPlainEntryR � z&ParseErrorTests.test_invalidPlainEntryc C s< | � t� t�� \}}}| �ttjd�|d ||g�� dS )z� If there are fewer than three whitespace-separated elements in an entry, or the hostname salt/hash portion has more than two elements, L{HashedEntry.fromString} should raise L{InvalidEntry}. rc s ||N)r` r rT �splitr] r r) rd �r! �a�b�cr" r" r# �test_invalidHashedEntryY s �z'ParseErrorTests.test_invalidHashedEntryc C rh )z� If the key portion of a plain entry is not decodable as base64, C{fromString} should raise L{BinasciiError}. N)rb r r r" r" r# �test_plainNotBase64e rj z#ParseErrorTests.test_plainNotBase64c C s� | � t� t�� \}}}| �ttjd�dtd�� � ||g�� | �ttjd�tj td�� � d ||g�� | �ttjd�d||g�� dS )z� If the key, host salt, or host hash portion of a hashed entry is not encoded, it will raise L{BinasciiError}. rc s |1|x|� stuffs |xs |1|x|xN) rb r rT rk r] ra r) rd r re �MAGICrl r" r" r# �test_hashedNotBase64l s � ��z$ParseErrorTests.test_hashedNotBase64c C s t �� \}}}| �t|� dS )z� If the key portion of the entry is valid base64, but is not actually an SSH key, C{HashedEntry.fromString} should raise L{BadKeyError}. N)rT rk rg r rl r" r" r# �test_hashedBadKey� s z!ParseErrorTests.test_hashedBadKeyc C s | � td� dS )z� If the key portion of the entry is valid base64, but is not actually an SSH key, C{PlainEntry.fromString} should raise L{BadKeyError}. s hostnameN)rg r r r" r" r# �test_plainBadKey� s z ParseErrorTests.test_plainBadKeyN) r7 r8 r9 r: r` rb rg ri rp rq rt ru rv r" r" r" r# r[ . s r[ c @ s< e Zd ZdZdd� Zee d fdd�Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z!d;d<� Z"d=d>� Z#d?d@� Z$dAdB� Z%dCdD� Z&dEdF� Z'dGdH� Z(dIdJ� Z)dKS )L�KnownHostsDatabaseTestsz& Tests for L{KnownHostsFile}. c C s t | �� �}|�|� |S )zC Return a FilePath with the given initial content. )r �mktemp� setContent)r! �content�fpr" r" r# �pathWithContent� s z'KnownHostsDatabaseTests.pathWithContentsN # That was a blank line. This is just unparseable. |1|This also unparseable. c C s t �| �|��S )zq Return a sample hosts file, with keys for www.twistedmatrix.com and divmod.com present. )r �fromPathr| )r! rz r" r" r# �loadSampleHostsFile� s z+KnownHostsDatabaseTests.loadSampleHostsFilec C sD t | �� �}t | �� �}t|�}| �tt|d|� | �||j� dS )z� L{KnownHostsFile.savePath} is read-only; if an assignment is made to it, L{AttributeError} is raised and the value is unchanged. �savePathN)r rx r r] �AttributeError�setattrr' r )r! �path�new� hostsFiler" r" r# �test_readOnlySavePath� s z-KnownHostsDatabaseTests.test_readOnlySavePathc C s& t | �t��}| �g t|�� �� dS )zz The default initializer for L{KnownHostsFile} disregards any existing contents in the save path. N)r r| rT r' �list� iterentries�r! r� r" r" r# �&test_defaultInitializerIgnoresExisting� s z>KnownHostsDatabaseTests.test_defaultInitializerIgnoresExistingc C s^ | � t�}t|�}|�dt�t��}|�� | �|gt |� � �� | �|� � d |� � � dS )z� After using the default initializer for L{KnownHostsFile}, the first use of L{KnownHostsFile.save} overwrites any existing contents in the save path. � www.example.comr N) r| rT r � addHostKeyr r) r- �saver' r� r� rA � getContent)r! r� r� r r" r" r# �'test_defaultInitializerClobbersExisting� s z?KnownHostsDatabaseTests.test_defaultInitializerClobbersExistingc C s^ t | �t��}|�dt�t��}|�� |�dt�t��}|�� | � ||gt |� � �� dS )z� After L{KnownHostsFile.save} is used once with an instance initialized by the default initializer, contents of the save path are respected and preserved. r� s another.example.comN) r r| rT r� r r) r- r� �thirdSampleKeyr' r� r� )r! r� �preSave�postSaver" r" r# �test_saveResetsClobberState� s � � z3KnownHostsDatabaseTests.test_saveResetsClobberStatec C s$ | � � }| �dtt|�� ��� dS )z� Loading a L{KnownHostsFile} from a path with six entries in it will result in a L{KnownHostsFile} object with six L{IKnownHostEntry} providers in it. � N)r~ r' �lenr� r� r� r" r" r# �test_loadFromPath� s z)KnownHostsDatabaseTests.test_loadFromPathc C s> t t| �� ��}|�dt�t�� | �dtt |� � ��� dS )z� If the save path for a L{KnownHostsFile} does not exist, L{KnownHostsFile.iterentries} still returns added but unsaved entries. r� � N) r r rx r� r r) r* r' r� r� r� r� r" r" r# �test_iterentriesUnsaved� s z/KnownHostsDatabaseTests.test_iterentriesUnsavedc C �N | � t�}t|�� �}| �|d t� | �|d �d�� | �dt |�� dS )z� Loading a L{KnownHostsFile} from a path containing a single valid L{HashedEntry} entry will result in a L{KnownHostsFile} object with one L{IKnownHostEntry} provider. r r2 r� N) r~ rT r� r� �assertIsInstancer r3 r4 r' r� �r! r� �entriesr" r" r# �test_verifyHashedEntry� � z.KnownHostsDatabaseTests.test_verifyHashedEntryc C r� )z� Loading a L{KnownHostsFile} from a path containing a single valid L{PlainEntry} entry will result in a L{KnownHostsFile} object with one L{IKnownHostEntry} provider. r � divmod.comr� N) r~ �otherSamplePlaintextLiner� r� r� r r3 r4 r' r� r� r" r" r# �test_verifyPlainEntry r� z-KnownHostsDatabaseTests.test_verifyPlainEntryc C �N | � d�}t|�� �}| �|d t� | �|d �� d� | �dt|�� dS )z� Loading a L{KnownHostsFile} from a path that only contains ' ' will result in a L{KnownHostsFile} object containing a L{UnparsedEntry} object. r r � r� N�r~ r� r� r� r r' rA r� r� r" r" r# �test_verifyUnparsedEntry r� z0KnownHostsDatabaseTests.test_verifyUnparsedEntryc C s> | � d�}t|�� �}| �|d t� | �|d �� d� dS )z� Loading a L{KnownHostsFile} from a path that contains a comment will result in a L{KnownHostsFile} object containing a L{UnparsedEntry} object. s # That was a blank line. r s # That was a blank line.N)r~ r� r� r� r r' rA r� r" r" r# �test_verifyUnparsedComment s z2KnownHostsDatabaseTests.test_verifyUnparsedCommentc C r� )z� Loading a L{KnownHostsFile} from a path that contains an unparseable line will be represented as an L{UnparsedEntry} instance. s This is just unparseable. r s This is just unparseable.r� Nr� r� r" r" r# �test_verifyUnparsableLine& s z1KnownHostsDatabaseTests.test_verifyUnparsableLinec C r� )z� Loading a L{KnownHostsFile} from a path containing an unparseable line that starts with an encryption marker will be represented as an L{UnparsedEntry} instance. s |1|This is unparseable. r s |1|This is unparseable.r� Nr� r� r" r" r# �%test_verifyUnparsableEncryptionMarker1 r� z=KnownHostsDatabaseTests.test_verifyUnparsableEncryptionMarkerc C s^ | � � }t�t|��}t|�� �}| �g |� | �t|��� � |� � | � t|��� � dS )z� Loading a L{KnownHostsFile} from a path that does not exist should result in an empty L{KnownHostsFile} that will save back to that path. N) rx r r} r r� r� r' r5 �existsr� r3 )r! �pn�knownHostsFiler� r" r" r# �test_loadNonExistent= s z,KnownHostsDatabaseTests.test_loadNonExistentc C sJ t | �� �}|�d��d�}t�|�}|�� |�d� | �|�� � dS )z� Loading a L{KnownHostsFile} from a path whose parent directory does not exist should result in an empty L{KnownHostsFile} that will save back to that path, creating its parent directory(ies) in the process. �foos known_hostsFN) r rx �childr r} r� �restatr3 r� )r! �thePath�knownHostsPathr� r" r" r# �test_loadNonExistentParentJ s z2KnownHostsDatabaseTests.test_loadNonExistentParentc C s� | � tt �}t�|�}|�dt�t��}tt t j t |j �� � d t |j�� � d t d }| �d|�d�� |�� | �||�� � dS )zq L{KnownHostsFile.save} will write out a new file with any entries that have been added. � some.example.com� |s ssh-rsa r � N)r| rT r� r r} r� r r) r� r rs r � _hostSaltre � _hostHash�thirdSampleEncodedKeyr' �countr� r� )r! r� r� �newEntry�expectedContentr" r" r# �test_savingAddsEntryW s4 ��� �� �����z,KnownHostsDatabaseTests.test_savingAddsEntryc C sZ t | �� �}t|�}|�dt�t��}|�� |�� t�|�}| � |gt |� � �� dS )z� L{KnownHostsFile.save} only writes new entries to the save path, not entries which were added and already written by a previous call to C{save}. r� N) r rx r r� r r) r* r� r} r' r� r� )r! r� � knownHostsr r" r" r# � test_savingAvoidsDuplications s z4KnownHostsDatabaseTests.test_savingAvoidsDuplicationc C s� | � t�}t�|�}|�d�� }|�t� W d � n1 s w Y t�t �}|� d|� |� � t�|�}| � g d�|� dt�t��|� dt�t��|� d|�g� dS )z� L{KnownHostsFile.save} will not overwrite existing entries in its save path, even if they were only added after the L{KnownHostsFile} instance was initialized. rm N� brandnew.example.com)TTTr2 r� )r| rT r r} �open�writer� r r) r� r� r� r' � hasHostKeyr* r- )r! r� r� � hostsFileObj�keyr" r" r# � test_savingsPreservesExisting� s$ � � ��z5KnownHostsDatabaseTests.test_savingsPreservesExistingc C s$ | � � }| �|�dt�t��� dS )z� L{KnownHostsFile.hasHostKey} returns C{True} when a key for the given hostname is present and matches the expected key. r2 N)r~ r3 r� r r) r* r� r" r" r# �test_hasPresentKey� s �z*KnownHostsDatabaseTests.test_hasPresentKeyc C sT | � � }| �|�dt�t��� | �|�dt�t��� | �|�dt�t��� dS )zy L{KnownHostsFile.hasHostKey} returns C{False} when a key for the given hostname is not present. s non-existent.example.comr2 N)r~ r5 r� r r) r* r3 �ecdsaSampleKeyr� r" r" r# �test_notPresentKey� s �� ��z*KnownHostsDatabaseTests.test_notPresentKeyc C s| t �t�}tdg|�� |d�}| �� }|j�d��}|�|� � d � W d � n1 s-w Y | � d|� d|�� dS )z� L{KnownHostsFile.hasHostKey} returns C{True} when a key for the given hostname is present in the file, even if it is only added to the file after the L{KnownHostsFile} instance is initialized. r� r� rm r NT) r r) r* r �sshTyper~ r r� r� rA r' r� )r! r� r r� r� r" r" r# �test_hasLaterAddedKey� s �z-KnownHostsDatabaseTests.test_hasLaterAddedKeyc C s` | � � }t|�� �}| �t|jdt�t��}| � |j |d � | � |j d� | � |j |j � dS )a L{KnownHostsFile.hasHostKey} raises L{HostKeyChanged} if the host key is present in the underlying file, but different from the expected one. The resulting exception should have an C{offendingEntry} indicating the given entry. r2 r r� N)r~ r� r� r] r r� r r) r- r' �offendingEntry�linenor� r )r! r� r� � exceptionr" r" r# � test_savedEntryHasKeyMismatch� s �z5KnownHostsDatabaseTests.test_savedEntryHasKeyMismatchc C sT | � � }|�dt�t�� | �t|jdt�t��}| �|j d� | �|j |j � dS )ai Even after a new entry has been added in memory but not yet saved, the L{HostKeyChanged} exception raised by L{KnownHostsFile.hasHostKey} has a C{lineno} attribute which indicates the 1-based line number of the offending entry in the underlying file when the given host key does not match the expected host key. r� r2 r� N) r~ r� r r) r- r] r r� r' r� r� r )r! r� r� r" r" r# �%test_savedEntryAfterAddHasKeyMismatch� s �z=KnownHostsDatabaseTests.test_savedEntryAfterAddHasKeyMismatchc C sd t t| �� ��}|�dt�t��}| �t|j dt�t ��}| � |j |� | � |j� | � |j� dS )az L{KnownHostsFile.hasHostKey} raises L{HostKeyChanged} if the host key is present in memory (but not yet saved), but different from the expected one. The resulting exception has a C{offendingEntry} indicating the given entry, but no filename or line number information (reflecting the fact that the entry exists only in memory). r� N)r r rx r� r r) r- r] r r� r� r' r� � assertIsNoner� r� )r! r� r r� r" r" r# � test_unsavedEntryHasKeyMismatch� s � z7KnownHostsDatabaseTests.test_unsavedEntryHasKeyMismatchc C s� | � � }t�t�}| �d|�d|�� |�d|�}| �dt|j�� | �d|� d�� | �|j d� | �||j � | �d|�d|�� dS )zr L{KnownHostsFile.addHostKey} adds a new L{HashedEntry} to the host file, and returns it. F� somewhere.example.com� Tr& N) r~ r r) r� r' r� r� r� r� r4 r+ r( )r! r� �aKeyr� r" r" r# �test_addHostKey s z'KnownHostsDatabaseTests.test_addHostKeyc C s6 | � � }t�t�}| �|�d|�j|�d|�j� dS )z� L{KnownHostsFile.addHostKey} generates a random salt for each new key, so subsequent salts will be different. r� s somewhere-else.example.comN)r~ r r) r� �assertNotEqualr� r� )r! r� r� r" r" r# �test_randomSalts s �z(KnownHostsDatabaseTests.test_randomSaltsc C sX | � � }|�dt�t�� t� }|�|ddt�t��}g }|�|j� | � |dg� dS )zb Verifying a valid key should return a L{Deferred} which fires with True. � 1.2.3.4r2 TN) r~ r� r r) r* �FakeUI� verifyHostKey� addCallback�appendr' )r! r� �ui�d�lr" r" r# �test_verifyValidKey% s � z+KnownHostsDatabaseTests.test_verifyValidKeyc C sF | � � }t�t�}t� }|�dt�t�� |�|dd|�}| �|t �S )z| Verifying an invalid key should return a L{Deferred} which fires with a L{HostKeyChanged} failure. r� r2 ) r~ r r) r� r� r� r* r� � assertFailurer �r! r� �wrongKeyr� r� r" r" r# �test_verifyInvalidKey4 s z-KnownHostsDatabaseTests.test_verifyInvalidKeyc C s\ | � � }t�t�}t� }g }|�|dd|�}|�|j� | �g |� | �|j d� |||fS )a Set up a test to verify a key that isn't present. Return a 3-tuple of the UI, a list set up to collect the result of the verifyHostKey call, and the sample L{KnownHostsFile} being used. This utility method avoids returning a L{Deferred}, and records results in the returned list instead, because the events which get generated here are pre-recorded in the 'ui' object. If the L{Deferred} in question does not fire, the it will fail quickly with an empty list. � sample-host.example.com� 4.3.2.1s� The authenticity of host 'sample-host.example.com (4.3.2.1)' can't be established. RSA key fingerprint is SHA256:mS7mDBGhewdzJkaKRkx+wMjUdZb/GzvgcdoYjX5Js9I=. Are you sure you want to continue connecting (yes/no)? ) r~ r r) r� r� r� �addBothr� r' � promptText)r! r� � absentKeyr� r� r� r" r" r# �verifyNonPresentKey@ s � � z+KnownHostsDatabaseTests.verifyNonPresentKeyc C sl | � � \}}}|j�d� | �dg|� t�|j�}| �d|�dt� t ��� | �d|�dt� t ��� dS )z� Verifying a key where neither the hostname nor the IP are present should result in the UI being prompted with a message explaining as much. If the UI says yes, the Deferred should fire with True. Tr� r� N) r� �promptDeferred�callbackr' r r} r r� r r) r� )r! r� r� r� �reloadedr" r" r# � test_verifyNonPresentKey_Yes^ s � ��z4KnownHostsDatabaseTests.test_verifyNonPresentKey_Yesc C s, | � � \}}}|j�d� |d �t� dS )z� Verifying a key where neither the hostname nor the IP are present should result in the UI being prompted with a message explaining as much. If the UI says no, the Deferred should fail with UserRejectedKey. Fr N)r� r� r� �trapr )r! r� r� r� r" r" r# �test_verifyNonPresentKey_Nor s z3KnownHostsDatabaseTests.test_verifyNonPresentKey_Noc C sv t jtjd tjd tjd tjd d�}| �� }t� }g }|�|dd|�}|�|j� | � g |� | � |j d� d S ) z� Set up a test to verify an ECDSA key that isn't present. Return a 3-tuple of the UI, a list set up to collect the result of the verifyHostKey call, and the sample L{KnownHostsFile} being used. �x�y� privateValue�curve)r� r� r� r� r� r� s� The authenticity of host 'sample-host.example.com (4.3.2.1)' can't be established. ECDSA key fingerprint is SHA256:fJnSpgCcYoYYsaBbnWj1YBghGh/QTDgfe4w4U5M5tEo=. Are you sure you want to continue connecting (yes/no)? N) r �_fromECComponentsr �ECDatanistp256r~ r� r� r� r� r' r� )r! �ecObjr� r� r� r� r" r" r# �test_verifyNonPresentECKey} s � �z2KnownHostsDatabaseTests.test_verifyNonPresentECKeyc C s4 | � � }t�t�}t� }|�|dd|�}| �|t�S )z� Verifying a key where the host is present (and correct), but the IP is present and different, should result the deferred firing in a HostKeyChanged failure. r2 r� )r~ r r) r� r� r� r� r r� r" r" r# �test_verifyHostIPMismatch� s z1KnownHostsDatabaseTests.test_verifyHostIPMismatchc C sX t � }| �� }t�t�}|�|dd|� | �dt�|j �� d|�� | �dg|j � dS )z� Verifying a key where the hostname is present but the IP is not should result in the key being added for the IP and the user being warned about the change. r2 s 5.4.3.2Tz`Warning: Permanently added the RSA host key for IP address '5.4.3.2' to the list of known hosts.N) r� r~ r r) r* r� r' r r} r r� � userWarnings)r! r� r� � expectedKeyr" r" r# �test_verifyKeyForHostAndIP� s ����z2KnownHostsDatabaseTests.test_verifyKeyForHostAndIPc C sh | � � }|�dt�t�� |�dt�t�� |�� i }|jj|d< t � d|�}ddg}| � ||� dS )zn For a given host, get the host key algorithms for that host in the known_hosts file. r2 � known-hostsr& s ecdsa-sha2-nistp256N) r~ r� r r) r- r� r� r r� r �getHostKeyAlgorithmsr' )r! r� �options� algorithms�expectedAlgorithmsr" r" r# �test_getHostKeyAlgorithms� s z1KnownHostsDatabaseTests.test_getHostKeyAlgorithmsN)*r7 r8 r9 r: r| rT r� r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r" r" r" r# rw � sV �� � ! rw c @ r ) �FakeFilez[ A fake file-like object that acts enough like a file for L{ConsoleUI.prompt}. c C s g | _ g | _d| _d S )NF)�inlines� outchunks�closedr r" r" r# �__init__� � zFakeFile.__init__c C s | j �d�S )z8 Return a line from the 'inlines' list. r )r �popr r" r" r# �readline� s zFakeFile.readlinec C s | j rtd��| j�|� dS )z@ Append the given item to the 'outchunks' list. zthe file was closedN)r �OSErrorr r� )r! �chunkr" r" r# r� � s zFakeFile.writec C s d| _ dS )zd Set the 'closed' flag to True, explicitly marking that it has been closed. TN)r r r" r" r# �close� s zFakeFile.closeN)r7 r8 r9 r: r r r� r r" r" r" r# r � s r c @ rZ )�ConsoleUITestsz& Test cases for L{ConsoleUI}. c C s t � | _t| j�| _dS )zA Create a L{ConsoleUI} pointed at a L{FakeFile}. N)r �fakeFiler �openFiler� r r" r" r# r> � s zConsoleUITests.setUpc C s | j S )z/ Return the current fake file. )r r r" r" r# r � s zConsoleUITests.openFilec C s t � | _|| j_dS )z� Create a new fake file (the next file that self.ui will open) with the given list of lines to be returned from readline(). N)r r r )r! �linesr" r" r# �newFile s zConsoleUITests.newFilec C �` dD ]+}| � |g� g }| j�d��|j� | �dg| jj� | �dg|� | �| jj � qdS )z� L{ConsoleUI.prompt} writes a message to the console, then reads a line. If that line is 'yes', then it returns a L{Deferred} that fires with True. )� yess Yess yes z Hello, world!TN� r r� �promptr� r� r' r r r3 r )r! �okYesr� r" r" r# �test_promptYes � �z ConsoleUITests.test_promptYesc C r )z� L{ConsoleUI.prompt} writes a message to the console, then reads a line. If that line is 'no', then it returns a L{Deferred} that fires with False. )� nos Nos no zGoodbye, world!FNr )r! �okNor� r" r" r# � test_promptNo r z ConsoleUITests.test_promptNoc C s� | � g d�� g }| j�d��|j� | �dg|� | �| jjdgdgd � | �| jj � | � g d�� g }| j�d��|j� | �dg|� | �| jjdgdgd � | �| jj � d S ) a, L{ConsoleUI.prompt} writes a message to the console, then reads a line. If that line is neither 'yes' nor 'no', then it says "Please enter 'yes' or 'no'" until it gets a 'yes' or a 'no', at which point it returns a Deferred that answers either True or False. )s whats uhs okayr s Please say something useful.Ts Please type 'yes' or 'no': r� )s blahrr s fehr s Please say something negative.FNr �r! r� r" r" r# �test_promptRepeatedly$ s$ ��z$ConsoleUITests.test_promptRepeatedlyc C s& dd� }t |�}|�d�}| �|t�S )z� If the C{opener} passed to L{ConsoleUI} raises an exception, that exception will fail the L{Deferred} returned from L{ConsoleUI.prompt}. c S s t � ��N)r r" r" r" r# �raiseItD s z5ConsoleUITests.test_promptOpenFailed.<locals>.raiseItzThis is a test.)r r r� �IOError)r! r r� r� r" r" r# �test_promptOpenFailed> s z$ConsoleUITests.test_promptOpenFailedc C s0 | j �d� | �dg| jj� | �| jj� dS )zR L{ConsoleUI.warn} should output a message to the console object. z Test message.N)r� �warnr' r r r3 r r r" r" r# � test_warnK s zConsoleUITests.test_warnc C s4 dd� }t |�}|�d� | �t| �t��d� dS )zY L{ConsoleUI.warn} should log a traceback if the output can't be opened. c S s dd d S )Nr� r r" r" r" r" r# r X s z3ConsoleUITests.test_warnOpenFailed.<locals>.raiseItz This message never makes it.r� N)r r" r' r� �flushLoggedErrors�ZeroDivisionError)r! r r� r" r" r# �test_warnOpenFailedS s z"ConsoleUITests.test_warnOpenFailedN) r7 r8 r9 r: r> r r r r r r! r# r& r" r" r" r# r � s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r� a A fake UI object, adhering to the interface expected by L{KnownHostsFile.verifyHostKey} @ivar userWarnings: inputs provided to 'warn'. @ivar promptDeferred: last result returned from 'prompt'. @ivar promptText: the last input provided to 'prompt'. c C s g | _ d | _d | _d S r )r� r� r� r r" r" r# r l r zFakeUI.__init__c C s || _ t� | _| jS )zV Issue the user an interactive prompt, which they can accept or deny. )r� r r� �r! �textr" r" r# r q s z FakeUI.promptc C s | j �|� dS )z> Issue a non-interactive warning to the user. N)r� r� r' r" r" r# r" y r% z FakeUI.warnN)r7 r8 r9 r: r r r" r" r" r" r# r� ` s r� c @ s e Zd ZdZdS )� FakeObjectzw A fake object that can have some attributes. Used to fake L{SSHClientTransport} and L{SSHClientFactory}. N)r7 r8 r9 r: r" r" r" r# r) � s r) �/dev/ttyzPlatform lacks /dev/ttyc @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )�DefaultAPITestsz� The API in L{twisted.conch.client.default.verifyHostKey} is the integration point between the code in the rest of conch and L{KnownHostsFile}. c K s. | � |d� | � |d� | � |d d� | jS )z The patched version of 'open'; this returns a L{FakeFile} that the instantiated L{ConsoleUI} can use. r* zr+b� bufferingr )r' r )r! �fname�mode�kwargsr" r" r# � patchedOpen� s zDefaultAPITests.patchedOpenc C s� t � | _| �td| j� | �� | _i | _tt | j��}dD ]}|� |t � t ��}|| j|< q |�� t� | _t� | j_d| jd� | _| jj_dS )z0 Patch 'open' in verifyHostKey. �_open)� exists.example.comr� r2 )�hostr� N)r r �patchr r0 rx � hostsOption� hashedEntriesr r r� r r) r* r� r) � fakeTransport�factoryr� )r! r� r3 r r" r" r# r> � s �zDefaultAPITests.setUpc C s0 g }t �| jdtd��|j� | �dg|� dS )z� L{default.verifyHostKey} should return a L{Deferred} which fires with C{1} when passed a host, IP, and key which already match the known_hosts file it is supposed to check. r� � I don't care.r� N)r r� r7 r* r� r� r' r r" r" r# �test_verifyOKKey� s � z DefaultAPITests.test_verifyOKKeyc s0 t j�d�� � fdd�}| �|� |t jd< dS )aC Replace the HOME environment variable until the end of the current test, with the given new home-directory, so that L{os.path.expanduser} will yield controllable, predictable results. @param tempHome: the pathname to replace the HOME variable with. @type tempHome: L{str} �HOMEc s" � d u r t jd= d S � t jd< d S )Nr; )�os�environr" ��oldHomer" r# � cleanupHome� s z0DefaultAPITests.replaceHome.<locals>.cleanupHomeN)r< r= �get� addCleanup)r! �tempHomer@ r" r> r# � replaceHome� s zDefaultAPITests.replaceHomec C s� g }| � � }| j}t|��d��d�}|�� �� t|��|� | �|� d| jd< t � | j dt d�� |j� | �dg|� dS )z� L{default.verifyHostKey} should find your known_hosts file in ~/.ssh/known_hosts if you don't specify one explicitly on the command line. z.ssh� known_hostsNr� r� r9 r� )rx r5 r r� �parent�makedirs�moveTorD r� r r� r7 r* r� r� r' )r! r� �tmpdir�oldHostsOption�hostsNonOptionr" r" r# �test_noKnownHostsOption� s � z'DefaultAPITests.test_noKnownHostsOptionc C sj g }t �| jdtd��|j� | �dg| jj� | �dg|� t � t | j ��}| � |�dt�t��� dS )a L{default.verifyHostKey} should return a L{Deferred} which fires with C{1} when passed a host which matches with an IP is not present in its known_hosts file, and should also warn the user that it has added the IP address. s 8.7.6.5s Fingerprint not required.z`Warning: Permanently added the RSA host key for IP address '8.7.6.5' to the list of known hosts.r� N)r r� r7 r* r� r� r' r r r r} r r5 r3 r� r r) )r! r� r� r" r" r# �test_verifyHostButNotIP� s � ���z'DefaultAPITests.test_verifyHostButNotIPc C sL d| j jjd< | jj�d� t�| j dtd�}| � dg| jj � | � |t �S )z� L{default.verifyHostKey} should return a L{Default} which fires with C{0} when passed an unknown host that the user refuses to acknowledge. s fake.example.comr3 r s 9.8.7.6s No fingerprint!s� The authenticity of host 'fake.example.com (9.8.7.6)' can't be established. RSA key fingerprint is SHA256:vD0YydsNIUYJa7yLZl3tIL8h0vZvQ8G+HPG7JLmQV0s=. Are you sure you want to continue connecting (yes/no)? ) r7 r8 r� r r r� r r� r- r' r r� r �r! r� r" r" r# �test_verifyQuestion� s ��� z#DefaultAPITests.test_verifyQuestionc C s t �| jdtd�}| �|t�S )z� L{default.verifyHostKey} should return a L{Deferred} which fails with L{HostKeyChanged} if the host key is incorrect. r� zAgain, not required.)r r� r7 r- r� r rN r" r" r# �test_verifyBadKey s � z!DefaultAPITests.test_verifyBadKeyc C �J | j d �� �� d }t�|t�t��� dt | j �j i�}| � d|� dS )zy L{default.isInKnownHosts} should return C{1} when a host with a key is in the known hosts file. r� r r� r� N) r6 rA rk r �isInKnownHostsr r) r* �blobr r5 r� r' �r! r3 �rr" r" r# �test_inKnownHosts � �z!DefaultAPITests.test_inKnownHostsc C s* t �dddt| j�ji�}| �d|� dS )z} L{default.isInKnownHosts} should return C{0} when a host with a key is not in the known hosts file. z not.theres irrelevantr� r N)r rR r r5 r� r' )r! rU r" r" r# �test_notInKnownHosts% s �z$DefaultAPITests.test_notInKnownHostsc C rQ )z� L{default.isInKnownHosts} should return C{2} when a host with a key other than the given one is in the known hosts file. r� r r� � N) r6 rA rk r rR r r) r- rS r r5 r� r' rT r" r" r# �test_inKnownHostsKeyChanged/ rW z+DefaultAPITests.test_inKnownHostsKeyChangedN)r7 r8 r9 r: r0 r> r: rD rL rM rO rP rV rX rZ r" r" r" r# r+ � s r+ )Er: r< �binasciir ra r r �unittestr �zope.interface.verifyr �twisted.conch.errorr r r �twisted.conch.interfacesr �twisted.internet.deferr �twisted.python.compatr �twisted.python.filepathr �twisted.python.reflectr �twisted.test.testutilsr �twisted.trial.unittestr �twisted.conch.clientr � twisted.conch.client.knownhostsr r r r r �twisted.conch.ssh.keysr r �twisted.conch.testr �skip�sampleEncodedKey�otherSampleEncodedKeyr� �ecdsaSampleEncodedKeyr* r- r� r� rE r� rF rT r r; rG rK rU rV r[ rw r r r� r) r� r+ r" r" r" r# �<module> sv � �� � ����0 O&f =!r PK F��Zh,i�� � + __pycache__/test_forwarding.cpython-310.pycnu �[��� o �br � @ sr d Z ddlmZ ed�ZerddlmZ ddlmZ ddlm Z ddl m Z m Z ddl mZ G d d � d ej�Zd S ) z, Tests for L{twisted.conch.ssh.forwarding}. � )� requireModule� cryptography)� forwarding)� IPv6Address)� deterministicResolvingReactor)�MemoryReactorClock�StringTransport)�unittestc @ s( e Zd ZdZes dZdd� Zdd� ZdS )� TestSSHConnectForwardingChannelzH Unit and integration tests for L{SSHConnectForwardingChannel}. z Cannot run without cryptographyc C s> |j d d }|jd }|�d�}t|�� d�}|�|� dS )z� Fake that connection was established for first connectTCP request made on C{reactor}. @param reactor: Reactor on which to fake the connection. @type reactor: A reactor. r � N)� peerAddress)� tcpClients� connectors� buildProtocolr �getDestination�makeConnection)�self�reactor�factory� connector�protocol� transport� r �D/usr/lib/python3/dist-packages/twisted/conch/test/test_forwarding.py�makeTCPConnection s z1TestSSHConnectForwardingChannel.makeTCPConnectionc C sp t jdd�}t� }t|dg�|_|�d� | �|� | �|j� | � |j t j � | � t ddd�|j j�� � dS )z� When a hostname is sent as part of forwarding requests, it is resolved using HostnameEndpoint's resolver. )zfwd.example.org�� )�hostportz::1N�TCPr )r �SSHConnectForwardingChannelr r �_reactor� channelOpenr �successResultOf�_channelOpenDeferred�assertIsInstance�client�SSHForwardingClient� assertEqualr r �getPeer)r �sut� memoryReactorr r r � test_channelOpenHostnameRequests+ s �z@TestSSHConnectForwardingChannel.test_channelOpenHostnameRequestsN)�__name__� __module__� __qualname__�__doc__r �skipr r* r r r r r s r N)r. �twisted.python.reflectr r �twisted.conch.sshr �twisted.internet.addressr �$twisted.internet.test.test_endpointsr �twisted.test.proto_helpersr r � twisted.trialr �TestCaser r r r r �<module> s PK F��Z�:'� �� ) __pycache__/test_checkers.cpython-310.pycnu �[��� o �blz � @ s� d Z zddlZW n ey dZY nw dZddlZddlmZ ddlmZ ddl m Z ddl m Z dd l mZ dd lmZmZmZmZ dd lmZmZ dd lmZ dd lmZ ddlm Z m Z ddl m Z ddl m!Z! ddl"m#Z# ddl$m%Z% e!d�r�e!d�r�dZ&ddl'm(Z( ddl)m*Z*m+Z+ ddl,m-Z- ddl.m/Z/ ndZ&e0edd�du r�dZ1ndZ1G d d � d e%�Z2G d d � d e%�Z3G d d!� d!e%�Z4G d"d#� d#e%�Z5G d$d%� d%e%�Z6G d&d'� d'e%�Z7G d(d)� d)e%�Z8ed*d+g�Z9G d,d-� d-e:�Z;G d.d/� d/e%�Z<dS )0z& Tests for L{twisted.conch.checkers}. � Nz cannot run without crypt module� )� encodebytes)� namedtuple)�BytesIO)� verifyObject)�'InMemoryUsernamePasswordDatabaseDontUse)�ISSHPrivateKey�IUsernamePassword� SSHPrivateKey�UsernamePassword)�UnauthorizedLogin�UnhandledCredentials)�util)�Failure)�ShadowDatabase� UserDatabase)�FilePath)� requireModule)�MockOS)�TestCase� cryptography�pyasn1)�checkers)�NotEnoughAuthentication�ValidPublicKey)�keys)�keydataz)can't run without cryptography and PyASN1�geteuidz0Cannot run without effective UIDs (questionable)c @ sX e Zd ZdZep eZdd� Zdd� Zdd� Z dd � Z d d � Z d d � Z dd� Z dd� ZdS )� HelperTestszl Tests for helper functions L{verifyCryptedPassword}, L{_pwdGetByName} and L{_shadowGetByName}. c C s t � | _d S �N)r �mockos��self� r# �B/usr/lib/python3/dist-packages/twisted/conch/test/test_checkers.py�setUp> s zHelperTests.setUpc C �4 d}d}t � ||�}| �t�||�d�||�� dS )z� L{verifyCryptedPassword} returns C{True} if the plaintext password passed to it matches the encrypted password passed to it. � secret string�saltyz5{!r} supposed to be valid encrypted password for {!r}N��crypt� assertTruer �verifyCryptedPassword�format�r" �password�salt�cryptedr# r# r$ �test_verifyCryptedPasswordA � ��z&HelperTests.test_verifyCryptedPasswordc C r& )z� L{verifyCryptedPassword} returns True if the provided cleartext password matches the provided MD5 password hash. r/ z$1$saltz3{!r} supposed to be valid encrypted password for {}Nr) r. r# r# r$ � test_verifyCryptedPasswordMD5P r3 z)HelperTests.test_verifyCryptedPasswordMD5c C s4 d}d}t � ||�}| �t�||�d�||�� dS )z� L{verifyCryptedPassword} returns C{False} if the plaintext password passed to it does not match the encrypted password passed to it. z string secretr' z7{!r} not supposed to be valid encrypted password for {}N)r* � assertFalser r, r- )r" r/ �wrongr1 r# r# r$ �test_refuteCryptedPassword_ r3 z&HelperTests.test_refuteCryptedPasswordc C sF t � }|�ddddddd� | �td|� | �t�d�|�d�� d S ) z� L{_pwdGetByName} returns a tuple of items from the UNIX /etc/passwd database if the L{pwd} module is present. �alice�secrit� � z first last�/foo�/bin/sh�pwdN)r �addUser�patchr � assertEqual� _pwdGetByName�getpwnam�r" �userdbr# r# r$ �test_pwdGetByNamen s z HelperTests.test_pwdGetByNamec C s" | � tdd� | �t�d�� dS )zW If the C{pwd} module isn't present, L{_pwdGetByName} returns L{None}. r> Nr8 )r@ r � assertIsNonerB r! r# r# r$ �test_pwdGetByNameWithoutPwdx s z'HelperTests.test_pwdGetByNameWithoutPwdc C s� t � }|�ddddddddd � | �td |� d | j_d | j_| �td | j� | �t� d�|� d�� | �| jj dd g� | �| jj dd g� dS )z� L{_shadowGetByName} returns a tuple of items from the UNIX /etc/shadow database if the L{spwd} is present. �bob� passphraser: r; � � � � � �spwd�) �� �osr N) r r? r@ r r �euid�egidr rA �_shadowGetByName�getspnam� seteuidCalls� setegidCallsrD r# r# r$ �test_shadowGetByName s z HelperTests.test_shadowGetByNamec C sB | � tdd� | �t�d�� | �| jjg � | �| jjg � dS )zP L{_shadowGetByName} returns L{None} if C{spwd} is not present. rP NrI )r@ r rG rV rA r rX rY r! r# r# r$ � test_shadowGetByNameWithoutSpwd� s z+HelperTests.test_shadowGetByNameWithoutSpwdN)�__name__� __module__� __qualname__�__doc__� cryptSkip�dependencySkip�skipr% r2 r4 r7 rF rH rZ r[ r# r# r# r$ r 6 s r c @ sp e Zd ZdZep eZdd� Zdd� Zdd� Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� Zdd� ZdS )�SSHPublicKeyDatabaseTestsz, Tests for L{SSHPublicKeyDatabase}. c C s� t �� | _td�| _td�| _d| j d | j d | _t� | _t | � � �| j_ | jj � � | � td| j� | jj �d�| _| j� � t� }|�dd d d d | jj j d � || j_d S )N� foobar� eggspams t1 s foo t2 s egg rS �.ssh� user� passwordr: r; s first last� /bin/shell)r �SSHPublicKeyDatabase�checkerr �key1�key2�contentr r r �mktemp�path�makedirsr@ r �child�sshDirr r? �_userdbrD r# r# r$ r% � s* � z SSHPublicKeyDatabaseTests.setUpc C sL | j | jgd�}| �|d d t� | �|d d d� | �t|�d� dS )zJ L{SSHPublicKeyDatabase} is deprecated as of version 15.0 )�offendingFunctionsr �category�messagez�twisted.conch.checkers.SSHPublicKeyDatabase was deprecated in Twisted 15.0.0: Please use twisted.conch.checkers.SSHPublicKeyChecker, initialized with an instance of twisted.conch.checkers.UNIXAuthorizedKeysFiles instead.r: N)� flushWarningsr% rA �DeprecationWarning�len)r" � warningsShownr# r# r$ �test_deprecated� s �z)SSHPublicKeyDatabaseTests.test_deprecatedc C sj | j �|��| j� tdd�}d|_| �| j�|�� d|_| �| j�|�� d|_| � | j�|�� d S )Nrg rh rd re s notallowed) rs rr � setContentrn r �blobr+ rk �checkKeyr5 )r" �filename�userr# r# r$ � _testCheckKey� s z'SSHPublicKeyDatabaseTests._testCheckKeyc C �. | � d� | �| jjg � | �| jjg � dS )z� L{SSHPublicKeyDatabase.checkKey} should retrieve the content of the authorized_keys file and check the keys against that file. �authorized_keysN�r� rA r rX rY r! r# r# r$ � test_checkKey� � z'SSHPublicKeyDatabaseTests.test_checkKeyc C r� )z� L{SSHPublicKeyDatabase.checkKey} should retrieve the content of the authorized_keys2 file and check the keys against that file. �authorized_keys2Nr� r! r# r# r$ �test_checkKey2� r� z(SSHPublicKeyDatabaseTests.test_checkKey2c s� | j �d�� � �| j� � �d� | �� jd� | jj�� �fdd�}d| j_d| j_ | � | jd|� | � t d | j� t d d �}d |_ | �| j�|�� | �| jjg d �� | �| jjddg� dS )z� If the key file is readable, L{SSHPublicKeyDatabase.checkKey} should switch its uid/gid to the ones of the authenticated user. r� r � c s � � d� �| �S )Nr� )�chmod)rT ��keyFile� savedSeteuidr# r$ �seteuid� s z>SSHPublicKeyDatabaseTests.test_checkKeyAsRoot.<locals>.seteuidrQ rR r� rS rg rh rd )r r: r rQ r; N)rs rr r} rn r� � addCleanupr r� rT rU r@ r r r~ r+ rk r rA rX rY )r" r� r� r# r� r$ �test_checkKeyAsRoot� s z-SSHPublicKeyDatabaseTests.test_checkKeyAsRootc s\ dd� }� � � jd|� tddtjdtj�tj�� d��}� j� |�}� fdd�}|� |�S ) z� L{SSHPublicKeyDatabase.requestAvatarId} should return the avatar id passed in if its C{_checkKey} method returns True. c S � dS �NTr# ��ignoredr# r# r$ � _checkKey � zASSHPublicKeyDatabaseTests.test_requestAvatarId.<locals>._checkKeyr � test� ssh-rsa� fooc � � � | d� d S �Nr� �rA ��avatarIdr! r# r$ �_verify � z?SSHPublicKeyDatabaseTests.test_requestAvatarId.<locals>._verify) r@ rk r r �publicRSA_opensshr �Key� fromString�privateRSA_openssh�sign�requestAvatarId� addCallback)r" r� � credentials�dr� r# r! r$ �test_requestAvatarId s � z.SSHPublicKeyDatabaseTests.test_requestAvatarIdc C sB dd� }| � | jd|� tddtjdd�}| j�|�}| �|t�S )a( L{SSHPublicKeyDatabase.requestAvatarId} should raise L{ValidPublicKey} if the credentials represent a valid key without a signature. This tells the user that the key is valid for login, but does not actually allow that user to do so without a signature. c S r� r� r# r� r# r# r$ r� " r� zQSSHPublicKeyDatabaseTests.test_requestAvatarIdWithoutSignature.<locals>._checkKeyr r� r� N)r@ rk r r r� r� � assertFailurer �r" r� r� r� r# r# r$ �$test_requestAvatarIdWithoutSignature s � z>SSHPublicKeyDatabaseTests.test_requestAvatarIdWithoutSignaturec C s0 dd� }| � | jd|� | j�d�}| �|t�S )z� If L{SSHPublicKeyDatabase.checkKey} returns False, C{_cbRequestAvatarId} should raise L{UnauthorizedLogin}. c S r� �NFr# r� r# r# r$ r� 2 r� zKSSHPublicKeyDatabaseTests.test_requestAvatarIdInvalidKey.<locals>._checkKeyr N)r@ rk r� r� r )r" r� r� r# r# r$ � test_requestAvatarIdInvalidKey, s z8SSHPublicKeyDatabaseTests.test_requestAvatarIdInvalidKeyc C sR dd� }| � | jd|� tddtjdtj�tj�� d��}| j� |�}| � |t �S )z� Valid keys with invalid signatures should cause L{SSHPublicKeyDatabase.requestAvatarId} to return a {UnauthorizedLogin} failure c S r� r� r# r� r# r# r$ r� @ r� zQSSHPublicKeyDatabaseTests.test_requestAvatarIdInvalidSignature.<locals>._checkKeyr r� r� r� ) r@ rk r r r� r r� r� �privateDSA_opensshr� r� r� r r� r# r# r$ �$test_requestAvatarIdInvalidSignature9 s � z>SSHPublicKeyDatabaseTests.test_requestAvatarIdInvalidSignaturec sV dd� }� � � jd|� tddddd�}� j�|�}� fd d �}|�|� � �|t�S ) z~ Exceptions raised while verifying the key should be normalized into an C{UnauthorizedLogin} failure. c S r� r� r# r� r# r# r$ r� T r� zSSSHPublicKeyDatabaseTests.test_requestAvatarIdNormalizeException.<locals>._checkKeyr r� Ns blobs sigDatas sigc s � � tj�}� �t|�d� | S )Nr: )�flushLoggedErrorsr � BadKeyErrorrA rz )�failure�errorsr! r# r$ �_verifyLoggedException[ � z`SSHPublicKeyDatabaseTests.test_requestAvatarIdNormalizeException.<locals>._verifyLoggedException)r@ rk r r� � addErrbackr� r )r" r� r� r� r� r# r! r$ �&test_requestAvatarIdNormalizeExceptionN s z@SSHPublicKeyDatabaseTests.test_requestAvatarIdNormalizeExceptionN)r\ r] r^ r_ �euidSkipra rb r% r| r� r� r� r� r� r� r� r� r� r# r# r# r$ rc � s rc c @ �D e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d d � Z d d � Z dS )�SSHProtocolCheckerTestsz* Tests for L{SSHProtocolChecker}. c C sL t �� }| �|jg � |�t �� � | �|jtg� | �|j t t j� dS )z� L{SSHProcotolChecker.registerChecker} should add the given checker to the list of registered checkers. N)r �SSHProtocolCheckerrA �credentialInterfaces�registerCheckerrj r �assertIsInstance�r" rk r# r# r$ �test_registerCheckerk s � �z,SSHProtocolCheckerTests.test_registerCheckerc C sN t �� }| �|jg � |�t �� t� | �|jtg� | �|j t t j� dS )z� If a specific interface is passed into L{SSHProtocolChecker.registerChecker}, that interface should be registered instead of what the checker specifies in credentialIntefaces. N)r r� rA r� r� rj r r� r� r# r# r$ �!test_registerCheckerWithInterfacez s �z9SSHProtocolCheckerTests.test_registerCheckerWithInterfacec sJ t �� }t� }|�dd� |�|� |�tdd��}� fdd�}|�|�S )z� L{SSHProtocolChecker.requestAvatarId} should defer to one if its registered checkers to authenticate a user. r� c r� r� r� r� r! r# r$ � _callback� r� z?SSHProtocolCheckerTests.test_requestAvatarId.<locals>._callback)r r� r r? r� r� r r� )r" rk �passwordDatabaser� r� r# r! r$ r� � s z,SSHProtocolCheckerTests.test_requestAvatarIdc C sV t �� }dd� }| �|d|� t� }|�dd� |�|� |�tdd��}| �|t �S )z� If the client indicates that it is never satisfied, by always returning False from _areDone, then L{SSHProtocolChecker} should raise L{NotEnoughAuthentication}. c S r� r� r# r� r# r# r$ �_areDone� r� zYSSHProtocolCheckerTests.test_requestAvatarIdWithNotEnoughAuthentication.<locals>._areDone�areDoner� ) r r� r@ r r? r� r� r r� r )r" rk r� r� r� r# r# r$ �/test_requestAvatarIdWithNotEnoughAuthentication� s zGSSHProtocolCheckerTests.test_requestAvatarIdWithNotEnoughAuthenticationc C s$ t �� }|�tdd��}| �|t�S )z� If the passed credentials aren't handled by any registered checker, L{SSHProtocolChecker} should raise L{UnhandledCredentials}. r� )r r� r� r r� r )r" rk r� r# r# r$ �%test_requestAvatarIdInvalidCredential� s z=SSHProtocolCheckerTests.test_requestAvatarIdInvalidCredentialc C s | � t�� �d�� dS )zV The default L{SSHProcotolChecker.areDone} should simply return True. N)r+ r r� r� r! r# r# r$ � test_areDone� s z$SSHProtocolCheckerTests.test_areDoneN) r\ r] r^ r_ ra rb r� r� r� r� r� r� r# r# r# r$ r� d s r� c @ s` e Zd ZdZep eZdd� Zdd� Zdd� Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� ZdS )�UNIXPasswordDatabaseTestsz, Tests for L{UNIXPasswordDatabase}. c C sP g }|� |j� | �t|�dd� t|d t�r |d �� | �|d |� dS )a� Assert that the L{Deferred} passed in is called back with the value 'username'. This represents a valid login for this TestCase. NOTE: To work, this method's return value must be returned from the test method, or otherwise hooked up to the test machinery. @param d: a L{Deferred} from an L{IChecker.requestAvatarId} method. @type d: L{Deferred} @rtype: L{Deferred} r: zlogin incompleter N)�addBoth�appendrA rz � isinstancer �raiseException)r" r� �username�resultr# r# r$ �assertLoggedIn� s z(UNIXPasswordDatabaseTests.assertLoggedInc C s< t �� }dd� }t� }|�d|dd�ddddd � |�d d ddd d d � t� }|�ddddddddd� |�d |d d�ddddddd� | �t d|� | �t d |� t� }| �td |� d |_d |_ t d d!�}| � |� |�d � | � |jg � | � |jg � d"|_| � |� |�d"� | � |jd#d g� | � |jd#d g� d$S )%z� L{UNIXPasswordDatabase} with no arguments has checks the C{pwd} database and then the C{spwd} database. c S s t � || �}t � |d| �}|S )Nz$1$)r* )r� r/ r0 r1 r# r# r$ r1 � r� z?UNIXPasswordDatabaseTests.test_defaultCheckers.<locals>.cryptedr8 r/ r: r; �foor< r= rI �x�bar�/barr6 rK rL rM rN rO � � � � � � � r> rP rS rQ rR � alicerh � bobr N)r �UNIXPasswordDatabaser r? r r@ r r rT rU r r� r� rA rX rY r� )r" rk r1 r> rP r �credr# r# r$ �test_defaultCheckers� s0 � z.UNIXPasswordDatabaseTests.test_defaultCheckersc C s | � tj| j|d� dS )a� Asserts that the L{Deferred} passed in is erred back with an L{UnauthorizedLogin} L{Failure}. This reprsents an invalid login for this TestCase. NOTE: To work, this method's return value must be returned from the test method, or otherwise hooked up to the test machinery. @param d: a L{Deferred} from an L{IChecker.requestAvatarId} method. @type d: L{Deferred} @rtype: L{None} z bogus valueN)� assertRaisesr r r� �r" r� r# r# r$ �assertUnauthorizedLogin� s �z1UNIXPasswordDatabaseTests.assertUnauthorizedLoginc C sR t � dd�}t� }|�d|ddddd� t�|jg�}| �|�tdd ��d� d S ) zo L{UNIXPasswordDatabase} takes a list of functions to check for UNIX user information. �secret�anybodyr: r; r� r� r= s anybodys secretN) r* r r? r r� rC r� r� r )r" r/ rE rk r# r# r$ �test_passInCheckers s �z-UNIXPasswordDatabaseTests.test_passInCheckersc C sJ dd� }dd� }| � td|� t�|g�}tdd�}| �|�|�d� dS )z� If the encrypted password provided by the getpwnam function is valid (verified by the L{verifyCryptedPassword} function), we callback the C{requestAvatarId} L{Deferred} with the username. c S � | |kS r r# �r1 �pwr# r# r$ r, # � zLUNIXPasswordDatabaseTests.test_verifyPassword.<locals>.verifyCryptedPasswordc S � | | gS r r# �r� r# r# r$ rC & r� z?UNIXPasswordDatabaseTests.test_verifyPassword.<locals>.getpwnamr, � usernameN�r@ r r� r r� r� �r" r, rC rk � credentialr# r# r$ �test_verifyPassword s z-UNIXPasswordDatabaseTests.test_verifyPasswordc C s2 dd� }t �|g�}tdd�}| �|�|�� dS )z} If the getpwnam function raises a KeyError, the login fails with an L{UnauthorizedLogin} exception. c S s t | ��r )�KeyErrorr� r# r# r$ rC 4 r� z?UNIXPasswordDatabaseTests.test_failOnKeyError.<locals>.getpwnamr� N)r r� r r� r� )r" rC rk r� r# r# r$ �test_failOnKeyError. s z-UNIXPasswordDatabaseTests.test_failOnKeyErrorc C sH dd� }dd� }| � td|� t�|g�}tdd�}| �|�|�� dS )z� If the verifyCryptedPassword function doesn't verify the password, the login fails with an L{UnauthorizedLogin} exception. c S r� r� r# r� r# r# r$ r, A r� zOUNIXPasswordDatabaseTests.test_failOnBadPassword.<locals>.verifyCryptedPasswordc S r� r r# r� r# r# r$ rC D r� zBUNIXPasswordDatabaseTests.test_failOnBadPassword.<locals>.getpwnamr, r� N)r@ r r� r r� r� r� r# r# r$ �test_failOnBadPassword; s z0UNIXPasswordDatabaseTests.test_failOnBadPasswordc C sT dd� }dd� }dd� }| � td|� t�||g�}tdd�}| �|�|�d� d S ) a UNIXPasswordDatabase.requestAvatarId loops through each getpwnam function associated with it and returns a L{Deferred} which fires with the result of the first one which returns a value other than None. ones do not verify the password. c S r� r r# r� r# r# r$ r, T r� zRUNIXPasswordDatabaseTests.test_loopThroughFunctions.<locals>.verifyCryptedPasswordc S s | dgS )Nznot the passwordr# r� r# r# r$ � getpwnam1W r� zFUNIXPasswordDatabaseTests.test_loopThroughFunctions.<locals>.getpwnam1c S r� r r# r� r# r# r$ � getpwnam2Z r� zFUNIXPasswordDatabaseTests.test_loopThroughFunctions.<locals>.getpwnam2r, r� Nr� )r" r, r� r� rk r� r# r# r$ �test_loopThroughFunctionsL s z3UNIXPasswordDatabaseTests.test_loopThroughFunctionsc C s� t � }|�ddddddd� |�ddddddd� |�d d ddddd� | �td |� t�tjg�}td d �}| �|�|�� tdd�}| �|�|�� tdd�}| �|�|�� dS )z� If the password returned by any function is C{""}, C{"x"}, or C{"*"} it is not compared against the supplied password. Instead it is skipped. r8 r r: r; r� r� rI r� �carol�*r> r� � r� � xs carol� *N) r r? r@ r r� rB r r� r� )r" r> rk r� r# r# r$ �test_failOnSpecialb s z,UNIXPasswordDatabaseTests.test_failOnSpecialN)r\ r] r^ r_ r` ra rb r� r� r� r� r� r� r� r� r r# r# r# r$ r� � s ( r� c @ �, e Zd ZdZeZdd� Zdd� Zdd� ZdS ) � AuthorizedKeyFileReaderTestsz5 Tests for L{checkers.readAuthorizedKeyFile} c C s0 t d�}t�|dd� �}| �ddgt|�� dS )zg L{checkers.readAuthorizedKeyFile} does not attempt to turn comments into keys sE # this comment is ignored this is not # this is again and this is notc S � | S r r# �r� r# r# r$ �<lambda>� � zCAuthorizedKeyFileReaderTests.test_ignoresComments.<locals>.<lambda>s this is nots and this is notN�r r �readAuthorizedKeyFilerA �list�r" �fileobjr� r# r# r$ �test_ignoresComments s �z1AuthorizedKeyFileReaderTests.test_ignoresCommentsc C s0 t d�}tj|dd� d�}| �dgt|�� dS )zw L{checkers.readAuthorizedKeyFile} ignores leading whitespace in lines, as well as empty lines sg # ignore not ignored c S r r r# r r# r# r$ r � r zYAuthorizedKeyFileReaderTests.test_ignoresLeadingWhitespaceAndEmptyLines.<locals>.<lambda>��parseKeys not ignoredNr r r# r# r$ �*test_ignoresLeadingWhitespaceAndEmptyLines� s �zGAuthorizedKeyFileReaderTests.test_ignoresLeadingWhitespaceAndEmptyLinesc C s4 dd� }t d�}tj||d�}| �dgt|�� dS )z� L{checkers.readAuthorizedKeyFile} does not raise an exception when a key fails to parse (raises a L{twisted.conch.ssh.keys.BadKeyError}), but rather just keeps going c S s | � d�r t�d��| S )N� fzfailed to parse)� startswithr r� )�liner# r# r$ � failOnSome� s zKAuthorizedKeyFileReaderTests.test_ignoresUnparsableKeys.<locals>.failOnSomes failed key good keyr s good keyNr )r" r r r� r# r# r$ �test_ignoresUnparsableKeys� s z7AuthorizedKeyFileReaderTests.test_ignoresUnparsableKeysN) r\ r] r^ r_ ra rb r r r r# r# r# r$ r x s r c @ r ) �InMemorySSHKeyDBTestsz0 Tests for L{checkers.InMemorySSHKeyDB} c C s t �ddgi�}tt j|� dS )z_ L{checkers.InMemorySSHKeyDB} implements L{checkers.IAuthorizedKeysDB} r� s keyN)r �InMemorySSHKeyDBr �IAuthorizedKeysDB�r" �keydbr# r# r$ �test_implementsInterface� s z.InMemorySSHKeyDBTests.test_implementsInterfacec C s* t �ddgi�}| �g t|�d��� dS )z� If the user is not in the mapping provided to L{checkers.InMemorySSHKeyDB}, an empty iterator is returned by L{checkers.InMemorySSHKeyDB.getAuthorizedKeys} r� s keysr� N�r r rA r �getAuthorizedKeysr r# r# r$ � test_noKeysForUnauthorizedUser� s z4InMemorySSHKeyDBTests.test_noKeysForUnauthorizedUserc C s0 t �dddgi�}| �ddgt|�d��� dS )z� If the user is in the mapping provided to L{checkers.InMemorySSHKeyDB}, an iterator with all the keys is returned by L{checkers.InMemorySSHKeyDB.getAuthorizedKeys} r� � a� bNr r r# r# r$ � test_allKeysForAuthorizedUser� s z3InMemorySSHKeyDBTests.test_allKeysForAuthorizedUserN) r\ r] r^ r_ ra rb r r r r# r# r# r$ r � s r c @ r� )� UNIXAuthorizedKeysFilesTestsz8 Tests for L{checkers.UNIXAuthorizedKeysFiles}. c C s~ t � }t| �� �|_|j�� t� | _| j�ddddd|jjd� |j�d�| _ | j �� | j �d�}|� d � d d g| _ d S ) Nr� rh r: r; s alice lastnameri rf r� s key 1 key 2s key 1s key 2) r r ro rp rq r rE r? rr rs r} � expectedKeys)r" r �authorizedKeysr# r# r$ r% � s$ � z"UNIXAuthorizedKeysFilesTests.setUpc C s t �| j�}tt j|� dS )zg L{checkers.UNIXAuthorizedKeysFiles} implements L{checkers.IAuthorizedKeysDB}. N)r �UNIXAuthorizedKeysFilesrE r r r r# r# r$ r � s z5UNIXAuthorizedKeysFilesTests.test_implementsInterfacec C s. t j| jdd� d�}| �g t|�d��� dS )z� If the user is not in the user database provided to L{checkers.UNIXAuthorizedKeysFiles}, an empty iterator is returned by L{checkers.UNIXAuthorizedKeysFiles.getAuthorizedKeys}. c S r r r# r r# r# r$ r � r zMUNIXAuthorizedKeysFilesTests.test_noKeysForUnauthorizedUser.<locals>.<lambda>r rI N)r r$ rE rA r r r r# r# r$ r � s z;UNIXAuthorizedKeysFilesTests.test_noKeysForUnauthorizedUserc C sH | j �d��d� tj| jdd� d�}| �| jdg t|� d��� dS )a If the user is in the user database provided to L{checkers.UNIXAuthorizedKeysFiles}, an iterator with all the keys in C{~/.ssh/authorized_keys} and C{~/.ssh/authorized_keys2} is returned by L{checkers.UNIXAuthorizedKeysFiles.getAuthorizedKeys}. r� s key 3c S r r r# r r# r# r$ r r z`UNIXAuthorizedKeysFilesTests.test_allKeysInAllAuthorizedFilesForAuthorizedUser.<locals>.<lambda>r r� N) rs rr r} r r$ rE rA r" r r r r# r# r$ �1test_allKeysInAllAuthorizedFilesForAuthorizedUser� s �zNUNIXAuthorizedKeysFilesTests.test_allKeysInAllAuthorizedFilesForAuthorizedUserc C s0 t j| jdd� d�}| �| jt|�d��� dS )z� L{checkers.UNIXAuthorizedKeysFiles.getAuthorizedKeys} returns only the keys in C{~/.ssh/authorized_keys} and C{~/.ssh/authorized_keys2} if they exist. c S r r r# r r# r# r$ r r zJUNIXAuthorizedKeysFilesTests.test_ignoresNonexistantFile.<locals>.<lambda>r r� N)r r$ rE rA r" r r r r# r# r$ �test_ignoresNonexistantFile s z8UNIXAuthorizedKeysFilesTests.test_ignoresNonexistantFilec C s@ | j �d��� tj| jdd� d�}| �| jt|� d��� dS )z� L{checkers.UNIXAuthorizedKeysFiles.getAuthorizedKeys} returns only the keys in C{~/.ssh/authorized_keys} and C{~/.ssh/authorized_keys2} if they are readable. r� c S r r r# r r# r# r$ r r zIUNIXAuthorizedKeysFilesTests.test_ignoresUnreadableFile.<locals>.<lambda>r r� N) rs rr rq r r$ rE rA r" r r r r# r# r$ �test_ignoresUnreadableFile s z7UNIXAuthorizedKeysFilesTests.test_ignoresUnreadableFileN) r\ r] r^ r_ ra rb r% r r r% r& r' r# r# r# r$ r! � s r! �_KeyDBr c @ s e Zd ZdZdS )�_DummyExceptionz0 Fake exception to be used for testing. N)r\ r] r^ r_ r# r# r# r$ r) ! s r) c @ sL e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dS )�SSHPublicKeyCheckerTestsz4 Tests for L{checkers.SSHPublicKeyChecker}. c C sD t ddtjdtj�tj��d��| _t dd� �| _ t � | j �| _ d S )Nr� r� r� c S s t j�tj�gS r )r r� r� r r� )�_r# r# r$ r 8 s z0SSHPublicKeyCheckerTests.setUp.<locals>.<lambda>)r r r� r r� r� r� r� r� r( r r �SSHPublicKeyCheckerrk r! r# r# r$ r% 0 s �z SSHPublicKeyCheckerTests.setUpc C s" d| j _| �| j�| j �t� dS )z� Calling L{checkers.SSHPublicKeyChecker.requestAvatarId} with credentials that do not have a signature fails with L{ValidPublicKey}. N)r� � signature�failureResultOfrk r� r r! r# r# r$ � test_credentialsWithoutSignature; s �z9SSHPublicKeyCheckerTests.test_credentialsWithoutSignaturec C s$ d| j _| �| j�| j �tj� dS )z� Calling L{checkers.SSHPublicKeyChecker.requestAvatarId} with credentials that have a bad key fails with L{keys.BadKeyError}. r� N)r� r~ r. rk r� r r� r! r# r# r$ �test_credentialsWithBadKeyE s �z3SSHPublicKeyCheckerTests.test_credentialsWithBadKeyc C s$ t j| j_| �| j�| j�t� dS )z� If L{checkers.IAuthorizedKeysDB.getAuthorizedKeys} returns no keys that match the credentials, L{checkers.SSHPublicKeyChecker.requestAvatarId} fails with L{UnauthorizedLogin}. N)r �publicDSA_opensshr� r~ r. rk r� r r! r# r# r$ � test_credentialsNoMatchingKeyO s �z6SSHPublicKeyCheckerTests.test_credentialsNoMatchingKeyc C s2 t j�tj��d�| j_| �| j � | j�t � dS )z� Calling L{checkers.SSHPublicKeyChecker.requestAvatarId} with credentials that are incorrectly signed fails with L{UnauthorizedLogin}. r� N) r r� r� r r� r� r� r- r. rk r� r r! r# r# r$ � test_credentialsInvalidSignature[ s ���z9SSHPublicKeyCheckerTests.test_credentialsInvalidSignaturec C s< dd� }| � tjd|� | �| j�| j�t� | �t � dS )z� If L{keys.Key.verify} raises an exception, L{checkers.SSHPublicKeyChecker.requestAvatarId} fails with L{UnauthorizedLogin}. c _ s t � �r )r) )�args�kwargsr# r# r$ �failo s z?SSHPublicKeyCheckerTests.test_failureVerifyingKey.<locals>.fail�verifyN) r@ r r� r. rk r� r� r r� r) )r" r6 r# r# r$ �test_failureVerifyingKeyh s �z1SSHPublicKeyCheckerTests.test_failureVerifyingKeyc C s$ | j �| j�}| �d| �|�� dS )zu L{checker.SSHPublicKeyChecker.requestAvatarId}, if successful, callbacks with the username. r� N)rk r� r� rA �successResultOfr� r# r# r$ � test_usernameReturnedOnSuccessy s z7SSHPublicKeyCheckerTests.test_usernameReturnedOnSuccessN) r\ r] r^ r_ ra rb r% r/ r0 r2 r3 r8 r: r# r# r# r$ r* ) s r* )=r_ r* � ImportErrorr` rS �base64r � collectionsr �ior �zope.interface.verifyr �twisted.cred.checkersr �twisted.cred.credentialsr r r r �twisted.cred.errorr r �twisted.pythonr �twisted.python.failurer �twisted.python.fakepwdr r �twisted.python.filepathr �twisted.python.reflectr �twisted.test.test_processr �twisted.trial.unittestr ra � twisted.conchr �twisted.conch.errorr r �twisted.conch.sshr �twisted.conch.testr �getattrr� r rc r� r� r r r! r( � Exceptionr) r* r# r# r# r$ �<module> sX � e JX =4" PPK F��Ze1��He He + __pycache__/test_connection.cpython-310.pycnu �[��� o �bIr � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ed�Z ddl m Z e r3dd lm Z mZ nG d d � d �ZG d d � d ej�ZG dd� d�ZG dd� dej�ZG dd� de j�ZG dd� de j�ZdS )z1 This module tests twisted.conch.ssh.connection. � N)�channel)� test_userauth)� requireModule)�unittest� cryptography)�error)�common� connectionc @ s e Zd ZG dd� d�ZdS )r c @ s e Zd ZdS )zconnection.SSHConnectionN)�__name__� __module__� __qualname__� r r �D/usr/lib/python3/dist-packages/twisted/conch/test/test_connection.py� SSHConnection s r N)r r r r r r r r r s r c @ sd e Zd ZdZdZdZdZdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zdd� ZdS )� TestChannela� A mocked-up version of twisted.conch.ssh.channel.SSHChannel. @ivar gotOpen: True if channelOpen has been called. @type gotOpen: L{bool} @ivar specificData: the specific channel open data passed to channelOpen. @type specificData: L{bytes} @ivar openFailureReason: the reason passed to openFailed. @type openFailed: C{error.ConchError} @ivar inBuffer: a C{list} of strings received by the channel. @type inBuffer: C{list} @ivar extBuffer: a C{list} of 2-tuples (type, extended data) of received by the channel. @type extBuffer: C{list} @ivar numberRequests: the number of requests that have been made to this channel. @type numberRequests: L{int} @ivar gotEOF: True if the other side sent EOF. @type gotEOF: L{bool} @ivar gotOneClose: True if the other side closed the connection. @type gotOneClose: L{bool} @ivar gotClosed: True if the channel is closed. @type gotClosed: L{bool} � TestChannelFc C s d| j S )NzTestChannel %i)�id��selfr r r � logPrefix: s zTestChannel.logPrefixc C s4 d| _ || _g | _g | _d| _d| _d| _d| _dS )zF The channel is open. Set up the instance variables. Tr FN)�gotOpen� specificData�inBuffer� extBuffer�numberRequests�gotEOF� gotOneClose� gotClosed)r r r r r � channelOpen= s zTestChannel.channelOpenc C s || _ dS )zD Opening the channel failed. Store the reason why. N)�openFailureReason)r �reasonr r r � openFailedJ � zTestChannel.openFailedc C s | j d7 _ |dkS )z_ A test request. Return True if data is 'data'. @type data: L{bytes} � � data)r �r �datar r r � request_testP s zTestChannel.request_testc C s | j �|� dS )z= Data was received. Store it in the buffer. N)r �appendr% r r r � dataReceivedY s zTestChannel.dataReceivedc C s | j �||f� dS )zF Extended data was received. Store it in the buffer. N)r r( )r �coder& r r r � extReceived_ s zTestChannel.extReceivedc C � d| _ dS )z1 EOF was received. Remember it. TN)r r r r r � eofReceivede r" zTestChannel.eofReceivedc C r, )z3 Close was received. Remember it. TN)r r r r r � closeReceivedk r" zTestChannel.closeReceivedc C r, )z7 The channel is closed. Rembember it. TN)r r r r r �closedq r" zTestChannel.closedN)r r r �__doc__�namer r r r r! r' r) r+ r- r. r/ r r r r r s r c @ s$ e Zd ZdZdZdd� Zdd� ZdS )� TestAvatarz? A mocked-up version of twisted.conch.avatar.ConchUser �{ c C s4 |t jkr t |||| d�S |dkrt�| jd��dS )z� The server wants us to return a channel. If the requested channel is our TestChannel, return it, otherwise return None. )� remoteWindow�remoteMaxPacketr& �avatar� conch-error-args�error args in wrong orderN)r r1 r � ConchError�_ARGS_ERROR_CODE)r � channelType� windowSize� maxPacketr& r r r � lookupChannel s ��zTestAvatar.lookupChannelc C s |dkrdS |dkrd|fS dS )z� The client has made a global request. If the global request is 'TestGlobal', return True. If the global request is 'TestData', return True and the request-specific data we received. Otherwise, return False. � TestGlobalT� TestDataFr )r � requestTyper& r r r �gotGlobalRequest� s zTestAvatar.gotGlobalRequestN)r r r r0 r: r> rB r r r r r2 x s r2 c @ s@ e Zd ZdZes dZdd� Zdd� Zdd� Zd d � Z d d � Z d S )�TestConnectionz} A subclass of SSHConnection for testing. @ivar channel: the current channel. @type channel. C{TestChannel} � Cannot run without cryptographyc C � dS )NrC r r r r r r � s zTestConnection.logPrefixc C rE )zT The other side made the 'TestGlobal' global request. Return True. Tr r% r r r �global_TestGlobal� s z TestConnection.global_TestGlobalc C s d|fS )zt The other side made the 'Test-Data' global request. Return True and the data we received. Tr r% r r r �global_Test_Data� s z TestConnection.global_Test_Datac C s t |||d�| _| jS )z� The other side is requesting the TestChannel. Create a C{TestChannel} instance, store it, and return it. )r4 r5 r& )r r �r r< r= r& r r r �channel_TestChannel� s �z"TestConnection.channel_TestChannelc C s t d��)zU The other side is requesting the ErrorChannel. Raise an exception. z no such thing)�AssertionErrorrH r r r �channel_ErrorChannel� s z#TestConnection.channel_ErrorChannelN) r r r r0 r �skipr rF rG rI rK r r r r rC � s rC c @ s e Zd ZesdZdd� Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z!d<d=� Z"d>d?� Z#d@dA� Z$dBdC� Z%dDS )E�ConnectionTestsrD c C s6 t �d �| _t� | j_t� | _| j| j_| j�� d S �N)r � FakeTransport� transportr2 r6 rC �conn�serviceStartedr r r r �setUp� s zConnectionTests.setUpc C s@ | j �|� | jjdd� | j_| j �t�d|jd�d � dS )z? Open the channel with the default connection. N���z>2L� s � )rQ � openChannelrP �packets� ssh_CHANNEL_OPEN_CONFIRMATION�struct�packr �r r r r r � _openChannel� s �z ConnectionTests._openChannelc C s | j �� d S rN )rQ �serviceStoppedr r r r �tearDown� � zConnectionTests.tearDownc C s | � | jjj| j� dS )z_ Test that the connection links itself to the avatar in the transport. N)�assertIsrP r6 rQ r r r r �test_linkAvatar� s z ConnectionTests.test_linkAvatarc C s� t � }t � }| j�|� | j�|� | j�d� | �|j� | �|j� | �|j� | �|j� | j�� | �|j� | �|j� | �|j� ddl m } | � |j |� dS )zF Test that serviceStopped() closes any open channels. s r )�ConnectionLostN) r rQ rV rX � assertTruer � assertFalser r] �twisted.internet.errorrb �assertIsInstancer )r �channel1�channel2rb r r r �test_serviceStopped� s z#ConnectionTests.test_serviceStoppedc C s� | j �t�d�d � | �| jjtjdfg� g | j_| j �t�d�d d � | �| jjtjdfg� g | j_| j �t�d�d � | �| jjtj dfg� g | j_| j �t�d�d � | �| jjg � dS ) z� Test that global request packets are dispatched to the global_* methods and the return values are translated into success or failure messages. r? � �� r@ s test datas TestBad� N) rQ �ssh_GLOBAL_REQUESTr �NS� assertEqualrP rW r �MSG_REQUEST_SUCCESS�MSG_REQUEST_FAILUREr r r r �test_GLOBAL_REQUEST s" ���z#ConnectionTests.test_GLOBAL_REQUESTc sB � j �ddd�}� j �d� � fdd�}|�|� |�� j� |S )zh Test that global request success packets cause the Deferred to be called back. � requestr$ Tc s � � | d� d S �Nr$ )ro )r& r r r �check" s z3ConnectionTests.test_REQUEST_SUCCESS.<locals>.check)rQ �sendGlobalRequest�ssh_REQUEST_SUCCESS� addCallback� addErrback�fail�r �dru r r r �test_REQUEST_SUCCESS s z$ConnectionTests.test_REQUEST_SUCCESSc sB � j �ddd�}� j �d� � fdd�}|�� j� |�|� |S )zg Test that global request failure packets cause the Deferred to be erred back. rs r$ Tc s � � | jjd� d S rt )ro �valuer& )�fr r r ru 1 � z3ConnectionTests.test_REQUEST_FAILURE.<locals>.check)rQ rv �ssh_REQUEST_FAILURErx rz ry r{ r r r �test_REQUEST_FAILURE) s z$ConnectionTests.test_REQUEST_FAILUREc C sH | j `| j�t�d�d � | �| jjj� | � | jjj| j� | � | jjj d� | � | jjj d� | � | jjj d� | � | jjj d� | � | j jtjdfg� g | j _| j�t�d�d � | �� | � | j jtjdt�d � t�d � fg� g | j _| j�t�d �d � | �� | � | j jtjd t�d � t�d � fg� dS )z� Test that open channel packets cause a channel to be created and opened or a failure message to be returned. r � � r# s � � BadChannels s s unknown channelrk s ErrorChannels s unknown failureN)rP r6 rQ �ssh_CHANNEL_OPENr rn rc r r ro r& r �remoteWindowLeftr5 rW r � MSG_CHANNEL_OPEN_CONFIRMATION�flushLoggedErrors�MSG_CHANNEL_OPEN_FAILUREr r r r �test_CHANNEL_OPEN8 sV ��� ����� �����z!ConnectionTests.test_CHANNEL_OPENc C s� || j j_| j�t�d�d � | �tj �}| � t |�dd|��� | � |d j j d� | � | j jtjdt�d� t�d � fg� d S ) z� Deliver a request for a channel open which will result in an exception being raised during channel lookup. Assert that an error response is delivered as a result. r7 r� r# zExpected one error, got: r )r3 r8 s {s error args in wrong orderrk N)rP r6 r: rQ r� r rn r� r r9 ro �lenr~ �argsrW r r� )r r* �errorsr r r �_lookupChannelErrorTestl s$ � �����z'ConnectionTests._lookupChannelErrorTestc C s | � d� dS )a� If a C{lookupChannel} implementation raises L{error.ConchError} with the arguments in the wrong order, a C{MSG_CHANNEL_OPEN} failure is still sent in response to the message. This is a temporary work-around until L{error.ConchError} is given better attributes and all of the Conch code starts constructing instances of it properly. Eventually this functionality should be deprecated and then removed. r3 N)r� r r r r �test_lookupChannelError� s z'ConnectionTests.test_lookupChannelErrorc C st t � }| j�|� | j�d� | �|jd� | �|jd� | �|jd� | �| jj| d� | �| jj d d� dS )zv Test that channel open confirmation packets cause the channel to be notified that it's open. s r � N) r rQ rV rX ro r� r5 r �channelsToRemoteChannel�localToRemoteChannelr[ r r r � test_CHANNEL_OPEN_CONFIRMATION� s z.ConnectionTests.test_CHANNEL_OPEN_CONFIRMATIONc C sP t � }| j�|� | j�dt�d� � | �|jjd� | � | jj � |�� dS )zz Test that channel open failure packets cause the channel to be notified that its opening failed. � � failure!)r� r# N) r rQ rV �ssh_CHANNEL_OPEN_FAILUREr rn ro r r� � assertIsNone�channels�getr[ r r r �test_CHANNEL_OPEN_FAILURE� s �z)ConnectionTests.test_CHANNEL_OPEN_FAILUREc C s8 t � }| �|� |j}| j�d� | �|j|d � dS )zc Test that channel window adjust messages add bytes to the channel window. r� r# N)r r\ r� rQ �ssh_CHANNEL_WINDOW_ADJUSTro )r r � oldWindowSizer r r �test_CHANNEL_WINDOW_ADJUST� s z*ConnectionTests.test_CHANNEL_WINDOW_ADJUSTc C s t ddd�}| �|� | j�dt�d� � | �|jdg� | �| jj t j dfg� g | j_ d|j d }| j�dt�|� � | �|jdg� | �| jj t j d fg� t � }| �|� d|jd }g | j_ | j�d t�|� � | �|jg � | �| jj t j d fg� d S ) z� Test that channel data messages are passed up to the channel, or cause the channel to be closed if the data is too large. � � �� localWindow�localMaxPacketr� r$ � � � ar# � �r� N)r r\ rQ �ssh_CHANNEL_DATAr rn ro r rP rW r �MSG_CHANNEL_WINDOW_ADJUST�localWindowLeft�MSG_CHANNEL_CLOSEr� �r r �longData�bigDatar r r �test_CHANNEL_DATA� s: ��� � �z!ConnectionTests.test_CHANNEL_DATAc C s t ddd�}| �|� | j�dt�d� � | �|jdg� | �| jj t j dfg� g | j_ d|j d }| j�dt�|� � | �|jdg� | �| jj t j d fg� t � }| �|� d|jd }g | j_ | j�d t�|� � | �|jg � | �| jj t j d fg� d S ) z� Test that channel extended data messages are passed up to the channel, or cause the channel to be closed if they're too big. r� r� r� s r$ )r r$ r� r� r# r� s N)r r\ rQ �ssh_CHANNEL_EXTENDED_DATAr rn ro r rP rW r r� r� r� r� r� r r r �test_CHANNEL_EXTENDED_DATA� sF ���� � � � �z*ConnectionTests.test_CHANNEL_EXTENDED_DATAc C s, t � }| �|� | j�d� | �|j� dS )zN Test that channel eof messages are passed up to the channel. r� N)r r\ rQ �ssh_CHANNEL_EOFrc r r[ r r r �test_CHANNEL_EOF s z ConnectionTests.test_CHANNEL_EOFc C sh t � }| �|� | �|j� | �|j� | �|j� | j�|� | j� d� | �|j� | �|j� dS )z� Test that channel close messages are passed up to the channel. Also, test that channel.close() is called if both sides are closed when this message is received. r� N) r r\ rc r rd r r rQ � sendClose�ssh_CHANNEL_CLOSEr[ r r r �test_CHANNEL_CLOSE s z"ConnectionTests.test_CHANNEL_CLOSEc sp t � }� �|� � j�dt�d� d � � �|jd� � j�dt�d� d d �}� fdd�}|�|� |S ) zS Test that channel requests that succeed send MSG_CHANNEL_SUCCESS. r� � testrl r# rj r$ c � � � � jjtjdfg� d S �Nr� )ro rP rW r �MSG_CHANNEL_SUCCESS��resultr r r ru 2 � �z;ConnectionTests.test_CHANNEL_REQUEST_success.<locals>.check) r r\ rQ �ssh_CHANNEL_REQUESTr rn ro r rx �r r r| ru r r r � test_CHANNEL_REQUEST_success$ s �� z,ConnectionTests.test_CHANNEL_REQUEST_successc sP t � }� �|� � j�dt�d� d �}� fdd�}|�� j� |�|� |S )zP Test that channel requests that fail send MSG_CHANNEL_FAILURE. r� r� rj c r� r� )ro rP rW r �MSG_CHANNEL_FAILUREr� r r r ru E r� z;ConnectionTests.test_CHANNEL_REQUEST_failure.<locals>.check) r r\ rQ r� r rn rx rz ry r� r r r � test_CHANNEL_REQUEST_failure; s � z,ConnectionTests.test_CHANNEL_REQUEST_failurec s> t � }� �|� � j�|ddd�}� j�d� � fdd�}|S )zj Test that channel request success messages cause the Deferred to be called back. r� r$ Tr� c s � � | � d S rN )rc r� r r r ru Y r_ z;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS.<locals>.check)r r\ rQ � sendRequest�ssh_CHANNEL_SUCCESSr� r r r � test_CHANNEL_REQUEST_SUCCESSO s z,ConnectionTests.test_CHANNEL_REQUEST_SUCCESSc sT t � }� �|� � j�|ddd�}� j�d� � fdd�}|�� j� |�|� |S )zi Test that channel request failure messages cause the Deferred to be erred back. r� rk Tr� c s � � | jjd� d S )Nzchannel request failed)ro r~ r� r r r ru h r� z;ConnectionTests.test_CHANNEL_REQUEST_FAILURE.<locals>.check)r r\ rQ r� �ssh_CHANNEL_FAILURErx rz ry r� r r r � test_CHANNEL_REQUEST_FAILURE^ s z,ConnectionTests.test_CHANNEL_REQUEST_FAILUREc C s| | j �ddd�}|�dd� � | j �ddd� | �| jjtjt� d�d ftjt� d�d fg� | �| j j d |gi� d S ) zQ Test that global request messages are sent in the right format. s wantReplyr$ Tc S � d S rN r ��failurer r r �<lambda>u � z8ConnectionTests.test_sendGlobalRequest.<locals>.<lambda>s noReplyrk Fs �datarl �globalN) rQ rv ry ro rP rW r �MSG_GLOBAL_REQUESTr rn � deferreds�r r| r r r �test_sendGlobalRequesto s ��z&ConnectionTests.test_sendGlobalRequestc C sX t � }| j�|d� | �| jjtjt� d�d fg� | �|j d� | �| jj d� dS )zO Test that open channel messages are sent in the right format. s aaaar s � aaaar r# N) r rQ rV ro rP rW r �MSG_CHANNEL_OPENr rn r �localChannelIDr[ r r r �test_openChannel� s ���� z ConnectionTests.test_openChannelc C s� t � }| �|� | j�|ddd�}|�dd� � | j�|ddd� d|_| j�|ddd� | �| jjt j d t � d� d ft j d t � d� d fg� | �| jj d |g� d S )zR Test that channel request messages are sent in the right format. r� Tc S r� rN r r� r r r r� � r� z2ConnectionTests.test_sendRequest.<locals>.<lambda>� test2rk Fs test3r� s testrl r N)r r\ rQ r� ry � localClosedro rP rW r �MSG_CHANNEL_REQUESTr rn r� �r r r| r r r �test_sendRequest� s$ ���� z ConnectionTests.test_sendRequestc C st t dd�}| �|� d|_| j�|d� | �|jd� d|_| j�|d� | �|jd� | �| jjt j dfg� dS ) zi Test that channel window adjust messages cause bytes to be added to the window. r� )r� r r# T� � � N) r r\ r� rQ � adjustWindowro r� rP rW r r� r[ r r r �test_adjustWindow� s ���z!ConnectionTests.test_adjustWindowc C sX t � }| �|� | j�|d� d|_| j�|d� | �| jjtj dt � d� fg� dS )zO Test that channel data messages are sent in the right format. r� T� br� N) r r\ rQ �sendDatar� ro rP rW r �MSG_CHANNEL_DATAr rn r[ r r r � test_sendData� s �z ConnectionTests.test_sendDatac C s\ t � }| �|� | j�|dd� d|_| j�|dd� | �| jjtj dt � d� fg� dS )zX Test that channel extended data messages are sent in the right format. r# r� Tr� r� r� N) r r\ rQ �sendExtendedDatar� ro rP rW r �MSG_CHANNEL_EXTENDED_DATAr rn r[ r r r �test_sendExtendedData� s ���z%ConnectionTests.test_sendExtendedDatac C sb t � }| �|� | j�|� | �| jjtjdfg� d|_ | j�|� | �| jjtjdfg� dS )zN Test that channel EOF messages are sent in the right format. r� TN) r r\ rQ �sendEOFro rP rW r �MSG_CHANNEL_EOFr� r[ r r r � test_sendEOF� s � �z ConnectionTests.test_sendEOFc C s� t � }| �|� | j�|� | �|j� | �| jjt j dfg� | j�|� | �| jjt j dfg� t � }| �|� | �|j � | � |j � d|_| j�|� | �|j � dS )zP Test that channel close messages are sent in the right format. r� TN)r r\ rQ r� rc r� ro rP rW r r� r rd r � remoteClosed)r r rh r r r �test_sendClose� s( � � z ConnectionTests.test_sendClosec C sZ | j �dddd�}| �|jd� | �|jd� | �|jd� | �tj| j jdddd� dS )z� Test that getChannel dispatches to the avatar when an avatar is present. Correct functioning without the avatar is verified in test_CHANNEL_OPEN. r �2 � r$ r� N) rQ � getChannelro r& r� r5 � assertRaisesr r9 r[ r r r �test_getChannelWithAvatar s �z)ConnectionTests.test_getChannelWithAvatarc C sH | j `| �| j�dd�� | �| j�dd�d� | �| j�dd�� dS )zW Test that gotGlobalRequests dispatches to global_* without an avatar. r? r$ s Test-Data)Tr$ s BadGlobalN)rP r6 rc rQ rB ro rd r r r r �"test_gotGlobalRequestWithoutAvatar s �z2ConnectionTests.test_gotGlobalRequestWithoutAvatarc C sB t � }| �|� | jj|dddd�}| �|tj�}| j�|� |S )z� Whenever an SSH channel gets closed any Deferred that was returned by a sendRequest() on its parent connection must be errbacked. � dummyrequest� dummydatar# �� wantReply)r r\ rQ r� � assertFailurer r9 � channelClosedr� r r r �9test_channelClosedCausesLeftoverChannelDeferredsToErrback) s zIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrbackN)&r r r r rL rS r\ r^ ra ri rr r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rM � sH 4 %+ rM c @ s( e Zd ZdZes dZdd� Zdd� ZdS )� CleanConnectionShutdownTestszL Check whether correct cleanup is performed on connection shutdown. rD c C s, t �d �| _t� | j_t� | _| j| j_d S rN )r rO rP r2 r6 rC rQ r r r r rS ? s z"CleanConnectionShutdownTests.setUpc C s8 | j �� | j jdddd�}| �|tj�}| j �� |S )z� Once the service is stopped any leftover global deferred returned by a sendGlobalRequest() call must be errbacked. r� r� r# r� )rQ rR rv r� r r9 r] r� r r r �9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackE s zVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrbackN)r r r r0 r rL rS r� r r r r r� 7 s r� )r0 rY �twisted.conch.sshr �twisted.conch.testr �twisted.python.reflectr � twisted.trialr r � twisted.conchr r r � SSHChannelr r2 r rC �TestCaserM r� r r r r �<module> s( \), nPK F��ZٳC'į į ( __pycache__/test_session.cpython-310.pycnu �[��� o �b�� � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m Z m Z m Z ddl mZ ddlmZmZ ddlmZmZ dd lmZ dd lmZ dd lmZ dd lm Z ed �Z e rgddl m Z m Z m!Z! nG dd� d�Z!G dd� d�Z"G dd� d�Z#ee!j$�G dd� d��Z%ee!j&�G dd� de%��Z'G dd� d�Z(G dd � d e j)�Z*G d d � d �Z+G d d � d �Z,G d!d"� d"e,�Z-G d#d$� d$�Z.G d%d&� d&ee �Z/G d'd(� d(ee �Z0G d)d*� d*e �Z1G d+d,� d,e �Z2G d-d.� d.e �Z3G d/d0� d0e �Z4dS )1zb Tests for the 'session' channel implementation in twisted.conch.ssh.session. See also RFC 4254. � N)�skipIf)� implementer)�defer�error�protocol�� IPv4Address)� ProcessDone�ProcessTerminated)� components�failure)�Failure)� requireModule)�RegistryUsingMixin)�TestCase� cryptography)�common� connection�sessionc @ s e Zd ZddlmZmZmZ dS )r r )� EnvironmentVariableNotPermitted�ISession�ISessionSetEnvN)�__name__� __module__� __qualname__�twisted.conch.interfacesr r r � r r �A/usr/lib/python3/dist-packages/twisted/conch/test/test_session.pyr " s r c @ � e Zd ZdZdd� ZdS )�SubsystemOnlyAvatarz^ A stub class representing an avatar that is only useful for getting a subsystem. c C s |dksJ �t � S )z� If the other side requests the 'subsystem' subsystem, allow it by returning a MockProtocol to implement it. Otherwise raise an assertion. � subsystem)� MockProtocol��self�name�datar r r �lookupSubsystem0 s z#SubsystemOnlyAvatar.lookupSubsystemN�r r r �__doc__r& r r r r r * � r c @ r )� StubAvatarz� A stub class representing the avatar representing the authenticated user. It implements the I{ISession} interface. c C s" |dkrt � | _|| j_| jS dS )z� If the user requests the TestSubsystem subsystem, connect them to a MockProtocol. If they request neither, then None is returned which is interpreted by SSHSession as a failure. � TestSubsystemN)r! � subsystem� packetDatar" r r r r& ? s �zStubAvatar.lookupSubsystemNr' r r r r r* 9 r) r* c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�StubSessionForStubAvatara2 A stub ISession implementation for our StubAvatar. The instance variables generally keep track of method invocations so that we can test that the methods were called. @ivar avatar: the L{StubAvatar} we are adapting. @ivar ptyRequest: if present, the terminal, window size, and modes passed to the getPty method. @ivar windowChange: if present, the window size passed to the windowChangned method. @ivar shellProtocol: if present, the L{SSHSessionProcessProtocol} passed to the openShell method. @ivar shellTransport: if present, the L{EchoTransport} connected to shellProtocol. @ivar execProtocol: if present, the L{SSHSessionProcessProtocol} passed to the execCommand method. @ivar execTransport: if present, the L{EchoTransport} connected to execProtocol. @ivar execCommandLine: if present, the command line passed to the execCommand method. @ivar gotEOF: if present, an EOF message was received. @ivar gotClosed: if present, a closed message was received. c C s || _ d| _dS )z2 Store the avatar we're adapting. N)�avatar� shellProtocol�r# r/ r r r �__init__e � z!StubSessionForStubAvatar.__init__c C s |dkr |||f| _ dS td��)zw If the terminal is 'bad', fail. Otherwise, store the information in the ptyRequest variable. � badznot getting a ptyN)� ptyRequest� RuntimeError)r# �terminal�window�modesr r r �getPtyl s z StubSessionForStubAvatar.getPtyc C s |dkrt d��|| _dS )zw If all the window sizes are 0, fail. Otherwise, store the size in the windowChange variable. �r r r r z not changing the window sizeN)r6 � windowChange)r# r8 r r r � windowChangedv s z&StubSessionForStubAvatar.windowChangedc C s& | j dur td��|| _ t|�| _dS )z� If we have gotten a shell request before, fail. Otherwise, store the process protocol in the shellProtocol variable, connect it to the EchoTransport and store that as shellTransport. Nz not getting a shell this time)r0 r6 � EchoTransport�shellTransport)r# �ppr r r � openShell� s z"StubSessionForStubAvatar.openShellc C sV || _ |dkr || _dS |dd� dkr'|| _t|�| _|�|dd� � dS td��)z� If the command is 'true', store the command, the process protocol, and the transport we connect to the process protocol. Otherwise, just store the command and raise an error. � successN� s repeat� znot getting a command)�execCommandLine� execProtocolr> � execTransport� outReceivedr6 )r# r@ �commandr r r � execCommand� s z$StubSessionForStubAvatar.execCommandc C � d| _ dS )z2 Note that EOF has been received. TN)�gotEOF�r# r r r � eofReceived� � z$StubSessionForStubAvatar.eofReceivedc C rK )z4 Note that close has been received. TN)� gotClosedrM r r r �closed� rO z StubSessionForStubAvatar.closedN) r r r r( r2 r: r= rA rJ rN rQ r r r r r. K s r. c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )� StubSessionForStubAvatarWithEnva� Same as StubSessionForStubAvatar, but supporting environment variables setting. End users would want to have the same class annotated with C{@implementer(session.ISession, session.ISessionSetEnv)}. The interfaces are split for backwards compatibility, so we split it here to test this compatibility too. @ivar environ: a L{dict} of environment variables passed to the setEnv method. c s t � �|� i | _i | _d S �N)�superr2 �environ� environAtPtyr1 �� __class__r r r2 � s z(StubSessionForStubAvatarWithEnv.__init__c C s0 |dkrt d��|dkrt�d��|| j|< dS )ak If the requested environment variable is 'FAIL', fail. If it is 'IGNORED', raise EnvironmentVariableNotPermitted, which should cause it to be silently ignored. Otherwise, store the requested environment variable. (Real applications should normally implement an allowed list rather than a blocked list.) � FAILz$disallowed environment variable name� IGNOREDz!ignored environment variable nameN)r6 r r rU )r# r$ �valuer r r �setEnv� s �z&StubSessionForStubAvatarWithEnv.setEnvc C s | j �� | _dS )zs Just a simple implementation which records the current environment when PTY is requested. N)rU �copyrV )r# �term� windowSizer9 r r r r: � s z&StubSessionForStubAvatarWithEnv.getPty)r r r r( r2 r\ r: � __classcell__r r rW r rR � s rR c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r> ay A transport for a ProcessProtocol which echos data that is sent to it with a Window newline (CR LF) appended to it. If a null byte is in the data, disconnect. When we are asked to disconnect, disconnect the C{ProcessProtocol} with a 0 exit code. @ivar proto: the C{ProcessProtocol} connected to us. @ivar data: a L{bytes} of data written to us. c C s || _ d| _d| _|�| � dS )z� Initialize our instance variables. @param processProtocol: a C{ProcessProtocol} to connect to ourself. F� N)�protorQ r% �makeConnection)r# �processProtocolr r r r2 � s zEchoTransport.__init__c C s> | j |7 _ | j�|� | j�d� d|v r | �� dS dS )z� We got some data. Give it back to our C{ProcessProtocol} with a newline attached. Disconnect if there's a null byte. � � N)r% rb rH �loseConnection�r# r% r r r �write� s �zEchoTransport.writec C sN | j rdS d| _ | j�� | j�� | j�� | j�t�t� ddd��� dS )z� If we're asked to disconnect (and we haven't already) shut down the C{ProcessProtocol} with a 0 exit code. N� r ) rQ rb �inConnectionLost�outConnectionLost�errConnectionLost� processEndedr r r r rM r r r rg � s z EchoTransport.loseConnectionN)r r r r( r2 ri rg r r r r r> � s r> c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S ) r! a> A sample Protocol which stores the data passed to it. @ivar packetData: a L{bytes} of data to be sent when the connection is made. @ivar data: a L{bytes} of the data passed to us. @ivar open: True if the channel is open. @ivar reason: if not None, the reason the protocol was closed. ra c C s, d| _ d| _d| _| jr| �| j� dS dS )zc Set up the instance variables. If we have any packetData, send it along. ra TN)r% �open�reasonr- � dataReceivedrM r r r �connectionMade s �zMockProtocol.connectionMadec C s" | j |7 _ | j�|d � dS )z� Store the received data and write it back with a tilde appended. The tilde is appended so that the tests can verify that we processed the data. � ~N)r% � transportri rh r r r rq " s zMockProtocol.dataReceivedc C s d| _ || _dS )z: Close the protocol and store the reason. FN)ro rp )r# rp r r r �connectionLost+ r3 zMockProtocol.connectionLostN)r r r r( r- rr rq ru r r r r r! s r! c @ sL e Zd ZdZddd�Zdd� Zdd� Zd d � Zdd d �Zdd� Z dd� Z dS )�StubConnectiona� A stub for twisted.conch.ssh.connection.SSHConnection. Record the data that channels send, and when they try to close the connection. @ivar data: a L{dict} mapping C{SSHChannel}s to a C{list} of L{bytes} of data they sent. @ivar extData: a L{dict} mapping L{SSHChannel}s to a C{list} of L{tuple} of (L{int}, L{bytes}) of extended data they sent. @ivar requests: a L{dict} mapping L{SSHChannel}s to a C{list} of L{tuple} of (L{str}, L{bytes}) of channel requests they made. @ivar eofs: a L{dict} mapping L{SSHChannel}s to C{true} if they have sent an EOF. @ivar closes: a L{dict} mapping L{SSHChannel}s to C{true} if they have sent a close. Nc C s( i | _ i | _i | _i | _i | _|| _dS )z4 Initialize our instance variables. N)r% �extData�requests�eofs�closesrt )r# rt r r r r2 D s zStubConnection.__init__c C � dS )z, Return our logging prefix. �MockConnectionr rM r r r � logPrefixO s zStubConnection.logPrefixc C s | j �|g ��|� dS )z' Record the sent data. N)r% � setdefault�append)r# �channelr% r r r �sendDataU s zStubConnection.sendDatac C s | j �|g ��||f� dS )z0 Record the sent extended data. N)rw r~ r )r# r� �typer% r r r �sendExtendedData[ � z StubConnection.sendExtendedDataFc C s, | j �|g ��|||f� |rt�d�S dS )z2 Record the sent channel request. N)rx r~ r r �succeed)r# r� �requestr% � wantReplyr r r � sendRequesta s �zStubConnection.sendRequestc C � d| j |< dS )z& Record the sent EOF. TN)ry �r# r� r r r �sendEOFi � zStubConnection.sendEOFc C r� )z( Record the sent close. TN)rz r� r r r � sendCloseo r� zStubConnection.sendCloserS )F) r r r r( r2 r} r� r� r� r� r� r r r r rv 3 s rv c @ s@ e Zd ZdZdZdZdd� Zdd� Zdd � Zd d � Z d d � Z dS )� StubTransportz� A stub transport which records the data written. @ivar buf: the data sent to the transport. @type buf: L{bytes} @ivar close: flags indicating if the transport has been closed. @type close: L{bool} ra Fc C � t ddd�S )�2 Return an arbitrary L{IAddress}. �TCP� remotehosti�" r rM r r r �getPeer� � zStubTransport.getPeerc C r� )r� r� � localhosti' r rM r r r �getHost� r� zStubTransport.getHostc C � | j |7 _ dS )z, Record data in the buffer. N)�bufrh r r r ri � s zStubTransport.writec C rK )z6 Note that the connection was closed. TN)�closerM r r r rg � rO z StubTransport.loseConnectionc C r{ )z0 Pretend to set C{TCP_NODELAY}. Nr )r# �enabledr r r � setTcpNoDelay� s zStubTransport.setTcpNoDelayN) r r r r( r� r� r� r� ri rg r� r r r r r� v s r� c @ s e Zd ZdZdZdd� ZdS )�StubTransportWithWriteErrz� A version of StubTransport which records the error data sent to it. @ivar err: the extended data sent to the transport. @type err: L{bytes} ra c C r� )z� Record the extended data in the buffer. This was an old interface that allowed the Transports from ISession.openShell() or ISession.execCommand() to receive extended data from the client. N)�errrh r r r �writeErr� s z"StubTransportWithWriteErr.writeErrN)r r r r( r� r� r r r r r� � s r� c @ r )� StubClientz� A stub class representing the client to a SSHSession. @ivar transport: A L{StubTransport} object which keeps track of the data passed to it. c C s t � | _d S rS )r� rt rM r r r r2 � s zStubClient.__init__N)r r r r( r2 r r r r r� � s r� c @ s� e Zd ZdZes dZd;dd�Zd;dd�Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z d:S )<�SessionInterfaceTestsz� Tests for the SSHSession class interface. This interface is not ideal, but it is tested in order to maintain backwards compatibility. � cannot run without cryptographyTc C s6 t �| � | �� | _|rt�tttj� | �� | _dS )z� Make an SSHSession object to test. Give the channel some window so that it's allowed to send packets. 500 and 100 are arbitrary values. N) r �setUp� getSSHSessionr r �registerAdapterrR r* r �r# �register_adaptersr r r r� � s �zSessionInterfaceTests.setUpc C s t jddt� t� d�S )z+ Return a new SSH session. �� �d )� remoteWindow�remoteMaxPacket�connr/ )r � SSHSessionrv r* r� r r r r� � s �z#SessionInterfaceTests.getSSHSessionc C s | � | jjt� dS )zj Asserts that self.session.session is an instance of StubSessionForStubOldAvatar. N)�assertIsInstancer r. rM r r r �assertSessionIsStubSession� s z0SessionInterfaceTests.assertSessionIsStubSessionc C s6 t jtd�}| �|jd� | �|j� | �|j � dS )z� SSHSession initializes its buffer (buf), client, and ISession adapter. The avatar should not need to be adaptable to an ISession immediately. )r/ ra N)r r� �object� assertEqualr� � assertIsNone�client)r# �sr r r � test_init� s z SessionInterfaceTests.test_initc C s: | j �d� t� | j _| j �d� | �| j jjjd� dS )z� SSHSession.dataReceived() passes data along to a client. If the data comes before there is a client, the data should be discarded. � 1� 2N)r rq r� r� r� rt r� rM r r r �test_client_dataReceived� s z.SessionInterfaceTests.test_client_dataReceivedc C sP | j �tjd� | j �dd� t� | j _| j �tjd� | �| j jjjd� dS )z� SSHSession.extReceived() passed data of type EXTENDED_DATA_STDERR along to the client. If the data comes before there is a client, or if the data is not of type EXTENDED_DATA_STDERR, it is discared. r� � r� � 3N) r � extReceivedr �EXTENDED_DATA_STDERRr� r� r� rt r� rM r r r �test_client_extReceived s z-SessionInterfaceTests.test_client_extReceivedc C s* t � }| j_t� |_| j�tjd� dS )z� SSHSession.extReceived() should handle the case where the transport on the client doesn't have a writeErr method. � ignoredN)r� r r� r� rt r� r r� �r# r� r r r �&test_client_extReceivedWithoutWriteErr s z<SessionInterfaceTests.test_client_extReceivedWithoutWriteErrc C s6 t � | j_| j�� | �| jjjj� d| jjj_dS )z} SSHSession.closed() should tell the transport connected to the client that the connection was lost. FN)r� r r� rQ � assertTruert r� rM r r r �test_client_closed s z(SessionInterfaceTests.test_client_closedc C s0 | j �dt�d��}| �|� | �| j j� dS )zV When a subsystem request fails, SSHSession.client should not be set. r s BadSubsystemN)r �requestReceivedr �NS� assertFalser� r� �r# �retr r r �$test_badSubsystemDoesNotCreateClient! s z:SessionInterfaceTests.test_badSubsystemDoesNotCreateClientc C sR | j �dt�d�d �}| �|� | �| j jtj� | � | j jj j | j j j � dS )z� When a client requests a subsystem, the SSHSession object should get the subsystem by calling avatar.lookupSubsystem, and attach it as the client. r r+ � dataN)r r� r r� r� r� r� r �ProcessProtocol�assertIsrt rb r/ r, r� r r r �test_lookupSubsystem) s � �z*SessionInterfaceTests.test_lookupSubsystemc C s� t jt� t� d�}|�t�d�d �}| �|� | �|j � | � |j j � |�� |�� | �|j j � |�� |�� |�� dS )z� Previously, if one only wanted to implement a subsystem, an ISession adapter wasn't needed because subsystems were looked up using the lookupSubsystem method on the avatar. )r/ r� r r� N)r r� r rv �request_subsystemr r� r� �assertIsNotNoner� r� r� rz �getrN rg rQ )r# r� r� r r r �'test_lookupSubsystemDoesNotNeedISession8 s z=SessionInterfaceTests.test_lookupSubsystemDoesNotNeedISessionc C s^ | j �dt�d�d � | �| j jj| j dg� | j �d� | �| j jj| j d d� dS ) aD After having looked up a subsystem, data should be passed along to the client. Additionally, subsystems were passed the entire request packet as data, instead of just the additional data. We check for the additional tidle to verify that the data passed through the client. r r+ r� s TestSubsystemdata~s more data���s more data~N)r r� r r� r� r� r% rq rM r r r �test_lookupSubsystem_dataI s �� z/SessionInterfaceTests.test_lookupSubsystem_datac C s< | j �dt�d�d � | j �� | �| j jj| j � dS )zd SSHSession.closeReceived() should sent a close message to the remote side. r r+ r� N)r r� r r� � closeReceivedr� r� rz rM r r r �"test_lookupSubsystem_closeReceiveda s � z8SessionInterfaceTests.test_lookupSubsystem_closeReceivedc C sB | � t�}| �t|�ddd�dd� |D �� � |d �t� dS )zo Assert that the request we just made raised a RuntimeError (and only a RuntimeError). rj z!Multiple RuntimeErrors raised: %s� c S s g | ]}t |��qS r )�repr)�.0r r r r � <listcomp>w s zISessionInterfaceTests.assertRequestRaisedRuntimeError.<locals>.<listcomp>r N)�flushLoggedErrorsr6 r� �len�join�trap)r# �errorsr r r � assertRequestRaisedRuntimeErrorm s ��z5SessionInterfaceTests.assertRequestRaisedRuntimeErrorc C sh | j �dd�}| �|� | �� | �| j jt j� | �| j j j| j j� | � | j �dd�� | � � dS )z� When a client requests a shell, the SSHSession object should get the shell by getting an ISession adapter for the avatar, then calling openShell() with a ProcessProtocol to attach. � shellra N) r r� r� r� r� r� �SSHSessionProcessProtocolr� r0 r� r� r� r r r �test_requestShell{ s z'SessionInterfaceTests.test_requestShellc C s� | j �dd�}| �|� | �� | j �d� | �| j j jjd� | �| j jj| j ddg� | �| j j jj � | �| j jj | j dg� dS )z� When a client executes a shell, it should be able to give pass data back and forth between the local and the remote side. r� ra s some data re �� exit-statuss FN) r r� r� r� rq r� r? r% r� rQ rx r� r r r �test_requestShellWithData� s ��z/SessionInterfaceTests.test_requestShellWithDatac C s� | j �dt�d��}| �|� | �� | �| j j� | �| j �dt�d��� | � � | � | j jt j � | � | j j j | j j� | �| j j jd� dS )z� When a client requests a command, the SSHSession object should get the command by getting an ISession adapter for the avatar, then calling execCommand with a ProcessProtocol to attach and the command line. � execs failurerB N)r r� r r� r� r� r� r� r� r� r� r� r� rF r� rE r� r r r �test_requestExec� s z&SessionInterfaceTests.test_requestExecc C s� | j �dt�d��}| �|� | �� | j �d� | �| j j jj d� | �| j j j | j g d�� | j � � | j � � | j � � | �| j j jj � | �| j j j| j dg� dS )zo When a client executes a command, it should be able to give pass data back and forth. r� s repeat hello� some data)s hellor� re r� N)r r� r r� r� r� rq r� rG r% r� rN r� rQ rx r� r r r �test_requestExecWithData� s � �z.SessionInterfaceTests.test_requestExecWithDatac C s� | � � | jdd� t�tttj� | �� }|� dt� ddd��}| � |� | � |jt� | � � | �|� dt� ddd��� | �|jjddg f� dS ) a When a client requests a PTY, the SSHSession object should make the request by getting an ISession adapter for the avatar, then calling getPty with the terminal type, the window size, and any modes the client gave us. F�r� � pty_reqr4 �rj � � � ra s goodN)� doCleanupsr� r r� r. r* r r r� r� �packRequest_pty_reqr� r� r� r� r� r5 )r# � test_sessionr� r r r �test_requestPty� s$ �� ��z%SessionInterfaceTests.test_requestPtyc C s� | � | j�dt�d�t�d� �� | �| jjt� | �� | � | j�dt�d�t�d� �� | �| � � g � | � | j�dt�d�t�d� �� | �| jjj ddi� dS ) a When a client requests passing an environment variable, the SSHSession object should make the request by getting an ISessionSetEnv adapter for the avatar, then calling setEnv with the environment variable name and value. � envrY r4 rZ r� � NAME� valueN) r� r r� r r� r� rR r� r� r� r� rU rM r r r � test_setEnv� s" �����z!SessionInterfaceTests.test_setEnvc C st | � � }| �|�dt�d�t�d� �� | �|�dt�d�t�d� �� | �|jt� | �ddd�|jj � dS )zG Multiple setenv requests will share the same session. r� � Key1� Value 1� Key2� Value2�r� r� N) r� r� r� r r� r� r rR r� rU �r# r� r r r �test_setEnvSessionShare s �����z-SessionInterfaceTests.test_setEnvSessionSharec C s~ | � � }|�dt�d�t�d� � |�dt�d�t�d� � |�dt�ddd �� | �|jt� | �ddd �|jj � d S ) z� Calling another session service after setenv will provide the previous session with the environment variables. r� r� r� r� r� r� � termr; ra r� N) r� r� r r� r r� r� rR r� rV r� r r r �test_setEnvMultiplexShare s ��z/SessionInterfaceTests.test_setEnvMultiplexSharec C sL | � � | jdd� t�tttj� | �| j� dt � d�t � d� �� dS )z� If the avatar does not have an ISessionSetEnv adapter, then a request to pass an environment variable fails gracefully. Fr� r� r� r� N) r� r� r r� r. r* r r r� r� r r� rM r r r �%test_setEnvNotProvidingISessionSetEnv) s ���z;SessionInterfaceTests.test_setEnvNotProvidingISessionSetEnvc C s^ | j �dt �d��}| �|� | �� | �� | �| j �dt �d��� | �| j j jd� dS )z� When the client requests to change the window size, the SSHSession object should make the request by getting an ISession adapter for the avatar, then calling windowChanged with the new window size. � window_changer; r� N) r r� �packRequest_window_changer� r� r� r� r� r< r� r r r �test_requestWindowChange; s � ��z.SessionInterfaceTests.test_requestWindowChangec C �0 t �| j j�| j _ | j �� | �| j j j� dS )z When an EOF is received and an ISession adapter is present, it should be notified of the EOF message. N)r r r/ rN r� rL rM r r r �test_eofReceivedN � z&SessionInterfaceTests.test_eofReceivedc C s. | j �� }| �|� | �| j jj| j � dS )zT When a close is received, the session should send a close message. N)r r� r� r� r� rz r� r r r �test_closeReceivedW s z(SessionInterfaceTests.test_closeReceivedc C r )z� When a close is received and an ISession adapter is present, it should be notified of the close message. N)r r r/ rQ r� rP rM r r r � test_closed_ r z!SessionInterfaceTests.test_closedN)T)!r r r r( r �skipr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r� � s> r� c @ sH e Zd ZdZes dZdd� Zdd� Zdd� Zd d � Z d d � Z d d� Z dS )�SessionWithNoAvatarTestsa Test for the SSHSession interface. Several of the methods (request_shell, request_exec, request_pty_req, request_env, request_window_change) would create a 'session' instance variable from the avatar if one didn't exist when they were called. r� c C s@ t �| � t�tttj� t�� | _t� | j_ | � | jj� d S rS ) r r� r r� r. r* r r r� r/ r� rM r r r r� t s � z SessionWithNoAvatarTests.setUpc C s$ | � tj�| jj�d| jj � dS )zB self.session.session should provide I{ISession}. zISession not provided by %rN)r� r r � providedByrM r r r � assertSessionProvidesISession} s �z6SessionWithNoAvatarTests.assertSessionProvidesISessionc C s | j �dd� | �� dS )ze If an ISession adapter isn't already present, request_shell should get one. r� ra N)r r� r rM r r r � test_requestShellGetsSession� s z5SessionWithNoAvatarTests.test_requestShellGetsSessionc C s | j �dt�d�� | �� dS )zd If an ISession adapter isn't already present, request_exec should get one. r� rB N)r r� r r� r rM r r r �test_requestExecGetsSession� s z4SessionWithNoAvatarTests.test_requestExecGetsSessionc C s$ | j �dt �ddd�� | �� dS )zg If an ISession adapter isn't already present, request_pty_req should get one. r� r� r; ra N)r r� r� r rM r r r � test_requestPtyReqGetsSession� s � z6SessionWithNoAvatarTests.test_requestPtyReqGetsSessionc C s | j �dt �d�� | �� dS )zm If an ISession adapter isn't already present, request_window_change should get one. r )rj rj rj rj N)r r� r r rM r r r �#test_requestWindowChangeGetsSession� s � z<SessionWithNoAvatarTests.test_requestWindowChangeGetsSessionN) r r r r( r r r� r r r r r r r r r r i s r c @ s( e Zd ZdZes dZdd� Zdd� ZdS )� WrappersTestszH A test for the wrapProtocol and wrapProcessProtocol functions. r� c C sz t � }t� |_|�� t�|�}|�d� | �|jjd� |� d� |� ddg� |� � | �|j d� |j �tj� dS )z� L{wrapProtocol}, when passed a L{Protocol} should return something that has write(), writeSequence(), loseConnection() methods which call the Protocol's dataReceived() and connectionLost() methods, respectively. s dataReceivedr� r� r� s data12N)r! r� rt rr r � wrapProtocolrq r� r� ri � writeSequencerg r% rp r� r �ConnectionDone)r# r �wrappedr r r �test_wrapProtocol� s z WrappersTests.test_wrapProtocolc C sf t � }t� |_t�|�}|�� |�d� | �|jjd� |� t � t � ddd��� |j�t j � dS )a. L{wrapPRocessProtocol}, when passed a L{Protocol} should return something that follows the L{IProcessProtocol} interface, with connectionMade() mapping to connectionMade(), outReceived() mapping to dataReceived() and processEnded() mapping to connectionLost(). r� s data~r N)r! r� rt r �wrapProcessProtocolrr rH r� r� rn r r r r rp r� )r# r �process_protocolr r r �!test_wrapProcessProtocol_Protocol� s �z/WrappersTests.test_wrapProcessProtocol_ProtocolN)r r r r( r r r r r r r r r � s r c @ s@ e Zd ZdZes dZdd� Zdd� Zdd� Zd d � Z d d � Z d S )� HelpersTestszM Tests for the 4 helper functions: parseRequest_* and packRequest_*. r� c C sJ | � t�t�d�t�ddddd� t�t�ddd �� �dd d gf� d S ) a1 The payload of a pty-req message is:: string terminal uint32 columns uint32 rows uint32 x pixels uint32 y pixels string modes Modes are:: byte mode number uint32 mode value � xterm�>4Lrj r� r� r� �>BL� rC �r� rj r� r� )r rC N)r� r �parseRequest_pty_reqr r� �struct�packrM r r r �test_parseRequest_pty_req� s ��� �z&HelpersTests.test_parseRequest_pty_reqc C �J t �ddd�}| �|t�d�t�ddddd� t�t�d d d �� � d S � zG See test_parseRequest_pty_req for the payload format. r r s r rj r� r� r� r r rC N�r r� r� r r� r r! �r# �packedr r r � test_packRequest_pty_req_old� s ����z)HelpersTests.test_packRequest_pty_req_oldc C r# r$ r% r& r r r �test_packRequest_pty_req s ����z%HelpersTests.test_packRequest_pty_reqc C s$ | � t�t�ddddd��d� dS )a The payload of a window_change request is:: uint32 columns uint32 rows uint32 x pixels uint32 y pixels parseRequest_window_change() returns (rows, columns, x pixels, y pixels). r rj r� r� r� r N)r� r �parseRequest_window_changer r! rM r r r � test_parseRequest_window_change s �z,HelpersTests.test_parseRequest_window_changec C s$ | � t�d�t�ddddd�� dS )zM See test_parseRequest_window_change for the payload format. r r rj r� r� r� N)r� r r r r! rM r r r � test_packRequest_window_change$ s �z+HelpersTests.test_packRequest_window_changeN) r r r r( r r r" r( r) r+ r, r r r r r � s r c @ s e Zd ZdZes dZdd� Zdd� Zdd� Zd d � Z d d � Z d d� Z dd� Z e eed� d�dd� �Ze eed� d�dd� �Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Ze eed)� d*�d+d,� �Z e eed)� d*�d-d.� �Z d/S )0� SSHSessionProcessProtocolTestsz1 Tests for L{SSHSessionProcessProtocol}. r� c C s@ t � | _tjt| j�ddd�| _t�| j�| _| j�| j� d S )Nr� r� )r� r� r� )r� rt r r� rv r� r@ rc rM r r r r� 6 s �z$SSHSessionProcessProtocolTests.setUpc C s | � | jjj| j � dS )z8 Assert that C{self.session} is closed. N)r� r r� rz rM r r r �assertSessionClosed> s z2SSHSessionProcessProtocolTests.assertSessionClosedc C s | � | jjj| j |� dS )zO Assert that C{self.session} has sent the C{expectedRequests}. N)r� r r� rx )r# �expectedRequestsr r r �assertRequestsEqualD r� z2SSHSessionProcessProtocolTests.assertRequestsEqualc C s | � | jj| j� dS )zy SSHSessionProcessProtocol should set self.session to the session passed to the __init__ method. N)r� r@ r rM r r r r� J s z(SSHSessionProcessProtocolTests.test_initc C � | � | jjj�� | j�� � dS )zu SSHSessionProcessProtocol.getHost() just delegates to its session.conn.transport.getHost(). N)r� r r� rt r� r@ rM r r r � test_getHostQ � z+SSHSessionProcessProtocolTests.test_getHostc C r1 )zu SSHSessionProcessProtocol.getPeer() just delegates to its session.conn.transport.getPeer(). N)r� r r� rt r� r@ rM r r r � test_getPeerX r3 z+SSHSessionProcessProtocolTests.test_getPeerc C s& d| j _| j�� | �| jjd� dS )z� SSHSessionProcessProtocol.connectionMade() should check if there's a 'buf' attribute on its session and write it to the transport if so. s bufferN)r r� r@ rr r� rt rM r r r �test_connectionMade_ s z2SSHSessionProcessProtocolTests.test_connectionMade�SIGALRMzNot all signals availablec C sF t jD ] }d| }tt|�}| j�|�}| �||d|||f � qdS )zi _getSignalName should return the name of a signal when given the signal number. �SIGz %i: %s != %sN)r �SUPPORTED_SIGNALS�getattr�signalr@ �_getSignalNamer� )r# � signalName� signalValue�sshNamer r r �test_getSignalNameh s ��z1SSHSessionProcessProtocolTests.test_getSignalNamec C s4 t jd t _d| j_| �| j�t j�dtj � dS )z� If there are signals in the signal module which aren't in the SSH RFC, we map their name to [signal name]@[platform]. rj NzSIGTwistedTest@) r: �NSIG�SIGTwistedTestr@ �_signalValuesToNamesr� r; �sys�platformrM r r r �!test_getSignalNameWithLocalSignalv s �z@SSHSessionProcessProtocolTests.test_getSignalNameWithLocalSignalc C �* | j �d� | �| jjj| j dg� dS )zy When data is passed to the outReceived method, it should be sent to the session's write method. � test dataN)r@ rH r� r r� r% rM r r r �test_outReceived� � z/SSHSessionProcessProtocolTests.test_outReceivedc C rF )z{ When data is passed to the write method, it should be sent to the session channel's write method. rG N)r@ ri r� r r� r% rM r r r � test_write� rI z)SSHSessionProcessProtocolTests.test_writec C s. | j �ddg� | �| jjj| j dg� dS )z� When a sequence is passed to the writeSequence method, it should be joined together and sent to the session channel's write method. s test r� rG N)r@ r r� r r� r% rM r r r �test_writeSequence� s z1SSHSessionProcessProtocolTests.test_writeSequencec C s* | j �d� | �| jjj| j dg� dS )z� When data is passed to the errReceived method, it should be sent to the session's writeExtended method. rG )rj rG N)r@ � errReceivedr� r r� rw rM r r r �test_errReceived� rI z/SSHSessionProcessProtocolTests.test_errReceivedc C �D | j �� | �| j| jjjv � | j �� | �| jjj| j � dS )zv When outConnectionLost and errConnectionLost are both called, we should send an EOF message. N)r@ rl r� r r� ry rm r� rM r r r �test_outConnectionLost� � z5SSHSessionProcessProtocolTests.test_outConnectionLostc C rN )zh Make sure reverse ordering of events in test_outConnectionLost also sends EOF. N)r@ rm r� r r� ry rl r� rM r r r �test_errConnectionLost� rP z5SSHSessionProcessProtocolTests.test_errConnectionLostc C s$ | j �� | �| jjj| j � dS )zp When loseConnection() is called, it should call loseConnection on the session channel. N)r@ rg r� r r� rz rM r r r �test_loseConnection� s z2SSHSessionProcessProtocolTests.test_loseConnectionc C s | j �t�td��� dS )zr When connectionLost() is called, it should call loseConnection() on the session channel. r N)r@ ru r r r rM r r r �test_connectionLost� s z2SSHSessionProcessProtocolTests.test_connectionLostc C s: | j �ttd��� | �dt�dd�dfg� | �� dS )z� When processEnded is called, if there is an exit code in the reason it should be sent in an exit-status method. The connection should be closed. Nr� z>Ir F)r@ rn r r r0 r r! r. rM r r r � test_processEndedWithExitCode� s z<SSHSessionProcessProtocolTests.test_processEndedWithExitCode� WCOREDUMPz can't run this w/o os.WCOREDUMPc C �V | j �ttdtjd��� | �dt�d�d t�d� t�d� dfg� | � � dS ) z� When processEnded is called, if there is an exit signal in the reason it should be sent in an exit-signal message. The connection should be closed. rj � � exit-signal� TERM� ra FN� r@ rn r r r: �SIGTERMr0 r r� r. rM r r r �'test_processEndedWithExitSignalCoreDump� s" ������� zFSSHSessionProcessProtocolTests.test_processEndedWithExitSignalCoreDumpc C rV ) z� When processEnded is called, if there is an exit signal in the reason it should be sent in an exit-signal message. If no core was dumped, don't set the core-dump bit. rj r rX rY rf ra FNr[ rM r r r �)test_processEndedWithExitSignalNoCoreDump� s ��� zHSSHSessionProcessProtocolTests.test_processEndedWithExitSignalNoCoreDumpN) r r r r( r r r� r. r0 r� r2 r4 r5 r �hasattrr: r? rE rH rJ rK rM rO rQ rR rS rT �osr] r^ r r r r r- . s8 r- c @ s e Zd ZdZes dZdd� ZdS )�SSHSessionClientTestsza SSHSessionClient is an obsolete class used to connect standard IO to an SSHSession. r� c C s. t �� }t� |_|�d� | �|jjd� dS )zL When data is received, it should be sent to the transport. rG N)r �SSHSessionClientr� rt rq r� r� r� r r r �test_dataReceived s z'SSHSessionClientTests.test_dataReceivedN)r r r r( r r rc r r r r ra � s ra )5r( r` r: r rC �unittestr �zope.interfacer �twisted.internetr r r �twisted.internet.addressr �twisted.internet.errorr r �twisted.pythonr r �twisted.python.failurer �twisted.python.reflectr �#twisted.python.test.test_componentsr �twisted.trial.unittestr r �twisted.conch.sshr r r r r* r r. r rR r> �Protocolr! rv r� r� r� r� r r r r- ra r r r r �<module> sR ]0/*C- *B-V RPK F��Z�5��� � loopback.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Loopback helper used in test_ssh and test_recvline """ from twisted.protocols import loopback class LoopbackRelay(loopback.LoopbackRelay): clearCall = None def logPrefix(self): return f"LoopbackRelay({self.target.__class__.__name__!r})" def write(self, data): loopback.LoopbackRelay.write(self, data) if self.clearCall is not None: self.clearCall.cancel() from twisted.internet import reactor self.clearCall = reactor.callLater(0, self._clearBuffer) def _clearBuffer(self): self.clearCall = None loopback.LoopbackRelay.clearBuffer(self) PK F��Z햞�� �� keydata.pynu �[��� # -*- test-case-name: twisted.conch.test.test_keys -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. # pylint: disable=I0011,C0103,W9401,W9402 """ Data used by test_keys as well as others. """ from base64 import decodebytes RSAData = { "n": int( "269413617238113438198661010376758399219880277968382122687862697" "296942471209955603071120391975773283844560230371884389952067978" "789684135947515341209478065209455427327369102356204259106807047" "964139525310539133073743116175821417513079706301100600025815509" "786721808719302671068052414466483676821987505720384645561708425" "794379383191274856941628512616355437197560712892001107828247792" "561858327085521991407807015047750218508971611590850575870321007" "991909043252470730134547038841839367764074379439843108550888709" "430958143271417044750314742880542002948053835745429446485015316" "60749404403945254975473896534482849256068133525751" ), "e": 65537, "d": int( "420335724286999695680502438485489819800002417295071059780489811" "840828351636754206234982682752076205397047218449504537476523960" "987613148307573487322720481066677105211155388802079519869249746" "774085882219244493290663802569201213676433159425782937159766786" "329742053214957933941260042101377175565683849732354700525628975" "239000548651346620826136200952740446562751690924335365940810658" "931238410612521441739702170503547025018016868116037053013935451" "477930426013703886193016416453215950072147440344656137718959053" "897268663969428680144841987624962928576808352739627262941675617" "7724661940425316604626522633351193810751757014073" ), "p": int( "152689878451107675391723141129365667732639179427453246378763774" "448531436802867910180261906924087589684175595016060014593521649" "964959248408388984465569934780790357826811592229318702991401054" "226302790395714901636384511513449977061729214247279176398290513" "085108930550446985490864812445551198848562639933888780317" ), "q": int( "176444974592327996338888725079951900172097062203378367409936859" "072670162290963119826394224277287608693818012745872307600855894" "647300295516866118620024751601329775653542084052616260193174546" "400544176890518564317596334518015173606460860373958663673307503" "231977779632583864454001476729233959405710696795574874403" ), "u": int( "936018002388095842969518498561007090965136403384715613439364803" "229386793506402222847415019772053080458257034241832795210460612" "924445085372678524176842007912276654532773301546269997020970818" "155956828553418266110329867222673040098885651348225673298948529" "93885224775891490070400861134282266967852120152546563278" ), } DSAData = { "g": int( "10253261326864117157640690761723586967382334319435778695" "29171533815411392477819921538350732400350395446211982054" "96512489289702949127531056893725702005035043292195216541" "11525058911428414042792836395195432445511200566318251789" "10575695836669396181746841141924498545494149998282951407" "18645344764026044855941864175" ), "p": int( "10292031726231756443208850082191198787792966516790381991" "77502076899763751166291092085666022362525614129374702633" "26262930887668422949051881895212412718444016917144560705" "45675251775747156453237145919794089496168502517202869160" "78674893099371444940800865897607102159386345313384716752" "18590012064772045092956919481" ), "q": 1393384845225358996250882900535419012502712821577, "x": 1220877188542930584999385210465204342686893855021, "y": int( "14604423062661947579790240720337570315008549983452208015" "39426429789435409684914513123700756086453120500041882809" "10283610277194188071619191739512379408443695946763554493" "86398594314468629823767964702559709430618263927529765769" "10270265745700231533660131769648708944711006508965764877" "684264272082256183140297951" ), } ECDatanistp256 = { "x": int( "762825130203920963171185031449647317742997734817505505433829043" "45687059013883" ), "y": int( "815431978646028526322656647694416475343443758943143196810611371" "59310646683104" ), "privateValue": int( "3463874347721034170096400845565569825355565567882605" "9678074967909361042656500" ), "curve": b"ecdsa-sha2-nistp256", } ECDatanistp384 = { "privateValue": int( "280814107134858470598753916394807521398239633534281" "633982576099083357871098966021020900021966162732114" "95718603965098" ), "x": int( "10036914308591746758780165503819213553101287571902957054148542" "504671046744460374996612408381962208627004841444205030" ), "y": int( "17337335659928075994560513699823544906448896792102247714689323" "575406618073069185107088229463828921069465902299522926" ), "curve": b"ecdsa-sha2-nistp384", } ECDatanistp521 = { "x": int( "12944742826257420846659527752683763193401384271391513286022917" "29910013082920512632908350502247952686156279140016049549948975" "670668730618745449113644014505462" ), "y": int( "10784108810271976186737587749436295782985563640368689081052886" "16296815984553198866894145509329328086635278430266482551941240" "591605833440825557820439734509311" ), "privateValue": int( "662751235215460886290293902658128847495347691199214" "706697089140769672273950767961331442265530524063943" "548846724348048614239791498442599782310681891569896" "0565" ), "curve": b"ecdsa-sha2-nistp521", } Ed25519Data = { "a": ( b"\xf1\x16\xd1\x15J\x1e\x15\x0e\x19^\x19F\xb5\xf2D\r\xb2R\xa0\xae*k" b"#\x13sE\xfd@\xd9W{\x8b" ), "k": ( b"7/%\xda\x8d\xd4\xa8\x9ax|a\xf0\x98\x01\xc6\xf4^mg\x05i17Li\r\x05U" b"\xbb\xc9DX" ), } privateECDSA_openssh521 = b"""-----BEGIN EC PRIVATE KEY----- MIHcAgEBBEIAjn0lSVF6QweS4bjOGP9RHwqxUiTastSE0MVuLtFvkxygZqQ712oZ ewMvqKkxthMQgxzSpGtRBcmkL7RqZ94+18qgBwYFK4EEACOhgYkDgYYABAFpX/6B mxxglwD+VpEvw0hcyxVzLxNnMGzxZGF7xmNj8nlF7M+TQctdlR2Xv/J+AgIeVGmB j2p84bkV9jBzrUNJEACsJjttZw8NbUrhxjkLT/3rMNtuwjE4vLja0P7DMTE0EV8X f09ETdku/z/1tOSSrSvRwmUcM9nQUJtHHAZlr5Q0fw== -----END EC PRIVATE KEY-----""" # New format introduced in OpenSSH 6.5 privateECDSA_openssh521_new = b"""-----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAArAAAABNlY2RzYS 1zaGEyLW5pc3RwNTIxAAAACG5pc3RwNTIxAAAAhQQBaV/+gZscYJcA/laRL8NIXMsVcy8T ZzBs8WRhe8ZjY/J5RezPk0HLXZUdl7/yfgICHlRpgY9qfOG5FfYwc61DSRAArCY7bWcPDW 1K4cY5C0/96zDbbsIxOLy42tD+wzExNBFfF39PRE3ZLv8/9bTkkq0r0cJlHDPZ0FCbRxwG Za+UNH8AAAEAeRISlnkSEpYAAAATZWNkc2Etc2hhMi1uaXN0cDUyMQAAAAhuaXN0cDUyMQ AAAIUEAWlf/oGbHGCXAP5WkS/DSFzLFXMvE2cwbPFkYXvGY2PyeUXsz5NBy12VHZe/8n4C Ah5UaYGPanzhuRX2MHOtQ0kQAKwmO21nDw1tSuHGOQtP/esw227CMTi8uNrQ/sMxMTQRXx d/T0RN2S7/P/W05JKtK9HCZRwz2dBQm0ccBmWvlDR/AAAAQgCOfSVJUXpDB5LhuM4Y/1Ef CrFSJNqy1ITQxW4u0W+THKBmpDvXahl7Ay+oqTG2ExCDHNKka1EFyaQvtGpn3j7XygAAAA ABAg== -----END OPENSSH PRIVATE KEY----- """ publicECDSA_openssh521 = ( b"ecdsa-sha2-nistp521 AAAAE2VjZHNhLXNoYTItbmlzdHA1MjEAAAAIbmlzdHA1MjEAAACF" b"BAFpX/6BmxxglwD+VpEvw0hcyxVzLxNnMGzxZGF7xmNj8nlF7M+TQctdlR2Xv/J+AgIeVGmB" b"j2p84bkV9jBzrUNJEACsJjttZw8NbUrhxjkLT/3rMNtuwjE4vLja0P7DMTE0EV8Xf09ETdku" b"/z/1tOSSrSvRwmUcM9nQUJtHHAZlr5Q0fw== comment" ) privateECDSA_openssh384 = b"""-----BEGIN EC PRIVATE KEY----- MIGkAgEBBDAtAi7I8j73WCX20qUM5hhHwHuFzYWYYILs2Sh8UZ+awNkARZ/Fu2LU LLl5RtOQpbWgBwYFK4EEACKhZANiAATU17sA9P5FRwSknKcFsjjsk0+E3CeXPYX0 Tk/M0HK3PpWQWgrO8JdRHP9eFE9O/23P8BumwFt7F/AvPlCzVd35VfraFT0o4cCW G0RqpQ+np31aKmeJshkcYALEchnU+tQ= -----END EC PRIVATE KEY-----""" # New format introduced in OpenSSH 6.5 privateECDSA_openssh384_new = b"""-----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAiAAAABNlY2RzYS 1zaGEyLW5pc3RwMzg0AAAACG5pc3RwMzg0AAAAYQTU17sA9P5FRwSknKcFsjjsk0+E3CeX PYX0Tk/M0HK3PpWQWgrO8JdRHP9eFE9O/23P8BumwFt7F/AvPlCzVd35VfraFT0o4cCWG0 RqpQ+np31aKmeJshkcYALEchnU+tQAAADIiktpWIpLaVgAAAATZWNkc2Etc2hhMi1uaXN0 cDM4NAAAAAhuaXN0cDM4NAAAAGEE1Ne7APT+RUcEpJynBbI47JNPhNwnlz2F9E5PzNBytz 6VkFoKzvCXURz/XhRPTv9tz/AbpsBbexfwLz5Qs1Xd+VX62hU9KOHAlhtEaqUPp6d9Wipn ibIZHGACxHIZ1PrUAAAAMC0CLsjyPvdYJfbSpQzmGEfAe4XNhZhgguzZKHxRn5rA2QBFn8 W7YtQsuXlG05CltQAAAAA= -----END OPENSSH PRIVATE KEY----- """ publicECDSA_openssh384 = ( b"ecdsa-sha2-nistp384 AAAAE2VjZHNhLXNoYTItbmlzdHAzODQAAAAIbmlzdHAzODQAAABh" b"BNTXuwD0/kVHBKScpwWyOOyTT4TcJ5c9hfROT8zQcrc+lZBaCs7wl1Ec/14UT07/bc/wG6bA" b"W3sX8C8+ULNV3flV+toVPSjhwJYbRGqlD6enfVoqZ4myGRxgAsRyGdT61A== comment" ) publicECDSA_openssh = ( b"ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABB" b"BKimX1DZ7+Qj0SpfePMbo1pb6yGkAb5l7duC1l855yD7tEfQfqk7bc7v46We1hLMyz6ObUBY" b"gkN/34n42F4vpeA= comment" ) privateECDSA_openssh = b"""-----BEGIN EC PRIVATE KEY----- MHcCAQEEIEyU1YOT2JxxofwbJXIjGftdNcJK55aQdNrhIt2xYQz0oAoGCCqGSM49 AwEHoUQDQgAEqKZfUNnv5CPRKl948xujWlvrIaQBvmXt24LWXznnIPu0R9B+qTtt zu/jpZ7WEszLPo5tQFiCQ3/fifjYXi+l4A== -----END EC PRIVATE KEY-----""" # New format introduced in OpenSSH 6.5 privateECDSA_openssh_new = b"""-----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAaAAAABNlY2RzYS 1zaGEyLW5pc3RwMjU2AAAACG5pc3RwMjU2AAAAQQSopl9Q2e/kI9EqX3jzG6NaW+shpAG+ Ze3bgtZfOecg+7RH0H6pO23O7+OlntYSzMs+jm1AWIJDf9+J+NheL6XgAAAAmCKU4hcilO IXAAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBKimX1DZ7+Qj0Spf ePMbo1pb6yGkAb5l7duC1l855yD7tEfQfqk7bc7v46We1hLMyz6ObUBYgkN/34n42F4vpe AAAAAgTJTVg5PYnHGh/BslciMZ+101wkrnlpB02uEi3bFhDPQAAAAA -----END OPENSSH PRIVATE KEY----- """ publicEd25519_openssh = ( b"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIPEW0RVKHhUOGV4ZRrXyRA2yUqCuKmsjE3NF" b"/UDZV3uL comment" ) # OpenSSH has only ever supported the "new" (v1) format for Ed25519. privateEd25519_openssh_new = b"""-----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW QyNTUxOQAAACDxFtEVSh4VDhleGUa18kQNslKgriprIxNzRf1A2Vd7iwAAAJA61eMLOtXj CwAAAAtzc2gtZWQyNTUxOQAAACDxFtEVSh4VDhleGUa18kQNslKgriprIxNzRf1A2Vd7iw AAAEA3LyXajdSomnh8YfCYAcb0Xm1nBWkxN0xpDQVVu8lEWPEW0RVKHhUOGV4ZRrXyRA2y UqCuKmsjE3NF/UDZV3uLAAAAB2NvbW1lbnQBAgMEBQY= -----END OPENSSH PRIVATE KEY-----""" publicRSA_openssh = ( b"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDVaqx4I9bWG+wloVDEd2NQhEUBVUIUKirg" b"0GDu1OmjrUr6OQZehFV1XwA2v2+qKj+DJjfBaS5b/fDz0n3WmM06QHjVyqgYwBGTJAkMgUyP" b"95ztExZqpATpSXfD5FVks3loniwI66zoBC0hdwWnju9TMA2l5bs9auIJNm/9NNN9b0b/h9qp" b"KSeq/631heY+Grh6HUqx6sBa9zDfH8Kk5O8/kUmWQNUZdy03w17snaY6RKXCpCnd1bqcPUWz" b"xiwYZNW6Pd+rf81CrKfxGAugWBViC6QqbkPD5ASfNaNHjkbtM6Vlvbw7KW4CC1ffdOgTtDc1" b"foNfICZgptyti8ZseZj3 comment" ) privateRSA_openssh = b"""-----BEGIN RSA PRIVATE KEY----- MIIEogIBAAKCAQEA1WqseCPW1hvsJaFQxHdjUIRFAVVCFCoq4NBg7tTpo61K+jkG XoRVdV8ANr9vqio/gyY3wWkuW/3w89J91pjNOkB41cqoGMARkyQJDIFMj/ec7RMW aqQE6Ul3w+RVZLN5aJ4sCOus6AQtIXcFp47vUzANpeW7PWriCTZv/TTTfW9G/4fa qSknqv+t9YXmPhq4eh1KserAWvcw3x/CpOTvP5FJlkDVGXctN8Ne7J2mOkSlwqQp 3dW6nD1Fs8YsGGTVuj3fq3/NQqyn8RgLoFgVYgukKm5Dw+QEnzWjR45G7TOlZb28 OyluAgtX33ToE7Q3NX6DXyAmYKbcrYvGbHmY9wIDAQABAoIBACFMCGaiKNW0+44P chuFCQC58k438BxXS+NRf54jp+Q6mFUb6ot6mB682Lqx+YkSGGCs6MwLTglaQGq6 L5n4syRghLnOaZWa+eL8H1FNJxXbKyet77RprL59EOuGR3BztACHlRU7N/nnFOeA u2geG+bdu3NjuWfmsid/z88wm8KY/dkYNi82LvE9gXqf4QMtR9s0UWI53U/prKiL 2dbzhMQXuXGdBghCeE27xSr0w1jNVSvtvjNfBOp75gQkY/It1z0bbNWcY0MvkoiN Pm7aGDfYDyVniR25RjReyc7Ei+2SWjMHD9+GCPmS6dvrOAg2yc3NCgFIWzk+esrG gKnc1DkCgYEA2XAG2OK81HiRUJTUwRuJOGxGZFpRoJoHPUiPA1HMaxKOfRqxZedx dTngMgV1jRhMr5OxSbFmX3hietEMyuZNQ7Oc9Gt95gyY3M8hYo7VLhLeBK7XJG6D MaIVokQ9IqliJiK5su1UCp0Ig6cHDf8ZGI7Yqx3aSJwxaBGhZm3j2B0CgYEA+0QX i6Q2vh43Haf2YWwExKrdeD4HjB4zAq4DFIeDeuWefQhnqPKqvxJwz3Kpp8cLHYjV IP2cY8pHMFVOi8TP9H8WpJISdKEJwsRunIwz76Xl9+ArrU9cEaoahDdb/Xrqw818 sMjkH1Rjtcev3/QJp/zHJfxc6ZHXksWYHlbTsSMCgYBRr+mSn5QLSoRlPpSzO5IQ tXS4jMnvyQ4BMvovaBKhAyauz1FoFEwmmyikAjMIX+GncJgBNHleUo7Ezza8H0tV rOvBU4TH4WGoStSi/0ANgB8SqVDAKhh1lAwGmxZQqEvsQc177/dLyXUCaMSYuIaI GFpD5wIzlyJkk4MMRSp87QKBgGlmN8ZA3SHFBPOwuD5HlHx2/C3rPzk8lcNDAVHE Qpfz6Bakxu7s1EkQUDgE7jvN19DMzDJpkAegG1qf/jHNHjp+cR4ZlBpOTwzfX1LV 0Rdu7NectlWd244hX7wkiLb8r6vw76QssNyfhrADEriL4t0PwO4jPUpQ/i+4KUZY v7YnAoGAZhb5IDTQVCW8YTGsgvvvnDUefkpVAmiVDQqTvh6/4UD6kKdUcDHpePzg Zrcid5rr3dXSMEbK4tdeQZvPtUg1Uaol3N7bNClIIdvWdPx+5S9T95wJcLnkoHam rXp0IjScTxfLP+Cq5V6lJ94/pX8Ppoj1FdZfNxeS4NYFSRA7kvY= -----END RSA PRIVATE KEY-----""" # Some versions of OpenSSH generate these (slightly different keys): the PKCS#1 # structure is wrapped in an extra ASN.1 SEQUENCE and there's an empty SEQUENCE # following it. It is not any standard key format and was probably a bug in # OpenSSH at some point. privateRSA_openssh_alternate = b"""-----BEGIN RSA PRIVATE KEY----- MIIEqDCCBKICAQACggEBANVqrHgj1tYb7CWhUMR3Y1CERQFVQhQqKuDQYO7U6aOt Svo5Bl6EVXVfADa/b6oqP4MmN8FpLlv98PPSfdaYzTpAeNXKqBjAEZMkCQyBTI/3 nO0TFmqkBOlJd8PkVWSzeWieLAjrrOgELSF3BaeO71MwDaXluz1q4gk2b/00031v Rv+H2qkpJ6r/rfWF5j4auHodSrHqwFr3MN8fwqTk7z+RSZZA1Rl3LTfDXuydpjpE pcKkKd3Vupw9RbPGLBhk1bo936t/zUKsp/EYC6BYFWILpCpuQ8PkBJ81o0eORu0z pWW9vDspbgILV9906BO0NzV+g18gJmCm3K2Lxmx5mPcCAwEAAQKCAQAhTAhmoijV tPuOD3IbhQkAufJON/AcV0vjUX+eI6fkOphVG+qLepgevNi6sfmJEhhgrOjMC04J WkBqui+Z+LMkYIS5zmmVmvni/B9RTScV2ysnre+0aay+fRDrhkdwc7QAh5UVOzf5 5xTngLtoHhvm3btzY7ln5rInf8/PMJvCmP3ZGDYvNi7xPYF6n+EDLUfbNFFiOd1P 6ayoi9nW84TEF7lxnQYIQnhNu8Uq9MNYzVUr7b4zXwTqe+YEJGPyLdc9G2zVnGND L5KIjT5u2hg32A8lZ4kduUY0XsnOxIvtklozBw/fhgj5kunb6zgINsnNzQoBSFs5 PnrKxoCp3NQ5AoGBANlwBtjivNR4kVCU1MEbiThsRmRaUaCaBz1IjwNRzGsSjn0a sWXncXU54DIFdY0YTK+TsUmxZl94YnrRDMrmTUOznPRrfeYMmNzPIWKO1S4S3gSu 1yRugzGiFaJEPSKpYiYiubLtVAqdCIOnBw3/GRiO2Ksd2kicMWgRoWZt49gdAoGB APtEF4ukNr4eNx2n9mFsBMSq3Xg+B4weMwKuAxSHg3rlnn0IZ6jyqr8ScM9yqafH Cx2I1SD9nGPKRzBVTovEz/R/FqSSEnShCcLEbpyMM++l5ffgK61PXBGqGoQ3W/16 6sPNfLDI5B9UY7XHr9/0Caf8xyX8XOmR15LFmB5W07EjAoGAUa/pkp+UC0qEZT6U szuSELV0uIzJ78kOATL6L2gSoQMmrs9RaBRMJpsopAIzCF/hp3CYATR5XlKOxM82 vB9LVazrwVOEx+FhqErUov9ADYAfEqlQwCoYdZQMBpsWUKhL7EHNe+/3S8l1AmjE mLiGiBhaQ+cCM5ciZJODDEUqfO0CgYBpZjfGQN0hxQTzsLg+R5R8dvwt6z85PJXD QwFRxEKX8+gWpMbu7NRJEFA4BO47zdfQzMwyaZAHoBtan/4xzR46fnEeGZQaTk8M 319S1dEXbuzXnLZVnduOIV+8JIi2/K+r8O+kLLDcn4awAxK4i+LdD8DuIz1KUP4v uClGWL+2JwKBgGYW+SA00FQlvGExrIL775w1Hn5KVQJolQ0Kk74ev+FA+pCnVHAx 6Xj84Ga3Inea693V0jBGyuLXXkGbz7VINVGqJdze2zQpSCHb1nT8fuUvU/ecCXC5 5KB2pq16dCI0nE8Xyz/gquVepSfeP6V/D6aI9RXWXzcXkuDWBUkQO5L2MAA= -----END RSA PRIVATE KEY-----""" # New format introduced in OpenSSH 6.5 privateRSA_openssh_new = b"""-----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAABFwAAAAdzc2gtcn NhAAAAAwEAAQAAAQEA1WqseCPW1hvsJaFQxHdjUIRFAVVCFCoq4NBg7tTpo61K+jkGXoRV dV8ANr9vqio/gyY3wWkuW/3w89J91pjNOkB41cqoGMARkyQJDIFMj/ec7RMWaqQE6Ul3w+ RVZLN5aJ4sCOus6AQtIXcFp47vUzANpeW7PWriCTZv/TTTfW9G/4faqSknqv+t9YXmPhq4 eh1KserAWvcw3x/CpOTvP5FJlkDVGXctN8Ne7J2mOkSlwqQp3dW6nD1Fs8YsGGTVuj3fq3 /NQqyn8RgLoFgVYgukKm5Dw+QEnzWjR45G7TOlZb28OyluAgtX33ToE7Q3NX6DXyAmYKbc rYvGbHmY9wAAA7gXkBoMF5AaDAAAAAdzc2gtcnNhAAABAQDVaqx4I9bWG+wloVDEd2NQhE UBVUIUKirg0GDu1OmjrUr6OQZehFV1XwA2v2+qKj+DJjfBaS5b/fDz0n3WmM06QHjVyqgY wBGTJAkMgUyP95ztExZqpATpSXfD5FVks3loniwI66zoBC0hdwWnju9TMA2l5bs9auIJNm /9NNN9b0b/h9qpKSeq/631heY+Grh6HUqx6sBa9zDfH8Kk5O8/kUmWQNUZdy03w17snaY6 RKXCpCnd1bqcPUWzxiwYZNW6Pd+rf81CrKfxGAugWBViC6QqbkPD5ASfNaNHjkbtM6Vlvb w7KW4CC1ffdOgTtDc1foNfICZgptyti8ZseZj3AAAAAwEAAQAAAQAhTAhmoijVtPuOD3Ib hQkAufJON/AcV0vjUX+eI6fkOphVG+qLepgevNi6sfmJEhhgrOjMC04JWkBqui+Z+LMkYI S5zmmVmvni/B9RTScV2ysnre+0aay+fRDrhkdwc7QAh5UVOzf55xTngLtoHhvm3btzY7ln 5rInf8/PMJvCmP3ZGDYvNi7xPYF6n+EDLUfbNFFiOd1P6ayoi9nW84TEF7lxnQYIQnhNu8 Uq9MNYzVUr7b4zXwTqe+YEJGPyLdc9G2zVnGNDL5KIjT5u2hg32A8lZ4kduUY0XsnOxIvt klozBw/fhgj5kunb6zgINsnNzQoBSFs5PnrKxoCp3NQ5AAAAgQCFSxt6mxIQN54frV7a/s aW/t81a7k04haXkiYJvb1wIAOnNb0tG6DSB0cr1N6oqAcHG7gEIKcnQTxsOTnpQc7nFx3R TFy8PdImJv5q1v1Icq5G+nvD0xlgRB2lE6eA9WMp1HpdBgcWXfaLPctkOuKEWk2MBi0tnR zrg0x4PXlUzgAAAIEA2XAG2OK81HiRUJTUwRuJOGxGZFpRoJoHPUiPA1HMaxKOfRqxZedx dTngMgV1jRhMr5OxSbFmX3hietEMyuZNQ7Oc9Gt95gyY3M8hYo7VLhLeBK7XJG6DMaIVok Q9IqliJiK5su1UCp0Ig6cHDf8ZGI7Yqx3aSJwxaBGhZm3j2B0AAACBAPtEF4ukNr4eNx2n 9mFsBMSq3Xg+B4weMwKuAxSHg3rlnn0IZ6jyqr8ScM9yqafHCx2I1SD9nGPKRzBVTovEz/ R/FqSSEnShCcLEbpyMM++l5ffgK61PXBGqGoQ3W/166sPNfLDI5B9UY7XHr9/0Caf8xyX8 XOmR15LFmB5W07EjAAAAAAEC -----END OPENSSH PRIVATE KEY----- """ # Encrypted with the passphrase 'encrypted' privateRSA_openssh_encrypted = b"""-----BEGIN RSA PRIVATE KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: DES-EDE3-CBC,FFFFFFFFFFFFFFFF p2A1YsHLXkpMVcsEqhh/nCYb5AqL0uMzfEIqc8hpZ/Ub8PtLsypilMkqzYTnZIGS ouyPjU/WgtR4VaDnutPWdgYaKdixSEmGhKghCtXFySZqCTJ4O8NCczsktYjUK3D4 Jtl90zL6O81WBY6xP76PBQo9lrI/heAetATeyqutc18bwQIGU+gKk32qvfo15DfS VYiY0Ds4D7F7fd9pz+f5+UbFUCgU+tfDvBrqodYrUgmH7jKoW/CRDCHHyeEIZDbF mcMwdcKOyw1sRLaPdihRSVx3kOMvIotHKVTkIDMp+0RTNeXzQnp5U2qzsxzTcG/M UyJN38XXkuvq5VMj2zmmjHzx34w3NK3ZxpZcoaFUqUBlNp2C8hkCLrAa/DWobKqN 5xA1ElrQvli9XXkT/RIuy4Gc10bbGEoJjuxNRibtSxxWd5Bd1E40ocOd4l1ebI8+ w69XvMTnsmHvkBEADGF2zfRszKnMelg+W5NER1UDuNT03i+1cuhp+2AZg8z7niTO M17XP3ScGVxrQAEYgtxPrPeIpFJvOx2j5Yt78U9Y2WlaAG6DrubbYv2RsMIibhOG yk139vMdD8FwCey6yMkkhFAJwnBtC22MAWgjmC5c6AF3SRQSjjQXepPsJcLgpOjy YwjhnL8w56x9kVDUNPw9A9Cqgxo2sty34ATnKrh4h59PsP83LOL6OC5WjbASgZRd OIBD8RloQPISo+RUF7X0i4kdaHVNPlR0KyapR+3M5BwhQuvEO99IArDV2LNKGzfc W4ssugm8iyAJlmwmb2yRXIDHXabInWY7XCdGk8J2qPFbDTvnPbiagJBimjVjgpWw tV3sVlJYqmOqmCDP78J6he04l0vaHtiOWTDEmNCrK7oFMXIIp3XWjOZGPSOJFdPs 6Go3YB+EGWfOQxqkFM28gcqmYfVPF2sa1FbZLz0ffO11Ma/rliZxZu7WdrAXe/tc BgIQ8etp2PwAK4jCwwVwjIO8FzqQGpS23Y9NY3rfi97ckgYXKESFtXPsMMA+drZd ThbXvccfh4EPmaqQXKf4WghHiVJ+/yuY1kUIDEl/O0jRZWT7STgBim/Aha1m6qRs zl1H7hkDbU4solb1GM5oPzbgGTzyBc+z0XxM9iFRM+fMzPB8+yYHTr4kPbVmKBjy SCovjQQVsHE4YeUGTq6k/NF5cVIRKTW/RlHvzxsky1Zj31MC736jrxGw4KG7VSLZ fP6F5jj+mXwS7m0v5to42JBZmRJdKUD88QaGE3ncyQ4yleW5bn9Lf9SuzQg1Dhao 3rSA1RuexsHlIAHvGxx/17X+pyygl8DJbt6TBfbLQk9wc707DJTfh5M/bnk9wwIX l/Hsa1WtylAMW/2MzgiVy83MbYz4+Ss6GQ5W66okWji+NxrnrYEy6q+WgVQanp7X D+D7oKykqE1Cdvvulvtfl5fh8wlAs8mrUnKPBBUru348u++2lfacLkxRXyT1ooqY uSNE5nlwFt08N2Ou/bl7yq6QNRMYrRkn+UEfHWCNYDoGMHln2/i6Z1RapQzNarik tJf7radBz5nBwBjP08YAEACNSQvpsUgdqiuYjLwX7efFXQva2RzqaQ== -----END RSA PRIVATE KEY-----""" # Encrypted with the passphrase 'encrypted', and using the new format # introduced in OpenSSH 6.5 privateRSA_openssh_encrypted_new = b"""-----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAACmFlczI1Ni1jdHIAAAAGYmNyeXB0AAAAGAAAABD0f9WAof DTbmwztb8pdrSeAAAAEAAAAAEAAAEXAAAAB3NzaC1yc2EAAAADAQABAAABAQDVaqx4I9bW G+wloVDEd2NQhEUBVUIUKirg0GDu1OmjrUr6OQZehFV1XwA2v2+qKj+DJjfBaS5b/fDz0n 3WmM06QHjVyqgYwBGTJAkMgUyP95ztExZqpATpSXfD5FVks3loniwI66zoBC0hdwWnju9T MA2l5bs9auIJNm/9NNN9b0b/h9qpKSeq/631heY+Grh6HUqx6sBa9zDfH8Kk5O8/kUmWQN UZdy03w17snaY6RKXCpCnd1bqcPUWzxiwYZNW6Pd+rf81CrKfxGAugWBViC6QqbkPD5ASf NaNHjkbtM6Vlvbw7KW4CC1ffdOgTtDc1foNfICZgptyti8ZseZj3AAADwPQaac8s1xX3af hQTQexj0vEAWDQsLYzDHN9G7W+UP5WHUu7igeu2GqAC/TOnjUXDP73I+EN3n7T3JFeDRfs U1Z6Zqb0NKHSRVYwDIdIi8qVohFv85g6+xQ01OpaoOzz+vI34OUvCRHQGTgR6L9fQShZyC McopYMYfbIse6KcqkfxX3KSdG1Pao6Njx/ShFRbgvmALpR/z0EaGCzHCDxpfUyAdnxm621 Jzaf+LverWdN7sfrfMptaS9//9iJb70sL67K+YIB64qhDnA/w9UOQfXGQFL+AEtdM0BPv8 thP1bs7T0yucBl+ZXdrDKVLZfaS3S/w85Jlgfu+a1DG73pOBOuag435iEJ9EnspjXiiydx GrfSRk2C+/c4fBDZVGFscK5bfQuUUZyU1qOagekxX7WLHFKk9xajnud+nrAN070SeNwlX8 FZ2CI4KGlQfDvVUpKanYn8Kkj3fZ+YBGyx4M+19clF65FKSM0x1Rrh5tAmNT/SNDbSc28m ASxrBhztzxUFTrIn3tp+uqkJniFLmFsUtiAUmj8fNyE9blykU7dqq+CqpLA872nQ9bOHHA JsS1oBYmQ0n6AJz8WrYMdcepqWVld6Q8QSD1zdrY/sAWUovuBA1s4oIEXZhpXSS4ZJiMfh PVktKBwj5bmoG/mmwYLbo0JHntK8N3TGTzTGLq5TpSBBdVvWSWo7tnfEkrFObmhi1uJSrQ 3zfPVP6BguboxBv+oxhaUBK8UOANe6ZwM4vfiu+QN+sZqWymHIfAktz7eWzwlToe4cKpdG Uv+e3/7Lo2dyMl3nke5HsSUrlsMGPREuGkBih8+o85ii6D+cuCiVtus3f5c78Cir80zLIr Z0wWvEAjciEvml00DWaA+JIaOrWwvXySaOzFGpCqC9SQjao379bvn9P3b7kVZsy6zBfHqm bNEJUOuhBZaY8Okz36chh1xqh4sz7m3nsZ3GYGcvM+3mvRY72QnqsQEG0Sp1XYIn2bHa29 tqp7CG9X8J6dqMcPeoPRDWIX9gw7EPl/M0LP6xgewGJ9bgxwle6Mnr9kNITIswjAJqrLec zx7dfixjAPc42ADqrw/tEdFQcSqxigcfJNKO1LbDBjh+Hk/cSBou2PoxbIcl0qfQfbGcqI Dbpd695IEuiW9pYR22txNoIi+7cbMsuFHxQ/OqbrX/jCsprGNNJLAjgGsVEI1JnHWDH0db 3UbqbOHAeY3ufoYXNY1utVOIACpW3r9wBw3FjRi04d70VcKr16OXvOAHGN2G++Y+kMya84 Hl/Kt/gA== -----END OPENSSH PRIVATE KEY----- """ # Encrypted with the passphrase 'testxp'. NB: this key was generated by # OpenSSH, so it doesn't use the same key data as the other keys here. privateRSA_openssh_encrypted_aes = b"""-----BEGIN RSA PRIVATE KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: AES-128-CBC,0673309A6ACCAB4B77DEE1C1E536AC26 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----""" publicRSA_lsh = ( b"{KDEwOnB1YmxpYy1rZXkoMTQ6cnNhLXBrY3MxLXNoYTEoMTpuMjU3OgDVaqx4I9bWG+wloVD" b"Ed2NQhEUBVUIUKirg0GDu1OmjrUr6OQZehFV1XwA2v2+qKj+DJjfBaS5b/fDz0n3WmM06QHj" b"VyqgYwBGTJAkMgUyP95ztExZqpATpSXfD5FVks3loniwI66zoBC0hdwWnju9TMA2l5bs9auI" b"JNm/9NNN9b0b/h9qpKSeq/631heY+Grh6HUqx6sBa9zDfH8Kk5O8/kUmWQNUZdy03w17snaY" b"6RKXCpCnd1bqcPUWzxiwYZNW6Pd+rf81CrKfxGAugWBViC6QqbkPD5ASfNaNHjkbtM6Vlvbw" b"7KW4CC1ffdOgTtDc1foNfICZgptyti8ZseZj3KSgxOmUzOgEAASkpKQ==}" ) privateRSA_lsh = ( b"(11:private-key(9:rsa-pkcs1(1:n257:\x00\xd5j\xacx#\xd6\xd6\x1b\xec%\xa1P" b"\xc4wcP\x84E\x01UB\x14**\xe0\xd0`\xee\xd4\xe9\xa3\xadJ\xfa9\x06^\x84Uu_" b"\x006\xbfo\xaa*?\x83&7\xc1i.[\xfd\xf0\xf3\xd2}\xd6\x98\xcd:@x\xd5\xca" b"\xa8\x18\xc0\x11\x93$\t\x0c\x81L\x8f\xf7\x9c\xed\x13\x16j\xa4\x04\xe9Iw" b"\xc3\xe4Ud\xb3yh\x9e,\x08\xeb\xac\xe8\x04-!w\x05\xa7\x8e\xefS0\r\xa5\xe5" b"\xbb=j\xe2\t6o\xfd4\xd3}oF\xff\x87\xda\xa9)'\xaa\xff\xad\xf5\x85\xe6>" b"\x1a\xb8z\x1dJ\xb1\xea\xc0Z\xf70\xdf\x1f\xc2\xa4\xe4\xef?\x91I\x96@\xd5" b"\x19w-7\xc3^\xec\x9d\xa6:D\xa5\xc2\xa4)\xdd\xd5\xba\x9c=E\xb3\xc6,\x18d" b"\xd5\xba=\xdf\xab\x7f\xcdB\xac\xa7\xf1\x18\x0b\xa0X\x15b\x0b\xa4*nC\xc3" b"\xe4\x04\x9f5\xa3G\x8eF\xed3\xa5e\xbd\xbc;)n\x02\x0bW\xdft\xe8\x13\xb475" b"~\x83_ &`\xa6\xdc\xad\x8b\xc6ly\x98\xf7)(1:e3:\x01\x00\x01)(1:d256:!L" b"\x08f\xa2(\xd5\xb4\xfb\x8e\x0fr\x1b\x85\t\x00\xb9\xf2N7\xf0\x1cWK\xe3Q" b"\x7f\x9e#\xa7\xe4:\x98U\x1b\xea\x8bz\x98\x1e\xbc\xd8\xba\xb1\xf9\x89\x12" b"\x18`\xac\xe8\xcc\x0bN\tZ@j\xba/\x99\xf8\xb3$`\x84\xb9\xcei\x95\x9a\xf9" b"\xe2\xfc\x1fQM'\x15\xdb+'\xad\xef\xb4i\xac\xbe}\x10\xeb\x86Gps\xb4\x00" b"\x87\x95\x15;7\xf9\xe7\x14\xe7\x80\xbbh\x1e\x1b\xe6\xdd\xbbsc\xb9g\xe6" b"\xb2'\x7f\xcf\xcf0\x9b\xc2\x98\xfd\xd9\x186/6.\xf1=\x81z\x9f\xe1\x03-G" b"\xdb4Qb9\xddO\xe9\xac\xa8\x8b\xd9\xd6\xf3\x84\xc4\x17\xb9q\x9d\x06\x08Bx" b"M\xbb\xc5*\xf4\xc3X\xcdU+\xed\xbe3_\x04\xea{\xe6\x04$c\xf2-\xd7=\x1bl" b"\xd5\x9ccC/\x92\x88\x8d>n\xda\x187\xd8\x0f%g\x89\x1d\xb9F4^\xc9\xce\xc4" b"\x8b\xed\x92Z3\x07\x0f\xdf\x86\x08\xf9\x92\xe9\xdb\xeb8\x086\xc9\xcd\xcd" b"\n\x01H[9>z\xca\xc6\x80\xa9\xdc\xd49)(1:p129:\x00\xfbD\x17\x8b\xa46\xbe" b"\x1e7\x1d\xa7\xf6al\x04\xc4\xaa\xddx>\x07\x8c\x1e3\x02\xae\x03\x14\x87" b"\x83z\xe5\x9e}\x08g\xa8\xf2\xaa\xbf\x12p\xcfr\xa9\xa7\xc7\x0b\x1d\x88" b"\xd5 \xfd\x9cc\xcaG0UN\x8b\xc4\xcf\xf4\x7f\x16\xa4\x92\x12t\xa1\t\xc2" b"\xc4n\x9c\x8c3\xef\xa5\xe5\xf7\xe0+\xadO\\\x11\xaa\x1a\x847[\xfdz\xea" b"\xc3\xcd|\xb0\xc8\xe4\x1fTc\xb5\xc7\xaf\xdf\xf4\t\xa7\xfc\xc7%\xfc\\\xe9" b"\x91\xd7\x92\xc5\x98\x1eV\xd3\xb1#)(1:q129:\x00\xd9p\x06\xd8\xe2\xbc\xd4" b"x\x91P\x94\xd4\xc1\x1b\x898lFdZQ\xa0\x9a\x07=H\x8f\x03Q\xcck\x12\x8e}" b"\x1a\xb1e\xe7qu9\xe02\x05u\x8d\x18L\xaf\x93\xb1I\xb1f_xbz\xd1\x0c\xca" b"\xe6MC\xb3\x9c\xf4k}\xe6\x0c\x98\xdc\xcf!b\x8e\xd5.\x12\xde\x04\xae\xd7$" b'n\x831\xa2\x15\xa2D="\xa9b&"\xb9\xb2\xedT\n\x9d\x08\x83\xa7\x07\r\xff' b"\x19\x18\x8e\xd8\xab\x1d\xdaH\x9c1h\x11\xa1fm\xe3\xd8\x1d)(1:a128:if7" b"\xc6@\xdd!\xc5\x04\xf3\xb0\xb8>G\x94|v\xfc-\xeb?9<\x95\xc3C\x01Q\xc4B" b"\x97\xf3\xe8\x16\xa4\xc6\xee\xec\xd4I\x10P8\x04\xee;\xcd\xd7\xd0\xcc\xcc" b"2i\x90\x07\xa0\x1bZ\x9f\xfe1\xcd\x1e:~q\x1e\x19\x94\x1aNO\x0c\xdf_R\xd5" b"\xd1\x17n\xec\xd7\x9c\xb6U\x9d\xdb\x8e!_\xbc$\x88\xb6\xfc\xaf\xab\xf0" b"\xef\xa4,\xb0\xdc\x9f\x86\xb0\x03\x12\xb8\x8b\xe2\xdd\x0f\xc0\xee#=JP" b"\xfe/\xb8)FX\xbf\xb6')(1:b128:Q\xaf\xe9\x92\x9f\x94\x0bJ\x84e>\x94\xb3;" b"\x92\x10\xb5t\xb8\x8c\xc9\xef\xc9\x0e\x012\xfa/h\x12\xa1\x03&\xae\xcfQh" b"\x14L&\x9b(\xa4\x023\x08_\xe1\xa7p\x98\x014y^R\x8e\xc4\xcf6\xbc\x1fKU" b"\xac\xeb\xc1S\x84\xc7\xe1a\xa8J\xd4\xa2\xff@\r\x80\x1f\x12\xa9P\xc0*\x18" b"u\x94\x0c\x06\x9b\x16P\xa8K\xecA\xcd{\xef\xf7K\xc9u\x02h\xc4\x98\xb8\x86" b'\x88\x18ZC\xe7\x023\x97"d\x93\x83\x0cE*|\xed)(1:c128:f\x16\xf9 4\xd0T%' b"\xbca1\xac\x82\xfb\xef\x9c5\x1e~JU\x02h\x95\r\n\x93\xbe\x1e\xbf\xe1@\xfa" b'\x90\xa7Tp1\xe9x\xfc\xe0f\xb7"w\x9a\xeb\xdd\xd5\xd20F\xca\xe2\xd7^A\x9b' b"\xcf\xb5H5Q\xaa%\xdc\xde\xdb4)H!\xdb\xd6t\xfc~\xe5/S\xf7\x9c\tp\xb9\xe4" b"\xa0v\xa6\xadzt\"4\x9cO\x17\xcb?\xe0\xaa\xe5^\xa5'\xde?\xa5\x7f\x0f\xa6" b"\x88\xf5\x15\xd6_7\x17\x92\xe0\xd6\x05I\x10;\x92\xf6)))" ) privateRSA_agentv3 = ( b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00\x03\x01\x00\x01\x00\x00\x01\x00!L" b"\x08f\xa2(\xd5\xb4\xfb\x8e\x0fr\x1b\x85\t\x00\xb9\xf2N7\xf0\x1cWK\xe3Q" b"\x7f\x9e#\xa7\xe4:\x98U\x1b\xea\x8bz\x98\x1e\xbc\xd8\xba\xb1\xf9\x89\x12" b"\x18`\xac\xe8\xcc\x0bN\tZ@j\xba/\x99\xf8\xb3$`\x84\xb9\xcei\x95\x9a\xf9" b"\xe2\xfc\x1fQM'\x15\xdb+'\xad\xef\xb4i\xac\xbe}\x10\xeb\x86Gps\xb4\x00" b"\x87\x95\x15;7\xf9\xe7\x14\xe7\x80\xbbh\x1e\x1b\xe6\xdd\xbbsc\xb9g\xe6" b"\xb2'\x7f\xcf\xcf0\x9b\xc2\x98\xfd\xd9\x186/6.\xf1=\x81z\x9f\xe1\x03-G" b"\xdb4Qb9\xddO\xe9\xac\xa8\x8b\xd9\xd6\xf3\x84\xc4\x17\xb9q\x9d\x06\x08Bx" b"M\xbb\xc5*\xf4\xc3X\xcdU+\xed\xbe3_\x04\xea{\xe6\x04$c\xf2-\xd7=\x1bl" b"\xd5\x9ccC/\x92\x88\x8d>n\xda\x187\xd8\x0f%g\x89\x1d\xb9F4^\xc9\xce\xc4" b"\x8b\xed\x92Z3\x07\x0f\xdf\x86\x08\xf9\x92\xe9\xdb\xeb8\x086\xc9\xcd\xcd" b"\n\x01H[9>z\xca\xc6\x80\xa9\xdc\xd49\x00\x00\x01\x01\x00\xd5j\xacx#\xd6" b"\xd6\x1b\xec%\xa1P\xc4wcP\x84E\x01UB\x14**\xe0\xd0`\xee\xd4\xe9\xa3\xadJ" b"\xfa9\x06^\x84Uu_\x006\xbfo\xaa*?\x83&7\xc1i.[\xfd\xf0\xf3\xd2}\xd6\x98" b"\xcd:@x\xd5\xca\xa8\x18\xc0\x11\x93$\t\x0c\x81L\x8f\xf7\x9c\xed\x13\x16j" b"\xa4\x04\xe9Iw\xc3\xe4Ud\xb3yh\x9e,\x08\xeb\xac\xe8\x04-!w\x05\xa7\x8e" b"\xefS0\r\xa5\xe5\xbb=j\xe2\t6o\xfd4\xd3}oF\xff\x87\xda\xa9)'\xaa\xff\xad" b"\xf5\x85\xe6>\x1a\xb8z\x1dJ\xb1\xea\xc0Z\xf70\xdf\x1f\xc2\xa4\xe4\xef?" b"\x91I\x96@\xd5\x19w-7\xc3^\xec\x9d\xa6:D\xa5\xc2\xa4)\xdd\xd5\xba\x9c=E" b"\xb3\xc6,\x18d\xd5\xba=\xdf\xab\x7f\xcdB\xac\xa7\xf1\x18\x0b\xa0X\x15b" b"\x0b\xa4*nC\xc3\xe4\x04\x9f5\xa3G\x8eF\xed3\xa5e\xbd\xbc;)n\x02\x0bW\xdf" b"t\xe8\x13\xb475~\x83_ &`\xa6\xdc\xad\x8b\xc6ly\x98\xf7\x00\x00\x00\x81" b"\x00\x85K\x1bz\x9b\x12\x107\x9e\x1f\xad^\xda\xfe\xc6\x96\xfe\xdf5k\xb94" b"\xe2\x16\x97\x92&\t\xbd\xbdp \x03\xa75\xbd-\x1b\xa0\xd2\x07G+\xd4\xde" b"\xa8\xa8\x07\x07\x1b\xb8\x04 \xa7'A<l99\xe9A\xce\xe7\x17\x1d\xd1L\\\xbc=" b"\xd2&&\xfej\xd6\xfdHr\xaeF\xfa{\xc3\xd3\x19`D\x1d\xa5\x13\xa7\x80\xf5c)" b"\xd4z]\x06\x07\x16]\xf6\x8b=\xcbd:\xe2\x84ZM\x8c\x06--\x9d\x1c\xeb\x83Lx" b"=yT\xce\x00\x00\x00\x81\x00\xd9p\x06\xd8\xe2\xbc\xd4x\x91P\x94\xd4\xc1" b"\x1b\x898lFdZQ\xa0\x9a\x07=H\x8f\x03Q\xcck\x12\x8e}\x1a\xb1e\xe7qu9\xe02" b"\x05u\x8d\x18L\xaf\x93\xb1I\xb1f_xbz\xd1\x0c\xca\xe6MC\xb3\x9c\xf4k}\xe6" b'\x0c\x98\xdc\xcf!b\x8e\xd5.\x12\xde\x04\xae\xd7$n\x831\xa2\x15\xa2D="' b'\xa9b&"\xb9\xb2\xedT\n\x9d\x08\x83\xa7\x07\r\xff\x19\x18\x8e\xd8\xab' b"\x1d\xdaH\x9c1h\x11\xa1fm\xe3\xd8\x1d\x00\x00\x00\x81\x00\xfbD\x17\x8b" b"\xa46\xbe\x1e7\x1d\xa7\xf6al\x04\xc4\xaa\xddx>\x07\x8c\x1e3\x02\xae\x03" b"\x14\x87\x83z\xe5\x9e}\x08g\xa8\xf2\xaa\xbf\x12p\xcfr\xa9\xa7\xc7\x0b" b"\x1d\x88\xd5 \xfd\x9cc\xcaG0UN\x8b\xc4\xcf\xf4\x7f\x16\xa4\x92\x12t\xa1" b"\t\xc2\xc4n\x9c\x8c3\xef\xa5\xe5\xf7\xe0+\xadO\\\x11\xaa\x1a\x847[\xfdz" b"\xea\xc3\xcd|\xb0\xc8\xe4\x1fTc\xb5\xc7\xaf\xdf\xf4\t\xa7\xfc\xc7%\xfc\\" b"\xe9\x91\xd7\x92\xc5\x98\x1eV\xd3\xb1#" ) publicDSA_openssh = b"""\ ssh-dss AAAAB3NzaC1kc3MAAACBAJKQOsVERVDQIpANHH+JAAylo9\ LvFYmFFVMIuHFGlZpIL7sh3IMkqy+cssINM/lnHD3fmsAyLlUXZtt6PD9LgZRazsPOgptuH+Gu48G\ +yFuE8l0fVVUivos/MmYVJ66qT99htcZKatrTWZnpVW7gFABoqw+he2LZ0gkeU0+Sx9a5AAAAFQD0\ EYmTNaFJ8CS0+vFSF4nYcyEnSQAAAIEAkgLjxHJAE7qFWdTqf7EZngu7jAGmdB9k3YzMHe1ldMxEB\ 7zNw5aOnxjhoYLtiHeoEcOk2XOyvnE+VfhIWwWAdOiKRTEZlmizkvhGbq0DCe2EPMXirjqWACI5nD\ ioQX1oEMonR8N3AEO5v9SfBqS2Q9R6OBr6lf04RvwpHZ0UGu8AAACAAhRpxGMIWEyaEh8YnjiazQT\ NEpklRZqeBGo1gotJggNmVaIQNIClGlLyCi359efEUuQcZ9SXxM59P+hecc/GU/GHakW5YWE4dP2G\ gdgMQWC7S6WFIXePGGXqNQDdWxlX8umhenvQqa1PnKrFRhDrJw8Z7GjdHxflsxCEmXPoLN8= \ comment\ """ privateDSA_openssh = b"""\ -----BEGIN DSA PRIVATE KEY----- MIIBvAIBAAKBgQCSkDrFREVQ0CKQDRx/iQAMpaPS7xWJhRVTCLhxRpWaSC+7IdyD JKsvnLLCDTP5Zxw935rAMi5VF2bbejw/S4GUWs7DzoKbbh/hruPBvshbhPJdH1VV Ir6LPzJmFSeuqk/fYbXGSmra01mZ6VVu4BQAaKsPoXti2dIJHlNPksfWuQIVAPQR iZM1oUnwJLT68VIXidhzISdJAoGBAJIC48RyQBO6hVnU6n+xGZ4Lu4wBpnQfZN2M zB3tZXTMRAe8zcOWjp8Y4aGC7Yh3qBHDpNlzsr5xPlX4SFsFgHToikUxGZZos5L4 Rm6tAwnthDzF4q46lgAiOZw4qEF9aBDKJ0fDdwBDub/UnwaktkPUejga+pX9OEb8 KR2dFBrvAoGAAhRpxGMIWEyaEh8YnjiazQTNEpklRZqeBGo1gotJggNmVaIQNICl GlLyCi359efEUuQcZ9SXxM59P+hecc/GU/GHakW5YWE4dP2GgdgMQWC7S6WFIXeP GGXqNQDdWxlX8umhenvQqa1PnKrFRhDrJw8Z7GjdHxflsxCEmXPoLN8CFQDV2gbL czUdxCus0pfEP1bddaXRLQ== -----END DSA PRIVATE KEY-----\ """ privateDSA_openssh_new = b"""\ -----BEGIN OPENSSH PRIVATE KEY----- b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAABsgAAAAdzc2gtZH NzAAAAgQCSkDrFREVQ0CKQDRx/iQAMpaPS7xWJhRVTCLhxRpWaSC+7IdyDJKsvnLLCDTP5 Zxw935rAMi5VF2bbejw/S4GUWs7DzoKbbh/hruPBvshbhPJdH1VVIr6LPzJmFSeuqk/fYb XGSmra01mZ6VVu4BQAaKsPoXti2dIJHlNPksfWuQAAABUA9BGJkzWhSfAktPrxUheJ2HMh J0kAAACBAJIC48RyQBO6hVnU6n+xGZ4Lu4wBpnQfZN2MzB3tZXTMRAe8zcOWjp8Y4aGC7Y h3qBHDpNlzsr5xPlX4SFsFgHToikUxGZZos5L4Rm6tAwnthDzF4q46lgAiOZw4qEF9aBDK J0fDdwBDub/UnwaktkPUejga+pX9OEb8KR2dFBrvAAAAgAIUacRjCFhMmhIfGJ44ms0EzR KZJUWangRqNYKLSYIDZlWiEDSApRpS8got+fXnxFLkHGfUl8TOfT/oXnHPxlPxh2pFuWFh OHT9hoHYDEFgu0ulhSF3jxhl6jUA3VsZV/LpoXp70KmtT5yqxUYQ6ycPGexo3R8X5bMQhJ lz6CzfAAAB2MVcBjzFXAY8AAAAB3NzaC1kc3MAAACBAJKQOsVERVDQIpANHH+JAAylo9Lv FYmFFVMIuHFGlZpIL7sh3IMkqy+cssINM/lnHD3fmsAyLlUXZtt6PD9LgZRazsPOgptuH+ Gu48G+yFuE8l0fVVUivos/MmYVJ66qT99htcZKatrTWZnpVW7gFABoqw+he2LZ0gkeU0+S x9a5AAAAFQD0EYmTNaFJ8CS0+vFSF4nYcyEnSQAAAIEAkgLjxHJAE7qFWdTqf7EZngu7jA GmdB9k3YzMHe1ldMxEB7zNw5aOnxjhoYLtiHeoEcOk2XOyvnE+VfhIWwWAdOiKRTEZlmiz kvhGbq0DCe2EPMXirjqWACI5nDioQX1oEMonR8N3AEO5v9SfBqS2Q9R6OBr6lf04RvwpHZ 0UGu8AAACAAhRpxGMIWEyaEh8YnjiazQTNEpklRZqeBGo1gotJggNmVaIQNIClGlLyCi35 9efEUuQcZ9SXxM59P+hecc/GU/GHakW5YWE4dP2GgdgMQWC7S6WFIXePGGXqNQDdWxlX8u mhenvQqa1PnKrFRhDrJw8Z7GjdHxflsxCEmXPoLN8AAAAVANXaBstzNR3EK6zSl8Q/Vt11 pdEtAAAAAAE= -----END OPENSSH PRIVATE KEY----- """ publicDSA_lsh = decodebytes( b"""\ e0tERXdPbkIxWW14cFl5MXJaWGtvTXpwa2MyRW9NVHB3TVRJNU9nQ1NrRHJGUkVWUTBDS1FEUngv aVFBTXBhUFM3eFdKaFJWVENMaHhScFdhU0MrN0lkeURKS3N2bkxMQ0RUUDVaeHc5MzVyQU1pNVZG MmJiZWp3L1M0R1VXczdEem9LYmJoL2hydVBCdnNoYmhQSmRIMVZWSXI2TFB6Sm1GU2V1cWsvZlli WEdTbXJhMDFtWjZWVnU0QlFBYUtzUG9YdGkyZElKSGxOUGtzZld1U2tvTVRweE1qRTZBUFFSaVpN MW9VbndKTFQ2OFZJWGlkaHpJU2RKS1NneE9tY3hNams2QUpJQzQ4UnlRQk82aFZuVTZuK3hHWjRM dTR3QnBuUWZaTjJNekIzdFpYVE1SQWU4emNPV2pwOFk0YUdDN1loM3FCSERwTmx6c3I1eFBsWDRT RnNGZ0hUb2lrVXhHWlpvczVMNFJtNnRBd250aER6RjRxNDZsZ0FpT1p3NHFFRjlhQkRLSjBmRGR3 QkR1Yi9Vbndha3RrUFVlamdhK3BYOU9FYjhLUjJkRkJydktTZ3hPbmt4TWpnNkFoUnB4R01JV0V5 YUVoOFluamlhelFUTkVwa2xSWnFlQkdvMWdvdEpnZ05tVmFJUU5JQ2xHbEx5Q2kzNTllZkVVdVFj WjlTWHhNNTlQK2hlY2MvR1UvR0hha1c1WVdFNGRQMkdnZGdNUVdDN1M2V0ZJWGVQR0dYcU5RRGRX eGxYOHVtaGVudlFxYTFQbktyRlJoRHJKdzhaN0dqZEh4ZmxzeENFbVhQb0xOOHBLU2s9fQ== """ ) privateDSA_lsh = decodebytes( b"""\ KDExOnByaXZhdGUta2V5KDM6ZHNhKDE6cDEyOToAkpA6xURFUNAikA0cf4kADKWj0u8ViYUVUwi4 cUaVmkgvuyHcgySrL5yywg0z+WccPd+awDIuVRdm23o8P0uBlFrOw86Cm24f4a7jwb7IW4TyXR9V VSK+iz8yZhUnrqpP32G1xkpq2tNZmelVbuAUAGirD6F7YtnSCR5TT5LH1rkpKDE6cTIxOgD0EYmT NaFJ8CS0+vFSF4nYcyEnSSkoMTpnMTI5OgCSAuPEckATuoVZ1Op/sRmeC7uMAaZ0H2TdjMwd7WV0 zEQHvM3Dlo6fGOGhgu2Id6gRw6TZc7K+cT5V+EhbBYB06IpFMRmWaLOS+EZurQMJ7YQ8xeKuOpYA IjmcOKhBfWgQyidHw3cAQ7m/1J8GpLZD1Ho4GvqV/ThG/CkdnRQa7ykoMTp5MTI4OgIUacRjCFhM mhIfGJ44ms0EzRKZJUWangRqNYKLSYIDZlWiEDSApRpS8got+fXnxFLkHGfUl8TOfT/oXnHPxlPx h2pFuWFhOHT9hoHYDEFgu0ulhSF3jxhl6jUA3VsZV/LpoXp70KmtT5yqxUYQ6ycPGexo3R8X5bMQ hJlz6CzfKSgxOngyMToA1doGy3M1HcQrrNKXxD9W3XWl0S0pKSk= """ ) privateDSA_agentv3 = decodebytes( b"""\ AAAAB3NzaC1kc3MAAACBAJKQOsVERVDQIpANHH+JAAylo9LvFYmFFVMIuHFGlZpIL7sh3IMkqy+c ssINM/lnHD3fmsAyLlUXZtt6PD9LgZRazsPOgptuH+Gu48G+yFuE8l0fVVUivos/MmYVJ66qT99h tcZKatrTWZnpVW7gFABoqw+he2LZ0gkeU0+Sx9a5AAAAFQD0EYmTNaFJ8CS0+vFSF4nYcyEnSQAA AIEAkgLjxHJAE7qFWdTqf7EZngu7jAGmdB9k3YzMHe1ldMxEB7zNw5aOnxjhoYLtiHeoEcOk2XOy vnE+VfhIWwWAdOiKRTEZlmizkvhGbq0DCe2EPMXirjqWACI5nDioQX1oEMonR8N3AEO5v9SfBqS2 Q9R6OBr6lf04RvwpHZ0UGu8AAACAAhRpxGMIWEyaEh8YnjiazQTNEpklRZqeBGo1gotJggNmVaIQ NIClGlLyCi359efEUuQcZ9SXxM59P+hecc/GU/GHakW5YWE4dP2GgdgMQWC7S6WFIXePGGXqNQDd WxlX8umhenvQqa1PnKrFRhDrJw8Z7GjdHxflsxCEmXPoLN8AAAAVANXaBstzNR3EK6zSl8Q/Vt11 pdEt """ ) __all__ = [ "DSAData", "RSAData", "privateDSA_agentv3", "privateDSA_lsh", "privateDSA_openssh", "privateRSA_agentv3", "privateRSA_lsh", "privateRSA_openssh", "publicDSA_lsh", "publicDSA_openssh", "publicRSA_lsh", "publicRSA_openssh", "privateRSA_openssh_alternate", ] PK F��Z���ilz lz test_checkers.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.checkers}. """ try: import crypt except ImportError: cryptSkip = "cannot run without crypt module" else: cryptSkip = "" import os from base64 import encodebytes from collections import namedtuple from io import BytesIO from zope.interface.verify import verifyObject from twisted.cred.checkers import InMemoryUsernamePasswordDatabaseDontUse from twisted.cred.credentials import ( ISSHPrivateKey, IUsernamePassword, SSHPrivateKey, UsernamePassword, ) from twisted.cred.error import UnauthorizedLogin, UnhandledCredentials from twisted.python import util from twisted.python.failure import Failure from twisted.python.fakepwd import ShadowDatabase, UserDatabase from twisted.python.filepath import FilePath from twisted.python.reflect import requireModule from twisted.test.test_process import MockOS from twisted.trial.unittest import TestCase if requireModule("cryptography") and requireModule("pyasn1"): dependencySkip = "" from twisted.conch import checkers from twisted.conch.error import NotEnoughAuthentication, ValidPublicKey from twisted.conch.ssh import keys from twisted.conch.test import keydata else: dependencySkip = "can't run without cryptography and PyASN1" if getattr(os, "geteuid", None) is None: euidSkip = "Cannot run without effective UIDs (questionable)" else: euidSkip = "" class HelperTests(TestCase): """ Tests for helper functions L{verifyCryptedPassword}, L{_pwdGetByName} and L{_shadowGetByName}. """ skip = cryptSkip or dependencySkip def setUp(self): self.mockos = MockOS() def test_verifyCryptedPassword(self): """ L{verifyCryptedPassword} returns C{True} if the plaintext password passed to it matches the encrypted password passed to it. """ password = "secret string" salt = "salty" crypted = crypt.crypt(password, salt) self.assertTrue( checkers.verifyCryptedPassword(crypted, password), "{!r} supposed to be valid encrypted password for {!r}".format( crypted, password ), ) def test_verifyCryptedPasswordMD5(self): """ L{verifyCryptedPassword} returns True if the provided cleartext password matches the provided MD5 password hash. """ password = "password" salt = "$1$salt" crypted = crypt.crypt(password, salt) self.assertTrue( checkers.verifyCryptedPassword(crypted, password), "{!r} supposed to be valid encrypted password for {}".format( crypted, password ), ) def test_refuteCryptedPassword(self): """ L{verifyCryptedPassword} returns C{False} if the plaintext password passed to it does not match the encrypted password passed to it. """ password = "string secret" wrong = "secret string" crypted = crypt.crypt(password, password) self.assertFalse( checkers.verifyCryptedPassword(crypted, wrong), "{!r} not supposed to be valid encrypted password for {}".format( crypted, wrong ), ) def test_pwdGetByName(self): """ L{_pwdGetByName} returns a tuple of items from the UNIX /etc/passwd database if the L{pwd} module is present. """ userdb = UserDatabase() userdb.addUser("alice", "secrit", 1, 2, "first last", "/foo", "/bin/sh") self.patch(checkers, "pwd", userdb) self.assertEqual(checkers._pwdGetByName("alice"), userdb.getpwnam("alice")) def test_pwdGetByNameWithoutPwd(self): """ If the C{pwd} module isn't present, L{_pwdGetByName} returns L{None}. """ self.patch(checkers, "pwd", None) self.assertIsNone(checkers._pwdGetByName("alice")) def test_shadowGetByName(self): """ L{_shadowGetByName} returns a tuple of items from the UNIX /etc/shadow database if the L{spwd} is present. """ userdb = ShadowDatabase() userdb.addUser("bob", "passphrase", 1, 2, 3, 4, 5, 6, 7) self.patch(checkers, "spwd", userdb) self.mockos.euid = 2345 self.mockos.egid = 1234 self.patch(util, "os", self.mockos) self.assertEqual(checkers._shadowGetByName("bob"), userdb.getspnam("bob")) self.assertEqual(self.mockos.seteuidCalls, [0, 2345]) self.assertEqual(self.mockos.setegidCalls, [0, 1234]) def test_shadowGetByNameWithoutSpwd(self): """ L{_shadowGetByName} returns L{None} if C{spwd} is not present. """ self.patch(checkers, "spwd", None) self.assertIsNone(checkers._shadowGetByName("bob")) self.assertEqual(self.mockos.seteuidCalls, []) self.assertEqual(self.mockos.setegidCalls, []) class SSHPublicKeyDatabaseTests(TestCase): """ Tests for L{SSHPublicKeyDatabase}. """ skip = euidSkip or dependencySkip def setUp(self): self.checker = checkers.SSHPublicKeyDatabase() self.key1 = encodebytes(b"foobar") self.key2 = encodebytes(b"eggspam") self.content = b"t1 " + self.key1 + b" foo\nt2 " + self.key2 + b" egg\n" self.mockos = MockOS() self.mockos.path = FilePath(self.mktemp()) self.mockos.path.makedirs() self.patch(util, "os", self.mockos) self.sshDir = self.mockos.path.child(".ssh") self.sshDir.makedirs() userdb = UserDatabase() userdb.addUser( b"user", b"password", 1, 2, b"first last", self.mockos.path.path, b"/bin/shell", ) self.checker._userdb = userdb def test_deprecated(self): """ L{SSHPublicKeyDatabase} is deprecated as of version 15.0 """ warningsShown = self.flushWarnings(offendingFunctions=[self.setUp]) self.assertEqual(warningsShown[0]["category"], DeprecationWarning) self.assertEqual( warningsShown[0]["message"], "twisted.conch.checkers.SSHPublicKeyDatabase " "was deprecated in Twisted 15.0.0: Please use " "twisted.conch.checkers.SSHPublicKeyChecker, " "initialized with an instance of " "twisted.conch.checkers.UNIXAuthorizedKeysFiles instead.", ) self.assertEqual(len(warningsShown), 1) def _testCheckKey(self, filename): self.sshDir.child(filename).setContent(self.content) user = UsernamePassword(b"user", b"password") user.blob = b"foobar" self.assertTrue(self.checker.checkKey(user)) user.blob = b"eggspam" self.assertTrue(self.checker.checkKey(user)) user.blob = b"notallowed" self.assertFalse(self.checker.checkKey(user)) def test_checkKey(self): """ L{SSHPublicKeyDatabase.checkKey} should retrieve the content of the authorized_keys file and check the keys against that file. """ self._testCheckKey("authorized_keys") self.assertEqual(self.mockos.seteuidCalls, []) self.assertEqual(self.mockos.setegidCalls, []) def test_checkKey2(self): """ L{SSHPublicKeyDatabase.checkKey} should retrieve the content of the authorized_keys2 file and check the keys against that file. """ self._testCheckKey("authorized_keys2") self.assertEqual(self.mockos.seteuidCalls, []) self.assertEqual(self.mockos.setegidCalls, []) def test_checkKeyAsRoot(self): """ If the key file is readable, L{SSHPublicKeyDatabase.checkKey} should switch its uid/gid to the ones of the authenticated user. """ keyFile = self.sshDir.child("authorized_keys") keyFile.setContent(self.content) # Fake permission error by changing the mode keyFile.chmod(0o000) self.addCleanup(keyFile.chmod, 0o777) # And restore the right mode when seteuid is called savedSeteuid = self.mockos.seteuid def seteuid(euid): keyFile.chmod(0o777) return savedSeteuid(euid) self.mockos.euid = 2345 self.mockos.egid = 1234 self.patch(self.mockos, "seteuid", seteuid) self.patch(util, "os", self.mockos) user = UsernamePassword(b"user", b"password") user.blob = b"foobar" self.assertTrue(self.checker.checkKey(user)) self.assertEqual(self.mockos.seteuidCalls, [0, 1, 0, 2345]) self.assertEqual(self.mockos.setegidCalls, [2, 1234]) def test_requestAvatarId(self): """ L{SSHPublicKeyDatabase.requestAvatarId} should return the avatar id passed in if its C{_checkKey} method returns True. """ def _checkKey(ignored): return True self.patch(self.checker, "checkKey", _checkKey) credentials = SSHPrivateKey( b"test", b"ssh-rsa", keydata.publicRSA_openssh, b"foo", keys.Key.fromString(keydata.privateRSA_openssh).sign(b"foo"), ) d = self.checker.requestAvatarId(credentials) def _verify(avatarId): self.assertEqual(avatarId, b"test") return d.addCallback(_verify) def test_requestAvatarIdWithoutSignature(self): """ L{SSHPublicKeyDatabase.requestAvatarId} should raise L{ValidPublicKey} if the credentials represent a valid key without a signature. This tells the user that the key is valid for login, but does not actually allow that user to do so without a signature. """ def _checkKey(ignored): return True self.patch(self.checker, "checkKey", _checkKey) credentials = SSHPrivateKey( b"test", b"ssh-rsa", keydata.publicRSA_openssh, None, None ) d = self.checker.requestAvatarId(credentials) return self.assertFailure(d, ValidPublicKey) def test_requestAvatarIdInvalidKey(self): """ If L{SSHPublicKeyDatabase.checkKey} returns False, C{_cbRequestAvatarId} should raise L{UnauthorizedLogin}. """ def _checkKey(ignored): return False self.patch(self.checker, "checkKey", _checkKey) d = self.checker.requestAvatarId(None) return self.assertFailure(d, UnauthorizedLogin) def test_requestAvatarIdInvalidSignature(self): """ Valid keys with invalid signatures should cause L{SSHPublicKeyDatabase.requestAvatarId} to return a {UnauthorizedLogin} failure """ def _checkKey(ignored): return True self.patch(self.checker, "checkKey", _checkKey) credentials = SSHPrivateKey( b"test", b"ssh-rsa", keydata.publicRSA_openssh, b"foo", keys.Key.fromString(keydata.privateDSA_openssh).sign(b"foo"), ) d = self.checker.requestAvatarId(credentials) return self.assertFailure(d, UnauthorizedLogin) def test_requestAvatarIdNormalizeException(self): """ Exceptions raised while verifying the key should be normalized into an C{UnauthorizedLogin} failure. """ def _checkKey(ignored): return True self.patch(self.checker, "checkKey", _checkKey) credentials = SSHPrivateKey(b"test", None, b"blob", b"sigData", b"sig") d = self.checker.requestAvatarId(credentials) def _verifyLoggedException(failure): errors = self.flushLoggedErrors(keys.BadKeyError) self.assertEqual(len(errors), 1) return failure d.addErrback(_verifyLoggedException) return self.assertFailure(d, UnauthorizedLogin) class SSHProtocolCheckerTests(TestCase): """ Tests for L{SSHProtocolChecker}. """ skip = dependencySkip def test_registerChecker(self): """ L{SSHProcotolChecker.registerChecker} should add the given checker to the list of registered checkers. """ checker = checkers.SSHProtocolChecker() self.assertEqual(checker.credentialInterfaces, []) checker.registerChecker( checkers.SSHPublicKeyDatabase(), ) self.assertEqual(checker.credentialInterfaces, [ISSHPrivateKey]) self.assertIsInstance( checker.checkers[ISSHPrivateKey], checkers.SSHPublicKeyDatabase ) def test_registerCheckerWithInterface(self): """ If a specific interface is passed into L{SSHProtocolChecker.registerChecker}, that interface should be registered instead of what the checker specifies in credentialIntefaces. """ checker = checkers.SSHProtocolChecker() self.assertEqual(checker.credentialInterfaces, []) checker.registerChecker(checkers.SSHPublicKeyDatabase(), IUsernamePassword) self.assertEqual(checker.credentialInterfaces, [IUsernamePassword]) self.assertIsInstance( checker.checkers[IUsernamePassword], checkers.SSHPublicKeyDatabase ) def test_requestAvatarId(self): """ L{SSHProtocolChecker.requestAvatarId} should defer to one if its registered checkers to authenticate a user. """ checker = checkers.SSHProtocolChecker() passwordDatabase = InMemoryUsernamePasswordDatabaseDontUse() passwordDatabase.addUser(b"test", b"test") checker.registerChecker(passwordDatabase) d = checker.requestAvatarId(UsernamePassword(b"test", b"test")) def _callback(avatarId): self.assertEqual(avatarId, b"test") return d.addCallback(_callback) def test_requestAvatarIdWithNotEnoughAuthentication(self): """ If the client indicates that it is never satisfied, by always returning False from _areDone, then L{SSHProtocolChecker} should raise L{NotEnoughAuthentication}. """ checker = checkers.SSHProtocolChecker() def _areDone(avatarId): return False self.patch(checker, "areDone", _areDone) passwordDatabase = InMemoryUsernamePasswordDatabaseDontUse() passwordDatabase.addUser(b"test", b"test") checker.registerChecker(passwordDatabase) d = checker.requestAvatarId(UsernamePassword(b"test", b"test")) return self.assertFailure(d, NotEnoughAuthentication) def test_requestAvatarIdInvalidCredential(self): """ If the passed credentials aren't handled by any registered checker, L{SSHProtocolChecker} should raise L{UnhandledCredentials}. """ checker = checkers.SSHProtocolChecker() d = checker.requestAvatarId(UsernamePassword(b"test", b"test")) return self.assertFailure(d, UnhandledCredentials) def test_areDone(self): """ The default L{SSHProcotolChecker.areDone} should simply return True. """ self.assertTrue(checkers.SSHProtocolChecker().areDone(None)) class UNIXPasswordDatabaseTests(TestCase): """ Tests for L{UNIXPasswordDatabase}. """ skip = cryptSkip or dependencySkip def assertLoggedIn(self, d, username): """ Assert that the L{Deferred} passed in is called back with the value 'username'. This represents a valid login for this TestCase. NOTE: To work, this method's return value must be returned from the test method, or otherwise hooked up to the test machinery. @param d: a L{Deferred} from an L{IChecker.requestAvatarId} method. @type d: L{Deferred} @rtype: L{Deferred} """ result = [] d.addBoth(result.append) self.assertEqual(len(result), 1, "login incomplete") if isinstance(result[0], Failure): result[0].raiseException() self.assertEqual(result[0], username) def test_defaultCheckers(self): """ L{UNIXPasswordDatabase} with no arguments has checks the C{pwd} database and then the C{spwd} database. """ checker = checkers.UNIXPasswordDatabase() def crypted(username, password): salt = crypt.crypt(password, username) crypted = crypt.crypt(password, "$1$" + salt) return crypted pwd = UserDatabase() pwd.addUser( "alice", crypted("alice", "password"), 1, 2, "foo", "/foo", "/bin/sh" ) # x and * are convention for "look elsewhere for the password" pwd.addUser("bob", "x", 1, 2, "bar", "/bar", "/bin/sh") spwd = ShadowDatabase() spwd.addUser("alice", "wrong", 1, 2, 3, 4, 5, 6, 7) spwd.addUser("bob", crypted("bob", "password"), 8, 9, 10, 11, 12, 13, 14) self.patch(checkers, "pwd", pwd) self.patch(checkers, "spwd", spwd) mockos = MockOS() self.patch(util, "os", mockos) mockos.euid = 2345 mockos.egid = 1234 cred = UsernamePassword(b"alice", b"password") self.assertLoggedIn(checker.requestAvatarId(cred), b"alice") self.assertEqual(mockos.seteuidCalls, []) self.assertEqual(mockos.setegidCalls, []) cred.username = b"bob" self.assertLoggedIn(checker.requestAvatarId(cred), b"bob") self.assertEqual(mockos.seteuidCalls, [0, 2345]) self.assertEqual(mockos.setegidCalls, [0, 1234]) def assertUnauthorizedLogin(self, d): """ Asserts that the L{Deferred} passed in is erred back with an L{UnauthorizedLogin} L{Failure}. This reprsents an invalid login for this TestCase. NOTE: To work, this method's return value must be returned from the test method, or otherwise hooked up to the test machinery. @param d: a L{Deferred} from an L{IChecker.requestAvatarId} method. @type d: L{Deferred} @rtype: L{None} """ self.assertRaises( checkers.UnauthorizedLogin, self.assertLoggedIn, d, "bogus value" ) def test_passInCheckers(self): """ L{UNIXPasswordDatabase} takes a list of functions to check for UNIX user information. """ password = crypt.crypt("secret", "secret") userdb = UserDatabase() userdb.addUser("anybody", password, 1, 2, "foo", "/bar", "/bin/sh") checker = checkers.UNIXPasswordDatabase([userdb.getpwnam]) self.assertLoggedIn( checker.requestAvatarId(UsernamePassword(b"anybody", b"secret")), b"anybody" ) def test_verifyPassword(self): """ If the encrypted password provided by the getpwnam function is valid (verified by the L{verifyCryptedPassword} function), we callback the C{requestAvatarId} L{Deferred} with the username. """ def verifyCryptedPassword(crypted, pw): return crypted == pw def getpwnam(username): return [username, username] self.patch(checkers, "verifyCryptedPassword", verifyCryptedPassword) checker = checkers.UNIXPasswordDatabase([getpwnam]) credential = UsernamePassword(b"username", b"username") self.assertLoggedIn(checker.requestAvatarId(credential), b"username") def test_failOnKeyError(self): """ If the getpwnam function raises a KeyError, the login fails with an L{UnauthorizedLogin} exception. """ def getpwnam(username): raise KeyError(username) checker = checkers.UNIXPasswordDatabase([getpwnam]) credential = UsernamePassword(b"username", b"username") self.assertUnauthorizedLogin(checker.requestAvatarId(credential)) def test_failOnBadPassword(self): """ If the verifyCryptedPassword function doesn't verify the password, the login fails with an L{UnauthorizedLogin} exception. """ def verifyCryptedPassword(crypted, pw): return False def getpwnam(username): return [username, username] self.patch(checkers, "verifyCryptedPassword", verifyCryptedPassword) checker = checkers.UNIXPasswordDatabase([getpwnam]) credential = UsernamePassword(b"username", b"username") self.assertUnauthorizedLogin(checker.requestAvatarId(credential)) def test_loopThroughFunctions(self): """ UNIXPasswordDatabase.requestAvatarId loops through each getpwnam function associated with it and returns a L{Deferred} which fires with the result of the first one which returns a value other than None. ones do not verify the password. """ def verifyCryptedPassword(crypted, pw): return crypted == pw def getpwnam1(username): return [username, "not the password"] def getpwnam2(username): return [username, username] self.patch(checkers, "verifyCryptedPassword", verifyCryptedPassword) checker = checkers.UNIXPasswordDatabase([getpwnam1, getpwnam2]) credential = UsernamePassword(b"username", b"username") self.assertLoggedIn(checker.requestAvatarId(credential), b"username") def test_failOnSpecial(self): """ If the password returned by any function is C{""}, C{"x"}, or C{"*"} it is not compared against the supplied password. Instead it is skipped. """ pwd = UserDatabase() pwd.addUser("alice", "", 1, 2, "", "foo", "bar") pwd.addUser("bob", "x", 1, 2, "", "foo", "bar") pwd.addUser("carol", "*", 1, 2, "", "foo", "bar") self.patch(checkers, "pwd", pwd) checker = checkers.UNIXPasswordDatabase([checkers._pwdGetByName]) cred = UsernamePassword(b"alice", b"") self.assertUnauthorizedLogin(checker.requestAvatarId(cred)) cred = UsernamePassword(b"bob", b"x") self.assertUnauthorizedLogin(checker.requestAvatarId(cred)) cred = UsernamePassword(b"carol", b"*") self.assertUnauthorizedLogin(checker.requestAvatarId(cred)) class AuthorizedKeyFileReaderTests(TestCase): """ Tests for L{checkers.readAuthorizedKeyFile} """ skip = dependencySkip def test_ignoresComments(self): """ L{checkers.readAuthorizedKeyFile} does not attempt to turn comments into keys """ fileobj = BytesIO( b"# this comment is ignored\n" b"this is not\n" b"# this is again\n" b"and this is not" ) result = checkers.readAuthorizedKeyFile(fileobj, lambda x: x) self.assertEqual([b"this is not", b"and this is not"], list(result)) def test_ignoresLeadingWhitespaceAndEmptyLines(self): """ L{checkers.readAuthorizedKeyFile} ignores leading whitespace in lines, as well as empty lines """ fileobj = BytesIO( b""" # ignore not ignored """ ) result = checkers.readAuthorizedKeyFile(fileobj, parseKey=lambda x: x) self.assertEqual([b"not ignored"], list(result)) def test_ignoresUnparsableKeys(self): """ L{checkers.readAuthorizedKeyFile} does not raise an exception when a key fails to parse (raises a L{twisted.conch.ssh.keys.BadKeyError}), but rather just keeps going """ def failOnSome(line): if line.startswith(b"f"): raise keys.BadKeyError("failed to parse") return line fileobj = BytesIO(b"failed key\ngood key") result = checkers.readAuthorizedKeyFile(fileobj, parseKey=failOnSome) self.assertEqual([b"good key"], list(result)) class InMemorySSHKeyDBTests(TestCase): """ Tests for L{checkers.InMemorySSHKeyDB} """ skip = dependencySkip def test_implementsInterface(self): """ L{checkers.InMemorySSHKeyDB} implements L{checkers.IAuthorizedKeysDB} """ keydb = checkers.InMemorySSHKeyDB({b"alice": [b"key"]}) verifyObject(checkers.IAuthorizedKeysDB, keydb) def test_noKeysForUnauthorizedUser(self): """ If the user is not in the mapping provided to L{checkers.InMemorySSHKeyDB}, an empty iterator is returned by L{checkers.InMemorySSHKeyDB.getAuthorizedKeys} """ keydb = checkers.InMemorySSHKeyDB({b"alice": [b"keys"]}) self.assertEqual([], list(keydb.getAuthorizedKeys(b"bob"))) def test_allKeysForAuthorizedUser(self): """ If the user is in the mapping provided to L{checkers.InMemorySSHKeyDB}, an iterator with all the keys is returned by L{checkers.InMemorySSHKeyDB.getAuthorizedKeys} """ keydb = checkers.InMemorySSHKeyDB({b"alice": [b"a", b"b"]}) self.assertEqual([b"a", b"b"], list(keydb.getAuthorizedKeys(b"alice"))) class UNIXAuthorizedKeysFilesTests(TestCase): """ Tests for L{checkers.UNIXAuthorizedKeysFiles}. """ skip = dependencySkip def setUp(self): mockos = MockOS() mockos.path = FilePath(self.mktemp()) mockos.path.makedirs() self.userdb = UserDatabase() self.userdb.addUser( b"alice", b"password", 1, 2, b"alice lastname", mockos.path.path, b"/bin/shell", ) self.sshDir = mockos.path.child(".ssh") self.sshDir.makedirs() authorizedKeys = self.sshDir.child("authorized_keys") authorizedKeys.setContent(b"key 1\nkey 2") self.expectedKeys = [b"key 1", b"key 2"] def test_implementsInterface(self): """ L{checkers.UNIXAuthorizedKeysFiles} implements L{checkers.IAuthorizedKeysDB}. """ keydb = checkers.UNIXAuthorizedKeysFiles(self.userdb) verifyObject(checkers.IAuthorizedKeysDB, keydb) def test_noKeysForUnauthorizedUser(self): """ If the user is not in the user database provided to L{checkers.UNIXAuthorizedKeysFiles}, an empty iterator is returned by L{checkers.UNIXAuthorizedKeysFiles.getAuthorizedKeys}. """ keydb = checkers.UNIXAuthorizedKeysFiles(self.userdb, parseKey=lambda x: x) self.assertEqual([], list(keydb.getAuthorizedKeys("bob"))) def test_allKeysInAllAuthorizedFilesForAuthorizedUser(self): """ If the user is in the user database provided to L{checkers.UNIXAuthorizedKeysFiles}, an iterator with all the keys in C{~/.ssh/authorized_keys} and C{~/.ssh/authorized_keys2} is returned by L{checkers.UNIXAuthorizedKeysFiles.getAuthorizedKeys}. """ self.sshDir.child("authorized_keys2").setContent(b"key 3") keydb = checkers.UNIXAuthorizedKeysFiles(self.userdb, parseKey=lambda x: x) self.assertEqual( self.expectedKeys + [b"key 3"], list(keydb.getAuthorizedKeys(b"alice")) ) def test_ignoresNonexistantFile(self): """ L{checkers.UNIXAuthorizedKeysFiles.getAuthorizedKeys} returns only the keys in C{~/.ssh/authorized_keys} and C{~/.ssh/authorized_keys2} if they exist. """ keydb = checkers.UNIXAuthorizedKeysFiles(self.userdb, parseKey=lambda x: x) self.assertEqual(self.expectedKeys, list(keydb.getAuthorizedKeys(b"alice"))) def test_ignoresUnreadableFile(self): """ L{checkers.UNIXAuthorizedKeysFiles.getAuthorizedKeys} returns only the keys in C{~/.ssh/authorized_keys} and C{~/.ssh/authorized_keys2} if they are readable. """ self.sshDir.child("authorized_keys2").makedirs() keydb = checkers.UNIXAuthorizedKeysFiles(self.userdb, parseKey=lambda x: x) self.assertEqual(self.expectedKeys, list(keydb.getAuthorizedKeys(b"alice"))) _KeyDB = namedtuple("_KeyDB", ["getAuthorizedKeys"]) class _DummyException(Exception): """ Fake exception to be used for testing. """ pass class SSHPublicKeyCheckerTests(TestCase): """ Tests for L{checkers.SSHPublicKeyChecker}. """ skip = dependencySkip def setUp(self): self.credentials = SSHPrivateKey( b"alice", b"ssh-rsa", keydata.publicRSA_openssh, b"foo", keys.Key.fromString(keydata.privateRSA_openssh).sign(b"foo"), ) self.keydb = _KeyDB(lambda _: [keys.Key.fromString(keydata.publicRSA_openssh)]) self.checker = checkers.SSHPublicKeyChecker(self.keydb) def test_credentialsWithoutSignature(self): """ Calling L{checkers.SSHPublicKeyChecker.requestAvatarId} with credentials that do not have a signature fails with L{ValidPublicKey}. """ self.credentials.signature = None self.failureResultOf( self.checker.requestAvatarId(self.credentials), ValidPublicKey ) def test_credentialsWithBadKey(self): """ Calling L{checkers.SSHPublicKeyChecker.requestAvatarId} with credentials that have a bad key fails with L{keys.BadKeyError}. """ self.credentials.blob = b"" self.failureResultOf( self.checker.requestAvatarId(self.credentials), keys.BadKeyError ) def test_credentialsNoMatchingKey(self): """ If L{checkers.IAuthorizedKeysDB.getAuthorizedKeys} returns no keys that match the credentials, L{checkers.SSHPublicKeyChecker.requestAvatarId} fails with L{UnauthorizedLogin}. """ self.credentials.blob = keydata.publicDSA_openssh self.failureResultOf( self.checker.requestAvatarId(self.credentials), UnauthorizedLogin ) def test_credentialsInvalidSignature(self): """ Calling L{checkers.SSHPublicKeyChecker.requestAvatarId} with credentials that are incorrectly signed fails with L{UnauthorizedLogin}. """ self.credentials.signature = keys.Key.fromString( keydata.privateDSA_openssh ).sign(b"foo") self.failureResultOf( self.checker.requestAvatarId(self.credentials), UnauthorizedLogin ) def test_failureVerifyingKey(self): """ If L{keys.Key.verify} raises an exception, L{checkers.SSHPublicKeyChecker.requestAvatarId} fails with L{UnauthorizedLogin}. """ def fail(*args, **kwargs): raise _DummyException() self.patch(keys.Key, "verify", fail) self.failureResultOf( self.checker.requestAvatarId(self.credentials), UnauthorizedLogin ) self.flushLoggedErrors(_DummyException) def test_usernameReturnedOnSuccess(self): """ L{checker.SSHPublicKeyChecker.requestAvatarId}, if successful, callbacks with the username. """ d = self.checker.requestAvatarId(self.credentials) self.assertEqual(b"alice", self.successResultOf(d)) PK F��Z��� test_address.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{SSHTransportAddrress} in ssh/address.py """ from twisted.conch.ssh.address import SSHTransportAddress from twisted.internet.address import IPv4Address from twisted.internet.test.test_address import AddressTestCaseMixin from twisted.trial import unittest class SSHTransportAddressTests(unittest.TestCase, AddressTestCaseMixin): """ L{twisted.conch.ssh.address.SSHTransportAddress} is what Conch transports use to represent the other side of the SSH connection. This tests the basic functionality of that class (string representation, comparison, &c). """ def _stringRepresentation(self, stringFunction): """ The string representation of C{SSHTransportAddress} should be "SSHTransportAddress(<stringFunction on address>)". """ addr = self.buildAddress() stringValue = stringFunction(addr) addressValue = stringFunction(addr.address) self.assertEqual(stringValue, "SSHTransportAddress(%s)" % addressValue) def buildAddress(self): """ Create an arbitrary new C{SSHTransportAddress}. A new instance is created for each call, but always for the same address. """ return SSHTransportAddress(IPv4Address("TCP", "127.0.0.1", 22)) def buildDifferentAddress(self): """ Like C{buildAddress}, but with a different fixed address. """ return SSHTransportAddress(IPv4Address("TCP", "127.0.0.2", 22)) PK F��Z�SP� P� test_endpoints.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.endpoints}. """ import os.path from errno import ENOSYS from struct import pack from zope.interface import implementer from zope.interface.verify import verifyClass, verifyObject import hamcrest from twisted.conch.error import ConchError, HostKeyChanged, UserRejectedKey from twisted.conch.interfaces import IConchUser from twisted.cred.checkers import InMemoryUsernamePasswordDatabaseDontUse from twisted.cred.portal import Portal from twisted.internet.address import IPv4Address from twisted.internet.defer import CancelledError, Deferred, fail, succeed from twisted.internet.error import ( ConnectingCancelledError, ConnectionDone, ConnectionRefusedError, ProcessTerminated, ) from twisted.internet.interfaces import IAddress, IStreamClientEndpoint from twisted.internet.protocol import Factory, Protocol from twisted.logger import LogLevel, globalLogPublisher from twisted.python.compat import networkString from twisted.python.failure import Failure from twisted.python.filepath import FilePath from twisted.python.log import msg from twisted.python.reflect import requireModule from twisted.test.proto_helpers import EventLoggingObserver, MemoryReactorClock from twisted.trial.unittest import TestCase if requireModule("cryptography") and requireModule("pyasn1.type"): from twisted.conch.avatar import ConchUser from twisted.conch.checkers import InMemorySSHKeyDB, SSHPublicKeyChecker from twisted.conch.client.knownhosts import ConsoleUI, KnownHostsFile from twisted.conch.endpoints import ( AuthenticationFailed, SSHCommandAddress, SSHCommandClientEndpoint, _ExistingConnectionHelper, _ISSHConnectionCreator, _NewConnectionHelper, _ReadFile, ) from twisted.conch.ssh import common from twisted.conch.ssh.agent import SSHAgentServer from twisted.conch.ssh.channel import SSHChannel from twisted.conch.ssh.connection import SSHConnection from twisted.conch.ssh.factory import SSHFactory from twisted.conch.ssh.keys import Key from twisted.conch.ssh.transport import SSHClientTransport from twisted.conch.ssh.userauth import SSHUserAuthServer from twisted.conch.test.keydata import ( privateDSA_openssh, privateRSA_openssh, privateRSA_openssh_encrypted_aes, publicRSA_openssh, ) else: skip = "can't run w/o cryptography and pyasn1" SSHFactory = object # type: ignore[assignment,misc] SSHUserAuthServer = object # type: ignore[assignment,misc] SSHConnection = object # type: ignore[assignment,misc] Key = object # type: ignore[assignment,misc,misc] SSHChannel = object # type: ignore[assignment,misc] SSHAgentServer = object # type: ignore[assignment,misc] KnownHostsFile = object # type: ignore[assignment,misc] SSHPublicKeyChecker = object # type: ignore[assignment,misc] ConchUser = object # type: ignore[assignment,misc] from twisted.test.iosim import FakeTransport, connect from twisted.test.proto_helpers import StringTransport class AbortableFakeTransport(FakeTransport): """ A L{FakeTransport} with added C{abortConnection} support. """ aborted = False def abortConnection(self): """ Abort the connection in a fake manner. This should really be implemented in the underlying module. """ self.aborted = True class BrokenExecSession(SSHChannel): """ L{BrokenExecSession} is a session on which exec requests always fail. """ def request_exec(self, data): """ Fail all exec requests. @param data: Information about what is being executed. @type data: L{bytes} @return: C{0} to indicate failure @rtype: L{int} """ return 0 class WorkingExecSession(SSHChannel): """ L{WorkingExecSession} is a session on which exec requests always succeed. """ def request_exec(self, data): """ Succeed all exec requests. @param data: Information about what is being executed. @type data: L{bytes} @return: C{1} to indicate success @rtype: L{int} """ return 1 class UnsatisfiedExecSession(SSHChannel): """ L{UnsatisfiedExecSession} is a session on which exec requests are always delayed indefinitely, never succeeding or failing. """ def request_exec(self, data): """ Delay all exec requests indefinitely. @param data: Information about what is being executed. @type data: L{bytes} @return: A L{Deferred} which will never fire. @rtype: L{Deferred} """ return Deferred() class TrivialRealm: def __init__(self): self.channelLookup = {} def requestAvatar(self, avatarId, mind, *interfaces): avatar = ConchUser() avatar.channelLookup = self.channelLookup return (IConchUser, avatar, lambda: None) class AddressSpyFactory(Factory): address = None def buildProtocol(self, address): self.address = address return Factory.buildProtocol(self, address) class FixedResponseUI: def __init__(self, result): self.result = result def prompt(self, text): return succeed(self.result) def warn(self, text): pass class FakeClockSSHUserAuthServer(SSHUserAuthServer): # Delegate this setting to the factory to simplify tweaking it @property def attemptsBeforeDisconnect(self): """ Use the C{attemptsBeforeDisconnect} value defined by the factory to make it easier to override. """ return self.transport.factory.attemptsBeforeDisconnect @property def clock(self): """ Use the reactor defined by the factory, rather than the default global reactor, to simplify testing (by allowing an alternate implementation to be supplied by tests). """ return self.transport.factory.reactor class CommandFactory(SSHFactory): @property def publicKeys(self): return {b"ssh-rsa": Key.fromString(data=publicRSA_openssh)} @property def privateKeys(self): return {b"ssh-rsa": Key.fromString(data=privateRSA_openssh)} services = { b"ssh-userauth": FakeClockSSHUserAuthServer, b"ssh-connection": SSHConnection, } # Simplify the tests by disconnecting after the first authentication # failure. One attempt should be sufficient to test authentication success # and failure. There is an off-by-one in the implementation of this # feature in Conch, so set it to 0 in order to allow 1 attempt. attemptsBeforeDisconnect = 0 @implementer(IAddress) class MemoryAddress: pass @implementer(IStreamClientEndpoint) class SingleUseMemoryEndpoint: """ L{SingleUseMemoryEndpoint} is a client endpoint which allows one connection to be set up and then exposes an API for moving around bytes related to that connection. @ivar pump: L{None} until a connection is attempted, then a L{IOPump} instance associated with the protocol which is connected. @type pump: L{IOPump} """ def __init__(self, server): """ @param server: An L{IProtocol} provider to which the client will be connected. @type server: L{IProtocol} provider """ self.pump = None self._server = server def connect(self, factory): if self.pump is not None: raise Exception("SingleUseMemoryEndpoint was already used") try: protocol = factory.buildProtocol(MemoryAddress()) except BaseException: return fail() else: self.pump = connect( self._server, AbortableFakeTransport(self._server, isServer=True), protocol, AbortableFakeTransport(protocol, isServer=False), ) return succeed(protocol) class SSHCommandClientEndpointTestsMixin: """ Tests for L{SSHCommandClientEndpoint}, an L{IStreamClientEndpoint} implementations which connects a protocol with the stdin and stdout of a command running in an SSH session. These tests apply to L{SSHCommandClientEndpoint} whether it is constructed using L{SSHCommandClientEndpoint.existingConnection} or L{SSHCommandClientEndpoint.newConnection}. Subclasses must override L{create}, L{assertClientTransportState}, and L{finishConnection}. """ def setUp(self): self.hostname = b"ssh.example.com" self.port = 42022 self.user = b"user" self.password = b"password" self.reactor = MemoryReactorClock() self.realm = TrivialRealm() self.portal = Portal(self.realm) self.passwdDB = InMemoryUsernamePasswordDatabaseDontUse() self.passwdDB.addUser(self.user, self.password) self.portal.registerChecker(self.passwdDB) self.factory = CommandFactory() self.factory.reactor = self.reactor self.factory.portal = self.portal self.factory.doStart() self.addCleanup(self.factory.doStop) self.clientAddress = IPv4Address("TCP", "10.0.0.1", 12345) self.serverAddress = IPv4Address("TCP", "192.168.100.200", 54321) def create(self): """ Create and return a new L{SSHCommandClientEndpoint} to be tested. Override this to implement creation in an interesting way the endpoint. """ raise NotImplementedError( f"{self.__class__.__name__!r} did not implement create" ) def assertClientTransportState(self, client, immediateClose): """ Make an assertion about the connectedness of the given protocol's transport. Override this to implement either a check for the connection still being open or having been closed as appropriate. @param client: The client whose state is being checked. @param immediateClose: Boolean indicating whether the connection was closed immediately or not. """ raise NotImplementedError( "%r did not implement assertClientTransportState" % (self.__class__.__name__,) ) def finishConnection(self): """ Do any remaining work necessary to complete an in-memory connection attempted initiated using C{self.reactor}. """ raise NotImplementedError( f"{self.__class__.__name__!r} did not implement finishConnection" ) def connectedServerAndClient(self, serverFactory, clientFactory): """ Set up an in-memory connection between protocols created by C{serverFactory} and C{clientFactory}. @return: A three-tuple. The first element is the protocol created by C{serverFactory}. The second element is the protocol created by C{clientFactory}. The third element is the L{IOPump} connecting them. """ clientProtocol = clientFactory.buildProtocol(None) serverProtocol = serverFactory.buildProtocol(None) clientTransport = AbortableFakeTransport( clientProtocol, isServer=False, hostAddress=self.clientAddress, peerAddress=self.serverAddress, ) serverTransport = AbortableFakeTransport( serverProtocol, isServer=True, hostAddress=self.serverAddress, peerAddress=self.clientAddress, ) pump = connect(serverProtocol, serverTransport, clientProtocol, clientTransport) return serverProtocol, clientProtocol, pump def test_channelOpenFailure(self): """ If a channel cannot be opened on the authenticated SSH connection, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping the reason given by the server. """ endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() # The server logs the channel open failure - this is expected. errors = self.flushLoggedErrors(ConchError) self.assertIn("unknown channel", (errors[0].value.data, errors[0].value.value)) self.assertEqual(1, len(errors)) # Now deal with the results on the endpoint side. f = self.failureResultOf(connected) f.trap(ConchError) self.assertEqual(b"unknown channel", f.value.value) self.assertClientTransportState(client, False) def test_execFailure(self): """ If execution of the command fails, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping the reason given by the server. """ self.realm.channelLookup[b"session"] = BrokenExecSession endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() f = self.failureResultOf(connected) f.trap(ConchError) self.assertEqual("channel request failed", f.value.value) self.assertClientTransportState(client, False) def test_execCancelled(self): """ If execution of the command is cancelled via the L{Deferred} returned by L{SSHCommandClientEndpoint.connect}, the connection is closed immediately. """ self.realm.channelLookup[b"session"] = UnsatisfiedExecSession endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() connected.cancel() f = self.failureResultOf(connected) f.trap(CancelledError) self.assertClientTransportState(client, True) def test_buildProtocol(self): """ Once the necessary SSH actions have completed successfully, L{SSHCommandClientEndpoint.connect} uses the factory passed to it to construct a protocol instance by calling its C{buildProtocol} method with an address object representing the SSH connection and command executed. """ self.realm.channelLookup[b"session"] = WorkingExecSession endpoint = self.create() factory = AddressSpyFactory() factory.protocol = Protocol endpoint.connect(factory) server, client, pump = self.finishConnection() self.assertIsInstance(factory.address, SSHCommandAddress) self.assertEqual(server.transport.getHost(), factory.address.server) self.assertEqual(self.user, factory.address.username) self.assertEqual(b"/bin/ls -l", factory.address.command) def test_makeConnection(self): """ L{SSHCommandClientEndpoint} establishes an SSH connection, creates a channel in it, runs a command in that channel, and uses the protocol's C{makeConnection} to associate it with a protocol representing that command's stdin and stdout. """ self.realm.channelLookup[b"session"] = WorkingExecSession endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() protocol = self.successResultOf(connected) self.assertIsNotNone(protocol.transport) def test_dataReceived(self): """ After establishing the connection, when the command on the SSH server produces output, it is delivered to the protocol's C{dataReceived} method. """ self.realm.channelLookup[b"session"] = WorkingExecSession endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() protocol = self.successResultOf(connected) dataReceived = [] protocol.dataReceived = dataReceived.append # Figure out which channel on the connection this protocol is # associated with so the test can do a write on it. channelId = protocol.transport.id server.service.channels[channelId].write(b"hello, world") pump.pump() self.assertEqual(b"hello, world", b"".join(dataReceived)) def test_connectionLost(self): """ When the command closes the channel, the protocol's C{connectionLost} method is called. """ self.realm.channelLookup[b"session"] = WorkingExecSession endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() protocol = self.successResultOf(connected) connectionLost = [] protocol.connectionLost = connectionLost.append # Figure out which channel on the connection this protocol is # associated with so the test can do a write on it. channelId = protocol.transport.id server.service.channels[channelId].loseConnection() pump.pump() connectionLost[0].trap(ConnectionDone) self.assertClientTransportState(client, False) def _exitStatusTest(self, request, requestArg): """ Test handling of non-zero exit statuses or exit signals. """ self.realm.channelLookup[b"session"] = WorkingExecSession endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() protocol = self.successResultOf(connected) connectionLost = [] protocol.connectionLost = connectionLost.append # Figure out which channel on the connection this protocol is # associated with so the test can simulate command exit and # channel close. channelId = protocol.transport.id channel = server.service.channels[channelId] server.service.sendRequest(channel, request, requestArg) channel.loseConnection() pump.pump() self.assertClientTransportState(client, False) return connectionLost[0] def test_zeroExitCode(self): """ When the command exits with a non-zero status, the protocol's C{connectionLost} method is called with a L{Failure} wrapping an exception which encapsulates that status. """ exitCode = 0 exc = self._exitStatusTest(b"exit-status", pack(">L", exitCode)) exc.trap(ConnectionDone) def test_nonZeroExitStatus(self): """ When the command exits with a non-zero status, the protocol's C{connectionLost} method is called with a L{Failure} wrapping an exception which encapsulates that status. """ exitCode = 123 signal = None exc = self._exitStatusTest(b"exit-status", pack(">L", exitCode)) exc.trap(ProcessTerminated) self.assertEqual(exitCode, exc.value.exitCode) self.assertEqual(signal, exc.value.signal) def test_nonZeroExitSignal(self): """ When the command exits with a non-zero signal, the protocol's C{connectionLost} method is called with a L{Failure} wrapping an exception which encapsulates that status. Additional packet contents are logged at the C{info} level. """ logObserver = EventLoggingObserver() globalLogPublisher.addObserver(logObserver) self.addCleanup(globalLogPublisher.removeObserver, logObserver) exitCode = None signal = 15 # See https://tools.ietf.org/html/rfc4254#section-6.10 packet = b"".join( [ common.NS(b"TERM"), # Signal name (without "SIG" prefix); # string b"\x01", # Core dumped; boolean common.NS(b"message"), # Error message; string (UTF-8 encoded) common.NS(b"en-US"), # Language tag; string ] ) exc = self._exitStatusTest(b"exit-signal", packet) exc.trap(ProcessTerminated) self.assertEqual(exitCode, exc.value.exitCode) self.assertEqual(signal, exc.value.signal) logNamespace = "twisted.conch.endpoints._CommandChannel" hamcrest.assert_that( logObserver, hamcrest.has_item( hamcrest.has_entries( { "log_level": hamcrest.equal_to(LogLevel.info), "log_namespace": logNamespace, "shortSignalName": b"TERM", "coreDumped": True, "errorMessage": "message", "languageTag": b"en-US", }, ) ), ) def record(self, server, protocol, event, noArgs=False): """ Hook into and record events which happen to C{protocol}. @param server: The SSH server protocol over which C{protocol} is running. @type server: L{IProtocol} provider @param protocol: @param event: @param noArgs: """ # Figure out which channel the test is going to send data over # so we can look for it to arrive at the right place on the server. channelId = protocol.transport.id recorder = [] if noArgs: f = lambda: recorder.append(None) else: f = recorder.append setattr(server.service.channels[channelId], event, f) return recorder def test_write(self): """ The transport connected to the protocol has a C{write} method which sends bytes to the input of the command executing on the SSH server. """ self.realm.channelLookup[b"session"] = WorkingExecSession endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() protocol = self.successResultOf(connected) dataReceived = self.record(server, protocol, "dataReceived") protocol.transport.write(b"hello, world") pump.pump() self.assertEqual(b"hello, world", b"".join(dataReceived)) def test_writeSequence(self): """ The transport connected to the protocol has a C{writeSequence} method which sends bytes to the input of the command executing on the SSH server. """ self.realm.channelLookup[b"session"] = WorkingExecSession endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() protocol = self.successResultOf(connected) dataReceived = self.record(server, protocol, "dataReceived") protocol.transport.writeSequence([b"hello, world"]) pump.pump() self.assertEqual(b"hello, world", b"".join(dataReceived)) class NewConnectionTests(TestCase, SSHCommandClientEndpointTestsMixin): """ Tests for L{SSHCommandClientEndpoint} when using the C{newConnection} constructor. """ def setUp(self): """ Configure an SSH server with password authentication enabled for a well-known (to the tests) account. """ SSHCommandClientEndpointTestsMixin.setUp(self) # Make the server's host key available to be verified by the client. self.hostKeyPath = FilePath(self.mktemp()) self.knownHosts = KnownHostsFile(self.hostKeyPath) self.knownHosts.addHostKey(self.hostname, self.factory.publicKeys[b"ssh-rsa"]) self.knownHosts.addHostKey( networkString(self.serverAddress.host), self.factory.publicKeys[b"ssh-rsa"] ) self.knownHosts.save() def create(self): """ Create and return a new L{SSHCommandClientEndpoint} using the C{newConnection} constructor. """ return SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", self.user, self.hostname, self.port, password=self.password, knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) def finishConnection(self): """ Establish the first attempted TCP connection using the SSH server which C{self.factory} can create. """ return self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) def loseConnectionToServer(self, server, client, protocol, pump): """ Lose the connection to a server and pump the L{IOPump} sufficiently for the client to handle the lost connection. Asserts that the client disconnects its transport. @param server: The SSH server protocol over which C{protocol} is running. @type server: L{IProtocol} provider @param client: The SSH client protocol over which C{protocol} is running. @type client: L{IProtocol} provider @param protocol: The protocol created by calling connect on the ssh endpoint under test. @type protocol: L{IProtocol} provider @param pump: The L{IOPump} connecting client to server. @type pump: L{IOPump} """ closed = self.record(server, protocol, "closed", noArgs=True) protocol.transport.loseConnection() pump.pump() self.assertEqual([None], closed) # Let the last bit of network traffic flow. This lets the server's # close acknowledgement through, at which point the client can close # the overall SSH connection. pump.pump() # Given that the client transport is disconnecting, report the # disconnect from up to the ssh protocol. client.transport.reportDisconnect() def assertClientTransportState(self, client, immediateClose): """ Assert that the transport for the given protocol has been disconnected. L{SSHCommandClientEndpoint.newConnection} creates a new dedicated SSH connection and cleans it up after the command exits. """ # Nothing useful can be done with the connection at this point, so the # endpoint should close it. if immediateClose: self.assertTrue(client.transport.aborted) else: self.assertTrue(client.transport.disconnecting) def test_interface(self): """ L{SSHCommandClientEndpoint} instances provide L{IStreamClientEndpoint}. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"dummy command", b"dummy user", self.hostname, self.port ) self.assertTrue(verifyObject(IStreamClientEndpoint, endpoint)) def test_defaultPort(self): """ L{SSHCommandClientEndpoint} uses the default port number for SSH when the C{port} argument is not specified. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"dummy command", b"dummy user", self.hostname ) self.assertEqual(22, endpoint._creator.port) def test_specifiedPort(self): """ L{SSHCommandClientEndpoint} uses the C{port} argument if specified. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"dummy command", b"dummy user", self.hostname, port=2222 ) self.assertEqual(2222, endpoint._creator.port) def test_destination(self): """ L{SSHCommandClientEndpoint} uses the L{IReactorTCP} passed to it to attempt a connection to the host/port address also passed to it. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", self.user, self.hostname, self.port, password=self.password, knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol endpoint.connect(factory) host, port, factory, timeout, bindAddress = self.reactor.tcpClients[0] self.assertEqual(self.hostname, networkString(host)) self.assertEqual(self.port, port) self.assertEqual(1, len(self.reactor.tcpClients)) def test_connectionFailed(self): """ If a connection cannot be established, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} representing the reason for the connection setup failure. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", b"dummy user", self.hostname, self.port, knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol d = endpoint.connect(factory) factory = self.reactor.tcpClients[0][2] factory.clientConnectionFailed(None, Failure(ConnectionRefusedError())) self.failureResultOf(d).trap(ConnectionRefusedError) def test_userRejectedHostKey(self): """ If the L{KnownHostsFile} instance used to construct L{SSHCommandClientEndpoint} rejects the SSH public key presented by the server, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{UserRejectedKey}. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", b"dummy user", self.hostname, self.port, knownHosts=KnownHostsFile(self.mktemp()), ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) f = self.failureResultOf(connected) f.trap(UserRejectedKey) def test_mismatchedHostKey(self): """ If the SSH public key presented by the SSH server does not match the previously remembered key, as reported by the L{KnownHostsFile} instance use to construct the endpoint, for that server, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{HostKeyChanged}. """ firstKey = Key.fromString(privateRSA_openssh).public() knownHosts = KnownHostsFile(FilePath(self.mktemp())) knownHosts.addHostKey(networkString(self.serverAddress.host), firstKey) # Add a different RSA key with the same hostname differentKey = Key.fromString( privateRSA_openssh_encrypted_aes, passphrase=b"testxp" ).public() knownHosts.addHostKey(self.hostname, differentKey) # The UI may answer true to any questions asked of it; they should # make no difference, since a *mismatched* key is not even optionally # allowed to complete a connection. ui = FixedResponseUI(True) endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", b"dummy user", self.hostname, self.port, password=b"dummy password", knownHosts=knownHosts, ui=ui, ) factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) f = self.failureResultOf(connected) f.trap(HostKeyChanged) def test_connectionClosedBeforeSecure(self): """ If the connection closes at any point before the SSH transport layer has finished key exchange (ie, gotten to the point where we may attempt to authenticate), the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping the reason for the lost connection. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", b"dummy user", self.hostname, self.port, knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol d = endpoint.connect(factory) transport = StringTransport() factory = self.reactor.tcpClients[0][2] client = factory.buildProtocol(None) client.makeConnection(transport) client.connectionLost(Failure(ConnectionDone())) self.failureResultOf(d).trap(ConnectionDone) def test_connectionCancelledBeforeSecure(self): """ If the connection is cancelled before the SSH transport layer has finished key exchange (ie, gotten to the point where we may attempt to authenticate), the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{CancelledError} and the connection is aborted. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", b"dummy user", self.hostname, self.port, knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol d = endpoint.connect(factory) transport = AbortableFakeTransport(None, isServer=False) factory = self.reactor.tcpClients[0][2] client = factory.buildProtocol(None) client.makeConnection(transport) d.cancel() self.failureResultOf(d).trap(CancelledError) self.assertTrue(transport.aborted) # Make sure the connection closing doesn't result in unexpected # behavior when due to cancellation: client.connectionLost(Failure(ConnectionDone())) def test_connectionCancelledBeforeConnected(self): """ If the connection is cancelled before it finishes connecting, the connection attempt is stopped. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", b"dummy user", self.hostname, self.port, knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol d = endpoint.connect(factory) d.cancel() self.failureResultOf(d).trap(ConnectingCancelledError) self.assertTrue(self.reactor.connectors[0].stoppedConnecting) def test_passwordAuthenticationFailure(self): """ If the SSH server rejects the password presented during authentication, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{AuthenticationFailed}. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", b"dummy user", self.hostname, self.port, password=b"dummy password", knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) # For security, the server delays password authentication failure # response. Advance the simulation clock so the client sees the # failure. self.reactor.advance(server.service.passwordDelay) # Let the failure response traverse the "network" pump.flush() f = self.failureResultOf(connected) f.trap(AuthenticationFailed) # XXX Should assert something specific about the arguments of the # exception self.assertClientTransportState(client, False) def setupKeyChecker(self, portal, users): """ Create an L{ISSHPrivateKey} checker which recognizes C{users} and add it to C{portal}. @param portal: A L{Portal} to which to add the checker. @type portal: L{Portal} @param users: The users and their keys the checker will recognize. Keys are byte strings giving user names. Values are byte strings giving OpenSSH-formatted private keys. @type users: L{dict} """ mapping = {k: [Key.fromString(v).public()] for k, v in users.items()} checker = SSHPublicKeyChecker(InMemorySSHKeyDB(mapping)) portal.registerChecker(checker) def test_publicKeyAuthenticationFailure(self): """ If the SSH server rejects the key pair presented during authentication, the L{Deferred} returned by L{SSHCommandClientEndpoint.connect} fires with a L{Failure} wrapping L{AuthenticationFailed}. """ badKey = Key.fromString(privateRSA_openssh) self.setupKeyChecker(self.portal, {self.user: privateDSA_openssh}) endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", self.user, self.hostname, self.port, keys=[badKey], knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) f = self.failureResultOf(connected) f.trap(AuthenticationFailed) # XXX Should assert something specific about the arguments of the # exception # Nothing useful can be done with the connection at this point, so the # endpoint should close it. self.assertTrue(client.transport.disconnecting) def test_authenticationFallback(self): """ If the SSH server does not accept any of the specified SSH keys, the specified password is tried. """ badKey = Key.fromString(privateRSA_openssh) self.setupKeyChecker(self.portal, {self.user: privateDSA_openssh}) endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", self.user, self.hostname, self.port, keys=[badKey], password=self.password, knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) # Exercising fallback requires a failed authentication attempt. Allow # one. self.factory.attemptsBeforeDisconnect += 1 server, client, pump = self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) pump.pump() # The server logs the channel open failure - this is expected. errors = self.flushLoggedErrors(ConchError) self.assertIn("unknown channel", (errors[0].value.data, errors[0].value.value)) self.assertEqual(1, len(errors)) # Now deal with the results on the endpoint side. f = self.failureResultOf(connected) f.trap(ConchError) self.assertEqual(b"unknown channel", f.value.value) # Nothing useful can be done with the connection at this point, so the # endpoint should close it. self.assertTrue(client.transport.disconnecting) def test_publicKeyAuthentication(self): """ If L{SSHCommandClientEndpoint} is initialized with any private keys, it will try to use them to authenticate with the SSH server. """ key = Key.fromString(privateDSA_openssh) self.setupKeyChecker(self.portal, {self.user: privateDSA_openssh}) self.realm.channelLookup[b"session"] = WorkingExecSession endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", self.user, self.hostname, self.port, keys=[key], knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) protocol = self.successResultOf(connected) self.assertIsNotNone(protocol.transport) def test_skipPasswordAuthentication(self): """ If the password is not specified, L{SSHCommandClientEndpoint} doesn't try it as an authentication mechanism. """ endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", self.user, self.hostname, self.port, knownHosts=self.knownHosts, ui=FixedResponseUI(False), ) factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) pump.pump() # Now deal with the results on the endpoint side. f = self.failureResultOf(connected) f.trap(AuthenticationFailed) # Nothing useful can be done with the connection at this point, so the # endpoint should close it. self.assertTrue(client.transport.disconnecting) def test_agentAuthentication(self): """ If L{SSHCommandClientEndpoint} is initialized with an L{SSHAgentClient}, the agent is used to authenticate with the SSH server. Once the connection with the SSH server has concluded, the connection to the agent is disconnected. """ key = Key.fromString(privateRSA_openssh) agentServer = SSHAgentServer() agentServer.factory = Factory() agentServer.factory.keys = {key.blob(): (key, b"")} self.setupKeyChecker(self.portal, {self.user: privateRSA_openssh}) agentEndpoint = SingleUseMemoryEndpoint(agentServer) endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", self.user, self.hostname, self.port, knownHosts=self.knownHosts, ui=FixedResponseUI(False), agentEndpoint=agentEndpoint, ) self.realm.channelLookup[b"session"] = WorkingExecSession factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) # Let the agent client talk with the agent server and the ssh client # talk with the ssh server. for i in range(14): agentEndpoint.pump.pump() pump.pump() protocol = self.successResultOf(connected) self.assertIsNotNone(protocol.transport) # Ensure the connection with the agent is cleaned up after the # connection with the server is lost. self.loseConnectionToServer(server, client, protocol, pump) self.assertTrue(client.transport.disconnecting) self.assertTrue(agentEndpoint.pump.clientIO.disconnecting) def test_loseConnection(self): """ The transport connected to the protocol has a C{loseConnection} method which causes the channel in which the command is running to close and the overall connection to be closed. """ self.realm.channelLookup[b"session"] = WorkingExecSession endpoint = self.create() factory = Factory() factory.protocol = Protocol connected = endpoint.connect(factory) server, client, pump = self.finishConnection() protocol = self.successResultOf(connected) self.loseConnectionToServer(server, client, protocol, pump) # Nothing useful can be done with the connection at this point, so the # endpoint should close it. self.assertTrue(client.transport.disconnecting) class ExistingConnectionTests(TestCase, SSHCommandClientEndpointTestsMixin): """ Tests for L{SSHCommandClientEndpoint} when using the C{existingConnection} constructor. """ def setUp(self): """ Configure an SSH server with password authentication enabled for a well-known (to the tests) account. """ SSHCommandClientEndpointTestsMixin.setUp(self) knownHosts = KnownHostsFile(FilePath(self.mktemp())) knownHosts.addHostKey(self.hostname, self.factory.publicKeys[b"ssh-rsa"]) knownHosts.addHostKey( networkString(self.serverAddress.host), self.factory.publicKeys[b"ssh-rsa"] ) self.endpoint = SSHCommandClientEndpoint.newConnection( self.reactor, b"/bin/ls -l", self.user, self.hostname, self.port, password=self.password, knownHosts=knownHosts, ui=FixedResponseUI(False), ) def create(self): """ Create and return a new L{SSHCommandClientEndpoint} using the C{existingConnection} constructor. """ factory = Factory() factory.protocol = Protocol connected = self.endpoint.connect(factory) # Please, let me in. This kinda sucks. channelLookup = self.realm.channelLookup.copy() try: self.realm.channelLookup[b"session"] = WorkingExecSession server, client, pump = self.connectedServerAndClient( self.factory, self.reactor.tcpClients[0][2] ) finally: self.realm.channelLookup.clear() self.realm.channelLookup.update(channelLookup) self._server = server self._client = client self._pump = pump protocol = self.successResultOf(connected) connection = protocol.transport.conn return SSHCommandClientEndpoint.existingConnection(connection, b"/bin/ls -l") def finishConnection(self): """ Give back the connection established in L{create} over which the new command channel being tested will exchange data. """ # The connection was set up and the first command channel set up, but # some more I/O needs to happen for the second command channel to be # ready. Make that I/O happen before giving back the objects. self._pump.pump() self._pump.pump() self._pump.pump() self._pump.pump() return self._server, self._client, self._pump def assertClientTransportState(self, client, immediateClose): """ Assert that the transport for the given protocol is still connected. L{SSHCommandClientEndpoint.existingConnection} re-uses an SSH connected created by some other code, so other code is responsible for cleaning it up. """ self.assertFalse(client.transport.disconnecting) self.assertFalse(client.transport.aborted) class ExistingConnectionHelperTests(TestCase): """ Tests for L{_ExistingConnectionHelper}. """ def test_interface(self): """ L{_ExistingConnectionHelper} implements L{_ISSHConnectionCreator}. """ self.assertTrue(verifyClass(_ISSHConnectionCreator, _ExistingConnectionHelper)) def test_secureConnection(self): """ L{_ExistingConnectionHelper.secureConnection} returns a L{Deferred} which fires with whatever object was fed to L{_ExistingConnectionHelper.__init__}. """ result = object() helper = _ExistingConnectionHelper(result) self.assertIs(result, self.successResultOf(helper.secureConnection())) def test_cleanupConnectionNotImmediately(self): """ L{_ExistingConnectionHelper.cleanupConnection} does nothing to the existing connection if called with C{immediate} set to C{False}. """ helper = _ExistingConnectionHelper(object()) # Bit hard to test nothing happens. However, since object() has no # relevant methods or attributes, if the code is incorrect we can # expect an AttributeError. helper.cleanupConnection(object(), False) def test_cleanupConnectionImmediately(self): """ L{_ExistingConnectionHelper.cleanupConnection} does nothing to the existing connection if called with C{immediate} set to C{True}. """ helper = _ExistingConnectionHelper(object()) # Bit hard to test nothing happens. However, since object() has no # relevant methods or attributes, if the code is incorrect we can # expect an AttributeError. helper.cleanupConnection(object(), True) class _PTYPath: """ A L{FilePath}-like object which can be opened to create a L{_ReadFile} with certain contents. """ def __init__(self, contents): """ @param contents: L{bytes} which will be the contents of the L{_ReadFile} this path can open. """ self.contents = contents def open(self, mode): """ If the mode is r+, return a L{_ReadFile} with the contents given to this path's initializer. @raise OSError: If the mode is unsupported. @return: A L{_ReadFile} instance """ if mode == "rb+": return _ReadFile(self.contents) raise OSError(ENOSYS, "Function not implemented") class NewConnectionHelperTests(TestCase): """ Tests for L{_NewConnectionHelper}. """ def test_interface(self): """ L{_NewConnectionHelper} implements L{_ISSHConnectionCreator}. """ self.assertTrue(verifyClass(_ISSHConnectionCreator, _NewConnectionHelper)) def test_defaultPath(self): """ The default I{known_hosts} path is I{~/.ssh/known_hosts}. """ self.assertEqual("~/.ssh/known_hosts", _NewConnectionHelper._KNOWN_HOSTS) def test_defaultKnownHosts(self): """ L{_NewConnectionHelper._knownHosts} is used to create a L{KnownHostsFile} if one is not passed to the initializer. """ result = object() self.patch(_NewConnectionHelper, "_knownHosts", lambda cls: result) helper = _NewConnectionHelper( None, None, None, None, None, None, None, None, None, None ) self.assertIs(result, helper.knownHosts) def test_readExisting(self): """ Existing entries in the I{known_hosts} file are reflected by the L{KnownHostsFile} created by L{_NewConnectionHelper} when none is supplied to it. """ key = CommandFactory().publicKeys[b"ssh-rsa"] path = FilePath(self.mktemp()) knownHosts = KnownHostsFile(path) knownHosts.addHostKey(b"127.0.0.1", key) knownHosts.save() msg(f"Created known_hosts file at {path.path!r}") # Unexpand ${HOME} to make sure ~ syntax is respected. home = os.path.expanduser("~/") default = path.path.replace(home, "~/") self.patch(_NewConnectionHelper, "_KNOWN_HOSTS", default) msg(f"Patched _KNOWN_HOSTS with {default!r}") loaded = _NewConnectionHelper._knownHosts() self.assertTrue(loaded.hasHostKey(b"127.0.0.1", key)) def test_defaultConsoleUI(self): """ If L{None} is passed for the C{ui} parameter to L{_NewConnectionHelper}, a L{ConsoleUI} is used. """ helper = _NewConnectionHelper( None, None, None, None, None, None, None, None, None, None ) self.assertIsInstance(helper.ui, ConsoleUI) def test_ttyConsoleUI(self): """ If L{None} is passed for the C{ui} parameter to L{_NewConnectionHelper} and /dev/tty is available, the L{ConsoleUI} used is associated with /dev/tty. """ tty = _PTYPath(b"yes") helper = _NewConnectionHelper( None, None, None, None, None, None, None, None, None, None, tty ) result = self.successResultOf(helper.ui.prompt(b"does this work?")) self.assertTrue(result) def test_nottyUI(self): """ If L{None} is passed for the C{ui} parameter to L{_NewConnectionHelper} and /dev/tty is not available, the L{ConsoleUI} used is associated with some file which always produces a C{b"no"} response. """ tty = FilePath(self.mktemp()) helper = _NewConnectionHelper( None, None, None, None, None, None, None, None, None, None, tty ) result = self.successResultOf(helper.ui.prompt(b"did this break?")) self.assertFalse(result) def test_defaultTTYFilename(self): """ If not passed the name of a tty in the filesystem, L{_NewConnectionHelper} uses C{b"/dev/tty"}. """ helper = _NewConnectionHelper( None, None, None, None, None, None, None, None, None, None ) self.assertEqual(FilePath(b"/dev/tty"), helper.tty) def test_cleanupConnectionNotImmediately(self): """ L{_NewConnectionHelper.cleanupConnection} closes the transport cleanly if called with C{immediate} set to C{False}. """ helper = _NewConnectionHelper( None, None, None, None, None, None, None, None, None, None ) connection = SSHConnection() connection.transport = StringTransport() helper.cleanupConnection(connection, False) self.assertTrue(connection.transport.disconnecting) def test_cleanupConnectionImmediately(self): """ L{_NewConnectionHelper.cleanupConnection} closes the transport with C{abortConnection} if called with C{immediate} set to C{True}. """ class Abortable: aborted = False def abortConnection(self): """ Abort the connection. """ self.aborted = True helper = _NewConnectionHelper( None, None, None, None, None, None, None, None, None, None ) connection = SSHConnection() connection.transport = SSHClientTransport() connection.transport.transport = Abortable() helper.cleanupConnection(connection, True) self.assertTrue(connection.transport.transport.aborted) PK F��Z�|�ї/ �/ test_channel.pynu �[��� # Copyright Twisted Matrix Laboratories. # See LICENSE for details. """ Test ssh/channel.py. """ from unittest import skipIf from zope.interface.verify import verifyObject try: from twisted.conch.ssh import channel from twisted.conch.ssh.address import SSHTransportAddress from twisted.conch.ssh.service import SSHService from twisted.conch.ssh.transport import SSHServerTransport from twisted.internet import interfaces from twisted.internet.address import IPv4Address from twisted.test.proto_helpers import StringTransport skipTest = "" except ImportError: skipTest = "Conch SSH not supported." SSHService = object # type: ignore[assignment,misc] from twisted.trial.unittest import TestCase class MockConnection(SSHService): """ A mock for twisted.conch.ssh.connection.SSHConnection. Record the data that channels send, and when they try to close the connection. @ivar data: a L{dict} mapping channel id #s to lists of data sent by that channel. @ivar extData: a L{dict} mapping channel id #s to lists of 2-tuples (extended data type, data) sent by that channel. @ivar closes: a L{dict} mapping channel id #s to True if that channel sent a close message. """ def __init__(self): self.data = {} self.extData = {} self.closes = {} def logPrefix(self): """ Return our logging prefix. """ return "MockConnection" def sendData(self, channel, data): """ Record the sent data. """ self.data.setdefault(channel, []).append(data) def sendExtendedData(self, channel, type, data): """ Record the sent extended data. """ self.extData.setdefault(channel, []).append((type, data)) def sendClose(self, channel): """ Record that the channel sent a close message. """ self.closes[channel] = True def connectSSHTransport(service, hostAddress=None, peerAddress=None): """ Connect a SSHTransport which is already connected to a remote peer to the channel under test. @param service: Service used over the connected transport. @type service: L{SSHService} @param hostAddress: Local address of the connected transport. @type hostAddress: L{interfaces.IAddress} @param peerAddress: Remote address of the connected transport. @type peerAddress: L{interfaces.IAddress} """ transport = SSHServerTransport() transport.makeConnection( StringTransport(hostAddress=hostAddress, peerAddress=peerAddress) ) transport.setService(service) @skipIf(skipTest, skipTest) class ChannelTests(TestCase): """ Tests for L{SSHChannel}. """ def setUp(self): """ Initialize the channel. remoteMaxPacket is 10 so that data is able to be sent (the default of 0 means no data is sent because no packets are made). """ self.conn = MockConnection() self.channel = channel.SSHChannel(conn=self.conn, remoteMaxPacket=10) self.channel.name = b"channel" def test_interface(self): """ L{SSHChannel} instances provide L{interfaces.ITransport}. """ self.assertTrue(verifyObject(interfaces.ITransport, self.channel)) def test_init(self): """ Test that SSHChannel initializes correctly. localWindowSize defaults to 131072 (2**17) and localMaxPacket to 32768 (2**15) as reasonable defaults (what OpenSSH uses for those variables). The values in the second set of assertions are meaningless; they serve only to verify that the instance variables are assigned in the correct order. """ c = channel.SSHChannel(conn=self.conn) self.assertEqual(c.localWindowSize, 131072) self.assertEqual(c.localWindowLeft, 131072) self.assertEqual(c.localMaxPacket, 32768) self.assertEqual(c.remoteWindowLeft, 0) self.assertEqual(c.remoteMaxPacket, 0) self.assertEqual(c.conn, self.conn) self.assertIsNone(c.data) self.assertIsNone(c.avatar) c2 = channel.SSHChannel(1, 2, 3, 4, 5, 6, 7) self.assertEqual(c2.localWindowSize, 1) self.assertEqual(c2.localWindowLeft, 1) self.assertEqual(c2.localMaxPacket, 2) self.assertEqual(c2.remoteWindowLeft, 3) self.assertEqual(c2.remoteMaxPacket, 4) self.assertEqual(c2.conn, 5) self.assertEqual(c2.data, 6) self.assertEqual(c2.avatar, 7) def test_str(self): """ Test that str(SSHChannel) works gives the channel name and local and remote windows at a glance.. """ self.assertEqual(str(self.channel), "<SSHChannel channel (lw 131072 rw 0)>") self.assertEqual( str(channel.SSHChannel(localWindow=1)), "<SSHChannel None (lw 1 rw 0)>" ) def test_bytes(self): """ Test that bytes(SSHChannel) works, gives the channel name and local and remote windows at a glance.. """ self.assertEqual( self.channel.__bytes__(), b"<SSHChannel channel (lw 131072 rw 0)>" ) self.assertEqual( channel.SSHChannel(localWindow=1).__bytes__(), b"<SSHChannel None (lw 1 rw 0)>", ) def test_logPrefix(self): """ Test that SSHChannel.logPrefix gives the name of the channel, the local channel ID and the underlying connection. """ self.assertEqual( self.channel.logPrefix(), "SSHChannel channel (unknown) on MockConnection" ) def test_addWindowBytes(self): """ Test that addWindowBytes adds bytes to the window and resumes writing if it was paused. """ cb = [False] def stubStartWriting(): cb[0] = True self.channel.startWriting = stubStartWriting self.channel.write(b"test") self.channel.writeExtended(1, b"test") self.channel.addWindowBytes(50) self.assertEqual(self.channel.remoteWindowLeft, 50 - 4 - 4) self.assertTrue(self.channel.areWriting) self.assertTrue(cb[0]) self.assertEqual(self.channel.buf, b"") self.assertEqual(self.conn.data[self.channel], [b"test"]) self.assertEqual(self.channel.extBuf, []) self.assertEqual(self.conn.extData[self.channel], [(1, b"test")]) cb[0] = False self.channel.addWindowBytes(20) self.assertFalse(cb[0]) self.channel.write(b"a" * 80) self.channel.loseConnection() self.channel.addWindowBytes(20) self.assertFalse(cb[0]) def test_requestReceived(self): """ Test that requestReceived handles requests by dispatching them to request_* methods. """ self.channel.request_test_method = lambda data: data == b"" self.assertTrue(self.channel.requestReceived(b"test-method", b"")) self.assertFalse(self.channel.requestReceived(b"test-method", b"a")) self.assertFalse(self.channel.requestReceived(b"bad-method", b"")) def test_closeReceieved(self): """ Test that the default closeReceieved closes the connection. """ self.assertFalse(self.channel.closing) self.channel.closeReceived() self.assertTrue(self.channel.closing) def test_write(self): """ Test that write handles data correctly. Send data up to the size of the remote window, splitting the data into packets of length remoteMaxPacket. """ cb = [False] def stubStopWriting(): cb[0] = True # no window to start with self.channel.stopWriting = stubStopWriting self.channel.write(b"d") self.channel.write(b"a") self.assertFalse(self.channel.areWriting) self.assertTrue(cb[0]) # regular write self.channel.addWindowBytes(20) self.channel.write(b"ta") data = self.conn.data[self.channel] self.assertEqual(data, [b"da", b"ta"]) self.assertEqual(self.channel.remoteWindowLeft, 16) # larger than max packet self.channel.write(b"12345678901") self.assertEqual(data, [b"da", b"ta", b"1234567890", b"1"]) self.assertEqual(self.channel.remoteWindowLeft, 5) # running out of window cb[0] = False self.channel.write(b"123456") self.assertFalse(self.channel.areWriting) self.assertTrue(cb[0]) self.assertEqual(data, [b"da", b"ta", b"1234567890", b"1", b"12345"]) self.assertEqual(self.channel.buf, b"6") self.assertEqual(self.channel.remoteWindowLeft, 0) def test_writeExtended(self): """ Test that writeExtended handles data correctly. Send extended data up to the size of the window, splitting the extended data into packets of length remoteMaxPacket. """ cb = [False] def stubStopWriting(): cb[0] = True # no window to start with self.channel.stopWriting = stubStopWriting self.channel.writeExtended(1, b"d") self.channel.writeExtended(1, b"a") self.channel.writeExtended(2, b"t") self.assertFalse(self.channel.areWriting) self.assertTrue(cb[0]) # regular write self.channel.addWindowBytes(20) self.channel.writeExtended(2, b"a") data = self.conn.extData[self.channel] self.assertEqual(data, [(1, b"da"), (2, b"t"), (2, b"a")]) self.assertEqual(self.channel.remoteWindowLeft, 16) # larger than max packet self.channel.writeExtended(3, b"12345678901") self.assertEqual( data, [(1, b"da"), (2, b"t"), (2, b"a"), (3, b"1234567890"), (3, b"1")] ) self.assertEqual(self.channel.remoteWindowLeft, 5) # running out of window cb[0] = False self.channel.writeExtended(4, b"123456") self.assertFalse(self.channel.areWriting) self.assertTrue(cb[0]) self.assertEqual( data, [ (1, b"da"), (2, b"t"), (2, b"a"), (3, b"1234567890"), (3, b"1"), (4, b"12345"), ], ) self.assertEqual(self.channel.extBuf, [[4, b"6"]]) self.assertEqual(self.channel.remoteWindowLeft, 0) def test_writeSequence(self): """ Test that writeSequence is equivalent to write(''.join(sequece)). """ self.channel.addWindowBytes(20) self.channel.writeSequence(b"%d" % (i,) for i in range(10)) self.assertEqual(self.conn.data[self.channel], [b"0123456789"]) def test_loseConnection(self): """ Tesyt that loseConnection() doesn't close the channel until all the data is sent. """ self.channel.write(b"data") self.channel.writeExtended(1, b"datadata") self.channel.loseConnection() self.assertIsNone(self.conn.closes.get(self.channel)) self.channel.addWindowBytes(4) # send regular data self.assertIsNone(self.conn.closes.get(self.channel)) self.channel.addWindowBytes(8) # send extended data self.assertTrue(self.conn.closes.get(self.channel)) def test_getPeer(self): """ L{SSHChannel.getPeer} returns the same object as the underlying transport's C{getPeer} method returns. """ peer = IPv4Address("TCP", "192.168.0.1", 54321) connectSSHTransport(service=self.channel.conn, peerAddress=peer) self.assertEqual(SSHTransportAddress(peer), self.channel.getPeer()) def test_getHost(self): """ L{SSHChannel.getHost} returns the same object as the underlying transport's C{getHost} method returns. """ host = IPv4Address("TCP", "127.0.0.1", 12345) connectSSHTransport(service=self.channel.conn, hostAddress=host) self.assertEqual(SSHTransportAddress(host), self.channel.getHost()) PK F��Z:���A A test_window.pynu �[��� """ Tests for the insults windowing module, L{twisted.conch.insults.window}. """ from twisted.conch.insults.window import ScrolledArea, TextOutput, TopWindow from twisted.trial.unittest import TestCase class TopWindowTests(TestCase): """ Tests for L{TopWindow}, the root window container class. """ def test_paintScheduling(self): """ Verify that L{TopWindow.repaint} schedules an actual paint to occur using the scheduling object passed to its initializer. """ paints = [] scheduled = [] root = TopWindow(lambda: paints.append(None), scheduled.append) # Nothing should have happened yet. self.assertEqual(paints, []) self.assertEqual(scheduled, []) # Cause a paint to be scheduled. root.repaint() self.assertEqual(paints, []) self.assertEqual(len(scheduled), 1) # Do another one to verify nothing else happens as long as the previous # one is still pending. root.repaint() self.assertEqual(paints, []) self.assertEqual(len(scheduled), 1) # Run the actual paint call. scheduled.pop()() self.assertEqual(len(paints), 1) self.assertEqual(scheduled, []) # Do one more to verify that now that the previous one is finished # future paints will succeed. root.repaint() self.assertEqual(len(paints), 1) self.assertEqual(len(scheduled), 1) class ScrolledAreaTests(TestCase): """ Tests for L{ScrolledArea}, a widget which creates a viewport containing another widget and can reposition that viewport using scrollbars. """ def test_parent(self): """ The parent of the widget passed to L{ScrolledArea} is set to a new L{Viewport} created by the L{ScrolledArea} which itself has the L{ScrolledArea} instance as its parent. """ widget = TextOutput() scrolled = ScrolledArea(widget) self.assertIs(widget.parent, scrolled._viewport) self.assertIs(scrolled._viewport.parent, scrolled) PK F��ZҢB��3 �3 test_manhole.pynu �[��� # -*- test-case-name: twisted.conch.test.test_manhole -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. # pylint: disable=I0011,W9401,W9402 """ Tests for L{twisted.conch.manhole}. """ import traceback from typing import Optional ssh: Optional[bool] = None from twisted.conch import manhole from twisted.conch.insults import insults from twisted.conch.test.test_recvline import ( _SSHMixin, _StdioMixin, _TelnetMixin, ssh, stdio, ) from twisted.internet import defer, error from twisted.test.proto_helpers import StringTransport from twisted.trial import unittest def determineDefaultFunctionName(): """ Return the string used by Python as the name for code objects which are compiled from interactive input or at the top-level of modules. """ try: 1 // 0 except BaseException: # The last frame is this function. The second to last frame is this # function's caller, which is module-scope, which is what we want, # so -2. return traceback.extract_stack()[-2][2] defaultFunctionName = determineDefaultFunctionName() class ManholeInterpreterTests(unittest.TestCase): """ Tests for L{manhole.ManholeInterpreter}. """ def test_resetBuffer(self): """ L{ManholeInterpreter.resetBuffer} should empty the input buffer. """ interpreter = manhole.ManholeInterpreter(None) interpreter.buffer.extend(["1", "2"]) interpreter.resetBuffer() self.assertFalse(interpreter.buffer) class ManholeProtocolTests(unittest.TestCase): """ Tests for L{manhole.Manhole}. """ def test_interruptResetsInterpreterBuffer(self): """ L{manhole.Manhole.handle_INT} should cause the interpreter input buffer to be reset. """ transport = StringTransport() terminal = insults.ServerProtocol(manhole.Manhole) terminal.makeConnection(transport) protocol = terminal.terminalProtocol interpreter = protocol.interpreter interpreter.buffer.extend(["1", "2"]) protocol.handle_INT() self.assertFalse(interpreter.buffer) class WriterTests(unittest.TestCase): def test_Integer(self): """ Colorize an integer. """ manhole.lastColorizedLine("1") def test_DoubleQuoteString(self): """ Colorize an integer in double quotes. """ manhole.lastColorizedLine('"1"') def test_SingleQuoteString(self): """ Colorize an integer in single quotes. """ manhole.lastColorizedLine("'1'") def test_TripleSingleQuotedString(self): """ Colorize an integer in triple quotes. """ manhole.lastColorizedLine("'''1'''") def test_TripleDoubleQuotedString(self): """ Colorize an integer in triple and double quotes. """ manhole.lastColorizedLine('"""1"""') def test_FunctionDefinition(self): """ Colorize a function definition. """ manhole.lastColorizedLine("def foo():") def test_ClassDefinition(self): """ Colorize a class definition. """ manhole.lastColorizedLine("class foo:") def test_unicode(self): """ Colorize a Unicode string. """ res = manhole.lastColorizedLine("\u0438") self.assertTrue(isinstance(res, bytes)) def test_bytes(self): """ Colorize a UTF-8 byte string. """ res = manhole.lastColorizedLine(b"\xd0\xb8") self.assertTrue(isinstance(res, bytes)) def test_identicalOutput(self): """ The output of UTF-8 bytestrings and Unicode strings are identical. """ self.assertEqual( manhole.lastColorizedLine(b"\xd0\xb8"), manhole.lastColorizedLine("\u0438") ) class ManholeLoopbackMixin: serverProtocol = manhole.ColoredManhole def wfd(self, d): return defer.waitForDeferred(d) def test_SimpleExpression(self): """ Evaluate simple expression. """ done = self.recvlineClient.expect(b"done") self._testwrite(b"1 + 1\n" b"done") def finished(ign): self._assertBuffer([b">>> 1 + 1", b"2", b">>> done"]) return done.addCallback(finished) def test_TripleQuoteLineContinuation(self): """ Evaluate line continuation in triple quotes. """ done = self.recvlineClient.expect(b"done") self._testwrite(b"'''\n'''\n" b"done") def finished(ign): self._assertBuffer([b">>> '''", b"... '''", b"'\\n'", b">>> done"]) return done.addCallback(finished) def test_FunctionDefinition(self): """ Evaluate function definition. """ done = self.recvlineClient.expect(b"done") self._testwrite(b"def foo(bar):\n" b"\tprint(bar)\n\n" b"foo(42)\n" b"done") def finished(ign): self._assertBuffer( [ b">>> def foo(bar):", b"... print(bar)", b"... ", b">>> foo(42)", b"42", b">>> done", ] ) return done.addCallback(finished) def test_ClassDefinition(self): """ Evaluate class definition. """ done = self.recvlineClient.expect(b"done") self._testwrite( b"class Foo:\n" b"\tdef bar(self):\n" b"\t\tprint('Hello, world!')\n\n" b"Foo().bar()\n" b"done" ) def finished(ign): self._assertBuffer( [ b">>> class Foo:", b"... def bar(self):", b"... print('Hello, world!')", b"... ", b">>> Foo().bar()", b"Hello, world!", b">>> done", ] ) return done.addCallback(finished) def test_Exception(self): """ Evaluate raising an exception. """ done = self.recvlineClient.expect(b"done") self._testwrite(b"raise Exception('foo bar baz')\n" b"done") def finished(ign): self._assertBuffer( [ b">>> raise Exception('foo bar baz')", b"Traceback (most recent call last):", b' File "<console>", line 1, in ' + defaultFunctionName.encode("utf-8"), b"Exception: foo bar baz", b">>> done", ] ) return done.addCallback(finished) def test_ControlC(self): """ Evaluate interrupting with CTRL-C. """ done = self.recvlineClient.expect(b"done") self._testwrite(b"cancelled line" + manhole.CTRL_C + b"done") def finished(ign): self._assertBuffer( [b">>> cancelled line", b"KeyboardInterrupt", b">>> done"] ) return done.addCallback(finished) def test_interruptDuringContinuation(self): """ Sending ^C to Manhole while in a state where more input is required to complete a statement should discard the entire ongoing statement and reset the input prompt to the non-continuation prompt. """ continuing = self.recvlineClient.expect(b"things") self._testwrite(b"(\nthings") def gotContinuation(ignored): self._assertBuffer([b">>> (", b"... things"]) interrupted = self.recvlineClient.expect(b">>> ") self._testwrite(manhole.CTRL_C) return interrupted continuing.addCallback(gotContinuation) def gotInterruption(ignored): self._assertBuffer([b">>> (", b"... things", b"KeyboardInterrupt", b">>> "]) continuing.addCallback(gotInterruption) return continuing def test_ControlBackslash(self): r""" Evaluate cancelling with CTRL-\. """ self._testwrite(b"cancelled line") partialLine = self.recvlineClient.expect(b"cancelled line") def gotPartialLine(ign): self._assertBuffer([b">>> cancelled line"]) self._testwrite(manhole.CTRL_BACKSLASH) d = self.recvlineClient.onDisconnection return self.assertFailure(d, error.ConnectionDone) def gotClearedLine(ign): self._assertBuffer([b""]) return partialLine.addCallback(gotPartialLine).addCallback(gotClearedLine) @defer.inlineCallbacks def test_controlD(self): """ A CTRL+D in the middle of a line doesn't close a connection, but at the beginning of a line it does. """ self._testwrite(b"1 + 1") yield self.recvlineClient.expect(br"\+ 1") self._assertBuffer([b">>> 1 + 1"]) self._testwrite(manhole.CTRL_D + b" + 1") yield self.recvlineClient.expect(br"\+ 1") self._assertBuffer([b">>> 1 + 1 + 1"]) self._testwrite(b"\n") yield self.recvlineClient.expect(b"3\n>>> ") self._testwrite(manhole.CTRL_D) d = self.recvlineClient.onDisconnection yield self.assertFailure(d, error.ConnectionDone) @defer.inlineCallbacks def test_ControlL(self): """ CTRL+L is generally used as a redraw-screen command in terminal applications. Manhole doesn't currently respect this usage of it, but it should at least do something reasonable in response to this event (rather than, say, eating your face). """ # Start off with a newline so that when we clear the display we can # tell by looking for the missing first empty prompt line. self._testwrite(b"\n1 + 1") yield self.recvlineClient.expect(br"\+ 1") self._assertBuffer([b">>> ", b">>> 1 + 1"]) self._testwrite(manhole.CTRL_L + b" + 1") yield self.recvlineClient.expect(br"1 \+ 1 \+ 1") self._assertBuffer([b">>> 1 + 1 + 1"]) def test_controlA(self): """ CTRL-A can be used as HOME - returning cursor to beginning of current line buffer. """ self._testwrite(b'rint "hello"' + b"\x01" + b"p") d = self.recvlineClient.expect(b'print "hello"') def cb(ignore): self._assertBuffer([b'>>> print "hello"']) return d.addCallback(cb) def test_controlE(self): """ CTRL-E can be used as END - setting cursor to end of current line buffer. """ self._testwrite(b'rint "hello' + b"\x01" + b"p" + b"\x05" + b'"') d = self.recvlineClient.expect(b'print "hello"') def cb(ignore): self._assertBuffer([b'>>> print "hello"']) return d.addCallback(cb) @defer.inlineCallbacks def test_deferred(self): """ When a deferred is returned to the manhole REPL, it is displayed with a sequence number, and when the deferred fires, the result is printed. """ self._testwrite( b"from twisted.internet import defer, reactor\n" b"d = defer.Deferred()\n" b"d\n" ) yield self.recvlineClient.expect(b"<Deferred #0>") self._testwrite(b"c = reactor.callLater(0.1, d.callback, 'Hi!')\n") yield self.recvlineClient.expect(b">>> ") yield self.recvlineClient.expect(b"Deferred #0 called back: 'Hi!'\n>>> ") self._assertBuffer( [ b">>> from twisted.internet import defer, reactor", b">>> d = defer.Deferred()", b">>> d", b"<Deferred #0>", b">>> c = reactor.callLater(0.1, d.callback, 'Hi!')", b"Deferred #0 called back: 'Hi!'", b">>> ", ] ) class ManholeLoopbackTelnetTests(_TelnetMixin, unittest.TestCase, ManholeLoopbackMixin): """ Test manhole loopback over Telnet. """ pass class ManholeLoopbackSSHTests(_SSHMixin, unittest.TestCase, ManholeLoopbackMixin): """ Test manhole loopback over SSH. """ if ssh is None: skip = "cryptography requirements missing" class ManholeLoopbackStdioTests(_StdioMixin, unittest.TestCase, ManholeLoopbackMixin): """ Test manhole loopback over standard IO. """ if stdio is None: skip = "Terminal requirements missing" else: serverProtocol = stdio.ConsoleManhole class ManholeMainTests(unittest.TestCase): """ Test the I{main} method from the I{manhole} module. """ if stdio is None: skip = "Terminal requirements missing" def test_mainClassNotFound(self): """ Will raise an exception when called with an argument which is a dotted patch which can not be imported.. """ exception = self.assertRaises( ValueError, stdio.main, argv=["no-such-class"], ) self.assertEqual("Empty module name", exception.args[0]) PK F��Z�G %Y� Y� test_knownhosts.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.client.knownhosts}. """ import os from binascii import Error as BinasciiError, a2b_base64, b2a_base64 from unittest import skipIf from zope.interface.verify import verifyObject from twisted.conch.error import HostKeyChanged, InvalidEntry, UserRejectedKey from twisted.conch.interfaces import IKnownHostEntry from twisted.internet.defer import Deferred from twisted.python.compat import networkString from twisted.python.filepath import FilePath from twisted.python.reflect import requireModule from twisted.test.testutils import ComparisonTestsMixin from twisted.trial.unittest import TestCase if requireModule("cryptography") and requireModule("pyasn1"): from twisted.conch.client import default from twisted.conch.client.knownhosts import ( ConsoleUI, HashedEntry, KnownHostsFile, PlainEntry, UnparsedEntry, ) from twisted.conch.ssh.keys import BadKeyError, Key from twisted.conch.test import keydata else: skip = "cryptography and PyASN1 required for twisted.conch.knownhosts." sampleEncodedKey = ( b"AAAAB3NzaC1yc2EAAAABIwAAAQEAsV0VMRbGmzhqxxayLRHmvnFvtyNqgbNKV46dU1bVFB+3y" b"tNvue4Riqv/SVkPRNwMb7eWH29SviXaBxUhYyzKkDoNUq3rTNnH1Vnif6d6X4JCrUb5d3W+Dm" b"YClyJrZ5HgD/hUpdSkTRqdbQ2TrvSAxRacj+vHHT4F4dm1bJSewm3B2D8HVOoi/CbVh3dsIiC" b"dp8VltdZx4qYVfYe2LwVINCbAa3d3tj9ma7RVfw3OH2Mfb+toLd1N5tBQFb7oqTt2nC6I/6Bd" b"4JwPUld+IEitw/suElq/AIJVQXXujeyiZlea90HE65U2mF1ytr17HTAIT2ySokJWyuBANGACk" b"6iIaw==" ) otherSampleEncodedKey = ( b"AAAAB3NzaC1yc2EAAAABIwAAAIEAwaeCZd3UCuPXhX39+/p9qO028jTF76DMVd9mPvYVDVXuf" b"WckKZauF7+0b7qm+ChT7kan6BzRVo4++gCVNfAlMzLysSt3ylmOR48tFpAfygg9UCX3DjHz0E" b"lOOUKh3iifc9aUShD0OPaK3pR5JJ8jfiBfzSYWt/hDi/iZ4igsSs8=" ) thirdSampleEncodedKey = ( b"AAAAB3NzaC1yc2EAAAABIwAAAQEAl/TQakPkePlnwCBRPitIVUTg6Z8VzN1en+DGkyo/evkmLw" b"7o4NWR5qbysk9A9jXW332nxnEuAnbcCam9SHe1su1liVfyIK0+3bdn0YRB0sXIbNEtMs2LtCho" b"/aV3cXPS+Cf1yut3wvIpaRnAzXxuKPCTXQ7/y0IXa8TwkRBH58OJa3RqfQ/NsSp5SAfdsrHyH2" b"aitiVKm2jfbTKzSEqOQG/zq4J9GXTkq61gZugory/Tvl5/yPgSnOR6C9jVOMHf27ZPoRtyj9SY" b"343Hd2QHiIE0KPZJEgCynKeWoKz8v6eTSK8n4rBnaqWdp8MnGZK1WGy05MguXbyCDuTC8AmJXQ" b"==" ) ecdsaSampleEncodedKey = ( b"AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBIFwh3/zBANyPPIE60" b"SMMfdKMYo3OvfvzGLZphzuKrzSt0q4uF+/iYqtYiHhryAwU/fDWlUQ9kck9f+IlpsNtY4=" ) sampleKey = a2b_base64(sampleEncodedKey) otherSampleKey = a2b_base64(otherSampleEncodedKey) thirdSampleKey = a2b_base64(thirdSampleEncodedKey) ecdsaSampleKey = a2b_base64(ecdsaSampleEncodedKey) samplePlaintextLine = b"www.twistedmatrix.com ssh-rsa " + sampleEncodedKey + b"\n" otherSamplePlaintextLine = b"divmod.com ssh-rsa " + otherSampleEncodedKey + b"\n" sampleHostIPLine = ( b"www.twistedmatrix.com,198.49.126.131 ssh-rsa " + sampleEncodedKey + b"\n" ) sampleHashedLine = ( b"|1|gJbSEPBG9ZSBoZpHNtZBD1bHKBA=|bQv+0Xa0dByrwkA1EB0E7Xop/Fo= ssh-rsa " + sampleEncodedKey + b"\n" ) class EntryTestsMixin: """ Tests for implementations of L{IKnownHostEntry}. Subclasses must set the 'entry' attribute to a provider of that interface, the implementation of that interface under test. @ivar entry: a provider of L{IKnownHostEntry} with a hostname of www.twistedmatrix.com and an RSA key of sampleKey. """ def test_providesInterface(self): """ The given entry should provide IKnownHostEntry. """ verifyObject(IKnownHostEntry, self.entry) def test_fromString(self): """ Constructing a plain text entry from an unhashed known_hosts entry will result in an L{IKnownHostEntry} provider with 'keyString', 'hostname', and 'keyType' attributes. While outside the interface in question, these attributes are held in common by L{PlainEntry} and L{HashedEntry} implementations; other implementations should override this method in subclasses. """ entry = self.entry self.assertEqual(entry.publicKey, Key.fromString(sampleKey)) self.assertEqual(entry.keyType, b"ssh-rsa") def test_matchesKey(self): """ L{IKnownHostEntry.matchesKey} checks to see if an entry matches a given SSH key. """ twistedmatrixDotCom = Key.fromString(sampleKey) divmodDotCom = Key.fromString(otherSampleKey) self.assertEqual(True, self.entry.matchesKey(twistedmatrixDotCom)) self.assertEqual(False, self.entry.matchesKey(divmodDotCom)) def test_matchesHost(self): """ L{IKnownHostEntry.matchesHost} checks to see if an entry matches a given hostname. """ self.assertTrue(self.entry.matchesHost(b"www.twistedmatrix.com")) self.assertFalse(self.entry.matchesHost(b"www.divmod.com")) class PlainEntryTests(EntryTestsMixin, TestCase): """ Test cases for L{PlainEntry}. """ plaintextLine = samplePlaintextLine hostIPLine = sampleHostIPLine def setUp(self): """ Set 'entry' to a sample plain-text entry with sampleKey as its key. """ self.entry = PlainEntry.fromString(self.plaintextLine) def test_matchesHostIP(self): """ A "hostname,ip" formatted line will match both the host and the IP. """ self.entry = PlainEntry.fromString(self.hostIPLine) self.assertTrue(self.entry.matchesHost(b"198.49.126.131")) self.test_matchesHost() def test_toString(self): """ L{PlainEntry.toString} generates the serialized OpenSSL format string for the entry, sans newline. """ self.assertEqual(self.entry.toString(), self.plaintextLine.rstrip(b"\n")) multiHostEntry = PlainEntry.fromString(self.hostIPLine) self.assertEqual(multiHostEntry.toString(), self.hostIPLine.rstrip(b"\n")) class PlainTextWithCommentTests(PlainEntryTests): """ Test cases for L{PlainEntry} when parsed from a line with a comment. """ plaintextLine = samplePlaintextLine[:-1] + b" plain text comment.\n" hostIPLine = sampleHostIPLine[:-1] + b" text following host/IP line\n" class HashedEntryTests(EntryTestsMixin, ComparisonTestsMixin, TestCase): """ Tests for L{HashedEntry}. This suite doesn't include any tests for host/IP pairs because hashed entries store IP addresses the same way as hostnames and does not support comma-separated lists. (If you hash the IP and host together you can't tell if you've got the key already for one or the other.) """ hashedLine = sampleHashedLine def setUp(self): """ Set 'entry' to a sample hashed entry for twistedmatrix.com with sampleKey as its key. """ self.entry = HashedEntry.fromString(self.hashedLine) def test_toString(self): """ L{HashedEntry.toString} generates the serialized OpenSSL format string for the entry, sans the newline. """ self.assertEqual(self.entry.toString(), self.hashedLine.rstrip(b"\n")) def test_equality(self): """ Two L{HashedEntry} instances compare equal if and only if they represent the same host and key in exactly the same way: the host salt, host hash, public key type, public key, and comment fields must all be equal. """ hostSalt = b"gJbSEPBG9ZSBoZpHNtZBD1bHKBA" hostHash = b"bQv+0Xa0dByrwkA1EB0E7Xop/Fo" publicKey = Key.fromString(sampleKey) keyType = networkString(publicKey.type()) comment = b"hello, world" entry = HashedEntry(hostSalt, hostHash, keyType, publicKey, comment) duplicate = HashedEntry(hostSalt, hostHash, keyType, publicKey, comment) # Vary the host salt self.assertNormalEqualityImplementation( entry, duplicate, HashedEntry(hostSalt[::-1], hostHash, keyType, publicKey, comment), ) # Vary the host hash self.assertNormalEqualityImplementation( entry, duplicate, HashedEntry(hostSalt, hostHash[::-1], keyType, publicKey, comment), ) # Vary the key type self.assertNormalEqualityImplementation( entry, duplicate, HashedEntry(hostSalt, hostHash, keyType[::-1], publicKey, comment), ) # Vary the key self.assertNormalEqualityImplementation( entry, duplicate, HashedEntry( hostSalt, hostHash, keyType, Key.fromString(otherSampleKey), comment ), ) # Vary the comment self.assertNormalEqualityImplementation( entry, duplicate, HashedEntry(hostSalt, hostHash, keyType, publicKey, comment[::-1]), ) class HashedEntryWithCommentTests(HashedEntryTests): """ Test cases for L{PlainEntry} when parsed from a line with a comment. """ hashedLine = sampleHashedLine[:-1] + b" plain text comment.\n" class UnparsedEntryTests(TestCase, EntryTestsMixin): """ Tests for L{UnparsedEntry} """ def setUp(self): """ Set up the 'entry' to be an unparsed entry for some random text. """ self.entry = UnparsedEntry(b" This is a bogus entry. \n") def test_fromString(self): """ Creating an L{UnparsedEntry} should simply record the string it was passed. """ self.assertEqual(b" This is a bogus entry. \n", self.entry._string) def test_matchesHost(self): """ An unparsed entry can't match any hosts. """ self.assertFalse(self.entry.matchesHost(b"www.twistedmatrix.com")) def test_matchesKey(self): """ An unparsed entry can't match any keys. """ self.assertFalse(self.entry.matchesKey(Key.fromString(sampleKey))) def test_toString(self): """ L{UnparsedEntry.toString} returns its input string, sans trailing newline. """ self.assertEqual(b" This is a bogus entry. ", self.entry.toString()) class ParseErrorTests(TestCase): """ L{HashedEntry.fromString} and L{PlainEntry.fromString} can raise a variety of errors depending on misformattings of certain strings. These tests make sure those errors are caught. Since many of the ways that this can go wrong are in the lower-level APIs being invoked by the parsing logic, several of these are integration tests with the C{base64} and L{twisted.conch.ssh.keys} modules. """ def invalidEntryTest(self, cls): """ If there are fewer than three elements, C{fromString} should raise L{InvalidEntry}. """ self.assertRaises(InvalidEntry, cls.fromString, b"invalid") def notBase64Test(self, cls): """ If the key is not base64, C{fromString} should raise L{BinasciiError}. """ self.assertRaises(BinasciiError, cls.fromString, b"x x x") def badKeyTest(self, cls, prefix): """ If the key portion of the entry is valid base64, but is not actually an SSH key, C{fromString} should raise L{BadKeyError}. """ self.assertRaises( BadKeyError, cls.fromString, b" ".join( [prefix, b"ssh-rsa", b2a_base64(b"Hey, this isn't an SSH key!").strip()] ), ) def test_invalidPlainEntry(self): """ If there are fewer than three whitespace-separated elements in an entry, L{PlainEntry.fromString} should raise L{InvalidEntry}. """ self.invalidEntryTest(PlainEntry) def test_invalidHashedEntry(self): """ If there are fewer than three whitespace-separated elements in an entry, or the hostname salt/hash portion has more than two elements, L{HashedEntry.fromString} should raise L{InvalidEntry}. """ self.invalidEntryTest(HashedEntry) a, b, c = sampleHashedLine.split() self.assertRaises( InvalidEntry, HashedEntry.fromString, b" ".join([a + b"||", b, c]) ) def test_plainNotBase64(self): """ If the key portion of a plain entry is not decodable as base64, C{fromString} should raise L{BinasciiError}. """ self.notBase64Test(PlainEntry) def test_hashedNotBase64(self): """ If the key, host salt, or host hash portion of a hashed entry is not encoded, it will raise L{BinasciiError}. """ self.notBase64Test(HashedEntry) a, b, c = sampleHashedLine.split() # Salt not valid base64. self.assertRaises( BinasciiError, HashedEntry.fromString, b" ".join([b"|1|x|" + b2a_base64(b"stuff").strip(), b, c]), ) # Host hash not valid base64. self.assertRaises( BinasciiError, HashedEntry.fromString, b" ".join([HashedEntry.MAGIC + b2a_base64(b"stuff").strip() + b"|x", b, c]), ) # Neither salt nor hash valid base64. self.assertRaises( BinasciiError, HashedEntry.fromString, b" ".join([b"|1|x|x", b, c]) ) def test_hashedBadKey(self): """ If the key portion of the entry is valid base64, but is not actually an SSH key, C{HashedEntry.fromString} should raise L{BadKeyError}. """ a, b, c = sampleHashedLine.split() self.badKeyTest(HashedEntry, a) def test_plainBadKey(self): """ If the key portion of the entry is valid base64, but is not actually an SSH key, C{PlainEntry.fromString} should raise L{BadKeyError}. """ self.badKeyTest(PlainEntry, b"hostname") class KnownHostsDatabaseTests(TestCase): """ Tests for L{KnownHostsFile}. """ def pathWithContent(self, content): """ Return a FilePath with the given initial content. """ fp = FilePath(self.mktemp()) fp.setContent(content) return fp def loadSampleHostsFile( self, content=( sampleHashedLine + otherSamplePlaintextLine + b"\n# That was a blank line.\n" b"This is just unparseable.\n" b"|1|This also unparseable.\n" ), ): """ Return a sample hosts file, with keys for www.twistedmatrix.com and divmod.com present. """ return KnownHostsFile.fromPath(self.pathWithContent(content)) def test_readOnlySavePath(self): """ L{KnownHostsFile.savePath} is read-only; if an assignment is made to it, L{AttributeError} is raised and the value is unchanged. """ path = FilePath(self.mktemp()) new = FilePath(self.mktemp()) hostsFile = KnownHostsFile(path) self.assertRaises(AttributeError, setattr, hostsFile, "savePath", new) self.assertEqual(path, hostsFile.savePath) def test_defaultInitializerIgnoresExisting(self): """ The default initializer for L{KnownHostsFile} disregards any existing contents in the save path. """ hostsFile = KnownHostsFile(self.pathWithContent(sampleHashedLine)) self.assertEqual([], list(hostsFile.iterentries())) def test_defaultInitializerClobbersExisting(self): """ After using the default initializer for L{KnownHostsFile}, the first use of L{KnownHostsFile.save} overwrites any existing contents in the save path. """ path = self.pathWithContent(sampleHashedLine) hostsFile = KnownHostsFile(path) entry = hostsFile.addHostKey(b"www.example.com", Key.fromString(otherSampleKey)) hostsFile.save() # Check KnownHostsFile to see what it thinks the state is self.assertEqual([entry], list(hostsFile.iterentries())) # And also directly check the underlying file itself self.assertEqual(entry.toString() + b"\n", path.getContent()) def test_saveResetsClobberState(self): """ After L{KnownHostsFile.save} is used once with an instance initialized by the default initializer, contents of the save path are respected and preserved. """ hostsFile = KnownHostsFile(self.pathWithContent(sampleHashedLine)) preSave = hostsFile.addHostKey( b"www.example.com", Key.fromString(otherSampleKey) ) hostsFile.save() postSave = hostsFile.addHostKey( b"another.example.com", Key.fromString(thirdSampleKey) ) hostsFile.save() self.assertEqual([preSave, postSave], list(hostsFile.iterentries())) def test_loadFromPath(self): """ Loading a L{KnownHostsFile} from a path with six entries in it will result in a L{KnownHostsFile} object with six L{IKnownHostEntry} providers in it. """ hostsFile = self.loadSampleHostsFile() self.assertEqual(6, len(list(hostsFile.iterentries()))) def test_iterentriesUnsaved(self): """ If the save path for a L{KnownHostsFile} does not exist, L{KnownHostsFile.iterentries} still returns added but unsaved entries. """ hostsFile = KnownHostsFile(FilePath(self.mktemp())) hostsFile.addHostKey(b"www.example.com", Key.fromString(sampleKey)) self.assertEqual(1, len(list(hostsFile.iterentries()))) def test_verifyHashedEntry(self): """ Loading a L{KnownHostsFile} from a path containing a single valid L{HashedEntry} entry will result in a L{KnownHostsFile} object with one L{IKnownHostEntry} provider. """ hostsFile = self.loadSampleHostsFile(sampleHashedLine) entries = list(hostsFile.iterentries()) self.assertIsInstance(entries[0], HashedEntry) self.assertTrue(entries[0].matchesHost(b"www.twistedmatrix.com")) self.assertEqual(1, len(entries)) def test_verifyPlainEntry(self): """ Loading a L{KnownHostsFile} from a path containing a single valid L{PlainEntry} entry will result in a L{KnownHostsFile} object with one L{IKnownHostEntry} provider. """ hostsFile = self.loadSampleHostsFile(otherSamplePlaintextLine) entries = list(hostsFile.iterentries()) self.assertIsInstance(entries[0], PlainEntry) self.assertTrue(entries[0].matchesHost(b"divmod.com")) self.assertEqual(1, len(entries)) def test_verifyUnparsedEntry(self): """ Loading a L{KnownHostsFile} from a path that only contains '\n' will result in a L{KnownHostsFile} object containing a L{UnparsedEntry} object. """ hostsFile = self.loadSampleHostsFile(b"\n") entries = list(hostsFile.iterentries()) self.assertIsInstance(entries[0], UnparsedEntry) self.assertEqual(entries[0].toString(), b"") self.assertEqual(1, len(entries)) def test_verifyUnparsedComment(self): """ Loading a L{KnownHostsFile} from a path that contains a comment will result in a L{KnownHostsFile} object containing a L{UnparsedEntry} object. """ hostsFile = self.loadSampleHostsFile(b"# That was a blank line.\n") entries = list(hostsFile.iterentries()) self.assertIsInstance(entries[0], UnparsedEntry) self.assertEqual(entries[0].toString(), b"# That was a blank line.") def test_verifyUnparsableLine(self): """ Loading a L{KnownHostsFile} from a path that contains an unparseable line will be represented as an L{UnparsedEntry} instance. """ hostsFile = self.loadSampleHostsFile(b"This is just unparseable.\n") entries = list(hostsFile.iterentries()) self.assertIsInstance(entries[0], UnparsedEntry) self.assertEqual(entries[0].toString(), b"This is just unparseable.") self.assertEqual(1, len(entries)) def test_verifyUnparsableEncryptionMarker(self): """ Loading a L{KnownHostsFile} from a path containing an unparseable line that starts with an encryption marker will be represented as an L{UnparsedEntry} instance. """ hostsFile = self.loadSampleHostsFile(b"|1|This is unparseable.\n") entries = list(hostsFile.iterentries()) self.assertIsInstance(entries[0], UnparsedEntry) self.assertEqual(entries[0].toString(), b"|1|This is unparseable.") self.assertEqual(1, len(entries)) def test_loadNonExistent(self): """ Loading a L{KnownHostsFile} from a path that does not exist should result in an empty L{KnownHostsFile} that will save back to that path. """ pn = self.mktemp() knownHostsFile = KnownHostsFile.fromPath(FilePath(pn)) entries = list(knownHostsFile.iterentries()) self.assertEqual([], entries) self.assertFalse(FilePath(pn).exists()) knownHostsFile.save() self.assertTrue(FilePath(pn).exists()) def test_loadNonExistentParent(self): """ Loading a L{KnownHostsFile} from a path whose parent directory does not exist should result in an empty L{KnownHostsFile} that will save back to that path, creating its parent directory(ies) in the process. """ thePath = FilePath(self.mktemp()) knownHostsPath = thePath.child("foo").child(b"known_hosts") knownHostsFile = KnownHostsFile.fromPath(knownHostsPath) knownHostsFile.save() knownHostsPath.restat(False) self.assertTrue(knownHostsPath.exists()) def test_savingAddsEntry(self): """ L{KnownHostsFile.save} will write out a new file with any entries that have been added. """ path = self.pathWithContent(sampleHashedLine + otherSamplePlaintextLine) knownHostsFile = KnownHostsFile.fromPath(path) newEntry = knownHostsFile.addHostKey( b"some.example.com", Key.fromString(thirdSampleKey) ) expectedContent = ( sampleHashedLine + otherSamplePlaintextLine + HashedEntry.MAGIC + b2a_base64(newEntry._hostSalt).strip() + b"|" + b2a_base64(newEntry._hostHash).strip() + b" ssh-rsa " + thirdSampleEncodedKey + b"\n" ) # Sanity check, let's make sure the base64 API being used for the test # isn't inserting spurious newlines. self.assertEqual(3, expectedContent.count(b"\n")) knownHostsFile.save() self.assertEqual(expectedContent, path.getContent()) def test_savingAvoidsDuplication(self): """ L{KnownHostsFile.save} only writes new entries to the save path, not entries which were added and already written by a previous call to C{save}. """ path = FilePath(self.mktemp()) knownHosts = KnownHostsFile(path) entry = knownHosts.addHostKey(b"some.example.com", Key.fromString(sampleKey)) knownHosts.save() knownHosts.save() knownHosts = KnownHostsFile.fromPath(path) self.assertEqual([entry], list(knownHosts.iterentries())) def test_savingsPreservesExisting(self): """ L{KnownHostsFile.save} will not overwrite existing entries in its save path, even if they were only added after the L{KnownHostsFile} instance was initialized. """ # Start off with one host/key pair in the file path = self.pathWithContent(sampleHashedLine) knownHosts = KnownHostsFile.fromPath(path) # After initializing the KnownHostsFile instance, add a second host/key # pair to the file directly - without the instance's help or knowledge. with path.open("a") as hostsFileObj: hostsFileObj.write(otherSamplePlaintextLine) # Add a third host/key pair using the KnownHostsFile instance key = Key.fromString(thirdSampleKey) knownHosts.addHostKey(b"brandnew.example.com", key) knownHosts.save() # Check that all three host/key pairs are present. knownHosts = KnownHostsFile.fromPath(path) self.assertEqual( [True, True, True], [ knownHosts.hasHostKey( b"www.twistedmatrix.com", Key.fromString(sampleKey) ), knownHosts.hasHostKey(b"divmod.com", Key.fromString(otherSampleKey)), knownHosts.hasHostKey(b"brandnew.example.com", key), ], ) def test_hasPresentKey(self): """ L{KnownHostsFile.hasHostKey} returns C{True} when a key for the given hostname is present and matches the expected key. """ hostsFile = self.loadSampleHostsFile() self.assertTrue( hostsFile.hasHostKey(b"www.twistedmatrix.com", Key.fromString(sampleKey)) ) def test_notPresentKey(self): """ L{KnownHostsFile.hasHostKey} returns C{False} when a key for the given hostname is not present. """ hostsFile = self.loadSampleHostsFile() self.assertFalse( hostsFile.hasHostKey(b"non-existent.example.com", Key.fromString(sampleKey)) ) self.assertTrue( hostsFile.hasHostKey(b"www.twistedmatrix.com", Key.fromString(sampleKey)) ) self.assertFalse( hostsFile.hasHostKey( b"www.twistedmatrix.com", Key.fromString(ecdsaSampleKey) ) ) def test_hasLaterAddedKey(self): """ L{KnownHostsFile.hasHostKey} returns C{True} when a key for the given hostname is present in the file, even if it is only added to the file after the L{KnownHostsFile} instance is initialized. """ key = Key.fromString(sampleKey) entry = PlainEntry([b"brandnew.example.com"], key.sshType(), key, b"") hostsFile = self.loadSampleHostsFile() with hostsFile.savePath.open("a") as hostsFileObj: hostsFileObj.write(entry.toString() + b"\n") self.assertEqual(True, hostsFile.hasHostKey(b"brandnew.example.com", key)) def test_savedEntryHasKeyMismatch(self): """ L{KnownHostsFile.hasHostKey} raises L{HostKeyChanged} if the host key is present in the underlying file, but different from the expected one. The resulting exception should have an C{offendingEntry} indicating the given entry. """ hostsFile = self.loadSampleHostsFile() entries = list(hostsFile.iterentries()) exception = self.assertRaises( HostKeyChanged, hostsFile.hasHostKey, b"www.twistedmatrix.com", Key.fromString(otherSampleKey), ) self.assertEqual(exception.offendingEntry, entries[0]) self.assertEqual(exception.lineno, 1) self.assertEqual(exception.path, hostsFile.savePath) def test_savedEntryAfterAddHasKeyMismatch(self): """ Even after a new entry has been added in memory but not yet saved, the L{HostKeyChanged} exception raised by L{KnownHostsFile.hasHostKey} has a C{lineno} attribute which indicates the 1-based line number of the offending entry in the underlying file when the given host key does not match the expected host key. """ hostsFile = self.loadSampleHostsFile() hostsFile.addHostKey(b"www.example.com", Key.fromString(otherSampleKey)) exception = self.assertRaises( HostKeyChanged, hostsFile.hasHostKey, b"www.twistedmatrix.com", Key.fromString(otherSampleKey), ) self.assertEqual(exception.lineno, 1) self.assertEqual(exception.path, hostsFile.savePath) def test_unsavedEntryHasKeyMismatch(self): """ L{KnownHostsFile.hasHostKey} raises L{HostKeyChanged} if the host key is present in memory (but not yet saved), but different from the expected one. The resulting exception has a C{offendingEntry} indicating the given entry, but no filename or line number information (reflecting the fact that the entry exists only in memory). """ hostsFile = KnownHostsFile(FilePath(self.mktemp())) entry = hostsFile.addHostKey(b"www.example.com", Key.fromString(otherSampleKey)) exception = self.assertRaises( HostKeyChanged, hostsFile.hasHostKey, b"www.example.com", Key.fromString(thirdSampleKey), ) self.assertEqual(exception.offendingEntry, entry) self.assertIsNone(exception.lineno) self.assertIsNone(exception.path) def test_addHostKey(self): """ L{KnownHostsFile.addHostKey} adds a new L{HashedEntry} to the host file, and returns it. """ hostsFile = self.loadSampleHostsFile() aKey = Key.fromString(thirdSampleKey) self.assertEqual(False, hostsFile.hasHostKey(b"somewhere.example.com", aKey)) newEntry = hostsFile.addHostKey(b"somewhere.example.com", aKey) # The code in OpenSSH requires host salts to be 20 characters long. # This is the required length of a SHA-1 HMAC hash, so it's just a # sanity check. self.assertEqual(20, len(newEntry._hostSalt)) self.assertEqual(True, newEntry.matchesHost(b"somewhere.example.com")) self.assertEqual(newEntry.keyType, b"ssh-rsa") self.assertEqual(aKey, newEntry.publicKey) self.assertEqual(True, hostsFile.hasHostKey(b"somewhere.example.com", aKey)) def test_randomSalts(self): """ L{KnownHostsFile.addHostKey} generates a random salt for each new key, so subsequent salts will be different. """ hostsFile = self.loadSampleHostsFile() aKey = Key.fromString(thirdSampleKey) self.assertNotEqual( hostsFile.addHostKey(b"somewhere.example.com", aKey)._hostSalt, hostsFile.addHostKey(b"somewhere-else.example.com", aKey)._hostSalt, ) def test_verifyValidKey(self): """ Verifying a valid key should return a L{Deferred} which fires with True. """ hostsFile = self.loadSampleHostsFile() hostsFile.addHostKey(b"1.2.3.4", Key.fromString(sampleKey)) ui = FakeUI() d = hostsFile.verifyHostKey( ui, b"www.twistedmatrix.com", b"1.2.3.4", Key.fromString(sampleKey) ) l = [] d.addCallback(l.append) self.assertEqual(l, [True]) def test_verifyInvalidKey(self): """ Verifying an invalid key should return a L{Deferred} which fires with a L{HostKeyChanged} failure. """ hostsFile = self.loadSampleHostsFile() wrongKey = Key.fromString(thirdSampleKey) ui = FakeUI() hostsFile.addHostKey(b"1.2.3.4", Key.fromString(sampleKey)) d = hostsFile.verifyHostKey(ui, b"www.twistedmatrix.com", b"1.2.3.4", wrongKey) return self.assertFailure(d, HostKeyChanged) def verifyNonPresentKey(self): """ Set up a test to verify a key that isn't present. Return a 3-tuple of the UI, a list set up to collect the result of the verifyHostKey call, and the sample L{KnownHostsFile} being used. This utility method avoids returning a L{Deferred}, and records results in the returned list instead, because the events which get generated here are pre-recorded in the 'ui' object. If the L{Deferred} in question does not fire, the it will fail quickly with an empty list. """ hostsFile = self.loadSampleHostsFile() absentKey = Key.fromString(thirdSampleKey) ui = FakeUI() l = [] d = hostsFile.verifyHostKey( ui, b"sample-host.example.com", b"4.3.2.1", absentKey ) d.addBoth(l.append) self.assertEqual([], l) self.assertEqual( ui.promptText, b"The authenticity of host 'sample-host.example.com (4.3.2.1)' " b"can't be established.\n" b"RSA key fingerprint is " b"SHA256:mS7mDBGhewdzJkaKRkx+wMjUdZb/GzvgcdoYjX5Js9I=.\n" b"Are you sure you want to continue connecting (yes/no)? ", ) return ui, l, hostsFile def test_verifyNonPresentKey_Yes(self): """ Verifying a key where neither the hostname nor the IP are present should result in the UI being prompted with a message explaining as much. If the UI says yes, the Deferred should fire with True. """ ui, l, knownHostsFile = self.verifyNonPresentKey() ui.promptDeferred.callback(True) self.assertEqual([True], l) reloaded = KnownHostsFile.fromPath(knownHostsFile.savePath) self.assertEqual( True, reloaded.hasHostKey(b"4.3.2.1", Key.fromString(thirdSampleKey)) ) self.assertEqual( True, reloaded.hasHostKey( b"sample-host.example.com", Key.fromString(thirdSampleKey) ), ) def test_verifyNonPresentKey_No(self): """ Verifying a key where neither the hostname nor the IP are present should result in the UI being prompted with a message explaining as much. If the UI says no, the Deferred should fail with UserRejectedKey. """ ui, l, knownHostsFile = self.verifyNonPresentKey() ui.promptDeferred.callback(False) l[0].trap(UserRejectedKey) def test_verifyNonPresentECKey(self): """ Set up a test to verify an ECDSA key that isn't present. Return a 3-tuple of the UI, a list set up to collect the result of the verifyHostKey call, and the sample L{KnownHostsFile} being used. """ ecObj = Key._fromECComponents( x=keydata.ECDatanistp256["x"], y=keydata.ECDatanistp256["y"], privateValue=keydata.ECDatanistp256["privateValue"], curve=keydata.ECDatanistp256["curve"], ) hostsFile = self.loadSampleHostsFile() ui = FakeUI() l = [] d = hostsFile.verifyHostKey(ui, b"sample-host.example.com", b"4.3.2.1", ecObj) d.addBoth(l.append) self.assertEqual([], l) self.assertEqual( ui.promptText, b"The authenticity of host 'sample-host.example.com (4.3.2.1)' " b"can't be established.\n" b"ECDSA key fingerprint is " b"SHA256:fJnSpgCcYoYYsaBbnWj1YBghGh/QTDgfe4w4U5M5tEo=.\n" b"Are you sure you want to continue connecting (yes/no)? ", ) def test_verifyHostIPMismatch(self): """ Verifying a key where the host is present (and correct), but the IP is present and different, should result the deferred firing in a HostKeyChanged failure. """ hostsFile = self.loadSampleHostsFile() wrongKey = Key.fromString(thirdSampleKey) ui = FakeUI() d = hostsFile.verifyHostKey(ui, b"www.twistedmatrix.com", b"4.3.2.1", wrongKey) return self.assertFailure(d, HostKeyChanged) def test_verifyKeyForHostAndIP(self): """ Verifying a key where the hostname is present but the IP is not should result in the key being added for the IP and the user being warned about the change. """ ui = FakeUI() hostsFile = self.loadSampleHostsFile() expectedKey = Key.fromString(sampleKey) hostsFile.verifyHostKey(ui, b"www.twistedmatrix.com", b"5.4.3.2", expectedKey) self.assertEqual( True, KnownHostsFile.fromPath(hostsFile.savePath).hasHostKey( b"5.4.3.2", expectedKey ), ) self.assertEqual( [ "Warning: Permanently added the RSA host key for IP address " "'5.4.3.2' to the list of known hosts." ], ui.userWarnings, ) def test_getHostKeyAlgorithms(self): """ For a given host, get the host key algorithms for that host in the known_hosts file. """ hostsFile = self.loadSampleHostsFile() hostsFile.addHostKey(b"www.twistedmatrix.com", Key.fromString(otherSampleKey)) hostsFile.addHostKey(b"www.twistedmatrix.com", Key.fromString(ecdsaSampleKey)) hostsFile.save() options = {} options["known-hosts"] = hostsFile.savePath.path algorithms = default.getHostKeyAlgorithms(b"www.twistedmatrix.com", options) expectedAlgorithms = [b"ssh-rsa", b"ecdsa-sha2-nistp256"] self.assertEqual(algorithms, expectedAlgorithms) class FakeFile: """ A fake file-like object that acts enough like a file for L{ConsoleUI.prompt}. """ def __init__(self): self.inlines = [] self.outchunks = [] self.closed = False def readline(self): """ Return a line from the 'inlines' list. """ return self.inlines.pop(0) def write(self, chunk): """ Append the given item to the 'outchunks' list. """ if self.closed: raise OSError("the file was closed") self.outchunks.append(chunk) def close(self): """ Set the 'closed' flag to True, explicitly marking that it has been closed. """ self.closed = True class ConsoleUITests(TestCase): """ Test cases for L{ConsoleUI}. """ def setUp(self): """ Create a L{ConsoleUI} pointed at a L{FakeFile}. """ self.fakeFile = FakeFile() self.ui = ConsoleUI(self.openFile) def openFile(self): """ Return the current fake file. """ return self.fakeFile def newFile(self, lines): """ Create a new fake file (the next file that self.ui will open) with the given list of lines to be returned from readline(). """ self.fakeFile = FakeFile() self.fakeFile.inlines = lines def test_promptYes(self): """ L{ConsoleUI.prompt} writes a message to the console, then reads a line. If that line is 'yes', then it returns a L{Deferred} that fires with True. """ for okYes in [b"yes", b"Yes", b"yes\n"]: self.newFile([okYes]) l = [] self.ui.prompt("Hello, world!").addCallback(l.append) self.assertEqual(["Hello, world!"], self.fakeFile.outchunks) self.assertEqual([True], l) self.assertTrue(self.fakeFile.closed) def test_promptNo(self): """ L{ConsoleUI.prompt} writes a message to the console, then reads a line. If that line is 'no', then it returns a L{Deferred} that fires with False. """ for okNo in [b"no", b"No", b"no\n"]: self.newFile([okNo]) l = [] self.ui.prompt("Goodbye, world!").addCallback(l.append) self.assertEqual(["Goodbye, world!"], self.fakeFile.outchunks) self.assertEqual([False], l) self.assertTrue(self.fakeFile.closed) def test_promptRepeatedly(self): """ L{ConsoleUI.prompt} writes a message to the console, then reads a line. If that line is neither 'yes' nor 'no', then it says "Please enter 'yes' or 'no'" until it gets a 'yes' or a 'no', at which point it returns a Deferred that answers either True or False. """ self.newFile([b"what", b"uh", b"okay", b"yes"]) l = [] self.ui.prompt(b"Please say something useful.").addCallback(l.append) self.assertEqual([True], l) self.assertEqual( self.fakeFile.outchunks, [b"Please say something useful."] + [b"Please type 'yes' or 'no': "] * 3, ) self.assertTrue(self.fakeFile.closed) self.newFile([b"blah", b"stuff", b"feh", b"no"]) l = [] self.ui.prompt(b"Please say something negative.").addCallback(l.append) self.assertEqual([False], l) self.assertEqual( self.fakeFile.outchunks, [b"Please say something negative."] + [b"Please type 'yes' or 'no': "] * 3, ) self.assertTrue(self.fakeFile.closed) def test_promptOpenFailed(self): """ If the C{opener} passed to L{ConsoleUI} raises an exception, that exception will fail the L{Deferred} returned from L{ConsoleUI.prompt}. """ def raiseIt(): raise OSError() ui = ConsoleUI(raiseIt) d = ui.prompt("This is a test.") return self.assertFailure(d, IOError) def test_warn(self): """ L{ConsoleUI.warn} should output a message to the console object. """ self.ui.warn("Test message.") self.assertEqual(["Test message."], self.fakeFile.outchunks) self.assertTrue(self.fakeFile.closed) def test_warnOpenFailed(self): """ L{ConsoleUI.warn} should log a traceback if the output can't be opened. """ def raiseIt(): 1 / 0 ui = ConsoleUI(raiseIt) ui.warn("This message never makes it.") self.assertEqual(len(self.flushLoggedErrors(ZeroDivisionError)), 1) class FakeUI: """ A fake UI object, adhering to the interface expected by L{KnownHostsFile.verifyHostKey} @ivar userWarnings: inputs provided to 'warn'. @ivar promptDeferred: last result returned from 'prompt'. @ivar promptText: the last input provided to 'prompt'. """ def __init__(self): self.userWarnings = [] self.promptDeferred = None self.promptText = None def prompt(self, text): """ Issue the user an interactive prompt, which they can accept or deny. """ self.promptText = text self.promptDeferred = Deferred() return self.promptDeferred def warn(self, text): """ Issue a non-interactive warning to the user. """ self.userWarnings.append(text) class FakeObject: """ A fake object that can have some attributes. Used to fake L{SSHClientTransport} and L{SSHClientFactory}. """ @skipIf(not FilePath("/dev/tty").exists(), "Platform lacks /dev/tty") class DefaultAPITests(TestCase): """ The API in L{twisted.conch.client.default.verifyHostKey} is the integration point between the code in the rest of conch and L{KnownHostsFile}. """ def patchedOpen(self, fname, mode, **kwargs): """ The patched version of 'open'; this returns a L{FakeFile} that the instantiated L{ConsoleUI} can use. """ self.assertEqual(fname, "/dev/tty") self.assertEqual(mode, "r+b") self.assertEqual(kwargs["buffering"], 0) return self.fakeFile def setUp(self): """ Patch 'open' in verifyHostKey. """ self.fakeFile = FakeFile() self.patch(default, "_open", self.patchedOpen) self.hostsOption = self.mktemp() self.hashedEntries = {} knownHostsFile = KnownHostsFile(FilePath(self.hostsOption)) for host in (b"exists.example.com", b"4.3.2.1"): entry = knownHostsFile.addHostKey(host, Key.fromString(sampleKey)) self.hashedEntries[host] = entry knownHostsFile.save() self.fakeTransport = FakeObject() self.fakeTransport.factory = FakeObject() self.options = self.fakeTransport.factory.options = { "host": b"exists.example.com", "known-hosts": self.hostsOption, } def test_verifyOKKey(self): """ L{default.verifyHostKey} should return a L{Deferred} which fires with C{1} when passed a host, IP, and key which already match the known_hosts file it is supposed to check. """ l = [] default.verifyHostKey( self.fakeTransport, b"4.3.2.1", sampleKey, b"I don't care." ).addCallback(l.append) self.assertEqual([1], l) def replaceHome(self, tempHome): """ Replace the HOME environment variable until the end of the current test, with the given new home-directory, so that L{os.path.expanduser} will yield controllable, predictable results. @param tempHome: the pathname to replace the HOME variable with. @type tempHome: L{str} """ oldHome = os.environ.get("HOME") def cleanupHome(): if oldHome is None: del os.environ["HOME"] else: os.environ["HOME"] = oldHome self.addCleanup(cleanupHome) os.environ["HOME"] = tempHome def test_noKnownHostsOption(self): """ L{default.verifyHostKey} should find your known_hosts file in ~/.ssh/known_hosts if you don't specify one explicitly on the command line. """ l = [] tmpdir = self.mktemp() oldHostsOption = self.hostsOption hostsNonOption = FilePath(tmpdir).child(".ssh").child("known_hosts") hostsNonOption.parent().makedirs() FilePath(oldHostsOption).moveTo(hostsNonOption) self.replaceHome(tmpdir) self.options["known-hosts"] = None default.verifyHostKey( self.fakeTransport, b"4.3.2.1", sampleKey, b"I don't care." ).addCallback(l.append) self.assertEqual([1], l) def test_verifyHostButNotIP(self): """ L{default.verifyHostKey} should return a L{Deferred} which fires with C{1} when passed a host which matches with an IP is not present in its known_hosts file, and should also warn the user that it has added the IP address. """ l = [] default.verifyHostKey( self.fakeTransport, b"8.7.6.5", sampleKey, b"Fingerprint not required." ).addCallback(l.append) self.assertEqual( [ "Warning: Permanently added the RSA host key for IP address " "'8.7.6.5' to the list of known hosts." ], self.fakeFile.outchunks, ) self.assertEqual([1], l) knownHostsFile = KnownHostsFile.fromPath(FilePath(self.hostsOption)) self.assertTrue( knownHostsFile.hasHostKey(b"8.7.6.5", Key.fromString(sampleKey)) ) def test_verifyQuestion(self): """ L{default.verifyHostKey} should return a L{Default} which fires with C{0} when passed an unknown host that the user refuses to acknowledge. """ self.fakeTransport.factory.options["host"] = b"fake.example.com" self.fakeFile.inlines.append(b"no") d = default.verifyHostKey( self.fakeTransport, b"9.8.7.6", otherSampleKey, b"No fingerprint!" ) self.assertEqual( [ b"The authenticity of host 'fake.example.com (9.8.7.6)' " b"can't be established.\n" b"RSA key fingerprint is " b"SHA256:vD0YydsNIUYJa7yLZl3tIL8h0vZvQ8G+HPG7JLmQV0s=.\n" b"Are you sure you want to continue connecting (yes/no)? " ], self.fakeFile.outchunks, ) return self.assertFailure(d, UserRejectedKey) def test_verifyBadKey(self): """ L{default.verifyHostKey} should return a L{Deferred} which fails with L{HostKeyChanged} if the host key is incorrect. """ d = default.verifyHostKey( self.fakeTransport, b"4.3.2.1", otherSampleKey, "Again, not required." ) return self.assertFailure(d, HostKeyChanged) def test_inKnownHosts(self): """ L{default.isInKnownHosts} should return C{1} when a host with a key is in the known hosts file. """ host = self.hashedEntries[b"4.3.2.1"].toString().split()[0] r = default.isInKnownHosts( host, Key.fromString(sampleKey).blob(), {"known-hosts": FilePath(self.hostsOption).path}, ) self.assertEqual(1, r) def test_notInKnownHosts(self): """ L{default.isInKnownHosts} should return C{0} when a host with a key is not in the known hosts file. """ r = default.isInKnownHosts( "not.there", b"irrelevant", {"known-hosts": FilePath(self.hostsOption).path} ) self.assertEqual(0, r) def test_inKnownHostsKeyChanged(self): """ L{default.isInKnownHosts} should return C{2} when a host with a key other than the given one is in the known hosts file. """ host = self.hashedEntries[b"4.3.2.1"].toString().split()[0] r = default.isInKnownHosts( host, Key.fromString(otherSampleKey).blob(), {"known-hosts": FilePath(self.hostsOption).path}, ) self.assertEqual(2, r) PK F��Z碢%Ir Ir test_connection.pynu �[��� # Copyright (c) 2007-2010 Twisted Matrix Laboratories. # See LICENSE for details """ This module tests twisted.conch.ssh.connection. """ import struct from twisted.conch.ssh import channel from twisted.conch.test import test_userauth from twisted.python.reflect import requireModule from twisted.trial import unittest cryptography = requireModule("cryptography") from twisted.conch import error if cryptography: from twisted.conch.ssh import common, connection else: class connection: # type: ignore[no-redef] class SSHConnection: pass class TestChannel(channel.SSHChannel): """ A mocked-up version of twisted.conch.ssh.channel.SSHChannel. @ivar gotOpen: True if channelOpen has been called. @type gotOpen: L{bool} @ivar specificData: the specific channel open data passed to channelOpen. @type specificData: L{bytes} @ivar openFailureReason: the reason passed to openFailed. @type openFailed: C{error.ConchError} @ivar inBuffer: a C{list} of strings received by the channel. @type inBuffer: C{list} @ivar extBuffer: a C{list} of 2-tuples (type, extended data) of received by the channel. @type extBuffer: C{list} @ivar numberRequests: the number of requests that have been made to this channel. @type numberRequests: L{int} @ivar gotEOF: True if the other side sent EOF. @type gotEOF: L{bool} @ivar gotOneClose: True if the other side closed the connection. @type gotOneClose: L{bool} @ivar gotClosed: True if the channel is closed. @type gotClosed: L{bool} """ name = b"TestChannel" gotOpen = False gotClosed = False def logPrefix(self): return "TestChannel %i" % self.id def channelOpen(self, specificData): """ The channel is open. Set up the instance variables. """ self.gotOpen = True self.specificData = specificData self.inBuffer = [] self.extBuffer = [] self.numberRequests = 0 self.gotEOF = False self.gotOneClose = False self.gotClosed = False def openFailed(self, reason): """ Opening the channel failed. Store the reason why. """ self.openFailureReason = reason def request_test(self, data): """ A test request. Return True if data is 'data'. @type data: L{bytes} """ self.numberRequests += 1 return data == b"data" def dataReceived(self, data): """ Data was received. Store it in the buffer. """ self.inBuffer.append(data) def extReceived(self, code, data): """ Extended data was received. Store it in the buffer. """ self.extBuffer.append((code, data)) def eofReceived(self): """ EOF was received. Remember it. """ self.gotEOF = True def closeReceived(self): """ Close was received. Remember it. """ self.gotOneClose = True def closed(self): """ The channel is closed. Rembember it. """ self.gotClosed = True class TestAvatar: """ A mocked-up version of twisted.conch.avatar.ConchUser """ _ARGS_ERROR_CODE = 123 def lookupChannel(self, channelType, windowSize, maxPacket, data): """ The server wants us to return a channel. If the requested channel is our TestChannel, return it, otherwise return None. """ if channelType == TestChannel.name: return TestChannel( remoteWindow=windowSize, remoteMaxPacket=maxPacket, data=data, avatar=self, ) elif channelType == b"conch-error-args": # Raise a ConchError with backwards arguments to make sure the # connection fixes it for us. This case should be deprecated and # deleted eventually, but only after all of Conch gets the argument # order right. raise error.ConchError(self._ARGS_ERROR_CODE, "error args in wrong order") def gotGlobalRequest(self, requestType, data): """ The client has made a global request. If the global request is 'TestGlobal', return True. If the global request is 'TestData', return True and the request-specific data we received. Otherwise, return False. """ if requestType == b"TestGlobal": return True elif requestType == b"TestData": return True, data else: return False class TestConnection(connection.SSHConnection): """ A subclass of SSHConnection for testing. @ivar channel: the current channel. @type channel. C{TestChannel} """ if not cryptography: skip = "Cannot run without cryptography" def logPrefix(self): return "TestConnection" def global_TestGlobal(self, data): """ The other side made the 'TestGlobal' global request. Return True. """ return True def global_Test_Data(self, data): """ The other side made the 'Test-Data' global request. Return True and the data we received. """ return True, data def channel_TestChannel(self, windowSize, maxPacket, data): """ The other side is requesting the TestChannel. Create a C{TestChannel} instance, store it, and return it. """ self.channel = TestChannel( remoteWindow=windowSize, remoteMaxPacket=maxPacket, data=data ) return self.channel def channel_ErrorChannel(self, windowSize, maxPacket, data): """ The other side is requesting the ErrorChannel. Raise an exception. """ raise AssertionError("no such thing") class ConnectionTests(unittest.TestCase): if not cryptography: skip = "Cannot run without cryptography" def setUp(self): self.transport = test_userauth.FakeTransport(None) self.transport.avatar = TestAvatar() self.conn = TestConnection() self.conn.transport = self.transport self.conn.serviceStarted() def _openChannel(self, channel): """ Open the channel with the default connection. """ self.conn.openChannel(channel) self.transport.packets = self.transport.packets[:-1] self.conn.ssh_CHANNEL_OPEN_CONFIRMATION( struct.pack(">2L", channel.id, 255) + b"\x00\x02\x00\x00\x00\x00\x80\x00" ) def tearDown(self): self.conn.serviceStopped() def test_linkAvatar(self): """ Test that the connection links itself to the avatar in the transport. """ self.assertIs(self.transport.avatar.conn, self.conn) def test_serviceStopped(self): """ Test that serviceStopped() closes any open channels. """ channel1 = TestChannel() channel2 = TestChannel() self.conn.openChannel(channel1) self.conn.openChannel(channel2) self.conn.ssh_CHANNEL_OPEN_CONFIRMATION(b"\x00\x00\x00\x00" * 4) self.assertTrue(channel1.gotOpen) self.assertFalse(channel1.gotClosed) self.assertFalse(channel2.gotOpen) self.assertFalse(channel2.gotClosed) self.conn.serviceStopped() self.assertTrue(channel1.gotClosed) self.assertFalse(channel2.gotOpen) self.assertFalse(channel2.gotClosed) from twisted.internet.error import ConnectionLost self.assertIsInstance(channel2.openFailureReason, ConnectionLost) def test_GLOBAL_REQUEST(self): """ Test that global request packets are dispatched to the global_* methods and the return values are translated into success or failure messages. """ self.conn.ssh_GLOBAL_REQUEST(common.NS(b"TestGlobal") + b"\xff") self.assertEqual( self.transport.packets, [(connection.MSG_REQUEST_SUCCESS, b"")] ) self.transport.packets = [] self.conn.ssh_GLOBAL_REQUEST(common.NS(b"TestData") + b"\xff" + b"test data") self.assertEqual( self.transport.packets, [(connection.MSG_REQUEST_SUCCESS, b"test data")] ) self.transport.packets = [] self.conn.ssh_GLOBAL_REQUEST(common.NS(b"TestBad") + b"\xff") self.assertEqual( self.transport.packets, [(connection.MSG_REQUEST_FAILURE, b"")] ) self.transport.packets = [] self.conn.ssh_GLOBAL_REQUEST(common.NS(b"TestGlobal") + b"\x00") self.assertEqual(self.transport.packets, []) def test_REQUEST_SUCCESS(self): """ Test that global request success packets cause the Deferred to be called back. """ d = self.conn.sendGlobalRequest(b"request", b"data", True) self.conn.ssh_REQUEST_SUCCESS(b"data") def check(data): self.assertEqual(data, b"data") d.addCallback(check) d.addErrback(self.fail) return d def test_REQUEST_FAILURE(self): """ Test that global request failure packets cause the Deferred to be erred back. """ d = self.conn.sendGlobalRequest(b"request", b"data", True) self.conn.ssh_REQUEST_FAILURE(b"data") def check(f): self.assertEqual(f.value.data, b"data") d.addCallback(self.fail) d.addErrback(check) return d def test_CHANNEL_OPEN(self): """ Test that open channel packets cause a channel to be created and opened or a failure message to be returned. """ del self.transport.avatar self.conn.ssh_CHANNEL_OPEN(common.NS(b"TestChannel") + b"\x00\x00\x00\x01" * 4) self.assertTrue(self.conn.channel.gotOpen) self.assertEqual(self.conn.channel.conn, self.conn) self.assertEqual(self.conn.channel.data, b"\x00\x00\x00\x01") self.assertEqual(self.conn.channel.specificData, b"\x00\x00\x00\x01") self.assertEqual(self.conn.channel.remoteWindowLeft, 1) self.assertEqual(self.conn.channel.remoteMaxPacket, 1) self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_OPEN_CONFIRMATION, b"\x00\x00\x00\x01\x00\x00\x00\x00\x00\x02\x00\x00" b"\x00\x00\x80\x00", ) ], ) self.transport.packets = [] self.conn.ssh_CHANNEL_OPEN(common.NS(b"BadChannel") + b"\x00\x00\x00\x02" * 4) self.flushLoggedErrors() self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_OPEN_FAILURE, b"\x00\x00\x00\x02\x00\x00\x00\x03" + common.NS(b"unknown channel") + common.NS(b""), ) ], ) self.transport.packets = [] self.conn.ssh_CHANNEL_OPEN(common.NS(b"ErrorChannel") + b"\x00\x00\x00\x02" * 4) self.flushLoggedErrors() self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_OPEN_FAILURE, b"\x00\x00\x00\x02\x00\x00\x00\x02" + common.NS(b"unknown failure") + common.NS(b""), ) ], ) def _lookupChannelErrorTest(self, code): """ Deliver a request for a channel open which will result in an exception being raised during channel lookup. Assert that an error response is delivered as a result. """ self.transport.avatar._ARGS_ERROR_CODE = code self.conn.ssh_CHANNEL_OPEN( common.NS(b"conch-error-args") + b"\x00\x00\x00\x01" * 4 ) errors = self.flushLoggedErrors(error.ConchError) self.assertEqual(len(errors), 1, f"Expected one error, got: {errors!r}") self.assertEqual(errors[0].value.args, (123, "error args in wrong order")) self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_OPEN_FAILURE, # The response includes some bytes which identifying the # associated request, as well as the error code (7b in hex) and # the error message. b"\x00\x00\x00\x01\x00\x00\x00\x7b" + common.NS(b"error args in wrong order") + common.NS(b""), ) ], ) def test_lookupChannelError(self): """ If a C{lookupChannel} implementation raises L{error.ConchError} with the arguments in the wrong order, a C{MSG_CHANNEL_OPEN} failure is still sent in response to the message. This is a temporary work-around until L{error.ConchError} is given better attributes and all of the Conch code starts constructing instances of it properly. Eventually this functionality should be deprecated and then removed. """ self._lookupChannelErrorTest(123) def test_CHANNEL_OPEN_CONFIRMATION(self): """ Test that channel open confirmation packets cause the channel to be notified that it's open. """ channel = TestChannel() self.conn.openChannel(channel) self.conn.ssh_CHANNEL_OPEN_CONFIRMATION(b"\x00\x00\x00\x00" * 5) self.assertEqual(channel.remoteWindowLeft, 0) self.assertEqual(channel.remoteMaxPacket, 0) self.assertEqual(channel.specificData, b"\x00\x00\x00\x00") self.assertEqual(self.conn.channelsToRemoteChannel[channel], 0) self.assertEqual(self.conn.localToRemoteChannel[0], 0) def test_CHANNEL_OPEN_FAILURE(self): """ Test that channel open failure packets cause the channel to be notified that its opening failed. """ channel = TestChannel() self.conn.openChannel(channel) self.conn.ssh_CHANNEL_OPEN_FAILURE( b"\x00\x00\x00\x00\x00\x00\x00" b"\x01" + common.NS(b"failure!") ) self.assertEqual(channel.openFailureReason.args, (b"failure!", 1)) self.assertIsNone(self.conn.channels.get(channel)) def test_CHANNEL_WINDOW_ADJUST(self): """ Test that channel window adjust messages add bytes to the channel window. """ channel = TestChannel() self._openChannel(channel) oldWindowSize = channel.remoteWindowLeft self.conn.ssh_CHANNEL_WINDOW_ADJUST(b"\x00\x00\x00\x00\x00\x00\x00" b"\x01") self.assertEqual(channel.remoteWindowLeft, oldWindowSize + 1) def test_CHANNEL_DATA(self): """ Test that channel data messages are passed up to the channel, or cause the channel to be closed if the data is too large. """ channel = TestChannel(localWindow=6, localMaxPacket=5) self._openChannel(channel) self.conn.ssh_CHANNEL_DATA(b"\x00\x00\x00\x00" + common.NS(b"data")) self.assertEqual(channel.inBuffer, [b"data"]) self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_WINDOW_ADJUST, b"\x00\x00\x00\xff" b"\x00\x00\x00\x04", ) ], ) self.transport.packets = [] longData = b"a" * (channel.localWindowLeft + 1) self.conn.ssh_CHANNEL_DATA(b"\x00\x00\x00\x00" + common.NS(longData)) self.assertEqual(channel.inBuffer, [b"data"]) self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_CLOSE, b"\x00\x00\x00\xff")], ) channel = TestChannel() self._openChannel(channel) bigData = b"a" * (channel.localMaxPacket + 1) self.transport.packets = [] self.conn.ssh_CHANNEL_DATA(b"\x00\x00\x00\x01" + common.NS(bigData)) self.assertEqual(channel.inBuffer, []) self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_CLOSE, b"\x00\x00\x00\xff")], ) def test_CHANNEL_EXTENDED_DATA(self): """ Test that channel extended data messages are passed up to the channel, or cause the channel to be closed if they're too big. """ channel = TestChannel(localWindow=6, localMaxPacket=5) self._openChannel(channel) self.conn.ssh_CHANNEL_EXTENDED_DATA( b"\x00\x00\x00\x00\x00\x00\x00" b"\x00" + common.NS(b"data") ) self.assertEqual(channel.extBuffer, [(0, b"data")]) self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_WINDOW_ADJUST, b"\x00\x00\x00\xff" b"\x00\x00\x00\x04", ) ], ) self.transport.packets = [] longData = b"a" * (channel.localWindowLeft + 1) self.conn.ssh_CHANNEL_EXTENDED_DATA( b"\x00\x00\x00\x00\x00\x00\x00" b"\x00" + common.NS(longData) ) self.assertEqual(channel.extBuffer, [(0, b"data")]) self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_CLOSE, b"\x00\x00\x00\xff")], ) channel = TestChannel() self._openChannel(channel) bigData = b"a" * (channel.localMaxPacket + 1) self.transport.packets = [] self.conn.ssh_CHANNEL_EXTENDED_DATA( b"\x00\x00\x00\x01\x00\x00\x00" b"\x00" + common.NS(bigData) ) self.assertEqual(channel.extBuffer, []) self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_CLOSE, b"\x00\x00\x00\xff")], ) def test_CHANNEL_EOF(self): """ Test that channel eof messages are passed up to the channel. """ channel = TestChannel() self._openChannel(channel) self.conn.ssh_CHANNEL_EOF(b"\x00\x00\x00\x00") self.assertTrue(channel.gotEOF) def test_CHANNEL_CLOSE(self): """ Test that channel close messages are passed up to the channel. Also, test that channel.close() is called if both sides are closed when this message is received. """ channel = TestChannel() self._openChannel(channel) self.assertTrue(channel.gotOpen) self.assertFalse(channel.gotOneClose) self.assertFalse(channel.gotClosed) self.conn.sendClose(channel) self.conn.ssh_CHANNEL_CLOSE(b"\x00\x00\x00\x00") self.assertTrue(channel.gotOneClose) self.assertTrue(channel.gotClosed) def test_CHANNEL_REQUEST_success(self): """ Test that channel requests that succeed send MSG_CHANNEL_SUCCESS. """ channel = TestChannel() self._openChannel(channel) self.conn.ssh_CHANNEL_REQUEST( b"\x00\x00\x00\x00" + common.NS(b"test") + b"\x00" ) self.assertEqual(channel.numberRequests, 1) d = self.conn.ssh_CHANNEL_REQUEST( b"\x00\x00\x00\x00" + common.NS(b"test") + b"\xff" + b"data" ) def check(result): self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_SUCCESS, b"\x00\x00\x00\xff")], ) d.addCallback(check) return d def test_CHANNEL_REQUEST_failure(self): """ Test that channel requests that fail send MSG_CHANNEL_FAILURE. """ channel = TestChannel() self._openChannel(channel) d = self.conn.ssh_CHANNEL_REQUEST( b"\x00\x00\x00\x00" + common.NS(b"test") + b"\xff" ) def check(result): self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_FAILURE, b"\x00\x00\x00\xff")], ) d.addCallback(self.fail) d.addErrback(check) return d def test_CHANNEL_REQUEST_SUCCESS(self): """ Test that channel request success messages cause the Deferred to be called back. """ channel = TestChannel() self._openChannel(channel) d = self.conn.sendRequest(channel, b"test", b"data", True) self.conn.ssh_CHANNEL_SUCCESS(b"\x00\x00\x00\x00") def check(result): self.assertTrue(result) return d def test_CHANNEL_REQUEST_FAILURE(self): """ Test that channel request failure messages cause the Deferred to be erred back. """ channel = TestChannel() self._openChannel(channel) d = self.conn.sendRequest(channel, b"test", b"", True) self.conn.ssh_CHANNEL_FAILURE(b"\x00\x00\x00\x00") def check(result): self.assertEqual(result.value.value, "channel request failed") d.addCallback(self.fail) d.addErrback(check) return d def test_sendGlobalRequest(self): """ Test that global request messages are sent in the right format. """ d = self.conn.sendGlobalRequest(b"wantReply", b"data", True) # must be added to prevent errbacking during teardown d.addErrback(lambda failure: None) self.conn.sendGlobalRequest(b"noReply", b"", False) self.assertEqual( self.transport.packets, [ (connection.MSG_GLOBAL_REQUEST, common.NS(b"wantReply") + b"\xffdata"), (connection.MSG_GLOBAL_REQUEST, common.NS(b"noReply") + b"\x00"), ], ) self.assertEqual(self.conn.deferreds, {"global": [d]}) def test_openChannel(self): """ Test that open channel messages are sent in the right format. """ channel = TestChannel() self.conn.openChannel(channel, b"aaaa") self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_OPEN, common.NS(b"TestChannel") + b"\x00\x00\x00\x00\x00\x02\x00\x00\x00\x00\x80\x00aaaa", ) ], ) self.assertEqual(channel.id, 0) self.assertEqual(self.conn.localChannelID, 1) def test_sendRequest(self): """ Test that channel request messages are sent in the right format. """ channel = TestChannel() self._openChannel(channel) d = self.conn.sendRequest(channel, b"test", b"test", True) # needed to prevent errbacks during teardown. d.addErrback(lambda failure: None) self.conn.sendRequest(channel, b"test2", b"", False) channel.localClosed = True # emulate sending a close message self.conn.sendRequest(channel, b"test3", b"", True) self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_REQUEST, b"\x00\x00\x00\xff" + common.NS(b"test") + b"\x01test", ), ( connection.MSG_CHANNEL_REQUEST, b"\x00\x00\x00\xff" + common.NS(b"test2") + b"\x00", ), ], ) self.assertEqual(self.conn.deferreds[0], [d]) def test_adjustWindow(self): """ Test that channel window adjust messages cause bytes to be added to the window. """ channel = TestChannel(localWindow=5) self._openChannel(channel) channel.localWindowLeft = 0 self.conn.adjustWindow(channel, 1) self.assertEqual(channel.localWindowLeft, 1) channel.localClosed = True self.conn.adjustWindow(channel, 2) self.assertEqual(channel.localWindowLeft, 1) self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_WINDOW_ADJUST, b"\x00\x00\x00\xff" b"\x00\x00\x00\x01", ) ], ) def test_sendData(self): """ Test that channel data messages are sent in the right format. """ channel = TestChannel() self._openChannel(channel) self.conn.sendData(channel, b"a") channel.localClosed = True self.conn.sendData(channel, b"b") self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_DATA, b"\x00\x00\x00\xff" + common.NS(b"a"))], ) def test_sendExtendedData(self): """ Test that channel extended data messages are sent in the right format. """ channel = TestChannel() self._openChannel(channel) self.conn.sendExtendedData(channel, 1, b"test") channel.localClosed = True self.conn.sendExtendedData(channel, 2, b"test2") self.assertEqual( self.transport.packets, [ ( connection.MSG_CHANNEL_EXTENDED_DATA, b"\x00\x00\x00\xff" + b"\x00\x00\x00\x01" + common.NS(b"test"), ) ], ) def test_sendEOF(self): """ Test that channel EOF messages are sent in the right format. """ channel = TestChannel() self._openChannel(channel) self.conn.sendEOF(channel) self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_EOF, b"\x00\x00\x00\xff")] ) channel.localClosed = True self.conn.sendEOF(channel) self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_EOF, b"\x00\x00\x00\xff")] ) def test_sendClose(self): """ Test that channel close messages are sent in the right format. """ channel = TestChannel() self._openChannel(channel) self.conn.sendClose(channel) self.assertTrue(channel.localClosed) self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_CLOSE, b"\x00\x00\x00\xff")], ) self.conn.sendClose(channel) self.assertEqual( self.transport.packets, [(connection.MSG_CHANNEL_CLOSE, b"\x00\x00\x00\xff")], ) channel2 = TestChannel() self._openChannel(channel2) self.assertTrue(channel2.gotOpen) self.assertFalse(channel2.gotClosed) channel2.remoteClosed = True self.conn.sendClose(channel2) self.assertTrue(channel2.gotClosed) def test_getChannelWithAvatar(self): """ Test that getChannel dispatches to the avatar when an avatar is present. Correct functioning without the avatar is verified in test_CHANNEL_OPEN. """ channel = self.conn.getChannel(b"TestChannel", 50, 30, b"data") self.assertEqual(channel.data, b"data") self.assertEqual(channel.remoteWindowLeft, 50) self.assertEqual(channel.remoteMaxPacket, 30) self.assertRaises( error.ConchError, self.conn.getChannel, b"BadChannel", 50, 30, b"data" ) def test_gotGlobalRequestWithoutAvatar(self): """ Test that gotGlobalRequests dispatches to global_* without an avatar. """ del self.transport.avatar self.assertTrue(self.conn.gotGlobalRequest(b"TestGlobal", b"data")) self.assertEqual( self.conn.gotGlobalRequest(b"Test-Data", b"data"), (True, b"data") ) self.assertFalse(self.conn.gotGlobalRequest(b"BadGlobal", b"data")) def test_channelClosedCausesLeftoverChannelDeferredsToErrback(self): """ Whenever an SSH channel gets closed any Deferred that was returned by a sendRequest() on its parent connection must be errbacked. """ channel = TestChannel() self._openChannel(channel) d = self.conn.sendRequest(channel, b"dummyrequest", b"dummydata", wantReply=1) d = self.assertFailure(d, error.ConchError) self.conn.channelClosed(channel) return d class CleanConnectionShutdownTests(unittest.TestCase): """ Check whether correct cleanup is performed on connection shutdown. """ if not cryptography: skip = "Cannot run without cryptography" def setUp(self): self.transport = test_userauth.FakeTransport(None) self.transport.avatar = TestAvatar() self.conn = TestConnection() self.conn.transport = self.transport def test_serviceStoppedCausesLeftoverGlobalDeferredsToErrback(self): """ Once the service is stopped any leftover global deferred returned by a sendGlobalRequest() call must be errbacked. """ self.conn.serviceStarted() d = self.conn.sendGlobalRequest(b"dummyrequest", b"dummydata", wantReply=1) d = self.assertFailure(d, error.ConchError) self.conn.serviceStopped() return d PK F��Z�T�vm� m� test_filetransfer.pynu �[��� # -*- test-case-name: twisted.conch.test.test_filetransfer -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE file for details. """ Tests for L{twisted.conch.ssh.filetransfer}. """ import os import re import struct from unittest import skipIf from hamcrest import assert_that, equal_to from twisted.internet import defer from twisted.internet.error import ConnectionLost from twisted.protocols import loopback from twisted.python import components from twisted.python.compat import _PY37PLUS from twisted.python.filepath import FilePath from twisted.test.proto_helpers import StringTransport from twisted.trial.unittest import TestCase try: from twisted.conch import unix as _unix except ImportError: unix = None else: unix = _unix try: from twisted.conch.unix import ( SFTPServerForUnixConchUser as _SFTPServerForUnixConchUser, ) except ImportError: SFTPServerForUnixConchUser = None else: SFTPServerForUnixConchUser = _SFTPServerForUnixConchUser try: import cryptography as _cryptography except ImportError: cryptography = None else: cryptography = _cryptography try: from twisted.conch.avatar import ConchUser as _ConchUser except ImportError: ConchUser = object else: ConchUser = _ConchUser # type: ignore[misc] try: from twisted.conch.ssh import common, connection, filetransfer, session except ImportError: pass class TestAvatar(ConchUser): def __init__(self): ConchUser.__init__(self) self.channelLookup[b"session"] = session.SSHSession self.subsystemLookup[b"sftp"] = filetransfer.FileTransferServer def _runAsUser(self, f, *args, **kw): try: f = iter(f) except TypeError: f = [(f, args, kw)] for i in f: func = i[0] args = len(i) > 1 and i[1] or () kw = len(i) > 2 and i[2] or {} r = func(*args, **kw) return r class FileTransferTestAvatar(TestAvatar): def __init__(self, homeDir): TestAvatar.__init__(self) self.homeDir = homeDir def getHomeDir(self): return FilePath(os.getcwd()).preauthChild(self.homeDir.path) class ConchSessionForTestAvatar: def __init__(self, avatar): self.avatar = avatar if SFTPServerForUnixConchUser is None: # unix should either be a fully working module, or None. I'm not sure # how this happens, but on win32 it does. Try to cope. --spiv. import warnings warnings.warn( ( "twisted.conch.unix imported %r, " "but doesn't define SFTPServerForUnixConchUser'" ) % (unix,) ) else: class FileTransferForTestAvatar(SFTPServerForUnixConchUser): # type: ignore[misc,valid-type] def gotVersion(self, version, otherExt): return {b"conchTest": b"ext data"} def extendedRequest(self, extName, extData): if extName == b"testExtendedRequest": return b"bar" raise NotImplementedError components.registerAdapter( FileTransferForTestAvatar, TestAvatar, filetransfer.ISFTPServer ) class SFTPTestBase(TestCase): def setUp(self): self.testDir = FilePath(self.mktemp()) # Give the testDir another level so we can safely "cd .." from it in # tests. self.testDir = self.testDir.child("extra") self.testDir.child("testDirectory").makedirs(True) with self.testDir.child("testfile1").open(mode="wb") as f: f.write(b"a" * 10 + b"b" * 10) with open("/dev/urandom", "rb") as f2: f.write(f2.read(1024 * 64)) # random data self.testDir.child("testfile1").chmod(0o644) with self.testDir.child("testRemoveFile").open(mode="wb") as f: f.write(b"a") with self.testDir.child("testRenameFile").open(mode="wb") as f: f.write(b"a") with self.testDir.child(".testHiddenFile").open(mode="wb") as f: f.write(b"a") @skipIf(not unix, "can't run on non-posix computers") class OurServerOurClientTests(SFTPTestBase): def setUp(self): SFTPTestBase.setUp(self) self.avatar = FileTransferTestAvatar(self.testDir) self.server = filetransfer.FileTransferServer(avatar=self.avatar) clientTransport = loopback.LoopbackRelay(self.server) self.client = filetransfer.FileTransferClient() self._serverVersion = None self._extData = None def _(serverVersion, extData): self._serverVersion = serverVersion self._extData = extData self.client.gotServerVersion = _ serverTransport = loopback.LoopbackRelay(self.client) self.client.makeConnection(clientTransport) self.server.makeConnection(serverTransport) self.clientTransport = clientTransport self.serverTransport = serverTransport self._emptyBuffers() def _emptyBuffers(self): while self.serverTransport.buffer or self.clientTransport.buffer: self.serverTransport.clearBuffer() self.clientTransport.clearBuffer() def tearDown(self): self.serverTransport.loseConnection() self.clientTransport.loseConnection() self.serverTransport.clearBuffer() self.clientTransport.clearBuffer() def test_serverVersion(self): self.assertEqual(self._serverVersion, 3) self.assertEqual(self._extData, {b"conchTest": b"ext data"}) def test_interface_implementation(self): """ It implements the ISFTPServer interface. """ self.assertTrue( filetransfer.ISFTPServer.providedBy(self.server.client), f"ISFTPServer not provided by {self.server.client!r}", ) def test_openedFileClosedWithConnection(self): """ A file opened with C{openFile} is closed when the connection is lost. """ d = self.client.openFile( b"testfile1", filetransfer.FXF_READ | filetransfer.FXF_WRITE, {} ) self._emptyBuffers() oldClose = os.close closed = [] def close(fd): closed.append(fd) oldClose(fd) self.patch(os, "close", close) def _fileOpened(openFile): fd = self.server.openFiles[openFile.handle[4:]].fd self.serverTransport.loseConnection() self.clientTransport.loseConnection() self.serverTransport.clearBuffer() self.clientTransport.clearBuffer() self.assertEqual(self.server.openFiles, {}) self.assertIn(fd, closed) d.addCallback(_fileOpened) return d def test_openedDirectoryClosedWithConnection(self): """ A directory opened with C{openDirectory} is close when the connection is lost. """ d = self.client.openDirectory("") self._emptyBuffers() def _getFiles(openDir): self.serverTransport.loseConnection() self.clientTransport.loseConnection() self.serverTransport.clearBuffer() self.clientTransport.clearBuffer() self.assertEqual(self.server.openDirs, {}) d.addCallback(_getFiles) return d def test_openFileIO(self): d = self.client.openFile( b"testfile1", filetransfer.FXF_READ | filetransfer.FXF_WRITE, {} ) self._emptyBuffers() def _fileOpened(openFile): self.assertEqual(openFile, filetransfer.ISFTPFile(openFile)) d = _readChunk(openFile) d.addCallback(_writeChunk, openFile) return d def _readChunk(openFile): d = openFile.readChunk(0, 20) self._emptyBuffers() d.addCallback(self.assertEqual, b"a" * 10 + b"b" * 10) return d def _writeChunk(_, openFile): d = openFile.writeChunk(20, b"c" * 10) self._emptyBuffers() d.addCallback(_readChunk2, openFile) return d def _readChunk2(_, openFile): d = openFile.readChunk(0, 30) self._emptyBuffers() d.addCallback(self.assertEqual, b"a" * 10 + b"b" * 10 + b"c" * 10) return d d.addCallback(_fileOpened) return d def test_closedFileGetAttrs(self): d = self.client.openFile( b"testfile1", filetransfer.FXF_READ | filetransfer.FXF_WRITE, {} ) self._emptyBuffers() def _getAttrs(_, openFile): d = openFile.getAttrs() self._emptyBuffers() return d def _err(f): self.flushLoggedErrors() return f def _close(openFile): d = openFile.close() self._emptyBuffers() d.addCallback(_getAttrs, openFile) d.addErrback(_err) return self.assertFailure(d, filetransfer.SFTPError) d.addCallback(_close) return d def test_openFileAttributes(self): d = self.client.openFile( b"testfile1", filetransfer.FXF_READ | filetransfer.FXF_WRITE, {} ) self._emptyBuffers() def _getAttrs(openFile): d = openFile.getAttrs() self._emptyBuffers() d.addCallback(_getAttrs2) return d def _getAttrs2(attrs1): d = self.client.getAttrs(b"testfile1") self._emptyBuffers() d.addCallback(self.assertEqual, attrs1) return d return d.addCallback(_getAttrs) def test_openFileSetAttrs(self): # XXX test setAttrs # Ok, how about this for a start? It caught a bug :) -- spiv. d = self.client.openFile( b"testfile1", filetransfer.FXF_READ | filetransfer.FXF_WRITE, {} ) self._emptyBuffers() def _getAttrs(openFile): d = openFile.getAttrs() self._emptyBuffers() d.addCallback(_setAttrs) return d def _setAttrs(attrs): attrs["atime"] = 0 d = self.client.setAttrs(b"testfile1", attrs) self._emptyBuffers() d.addCallback(_getAttrs2) d.addCallback(self.assertEqual, attrs) return d def _getAttrs2(_): d = self.client.getAttrs(b"testfile1") self._emptyBuffers() return d d.addCallback(_getAttrs) return d def test_openFileExtendedAttributes(self): """ Check that L{filetransfer.FileTransferClient.openFile} can send extended attributes, that should be extracted server side. By default, they are ignored, so we just verify they are correctly parsed. """ savedAttributes = {} oldOpenFile = self.server.client.openFile def openFile(filename, flags, attrs): savedAttributes.update(attrs) return oldOpenFile(filename, flags, attrs) self.server.client.openFile = openFile d = self.client.openFile( b"testfile1", filetransfer.FXF_READ | filetransfer.FXF_WRITE, {"ext_foo": b"bar"}, ) self._emptyBuffers() def check(ign): self.assertEqual(savedAttributes, {"ext_foo": b"bar"}) return d.addCallback(check) def test_removeFile(self): d = self.client.getAttrs(b"testRemoveFile") self._emptyBuffers() def _removeFile(ignored): d = self.client.removeFile(b"testRemoveFile") self._emptyBuffers() return d d.addCallback(_removeFile) d.addCallback(_removeFile) return self.assertFailure(d, filetransfer.SFTPError) def test_renameFile(self): d = self.client.getAttrs(b"testRenameFile") self._emptyBuffers() def _rename(attrs): d = self.client.renameFile(b"testRenameFile", b"testRenamedFile") self._emptyBuffers() d.addCallback(_testRenamed, attrs) return d def _testRenamed(_, attrs): d = self.client.getAttrs(b"testRenamedFile") self._emptyBuffers() d.addCallback(self.assertEqual, attrs) return d.addCallback(_rename) def test_directoryBad(self): d = self.client.getAttrs(b"testMakeDirectory") self._emptyBuffers() return self.assertFailure(d, filetransfer.SFTPError) def test_directoryCreation(self): d = self.client.makeDirectory(b"testMakeDirectory", {}) self._emptyBuffers() def _getAttrs(_): d = self.client.getAttrs(b"testMakeDirectory") self._emptyBuffers() return d # XXX not until version 4/5 # self.assertEqual(filetransfer.FILEXFER_TYPE_DIRECTORY&attrs['type'], # filetransfer.FILEXFER_TYPE_DIRECTORY) def _removeDirectory(_): d = self.client.removeDirectory(b"testMakeDirectory") self._emptyBuffers() return d d.addCallback(_getAttrs) d.addCallback(_removeDirectory) d.addCallback(_getAttrs) return self.assertFailure(d, filetransfer.SFTPError) def test_openDirectory(self): d = self.client.openDirectory(b"") self._emptyBuffers() files = [] def _getFiles(openDir): def append(f): files.append(f) return openDir d = defer.maybeDeferred(openDir.next) self._emptyBuffers() d.addCallback(append) d.addCallback(_getFiles) d.addErrback(_close, openDir) return d def _checkFiles(ignored): fs = list(list(zip(*files))[0]) fs.sort() self.assertEqual( fs, [ b".testHiddenFile", b"testDirectory", b"testRemoveFile", b"testRenameFile", b"testfile1", ], ) def _close(_, openDir): d = openDir.close() self._emptyBuffers() return d d.addCallback(_getFiles) d.addCallback(_checkFiles) return d def test_linkDoesntExist(self): d = self.client.getAttrs(b"testLink") self._emptyBuffers() return self.assertFailure(d, filetransfer.SFTPError) def test_linkSharesAttrs(self): d = self.client.makeLink(b"testLink", b"testfile1") self._emptyBuffers() def _getFirstAttrs(_): d = self.client.getAttrs(b"testLink", 1) self._emptyBuffers() return d def _getSecondAttrs(firstAttrs): d = self.client.getAttrs(b"testfile1") self._emptyBuffers() d.addCallback(self.assertEqual, firstAttrs) return d d.addCallback(_getFirstAttrs) return d.addCallback(_getSecondAttrs) def test_linkPath(self): d = self.client.makeLink(b"testLink", b"testfile1") self._emptyBuffers() def _readLink(_): d = self.client.readLink(b"testLink") self._emptyBuffers() testFile = FilePath(os.getcwd()).preauthChild(self.testDir.path) testFile = testFile.child("testfile1") d.addCallback(self.assertEqual, testFile.path) return d def _realPath(_): d = self.client.realPath(b"testLink") self._emptyBuffers() testLink = FilePath(os.getcwd()).preauthChild(self.testDir.path) testLink = testLink.child("testfile1") d.addCallback(self.assertEqual, testLink.path) return d d.addCallback(_readLink) d.addCallback(_realPath) return d def test_extendedRequest(self): d = self.client.extendedRequest(b"testExtendedRequest", b"foo") self._emptyBuffers() d.addCallback(self.assertEqual, b"bar") d.addCallback(self._cbTestExtendedRequest) return d def _cbTestExtendedRequest(self, ignored): d = self.client.extendedRequest(b"testBadRequest", b"") self._emptyBuffers() return self.assertFailure(d, NotImplementedError) @defer.inlineCallbacks @skipIf(_PY37PLUS, "Broken by PEP 479 and deprecated.") def test_openDirectoryIterator(self): """ Check that the object returned by L{filetransfer.FileTransferClient.openDirectory} can be used as an iterator. """ # This function is a little more complicated than it would be # normally, since we need to call _emptyBuffers() after # creating any SSH-related Deferreds, but before waiting on # them via yield. d = self.client.openDirectory(b"") self._emptyBuffers() openDir = yield d filenames = set() try: for f in openDir: self._emptyBuffers() (filename, _, fileattrs) = yield f filenames.add(filename) finally: d = openDir.close() self._emptyBuffers() yield d self._emptyBuffers() self.assertEqual( filenames, { b".testHiddenFile", b"testDirectory", b"testRemoveFile", b"testRenameFile", b"testfile1", }, ) @defer.inlineCallbacks def test_openDirectoryIteratorDeprecated(self): """ Using client.openDirectory as an iterator is deprecated. """ d = self.client.openDirectory(b"") self._emptyBuffers() openDir = yield d oneFile = openDir.next() self._emptyBuffers() yield oneFile warnings = self.flushWarnings() message = ( "Using twisted.conch.ssh.filetransfer.ClientDirectory" " as an iterator was deprecated in Twisted 18.9.0." ) self.assertEqual(1, len(warnings)) self.assertEqual(DeprecationWarning, warnings[0]["category"]) self.assertEqual(message, warnings[0]["message"]) @defer.inlineCallbacks def test_closedConnectionCancelsRequests(self): """ If there are requests outstanding when the connection is closed for any reason, they should fail. """ d = self.client.openFile(b"testfile1", filetransfer.FXF_READ, {}) self._emptyBuffers() fh = yield d # Intercept the handling of the read request on the server side gotReadRequest = [] def _slowRead(offset, length): self.assertEqual(gotReadRequest, []) d = defer.Deferred() gotReadRequest.append(offset) return d [serverSideFh] = self.server.openFiles.values() serverSideFh.readChunk = _slowRead del serverSideFh # Make a read request, dropping the connection before the reply # is sent d = fh.readChunk(100, 200) self._emptyBuffers() self.assertEqual(len(gotReadRequest), 1) self.assertNoResult(d) # Lost connection should cause an errback self.serverTransport.loseConnection() self.serverTransport.clearBuffer() self.clientTransport.clearBuffer() self._emptyBuffers() self.assertFalse(self.client.connected) self.failureResultOf(d, ConnectionLost) # Further attempts to use the filetransfer session should fail # immediately d = fh.getAttrs() self.failureResultOf(d, ConnectionLost) class FakeConn: def sendClose(self, channel): pass @skipIf(not unix, "can't run on non-posix computers") class FileTransferCloseTests(TestCase): def setUp(self): self.avatar = TestAvatar() def buildServerConnection(self): # make a server connection conn = connection.SSHConnection() # server connections have a 'self.transport.avatar'. class DummyTransport: def __init__(self): self.transport = self def sendPacket(self, kind, data): pass def logPrefix(self): return "dummy transport" conn.transport = DummyTransport() conn.transport.avatar = self.avatar return conn def interceptConnectionLost(self, sftpServer): self.connectionLostFired = False origConnectionLost = sftpServer.connectionLost def connectionLost(reason): self.connectionLostFired = True origConnectionLost(reason) sftpServer.connectionLost = connectionLost def assertSFTPConnectionLost(self): self.assertTrue( self.connectionLostFired, "sftpServer's connectionLost was not called" ) def test_sessionClose(self): """ Closing a session should notify an SFTP subsystem launched by that session. """ # make a session testSession = session.SSHSession(conn=FakeConn(), avatar=self.avatar) # start an SFTP subsystem on the session testSession.request_subsystem(common.NS(b"sftp")) sftpServer = testSession.client.transport.proto # intercept connectionLost so we can check that it's called self.interceptConnectionLost(sftpServer) # close session testSession.closeReceived() self.assertSFTPConnectionLost() def test_clientClosesChannelOnConnnection(self): """ A client sending CHANNEL_CLOSE should trigger closeReceived on the associated channel instance. """ conn = self.buildServerConnection() # somehow get a session packet = common.NS(b"session") + struct.pack(">L", 0) * 3 conn.ssh_CHANNEL_OPEN(packet) sessionChannel = conn.channels[0] sessionChannel.request_subsystem(common.NS(b"sftp")) sftpServer = sessionChannel.client.transport.proto self.interceptConnectionLost(sftpServer) # intercept closeReceived self.interceptConnectionLost(sftpServer) # close the connection conn.ssh_CHANNEL_CLOSE(struct.pack(">L", 0)) self.assertSFTPConnectionLost() def test_stopConnectionServiceClosesChannel(self): """ Closing an SSH connection should close all sessions within it. """ conn = self.buildServerConnection() # somehow get a session packet = common.NS(b"session") + struct.pack(">L", 0) * 3 conn.ssh_CHANNEL_OPEN(packet) sessionChannel = conn.channels[0] sessionChannel.request_subsystem(common.NS(b"sftp")) sftpServer = sessionChannel.client.transport.proto self.interceptConnectionLost(sftpServer) # close the connection conn.serviceStopped() self.assertSFTPConnectionLost() @skipIf(not cryptography, "Cannot run without cryptography") class ConstantsTests(TestCase): """ Tests for the constants used by the SFTP protocol implementation. @ivar filexferSpecExcerpts: Excerpts from the draft-ietf-secsh-filexfer-02.txt (draft) specification of the SFTP protocol. There are more recent drafts of the specification, but this one describes version 3, which is what conch (and OpenSSH) implements. """ filexferSpecExcerpts = [ """ The following values are defined for packet types. #define SSH_FXP_INIT 1 #define SSH_FXP_VERSION 2 #define SSH_FXP_OPEN 3 #define SSH_FXP_CLOSE 4 #define SSH_FXP_READ 5 #define SSH_FXP_WRITE 6 #define SSH_FXP_LSTAT 7 #define SSH_FXP_FSTAT 8 #define SSH_FXP_SETSTAT 9 #define SSH_FXP_FSETSTAT 10 #define SSH_FXP_OPENDIR 11 #define SSH_FXP_READDIR 12 #define SSH_FXP_REMOVE 13 #define SSH_FXP_MKDIR 14 #define SSH_FXP_RMDIR 15 #define SSH_FXP_REALPATH 16 #define SSH_FXP_STAT 17 #define SSH_FXP_RENAME 18 #define SSH_FXP_READLINK 19 #define SSH_FXP_SYMLINK 20 #define SSH_FXP_STATUS 101 #define SSH_FXP_HANDLE 102 #define SSH_FXP_DATA 103 #define SSH_FXP_NAME 104 #define SSH_FXP_ATTRS 105 #define SSH_FXP_EXTENDED 200 #define SSH_FXP_EXTENDED_REPLY 201 Additional packet types should only be defined if the protocol version number (see Section ``Protocol Initialization'') is incremented, and their use MUST be negotiated using the version number. However, the SSH_FXP_EXTENDED and SSH_FXP_EXTENDED_REPLY packets can be used to implement vendor-specific extensions. See Section ``Vendor-Specific-Extensions'' for more details. """, """ The flags bits are defined to have the following values: #define SSH_FILEXFER_ATTR_SIZE 0x00000001 #define SSH_FILEXFER_ATTR_UIDGID 0x00000002 #define SSH_FILEXFER_ATTR_PERMISSIONS 0x00000004 #define SSH_FILEXFER_ATTR_ACMODTIME 0x00000008 #define SSH_FILEXFER_ATTR_EXTENDED 0x80000000 """, """ The `pflags' field is a bitmask. The following bits have been defined. #define SSH_FXF_READ 0x00000001 #define SSH_FXF_WRITE 0x00000002 #define SSH_FXF_APPEND 0x00000004 #define SSH_FXF_CREAT 0x00000008 #define SSH_FXF_TRUNC 0x00000010 #define SSH_FXF_EXCL 0x00000020 """, """ Currently, the following values are defined (other values may be defined by future versions of this protocol): #define SSH_FX_OK 0 #define SSH_FX_EOF 1 #define SSH_FX_NO_SUCH_FILE 2 #define SSH_FX_PERMISSION_DENIED 3 #define SSH_FX_FAILURE 4 #define SSH_FX_BAD_MESSAGE 5 #define SSH_FX_NO_CONNECTION 6 #define SSH_FX_CONNECTION_LOST 7 #define SSH_FX_OP_UNSUPPORTED 8 """, ] def test_constantsAgainstSpec(self): """ The constants used by the SFTP protocol implementation match those found by searching through the spec. """ constants = {} for excerpt in self.filexferSpecExcerpts: for line in excerpt.splitlines(): m = re.match(r"^\s*#define SSH_([A-Z_]+)\s+([0-9x]*)\s*$", line) if m: constants[m.group(1)] = int(m.group(2), 0) self.assertTrue( len(constants) > 0, "No constants found (the test must be buggy)." ) for k, v in constants.items(): self.assertEqual(v, getattr(filetransfer, k)) @skipIf(not cryptography, "Cannot run without cryptography") class RawPacketDataServerTests(TestCase): """ Tests for L{filetransfer.FileTransferServer} which explicitly craft certain less common situations to exercise their handling. """ def setUp(self): self.fts = filetransfer.FileTransferServer(avatar=TestAvatar()) def test_closeInvalidHandle(self): """ A close request with an unknown handle receives an FX_NO_SUCH_FILE error response. """ transport = StringTransport() self.fts.makeConnection(transport) # any four bytes requestId = b"1234" # The handle to close, arbitrary bytes. handle = b"invalid handle" # Construct a message packet # https://datatracker.ietf.org/doc/html/draft-ietf-secsh-filexfer-13#section-4 close = common.NS( # Packet type - SSH_FXP_CLOSE # https://datatracker.ietf.org/doc/html/draft-ietf-secsh-filexfer-13#section-4.3 bytes([4]) + requestId + common.NS(handle) ) self.fts.dataReceived(close) # An SSH_FXP_STATUS message # https://datatracker.ietf.org/doc/html/draft-ietf-secsh-filexfer-13#section-9.1 expected = common.NS( # Packet type SSH_FXP_STATUS # https://datatracker.ietf.org/doc/html/draft-ietf-secsh-filexfer-13#section-4.3 bytes([101]) + # The same request id requestId + # A four byte status code. SSH_FX_NO_SUCH_FILE in this case. bytes([0, 0, 0, 2]) + # Error message common.NS(b"No such file or directory") + # error message language tag - conch doesn't send one at all, # though maybe it should common.NS(b"") ) assert_that( transport.value(), equal_to(expected), ) @skipIf(not cryptography, "Cannot run without cryptography") class RawPacketDataTests(TestCase): """ Tests for L{filetransfer.FileTransferClient} which explicitly craft certain less common protocol messages to exercise their handling. """ def setUp(self): self.ftc = filetransfer.FileTransferClient() def test_packetSTATUS(self): """ A STATUS packet containing a result code, a message, and a language is parsed to produce the result of an outstanding request L{Deferred}. @see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>} of the SFTP Internet-Draft. """ d = defer.Deferred() d.addCallback(self._cbTestPacketSTATUS) self.ftc.openRequests[1] = d data = ( struct.pack("!LL", 1, filetransfer.FX_OK) + common.NS(b"msg") + common.NS(b"lang") ) self.ftc.packet_STATUS(data) return d def _cbTestPacketSTATUS(self, result): """ Assert that the result is a two-tuple containing the message and language from the STATUS packet. """ self.assertEqual(result[0], b"msg") self.assertEqual(result[1], b"lang") def test_packetSTATUSShort(self): """ A STATUS packet containing only a result code can also be parsed to produce the result of an outstanding request L{Deferred}. Such packets are sent by some SFTP implementations, though not strictly legal. @see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>} of the SFTP Internet-Draft. """ d = defer.Deferred() d.addCallback(self._cbTestPacketSTATUSShort) self.ftc.openRequests[1] = d data = struct.pack("!LL", 1, filetransfer.FX_OK) self.ftc.packet_STATUS(data) return d def _cbTestPacketSTATUSShort(self, result): """ Assert that the result is a two-tuple containing empty strings, since the STATUS packet had neither a message nor a language. """ self.assertEqual(result[0], b"") self.assertEqual(result[1], b"") def test_packetSTATUSWithoutLang(self): """ A STATUS packet containing a result code and a message but no language can also be parsed to produce the result of an outstanding request L{Deferred}. Such packets are sent by some SFTP implementations, though not strictly legal. @see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>} of the SFTP Internet-Draft. """ d = defer.Deferred() d.addCallback(self._cbTestPacketSTATUSWithoutLang) self.ftc.openRequests[1] = d data = struct.pack("!LL", 1, filetransfer.FX_OK) + common.NS(b"msg") self.ftc.packet_STATUS(data) return d def _cbTestPacketSTATUSWithoutLang(self, result): """ Assert that the result is a two-tuple containing the message from the STATUS packet and an empty string, since the language was missing. """ self.assertEqual(result[0], b"msg") self.assertEqual(result[1], b"") PK F��Z�r��( ( test_unix.pynu �[��� # -*- test-case-name: twisted.conch.test.test_unix -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. from zope.interface import implementer from twisted.internet.interfaces import IReactorProcess from twisted.python.reflect import requireModule from twisted.trial import unittest cryptography = requireModule("cryptography") unix = requireModule("twisted.conch.unix") @implementer(IReactorProcess) class MockProcessSpawner: """ An L{IReactorProcess} that logs calls to C{spawnProcess}. """ def __init__(self): self._spawnProcessCalls = [] def spawnProcess( self, processProtocol, executable, args=(), env={}, path=None, uid=None, gid=None, usePTY=0, childFDs=None, ): """ Log a call to C{spawnProcess}. Do not actually spawn a process. """ self._spawnProcessCalls.append( { "processProtocol": processProtocol, "executable": executable, "args": args, "env": env, "path": path, "uid": uid, "gid": gid, "usePTY": usePTY, "childFDs": childFDs, } ) class StubUnixConchUser: """ Enough of UnixConchUser to exercise SSHSessionForUnixConchUser in the tests below. """ def __init__(self, homeDirectory): from .test_session import StubClient, StubConnection self._homeDirectory = homeDirectory self.conn = StubConnection(transport=StubClient()) def getUserGroupId(self): return (None, None) def getHomeDir(self): return self._homeDirectory def getShell(self): pass class TestSSHSessionForUnixConchUser(unittest.TestCase): if cryptography is None: skip = "Cannot run without cryptography" elif unix is None: skip = "Unix system required" def testExecCommandEnvironment(self): """ C{execCommand} sets the C{HOME} environment variable to the avatar's home directory. """ mockReactor = MockProcessSpawner() homeDirectory = "/made/up/path/" avatar = StubUnixConchUser(homeDirectory) session = unix.SSHSessionForUnixConchUser(avatar, reactor=mockReactor) protocol = None command = ["not-actually-executed"] session.execCommand(protocol, command) [call] = mockReactor._spawnProcessCalls self.assertEqual(homeDirectory, call["env"]["HOME"]) PK F��Z F�� � test_transport.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for ssh/transport.py and the classes therein. """ import binascii import re import string import struct import types from hashlib import md5, sha1, sha256, sha384, sha512 from typing import Optional, Type from twisted import __version__ as twisted_version from twisted.conch.error import ConchError from twisted.conch.ssh import _kex, address, service from twisted.internet import defer from twisted.protocols import loopback from twisted.python import randbytes from twisted.python.compat import iterbytes from twisted.python.randbytes import insecureRandom from twisted.python.reflect import requireModule from twisted.test import proto_helpers from twisted.trial.unittest import TestCase pyasn1 = requireModule("pyasn1") cryptography = requireModule("cryptography") dependencySkip: Optional[str] if pyasn1 and cryptography: dependencySkip = None from cryptography.exceptions import UnsupportedAlgorithm from cryptography.hazmat.backends import default_backend from cryptography.hazmat.primitives import serialization from cryptography.hazmat.primitives.asymmetric import dh, ec from twisted.conch.ssh import common, factory, keys, transport from twisted.conch.test import keydata X25519_SUPPORTED = default_backend().x25519_supported() else: if not pyasn1: dependencySkip = "Cannot run without PyASN1" elif not cryptography: dependencySkip = "can't run without cryptography" X25519_SUPPORTED = False # fictional modules to make classes work class transport: # type: ignore[no-redef] class SSHTransportBase: pass class SSHServerTransport: pass class SSHClientTransport: pass class factory: # type: ignore[no-redef] class SSHFactory: pass class common: # type: ignore[no-redef] @classmethod def NS(self, arg): return b"" def skipWithoutX25519(f): if not X25519_SUPPORTED: f.skip = "x25519 not supported on this system" return f def _MPpow(x, y, z): """ Return the MP version of C{(x ** y) % z}. """ return common.MP(pow(x, y, z)) class MockTransportBase(transport.SSHTransportBase): """ A base class for the client and server protocols. Stores the messages it receives instead of ignoring them. @ivar errors: a list of tuples: (reasonCode, description) @ivar unimplementeds: a list of integers: sequence number @ivar debugs: a list of tuples: (alwaysDisplay, message, lang) @ivar ignoreds: a list of strings: ignored data """ def connectionMade(self): """ Set up instance variables. """ transport.SSHTransportBase.connectionMade(self) self.errors = [] self.unimplementeds = [] self.debugs = [] self.ignoreds = [] self.gotUnsupportedVersion = None def _unsupportedVersionReceived(self, remoteVersion): """ Intercept unsupported version call. @type remoteVersion: L{str} """ self.gotUnsupportedVersion = remoteVersion return transport.SSHTransportBase._unsupportedVersionReceived( self, remoteVersion ) def receiveError(self, reasonCode, description): """ Store any errors received. @type reasonCode: L{int} @type description: L{str} """ self.errors.append((reasonCode, description)) def receiveUnimplemented(self, seqnum): """ Store any unimplemented packet messages. @type seqnum: L{int} """ self.unimplementeds.append(seqnum) def receiveDebug(self, alwaysDisplay, message, lang): """ Store any debug messages. @type alwaysDisplay: L{bool} @type message: L{str} @type lang: L{str} """ self.debugs.append((alwaysDisplay, message, lang)) def ssh_IGNORE(self, packet): """ Store any ignored data. @type packet: L{str} """ self.ignoreds.append(packet) class MockCipher: """ A mocked-up version of twisted.conch.ssh.transport.SSHCiphers. """ outCipType = b"test" encBlockSize = 6 inCipType = b"test" decBlockSize = 6 inMACType = b"test" outMACType = b"test" verifyDigestSize = 1 usedEncrypt = False usedDecrypt = False outMAC = (None, b"", b"", 1) inMAC = (None, b"", b"", 1) keys = () def encrypt(self, x): """ Called to encrypt the packet. Simply record that encryption was used and return the data unchanged. """ self.usedEncrypt = True if (len(x) % self.encBlockSize) != 0: raise RuntimeError( "length %i modulo blocksize %i is not 0: %i" % (len(x), self.encBlockSize, len(x) % self.encBlockSize) ) return x def decrypt(self, x): """ Called to decrypt the packet. Simply record that decryption was used and return the data unchanged. """ self.usedDecrypt = True if (len(x) % self.encBlockSize) != 0: raise RuntimeError( "length %i modulo blocksize %i is not 0: %i" % (len(x), self.decBlockSize, len(x) % self.decBlockSize) ) return x def makeMAC(self, outgoingPacketSequence, payload): """ Make a Message Authentication Code by sending the character value of the outgoing packet. """ return bytes((outgoingPacketSequence,)) def verify(self, incomingPacketSequence, packet, macData): """ Verify the Message Authentication Code by checking that the packet sequence number is the same. """ return bytes((incomingPacketSequence,)) == macData def setKeys(self, ivOut, keyOut, ivIn, keyIn, macIn, macOut): """ Record the keys. """ self.keys = (ivOut, keyOut, ivIn, keyIn, macIn, macOut) class MockCompression: """ A mocked-up compression, based on the zlib interface. Instead of compressing, it reverses the data and adds a 0x66 byte to the end. """ def compress(self, payload): return payload[::-1] # reversed def decompress(self, payload): return payload[:-1][::-1] def flush(self, kind): return b"\x66" class MockService(service.SSHService): """ A mocked-up service, based on twisted.conch.ssh.service.SSHService. @ivar started: True if this service has been started. @ivar stopped: True if this service has been stopped. """ name = b"MockService" started = False stopped = False protocolMessages = {0xFF: "MSG_TEST", 71: "MSG_fiction"} def logPrefix(self): return "MockService" def serviceStarted(self): """ Record that the service was started. """ self.started = True def serviceStopped(self): """ Record that the service was stopped. """ self.stopped = True def ssh_TEST(self, packet): """ A message that this service responds to. """ self.transport.sendPacket(0xFF, packet) class MockFactory(factory.SSHFactory): """ A mocked-up factory based on twisted.conch.ssh.factory.SSHFactory. """ services = {b"ssh-userauth": MockService} def getPublicKeys(self): """ Return the public keys that authenticate this server. """ return { b"ssh-rsa": keys.Key.fromString(keydata.publicRSA_openssh), b"ssh-dsa": keys.Key.fromString(keydata.publicDSA_openssh), } def getPrivateKeys(self): """ Return the private keys that authenticate this server. """ return { b"ssh-rsa": keys.Key.fromString(keydata.privateRSA_openssh), b"ssh-dsa": keys.Key.fromString(keydata.privateDSA_openssh), } def getPrimes(self): """ Diffie-Hellman primes that can be used for key exchange algorithms that use group exchange to establish a prime / generator group. @return: The primes and generators. @rtype: L{dict} mapping the key size to a C{list} of C{(generator, prime)} tuple. """ # In these tests, we hardwire the prime values to those # defined by the diffie-hellman-group14-sha1 key exchange # algorithm, to avoid requiring a moduli file when running # tests. # See OpenSSHFactory.getPrimes. group14 = _kex.getDHGeneratorAndPrime(b"diffie-hellman-group14-sha1") return {2048: (group14,), 4096: ((5, 7),)} class MockOldFactoryPublicKeys(MockFactory): """ The old SSHFactory returned mappings from key names to strings from getPublicKeys(). We return those here for testing. """ def getPublicKeys(self): """ We used to map key types to public key blobs as strings. """ keys = MockFactory.getPublicKeys(self) for name, key in keys.items()[:]: keys[name] = key.blob() return keys class MockOldFactoryPrivateKeys(MockFactory): """ The old SSHFactory returned mappings from key names to cryptography key objects from getPrivateKeys(). We return those here for testing. """ def getPrivateKeys(self): """ We used to map key types to cryptography key objects. """ keys = MockFactory.getPrivateKeys(self) for name, key in keys.items()[:]: keys[name] = key.keyObject return keys def generatePredictableKey(transport): p = transport.p g = transport.g bits = p.bit_length() x = sum(0x9 << x for x in range(0, bits - 3, 4)) # The cryptography module doesn't let us create a secret key directly from # an "x" value; we need to compute the public value ourselves. y = pow(g, x, p) try: transport.dhSecretKey = dh.DHPrivateNumbers( x, dh.DHPublicNumbers(y, dh.DHParameterNumbers(p, g)) ).private_key(default_backend()) except ValueError: print(f"\np={p}\ng={g}\nx={x}\n") raise transport.dhSecretKeyPublicMP = common.MP( transport.dhSecretKey.public_key().public_numbers().y ) class TransportTestCase(TestCase): """ Base class for transport test cases. """ klass: Optional[Type[transport.SSHTransportBase]] = None if dependencySkip: skip = dependencySkip def setUp(self): self.transport = proto_helpers.StringTransport() self.proto = self.klass() self.packets = [] def secureRandom(len): """ Return a consistent entropy value """ return b"\x99" * len self.patch(randbytes, "secureRandom", secureRandom) self.proto._startEphemeralDH = types.MethodType( generatePredictableKey, self.proto ) def stubSendPacket(messageType, payload): self.packets.append((messageType, payload)) self.proto.makeConnection(self.transport) # we just let the kex packet go into the transport self.proto.sendPacket = stubSendPacket def finishKeyExchange(self, proto): """ Deliver enough additional messages to C{proto} so that the key exchange which is started in L{SSHTransportBase.connectionMade} completes and non-key exchange messages can be sent and received. """ proto.dataReceived(b"SSH-2.0-BogoClient-1.2i\r\n") proto.dispatchMessage(transport.MSG_KEXINIT, self._A_KEXINIT_MESSAGE) proto._keySetup(b"foo", b"bar") # SSHTransportBase can't handle MSG_NEWKEYS, or it would be the right # thing to deliver next. _newKeys won't work either, because # sendKexInit (probably) hasn't been called. sendKexInit is # responsible for setting up certain state _newKeys relies on. So, # just change the key exchange state to what it would be when key # exchange is finished. proto._keyExchangeState = proto._KEY_EXCHANGE_NONE def simulateKeyExchange(self, sharedSecret, exchangeHash): """ Finish a key exchange by calling C{_keySetup} with the given arguments. Also do extra whitebox stuff to satisfy that method's assumption that some kind of key exchange has actually taken place. """ self.proto._keyExchangeState = self.proto._KEY_EXCHANGE_REQUESTED self.proto._blockedByKeyExchange = [] self.proto._keySetup(sharedSecret, exchangeHash) class DHGroupExchangeSHA1Mixin: """ Mixin for diffie-hellman-group-exchange-sha1 tests. """ kexAlgorithm = b"diffie-hellman-group-exchange-sha1" hashProcessor = sha1 class DHGroupExchangeSHA256Mixin: """ Mixin for diffie-hellman-group-exchange-sha256 tests. """ kexAlgorithm = b"diffie-hellman-group-exchange-sha256" hashProcessor = sha256 class ECDHMixin: """ Mixin for elliptic curve diffie-hellman tests. """ kexAlgorithm = b"ecdh-sha2-nistp256" hashProcessor = sha256 class Curve25519SHA256Mixin: """ Mixin for curve25519-sha256 tests. """ kexAlgorithm = b"curve25519-sha256" hashProcessor = sha256 class BaseSSHTransportBaseCase: """ Base case for TransportBase tests. """ klass: Optional[Type[transport.SSHTransportBase]] = MockTransportBase class BaseSSHTransportTests(BaseSSHTransportBaseCase, TransportTestCase): """ Test TransportBase. It implements the non-server/client specific parts of the SSH transport protocol. """ if dependencySkip: skip = dependencySkip _A_KEXINIT_MESSAGE = ( b"\xAA" * 16 + common.NS(b"diffie-hellman-group14-sha1") + common.NS(b"ssh-rsa") + common.NS(b"aes256-ctr") + common.NS(b"aes256-ctr") + common.NS(b"hmac-sha1") + common.NS(b"hmac-sha1") + common.NS(b"none") + common.NS(b"none") + common.NS(b"") + common.NS(b"") + b"\x00" + b"\x00\x00\x00\x00" ) def test_sendVersion(self): """ Test that the first thing sent over the connection is the version string. The 'softwareversion' part must consist of printable US-ASCII characters, with the exception of whitespace characters and the minus sign. RFC 4253, section 4.2. """ # the other setup was done in the setup method version = self.transport.value().split(b"\r\n", 1)[0] self.assertEqual(version, b"SSH-2.0-Twisted_" + twisted_version.encode("ascii")) softwareVersion = version.decode("ascii")[len("SSH-2.0-") :] # This is an inefficient regex, but it's simple to build. softwareVersionRegex = ( r"^(" + "|".join( re.escape(c) for c in string.printable if c != "-" and not c.isspace() ) + r")*$" ) self.assertRegex(softwareVersion, softwareVersionRegex) def test_dataReceiveVersionNotSentMemoryDOS(self): """ When the peer is not sending its SSH version but keeps sending data, the connection is disconnected after 4KB to prevent buffering too much and running our of memory. """ sut = MockTransportBase() sut.makeConnection(self.transport) # Data can be received over multiple chunks. sut.dataReceived(b"SSH-2-Server-Identifier") sut.dataReceived(b"1234567890" * 406) sut.dataReceived(b"1235678") self.assertFalse(self.transport.disconnecting) # Here we are going over the limit. sut.dataReceived(b"1234567") # Once a lot of data is received without an SSH version string, # the transport is disconnected. self.assertTrue(self.transport.disconnecting) self.assertIn(b"Preventing a denial of service attack", self.transport.value()) def test_sendPacketPlain(self): """ Test that plain (unencrypted, uncompressed) packets are sent correctly. The format is:: uint32 length (including type and padding length) byte padding length byte type bytes[length-padding length-2] data bytes[padding length] padding """ proto = MockTransportBase() proto.makeConnection(self.transport) self.finishKeyExchange(proto) self.transport.clear() message = ord("A") payload = b"BCDEFG" proto.sendPacket(message, payload) value = self.transport.value() self.assertEqual(value, b"\x00\x00\x00\x0c\x04ABCDEFG\x99\x99\x99\x99") def test_sendPacketEncrypted(self): """ Test that packets sent while encryption is enabled are sent correctly. The whole packet should be encrypted. """ proto = MockTransportBase() proto.makeConnection(self.transport) self.finishKeyExchange(proto) proto.currentEncryptions = testCipher = MockCipher() message = ord("A") payload = b"BC" self.transport.clear() proto.sendPacket(message, payload) self.assertTrue(testCipher.usedEncrypt) value = self.transport.value() self.assertEqual( value, # Four byte length prefix b"\x00\x00\x00\x08" # One byte padding length b"\x04" # The actual application data b"ABC" # "Random" padding - see the secureRandom monkeypatch in setUp b"\x99\x99\x99\x99" # The MAC b"\x02", ) def test_sendPacketCompressed(self): """ Test that packets sent while compression is enabled are sent correctly. The packet type and data should be encrypted. """ proto = MockTransportBase() proto.makeConnection(self.transport) self.finishKeyExchange(proto) proto.outgoingCompression = MockCompression() self.transport.clear() proto.sendPacket(ord("A"), b"B") value = self.transport.value() self.assertEqual( value, b"\x00\x00\x00\x0c\x08BA\x66\x99\x99\x99\x99\x99\x99\x99\x99" ) def test_sendPacketBoth(self): """ Test that packets sent while compression and encryption are enabled are sent correctly. The packet type and data should be compressed and then the whole packet should be encrypted. """ proto = MockTransportBase() proto.makeConnection(self.transport) self.finishKeyExchange(proto) proto.currentEncryptions = testCipher = MockCipher() proto.outgoingCompression = MockCompression() message = ord("A") payload = b"BC" self.transport.clear() proto.sendPacket(message, payload) self.assertTrue(testCipher.usedEncrypt) value = self.transport.value() self.assertEqual( value, # Four byte length prefix b"\x00\x00\x00\x0e" # One byte padding length b"\x09" # Compressed application data b"CBA\x66" # "Random" padding - see the secureRandom monkeypatch in setUp b"\x99\x99\x99\x99\x99\x99\x99\x99\x99" # The MAC b"\x02", ) def test_getPacketPlain(self): """ Test that packets are retrieved correctly out of the buffer when no encryption is enabled. """ proto = MockTransportBase() proto.makeConnection(self.transport) self.finishKeyExchange(proto) self.transport.clear() proto.sendPacket(ord("A"), b"BC") proto.buf = self.transport.value() + b"extra" self.assertEqual(proto.getPacket(), b"ABC") self.assertEqual(proto.buf, b"extra") def test_getPacketEncrypted(self): """ Test that encrypted packets are retrieved correctly. See test_sendPacketEncrypted. """ proto = MockTransportBase() proto.sendKexInit = lambda: None # don't send packets proto.makeConnection(self.transport) self.transport.clear() proto.currentEncryptions = testCipher = MockCipher() proto.sendPacket(ord("A"), b"BCD") value = self.transport.value() proto.buf = value[: MockCipher.decBlockSize] self.assertIsNone(proto.getPacket()) self.assertTrue(testCipher.usedDecrypt) self.assertEqual(proto.first, b"\x00\x00\x00\x0e\x09A") proto.buf += value[MockCipher.decBlockSize :] self.assertEqual(proto.getPacket(), b"ABCD") self.assertEqual(proto.buf, b"") def test_getPacketCompressed(self): """ Test that compressed packets are retrieved correctly. See test_sendPacketCompressed. """ proto = MockTransportBase() proto.makeConnection(self.transport) self.finishKeyExchange(proto) self.transport.clear() proto.outgoingCompression = MockCompression() proto.incomingCompression = proto.outgoingCompression proto.sendPacket(ord("A"), b"BCD") proto.buf = self.transport.value() self.assertEqual(proto.getPacket(), b"ABCD") def test_getPacketBoth(self): """ Test that compressed and encrypted packets are retrieved correctly. See test_sendPacketBoth. """ proto = MockTransportBase() proto.sendKexInit = lambda: None proto.makeConnection(self.transport) self.transport.clear() proto.currentEncryptions = MockCipher() proto.outgoingCompression = MockCompression() proto.incomingCompression = proto.outgoingCompression proto.sendPacket(ord("A"), b"BCDEFG") proto.buf = self.transport.value() self.assertEqual(proto.getPacket(), b"ABCDEFG") def test_ciphersAreValid(self): """ Test that all the supportedCiphers are valid. """ ciphers = transport.SSHCiphers(b"A", b"B", b"C", b"D") iv = key = b"\x00" * 16 for cipName in self.proto.supportedCiphers: self.assertTrue(ciphers._getCipher(cipName, iv, key)) def test_sendKexInit(self): """ Test that the KEXINIT (key exchange initiation) message is sent correctly. Payload:: bytes[16] cookie string key exchange algorithms string public key algorithms string outgoing ciphers string incoming ciphers string outgoing MACs string incoming MACs string outgoing compressions string incoming compressions bool first packet follows uint32 0 """ value = self.transport.value().split(b"\r\n", 1)[1] self.proto.buf = value packet = self.proto.getPacket() self.assertEqual(packet[0:1], bytes((transport.MSG_KEXINIT,))) self.assertEqual(packet[1:17], b"\x99" * 16) ( keyExchanges, pubkeys, ciphers1, ciphers2, macs1, macs2, compressions1, compressions2, languages1, languages2, buf, ) = common.getNS(packet[17:], 10) self.assertEqual( keyExchanges, b",".join(self.proto.supportedKeyExchanges + [b"ext-info-s"]) ) self.assertEqual(pubkeys, b",".join(self.proto.supportedPublicKeys)) self.assertEqual(ciphers1, b",".join(self.proto.supportedCiphers)) self.assertEqual(ciphers2, b",".join(self.proto.supportedCiphers)) self.assertEqual(macs1, b",".join(self.proto.supportedMACs)) self.assertEqual(macs2, b",".join(self.proto.supportedMACs)) self.assertEqual(compressions1, b",".join(self.proto.supportedCompressions)) self.assertEqual(compressions2, b",".join(self.proto.supportedCompressions)) self.assertEqual(languages1, b",".join(self.proto.supportedLanguages)) self.assertEqual(languages2, b",".join(self.proto.supportedLanguages)) self.assertEqual(buf, b"\x00" * 5) def test_receiveKEXINITReply(self): """ Immediately after connecting, the transport expects a KEXINIT message and does not reply to it. """ self.transport.clear() self.proto.dispatchMessage(transport.MSG_KEXINIT, self._A_KEXINIT_MESSAGE) self.assertEqual(self.packets, []) def test_sendKEXINITReply(self): """ When a KEXINIT message is received which is not a reply to an earlier KEXINIT message which was sent, a KEXINIT reply is sent. """ self.finishKeyExchange(self.proto) del self.packets[:] self.proto.dispatchMessage(transport.MSG_KEXINIT, self._A_KEXINIT_MESSAGE) self.assertEqual(len(self.packets), 1) self.assertEqual(self.packets[0][0], transport.MSG_KEXINIT) def test_sendKexInitTwiceFails(self): """ A new key exchange cannot be started while a key exchange is already in progress. If an attempt is made to send a I{KEXINIT} message using L{SSHTransportBase.sendKexInit} while a key exchange is in progress causes that method to raise a L{RuntimeError}. """ self.assertRaises(RuntimeError, self.proto.sendKexInit) def test_sendKexInitBlocksOthers(self): """ After L{SSHTransportBase.sendKexInit} has been called, messages types other than the following are queued and not sent until after I{NEWKEYS} is sent by L{SSHTransportBase._keySetup}. RFC 4253, section 7.1. """ # sendKexInit is called by connectionMade, which is called in setUp. # So we're in the state already. disallowedMessageTypes = [ transport.MSG_SERVICE_REQUEST, transport.MSG_KEXINIT, ] # Drop all the bytes sent by setUp, they're not relevant to this test. self.transport.clear() # Get rid of the sendPacket monkey patch, we are testing the behavior # of sendPacket. del self.proto.sendPacket for messageType in disallowedMessageTypes: self.proto.sendPacket(messageType, b"foo") self.assertEqual(self.transport.value(), b"") self.finishKeyExchange(self.proto) # Make the bytes written to the transport cleartext so it's easier to # make an assertion about them. self.proto.nextEncryptions = MockCipher() # Pseudo-deliver the peer's NEWKEYS message, which should flush the # messages which were queued above. self.proto._newKeys() self.assertEqual(self.transport.value().count(b"foo"), 2) def test_sendExtInfo(self): """ Test that EXT_INFO messages are sent correctly. See RFC 8308, section 2.3. """ self.proto._peerSupportsExtensions = True self.proto.sendExtInfo( [ (b"server-sig-algs", b"ssh-rsa,rsa-sha2-256"), (b"elevation", b"d"), ] ) self.assertEqual( self.packets, [ ( transport.MSG_EXT_INFO, b"\x00\x00\x00\x02" + common.NS(b"server-sig-algs") + common.NS(b"ssh-rsa,rsa-sha2-256") + common.NS(b"elevation") + common.NS(b"d"), ) ], ) def test_sendExtInfoUnsupported(self): """ If the peer has not advertised support for extension negotiation, no EXT_INFO message is sent, since RFC 8308 only guarantees that the peer will be prepared to accept it if it has advertised support. """ self.proto.sendExtInfo([(b"server-sig-algs", b"ssh-rsa,rsa-sha2-256")]) self.assertEqual(self.packets, []) def test_EXT_INFO(self): """ When an EXT_INFO message is received, the transport stores a mapping of the peer's advertised extensions. See RFC 8308, section 2.3. """ self.proto.dispatchMessage( transport.MSG_EXT_INFO, b"\x00\x00\x00\x02" + common.NS(b"server-sig-algs") + common.NS(b"ssh-rsa,rsa-sha2-256,rsa-sha2-512") + common.NS(b"no-flow-control") + common.NS(b"s"), ) self.assertEqual( self.proto.peerExtensions, { b"server-sig-algs": b"ssh-rsa,rsa-sha2-256,rsa-sha2-512", b"no-flow-control": b"s", }, ) def test_sendDebug(self): """ Test that debug messages are sent correctly. Payload:: bool always display string debug message string language """ self.proto.sendDebug(b"test", True, b"en") self.assertEqual( self.packets, [(transport.MSG_DEBUG, b"\x01\x00\x00\x00\x04test\x00\x00\x00\x02en")], ) def test_receiveDebug(self): """ Test that debug messages are received correctly. See test_sendDebug. """ self.proto.dispatchMessage( transport.MSG_DEBUG, b"\x01\x00\x00\x00\x04test\x00\x00\x00\x02en" ) self.proto.dispatchMessage( transport.MSG_DEBUG, b"\x00\x00\x00\x00\x06silent\x00\x00\x00\x02en" ) self.assertEqual( self.proto.debugs, [(True, b"test", b"en"), (False, b"silent", b"en")] ) def test_sendIgnore(self): """ Test that ignored messages are sent correctly. Payload:: string ignored data """ self.proto.sendIgnore(b"test") self.assertEqual( self.packets, [(transport.MSG_IGNORE, b"\x00\x00\x00\x04test")] ) def test_receiveIgnore(self): """ Test that ignored messages are received correctly. See test_sendIgnore. """ self.proto.dispatchMessage(transport.MSG_IGNORE, b"test") self.assertEqual(self.proto.ignoreds, [b"test"]) def test_sendUnimplemented(self): """ Test that unimplemented messages are sent correctly. Payload:: uint32 sequence number """ self.proto.sendUnimplemented() self.assertEqual( self.packets, [(transport.MSG_UNIMPLEMENTED, b"\x00\x00\x00\x00")] ) def test_receiveUnimplemented(self): """ Test that unimplemented messages are received correctly. See test_sendUnimplemented. """ self.proto.dispatchMessage(transport.MSG_UNIMPLEMENTED, b"\x00\x00\x00\xff") self.assertEqual(self.proto.unimplementeds, [255]) def test_sendDisconnect(self): """ Test that disconnection messages are sent correctly. Payload:: uint32 reason code string reason description string language """ disconnected = [False] def stubLoseConnection(): disconnected[0] = True self.transport.loseConnection = stubLoseConnection self.proto.sendDisconnect(0xFF, b"test") self.assertEqual( self.packets, [ ( transport.MSG_DISCONNECT, b"\x00\x00\x00\xff\x00\x00\x00\x04test\x00\x00\x00\x00", ) ], ) self.assertTrue(disconnected[0]) def test_receiveDisconnect(self): """ Test that disconnection messages are received correctly. See test_sendDisconnect. """ disconnected = [False] def stubLoseConnection(): disconnected[0] = True self.transport.loseConnection = stubLoseConnection self.proto.dispatchMessage( transport.MSG_DISCONNECT, b"\x00\x00\x00\xff\x00\x00\x00\x04test" ) self.assertEqual(self.proto.errors, [(255, b"test")]) self.assertTrue(disconnected[0]) def test_dataReceived(self): """ Test that dataReceived parses packets and dispatches them to ssh_* methods. """ kexInit = [False] def stubKEXINIT(packet): kexInit[0] = True self.proto.ssh_KEXINIT = stubKEXINIT self.proto.dataReceived(self.transport.value()) self.assertTrue(self.proto.gotVersion) self.assertEqual(self.proto.ourVersionString, self.proto.otherVersionString) self.assertTrue(kexInit[0]) def test_service(self): """ Test that the transport can set the running service and dispatches packets to the service's packetReceived method. """ service = MockService() self.proto.setService(service) self.assertEqual(self.proto.service, service) self.assertTrue(service.started) self.proto.dispatchMessage(0xFF, b"test") self.assertEqual(self.packets, [(0xFF, b"test")]) service2 = MockService() self.proto.setService(service2) self.assertTrue(service2.started) self.assertTrue(service.stopped) self.proto.connectionLost(None) self.assertTrue(service2.stopped) def test_avatar(self): """ Test that the transport notifies the avatar of disconnections. """ disconnected = [False] def logout(): disconnected[0] = True self.proto.logoutFunction = logout self.proto.avatar = True self.proto.connectionLost(None) self.assertTrue(disconnected[0]) def test_isEncrypted(self): """ Test that the transport accurately reflects its encrypted status. """ self.assertFalse(self.proto.isEncrypted("in")) self.assertFalse(self.proto.isEncrypted("out")) self.assertFalse(self.proto.isEncrypted("both")) self.proto.currentEncryptions = MockCipher() self.assertTrue(self.proto.isEncrypted("in")) self.assertTrue(self.proto.isEncrypted("out")) self.assertTrue(self.proto.isEncrypted("both")) self.proto.currentEncryptions = transport.SSHCiphers( b"none", b"none", b"none", b"none" ) self.assertFalse(self.proto.isEncrypted("in")) self.assertFalse(self.proto.isEncrypted("out")) self.assertFalse(self.proto.isEncrypted("both")) self.assertRaises(TypeError, self.proto.isEncrypted, "bad") def test_isVerified(self): """ Test that the transport accurately reflects its verified status. """ self.assertFalse(self.proto.isVerified("in")) self.assertFalse(self.proto.isVerified("out")) self.assertFalse(self.proto.isVerified("both")) self.proto.currentEncryptions = MockCipher() self.assertTrue(self.proto.isVerified("in")) self.assertTrue(self.proto.isVerified("out")) self.assertTrue(self.proto.isVerified("both")) self.proto.currentEncryptions = transport.SSHCiphers( b"none", b"none", b"none", b"none" ) self.assertFalse(self.proto.isVerified("in")) self.assertFalse(self.proto.isVerified("out")) self.assertFalse(self.proto.isVerified("both")) self.assertRaises(TypeError, self.proto.isVerified, "bad") def test_loseConnection(self): """ Test that loseConnection sends a disconnect message and closes the connection. """ disconnected = [False] def stubLoseConnection(): disconnected[0] = True self.transport.loseConnection = stubLoseConnection self.proto.loseConnection() self.assertEqual(self.packets[0][0], transport.MSG_DISCONNECT) self.assertEqual( self.packets[0][1][3:4], bytes((transport.DISCONNECT_CONNECTION_LOST,)) ) def test_badVersion(self): """ Test that the transport disconnects when it receives a bad version. """ def testBad(version): self.packets = [] self.proto.gotVersion = False disconnected = [False] def stubLoseConnection(): disconnected[0] = True self.transport.loseConnection = stubLoseConnection for c in iterbytes(version + b"\r\n"): self.proto.dataReceived(c) self.assertTrue(disconnected[0]) self.assertEqual(self.packets[0][0], transport.MSG_DISCONNECT) self.assertEqual( self.packets[0][1][3:4], bytes((transport.DISCONNECT_PROTOCOL_VERSION_NOT_SUPPORTED,)), ) testBad(b"SSH-1.5-OpenSSH") testBad(b"SSH-3.0-Twisted") testBad(b"GET / HTTP/1.1") def test_dataBeforeVersion(self): """ Test that the transport ignores data sent before the version string. """ proto = MockTransportBase() proto.makeConnection(proto_helpers.StringTransport()) data = ( b"""here's some stuff beforehand here's some other stuff """ + proto.ourVersionString + b"\r\n" ) [proto.dataReceived(c) for c in iterbytes(data)] self.assertTrue(proto.gotVersion) self.assertEqual(proto.otherVersionString, proto.ourVersionString) def test_compatabilityVersion(self): """ Test that the transport treats the compatibility version (1.99) as equivalent to version 2.0. """ proto = MockTransportBase() proto.makeConnection(proto_helpers.StringTransport()) proto.dataReceived(b"SSH-1.99-OpenSSH\n") self.assertTrue(proto.gotVersion) self.assertEqual(proto.otherVersionString, b"SSH-1.99-OpenSSH") def test_dataReceivedSSHVersionUnixNewline(self): """ It can parse the SSH version string even when it ends only in Unix newlines (CR) and does not follows the RFC 4253 to use network newlines (CR LF). """ sut = MockTransportBase() sut.makeConnection(proto_helpers.StringTransport()) sut.dataReceived(b"SSH-2.0-PoorSSHD Some-comment here\n" b"more-data") self.assertTrue(sut.gotVersion) self.assertEqual(sut.otherVersionString, b"SSH-2.0-PoorSSHD Some-comment here") def test_dataReceivedSSHVersionTrailingSpaces(self): """ The trailing spaces from SSH version comment are not removed. The SSH version string needs to be kept as received (without CR LF end of line) as they are used in the host authentication process. This can happen with a Bitvise SSH server which hides its version. """ sut = MockTransportBase() sut.makeConnection(proto_helpers.StringTransport()) sut.dataReceived( b"SSH-2.0-9.99 FlowSsh: Bitvise SSH Server (WinSSHD) \r\n" b"more-data" ) self.assertTrue(sut.gotVersion) self.assertEqual( sut.otherVersionString, b"SSH-2.0-9.99 FlowSsh: Bitvise SSH Server (WinSSHD) ", ) def test_supportedVersionsAreAllowed(self): """ If an unusual SSH version is received and is included in C{supportedVersions}, an unsupported version error is not emitted. """ proto = MockTransportBase() proto.supportedVersions = (b"9.99",) proto.makeConnection(proto_helpers.StringTransport()) proto.dataReceived(b"SSH-9.99-OpenSSH\n") self.assertFalse(proto.gotUnsupportedVersion) def test_unsupportedVersionsCallUnsupportedVersionReceived(self): """ If an unusual SSH version is received and is not included in C{supportedVersions}, an unsupported version error is emitted. """ proto = MockTransportBase() proto.supportedVersions = (b"2.0",) proto.makeConnection(proto_helpers.StringTransport()) proto.dataReceived(b"SSH-9.99-OpenSSH\n") self.assertEqual(b"9.99", proto.gotUnsupportedVersion) def test_badPackets(self): """ Test that the transport disconnects with an error when it receives bad packets. """ def testBad(packet, error=transport.DISCONNECT_PROTOCOL_ERROR): self.packets = [] self.proto.buf = packet self.assertIsNone(self.proto.getPacket()) self.assertEqual(len(self.packets), 1) self.assertEqual(self.packets[0][0], transport.MSG_DISCONNECT) self.assertEqual(self.packets[0][1][3:4], bytes((error,))) testBad(b"\xff" * 8) # big packet testBad(b"\x00\x00\x00\x05\x00BCDE") # length not modulo blocksize oldEncryptions = self.proto.currentEncryptions self.proto.currentEncryptions = MockCipher() testBad( b"\x00\x00\x00\x08\x06AB123456", transport.DISCONNECT_MAC_ERROR # bad MAC ) self.proto.currentEncryptions.decrypt = lambda x: x[:-1] testBad(b"\x00\x00\x00\x08\x06BCDEFGHIJK") # bad decryption self.proto.currentEncryptions = oldEncryptions self.proto.incomingCompression = MockCompression() def stubDecompress(payload): raise Exception("bad compression") self.proto.incomingCompression.decompress = stubDecompress testBad( b"\x00\x00\x00\x04\x00BCDE", # bad decompression transport.DISCONNECT_COMPRESSION_ERROR, ) self.flushLoggedErrors() def test_unimplementedPackets(self): """ Test that unimplemented packet types cause MSG_UNIMPLEMENTED packets to be sent. """ seqnum = self.proto.incomingPacketSequence def checkUnimplemented(seqnum=seqnum): self.assertEqual(self.packets[0][0], transport.MSG_UNIMPLEMENTED) self.assertEqual(self.packets[0][1][3:4], bytes((seqnum,))) self.proto.packets = [] seqnum += 1 self.proto.dispatchMessage(40, b"") checkUnimplemented() transport.messages[41] = b"MSG_fiction" self.proto.dispatchMessage(41, b"") checkUnimplemented() self.proto.dispatchMessage(60, b"") checkUnimplemented() self.proto.setService(MockService()) self.proto.dispatchMessage(70, b"") checkUnimplemented() self.proto.dispatchMessage(71, b"") checkUnimplemented() def test_multipleClasses(self): """ Test that multiple instances have distinct states. """ proto = self.proto proto.dataReceived(self.transport.value()) proto.currentEncryptions = MockCipher() proto.outgoingCompression = MockCompression() proto.incomingCompression = MockCompression() proto.setService(MockService()) proto2 = MockTransportBase() proto2.makeConnection(proto_helpers.StringTransport()) proto2.sendIgnore(b"") self.assertNotEqual(proto.gotVersion, proto2.gotVersion) self.assertNotEqual(proto.transport, proto2.transport) self.assertNotEqual(proto.outgoingPacketSequence, proto2.outgoingPacketSequence) self.assertNotEqual(proto.incomingPacketSequence, proto2.incomingPacketSequence) self.assertNotEqual(proto.currentEncryptions, proto2.currentEncryptions) self.assertNotEqual(proto.service, proto2.service) class BaseSSHTransportDHGroupExchangeBaseCase(BaseSSHTransportBaseCase): """ Diffie-Hellman group exchange tests for TransportBase. """ def test_getKey(self): """ Test that _getKey generates the correct keys. """ self.proto.kexAlg = self.kexAlgorithm self.proto.sessionID = b"EF" k1 = self.hashProcessor(b"AB" + b"CD" + b"K" + self.proto.sessionID).digest() k2 = self.hashProcessor(b"ABCD" + k1).digest() k3 = self.hashProcessor(b"ABCD" + k1 + k2).digest() k4 = self.hashProcessor(b"ABCD" + k1 + k2 + k3).digest() self.assertEqual(self.proto._getKey(b"K", b"AB", b"CD"), k1 + k2 + k3 + k4) class BaseSSHTransportDHGroupExchangeSHA1Tests( BaseSSHTransportDHGroupExchangeBaseCase, DHGroupExchangeSHA1Mixin, TransportTestCase ): """ diffie-hellman-group-exchange-sha1 tests for TransportBase. """ class BaseSSHTransportDHGroupExchangeSHA256Tests( BaseSSHTransportDHGroupExchangeBaseCase, DHGroupExchangeSHA256Mixin, TransportTestCase, ): """ diffie-hellman-group-exchange-sha256 tests for TransportBase. """ class BaseSSHTransportEllipticCurveTests( BaseSSHTransportDHGroupExchangeBaseCase, ECDHMixin, TransportTestCase ): """ ecdh-sha2-nistp256 tests for TransportBase """ @skipWithoutX25519 class BaseSSHTransportCurve25519SHA256Tests( BaseSSHTransportDHGroupExchangeBaseCase, Curve25519SHA256Mixin, TransportTestCase ): """ curve25519-sha256 tests for TransportBase """ class ServerAndClientSSHTransportBaseCase: """ Tests that need to be run on both the server and the client. """ def checkDisconnected(self, kind=None): """ Helper function to check if the transport disconnected. """ if kind is None: kind = transport.DISCONNECT_PROTOCOL_ERROR self.assertEqual(self.packets[-1][0], transport.MSG_DISCONNECT) self.assertEqual(self.packets[-1][1][3:4], bytes((kind,))) def connectModifiedProtocol(self, protoModification, kind=None): """ Helper function to connect a modified protocol to the test protocol and test for disconnection. """ if kind is None: kind = transport.DISCONNECT_KEY_EXCHANGE_FAILED proto2 = self.klass() protoModification(proto2) proto2.makeConnection(proto_helpers.StringTransport()) self.proto.dataReceived(proto2.transport.value()) if kind: self.checkDisconnected(kind) return proto2 def test_disconnectIfCantMatchKex(self): """ Test that the transport disconnects if it can't match the key exchange """ def blankKeyExchanges(proto2): proto2.supportedKeyExchanges = [] self.connectModifiedProtocol(blankKeyExchanges) def test_disconnectIfCantMatchKeyAlg(self): """ Like test_disconnectIfCantMatchKex, but for the key algorithm. """ def blankPublicKeys(proto2): proto2.supportedPublicKeys = [] self.connectModifiedProtocol(blankPublicKeys) def test_disconnectIfCantMatchCompression(self): """ Like test_disconnectIfCantMatchKex, but for the compression. """ def blankCompressions(proto2): proto2.supportedCompressions = [] self.connectModifiedProtocol(blankCompressions) def test_disconnectIfCantMatchCipher(self): """ Like test_disconnectIfCantMatchKex, but for the encryption. """ def blankCiphers(proto2): proto2.supportedCiphers = [] self.connectModifiedProtocol(blankCiphers) def test_disconnectIfCantMatchMAC(self): """ Like test_disconnectIfCantMatchKex, but for the MAC. """ def blankMACs(proto2): proto2.supportedMACs = [] self.connectModifiedProtocol(blankMACs) def test_getPeer(self): """ Test that the transport's L{getPeer} method returns an L{SSHTransportAddress} with the L{IAddress} of the peer. """ self.assertEqual( self.proto.getPeer(), address.SSHTransportAddress(self.proto.transport.getPeer()), ) def test_getHost(self): """ Test that the transport's L{getHost} method returns an L{SSHTransportAddress} with the L{IAddress} of the host. """ self.assertEqual( self.proto.getHost(), address.SSHTransportAddress(self.proto.transport.getHost()), ) class ServerSSHTransportBaseCase(ServerAndClientSSHTransportBaseCase): """ Base case for SSHServerTransport tests. """ klass: Optional[Type[transport.SSHTransportBase]] = transport.SSHServerTransport def setUp(self): TransportTestCase.setUp(self) self.proto.factory = MockFactory() self.proto.factory.startFactory() def tearDown(self): TransportTestCase.tearDown(self) self.proto.factory.stopFactory() del self.proto.factory class ServerSSHTransportTests(ServerSSHTransportBaseCase, TransportTestCase): """ Tests for SSHServerTransport. """ def test_KEXINITMultipleAlgorithms(self): """ Receiving a KEXINIT packet listing multiple supported algorithms will set up the first common algorithm found in the client's preference list. """ self.proto.dataReceived( b"SSH-2.0-Twisted\r\n\x00\x00\x01\xf4\x04\x14" b"\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99" b"\x99\x00\x00\x00bdiffie-hellman-group1-sha1,diffie-hellman-g" b"roup-exchange-sha1,diffie-hellman-group-exchange-sha256\x00" b"\x00\x00\x0fssh-dss,ssh-rsa\x00\x00\x00\x85aes128-ctr,aes128-" b"cbc,aes192-ctr,aes192-cbc,aes256-ctr,aes256-cbc,cast128-ctr,c" b"ast128-cbc,blowfish-ctr,blowfish-cbc,3des-ctr,3des-cbc\x00" b"\x00\x00\x85aes128-ctr,aes128-cbc,aes192-ctr,aes192-cbc,aes25" b"6-ctr,aes256-cbc,cast128-ctr,cast128-cbc,blowfish-ctr,blowfis" b"h-cbc,3des-ctr,3des-cbc\x00\x00\x00\x12hmac-md5,hmac-sha1\x00" b"\x00\x00\x12hmac-md5,hmac-sha1\x00\x00\x00\tnone,zlib\x00\x00" b"\x00\tnone,zlib\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00" b"\x00\x00\x99\x99\x99\x99" ) # Even if as server we prefer diffie-hellman-group-exchange-sha256 the # client preference is used, after skipping diffie-hellman-group1-sha1 self.assertEqual(self.proto.kexAlg, b"diffie-hellman-group-exchange-sha1") self.assertEqual(self.proto.keyAlg, b"ssh-dss") self.assertEqual(self.proto.outgoingCompressionType, b"none") self.assertEqual(self.proto.incomingCompressionType, b"none") self.assertFalse(self.proto._peerSupportsExtensions) ne = self.proto.nextEncryptions self.assertEqual(ne.outCipType, b"aes128-ctr") self.assertEqual(ne.inCipType, b"aes128-ctr") self.assertEqual(ne.outMACType, b"hmac-md5") self.assertEqual(ne.inMACType, b"hmac-md5") def test_KEXINITExtensionNegotiation(self): """ If the client sends "ext-info-c" in its key exchange algorithms, then the server notes that the client supports extension negotiation. See RFC 8308, section 2.1. """ kexInitPacket = ( b"\x00" * 16 + common.NS(b"diffie-hellman-group-exchange-sha256,ext-info-c") + common.NS(b"ssh-rsa") + common.NS(b"aes256-ctr") + common.NS(b"aes256-ctr") + common.NS(b"hmac-sha1") + common.NS(b"hmac-sha1") + common.NS(b"none") + common.NS(b"none") + common.NS(b"") + common.NS(b"") + b"\x00" + b"\x00\x00\x00\x00" ) self.proto.ssh_KEXINIT(kexInitPacket) self.assertEqual(self.proto.kexAlg, b"diffie-hellman-group-exchange-sha256") self.assertTrue(self.proto._peerSupportsExtensions) def test_ignoreGuessPacketKex(self): """ The client is allowed to send a guessed key exchange packet after it sends the KEXINIT packet. However, if the key exchanges do not match, that guess packet must be ignored. This tests that the packet is ignored in the case of the key exchange method not matching. """ kexInitPacket = ( b"\x00" * 16 + ( b"".join( [ common.NS(x) for x in [ b",".join(y) for y in [ self.proto.supportedKeyExchanges[::-1], self.proto.supportedPublicKeys, self.proto.supportedCiphers, self.proto.supportedCiphers, self.proto.supportedMACs, self.proto.supportedMACs, self.proto.supportedCompressions, self.proto.supportedCompressions, self.proto.supportedLanguages, self.proto.supportedLanguages, ] ] ] ) ) + (b"\xff\x00\x00\x00\x00") ) self.proto.ssh_KEXINIT(kexInitPacket) self.assertTrue(self.proto.ignoreNextPacket) self.proto.ssh_DEBUG(b"\x01\x00\x00\x00\x04test\x00\x00\x00\x00") self.assertTrue(self.proto.ignoreNextPacket) self.proto.ssh_KEX_DH_GEX_REQUEST_OLD(b"\x00\x00\x08\x00") self.assertFalse(self.proto.ignoreNextPacket) self.assertEqual(self.packets, []) self.proto.ignoreNextPacket = True self.proto.ssh_KEX_DH_GEX_REQUEST(b"\x00\x00\x08\x00" * 3) self.assertFalse(self.proto.ignoreNextPacket) self.assertEqual(self.packets, []) def test_ignoreGuessPacketKey(self): """ Like test_ignoreGuessPacketKex, but for an incorrectly guessed public key format. """ kexInitPacket = ( b"\x00" * 16 + ( b"".join( [ common.NS(x) for x in [ b",".join(y) for y in [ self.proto.supportedKeyExchanges, self.proto.supportedPublicKeys[::-1], self.proto.supportedCiphers, self.proto.supportedCiphers, self.proto.supportedMACs, self.proto.supportedMACs, self.proto.supportedCompressions, self.proto.supportedCompressions, self.proto.supportedLanguages, self.proto.supportedLanguages, ] ] ] ) ) + (b"\xff\x00\x00\x00\x00") ) self.proto.ssh_KEXINIT(kexInitPacket) self.assertTrue(self.proto.ignoreNextPacket) self.proto.ssh_DEBUG(b"\x01\x00\x00\x00\x04test\x00\x00\x00\x00") self.assertTrue(self.proto.ignoreNextPacket) self.proto.ssh_KEX_DH_GEX_REQUEST_OLD(b"\x00\x00\x08\x00") self.assertFalse(self.proto.ignoreNextPacket) self.assertEqual(self.packets, []) self.proto.ignoreNextPacket = True self.proto.ssh_KEX_DH_GEX_REQUEST(b"\x00\x00\x08\x00" * 3) self.assertFalse(self.proto.ignoreNextPacket) self.assertEqual(self.packets, []) def assertKexDHInitResponse(self, kexAlgorithm, bits): """ Test that the KEXDH_INIT packet causes the server to send a KEXDH_REPLY with the server's public key and a signature. @param kexAlgorithm: The key exchange algorithm to use. @type kexAlgorithm: L{str} """ self.proto.supportedKeyExchanges = [kexAlgorithm] self.proto.supportedPublicKeys = [b"ssh-rsa"] self.proto.dataReceived(self.transport.value()) g, p = _kex.getDHGeneratorAndPrime(kexAlgorithm) e = pow(g, 5000, p) self.proto.ssh_KEX_DH_GEX_REQUEST_OLD(common.MP(e)) y = common.getMP(common.NS(b"\x99" * (bits // 8)))[0] f = _MPpow(self.proto.g, y, self.proto.p) self.assertEqual(self.proto.dhSecretKeyPublicMP, f) sharedSecret = _MPpow(e, y, self.proto.p) h = sha1() h.update(common.NS(self.proto.ourVersionString) * 2) h.update(common.NS(self.proto.ourKexInitPayload) * 2) h.update(common.NS(self.proto.factory.publicKeys[b"ssh-rsa"].blob())) h.update(common.MP(e)) h.update(f) h.update(sharedSecret) exchangeHash = h.digest() signature = self.proto.factory.privateKeys[b"ssh-rsa"].sign(exchangeHash) self.assertEqual( self.packets, [ ( transport.MSG_KEXDH_REPLY, common.NS(self.proto.factory.publicKeys[b"ssh-rsa"].blob()) + f + common.NS(signature), ), (transport.MSG_NEWKEYS, b""), ], ) def test_checkBad_KEX_ECDH_INIT_CurveName(self): """ Test that if the server receives a KEX_DH_GEX_REQUEST_OLD message and the key exchange algorithm is not set, we raise a ConchError. """ self.proto.kexAlg = b"bad-curve" self.proto.keyAlg = b"ssh-rsa" self.assertRaises( UnsupportedAlgorithm, self.proto._ssh_KEX_ECDH_INIT, common.NS(b"unused-key"), ) def test_checkBad_KEX_INIT_CurveName(self): """ Test that if the server received a bad name for a curve we raise an UnsupportedAlgorithm error. """ kexmsg = ( b"\xAA" * 16 + common.NS(b"ecdh-sha2-nistp256") + common.NS(b"ssh-rsa") + common.NS(b"aes256-ctr") + common.NS(b"aes256-ctr") + common.NS(b"hmac-sha1") + common.NS(b"hmac-sha1") + common.NS(b"none") + common.NS(b"none") + common.NS(b"") + common.NS(b"") + b"\x00" + b"\x00\x00\x00\x00" ) self.proto.ssh_KEXINIT(kexmsg) self.assertRaises(AttributeError) self.assertRaises(UnsupportedAlgorithm) def test_KEXDH_INIT_GROUP14(self): """ KEXDH_INIT messages are processed when the diffie-hellman-group14-sha1 key exchange algorithm is requested. """ self.assertKexDHInitResponse(b"diffie-hellman-group14-sha1", 2048) def test_keySetup(self): """ Test that _keySetup sets up the next encryption keys. """ self.proto.kexAlg = b"diffie-hellman-group14-sha1" self.proto.nextEncryptions = MockCipher() self.simulateKeyExchange(b"AB", b"CD") self.assertEqual(self.proto.sessionID, b"CD") self.simulateKeyExchange(b"AB", b"EF") self.assertEqual(self.proto.sessionID, b"CD") self.assertEqual(self.packets[-1], (transport.MSG_NEWKEYS, b"")) newKeys = [self.proto._getKey(c, b"AB", b"EF") for c in iterbytes(b"ABCDEF")] self.assertEqual( self.proto.nextEncryptions.keys, (newKeys[1], newKeys[3], newKeys[0], newKeys[2], newKeys[5], newKeys[4]), ) def test_keySetupWithExtInfo(self): """ If the client advertised support for extension negotiation, then _keySetup sends SSH_MSG_EXT_INFO with the "server-sig-algs" extension as the next packet following the server's first SSH_MSG_NEWKEYS. See RFC 8308, sections 2.4 and 3.1. """ self.proto.supportedPublicKeys = [b"ssh-rsa", b"rsa-sha2-256", b"rsa-sha2-512"] self.proto.kexAlg = b"diffie-hellman-group14-sha1" self.proto.nextEncryptions = MockCipher() self.proto._peerSupportsExtensions = True self.simulateKeyExchange(b"AB", b"CD") self.assertEqual(self.packets[-2], (transport.MSG_NEWKEYS, b"")) self.assertEqual( self.packets[-1], ( transport.MSG_EXT_INFO, b"\x00\x00\x00\x01" + common.NS(b"server-sig-algs") + common.NS(b"ssh-rsa,rsa-sha2-256,rsa-sha2-512"), ), ) self.simulateKeyExchange(b"AB", b"EF") self.assertEqual(self.packets[-1], (transport.MSG_NEWKEYS, b"")) def test_ECDH_keySetup(self): """ Test that _keySetup sets up the next encryption keys. """ self.proto.kexAlg = b"ecdh-sha2-nistp256" self.proto.nextEncryptions = MockCipher() self.simulateKeyExchange(b"AB", b"CD") self.assertEqual(self.proto.sessionID, b"CD") self.simulateKeyExchange(b"AB", b"EF") self.assertEqual(self.proto.sessionID, b"CD") self.assertEqual(self.packets[-1], (transport.MSG_NEWKEYS, b"")) newKeys = [self.proto._getKey(c, b"AB", b"EF") for c in iterbytes(b"ABCDEF")] self.assertEqual( self.proto.nextEncryptions.keys, (newKeys[1], newKeys[3], newKeys[0], newKeys[2], newKeys[5], newKeys[4]), ) def test_NEWKEYS(self): """ Test that NEWKEYS transitions the keys in nextEncryptions to currentEncryptions. """ self.test_KEXINITMultipleAlgorithms() self.proto.nextEncryptions = transport.SSHCiphers( b"none", b"none", b"none", b"none" ) self.proto.ssh_NEWKEYS(b"") self.assertIs(self.proto.currentEncryptions, self.proto.nextEncryptions) self.assertIsNone(self.proto.outgoingCompression) self.assertIsNone(self.proto.incomingCompression) self.proto.outgoingCompressionType = b"zlib" self.simulateKeyExchange(b"AB", b"CD") self.proto.ssh_NEWKEYS(b"") self.assertIsNotNone(self.proto.outgoingCompression) self.proto.incomingCompressionType = b"zlib" self.simulateKeyExchange(b"AB", b"EF") self.proto.ssh_NEWKEYS(b"") self.assertIsNotNone(self.proto.incomingCompression) def test_SERVICE_REQUEST(self): """ Test that the SERVICE_REQUEST message requests and starts a service. """ self.proto.ssh_SERVICE_REQUEST(common.NS(b"ssh-userauth")) self.assertEqual( self.packets, [(transport.MSG_SERVICE_ACCEPT, common.NS(b"ssh-userauth"))] ) self.assertEqual(self.proto.service.name, b"MockService") def test_disconnectNEWKEYSData(self): """ Test that NEWKEYS disconnects if it receives data. """ self.proto.ssh_NEWKEYS(b"bad packet") self.checkDisconnected() def test_disconnectSERVICE_REQUESTBadService(self): """ Test that SERVICE_REQUESTS disconnects if an unknown service is requested. """ self.proto.ssh_SERVICE_REQUEST(common.NS(b"no service")) self.checkDisconnected(transport.DISCONNECT_SERVICE_NOT_AVAILABLE) class ServerSSHTransportDHGroupExchangeBaseCase(ServerSSHTransportBaseCase): """ Diffie-Hellman group exchange tests for SSHServerTransport. """ def test_KEX_DH_GEX_REQUEST_OLD(self): """ Test that the KEX_DH_GEX_REQUEST_OLD message causes the server to reply with a KEX_DH_GEX_GROUP message with the correct Diffie-Hellman group. """ self.proto.supportedKeyExchanges = [self.kexAlgorithm] self.proto.supportedPublicKeys = [b"ssh-rsa"] self.proto.dataReceived(self.transport.value()) self.proto.ssh_KEX_DH_GEX_REQUEST_OLD(b"\x00\x00\x04\x00") dhGenerator, dhPrime = self.proto.factory.getPrimes().get(2048)[0] self.assertEqual( self.packets, [ ( transport.MSG_KEX_DH_GEX_GROUP, common.MP(dhPrime) + b"\x00\x00\x00\x01\x02", ) ], ) self.assertEqual(self.proto.g, 2) self.assertEqual(self.proto.p, dhPrime) def test_KEX_DH_GEX_REQUEST_OLD_badKexAlg(self): """ Test that if the server receives a KEX_DH_GEX_REQUEST_OLD message and the key exchange algorithm is not set, we raise a ConchError. """ self.proto.kexAlg = None self.assertRaises(ConchError, self.proto.ssh_KEX_DH_GEX_REQUEST_OLD, None) def test_KEX_DH_GEX_REQUEST(self): """ Test that the KEX_DH_GEX_REQUEST message causes the server to reply with a KEX_DH_GEX_GROUP message with the correct Diffie-Hellman group. """ self.proto.supportedKeyExchanges = [self.kexAlgorithm] self.proto.supportedPublicKeys = [b"ssh-rsa"] self.proto.dataReceived(self.transport.value()) self.proto.ssh_KEX_DH_GEX_REQUEST( b"\x00\x00\x04\x00\x00\x00\x08\x00" + b"\x00\x00\x0c\x00" ) dhGenerator, dhPrime = self.proto.factory.getPrimes().get(2048)[0] self.assertEqual( self.packets, [ ( transport.MSG_KEX_DH_GEX_GROUP, common.MP(dhPrime) + b"\x00\x00\x00\x01\x02", ) ], ) self.assertEqual(self.proto.g, 2) self.assertEqual(self.proto.p, dhPrime) def test_KEX_DH_GEX_INIT_after_REQUEST_OLD(self): """ Test that the KEX_DH_GEX_INIT message after the client sends KEX_DH_GEX_REQUEST_OLD causes the server to send a KEX_DH_GEX_INIT message with a public key and signature. """ self.test_KEX_DH_GEX_REQUEST_OLD() e = pow(self.proto.g, 3, self.proto.p) y = common.getMP(b"\x00\x00\x01\x00" + b"\x99" * 512)[0] self.assertEqual(self.proto.dhSecretKey.private_numbers().x, y) f = _MPpow(self.proto.g, y, self.proto.p) self.assertEqual(self.proto.dhSecretKeyPublicMP, f) sharedSecret = _MPpow(e, y, self.proto.p) h = self.hashProcessor() h.update(common.NS(self.proto.ourVersionString) * 2) h.update(common.NS(self.proto.ourKexInitPayload) * 2) h.update(common.NS(self.proto.factory.publicKeys[b"ssh-rsa"].blob())) h.update(b"\x00\x00\x04\x00") h.update(common.MP(self.proto.p)) h.update(common.MP(self.proto.g)) h.update(common.MP(e)) h.update(f) h.update(sharedSecret) exchangeHash = h.digest() self.proto.ssh_KEX_DH_GEX_INIT(common.MP(e)) self.assertEqual( self.packets[1:], [ ( transport.MSG_KEX_DH_GEX_REPLY, common.NS(self.proto.factory.publicKeys[b"ssh-rsa"].blob()) + f + common.NS( self.proto.factory.privateKeys[b"ssh-rsa"].sign(exchangeHash) ), ), (transport.MSG_NEWKEYS, b""), ], ) def test_KEX_DH_GEX_INIT_after_REQUEST(self): """ Test that the KEX_DH_GEX_INIT message after the client sends KEX_DH_GEX_REQUEST causes the server to send a KEX_DH_GEX_INIT message with a public key and signature. """ self.test_KEX_DH_GEX_REQUEST() e = pow(self.proto.g, 3, self.proto.p) y = common.getMP(b"\x00\x00\x01\x00" + b"\x99" * 256)[0] f = _MPpow(self.proto.g, y, self.proto.p) sharedSecret = _MPpow(e, y, self.proto.p) h = self.hashProcessor() h.update(common.NS(self.proto.ourVersionString) * 2) h.update(common.NS(self.proto.ourKexInitPayload) * 2) h.update(common.NS(self.proto.factory.publicKeys[b"ssh-rsa"].blob())) h.update(b"\x00\x00\x04\x00\x00\x00\x08\x00\x00\x00\x0c\x00") h.update(common.MP(self.proto.p)) h.update(common.MP(self.proto.g)) h.update(common.MP(e)) h.update(f) h.update(sharedSecret) exchangeHash = h.digest() self.proto.ssh_KEX_DH_GEX_INIT(common.MP(e)) self.assertEqual( self.packets[1], ( transport.MSG_KEX_DH_GEX_REPLY, common.NS(self.proto.factory.publicKeys[b"ssh-rsa"].blob()) + f + common.NS( self.proto.factory.privateKeys[b"ssh-rsa"].sign(exchangeHash) ), ), ) class ServerSSHTransportDHGroupExchangeSHA1Tests( ServerSSHTransportDHGroupExchangeBaseCase, DHGroupExchangeSHA1Mixin, TransportTestCase, ): """ diffie-hellman-group-exchange-sha1 tests for SSHServerTransport. """ class ServerSSHTransportDHGroupExchangeSHA256Tests( ServerSSHTransportDHGroupExchangeBaseCase, DHGroupExchangeSHA256Mixin, TransportTestCase, ): """ diffie-hellman-group-exchange-sha256 tests for SSHServerTransport. """ class ServerSSHTransportECDHBaseCase(ServerSSHTransportBaseCase): """ Elliptic Curve Diffie-Hellman tests for SSHServerTransport. """ def test_KEX_ECDH_INIT(self): """ Test that the KEXDH_INIT message causes the server to send a KEXDH_REPLY with the server's public key and a signature. """ self.proto.supportedKeyExchanges = [self.kexAlgorithm] self.proto.supportedPublicKeys = [b"ssh-rsa"] self.proto.dataReceived(self.transport.value()) privKey = self.proto.factory.privateKeys[b"ssh-rsa"] pubKey = self.proto.factory.publicKeys[b"ssh-rsa"] ecPriv = self.proto._generateECPrivateKey() ecPub = ecPriv.public_key() encPub = self.proto._encodeECPublicKey(ecPub) self.proto.ssh_KEX_DH_GEX_REQUEST_OLD(common.NS(encPub)) sharedSecret = self.proto._generateECSharedSecret( ecPriv, self.proto._encodeECPublicKey(self.proto.ecPub) ) h = self.hashProcessor() h.update(common.NS(self.proto.otherVersionString)) h.update(common.NS(self.proto.ourVersionString)) h.update(common.NS(self.proto.otherKexInitPayload)) h.update(common.NS(self.proto.ourKexInitPayload)) h.update(common.NS(pubKey.blob())) h.update(common.NS(encPub)) h.update(common.NS(self.proto._encodeECPublicKey(self.proto.ecPub))) h.update(sharedSecret) exchangeHash = h.digest() signature = privKey.sign(exchangeHash) self.assertEqual( self.packets, [ ( transport.MSG_KEXDH_REPLY, common.NS(pubKey.blob()) + common.NS(self.proto._encodeECPublicKey(self.proto.ecPub)) + common.NS(signature), ), (transport.MSG_NEWKEYS, b""), ], ) class ServerSSHTransportECDHTests( ServerSSHTransportECDHBaseCase, ECDHMixin, TransportTestCase ): """ ecdh-sha2-nistp256 tests for SSHServerTransport. """ @skipWithoutX25519 class ServerSSHTransportCurve25519SHA256Tests( ServerSSHTransportECDHBaseCase, Curve25519SHA256Mixin, TransportTestCase ): """ curve25519-sha256 tests for SSHServerTransport. """ class ClientSSHTransportBaseCase(ServerAndClientSSHTransportBaseCase): """ Base case for SSHClientTransport tests. """ klass: Optional[Type[transport.SSHTransportBase]] = transport.SSHClientTransport def verifyHostKey(self, pubKey, fingerprint): """ Mock version of SSHClientTransport.verifyHostKey. """ self.calledVerifyHostKey = True self.assertEqual(pubKey, self.blob) self.assertEqual( fingerprint.replace(b":", b""), binascii.hexlify(md5(pubKey).digest()) ) return defer.succeed(True) def setUp(self): TransportTestCase.setUp(self) self.blob = keys.Key.fromString(keydata.publicRSA_openssh).blob() self.privObj = keys.Key.fromString(keydata.privateRSA_openssh) self.calledVerifyHostKey = False self.proto.verifyHostKey = self.verifyHostKey class ClientSSHTransportTests(ClientSSHTransportBaseCase, TransportTestCase): """ Tests for SSHClientTransport. """ def test_KEXINITMultipleAlgorithms(self): """ Receiving a KEXINIT packet listing multiple supported algorithms will set up the first common algorithm, ordered after our preference. """ self.proto.dataReceived( b"SSH-2.0-Twisted\r\n\x00\x00\x01\xf4\x04\x14" b"\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99\x99" b"\x99\x00\x00\x00bdiffie-hellman-group1-sha1,diffie-hellman-g" b"roup-exchange-sha1,diffie-hellman-group-exchange-sha256\x00" b"\x00\x00\x0fssh-dss,ssh-rsa\x00\x00\x00\x85aes128-ctr,aes128-" b"cbc,aes192-ctr,aes192-cbc,aes256-ctr,aes256-cbc,cast128-ctr,c" b"ast128-cbc,blowfish-ctr,blowfish-cbc,3des-ctr,3des-cbc\x00" b"\x00\x00\x85aes128-ctr,aes128-cbc,aes192-ctr,aes192-cbc,aes25" b"6-ctr,aes256-cbc,cast128-ctr,cast128-cbc,blowfish-ctr,blowfis" b"h-cbc,3des-ctr,3des-cbc\x00\x00\x00\x12hmac-md5,hmac-sha1\x00" b"\x00\x00\x12hmac-md5,hmac-sha1\x00\x00\x00\tzlib,none\x00\x00" b"\x00\tzlib,none\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00" b"\x00\x00\x99\x99\x99\x99" ) # Even if client prefer diffie-hellman-group1-sha1, we will go for # diffie-hellman-group-exchange-sha256 as this what we prefer and is # also supported by the server. self.assertEqual(self.proto.kexAlg, b"diffie-hellman-group-exchange-sha256") self.assertEqual(self.proto.keyAlg, b"ssh-rsa") self.assertEqual(self.proto.outgoingCompressionType, b"none") self.assertEqual(self.proto.incomingCompressionType, b"none") ne = self.proto.nextEncryptions self.assertEqual(ne.outCipType, b"aes256-ctr") self.assertEqual(ne.inCipType, b"aes256-ctr") self.assertEqual(ne.outMACType, b"hmac-sha1") self.assertEqual(ne.inMACType, b"hmac-sha1") def test_notImplementedClientMethods(self): """ verifyHostKey() should return a Deferred which fails with a NotImplementedError exception. connectionSecure() should raise NotImplementedError(). """ self.assertRaises(NotImplementedError, self.klass().connectionSecure) def _checkRaises(f): f.trap(NotImplementedError) d = self.klass().verifyHostKey(None, None) return d.addCallback(self.fail).addErrback(_checkRaises) def assertKexInitResponseForDH(self, kexAlgorithm, bits): """ Test that a KEXINIT packet with a group1 or group14 key exchange results in a correct KEXDH_INIT response. @param kexAlgorithm: The key exchange algorithm to use @type kexAlgorithm: L{str} """ self.proto.supportedKeyExchanges = [kexAlgorithm] # Imitate reception of server key exchange request contained # in data returned by self.transport.value() self.proto.dataReceived(self.transport.value()) x = self.proto.dhSecretKey.private_numbers().x self.assertEqual(common.MP(x)[5:], b"\x99" * (bits // 8)) # Data sent to server should be a transport.MSG_KEXDH_INIT # message containing our public key. self.assertEqual( self.packets, [(transport.MSG_KEXDH_INIT, self.proto.dhSecretKeyPublicMP)] ) def test_KEXINIT_group14(self): """ KEXINIT messages requesting diffie-hellman-group14-sha1 result in KEXDH_INIT responses. """ self.assertKexInitResponseForDH(b"diffie-hellman-group14-sha1", 2048) def test_KEXINIT_badKexAlg(self): """ Test that the client raises a ConchError if it receives a KEXINIT message but doesn't have a key exchange algorithm that we understand. """ self.proto.supportedKeyExchanges = [b"diffie-hellman-group24-sha1"] data = self.transport.value().replace(b"group14", b"group24") self.assertRaises(ConchError, self.proto.dataReceived, data) def test_KEXINITExtensionNegotiation(self): """ If the server sends "ext-info-s" in its key exchange algorithms, then the client notes that the server supports extension negotiation. See RFC 8308, section 2.1. """ kexInitPacket = ( b"\x00" * 16 + common.NS(b"diffie-hellman-group-exchange-sha256,ext-info-s") + common.NS(b"ssh-rsa") + common.NS(b"aes256-ctr") + common.NS(b"aes256-ctr") + common.NS(b"hmac-sha1") + common.NS(b"hmac-sha1") + common.NS(b"none") + common.NS(b"none") + common.NS(b"") + common.NS(b"") + b"\x00" + b"\x00\x00\x00\x00" ) self.proto.ssh_KEXINIT(kexInitPacket) self.assertTrue(self.proto._peerSupportsExtensions) def begin_KEXDH_REPLY(self): """ Utility for test_KEXDH_REPLY and test_disconnectKEXDH_REPLYBadSignature. Begins a Diffie-Hellman key exchange in the named group Group-14 and computes information needed to return either a correct or incorrect signature. """ self.test_KEXINIT_group14() f = 2 fMP = common.MP(f) x = self.proto.dhSecretKey.private_numbers().x p = self.proto.p sharedSecret = _MPpow(f, x, p) h = sha1() h.update(common.NS(self.proto.ourVersionString) * 2) h.update(common.NS(self.proto.ourKexInitPayload) * 2) h.update(common.NS(self.blob)) h.update(self.proto.dhSecretKeyPublicMP) h.update(fMP) h.update(sharedSecret) exchangeHash = h.digest() signature = self.privObj.sign(exchangeHash) return (exchangeHash, signature, common.NS(self.blob) + fMP) def test_KEXDH_REPLY(self): """ Test that the KEXDH_REPLY message verifies the server. """ (exchangeHash, signature, packetStart) = self.begin_KEXDH_REPLY() def _cbTestKEXDH_REPLY(value): self.assertIsNone(value) self.assertTrue(self.calledVerifyHostKey) self.assertEqual(self.proto.sessionID, exchangeHash) d = self.proto.ssh_KEX_DH_GEX_GROUP(packetStart + common.NS(signature)) d.addCallback(_cbTestKEXDH_REPLY) return d def test_keySetup(self): """ Test that _keySetup sets up the next encryption keys. """ self.proto.kexAlg = b"diffie-hellman-group14-sha1" self.proto.nextEncryptions = MockCipher() self.simulateKeyExchange(b"AB", b"CD") self.assertEqual(self.proto.sessionID, b"CD") self.simulateKeyExchange(b"AB", b"EF") self.assertEqual(self.proto.sessionID, b"CD") self.assertEqual(self.packets[-1], (transport.MSG_NEWKEYS, b"")) newKeys = [self.proto._getKey(c, b"AB", b"EF") for c in iterbytes(b"ABCDEF")] self.assertEqual( self.proto.nextEncryptions.keys, (newKeys[0], newKeys[2], newKeys[1], newKeys[3], newKeys[4], newKeys[5]), ) def test_NEWKEYS(self): """ Test that NEWKEYS transitions the keys from nextEncryptions to currentEncryptions. """ self.test_KEXINITMultipleAlgorithms() secure = [False] def stubConnectionSecure(): secure[0] = True self.proto.connectionSecure = stubConnectionSecure self.proto.nextEncryptions = transport.SSHCiphers( b"none", b"none", b"none", b"none" ) self.simulateKeyExchange(b"AB", b"CD") self.assertIsNot(self.proto.currentEncryptions, self.proto.nextEncryptions) self.proto.nextEncryptions = MockCipher() self.proto.ssh_NEWKEYS(b"") self.assertIsNone(self.proto.outgoingCompression) self.assertIsNone(self.proto.incomingCompression) self.assertIs(self.proto.currentEncryptions, self.proto.nextEncryptions) self.assertTrue(secure[0]) self.proto.outgoingCompressionType = b"zlib" self.simulateKeyExchange(b"AB", b"GH") self.proto.ssh_NEWKEYS(b"") self.assertIsNotNone(self.proto.outgoingCompression) self.proto.incomingCompressionType = b"zlib" self.simulateKeyExchange(b"AB", b"IJ") self.proto.ssh_NEWKEYS(b"") self.assertIsNotNone(self.proto.incomingCompression) def test_SERVICE_ACCEPT(self): """ Test that the SERVICE_ACCEPT packet starts the requested service. """ self.proto.instance = MockService() self.proto.ssh_SERVICE_ACCEPT(b"\x00\x00\x00\x0bMockService") self.assertTrue(self.proto.instance.started) def test_requestService(self): """ Test that requesting a service sends a SERVICE_REQUEST packet. """ self.proto.requestService(MockService()) self.assertEqual( self.packets, [(transport.MSG_SERVICE_REQUEST, b"\x00\x00\x00\x0bMockService")], ) def test_disconnectKEXDH_REPLYBadSignature(self): """ Test that KEXDH_REPLY disconnects if the signature is bad. """ (exchangeHash, signature, packetStart) = self.begin_KEXDH_REPLY() d = self.proto.ssh_KEX_DH_GEX_GROUP(packetStart + common.NS(b"bad signature")) return d.addCallback( lambda _: self.checkDisconnected(transport.DISCONNECT_KEY_EXCHANGE_FAILED) ) def test_disconnectKEX_ECDH_REPLYBadSignature(self): """ Test that KEX_ECDH_REPLY disconnects if the signature is bad. """ kexmsg = ( b"\xAA" * 16 + common.NS(b"ecdh-sha2-nistp256") + common.NS(b"ssh-rsa") + common.NS(b"aes256-ctr") + common.NS(b"aes256-ctr") + common.NS(b"hmac-sha1") + common.NS(b"hmac-sha1") + common.NS(b"none") + common.NS(b"none") + common.NS(b"") + common.NS(b"") + b"\x00" + b"\x00\x00\x00\x00" ) self.proto.ssh_KEXINIT(kexmsg) self.proto.dataReceived(b"SSH-2.0-OpenSSH\r\n") self.proto.ecPriv = ec.generate_private_key(ec.SECP256R1(), default_backend()) self.proto.ecPub = self.proto.ecPriv.public_key() # Generate the private key thisPriv = ec.generate_private_key(ec.SECP256R1(), default_backend()) # Get the public key thisPub = thisPriv.public_key() encPub = thisPub.public_bytes( serialization.Encoding.X962, serialization.PublicFormat.UncompressedPoint ) self.proto.curve = ec.SECP256R1() self.proto.kexAlg = b"ecdh-sha2-nistp256" self.proto._ssh_KEX_ECDH_REPLY( common.NS(MockFactory().getPublicKeys()[b"ssh-rsa"].blob()) + common.NS(encPub) + common.NS(b"bad-signature") ) self.checkDisconnected(transport.DISCONNECT_KEY_EXCHANGE_FAILED) def test_disconnectNEWKEYSData(self): """ Test that NEWKEYS disconnects if it receives data. """ self.proto.ssh_NEWKEYS(b"bad packet") self.checkDisconnected() def test_disconnectSERVICE_ACCEPT(self): """ Test that SERVICE_ACCEPT disconnects if the accepted protocol is differet from the asked-for protocol. """ self.proto.instance = MockService() self.proto.ssh_SERVICE_ACCEPT(b"\x00\x00\x00\x03bad") self.checkDisconnected() def test_noPayloadSERVICE_ACCEPT(self): """ Some commercial SSH servers don't send a payload with the SERVICE_ACCEPT message. Conch pretends that it got the correct name of the service. """ self.proto.instance = MockService() self.proto.ssh_SERVICE_ACCEPT(b"") # no payload self.assertTrue(self.proto.instance.started) self.assertEqual(len(self.packets), 0) # not disconnected class ClientSSHTransportDHGroupExchangeBaseCase(ClientSSHTransportBaseCase): """ Diffie-Hellman group exchange tests for SSHClientTransport. """ """ 1536-bit modulus from RFC 3526 """ P1536 = int( "FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1" "29024E088A67CC74020BBEA63B139B22514A08798E3404DD" "EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245" "E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED" "EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3D" "C2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F" "83655D23DCA3AD961C62F356208552BB9ED529077096966D" "670C354E4ABC9804F1746C08CA237327FFFFFFFFFFFFFFFF", 16, ) def test_KEXINIT_groupexchange(self): """ KEXINIT packet with a group-exchange key exchange results in a KEX_DH_GEX_REQUEST message. """ self.proto.supportedKeyExchanges = [self.kexAlgorithm] self.proto.dataReceived(self.transport.value()) # The response will include our advertised group sizes. self.assertEqual( self.packets, [ ( transport.MSG_KEX_DH_GEX_REQUEST, b"\x00\x00\x04\x00\x00\x00\x08\x00\x00\x00\x20\x00", ) ], ) def test_KEX_DH_GEX_GROUP(self): """ Test that the KEX_DH_GEX_GROUP message results in a KEX_DH_GEX_INIT message with the client's Diffie-Hellman public key. """ self.test_KEXINIT_groupexchange() self.proto.ssh_KEX_DH_GEX_GROUP(common.MP(self.P1536) + common.MP(2)) self.assertEqual(self.proto.p, self.P1536) self.assertEqual(self.proto.g, 2) x = self.proto.dhSecretKey.private_numbers().x self.assertEqual(common.MP(x)[5:], b"\x99" * 192) self.assertEqual( self.proto.dhSecretKeyPublicMP, common.MP(pow(2, x, self.P1536)) ) self.assertEqual( self.packets[1:], [(transport.MSG_KEX_DH_GEX_INIT, self.proto.dhSecretKeyPublicMP)], ) def begin_KEX_DH_GEX_REPLY(self): """ Utility for test_KEX_DH_GEX_REPLY and test_disconnectGEX_REPLYBadSignature. Begins a Diffie-Hellman key exchange in an unnamed (server-specified) group and computes information needed to return either a correct or incorrect signature. """ self.test_KEX_DH_GEX_GROUP() p = self.proto.p f = 3 fMP = common.MP(f) sharedSecret = _MPpow(f, self.proto.dhSecretKey.private_numbers().x, p) h = self.hashProcessor() h.update(common.NS(self.proto.ourVersionString) * 2) h.update(common.NS(self.proto.ourKexInitPayload) * 2) h.update(common.NS(self.blob)) # Here is the wire format for advertised min, pref and max DH sizes. h.update(b"\x00\x00\x04\x00\x00\x00\x08\x00\x00\x00\x20\x00") # And the selected group parameters. h.update(common.MP(self.P1536) + common.MP(2)) h.update(self.proto.dhSecretKeyPublicMP) h.update(fMP) h.update(sharedSecret) exchangeHash = h.digest() signature = self.privObj.sign(exchangeHash) return (exchangeHash, signature, common.NS(self.blob) + fMP) def test_KEX_DH_GEX_REPLY(self): """ Test that the KEX_DH_GEX_REPLY message results in a verified server. """ (exchangeHash, signature, packetStart) = self.begin_KEX_DH_GEX_REPLY() def _cbTestKEX_DH_GEX_REPLY(value): self.assertIsNone(value) self.assertTrue(self.calledVerifyHostKey) self.assertEqual(self.proto.sessionID, exchangeHash) d = self.proto.ssh_KEX_DH_GEX_REPLY(packetStart + common.NS(signature)) d.addCallback(_cbTestKEX_DH_GEX_REPLY) return d def test_disconnectGEX_REPLYBadSignature(self): """ Test that KEX_DH_GEX_REPLY disconnects if the signature is bad. """ (exchangeHash, signature, packetStart) = self.begin_KEX_DH_GEX_REPLY() d = self.proto.ssh_KEX_DH_GEX_REPLY(packetStart + common.NS(b"bad signature")) return d.addCallback( lambda _: self.checkDisconnected(transport.DISCONNECT_KEY_EXCHANGE_FAILED) ) def test_disconnectKEX_ECDH_REPLYBadSignature(self): """ Test that KEX_ECDH_REPLY disconnects if the signature is bad. """ kexmsg = ( b"\xAA" * 16 + common.NS(b"ecdh-sha2-nistp256") + common.NS(b"ssh-rsa") + common.NS(b"aes256-ctr") + common.NS(b"aes256-ctr") + common.NS(b"hmac-sha1") + common.NS(b"hmac-sha1") + common.NS(b"none") + common.NS(b"none") + common.NS(b"") + common.NS(b"") + b"\x00" + b"\x00\x00\x00\x00" ) self.proto.ssh_KEXINIT(kexmsg) self.proto.dataReceived(b"SSH-2.0-OpenSSH\r\n") self.proto.ecPriv = ec.generate_private_key(ec.SECP256R1(), default_backend()) self.proto.ecPub = self.proto.ecPriv.public_key() # Generate the private key thisPriv = ec.generate_private_key(ec.SECP256R1(), default_backend()) # Get the public key thisPub = thisPriv.public_key() encPub = thisPub.public_bytes( serialization.Encoding.X962, serialization.PublicFormat.UncompressedPoint ) self.proto.curve = ec.SECP256R1() self.proto.kexAlg = b"ecdh-sha2-nistp256" self.proto._ssh_KEX_ECDH_REPLY( common.NS(MockFactory().getPublicKeys()[b"ssh-rsa"].blob()) + common.NS(encPub) + common.NS(b"bad-signature") ) self.checkDisconnected(transport.DISCONNECT_KEY_EXCHANGE_FAILED) class ClientSSHTransportDHGroupExchangeSHA1Tests( ClientSSHTransportDHGroupExchangeBaseCase, DHGroupExchangeSHA1Mixin, TransportTestCase, ): """ diffie-hellman-group-exchange-sha1 tests for SSHClientTransport. """ class ClientSSHTransportDHGroupExchangeSHA256Tests( ClientSSHTransportDHGroupExchangeBaseCase, DHGroupExchangeSHA256Mixin, TransportTestCase, ): """ diffie-hellman-group-exchange-sha256 tests for SSHClientTransport. """ class ClientSSHTransportECDHBaseCase(ClientSSHTransportBaseCase): """ Elliptic Curve Diffie-Hellman tests for SSHClientTransport. """ def test_KEXINIT(self): """ KEXINIT packet with an elliptic curve key exchange results in a KEXDH_INIT message. """ self.proto.supportedKeyExchanges = [self.kexAlgorithm] self.proto.dataReceived(self.transport.value()) # The response will include the client's ephemeral public key. self.assertEqual( self.packets, [ ( transport.MSG_KEXDH_INIT, common.NS(self.proto._encodeECPublicKey(self.proto.ecPub)), ) ], ) def begin_KEXDH_REPLY(self): """ Utility for test_KEXDH_REPLY and test_disconnectKEXDH_REPLYBadSignature. Begins an Elliptic Curve Diffie-Hellman key exchange and computes information needed to return either a correct or incorrect signature. """ self.test_KEXINIT() privKey = MockFactory().getPrivateKeys()[b"ssh-rsa"] pubKey = MockFactory().getPublicKeys()[b"ssh-rsa"] ecPriv = self.proto._generateECPrivateKey() ecPub = ecPriv.public_key() encPub = self.proto._encodeECPublicKey(ecPub) sharedSecret = self.proto._generateECSharedSecret( ecPriv, self.proto._encodeECPublicKey(self.proto.ecPub) ) h = self.hashProcessor() h.update(common.NS(self.proto.ourVersionString)) h.update(common.NS(self.proto.otherVersionString)) h.update(common.NS(self.proto.ourKexInitPayload)) h.update(common.NS(self.proto.otherKexInitPayload)) h.update(common.NS(pubKey.blob())) h.update(common.NS(self.proto._encodeECPublicKey(self.proto.ecPub))) h.update(common.NS(encPub)) h.update(sharedSecret) exchangeHash = h.digest() signature = privKey.sign(exchangeHash) return (exchangeHash, signature, common.NS(pubKey.blob()) + common.NS(encPub)) def test_KEXDH_REPLY(self): """ Test that the KEXDH_REPLY message completes the key exchange. """ (exchangeHash, signature, packetStart) = self.begin_KEXDH_REPLY() def _cbTestKEXDH_REPLY(value): self.assertIsNone(value) self.assertTrue(self.calledVerifyHostKey) self.assertEqual(self.proto.sessionID, exchangeHash) d = self.proto.ssh_KEX_DH_GEX_GROUP(packetStart + common.NS(signature)) d.addCallback(_cbTestKEXDH_REPLY) return d def test_disconnectKEXDH_REPLYBadSignature(self): """ Test that KEX_ECDH_REPLY disconnects if the signature is bad. """ (exchangeHash, signature, packetStart) = self.begin_KEXDH_REPLY() d = self.proto.ssh_KEX_DH_GEX_GROUP(packetStart + common.NS(b"bad signature")) return d.addCallback( lambda _: self.checkDisconnected(transport.DISCONNECT_KEY_EXCHANGE_FAILED) ) class ClientSSHTransportECDHTests( ClientSSHTransportECDHBaseCase, ECDHMixin, TransportTestCase ): """ ecdh-sha2-nistp256 tests for SSHClientTransport. """ @skipWithoutX25519 class ClientSSHTransportCurve25519SHA256Tests( ClientSSHTransportECDHBaseCase, Curve25519SHA256Mixin, TransportTestCase ): """ curve25519-sha256 tests for SSHClientTransport. """ class GetMACTests(TestCase): """ Tests for L{SSHCiphers._getMAC}. """ if dependencySkip: skip = dependencySkip def setUp(self): self.ciphers = transport.SSHCiphers(b"A", b"B", b"C", b"D") def getSharedSecret(self): """ Generate a new shared secret to be used with the tests. @return: A new secret. @rtype: L{bytes} """ return insecureRandom(64) def assertGetMAC(self, hmacName, hashProcessor, digestSize, blockPadSize): """ Check that when L{SSHCiphers._getMAC} is called with a supportd HMAC algorithm name it returns a tuple of (digest object, inner pad, outer pad, digest size) with a C{key} attribute set to the value of the key supplied. @param hmacName: Identifier of HMAC algorithm. @type hmacName: L{bytes} @param hashProcessor: Callable for the hash algorithm. @type hashProcessor: C{callable} @param digestSize: Size of the digest for algorithm. @type digestSize: L{int} @param blockPadSize: Size of padding applied to the shared secret to match the block size. @type blockPadSize: L{int} """ secret = self.getSharedSecret() params = self.ciphers._getMAC(hmacName, secret) key = secret[:digestSize] + b"\x00" * blockPadSize innerPad = bytes(ord(b) ^ 0x36 for b in iterbytes(key)) outerPad = bytes(ord(b) ^ 0x5C for b in iterbytes(key)) self.assertEqual((hashProcessor, innerPad, outerPad, digestSize), params) self.assertEqual(key, params.key) def test_hmacsha2512(self): """ When L{SSHCiphers._getMAC} is called with the C{b"hmac-sha2-512"} MAC algorithm name it returns a tuple of (sha512 digest object, inner pad, outer pad, sha512 digest size) with a C{key} attribute set to the value of the key supplied. """ self.assertGetMAC(b"hmac-sha2-512", sha512, digestSize=64, blockPadSize=64) def test_hmacsha2384(self): """ When L{SSHCiphers._getMAC} is called with the C{b"hmac-sha2-384"} MAC algorithm name it returns a tuple of (sha384 digest object, inner pad, outer pad, sha384 digest size) with a C{key} attribute set to the value of the key supplied. """ self.assertGetMAC(b"hmac-sha2-384", sha384, digestSize=48, blockPadSize=80) def test_hmacsha2256(self): """ When L{SSHCiphers._getMAC} is called with the C{b"hmac-sha2-256"} MAC algorithm name it returns a tuple of (sha256 digest object, inner pad, outer pad, sha256 digest size) with a C{key} attribute set to the value of the key supplied. """ self.assertGetMAC(b"hmac-sha2-256", sha256, digestSize=32, blockPadSize=32) def test_hmacsha1(self): """ When L{SSHCiphers._getMAC} is called with the C{b"hmac-sha1"} MAC algorithm name it returns a tuple of (sha1 digest object, inner pad, outer pad, sha1 digest size) with a C{key} attribute set to the value of the key supplied. """ self.assertGetMAC(b"hmac-sha1", sha1, digestSize=20, blockPadSize=44) def test_hmacmd5(self): """ When L{SSHCiphers._getMAC} is called with the C{b"hmac-md5"} MAC algorithm name it returns a tuple of (md5 digest object, inner pad, outer pad, md5 digest size) with a C{key} attribute set to the value of the key supplied. """ self.assertGetMAC(b"hmac-md5", md5, digestSize=16, blockPadSize=48) def test_none(self): """ When L{SSHCiphers._getMAC} is called with the C{b"none"} MAC algorithm name it returns a tuple of (None, "", "", 0). """ key = self.getSharedSecret() params = self.ciphers._getMAC(b"none", key) self.assertEqual((None, b"", b"", 0), params) class SSHCiphersTests(TestCase): """ Tests for the SSHCiphers helper class. """ if dependencySkip: skip = dependencySkip def test_init(self): """ Test that the initializer sets up the SSHCiphers object. """ ciphers = transport.SSHCiphers(b"A", b"B", b"C", b"D") self.assertEqual(ciphers.outCipType, b"A") self.assertEqual(ciphers.inCipType, b"B") self.assertEqual(ciphers.outMACType, b"C") self.assertEqual(ciphers.inMACType, b"D") def test_getCipher(self): """ Test that the _getCipher method returns the correct cipher. """ ciphers = transport.SSHCiphers(b"A", b"B", b"C", b"D") iv = key = b"\x00" * 16 for cipName, (algClass, keySize, counter) in ciphers.cipherMap.items(): cip = ciphers._getCipher(cipName, iv, key) if cipName == b"none": self.assertIsInstance(cip, transport._DummyCipher) else: self.assertIsInstance(cip.algorithm, algClass) def test_setKeysCiphers(self): """ Test that setKeys sets up the ciphers. """ key = b"\x00" * 64 for cipName in transport.SSHTransportBase.supportedCiphers: modName, keySize, counter = transport.SSHCiphers.cipherMap[cipName] encCipher = transport.SSHCiphers(cipName, b"none", b"none", b"none") decCipher = transport.SSHCiphers(b"none", cipName, b"none", b"none") cip = encCipher._getCipher(cipName, key, key) bs = cip.algorithm.block_size // 8 encCipher.setKeys(key, key, b"", b"", b"", b"") decCipher.setKeys(b"", b"", key, key, b"", b"") self.assertEqual(encCipher.encBlockSize, bs) self.assertEqual(decCipher.decBlockSize, bs) encryptor = cip.encryptor() enc = encryptor.update(key[:bs]) enc2 = encryptor.update(key[:bs]) self.assertEqual(encCipher.encrypt(key[:bs]), enc) self.assertEqual(encCipher.encrypt(key[:bs]), enc2) self.assertEqual(decCipher.decrypt(enc), key[:bs]) self.assertEqual(decCipher.decrypt(enc2), key[:bs]) def test_setKeysMACs(self): """ Test that setKeys sets up the MACs. """ key = b"\x00" * 64 for macName, mod in transport.SSHCiphers.macMap.items(): outMac = transport.SSHCiphers(b"none", b"none", macName, b"none") inMac = transport.SSHCiphers(b"none", b"none", b"none", macName) outMac.setKeys(b"", b"", b"", b"", key, b"") inMac.setKeys(b"", b"", b"", b"", b"", key) if mod: ds = mod().digest_size else: ds = 0 self.assertEqual(inMac.verifyDigestSize, ds) if mod: mod, i, o, ds = outMac._getMAC(macName, key) seqid = 0 data = key packet = b"\x00" * 4 + key if mod: mac = mod(o + mod(i + packet).digest()).digest() else: mac = b"" self.assertEqual(outMac.makeMAC(seqid, data), mac) self.assertTrue(inMac.verify(seqid, data, mac)) def test_makeMAC(self): """ L{SSHCiphers.makeMAC} computes the HMAC of an outgoing SSH message with a particular sequence id and content data. """ # Use the test vectors given in the appendix of RFC 2104. vectors = [ (b"\x0b" * 16, b"Hi There", b"9294727a3638bb1c13f48ef8158bfc9d"), ( b"Jefe", b"what do ya want for nothing?", b"750c783e6ab0b503eaa86e310a5db738", ), (b"\xAA" * 16, b"\xDD" * 50, b"56be34521d144c88dbb8c733f0e8b3f6"), ] for key, data, mac in vectors: outMAC = transport.SSHCiphers(b"none", b"none", b"hmac-md5", b"none") outMAC.outMAC = outMAC._getMAC(b"hmac-md5", key) (seqid,) = struct.unpack(">L", data[:4]) shortened = data[4:] self.assertEqual( mac, binascii.hexlify(outMAC.makeMAC(seqid, shortened)), f"Failed HMAC test vector; key={key!r} data={data!r}", ) class TransportLoopbackTests(TestCase): """ Test the server transport and client transport against each other, """ if dependencySkip: skip = dependencySkip def _runClientServer(self, mod): """ Run an async client and server, modifying each using the mod function provided. Returns a Deferred called back when both Protocols have disconnected. @type mod: C{func} @rtype: C{defer.Deferred} """ factory = MockFactory() server = transport.SSHServerTransport() server.factory = factory factory.startFactory() server.errors = [] server.receiveError = lambda code, desc: server.errors.append((code, desc)) client = transport.SSHClientTransport() client.verifyHostKey = lambda x, y: defer.succeed(None) client.errors = [] client.receiveError = lambda code, desc: client.errors.append((code, desc)) client.connectionSecure = lambda: client.loseConnection() server.supportedPublicKeys = list(server.factory.getPublicKeys().keys()) server = mod(server) client = mod(client) def check(ignored, server, client): name = repr( [ server.supportedCiphers[0], server.supportedMACs[0], server.supportedKeyExchanges[0], server.supportedCompressions[0], ] ) self.assertEqual(client.errors, []) self.assertEqual( server.errors, [(transport.DISCONNECT_CONNECTION_LOST, b"user closed connection")], ) if server.supportedCiphers[0] == b"none": self.assertFalse(server.isEncrypted(), name) self.assertFalse(client.isEncrypted(), name) else: self.assertTrue(server.isEncrypted(), name) self.assertTrue(client.isEncrypted(), name) if server.supportedMACs[0] == b"none": self.assertFalse(server.isVerified(), name) self.assertFalse(client.isVerified(), name) else: self.assertTrue(server.isVerified(), name) self.assertTrue(client.isVerified(), name) d = loopback.loopbackAsync(server, client) d.addCallback(check, server, client) return d def test_ciphers(self): """ Test that the client and server play nicely together, in all the various combinations of ciphers. """ deferreds = [] for cipher in transport.SSHTransportBase.supportedCiphers + [b"none"]: def setCipher(proto): proto.supportedCiphers = [cipher] return proto deferreds.append(self._runClientServer(setCipher)) return defer.DeferredList(deferreds, fireOnOneErrback=True) def test_macs(self): """ Like test_ciphers, but for the various MACs. """ deferreds = [] for mac in transport.SSHTransportBase.supportedMACs + [b"none"]: def setMAC(proto): proto.supportedMACs = [mac] return proto deferreds.append(self._runClientServer(setMAC)) return defer.DeferredList(deferreds, fireOnOneErrback=True) def test_keyexchanges(self): """ Like test_ciphers, but for the various key exchanges. """ deferreds = [] for kexAlgorithm in transport.SSHTransportBase.supportedKeyExchanges: def setKeyExchange(proto): proto.supportedKeyExchanges = [kexAlgorithm] return proto deferreds.append(self._runClientServer(setKeyExchange)) return defer.DeferredList(deferreds, fireOnOneErrback=True) def test_compressions(self): """ Like test_ciphers, but for the various compressions. """ deferreds = [] for compression in transport.SSHTransportBase.supportedCompressions: def setCompression(proto): proto.supportedCompressions = [compression] return proto deferreds.append(self._runClientServer(setCompression)) return defer.DeferredList(deferreds, fireOnOneErrback=True) PK F��Z�7�d�3 �3 test_agent.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.ssh.agent}. """ import struct from twisted.test import iosim from twisted.trial import unittest try: import cryptography as _cryptography except ImportError: cryptography = None else: cryptography = _cryptography try: import pyasn1 as _pyasn1 # type: ignore[import] except ImportError: pyasn1 = None else: pyasn1 = _pyasn1 try: from twisted.conch.ssh import agent as _agent, keys as _keys except ImportError: keys = agent = None else: keys, agent = _keys, _agent from twisted.conch.error import ConchError, MissingKeyStoreError from twisted.conch.test import keydata class StubFactory: """ Mock factory that provides the keys attribute required by the SSHAgentServerProtocol """ def __init__(self): self.keys = {} class AgentTestBase(unittest.TestCase): """ Tests for SSHAgentServer/Client. """ if agent is None or keys is None: skip = "Cannot run without cryptography or PyASN1" def setUp(self): # wire up our client <-> server self.client, self.server, self.pump = iosim.connectedServerAndClient( agent.SSHAgentServer, agent.SSHAgentClient ) # the server's end of the protocol is stateful and we store it on the # factory, for which we only need a mock self.server.factory = StubFactory() # pub/priv keys of each kind self.rsaPrivate = keys.Key.fromString(keydata.privateRSA_openssh) self.dsaPrivate = keys.Key.fromString(keydata.privateDSA_openssh) self.rsaPublic = keys.Key.fromString(keydata.publicRSA_openssh) self.dsaPublic = keys.Key.fromString(keydata.publicDSA_openssh) class ServerProtocolContractWithFactoryTests(AgentTestBase): """ The server protocol is stateful and so uses its factory to track state across requests. This test asserts that the protocol raises if its factory doesn't provide the necessary storage for that state. """ def test_factorySuppliesKeyStorageForServerProtocol(self): # need a message to send into the server msg = struct.pack("!LB", 1, agent.AGENTC_REQUEST_IDENTITIES) del self.server.factory.__dict__["keys"] self.assertRaises(MissingKeyStoreError, self.server.dataReceived, msg) class UnimplementedVersionOneServerTests(AgentTestBase): """ Tests for methods with no-op implementations on the server. We need these for clients, such as openssh, that try v1 methods before going to v2. Because the client doesn't expose these operations with nice method names, we invoke sendRequest directly with an op code. """ def test_agentc_REQUEST_RSA_IDENTITIES(self): """ assert that we get the correct op code for an RSA identities request """ d = self.client.sendRequest(agent.AGENTC_REQUEST_RSA_IDENTITIES, b"") self.pump.flush() def _cb(packet): self.assertEqual(agent.AGENT_RSA_IDENTITIES_ANSWER, ord(packet[0:1])) return d.addCallback(_cb) def test_agentc_REMOVE_RSA_IDENTITY(self): """ assert that we get the correct op code for an RSA remove identity request """ d = self.client.sendRequest(agent.AGENTC_REMOVE_RSA_IDENTITY, b"") self.pump.flush() return d.addCallback(self.assertEqual, b"") def test_agentc_REMOVE_ALL_RSA_IDENTITIES(self): """ assert that we get the correct op code for an RSA remove all identities request. """ d = self.client.sendRequest(agent.AGENTC_REMOVE_ALL_RSA_IDENTITIES, b"") self.pump.flush() return d.addCallback(self.assertEqual, b"") if agent is not None: class CorruptServer(agent.SSHAgentServer): # type: ignore[name-defined] """ A misbehaving server that returns bogus response op codes so that we can verify that our callbacks that deal with these op codes handle such miscreants. """ def agentc_REQUEST_IDENTITIES(self, data): self.sendResponse(254, b"") def agentc_SIGN_REQUEST(self, data): self.sendResponse(254, b"") class ClientWithBrokenServerTests(AgentTestBase): """ verify error handling code in the client using a misbehaving server """ def setUp(self): AgentTestBase.setUp(self) self.client, self.server, self.pump = iosim.connectedServerAndClient( CorruptServer, agent.SSHAgentClient ) # the server's end of the protocol is stateful and we store it on the # factory, for which we only need a mock self.server.factory = StubFactory() def test_signDataCallbackErrorHandling(self): """ Assert that L{SSHAgentClient.signData} raises a ConchError if we get a response from the server whose opcode doesn't match the protocol for data signing requests. """ d = self.client.signData(self.rsaPublic.blob(), b"John Hancock") self.pump.flush() return self.assertFailure(d, ConchError) def test_requestIdentitiesCallbackErrorHandling(self): """ Assert that L{SSHAgentClient.requestIdentities} raises a ConchError if we get a response from the server whose opcode doesn't match the protocol for identity requests. """ d = self.client.requestIdentities() self.pump.flush() return self.assertFailure(d, ConchError) class AgentKeyAdditionTests(AgentTestBase): """ Test adding different flavors of keys to an agent. """ def test_addRSAIdentityNoComment(self): """ L{SSHAgentClient.addIdentity} adds the private key it is called with to the SSH agent server to which it is connected, associating it with the comment it is called with. This test asserts that omitting the comment produces an empty string for the comment on the server. """ d = self.client.addIdentity(self.rsaPrivate.privateBlob()) self.pump.flush() def _check(ignored): serverKey = self.server.factory.keys[self.rsaPrivate.blob()] self.assertEqual(self.rsaPrivate, serverKey[0]) self.assertEqual(b"", serverKey[1]) return d.addCallback(_check) def test_addDSAIdentityNoComment(self): """ L{SSHAgentClient.addIdentity} adds the private key it is called with to the SSH agent server to which it is connected, associating it with the comment it is called with. This test asserts that omitting the comment produces an empty string for the comment on the server. """ d = self.client.addIdentity(self.dsaPrivate.privateBlob()) self.pump.flush() def _check(ignored): serverKey = self.server.factory.keys[self.dsaPrivate.blob()] self.assertEqual(self.dsaPrivate, serverKey[0]) self.assertEqual(b"", serverKey[1]) return d.addCallback(_check) def test_addRSAIdentityWithComment(self): """ L{SSHAgentClient.addIdentity} adds the private key it is called with to the SSH agent server to which it is connected, associating it with the comment it is called with. This test asserts that the server receives/stores the comment as sent by the client. """ d = self.client.addIdentity( self.rsaPrivate.privateBlob(), comment=b"My special key" ) self.pump.flush() def _check(ignored): serverKey = self.server.factory.keys[self.rsaPrivate.blob()] self.assertEqual(self.rsaPrivate, serverKey[0]) self.assertEqual(b"My special key", serverKey[1]) return d.addCallback(_check) def test_addDSAIdentityWithComment(self): """ L{SSHAgentClient.addIdentity} adds the private key it is called with to the SSH agent server to which it is connected, associating it with the comment it is called with. This test asserts that the server receives/stores the comment as sent by the client. """ d = self.client.addIdentity( self.dsaPrivate.privateBlob(), comment=b"My special key" ) self.pump.flush() def _check(ignored): serverKey = self.server.factory.keys[self.dsaPrivate.blob()] self.assertEqual(self.dsaPrivate, serverKey[0]) self.assertEqual(b"My special key", serverKey[1]) return d.addCallback(_check) class AgentClientFailureTests(AgentTestBase): def test_agentFailure(self): """ verify that the client raises ConchError on AGENT_FAILURE """ d = self.client.sendRequest(254, b"") self.pump.flush() return self.assertFailure(d, ConchError) class AgentIdentityRequestsTests(AgentTestBase): """ Test operations against a server with identities already loaded. """ def setUp(self): AgentTestBase.setUp(self) self.server.factory.keys[self.dsaPrivate.blob()] = ( self.dsaPrivate, b"a comment", ) self.server.factory.keys[self.rsaPrivate.blob()] = ( self.rsaPrivate, b"another comment", ) def test_signDataRSA(self): """ Sign data with an RSA private key and then verify it with the public key. """ d = self.client.signData(self.rsaPublic.blob(), b"John Hancock") self.pump.flush() signature = self.successResultOf(d) expected = self.rsaPrivate.sign(b"John Hancock") self.assertEqual(expected, signature) self.assertTrue(self.rsaPublic.verify(signature, b"John Hancock")) def test_signDataDSA(self): """ Sign data with a DSA private key and then verify it with the public key. """ d = self.client.signData(self.dsaPublic.blob(), b"John Hancock") self.pump.flush() def _check(sig): # Cannot do this b/c DSA uses random numbers when signing # expected = self.dsaPrivate.sign("John Hancock") # self.assertEqual(expected, sig) self.assertTrue(self.dsaPublic.verify(sig, b"John Hancock")) return d.addCallback(_check) def test_signDataRSAErrbackOnUnknownBlob(self): """ Assert that we get an errback if we try to sign data using a key that wasn't added. """ del self.server.factory.keys[self.rsaPublic.blob()] d = self.client.signData(self.rsaPublic.blob(), b"John Hancock") self.pump.flush() return self.assertFailure(d, ConchError) def test_requestIdentities(self): """ Assert that we get all of the keys/comments that we add when we issue a request for all identities. """ d = self.client.requestIdentities() self.pump.flush() def _check(keyt): expected = {} expected[self.dsaPublic.blob()] = b"a comment" expected[self.rsaPublic.blob()] = b"another comment" received = {} for k in keyt: received[keys.Key.fromString(k[0], type="blob").blob()] = k[1] self.assertEqual(expected, received) return d.addCallback(_check) class AgentKeyRemovalTests(AgentTestBase): """ Test support for removing keys in a remote server. """ def setUp(self): AgentTestBase.setUp(self) self.server.factory.keys[self.dsaPrivate.blob()] = ( self.dsaPrivate, b"a comment", ) self.server.factory.keys[self.rsaPrivate.blob()] = ( self.rsaPrivate, b"another comment", ) def test_removeRSAIdentity(self): """ Assert that we can remove an RSA identity. """ # only need public key for this d = self.client.removeIdentity(self.rsaPrivate.blob()) self.pump.flush() def _check(ignored): self.assertEqual(1, len(self.server.factory.keys)) self.assertIn(self.dsaPrivate.blob(), self.server.factory.keys) self.assertNotIn(self.rsaPrivate.blob(), self.server.factory.keys) return d.addCallback(_check) def test_removeDSAIdentity(self): """ Assert that we can remove a DSA identity. """ # only need public key for this d = self.client.removeIdentity(self.dsaPrivate.blob()) self.pump.flush() def _check(ignored): self.assertEqual(1, len(self.server.factory.keys)) self.assertIn(self.rsaPrivate.blob(), self.server.factory.keys) return d.addCallback(_check) def test_removeAllIdentities(self): """ Assert that we can remove all identities. """ d = self.client.removeAllIdentities() self.pump.flush() def _check(ignored): self.assertEqual(0, len(self.server.factory.keys)) return d.addCallback(_check) PK F��Z[���r r test_forwarding.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.ssh.forwarding}. """ from twisted.python.reflect import requireModule cryptography = requireModule("cryptography") if cryptography: from twisted.conch.ssh import forwarding from twisted.internet.address import IPv6Address from twisted.internet.test.test_endpoints import deterministicResolvingReactor from twisted.test.proto_helpers import MemoryReactorClock, StringTransport from twisted.trial import unittest class TestSSHConnectForwardingChannel(unittest.TestCase): """ Unit and integration tests for L{SSHConnectForwardingChannel}. """ if not cryptography: skip = "Cannot run without cryptography" def makeTCPConnection(self, reactor): """ Fake that connection was established for first connectTCP request made on C{reactor}. @param reactor: Reactor on which to fake the connection. @type reactor: A reactor. """ factory = reactor.tcpClients[0][2] connector = reactor.connectors[0] protocol = factory.buildProtocol(None) transport = StringTransport(peerAddress=connector.getDestination()) protocol.makeConnection(transport) def test_channelOpenHostnameRequests(self): """ When a hostname is sent as part of forwarding requests, it is resolved using HostnameEndpoint's resolver. """ sut = forwarding.SSHConnectForwardingChannel(hostport=("fwd.example.org", 1234)) # Patch channel and resolver to not touch the network. memoryReactor = MemoryReactorClock() sut._reactor = deterministicResolvingReactor(memoryReactor, ["::1"]) sut.channelOpen(None) self.makeTCPConnection(memoryReactor) self.successResultOf(sut._channelOpenDeferred) # Channel is connected using a forwarding client to the resolved # address of the requested host. self.assertIsInstance(sut.client, forwarding.SSHForwardingClient) self.assertEqual( IPv6Address("TCP", "::1", 1234), sut.client.transport.getPeer() ) PK F��Z!�*IXc Xc test_recvline.pynu �[��� # -*- test-case-name: twisted.conch.test.test_recvline -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.recvline} and fixtures for testing related functionality. """ import os import sys from unittest import skipIf from twisted.conch import recvline from twisted.conch.insults import insults from twisted.cred import portal from twisted.internet import defer, error from twisted.python import components, filepath, reflect from twisted.python.compat import iterbytes from twisted.python.reflect import requireModule from twisted.test.proto_helpers import StringTransport from twisted.trial.unittest import SkipTest, TestCase stdio = requireModule("twisted.conch.stdio") properEnv = dict(os.environ) properEnv["PYTHONPATH"] = os.pathsep.join(sys.path) class ArrowsTests(TestCase): def setUp(self): self.underlyingTransport = StringTransport() self.pt = insults.ServerProtocol() self.p = recvline.HistoricRecvLine() self.pt.protocolFactory = lambda: self.p self.pt.factory = self self.pt.makeConnection(self.underlyingTransport) def test_printableCharacters(self): """ When L{HistoricRecvLine} receives a printable character, it adds it to the current line buffer. """ self.p.keystrokeReceived(b"x", None) self.p.keystrokeReceived(b"y", None) self.p.keystrokeReceived(b"z", None) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) def test_horizontalArrows(self): """ When L{HistoricRecvLine} receives a LEFT_ARROW or RIGHT_ARROW keystroke it moves the cursor left or right in the current line buffer, respectively. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) for ch in iterbytes(b"xyz"): kR(ch) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) kR(self.pt.RIGHT_ARROW) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) kR(self.pt.LEFT_ARROW) self.assertEqual(self.p.currentLineBuffer(), (b"xy", b"z")) kR(self.pt.LEFT_ARROW) self.assertEqual(self.p.currentLineBuffer(), (b"x", b"yz")) kR(self.pt.LEFT_ARROW) self.assertEqual(self.p.currentLineBuffer(), (b"", b"xyz")) kR(self.pt.LEFT_ARROW) self.assertEqual(self.p.currentLineBuffer(), (b"", b"xyz")) kR(self.pt.RIGHT_ARROW) self.assertEqual(self.p.currentLineBuffer(), (b"x", b"yz")) kR(self.pt.RIGHT_ARROW) self.assertEqual(self.p.currentLineBuffer(), (b"xy", b"z")) kR(self.pt.RIGHT_ARROW) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) kR(self.pt.RIGHT_ARROW) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) def test_newline(self): """ When {HistoricRecvLine} receives a newline, it adds the current line buffer to the end of its history buffer. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) for ch in iterbytes(b"xyz\nabc\n123\n"): kR(ch) self.assertEqual(self.p.currentHistoryBuffer(), ((b"xyz", b"abc", b"123"), ())) kR(b"c") kR(b"b") kR(b"a") self.assertEqual(self.p.currentHistoryBuffer(), ((b"xyz", b"abc", b"123"), ())) kR(b"\n") self.assertEqual( self.p.currentHistoryBuffer(), ((b"xyz", b"abc", b"123", b"cba"), ()) ) def test_verticalArrows(self): """ When L{HistoricRecvLine} receives UP_ARROW or DOWN_ARROW keystrokes it move the current index in the current history buffer up or down, and resets the current line buffer to the previous or next line in history, respectively for each. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) for ch in iterbytes(b"xyz\nabc\n123\n"): kR(ch) self.assertEqual(self.p.currentHistoryBuffer(), ((b"xyz", b"abc", b"123"), ())) self.assertEqual(self.p.currentLineBuffer(), (b"", b"")) kR(self.pt.UP_ARROW) self.assertEqual(self.p.currentHistoryBuffer(), ((b"xyz", b"abc"), (b"123",))) self.assertEqual(self.p.currentLineBuffer(), (b"123", b"")) kR(self.pt.UP_ARROW) self.assertEqual(self.p.currentHistoryBuffer(), ((b"xyz",), (b"abc", b"123"))) self.assertEqual(self.p.currentLineBuffer(), (b"abc", b"")) kR(self.pt.UP_ARROW) self.assertEqual(self.p.currentHistoryBuffer(), ((), (b"xyz", b"abc", b"123"))) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) kR(self.pt.UP_ARROW) self.assertEqual(self.p.currentHistoryBuffer(), ((), (b"xyz", b"abc", b"123"))) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) for i in range(4): kR(self.pt.DOWN_ARROW) self.assertEqual(self.p.currentHistoryBuffer(), ((b"xyz", b"abc", b"123"), ())) def test_home(self): """ When L{HistoricRecvLine} receives a HOME keystroke it moves the cursor to the beginning of the current line buffer. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) for ch in iterbytes(b"hello, world"): kR(ch) self.assertEqual(self.p.currentLineBuffer(), (b"hello, world", b"")) kR(self.pt.HOME) self.assertEqual(self.p.currentLineBuffer(), (b"", b"hello, world")) def test_end(self): """ When L{HistoricRecvLine} receives an END keystroke it moves the cursor to the end of the current line buffer. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) for ch in iterbytes(b"hello, world"): kR(ch) self.assertEqual(self.p.currentLineBuffer(), (b"hello, world", b"")) kR(self.pt.HOME) kR(self.pt.END) self.assertEqual(self.p.currentLineBuffer(), (b"hello, world", b"")) def test_backspace(self): """ When L{HistoricRecvLine} receives a BACKSPACE keystroke it deletes the character immediately before the cursor. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) for ch in iterbytes(b"xyz"): kR(ch) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) kR(self.pt.BACKSPACE) self.assertEqual(self.p.currentLineBuffer(), (b"xy", b"")) kR(self.pt.LEFT_ARROW) kR(self.pt.BACKSPACE) self.assertEqual(self.p.currentLineBuffer(), (b"", b"y")) kR(self.pt.BACKSPACE) self.assertEqual(self.p.currentLineBuffer(), (b"", b"y")) def test_delete(self): """ When L{HistoricRecvLine} receives a DELETE keystroke, it delets the character immediately after the cursor. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) for ch in iterbytes(b"xyz"): kR(ch) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) kR(self.pt.DELETE) self.assertEqual(self.p.currentLineBuffer(), (b"xyz", b"")) kR(self.pt.LEFT_ARROW) kR(self.pt.DELETE) self.assertEqual(self.p.currentLineBuffer(), (b"xy", b"")) kR(self.pt.LEFT_ARROW) kR(self.pt.DELETE) self.assertEqual(self.p.currentLineBuffer(), (b"x", b"")) kR(self.pt.LEFT_ARROW) kR(self.pt.DELETE) self.assertEqual(self.p.currentLineBuffer(), (b"", b"")) kR(self.pt.DELETE) self.assertEqual(self.p.currentLineBuffer(), (b"", b"")) def test_insert(self): """ When not in INSERT mode, L{HistoricRecvLine} inserts the typed character at the cursor before the next character. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) for ch in iterbytes(b"xyz"): kR(ch) kR(self.pt.LEFT_ARROW) kR(b"A") self.assertEqual(self.p.currentLineBuffer(), (b"xyA", b"z")) kR(self.pt.LEFT_ARROW) kR(b"B") self.assertEqual(self.p.currentLineBuffer(), (b"xyB", b"Az")) def test_typeover(self): """ When in INSERT mode and upon receiving a keystroke with a printable character, L{HistoricRecvLine} replaces the character at the cursor with the typed character rather than inserting before. Ah, the ironies of INSERT mode. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) for ch in iterbytes(b"xyz"): kR(ch) kR(self.pt.INSERT) kR(self.pt.LEFT_ARROW) kR(b"A") self.assertEqual(self.p.currentLineBuffer(), (b"xyA", b"")) kR(self.pt.LEFT_ARROW) kR(b"B") self.assertEqual(self.p.currentLineBuffer(), (b"xyB", b"")) def test_unprintableCharacters(self): """ When L{HistoricRecvLine} receives a keystroke for an unprintable function key with no assigned behavior, the line buffer is unmodified. """ kR = lambda ch: self.p.keystrokeReceived(ch, None) pt = self.pt for ch in ( pt.F1, pt.F2, pt.F3, pt.F4, pt.F5, pt.F6, pt.F7, pt.F8, pt.F9, pt.F10, pt.F11, pt.F12, pt.PGUP, pt.PGDN, ): kR(ch) self.assertEqual(self.p.currentLineBuffer(), (b"", b"")) from twisted.conch import telnet from twisted.conch.insults import helper from twisted.conch.test.loopback import LoopbackRelay class EchoServer(recvline.HistoricRecvLine): def lineReceived(self, line): self.terminal.write(line + b"\n" + self.ps[self.pn]) # An insults API for this would be nice. left = b"\x1b[D" right = b"\x1b[C" up = b"\x1b[A" down = b"\x1b[B" insert = b"\x1b[2~" home = b"\x1b[1~" delete = b"\x1b[3~" end = b"\x1b[4~" backspace = b"\x7f" from twisted.cred import checkers try: from twisted.conch.manhole_ssh import ( ConchFactory, TerminalRealm, TerminalSession, TerminalSessionTransport, TerminalUser, ) from twisted.conch.ssh import ( channel, connection, keys, session, transport, userauth, ) except ImportError: ssh = False else: ssh = True class SessionChannel(channel.SSHChannel): name = b"session" def __init__( self, protocolFactory, protocolArgs, protocolKwArgs, width, height, *a, **kw ): channel.SSHChannel.__init__(self, *a, **kw) self.protocolFactory = protocolFactory self.protocolArgs = protocolArgs self.protocolKwArgs = protocolKwArgs self.width = width self.height = height def channelOpen(self, data): term = session.packRequest_pty_req( b"vt102", (self.height, self.width, 0, 0), b"" ) self.conn.sendRequest(self, b"pty-req", term) self.conn.sendRequest(self, b"shell", b"") self._protocolInstance = self.protocolFactory( *self.protocolArgs, **self.protocolKwArgs ) self._protocolInstance.factory = self self._protocolInstance.makeConnection(self) def closed(self): self._protocolInstance.connectionLost(error.ConnectionDone()) def dataReceived(self, data): self._protocolInstance.dataReceived(data) class TestConnection(connection.SSHConnection): def __init__( self, protocolFactory, protocolArgs, protocolKwArgs, width, height, *a, **kw ): connection.SSHConnection.__init__(self, *a, **kw) self.protocolFactory = protocolFactory self.protocolArgs = protocolArgs self.protocolKwArgs = protocolKwArgs self.width = width self.height = height def serviceStarted(self): self.__channel = SessionChannel( self.protocolFactory, self.protocolArgs, self.protocolKwArgs, self.width, self.height, ) self.openChannel(self.__channel) def write(self, data): return self.__channel.write(data) class TestAuth(userauth.SSHUserAuthClient): def __init__(self, username, password, *a, **kw): userauth.SSHUserAuthClient.__init__(self, username, *a, **kw) self.password = password def getPassword(self): return defer.succeed(self.password) class TestTransport(transport.SSHClientTransport): def __init__( self, protocolFactory, protocolArgs, protocolKwArgs, username, password, width, height, *a, **kw, ): self.protocolFactory = protocolFactory self.protocolArgs = protocolArgs self.protocolKwArgs = protocolKwArgs self.username = username self.password = password self.width = width self.height = height def verifyHostKey(self, hostKey, fingerprint): return defer.succeed(True) def connectionSecure(self): self.__connection = TestConnection( self.protocolFactory, self.protocolArgs, self.protocolKwArgs, self.width, self.height, ) self.requestService( TestAuth(self.username, self.password, self.__connection) ) def write(self, data): return self.__connection.write(data) class TestSessionTransport(TerminalSessionTransport): def protocolFactory(self): return self.avatar.conn.transport.factory.serverProtocol() class TestSession(TerminalSession): transportFactory = TestSessionTransport class TestUser(TerminalUser): pass components.registerAdapter(TestSession, TestUser, session.ISession) class NotifyingExpectableBuffer(helper.ExpectableBuffer): def __init__(self): self.onConnection = defer.Deferred() self.onDisconnection = defer.Deferred() def connectionMade(self): helper.ExpectableBuffer.connectionMade(self) self.onConnection.callback(self) def connectionLost(self, reason): self.onDisconnection.errback(reason) class _BaseMixin: WIDTH = 80 HEIGHT = 24 def _assertBuffer(self, lines): receivedLines = self.recvlineClient.__bytes__().splitlines() expectedLines = lines + ([b""] * (self.HEIGHT - len(lines) - 1)) self.assertEqual(len(receivedLines), len(expectedLines)) for i in range(len(receivedLines)): self.assertEqual( receivedLines[i], expectedLines[i], b"".join(receivedLines[max(0, i - 1) : i + 1]) + b" != " + b"".join(expectedLines[max(0, i - 1) : i + 1]), ) def _trivialTest(self, inputLine, output): done = self.recvlineClient.expect(b"done") self._testwrite(inputLine) def finished(ign): self._assertBuffer(output) return done.addCallback(finished) class _SSHMixin(_BaseMixin): def setUp(self): if not ssh: raise SkipTest( "cryptography requirements missing, can't run historic " "recvline tests over ssh" ) u, p = b"testuser", b"testpass" rlm = TerminalRealm() rlm.userFactory = TestUser rlm.chainedProtocolFactory = lambda: insultsServer checker = checkers.InMemoryUsernamePasswordDatabaseDontUse() checker.addUser(u, p) ptl = portal.Portal(rlm) ptl.registerChecker(checker) sshFactory = ConchFactory(ptl) sshKey = keys._getPersistentRSAKey( filepath.FilePath(self.mktemp()), keySize=512 ) sshFactory.publicKeys[b"ssh-rsa"] = sshKey sshFactory.privateKeys[b"ssh-rsa"] = sshKey sshFactory.serverProtocol = self.serverProtocol sshFactory.startFactory() recvlineServer = self.serverProtocol() insultsServer = insults.ServerProtocol(lambda: recvlineServer) sshServer = sshFactory.buildProtocol(None) clientTransport = LoopbackRelay(sshServer) recvlineClient = NotifyingExpectableBuffer() insultsClient = insults.ClientProtocol(lambda: recvlineClient) sshClient = TestTransport( lambda: insultsClient, (), {}, u, p, self.WIDTH, self.HEIGHT ) serverTransport = LoopbackRelay(sshClient) sshClient.makeConnection(clientTransport) sshServer.makeConnection(serverTransport) self.recvlineClient = recvlineClient self.sshClient = sshClient self.sshServer = sshServer self.clientTransport = clientTransport self.serverTransport = serverTransport return recvlineClient.onConnection def _testwrite(self, data): self.sshClient.write(data) from twisted.conch.test import test_telnet class TestInsultsClientProtocol(insults.ClientProtocol, test_telnet.TestProtocol): pass class TestInsultsServerProtocol(insults.ServerProtocol, test_telnet.TestProtocol): pass class _TelnetMixin(_BaseMixin): def setUp(self): recvlineServer = self.serverProtocol() insultsServer = TestInsultsServerProtocol(lambda: recvlineServer) telnetServer = telnet.TelnetTransport(lambda: insultsServer) clientTransport = LoopbackRelay(telnetServer) recvlineClient = NotifyingExpectableBuffer() insultsClient = TestInsultsClientProtocol(lambda: recvlineClient) telnetClient = telnet.TelnetTransport(lambda: insultsClient) serverTransport = LoopbackRelay(telnetClient) telnetClient.makeConnection(clientTransport) telnetServer.makeConnection(serverTransport) serverTransport.clearBuffer() clientTransport.clearBuffer() self.recvlineClient = recvlineClient self.telnetClient = telnetClient self.clientTransport = clientTransport self.serverTransport = serverTransport return recvlineClient.onConnection def _testwrite(self, data): self.telnetClient.write(data) class _StdioMixin(_BaseMixin): def setUp(self): # A memory-only terminal emulator, into which the server will # write things and make other state changes. What ends up # here is basically what a user would have seen on their # screen. testTerminal = NotifyingExpectableBuffer() # An insults client protocol which will translate bytes # received from the child process into keystroke commands for # an ITerminalProtocol. insultsClient = insults.ClientProtocol(lambda: testTerminal) # A process protocol which will translate stdout and stderr # received from the child process to dataReceived calls and # error reporting on an insults client protocol. processClient = stdio.TerminalProcessProtocol(insultsClient) # Run twisted/conch/stdio.py with the name of a class # implementing ITerminalProtocol. This class will be used to # handle bytes we send to the child process. exe = sys.executable module = stdio.__file__ if module.endswith(".pyc") or module.endswith(".pyo"): module = module[:-1] args = [exe, module, reflect.qual(self.serverProtocol)] from twisted.internet import reactor clientTransport = reactor.spawnProcess( processClient, exe, args, env=properEnv, usePTY=True ) self.recvlineClient = self.testTerminal = testTerminal self.processClient = processClient self.clientTransport = clientTransport # Wait for the process protocol and test terminal to become # connected before proceeding. The former should always # happen first, but it doesn't hurt to be safe. return defer.gatherResults( filter(None, [processClient.onConnection, testTerminal.expect(b">>> ")]) ) def tearDown(self): # Kill the child process. We're done with it. try: self.clientTransport.signalProcess("KILL") except (error.ProcessExitedAlready, OSError): pass def trap(failure): failure.trap(error.ProcessTerminated) self.assertIsNone(failure.value.exitCode) self.assertEqual(failure.value.status, 9) return self.testTerminal.onDisconnection.addErrback(trap) def _testwrite(self, data): self.clientTransport.write(data) class RecvlineLoopbackMixin: serverProtocol = EchoServer def testSimple(self): return self._trivialTest( b"first line\ndone", [b">>> first line", b"first line", b">>> done"] ) def testLeftArrow(self): return self._trivialTest( insert + b"first line" + left * 4 + b"xxxx\ndone", [b">>> first xxxx", b"first xxxx", b">>> done"], ) def testRightArrow(self): return self._trivialTest( insert + b"right line" + left * 4 + right * 2 + b"xx\ndone", [b">>> right lixx", b"right lixx", b">>> done"], ) def testBackspace(self): return self._trivialTest( b"second line" + backspace * 4 + b"xxxx\ndone", [b">>> second xxxx", b"second xxxx", b">>> done"], ) def testDelete(self): return self._trivialTest( b"delete xxxx" + left * 4 + delete * 4 + b"line\ndone", [b">>> delete line", b"delete line", b">>> done"], ) def testInsert(self): return self._trivialTest( b"third ine" + left * 3 + b"l\ndone", [b">>> third line", b"third line", b">>> done"], ) def testTypeover(self): return self._trivialTest( b"fourth xine" + left * 4 + insert + b"l\ndone", [b">>> fourth line", b"fourth line", b">>> done"], ) def testHome(self): return self._trivialTest( insert + b"blah line" + home + b"home\ndone", [b">>> home line", b"home line", b">>> done"], ) def testEnd(self): return self._trivialTest( b"end " + left * 4 + end + b"line\ndone", [b">>> end line", b"end line", b">>> done"], ) class RecvlineLoopbackTelnetTests(_TelnetMixin, TestCase, RecvlineLoopbackMixin): pass class RecvlineLoopbackSSHTests(_SSHMixin, TestCase, RecvlineLoopbackMixin): pass @skipIf(not stdio, "Terminal requirements missing, can't run recvline tests over stdio") class RecvlineLoopbackStdioTests(_StdioMixin, TestCase, RecvlineLoopbackMixin): pass class HistoricRecvlineLoopbackMixin: serverProtocol = EchoServer def testUpArrow(self): return self._trivialTest( b"first line\n" + up + b"\ndone", [ b">>> first line", b"first line", b">>> first line", b"first line", b">>> done", ], ) def test_DownArrowToPartialLineInHistory(self): """ Pressing down arrow to visit an entry that was added to the history by pressing the up arrow instead of return does not raise a L{TypeError}. @see: U{http://twistedmatrix.com/trac/ticket/9031} @return: A L{defer.Deferred} that fires when C{b"done"} is echoed back. """ return self._trivialTest( b"first line\n" + b"partial line" + up + down + b"\ndone", [ b">>> first line", b"first line", b">>> partial line", b"partial line", b">>> done", ], ) def testDownArrow(self): return self._trivialTest( b"first line\nsecond line\n" + up * 2 + down + b"\ndone", [ b">>> first line", b"first line", b">>> second line", b"second line", b">>> second line", b"second line", b">>> done", ], ) class HistoricRecvlineLoopbackTelnetTests( _TelnetMixin, TestCase, HistoricRecvlineLoopbackMixin ): pass class HistoricRecvlineLoopbackSSHTests( _SSHMixin, TestCase, HistoricRecvlineLoopbackMixin ): pass @skipIf( not stdio, "Terminal requirements missing, " "can't run historic recvline tests over stdio", ) class HistoricRecvlineLoopbackStdioTests( _StdioMixin, TestCase, HistoricRecvlineLoopbackMixin ): pass class TransportSequenceTests(TestCase): """ L{twisted.conch.recvline.TransportSequence} """ def test_invalidSequence(self): """ Initializing a L{recvline.TransportSequence} with no args raises an assertion. """ self.assertRaises(AssertionError, recvline.TransportSequence) PK F��Z�i��G� G� test_cftp.pynu �[��� # -*- test-case-name: twisted.conch.test.test_cftp -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE file for details. """ Tests for L{twisted.conch.scripts.cftp}. """ import getpass import locale import operator import os import struct import sys import time from io import BytesIO from unittest import skipIf from zope.interface import implementer from twisted.conch import ls from twisted.conch.interfaces import ISFTPFile from twisted.conch.test.test_filetransfer import FileTransferTestAvatar, SFTPTestBase from twisted.cred import portal from twisted.internet import defer, error, interfaces, protocol, reactor from twisted.internet.task import Clock from twisted.internet.utils import getProcessOutputAndValue, getProcessValue from twisted.python import log from twisted.python.fakepwd import UserDatabase from twisted.python.filepath import FilePath from twisted.python.procutils import which from twisted.python.reflect import requireModule from twisted.test.proto_helpers import StringTransport from twisted.trial.unittest import TestCase pyasn1 = requireModule("pyasn1") cryptography = requireModule("cryptography") unix = requireModule("twisted.conch.unix") if cryptography and pyasn1: try: from twisted.conch.scripts import cftp from twisted.conch.scripts.cftp import SSHSession from twisted.conch.ssh import filetransfer from twisted.conch.test import test_conch, test_ssh from twisted.conch.test.test_conch import FakeStdio from twisted.conch.test.test_filetransfer import FileTransferForTestAvatar except ImportError: pass skipTests = False if None in (unix, cryptography, pyasn1, interfaces.IReactorProcess(reactor, None)): skipTests = True @skipIf(skipTests, "don't run w/o spawnProcess or cryptography or pyasn1") class SSHSessionTests(TestCase): """ Tests for L{twisted.conch.scripts.cftp.SSHSession}. """ def test_eofReceived(self): """ L{twisted.conch.scripts.cftp.SSHSession.eofReceived} loses the write half of its stdio connection. """ stdio = FakeStdio() channel = SSHSession() channel.stdio = stdio channel.eofReceived() self.assertTrue(stdio.writeConnLost) class ListingTests(TestCase): """ Tests for L{lsLine}, the function which generates an entry for a file or directory in an SFTP I{ls} command's output. """ if getattr(time, "tzset", None) is None: skip = "Cannot test timestamp formatting code without time.tzset" def setUp(self): """ Patch the L{ls} module's time function so the results of L{lsLine} are deterministic. """ self.now = 123456789 def fakeTime(): return self.now self.patch(ls, "time", fakeTime) # Make sure that the timezone ends up the same after these tests as # it was before. if "TZ" in os.environ: self.addCleanup(operator.setitem, os.environ, "TZ", os.environ["TZ"]) self.addCleanup(time.tzset) else: def cleanup(): # os.environ.pop is broken! Don't use it! Ever! Or die! try: del os.environ["TZ"] except KeyError: pass time.tzset() self.addCleanup(cleanup) def _lsInTimezone(self, timezone, stat): """ Call L{ls.lsLine} after setting the timezone to C{timezone} and return the result. """ # Set the timezone to a well-known value so the timestamps are # predictable. os.environ["TZ"] = timezone time.tzset() return ls.lsLine("foo", stat) def test_oldFile(self): """ A file with an mtime six months (approximately) or more in the past has a listing including a low-resolution timestamp. """ # Go with 7 months. That's more than 6 months. then = self.now - (60 * 60 * 24 * 31 * 7) stat = os.stat_result((0, 0, 0, 0, 0, 0, 0, 0, then, 0)) self.assertEqual( self._lsInTimezone("America/New_York", stat), "!--------- 0 0 0 0 Apr 26 1973 foo", ) self.assertEqual( self._lsInTimezone("Pacific/Auckland", stat), "!--------- 0 0 0 0 Apr 27 1973 foo", ) def test_oldSingleDigitDayOfMonth(self): """ A file with a high-resolution timestamp which falls on a day of the month which can be represented by one decimal digit is formatted with one padding 0 to preserve the columns which come after it. """ # A point about 7 months in the past, tweaked to fall on the first of a # month so we test the case we want to test. then = self.now - (60 * 60 * 24 * 31 * 7) + (60 * 60 * 24 * 5) stat = os.stat_result((0, 0, 0, 0, 0, 0, 0, 0, then, 0)) self.assertEqual( self._lsInTimezone("America/New_York", stat), "!--------- 0 0 0 0 May 01 1973 foo", ) self.assertEqual( self._lsInTimezone("Pacific/Auckland", stat), "!--------- 0 0 0 0 May 02 1973 foo", ) def test_newFile(self): """ A file with an mtime fewer than six months (approximately) in the past has a listing including a high-resolution timestamp excluding the year. """ # A point about three months in the past. then = self.now - (60 * 60 * 24 * 31 * 3) stat = os.stat_result((0, 0, 0, 0, 0, 0, 0, 0, then, 0)) self.assertEqual( self._lsInTimezone("America/New_York", stat), "!--------- 0 0 0 0 Aug 28 17:33 foo", ) self.assertEqual( self._lsInTimezone("Pacific/Auckland", stat), "!--------- 0 0 0 0 Aug 29 09:33 foo", ) # If alternate locale is not available, the next test will be # skipped, please install this locale for it to run currentLocale = locale.getlocale() try: try: locale.setlocale(locale.LC_ALL, "es_AR.UTF8") except locale.Error: localeSkip = True else: locale.setlocale(locale.LC_ALL, currentLocale) except: pass @skipIf(localeSkip, "The es_AR.UTF8 locale is not installed.") def test_localeIndependent(self): """ The month name in the date is locale independent. """ # A point about three months in the past. then = self.now - (60 * 60 * 24 * 31 * 3) stat = os.stat_result((0, 0, 0, 0, 0, 0, 0, 0, then, 0)) # Fake that we're in a language where August is not Aug (e.g.: Spanish) currentLocale = locale.getlocale() locale.setlocale(locale.LC_ALL, "es_AR.UTF8") self.addCleanup(locale.setlocale, locale.LC_ALL, currentLocale) self.assertEqual( self._lsInTimezone("America/New_York", stat), "!--------- 0 0 0 0 Aug 28 17:33 foo", ) self.assertEqual( self._lsInTimezone("Pacific/Auckland", stat), "!--------- 0 0 0 0 Aug 29 09:33 foo", ) def test_newSingleDigitDayOfMonth(self): """ A file with a high-resolution timestamp which falls on a day of the month which can be represented by one decimal digit is formatted with one padding 0 to preserve the columns which come after it. """ # A point about three months in the past, tweaked to fall on the first # of a month so we test the case we want to test. then = self.now - (60 * 60 * 24 * 31 * 3) + (60 * 60 * 24 * 4) stat = os.stat_result((0, 0, 0, 0, 0, 0, 0, 0, then, 0)) self.assertEqual( self._lsInTimezone("America/New_York", stat), "!--------- 0 0 0 0 Sep 01 17:33 foo", ) self.assertEqual( self._lsInTimezone("Pacific/Auckland", stat), "!--------- 0 0 0 0 Sep 02 09:33 foo", ) class InMemorySSHChannel(StringTransport): """ Minimal implementation of a L{SSHChannel} like class which only reads and writes data from memory. """ def __init__(self, conn): """ @param conn: The SSH connection associated with this channel. @type conn: L{SSHConnection} """ self.conn = conn self.localClosed = 0 super().__init__() class FilesystemAccessExpectations: """ A test helper used to support expected filesystem access. """ def __init__(self): self._cache = {} def put(self, path, flags, stream): """ @param path: Path at which the stream is requested. @type path: L{str} @param path: Flags with which the stream is requested. @type path: L{str} @param stream: A stream. @type stream: C{File} """ self._cache[(path, flags)] = stream def pop(self, path, flags): """ Remove a stream from the memory. @param path: Path at which the stream is requested. @type path: L{str} @param path: Flags with which the stream is requested. @type path: L{str} @return: A stream. @rtype: C{File} """ return self._cache.pop((path, flags)) class InMemorySFTPClient: """ A L{filetransfer.FileTransferClient} which does filesystem operations in memory, without touching the local disc or the network interface. @ivar _availableFiles: File like objects which are available to the SFTP client. @type _availableFiles: L{FilesystemRegister} """ def __init__(self, availableFiles): self.transport = InMemorySSHChannel(self) self.options = { "requests": 1, "buffersize": 10, } self._availableFiles = availableFiles def openFile(self, filename, flags, attrs): """ @see: L{filetransfer.FileTransferClient.openFile}. Retrieve and remove cached file based on flags. """ return self._availableFiles.pop(filename, flags) @implementer(ISFTPFile) class InMemoryRemoteFile(BytesIO): """ An L{ISFTPFile} which handles all data in memory. """ def __init__(self, name): """ @param name: Name of this file. @type name: L{str} """ self.name = name BytesIO.__init__(self) def writeChunk(self, start, data): """ @see: L{ISFTPFile.writeChunk} """ self.seek(start) self.write(data) return defer.succeed(self) def close(self): """ @see: L{ISFTPFile.writeChunk} Keeps data after file was closed to help with testing. """ self._closed = True def getAttrs(self): # ISFTPFile.getAttrs pass def readChunk(self, offset, length): # ISFTPFile.readChunk pass def setAttrs(self, attrs): # ISFTPFile.getAttrs pass def getvalue(self): """ Get current data of file. Allow reading data event when file is closed. """ return BytesIO.getvalue(self) @skipIf(skipTests, "don't run w/o spawnProcess or cryptography or pyasn1") class StdioClientTests(TestCase): """ Tests for L{cftp.StdioClient}. """ def setUp(self): """ Create a L{cftp.StdioClient} hooked up to dummy transport and a fake user database. """ self.fakeFilesystem = FilesystemAccessExpectations() sftpClient = InMemorySFTPClient(self.fakeFilesystem) self.client = cftp.StdioClient(sftpClient) self.client.currentDirectory = "/" self.database = self.client._pwd = UserDatabase() # Use a fixed width for all tests so that we get the same results when # running these tests from different terminals. # Run tests in a wide console so that all items are delimited by at # least one space character. self.setKnownConsoleSize(500, 24) # Intentionally bypassing makeConnection - that triggers some code # which uses features not provided by our dumb Connection fake. self.client.transport = self.client.client.transport def test_exec(self): """ The I{exec} command runs its arguments locally in a child process using the user's shell. """ self.database.addUser( getpass.getuser(), "secret", os.getuid(), 1234, "foo", "bar", sys.executable ) d = self.client._dispatchCommand("exec print(1 + 2)") d.addCallback(self.assertEqual, b"3\n") return d def test_execWithoutShell(self): """ If the local user has no shell, the I{exec} command runs its arguments using I{/bin/sh}. """ self.database.addUser( getpass.getuser(), "secret", os.getuid(), 1234, "foo", "bar", "" ) d = self.client._dispatchCommand("exec echo hello") d.addCallback(self.assertEqual, b"hello\n") return d def test_bang(self): """ The I{exec} command is run for lines which start with C{"!"}. """ self.database.addUser( getpass.getuser(), "secret", os.getuid(), 1234, "foo", "bar", "/bin/sh" ) d = self.client._dispatchCommand("!echo hello") d.addCallback(self.assertEqual, b"hello\n") return d def setKnownConsoleSize(self, width, height): """ For the duration of this test, patch C{cftp}'s C{fcntl} module to return a fixed width and height. @param width: the width in characters @type width: L{int} @param height: the height in characters @type height: L{int} """ # Local import to avoid win32 issues. import tty class FakeFcntl: def ioctl(self, fd, opt, mutate): if opt != tty.TIOCGWINSZ: self.fail("Only window-size queries supported.") return struct.pack("4H", height, width, 0, 0) self.patch(cftp, "fcntl", FakeFcntl()) def test_printProgressBarReporting(self): """ L{StdioClient._printProgressBar} prints a progress description, including percent done, amount transferred, transfer rate, and time remaining, all based the given start time, the given L{FileWrapper}'s progress information and the reactor's current time. """ # Use a short, known console width because this simple test doesn't # need to test the console padding. self.setKnownConsoleSize(10, 34) clock = self.client.reactor = Clock() wrapped = BytesIO(b"x") wrapped.name = b"sample" wrapper = cftp.FileWrapper(wrapped) wrapper.size = 1024 * 10 startTime = clock.seconds() clock.advance(2.0) wrapper.total += 4096 self.client._printProgressBar(wrapper, startTime) result = b"\rb'sample' 40% 4.0kB 2.0kBps 00:03 " self.assertEqual(self.client.transport.value(), result) def test_printProgressBarNoProgress(self): """ L{StdioClient._printProgressBar} prints a progress description that indicates 0 bytes transferred if no bytes have been transferred and no time has passed. """ self.setKnownConsoleSize(10, 34) clock = self.client.reactor = Clock() wrapped = BytesIO(b"x") wrapped.name = b"sample" wrapper = cftp.FileWrapper(wrapped) startTime = clock.seconds() self.client._printProgressBar(wrapper, startTime) result = b"\rb'sample' 0% 0.0B 0.0Bps 00:00 " self.assertEqual(self.client.transport.value(), result) def test_printProgressBarEmptyFile(self): """ Print the progress for empty files. """ self.setKnownConsoleSize(10, 34) wrapped = BytesIO() wrapped.name = b"empty-file" wrapper = cftp.FileWrapper(wrapped) self.client._printProgressBar(wrapper, 0) result = b"\rb'empty-file'100% 0.0B 0.0Bps 00:00 " self.assertEqual(result, self.client.transport.value()) def test_getFilenameEmpty(self): """ Returns empty value for both filename and remaining data. """ result = self.client._getFilename(" ") self.assertEqual(("", ""), result) def test_getFilenameOnlyLocal(self): """ Returns empty value for remaining data when line contains only a filename. """ result = self.client._getFilename("only-local") self.assertEqual(("only-local", ""), result) def test_getFilenameNotQuoted(self): """ Returns filename and remaining data striped of leading and trailing spaces. """ result = self.client._getFilename(" local remote file ") self.assertEqual(("local", "remote file"), result) def test_getFilenameQuoted(self): """ Returns filename and remaining data not striped of leading and trailing spaces when quoted paths are requested. """ result = self.client._getFilename(' " local file " " remote file " ') self.assertEqual((" local file ", '" remote file "'), result) def makeFile(self, path=None, content=b""): """ Create a local file and return its path. When `path` is L{None}, it will create a new temporary file. @param path: Optional path for the new file. @type path: L{str} @param content: Content to be written in the new file. @type content: L{bytes} @return: Path to the newly create file. """ if path is None: path = self.mktemp() with open(path, "wb") as file: file.write(content) return path def checkPutMessage(self, transfers, randomOrder=False): """ Check output of cftp client for a put request. @param transfers: List with tuple of (local, remote, progress). @param randomOrder: When set to C{True}, it will ignore the order in which put reposes are received """ output = self.client.transport.value() output = output.decode("utf-8") output = output.split("\n\r") expectedOutput = [] actualOutput = [] for local, remote, expected in transfers: # For each transfer we have a list of reported progress which # ends with the final message informing that file was transferred. expectedTransfer = [] for line in expected: expectedTransfer.append(f"{local} {line}") expectedTransfer.append(f"Transferred {local} to {remote}") expectedOutput.append(expectedTransfer) progressParts = output.pop(0).strip("\r").split("\r") actual = progressParts[:-1] last = progressParts[-1].strip("\n").split("\n") actual.extend(last) actualTransfer = [] # Each transferred file is on a line with summary on the last # line. Summary is copying at the end. for line in actual[:-1]: # Output line is in the format # NAME PROGRESS_PERCENTAGE PROGRESS_BYTES SPEED ETA. # For testing we only care about the # PROGRESS_PERCENTAGE and PROGRESS values. # Ignore SPPED and ETA. line = line.strip().rsplit(" ", 2)[0] # NAME can be followed by a lot of spaces so we need to # reduce them to single space. line = line.strip().split(" ", 1) actualTransfer.append(f"{line[0]} {line[1].strip()}") actualTransfer.append(actual[-1]) actualOutput.append(actualTransfer) if randomOrder: self.assertEqual(sorted(expectedOutput), sorted(actualOutput)) else: self.assertEqual(expectedOutput, actualOutput) self.assertEqual( 0, len(output), "There are still put responses which were not checked.", ) def test_cmd_PUTSingleNoRemotePath(self): """ A name based on local path is used when remote path is not provided. The progress is updated while chunks are transferred. """ content = b"Test\r\nContent" localPath = self.makeFile(content=content) flags = filetransfer.FXF_WRITE | filetransfer.FXF_CREAT | filetransfer.FXF_TRUNC remoteName = os.path.join("/", os.path.basename(localPath)) remoteFile = InMemoryRemoteFile(remoteName) self.fakeFilesystem.put(remoteName, flags, defer.succeed(remoteFile)) self.client.client.options["buffersize"] = 10 deferred = self.client.cmd_PUT(localPath) self.successResultOf(deferred) self.assertEqual(content, remoteFile.getvalue()) self.assertTrue(remoteFile._closed) self.checkPutMessage( [(localPath, remoteName, ["76% 10.0B", "100% 13.0B", "100% 13.0B"])] ) def test_cmd_PUTSingleRemotePath(self): """ Remote path is extracted from first filename after local file. Any other data in the line is ignored. """ localPath = self.makeFile() flags = filetransfer.FXF_WRITE | filetransfer.FXF_CREAT | filetransfer.FXF_TRUNC remoteName = "/remote-path" remoteFile = InMemoryRemoteFile(remoteName) self.fakeFilesystem.put(remoteName, flags, defer.succeed(remoteFile)) deferred = self.client.cmd_PUT(f"{localPath} {remoteName} ignored") self.successResultOf(deferred) self.checkPutMessage([(localPath, remoteName, ["100% 0.0B"])]) self.assertTrue(remoteFile._closed) self.assertEqual(b"", remoteFile.getvalue()) def test_cmd_PUTMultipleNoRemotePath(self): """ When a gobbing expression is used local files are transferred with remote file names based on local names. """ first = self.makeFile() firstName = os.path.basename(first) secondName = "second-name" parent = os.path.dirname(first) second = self.makeFile(path=os.path.join(parent, secondName)) flags = filetransfer.FXF_WRITE | filetransfer.FXF_CREAT | filetransfer.FXF_TRUNC firstRemotePath = f"/{firstName}" secondRemotePath = f"/{secondName}" firstRemoteFile = InMemoryRemoteFile(firstRemotePath) secondRemoteFile = InMemoryRemoteFile(secondRemotePath) self.fakeFilesystem.put(firstRemotePath, flags, defer.succeed(firstRemoteFile)) self.fakeFilesystem.put( secondRemotePath, flags, defer.succeed(secondRemoteFile) ) deferred = self.client.cmd_PUT(os.path.join(parent, "*")) self.successResultOf(deferred) self.assertTrue(firstRemoteFile._closed) self.assertEqual(b"", firstRemoteFile.getvalue()) self.assertTrue(secondRemoteFile._closed) self.assertEqual(b"", secondRemoteFile.getvalue()) self.checkPutMessage( [ (first, firstRemotePath, ["100% 0.0B"]), (second, secondRemotePath, ["100% 0.0B"]), ], randomOrder=True, ) def test_cmd_PUTMultipleWithRemotePath(self): """ When a gobbing expression is used local files are transferred with remote file names based on local names. when a remote folder is requested remote paths are composed from remote path and local filename. """ first = self.makeFile() firstName = os.path.basename(first) secondName = "second-name" parent = os.path.dirname(first) second = self.makeFile(path=os.path.join(parent, secondName)) flags = filetransfer.FXF_WRITE | filetransfer.FXF_CREAT | filetransfer.FXF_TRUNC firstRemoteFile = InMemoryRemoteFile(firstName) secondRemoteFile = InMemoryRemoteFile(secondName) firstRemotePath = f"/remote/{firstName}" secondRemotePath = f"/remote/{secondName}" self.fakeFilesystem.put(firstRemotePath, flags, defer.succeed(firstRemoteFile)) self.fakeFilesystem.put( secondRemotePath, flags, defer.succeed(secondRemoteFile) ) deferred = self.client.cmd_PUT("{} remote".format(os.path.join(parent, "*"))) self.successResultOf(deferred) self.assertTrue(firstRemoteFile._closed) self.assertEqual(b"", firstRemoteFile.getvalue()) self.assertTrue(secondRemoteFile._closed) self.assertEqual(b"", secondRemoteFile.getvalue()) self.checkPutMessage( [ (first, firstName, ["100% 0.0B"]), (second, secondName, ["100% 0.0B"]), ], randomOrder=True, ) class FileTransferTestRealm: def __init__(self, testDir): self.testDir = testDir def requestAvatar(self, avatarID, mind, *interfaces): a = FileTransferTestAvatar(self.testDir) return interfaces[0], a, lambda: None class SFTPTestProcess(protocol.ProcessProtocol): """ Protocol for testing cftp. Provides an interface between Python (where all the tests are) and the cftp client process (which does the work that is being tested). """ def __init__(self, onOutReceived): """ @param onOutReceived: A L{Deferred} to be fired as soon as data is received from stdout. """ self.clearBuffer() self.onOutReceived = onOutReceived self.onProcessEnd = None self._expectingCommand = None self._processEnded = False def clearBuffer(self): """ Clear any buffered data received from stdout. Should be private. """ self.buffer = b"" self._linesReceived = [] self._lineBuffer = b"" def outReceived(self, data): """ Called by Twisted when the cftp client prints data to stdout. """ log.msg("got %r" % data) lines = (self._lineBuffer + data).split(b"\n") self._lineBuffer = lines.pop(-1) self._linesReceived.extend(lines) # XXX - not strictly correct. # We really want onOutReceived to fire after the first 'cftp>' prompt # has been received. (See use in OurServerCmdLineClientTests.setUp) if self.onOutReceived is not None: d, self.onOutReceived = self.onOutReceived, None d.callback(data) self.buffer += data self._checkForCommand() def _checkForCommand(self): prompt = b"cftp> " if self._expectingCommand and self._lineBuffer == prompt: buf = b"\n".join(self._linesReceived) if buf.startswith(prompt): buf = buf[len(prompt) :] self.clearBuffer() d, self._expectingCommand = self._expectingCommand, None d.callback(buf) def errReceived(self, data): """ Called by Twisted when the cftp client prints data to stderr. """ log.msg("err: %s" % data) def getBuffer(self): """ Return the contents of the buffer of data received from stdout. """ return self.buffer def runCommand(self, command): """ Issue the given command via the cftp client. Return a C{Deferred} that fires when the server returns a result. Note that the C{Deferred} will callback even if the server returns some kind of error. @param command: A string containing an sftp command. @return: A C{Deferred} that fires when the sftp server returns a result. The payload is the server's response string. """ self._expectingCommand = defer.Deferred() self.clearBuffer() if isinstance(command, str): command = command.encode("utf-8") self.transport.write(command + b"\n") return self._expectingCommand def runScript(self, commands): """ Run each command in sequence and return a Deferred that fires when all commands are completed. @param commands: A list of strings containing sftp commands. @return: A C{Deferred} that fires when all commands are completed. The payload is a list of response strings from the server, in the same order as the commands. """ sem = defer.DeferredSemaphore(1) dl = [sem.run(self.runCommand, command) for command in commands] return defer.gatherResults(dl) def killProcess(self): """ Kill the process if it is still running. If the process is still running, sends a KILL signal to the transport and returns a C{Deferred} which fires when L{processEnded} is called. @return: a C{Deferred}. """ if self._processEnded: return defer.succeed(None) self.onProcessEnd = defer.Deferred() self.transport.signalProcess("KILL") return self.onProcessEnd def processEnded(self, reason): """ Called by Twisted when the cftp client process ends. """ self._processEnded = True if self.onProcessEnd: d, self.onProcessEnd = self.onProcessEnd, None d.callback(None) class CFTPClientTestBase(SFTPTestBase): def setUp(self): with open("dsa_test.pub", "wb") as f: f.write(test_ssh.publicDSA_openssh) with open("dsa_test", "wb") as f: f.write(test_ssh.privateDSA_openssh) os.chmod("dsa_test", 33152) with open("kh_test", "wb") as f: f.write(b"127.0.0.1 " + test_ssh.publicRSA_openssh) return SFTPTestBase.setUp(self) def startServer(self): realm = FileTransferTestRealm(self.testDir) p = portal.Portal(realm) p.registerChecker(test_ssh.conchTestPublicKeyChecker()) fac = test_ssh.ConchTestServerFactory() fac.portal = p self.server = reactor.listenTCP(0, fac, interface="127.0.0.1") def stopServer(self): if not hasattr(self.server.factory, "proto"): return self._cbStopServer(None) self.server.factory.proto.expectedLoseConnection = 1 d = defer.maybeDeferred(self.server.factory.proto.transport.loseConnection) d.addCallback(self._cbStopServer) return d def _cbStopServer(self, ignored): return defer.maybeDeferred(self.server.stopListening) def tearDown(self): for f in ["dsa_test.pub", "dsa_test", "kh_test"]: try: os.remove(f) except BaseException: pass return SFTPTestBase.tearDown(self) @skipIf(skipTests, "don't run w/o spawnProcess or cryptography or pyasn1") class OurServerCmdLineClientTests(CFTPClientTestBase): """ Functional tests which launch a SFTP server over TCP on localhost and check cftp command line interface using a spawned process. Due to the spawned process you can not add a debugger breakpoint for the client code. """ def setUp(self): CFTPClientTestBase.setUp(self) self.startServer() cmds = ( "-p %i -l testuser " "--known-hosts kh_test " "--user-authentications publickey " "--host-key-algorithms ssh-rsa " "-i dsa_test " "-a " "-v " "127.0.0.1" ) port = self.server.getHost().port cmds = test_conch._makeArgs((cmds % port).split(), mod="cftp") log.msg(f"running {sys.executable} {cmds}") d = defer.Deferred() self.processProtocol = SFTPTestProcess(d) d.addCallback(lambda _: self.processProtocol.clearBuffer()) env = os.environ.copy() env["PYTHONPATH"] = os.pathsep.join(sys.path) encodedCmds = [] encodedEnv = {} for cmd in cmds: if isinstance(cmd, str): cmd = cmd.encode("utf-8") encodedCmds.append(cmd) for var in env: val = env[var] if isinstance(var, str): var = var.encode("utf-8") if isinstance(val, str): val = val.encode("utf-8") encodedEnv[var] = val log.msg(encodedCmds) log.msg(encodedEnv) reactor.spawnProcess( self.processProtocol, sys.executable, encodedCmds, env=encodedEnv ) return d def tearDown(self): d = self.stopServer() d.addCallback(lambda _: self.processProtocol.killProcess()) return d def _killProcess(self, ignored): try: self.processProtocol.transport.signalProcess("KILL") except error.ProcessExitedAlready: pass def runCommand(self, command): """ Run the given command with the cftp client. Return a C{Deferred} that fires when the command is complete. Payload is the server's output for that command. """ return self.processProtocol.runCommand(command) def runScript(self, *commands): """ Run the given commands with the cftp client. Returns a C{Deferred} that fires when the commands are all complete. The C{Deferred}'s payload is a list of output for each command. """ return self.processProtocol.runScript(commands) def testCdPwd(self): """ Test that 'pwd' reports the current remote directory, that 'lpwd' reports the current local directory, and that changing to a subdirectory then changing to its parent leaves you in the original remote directory. """ # XXX - not actually a unit test, see docstring. homeDir = self.testDir d = self.runScript("pwd", "lpwd", "cd testDirectory", "cd ..", "pwd") def cmdOutput(output): """ Callback function for handling command output. """ cmds = [] for cmd in output: if isinstance(cmd, bytes): cmd = cmd.decode("utf-8") cmds.append(cmd) return cmds[:3] + cmds[4:] d.addCallback(cmdOutput) d.addCallback(self.assertEqual, [homeDir.path, os.getcwd(), "", homeDir.path]) return d def testChAttrs(self): """ Check that 'ls -l' output includes the access permissions and that this output changes appropriately with 'chmod'. """ def _check(results): self.flushLoggedErrors() self.assertTrue(results[0].startswith(b"-rw-r--r--")) self.assertEqual(results[1], b"") self.assertTrue(results[2].startswith(b"----------"), results[2]) self.assertEqual(results[3], b"") d = self.runScript( "ls -l testfile1", "chmod 0 testfile1", "ls -l testfile1", "chmod 644 testfile1", ) return d.addCallback(_check) # XXX test chgrp/own def testList(self): """ Check 'ls' works as expected. Checks for wildcards, hidden files, listing directories and listing empty directories. """ def _check(results): self.assertEqual( results[0], [b"testDirectory", b"testRemoveFile", b"testRenameFile", b"testfile1"], ) self.assertEqual( results[1], [b"testDirectory", b"testRemoveFile", b"testRenameFile", b"testfile1"], ) self.assertEqual(results[2], [b"testRemoveFile", b"testRenameFile"]) self.assertEqual( results[3], [b".testHiddenFile", b"testRemoveFile", b"testRenameFile"] ) self.assertEqual(results[4], [b""]) d = self.runScript( "ls", "ls ../" + self.testDir.basename(), "ls *File", "ls -a *File", "ls -l testDirectory", ) d.addCallback(lambda xs: [x.split(b"\n") for x in xs]) return d.addCallback(_check) def testHelp(self): """ Check that running the '?' command returns help. """ d = self.runCommand("?") helpText = cftp.StdioClient(None).cmd_HELP("").strip() if isinstance(helpText, str): helpText = helpText.encode("utf-8") d.addCallback(self.assertEqual, helpText) return d def assertFilesEqual(self, name1, name2, msg=None): """ Assert that the files at C{name1} and C{name2} contain exactly the same data. """ self.assertEqual(name1.getContent(), name2.getContent(), msg) def testGet(self): """ Test that 'get' saves the remote file to the correct local location, that the output of 'get' is correct and that 'rm' actually removes the file. """ # XXX - not actually a unit test expectedOutput = "Transferred {}/testfile1 to {}/test file2".format( self.testDir.path, self.testDir.path, ) if isinstance(expectedOutput, str): expectedOutput = expectedOutput.encode("utf-8") def _checkGet(result): self.assertTrue(result.endswith(expectedOutput)) self.assertFilesEqual( self.testDir.child("testfile1"), self.testDir.child("test file2"), "get failed", ) return self.runCommand('rm "test file2"') d = self.runCommand(f'get testfile1 "{self.testDir.path}/test file2"') d.addCallback(_checkGet) d.addCallback( lambda _: self.assertFalse(self.testDir.child("test file2").exists()) ) return d def testWildcardGet(self): """ Test that 'get' works correctly when given wildcard parameters. """ def _check(ignored): self.assertFilesEqual( self.testDir.child("testRemoveFile"), FilePath("testRemoveFile"), "testRemoveFile get failed", ) self.assertFilesEqual( self.testDir.child("testRenameFile"), FilePath("testRenameFile"), "testRenameFile get failed", ) d = self.runCommand("get testR*") return d.addCallback(_check) def testPut(self): """ Check that 'put' uploads files correctly and that they can be successfully removed. Also check the output of the put command. """ # XXX - not actually a unit test expectedOutput = ( b"Transferred " + self.testDir.asBytesMode().path + b"/testfile1 to " + self.testDir.asBytesMode().path + b'/test"file2' ) def _checkPut(result): self.assertFilesEqual( self.testDir.child("testfile1"), self.testDir.child('test"file2') ) self.assertTrue(result.endswith(expectedOutput)) return self.runCommand('rm "test\\"file2"') d = self.runCommand(f'put {self.testDir.path}/testfile1 "test\\"file2"') d.addCallback(_checkPut) d.addCallback( lambda _: self.assertFalse(self.testDir.child('test"file2').exists()) ) return d def test_putOverLongerFile(self): """ Check that 'put' uploads files correctly when overwriting a longer file. """ # XXX - not actually a unit test with self.testDir.child("shorterFile").open(mode="w") as f: f.write(b"a") with self.testDir.child("longerFile").open(mode="w") as f: f.write(b"bb") def _checkPut(result): self.assertFilesEqual( self.testDir.child("shorterFile"), self.testDir.child("longerFile") ) d = self.runCommand(f"put {self.testDir.path}/shorterFile longerFile") d.addCallback(_checkPut) return d def test_putMultipleOverLongerFile(self): """ Check that 'put' uploads files correctly when overwriting a longer file and you use a wildcard to specify the files to upload. """ # XXX - not actually a unit test someDir = self.testDir.child("dir") someDir.createDirectory() with someDir.child("file").open(mode="w") as f: f.write(b"a") with self.testDir.child("file").open(mode="w") as f: f.write(b"bb") def _checkPut(result): self.assertFilesEqual(someDir.child("file"), self.testDir.child("file")) d = self.runCommand(f"put {self.testDir.path}/dir/*") d.addCallback(_checkPut) return d def testWildcardPut(self): """ What happens if you issue a 'put' command and include a wildcard (i.e. '*') in parameter? Check that all files matching the wildcard are uploaded to the correct directory. """ def check(results): self.assertEqual(results[0], b"") self.assertEqual(results[2], b"") self.assertFilesEqual( self.testDir.child("testRemoveFile"), self.testDir.parent().child("testRemoveFile"), "testRemoveFile get failed", ) self.assertFilesEqual( self.testDir.child("testRenameFile"), self.testDir.parent().child("testRenameFile"), "testRenameFile get failed", ) d = self.runScript( "cd ..", f"put {self.testDir.path}/testR*", "cd %s" % self.testDir.basename(), ) d.addCallback(check) return d def testLink(self): """ Test that 'ln' creates a file which appears as a link in the output of 'ls'. Check that removing the new file succeeds without output. """ def _check(results): self.flushLoggedErrors() self.assertEqual(results[0], b"") self.assertTrue(results[1].startswith(b"l"), "link failed") return self.runCommand("rm testLink") d = self.runScript("ln testLink testfile1", "ls -l testLink") d.addCallback(_check) d.addCallback(self.assertEqual, b"") return d def testRemoteDirectory(self): """ Test that we can create and remove directories with the cftp client. """ def _check(results): self.assertEqual(results[0], b"") self.assertTrue(results[1].startswith(b"d")) return self.runCommand("rmdir testMakeDirectory") d = self.runScript("mkdir testMakeDirectory", "ls -l testMakeDirector?") d.addCallback(_check) d.addCallback(self.assertEqual, b"") return d def test_existingRemoteDirectory(self): """ Test that a C{mkdir} on an existing directory fails with the appropriate error, and doesn't log an useless error server side. """ def _check(results): self.assertEqual(results[0], b"") self.assertEqual(results[1], b"remote error 11: mkdir failed") d = self.runScript("mkdir testMakeDirectory", "mkdir testMakeDirectory") d.addCallback(_check) return d def testLocalDirectory(self): """ Test that we can create a directory locally and remove it with the cftp client. This test works because the 'remote' server is running out of a local directory. """ d = self.runCommand(f"lmkdir {self.testDir.path}/testLocalDirectory") d.addCallback(self.assertEqual, b"") d.addCallback(lambda _: self.runCommand("rmdir testLocalDirectory")) d.addCallback(self.assertEqual, b"") return d def testRename(self): """ Test that we can rename a file. """ def _check(results): self.assertEqual(results[0], b"") self.assertEqual(results[1], b"testfile2") return self.runCommand("rename testfile2 testfile1") d = self.runScript("rename testfile1 testfile2", "ls testfile?") d.addCallback(_check) d.addCallback(self.assertEqual, b"") return d @skipIf(skipTests, "don't run w/o spawnProcess or cryptography or pyasn1") class OurServerBatchFileTests(CFTPClientTestBase): """ Functional tests which launch a SFTP server over localhost and checks csftp in batch interface. """ def setUp(self): CFTPClientTestBase.setUp(self) self.startServer() def tearDown(self): CFTPClientTestBase.tearDown(self) return self.stopServer() def _getBatchOutput(self, f): fn = self.mktemp() with open(fn, "w") as fp: fp.write(f) port = self.server.getHost().port cmds = ( "-p %i -l testuser " "--known-hosts kh_test " "--user-authentications publickey " "--host-key-algorithms ssh-rsa " "-i dsa_test " "-a " "-v -b %s 127.0.0.1" ) % (port, fn) cmds = test_conch._makeArgs(cmds.split(), mod="cftp")[1:] log.msg(f"running {sys.executable} {cmds}") env = os.environ.copy() env["PYTHONPATH"] = os.pathsep.join(sys.path) self.server.factory.expectedLoseConnection = 1 d = getProcessOutputAndValue(sys.executable, cmds, env=env) def _cleanup(res): os.remove(fn) return res d.addCallback(lambda res: res[0]) d.addBoth(_cleanup) return d def testBatchFile(self): """ Test whether batch file function of cftp ('cftp -b batchfile'). This works by treating the file as a list of commands to be run. """ cmds = """pwd ls exit """ def _cbCheckResult(res): res = res.split(b"\n") log.msg("RES %s" % repr(res)) self.assertIn(self.testDir.asBytesMode().path, res[1]) self.assertEqual( res[3:-2], [b"testDirectory", b"testRemoveFile", b"testRenameFile", b"testfile1"], ) d = self._getBatchOutput(cmds) d.addCallback(_cbCheckResult) return d def testError(self): """ Test that an error in the batch file stops running the batch. """ cmds = """chown 0 missingFile pwd exit """ def _cbCheckResult(res): self.assertNotIn(self.testDir.asBytesMode().path, res) d = self._getBatchOutput(cmds) d.addCallback(_cbCheckResult) return d def testIgnoredError(self): """ Test that a minus sign '-' at the front of a line ignores any errors. """ cmds = """-chown 0 missingFile pwd exit """ def _cbCheckResult(res): self.assertIn(self.testDir.asBytesMode().path, res) d = self._getBatchOutput(cmds) d.addCallback(_cbCheckResult) return d @skipIf(skipTests, "don't run w/o spawnProcess or cryptography or pyasn1") @skipIf(not which("sftp"), "no sftp command-line client available") class OurServerSftpClientTests(CFTPClientTestBase): """ Test the sftp server against sftp command line client. """ def setUp(self): CFTPClientTestBase.setUp(self) return self.startServer() def tearDown(self): return self.stopServer() def test_extendedAttributes(self): """ Test the return of extended attributes by the server: the sftp client should ignore them, but still be able to parse the response correctly. This test is mainly here to check that L{filetransfer.FILEXFER_ATTR_EXTENDED} has the correct value. """ fn = self.mktemp() with open(fn, "w") as f: f.write("ls .\nexit") port = self.server.getHost().port oldGetAttr = FileTransferForTestAvatar._getAttrs def _getAttrs(self, s): attrs = oldGetAttr(self, s) attrs["ext_foo"] = "bar" return attrs self.patch(FileTransferForTestAvatar, "_getAttrs", _getAttrs) self.server.factory.expectedLoseConnection = True # PubkeyAcceptedKeyTypes does not exist prior to OpenSSH 7.0 so we # first need to check if we can set it. If we can, -V will just print # the version without doing anything else; if we can't, we will get a # configuration error. d = getProcessValue("ssh", ("-o", "PubkeyAcceptedKeyTypes=ssh-dss", "-V")) def hasPAKT(status): if status == 0: args = ("-o", "PubkeyAcceptedKeyTypes=ssh-dss") else: args = () # Pass -F /dev/null to avoid the user's configuration file from # being loaded, as it may contain settings that cause our tests to # fail or hang. args += ( "-F", "/dev/null", "-o", "IdentityFile=dsa_test", "-o", "UserKnownHostsFile=kh_test", "-o", "HostKeyAlgorithms=ssh-rsa", "-o", "Port=%i" % (port,), "-b", fn, "testuser@127.0.0.1", ) return args def check(result): self.assertEqual(result[2], 0) for i in [ b"testDirectory", b"testRemoveFile", b"testRenameFile", b"testfile1", ]: self.assertIn(i, result[0]) d.addCallback(hasPAKT) d.addCallback(lambda args: getProcessOutputAndValue("sftp", args)) return d.addCallback(check) PK F��Z����AN AN test_helper.pynu �[��� # -*- test-case-name: twisted.conch.test.test_helper -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. from twisted.conch.insults import helper from twisted.conch.insults.insults import ( BLINK, BOLD, G0, G1, G2, G3, NORMAL, REVERSE_VIDEO, UNDERLINE, modes, privateModes, ) from twisted.trial import unittest WIDTH = 80 HEIGHT = 24 class BufferTests(unittest.TestCase): def setUp(self): self.term = helper.TerminalBuffer() self.term.connectionMade() def testInitialState(self): self.assertEqual(self.term.width, WIDTH) self.assertEqual(self.term.height, HEIGHT) self.assertEqual(self.term.__bytes__(), b"\n" * (HEIGHT - 1)) self.assertEqual(self.term.reportCursorPosition(), (0, 0)) def test_initialPrivateModes(self): """ Verify that only DEC Auto Wrap Mode (DECAWM) and DEC Text Cursor Enable Mode (DECTCEM) are initially in the Set Mode (SM) state. """ self.assertEqual( {privateModes.AUTO_WRAP: True, privateModes.CURSOR_MODE: True}, self.term.privateModes, ) def test_carriageReturn(self): """ C{"\r"} moves the cursor to the first column in the current row. """ self.term.cursorForward(5) self.term.cursorDown(3) self.assertEqual(self.term.reportCursorPosition(), (5, 3)) self.term.insertAtCursor(b"\r") self.assertEqual(self.term.reportCursorPosition(), (0, 3)) def test_linefeed(self): """ C{"\n"} moves the cursor to the next row without changing the column. """ self.term.cursorForward(5) self.assertEqual(self.term.reportCursorPosition(), (5, 0)) self.term.insertAtCursor(b"\n") self.assertEqual(self.term.reportCursorPosition(), (5, 1)) def test_newline(self): """ C{write} transforms C{"\n"} into C{"\r\n"}. """ self.term.cursorForward(5) self.term.cursorDown(3) self.assertEqual(self.term.reportCursorPosition(), (5, 3)) self.term.write(b"\n") self.assertEqual(self.term.reportCursorPosition(), (0, 4)) def test_setPrivateModes(self): """ Verify that L{helper.TerminalBuffer.setPrivateModes} changes the Set Mode (SM) state to "set" for the private modes it is passed. """ expected = self.term.privateModes.copy() self.term.setPrivateModes([privateModes.SCROLL, privateModes.SCREEN]) expected[privateModes.SCROLL] = True expected[privateModes.SCREEN] = True self.assertEqual(expected, self.term.privateModes) def test_resetPrivateModes(self): """ Verify that L{helper.TerminalBuffer.resetPrivateModes} changes the Set Mode (SM) state to "reset" for the private modes it is passed. """ expected = self.term.privateModes.copy() self.term.resetPrivateModes([privateModes.AUTO_WRAP, privateModes.CURSOR_MODE]) del expected[privateModes.AUTO_WRAP] del expected[privateModes.CURSOR_MODE] self.assertEqual(expected, self.term.privateModes) def testCursorDown(self): self.term.cursorDown(3) self.assertEqual(self.term.reportCursorPosition(), (0, 3)) self.term.cursorDown() self.assertEqual(self.term.reportCursorPosition(), (0, 4)) self.term.cursorDown(HEIGHT) self.assertEqual(self.term.reportCursorPosition(), (0, HEIGHT - 1)) def testCursorUp(self): self.term.cursorUp(5) self.assertEqual(self.term.reportCursorPosition(), (0, 0)) self.term.cursorDown(20) self.term.cursorUp(1) self.assertEqual(self.term.reportCursorPosition(), (0, 19)) self.term.cursorUp(19) self.assertEqual(self.term.reportCursorPosition(), (0, 0)) def testCursorForward(self): self.term.cursorForward(2) self.assertEqual(self.term.reportCursorPosition(), (2, 0)) self.term.cursorForward(2) self.assertEqual(self.term.reportCursorPosition(), (4, 0)) self.term.cursorForward(WIDTH) self.assertEqual(self.term.reportCursorPosition(), (WIDTH, 0)) def testCursorBackward(self): self.term.cursorForward(10) self.term.cursorBackward(2) self.assertEqual(self.term.reportCursorPosition(), (8, 0)) self.term.cursorBackward(7) self.assertEqual(self.term.reportCursorPosition(), (1, 0)) self.term.cursorBackward(1) self.assertEqual(self.term.reportCursorPosition(), (0, 0)) self.term.cursorBackward(1) self.assertEqual(self.term.reportCursorPosition(), (0, 0)) def testCursorPositioning(self): self.term.cursorPosition(3, 9) self.assertEqual(self.term.reportCursorPosition(), (3, 9)) def testSimpleWriting(self): s = b"Hello, world." self.term.write(s) self.assertEqual(self.term.__bytes__(), s + b"\n" + b"\n" * (HEIGHT - 2)) def testOvertype(self): s = b"hello, world." self.term.write(s) self.term.cursorBackward(len(s)) self.term.resetModes([modes.IRM]) self.term.write(b"H") self.assertEqual( self.term.__bytes__(), (b"H" + s[1:]) + b"\n" + b"\n" * (HEIGHT - 2) ) def testInsert(self): s = b"ello, world." self.term.write(s) self.term.cursorBackward(len(s)) self.term.setModes([modes.IRM]) self.term.write(b"H") self.assertEqual( self.term.__bytes__(), (b"H" + s) + b"\n" + b"\n" * (HEIGHT - 2) ) def testWritingInTheMiddle(self): s = b"Hello, world." self.term.cursorDown(5) self.term.cursorForward(5) self.term.write(s) self.assertEqual( self.term.__bytes__(), b"\n" * 5 + (self.term.fill * 5) + s + b"\n" + b"\n" * (HEIGHT - 7), ) def testWritingWrappedAtEndOfLine(self): s = b"Hello, world." self.term.cursorForward(WIDTH - 5) self.term.write(s) self.assertEqual( self.term.__bytes__(), s[:5].rjust(WIDTH) + b"\n" + s[5:] + b"\n" + b"\n" * (HEIGHT - 3), ) def testIndex(self): self.term.index() self.assertEqual(self.term.reportCursorPosition(), (0, 1)) self.term.cursorDown(HEIGHT) self.assertEqual(self.term.reportCursorPosition(), (0, HEIGHT - 1)) self.term.index() self.assertEqual(self.term.reportCursorPosition(), (0, HEIGHT - 1)) def testReverseIndex(self): self.term.reverseIndex() self.assertEqual(self.term.reportCursorPosition(), (0, 0)) self.term.cursorDown(2) self.assertEqual(self.term.reportCursorPosition(), (0, 2)) self.term.reverseIndex() self.assertEqual(self.term.reportCursorPosition(), (0, 1)) def test_nextLine(self): """ C{nextLine} positions the cursor at the beginning of the row below the current row. """ self.term.nextLine() self.assertEqual(self.term.reportCursorPosition(), (0, 1)) self.term.cursorForward(5) self.assertEqual(self.term.reportCursorPosition(), (5, 1)) self.term.nextLine() self.assertEqual(self.term.reportCursorPosition(), (0, 2)) def testSaveCursor(self): self.term.cursorDown(5) self.term.cursorForward(7) self.assertEqual(self.term.reportCursorPosition(), (7, 5)) self.term.saveCursor() self.term.cursorDown(7) self.term.cursorBackward(3) self.assertEqual(self.term.reportCursorPosition(), (4, 12)) self.term.restoreCursor() self.assertEqual(self.term.reportCursorPosition(), (7, 5)) def testSingleShifts(self): self.term.singleShift2() self.term.write(b"Hi") ch = self.term.getCharacter(0, 0) self.assertEqual(ch[0], b"H") self.assertEqual(ch[1].charset, G2) ch = self.term.getCharacter(1, 0) self.assertEqual(ch[0], b"i") self.assertEqual(ch[1].charset, G0) self.term.singleShift3() self.term.write(b"!!") ch = self.term.getCharacter(2, 0) self.assertEqual(ch[0], b"!") self.assertEqual(ch[1].charset, G3) ch = self.term.getCharacter(3, 0) self.assertEqual(ch[0], b"!") self.assertEqual(ch[1].charset, G0) def testShifting(self): s1 = b"Hello" s2 = b"World" s3 = b"Bye!" self.term.write(b"Hello\n") self.term.shiftOut() self.term.write(b"World\n") self.term.shiftIn() self.term.write(b"Bye!\n") g = G0 h = 0 for s in (s1, s2, s3): for i in range(len(s)): ch = self.term.getCharacter(i, h) self.assertEqual(ch[0], s[i : i + 1]) self.assertEqual(ch[1].charset, g) g = g == G0 and G1 or G0 h += 1 def testGraphicRendition(self): self.term.selectGraphicRendition(BOLD, UNDERLINE, BLINK, REVERSE_VIDEO) self.term.write(b"W") self.term.selectGraphicRendition(NORMAL) self.term.write(b"X") self.term.selectGraphicRendition(BLINK) self.term.write(b"Y") self.term.selectGraphicRendition(BOLD) self.term.write(b"Z") ch = self.term.getCharacter(0, 0) self.assertEqual(ch[0], b"W") self.assertTrue(ch[1].bold) self.assertTrue(ch[1].underline) self.assertTrue(ch[1].blink) self.assertTrue(ch[1].reverseVideo) ch = self.term.getCharacter(1, 0) self.assertEqual(ch[0], b"X") self.assertFalse(ch[1].bold) self.assertFalse(ch[1].underline) self.assertFalse(ch[1].blink) self.assertFalse(ch[1].reverseVideo) ch = self.term.getCharacter(2, 0) self.assertEqual(ch[0], b"Y") self.assertTrue(ch[1].blink) self.assertFalse(ch[1].bold) self.assertFalse(ch[1].underline) self.assertFalse(ch[1].reverseVideo) ch = self.term.getCharacter(3, 0) self.assertEqual(ch[0], b"Z") self.assertTrue(ch[1].blink) self.assertTrue(ch[1].bold) self.assertFalse(ch[1].underline) self.assertFalse(ch[1].reverseVideo) def testColorAttributes(self): s1 = b"Merry xmas" s2 = b"Just kidding" self.term.selectGraphicRendition( helper.FOREGROUND + helper.RED, helper.BACKGROUND + helper.GREEN ) self.term.write(s1 + b"\n") self.term.selectGraphicRendition(NORMAL) self.term.write(s2 + b"\n") for i in range(len(s1)): ch = self.term.getCharacter(i, 0) self.assertEqual(ch[0], s1[i : i + 1]) self.assertEqual(ch[1].charset, G0) self.assertFalse(ch[1].bold) self.assertFalse(ch[1].underline) self.assertFalse(ch[1].blink) self.assertFalse(ch[1].reverseVideo) self.assertEqual(ch[1].foreground, helper.RED) self.assertEqual(ch[1].background, helper.GREEN) for i in range(len(s2)): ch = self.term.getCharacter(i, 1) self.assertEqual(ch[0], s2[i : i + 1]) self.assertEqual(ch[1].charset, G0) self.assertFalse(ch[1].bold) self.assertFalse(ch[1].underline) self.assertFalse(ch[1].blink) self.assertFalse(ch[1].reverseVideo) self.assertEqual(ch[1].foreground, helper.WHITE) self.assertEqual(ch[1].background, helper.BLACK) def testEraseLine(self): s1 = b"line 1" s2 = b"line 2" s3 = b"line 3" self.term.write(b"\n".join((s1, s2, s3)) + b"\n") self.term.cursorPosition(1, 1) self.term.eraseLine() self.assertEqual( self.term.__bytes__(), s1 + b"\n" + b"\n" + s3 + b"\n" + b"\n" * (HEIGHT - 4), ) def testEraseToLineEnd(self): s = b"Hello, world." self.term.write(s) self.term.cursorBackward(5) self.term.eraseToLineEnd() self.assertEqual(self.term.__bytes__(), s[:-5] + b"\n" + b"\n" * (HEIGHT - 2)) def testEraseToLineBeginning(self): s = b"Hello, world." self.term.write(s) self.term.cursorBackward(5) self.term.eraseToLineBeginning() self.assertEqual( self.term.__bytes__(), s[-4:].rjust(len(s)) + b"\n" + b"\n" * (HEIGHT - 2) ) def testEraseDisplay(self): self.term.write(b"Hello world\n") self.term.write(b"Goodbye world\n") self.term.eraseDisplay() self.assertEqual(self.term.__bytes__(), b"\n" * (HEIGHT - 1)) def testEraseToDisplayEnd(self): s1 = b"Hello world" s2 = b"Goodbye world" self.term.write(b"\n".join((s1, s2, b""))) self.term.cursorPosition(5, 1) self.term.eraseToDisplayEnd() self.assertEqual( self.term.__bytes__(), s1 + b"\n" + s2[:5] + b"\n" + b"\n" * (HEIGHT - 3) ) def testEraseToDisplayBeginning(self): s1 = b"Hello world" s2 = b"Goodbye world" self.term.write(b"\n".join((s1, s2))) self.term.cursorPosition(5, 1) self.term.eraseToDisplayBeginning() self.assertEqual( self.term.__bytes__(), b"\n" + s2[6:].rjust(len(s2)) + b"\n" + b"\n" * (HEIGHT - 3), ) def testLineInsertion(self): s1 = b"Hello world" s2 = b"Goodbye world" self.term.write(b"\n".join((s1, s2))) self.term.cursorPosition(7, 1) self.term.insertLine() self.assertEqual( self.term.__bytes__(), s1 + b"\n" + b"\n" + s2 + b"\n" + b"\n" * (HEIGHT - 4), ) def testLineDeletion(self): s1 = b"Hello world" s2 = b"Middle words" s3 = b"Goodbye world" self.term.write(b"\n".join((s1, s2, s3))) self.term.cursorPosition(9, 1) self.term.deleteLine() self.assertEqual( self.term.__bytes__(), s1 + b"\n" + s3 + b"\n" + b"\n" * (HEIGHT - 3) ) class FakeDelayedCall: called = False cancelled = False def __init__(self, fs, timeout, f, a, kw): self.fs = fs self.timeout = timeout self.f = f self.a = a self.kw = kw def active(self): return not (self.cancelled or self.called) def cancel(self): self.cancelled = True # self.fs.calls.remove(self) def call(self): self.called = True self.f(*self.a, **self.kw) class FakeScheduler: def __init__(self): self.calls = [] def callLater(self, timeout, f, *a, **kw): self.calls.append(FakeDelayedCall(self, timeout, f, a, kw)) return self.calls[-1] class ExpectTests(unittest.TestCase): def setUp(self): self.term = helper.ExpectableBuffer() self.term.connectionMade() self.fs = FakeScheduler() def testSimpleString(self): result = [] d = self.term.expect(b"hello world", timeout=1, scheduler=self.fs) d.addCallback(result.append) self.term.write(b"greeting puny earthlings\n") self.assertFalse(result) self.term.write(b"hello world\n") self.assertTrue(result) self.assertEqual(result[0].group(), b"hello world") self.assertEqual(len(self.fs.calls), 1) self.assertFalse(self.fs.calls[0].active()) def testBrokenUpString(self): result = [] d = self.term.expect(b"hello world") d.addCallback(result.append) self.assertFalse(result) self.term.write(b"hello ") self.assertFalse(result) self.term.write(b"worl") self.assertFalse(result) self.term.write(b"d") self.assertTrue(result) self.assertEqual(result[0].group(), b"hello world") def testMultiple(self): result = [] d1 = self.term.expect(b"hello ") d1.addCallback(result.append) d2 = self.term.expect(b"world") d2.addCallback(result.append) self.assertFalse(result) self.term.write(b"hello") self.assertFalse(result) self.term.write(b" ") self.assertEqual(len(result), 1) self.term.write(b"world") self.assertEqual(len(result), 2) self.assertEqual(result[0].group(), b"hello ") self.assertEqual(result[1].group(), b"world") def testSynchronous(self): self.term.write(b"hello world") result = [] d = self.term.expect(b"hello world") d.addCallback(result.append) self.assertTrue(result) self.assertEqual(result[0].group(), b"hello world") def testMultipleSynchronous(self): self.term.write(b"goodbye world") result = [] d1 = self.term.expect(b"bye") d1.addCallback(result.append) d2 = self.term.expect(b"world") d2.addCallback(result.append) self.assertEqual(len(result), 2) self.assertEqual(result[0].group(), b"bye") self.assertEqual(result[1].group(), b"world") def _cbTestTimeoutFailure(self, res): self.assertTrue(hasattr(res, "type")) self.assertEqual(res.type, helper.ExpectationTimeout) def testTimeoutFailure(self): d = self.term.expect(b"hello world", timeout=1, scheduler=self.fs) d.addBoth(self._cbTestTimeoutFailure) self.fs.calls[0].call() def testOverlappingTimeout(self): self.term.write(b"not zoomtastic") result = [] d1 = self.term.expect(b"hello world", timeout=1, scheduler=self.fs) d1.addBoth(self._cbTestTimeoutFailure) d2 = self.term.expect(b"zoom") d2.addCallback(result.append) self.fs.calls[0].call() self.assertEqual(len(result), 1) self.assertEqual(result[0].group(), b"zoom") class CharacterAttributeTests(unittest.TestCase): """ Tests for L{twisted.conch.insults.helper.CharacterAttribute}. """ def test_equality(self): """ L{CharacterAttribute}s must have matching character attribute values (bold, blink, underline, etc) with the same values to be considered equal. """ self.assertEqual(helper.CharacterAttribute(), helper.CharacterAttribute()) self.assertEqual( helper.CharacterAttribute(), helper.CharacterAttribute(charset=G0) ) self.assertEqual( helper.CharacterAttribute( bold=True, underline=True, blink=False, reverseVideo=True, foreground=helper.BLUE, ), helper.CharacterAttribute( bold=True, underline=True, blink=False, reverseVideo=True, foreground=helper.BLUE, ), ) self.assertNotEqual( helper.CharacterAttribute(), helper.CharacterAttribute(charset=G1) ) self.assertNotEqual( helper.CharacterAttribute(bold=True), helper.CharacterAttribute(bold=False) ) def test_wantOneDeprecated(self): """ L{twisted.conch.insults.helper.CharacterAttribute.wantOne} emits a deprecation warning when invoked. """ # Trigger the deprecation warning. helper._FormattingState().wantOne(bold=True) warningsShown = self.flushWarnings([self.test_wantOneDeprecated]) self.assertEqual(len(warningsShown), 1) self.assertEqual(warningsShown[0]["category"], DeprecationWarning) deprecatedClass = "twisted.conch.insults.helper._FormattingState.wantOne" self.assertEqual( warningsShown[0]["message"], "%s was deprecated in Twisted 13.1.0" % (deprecatedClass), ) PK F��Z��E�� � test_userauth.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for the implementation of the ssh-userauth service. Maintainer: Paul Swartz """ from types import ModuleType from typing import Optional from zope.interface import implementer from twisted.conch.error import ConchError, ValidPublicKey from twisted.cred.checkers import ICredentialsChecker from twisted.cred.credentials import IAnonymous, ISSHPrivateKey, IUsernamePassword from twisted.cred.error import UnauthorizedLogin from twisted.cred.portal import IRealm, Portal from twisted.internet import defer, task from twisted.protocols import loopback from twisted.python.reflect import requireModule from twisted.trial import unittest keys: Optional[ModuleType] = None if requireModule("cryptography") and requireModule("pyasn1"): from twisted.conch.checkers import SSHProtocolChecker from twisted.conch.ssh import keys, transport, userauth from twisted.conch.ssh.common import NS from twisted.conch.test import keydata else: class transport: # type: ignore[no-redef] class SSHTransportBase: """ A stub class so that later class definitions won't die. """ class userauth: # type: ignore[no-redef] class SSHUserAuthClient: """ A stub class so that later class definitions won't die. """ class ClientUserAuth(userauth.SSHUserAuthClient): """ A mock user auth client. """ def getPublicKey(self): """ If this is the first time we've been called, return a blob for the DSA key. Otherwise, return a blob for the RSA key. """ if self.lastPublicKey: return keys.Key.fromString(keydata.publicRSA_openssh) else: return defer.succeed(keys.Key.fromString(keydata.publicDSA_openssh)) def getPrivateKey(self): """ Return the private key object for the RSA key. """ return defer.succeed(keys.Key.fromString(keydata.privateRSA_openssh)) def getPassword(self, prompt=None): """ Return 'foo' as the password. """ return defer.succeed(b"foo") def getGenericAnswers(self, name, information, answers): """ Return 'foo' as the answer to two questions. """ return defer.succeed(("foo", "foo")) class OldClientAuth(userauth.SSHUserAuthClient): """ The old SSHUserAuthClient returned a cryptography key object from getPrivateKey() and a string from getPublicKey """ def getPrivateKey(self): return defer.succeed(keys.Key.fromString(keydata.privateRSA_openssh).keyObject) def getPublicKey(self): return keys.Key.fromString(keydata.publicRSA_openssh).blob() class ClientAuthWithoutPrivateKey(userauth.SSHUserAuthClient): """ This client doesn't have a private key, but it does have a public key. """ def getPrivateKey(self): return def getPublicKey(self): return keys.Key.fromString(keydata.publicRSA_openssh) class FakeTransport(transport.SSHTransportBase): """ L{userauth.SSHUserAuthServer} expects an SSH transport which has a factory attribute which has a portal attribute. Because the portal is important for testing authentication, we need to be able to provide an interesting portal object to the L{SSHUserAuthServer}. In addition, we want to be able to capture any packets sent over the transport. @ivar packets: a list of 2-tuples: (messageType, data). Each 2-tuple is a sent packet. @type packets: C{list} @param lostConnecion: True if loseConnection has been called on us. @type lostConnection: L{bool} """ class Service: """ A mock service, representing the other service offered by the server. """ name = b"nancy" def serviceStarted(self): pass class Factory: """ A mock factory, representing the factory that spawned this user auth service. """ def getService(self, transport, service): """ Return our fake service. """ if service == b"none": return FakeTransport.Service def __init__(self, portal): self.factory = self.Factory() self.factory.portal = portal self.lostConnection = False self.transport = self self.packets = [] def sendPacket(self, messageType, message): """ Record the packet sent by the service. """ self.packets.append((messageType, message)) def isEncrypted(self, direction): """ Pretend that this transport encrypts traffic in both directions. The SSHUserAuthServer disables password authentication if the transport isn't encrypted. """ return True def loseConnection(self): self.lostConnection = True @implementer(IRealm) class Realm: """ A mock realm for testing L{userauth.SSHUserAuthServer}. This realm is not actually used in the course of testing, so it returns the simplest thing that could possibly work. """ def requestAvatar(self, avatarId, mind, *interfaces): return defer.succeed((interfaces[0], None, lambda: None)) @implementer(ICredentialsChecker) class PasswordChecker: """ A very simple username/password checker which authenticates anyone whose password matches their username and rejects all others. """ credentialInterfaces = (IUsernamePassword,) def requestAvatarId(self, creds): if creds.username == creds.password: return defer.succeed(creds.username) return defer.fail(UnauthorizedLogin("Invalid username/password pair")) @implementer(ICredentialsChecker) class PrivateKeyChecker: """ A very simple public key checker which authenticates anyone whose public/private keypair is the same keydata.public/privateRSA_openssh. """ credentialInterfaces = (ISSHPrivateKey,) def requestAvatarId(self, creds): if creds.blob == keys.Key.fromString(keydata.publicRSA_openssh).blob(): if creds.signature is not None: obj = keys.Key.fromString(creds.blob) if obj.verify(creds.signature, creds.sigData): return creds.username else: raise ValidPublicKey() raise UnauthorizedLogin() @implementer(ICredentialsChecker) class AnonymousChecker: """ A simple checker which isn't supported by L{SSHUserAuthServer}. """ credentialInterfaces = (IAnonymous,) def requestAvatarId(self, credentials): # ICredentialsChecker.requestAvatarId pass class SSHUserAuthServerTests(unittest.TestCase): """ Tests for SSHUserAuthServer. """ if keys is None: skip = "cannot run without cryptography" def setUp(self): self.realm = Realm() self.portal = Portal(self.realm) self.portal.registerChecker(PasswordChecker()) self.portal.registerChecker(PrivateKeyChecker()) self.authServer = userauth.SSHUserAuthServer() self.authServer.transport = FakeTransport(self.portal) self.authServer.serviceStarted() self.authServer.supportedAuthentications.sort() # give a consistent # order def tearDown(self): self.authServer.serviceStopped() self.authServer = None def _checkFailed(self, ignored): """ Check that the authentication has failed. """ self.assertEqual( self.authServer.transport.packets[-1], (userauth.MSG_USERAUTH_FAILURE, NS(b"password,publickey") + b"\x00"), ) def test_noneAuthentication(self): """ A client may request a list of authentication 'method name' values that may continue by using the "none" authentication 'method name'. See RFC 4252 Section 5.2. """ d = self.authServer.ssh_USERAUTH_REQUEST( NS(b"foo") + NS(b"service") + NS(b"none") ) return d.addCallback(self._checkFailed) def test_successfulPasswordAuthentication(self): """ When provided with correct password authentication information, the server should respond by sending a MSG_USERAUTH_SUCCESS message with no other data. See RFC 4252, Section 5.1. """ packet = b"".join([NS(b"foo"), NS(b"none"), NS(b"password"), b"\0", NS(b"foo")]) d = self.authServer.ssh_USERAUTH_REQUEST(packet) def check(ignored): self.assertEqual( self.authServer.transport.packets, [(userauth.MSG_USERAUTH_SUCCESS, b"")], ) return d.addCallback(check) def test_failedPasswordAuthentication(self): """ When provided with invalid authentication details, the server should respond by sending a MSG_USERAUTH_FAILURE message which states whether the authentication was partially successful, and provides other, open options for authentication. See RFC 4252, Section 5.1. """ # packet = username, next_service, authentication type, FALSE, password packet = b"".join([NS(b"foo"), NS(b"none"), NS(b"password"), b"\0", NS(b"bar")]) self.authServer.clock = task.Clock() d = self.authServer.ssh_USERAUTH_REQUEST(packet) self.assertEqual(self.authServer.transport.packets, []) self.authServer.clock.advance(2) return d.addCallback(self._checkFailed) def test_successfulPrivateKeyAuthentication(self): """ Test that private key authentication completes successfully, """ blob = keys.Key.fromString(keydata.publicRSA_openssh).blob() obj = keys.Key.fromString(keydata.privateRSA_openssh) packet = ( NS(b"foo") + NS(b"none") + NS(b"publickey") + b"\xff" + NS(obj.sshType()) + NS(blob) ) self.authServer.transport.sessionID = b"test" signature = obj.sign( NS(b"test") + bytes((userauth.MSG_USERAUTH_REQUEST,)) + packet ) packet += NS(signature) d = self.authServer.ssh_USERAUTH_REQUEST(packet) def check(ignored): self.assertEqual( self.authServer.transport.packets, [(userauth.MSG_USERAUTH_SUCCESS, b"")], ) return d.addCallback(check) def test_requestRaisesConchError(self): """ ssh_USERAUTH_REQUEST should raise a ConchError if tryAuth returns None. Added to catch a bug noticed by pyflakes. """ d = defer.Deferred() def mockCbFinishedAuth(self, ignored): self.fail("request should have raised ConochError") def mockTryAuth(kind, user, data): return None def mockEbBadAuth(reason): d.errback(reason.value) self.patch(self.authServer, "tryAuth", mockTryAuth) self.patch(self.authServer, "_cbFinishedAuth", mockCbFinishedAuth) self.patch(self.authServer, "_ebBadAuth", mockEbBadAuth) packet = NS(b"user") + NS(b"none") + NS(b"public-key") + NS(b"data") # If an error other than ConchError is raised, this will trigger an # exception. self.authServer.ssh_USERAUTH_REQUEST(packet) return self.assertFailure(d, ConchError) def test_verifyValidPrivateKey(self): """ Test that verifying a valid private key works. """ blob = keys.Key.fromString(keydata.publicRSA_openssh).blob() packet = ( NS(b"foo") + NS(b"none") + NS(b"publickey") + b"\x00" + NS(b"ssh-rsa") + NS(blob) ) d = self.authServer.ssh_USERAUTH_REQUEST(packet) def check(ignored): self.assertEqual( self.authServer.transport.packets, [(userauth.MSG_USERAUTH_PK_OK, NS(b"ssh-rsa") + NS(blob))], ) return d.addCallback(check) def test_failedPrivateKeyAuthenticationWithoutSignature(self): """ Test that private key authentication fails when the public key is invalid. """ blob = keys.Key.fromString(keydata.publicDSA_openssh).blob() packet = ( NS(b"foo") + NS(b"none") + NS(b"publickey") + b"\x00" + NS(b"ssh-dsa") + NS(blob) ) d = self.authServer.ssh_USERAUTH_REQUEST(packet) return d.addCallback(self._checkFailed) def test_failedPrivateKeyAuthenticationWithSignature(self): """ Test that private key authentication fails when the public key is invalid. """ blob = keys.Key.fromString(keydata.publicRSA_openssh).blob() obj = keys.Key.fromString(keydata.privateRSA_openssh) packet = ( NS(b"foo") + NS(b"none") + NS(b"publickey") + b"\xff" + NS(b"ssh-rsa") + NS(blob) + NS(obj.sign(blob)) ) self.authServer.transport.sessionID = b"test" d = self.authServer.ssh_USERAUTH_REQUEST(packet) return d.addCallback(self._checkFailed) def test_unsupported_publickey(self): """ Private key authentication fails when the public key type is unsupported or the public key is corrupt. """ blob = keys.Key.fromString(keydata.publicDSA_openssh).blob() # Change the blob to a bad type blob = NS(b"ssh-bad-type") + blob[11:] packet = ( NS(b"foo") + NS(b"none") + NS(b"publickey") + b"\x00" + NS(b"ssh-rsa") + NS(blob) ) d = self.authServer.ssh_USERAUTH_REQUEST(packet) return d.addCallback(self._checkFailed) def test_ignoreUnknownCredInterfaces(self): """ L{SSHUserAuthServer} sets up C{SSHUserAuthServer.supportedAuthentications} by checking the portal's credentials interfaces and mapping them to SSH authentication method strings. If the Portal advertises an interface that L{SSHUserAuthServer} can't map, it should be ignored. This is a white box test. """ server = userauth.SSHUserAuthServer() server.transport = FakeTransport(self.portal) self.portal.registerChecker(AnonymousChecker()) server.serviceStarted() server.serviceStopped() server.supportedAuthentications.sort() # give a consistent order self.assertEqual(server.supportedAuthentications, [b"password", b"publickey"]) def test_removePasswordIfUnencrypted(self): """ Test that the userauth service does not advertise password authentication if the password would be send in cleartext. """ self.assertIn(b"password", self.authServer.supportedAuthentications) # no encryption clearAuthServer = userauth.SSHUserAuthServer() clearAuthServer.transport = FakeTransport(self.portal) clearAuthServer.transport.isEncrypted = lambda x: False clearAuthServer.serviceStarted() clearAuthServer.serviceStopped() self.assertNotIn(b"password", clearAuthServer.supportedAuthentications) # only encrypt incoming (the direction the password is sent) halfAuthServer = userauth.SSHUserAuthServer() halfAuthServer.transport = FakeTransport(self.portal) halfAuthServer.transport.isEncrypted = lambda x: x == "in" halfAuthServer.serviceStarted() halfAuthServer.serviceStopped() self.assertIn(b"password", halfAuthServer.supportedAuthentications) def test_unencryptedConnectionWithoutPasswords(self): """ If the L{SSHUserAuthServer} is not advertising passwords, then an unencrypted connection should not cause any warnings or exceptions. This is a white box test. """ # create a Portal without password authentication portal = Portal(self.realm) portal.registerChecker(PrivateKeyChecker()) # no encryption clearAuthServer = userauth.SSHUserAuthServer() clearAuthServer.transport = FakeTransport(portal) clearAuthServer.transport.isEncrypted = lambda x: False clearAuthServer.serviceStarted() clearAuthServer.serviceStopped() self.assertEqual(clearAuthServer.supportedAuthentications, [b"publickey"]) # only encrypt incoming (the direction the password is sent) halfAuthServer = userauth.SSHUserAuthServer() halfAuthServer.transport = FakeTransport(portal) halfAuthServer.transport.isEncrypted = lambda x: x == "in" halfAuthServer.serviceStarted() halfAuthServer.serviceStopped() self.assertEqual(clearAuthServer.supportedAuthentications, [b"publickey"]) def test_loginTimeout(self): """ Test that the login times out. """ timeoutAuthServer = userauth.SSHUserAuthServer() timeoutAuthServer.clock = task.Clock() timeoutAuthServer.transport = FakeTransport(self.portal) timeoutAuthServer.serviceStarted() timeoutAuthServer.clock.advance(11 * 60 * 60) timeoutAuthServer.serviceStopped() self.assertEqual( timeoutAuthServer.transport.packets, [ ( transport.MSG_DISCONNECT, b"\x00" * 3 + bytes((transport.DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE,)) + NS(b"you took too long") + NS(b""), ) ], ) self.assertTrue(timeoutAuthServer.transport.lostConnection) def test_cancelLoginTimeout(self): """ Test that stopping the service also stops the login timeout. """ timeoutAuthServer = userauth.SSHUserAuthServer() timeoutAuthServer.clock = task.Clock() timeoutAuthServer.transport = FakeTransport(self.portal) timeoutAuthServer.serviceStarted() timeoutAuthServer.serviceStopped() timeoutAuthServer.clock.advance(11 * 60 * 60) self.assertEqual(timeoutAuthServer.transport.packets, []) self.assertFalse(timeoutAuthServer.transport.lostConnection) def test_tooManyAttempts(self): """ Test that the server disconnects if the client fails authentication too many times. """ packet = b"".join([NS(b"foo"), NS(b"none"), NS(b"password"), b"\0", NS(b"bar")]) self.authServer.clock = task.Clock() for i in range(21): d = self.authServer.ssh_USERAUTH_REQUEST(packet) self.authServer.clock.advance(2) def check(ignored): self.assertEqual( self.authServer.transport.packets[-1], ( transport.MSG_DISCONNECT, b"\x00" * 3 + bytes((transport.DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE,)) + NS(b"too many bad auths") + NS(b""), ), ) return d.addCallback(check) def test_failIfUnknownService(self): """ If the user requests a service that we don't support, the authentication should fail. """ packet = NS(b"foo") + NS(b"") + NS(b"password") + b"\0" + NS(b"foo") self.authServer.clock = task.Clock() d = self.authServer.ssh_USERAUTH_REQUEST(packet) return d.addCallback(self._checkFailed) def test_tryAuthEdgeCases(self): """ tryAuth() has two edge cases that are difficult to reach. 1) an authentication method auth_* returns None instead of a Deferred. 2) an authentication type that is defined does not have a matching auth_* method. Both these cases should return a Deferred which fails with a ConchError. """ def mockAuth(packet): return None self.patch(self.authServer, "auth_publickey", mockAuth) # first case self.patch(self.authServer, "auth_password", None) # second case def secondTest(ignored): d2 = self.authServer.tryAuth(b"password", None, None) return self.assertFailure(d2, ConchError) d1 = self.authServer.tryAuth(b"publickey", None, None) return self.assertFailure(d1, ConchError).addCallback(secondTest) class SSHUserAuthClientTests(unittest.TestCase): """ Tests for SSHUserAuthClient. """ if keys is None: skip = "cannot run without cryptography" def setUp(self): self.authClient = ClientUserAuth(b"foo", FakeTransport.Service()) self.authClient.transport = FakeTransport(None) self.authClient.transport.sessionID = b"test" self.authClient.serviceStarted() def tearDown(self): self.authClient.serviceStopped() self.authClient = None def test_init(self): """ Test that client is initialized properly. """ self.assertEqual(self.authClient.user, b"foo") self.assertEqual(self.authClient.instance.name, b"nancy") self.assertEqual( self.authClient.transport.packets, [(userauth.MSG_USERAUTH_REQUEST, NS(b"foo") + NS(b"nancy") + NS(b"none"))], ) def test_USERAUTH_SUCCESS(self): """ Test that the client succeeds properly. """ instance = [None] def stubSetService(service): instance[0] = service self.authClient.transport.setService = stubSetService self.authClient.ssh_USERAUTH_SUCCESS(b"") self.assertEqual(instance[0], self.authClient.instance) def test_publickey(self): """ Test that the client can authenticate with a public key. """ self.authClient.ssh_USERAUTH_FAILURE(NS(b"publickey") + b"\x00") self.assertEqual( self.authClient.transport.packets[-1], ( userauth.MSG_USERAUTH_REQUEST, NS(b"foo") + NS(b"nancy") + NS(b"publickey") + b"\x00" + NS(b"ssh-dss") + NS(keys.Key.fromString(keydata.publicDSA_openssh).blob()), ), ) # that key isn't good self.authClient.ssh_USERAUTH_FAILURE(NS(b"publickey") + b"\x00") blob = NS(keys.Key.fromString(keydata.publicRSA_openssh).blob()) self.assertEqual( self.authClient.transport.packets[-1], ( userauth.MSG_USERAUTH_REQUEST, ( NS(b"foo") + NS(b"nancy") + NS(b"publickey") + b"\x00" + NS(b"ssh-rsa") + blob ), ), ) self.authClient.ssh_USERAUTH_PK_OK( NS(b"ssh-rsa") + NS(keys.Key.fromString(keydata.publicRSA_openssh).blob()) ) sigData = ( NS(self.authClient.transport.sessionID) + bytes((userauth.MSG_USERAUTH_REQUEST,)) + NS(b"foo") + NS(b"nancy") + NS(b"publickey") + b"\x01" + NS(b"ssh-rsa") + blob ) obj = keys.Key.fromString(keydata.privateRSA_openssh) self.assertEqual( self.authClient.transport.packets[-1], ( userauth.MSG_USERAUTH_REQUEST, NS(b"foo") + NS(b"nancy") + NS(b"publickey") + b"\x01" + NS(b"ssh-rsa") + blob + NS(obj.sign(sigData)), ), ) def test_publickey_without_privatekey(self): """ If the SSHUserAuthClient doesn't return anything from signData, the client should start the authentication over again by requesting 'none' authentication. """ authClient = ClientAuthWithoutPrivateKey(b"foo", FakeTransport.Service()) authClient.transport = FakeTransport(None) authClient.transport.sessionID = b"test" authClient.serviceStarted() authClient.tryAuth(b"publickey") authClient.transport.packets = [] self.assertIsNone(authClient.ssh_USERAUTH_PK_OK(b"")) self.assertEqual( authClient.transport.packets, [(userauth.MSG_USERAUTH_REQUEST, NS(b"foo") + NS(b"nancy") + NS(b"none"))], ) def test_no_publickey(self): """ If there's no public key, auth_publickey should return a Deferred called back with a False value. """ self.authClient.getPublicKey = lambda x: None d = self.authClient.tryAuth(b"publickey") def check(result): self.assertFalse(result) return d.addCallback(check) def test_password(self): """ Test that the client can authentication with a password. This includes changing the password. """ self.authClient.ssh_USERAUTH_FAILURE(NS(b"password") + b"\x00") self.assertEqual( self.authClient.transport.packets[-1], ( userauth.MSG_USERAUTH_REQUEST, NS(b"foo") + NS(b"nancy") + NS(b"password") + b"\x00" + NS(b"foo"), ), ) self.authClient.ssh_USERAUTH_PK_OK(NS(b"") + NS(b"")) self.assertEqual( self.authClient.transport.packets[-1], ( userauth.MSG_USERAUTH_REQUEST, NS(b"foo") + NS(b"nancy") + NS(b"password") + b"\xff" + NS(b"foo") * 2, ), ) def test_no_password(self): """ If getPassword returns None, tryAuth should return False. """ self.authClient.getPassword = lambda: None self.assertFalse(self.authClient.tryAuth(b"password")) def test_keyboardInteractive(self): """ Make sure that the client can authenticate with the keyboard interactive method. """ self.authClient.ssh_USERAUTH_PK_OK_keyboard_interactive( NS(b"") + NS(b"") + NS(b"") + b"\x00\x00\x00\x01" + NS(b"Password: ") + b"\x00" ) self.assertEqual( self.authClient.transport.packets[-1], ( userauth.MSG_USERAUTH_INFO_RESPONSE, b"\x00\x00\x00\x02" + NS(b"foo") + NS(b"foo"), ), ) def test_USERAUTH_PK_OK_unknown_method(self): """ If C{SSHUserAuthClient} gets a MSG_USERAUTH_PK_OK packet when it's not expecting it, it should fail the current authentication and move on to the next type. """ self.authClient.lastAuth = b"unknown" self.authClient.transport.packets = [] self.authClient.ssh_USERAUTH_PK_OK(b"") self.assertEqual( self.authClient.transport.packets, [(userauth.MSG_USERAUTH_REQUEST, NS(b"foo") + NS(b"nancy") + NS(b"none"))], ) def test_USERAUTH_FAILURE_sorting(self): """ ssh_USERAUTH_FAILURE should sort the methods by their position in SSHUserAuthClient.preferredOrder. Methods that are not in preferredOrder should be sorted at the end of that list. """ def auth_firstmethod(): self.authClient.transport.sendPacket(255, b"here is data") def auth_anothermethod(): self.authClient.transport.sendPacket(254, b"other data") return True self.authClient.auth_firstmethod = auth_firstmethod self.authClient.auth_anothermethod = auth_anothermethod # although they shouldn't get called, method callbacks auth_* MUST # exist in order for the test to work properly. self.authClient.ssh_USERAUTH_FAILURE(NS(b"anothermethod,password") + b"\x00") # should send password packet self.assertEqual( self.authClient.transport.packets[-1], ( userauth.MSG_USERAUTH_REQUEST, NS(b"foo") + NS(b"nancy") + NS(b"password") + b"\x00" + NS(b"foo"), ), ) self.authClient.ssh_USERAUTH_FAILURE( NS(b"firstmethod,anothermethod,password") + b"\xff" ) self.assertEqual( self.authClient.transport.packets[-2:], [(255, b"here is data"), (254, b"other data")], ) def test_disconnectIfNoMoreAuthentication(self): """ If there are no more available user authentication messages, the SSHUserAuthClient should disconnect with code DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE. """ self.authClient.ssh_USERAUTH_FAILURE(NS(b"password") + b"\x00") self.authClient.ssh_USERAUTH_FAILURE(NS(b"password") + b"\xff") self.assertEqual( self.authClient.transport.packets[-1], ( transport.MSG_DISCONNECT, b"\x00\x00\x00\x0e" + NS(b"no more authentication methods available") + b"\x00\x00\x00\x00", ), ) def test_ebAuth(self): """ _ebAuth (the generic authentication error handler) should send a request for the 'none' authentication method. """ self.authClient.transport.packets = [] self.authClient._ebAuth(None) self.assertEqual( self.authClient.transport.packets, [(userauth.MSG_USERAUTH_REQUEST, NS(b"foo") + NS(b"nancy") + NS(b"none"))], ) def test_defaults(self): """ getPublicKey() should return None. getPrivateKey() should return a failed Deferred. getPassword() should return a failed Deferred. getGenericAnswers() should return a failed Deferred. """ authClient = userauth.SSHUserAuthClient(b"foo", FakeTransport.Service()) self.assertIsNone(authClient.getPublicKey()) def check(result): result.trap(NotImplementedError) d = authClient.getPassword() return d.addCallback(self.fail).addErrback(check2) def check2(result): result.trap(NotImplementedError) d = authClient.getGenericAnswers(None, None, None) return d.addCallback(self.fail).addErrback(check3) def check3(result): result.trap(NotImplementedError) d = authClient.getPrivateKey() return d.addCallback(self.fail).addErrback(check) class LoopbackTests(unittest.TestCase): if keys is None: skip = "cannot run without cryptography or PyASN1" class Factory: class Service: name = b"TestService" def serviceStarted(self): self.transport.loseConnection() def serviceStopped(self): pass def getService(self, avatar, name): return self.Service def test_loopback(self): """ Test that the userauth server and client play nicely with each other. """ server = userauth.SSHUserAuthServer() client = ClientUserAuth(b"foo", self.Factory.Service()) # set up transports server.transport = transport.SSHTransportBase() server.transport.service = server server.transport.isEncrypted = lambda x: True client.transport = transport.SSHTransportBase() client.transport.service = client server.transport.sessionID = client.transport.sessionID = b"" # don't send key exchange packet server.transport.sendKexInit = client.transport.sendKexInit = lambda: None # set up server authentication server.transport.factory = self.Factory() server.passwordDelay = 0 # remove bad password delay realm = Realm() portal = Portal(realm) checker = SSHProtocolChecker() checker.registerChecker(PasswordChecker()) checker.registerChecker(PrivateKeyChecker()) checker.areDone = lambda aId: (len(checker.successfulCredentials[aId]) == 2) portal.registerChecker(checker) server.transport.factory.portal = portal d = loopback.loopbackAsync(server.transport, client.transport) server.transport.transport.logPrefix = lambda: "_ServerLoopback" client.transport.transport.logPrefix = lambda: "_ClientLoopback" server.serviceStarted() client.serviceStarted() def check(ignored): self.assertEqual(server.transport.service.name, b"TestService") return d.addCallback(check) class ModuleInitializationTests(unittest.TestCase): if keys is None: skip = "cannot run without cryptography or PyASN1" def test_messages(self): # Several message types have value 60, check that MSG_USERAUTH_PK_OK # is always the one which is mapped. self.assertEqual( userauth.SSHUserAuthServer.protocolMessages[60], "MSG_USERAUTH_PK_OK" ) self.assertEqual( userauth.SSHUserAuthClient.protocolMessages[60], "MSG_USERAUTH_PK_OK" ) PK F��Z2��i�h �h test_telnet.pynu �[��� # -*- test-case-name: twisted.conch.test.test_telnet -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.telnet}. """ from zope.interface import implementer from zope.interface.verify import verifyObject from twisted.conch import telnet from twisted.internet import defer from twisted.python.compat import iterbytes from twisted.test import proto_helpers from twisted.trial import unittest @implementer(telnet.ITelnetProtocol) class TestProtocol: localEnableable = () remoteEnableable = () def __init__(self): self.data = b"" self.subcmd = [] self.calls = [] self.enabledLocal = [] self.enabledRemote = [] self.disabledLocal = [] self.disabledRemote = [] def makeConnection(self, transport): d = transport.negotiationMap = {} d[b"\x12"] = self.neg_TEST_COMMAND d = transport.commandMap = transport.commandMap.copy() for cmd in ("EOR", "NOP", "DM", "BRK", "IP", "AO", "AYT", "EC", "EL", "GA"): d[getattr(telnet, cmd)] = lambda arg, cmd=cmd: self.calls.append(cmd) def dataReceived(self, data): self.data += data def connectionLost(self, reason): pass def neg_TEST_COMMAND(self, payload): self.subcmd = payload def enableLocal(self, option): if option in self.localEnableable: self.enabledLocal.append(option) return True return False def disableLocal(self, option): self.disabledLocal.append(option) def enableRemote(self, option): if option in self.remoteEnableable: self.enabledRemote.append(option) return True return False def disableRemote(self, option): self.disabledRemote.append(option) def connectionMade(self): # IProtocol.connectionMade pass def unhandledCommand(self, command, argument): # ITelnetProtocol.unhandledCommand pass def unhandledSubnegotiation(self, command, data): # ITelnetProtocol.unhandledSubnegotiation pass class InterfacesTests(unittest.TestCase): def test_interface(self): """ L{telnet.TelnetProtocol} implements L{telnet.ITelnetProtocol} """ p = telnet.TelnetProtocol() verifyObject(telnet.ITelnetProtocol, p) class TelnetTransportTests(unittest.TestCase): """ Tests for L{telnet.TelnetTransport}. """ def setUp(self): self.p = telnet.TelnetTransport(TestProtocol) self.t = proto_helpers.StringTransport() self.p.makeConnection(self.t) def testRegularBytes(self): # Just send a bunch of bytes. None of these do anything # with telnet. They should pass right through to the # application layer. h = self.p.protocol L = [ b"here are some bytes la la la", b"some more arrive here", b"lots of bytes to play with", b"la la la", b"ta de da", b"dum", ] for b in L: self.p.dataReceived(b) self.assertEqual(h.data, b"".join(L)) def testNewlineHandling(self): # Send various kinds of newlines and make sure they get translated # into \n. h = self.p.protocol L = [ b"here is the first line\r\n", b"here is the second line\r\0", b"here is the third line\r\n", b"here is the last line\r\0", ] for b in L: self.p.dataReceived(b) self.assertEqual( h.data, L[0][:-2] + b"\n" + L[1][:-2] + b"\r" + L[2][:-2] + b"\n" + L[3][:-2] + b"\r", ) def testIACEscape(self): # Send a bunch of bytes and a couple quoted \xFFs. Unquoted, # \xFF is a telnet command. Quoted, one of them from each pair # should be passed through to the application layer. h = self.p.protocol L = [ b"here are some bytes\xff\xff with an embedded IAC", b"and here is a test of a border escape\xff", b"\xff did you get that IAC?", ] for b in L: self.p.dataReceived(b) self.assertEqual(h.data, b"".join(L).replace(b"\xff\xff", b"\xff")) def _simpleCommandTest(self, cmdName): # Send a single simple telnet command and make sure # it gets noticed and the appropriate method gets # called. h = self.p.protocol cmd = telnet.IAC + getattr(telnet, cmdName) L = [b"Here's some bytes, tra la la", b"But ono!" + cmd + b" an interrupt"] for b in L: self.p.dataReceived(b) self.assertEqual(h.calls, [cmdName]) self.assertEqual(h.data, b"".join(L).replace(cmd, b"")) def testInterrupt(self): self._simpleCommandTest("IP") def testEndOfRecord(self): self._simpleCommandTest("EOR") def testNoOperation(self): self._simpleCommandTest("NOP") def testDataMark(self): self._simpleCommandTest("DM") def testBreak(self): self._simpleCommandTest("BRK") def testAbortOutput(self): self._simpleCommandTest("AO") def testAreYouThere(self): self._simpleCommandTest("AYT") def testEraseCharacter(self): self._simpleCommandTest("EC") def testEraseLine(self): self._simpleCommandTest("EL") def testGoAhead(self): self._simpleCommandTest("GA") def testSubnegotiation(self): # Send a subnegotiation command and make sure it gets # parsed and that the correct method is called. h = self.p.protocol cmd = telnet.IAC + telnet.SB + b"\x12hello world" + telnet.IAC + telnet.SE L = [b"These are some bytes but soon" + cmd, b"there will be some more"] for b in L: self.p.dataReceived(b) self.assertEqual(h.data, b"".join(L).replace(cmd, b"")) self.assertEqual(h.subcmd, list(iterbytes(b"hello world"))) def testSubnegotiationWithEmbeddedSE(self): # Send a subnegotiation command with an embedded SE. Make sure # that SE gets passed to the correct method. h = self.p.protocol cmd = telnet.IAC + telnet.SB + b"\x12" + telnet.SE + telnet.IAC + telnet.SE L = [b"Some bytes are here" + cmd + b"and here", b"and here"] for b in L: self.p.dataReceived(b) self.assertEqual(h.data, b"".join(L).replace(cmd, b"")) self.assertEqual(h.subcmd, [telnet.SE]) def testBoundarySubnegotiation(self): # Send a subnegotiation command. Split it at every possible byte boundary # and make sure it always gets parsed and that it is passed to the correct # method. cmd = ( telnet.IAC + telnet.SB + b"\x12" + telnet.SE + b"hello" + telnet.IAC + telnet.SE ) for i in range(len(cmd)): h = self.p.protocol = TestProtocol() h.makeConnection(self.p) a, b = cmd[:i], cmd[i:] L = [b"first part" + a, b + b"last part"] for data in L: self.p.dataReceived(data) self.assertEqual(h.data, b"".join(L).replace(cmd, b"")) self.assertEqual(h.subcmd, [telnet.SE] + list(iterbytes(b"hello"))) def _enabledHelper(self, o, eL=[], eR=[], dL=[], dR=[]): self.assertEqual(o.enabledLocal, eL) self.assertEqual(o.enabledRemote, eR) self.assertEqual(o.disabledLocal, dL) self.assertEqual(o.disabledRemote, dR) def testRefuseWill(self): # Try to enable an option. The server should refuse to enable it. cmd = telnet.IAC + telnet.WILL + b"\x12" data = b"surrounding bytes" + cmd + b"to spice things up" self.p.dataReceived(data) self.assertEqual(self.p.protocol.data, data.replace(cmd, b"")) self.assertEqual(self.t.value(), telnet.IAC + telnet.DONT + b"\x12") self._enabledHelper(self.p.protocol) def testRefuseDo(self): # Try to enable an option. The server should refuse to enable it. cmd = telnet.IAC + telnet.DO + b"\x12" data = b"surrounding bytes" + cmd + b"to spice things up" self.p.dataReceived(data) self.assertEqual(self.p.protocol.data, data.replace(cmd, b"")) self.assertEqual(self.t.value(), telnet.IAC + telnet.WONT + b"\x12") self._enabledHelper(self.p.protocol) def testAcceptDo(self): # Try to enable an option. The option is in our allowEnable # list, so we will allow it to be enabled. cmd = telnet.IAC + telnet.DO + b"\x19" data = b"padding" + cmd + b"trailer" h = self.p.protocol h.localEnableable = (b"\x19",) self.p.dataReceived(data) self.assertEqual(self.t.value(), telnet.IAC + telnet.WILL + b"\x19") self._enabledHelper(h, eL=[b"\x19"]) def testAcceptWill(self): # Same as testAcceptDo, but reversed. cmd = telnet.IAC + telnet.WILL + b"\x91" data = b"header" + cmd + b"padding" h = self.p.protocol h.remoteEnableable = (b"\x91",) self.p.dataReceived(data) self.assertEqual(self.t.value(), telnet.IAC + telnet.DO + b"\x91") self._enabledHelper(h, eR=[b"\x91"]) def testAcceptWont(self): # Try to disable an option. The server must allow any option to # be disabled at any time. Make sure it disables it and sends # back an acknowledgement of this. cmd = telnet.IAC + telnet.WONT + b"\x29" # Jimmy it - after these two lines, the server will be in a state # such that it believes the option to have been previously enabled # via normal negotiation. s = self.p.getOptionState(b"\x29") s.him.state = "yes" data = b"fiddle dee" + cmd self.p.dataReceived(data) self.assertEqual(self.p.protocol.data, data.replace(cmd, b"")) self.assertEqual(self.t.value(), telnet.IAC + telnet.DONT + b"\x29") self.assertEqual(s.him.state, "no") self._enabledHelper(self.p.protocol, dR=[b"\x29"]) def testAcceptDont(self): # Try to disable an option. The server must allow any option to # be disabled at any time. Make sure it disables it and sends # back an acknowledgement of this. cmd = telnet.IAC + telnet.DONT + b"\x29" # Jimmy it - after these two lines, the server will be in a state # such that it believes the option to have beenp previously enabled # via normal negotiation. s = self.p.getOptionState(b"\x29") s.us.state = "yes" data = b"fiddle dum " + cmd self.p.dataReceived(data) self.assertEqual(self.p.protocol.data, data.replace(cmd, b"")) self.assertEqual(self.t.value(), telnet.IAC + telnet.WONT + b"\x29") self.assertEqual(s.us.state, "no") self._enabledHelper(self.p.protocol, dL=[b"\x29"]) def testIgnoreWont(self): # Try to disable an option. The option is already disabled. The # server should send nothing in response to this. cmd = telnet.IAC + telnet.WONT + b"\x47" data = b"dum de dum" + cmd + b"tra la la" self.p.dataReceived(data) self.assertEqual(self.p.protocol.data, data.replace(cmd, b"")) self.assertEqual(self.t.value(), b"") self._enabledHelper(self.p.protocol) def testIgnoreDont(self): # Try to disable an option. The option is already disabled. The # server should send nothing in response to this. Doing so could # lead to a negotiation loop. cmd = telnet.IAC + telnet.DONT + b"\x47" data = b"dum de dum" + cmd + b"tra la la" self.p.dataReceived(data) self.assertEqual(self.p.protocol.data, data.replace(cmd, b"")) self.assertEqual(self.t.value(), b"") self._enabledHelper(self.p.protocol) def testIgnoreWill(self): # Try to enable an option. The option is already enabled. The # server should send nothing in response to this. Doing so could # lead to a negotiation loop. cmd = telnet.IAC + telnet.WILL + b"\x56" # Jimmy it - after these two lines, the server will be in a state # such that it believes the option to have been previously enabled # via normal negotiation. s = self.p.getOptionState(b"\x56") s.him.state = "yes" data = b"tra la la" + cmd + b"dum de dum" self.p.dataReceived(data) self.assertEqual(self.p.protocol.data, data.replace(cmd, b"")) self.assertEqual(self.t.value(), b"") self._enabledHelper(self.p.protocol) def testIgnoreDo(self): # Try to enable an option. The option is already enabled. The # server should send nothing in response to this. Doing so could # lead to a negotiation loop. cmd = telnet.IAC + telnet.DO + b"\x56" # Jimmy it - after these two lines, the server will be in a state # such that it believes the option to have been previously enabled # via normal negotiation. s = self.p.getOptionState(b"\x56") s.us.state = "yes" data = b"tra la la" + cmd + b"dum de dum" self.p.dataReceived(data) self.assertEqual(self.p.protocol.data, data.replace(cmd, b"")) self.assertEqual(self.t.value(), b"") self._enabledHelper(self.p.protocol) def testAcceptedEnableRequest(self): # Try to enable an option through the user-level API. This # returns a Deferred that fires when negotiation about the option # finishes. Make sure it fires, make sure state gets updated # properly, make sure the result indicates the option was enabled. d = self.p.do(b"\x42") h = self.p.protocol h.remoteEnableable = (b"\x42",) self.assertEqual(self.t.value(), telnet.IAC + telnet.DO + b"\x42") self.p.dataReceived(telnet.IAC + telnet.WILL + b"\x42") d.addCallback(self.assertEqual, True) d.addCallback(lambda _: self._enabledHelper(h, eR=[b"\x42"])) return d def test_refusedEnableRequest(self): """ If the peer refuses to enable an option we request it to enable, the L{Deferred} returned by L{TelnetProtocol.do} fires with an L{OptionRefused} L{Failure}. """ # Try to enable an option through the user-level API. This returns a # Deferred that fires when negotiation about the option finishes. Make # sure it fires, make sure state gets updated properly, make sure the # result indicates the option was enabled. self.p.protocol.remoteEnableable = (b"\x42",) d = self.p.do(b"\x42") self.assertEqual(self.t.value(), telnet.IAC + telnet.DO + b"\x42") s = self.p.getOptionState(b"\x42") self.assertEqual(s.him.state, "no") self.assertEqual(s.us.state, "no") self.assertTrue(s.him.negotiating) self.assertFalse(s.us.negotiating) self.p.dataReceived(telnet.IAC + telnet.WONT + b"\x42") d = self.assertFailure(d, telnet.OptionRefused) d.addCallback(lambda ignored: self._enabledHelper(self.p.protocol)) d.addCallback(lambda ignored: self.assertFalse(s.him.negotiating)) return d def test_refusedEnableOffer(self): """ If the peer refuses to allow us to enable an option, the L{Deferred} returned by L{TelnetProtocol.will} fires with an L{OptionRefused} L{Failure}. """ # Try to offer an option through the user-level API. This returns a # Deferred that fires when negotiation about the option finishes. Make # sure it fires, make sure state gets updated properly, make sure the # result indicates the option was enabled. self.p.protocol.localEnableable = (b"\x42",) d = self.p.will(b"\x42") self.assertEqual(self.t.value(), telnet.IAC + telnet.WILL + b"\x42") s = self.p.getOptionState(b"\x42") self.assertEqual(s.him.state, "no") self.assertEqual(s.us.state, "no") self.assertFalse(s.him.negotiating) self.assertTrue(s.us.negotiating) self.p.dataReceived(telnet.IAC + telnet.DONT + b"\x42") d = self.assertFailure(d, telnet.OptionRefused) d.addCallback(lambda ignored: self._enabledHelper(self.p.protocol)) d.addCallback(lambda ignored: self.assertFalse(s.us.negotiating)) return d def testAcceptedDisableRequest(self): # Try to disable an option through the user-level API. This # returns a Deferred that fires when negotiation about the option # finishes. Make sure it fires, make sure state gets updated # properly, make sure the result indicates the option was enabled. s = self.p.getOptionState(b"\x42") s.him.state = "yes" d = self.p.dont(b"\x42") self.assertEqual(self.t.value(), telnet.IAC + telnet.DONT + b"\x42") self.p.dataReceived(telnet.IAC + telnet.WONT + b"\x42") d.addCallback(self.assertEqual, True) d.addCallback(lambda _: self._enabledHelper(self.p.protocol, dR=[b"\x42"])) return d def testNegotiationBlocksFurtherNegotiation(self): # Try to disable an option, then immediately try to enable it, then # immediately try to disable it. Ensure that the 2nd and 3rd calls # fail quickly with the right exception. s = self.p.getOptionState(b"\x24") s.him.state = "yes" self.p.dont(b"\x24") # fires after the first line of _final def _do(x): d = self.p.do(b"\x24") return self.assertFailure(d, telnet.AlreadyNegotiating) def _dont(x): d = self.p.dont(b"\x24") return self.assertFailure(d, telnet.AlreadyNegotiating) def _final(x): self.p.dataReceived(telnet.IAC + telnet.WONT + b"\x24") # an assertion that only passes if d2 has fired self._enabledHelper(self.p.protocol, dR=[b"\x24"]) # Make sure we allow this self.p.protocol.remoteEnableable = (b"\x24",) d = self.p.do(b"\x24") self.p.dataReceived(telnet.IAC + telnet.WILL + b"\x24") d.addCallback(self.assertEqual, True) d.addCallback( lambda _: self._enabledHelper( self.p.protocol, eR=[b"\x24"], dR=[b"\x24"] ) ) return d d = _do(None) d.addCallback(_dont) d.addCallback(_final) return d def testSuperfluousDisableRequestRaises(self): # Try to disable a disabled option. Make sure it fails properly. d = self.p.dont(b"\xab") return self.assertFailure(d, telnet.AlreadyDisabled) def testSuperfluousEnableRequestRaises(self): # Try to disable a disabled option. Make sure it fails properly. s = self.p.getOptionState(b"\xab") s.him.state = "yes" d = self.p.do(b"\xab") return self.assertFailure(d, telnet.AlreadyEnabled) def testLostConnectionFailsDeferreds(self): d1 = self.p.do(b"\x12") d2 = self.p.do(b"\x23") d3 = self.p.do(b"\x34") class TestException(Exception): pass self.p.connectionLost(TestException("Total failure!")) d1 = self.assertFailure(d1, TestException) d2 = self.assertFailure(d2, TestException) d3 = self.assertFailure(d3, TestException) return defer.gatherResults([d1, d2, d3]) class TestTelnet(telnet.Telnet): """ A trivial extension of the telnet protocol class useful to unit tests. """ def __init__(self): telnet.Telnet.__init__(self) self.events = [] def applicationDataReceived(self, data): """ Record the given data in C{self.events}. """ self.events.append(("bytes", data)) def unhandledCommand(self, command, data): """ Record the given command in C{self.events}. """ self.events.append(("command", command, data)) def unhandledSubnegotiation(self, command, data): """ Record the given subnegotiation command in C{self.events}. """ self.events.append(("negotiate", command, data)) class TelnetTests(unittest.TestCase): """ Tests for L{telnet.Telnet}. L{telnet.Telnet} implements the TELNET protocol (RFC 854), including option and suboption negotiation, and option state tracking. """ def setUp(self): """ Create an unconnected L{telnet.Telnet} to be used by tests. """ self.protocol = TestTelnet() def test_enableLocal(self): """ L{telnet.Telnet.enableLocal} should reject all options, since L{telnet.Telnet} does not know how to implement any options. """ self.assertFalse(self.protocol.enableLocal(b"\0")) def test_enableRemote(self): """ L{telnet.Telnet.enableRemote} should reject all options, since L{telnet.Telnet} does not know how to implement any options. """ self.assertFalse(self.protocol.enableRemote(b"\0")) def test_disableLocal(self): """ It is an error for L{telnet.Telnet.disableLocal} to be called, since L{telnet.Telnet.enableLocal} will never allow any options to be enabled locally. If a subclass overrides enableLocal, it must also override disableLocal. """ self.assertRaises(NotImplementedError, self.protocol.disableLocal, b"\0") def test_disableRemote(self): """ It is an error for L{telnet.Telnet.disableRemote} to be called, since L{telnet.Telnet.enableRemote} will never allow any options to be enabled remotely. If a subclass overrides enableRemote, it must also override disableRemote. """ self.assertRaises(NotImplementedError, self.protocol.disableRemote, b"\0") def test_requestNegotiation(self): """ L{telnet.Telnet.requestNegotiation} formats the feature byte and the payload bytes into the subnegotiation format and sends them. See RFC 855. """ transport = proto_helpers.StringTransport() self.protocol.makeConnection(transport) self.protocol.requestNegotiation(b"\x01", b"\x02\x03") self.assertEqual( transport.value(), # IAC SB feature bytes IAC SE b"\xff\xfa\x01\x02\x03\xff\xf0", ) def test_requestNegotiationEscapesIAC(self): """ If the payload for a subnegotiation includes I{IAC}, it is escaped by L{telnet.Telnet.requestNegotiation} with another I{IAC}. See RFC 855. """ transport = proto_helpers.StringTransport() self.protocol.makeConnection(transport) self.protocol.requestNegotiation(b"\x01", b"\xff") self.assertEqual(transport.value(), b"\xff\xfa\x01\xff\xff\xff\xf0") def _deliver(self, data, *expected): """ Pass the given bytes to the protocol's C{dataReceived} method and assert that the given events occur. """ received = self.protocol.events = [] self.protocol.dataReceived(data) self.assertEqual(received, list(expected)) def test_oneApplicationDataByte(self): """ One application-data byte in the default state gets delivered right away. """ self._deliver(b"a", ("bytes", b"a")) def test_twoApplicationDataBytes(self): """ Two application-data bytes in the default state get delivered together. """ self._deliver(b"bc", ("bytes", b"bc")) def test_threeApplicationDataBytes(self): """ Three application-data bytes followed by a control byte get delivered, but the control byte doesn't. """ self._deliver(b"def" + telnet.IAC, ("bytes", b"def")) def test_escapedControl(self): """ IAC in the escaped state gets delivered and so does another application-data byte following it. """ self._deliver(telnet.IAC) self._deliver(telnet.IAC + b"g", ("bytes", telnet.IAC + b"g")) def test_carriageReturn(self): """ A carriage return only puts the protocol into the newline state. A linefeed in the newline state causes just the newline to be delivered. A nul in the newline state causes a carriage return to be delivered. An IAC in the newline state causes a carriage return to be delivered and puts the protocol into the escaped state. Anything else causes a carriage return and that thing to be delivered. """ self._deliver(b"\r") self._deliver(b"\n", ("bytes", b"\n")) self._deliver(b"\r\n", ("bytes", b"\n")) self._deliver(b"\r") self._deliver(b"\0", ("bytes", b"\r")) self._deliver(b"\r\0", ("bytes", b"\r")) self._deliver(b"\r") self._deliver(b"a", ("bytes", b"\ra")) self._deliver(b"\ra", ("bytes", b"\ra")) self._deliver(b"\r") self._deliver( telnet.IAC + telnet.IAC + b"x", ("bytes", b"\r" + telnet.IAC + b"x") ) def test_applicationDataBeforeSimpleCommand(self): """ Application bytes received before a command are delivered before the command is processed. """ self._deliver( b"x" + telnet.IAC + telnet.NOP, ("bytes", b"x"), ("command", telnet.NOP, None), ) def test_applicationDataBeforeCommand(self): """ Application bytes received before a WILL/WONT/DO/DONT are delivered before the command is processed. """ self.protocol.commandMap = {} self._deliver( b"y" + telnet.IAC + telnet.WILL + b"\x00", ("bytes", b"y"), ("command", telnet.WILL, b"\x00"), ) def test_applicationDataBeforeSubnegotiation(self): """ Application bytes received before a subnegotiation command are delivered before the negotiation is processed. """ self._deliver( b"z" + telnet.IAC + telnet.SB + b"Qx" + telnet.IAC + telnet.SE, ("bytes", b"z"), ("negotiate", b"Q", [b"x"]), ) PK F��Z�@D- D- test_default.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.client.default}. """ import sys from unittest import skipIf from twisted.conch.error import ConchError from twisted.conch.test import keydata from twisted.python.compat import nativeString from twisted.python.filepath import FilePath from twisted.python.reflect import requireModule from twisted.python.runtime import platform from twisted.test.proto_helpers import StringTransport from twisted.trial.unittest import TestCase doSkip = False skipReason = "" if requireModule("cryptography") and requireModule("pyasn1"): from twisted.conch.client import default from twisted.conch.client.agent import SSHAgentClient from twisted.conch.client.default import SSHUserAuthClient from twisted.conch.client.options import ConchOptions from twisted.conch.ssh.keys import Key else: doSkip = True skipReason = "cryptography and PyASN1 required for twisted.conch.client.default." skip = skipReason # no SSL available, skip the entire module if platform.isWindows(): doSkip = True skipReason = ( "genericAnswers and getPassword does not work on Windows." " Should be fixed as part of fixing bug 6409 and 6410" ) if not sys.stdin.isatty(): doSkip = True skipReason = "sys.stdin is not an interactive tty" if not sys.stdout.isatty(): doSkip = True skipReason = "sys.stdout is not an interactive tty" class SSHUserAuthClientTests(TestCase): """ Tests for L{SSHUserAuthClient}. @type rsaPublic: L{Key} @ivar rsaPublic: A public RSA key. """ def setUp(self): self.rsaPublic = Key.fromString(keydata.publicRSA_openssh) self.tmpdir = FilePath(self.mktemp()) self.tmpdir.makedirs() self.rsaFile = self.tmpdir.child("id_rsa") self.rsaFile.setContent(keydata.privateRSA_openssh) self.tmpdir.child("id_rsa.pub").setContent(keydata.publicRSA_openssh) def test_signDataWithAgent(self): """ When connected to an agent, L{SSHUserAuthClient} can use it to request signatures of particular data with a particular L{Key}. """ client = SSHUserAuthClient(b"user", ConchOptions(), None) agent = SSHAgentClient() transport = StringTransport() agent.makeConnection(transport) client.keyAgent = agent cleartext = b"Sign here" client.signData(self.rsaPublic, cleartext) self.assertEqual( transport.value(), b"\x00\x00\x01\x2d\r\x00\x00\x01\x17" + self.rsaPublic.blob() + b"\x00\x00\x00\t" + cleartext + b"\x00\x00\x00\x00", ) def test_agentGetPublicKey(self): """ L{SSHUserAuthClient} looks up public keys from the agent using the L{SSHAgentClient} class. That L{SSHAgentClient.getPublicKey} returns a L{Key} object with one of the public keys in the agent. If no more keys are present, it returns L{None}. """ agent = SSHAgentClient() agent.blobs = [self.rsaPublic.blob()] key = agent.getPublicKey() self.assertTrue(key.isPublic()) self.assertEqual(key, self.rsaPublic) self.assertIsNone(agent.getPublicKey()) def test_getPublicKeyFromFile(self): """ L{SSHUserAuthClient.getPublicKey()} is able to get a public key from the first file described by its options' C{identitys} list, and return the corresponding public L{Key} object. """ options = ConchOptions() options.identitys = [self.rsaFile.path] client = SSHUserAuthClient(b"user", options, None) key = client.getPublicKey() self.assertTrue(key.isPublic()) self.assertEqual(key, self.rsaPublic) def test_getPublicKeyAgentFallback(self): """ If an agent is present, but doesn't return a key, L{SSHUserAuthClient.getPublicKey} continue with the normal key lookup. """ options = ConchOptions() options.identitys = [self.rsaFile.path] agent = SSHAgentClient() client = SSHUserAuthClient(b"user", options, None) client.keyAgent = agent key = client.getPublicKey() self.assertTrue(key.isPublic()) self.assertEqual(key, self.rsaPublic) def test_getPublicKeyBadKeyError(self): """ If L{keys.Key.fromFile} raises a L{keys.BadKeyError}, the L{SSHUserAuthClient.getPublicKey} tries again to get a public key by calling itself recursively. """ options = ConchOptions() self.tmpdir.child("id_dsa.pub").setContent(keydata.publicDSA_openssh) dsaFile = self.tmpdir.child("id_dsa") dsaFile.setContent(keydata.privateDSA_openssh) options.identitys = [self.rsaFile.path, dsaFile.path] self.tmpdir.child("id_rsa.pub").setContent(b"not a key!") client = SSHUserAuthClient(b"user", options, None) key = client.getPublicKey() self.assertTrue(key.isPublic()) self.assertEqual(key, Key.fromString(keydata.publicDSA_openssh)) self.assertEqual(client.usedFiles, [self.rsaFile.path, dsaFile.path]) def test_getPrivateKey(self): """ L{SSHUserAuthClient.getPrivateKey} will load a private key from the last used file populated by L{SSHUserAuthClient.getPublicKey}, and return a L{Deferred} which fires with the corresponding private L{Key}. """ rsaPrivate = Key.fromString(keydata.privateRSA_openssh) options = ConchOptions() options.identitys = [self.rsaFile.path] client = SSHUserAuthClient(b"user", options, None) # Populate the list of used files client.getPublicKey() def _cbGetPrivateKey(key): self.assertFalse(key.isPublic()) self.assertEqual(key, rsaPrivate) return client.getPrivateKey().addCallback(_cbGetPrivateKey) def test_getPrivateKeyPassphrase(self): """ L{SSHUserAuthClient} can get a private key from a file, and return a Deferred called back with a private L{Key} object, even if the key is encrypted. """ rsaPrivate = Key.fromString(keydata.privateRSA_openssh) passphrase = b"this is the passphrase" self.rsaFile.setContent(rsaPrivate.toString("openssh", passphrase=passphrase)) options = ConchOptions() options.identitys = [self.rsaFile.path] client = SSHUserAuthClient(b"user", options, None) # Populate the list of used files client.getPublicKey() def _getPassword(prompt): self.assertEqual( prompt, f"Enter passphrase for key '{self.rsaFile.path}': " ) return nativeString(passphrase) def _cbGetPrivateKey(key): self.assertFalse(key.isPublic()) self.assertEqual(key, rsaPrivate) self.patch(client, "_getPassword", _getPassword) return client.getPrivateKey().addCallback(_cbGetPrivateKey) @skipIf(doSkip, skipReason) def test_getPassword(self): """ Get the password using L{twisted.conch.client.default.SSHUserAuthClient.getPassword} """ class FakeTransport: def __init__(self, host): self.transport = self self.host = host def getPeer(self): return self options = ConchOptions() client = SSHUserAuthClient(b"user", options, None) client.transport = FakeTransport("127.0.0.1") def getpass(prompt): self.assertEqual(prompt, "user@127.0.0.1's password: ") return "bad password" self.patch(default.getpass, "getpass", getpass) d = client.getPassword() d.addCallback(self.assertEqual, b"bad password") return d @skipIf(doSkip, skipReason) def test_getPasswordPrompt(self): """ Get the password using L{twisted.conch.client.default.SSHUserAuthClient.getPassword} using a different prompt. """ options = ConchOptions() client = SSHUserAuthClient(b"user", options, None) prompt = b"Give up your password" def getpass(p): self.assertEqual(p, nativeString(prompt)) return "bad password" self.patch(default.getpass, "getpass", getpass) d = client.getPassword(prompt) d.addCallback(self.assertEqual, b"bad password") return d @skipIf(doSkip, skipReason) def test_getPasswordConchError(self): """ Get the password using L{twisted.conch.client.default.SSHUserAuthClient.getPassword} and trigger a {twisted.conch.error import ConchError}. """ options = ConchOptions() client = SSHUserAuthClient(b"user", options, None) def getpass(prompt): raise KeyboardInterrupt("User pressed CTRL-C") self.patch(default.getpass, "getpass", getpass) stdout, stdin = sys.stdout, sys.stdin d = client.getPassword(b"?") @d.addErrback def check_sys(fail): self.assertEqual([stdout, stdin], [sys.stdout, sys.stdin]) return fail self.assertFailure(d, ConchError) @skipIf(doSkip, skipReason) def test_getGenericAnswers(self): """ L{twisted.conch.client.default.SSHUserAuthClient.getGenericAnswers} """ options = ConchOptions() client = SSHUserAuthClient(b"user", options, None) def getpass(prompt): self.assertEqual(prompt, "pass prompt") return "getpass" self.patch(default.getpass, "getpass", getpass) def raw_input(prompt): self.assertEqual(prompt, "raw_input prompt") return "raw_input" self.patch(default, "_input", raw_input) d = client.getGenericAnswers( b"Name", b"Instruction", [(b"pass prompt", False), (b"raw_input prompt", True)], ) d.addCallback(self.assertListEqual, ["getpass", "raw_input"]) return d class ConchOptionsParsing(TestCase): """ Options parsing. """ def test_macs(self): """ Specify MAC algorithms. """ opts = ConchOptions() e = self.assertRaises(SystemExit, opts.opt_macs, "invalid-mac") self.assertIn("Unknown mac type", e.code) opts = ConchOptions() opts.opt_macs("hmac-sha2-512") self.assertEqual(opts["macs"], [b"hmac-sha2-512"]) opts.opt_macs(b"hmac-sha2-512") self.assertEqual(opts["macs"], [b"hmac-sha2-512"]) opts.opt_macs("hmac-sha2-256,hmac-sha1,hmac-md5") self.assertEqual(opts["macs"], [b"hmac-sha2-256", b"hmac-sha1", b"hmac-md5"]) def test_host_key_algorithms(self): """ Specify host key algorithms. """ opts = ConchOptions() e = self.assertRaises(SystemExit, opts.opt_host_key_algorithms, "invalid-key") self.assertIn("Unknown host key type", e.code) opts = ConchOptions() opts.opt_host_key_algorithms("ssh-rsa") self.assertEqual(opts["host-key-algorithms"], [b"ssh-rsa"]) opts.opt_host_key_algorithms(b"ssh-dss") self.assertEqual(opts["host-key-algorithms"], [b"ssh-dss"]) opts.opt_host_key_algorithms("ssh-rsa,ssh-dss") self.assertEqual(opts["host-key-algorithms"], [b"ssh-rsa", b"ssh-dss"]) PK F��Z�B9� � test_tap.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.tap}. """ from twisted.application.internet import StreamServerEndpointService from twisted.cred import error from twisted.cred.credentials import ISSHPrivateKey, IUsernamePassword, UsernamePassword from twisted.python.reflect import requireModule from twisted.trial.unittest import TestCase cryptography = requireModule("cryptography") pyasn1 = requireModule("pyasn1") unix = requireModule("twisted.conch.unix") if cryptography and pyasn1 and unix: from twisted.conch import tap from twisted.conch.openssh_compat.factory import OpenSSHFactory class MakeServiceTests(TestCase): """ Tests for L{tap.makeService}. """ if not cryptography: skip = "can't run without cryptography" if not pyasn1: skip = "Cannot run without PyASN1" if not unix: skip = "can't run on non-posix computers" usernamePassword = (b"iamuser", b"thisispassword") def setUp(self): """ Create a file with two users. """ self.filename = self.mktemp() with open(self.filename, "wb+") as f: f.write(b":".join(self.usernamePassword)) self.options = tap.Options() def test_basic(self): """ L{tap.makeService} returns a L{StreamServerEndpointService} instance running on TCP port 22, and the linked protocol factory is an instance of L{OpenSSHFactory}. """ config = tap.Options() service = tap.makeService(config) self.assertIsInstance(service, StreamServerEndpointService) self.assertEqual(service.endpoint._port, 22) self.assertIsInstance(service.factory, OpenSSHFactory) def test_defaultAuths(self): """ Make sure that if the C{--auth} command-line option is not passed, the default checkers are (for backwards compatibility): SSH and UNIX """ numCheckers = 2 self.assertIn( ISSHPrivateKey, self.options["credInterfaces"], "SSH should be one of the default checkers", ) self.assertIn( IUsernamePassword, self.options["credInterfaces"], "UNIX should be one of the default checkers", ) self.assertEqual( numCheckers, len(self.options["credCheckers"]), "There should be %d checkers by default" % (numCheckers,), ) def test_authAdded(self): """ The C{--auth} command-line option will add a checker to the list of checkers, and it should be the only auth checker """ self.options.parseOptions(["--auth", "file:" + self.filename]) self.assertEqual(len(self.options["credCheckers"]), 1) def test_multipleAuthAdded(self): """ Multiple C{--auth} command-line options will add all checkers specified to the list ofcheckers, and there should only be the specified auth checkers (no default checkers). """ self.options.parseOptions( [ "--auth", "file:" + self.filename, "--auth", "memory:testuser:testpassword", ] ) self.assertEqual(len(self.options["credCheckers"]), 2) def test_authFailure(self): """ The checker created by the C{--auth} command-line option returns a L{Deferred} that fails with L{UnauthorizedLogin} when presented with credentials that are unknown to that checker. """ self.options.parseOptions(["--auth", "file:" + self.filename]) checker = self.options["credCheckers"][-1] invalid = UsernamePassword(self.usernamePassword[0], "fake") # Wrong password should raise error return self.assertFailure( checker.requestAvatarId(invalid), error.UnauthorizedLogin ) def test_authSuccess(self): """ The checker created by the C{--auth} command-line option returns a L{Deferred} that returns the avatar id when presented with credentials that are known to that checker. """ self.options.parseOptions(["--auth", "file:" + self.filename]) checker = self.options["credCheckers"][-1] correct = UsernamePassword(*self.usernamePassword) d = checker.requestAvatarId(correct) def checkSuccess(username): self.assertEqual(username, correct.username) return d.addCallback(checkSuccess) def test_checkers(self): """ The L{OpenSSHFactory} built by L{tap.makeService} has a portal with L{ISSHPrivateKey} and L{IUsernamePassword} interfaces registered as checkers. """ config = tap.Options() service = tap.makeService(config) portal = service.factory.portal self.assertEqual( set(portal.checkers.keys()), {ISSHPrivateKey, IUsernamePassword} ) PK F��Z>�t) test_openssh_compat.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.openssh_compat}. """ import os from unittest import skipIf from twisted.conch.ssh._kex import getDHGeneratorAndPrime from twisted.conch.test import keydata from twisted.python.filepath import FilePath from twisted.python.reflect import requireModule from twisted.test.test_process import MockOS from twisted.trial.unittest import TestCase doSkip = False skipReason = "" if requireModule("cryptography") and requireModule("pyasn1"): from twisted.conch.openssh_compat.factory import OpenSSHFactory else: doSkip = True skipReason = "Cannot run without cryptography or PyASN1" if not hasattr(os, "geteuid"): doSkip = True skipReason = "geteuid/seteuid not available" @skipIf(doSkip, skipReason) class OpenSSHFactoryTests(TestCase): """ Tests for L{OpenSSHFactory}. """ def setUp(self): self.factory = OpenSSHFactory() self.keysDir = FilePath(self.mktemp()) self.keysDir.makedirs() self.factory.dataRoot = self.keysDir.path self.moduliDir = FilePath(self.mktemp()) self.moduliDir.makedirs() self.factory.moduliRoot = self.moduliDir.path self.keysDir.child("ssh_host_foo").setContent(b"foo") self.keysDir.child("bar_key").setContent(b"foo") self.keysDir.child("ssh_host_one_key").setContent(keydata.privateRSA_openssh) self.keysDir.child("ssh_host_two_key").setContent(keydata.privateDSA_openssh) self.keysDir.child("ssh_host_three_key").setContent(b"not a key content") self.keysDir.child("ssh_host_one_key.pub").setContent(keydata.publicRSA_openssh) self.moduliDir.child("moduli").setContent( b"\n" b"# $OpenBSD: moduli,v 1.xx 2016/07/26 12:34:56 jhacker Exp $i\n" b"# Time Type Tests Tries Size Generator Modulus\n" b"20030501000000 2 6 100 2047 2 " b"FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74" b"020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F1437" b"4FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED" b"EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF05" b"98DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB" b"9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3B" b"E39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF695581718" b"3995497CEA956AE515D2261898FA051015728E5A8AACAA68FFFFFFFFFFFFFFFF" b"\n" ) self.mockos = MockOS() self.patch(os, "seteuid", self.mockos.seteuid) self.patch(os, "setegid", self.mockos.setegid) def test_getPublicKeys(self): """ L{OpenSSHFactory.getPublicKeys} should return the available public keys in the data directory """ keys = self.factory.getPublicKeys() self.assertEqual(len(keys), 1) keyTypes = keys.keys() self.assertEqual(list(keyTypes), [b"ssh-rsa"]) def test_getPrivateKeys(self): """ Will return the available private keys in the data directory, ignoring key files which failed to be loaded. """ keys = self.factory.getPrivateKeys() self.assertEqual(len(keys), 2) keyTypes = keys.keys() self.assertEqual(set(keyTypes), {b"ssh-rsa", b"ssh-dss"}) self.assertEqual(self.mockos.seteuidCalls, []) self.assertEqual(self.mockos.setegidCalls, []) def test_getPrivateKeysAsRoot(self): """ L{OpenSSHFactory.getPrivateKeys} should switch to root if the keys aren't readable by the current user. """ keyFile = self.keysDir.child("ssh_host_two_key") # Fake permission error by changing the mode keyFile.chmod(0000) self.addCleanup(keyFile.chmod, 0o777) # And restore the right mode when seteuid is called savedSeteuid = os.seteuid def seteuid(euid): keyFile.chmod(0o777) return savedSeteuid(euid) self.patch(os, "seteuid", seteuid) keys = self.factory.getPrivateKeys() self.assertEqual(len(keys), 2) keyTypes = keys.keys() self.assertEqual(set(keyTypes), {b"ssh-rsa", b"ssh-dss"}) self.assertEqual(self.mockos.seteuidCalls, [0, os.geteuid()]) self.assertEqual(self.mockos.setegidCalls, [0, os.getegid()]) def test_getPrimes(self): """ L{OpenSSHFactory.getPrimes} should return the available primes in the moduli directory. """ primes = self.factory.getPrimes() self.assertEqual( primes, { 2048: [getDHGeneratorAndPrime(b"diffie-hellman-group14-sha1")], }, ) PK F��Z��n�7� 7� test_keys.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.ssh.keys}. """ import base64 import os from textwrap import dedent from twisted.conch.test import keydata from twisted.python import randbytes from twisted.python.filepath import FilePath from twisted.python.reflect import requireModule from twisted.trial import unittest cryptography = requireModule("cryptography") if cryptography is None: skipCryptography = "Cannot run without cryptography." pyasn1 = requireModule("pyasn1") if cryptography and pyasn1: from cryptography.hazmat.backends import default_backend from twisted.conch.ssh import common, keys, sexpy ED25519_SUPPORTED = default_backend().ed25519_supported() else: ED25519_SUPPORTED = False def skipWithoutEd25519(f): if not ED25519_SUPPORTED: f.skip = "ed25519 not supported on this system" return f class KeyTests(unittest.TestCase): if cryptography is None: skip = skipCryptography if pyasn1 is None: skip = "Cannot run without PyASN1" def setUp(self): self.rsaObj = keys.Key._fromRSAComponents( n=keydata.RSAData["n"], e=keydata.RSAData["e"], d=keydata.RSAData["d"], p=keydata.RSAData["p"], q=keydata.RSAData["q"], u=keydata.RSAData["u"], )._keyObject self.dsaObj = keys.Key._fromDSAComponents( y=keydata.DSAData["y"], p=keydata.DSAData["p"], q=keydata.DSAData["q"], g=keydata.DSAData["g"], x=keydata.DSAData["x"], )._keyObject self.ecObj = keys.Key._fromECComponents( x=keydata.ECDatanistp256["x"], y=keydata.ECDatanistp256["y"], privateValue=keydata.ECDatanistp256["privateValue"], curve=keydata.ECDatanistp256["curve"], )._keyObject self.ecObj384 = keys.Key._fromECComponents( x=keydata.ECDatanistp384["x"], y=keydata.ECDatanistp384["y"], privateValue=keydata.ECDatanistp384["privateValue"], curve=keydata.ECDatanistp384["curve"], )._keyObject self.ecObj521 = keys.Key._fromECComponents( x=keydata.ECDatanistp521["x"], y=keydata.ECDatanistp521["y"], privateValue=keydata.ECDatanistp521["privateValue"], curve=keydata.ECDatanistp521["curve"], )._keyObject if ED25519_SUPPORTED: self.ed25519Obj = keys.Key._fromEd25519Components( a=keydata.Ed25519Data["a"], k=keydata.Ed25519Data["k"] )._keyObject self.rsaSignature = ( b"\x00\x00\x00\x07ssh-rsa\x00\x00\x01\x00~Y\xa3\xd7\xfdW\xc6pu@" b"\xd81\xa1S\xf3O\xdaE\xf4/\x1ex\x1d\xf1\x9a\xe1G3\xd9\xd6U\x1f" b"\x8c\xd9\x1b\x8b\x90\x0e\x8a\xc1\x91\xd8\x0cd\xc9\x0c\xe7\xb2" b"\xc9,'=\x15\x1cQg\xe7x\xb5j\xdbI\xc0\xde\xafb\xd7@\xcar\x0b" b"\xce\xa3zM\x151q5\xde\xfa\x0c{wjKN\x88\xcbC\xe5\x89\xc3\xf9i" b"\x96\x91\xdb\xca}\xdbR\x1a\x13T\xf9\x0cDJH\x0b\x06\xcfl\xf3" b"\x13[\x82\xa2\x9d\x93\xfd\x8e\xce|\xfb^n\xd4\xed\xe2\xd1\x8a" b"\xb7aY\x9bB\x8f\xa4\xc7\xbe7\xb5\x0b9j\xa4.\x87\x13\xf7\xf0" b"\xda\xd7\xd2\xf9\x1f9p\xfd?\x18\x0f\xf2N\x9b\xcf/\x1e)\n>A\x19" b"\xc2\xb5j\xf9UW\xd4\xae\x87B\xe6\x99t\xa2y\x90\x98\xa2\xaaf\xcb" b"\x86\xe5k\xe3\xce\xe0u\x1c\xeb\x93\x1aN\x88\xc9\x93Y\xc3.V\xb1L" b"44`C\xc7\xa66\xaf\xfa\x7f\x04Y\x92\xfa\xa4\x1a\x18%\x19\xd5 4^" b"\xb9rY\xba \x01\xf9.\x89%H\xbe\x1c\x83A\x96" ) self.dsaSignature = ( b"\x00\x00\x00\x07ssh-dss\x00\x00\x00(?\xc7\xeb\x86;\xd5TFA\xb4" b"\xdf\x0c\xc4E@4,d\xbc\t\xd9\xae\xdd[\xed-\x82nQ\x8cf\x9b\xe8\xe1" b"jrg\x84p<" ) self.patch(randbytes, "secureRandom", lambda x: b"\xff" * x) self.keyFile = self.mktemp() with open(self.keyFile, "wb") as f: f.write(keydata.privateRSA_lsh) def tearDown(self): os.unlink(self.keyFile) def test_size(self): """ The L{keys.Key.size} method returns the size of key object in bits. """ self.assertEqual(keys.Key(self.rsaObj).size(), 2048) self.assertEqual(keys.Key(self.dsaObj).size(), 1024) self.assertEqual(keys.Key(self.ecObj).size(), 256) self.assertEqual(keys.Key(self.ecObj384).size(), 384) self.assertEqual(keys.Key(self.ecObj521).size(), 521) if ED25519_SUPPORTED: self.assertEqual(keys.Key(self.ed25519Obj).size(), 256) def test__guessStringType(self): """ Test that the _guessStringType method guesses string types correctly. """ self.assertEqual( keys.Key._guessStringType(keydata.publicRSA_openssh), "public_openssh" ) self.assertEqual( keys.Key._guessStringType(keydata.publicDSA_openssh), "public_openssh" ) self.assertEqual( keys.Key._guessStringType(keydata.publicECDSA_openssh), "public_openssh" ) if ED25519_SUPPORTED: self.assertEqual( keys.Key._guessStringType(keydata.publicEd25519_openssh), "public_openssh", ) self.assertEqual( keys.Key._guessStringType(keydata.privateRSA_openssh), "private_openssh" ) self.assertEqual( keys.Key._guessStringType(keydata.privateRSA_openssh_new), "private_openssh" ) self.assertEqual( keys.Key._guessStringType(keydata.privateDSA_openssh), "private_openssh" ) self.assertEqual( keys.Key._guessStringType(keydata.privateDSA_openssh_new), "private_openssh" ) self.assertEqual( keys.Key._guessStringType(keydata.privateECDSA_openssh), "private_openssh" ) self.assertEqual( keys.Key._guessStringType(keydata.privateECDSA_openssh_new), "private_openssh", ) if ED25519_SUPPORTED: self.assertEqual( keys.Key._guessStringType(keydata.privateEd25519_openssh_new), "private_openssh", ) self.assertEqual(keys.Key._guessStringType(keydata.publicRSA_lsh), "public_lsh") self.assertEqual(keys.Key._guessStringType(keydata.publicDSA_lsh), "public_lsh") self.assertEqual( keys.Key._guessStringType(keydata.privateRSA_lsh), "private_lsh" ) self.assertEqual( keys.Key._guessStringType(keydata.privateDSA_lsh), "private_lsh" ) self.assertEqual( keys.Key._guessStringType(keydata.privateRSA_agentv3), "agentv3" ) self.assertEqual( keys.Key._guessStringType(keydata.privateDSA_agentv3), "agentv3" ) self.assertEqual( keys.Key._guessStringType(b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00\x01\x01"), "blob", ) self.assertEqual( keys.Key._guessStringType(b"\x00\x00\x00\x07ssh-dss\x00\x00\x00\x01\x01"), "blob", ) self.assertEqual(keys.Key._guessStringType(b"not a key"), None) def test_public(self): """ The L{keys.Key.public} method returns a public key for both public and private keys. """ # NB: This assumes that the private and public keys correspond # to each other. privateRSAKey = keys.Key.fromString(keydata.privateRSA_openssh) publicRSAKey = keys.Key.fromString(keydata.publicRSA_openssh) self.assertEqual(privateRSAKey.public(), publicRSAKey.public()) privateDSAKey = keys.Key.fromString(keydata.privateDSA_openssh) publicDSAKey = keys.Key.fromString(keydata.publicDSA_openssh) self.assertEqual(privateDSAKey.public(), publicDSAKey.public()) privateECDSAKey = keys.Key.fromString(keydata.privateECDSA_openssh) publicECDSAKey = keys.Key.fromString(keydata.publicECDSA_openssh) self.assertEqual(privateECDSAKey.public(), publicECDSAKey.public()) if ED25519_SUPPORTED: privateEd25519Key = keys.Key.fromString(keydata.privateEd25519_openssh_new) publicEd25519Key = keys.Key.fromString(keydata.publicEd25519_openssh) self.assertEqual(privateEd25519Key.public(), publicEd25519Key.public()) def test_isPublic(self): """ The L{keys.Key.isPublic} method returns True for public keys otherwise False. """ rsaKey = keys.Key.fromString(keydata.privateRSA_openssh) dsaKey = keys.Key.fromString(keydata.privateDSA_openssh) ecdsaKey = keys.Key.fromString(keydata.privateECDSA_openssh) self.assertTrue(rsaKey.public().isPublic()) self.assertFalse(rsaKey.isPublic()) self.assertTrue(dsaKey.public().isPublic()) self.assertFalse(dsaKey.isPublic()) self.assertTrue(ecdsaKey.public().isPublic()) self.assertFalse(ecdsaKey.isPublic()) if ED25519_SUPPORTED: ed25519Key = keys.Key.fromString(keydata.privateEd25519_openssh_new) self.assertTrue(ed25519Key.public().isPublic()) self.assertFalse(ed25519Key.isPublic()) def _testPublicPrivateFromString(self, public, private, type, data): self._testPublicFromString(public, type, data) self._testPrivateFromString(private, type, data) def _testPublicFromString(self, public, type, data): publicKey = keys.Key.fromString(public) self.assertTrue(publicKey.isPublic()) self.assertEqual(publicKey.type(), type) for k, v in publicKey.data().items(): self.assertEqual(data[k], v) def _testPrivateFromString(self, private, type, data): privateKey = keys.Key.fromString(private) self.assertFalse(privateKey.isPublic()) self.assertEqual(privateKey.type(), type) for k, v in data.items(): self.assertEqual(privateKey.data()[k], v) def test_fromOpenSSH(self): """ Test that keys are correctly generated from OpenSSH strings. """ self._testPublicPrivateFromString( keydata.publicECDSA_openssh, keydata.privateECDSA_openssh, "EC", keydata.ECDatanistp256, ) self._testPublicPrivateFromString( keydata.publicRSA_openssh, keydata.privateRSA_openssh, "RSA", keydata.RSAData, ) self.assertEqual( keys.Key.fromString( keydata.privateRSA_openssh_encrypted, passphrase=b"encrypted" ), keys.Key.fromString(keydata.privateRSA_openssh), ) self.assertEqual( keys.Key.fromString(keydata.privateRSA_openssh_alternate), keys.Key.fromString(keydata.privateRSA_openssh), ) self._testPublicPrivateFromString( keydata.publicDSA_openssh, keydata.privateDSA_openssh, "DSA", keydata.DSAData, ) if ED25519_SUPPORTED: self._testPublicPrivateFromString( keydata.publicEd25519_openssh, keydata.privateEd25519_openssh_new, "Ed25519", keydata.Ed25519Data, ) def test_fromOpenSSHErrors(self): """ Tests for invalid key types. """ badKey = b"""-----BEGIN FOO PRIVATE KEY----- MIGkAgEBBDAtAi7I8j73WCX20qUM5hhHwHuFzYWYYILs2Sh8UZ+awNkARZ/Fu2LU LLl5RtOQpbWgBwYFK4EEACKhZANiAATU17sA9P5FRwSknKcFsjjsk0+E3CeXPYX0 Tk/M0HK3PpWQWgrO8JdRHP9eFE9O/23P8BumwFt7F/AvPlCzVd35VfraFT0o4cCW G0RqpQ+np31aKmeJshkcYALEchnU+tQ= -----END EC PRIVATE KEY-----""" self.assertRaises( keys.BadKeyError, keys.Key._fromString_PRIVATE_OPENSSH, badKey, None ) def test_fromOpenSSH_with_whitespace(self): """ If key strings have trailing whitespace, it should be ignored. """ # from bug #3391, since our test key data doesn't have # an issue with appended newlines privateDSAData = b"""-----BEGIN DSA PRIVATE KEY----- MIIBuwIBAAKBgQDylESNuc61jq2yatCzZbenlr9llG+p9LhIpOLUbXhhHcwC6hrh EZIdCKqTO0USLrGoP5uS9UHAUoeN62Z0KXXWTwOWGEQn/syyPzNJtnBorHpNUT9D Qzwl1yUa53NNgEctpo4NoEFOx8PuU6iFLyvgHCjNn2MsuGuzkZm7sI9ZpQIVAJiR 9dPc08KLdpJyRxz8T74b4FQRAoGAGBc4Z5Y6R/HZi7AYM/iNOM8su6hrk8ypkBwR a3Dbhzk97fuV3SF1SDrcQu4zF7c4CtH609N5nfZs2SUjLLGPWln83Ysb8qhh55Em AcHXuROrHS/sDsnqu8FQp86MaudrqMExCOYyVPE7jaBWW+/JWFbKCxmgOCSdViUJ esJpBFsCgYEA7+jtVvSt9yrwsS/YU1QGP5wRAiDYB+T5cK4HytzAqJKRdC5qS4zf C7R0eKcDHHLMYO39aPnCwXjscisnInEhYGNblTDyPyiyNxAOXuC8x7luTmwzMbNJ /ow0IqSj0VF72VJN9uSoPpFd4lLT0zN8v42RWja0M8ohWNf+YNJluPgCFE0PT4Vm SUrCyZXsNh6VXwjs3gKQ -----END DSA PRIVATE KEY-----""" self.assertEqual( keys.Key.fromString(privateDSAData), keys.Key.fromString(privateDSAData + b"\n"), ) def test_fromNewerOpenSSH(self): """ Newer versions of OpenSSH generate encrypted keys which have a longer IV than the older versions. These newer keys are also loaded. """ key = keys.Key.fromString( keydata.privateRSA_openssh_encrypted_aes, passphrase=b"testxp" ) self.assertEqual(key.type(), "RSA") key2 = keys.Key.fromString( keydata.privateRSA_openssh_encrypted_aes + b"\n", passphrase=b"testxp" ) self.assertEqual(key, key2) def test_fromOpenSSH_v1_format(self): """ OpenSSH 6.5 introduced a newer "openssh-key-v1" private key format (made the default in OpenSSH 7.8). Loading keys in this format produces identical results to loading the same keys in the old PEM-based format. """ for old, new in ( (keydata.privateRSA_openssh, keydata.privateRSA_openssh_new), (keydata.privateDSA_openssh, keydata.privateDSA_openssh_new), (keydata.privateECDSA_openssh, keydata.privateECDSA_openssh_new), (keydata.privateECDSA_openssh384, keydata.privateECDSA_openssh384_new), (keydata.privateECDSA_openssh521, keydata.privateECDSA_openssh521_new), ): self.assertEqual(keys.Key.fromString(new), keys.Key.fromString(old)) self.assertEqual( keys.Key.fromString( keydata.privateRSA_openssh_encrypted_new, passphrase=b"encrypted" ), keys.Key.fromString( keydata.privateRSA_openssh_encrypted, passphrase=b"encrypted" ), ) def test_fromOpenSSH_windows_line_endings(self): """ Test that keys are correctly generated from OpenSSH strings with Windows line endings. """ privateDSAData = b"""-----BEGIN DSA PRIVATE KEY----- MIIBuwIBAAKBgQDylESNuc61jq2yatCzZbenlr9llG+p9LhIpOLUbXhhHcwC6hrh EZIdCKqTO0USLrGoP5uS9UHAUoeN62Z0KXXWTwOWGEQn/syyPzNJtnBorHpNUT9D Qzwl1yUa53NNgEctpo4NoEFOx8PuU6iFLyvgHCjNn2MsuGuzkZm7sI9ZpQIVAJiR 9dPc08KLdpJyRxz8T74b4FQRAoGAGBc4Z5Y6R/HZi7AYM/iNOM8su6hrk8ypkBwR a3Dbhzk97fuV3SF1SDrcQu4zF7c4CtH609N5nfZs2SUjLLGPWln83Ysb8qhh55Em AcHXuROrHS/sDsnqu8FQp86MaudrqMExCOYyVPE7jaBWW+/JWFbKCxmgOCSdViUJ esJpBFsCgYEA7+jtVvSt9yrwsS/YU1QGP5wRAiDYB+T5cK4HytzAqJKRdC5qS4zf C7R0eKcDHHLMYO39aPnCwXjscisnInEhYGNblTDyPyiyNxAOXuC8x7luTmwzMbNJ /ow0IqSj0VF72VJN9uSoPpFd4lLT0zN8v42RWja0M8ohWNf+YNJluPgCFE0PT4Vm SUrCyZXsNh6VXwjs3gKQ -----END DSA PRIVATE KEY-----""" self.assertEqual( keys.Key.fromString(privateDSAData), keys.Key.fromString(privateDSAData.replace(b"\n", b"\r\n")), ) def test_fromLSHPublicUnsupportedType(self): """ C{BadKeyError} exception is raised when public key has an unknown type. """ sexp = sexpy.pack([[b"public-key", [b"bad-key", [b"p", b"2"]]]]) self.assertRaises( keys.BadKeyError, keys.Key.fromString, data=b"{" + base64.b64encode(sexp) + b"}", ) def test_fromLSHPrivateUnsupportedType(self): """ C{BadKeyError} exception is raised when private key has an unknown type. """ sexp = sexpy.pack([[b"private-key", [b"bad-key", [b"p", b"2"]]]]) self.assertRaises( keys.BadKeyError, keys.Key.fromString, sexp, ) def test_fromLSHRSA(self): """ RSA public and private keys can be generated from a LSH strings. """ self._testPublicPrivateFromString( keydata.publicRSA_lsh, keydata.privateRSA_lsh, "RSA", keydata.RSAData, ) def test_fromLSHDSA(self): """ DSA public and private key can be generated from LSHs. """ self._testPublicPrivateFromString( keydata.publicDSA_lsh, keydata.privateDSA_lsh, "DSA", keydata.DSAData, ) def test_fromAgentv3(self): """ Test that keys are correctly generated from Agent v3 strings. """ self._testPrivateFromString(keydata.privateRSA_agentv3, "RSA", keydata.RSAData) self._testPrivateFromString(keydata.privateDSA_agentv3, "DSA", keydata.DSAData) self.assertRaises( keys.BadKeyError, keys.Key.fromString, b"\x00\x00\x00\x07ssh-foo" + b"\x00\x00\x00\x01\x01" * 5, ) def test_fromStringNormalizesUnicodePassphrase(self): """ L{keys.Key.fromString} applies Normalization Form KC to Unicode passphrases. """ key = keys.Key(self.rsaObj) key_data = key.toString("openssh", passphrase="verschl\u00FCsselt".encode()) self.assertEqual( keys.Key.fromString(key_data, passphrase="verschlu\u0308sselt"), key ) # U+FFFF is a "noncharacter" and guaranteed to have General_Category # Cn (Unassigned). self.assertRaises( keys.PassphraseNormalizationError, keys.Key.fromString, key_data, passphrase="unassigned \uFFFF", ) def test_fromStringErrors(self): """ keys.Key.fromString should raise BadKeyError when the key is invalid. """ self.assertRaises(keys.BadKeyError, keys.Key.fromString, b"") # no key data with a bad key type self.assertRaises(keys.BadKeyError, keys.Key.fromString, b"", "bad_type") # trying to decrypt a key which doesn't support encryption self.assertRaises( keys.BadKeyError, keys.Key.fromString, keydata.publicRSA_lsh, passphrase=b"unencrypted", ) # trying to decrypt a key with the wrong passphrase self.assertRaises( keys.EncryptedKeyError, keys.Key.fromString, keys.Key(self.rsaObj).toString("openssh", passphrase=b"encrypted"), ) # key with no key data self.assertRaises( keys.BadKeyError, keys.Key.fromString, b"-----BEGIN RSA KEY-----\nwA==\n" ) # key with invalid DEK Info self.assertRaises( keys.BadKeyError, keys.Key.fromString, b"""-----BEGIN ENCRYPTED RSA KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: weird type 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----""", passphrase="encrypted", ) # key with invalid encryption type self.assertRaises( keys.BadKeyError, keys.Key.fromString, b"""-----BEGIN ENCRYPTED RSA KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: FOO-123-BAR,01234567 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----""", passphrase="encrypted", ) # key with bad IV (AES) self.assertRaises( keys.BadKeyError, keys.Key.fromString, b"""-----BEGIN ENCRYPTED RSA KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: AES-128-CBC,01234 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----""", passphrase="encrypted", ) # key with bad IV (DES3) self.assertRaises( keys.BadKeyError, keys.Key.fromString, b"""-----BEGIN ENCRYPTED RSA KEY----- Proc-Type: 4,ENCRYPTED DEK-Info: DES-EDE3-CBC,01234 4Ed/a9OgJWHJsne7yOGWeWMzHYKsxuP9w1v0aYcp+puS75wvhHLiUnNwxz0KDi6n T3YkKLBsoCWS68ApR2J9yeQ6R+EyS+UQDrO9nwqo3DB5BT3Ggt8S1wE7vjNLQD0H g/SJnlqwsECNhh8aAx+Ag0m3ZKOZiRD5mCkcDQsZET7URSmFytDKOjhFn3u6ZFVB sXrfpYc6TJtOQlHd/52JB6aAbjt6afSv955Z7enIi+5yEJ5y7oYQTaE5zrFMP7N5 9LbfJFlKXxEddy/DErRLxEjmC+t4svHesoJKc2jjjyNPiOoGGF3kJXea62vsjdNV gMK5Eged3TBVIk2dv8rtJUvyFeCUtjQ1UJZIebScRR47KrbsIpCmU8I4/uHWm5hW 0mOwvdx1L/mqx/BHqVU9Dw2COhOdLbFxlFI92chkovkmNk4P48ziyVnpm7ME22sE vfCMsyirdqB1mrL4CSM7FXONv+CgfBfeYVkYW8RfJac9U1L/O+JNn7yee414O/rS hRYw4UdWnH6Gg6niklVKWNY0ZwUZC8zgm2iqy8YCYuneS37jC+OEKP+/s6HSKuqk 2bzcl3/TcZXNSM815hnFRpz0anuyAsvwPNRyvxG2/DacJHL1f6luV4B0o6W410yf qXQx01DLo7nuyhJqoH3UGCyyXB+/QUs0mbG2PAEn3f5dVs31JMdbt+PrxURXXjKk 4cexpUcIpqqlfpIRe3RD0sDVbH4OXsGhi2kiTfPZu7mgyFxKopRbn1KwU1qKinfY EU9O4PoTak/tPT+5jFNhaP+HrURoi/pU8EAUNSktl7xAkHYwkN/9Cm7DeBghgf3n 8+tyCGYDsB5utPD0/Xe9yx0Qhc/kMm4xIyQDyA937dk3mUvLC9vulnAP8I+Izim0 fZ182+D1bWwykoD0997mUHG/AUChWR01V1OLwRyPv2wUtiS8VNG76Y2aqKlgqP1P V+IvIEqR4ERvSBVFzXNF8Y6j/sVxo8+aZw+d0L1Ns/R55deErGg3B8i/2EqGd3r+ 0jps9BqFHHWW87n3VyEB3jWCMj8Vi2EJIfa/7pSaViFIQn8LiBLf+zxG5LTOToK5 xkN42fReDcqi3UNfKNGnv4dsplyTR2hyx65lsj4bRKDGLKOuB1y7iB0AGb0LtcAI dcsVlcCeUquDXtqKvRnwfIMg+ZunyjqHBhj3qgRgbXbT6zjaSdNnih569aTg0Vup VykzZ7+n/KVcGLmvX0NesdoI7TKbq4TnEIOynuG5Sf+2GpARO5bjcWKSZeN/Ybgk gccf8Cqf6XWqiwlWd0B7BR3SymeHIaSymC45wmbgdstrbk7Ppa2Tp9AZku8M2Y7c 8mY9b+onK075/ypiwBm4L4GRNTFLnoNQJXx0OSl4FNRWsn6ztbD+jZhu8Seu10Jw SEJVJ+gmTKdRLYORJKyqhDet6g7kAxs4EoJ25WsOnX5nNr00rit+NkMPA7xbJT+7 CfI51GQLw7pUPeO2WNt6yZO/YkzZrqvTj5FEwybkUyBv7L0gkqu9wjfDdUw0fVHE xEm4DxjEoaIp8dW/JOzXQ2EF+WaSOgdYsw3Ac+rnnjnNptCdOEDGP6QBkt+oXj4P -----END RSA PRIVATE KEY-----""", passphrase="encrypted", ) def test_fromFile(self): """ Test that fromFile works correctly. """ self.assertEqual( keys.Key.fromFile(self.keyFile), keys.Key.fromString(keydata.privateRSA_lsh) ) self.assertRaises(keys.BadKeyError, keys.Key.fromFile, self.keyFile, "bad_type") self.assertRaises( keys.BadKeyError, keys.Key.fromFile, self.keyFile, passphrase="unencrypted" ) def test_init(self): """ Test that the PublicKey object is initialized correctly. """ obj = keys.Key._fromRSAComponents(n=5, e=3)._keyObject key = keys.Key(obj) self.assertEqual(key._keyObject, obj) def test_equal(self): """ Test that Key objects are compared correctly. """ rsa1 = keys.Key(self.rsaObj) rsa2 = keys.Key(self.rsaObj) rsa3 = keys.Key(keys.Key._fromRSAComponents(n=5, e=3)._keyObject) dsa = keys.Key(self.dsaObj) self.assertTrue(rsa1 == rsa2) self.assertFalse(rsa1 == rsa3) self.assertFalse(rsa1 == dsa) self.assertFalse(rsa1 == object) self.assertFalse(rsa1 == None) def test_notEqual(self): """ Test that Key objects are not-compared correctly. """ rsa1 = keys.Key(self.rsaObj) rsa2 = keys.Key(self.rsaObj) rsa3 = keys.Key(keys.Key._fromRSAComponents(n=5, e=3)._keyObject) dsa = keys.Key(self.dsaObj) self.assertFalse(rsa1 != rsa2) self.assertTrue(rsa1 != rsa3) self.assertTrue(rsa1 != dsa) self.assertTrue(rsa1 != object) self.assertTrue(rsa1 != None) def test_dataError(self): """ The L{keys.Key.data} method raises RuntimeError for bad keys. """ badKey = keys.Key(b"") self.assertRaises(RuntimeError, badKey.data) def test_fingerprintdefault(self): """ Test that the fingerprint method returns fingerprint in L{FingerprintFormats.MD5-HEX} format by default. """ self.assertEqual( keys.Key(self.rsaObj).fingerprint(), "85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:da", ) self.assertEqual( keys.Key(self.dsaObj).fingerprint(), "63:15:b3:0e:e6:4f:50:de:91:48:3d:01:6b:b3:13:c1", ) def test_fingerprint_md5_hex(self): """ fingerprint method generates key fingerprint in L{FingerprintFormats.MD5-HEX} format if explicitly specified. """ self.assertEqual( keys.Key(self.rsaObj).fingerprint(keys.FingerprintFormats.MD5_HEX), "85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:da", ) self.assertEqual( keys.Key(self.dsaObj).fingerprint(keys.FingerprintFormats.MD5_HEX), "63:15:b3:0e:e6:4f:50:de:91:48:3d:01:6b:b3:13:c1", ) def test_fingerprintsha256(self): """ fingerprint method generates key fingerprint in L{FingerprintFormats.SHA256-BASE64} format if explicitly specified. """ self.assertEqual( keys.Key(self.rsaObj).fingerprint(keys.FingerprintFormats.SHA256_BASE64), "FBTCOoknq0mHy+kpfnY9tDdcAJuWtCpuQMaV3EsvbUI=", ) self.assertEqual( keys.Key(self.dsaObj).fingerprint(keys.FingerprintFormats.SHA256_BASE64), "Wz5o2YbKyxOEcJn1au/UaALSVruUzfz0vaLI1xiIGyY=", ) def test_fingerprintBadFormat(self): """ A C{BadFingerPrintFormat} error is raised when unsupported formats are requested. """ with self.assertRaises(keys.BadFingerPrintFormat) as em: keys.Key(self.rsaObj).fingerprint("sha256-base") self.assertEqual( "Unsupported fingerprint format: sha256-base", em.exception.args[0] ) def test_type(self): """ Test that the type method returns the correct type for an object. """ self.assertEqual(keys.Key(self.rsaObj).type(), "RSA") self.assertEqual(keys.Key(self.rsaObj).sshType(), b"ssh-rsa") self.assertEqual(keys.Key(self.dsaObj).type(), "DSA") self.assertEqual(keys.Key(self.dsaObj).sshType(), b"ssh-dss") self.assertEqual(keys.Key(self.ecObj).type(), "EC") self.assertEqual( keys.Key(self.ecObj).sshType(), keydata.ECDatanistp256["curve"] ) if ED25519_SUPPORTED: self.assertEqual(keys.Key(self.ed25519Obj).type(), "Ed25519") self.assertEqual(keys.Key(self.ed25519Obj).sshType(), b"ssh-ed25519") self.assertRaises(RuntimeError, keys.Key(None).type) self.assertRaises(RuntimeError, keys.Key(None).sshType) self.assertRaises(RuntimeError, keys.Key(self).type) self.assertRaises(RuntimeError, keys.Key(self).sshType) def test_fromBlobUnsupportedType(self): """ A C{BadKeyError} error is raised whey the blob has an unsupported key type. """ badBlob = common.NS(b"ssh-bad") self.assertRaises(keys.BadKeyError, keys.Key.fromString, badBlob) def test_fromBlobRSA(self): """ A public RSA key is correctly generated from a public key blob. """ rsaPublicData = { "n": keydata.RSAData["n"], "e": keydata.RSAData["e"], } rsaBlob = ( common.NS(b"ssh-rsa") + common.MP(rsaPublicData["e"]) + common.MP(rsaPublicData["n"]) ) rsaKey = keys.Key.fromString(rsaBlob) self.assertTrue(rsaKey.isPublic()) self.assertEqual(rsaPublicData, rsaKey.data()) def test_fromBlobDSA(self): """ A public DSA key is correctly generated from a public key blob. """ dsaPublicData = { "p": keydata.DSAData["p"], "q": keydata.DSAData["q"], "g": keydata.DSAData["g"], "y": keydata.DSAData["y"], } dsaBlob = ( common.NS(b"ssh-dss") + common.MP(dsaPublicData["p"]) + common.MP(dsaPublicData["q"]) + common.MP(dsaPublicData["g"]) + common.MP(dsaPublicData["y"]) ) dsaKey = keys.Key.fromString(dsaBlob) self.assertTrue(dsaKey.isPublic()) self.assertEqual(dsaPublicData, dsaKey.data()) def test_fromBlobECDSA(self): """ Key.fromString generates ECDSA keys from blobs. """ from cryptography import utils ecPublicData = { "x": keydata.ECDatanistp256["x"], "y": keydata.ECDatanistp256["y"], "curve": keydata.ECDatanistp256["curve"], } ecblob = ( common.NS(ecPublicData["curve"]) + common.NS(ecPublicData["curve"][-8:]) + common.NS( b"\x04" + utils.int_to_bytes(ecPublicData["x"], 32) + utils.int_to_bytes(ecPublicData["y"], 32) ) ) eckey = keys.Key.fromString(ecblob) self.assertTrue(eckey.isPublic()) self.assertEqual(ecPublicData, eckey.data()) @skipWithoutEd25519 def test_fromBlobEd25519(self): """ A public Ed25519 key is correctly generated from a public key blob. """ ed25519PublicData = { "a": keydata.Ed25519Data["a"], } ed25519Blob = common.NS(b"ssh-ed25519") + common.NS(ed25519PublicData["a"]) ed25519Key = keys.Key.fromString(ed25519Blob) self.assertTrue(ed25519Key.isPublic()) self.assertEqual(ed25519PublicData, ed25519Key.data()) def test_fromPrivateBlobUnsupportedType(self): """ C{BadKeyError} is raised when loading a private blob with an unsupported type. """ badBlob = common.NS(b"ssh-bad") self.assertRaises(keys.BadKeyError, keys.Key._fromString_PRIVATE_BLOB, badBlob) def test_fromPrivateBlobRSA(self): """ A private RSA key is correctly generated from a private key blob. """ rsaBlob = ( common.NS(b"ssh-rsa") + common.MP(keydata.RSAData["n"]) + common.MP(keydata.RSAData["e"]) + common.MP(keydata.RSAData["d"]) + common.MP(keydata.RSAData["u"]) + common.MP(keydata.RSAData["p"]) + common.MP(keydata.RSAData["q"]) ) rsaKey = keys.Key._fromString_PRIVATE_BLOB(rsaBlob) self.assertFalse(rsaKey.isPublic()) self.assertEqual(keydata.RSAData, rsaKey.data()) self.assertEqual( rsaKey, keys.Key._fromString_PRIVATE_BLOB(rsaKey.privateBlob()) ) def test_fromPrivateBlobDSA(self): """ A private DSA key is correctly generated from a private key blob. """ dsaBlob = ( common.NS(b"ssh-dss") + common.MP(keydata.DSAData["p"]) + common.MP(keydata.DSAData["q"]) + common.MP(keydata.DSAData["g"]) + common.MP(keydata.DSAData["y"]) + common.MP(keydata.DSAData["x"]) ) dsaKey = keys.Key._fromString_PRIVATE_BLOB(dsaBlob) self.assertFalse(dsaKey.isPublic()) self.assertEqual(keydata.DSAData, dsaKey.data()) self.assertEqual( dsaKey, keys.Key._fromString_PRIVATE_BLOB(dsaKey.privateBlob()) ) def test_fromPrivateBlobECDSA(self): """ A private EC key is correctly generated from a private key blob. """ from cryptography.hazmat.primitives import serialization from cryptography.hazmat.primitives.asymmetric import ec publicNumbers = ec.EllipticCurvePublicNumbers( x=keydata.ECDatanistp256["x"], y=keydata.ECDatanistp256["y"], curve=ec.SECP256R1(), ) ecblob = ( common.NS(keydata.ECDatanistp256["curve"]) + common.NS(keydata.ECDatanistp256["curve"][-8:]) + common.NS( publicNumbers.public_key(default_backend()).public_bytes( serialization.Encoding.X962, serialization.PublicFormat.UncompressedPoint, ) ) + common.MP(keydata.ECDatanistp256["privateValue"]) ) eckey = keys.Key._fromString_PRIVATE_BLOB(ecblob) self.assertFalse(eckey.isPublic()) self.assertEqual(keydata.ECDatanistp256, eckey.data()) self.assertEqual(eckey, keys.Key._fromString_PRIVATE_BLOB(eckey.privateBlob())) @skipWithoutEd25519 def test_fromPrivateBlobEd25519(self): """ A private Ed25519 key is correctly generated from a private key blob. """ ed25519Blob = ( common.NS(b"ssh-ed25519") + common.NS(keydata.Ed25519Data["a"]) + common.NS(keydata.Ed25519Data["k"] + keydata.Ed25519Data["a"]) ) ed25519Key = keys.Key._fromString_PRIVATE_BLOB(ed25519Blob) self.assertFalse(ed25519Key.isPublic()) self.assertEqual(keydata.Ed25519Data, ed25519Key.data()) self.assertEqual( ed25519Key, keys.Key._fromString_PRIVATE_BLOB(ed25519Key.privateBlob()) ) def test_blobRSA(self): """ Return the over-the-wire SSH format of the RSA public key. """ self.assertEqual( keys.Key(self.rsaObj).blob(), common.NS(b"ssh-rsa") + common.MP(self.rsaObj.private_numbers().public_numbers.e) + common.MP(self.rsaObj.private_numbers().public_numbers.n), ) def test_blobDSA(self): """ Return the over-the-wire SSH format of the DSA public key. """ publicNumbers = self.dsaObj.private_numbers().public_numbers self.assertEqual( keys.Key(self.dsaObj).blob(), common.NS(b"ssh-dss") + common.MP(publicNumbers.parameter_numbers.p) + common.MP(publicNumbers.parameter_numbers.q) + common.MP(publicNumbers.parameter_numbers.g) + common.MP(publicNumbers.y), ) def test_blobEC(self): """ Return the over-the-wire SSH format of the EC public key. """ from cryptography import utils byteLength = (self.ecObj.curve.key_size + 7) // 8 self.assertEqual( keys.Key(self.ecObj).blob(), common.NS(keydata.ECDatanistp256["curve"]) + common.NS(keydata.ECDatanistp256["curve"][-8:]) + common.NS( b"\x04" + utils.int_to_bytes( self.ecObj.private_numbers().public_numbers.x, byteLength ) + utils.int_to_bytes( self.ecObj.private_numbers().public_numbers.y, byteLength ) ), ) @skipWithoutEd25519 def test_blobEd25519(self): """ Return the over-the-wire SSH format of the Ed25519 public key. """ from cryptography.hazmat.primitives import serialization publicBytes = self.ed25519Obj.public_key().public_bytes( serialization.Encoding.Raw, serialization.PublicFormat.Raw ) self.assertEqual( keys.Key(self.ed25519Obj).blob(), common.NS(b"ssh-ed25519") + common.NS(publicBytes), ) def test_blobNoKey(self): """ C{RuntimeError} is raised when the blob is requested for a Key which is not wrapping anything. """ badKey = keys.Key(None) self.assertRaises(RuntimeError, badKey.blob) def test_privateBlobRSA(self): """ L{keys.Key.privateBlob} returns the SSH protocol-level format of an RSA private key. """ numbers = self.rsaObj.private_numbers() self.assertEqual( keys.Key(self.rsaObj).privateBlob(), common.NS(b"ssh-rsa") + common.MP(numbers.public_numbers.n) + common.MP(numbers.public_numbers.e) + common.MP(numbers.d) + common.MP(numbers.iqmp) + common.MP(numbers.p) + common.MP(numbers.q), ) def test_privateBlobDSA(self): """ L{keys.Key.privateBlob} returns the SSH protocol-level format of a DSA private key. """ publicNumbers = self.dsaObj.private_numbers().public_numbers self.assertEqual( keys.Key(self.dsaObj).privateBlob(), common.NS(b"ssh-dss") + common.MP(publicNumbers.parameter_numbers.p) + common.MP(publicNumbers.parameter_numbers.q) + common.MP(publicNumbers.parameter_numbers.g) + common.MP(publicNumbers.y) + common.MP(self.dsaObj.private_numbers().x), ) def test_privateBlobEC(self): """ L{keys.Key.privateBlob} returns the SSH ptotocol-level format of EC private key. """ from cryptography.hazmat.primitives import serialization self.assertEqual( keys.Key(self.ecObj).privateBlob(), common.NS(keydata.ECDatanistp256["curve"]) + common.NS(keydata.ECDatanistp256["curve"][-8:]) + common.NS( self.ecObj.public_key().public_bytes( serialization.Encoding.X962, serialization.PublicFormat.UncompressedPoint, ) ) + common.MP(self.ecObj.private_numbers().private_value), ) @skipWithoutEd25519 def test_privateBlobEd25519(self): """ L{keys.Key.privateBlob} returns the SSH protocol-level format of an Ed25519 private key. """ from cryptography.hazmat.primitives import serialization publicBytes = self.ed25519Obj.public_key().public_bytes( serialization.Encoding.Raw, serialization.PublicFormat.Raw ) privateBytes = self.ed25519Obj.private_bytes( serialization.Encoding.Raw, serialization.PrivateFormat.Raw, serialization.NoEncryption(), ) self.assertEqual( keys.Key(self.ed25519Obj).privateBlob(), common.NS(b"ssh-ed25519") + common.NS(publicBytes) + common.NS(privateBytes + publicBytes), ) def test_privateBlobNoKeyObject(self): """ Raises L{RuntimeError} if the underlying key object does not exists. """ badKey = keys.Key(None) self.assertRaises(RuntimeError, badKey.privateBlob) def test_toOpenSSHRSA(self): """ L{keys.Key.toString} serializes an RSA key in OpenSSH format. """ key = keys.Key.fromString(keydata.privateRSA_agentv3) self.assertEqual(key.toString("openssh"), keydata.privateRSA_openssh) self.assertEqual( key.toString("openssh", passphrase=b"encrypted"), keydata.privateRSA_openssh_encrypted, ) self.assertEqual( key.public().toString("openssh"), keydata.publicRSA_openssh[:-8] ) # no comment self.assertEqual( key.public().toString("openssh", comment=b"comment"), keydata.publicRSA_openssh, ) def test_toOpenSSHRSA_v1_format(self): """ L{keys.Key.toString} serializes an RSA key in OpenSSH's v1 format. """ key = keys.Key.fromString(keydata.privateRSA_openssh) new_key_data = key.toString("openssh", subtype="v1") new_enc_key_data = key.toString("openssh", subtype="v1", passphrase="encrypted") self.assertEqual( b"-----BEGIN OPENSSH PRIVATE KEY-----", new_key_data.splitlines()[0] ) self.assertEqual( b"-----BEGIN OPENSSH PRIVATE KEY-----", new_enc_key_data.splitlines()[0] ) self.assertEqual(key, keys.Key.fromString(new_key_data)) self.assertEqual( key, keys.Key.fromString(new_enc_key_data, passphrase="encrypted") ) def test_toOpenSSHDSA(self): """ L{keys.Key.toString} serializes a DSA key in OpenSSH format. """ key = keys.Key.fromString(keydata.privateDSA_lsh) self.assertEqual(key.toString("openssh"), keydata.privateDSA_openssh) self.assertEqual( key.public().toString("openssh", comment=b"comment"), keydata.publicDSA_openssh, ) self.assertEqual( key.public().toString("openssh"), keydata.publicDSA_openssh[:-8] ) # no comment def test_toOpenSSHDSA_v1_format(self): """ L{keys.Key.toString} serializes a DSA key in OpenSSH's v1 format. """ key = keys.Key.fromString(keydata.privateDSA_openssh) new_key_data = key.toString("openssh", subtype="v1") new_enc_key_data = key.toString("openssh", subtype="v1", passphrase="encrypted") self.assertEqual( b"-----BEGIN OPENSSH PRIVATE KEY-----", new_key_data.splitlines()[0] ) self.assertEqual( b"-----BEGIN OPENSSH PRIVATE KEY-----", new_enc_key_data.splitlines()[0] ) self.assertEqual(key, keys.Key.fromString(new_key_data)) self.assertEqual( key, keys.Key.fromString(new_enc_key_data, passphrase="encrypted") ) def test_toOpenSSHECDSA(self): """ L{keys.Key.toString} serializes an ECDSA key in OpenSSH format. """ key = keys.Key.fromString(keydata.privateECDSA_openssh) self.assertEqual( key.public().toString("openssh", comment=b"comment"), keydata.publicECDSA_openssh, ) self.assertEqual( key.public().toString("openssh"), keydata.publicECDSA_openssh[:-8] ) # no comment def test_toOpenSSHECDSA_v1_format(self): """ L{keys.Key.toString} serializes an ECDSA key in OpenSSH's v1 format. """ key = keys.Key.fromString(keydata.privateECDSA_openssh) new_key_data = key.toString("openssh", subtype="v1") new_enc_key_data = key.toString("openssh", subtype="v1", passphrase="encrypted") self.assertEqual( b"-----BEGIN OPENSSH PRIVATE KEY-----", new_key_data.splitlines()[0] ) self.assertEqual( b"-----BEGIN OPENSSH PRIVATE KEY-----", new_enc_key_data.splitlines()[0] ) self.assertEqual(key, keys.Key.fromString(new_key_data)) self.assertEqual( key, keys.Key.fromString(new_enc_key_data, passphrase="encrypted") ) @skipWithoutEd25519 def test_toOpenSSHEd25519(self): """ L{keys.Key.toString} serializes an Ed25519 key in OpenSSH's v1 format. """ key = keys.Key.fromString(keydata.privateEd25519_openssh_new) new_key_data = key.toString("openssh") new_enc_key_data = key.toString("openssh", passphrase="encrypted") self.assertEqual( b"-----BEGIN OPENSSH PRIVATE KEY-----", new_key_data.splitlines()[0] ) self.assertEqual( b"-----BEGIN OPENSSH PRIVATE KEY-----", new_enc_key_data.splitlines()[0] ) self.assertEqual(key, keys.Key.fromString(new_key_data)) self.assertEqual( key, keys.Key.fromString(new_enc_key_data, passphrase="encrypted") ) self.assertEqual(new_key_data, key.toString("openssh", subtype="v1")) @skipWithoutEd25519 def test_toOpenSSHEd25519_PEM_format(self): """ L{keys.Key.toString} refuses to serialize an Ed25519 key in OpenSSH's old PEM format, as no encoding of Ed25519 is defined for that format. """ key = keys.Key.fromString(keydata.privateEd25519_openssh_new) self.assertRaises(ValueError, key.toString, "openssh", subtype="PEM") def test_toLSHRSA(self): """ L{keys.Key.toString} serializes an RSA key in LSH format. """ key = keys.Key.fromString(keydata.privateRSA_openssh) self.assertEqual(key.toString("lsh"), keydata.privateRSA_lsh) self.assertEqual(key.public().toString("lsh"), keydata.publicRSA_lsh) def test_toLSHDSA(self): """ L{keys.Key.toString} serializes a DSA key in LSH format. """ key = keys.Key.fromString(keydata.privateDSA_openssh) self.assertEqual(key.toString("lsh"), keydata.privateDSA_lsh) self.assertEqual(key.public().toString("lsh"), keydata.publicDSA_lsh) def test_toAgentv3RSA(self): """ L{keys.Key.toString} serializes an RSA key in Agent v3 format. """ key = keys.Key.fromString(keydata.privateRSA_openssh) self.assertEqual(key.toString("agentv3"), keydata.privateRSA_agentv3) def test_toAgentv3DSA(self): """ L{keys.Key.toString} serializes a DSA key in Agent v3 format. """ key = keys.Key.fromString(keydata.privateDSA_openssh) self.assertEqual(key.toString("agentv3"), keydata.privateDSA_agentv3) def test_toStringNormalizesUnicodePassphrase(self): """ L{keys.Key.toString} applies Normalization Form KC to Unicode passphrases. """ key = keys.Key(self.rsaObj) key_data = key.toString("openssh", passphrase="verschlu\u0308sselt") self.assertEqual( keys.Key.fromString(key_data, passphrase="verschl\u00FCsselt".encode()), key, ) # U+FFFF is a "noncharacter" and guaranteed to have General_Category # Cn (Unassigned). self.assertRaises( keys.PassphraseNormalizationError, key.toString, "openssh", passphrase="unassigned \uFFFF", ) def test_toStringErrors(self): """ L{keys.Key.toString} raises L{keys.BadKeyError} when passed an invalid format type. """ self.assertRaises(keys.BadKeyError, keys.Key(self.rsaObj).toString, "bad_type") def test_signAndVerifyRSA(self): """ Signed data can be verified using RSA. """ data = b"some-data" key = keys.Key.fromString(keydata.privateRSA_openssh) signature = key.sign(data) self.assertTrue(key.public().verify(signature, data)) self.assertTrue(key.verify(signature, data)) def test_signAndVerifyDSA(self): """ Signed data can be verified using DSA. """ data = b"some-data" key = keys.Key.fromString(keydata.privateDSA_openssh) signature = key.sign(data) self.assertTrue(key.public().verify(signature, data)) self.assertTrue(key.verify(signature, data)) def test_signAndVerifyEC(self): """ Signed data can be verified using EC. """ data = b"some-data" key = keys.Key.fromString(keydata.privateECDSA_openssh) signature = key.sign(data) key384 = keys.Key.fromString(keydata.privateECDSA_openssh384) signature384 = key384.sign(data) key521 = keys.Key.fromString(keydata.privateECDSA_openssh521) signature521 = key521.sign(data) self.assertTrue(key.public().verify(signature, data)) self.assertTrue(key.verify(signature, data)) self.assertTrue(key384.public().verify(signature384, data)) self.assertTrue(key384.verify(signature384, data)) self.assertTrue(key521.public().verify(signature521, data)) self.assertTrue(key521.verify(signature521, data)) @skipWithoutEd25519 def test_signAndVerifyEd25519(self): """ Signed data can be verified using Ed25519. """ data = b"some-data" key = keys.Key.fromString(keydata.privateEd25519_openssh_new) signature = key.sign(data) self.assertTrue(key.public().verify(signature, data)) self.assertTrue(key.verify(signature, data)) def test_verifyRSA(self): """ A known-good RSA signature verifies successfully. """ key = keys.Key.fromString(keydata.publicRSA_openssh) self.assertTrue(key.verify(self.rsaSignature, b"")) self.assertFalse(key.verify(self.rsaSignature, b"a")) self.assertFalse(key.verify(self.dsaSignature, b"")) def test_verifyDSA(self): """ A known-good DSA signature verifies successfully. """ key = keys.Key.fromString(keydata.publicDSA_openssh) self.assertTrue(key.verify(self.dsaSignature, b"")) self.assertFalse(key.verify(self.dsaSignature, b"a")) self.assertFalse(key.verify(self.rsaSignature, b"")) def test_verifyDSANoPrefix(self): """ Some commercial SSH servers send DSA keys as 2 20-byte numbers; they are still verified as valid keys. """ key = keys.Key.fromString(keydata.publicDSA_openssh) self.assertTrue(key.verify(self.dsaSignature[-40:], b"")) def test_reprPrivateRSA(self): """ The repr of a L{keys.Key} contains all of the RSA components for an RSA private key. """ self.assertEqual( repr(keys.Key(self.rsaObj)), """<RSA Private Key (2048 bits) attr d: \t21:4c:08:66:a2:28:d5:b4:fb:8e:0f:72:1b:85:09: \t00:b9:f2:4e:37:f0:1c:57:4b:e3:51:7f:9e:23:a7: \te4:3a:98:55:1b:ea:8b:7a:98:1e:bc:d8:ba:b1:f9: \t89:12:18:60:ac:e8:cc:0b:4e:09:5a:40:6a:ba:2f: \t99:f8:b3:24:60:84:b9:ce:69:95:9a:f9:e2:fc:1f: \t51:4d:27:15:db:2b:27:ad:ef:b4:69:ac:be:7d:10: \teb:86:47:70:73:b4:00:87:95:15:3b:37:f9:e7:14: \te7:80:bb:68:1e:1b:e6:dd:bb:73:63:b9:67:e6:b2: \t27:7f:cf:cf:30:9b:c2:98:fd:d9:18:36:2f:36:2e: \tf1:3d:81:7a:9f:e1:03:2d:47:db:34:51:62:39:dd: \t4f:e9:ac:a8:8b:d9:d6:f3:84:c4:17:b9:71:9d:06: \t08:42:78:4d:bb:c5:2a:f4:c3:58:cd:55:2b:ed:be: \t33:5f:04:ea:7b:e6:04:24:63:f2:2d:d7:3d:1b:6c: \td5:9c:63:43:2f:92:88:8d:3e:6e:da:18:37:d8:0f: \t25:67:89:1d:b9:46:34:5e:c9:ce:c4:8b:ed:92:5a: \t33:07:0f:df:86:08:f9:92:e9:db:eb:38:08:36:c9: \tcd:cd:0a:01:48:5b:39:3e:7a:ca:c6:80:a9:dc:d4: \t39 attr e: \t01:00:01 attr n: \t00:d5:6a:ac:78:23:d6:d6:1b:ec:25:a1:50:c4:77: \t63:50:84:45:01:55:42:14:2a:2a:e0:d0:60:ee:d4: \te9:a3:ad:4a:fa:39:06:5e:84:55:75:5f:00:36:bf: \t6f:aa:2a:3f:83:26:37:c1:69:2e:5b:fd:f0:f3:d2: \t7d:d6:98:cd:3a:40:78:d5:ca:a8:18:c0:11:93:24: \t09:0c:81:4c:8f:f7:9c:ed:13:16:6a:a4:04:e9:49: \t77:c3:e4:55:64:b3:79:68:9e:2c:08:eb:ac:e8:04: \t2d:21:77:05:a7:8e:ef:53:30:0d:a5:e5:bb:3d:6a: \te2:09:36:6f:fd:34:d3:7d:6f:46:ff:87:da:a9:29: \t27:aa:ff:ad:f5:85:e6:3e:1a:b8:7a:1d:4a:b1:ea: \tc0:5a:f7:30:df:1f:c2:a4:e4:ef:3f:91:49:96:40: \td5:19:77:2d:37:c3:5e:ec:9d:a6:3a:44:a5:c2:a4: \t29:dd:d5:ba:9c:3d:45:b3:c6:2c:18:64:d5:ba:3d: \tdf:ab:7f:cd:42:ac:a7:f1:18:0b:a0:58:15:62:0b: \ta4:2a:6e:43:c3:e4:04:9f:35:a3:47:8e:46:ed:33: \ta5:65:bd:bc:3b:29:6e:02:0b:57:df:74:e8:13:b4: \t37:35:7e:83:5f:20:26:60:a6:dc:ad:8b:c6:6c:79: \t98:f7 attr p: \t00:d9:70:06:d8:e2:bc:d4:78:91:50:94:d4:c1:1b: \t89:38:6c:46:64:5a:51:a0:9a:07:3d:48:8f:03:51: \tcc:6b:12:8e:7d:1a:b1:65:e7:71:75:39:e0:32:05: \t75:8d:18:4c:af:93:b1:49:b1:66:5f:78:62:7a:d1: \t0c:ca:e6:4d:43:b3:9c:f4:6b:7d:e6:0c:98:dc:cf: \t21:62:8e:d5:2e:12:de:04:ae:d7:24:6e:83:31:a2: \t15:a2:44:3d:22:a9:62:26:22:b9:b2:ed:54:0a:9d: \t08:83:a7:07:0d:ff:19:18:8e:d8:ab:1d:da:48:9c: \t31:68:11:a1:66:6d:e3:d8:1d attr q: \t00:fb:44:17:8b:a4:36:be:1e:37:1d:a7:f6:61:6c: \t04:c4:aa:dd:78:3e:07:8c:1e:33:02:ae:03:14:87: \t83:7a:e5:9e:7d:08:67:a8:f2:aa:bf:12:70:cf:72: \ta9:a7:c7:0b:1d:88:d5:20:fd:9c:63:ca:47:30:55: \t4e:8b:c4:cf:f4:7f:16:a4:92:12:74:a1:09:c2:c4: \t6e:9c:8c:33:ef:a5:e5:f7:e0:2b:ad:4f:5c:11:aa: \t1a:84:37:5b:fd:7a:ea:c3:cd:7c:b0:c8:e4:1f:54: \t63:b5:c7:af:df:f4:09:a7:fc:c7:25:fc:5c:e9:91: \td7:92:c5:98:1e:56:d3:b1:23 attr u: \t00:85:4b:1b:7a:9b:12:10:37:9e:1f:ad:5e:da:fe: \tc6:96:fe:df:35:6b:b9:34:e2:16:97:92:26:09:bd: \tbd:70:20:03:a7:35:bd:2d:1b:a0:d2:07:47:2b:d4: \tde:a8:a8:07:07:1b:b8:04:20:a7:27:41:3c:6c:39: \t39:e9:41:ce:e7:17:1d:d1:4c:5c:bc:3d:d2:26:26: \tfe:6a:d6:fd:48:72:ae:46:fa:7b:c3:d3:19:60:44: \t1d:a5:13:a7:80:f5:63:29:d4:7a:5d:06:07:16:5d: \tf6:8b:3d:cb:64:3a:e2:84:5a:4d:8c:06:2d:2d:9d: \t1c:eb:83:4c:78:3d:79:54:ce>""", ) def test_reprPublicRSA(self): """ The repr of a L{keys.Key} contains all of the RSA components for an RSA public key. """ self.assertEqual( repr(keys.Key(self.rsaObj).public()), """<RSA Public Key (2048 bits) attr e: \t01:00:01 attr n: \t00:d5:6a:ac:78:23:d6:d6:1b:ec:25:a1:50:c4:77: \t63:50:84:45:01:55:42:14:2a:2a:e0:d0:60:ee:d4: \te9:a3:ad:4a:fa:39:06:5e:84:55:75:5f:00:36:bf: \t6f:aa:2a:3f:83:26:37:c1:69:2e:5b:fd:f0:f3:d2: \t7d:d6:98:cd:3a:40:78:d5:ca:a8:18:c0:11:93:24: \t09:0c:81:4c:8f:f7:9c:ed:13:16:6a:a4:04:e9:49: \t77:c3:e4:55:64:b3:79:68:9e:2c:08:eb:ac:e8:04: \t2d:21:77:05:a7:8e:ef:53:30:0d:a5:e5:bb:3d:6a: \te2:09:36:6f:fd:34:d3:7d:6f:46:ff:87:da:a9:29: \t27:aa:ff:ad:f5:85:e6:3e:1a:b8:7a:1d:4a:b1:ea: \tc0:5a:f7:30:df:1f:c2:a4:e4:ef:3f:91:49:96:40: \td5:19:77:2d:37:c3:5e:ec:9d:a6:3a:44:a5:c2:a4: \t29:dd:d5:ba:9c:3d:45:b3:c6:2c:18:64:d5:ba:3d: \tdf:ab:7f:cd:42:ac:a7:f1:18:0b:a0:58:15:62:0b: \ta4:2a:6e:43:c3:e4:04:9f:35:a3:47:8e:46:ed:33: \ta5:65:bd:bc:3b:29:6e:02:0b:57:df:74:e8:13:b4: \t37:35:7e:83:5f:20:26:60:a6:dc:ad:8b:c6:6c:79: \t98:f7>""", ) def test_reprPublicECDSA(self): """ The repr of a L{keys.Key} contains all the OpenSSH format for an ECDSA public key. """ self.assertEqual( repr(keys.Key(self.ecObj).public()), dedent( """\ <Elliptic Curve Public Key (256 bits) curve: \tecdsa-sha2-nistp256 x: \t{x} y: \t{y}> """ ).format(**keydata.ECDatanistp256), ) def test_reprPrivateECDSA(self): """ The repr of a L{keys.Key} contains all the OpenSSH format for an ECDSA private key. """ self.assertEqual( repr(keys.Key(self.ecObj)), dedent( """\ <Elliptic Curve Private Key (256 bits) curve: \tecdsa-sha2-nistp256 privateValue: \t{privateValue} x: \t{x} y: \t{y}> """ ).format(**keydata.ECDatanistp256), ) @skipWithoutEd25519 def test_reprPublicEd25519(self): """ The repr of a L{keys.Key} contains all the OpenSSH format for an Ed25519 public key. """ self.assertEqual( repr(keys.Key(self.ed25519Obj).public()), dedent( """\ <Ed25519 Public Key (256 bits) attr a: \tf1:16:d1:15:4a:1e:15:0e:19:5e:19:46:b5:f2:44: \t0d:b2:52:a0:ae:2a:6b:23:13:73:45:fd:40:d9:57: \t7b:8b>""" ), ) @skipWithoutEd25519 def test_reprPrivateEd25519(self): """ The repr of a L{keys.Key} contains all the OpenSSH format for an Ed25519 private key. """ self.assertEqual( repr(keys.Key(self.ed25519Obj)), dedent( """\ <Ed25519 Private Key (256 bits) attr a: \tf1:16:d1:15:4a:1e:15:0e:19:5e:19:46:b5:f2:44: \t0d:b2:52:a0:ae:2a:6b:23:13:73:45:fd:40:d9:57: \t7b:8b attr k: \t37:2f:25:da:8d:d4:a8:9a:78:7c:61:f0:98:01:c6: \tf4:5e:6d:67:05:69:31:37:4c:69:0d:05:55:bb:c9: \t44:58>""" ), ) class PersistentRSAKeyTests(unittest.TestCase): """ Tests for L{keys._getPersistentRSAKey}. """ if cryptography is None: skip = skipCryptography def test_providedArguments(self): """ L{keys._getPersistentRSAKey} will put the key in C{directory}/C{filename}, with the key length of C{keySize}. """ tempDir = FilePath(self.mktemp()) keyFile = tempDir.child("mykey.pem") key = keys._getPersistentRSAKey(keyFile, keySize=512) self.assertEqual(key.size(), 512) self.assertTrue(keyFile.exists()) def test_noRegeneration(self): """ L{keys._getPersistentRSAKey} will not regenerate the key if the key already exists. """ tempDir = FilePath(self.mktemp()) keyFile = tempDir.child("mykey.pem") key = keys._getPersistentRSAKey(keyFile, keySize=512) self.assertEqual(key.size(), 512) self.assertTrue(keyFile.exists()) keyContent = keyFile.getContent() # Set the key size to 1024 bits. Since it exists already, it will find # the 512 bit key, and not generate a 1024 bit key. key = keys._getPersistentRSAKey(keyFile, keySize=1024) self.assertEqual(key.size(), 512) self.assertEqual(keyFile.getContent(), keyContent) def test_keySizeZero(self): """ If the key generated by L{keys.getPersistentRSAKey} is set to None the key size should then become 0. """ tempDir = FilePath(self.mktemp()) keyFile = tempDir.child("mykey.pem") key = keys._getPersistentRSAKey(keyFile, keySize=512) key._keyObject = None self.assertEqual(key.size(), 0) PK F��ZӉ_Ȫ� �� test_ssh.pynu �[��� # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Tests for L{twisted.conch.ssh}. """ import struct from twisted.conch.test.keydata import ( privateDSA_openssh, privateRSA_openssh, publicDSA_openssh, publicRSA_openssh, ) from twisted.conch.test.loopback import LoopbackRelay from twisted.cred import portal from twisted.cred.error import UnauthorizedLogin from twisted.internet import defer, protocol, reactor from twisted.internet.error import ProcessTerminated from twisted.python import failure, log from twisted.python.reflect import requireModule from twisted.trial import unittest cryptography = requireModule("cryptography") pyasn1 = requireModule("pyasn1") if cryptography: from twisted.conch import avatar, error from twisted.conch.ssh import _kex, common, forwarding, session else: class avatar: # type: ignore[no-redef] class ConchUser: pass class ConchTestRealm: """ A realm which expects a particular avatarId to log in once and creates a L{ConchTestAvatar} for that request. @ivar expectedAvatarID: The only avatarID that this realm will produce an avatar for. @ivar avatar: A reference to the avatar after it is requested. """ avatar = None def __init__(self, expectedAvatarID): self.expectedAvatarID = expectedAvatarID def requestAvatar(self, avatarID, mind, *interfaces): """ Return a new L{ConchTestAvatar} if the avatarID matches the expected one and this is the first avatar request. """ if avatarID == self.expectedAvatarID: if self.avatar is not None: raise UnauthorizedLogin("Only one login allowed") self.avatar = ConchTestAvatar() return interfaces[0], self.avatar, self.avatar.logout raise UnauthorizedLogin( f"Only {self.expectedAvatarID!r} may log in, not {avatarID!r}" ) class ConchTestAvatar(avatar.ConchUser): """ An avatar against which various SSH features can be tested. @ivar loggedOut: A flag indicating whether the avatar logout method has been called. """ if not cryptography: skip = "cannot run without cryptography" loggedOut = False def __init__(self): avatar.ConchUser.__init__(self) self.listeners = {} self.globalRequests = {} self.channelLookup.update( { b"session": session.SSHSession, b"direct-tcpip": forwarding.openConnectForwardingClient, } ) self.subsystemLookup.update({b"crazy": CrazySubsystem}) def global_foo(self, data): self.globalRequests["foo"] = data return 1 def global_foo_2(self, data): self.globalRequests["foo_2"] = data return 1, b"data" def global_tcpip_forward(self, data): host, port = forwarding.unpackGlobal_tcpip_forward(data) try: listener = reactor.listenTCP( port, forwarding.SSHListenForwardingFactory( self.conn, (host, port), forwarding.SSHListenServerForwardingChannel ), interface=host, ) except BaseException: log.err(None, "something went wrong with remote->local forwarding") return 0 else: self.listeners[(host, port)] = listener return 1 def global_cancel_tcpip_forward(self, data): host, port = forwarding.unpackGlobal_tcpip_forward(data) listener = self.listeners.get((host, port), None) if not listener: return 0 del self.listeners[(host, port)] listener.stopListening() return 1 def logout(self): self.loggedOut = True for listener in self.listeners.values(): log.msg("stopListening %s" % listener) listener.stopListening() class ConchSessionForTestAvatar: """ An ISession adapter for ConchTestAvatar. """ def __init__(self, avatar): """ Initialize the session and create a reference to it on the avatar for later inspection. """ self.avatar = avatar self.avatar._testSession = self self.cmd = None self.proto = None self.ptyReq = False self.eof = 0 self.onClose = defer.Deferred() def getPty(self, term, windowSize, attrs): log.msg("pty req") self._terminalType = term self._windowSize = windowSize self.ptyReq = True def openShell(self, proto): log.msg("opening shell") self.proto = proto EchoTransport(proto) self.cmd = b"shell" def execCommand(self, proto, cmd): self.cmd = cmd self.proto = proto f = cmd.split()[0] if f == b"false": t = FalseTransport(proto) # Avoid disconnecting this immediately. If the channel is closed # before execCommand even returns the caller gets confused. reactor.callLater(0, t.loseConnection) elif f == b"echo": t = EchoTransport(proto) t.write(cmd[5:]) t.loseConnection() elif f == b"secho": t = SuperEchoTransport(proto) t.write(cmd[6:]) t.loseConnection() elif f == b"eecho": t = ErrEchoTransport(proto) t.write(cmd[6:]) t.loseConnection() else: raise error.ConchError("bad exec") self.avatar.conn.transport.expectedLoseConnection = 1 def eofReceived(self): self.eof = 1 def closed(self): log.msg('closed cmd "%s"' % self.cmd) self.remoteWindowLeftAtClose = self.proto.session.remoteWindowLeft self.onClose.callback(None) from twisted.python import components if cryptography: components.registerAdapter( ConchSessionForTestAvatar, ConchTestAvatar, session.ISession ) class CrazySubsystem(protocol.Protocol): def __init__(self, *args, **kw): pass def connectionMade(self): """ good ... good """ class FalseTransport: """ False transport should act like a /bin/false execution, i.e. just exit with nonzero status, writing nothing to the terminal. @ivar proto: The protocol associated with this transport. @ivar closed: A flag tracking whether C{loseConnection} has been called yet. """ def __init__(self, p): """ @type p L{twisted.conch.ssh.session.SSHSessionProcessProtocol} instance """ self.proto = p p.makeConnection(self) self.closed = 0 def loseConnection(self): """ Disconnect the protocol associated with this transport. """ if self.closed: return self.closed = 1 self.proto.inConnectionLost() self.proto.outConnectionLost() self.proto.errConnectionLost() self.proto.processEnded(failure.Failure(ProcessTerminated(255, None, None))) class EchoTransport: def __init__(self, p): self.proto = p p.makeConnection(self) self.closed = 0 def write(self, data): log.msg(repr(data)) self.proto.outReceived(data) self.proto.outReceived(b"\r\n") if b"\x00" in data: # mimic 'exit' for the shell test self.loseConnection() def loseConnection(self): if self.closed: return self.closed = 1 self.proto.inConnectionLost() self.proto.outConnectionLost() self.proto.errConnectionLost() self.proto.processEnded(failure.Failure(ProcessTerminated(0, None, None))) class ErrEchoTransport: def __init__(self, p): self.proto = p p.makeConnection(self) self.closed = 0 def write(self, data): self.proto.errReceived(data) self.proto.errReceived(b"\r\n") def loseConnection(self): if self.closed: return self.closed = 1 self.proto.inConnectionLost() self.proto.outConnectionLost() self.proto.errConnectionLost() self.proto.processEnded(failure.Failure(ProcessTerminated(0, None, None))) class SuperEchoTransport: def __init__(self, p): self.proto = p p.makeConnection(self) self.closed = 0 def write(self, data): self.proto.outReceived(data) self.proto.outReceived(b"\r\n") self.proto.errReceived(data) self.proto.errReceived(b"\r\n") def loseConnection(self): if self.closed: return self.closed = 1 self.proto.inConnectionLost() self.proto.outConnectionLost() self.proto.errConnectionLost() self.proto.processEnded(failure.Failure(ProcessTerminated(0, None, None))) if cryptography is not None and pyasn1 is not None: from twisted.conch import checkers from twisted.conch.ssh import ( channel, connection, factory, keys, transport, userauth, ) class ConchTestPasswordChecker: credentialInterfaces = (checkers.IUsernamePassword,) def requestAvatarId(self, credentials): if ( credentials.username == b"testuser" and credentials.password == b"testpass" ): return defer.succeed(credentials.username) return defer.fail(Exception("Bad credentials")) class ConchTestSSHChecker(checkers.SSHProtocolChecker): def areDone(self, avatarId): if avatarId != b"testuser" or len(self.successfulCredentials[avatarId]) < 2: return False return True class ConchTestServerFactory(factory.SSHFactory): noisy = False services = { b"ssh-userauth": userauth.SSHUserAuthServer, b"ssh-connection": connection.SSHConnection, } def buildProtocol(self, addr): proto = ConchTestServer() proto.supportedPublicKeys = self.privateKeys.keys() proto.factory = self if hasattr(self, "expectedLoseConnection"): proto.expectedLoseConnection = self.expectedLoseConnection self.proto = proto return proto def getPublicKeys(self): return { b"ssh-rsa": keys.Key.fromString(publicRSA_openssh), b"ssh-dss": keys.Key.fromString(publicDSA_openssh), } def getPrivateKeys(self): return { b"ssh-rsa": keys.Key.fromString(privateRSA_openssh), b"ssh-dss": keys.Key.fromString(privateDSA_openssh), } def getPrimes(self): """ Diffie-Hellman primes that can be used for the diffie-hellman-group-exchange-sha1 key exchange. @return: The primes and generators. @rtype: L{dict} mapping the key size to a C{list} of C{(generator, prime)} tupple. """ # In these tests, we hardwire the prime values to those defined by # the diffie-hellman-group14-sha1 key exchange algorithm, to avoid # requiring a moduli file when running tests. # See OpenSSHFactory.getPrimes. return {2048: [_kex.getDHGeneratorAndPrime(b"diffie-hellman-group14-sha1")]} def getService(self, trans, name): return factory.SSHFactory.getService(self, trans, name) class ConchTestBase: done = 0 def connectionLost(self, reason): if self.done: return if not hasattr(self, "expectedLoseConnection"): raise unittest.FailTest( f"unexpectedly lost connection {self}\n{reason}" ) self.done = 1 def receiveError(self, reasonCode, desc): self.expectedLoseConnection = 1 # Some versions of OpenSSH (for example, OpenSSH_5.3p1) will # send a DISCONNECT_BY_APPLICATION error before closing the # connection. Other, older versions (for example, # OpenSSH_5.1p1), won't. So accept this particular error here, # but no others. if reasonCode != transport.DISCONNECT_BY_APPLICATION: log.err( Exception( "got disconnect for %s: reason %s, desc: %s" % (self, reasonCode, desc) ) ) self.loseConnection() def receiveUnimplemented(self, seqID): raise unittest.FailTest(f"got unimplemented: seqid {seqID}") class ConchTestServer(ConchTestBase, transport.SSHServerTransport): def connectionLost(self, reason): ConchTestBase.connectionLost(self, reason) transport.SSHServerTransport.connectionLost(self, reason) class ConchTestClient(ConchTestBase, transport.SSHClientTransport): """ @ivar _channelFactory: A callable which accepts an SSH connection and returns a channel which will be attached to a new channel on that connection. """ def __init__(self, channelFactory): self._channelFactory = channelFactory def connectionLost(self, reason): ConchTestBase.connectionLost(self, reason) transport.SSHClientTransport.connectionLost(self, reason) def verifyHostKey(self, key, fp): keyMatch = key == keys.Key.fromString(publicRSA_openssh).blob() fingerprintMatch = fp == b"85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:da" if keyMatch and fingerprintMatch: return defer.succeed(1) return defer.fail(Exception("Key or fingerprint mismatch")) def connectionSecure(self): self.requestService( ConchTestClientAuth( b"testuser", ConchTestClientConnection(self._channelFactory) ) ) class ConchTestClientAuth(userauth.SSHUserAuthClient): hasTriedNone = 0 # have we tried the 'none' auth yet? canSucceedPublicKey = 0 # can we succeed with this yet? canSucceedPassword = 0 def ssh_USERAUTH_SUCCESS(self, packet): if not self.canSucceedPassword and self.canSucceedPublicKey: raise unittest.FailTest( "got USERAUTH_SUCCESS before password and publickey" ) userauth.SSHUserAuthClient.ssh_USERAUTH_SUCCESS(self, packet) def getPassword(self): self.canSucceedPassword = 1 return defer.succeed(b"testpass") def getPrivateKey(self): self.canSucceedPublicKey = 1 return defer.succeed(keys.Key.fromString(privateDSA_openssh)) def getPublicKey(self): return keys.Key.fromString(publicDSA_openssh) class ConchTestClientConnection(connection.SSHConnection): """ @ivar _completed: A L{Deferred} which will be fired when the number of results collected reaches C{totalResults}. """ name = b"ssh-connection" results = 0 totalResults = 8 def __init__(self, channelFactory): connection.SSHConnection.__init__(self) self._channelFactory = channelFactory def serviceStarted(self): self.openChannel(self._channelFactory(conn=self)) class SSHTestChannel(channel.SSHChannel): def __init__(self, name, opened, *args, **kwargs): self.name = name self._opened = opened self.received = [] self.receivedExt = [] self.onClose = defer.Deferred() channel.SSHChannel.__init__(self, *args, **kwargs) def openFailed(self, reason): self._opened.errback(reason) def channelOpen(self, ignore): self._opened.callback(self) def dataReceived(self, data): self.received.append(data) def extReceived(self, dataType, data): if dataType == connection.EXTENDED_DATA_STDERR: self.receivedExt.append(data) else: log.msg(f"Unrecognized extended data: {dataType!r}") def request_exit_status(self, status): [self.status] = struct.unpack(">L", status) def eofReceived(self): self.eofCalled = True def closed(self): self.onClose.callback(None) def conchTestPublicKeyChecker(): """ Produces a SSHPublicKeyChecker with an in-memory key mapping with a single use: 'testuser' @return: L{twisted.conch.checkers.SSHPublicKeyChecker} """ conchTestPublicKeyDB = checkers.InMemorySSHKeyDB( {b"testuser": [keys.Key.fromString(publicDSA_openssh)]} ) return checkers.SSHPublicKeyChecker(conchTestPublicKeyDB) class SSHProtocolTests(unittest.TestCase): """ Tests for communication between L{SSHServerTransport} and L{SSHClientTransport}. """ if not cryptography: skip = "can't run without cryptography" if not pyasn1: skip = "Cannot run without PyASN1" def _ourServerOurClientTest(self, name=b"session", **kwargs): """ Create a connected SSH client and server protocol pair and return a L{Deferred} which fires with an L{SSHTestChannel} instance connected to a channel on that SSH connection. """ result = defer.Deferred() self.realm = ConchTestRealm(b"testuser") p = portal.Portal(self.realm) sshpc = ConchTestSSHChecker() sshpc.registerChecker(ConchTestPasswordChecker()) sshpc.registerChecker(conchTestPublicKeyChecker()) p.registerChecker(sshpc) fac = ConchTestServerFactory() fac.portal = p fac.startFactory() self.server = fac.buildProtocol(None) self.clientTransport = LoopbackRelay(self.server) self.client = ConchTestClient( lambda conn: SSHTestChannel(name, result, conn=conn, **kwargs) ) self.serverTransport = LoopbackRelay(self.client) self.server.makeConnection(self.serverTransport) self.client.makeConnection(self.clientTransport) return result def test_subsystemsAndGlobalRequests(self): """ Run the Conch server against the Conch client. Set up several different channels which exercise different behaviors and wait for them to complete. Verify that the channels with errors log them. """ channel = self._ourServerOurClientTest() def cbSubsystem(channel): self.channel = channel return self.assertFailure( channel.conn.sendRequest( channel, b"subsystem", common.NS(b"not-crazy"), 1 ), Exception, ) channel.addCallback(cbSubsystem) def cbNotCrazyFailed(ignored): channel = self.channel return channel.conn.sendRequest( channel, b"subsystem", common.NS(b"crazy"), 1 ) channel.addCallback(cbNotCrazyFailed) def cbGlobalRequests(ignored): channel = self.channel d1 = channel.conn.sendGlobalRequest(b"foo", b"bar", 1) d2 = channel.conn.sendGlobalRequest(b"foo-2", b"bar2", 1) d2.addCallback(self.assertEqual, b"data") d3 = self.assertFailure( channel.conn.sendGlobalRequest(b"bar", b"foo", 1), Exception ) return defer.gatherResults([d1, d2, d3]) channel.addCallback(cbGlobalRequests) def disconnect(ignored): self.assertEqual( self.realm.avatar.globalRequests, {"foo": b"bar", "foo_2": b"bar2"} ) channel = self.channel channel.conn.transport.expectedLoseConnection = True channel.conn.serviceStopped() channel.loseConnection() channel.addCallback(disconnect) return channel def test_shell(self): """ L{SSHChannel.sendRequest} can open a shell with a I{pty-req} request, specifying a terminal type and window size. """ channel = self._ourServerOurClientTest() data = session.packRequest_pty_req(b"conch-test-term", (24, 80, 0, 0), b"") def cbChannel(channel): self.channel = channel return channel.conn.sendRequest(channel, b"pty-req", data, 1) channel.addCallback(cbChannel) def cbPty(ignored): # The server-side object corresponding to our client side channel. session = self.realm.avatar.conn.channels[0].session self.assertIs(session.avatar, self.realm.avatar) self.assertEqual(session._terminalType, b"conch-test-term") self.assertEqual(session._windowSize, (24, 80, 0, 0)) self.assertTrue(session.ptyReq) channel = self.channel return channel.conn.sendRequest(channel, b"shell", b"", 1) channel.addCallback(cbPty) def cbShell(ignored): self.channel.write(b"testing the shell!\x00") self.channel.conn.sendEOF(self.channel) return defer.gatherResults( [self.channel.onClose, self.realm.avatar._testSession.onClose] ) channel.addCallback(cbShell) def cbExited(ignored): if self.channel.status != 0: log.msg("shell exit status was not 0: %i" % (self.channel.status,)) self.assertEqual( b"".join(self.channel.received), b"testing the shell!\x00\r\n" ) self.assertTrue(self.channel.eofCalled) self.assertTrue(self.realm.avatar._testSession.eof) channel.addCallback(cbExited) return channel def test_failedExec(self): """ If L{SSHChannel.sendRequest} issues an exec which the server responds to with an error, the L{Deferred} it returns fires its errback. """ channel = self._ourServerOurClientTest() def cbChannel(channel): self.channel = channel return self.assertFailure( channel.conn.sendRequest(channel, b"exec", common.NS(b"jumboliah"), 1), Exception, ) channel.addCallback(cbChannel) def cbFailed(ignored): # The server logs this exception when it cannot perform the # requested exec. errors = self.flushLoggedErrors(error.ConchError) self.assertEqual(errors[0].value.args, ("bad exec", None)) channel.addCallback(cbFailed) return channel def test_falseChannel(self): """ When the process started by a L{SSHChannel.sendRequest} exec request exits, the exit status is reported to the channel. """ channel = self._ourServerOurClientTest() def cbChannel(channel): self.channel = channel return channel.conn.sendRequest(channel, b"exec", common.NS(b"false"), 1) channel.addCallback(cbChannel) def cbExec(ignored): return self.channel.onClose channel.addCallback(cbExec) def cbClosed(ignored): # No data is expected self.assertEqual(self.channel.received, []) self.assertNotEqual(self.channel.status, 0) channel.addCallback(cbClosed) return channel def test_errorChannel(self): """ Bytes sent over the extended channel for stderr data are delivered to the channel's C{extReceived} method. """ channel = self._ourServerOurClientTest(localWindow=4, localMaxPacket=5) def cbChannel(channel): self.channel = channel return channel.conn.sendRequest( channel, b"exec", common.NS(b"eecho hello"), 1 ) channel.addCallback(cbChannel) def cbExec(ignored): return defer.gatherResults( [self.channel.onClose, self.realm.avatar._testSession.onClose] ) channel.addCallback(cbExec) def cbClosed(ignored): self.assertEqual(self.channel.received, []) self.assertEqual(b"".join(self.channel.receivedExt), b"hello\r\n") self.assertEqual(self.channel.status, 0) self.assertTrue(self.channel.eofCalled) self.assertEqual(self.channel.localWindowLeft, 4) self.assertEqual( self.channel.localWindowLeft, self.realm.avatar._testSession.remoteWindowLeftAtClose, ) channel.addCallback(cbClosed) return channel def test_unknownChannel(self): """ When an attempt is made to open an unknown channel type, the L{Deferred} returned by L{SSHChannel.sendRequest} fires its errback. """ d = self.assertFailure( self._ourServerOurClientTest(b"crazy-unknown-channel"), Exception ) def cbFailed(ignored): errors = self.flushLoggedErrors(error.ConchError) self.assertEqual(errors[0].value.args, (3, "unknown channel")) self.assertEqual(len(errors), 1) d.addCallback(cbFailed) return d def test_maxPacket(self): """ An L{SSHChannel} can be configured with a maximum packet size to receive. """ # localWindow needs to be at least 11 otherwise the assertion about it # in cbClosed is invalid. channel = self._ourServerOurClientTest(localWindow=11, localMaxPacket=1) def cbChannel(channel): self.channel = channel return channel.conn.sendRequest( channel, b"exec", common.NS(b"secho hello"), 1 ) channel.addCallback(cbChannel) def cbExec(ignored): return self.channel.onClose channel.addCallback(cbExec) def cbClosed(ignored): self.assertEqual(self.channel.status, 0) self.assertEqual(b"".join(self.channel.received), b"hello\r\n") self.assertEqual(b"".join(self.channel.receivedExt), b"hello\r\n") self.assertEqual(self.channel.localWindowLeft, 11) self.assertTrue(self.channel.eofCalled) channel.addCallback(cbClosed) return channel def test_echo(self): """ Normal standard out bytes are sent to the channel's C{dataReceived} method. """ channel = self._ourServerOurClientTest(localWindow=4, localMaxPacket=5) def cbChannel(channel): self.channel = channel return channel.conn.sendRequest( channel, b"exec", common.NS(b"echo hello"), 1 ) channel.addCallback(cbChannel) def cbEcho(ignored): return defer.gatherResults( [self.channel.onClose, self.realm.avatar._testSession.onClose] ) channel.addCallback(cbEcho) def cbClosed(ignored): self.assertEqual(self.channel.status, 0) self.assertEqual(b"".join(self.channel.received), b"hello\r\n") self.assertEqual(self.channel.localWindowLeft, 4) self.assertTrue(self.channel.eofCalled) self.assertEqual( self.channel.localWindowLeft, self.realm.avatar._testSession.remoteWindowLeftAtClose, ) channel.addCallback(cbClosed) return channel class SSHFactoryTests(unittest.TestCase): if not cryptography: skip = "can't run without cryptography" if not pyasn1: skip = "Cannot run without PyASN1" def makeSSHFactory(self, primes=None): sshFactory = factory.SSHFactory() gpk = lambda: {"ssh-rsa": keys.Key(None)} sshFactory.getPrimes = lambda: primes sshFactory.getPublicKeys = sshFactory.getPrivateKeys = gpk sshFactory.startFactory() return sshFactory def test_buildProtocol(self): """ By default, buildProtocol() constructs an instance of SSHServerTransport. """ factory = self.makeSSHFactory() protocol = factory.buildProtocol(None) self.assertIsInstance(protocol, transport.SSHServerTransport) def test_buildProtocolRespectsProtocol(self): """ buildProtocol() calls 'self.protocol()' to construct a protocol instance. """ calls = [] def makeProtocol(*args): calls.append(args) return transport.SSHServerTransport() factory = self.makeSSHFactory() factory.protocol = makeProtocol factory.buildProtocol(None) self.assertEqual([()], calls) def test_buildProtocolNoPrimes(self): """ Group key exchanges are not supported when we don't have the primes database. """ f1 = self.makeSSHFactory(primes=None) p1 = f1.buildProtocol(None) self.assertNotIn( b"diffie-hellman-group-exchange-sha1", p1.supportedKeyExchanges ) self.assertNotIn( b"diffie-hellman-group-exchange-sha256", p1.supportedKeyExchanges ) def test_buildProtocolWithPrimes(self): """ Group key exchanges are supported when we have the primes database. """ f2 = self.makeSSHFactory(primes={1: (2, 3)}) p2 = f2.buildProtocol(None) self.assertIn(b"diffie-hellman-group-exchange-sha1", p2.supportedKeyExchanges) self.assertIn(b"diffie-hellman-group-exchange-sha256", p2.supportedKeyExchanges) def test_buildProtocolKexECDSA(self): """ ECDSA key exchanges are listed with 256 having a higher priority among ECDSA. """ f2 = self.makeSSHFactory() p2 = f2.buildProtocol(None) # The list might contain other algorightm. # For this test just check the order for ECDSA KEX. self.assertIn( b"ecdh-sha2-nistp256,ecdh-sha2-nistp384,ecdh-sha2-nistp521", b",".join(p2.supportedKeyExchanges), ) class MPTests(unittest.TestCase): """ Tests for L{common.getMP}. @cvar getMP: a method providing a MP parser. @type getMP: C{callable} """ if not cryptography: skip = "can't run without cryptography" if not pyasn1: skip = "Cannot run without PyASN1" if cryptography: getMP = staticmethod(common.getMP) def test_getMP(self): """ L{common.getMP} should parse the a multiple precision integer from a string: a 4-byte length followed by length bytes of the integer. """ self.assertEqual(self.getMP(b"\x00\x00\x00\x04\x00\x00\x00\x01"), (1, b"")) def test_getMPBigInteger(self): """ L{common.getMP} should be able to parse a big enough integer (that doesn't fit on one byte). """ self.assertEqual( self.getMP(b"\x00\x00\x00\x04\x01\x02\x03\x04"), (16909060, b"") ) def test_multipleGetMP(self): """ L{common.getMP} has the ability to parse multiple integer in the same string. """ self.assertEqual( self.getMP( b"\x00\x00\x00\x04\x00\x00\x00\x01" b"\x00\x00\x00\x04\x00\x00\x00\x02", 2, ), (1, 2, b""), ) def test_getMPRemainingData(self): """ When more data than needed is sent to L{common.getMP}, it should return the remaining data. """ self.assertEqual( self.getMP(b"\x00\x00\x00\x04\x00\x00\x00\x01foo"), (1, b"foo") ) def test_notEnoughData(self): """ When the string passed to L{common.getMP} doesn't even make 5 bytes, it should raise a L{struct.error}. """ self.assertRaises(struct.error, self.getMP, b"\x02\x00") class GMPYInstallDeprecationTests(unittest.TestCase): """ Tests for the deprecation of former GMPY accidental public API. """ if not cryptography: skip = "cannot run without cryptography" def test_deprecated(self): """ L{twisted.conch.ssh.common.install} is deprecated. """ common.install() warnings = self.flushWarnings([self.test_deprecated]) self.assertEqual(len(warnings), 1) self.assertEqual( warnings[0]["message"], "twisted.conch.ssh.common.install was deprecated in Twisted 16.5.0", ) PK ��Z��-�)� )� ntb_tool.konu �[��� ELF > � @ @ ) ( GNU ����h[ ,^4KE��=�}+� Linux Linux � UH��SH��H�?H��� H��P H��tA� H��f�H�{81ɺ � � H�]��� H�3H�� H�� � ��1��� � � UH��AUATA��SH��H�?H��� H��� H��ta� I��H�;H��� H��� � f�H�{ 1ɺ � � [A\A]]� H�3I��M��D��H�� H�� � �H��� � I��� @ � UH��SH��H�U�H�u�H��H�?eH�%( H�E�1�H��� H�@ � f�1�H�{� � � H�E�eH+%( u>H�]��� ��H� D�M�H�3H� D�E�H�� H�� HD�� �� ff. � �� UH��AWAVAUATSH��H�?H��� H�@X� �0 H�;Hc�CPH��H���� H�ƺ� � H�CXH���� �sP��~31��H�CXHc�H�RH��� H�CX���D����H�CXH�\9KPыK`���j H�KhE1�A�0 H�;Ic�D��H��L�$PH��� I��N�<!H�@8� A�GL�{hH�;M�IcGI��H���B �� � I�GL�{hM�I�OH��� A�W��~B1��I�OHc�H�@H���H�Kh��J�L!D�tH�KhJ�L!H�\L�{hM�A9W�H�;H��� H�@X� A�G L�{hH�;M�IcG H��H��H��<�� �� � I�G(H�KhJ�!H�p(H��tf�@ ��~@1��H�p(Hc�H��H�@H��H��D�4H�shHCXJ�t&(H�DH�KhJ�!9P �A��D9s`�����[1�A\A]A^A_]� ��[A\A]A^A_]� 1��j���1����H�CX ���� � UH��AVAUATSH��H�?H��� H�� H���� � H�;Hc� H��H��<��H����H���q �� � H��� H���4 ��� ��~;1��H��� Hc�H��� H��� ���D����H��� H�\9�� ɋS`���� H�KhE1�Ic�H�;H��L�$PH��� I��H�� N�,!H���� � A�E@L�khH�;M�IcE@H��H��H��<�� �� � I�EHH�KhJ�!H�pHH��ti�@@��~>1��H�pHHc�H���H�Kh��J�L!HD�tH�KhJ�L!HH�\H�KhJ�!9P@�A��D9s`�9���[1�A\A]A^]� 1��U���[��A\A]A^]� ǃ� 1����Hǃ� ��[A\A]A^]� 1��;��� � UH��AVAUATSH��H�?H��� H��8 H���v � H�;Hc�CpH��H��<��H����H���\ �� � H�CxH���% �Kp��~/1��H�CxHc�H��� H�Cx���D����H�CxH�\9KpՋS`���� H�KhE1�Ic�H�;H��L�$PH��� I��H��8 N�,!H���� � A�E0L�khH�;M�IcE0H��H��H��<�� �� � I�E8H�KhJ�!H�p8H��ti�@0��~>1��H�p8Hc�H���H�Kh��J�L!8D�tH�KhJ�L!8H�\H�KhJ�!9P0�A��D9s`�9���[1�A\A]A^]� 1��U���[��A\A]A^]� �Cp 1����H�Cx ��[A\A]A^]� 1��A���ff. � � UH��AWAVAUATSH��H��(H�5 eH�%( H�E�1�H��� H�H�xPH���� � H��� H��H���� H�پ� L�m�E1�I� H�� � H��� H��I� �� H�� � H��� H��I� �� H�� � H��� H��I� �� H�� � H��� H��I� �� H�� � H��� H��I� �� H�� � H��� H��I� �� H�� � H��� H��I� �� H�� � D��� E��~MD��H�� � L��� Ic̾� L��H��� H��I� A��H�� � D9�� �D�CpE���p �{`���7 E1�L�m�D��H�� � L��� Ic�L�{hL��H��H��� L�$P� I��I� �� M�H�� I�GPH�KhL�H�QP� H�KhI� �� H�� L�H�QP� H�Kh�� I� H�� L�H�QP� H�ChL�p��~YE1�D��H�� � L��� H�ChIc�L��H� RI� �� A��L�H��H�PPHH� H�ChL�D9x��H ��~UE1�D��H�� � L��� H�ChIc�L��H��I� �� A��L�H�PPHH(� H�ChL�D9x ��P@��~UE1�D��H�� � L��� H�ChIc�L��H��I� �� A��L�H�PPHHH� H�ChL�D9x@��@0E1���~PD��H�� � L��� H�ChIc�L��H��I� �� A��L�H�PPHH8� H�ChF9| 0�A��D9s`����H�E�eH+%( �- H��([A\A]A^A_]� H�8����A� L�m�~GD��H�� � L��� Ic̾� L��H��� H��HKxA��I� � D9cp�H��� H��I� �� H�� � H��� H��I� �� H�� � H��� H��I� �� H�� � H��� H��I� �� H�� � H��� H��I� �� H�� � ���HLJ� ���� � UI��H��H��ATSH��H�M�1�H��H��M��� eH�%( H�E�1�� ��uRI�D$�M�A�$A�t$H�8H��� H��x H��t/� ��H���HD�H�U�eH+%( uH��[A\]� H���H������ ff. � � UI��H��H��ATSH��M��� eH�%( H�E�1�I�D$H�0H��� H��0 tkH��H�M�1�H��� ��uRI�D$�M�A�$A�t$H�8H��� H��0 H��tV� ��H���HD�H�U�eH+%( uAH��[A\]� H���f�H�����H�� H�� � H����H����� ff. � � UI��H��H��ATSH��M��� eH�%( H�E�1�I�D$H�0H��� H�� tfH��H�M�1�H��� ��uMI�D$�U�A�4$H�8H��� H�� H��tV� ��H���HD�H�U�eH+%( uAH��[A\]� H���f�H�����H�� H�� � H����H����� � UH��AVI��AUI��ATH��(eH�%( H�E�H��� H�PH�:H��� H��( H��tX�I��p� H�� � H�}���� H�M�L��L��Lc�L��� H�U�eH+%( uH��(A\A]A^]� H������ � I��H��H��H��I��� H�HH��tUL�@ H��� ]� H������ � UH��AVAUATH��(eH�%( H�E�1�M��teI��I��I��H�?.� � H�� H�}�I9�H��IF�� H�M�L��L��Lc�L��� H�U�eH+%( uH��(A\A]A^]� H������ f D � UH��� H�H��H��� L��H �:���]� @ � UH��� H�H��H��� L��@ � ���]� @ � UH��� H�H��H��� L��P ����]� @ � UH��� H�H��H��� L��� ���]� @ � UH��� H�H��H��� L��� �z���]� @ � UH��� H�H��H��� L��� �J���]� @ � UH��� H�H��H��� L��� ����]� @ � UH��� H�H��H��� L��� ���]� @ � UH��AVI��AUI��ATI��H��0eH�%( H�E�H��� �H�@H�8H��� H��p H��tXH�u�� ��D�E�H�� � H�}�� L��H�M�L��Lc�L��� H�U�eH+%( uH��0A\A]A^]� ������� � UH��AVAUI��ATH��(H��� eH�%( H�E�1�H�BH�8H��� H�� H��tXI��I��2� H�� � H�}���� H�M�L��L��Lc�L��� H�U�eH+%( uH��(A\A]A^]� H������ @ � UH��AWAVAUATSH��L��� L�9I�~ �� I�^ H���� L9��� L)�I���� I��H9�HG�H��� I��H���� H�����ww1�H��H��H�E�� H�}�L��H��I���� L�E�H9�t$H)�L��I� H��I��I�$I�~L�� L�E�L��� �E1�H��L��[A\A]A^A_]� I�����I�������I�����f � � UH��AWAVAUATSH��L��� L�1H�u�I� �� I�_ H���� L9�v|L)� I��H9�HG�H��� I��H���� I�wH��H��L�� H�����w]� H��L��� H�}�H��L��� H9�t0H)�I�M�4$I��L��� �E1�H��L��[A\A]A^A_]� I����� I�����I�������I����� @ � H��� H�xH��t UL�H I��H��H��L��H��� ]� H������ � UH��AWAVI��AUI��ATI��SH��eH�%( H�E�H��� �L�x��?� I�?1�1�H��� H�@ � ��L��L��I��� L��H��H�M�L!�A� H������Y�E f�E�� H�U�eH+%( uH��[A\A]A^A_]� � � UE1�H��AWI��AVA��Hc�AUI��Ic�ATA��SH� IH��H��PI�uheH�%( H�E�1�H��H�BH\�H�<� H�{ H�}�t)H�E�eH+%( �� H��PD��[A\A]A^A_]� M�] A��?� I��� 1�1�L�]�L��H�@ � L���� L�]�L�M�L�E�D��H�M�D��I��� L��H�@@� A�����p����E�H�E�H�S�� H�p�H�E�H��H �H�E�I9�IF�E1�I��I��L �H��H�s I�E H��� H��� � H�CH��H���� H�E�H�KH�s I�} H��H��u|H��� D�M�H�@HH��tI��D��D��� D�M����� D�� H�}�D�M�H�� � H�M�I�EhI� �� H�}�H�TPH��� D�M�H�C(�u���A��H��� E1�D�M�H��� � H�C D�M�H�C H�C �4���H�KH�SA��I�} H�s �A���������A��� ���� ff. � � UI��H��H��AVAUH�M�ATI��1�SL��H��8M��� eH�%( H�E�1�� ��t&H�H�U�eH+%( �� H��8[A\A]A^]� � I�} H��� H��P H���� � H9E�te1�H�}�M�u8� �� H9E�t=H��uL� L��� H�u�� I�} H��H��� H��P H��u�1�H9E�u�H�u�L��� L���D�����t�H������4���1��v���� �� UI��H��H��AVAUH�M�ATI��1�SL��H��8M��� eH�%( H�E�1�� ��t&H�H�U�eH+%( �� H��8[A\A]A^]� � I�} H��� H��� � H9E�tX1�H�}�M�u � � H��uJ� L��� H�u�� I�} H��H��� H��� � H9E�u�H�u�L��� L���Z�����t�H������J���� ff. � � UH��AVI��AUI��ATI��H��(eH�%( H�E�H��� H�8H��� H� H��tP� ��H�� � H�}�� L��H�M�L��Lc�L��� H�U�eH+%( uH��(A\A]A^]� � ���� � UI��H��H��AUATSH��H�U�H��H��M��� eH�%( H�E�1�� ��uiD�m�A��� I�<$A��H��� u2H�@0� ��Hc �HD�H�U�eH+%( u*H��[A\A]]� �����H�@(������ ����H���� ff. � f�� UI��H��H��AWAVAUATI��H�U�SL��H��@I��� eH�%( H�E�1�L�s� ���� � ��� �����?)Ѓ�?� A� I��� I�>1�1�H��� H�@ � �]�I��A�ǀ�� ��A8�u)L��H�U�eH+%( �� H��@[A\A]A^A_]� 1�H�}�M�~� � H��ua� � H�u�L��� I�>1�1�H��H��� H�@ � I���E���<� ��8�u�H�u�L��� �g���H��c������X���H������O���� ff. � � UH��AWAVI��H�zAUI���� ATM��SH��H�� H�M�eH�%( H�E�1�� H���� I��H������� 1�H��H��� H��L��L��� H���� A� L��H�M�H�U�H�� � L��A��� A��uu�E�<stAM��th<cudH�u�I�>.� H�H��HD�H�U�eH+%( uHH�� [A\A]A^A_]� H�E�H��t#H�u�I�>� H�� L��� H���H����� H��� � � H�9 u.UH��� H�H��H��� H��` L��h ���]� 1�� � H�9 u)UH��� 1�H�H��H��� L��X �R���]� 1�� D � H�9 u.UH��� H�H��H��� H��� L��� � ���]� 1�� � H�9 u.UH��� H�H��H��� H��� L��� ����]� 1�� � H�9 u.UH��� H�H��H��� H��� L��� ���]� 1�� � H�9 u.UH��� H�H��H��� H��� L��� �M���]� 1�� H��� UH�ApH��H��t � ]� H�AhH��t�E1�1�� ]� � � UHc�H��AVA��H��AUI��H�PATI��SH�WhH��Ic�H� RH��HXH�{(� H�SH��tEI�<$H��� H�APH��t^D��D��� H�SI�<$H��� H�KE1�H�s H��� � H�C H�C H�C H�C( [A\A]A^]� H�AHH��t�D��E1�1�D��� H�SI�<$�ff. � � UI��H��H��ATSH��H�M�1�H��H��M��� eH�%( H�E�1�� ��t H�H�U�eH+%( uEH��[A\]� A�t$I�|$A�$����M��uH����A�t$I�|$A�$���t��� @ � UH��AVI��AUI��ATI��H��(eH�%( H�E�H��� �0H�@H�8H��� H�@H��tP� ��H�� � H�}�� L��H�M�L��Lc�L��� H�U�eH+%( uH��(A\A]A^]� � ���� f � � UHc�H��AVI��AUL�,@ATI��I��SH�_XL�H�{(� H�{ t.I�D$XJ�|(� I�<$�sH��� H�BpH��t<D��� H�C H�C H�C �C����H�C( [A\A]A^]� H�BhH��t�E1�1�D��� �ff. � � UH��AVAUATI��S�wP��~1ۉ�L�������A9\$P�A�L$`��~LI�D$hE1�Ic�1�H� �H�JL�4� �T���~ ��D��L������I�D$hB9\0�A��E9l$`�[A\A]A^]� ff. � � U�� H��AWAVAUATSH�� H��� H���! H�I��L�xH�� H�� L��M�t$ M�l$8� H�� H�� L��� H�� H�� L��� H��� H�PxH��tH��� ���3 H��� H�� H��tH��� ��� I�� ��� I�<$H��� H�@H���~ � Hc�A�D$`�X I�<$H��H��H���m �� � I�D$hH���K A�T$`��~+1��I�D$hHc�H�4�H� q��I�D$h��L�d�A;T$`|�L��� ���Å�tM1ɺ � L��� 1ɺ L��� � 1ɺ L�� � ��[A\A]A^A_]� L������Å�tUL������f���f�����H�� H��H�� � ����H�� H��H�� � H��� ���L������Å�u�I�<$H�� L��� �Å�u�L���D��X���I���D���[A\A]A^A_]� � �}��������I�D$h ������ff. � f�� UH��AWI��AVAUATI��S� H��xL��� H�}�eH�%( H�E�1�H9�HF�1�M�nH��� H��L��H�}�� H���� H�� �� H��p���H��x���H�� �D � H�}�� ���� I�}A�u �U���L��p���M��u)L��H�U�eH+%( �~ H��x[A\A]A^A_]� H��x���Mc6L��`���H��h���IcE M�mH� @I��H��I]XH�{ u�I�} H�M�H�U�D��H��� H�@`� ���� I�} L��`���H��� H�@hH��t(L��`���H��h���D��D��� L��`������� H�u�H�}�L��`���� L��`���H��H�Ct`L�C D��� H�}�H��h���D�sH�� H�C� I�UhK��H��I�FI� �� H��H�}�H�PP� H�C(���I�} D��D���m�����H����H������H�����H��H�� � �'���� � UH��SH�� H��� � H�;� H�;H��� H�@0� H���0���H�{1ɺ � � H�{ 1ɺ � � H�{81ɺ � � H�]��� � � UH��AWAVI��AUATS� H��8H�u��� L��� H�M�eH�%( H�E�1�H9�HF�H��� H��� I��E�UI�EE�] L�8A��?� I��� D�U�1�1�D�]�L��H�@ � D�U�L���� I��� L���U�D��L�M�L�E�H�@@H�M�� Lc�M��t1L��� H�E�eH+%( �| H��8L��[A\A]A^A_]� A�M H��L��H�� � E�ELc�I�ED�E�D��H�8H��� H�@H���� � D�E���H��K�<<H�� L)�� I�MH��H�� H�I�L)�K�<<� H��I�MH�� H�I�L)�K�<<� H��I�M H�� H�I�L)�K�<<� H��H�M�H�� Lc�M�L)�K�<,� H��H�M�H�� Lc�M�L)�K�<<� H�M�H�� Lc�M�L)�K�<,H��� H�U�H�}�L��Lc�L��M�� I������ D�E������I�Ǖ����u���I����q���� D � UH��AWAVAUATI��H�U�SH��0H�u�H�M�H�M�eH�%( H�E�H��� L�hI�EA�u H�8H��� H�@`� ���� � �� I9�IF�H��� I��H���� A�M H�� H��H��� I�} Lc��\ I�EE�EH�8D�E�D��H��� H�@H���W � D�E���H��K�<7H�� L)�� Lc�M�I�MH��K�<L�U�L)�H�� � L�U�H��H�M�Lc�H�� M�L)�K�<L�E�� L�E�H��H�M�Lc�H�� M�L)�K�<L�U�� L�U�I�MH��Lc�H�� M�L)�K�<L�E�� L�E�I�M H�� Lc�M�L)�K�<L�U�H��� L�U�H�U�L��H�}�Lc�L��M�� L��H��� Hc�H�U�eH+%( uHH��0[A\A]A^A_]� H���H��K�<7H�� L)�� Lc�M������ D�E������� H���HcӾ H�� � � � H�� L�]�� L�]�� A��H�� � � ��H�� �U��E�� �E��U�� ��H�� � � Hcо H�� �M�� �M�� IcҾ H�� D�]�D�U�� D�U�D�]�� � UH��AT� ��t1�H�� � H� H�� H�� H�� � A�ą�t H�= � D��A\]� UH�� H��� H�= � ]� message bits %#llx up down link is %s speed %d width %d port link db db_valid_mask db_mask db_event peer_db peer_db_mask spad%d msg_sts msg_inbits msg_outbits msg_mask msg_event msg%d peer%d link_event mw_trans%d peer_mw_trans%d no spad write fn %#x %#llx 0x%08x<-%d include/linux/thread_info.h mw%d %d %c %lli &tc->link_wq &tc->db_wq &tc->msg_wq doorbell is unsafe scratchpad is unsafe %lli:%zi peer_mw%d Inbound MW %d Port %d (%d) Window Address 0x%pK DMA Address %pad Window Size %pap Alignment %pap Size Alignment %pap Size Max %pap Outbound MW: %d Port attached %d (%d) Port attached -1 (-1) Virtual address 0x%pK Phys Address %pap Mapping Size %pap Window Size %pap ntb_tool drivers/ntb/test/ntb_tool.c doorbell vec %d mask %#llx bits %#llx Translation Address 0x%016llx /build/linux-mJOtZy/linux-5.15.0/drivers/ntb/test/ntb_tool.c /build/linux-mJOtZy/linux-5.15.0/include/linux/ntb.h tool_spad_write tool_peer_spad_write tool_link_event tool_db_event tool_msg_event tool_create_data description=PCIe NTB Debugging Tool author=Allen Hubbe <Allen.Hubbe@emc.com> version=2.0 license=Dual BSD/GPL srcversion=38361C1C48D1F1D6483A076 depends=ntb retpoline=Y intree=Y name=ntb_tool vermagic=5.15.0-133-generic SMP mod_unload modversions S �� module_layout �� simple_open %�j� ntb_unregister_client K�A __ntb_register_client �ebq debugfs_initialized p�� ntb_clear_ctx � �� __ubsan_handle_out_of_bounds 5� ioremap_wc P�}� ntb_default_peer_port_count �U� ntb_set_ctx |��c __x86_indirect_thunk_rdx T�� __init_waitqueue_head S9�� iounmap �4:) ntb_default_peer_port_number �U� debugfs_remove ��b __x86_indirect_thunk_r12 �n�� sscanf ��T __ubsan_handle_load_invalid_value ��H kstrtobool_from_user ��| ntb_default_port_number �T� finish_wait ��&� prepare_to_wait_event Q schedule uyH� init_wait_entry ?G __cond_resched "�t} kstrtoull_from_user �ܯF dma_free_attrs T�� dma_alloc_attrs a�H� __ubsan_handle_shift_out_of_bounds fGO� simple_write_to_buffer �k _copy_to_user D�b} memcpy_fromio � z kfree �B� memcpy_toio � _copy_from_user H�ۈ __check_object_size E:#� __kmalloc � � __x86_indirect_thunk_r8 ݷ�a simple_read_from_buffer ���� scnprintf ���f __x86_indirect_thunk_rcx %��G kstrtouint_from_user nJne snprintf ���� debugfs_create_file ?Fp debugfs_create_dir ���� devm_kmalloc ke�� __stack_chk_fail >:q� __dynamic_dev_dbg �9�[ __x86_return_thunk "#�> __wake_up �pHe __x86_indirect_thunk_rax �m�� __fentry__ � � � � 'char [32]' � '_Bool' � , < * 'long long unsigned int' 'long unsigned int' 'int' � � � � E 8 * ntb_tool GCC: (Ubuntu 11.4.0-1ubuntu1~22.04) 11.4.0 GCC: (Ubuntu 11.4.0-1ubuntu1~22.04) 11.4.0 �� � � � � � ����)� 6� C� T� e� u� �� �� ������ �� �� �� �� �� �� ����� � � ,� 8� D� Q� ^� k� � &� � (� @ � "� $� #� z� � t � �- � � %� �� @ IT ~ �L # @ �� � � �- U� � �. S �� V� @ �{ j � �� � � �� X� � !� "� $� '� �� �� ` �� R @ �� ` � )� �� 0 � �� .� �� .� @ �� 0� � �� 0� � � 4� � 6� @ � .� � '� 0� � 0� 8� =� :� @ J� <� � Y� .� � g� >� x� @� @ �� <� � �� .� � �� B� �� .� @ �� D� � �� B� � �� F� �� F� @ �� B� � �� F� � � F� � H� @ $� B� � 1� F� � =� F� K� B� @ ]� F� � n� F� � �� .� �� .� @ �� J� � �� L� � �� N� �� P� @ �� R� � �\ .� � �� B� �� B� @ �� B� � � F� � � F� � F� @ ,� T� � 5� R� � +� $� -� $� /� $� 2� 3� � � 1� $� � � 5� $� � � � 7� $� = E 9� $� ;� $� Z � =� $� E ?� $� A� $� C� $� E� $� Z � � G� $� I� $� K� $� Z M� $� O� $� Q� $� � S� ,� *� D� W� L� 4 �� 4 �� = T� \� 0 d� �C � ]� @ Y� � Z� � � E i� � @ t� � }� $� �� k @ �� k �� k � �� � �� _� � �� �( f� @ �� � �� d� � �� �� e� @ Dr � � \� �� �C �� _� @ [� �� �� �C � ]� @ �� �� �C � ]� @ � X �C � ]� @ � � � _� � �� �� c� @ � � %� d� � -� 9� e� @ Dr � � ^� `� a� b� �* g� 4 h� }� $� �C d� B� j� � ]� Z� l� �. S �� i� n� < �� � �� � ; � � w� p� < �� � �� / � ; � � �� r� �� p� �� r� �� r� �� p� �� W < � ]� �� / � ; � � �� B� �� y� < � ]� �� � ; � � � F� � F� � {� � ]� �C d� � }� � ]� d� -� � ?� � � ]� I� �� X� �� f� �� v� r� �� 6� �� p� �� 6� �� p� �� r� �� p� �� r� � r� � p� .� r� ;� r� N� p� b� p� p� p� �� r� �� p� �� r� � p� � p� � r� � r� � r� /� p� H� p� [� r� o� r� �� p� �� r� � "� }� $� �� �� � "� }� $� �� �� �� l� � ]� �C d� ; � �� � r� � p� ntb_topo NTB_TOPO_NONE NTB_TOPO_PRI NTB_TOPO_SEC NTB_TOPO_B2B_USD NTB_TOPO_B2B_DSD NTB_TOPO_SWITCH NTB_TOPO_CROSSLINK ntb_speed NTB_SPEED_AUTO NTB_SPEED_NONE NTB_SPEED_GEN1 NTB_SPEED_GEN2 NTB_SPEED_GEN3 NTB_SPEED_GEN4 ntb_width NTB_WIDTH_AUTO NTB_WIDTH_NONE NTB_WIDTH_1 NTB_WIDTH_2 NTB_WIDTH_4 NTB_WIDTH_8 NTB_WIDTH_12 NTB_WIDTH_16 NTB_WIDTH_32 ntb_client_ops ntb_client ntb_dev topo ctx_ops msi ntb_ctx_ops link_event db_event msg_event ntb_dev_ops peer_port_count peer_port_number peer_port_idx link_is_up link_enable link_disable mw_count mw_get_align mw_set_trans mw_clear_trans peer_mw_count peer_mw_get_addr peer_mw_set_trans peer_mw_clear_trans db_is_unsafe db_valid_mask db_vector_count db_vector_mask db_read db_set db_clear db_read_mask db_set_mask db_clear_mask peer_db_addr peer_db_read peer_db_set peer_db_clear peer_db_read_mask peer_db_set_mask peer_db_clear_mask spad_is_unsafe spad_count spad_read spad_write peer_spad_addr peer_spad_read peer_spad_write msg_inbits msg_outbits msg_read_sts msg_clear_sts msg_set_mask msg_clear_mask msg_read peer_msg_write ntb_msi mm_base tr_base tool_mw widx dbgfs_file tool_ctx ntb link_wq db_wq msg_wq outmw_cnt outmws peer_cnt inmsg_cnt inmsgs inspad_cnt inspads tool_mw_wrap mw tool_msg midx tool_spad sidx tool_peer inmw_cnt inmws outmsg_cnt outmsgs outspad_cnt outspads ntb_peer_mw_clear_trans tool_clear_mws tool_db_event tool_db_event_write tool_db_mask_read tool_db_mask_write tool_db_read tool_db_valid_mask_read tool_db_write tool_exit fn_read tool_fn_read fn_set fn_clear tool_fn_write tool_free_mw tool_free_peer_mw tool_init tool_init_msgs tool_init_mws tool_init_spads tool_inmsg_read tool_link_event tool_link_write tool_msg_event tool_msg_event_write tool_msg_inbits_read tool_msg_mask_write tool_msg_outbits_read tool_msg_sts_read tool_msg_sts_write tool_mw_read tool_mw_trans_read tool_mw_trans_write tool_mw_write tool_outmsg_write tool_peer_db_mask_read tool_peer_db_mask_write tool_peer_db_read tool_peer_db_write tool_peer_link_event_write tool_peer_link_read tool_peer_mw_read tool_peer_mw_trans_read tool_peer_mw_trans_write tool_peer_mw_write tool_peer_port_read tool_peer_spad_read tool_peer_spad_write tool_port_read tool_probe tool_remove tool_setup_dbgfs tool_setup_mw tool_spad_read tool_spad_write ntb_tool.ko }� ! n # � 2 ? � X � n � � � 8 � � � h � � 8 � p � � 8 � � P 8 � M - � = � L � ] ] ! o ` ~ � @ � � � � � � � � � @ # 7 � Q � d ` | ` � � @ � � � � � � � � @ � ' � � < p 8 T � � d � 8 | � � � p @ � � � � P , � � , � � , � � , � , @ , " p , : � , G � � W � � f 0 y @ � � @ @ � � � � � @ b � � � � � � � � , � � < 7 Q @ b l L U � � 9 � � @ � 0 ; � p @ � � @ � � @ 0! @ p! 8 3 �! � @ �" � T @# � h �# � z �$ � � P% � � ! � ! � ! � 8 � 8 8 � p � �' P � @* � �* k 2 � ) J @- H b a l � � x " � � � 0 � @ � P � ` � $ � $ ) � M Y ) / ; X l | � � � � � � � � " � � � & ; M a l a x � � � � � � 4 C S _ u � � � � � � � ! 0 I b v � � � � � __UNIQUE_ID_srcversion124 __UNIQUE_ID_depends123 ____versions __UNIQUE_ID_retpoline122 __UNIQUE_ID_intree121 __UNIQUE_ID_name120 __UNIQUE_ID_vermagic119 _note_9 _note_8 tool_msg_event __UNIQUE_ID_ddebug263.4 tool_db_event __UNIQUE_ID_ddebug262.5 tool_link_event __UNIQUE_ID_ddebug261.6 tool_init_mws tool_init_spads tool_init_msgs tool_setup_dbgfs tool_dbgfs_topdir tool_port_fops tool_link_fops tool_db_fops tool_db_valid_mask_fops tool_db_mask_fops tool_db_event_fops tool_peer_db_fops tool_peer_db_mask_fops tool_spad_fops tool_peer_port_fops tool_peer_link_fops tool_peer_link_event_fops tool_mw_trans_fops tool_peer_mw_trans_fops tool_peer_spad_fops tool_outmsg_fops tool_inmsg_fops tool_msg_sts_fops tool_msg_inbits_fops tool_msg_outbits_fops tool_msg_mask_fops tool_msg_event_fops tool_outmsg_write tool_peer_spad_write __UNIQUE_ID_ddebug275.2 tool_spad_write __UNIQUE_ID_ddebug274.3 tool_peer_spad_read tool_mw_read tool_fn_read tool_msg_outbits_read tool_msg_inbits_read tool_msg_sts_read tool_peer_db_mask_read tool_peer_db_read tool_db_mask_read tool_db_valid_mask_read tool_db_read tool_inmsg_read tool_spad_read tool_peer_mw_write tool_peer_mw_read tool_mw_write tool_peer_link_read tool_peer_link_read.cold tool_setup_mw tool_mw_fops tool_setup_mw.cold tool_msg_event_write tool_db_event_write tool_port_read tool_link_write tool_link_write.cold tool_peer_link_event_write tool_peer_link_event_write.cold tool_fn_write.part.0 tool_msg_mask_write tool_msg_sts_write tool_peer_db_mask_write tool_peer_db_write tool_db_mask_write tool_db_write ntb_peer_mw_clear_trans.isra.0 tool_free_mw tool_mw_trans_write tool_peer_port_read tool_free_peer_mw tool_clear_mws tool_probe __key.10 __key.11 __key.12 __UNIQUE_ID_ddebug277.0 __UNIQUE_ID_ddebug276.1 tool_ops tool_peer_mw_trans_write tool_peer_mw_fops tool_remove tool_mw_trans_read tool_mw_trans_read.cold tool_peer_mw_trans_read tool_init tool_client tool_exit __func__.18 __func__.17 __func__.16 __func__.15 __func__.14 __func__.13 __UNIQUE_ID_description260 __UNIQUE_ID_author259 __UNIQUE_ID_version258 __UNIQUE_ID_license257 .LC28 ntb_set_ctx ntb_default_peer_port_number __check_object_size _copy_from_user devm_kmalloc __this_module snprintf finish_wait scnprintf memcpy_fromio iounmap cleanup_module ntb_clear_ctx kfree simple_write_to_buffer prepare_to_wait_event __wake_up kstrtobool_from_user __dynamic_dev_dbg debugfs_initialized __fentry__ init_module __ntb_register_client __x86_indirect_thunk_rax memcpy_toio ntb_default_peer_port_count schedule __stack_chk_fail __x86_indirect_thunk_rdx kstrtouint_from_user __ubsan_handle_out_of_bounds __cond_resched init_wait_entry simple_open ntb_unregister_client dma_alloc_attrs debugfs_remove sscanf __x86_return_thunk ntb_default_port_number _copy_to_user __init_waitqueue_head __ubsan_handle_shift_out_of_bounds debugfs_create_file dma_free_attrs __x86_indirect_thunk_rcx __x86_indirect_thunk_r12 kstrtoull_from_user __ubsan_handle_load_invalid_value ioremap_wc debugfs_create_dir simple_read_from_buffer __x86_indirect_thunk_r8 __kmalloc � ��������$ � ��������> � ��������H � ��������R Y � ^ � ��������q � ��������� � ��������� � ��������� � ��������� � ��������� � � � �������� � �������� � ��������G � ��������^ � ��������w � ��������� � � � P � � ��������� � ��������� � ��������� � �������� � ��������� � ��������� � ��������0 � ��������] � ��������� � ��������� � �������� � ��������> � ��������n � �������� � ��������4 � ��������� � ��������� � ��������� � �������� � ��������. � ��������[ � ��������� � �������� � ��������� � ��������� � ��������� � ��������� � ��������� ��������) � ��������R ` Y : ^ � ��������o { ? � � ��������� @ � D � � ��������� � G � � ��������� � U � � ��������� � ] � �������� � % f * � ��������; � G n L � ��������b { o � ��������� � � � ��������� � � � ��������� � �������� @ : ' � ��������2 > ? J � ��������Z � a � m � ��������� � � � ��������� � � � ��������� � � � �������� ` # � ��������D { Q � ��������f ` � ��������� � � � ��������� � � �������� � ��������8 � E � ��������j @ o � ��������� � � � � ��������� � � � � � ��������� � � � � � ��������� � � � � � �������� � � � ��������9 � ��������A � ��������y � ��������� � ��������� � ��������� � ��������� � ��������B � ��������r � ��������� � ��������� � � p � � ��������� � ��������� � ��������2 � ��������] � ��������� � ��������� � � � � � ��������� � ��������� � �������� � �������� � ' � ��������< � ��������[ � ��������i � ��������q � ��������� � ��������� � ��������� � ��������� � ��������� � ��������� � � �������� � ��������8 � ��������F � ��������Q � ��������x � ��������� � ��������� � ��������� � ��������� � ��������� � �������� � �������� � ��������8 � ��������A � ��������h � ��������q � ��������� � ��������� � ��������� � ��������� � ��������! � ��������. < � ��������Q � ��������p � ��������| � ��������� � ��������� � ��������� � � � ��������� � �������� � ��������( � ��������1 � ��������� � ��������� � ��������� � ��������� � ��������� � �������� � ��������A � ��������� � ��������� � ��������� � ��������� � ��������� � �������� � ��������A � ��������j � ��������p � ��������| � ��������� � ��������� ��������� � �������� � ��������7 � ��������< � ��������A � ��������� � ��������� � � ��������' � ��������� � ��������� � ��������� - � � �������� � � ��������M � ��������� � ��������� � ��������� � �������� � �������� � ��������< � ��������Q � ��������X � ��������h � ��������y � ��������� � ��������� � ��������� � �������� � ��������7 � ��������< � ��������S � ��������h � ��������t � ��������� � ��������� � ��������� � ��������� � ��������� � ��������# � ��������, 2 : � ��������O � ��������n � ��������s � ��������| � ��������� � ��������� � ��������� 3 � � �������� � ��������" � ��������/ � ��������A � ��������� � ��������� ~ � � ��������� � ��������� j � �������� � ��������) � ��������: � ��������T � ��������f H y � ��������� � ��������� � ��������� � �������� � ��������# � ��������C 6 H � ��������S � ��������w � ��������� � ��������� � ��������� � ��������� � ��������� � ��������% � ��������, � ��������1 � ��������` � ��������g � ��������q � ��������� � ��������� � ��������� � ��������� � ��������� � ��������� � ��������%! � ��������,! � ��������1! � ��������e! � ��������l! � ���������! � ���������! � ���������! � ���������! � ���������! � ���������! � ��������" � ��������>" � ��������k" � ���������" � ���������" � ���������" � ���������" � ��������8# � ��������A# � ���������# � ���������# 2 �# � ���������# � ���������# � ���������# ���������# � ���������# � �������� $ � ��������3$ � ��������R$ � ���������$ � ���������$ � ���������$ � ��������>% � ��������Q% � ��������s% � ���������% �% > �% � ���������% �% K �% � ���������% �% V �% � ���������% � �������� & � ��������?& � ��������i& � ���������& � ���������& � ���������& � ��������' � ��������3' v =' B' � ��������N' b X' 8 ]' � ���������' p �' ~ ���������' � ���������' � ���������' � ��������<( � ��������K( � ��������s( � �( � ���������( � ��������!) � ��������]) � ���������) � ���������) � �) � ���������) � �) � ��������-* 2* � ��������<* � ��������A* � ��������Y* � ��������a* � ��������t* � ���������* � ���������* � ���������* � ���������* � ���������* � �������� + � ��������A+ � `+ � ���������+ � ���������+ � ���������+ � ���������+ � �+ � ��������, � ��������$, � ,, � ��������:, � K, � ��������Y, � j, � ��������x, � �, � ���������, �, � ���������, % �, � ���������, ; �, � ��������- � ��������- ��������7- � ��������A- � ���������- � ���������- � ���������- Q �- � ��������. � ��������(. j 0. � ��������O. � T. � ��������i. � |. � ���������. � �. � ���������. ( �. � ���������. � �. � ��������/ � �������� / � ��������A/ � ��������S/ � [/ � ��������k/ ��������{/ � �������� ` � �������� � ! � * � ��������3 � > p C � ��������H � R 0 ] � ��������j f t P y � ��������~ � � @ � � ��������� � � � � � ��������� A+ � �������� � �������� � ��������% ��������, 3 � : � ? � ��������M ��������R � ��������] � �������� � � �������� �������� � �������� � �������� p � ( 0 � 8 @ @ � H � P � X p ` � h P p � x � � � � � @ � p � � � � � � � 0 � @ � @ � � � @ � � � � � � � � @ � � 0 p ( � 0 � 8 0! @ �! H �" P @# X �# ` �$ h P% p �' x @* � �* � @- � � � � � � � � � � � � � � � P 0 � � � � � P � � � 0 p � � 8 @ � � @ � P � � � ` � p � x � � � � � � � � � � � � � � � � � � � � � p 0 � � � � � � P � � @ � p � � 0 p � � @ � P � X 0! � � ` � p @- x �' � � � � � @ � 0 � � � � � �* � �" � � � � p � @ 0 � � � � @ P � � � p � � 8 � � � @ � P @# � � ` � p � � � p x p � # � � F � � / = $ ( - , � 0 � 4 q 8 \ < @ � D H � L � P � T & X � \ ; ` W d R h � l " p � t ! x � | S � u � � � �! � �! � " � �" � �# � Q$ � �$ � �% � & � >& � ) � \) � s* � _+ � �+ � , � �- � . G � v � � � � � � $ � ( � , 0 � 4 � 8 � < Z @ � D � H 7 L w P � T � X \ 7 ` g d � h � l o p t x | o � { � 6 � � � � 6 � m � � � � � $ � + � _ � f � � � � � � � � � $! � +! � d! � k! � �! � �! � j" � �" � �# � �$ � =% � ' � �' � �( � �* �+ @/ \ } } � } + L � � B K $ { ( z 0 � 4 � 8 � @ � D � H � P "' T 0' X * ` )' d K' h b H � 0 H @ � P H ` � p H � � 0 P% 8 @* @ H P � X � ` H p � x � � � � � � � ` v 8 @ ` H P b p x � � � � � � � � � � P � � @ ( 0 P X 0 ` h 8 � 8 � .symtab .strtab .shstrtab .note.gnu.build-id .note.Linux .rela.text .rela.text.unlikely .rela.init.text .rela.exit.text .rodata.str1.1 .rela__mcount_loc .rodata.str1.8 .rela.rodata .modinfo .rela.retpoline_sites .rela.return_sites __versions .rela__bug_table .rela__jump_table .rela.data .rela__dyndbg .rela.gnu.linkonce.this_module .bss .comment .note.GNU-stack .BTF .gnu_debuglink @ $ . d 0 ? � �/ : @ @� x* & J (0 � E @ �� � & ^ �0 a Y @ �� & n S1 " i @ �� x & y 2 u1 8 � �4 � � @ H� � & � 2 H6 � � 7 � � @ � & � �R � � �S � � @ � � & � pT � @ � ` & � �U � �b $ � @ `� � & �b p @ �� � & c � @ �� � & & e � ! @ �� � & 4 �f � @ / @ h� 0 & N @j S 0 @j X \ �j l �j � q \� p� � ' ~ P� � �� � 0�� *�H�� ���0��1 0 `�He0 *�H�� 1�o0�k0F0.1,0*U #Build time autogenerated kernel key.;��ڽ��> R�i8B0�`0 `�He0 *�H�� � /"d"��D<�=�m�-��UB+��.�yǼ� Ӊ��u٢y��e�� -� �k7,���1<��E� ��˻w��fo�:��[2��§� ��B$Y�G��¿D�+�S����Bp ���ѯ���X����G�A}XO� �e N0߯Jq��Ș|�Xf X ���9N��֗�o��c5���q4��U8T�5�Uql���fm��W��}S� u9{��*,���,u) �?�X/�3*{r� `�IQX��7ѳp����V�>T�s�z�)� ��R����� ��;ǁƺ� �C�:Ąɱ�m��Wt� � 0���3j�ű\� m�#S���.z��ӑ�,5DTHr~���=�>�y/Tw����w^g�)��K��A&!��S�?酛�gᐭ,����=a;L��q�ۈ���Q�m� %*f �I�j�V�R���r�':%�bRb� C��8X �bٵZ�җ�>p[G�Ϸ�����Gb���G���H��]�ַE/�� �~Module signature appended~ PK ��ZE�)�� � ntb_perf.konu �[��� ELF > � @ @ / . GNU (]̤,��-:Iã�bQT�e=� Linux Linux � H�U��� � H���tH�H�@p9�� ��]� ff. � � UH��ATSH��H��H�?H��� H�@X� 9C � H�;H��� H��8 H��t� ���� D �? +K ��?� H�;I�����I��H��� H�� H���� � H�;�S �TR9��� H��� H��� � L!�L9��� HǃH HǃP f�1�H��[A\]� H�3H�� H�� � �S���H�3H�� H�� � 1�H��[A\]� HǃH HǃP f�H��1�[A\]� H�3H�� H�� � 1�H��[A\]� 1�����f�H�;H�� � ����E���H�3H�� H�� � ��� UH���O 1�H��� � � ]� � � U1ɺ � H��AUATE1�SH��H� �G����H��X� Mc�I�� w<1ɺ � H��� I�� w3H�{@A��H��p� A�� u�[A\A]]� L��H�� � �L��H�� � � D � UH��AVI��AUI��H�� ATI�� H��eH�%( H�E�H��� H�}��H<� L��H�M�L��Lc�L��� H�U�eH+%( uH��A\A]A^]� � ff. � � UH��AUATSH��H��0D�- L�'eH�%( H�E�1�A��� A���� � H+C@H�C@f� D f�1�H�U�eH+%( �D H��0[A\A]]� �KL�C8H�� H�� I�4$� �H�K@I�4$H��S㥛� H�� H��H��H��H��?H��H��1�H)�H�C8H��KH�� I��� 1��l���H�K@D�KH��S㥛� H�� I�4$H��H��?H��H��H)�D��I��H�� � �#���� �C ��t A�D$8��yA�D$8�������������1�H�}�L�k� �A�D$8��x � � H�u�L��� �C ��u�H�u�L��� A�D$8��������� f � � UH��AWI��AVI��AUI��ATSH��H��H�L�`@� <� ��� I�E(I��A�� H� �HXL�Pp�� � L�� L�]�H ���� ��L �H�H���� � � L��L�U�M+t$ Mt$0� I��H���� L�U�L�]�L�x� �H��� H� �H+ H�4L��E1�A� H�� L��L��H��H5 � I�D$ I��H���� A�D$�e I�}(H���� H�H���� L��� M���� A� L��L��L��.� H���� L�h8I�|$� H�@( �A�T$���� �J ��� H��L�`@H�P� L��A�ƅ��� � �A�E I�}(H�H��� � I�E E1��@8��A��F�4�����H��D��[A\A]A^A_]� � � ��tM�T$ ����L��H��L��� �A�����L��� �� �� H�E�� H�E��@���� H�E�� H�E��)���A������q���A���f���H� �I���ff. � �� U�� H��AVAUATSH��H��L�'eH�%( H�E�1�M�l$@I�$I�}��� � H�C0H���� H��I�D$@H�p� D�5 A��� 1�A��u$H�U�eH+%( �� H��[A\A]A^]� H�E� �M�1�L��H�}�H�� � H�C(H��� H� I�UE1�� I�u(H�xp� I�E0H�S(H�L�RpH���� f��C 1��h����KM�M0M�E(L��H�� H�� � �ϸ��;���� � � UH��AWAVI��AUI��ATI��SH��H�?H��� H��@ H���3 � H�;I��H��� H��P H��� � L!�H��� H��� H�;H��� H��p H���� 1�L��� A�H�;�����H��� H��p H��t� L��� ��I�$H�;H��� H��p H� ����H��t� L��� I�$H�� H �I�$H�;H��� H��X H��tL��� f�1�[A\A]A^A_]� M�$A�H�� H�� H�3� 1�[A\A]A^A_]� 1����E1�������������������ff. � �� UH��SH�H��D�O �wH�8H��� H�BPH��tBD��� H�H�8H��� H�K8E1�H�SHH�s@H��� � H�CH H�]��� H�BHH��t�E1�1�D��� H�H�8�f D � UH��AWAVL�w�AUATSH��L��H�� �k�����t�C�H�� [A\A]A^A_]� H�C�� H�E�H�@@H�EЀ�?� � H��� H�EȀ�?� H�u�A� L�k�I��H�FL�~ I9�LG�L�H�E�� H�C�H�E�H;C�w �dH�E�L�x L�k�� H�E�H;C�vIL��L��L��L��� ������ Lc�K�'H9E�v�H�u�L�~ L9�r�I��M�� H�E�H;C�w�L��� ����C�D�- L�c�A��� A��t=L�k�I�E H��h H��tL��� ��tUL�k�H�C�H�@@H�p0H��u_L��� �A�L$81�I�|$ � � � H�{�� H�� [A\A]A^A_]� � I�E H��p H��t�L��� �H�PI�E E1�� H�xp� L�k��ff. � � UHc�H��SH���H�Wpf�H�5 H�SP� � H�]��� H�H��H�� H�� H�0� ��ff. � � UH��SH��H�M�H�U�H�u�H��eH�%( H�E�H��P � ��u2HcE�H��H�PH�CH�<ЋE�t0��uDH�E� H�G�4���H�E�eH+%( u H�]��� H�E� H�G@� ������ � ff. � f�� UH��AUATI��Sf�L���5���[A\A]]� H�?A��H��� H��� � I�<$H��H��� H��� H��t;D��� I��I�4$I��D��H�� H�� � L������[A\A]]� H��� � I���ff. � � UH��ATI��f�L�����L�e��� H�7H��� H��P H��t4H��� I�4$H��H�� H�� � L���>���L�e��� 1���� UH��AWAVAUATI��SH�� H� D�weH�%( H�E�1�D�{L�+A��?� I��� 1�1�L��H�@ � L��� I��� D��D��L��L�M�L�E�H�@@H�M�� ��� I�D$@H;E�� H�M�H��H�q�H �H��I�|$H I�t$@t L���{���I�t$@H�I�T$8E1��� H��� H��� � I�D$HH��� H�E�I�L$8H�;H��H��� H��� M�D$@A�T$ A�t$H�@HH��t � ��� � L�����H�E�eH+%( uH�� [A\A]A^A_]� � ff. � �� UH��ATSH��H��H��H �" �O��?� H�;� H��H��� H��� � H�;� �K ���� E1�Icľ A��H��H�PH�CH�<������C D9�ׅ�~oE1�Ic�A��H��H�PH�CH�|�P� �C D9�ۅ�~CE1�Ic�H�;H��H�PH�SH�@ �4@H��� H�� H��t1�� A��D9c ��K��?wH�;� H��H��� H��� � H�;H��� H�@0� H��[A\]� H�?H��� H��@ H��t'� H�;H��H��� H��` H����������1���� @ � UH��AUATL�g�SH���H�w r;�H�s �( �H�s �� �H�s �� �H�s r&[A\A]]� H�G�H�W�1�L��H��H � ��C( H�{0H�� H�� � �c �H�C�H�P���� H�8�s�D�K�H��� H�BpH���� D��� H�{� �u���L���O���[A\A]]� L�k�L�C�H�K��S�I�} �s�H��� H�@hH��t � ��� �K H�{(� ����H�C�H�S� L��H��H � ���L���������L9`@�>���� H�BhH���R���E1�1�D��� �@���f�� UH��AUATSH��H��H�?H��� H��� � H�;H��H��� H��� � H�;H�� H��� A�ą��� H�;H��� H��H �� H�� H��t8� A�Ņ�~,H�;H��� H�� H��t �����D��� A��E9�uԋK��?�f H�;A� I��H��� L��H��� � A��f�H�;E��uUH��� ����������H�@(� H�;� H��D��[A\A]]� H�3H�� L��H�� � H�;E��t�� D��H��[A\A]]� H��@ H���� � H�;I��H��� H��H H���� � H�;L �H��� H��H��` H��t� �I��H��?� H�;A� I��H��� H��h H��t=L��� A��f���H�3L��H�� H�� � ����E1��d���L���A������ f � � UI��H��H��ATSH��H�M�1�H��H��M��� eH�%( H�E�1�� ��t H�H�U�eH+%( uFH��[A\]� �E�P��� w$�I�l$ rA�D$<�A�d$�H���H���H����� � UH��AWAVAUATSH�� H��� H�= H�u�H�U��� � H�M�H�]�� H���1 I���H�k � H�C@� L��E1�H�� L��� H��S㥛� �H� Lc��HI�MM�E H��H��?H��L��H��I��1�I)�D��I��H�� P� H�I�XA��I��pA�� tJIc�H�� �� E�EA���t۾ K�<4L)�E��t�D��H�� A��I��p� H�I�A�� u�H�E��`�H�U�H�u�M��L��H�}�� I��L��� H�e�L��[A\A]A^A_]� I�����H�� � �h���I�����f. � � UH��AWAVA��AUI��ATSH��H��L�' D L��A�� H�81�1�H��� H�@ � �KI����� �����?)Ѓ�?� I���p A�D$I�<$�s�@H��� H��( H��t$� ���� A�D$I�<$�s�@H��� H��0 D���H��t^� A�D$I�<$I�� �sL��T@H��� H��0 H��t/� A�D$I�<$�s�@H��� H��0 H��tD��� �K ��?� I�<$� H��H��� H��� � f�1�H��[A\A]A^A_]� ��I�4$I��H�� H�� � H�����K ��?� � I�4$H�� H�� H��H��� 1�뗺 ��� �@B � A��tH��P���������k���������a���ff. � �� UH��AWAVAUATI��SH��H�?eH�%( H�E�1�H��� H� H���B � �E�A�D$����A�D$ ���� I��� E1��E��Ic�D��H��H�PI�D$H� �L�#D�{I�<$H��� H�@H���A � E�o9E�}A�|$�uE�|$E��H�CXD�{ H��� H�� H�CXH�� H�C`L�sPH�Ch �Cx � A�D$ D9��^���A�|$��� ��� A�D$ E1����� Ic�H��H�PI�D$H� �H�U�L�;L�kL��I�?A�wH��� H�@`� ��t&H�U�eH+%( �# H��[A\A]A^A_]� H�SH�u�I�?� H�C H���� H�E�H�C(H� H��t H;C� A��E9t$ �S���f�1��� ���A�L$I�4$H�� H�� � 1��]���E1�E�l$�������I�<$H��� H� H��tW� ������I�<$1�H��� H�@H��tE� �������A�D$ I�D$�@ ���� �E����� 맸������ �� f D � U�� H��AVAUATSH��` H��� H���F I��H��� H��H�@H���" � Hc�A�E �� H��H��H���0 �� H��� I�EH���� I�] I�� ��� L�����A�ą�t[D��A\A]A^]� I�EA�E<I�} H�� H�� I�]XI��� I�E@� Ic�H�� �� D�c�H��H�� A��L�k�H�� �C8���� H�CHL�s@H��pH� H�@H�C� A�� u�L�����A�ą��T���L�����A�ą��A���I�E H�5 H��� H�� H��uH��� � I��X H��H��� L��I� �� H�� � I��X L��I� �� H�� � I��X L��I� �� H�� � I��X H� �@ H�� � I��X H� �@ H�� � I��X H� �@ H�� � D��[A\A]A^]� � ����A���#���H�� � �a���E��� ���I�E A����� D � �G ���� UH��AVI��AUATS1�H����I�t$p�� ��A9^ ~rHc�1�H��H�PI�FL�$�1�I�$H�8H��� H�@ � E�l$D�� ��A�T ��?)ʃ�?� L��s��I�l$pr�1�L�����A9^ �H��[A\A]A^]� � L�����b���� f � � UH��AUATE1�SH�� H��X � H����H�{ 1ɺ �C8����� H��X� Mc�I�� w<1ɺ � H��� I�� w3H�{@A��H��p� A�� u�[A\A]]� L��H�� � �L��H�� � �ff. � f�� UH��AWAVI��AUI��ATI��SH��H���O��?� I�>� H��H��� H��� � A�$ A�V 1����C H�1�H��H�PI�FL�<�1�I�H�8H��� H�@ � A�OI����� �����?)Ѓ�?� I���� A�G I�>�4@H��� H�� H���� � ����� A�E A�G �����I�>�t@H��� H�� H��t� ��H�A�G I�>�t@H��� H�� H� ����H��t � H�H�� H �H�A�G I�>�4@H��� H�� H��t ������ f�H��1�[A\A]A^A_]� A�$��A�$A;F �����H������[A\A]A^A_]� L�A�M H�� H�� I�6� 1�H��[A\A]A^A_]� �� UH��AWAVAUA� ATSH��(L9�H�u�L��� LF� H�U�H�M�L��� H��� H��H�� L��I��� I�?E�wHc�H��� H� H���� � ��L��I�< E��H�� H)�� L��H�� Lc�L�H)�I�< � Lc�I�GLӨ� L��I�< H�� H)�� H�L�4A�G �E� ��e� H�� � L��H�K@H�� H�I�L)�K�<4� L��H�K8H�� H�I�L)�K�<4� �E�H�IƋE�A9G �Q HcE�I�?H��H�PI�GH� �H��� D�CD�K H�@D�M�D��D�E�H���� � D�M�D�EЉ�L��K�<4H�� L)�� H� L��H�� Lc�H�CpMƨ@H� K�<4HE�L)�� H�K L��H�� H�I�L)�K�<4� L��H�K(H�� H�I�L)�K�<4� L��H�KH�� H�I�L)�K�<4� H�KL��H�� H�I�L)�K�<4� H�KHL��H�I�L)�K�<4H���q���H�� � �E�H�IƋE�A9G �����H�U�H�u�M��L��H�}�� L��H��� Hc�H��([A\A]A^A_]� I�G@I�?D�pH��� H�@D��H��tK� ��L��E��I�< H�� H)�� H�L�4���� D�E�D�M̉��g���� ���7���� ���H���o���ff. � f�� UH��AWAVAUI��ATSH��H���u�L�' D I�<$H��� H��H H���� � I���E�� D�m�H�1�1�H�8H��� H�@ � D�sH��D�� ��A���?)Ѓ�?� L���0 I�<$H��� H��X H��� L��� A�ƅ��� I�<$�sH��� L��x M��t�Mк .� I�<$H��� H��P H��t� I���� I�<$H��� H��x I�� �sL��H��t.� � I�<$�sH��� H��x H��t �M�1�� H��D��[A\A]A^A_]� �M�I�4$I��H�� H�� � ���� ��� �@B � �m������A������E1�����A������A���� D � UI��H��H��AWI��1�AVH�M�L��AUATSH��8I��� eH�%( H�E�1�� A�ƅ�� HcE����� ;C �� H��H�PH�CH� �H�{xL�+� ���� �I�m �� A�E<I�]@E1�I��� A�E8�A��H��pA�� tPIc�H�� �P �C����H�C� H�C� A�E<D9�~�H�5 H�ڿ A��H��p� A�� u�� A�E8��>�A�e�Ic�E��ID�H�U�eH+%( �� H��8[A\A]A^A_]� H���H���1�H�}�M�e � � H��u,� � H�u�L��� A�U8���H�u�L��� �= ���tA���q���1ɺ � L��A�E8����I�]X� Mc�I�� w{1ɺ � H��� I�� wPH�{@A��H��p� A�� u�A������ ���H�������H�� � ���H����� L��H�� � �L��H�� � �q���H�;H�� � ���� Hc�H��H�� �M�� �M�� A��H�� � � ��H�� �E�� �E�� �ʾ H�� L�U�L�]ȉM�� �M�L�]�L�U�� A��H�� � � �SI�<$H�� � �A�L$81�I�|$ � � � H�{0� ���� �SL��H�� � I�E0 H�{(� �A��H�� � � �Ѿ H�� �M�� �M�� �EЉ�H�E�H�� �S�H�8� �E�� �Ѿ H�� �M�� �M�� H�;H�� � � H�;H�� � I�|$H � L��� � H�� � ��H�;I�T$@H�� � � H�;I�T$@H�M�H�� � � H�;H�� � � Ic־ H�� � � HcѾ H�� �M�� �M�� HcѾ H�� �M�� �M�� H�8H�� � H�{�� H�C�� I�} H�� � � H�ʾ H�� H�M�� H�M�� HcѾ H�� �M�� �M�� HcѾ H�� �M�� �M�� HcѾ H�� �M�� �M�� Hcо H�� �M�L�E�� L�EЋM�� H�CH�L��H�� �SH�8� � I�} H�� � � HcҾ H�� H�E�� H�E�� HcѾ H�� �M�� �M�� Hcо H�� �M�L�E�� L�EЋM�� Hcо H�� H�M�� H�M�� �H��� U�= H��ATv� H�� � � �5 @85 sH�� @�5 � 1ҾB H�� A��� H� H��tW� ��t1�H�� � H� H�� H�� H�� � A�ą�tH�= � H�= � D��A\]� UH�� H��� H�= � H�= � ]� � Not enough memory windows Message service initialized Message service unsupported Command services unsupported %hhu %d: copied %llu bytes %d: lasted %llu usecs %d: %llu MBytes/s %d: Failed to map DMA addr CMD recv: %d 0x%llx %d: Got error %d on test CMD exec: %d Recv invalid command Msg status bits %#llx Too big inbuf size %pa > %pa Failed to alloc inbuf of %pa Unaligned inbuf allocated &x->wait Freeing while test on-fly MSG sts unmasked %#llx DB bits unmasked %#llx Peer %d test statistics: %d: error status %d CMD send: %d 0x%llx DB ring peer %#llx Global port index %d &perf->twait &pthr->dma_wait DebugFS unsupported info run threads_count chunk_order total_order use_dma up down Test status: on-fly with port %d (%d) idle Link status: %s Out buffer addr 0x%pK Out buff phys addr %pap Out buffer size %pa In buffer addr: unallocated In buffer addr 0x%pK In buffer size %pa In buffer xlat %pad[p] perf_wq ntb_perf drivers/ntb/test/ntb_perf.c Scratchpad service initialized Scratchpad service unsupported %d: Failed to get DMA channel %d: Map MMIO %pa to DMA addr %pad DB vec %d mask %#llx bits %#llx Couldn't get inbuf restrictions Failed to set inbuf translation Failed to set outbuf translation %d: copied %llu bytes in %llu usecs, %llu MBytes/s Peer %d outbuf reduced to %pa Performance measuring tool info: Local port %d, Global index %d Port %d (%d), Global index %d: Out buffer xlat 0x%016llx[p] 6Chunk order reduced to %hhu 6Total data order reduced to %hhu /build/linux-mJOtZy/linux-5.15.0/drivers/ntb/test/ntb_perf.c /build/linux-mJOtZy/linux-5.15.0/include/linux/ntb.h /build/linux-mJOtZy/linux-5.15.0/include/linux/dmaengine.h perf_db_event perf_msg_event perf_enable_service perf_msg_cmd_recv perf_spad_cmd_send perf_spad_cmd_recv perf_init_service perf_sync_test perf_init_test perf_init_peers perf_cmd_exec perf_msg_cmd_send use_dma total_order chunk_order max_mw_size parm=use_dma:Use DMA engine to measure performance parmtype=use_dma:bool parm=total_order:Total data order [2^n] to transfer parmtype=total_order:byte parm=chunk_order:Data chunk order [2^n] to transfer parmtype=chunk_order:byte parm=max_mw_size:Upper limit of memory window size parmtype=max_mw_size:ulong description=PCIe NTB Performance Measurement Tool author=Dave Jiang <dave.jiang@intel.com> version=2.0 license=Dual BSD/GPL srcversion=E659F54652AB1658585E18D depends=ntb retpoline=Y intree=Y name=ntb_perf vermagic=5.15.0-133-generic SMP mod_unload modversions �� �� �� �� S �� module_layout F?�� param_ops_ulong ��� param_ops_byte �Wz param_ops_bool �� simple_open %�j� ntb_unregister_client �� destroy_workqueue K�A __ntb_register_client �ebq debugfs_initialized �%�I alloc_workqueue �~�� _printk ��;. wait_for_completion_interruptible �.� kstrtoint_from_user � � __x86_indirect_thunk_r8 E:#� __kmalloc �U� debugfs_remove P�}� ntb_default_peer_port_count �"} debugfs_create_bool W�EZ debugfs_create_u8 ���� debugfs_create_file ?Fp debugfs_create_dir T�� __init_waitqueue_head ���� devm_kmalloc ��| ntb_default_port_number �4:) ntb_default_peer_port_number ��~� devm_ioremap_wc ]� usleep_range_state �戯 kmem_cache_alloc_trace 1� kmalloc_caches 7`%� kstrtou8_from_user �d� ntb_link_event �U� ntb_set_ctx DRf _dev_warn ��֓ complete_all �A�` __init_swait_queue_head ĕ,/ flush_work p�� ntb_clear_ctx T�� dma_alloc_attrs 6�� queue_work_on '��� system_highpri_wq ��P dma_unmap_resource �ܯF dma_free_attrs ���f __x86_indirect_thunk_rcx ��F� __sw_hweight64 �+;' dma_release_channel � z kfree g-Ob dma_map_resource ��A� __dma_request_channel 7�A get_random_bytes '�� __kmalloc_node �(�L phys_base _i� refcount_warn_saturate �B� memcpy_toio ���� msleep �K&� dmaengine_unmap_put |��c __x86_indirect_thunk_rdx �/� __x86_indirect_thunk_r9 ��h? dma_map_page_attrs l e� vmemmap_base ^��| page_offset_base ��� dmaengine_get_unmap_data ��T __ubsan_handle_load_invalid_value �T� finish_wait ��&� prepare_to_wait_event Q schedule uyH� init_wait_entry ?G __cond_resched e�?� ktime_get ke�� __stack_chk_fail ݷ�a simple_read_from_buffer ���� scnprintf � �� __ubsan_handle_out_of_bounds �-� cancel_work_sync "#�> __wake_up a�H� __ubsan_handle_shift_out_of_bounds �-� _dev_err >:q� __dynamic_dev_dbg �pHe __x86_indirect_thunk_rax �9�[ __x86_return_thunk �m�� __fentry__ 3 " 2 H 3 " 2 � j 3 " 2 �� 'perf_thread [32]' � � t '_Bool' � � � � $ � � � 6 ' % � < * � 'long long unsigned int' 'unsigned int' 'long unsigned int' 'int' � � � � � � � � � � � � � � c T ' ntb_perf GCC: (Ubuntu 11.4.0-1ubuntu1~22.04) 11.4.0 GCC: (Ubuntu 11.4.0-1ubuntu1~22.04) 11.4.0 �� � � � � � ����)� 6� C� T� e� u� �� �� ������ �� �� �� �� �� �� ����� � � ,� 8� D� Q� ^� k� � &� � (� @ � "� $� #� z� � t � �- � � %� �� @ IT ~ �L # @ �� � � �- U� � �. S �� V� @ �{ j � �� � � �� X� � !� "� $� '� �� �� ` �� R @ �� ` � )� �� 0 � �� .� �� .� @ �� 0� � �� 0� � � 4� � 6� @ � .� � '� 0� � 0� 8� =� :� @ J� <� � Y� .� � g� >� x� @� @ �� <� � �� .� � �� B� �� .� @ �� D� � �� B� � �� F� �� F� @ �� B� � �� F� � � F� � H� @ $� B� � 1� F� � =� F� K� B� @ ]� F� � n� F� � �� .� �� .� @ �� J� � �� L� � �� N� �� P� @ �� R� � �\ .� � �� B� �� B� @ �� B� � � F� � � F� � F� @ ,� T� � 5� R� � +� $� -� $� /� $� 2� 3� � � 1� $� � � 5� $� � � � 7� $� = E 9� $� ;� $� Z � =� $� E ?� $� A� $� C� $� E� $� Z � � G� $� I� $� K� $� Z M� $� O� $� Q� $� � S� ,� *� D� W� L� U� ����d� s� �� �� �� �� �� �� � 0[ \� �C @ � ` �� � =� E � �� S �� C @ �� = � �� = � � E �� S @ ^� � � G& � � � � � ` &� $� � @ *� ` �( ^� � /� � 9� k ?� E � E� � J� ^� <. _� @ T� a� @r ]� d� �r Dr � �r [� f� p 0[ \� r� @ w� E ` �� k � = �e @ {& S � | � Gg � �# @ �@ � � Z� ]� ^� Y� `� \� � c� � Y� b� ]� �* f� P g� F� ^� �\ Y� �� i� 0[ \� �� k� �� e� & S {& S @ ; �� m� �. S �� �� o� < �� � �� / � ; � � �� q� �� q� �� q� < �� � �� � ; � � � u� � u� 0[ \� 1� x� F� +� 6 �1 �e �) S ]� {� m� k� �� W F� ^� �� � �� � �� k� �� k� �� e� �� �� �� 6� 0[ \� �C � �\ c� �) � �� �� F� ^� �\ Y� �) �� �� � 6� �� "� &� $� � �� �� "� &� $� � �� %� �� F� ^� 7� �� H� �� [� �� n� �� }� x� �� �� ntb_topo NTB_TOPO_NONE NTB_TOPO_PRI NTB_TOPO_SEC NTB_TOPO_B2B_USD NTB_TOPO_B2B_DSD NTB_TOPO_SWITCH NTB_TOPO_CROSSLINK ntb_speed NTB_SPEED_AUTO NTB_SPEED_NONE NTB_SPEED_GEN1 NTB_SPEED_GEN2 NTB_SPEED_GEN3 NTB_SPEED_GEN4 ntb_width NTB_WIDTH_AUTO NTB_WIDTH_NONE NTB_WIDTH_1 NTB_WIDTH_2 NTB_WIDTH_4 NTB_WIDTH_8 NTB_WIDTH_12 NTB_WIDTH_16 NTB_WIDTH_32 ntb_client_ops ntb_client ntb_dev topo ctx_ops msi ntb_ctx_ops link_event db_event msg_event ntb_dev_ops peer_port_count peer_port_number peer_port_idx link_is_up link_enable link_disable mw_count mw_get_align mw_set_trans mw_clear_trans peer_mw_count peer_mw_get_addr peer_mw_set_trans peer_mw_clear_trans db_is_unsafe db_valid_mask db_vector_count db_vector_mask db_read db_set db_clear db_read_mask db_set_mask db_clear_mask peer_db_addr peer_db_read peer_db_set peer_db_clear peer_db_read_mask peer_db_set_mask peer_db_clear_mask spad_is_unsafe spad_count spad_read spad_write peer_spad_addr peer_spad_read peer_spad_write msg_inbits msg_outbits msg_read_sts msg_clear_sts msg_set_mask msg_clear_mask msg_read peer_msg_write ntb_msi perf_cmd PERF_CMD_INVAL PERF_CMD_SSIZE PERF_CMD_RSIZE PERF_CMD_SXLAT PERF_CMD_RXLAT PERF_CMD_CLEAR PERF_STS_DONE PERF_STS_LNKUP perf_peer outbuf_xlat out_phys_addr dma_dst_addr inbuf_xlat inbuf_size init_comp perf_ctx ntb pcnt busy_flag twait tsync tcnt test_peer cmd_send cmd_recv perf_thread tidx dma_sync dma_wait perf_cmd_exec perf_cmd_recv pthr perf_copy_chunk perf_db_event perf_dbgfs_read_info perf_dbgfs_read_run perf_dbgfs_read_tcnt perf_dbgfs_write_run perf_dbgfs_write_tcnt perf_disable_service perf_dma_copy_callback perf_dma_filter perf_enable_service perf_exit perf_free_inbuf perf_init perf_init_peers perf_init_service perf_init_test perf_link_event perf_msg_cmd_recv perf_msg_cmd_send perf_msg_event perf_remove perf_service_work perf_setup_inbuf perf_spad_cmd_recv perf_spad_cmd_send perf_sync_test perf_terminate_test perf_thread_work ntb_perf.ko tY� " $ ' � # � 2 ? � X � n � � 8 � � � 2 � @ � � " � 8 � ` � � `! � " � 8 " �& � 4 � F " 0 8 _ " � 8 w 7 � � ( � � � � � � P � e ' � " � 8 � " p 8 " 8 8 % 7 9 � I L I ^ � H m " � 8 � � | � " 8 � � � � � � � ) � � x � U " � 8 � � 3 � M 0 � [ " � 8 t � p � " h 8 � P t � � � � � � � 6 > ` � ' t 6 2 � 7 F p O " 8 g " P 8 � @ � 0 � � � � � " � 8 � " � 8 � � e ` � � ' " 8 4 O ! I � T ' ] ' f ' x ` � @ � � p � � � � � ! � � � � " � 8 � F P# c . " H 8 G � ! ^ �( _ s $ { � � � � � . � � � � @ � ` � � � � � � � � � � + 3 B 3 ] ( m � � I 4 � } � ( ( � � � � 4 � ! P ( 5 � M � 3 h � x ( � � � 3 2 � e ) � � � � ; J Z j w % � � � � � � � . � � � 0 : I [ n � � � � � � � � - 9 A ] f w � � � � � � , > T d s � � � � � � � + 8 E h | � � � � � � � : L _ w � � � � __UNIQUE_ID_srcversion124 __UNIQUE_ID_depends123 ____versions __UNIQUE_ID_retpoline122 __UNIQUE_ID_intree121 __UNIQUE_ID_name120 __UNIQUE_ID_vermagic119 _note_9 _note_8 perf_dma_filter perf_init_service __UNIQUE_ID_ddebug281.9 perf_spad_cmd_send perf_spad_cmd_recv __UNIQUE_ID_ddebug282.8 perf_msg_cmd_send perf_msg_cmd_recv __UNIQUE_ID_ddebug280.10 __UNIQUE_ID_ddebug283.7 perf_init_service.cold perf_dma_copy_callback perf_terminate_test perf_dbgfs_read_tcnt perf_sync_test __UNIQUE_ID_ddebug289.3 __UNIQUE_ID_ddebug290.2 __UNIQUE_ID_ddebug291.1 perf_sync_test.cold perf_copy_chunk perf_copy_chunk.cold perf_init_test __UNIQUE_ID_ddebug286.4 perf_init_test.cold __UNIQUE_ID_ddebug275.14 perf_free_inbuf.part.0 perf_thread_work perf_thread_work.cold perf_cmd_exec.isra.0 __UNIQUE_ID_ddebug276.13 perf_cmd_recv.isra.0 perf_cmd_recv.isra.0.cold perf_db_event __UNIQUE_ID_ddebug277.12 perf_msg_event __UNIQUE_ID_ddebug278.11 perf_setup_inbuf.isra.0 perf_setup_inbuf.isra.0.cold perf_disable_service perf_disable_service.cold perf_service_work __key.26 perf_service_work.cold perf_enable_service perf_ops __UNIQUE_ID_ddebug285.5 __UNIQUE_ID_ddebug284.6 perf_enable_service.cold perf_dbgfs_write_tcnt perf_dbgfs_read_run __UNIQUE_ID_ddebug271.18 __UNIQUE_ID_ddebug272.17 perf_spad_cmd_send.cold perf_init_peers __UNIQUE_ID_ddebug295.0 perf_init_peers.cold perf_probe __key.27 __key.28 perf_dbgfs_topdir perf_dbgfs_info perf_dbgfs_run perf_dbgfs_tcnt perf_probe.cold perf_link_event perf_link_event.cold perf_remove __UNIQUE_ID_ddebug273.16 perf_spad_cmd_recv.cold perf_dbgfs_read_info __UNIQUE_ID_ddebug274.15 perf_msg_cmd_send.cold perf_dbgfs_write_run perf_wq perf_init perf_client perf_exit __func__.39 __func__.38 __func__.37 __func__.36 __func__.35 __func__.34 __func__.33 __func__.32 __func__.30 __func__.25 __func__.23 __func__.22 __UNIQUE_ID_use_dma270 __UNIQUE_ID_use_dmatype269 __param_use_dma __param_str_use_dma __UNIQUE_ID_total_order268 __UNIQUE_ID_total_ordertype267 __param_total_order __param_str_total_order __UNIQUE_ID_chunk_order266 __UNIQUE_ID_chunk_ordertype265 __param_chunk_order __param_str_chunk_order __UNIQUE_ID_max_mw_size264 __UNIQUE_ID_max_mw_sizetype263 __param_max_mw_size __param_str_max_mw_size __UNIQUE_ID_description262 __UNIQUE_ID_author261 __UNIQUE_ID_version260 __UNIQUE_ID_license259 ntb_set_ctx ntb_default_peer_port_number ntb_link_event alloc_workqueue param_ops_ulong devm_kmalloc __this_module queue_work_on __kmalloc_node __init_swait_queue_head finish_wait scnprintf cleanup_module ntb_clear_ctx kfree dma_release_channel usleep_range_state prepare_to_wait_event __wake_up param_ops_byte system_highpri_wq dma_unmap_resource __dynamic_dev_dbg kmem_cache_alloc_trace debugfs_initialized __dma_request_channel __fentry__ init_module kstrtou8_from_user __ntb_register_client __x86_indirect_thunk_rax dma_map_page_attrs memcpy_toio _printk ntb_default_peer_port_count schedule __stack_chk_fail refcount_warn_saturate __x86_indirect_thunk_rdx __ubsan_handle_out_of_bounds page_offset_base __cond_resched __sw_hweight64 init_wait_entry _dev_err debugfs_create_bool simple_open destroy_workqueue ntb_unregister_client dma_alloc_attrs debugfs_remove wait_for_completion_interruptible dmaengine_get_unmap_data phys_base _dev_warn kstrtoint_from_user dmaengine_unmap_put __x86_return_thunk ntb_default_port_number __init_waitqueue_head complete_all vmemmap_base __ubsan_handle_shift_out_of_bounds debugfs_create_file dma_free_attrs param_ops_bool __x86_indirect_thunk_rcx ktime_get cancel_work_sync __x86_indirect_thunk_r9 dma_map_resource devm_ioremap_wc get_random_bytes __ubsan_handle_load_invalid_value debugfs_create_u8 debugfs_create_dir simple_read_from_buffer __x86_indirect_thunk_r8 msleep __kmalloc kmalloc_caches flush_work � ��������. � ��������A � ��������b � ��������k ��������� � ��������� � � ��������� � �������� ` `! & � ��������0 8 7 � < � ��������K R � W � ��������f � ��������q �& | � � ��������� � 0 � � ��������� � ��������� U � � ��������� � � � � ��������� � �������� � �������� � ��������! � ��������P � ��������m � ��������� � ��������� � ��������� � � ��������� � � � ��������� � ��������� s � �������� � ��������9 � ��������> � ��������Q � ��������i � 3 � � ��������� � ��������� y � � � � �������� 8 , � 4 � ��������T p v � { � ��������� � ��������� � ��������� � ��������� � ��������� � �������� � �������� � ��������< D H m c � � ��������� � �������� � �������� � ��������n � ��������� � � � ��������� � ��������� � �������� � �������� � ��������7 � ��������G � ��������N � ��������^ � ��������u � ��������� � ��������� � ��������� � �������� � �������� " � N � ��������k p � ��������} � � � ��������� � � ` � � � � ��������� � �������� � ��������9 � ��������[ � ��������f � ��������� � ��������� � ��������� � �������� � ��������2 � ��������@ � G O � ��������` � ��������� � ��������� � ��������� � ��������� � �������� � ��������! � ��������[ � ��������f � { e � � � " � � ��������� � �������� B - � ��������J X z � ��������� � ��������� � ��������� � ��������� � ��������� � ��������� � �������� � ��������! � ��������; � ��������I � ��������S � ��������` � g o � ��������� � ��������� � �������� � �������� � �������� � 1 � ��������R � ��������k � ��������� � ��������� � � � � � ��������� � ��������� � ��������� � ��������� � �������� � ��������* ) 1 h 6 � ��������H � ��������Q � ��������� � � ��������� � � ��������� � � 9 � ��������G � ` � � � ��������� � � � ��������� � ��������� � ��������� �& � 2 � ��������" � ��������} � ��������� � ��������� � �������� � �������� � ��������- � ��������X Q a � ��������� � ��������� � ��������� � � � � ��������* � ��������H � ��������s � ��������{ � � � ��������� � ��������� p � � ��������� � �������� � ��������0 � ��������: p B � ��������b �& y � ��������� � ��������� � �������� � �������� � �������� � ��������% / � 4 � ��������A � ��������S � ��������h � ��������� � ��������� � ��������� � � � ��������� � � P � ��������# � 1 � ��������i � ��������� � ��������� � ��������� � ��������� � L � ��������> � W � ��������� ( � � ��������� � � �������� � �������� � ��������/ � ��������? � D � ��������a � ��������� � ��������� $ � � ��������) � ��������X � ��������� � ��������� � � ��������� � ��������� � � � � ��������� � 3 � � ��������. � ��������a � ��������� � �������� � ��������4 ? � O ` [ � ��������� � ��������� � �������� � �������� , K F � ��������[ G b g � ��������� � ��������� � ��������� � ��������� � ��������� � �������� � �������� � ��������1 � ��������Y � ��������� � ��������� � ��������� � ] � � �������� j & � ��������A x ��������� � ��������� l � ` � � � � ��������� @ � � � � ��������� � � �������� � # � ( � ��������6 � B � G � ��������U a � f � ��������v � ��������{ � ��������� � � � ��������� � ��������' � ��������H � | � ��������� � ��������� � ��������� � ��������� � ��������! � �������� ! � ��������-! � ��������7! � <! � ��������H! ` M! � ��������a! � ���������! � �! � ���������! � �������� " � :" � ��������o" � ���������" � ���������" � ���������" � ��������# � ��������'# � .# � 6# � ��������K# � ��������Q# � ���������# � ���������# � �# � ���������# � ���������# � �# � ���������# � $ � ��������&$ � .$ � ��������O$ t T$ � ��������b$ � s$ � ���������$ � �$ � ���������$ � ��������% � % � ��������% � % � /% � ?% � ��������M% ^% � ��������l% & }% � ���������% @ �% � ���������% � �% � ���������% V �% � ��������& � ��������& � ��������1& � ��������T& � ��������g& � o& � ��������& � ���������& � ���������& � ���������& � ��������' � ��������+' � ��������M' � z' � ���������' � ���������' � ���������' � ��������!( � ��������7( � ��������H( H O( T( � ��������m( � ���������( � ���������( � ��������) � ���������) ���������) � ���������) � ���������) � ��������* � �������� * � �������� * � ��������2* � ��������c* � ���������* � ���������* � ���������* @ �* � ���������* � ���������* �* � ���������* �* � �������� � �������� # @ + � ��������3 � > ` C � ��������H � R � Z � ��������c D q � � ��������� m � � � � ��������� " � @ � � ��������� � ��������� � ��������� * � � � � �������� � �������� @ � ��������" X 1 � 9 � ��������B � R � ] � ��������e ? t � | � ��������� { � � � ��������� � � � � � ��������� � � � � � � | � � ��������� ^ � � ��������� � � @ � �������� � � � �������� � ( - � ��������2 � A � I � ��������Q � ` � h � ��������p � z � � ��������� � � � � ��������� � � � � ��������� � � � � ��������� � � � � � �������� � � � ��������$ � 3 � ? � ��������K � \ ` g � ��������l , w z | � ��������� � � � � � ��������� H � ` � � ��������� �! � � � � ��������� " � � � �������� M' � �������� � " � ( � ��������/ � 6 � ? ( F � K � ��������Y � d � ��������k ��������u � ��������� � � � ��������� ��������� � � � � � � � ��������� ��������� � ��������� ��������� � ��������� � �������� � � �������� �������� � �������� ��������$ � ��������* � �������� e @ � � ( P 0 8 � @ H � P X ` � h 0 p � x P � � � ` � � � 0 � � � ` � ` � � � � � � `! � P# � �& � �( � � � � Z � � 0 u ~ $ � ( � , � 0 � 4 8 � < � @ D � H � L R P ') T �) 0 � 8 0 � � P � X �( � � p P# � � p � x 0 � � � � � ( � 0 � 8 � H � P � X � ` � p � x � � � � � � a � � � l � � 8 Z $ � ( � , � 0 4 � 8 < y @ � D � H j L � P � T X � \ � ` � d h � l � p t , x � | ) � r � � � � � � / � x � � � � � � g � � � � � � � � � � � ( � W � � � � � � � � � � � � � � X � & � �! � �! � 9" � n" � �" � �" �# �$ S& ' *' y' �' �' �' $ ( - % e � � � 8 � $ ( M , 1 0 _ 4 � 8 Z < � @ R D H Q L � P � T G X � \ ` � d G h l R p � t . x � | � � � � u � { � � � ,! � �" � # � J# � 0& � 6( � �) � � � ) � * " E � � $ � ( Z 0 � 4 � 8 � @ � D � H � P � T ? X � ` � d � h b p � t � x � # � 6 � : � 6 � W � � A � V � � � � � � � � � � � B � � � � � z � � � � � � � � � � � ? O * �" # � �& $ ;( ( r P � P 0 8 � @ P P X � ` P p x � � P � � � � P � � � � P � � � � P � � � P � @ P P � ` P p � � P � � � P � � p x � � P � @ � � � P � @ � � � P � @ � � � P � @ � � P @ � P 0 @ 8 r @ P P @ X � ` P p @ x � � P � @ � � � P � @ � � � P � @ � � � @ � � 0 � 8 ^ � � � G 8 � @ � H � P � p � x � � � � � � � � � � � � y � � � � � � � ` � ( � 0 � P � X ` � h � � � � � � � � � � � � � � � � � � � 8 0 � 8 � @ � H h � p x � � ) � � � � � � � � � � � � � � � @ � ( � H � P X � ` � � � � � � � � � � � ` � � � 3 � � � ` � 8 � 8 � .symtab .strtab .shstrtab .note.gnu.build-id .note.Linux .rela.text .rela.text.unlikely .altinstr_replacement .rela.init.text .rela.exit.text .rela.altinstructions .rela__mcount_loc .rodata.str1.1 .rodata.str1.8 .rela.smp_locks .rela.rodata .modinfo .rela__param .rela.retpoline_sites .rela.return_sites __versions .rela__jump_table .rela.data .rela__dyndbg .data..read_mostly .rela.gnu.linkonce.this_module .bss .comment .note.GNU-stack .BTF .gnu_debuglink @ $ . d 0 ? � �* : @ P� @) , J �+ E @ �� � , Y �/ t �/ � o @ �� p , � �0 . @ �� � , � �0 � @ �� 0 , � �0 � � @ �� � , � 2 �1 � � 2 �5 � �8 X � @ �� , � 9 � � @ �� , � �= 9 @ � � @ �� � , �@ ( @ H� � , # �A � @ 8� 8 , 1 �B A �V 0 < @ p� X , S �W � N @ �� , ^ x\ ( Y @ �� , " g �` �` � @ z @ � 0 , % � @d � 0 @d X � �d � �d � � ${ 8{ X - � �� � 8� � 0�� *�H�� ���0��1 0 `�He0 *�H�� 1�o0�k0F0.1,0*U #Build time autogenerated kernel key.;��ڽ��> R�i8B0�`0 `�He0 *�H�� � �T�� 3�v�y����(6Q�++��i��ZAF"�4���ϻ焀KI������<���W亹Ia?w�9� g=",YC���D�oYԣ��K�nkJ*]'L��|�� ]��pjx�Uo�&�@^�M����f����o���Y�قO��� 1�T�� }�_dxD�Ɠ~ a?�r\��- T+�͚ٖ�,Xi�Ë��vU�%)ͧg�cV�%���L�(t�>��KD�- �IhTjT�W!�r`\� &y� ��\xD#�\ ZQqj5_BO��`�/�}ȇ��,���3N)��2L(�Y���'�ܟ����5�eO��d�K>�|{(ϔL��&�� i+��עk%�T�. _� 1� ��u�ι��6����W��Q�0�W��C0_+!^ݣo�t�σ���x���.��+ں���W�2?C�T�N���( #vls�)e�~��{� F/c���c=��+���n©�;� k�ѷ��71������0ۭk]�{�� �~Module signature appended~ PK ��Z�Y��r �r ntb_pingpong.konu �[��� ELF > �c @ @ 0 / GNU M��U�He�Ғ��kj� Linux Linux � UH��SH�� H�{x� H�;H��� H�@0� H�;� H�{� H�;H�sHH��� H��� � f�H�]��� H�3H�� H�� � H�]��� ff. � �� UH��ATI��SH��H��� H�PxH��t$� ��t D � ���C I��$� H�� H��tL��� ��� I��$� H�@L��H���� � �? )���?� I��$� H�����L��H��H��� � H!�H9�� I��$� H�� H��t|L��� ��~hI��$� H�� H��tCL��� ��~7I��$� H��8 H���� L��� ���� 1�H��[A\]� f�1�H��[A\]� I��$� H��8 H��� L��� ���p���� f�� ���������L��H�� H�� � 1�H��[A\]� f��H�� L��H�� � �d���H�� L��H�� � �H�� L��H�� � 1��%���� �t��� @ � UH��AWAVAUATI��SH��H�?H��� H� H���' � A��I�<$H��� H�@H��� � A��A� D��I��E��� 1��� A9�|(��D9�t I�<$��H��� H�@H��u�� A9�}�A��?�� �? M�|$HI��������I�����)؉�I���I��?� M�|$h��?� �@ D)�?� I�����I��M!�M�L$`f�H��[A\A]A^A_]� I�L$HM�D$hH�� H�� I�4$� H��[A\A]A^A_]� � A������� A�����A��?w �@ M�|$HI��I�D$h D)�� � � �� U�� �� H��ATI��S� H���� L� H��H�x� �@p � �@t � H�C0 H�� ��b H�����H�;H��� H��� � H�;H��H��� H��� � A�ą�� H�;H��� H��8 H��tu� H�;��~iH��� H��H H���� � H�;I��H��� H��@ H��t� H�;I �H��� H��` H���� L��� A�ą�u|H�;H�� H��� A�ą�ucH�;����������H��� H�@(� H�;� H�H��� H�� H��t4H�5 � H�Kp�� H�� H�CxH��� D��[A\]� H��� ��H���A��[D��A\]� E1�����A�����ff. � �� UH��H��ATI���J�����uL�����L�e��� @ � UH��AWAVAUATSH��H�?L�cH�sHH��� H��� � L��� H�;1�1�H��� H�@ � I��L#k`�� �M��H�;D��H��� H�@H���� � A��A��?� L�{tL��� D�� L��H��D�kXH�CP�% f��5 L�� 1�Hi�@B � [A\A]A^A_]� H#ChtI�H��A���k���H�;L�kP�sXH��� H�@H��te� ��H�3M��H�� H�� � �f�[A\A]A^A_]� H�3H�� H�� � [A\A]A^A_]� � A������� ���ff. � @ � UH���b���]� ff. � �� UH��AUATSH��H��H�?eH�%( H�E�1��E� H��� H�� H��� 1�� H�;A��H��� H��p H���� 1�H�u�� H�;A��H��� H��X H��t H������ f��CpH�;H�sHH��� H��� � H�;H�sHH��� H��� � �5 1�H�{� Hi�@B � H�E�eH+%( unH��[A\A]]� H�;�u�H��� H�@H��t?� A��H�3E��D��H�� H�� � �N���A������)���A��������� A���� ff. � � UH���r���]� ff. � �� UH��AUATL�glSH��D�ohL��� H�{��sPH��� L��0 M��tD��1�.� H�{��sPH��� L��x M��tD��1�.� H�{�H��� f�H�sHH��� � H�{�H�s@H��� H��� � L��% 1�[A\A]]� H�@�sPH��t1� ��H�s�H�� E��E��H�� � H�{�H��� �� ����H�� L��� ���� Hc�H��H�� �M�� �M�� H�� L��� ���� Hc�H��H�� �M�� �M�� Hc�H��H�� � � HcӾ H�� � � 1�Icվ H�� � � Ic־ H�� � � � � � UH��AT� ��t1�H�� � H� H�� H�� H�� � A�ą�t H�= � D��A\]� UH�� H��� H�= � ]� � �Ping-pong cancelled Doorbell is unsafe Scratchpad is unsafe Unsupported DB configuration Scratchpads unsupported Messages unsupported count Got no peers, so cancel ntb_pingpong Scratchpads and messages unsupported Inbound db %#llx, prev %#llx, next %#llx Ping-pong started with port %d, db %#llx Pong spad %#x, msg %#x (port %d) Ping port %d spad %#x, msg %#x drivers/ntb/test/ntb_pingpong.c /build/linux-mJOtZy/linux-5.15.0/drivers/ntb/test/ntb_pingpong.c O O pp_setup pp_pong pp_init_flds pp_ping pp_check_ntb pp_clear delay_ms unsafe parm=delay_ms:Milliseconds to delay the response to peer parmtype=delay_ms:uint parm=unsafe:Run even though ntb operations may be unsafe parmtype=unsafe:uint description=PCIe NTB Simple Pingpong Client author=Allen Hubbe <Allen.Hubbe@emc.com> version=2.0 license=Dual BSD/GPL srcversion=8E61D69C1F94C0C31082466 depends=ntb retpoline=Y intree=Y name=ntb_pingpong vermagic=5.15.0-133-generic SMP mod_unload modversions �� �� S �� module_layout ��z param_ops_uint %�j� ntb_unregister_client K�A __ntb_register_client �ebq debugfs_initialized � � __x86_indirect_thunk_r8 ke�� __stack_chk_fail ���f __x86_indirect_thunk_rcx ��U� hrtimer_start_range_ns NJq pv_ops d��� _raw_spin_lock D��� debugfs_create_atomic_t ?Fp debugfs_create_dir �d� ntb_link_event �U� ntb_set_ctx ��H hrtimer_init ���� devm_kmalloc ��| ntb_default_port_number �4:) ntb_default_peer_port_number a�H� __ubsan_handle_shift_out_of_bounds �-� _dev_err P�}� ntb_default_peer_port_count |��c __x86_indirect_thunk_rdx >:q� __dynamic_dev_dbg �9�[ __x86_return_thunk VmfJ hrtimer_cancel p�� ntb_clear_ctx �pHe __x86_indirect_thunk_rax �U� debugfs_remove �m�� __fentry__ 3 3 2 1 � � 'long long unsigned int' 'int' 5 � � � � � � � ntb_pingpong GCC: (Ubuntu 11.4.0-1ubuntu1~22.04) 11.4.0 GCC: (Ubuntu 11.4.0-1ubuntu1~22.04) 11.4.0 �� T T � � � ����)� 6� C� T� e� u� �� �� ������ �� �� �� �� �� �� ����� � � ,� 8� D� Q� ^� k� � &� � (� @ � "� $� #� z� � t � �- � � %� �� @ IT ~ �L # @ �� � � �- U� � �. S �� V� @ �{ j � �� � � �� X� � !� "� $� '� �� �� ` �� R @ �� ` � )� �� 0 � �� .� �� .� @ �� 0� � �� 0� � � 4� � 6� @ � .� � '� 0� � 0� 8� =� :� @ J� <� � Y� .� � g� >� x� @� @ �� <� � �� .� � �� B� �� .� @ �� D� � �� B� � �� F� �� F� @ �� B� � �� F� � � F� � H� @ $� B� � 1� F� � =� F� K� B� @ ]� F� � n� F� � �� .� �� .� @ �� J� � �� L� � �� N� �� P� @ �� R� � �\ .� � �� B� �� B� @ �� B� � � F� � � F� � F� @ ,� T� � 5� R� � +� $� -� $� /� $� 2� 3� � � 1� $� � � 5� $� � � � 7� $� = E 9� $� ;� $� Z � =� $� E ?� $� A� $� C� $� E� $� Z � � G� $� I� $� K� $� Z M� $� O� $� Q� $� � S� ,� *� D� W� L� � S� $� x{ @ W� @ ]� � d� � [ M� @ �( E � T j � Dr � � Y� �* [� \� S� $� m� ^� �. S �� z� `� �� W �� � o, Z� �� d� �� 6� �� d� �� "� S� $� �� h� �� "� S� $� �� j� �� d� �� � ntb_topo NTB_TOPO_NONE NTB_TOPO_PRI NTB_TOPO_SEC NTB_TOPO_B2B_USD NTB_TOPO_B2B_DSD NTB_TOPO_SWITCH NTB_TOPO_CROSSLINK ntb_speed NTB_SPEED_AUTO NTB_SPEED_NONE NTB_SPEED_GEN1 NTB_SPEED_GEN2 NTB_SPEED_GEN3 NTB_SPEED_GEN4 ntb_width NTB_WIDTH_AUTO NTB_WIDTH_NONE NTB_WIDTH_1 NTB_WIDTH_2 NTB_WIDTH_4 NTB_WIDTH_8 NTB_WIDTH_12 NTB_WIDTH_16 NTB_WIDTH_32 ntb_client_ops ntb_client ntb_dev topo ctx_ops msi ntb_ctx_ops link_event db_event msg_event ntb_dev_ops peer_port_count peer_port_number peer_port_idx link_is_up link_enable link_disable mw_count mw_get_align mw_set_trans mw_clear_trans peer_mw_count peer_mw_get_addr peer_mw_set_trans peer_mw_clear_trans db_is_unsafe db_valid_mask db_vector_count db_vector_mask db_read db_set db_clear db_read_mask db_set_mask db_clear_mask peer_db_addr peer_db_read peer_db_set peer_db_clear peer_db_read_mask peer_db_set_mask peer_db_clear_mask spad_is_unsafe spad_count spad_read spad_write peer_spad_addr peer_spad_read peer_spad_write msg_inbits msg_outbits msg_read_sts msg_clear_sts msg_set_mask msg_clear_mask msg_read peer_msg_write ntb_msi pp_ctx ntb in_db out_db out_pidx pp_check_ntb pp_db_event pp_exit pp_init pp_init_flds pp_link_event pp_pong pp_probe pp_remove pp_setup pp_timer_func ntb_pingpong.ko kt�� " $ ( # 7 2 � ? C X O n X � j 8 � � � � � $ � 8 � � � $ � 8 M ( � $ p 8 $ � 8 ! $ 8 8 9 P K � � X $ 8 p P j � @ � � � � P � ( � 0 , � ` � � $ � 8 � " ` � $ � 8 � � � r $ 8 5 � A $ P 8 Y a a " � k " s � � ( � 0 � @ � 9 � 9 � ( h P 9 + � E ( ( T q g � , � � ) � � � � � � � � & � ( 5 M " \ j | � � � a � � � + 4 J Y p � � � � � � __UNIQUE_ID_srcversion124 __UNIQUE_ID_depends123 ____versions __UNIQUE_ID_retpoline122 __UNIQUE_ID_intree121 __UNIQUE_ID_name120 __UNIQUE_ID_vermagic119 _note_9 _note_8 pp_remove __UNIQUE_ID_ddebug267.7 pp_check_ntb __UNIQUE_ID_ddebug271.4 __UNIQUE_ID_ddebug273.2 __UNIQUE_ID_ddebug272.3 __UNIQUE_ID_ddebug274.1 pp_check_ntb.cold pp_init_flds __UNIQUE_ID_ddebug275.0 pp_init_flds.cold pp_probe.part.0 pp_timer_func pp_ops pp_dbgfs_topdir pp_probe pp_setup __UNIQUE_ID_ddebug266.8 __UNIQUE_ID_ddebug265.9 pp_setup.cold pp_link_event pp_pong __UNIQUE_ID_ddebug269.5 pp_db_event __UNIQUE_ID_ddebug268.6 pp_init pp_client pp_exit __func__.18 __func__.17 __func__.16 __func__.15 __func__.14 __func__.13 __UNIQUE_ID_delay_ms264 __UNIQUE_ID_delay_mstype263 __param_delay_ms __param_str_delay_ms __UNIQUE_ID_unsafe262 __UNIQUE_ID_unsafetype261 __param_unsafe __param_str_unsafe __UNIQUE_ID_description260 __UNIQUE_ID_author259 __UNIQUE_ID_version258 __UNIQUE_ID_license257 ntb_set_ctx ntb_default_peer_port_number ntb_link_event param_ops_uint devm_kmalloc __this_module hrtimer_init debugfs_create_atomic_t cleanup_module ntb_clear_ctx __dynamic_dev_dbg _raw_spin_lock debugfs_initialized __fentry__ init_module __ntb_register_client __x86_indirect_thunk_rax ntb_default_peer_port_count __stack_chk_fail __x86_indirect_thunk_rdx _dev_err ntb_unregister_client debugfs_remove hrtimer_start_range_ns __x86_return_thunk ntb_default_port_number pv_ops __ubsan_handle_shift_out_of_bounds __x86_indirect_thunk_rcx hrtimer_cancel debugfs_create_dir __x86_indirect_thunk_r8 R �������� [ ��������) U ��������1 N ��������: b ��������T U ��������` ] ��������j q � v O ��������� ] ��������� R ��������� X ��������� � X �������� U �������� = U ��������I ��������e X ��������� U ��������� U ��������� ] ��������� ] ��������� 3 � U ��������� 3 ] & p + O ��������: ] ��������E O � T O ��������` ) j � o O ��������x v � 8 � O ��������� V ��������� R ��������� U ��������� U �������� U ��������9 F ��������x j � � � L � ] ��������� ( � � O ��������� ] ��������� ^ �������� V ��������1 L 6 � ; � A R ��������Z I ��������� K ��������� � � U ��������� U ��������� U �������� U ��������= U ��������b U ��������s P { E ��������� U ��������� G ��������� ��������� c ��������� � � L ��������� ] �������� ] ��������1 R ��������X ] ��������a R ��������� U ��������� b ��������� U ��������� U ��������� � � P �������� _ x \ 3 \ ��������B ] ��������t U ��������� X � � � O ��������� ] ��������� � � � � O ��������� ] ��������� F ��������� F ��������� R �������� ] �������� R ��������X X ��������� a ��������� U ��������� U ��������� U ��������� \ � \ �������� ] ��������5 U ��������H � O T O ��������t F ��������~ W ��������� R ��������� ] ��������� R ��������� P ��������� d �������� d ��������4 U ��������O U ��������Y _ x f ] ��������w U ��������� P � � � O ��������� F �������� ? Y �������� � # + ` ��������3 : B Y ��������L � Z � b ` ��������j � x � } ` ��������� x � � � ` ��������� � � � ` ��������� H � @ � ` ��������� � R �������� Q �������� � c ��������% ��������, � 3 J : ? T ��������M ��������R [ ��������] ] �������� Z �������� �������� [ �������� ] �������� V � � @ 0 ( ` 0 � 8 @ � H � P V � P � X � h J H ` ( q 0 J 8 H H ( S � � < d � � $ � ( � , � 0 4 � 8 � < � @ D < H a L � P � T � X � \ s ` W d h � l � p � t 4 x � | � 3 � N � v _ � � 9 � � � $ W ( A , � 0 � 4 � 8 < � @ e D \ H X d � � B � $ ( � 0 4 ] 8 � @ > D u H b P � T � X * ` d Y h � p � t � x " � � � � B � & � j � z � 0 � � � � d � � � � � d � � � � � d � � � d � � 0 d 8 � @ � P d X � � � ( 8 � @ 0 H � P v p � x 0 � � � ] � � � 0 � � � ) � � � 0 � � � � ( � 0 � P � X ( ` � h � � � � @ � � � � � � � � � X � � � � 8 S 8 M .symtab .strtab .shstrtab .note.gnu.build-id .note.Linux .rela.text .rela.text.unlikely .altinstr_replacement .rela.init.text .rela.exit.text .rela.altinstructions .rodata.str1.1 .rela__mcount_loc .rodata.str1.8 .rela.parainstructions .rela.smp_locks .rela.rodata .modinfo .rela__param .rela.retpoline_sites .rela.return_sites __versions .rela__jump_table .rela.data .rela__dyndbg .rela.gnu.linkonce.this_module .bss .comment .note.GNU-stack .BTF .gnu_debuglink @ $ . d 0 ? � � : @ @A � - J P � E @ N @ - Y # t ) a o @ @P - � � " @ `Q x - � � � @ �Q ` - � 2 � � � � X � @ 8R - � 2 � 1 � � @ @S 0 - � , � @ pS - 0 x � @ �S 0 - � � P P @ �S � - $ � � @ xT H - : , L 5 @ �W � - H � � X � S @ �Y � - j � � e @ X\ � - " u 0 0 p @ 8^ � - $ � � � @ ~ @ �a 0 - & � " � 0 " X � X" � X" = � �1 �1 x . E (; (b � 0�� *�H�� ���0��1 0 `�He0 *�H�� 1�o0�k0F0.1,0*U #Build time autogenerated kernel key.;��ڽ��> R�i8B0�`0 `�He0 *�H�� � Ld1 �~;�bX�O���� �-w�,8 b��x �x�tá�WF7�}�P�'Hb�6�)ހ��L��Z �[}i����W���� ���A/�� �'�{�<i�;m� ��y�W��rسv�.�W �?`aX=%��x_�{��9o�w�2�o�����q�`���qK���]� ��?������LxNp���|�Ke���м� #�j� ꢍ����KQ7�L���ڀ���f��� Y� D�� fLɦ<ÖQ�7Sl���`6����� $˖�;i�0B$��,���}]�Y=�&�DwH�j�Fj�h�[^�a �U�M��z �<�ěQ��k���P34d�>��{Ԓ�g|��#]W���l%�>~_�%q�&�gp��v�L.�";�-9s������yʔMY.� �)E'fC���. ��-���x�ܜW��y�j&�} T7Z�+�8��`~ �����}v� <�f���F���Q�FXٖXEC} o?�5 �~Module signature appended~ PK ə�Z][�K� � test_scripts.pynu �[��� PK ə�Z� ��� � $ 4 __pycache__/__init__.cpython-310.pycnu �[��� PK ə�Z� �� � ( 9 __pycache__/test_scripts.cpython-310.pycnu �[��� PK ə�Z�}9 [ __init__.pynu �[��� PK ���Z���^� � ) � __pycache__/test_sentence.cpython-310.pycnu �[��� PK ���Z�U5{ % �' __pycache__/test_base.cpython-310.pycnu �[��� PK ���Z�"b}� � $ � __pycache__/receiver.cpython-310.pycnu �[��� PK ���Z��t� � % � __pycache__/test_nmea.cpython-310.pycnu �[��� PK ���Z��ç�p �p PN test_base.pynu �[��� PK ���Z�}_�`� `� H� test_nmea.pynu �[��� PK ���Z�v^] ] �Y receiver.pynu �[��� PK ���Z�`� Z Z |^ test_sentence.pynu �[��� PK ͪ�Z��|� q test_logger.pynu �[��� PK ͪ�Z*�8�3 �3 t� test_filter.pynu �[��� PK ͪ�Z��.� * * ( �� __pycache__/test_flatten.cpython-310.pycnu �[��� PK ͪ�ZHf/� � ' � __pycache__/test_levels.cpython-310.pycnu �[��� PK ͪ�Z��>��1 �1 ' R� __pycache__/test_filter.cpython-310.pycnu �[��� PK ͪ�Z��ک� � ( �' __pycache__/test_capture.cpython-310.pycnu �[��� PK ͪ�Z��{�$ �$ ' �, __pycache__/test_logger.cpython-310.pycnu �[��� PK ͪ�ZrIxFwS wS ' R __pycache__/test_format.cpython-310.pycnu �[��� PK ͪ�Z�1. # # ' ҥ __pycache__/test_stdlib.cpython-310.pycnu �[��� PK ͪ�Z����t t % )� __pycache__/test_util.cpython-310.pycnu �[��� PK ͪ�Z�Dq�/# /# # �� __pycache__/test_io.cpython-310.pycnu �[��� PK ͪ�ZV1�-0 -0 ' t� __pycache__/test_global.cpython-310.pycnu �[��� PK ͪ�Z��^� % �* __pycache__/test_file.cpython-310.pycnu �[��� PK ͪ�Z��2�2 2 ' \I __pycache__/test_buffer.cpython-310.pycnu �[��� PK ͪ�Z[h�H H ) �R __pycache__/test_observer.cpython-310.pycnu �[��� PK ͪ�Z��ʕ�> �> ' �p __pycache__/test_legacy.cpython-310.pycnu �[��� PK ͪ�ZDZ�J E E % �� __pycache__/test_json.cpython-310.pycnu �[��� PK ͪ�Z! XPU U � test_format.pynu �[��� PK ͪ�Z���� 9 9 MJ test_legacy.pynu �[��� PK ͪ�Z��Is�! �! �� test_stdlib.pynu �[��� PK ͪ�Z�g�bG bG �� test_json.pynu �[��� PK ͪ�Z&����1 �1 _� test_global.pynu �[��� PK ͪ�Z����@ @ r test_capture.pynu �[��� PK ͪ�Zj �� � �# test_util.pynu �[��� PK ͪ�Zo6K=�% �% *2 test_flatten.pynu �[��� PK ͪ�Zu�:��" �" X test_io.pynu �[��� PK ͪ�Z��c c { test_levels.pynu �[��� PK ͪ�Z3=� �~ test_buffer.pynu �[��� PK ͪ�Z�x�� � � test_file.pynu �[��� PK ͪ�Z8»�* * � test_observer.pynu �[��� PK F��Z�!��} �} q� test_insults.pynu �[��� PK F��Z`l�_ _ 43 test_ckeygen.pynu �[��� PK F��ZCXF@�d �d �� test_conch.pynu �[��� PK F��Zή�� �� O� test_session.pynu �[��� PK F��Zgv[ �� test_mixin.pynu �[��� PK F��ZG�~�� � � test_manhole_tap.pynu �[��� PK F��ZL2� � $� test_text.pynu �[��� PK F��Z<���' ' / � __pycache__/test_openssh_compat.cpython-310.pycnu �[��� PK F��Z ��m % �� __pycache__/test_unix.cpython-310.pycnu �[��� PK F��Z��F�A A $ � __pycache__/loopback.cpython-310.pycnu �[��� PK F��Z�H�FE� E� $ �� __pycache__/test_ssh.cpython-310.pycnu �[��� PK F��Z��k�i �i & /v __pycache__/test_conch.cpython-310.pycnu �[��� PK F��Z�c�ngM gM ( ;� __pycache__/test_ckeygen.cpython-310.pycnu �[��� PK F��ZE8w�& & , �- __pycache__/test_manhole_tap.cpython-310.pycnu �[��� PK F��Zd�O^�6 �6 & |= __pycache__/test_agent.cpython-310.pycnu �[��� PK F��Z�?�� � & �t __pycache__/test_mixin.cpython-310.pycnu �[��� PK F��Z"�� ̹ ̹ * �z __pycache__/test_endpoints.cpython-310.pycnu �[��� PK F��Z�x�Xn Xn # �4 __pycache__/keydata.cpython-310.pycnu �[��� PK F��Z�%i�. �. ( �� __pycache__/test_channel.cpython-310.pycnu �[��� PK F��Z x��J �J ' �� __pycache__/test_helper.cpython-310.pycnu �[��� PK F��Z� ��B/ B/ ( 1 __pycache__/test_default.cpython-310.pycnu �[��� PK F��Z��mB B % �M __pycache__/test_text.cpython-310.pycnu �[��� PK F��Z�� l l $ b^ __pycache__/test_tap.cpython-310.pycnu �[��� PK F��ZSx�6� 6� % "s __pycache__/test_keys.cpython-310.pycnu �[��� PK F��Z��OS S ' �8 __pycache__/test_window.cpython-310.pycnu �[��� PK F��Z�� �� � % WA __pycache__/test_cftp.cpython-310.pycnu �[��� PK F��Z�n�+�� �� - � __pycache__/test_filetransfer.cpython-310.pycnu �[��� PK F��Z?��8 8 ( �� __pycache__/test_manhole.cpython-310.pycnu �[��� PK F��Z�B`-n� n� ) � __pycache__/test_userauth.cpython-310.pycnu �[��� PK F��Z_��u� u� * �C __pycache__/test_transport.cpython-310.pycnu �[��� PK F��Z��ղ�` �` ' �� __pycache__/test_telnet.cpython-310.pycnu �[��� PK F��Z^a� d d ) t( __pycache__/test_recvline.cpython-310.pycnu �[��� PK F��ZJ�t4� � ( � __pycache__/test_address.cpython-310.pycnu �[��� PK F��Z�1�|x |x ( ؔ __pycache__/test_insults.cpython-310.pycnu �[��� PK F��Z�� 4� 4� + � __pycache__/test_knownhosts.cpython-310.pycnu �[��� PK F��Zh,i�� � + ;� __pycache__/test_forwarding.cpython-310.pycnu �[��� PK F��Z�:'� �� ) �� __pycache__/test_checkers.cpython-310.pycnu �[��� PK F��Ze1��He He + bT __pycache__/test_connection.cpython-310.pycnu �[��� PK F��ZٳC'į į ( � __pycache__/test_session.cpython-310.pycnu �[��� PK F��Z�5��� � !j loopback.pynu �[��� PK F��Z햞�� �� m keydata.pynu �[��� PK F��Z���ilz lz � test_checkers.pynu �[��� PK F��Z��� �q test_address.pynu �[��� PK F��Z�SP� P� �w test_endpoints.pynu �[��� PK F��Z�|�ї/ �/ jR test_channel.pynu �[��� PK F��Z:���A A @� test_window.pynu �[��� PK F��ZҢB��3 �3 �� test_manhole.pynu �[��� PK F��Z�G %Y� Y� Ǿ test_knownhosts.pynu �[��� PK F��Z碢%Ir Ir b� test_connection.pynu �[��� PK F��Z�T�vm� m� �� test_filetransfer.pynu �[��� PK F��Z�r��( ( �t test_unix.pynu �[��� PK F��Z F�� � test_transport.pynu �[��� PK F��Z�7�d�3 �3 J" test_agent.pynu �[��� PK F��Z[���r r *V test_forwarding.pynu �[��� PK F��Z!�*IXc Xc �^ test_recvline.pynu �[��� PK F��Z�i��G� G� v� test_cftp.pynu �[��� PK F��Z����AN AN �� test_helper.pynu �[��� PK F��Z��E�� � x� test_userauth.pynu �[��� PK F��Z2��i�h �h �Y test_telnet.pynu �[��� PK F��Z�@D- D- �� test_default.pynu �[��� PK F��Z�B9� � C� test_tap.pynu �[��� PK F��Z>�t) + test_openssh_compat.pynu �[��� PK F��Z��n�7� 7� } test_keys.pynu �[��� PK F��ZӉ_Ȫ� �� � test_ssh.pynu �[��� PK ��Z��-�)� )� Ռ ntb_tool.konu �[��� PK ��ZE�)�� � 9� ntb_perf.konu �[��� PK ��Z�Y��r �r � ntb_pingpong.konu �[��� PK m m x& !
| ver. 1.4 |
Github
|
.
| PHP 8.2.28 | Generation time: 0.06 |
proxy
|
phpinfo
|
Settings