U [e*@sdZz ddlZWnek r(dZYnXddlZddlZddlmZmZddlmZddl m Z m Z m Z m Z mZmZddlmZmZmZmZmZddlmZddlmZdd lmZdd lmZGd d d ZGd ddZdS)zo maxminddb.reader ~~~~~~~~~~~~~~~~ This module contains the pure Python database reader and related classes. N) IPv4Address IPv6Address)PathLike)AnyAnyStrIOOptionalTupleUnion) MODE_AUTO MODE_MMAP MODE_FILE MODE_MEMORYMODE_FD)Decoder)InvalidDatabaseError) FileBuffer)Recordc@s$eZdZUdZdZdZeeedfe d<dZ e e e d<e feee eefe ddd d Zd d d dZeeeefe edddZeeeefee ee fdddZeee e fdddZe e dddZe e e dddZe edddZdd d d!Zdd d"d#Z dd d$d%Z!dS)&Readerz Instances of this class provide a reader for the MaxMind DB format. IP addresses can be looked up using the ``get`` method. sMaxMind.comz mmap.mmap_bufferN _ipv4_start)databasemodereturnc Cs|tkr ts|tkrTt|d*}tj|dtjd|_|j|_W5QRX|}n|tt fkr|t ||_|j|_|}np|t krt|d}| |_t |j|_W5QRX|}n6|tkr| |_t |j|_|j}ntd|d|j|jtd|jd}|dkr*|td|d |t |j7}t|j|}||\}}t|tsntd |d tf||_t|j|jj|j|_d |_d S)aReader for the MaxMind DB file format Arguments: database -- A path to a valid MaxMind DB file such as a GeoIP2 database file, or a file descriptor in the case of MODE_FD. mode -- mode to open the database with. Valid mode are: * MODE_MMAP - read from memory map. * MODE_FILE - read database as standard file. * MODE_MEMORY - load database into memory. * MODE_AUTO - tries MODE_MMAP and then MODE_FILE. Default. * MODE_FD - the param passed via database is a file descriptor, not a path. This mode implies MODE_MEMORY. rbr)accesszUnsupported open mode (z]). Only MODE_AUTO, MODE_FILE, MODE_MEMORY and MODE_FD are supported by the pure Python ReaderizError opening database file (z#). Is this a valid MaxMind DB file?z)Error reading metadata in database file (z).FN) r mmapr openfileno ACCESS_READrsize _buffer_sizer rrreadlenrname ValueErrorrfind_METADATA_START_MARKERmaxcloserrdecode isinstancedictMetadata _metadatasearch_tree_size_DATA_SECTION_SEPARATOR_SIZE_decoderclosed) selfrrZdb_filefilenameZmetadata_startZmetadata_decodermetadata_r9R/var/www/html/services/stratfitenv/lib/python3.8/site-packages/maxminddb/reader.py__init__'sV                zReader.__init__r/rcCs|jS)z7Return the metadata associated with the MaxMind DB file)r0r5r9r9r:r7mszReader.metadata) ip_addressrcCs||\}}|S)zReturn the record for the ip_address in the MaxMind DB Arguments: ip_address -- an IP address in the standard string notation )get_with_prefix_len)r5r>recordr8r9r9r:getqsz Reader.getc Cst|trt|}n|}zt|j}Wn,tk rT}ztd|W5d}~XYnX|jdkr||j j dkr|t d|d| |\}}|r| ||fSd|fS)zReturn a tuple with the record and the associated prefix length Arguments: ip_address -- an IP address in the standard string notation z/argument 1 must be a string or ipaddress objectNzError looking up zD. You attempted to look up an IPv6 address in an IPv4-only database.)r-str ipaddressr> bytearraypackedAttributeError TypeErrorversionr0 ip_versionr'_find_address_in_tree_resolve_data_pointer)r5r>addressZpacked_addressexpointer prefix_lenr9r9r:r?{s   zReader.get_with_prefix_len)rGrcCst|d}||}|jj}d}||krd||krdd||d?d|d?@}|||}|d}q"||krtd|fS||kr||fStddS)NrzInvalid node in search tree)r% _start_noder0 node_count _read_noder)r5rGZ bit_countnoderWibitr9r9r:rLs    zReader._find_address_in_tree)lengthrcCs\|jjdks|dkrdS|jr$|jSd}tdD] }||jjkrDqR||d}q0||_|S)NrBr`)r0rKrrangerWrX)r5r\rYr8r9r9r:rVs  zReader._start_node) node_numberindexrcCs||jj}|jj}|dkr@||d}d|j||d}n|dkr|d|}t|j||d}|rd|d@|d<qd|@d?}|d|n6|d kr||d}|j||d}ntd |t d |dS) NrTrCr zUnknown record size: s!I) r0node_byte_size record_sizerrFpopinsertrstructunpack)r5r`raZ base_offsetrioffsetZ node_bytesmiddler9r9r:rXs"    zReader._read_node)rPrcCs:||jj|jj}||jkr&td|j|\}}|S)Nz,The MaxMind DB file's search tree is corrupt)r0rWr1r#rr3r,)r5rPresolveddatar8r9r9r:rMs  zReader._resolve_data_pointercCs.z|jWntk r"YnXd|_dS)zBCloses the MaxMind DB file and returns the resources to the systemTN)rr+rHr4r=r9r9r:r+s z Reader.closecGs |dS)N)r+r5argsr9r9r:__exit__szReader.__exit__cCs|jrtd|S)Nz%Attempt to reopen a closed MaxMind DB)r4r'r=r9r9r: __enter__szReader.__enter__)"__name__ __module__ __qualname____doc__r2r)r bytesr__annotations__rrintr rrrr;r7rDrrrrAr r?rFrLrVrXrMr+rtrur9r9r9r:rs.  F   rc@sJeZdZdZddddZeedddZeeddd Zd d Z dS) r/aMetadata for the MaxMind DB reader .. attribute:: binary_format_major_version The major version number of the binary format used when creating the database. :type: int .. attribute:: binary_format_minor_version The minor version number of the binary format used when creating the database. :type: int .. attribute:: build_epoch The Unix epoch for the build time of the database. :type: int .. attribute:: database_type A string identifying the database type, e.g., "GeoIP2-City". :type: str .. attribute:: description A map from locales to text descriptions of the database. :type: dict(str, str) .. attribute:: ip_version The IP version of the data in a database. A value of "4" means the database only supports IPv4. A database with a value of "6" may support both IPv4 and IPv6 lookups. :type: int .. attribute:: languages A list of locale codes supported by the databse. :type: list(str) .. attribute:: node_count The number of nodes in the database. :type: int .. attribute:: record_size The bit size of a record in the search tree. :type: int Nr<cKs^|d|_|d|_|d|_|d|_|d|_|d|_|d|_|d|_|d |_d S) zACreates new Metadata object. kwargs are key/value pairs from specrWrirK database_type languagesbinary_format_major_versionbinary_format_minor_version build_epoch descriptionN) rWrirKr}r~rrrr)r5kwargsr9r9r:r;/s        zMetadata.__init__cCs |jdS)z8The size of a node in bytes :type: int rC)rir=r9r9r:rh=szMetadata.node_byte_sizecCs |j|jS)z8The size of the search tree :type: int )rWrhr=r9r9r:r1EszMetadata.search_tree_sizecCs6ddd|jD}|jd|jjd|dS)Nz, css |]\}}|d|VqdS)=Nr9).0kvr9r9r: Nsz$Metadata.__repr__...())join__dict__itemsrw __class__rvrrr9r9r:__repr__MszMetadata.__repr__) rvrwrxryr;propertyr|rhr1rr9r9r9r:r/s@r/) ryr ImportErrorrErlrrosrtypingrrrrr r Zmaxminddb.constr r r rrZmaxminddb.decoderrZmaxminddb.errorsrZmaxminddb.filerZmaxminddb.typesrrr/r9r9r9r:s"        T