Fixed handling of null-termination in the string and unicode string readers.

[~jspiros/python-ebml.git] / ebml / core.py

diff --git a/ebml/core.py b/ebml/core.py
index 0f390a8..8c01068 100644 (file)
--- a/ebml/core.py
+++ b/ebml/core.py
@@ -189,7 +189,7 @@ def read_string(stream, size):
        value = ''
        if size > 0:
                value = stream.read(size)
-               value = value.rstrip(chr(0))
+               value = value.partition(chr(0))[0]
        return value
 
 
@@ -209,7 +209,7 @@ def read_unicode_string(stream, size):
        value = u''
        if size > 0:
                data = stream.read(size)
-               data = data.rstrip(chr(0))
+               data = data.partition(chr(0))[0]
                value = unicode(data, 'utf_8')
        return value