U vù¾f…$ã@sVdZddlZddlmZddlmZddlmZmZm Z ddl mZGdd„dƒZdS) z& Implementation of an SSH2 "message". éN)ÚBytesIO)Úutil)Ú zero_byteÚmax_byteÚone_byte)Úuc@sþeZdZdZdZd>dd„Zdd„Zdd „Zd d„Zdd „Z dd„Z dd„Zdd„Zdd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Zd2d3„Zd4d5„Zd6d7„Zd8d9„Zd:d;„Z d`. :return: the next (`bytes`) byte of the message, or ``b''`` if there aren't any bytes remaining. é)r"rrrr Úget_byteqs zMessage.get_bytecCs| d¡}|tkS)z2 Fetch a boolean from the stream. r#)r"r©r r!rrr Úget_boolean|s zMessage.get_booleancCs>| d¡}|tkr t | ¡¡S|| d¡7}t d|¡dS)zZ Fetch an int from the stream. :return: a 32-bit unsigned `int`. r#éú>Ir)r"rrÚinflate_longÚ get_binaryÚstructÚunpack)r Úbyterrr Úget_adaptive_intƒs zMessage.get_adaptive_intcCst d| d¡¡dS)z/ Fetch an int from the stream. r(ér©r+r,r"rrrr Úget_intszMessage.get_intcCst d| d¡¡dS)zj Fetch a 64-bit int from the stream. :return: a 64-bit unsigned integer (`int`). ú>Qérr0rrrr Ú get_int64•szMessage.get_int64cCst | ¡¡S)zr Fetch a long int (mpint) from the stream. :return: an arbitrary-length integer (`int`). )rr)r*rrrr Ú 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"r1rrrr Ú get_string§szMessage.get_stringcCst| ¡ƒS)zš Fetch a Unicode string from the stream. This currently operates by attempting to encode the next "string" as ``utf-8``. )rr7rrrr Úget_text±szMessage.get_textcCs| | ¡¡S)z@ Alias for `get_string` (obtains a bytestring). r6rrrr r*ºszMessage.get_binarycCs| ¡ d¡S)zŽ Fetch a list of `strings ` from the stream. These are trivially encoded as comma-separated values in a string. ú,)r8Úsplitrrrr Úget_listÀszMessage.get_listcCs|j |¡|S)zj Write bytes to the stream, without any formatting. :param bytes b: bytes to add ©r Úwriter%rrr Ú add_bytesÈszMessage.add_bytescCs|j |¡|S)zq Write a single byte to the stream, without any formatting. :param bytes b: byte to add r<r%rrr Úadd_byteÑszMessage.add_bytecCs"|r|j t¡n|j t¡|S)za Add a boolean value to the stream. :param bool b: boolean value to add )r r=rrr%rrr Úadd_booleanÚszMessage.add_booleancCs|j t d|¡¡|S©zU Add an integer to the stream. :param int n: integer to add r(©r r=r+Úpack©r r rrr Úadd_intæszMessage.add_intcCs@|tjkr(|j t¡| t |¡¡n|j t d|¡¡|SrA) rÚbig_intr r=rÚ add_stringrÚdeflate_longr+rCrDrrr Úadd_adaptive_intïs zMessage.add_adaptive_intcCs|j t d|¡¡|S)zX Add a 64-bit int to the stream. :param int n: long int to add r2rBrDrrr Ú add_int64üszMessage.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 Úzrrr Ú add_mpintszMessage.add_mpintcCs(t |¡}| t|ƒ¡|j |¡|S)z[ Add a bytestring to the stream. :param byte s: bytestring to add )rrrErr r=)r Úsrrr 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)rGÚjoin)r Úlrrr Úadd_listszMessage.add_listcCsNt|ƒtkr| |¡St|tƒr*| |¡St|ƒtkr@| |¡S| |¡SdSr) ÚtypeÚboolr@Ú isinstanceÚintrIÚlistrPrG)r Úirrr Ú_add's zMessage._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 ÚseqÚitemrrr Úadd3s zMessage.add)N)"Ú__name__Ú __module__Ú__qualname__Ú__doc__rFrrrrrrrr"r$r&r.r1r4r5r7r8r*r;r>r?r@rErIrJrLrGrPrWrZrrrr rs> r) r^r+ÚiorZparamikorZparamiko.commonrrrZ paramiko.utilrrrrrr Ús