codekingpro/portable-devtools
114k
1�
2 ��j� � �� � S SK r S SKrS SKrS SKrS SKJr S SKJr S SKJ r S SKJ3r4 S SKJ r S SKJ r S SKJ
r
S S KJr \R " \5 r " S5 S 5 r " S S
\5 rg)� N)�BinaryIO)�Optional)�command)�ctx)�6exceptions)�7flowfilter)�io)�
asyncio_utilsc � � \ rS rSrSrS rS rS rS\S\ 4S jr8S \ S\ 4S9 jr S \ SS 4S
jr
S r\R " S5 S\4S j5 rSrg )�ReadFile� z510An addon that handles reading from file on startup.11c � � S U l S U l g �N)�filter�12_read_task��selfs �_D:\code\apps\devtools\python\user_packages\Python313\site-packages\mitmproxy/addons/readfile.py�__init__�ReadFile.__init__ s � ��� �/3��� c � � UR S[ [ S S5 UR S[ [ S S5 g )N�rfilezRead flows from file.�readfile_filterzRead only matching flows.)�13add_optionr �str)r �loaders r �load�
ReadFile.load s7 � ����'�8�C�=�$�8O�P���� �x��}�d�4O� 14r c �, � SU; a\ [ R R ( a5 [ R " [ R R 5 U l g S U l g g ! [ a% n[ R " [ U5 5 UeS nAff = f)Nr )15r �optionsr r �parser �16ValueErrorr � OptionsErrorr )r �updated�es r � configure�ReadFile.configure! sr � � �� '��{�{�*�*�A�",�"2�"2�3�;�;�3N�3N�"O�D�K� #�� � (�� "� A�$�1�1�#�a�&�9�q�@��A�s �3A$ �$17B�. B�B�fo�returnc � � # � Sn[ R " U5 n UR 5 HX nU R ( a U R U5 ( d M, [ R18 R
U5 I S h v�N US-
nMZ U$ N! [ [ R 4 a\ nU( a [ R " SU- 5 O[ R " S5 [ R " [ U5 5 UeS nAff = f7f)Nr � z&Flow file corrupted - loaded %i flows.zFlow file corrupted.)r �19FlowReader�streamr r �master� load_flow�OSErrorr �FlowReadException�logging�warning�errorr )r r) �cnt�freader�flowr& s r �20load_flows�ReadFile.load_flows+ s� � � ����-�-��#��
� ���(���;�;�t�{�{�4�'8�'8� ��j�j�*�*�4�0�0�0��q��� )� �J� 1����5�5�6� >����� H�3� N�O��
�
� 4�5� �.�.�s�1�v�6�A� =�� >�s< �D�A B �:B21�;
B �D�22B � D�&AC=�=D�D�pathc � �d # � [ R R U5 n [ US5 nU R U5 I S h v�N sS S S 5 $ N
! , ( d f g = f! [23 a> n[ R " SU 35 [ R " [ U5 5 UeS nAff = f7f)N�rbzCannot load flows: ) �osr; �24expanduser�openr9 r1 r3 r5 r r2 r )r r; �fr& s r �load_flows_from_path� ReadFile.load_flows_from_path= s� � � ��w�w�!�!�$�'�� >��d�D�!�Q� !�_�_�Q� /�/� "�!�/� "�!��� >� �M�M� /��s�3� 4� �.�.�s�1�v�6�A� =�� >�s] � B0� A% �A�A�A� A% �B0�A�25A"� A% �!B0�"A% �%26B-�/9B(�(B-�-B0r Nc � �� # � U R U5 I S h v�N g N! [ R a= n[ R S[27 R R SU 35 S nAg S nAff = f7f)NzFailed to read z: )rB r r2 �logger� exceptionr r! r )r r r& s r �doread�ReadFile.doreadF s_ � � � I��+�+�E�2� 2� 2���+�+� I� � � ��s�{�{�/@�/@�.A��A�3� G� H� H�� I�s6 �A4� � � �A4� �A1�3A,�'A4�,A1�1A4c �� � [ R R ( aC [ R " U R [ R R 5 SSS9U l g g )N�readfileF)�name�keep_ref)r r! r r28 � create_taskrG r r s r �running�ReadFile.runningL sB � � �;�;� � � +� 7� 7�� � �C�K�K� -� -�.� � � �D�O� r zreadfile.readingc �x � [ U R =( a U R R 5 ( + 5 $ r )�boolr �doner s r �reading�ReadFile.readingT s& � ��D�O�O�B�D�O�O�,@�,@�,B�(B�C�Cr )r r )�__name__�29__module__� __qualname__�__firstlineno__�__doc__r r r' r �intr9 r rB rG rN r rQ rS �__static_attributes__� r r r r s � ��4�30� #��8� �� �$>�s� >�s� >�I�#� I�$� I� �
�_�_�'�(�D�� D� )�Dr r c �8 ^ � \ rS rSrSrS\S\4U 4S jjrSrU =r $ )�
ReadFileStdin�Y z6Support the special case of "-" for reading from stdinr; r* c � � ># � US:X a1 U R [ R R 5 I S h v�N $ [ TU ] U5 I S h v�N $ N N7f)N�-)r9 �sys�stdin�buffer�superrB )r r; � __class__s �r rB �"ReadFileStdin.load_flows_from_path\ sF �� � � �3�;� � � ����)9�)9�:�:� :� ��5�d�;�;� ;� ;�;�s! �3A�A�A�
A�A�Ar\ )31rU rV rW rX rY r rZ rB r[ �
__classcell__)rf s @r r^ r^ Y s �� �@�<�s� <�s� <� <r r^ )�asyncior3 �os.pathr> rb �typingr r � mitmproxyr r r r r �mitmproxy.utilsr32 � getLoggerrU rE r r^ r\ r r �<module>ro sT �� � � � 33� � � � � � � � )� � � �8� $��DD� DD�N <�H� <r