codekingpro/portable-devtools
114k
1�
2 ��j( � � � S SK r S SKrS SKJr S SKJr S SKJr S SKJr S SKJ r S SKJ3r4 S SK J r S S K J
r
S S5K Jr S S KJr S S KJr S S
KJr " S S5 r " S S5 r " S S5 rS\\
R0 4S jrg)� N)�Iterable)�BufferedReader)�Any)�BinaryIO)�cast)�Union)�6exceptions)�flow)�7flowfilter)�compat)�8tnetstring)�request_to_flowc �@ � \ rS rSrS rS\R SS4S jrSrg)�9FlowWriter� c � � Xl g �N��fo��selfr s �UD:\code\apps\devtools\python\user_packages\Python313\site-packages\mitmproxy/io/io.py�__init__�FlowWriter.__init__ � � ��� �f�returnNc �d � UR 5 n[ R " X R 5 g r )� get_stater
�dumpr �r r �ds r �add�FlowWriter.add s � �
�K�K�M������7�7�#r r ) �__name__�10__module__� __qualname__�__firstlineno__r r11 �Flowr$ �__static_attributes__� r r r r s � ��$�T�Y�Y� $�4� $r r c �h � \ rS rSr% \\S' S\4S jrS\S\4S jr S\12\ R 4S jr
Srg )13�14FlowReader� r c � � Xl g r r r s r r �FlowReader.__init__ r r �nr c � � [ [ U R 5 R U5 $ ! [ aU U R R 5 nU R R
U5 nU R R U5 Us $ f = fr )r r r �peek�AttributeError�tell�read�seek)r r2 �pos�rets r r4 �FlowReader.peek! sg � � ������0�5�5�a�8� 8�� � ��'�'�,�,�.�C��'�'�,�,�q�/�C� �G�G�L�L�� ��J� �s �(+ �A B15� B16c # � # � U R S5 R S5 ( a U R R S5 U R S5 R S5 ( a[ [ R17 " U R R 5 R
S5 5 nUS S H n[ U5 v � M g [ [ [ [ [ 4 [ 4 [" R$ " U R 5 5 n [' U[ 5 ( d [) S18U< 35 e[* R, R/ [0 R2 " U5 5 v � M� ! [ a [ R " S 5 ef = f! [( a n[ R " U5 UeS nAff = f! [( [4 [6 4 a0 n[ U5 S :X a S nAg [ R " S
5 UeS nAff = f7f)z$19Yields Flow objects from the dump.20� s {� � � {zutf-8�log�entriesz8Unable to read HAR file. Please provide a valid HAR filezInvalid flow: loaded=Nz not a tnetstring: empty filezInvalid data format.) r4 �21startswithr r7 �json�loads�decoder � Exceptionr �FlowReadExceptionr �dictr �bytes�strr r
�load�22isinstance�23ValueErrorr24 r* �25from_stater � migrate_flow� TypeError�26IndexError)r �har_file� request_json�loaded�es r �stream�FlowReader.stream+ s� � � �27 �9�9�Q�<� "� "� � 28� 29�
�G�G�L�L��O� �9�9�Q�<� "� "�4� (� (�
� �:�:�d�g�g�l�l�n�&;�&;�G�&D�E��$,�U�O�I�$>�L�)�,�7�7� %?�
R�� !� �U�5�#�:� .�� 3�4�"������0� �F�E� )�&�$� 7� 7�",�/E�f�Y�-G�"H� H� "�i�i� 2� 2�6�3F�3F�v�3N� O�O� �� �
� �2�2�N�� �
��" &� E� (� :� :�1� =�1�D��E�� � �:�6�
R��q�6� ;�;�� �2�2�3I�J�PQ�Q��
R�so �A&G>�)AE) �G>�A F6 �AF
�'F6 �)!F30�31G>�
32F3�F.�.F3�3F6 �6G;� G6�G>� G6�6G;�;G>r N)r& r' r( r) r �__annotations__r �intrJ r4 r r33 r* rW r+ r, r r r. r. s= � ��L��8� ��c� �e� �&R�����+� &Rr r. c �f � \ rS rSrS\S\R S- 4S jrS\R SS4S jr34S r g)35�FilteredFlowWriter�T r �fltNc � � Xl X l g r )r r^ )r r r^ s r r �FilteredFlowWriter.__init__U s � ����r r r c � � U R ( a' [ R " U R U5 ( d g UR 5 n[ R36 " X R 5 U R R 5 g r )r^ r �matchr r
r! r �flushr" s r r$ �FilteredFlowWriter.addY sJ � � �8�8�J� ,� ,�T�X�X�q� 9� 9� �
�K�K�M������7�7�#� ���
�
�r )r^ r ) r& r' r( r) r r �TFilterr r37 r* r$ r+ r, r r r\ r\ T s7 � ��8� �*�*<�*<�t�*C� ��T�Y�Y� �4� r r\ r c �^ � / nU H_ n[ R R U5 n[ US5 nUR [ U5 R
5 5 SSS5 Ma U$ ! , ( d f Mt = f! [ a% n[ R " UR 5 eSnAff = f)a 38Given a list of filepaths, read all flows and return a list of them.39From a performance perspective, streaming would be advisable -40however, if there's an error with one of the files, we want it to be raised immediately.41 42Raises:43 FlowReadException, if any error occurs.44�rbN) �os�path�45expanduser�open�extendr. rW �OSErrorr rH �strerror)�paths�flowsri r rV s r �read_flows_from_pathsrq a s� � �7�!#���D��7�7�%�%�d�+�D��d�D�!�Q�� � �Z��]� 1� 1� 3�4� "�!� � �L� "�!�� � 7��*�*�1�:�:�6�6��7�s. �2A= �)A+� A= �+46A: �5A= �=47B,� B'�'B,)rD rh �collections.abcr �ior �typingr r r r � mitmproxyr r48 r � mitmproxy.ior r
�mitmproxy.io.harr r r. r\ �listr* rq r, r r �<module>ry sd �� � � $� � � � � � � � � � #� ,�$� $�6R� 6R�r49� 50��D����O� r