o g$@sVdZddlZddlmZddlmZddlmZmZm Z ddl m Z GdddZ dS) z& Implementation of an SSH2 "message". N)BytesIO)util) zero_bytemax_byteone_byte)uc@seZdZdZdZd>ddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Z d`. :return: the next (`bytes`) byte of the message, or ``b''`` if there aren't any bytes remaining. )r$rr r r get_byteqs zMessage.get_bytecCs|d}|tkS)z2 Fetch a boolean from the stream. r%)r$rr r"r r r get_boolean|s zMessage.get_booleancCs>|d}|tkrt|S||d7}td|dS)zZ Fetch an int from the stream. :return: a 32-bit unsigned `int`. r%>Ir)r$rr inflate_long get_binarystructunpack)r byter r r get_adaptive_ints zMessage.get_adaptive_intcCtd|ddS)z/ Fetch an int from the stream. r*rr-r.r$rr r r get_intrzMessage.get_intcCr1)zj Fetch a 64-bit int from the stream. :return: a 64-bit unsigned integer (`int`). >Qrr3rr r r get_int64szMessage.get_int64cCst|S)zr Fetch a long int (mpint) from the stream. :return: an arbitrary-length integer (`int`). )rr+r,rr r r get_mpintzMessage.get_mpintcC||S)z Fetch a "string" from the stream. This will actually be a `bytes` object, and may contain unprintable characters. (It's not unheard of for a string to contain another byte-stream message.) r$r4rr r r get_stringr9zMessage.get_stringcCs t|S)z Fetch a Unicode string from the stream. This currently operates by attempting to encode the next "string" as ``utf-8``. )rr<rr r r get_texts zMessage.get_textcCr:)z@ Alias for `get_string` (obtains a bytestring). r;rr r r r,szMessage.get_binarycCs|dS)z Fetch a list of `strings ` from the stream. These are trivially encoded as comma-separated values in a string. ,)r=splitrr r r get_listr9zMessage.get_listcC|j||S)zj Write bytes to the stream, without any formatting. :param bytes b: bytes to add r writer'r r r add_bytes zMessage.add_bytescCrA)zq Write a single byte to the stream, without any formatting. :param bytes b: byte to add rBr'r r r add_byterEzMessage.add_bytecCs$|r |jt|S|jt|S)za Add a boolean value to the stream. :param bool b: boolean value to add )r rCrrr'r r r add_booleans   zMessage.add_booleancC|jtd||SzU Add an integer to the stream. :param int n: integer to add r*r rCr-packr r!r r r add_intzMessage.add_intcCsB|tjkr|jt|t||S|jt d||SrI) rbig_intr rCr add_stringr deflate_longr-rKrLr r r add_adaptive_ints  zMessage.add_adaptive_intcCrH)zX Add a 64-bit int to the stream. :param int n: long int to add r5rJrLr r r add_int64rNzMessage.add_int64cCs|t||S)z Add a long int to the stream, encoded as an infinite-precision integer. This method only works on positive numbers. :param int z: long int to add )rPrrQ)r zr r r add_mpintszMessage.add_mpintcCs(t|}|t||j||S)z[ Add a bytestring to the stream. :param byte s: bytestring to add )rrrMr r rC)r sr r r rPs  zMessage.add_stringcCs|d||S)z Add a list of strings to the stream. They are encoded identically to a single string of values separated by commas. (Yes, really, that's how SSH2 does it.) :param l: list of strings to add r>)rPjoin)r lr r r add_listszMessage.add_listcCsJt|tur ||St|tr||St|tur ||S||Sr) typeboolrG isinstanceintrRlistrYrP)r ir r r _add's       z Message._addcGs|D]}||qdS)a Add a sequence of items to the stream. The values are encoded based on their type: bytes, str, int, bool, or list. .. warning:: Longs are encoded non-deterministically. Don't use this method. :param seq: the sequence of items N)r`)r seqitemr r r add3s z Message.addr)"__name__ __module__ __qualname____doc__rOrrrrrrrr$r&r(r0r4r7r8r<r=r,r@rDrFrGrMrRrSrUrPrYr`rcr r r r rs@                   r) rgr-iorparamikorparamiko.commonrrr paramiko.utilrrr r r r s