3 [_]xF@sdZddlmZddlZddlZddlZddlZddlm Z m Z m Z ddddddddZ ddiZ dZdZd Zd Zd Zd Zd ZGdddejjZGdddeZGdddeZdS)zTokenize DNS master file format)StringION)long text_type binary_typeT)   ;()"r c@seZdZdZdS)UngetBufferFullzDAn attempt was made to unget a token when the unget buffer was full.N)__name__ __module__ __qualname____doc__rr2/tmp/pip-build-yz_zf6az/dnspython/dns/tokenizer.pyr0src@seZdZdZd%ddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$S)&TokenzA DNS master file format token. ttype: The token type value: The token value has_escape: Does the token value contain escapes? FcCs||_||_||_dS)zInitialize a token instance.N)ttypevalue has_escape)selfrrrrrr__init__<szToken.__init__cCs |jtkS)N)rEOF)rrrris_eofCsz Token.is_eofcCs |jtkS)N)rEOL)rrrris_eolFsz Token.is_eolcCs |jtkS)N)r WHITESPACE)rrrr is_whitespaceIszToken.is_whitespacecCs |jtkS)N)r IDENTIFIER)rrrr is_identifierLszToken.is_identifiercCs |jtkS)N)r QUOTED_STRING)rrrris_quoted_stringOszToken.is_quoted_stringcCs |jtkS)N)rCOMMENT)rrrr is_commentRszToken.is_commentcCs |jtkS)N)r DELIMITER)rrrr is_delimiterUszToken.is_delimitercCs|jtkp|jtkS)N)rr#r!)rrrr is_eol_or_eofXszToken.is_eol_or_eofcCs&t|tsdS|j|jko$|j|jkS)NF) isinstancerrr)rotherrrr__eq__[s  z Token.__eq__cCs&t|tsdS|j|jkp$|j|jkS)NT)r0rrr)rr1rrr__ne__as  z Token.__ne__cCsd|j|jfS)Nz%d "%s")rr)rrrr__str__gsz Token.__str__cCs|js |Sd}t|j}d}x||kr|j|}|d7}|dkr||krPtjj|j|}|d7}|jr||krztjj|j|}|d7}||krtjj|j|}|d7}|jo|jstjjtt |dt |dt |}||7}qWt |j |S)Nrrr\d ) rlenrdns exception UnexpectedEndisdigit SyntaxErrorchrintrr)rZ unescapedlicc2c3rrrunescapejs6      $ zToken.unescapecCsdS)Nrr)rrrr__len__sz Token.__len__cCst|j|jfS)N)iterrr)rrrr__iter__szToken.__iter__cCs$|dkr|jS|dkr|jStdS)Nrr)rr IndexError)rrArrr __getitem__s zToken.__getitem__N)rF)rrrrr r"r$r&r(r*r,r.r/r2r3r4rErFrHrJrrrrr4s" rc@seZdZdZejdfddZddZddZd d Z d d Z d)ddZ ddZ ddZ e ZddZd*ddZddZd+ddZddZd,dd Zd-d!d"Zd.d#d$Zd%d&Zd'd(ZdS)/ TokenizeradA DNS master file format tokenizer. A token object is basically a (type, value) tuple. The valid types are EOF, EOL, WHITESPACE, IDENTIFIER, QUOTED_STRING, COMMENT, and DELIMITER. file: The file to tokenize ungotten_char: The most recently ungotten character, or None. ungotten_token: The most recently ungotten token, or None. multiline: The current multiline level. This value is increased by one every time a '(' delimiter is read, and decreased by one every time a ')' delimiter is read. quoting: This variable is true if the tokenizer is currently reading a quoted string. eof: This variable is true if the tokenizer has encountered EOF. delimiters: The current delimiter dictionary. line_number: The current line number filename: A filename that will be returned by the where() method. NcCst|tr t|}|dkr`d}n@t|trDt|j}|dkr`d}n|dkr`|tjkr\d}nd}||_d|_d|_ d|_ d|_ d|_ t |_d|_||_dS)aEInitialize a tokenizer instance. f: The file to tokenize. The default is sys.stdin. This parameter may also be a string, in which case the tokenizer will take its input from the contents of the string. filename: the name of the filename that the where() method will return. NzzzrFr)r0rrrdecodesysstdinfile ungotten_charungotten_token multilinequotingeof _DELIMITERS delimiters line_numberfilename)rfrXrrrr s*    zTokenizer.__init__cCsZ|jdkrJ|jrd}qV|jjd}|dkr2d|_qV|dkrV|jd7_n |j}d|_|S)z%Read a character from input. NrrTr )rPrTrOreadrW)rrBrrr _get_chars  zTokenizer._get_charcCs |j|jfS)zReturn the current location in the input. Returns a (string, int) tuple. The first item is the filename of the input, the second is the current line number. )rXrW)rrrrwhereszTokenizer.wherecCs|jdk rt||_dS)a%Unget a character. The unget buffer for characters is only one character large; it is an error to try to unget a character when the unget buffer is not empty. c: the character to unget raises UngetBufferFull: there is already an ungotten char N)rPr)rrBrrr _unget_chars zTokenizer._unget_charcCsLd}xB|j}|dkr<|dkr<|dks.|j r<|j||S|d7}qWdS)aFConsume input until a non-whitespace character is encountered. The non-whitespace character is then ungotten, and the number of whitespace characters consumed is returned. If the tokenizer is in multiline mode, then newlines are whitespace. Returns the number of characters skipped. rrrr rN)r[rRr])rskippedrBrrrskip_whitespaces  zTokenizer.skip_whitespaceFc CsT|jdk r>|j}d|_|jr(|r>|Sn|jr:|r>|Sn|S|j}|r\|dkr\ttdSd}t}d}x|j}|dks||jkr|dkr|j rt j j |dko|t kr|dkr|jd7_|jqlq |dkr |jdkrt j j|jd8_|jqln|d krH|j s0d |_ t|_t }qlnd|_ t|_|jqln|d kr\ttd S|d krx,|j}|d ks|dkrP||7}qhW|r|j|tt|S|dkr|jrt j jd ttS|jr|jd}qln ttd Sn|}t}n |j|Pn|j r|dkr|j}|dkr>t j j |jr|j}|dkrbt j j |j} |dkr|t j j |jo| jst j jtt|dt|dt| }n|d krt j jdn:|dkr||7}d }|j}|dks|d krt j j ||7}qlW|dkrH|t krH|jrDt j jd t}t|||S)aGet the next token. want_leading: If True, return a WHITESPACE token if the first character read is whitespace. The default is False. want_comment: If True, return a COMMENT token if the first token read is a comment. The default is False. Raises dns.exception.UnexpectedEnd: input ended prematurely Raises dns.exception.SyntaxError: input was badly formed Returns a Token. NrrrFr rr r Tr r zunbalanced parenthesesr5r6r7znewline in quoted string)rQr&r,r_rr%r'r[rVrSr9r:r;r)rRr=_QUOTING_DELIMITERSrUr#r]r+r!r-r<r>r?) rZ want_leadingZ want_commenttokenr^rrrBrCrDrrrgets                   &    z Tokenizer.getcCs|jdk rt||_dS)aUnget a token. The unget buffer for tokens is only one token large; it is an error to try to unget a token when the unget buffer is not empty. token: the token to unget Raises UngetBufferFull: there is already an ungotten token N)rQr)rrarrrungets zTokenizer.ungetcCs|j}|jrt|S)zHReturn the next item in an iteration. Returns a Token. )rbr" StopIteration)rrarrrnextszTokenizer.nextcCs|S)Nr)rrrrrHszTokenizer.__iter__r7cCsB|jj}|js tjjd|jjs6tjjdt|j|S)zRead the next token and interpret it as an integer. Raises dns.exception.SyntaxError if not an integer. Returns an int. zexpecting an identifierzexpecting an integer) rbrEr(r9r:r=rr<r?)rbaserarrrget_ints     zTokenizer.get_intcCs,|j}|dks|dkr(tjjd||S)zRead the next token and interpret it as an 8-bit unsigned integer. Raises dns.exception.SyntaxError if not an 8-bit unsigned integer. Returns an int. rz#%d is not an unsigned 8-bit integer)rgr9r:r=)rrrrr get_uint8s  zTokenizer.get_uint8cCsJ|j|d}|dks|dkrF|dkr6tjjd|ntjjd||S)zRead the next token and interpret it as a 16-bit unsigned integer. Raises dns.exception.SyntaxError if not a 16-bit unsigned integer. Returns an int. )rfriz*%o is not an octal unsigned 16-bit integerz$%d is not an unsigned 16-bit integer)rgr9r:r=)rrfrrrr get_uint16s   zTokenizer.get_uint16cCsh|jj}|js tjjd|jjs6tjjdt|j}|dksT|tdkrdtjjd||S)zRead the next token and interpret it as a 32-bit unsigned integer. Raises dns.exception.SyntaxError if not a 32-bit unsigned integer. Returns an int. zexpecting an identifierzexpecting an integerrlz$%d is not an unsigned 32-bit integer) rbrEr(r9r:r=rr<r)rrarrrr get_uint32s      zTokenizer.get_uint32cCs.|jj}|jp|js(tjjd|jS)zRead the next token and interpret it as a string. Raises dns.exception.SyntaxError if not a string. Returns a string. zexpecting a string)rbrEr(r*r9r:r=r)roriginrarrr get_strings  zTokenizer.get_stringcCs&|jj}|js tjjd|jS)zRead the next token, which should be an identifier. Raises dns.exception.SyntaxError if not an identifier. Returns a string. zexpecting an identifier)rbrEr(r9r:r=r)rrmrarrrget_identifiers  zTokenizer.get_identifiercCs,|j}|jstjjdtjj|j|S)zRead the next token and interpret it as a DNS name. Raises dns.exception.SyntaxError if not a name. Returns a dns.name.Name. zexpecting an identifier)rbr(r9r:r=name from_textr)rrmrarrrget_names zTokenizer.get_namecCs.|j}|js(tjjd|j|jf|jS)znRead the next token and raise an exception if it isn't EOL or EOF. Returns a string. z expected EOL or EOF, got %d "%s")rbr/r9r:r=rr)rrarrrget_eol!s  zTokenizer.get_eolcCs.|jj}|js tjjdtjj|jS)zRead the next token and interpret it as a DNS TTL. Raises dns.exception.SyntaxError or dns.ttl.BadTTL if not an identifier or badly formed. Returns an int. zexpecting an identifier) rbrEr(r9r:r=Zttlrqr)rrarrrget_ttl/s  zTokenizer.get_ttl)FF)r7)r7)N)N)N)rrrrrMrNr r[r\r]r_rbrcre__next__rHrgrirkrlrnrorrrsrtrrrrrKs(#  }    rK)riorrMZ dns.exceptionr9Zdns.nameZdns.ttl_compatrrrrUr`r!r#r%r'r)r+r-r:Z DNSExceptionrobjectrrKrrrrs0 d