Current File : //opt/alt/python38/lib64/python3.8/site-packages/maxminddb/__pycache__/decoder.cpython-38.pyc
U
��^� � @ sL d Z ddlmZ ddlZddlmZmZmZ ddlm Z G dd� de
�Z dS )zl
maxminddb.decoder
~~~~~~~~~~~~~~~~~
This package contains code for decoding the MaxMind DB data section.
� )�unicode_literalsN)�
byte_from_int�
int_from_byte�int_from_bytes��InvalidDatabaseErrorc @ s� e Zd ZdZd$dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Ze eeee
e
e e
e
e
eee d�
Zdd � Zd d � Zd d � Zd!d"� Zd#S )%�Decoderz.Decoder for the data section of the MaxMind DBr Fc C s || _ || _|| _dS )a Created a Decoder for a MaxMind DB
Arguments:
database_buffer -- an mmap'd MaxMind DB file.
pointer_base -- the base number to use when decoding a pointer
pointer_test -- used for internal unit testing of pointer code
N)�
_pointer_test�_buffer�
_pointer_base)�selfZdatabase_bufferZ pointer_baseZ pointer_test� r
� /decoder.py�__init__ s zDecoder.__init__c C s2 g }t |�D ] }| �|�\}}|�|� q ||fS �N)�range�decode�append)r �size�offsetZarray�_�valuer
r
r �
_decode_array s
zDecoder._decode_arrayc C s |dk|fS )Nr r
)r r r r
r
r �_decode_boolean& s zDecoder._decode_booleanc C s || }| j ||� |fS r )r
�r r r �
new_offsetr
r
r �
_decode_bytes) s zDecoder._decode_bytesc C s8 | � |d� || }| j||� }t�d|�\}||fS )N� s !d�� _verify_sizer
�struct�unpack�r r r r Z packed_bytesr r
r
r �_decode_double- s
zDecoder._decode_doublec C s8 | � |d� || }| j||� }t�d|�\}||fS )N� s !fr r" r
r
r �
_decode_float4 s
zDecoder._decode_floatc C sP |dkrd|fS || }| j ||� }|dkr:|�dd�}t�d|�\}||fS )Nr r$ � s !i)r
�rjustr r! r" r
r
r �
_decode_int32; s zDecoder._decode_int32c C s> i }t |�D ](}| �|�\}}| �|�\}}|||< q ||fS r )r r )r r r Z containerr �keyr r
r
r � _decode_mapF s
zDecoder._decode_mapc C s� |d? d }| j ||| � }|| }|dkrVt|d@ �| }t�d|�d | j }n�|dkr�dt|d@ � | }t�d|�d d | j }nJ|dkr�t|d@ �| }t�d|�d d
| j }nt�d|�d | j }| jr�||fS | �|�\}}||fS ) N� � � � !Hr � r& � !Ii i )r
r r r! r r r ) r r r Z pointer_sizeZbufr Zpointerr r r
r
r �_decode_pointerN s" zDecoder._decode_pointerc C s" || }| j ||� }t|�|fS r )r
r )r r r r Z
uint_bytesr
r
r � _decode_uinte s zDecoder._decode_uintc C s || }| j ||� �d�|fS )Nzutf-8)r
r r r
r
r �_decode_utf8_stringj s zDecoder._decode_utf8_string)
r, r/ r+ r$ � � r- r � �
� � � c C s� |d }t | j| �}|d? }|s0| �|�\}}z| j| }W n$ tk
rb tdj|d���Y nX | �|||�\}}|| ||�S )z�Decode a section of the data section starting at offset
Arguments:
offset -- the location of the data structure to decode
r, r4 z+Unexpected type number ({type}) encountered��type)r r
�_read_extended�
_type_decoder�KeyErrorr �format�_size_from_ctrl_byte)r r r � ctrl_byte�type_num�decoderr r
r
r r ~ s
�
zDecoder.decodec C s: t | j| �}|d }|dk r.tdj|d���||d fS )Nr- zeSomething went horribly wrong in the decoder. An extended type resolved to a type number < 8 ({type})r; r, )r r
r r@ )r r Z next_byterC r
r
r r= � s ��zDecoder._read_extendedc C s ||krt d��d S )NzXThe MaxMind DB file's data section contains bad data (unknown data type or corrupt data)r )r Zexpected�actualr
r
r r � s �zDecoder._verify_sizec C s� |d@ }|dks|dk r ||fS |dkrFdt | j| � }||d fS |dkr�|d }| j||� }dt�d|�d }||fS |d }| j||� }t�d
d | �d d }||fS )
N� r, � � r/ i r. r r+ r0 r& i )r r
r r! )r rB r rC r r Z
size_bytesr
r
r rA � s z Decoder._size_from_ctrl_byteN)r F)�__name__�
__module__� __qualname__�__doc__r r r r r# r% r( r* r1 r2 r3 r> r r= r rA r
r
r
r r s<
� r ) rL Z
__future__r r Zmaxminddb.compatr r r Zmaxminddb.errorsr �objectr r
r
r
r �<module> s
Mini Shell