File: //lib/python3.9/site-packages/elftools/construct/__pycache__/debug.cpython-39.pyc
a
�#�_� � @ sx d Z ddlmZ ddlZddlZddlZddlZddlmZm Z ddl
m Z m Z m
Z
G dd� de�ZG d d
� d
e �ZdS ) z$
Debugging utilities for constructs
� )�print_functionN� )� Construct� Subconstruct)� HexString� Container�
ListContainerc @ sN e Zd ZdZg d�ZdZddd�Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�Probea�
A probe: dumps the context, stack frames, and stream content to the screen
to aid the debugging process.
See also Debugger.
Parameters:
* name - the display name
* show_stream - whether or not to show stream contents. default is True.
the stream must be seekable.
* show_context - whether or not to show the context. default is True.
* show_stack - whether or not to show the upper stack frames. default
is True.
* stream_lookahead - the number of bytes to dump when show_stack is set.
default is 100.
Example:
Struct("foo",
UBInt8("a"),
Probe("between a and b"),
UBInt8("b"),
)
)� printname� show_stream� show_context�
show_stack�stream_lookaheadr NT�d c C sP t �| d � |d u r.t jd7 _dtjf }|| _|| _|| _|| _|| _d S )Nr z <unnamed %d>) r �__init__r �counterr
r r r
r )�self�namer r r
r � r �</usr/lib/python3.9/site-packages/elftools/construct/debug.pyr * s zProbe.__init__c C s d| j j| jf S )Nz%s(%r))� __class__�__name__r
)r r r r �__repr__6 s zProbe.__repr__c C s | � ||� d S �N��printout�r �stream�contextr r r �_parse8 s z Probe._parsec C s | � ||� d S r r �r �objr r r r r �_build: s z Probe._buildc C s dS )Nr r )r r r r r �_sizeof< s z
Probe._sizeofc C s� t � }| jrN|�� |_|�| j�}|s.d|_n |�t|� d� t |�|_t
| j rZ||_ | j
r�t� |_dd� t�� D �dd� }|�� |D ]"}t � }|�|j� |j�|� q�t
d� t
d| j� t
|� t
d� d S )Nz EOF reachedr c S s g | ] }|d �qS )r r )�.0�sr r r �
<listcomp>P � z"Probe.printout.<locals>.<listcomp>����P================================================================================r )r r �tellZstream_position�readr Zfollowing_stream_data�seek�lenr �printr r r
r �stack�inspect�reverseZ
__update__�f_locals�appendr
)r r r r! ZfollowsZframes�f�ar r r r ? s.
zProbe.printout)NTTTr ) r �
__module__� __qualname__�__doc__� __slots__r r r r r" r# r r r r r r
s �
r c @ s0 e Zd ZdZdgZdd� Zdd� Zd
dd �ZdS ) �DebuggeraR
A pdb-based debugger. When an exception occurs in the subcon, a debugger
will appear and allow you to debug the error (and even fix on-the-fly).
Parameters:
* subcon - the subcon to debug
Example:
Debugger(
Enum(UBInt8("foo"),
a = 1,
b = 2,
c = 3
)
)
�retvalc C sP z| j �||�W S tyJ t| _| �d� | jtu r<� n
| j Y S Y n0 d S )Nz@(you can set the value of 'self.retval', which will be returned))�subconr � Exception�NotImplementedr; �
handle_excr r r r r n s
zDebugger._parsec C s4 z| j �|||� W n ty. | �� Y n0 d S r )r< r" r= r? r r r r r" y s zDebugger._buildNc C sb t d� t d| jf � t d�tjt�� � dd � �� |rDt |� t�t�� d � t d� d S )Nr) zDebugging exception of %s:� r � ) r. r< �join� traceback�format_exception�sys�exc_info�pdbZ post_mortem)r �msgr r r r? ~ s zDebugger.handle_exc)N)r r6 r7 r8 r9 r r" r? r r r r r: \ s
r: )r8 Z
__future__r rE rC rG r0 Zcorer r �libr r r r r: r r r r �<module> s O