U ςd$@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}|tkr t|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_intcCstd|ddS)z/ Fetch an int from the stream. r(rr+r,r"rr r r get_intszMessage.get_intcCstd|ddS)zj Fetch a 64-bit int from the stream. :return: a 64-bit unsigned integer (`int`). >Qrr0rr 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_mpintszMessage.get_mpintcCs||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"r1rr r r get_stringszMessage.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``. )rr7rr r r get_textszMessage.get_textcCs||S)z@ Alias for `get_string` (obtains a bytestring). r6rr 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. ,)r8splitrr r r get_listszMessage.get_listcCs|j||S)zj Write bytes to the stream, without any formatting. :param bytes b: bytes to add r writer%r r r add_bytess zMessage.add_bytescCs|j||S)zq Write a single byte to the stream, without any formatting. :param bytes b: byte to add r<r%r r r add_bytes zMessage.add_bytecCs"|r|jtn |jt|S)za Add a boolean value to the stream. :param bool b: boolean value to add )r r=rrr%r r r add_booleans zMessage.add_booleancCs|jtd||SzU Add an integer to the stream. :param int n: integer to add r(r r=r+packr r r r r add_intszMessage.add_intcCs@|tjkr(|jt|t|n|jt d||SrA) rbig_intr r=r add_stringr deflate_longr+rCrDr r r add_adaptive_ints   zMessage.add_adaptive_intcCs|jtd||S)zX Add a 64-bit int to the stream. :param int n: long int to add r2rBrDr r r add_int64szMessage.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 )rGrrH)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 )rrrErr r=)r sr r r rGs  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 r9)rGjoin)r lr r r add_listszMessage.add_listcCsNt|tkr||St|tr*||St|tkr@||S||SdSr) typeboolr@ isinstanceintrIlistrPrG)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)rW)r seqitemr r r add3s z Message.add)N)"__name__ __module__ __qualname____doc__rFrrrrrrrr"r$r&r.r1r4r5r7r8r*r;r>r?r@rErIrJrLrGrPrWrZr r r r rs>                    r) r^r+iorZparamikorZparamiko.commonrrrZ paramiko.utilrrr r r r s