codekingpro/portable-devtools
114k
1�
2 ��j�$ � � � S SK r S SKrS SKJrJrJrJrJr S SKJ r S SK3J r J r J
r
S SKJr S SKJr Sr " S S \5 rg)4� N)�byref�c_uint64�c_uint�c_char�c_char_p)�
windivert_dll)�Layer� Direction�Flag)�Packet)�PY2i� c �, � \ rS rSrSrS\R S\R 4S jr S r5S r S r S r
S6 r\R S : a \r\S 5 r\S
5 r\S 5 r\\R 4S j5 rS r\S 5 rS r\4S jr SS jr S r S r Sr g)� WinDivert� a 7A WinDivert handle that can be used to capture packets.8The main methods are `.open()`, `.recv()`, `.send()` and `.close()`.9 10Use it like so::11 12 with pydivert.WinDivert() as w:13 for packet in w:14 print(packet)15 w.send(packet)16 17�truer c �` � S U l UR 5 U l X l X0l X@l g �N)�_handle�encode�_filter�_layer� _priority�_flags)�self�filter�layer�priority�flagss �XD:\code\apps\devtools\python\user_packages\Python313\site-packages\pydivert/windivert.py�__init__�WinDivert.__init__* s% � ��� � �}�}��� �� �!��� � c � � SR U R b SOSU R R 5 U R U R18 U R 5 $ )NzH<WinDivert state="{}" filter="{}" layer="{}" priority="{}" flags="{}" />�open�closed)�formatr r �decoder r r �r s r �__repr__�WinDivert.__repr__1 sH � �Y�`�`��l�l�.�F�H� �L�L� � � !� �K�K� �N�N� �K�K� 19� 20r" c �& � U R 5 U $ r )r$ r( s r � __enter__�WinDivert.__enter__: s � � � � � �� r" c �$ � U R 5 g r )�close)r �argss r �__exit__�WinDivert.__exit__> s � � �21�22� r" c � � U $ r � r( s r �__iter__�WinDivert.__iter__A s � �� r" c �" � U R 5 $ r )�recvr( s r �__next__�WinDivert.__next__D s � ��y�y�{�r" )� r c �P � [ S5 SSS5 g! , ( d f g= f)z�23An utility method to register the service the first time.24It is usually not required to call this function, as WinDivert will register itself when opening a handle.25�falseN)r r4 r" r �register�WinDivert.registerJ s � � �w�
� � �
�
�s ��26%c �l � [ R " S[ R [ R S9S:H $ )zF27Check if the WinDivert service is currently installed on the system.28zsc query WinDivert1.3��stdout�stderrr )�29subprocess�call�PIPEr4 r" r �
is_registered�WinDivert.is_registeredS s- � �30 ��� 6�z���&0�o�o�7�:;�<� <r" c �h � [ R " S[ R [ R S9 g)z�31Unregisters the WinDivert service.32This function only requests a service stop, which may not be processed immediately if there are still open33handles.34zsc stop WinDivert1.3rA N)rD �35check_callrF r4 r" r �36unregister�WinDivert.unregister[ s" � � � � � 4�Z�_�_�%/�_�_� 6r" c � � S[ 5 [ 5 pCn [ R " U R 5 U[ U5 [ U5 5 nX#R UR R 5 4$ ! [ a N2f = f)a$ 37Checks if the given packet filter string is valid with respect to the filter language.38 39The remapped function is WinDivertHelperCheckFilter::40 41 BOOL WinDivertHelperCheckFilter(42 __in const char *filter,43 __in WINDIVERT_LAYER layer,44 __out_opt const char **errorStr,45 __out_opt UINT *errorPos46 );47 48See: https://reqrypt.org/windivert-doc.html#divert_helper_check_filter49 50:return: A tuple (res, pos, msg) with check result in 'res' human readable description of the error in 'msg' and the error's position in 'pos'.51F) r r r �WinDivertHelperCheckFilterr r �OSError�valuer' )r r �res�pos�msgs r � check_filter�WinDivert.check_filtere st � �$ �v�x���#�� � �:�:�6�=�=�?�E�SX�Y\�S]�_d�eh�_i�j�C� �I�I�s�y�y� /� /� 1�1�1�� � � � �s �9A6 �652B�Bc �� � U R ( a [ S5 e[ R " U R U R53 U R U R 5 U l g)a� 54Opens a WinDivert handle for the given filter.55Unless otherwise specified by flags, any packet that matches the filter will be diverted to the handle.56Diverted packets can be read by the application with receive().57 58The remapped function is WinDivertOpen::59 60 HANDLE WinDivertOpen(61 __in const char *filter,62 __in WINDIVERT_LAYER layer,63 __in INT16 priority,64 __in UINT64 flags65 );66 67For more info on the C call visit: http://reqrypt.org/windivert-doc.html#divert_open68z!WinDivert handle is already open.N) �is_open� RuntimeErrorr �
WinDivertOpenr r r r r r( s r r$ �WinDivert.open~ sC � �" �<�<� � B�C� C�$�2�2�4�<�<����d�n�n�37�;�;�@�� r" c �, � [ U R 5 $ )z169Indicates if there is currently an open handle.70)�boolr r( s r rW �WinDivert.is_open� s � �71 �D�L�L�!�!r" c � � U R ( d [ S5 e[ R " U R 5 SU l g)z�72Closes the handle opened by open().73 74The remapped function is WinDivertClose::75 76 BOOL WinDivertClose(77 __in HANDLE handle78 );79 80For more info on the C call visit: http://reqrypt.org/windivert-doc.html#divert_close81z WinDivert handle is not open.N)rW rX r �WinDivertCloser r( s r r/ �WinDivert.close� s0 � � �|�|� � >�?� ?��$�$�T�\�\�2��� r" c � � U R c [ S5 e[ U5 n[ U- R U5 n[82 R " 5 n[ S5 n[83 R " U R X1[ U5 [ U5 5 [ [ U5 SUR UR UR 4[ UR 5 5 $ )a� 84Receives a diverted packet that matched the filter.85 86The remapped function is WinDivertRecv::87 88 BOOL WinDivertRecv(89 __in HANDLE handle,90 __out PVOID pPacket,91 __in UINT packetLen,92 __out_opt PWINDIVERT_ADDRESS pAddr,93 __out_opt UINT *recvLen94 );95 96For more info on the C call visit: http://reqrypt.org/windivert-doc.html#divert_recv97 98:return: The return value is a `pydivert.Packet`.99Nz WinDivert handle is not openr )r rX � bytearrayr � from_bufferr �WinDivertAddressr �
WinDivertRecvr r �100memoryviewrP �IfIdx�SubIfIdxr101 )r �bufsize�packet�packet_�address�recv_lens r r8 �WinDivert.recv� s� � �$ �<�<� � � =�>� >��7�#���G�#�0�0��8�� �0�0�2���!�9���#�#�D�L�L�'�E�'�N�TY�Zb�Tc�d�� �v� ����� /�
�]�]�G� ,� ,� -� �g�'�'� (�102� 103r" c � � U( a UR 5 [ S5 n[ ( a [ UR 5 nO UR n[104 [
UR 5 - R U5 n[ R " U R U[
UR 5 [ UR 5 [ U5 5 U$ )a� 105Injects a packet into the network stack.106Recalculates the checksum before sending unless recalculate_checksum=False is passed.107 108The injected packet may be one received from recv(), or a modified version, or a completely new packet.109Injected packets can be captured and diverted again by other WinDivert handles with lower priorities.110 111The remapped function is WinDivertSend::112 113 BOOL WinDivertSend(114 __in HANDLE handle,115 __in PVOID pPacket,116 __in UINT packetLen,117 __in PWINDIVERT_ADDRESS pAddr,118 __out_opt UINT *sendLen119 );120 121For more info on the C call visit: http://reqrypt.org/windivert-doc.html#divert_send122 123:return: The return value is the number of bytes actually sent.124r )
�recalculate_checksumsr r
rb �rawr �lenrc r �
WinDivertSendr r �wd_addr)r rj �recalculate_checksum�send_len�buffs r �send�WinDivert.send� s� � �, � � (� (� *��!�9�� �3� �V�Z�Z�(�D��:�:�D���V�Z�Z� �(�5�5�d�;���#�#�D�L�L�$��F�J�J���v�~�~�I^�$)�(�O� 5��r" c � � [ S5 n[ R " U R U[ U5 5 UR125 $ )at 126Get a WinDivert parameter. See pydivert.Param for the list of parameters.127 128The remapped function is WinDivertGetParam::129 130 BOOL WinDivertGetParam(131 __in HANDLE handle,132 __in WINDIVERT_PARAM param,133 __out UINT64 *pValue134 );135 136For more info on the C call visit: http://reqrypt.org/windivert-doc.html#divert_get_param137 138:return: The parameter value.139r )r r �WinDivertGetParamr r rP �r �namerP s r � get_param�WinDivert.get_param� s1 � � �� ���'�'�� � �d�E�%�L�I��{�{�r" c �D � [ R " U R X5 $ )aO 140Set a WinDivert parameter. See pydivert.Param for the list of parameters.141 142The remapped function is DivertSetParam::143 144 BOOL WinDivertSetParam(145 __in HANDLE handle,146 __in WINDIVERT_PARAM param,147 __in UINT64 value148 );149 150For more info on the C call visit: http://reqrypt.org/windivert-doc.html#divert_set_param151)r �WinDivertSetParamr r| s r � set_param�WinDivert.set_param s � � �.�.�t�|�|�T�I�Ir" )r r r r r N)T)!�__name__�152__module__� __qualname__�__firstlineno__�__doc__r �NETWORKr �DEFAULTr r) r, r1 r5 r9 �sys� version_info�next� staticmethodr> rG rK rT r$ �propertyrW r/ �DEFAULT_PACKET_BUFFER_SIZEr8 rx r~ r� �__static_attributes__r4 r" r r r s� � � � %�E�M�M�A�T�\�\� �153����� ���&� ����� �� �<� �<� �6� �6� �#(�=�=� 2� �2�0@�, �"� �"� �" 6� 154�@"�H�(Jr" r )rD r� �ctypesr r r r r �pydivertr �pydivert.constsr r155 r �pydivert.packetr �
pydivert.utilr
r� �objectr r4 r" r �<module>r� s8 �� � 156� <� <� "� 2� 2� "� � !� �vJ�� vJr"