Your IP : 216.73.216.213
�jvX+ � @ sb d Z d d l m Z d d l Z d d l m Z m Z d d l m Z Gd d � d e � Z
d S)zl
maxminddb.decoder
~~~~~~~~~~~~~~~~~
This package contains code for decoding the MaxMind DB data section.
� )�unicode_literalsN)�
byte_from_int�int_from_bytes)�InvalidDatabaseErrorc @ sN e Z d Z d Z d d d d � Z d d � Z d d � Z d
d � Z d d d
� Z d d � Z d d d d d d d d i Z
d d � Z d d � Z d d � Z
d e d e
d e d d � d e d e d e d e d e d! d d" d# �d$ e d% e d& e d' e d( e d) d � i
Z d* d+ � Z d, d- � Z d. d/ � Z d0 d1 � Z d2 S)3�Decoderz.Decoder for the data section of the MaxMind DBr Fc C s | | _ | | _ | | _ d S)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 �D/opt/alt/python35/lib64/python3.5/site-packages/maxminddb/decoder.py�__init__ s zDecoder.__init__c C sI g } x6 t | � D]( } | j | � \ } } | j | � q W| | f S)N)�range�decode�append)r
�size�offset�array�_�valuer r r �
_decode_array s
zDecoder._decode_arrayc C s | d k | f S)Nr r )r
r r r r r �_decode_boolean& s zDecoder._decode_booleanc C s! | | } | j | | � | f S)N)r )r
r r �
new_offsetr r r �
_decode_bytes) s
zDecoder._decode_bytesc s � � � f d d � } | S)Nc sj � s | j | � � | � } | j | | � } � rK | j � d � } t j � | � \ } | | f S)Ns )� _verify_sizer �rjust�struct�unpack)r
r r r Z packed_bytesr )�pad� type_code� type_sizer r � unpack_type2 s
z0Decoder._decode_packed_type.<locals>.unpack_typer )r r r r! r )r r r r �_decode_packed_type0 s
zDecoder._decode_packed_typec C s[ i } xH t | � D]: } | j | � \ } } | j | � \ } } | | | <q W| | f S)N)r r )r
r r Z containerr �keyr r r r � _decode_map> s zDecoder._decode_map� � i � i � c C s� | d ?d @d } | | } | j | | � } | d k rA | n t j d t | d @� � | } t | � } | | j | j | } | j r� | | f S| j | � \ } }
| | f S)Nr' r% r( s !c� ) r r �packr r r �_pointer_value_offsetr r ) r
r r Z pointer_sizer Z
pointer_bytesZpackedZunpacked�pointerr r r r r �_decode_pointerM s
zDecoder._decode_pointerc C s- | | } | j | | � } t | � | f S)N)r r )r
r r r Z
uint_bytesr r r � _decode_uint[ s
zDecoder._decode_uintc C s* | | } | j | | � j d � | f S)Nzutf-8)r r )r
r r r r r r �_decode_utf8_string` s
zDecoder._decode_utf8_strings !d� � � r) s !ir T� �
� � � s !fc C s� | d } t j d | j | | � � \ } | d ?} | sQ | j | � \ } } | | j k rx t d j d | � � � | j | | | � \ } } | j | | | | � S)z�Decode a section of the data section starting at offset
Arguments:
offset -- the location of the data structure to decode
r% s !Br1 z+Unexpected type number ({type}) encountered�type)r r r �_read_extended�
_type_decoderr �format�_size_from_ctrl_byte)r
r r � ctrl_byte�type_numr r r r r u s
"
zDecoder.decodec C sb t j d | j | | d � � \ } | d } | d k rT t d j d | � � � | | d f S)Ns !Br% r) zeSomething went horribly wrong in the decoder. An extended type resolved to a type number < 8 ({type})r8 )r r r r r; )r
r Z next_byter> r r r r9 � s &
zDecoder._read_extendedc C s | | k r t 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 @} | d k r | | f S| d k r2 d n | d } | | } | j | | � } | d k r� d t j d | � d } n[ | d k r� d t j d | � d } n2 | d k r� t j d
| j d d � � d d
} | | f S)N� r% � r � s !B� i s !Hs !Ir( s i )r r r r )r
r= r r> r Z
bytes_to_readr Z
size_bytesr r r r< � s
z Decoder._size_from_ctrl_byteN)�__name__�
__module__� __qualname__�__doc__r
r r r r" r$ r+ r- r. r/ r: r r9 r r< r r r r r s@
r ) rG �
__future__r r Zmaxminddb.compatr r Zmaxminddb.errorsr �objectr r r r r �<module> s