elie/ssh-audit
1
0
Fork 0
mirror of https://github.com/jtesta/ssh-audit.git synced 2025-06-23 11:04:31 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix and update write buffer. Add buffer tests.

Browse Source
This commit is contained in:
Andris Raugulis
2016-10-05 06:06:26 +03:00
parent 262c65b7be
commit 7959c7448a
3 changed files with 124 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ssh-audit.py
View File

@ -416,7 +416,7 @@ class WriteBuf(object):
return self.write(v)
def write_list(self, v):
self.write_string(u','.join(v))
return self.write_string(u','.join(v))
@classmethod
def _bitlength(cls, n):
@ -454,6 +454,12 @@ class WriteBuf(object):
data = self._create_mpint(n)
return self.write_string(data)
def write_line(self, v):
if not isinstance(v, bytes):
v = bytes(bytearray(v, 'utf-8'))
v += b'\r\n'
return self.write(v)
def write_flush(self):
payload = self._wbuf.getvalue()
self._wbuf.truncate(0)