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

Use safer UTF-8 decoding (with replace) and add related tests.

Browse Source
This commit is contained in:
Andris Raugulis
2016-10-25 13:53:51 +03:00
parent 66bd6c3ef0
commit 4bbb1f4d11
2 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ssh-audit.py
View File

@ -40,12 +40,12 @@ else:
try:
# pylint: disable=unused-import
from typing import List, Tuple, Optional, Callable, Union, Any
except ImportError:
except ImportError: # pragma: nocover
pass
try:
from colorama import init as colorama_init
colorama_init()
except ImportError:
except ImportError: # pragma: nocover
pass
@ -572,7 +572,7 @@ class ReadBuf(object):
def read_list(self):
# type: () -> List[text_type]
list_size = self.read_int()
return self.read(list_size).decode().split(',')
return self.read(list_size).decode('utf-8', 'replace').split(',')
def read_string(self):
# type: () -> binary_type
@ -607,7 +607,7 @@ class ReadBuf(object):
def read_line(self):
# type: () -> text_type
return self._buf.readline().rstrip().decode('utf-8')
return self._buf.readline().rstrip().decode('utf-8', 'replace')
class WriteBuf(object):