a Oc*@sdZz ddlZWney&dZYn0ddlZddlZddlmZmZddlmZddl 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|tkrht|d4}tj|dtjd|_|j|_Wdn1sX0Y|}n|tt fvrt ||_|j|_|}n|t krt|d&}| |_t |j|_Wdn1s0Y|}n8|tkr| |_t |j|_|j}ntd|d|j|jtd|jd}|dkrT|td |d |t |j7}t|j|}||\}}t|tstd |d tfi||_t|j|jj|j|_d |_dS)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)accessNzUnsupported 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).F) r mmapr openfilenoZ ACCESS_READrsize _buffer_sizer rrreadlenrname ValueErrorrfind_METADATA_START_MARKERmaxcloserrdecode isinstancedictMetadata _metadatasearch_tree_size_DATA_SECTION_SEPARATOR_SIZE_decoderclosed) selfrrZdb_filefilenameZmetadata_startZmetadata_decodermetadata_r8G/var/www/html/gsl.tools/lib/python3.9/site-packages/maxminddb/reader.py__init__'sT *     *          zReader.__init__r.rcCs|jS)z7Return the metadata associated with the MaxMind DB file)r/r4r8r8r9r6mszReader.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)r4r=recordr7r8r8r9getqsz Reader.getc Cst|trt|}n|}zt|j}Wn.tyV}ztd|WYd}~n d}~00|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 TypeErrorversionr/ ip_versionr&_find_address_in_tree_resolve_data_pointer)r4r=addressZpacked_addressexpointer prefix_lenr8r8r9r>{s    zReader.get_with_prefix_len)rFrcCst|d}||}|jj}d}||krd||krdd||d?d|d?@}|||}|d}q"||krtd|fS||kr||fStddS)NrzInvalid node in search tree)r$ _start_noder/ node_count _read_noder)r4rFZ bit_countnoderVibitr8r8r9rKs    zReader._find_address_in_tree)lengthrcCs\|jjdks|dkrdS|jr$|jSd}tdD] }||jjkrDqR||d}q0||_|S)NrAr`)r/rJrrangerVrW)r4r[rXr7r8r8r9rUs  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) NrSrBr zUnknown record size: s!I) r/node_byte_size record_sizerrEpopinsertrstructunpack)r4r_r`Z base_offsetrhoffsetZ node_bytesZmiddler8r8r9rWs"    zReader._read_node)rOrcCs:||jj|jj}||jkr&td|j|\}}|S)Nz,The MaxMind DB file's search tree is corrupt)r/rVr0r"rr2r+)r4rOZresolveddatar7r8r8r9rLs  zReader._resolve_data_pointercCs,z|jWnty Yn0d|_dS)zBCloses the MaxMind DB file and returns the resources to the systemTN)rr*rGr3r<r8r8r9r*s  z Reader.closecGs |dS)N)r*r4argsr8r8r9__exit__szReader.__exit__cCs|jrtd|S)Nz%Attempt to reopen a closed MaxMind DB)r3r&r<r8r8r9 __enter__szReader.__enter__)"__name__ __module__ __qualname____doc__r1r(r bytesr__annotations__rrintr rrrr:r6rCrrrr@r r>rErKrUrWrLr*rqrrr8r8r8r9rs,  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 specrVrhrJ database_type languagesbinary_format_major_versionbinary_format_minor_version build_epoch descriptionN) rVrhrJrzr{r|r}r~r)r4kwargsr8r8r9r:/s        zMetadata.__init__cCs |jdS)z8The size of a node in bytes :type: int rB)rhr<r8r8r9rg=szMetadata.node_byte_sizecCs |j|jS)z8The size of the search tree :type: int )rVrgr<r8r8r9r0EszMetadata.search_tree_sizecCs6ddd|jD}|jd|jjd|dS)Nz, css |]\}}|d|VqdS)=Nr8).0kvr8r8r9 Nz$Metadata.__repr__...())join__dict__itemsrt __class__rsror8r8r9__repr__MszMetadata.__repr__) rsrtrurvr:propertyryrgr0rr8r8r8r9r.s@r.) rvr ImportErrorrDrkrrosrtypingrrrrr r Zmaxminddb.constr r r rrZmaxminddb.decoderrZmaxminddb.errorsrZmaxminddb.filerZmaxminddb.typesrrr.r8r8r8r9s"         T