Fix invalid variables

This commit is contained in:
Peter Šurda 2016-08-31 16:33:25 +08:00
parent cc90747da1
commit 029b0525de
Signed by untrusted user: PeterSurda
GPG Key ID: 0C5F50C0B5F37D87

View File

@ -1,5 +1,6 @@
#!/usr/bin/python2.7 #!/usr/bin/python2.7
import string
import msgpack import msgpack
import zlib import zlib
@ -8,7 +9,7 @@ from debug import logger
BITMESSAGE_ENCODING_IGNORE = 0 BITMESSAGE_ENCODING_IGNORE = 0
BITMESSAGE_ENCODING_TRIVIAL = 1 BITMESSAGE_ENCODING_TRIVIAL = 1
BITMESSAGE_ENCODING_SIMPLE = 2 BITMESSAGE_ENCODING_SIMPLE = 2
BITMESSAGE.ENCODING_EXTENDED = 3 BITMESSAGE_ENCODING_EXTENDED = 3
class MsgEncode(object): class MsgEncode(object):
def __init__(self, message, encoding = BITMESSAGE_ENCODING_SIMPLE): def __init__(self, message, encoding = BITMESSAGE_ENCODING_SIMPLE):
@ -71,14 +72,14 @@ class MsgDecode(object):
def decodeSimple(self, data): def decodeSimple(self, data):
bodyPositionIndex = string.find(data, '\nBody:') bodyPositionIndex = string.find(data, '\nBody:')
if bodyPositionIndex > 1: if bodyPositionIndex > 1:
subject = message[8:bodyPositionIndex] subject = data[8:bodyPositionIndex]
# Only save and show the first 500 characters of the subject. # Only save and show the first 500 characters of the subject.
# Any more is probably an attack. # Any more is probably an attack.
subject = subject[:500] subject = subject[:500]
body = message[bodyPositionIndex + 6:] body = data[bodyPositionIndex + 6:]
else: else:
subject = '' subject = ''
body = message body = data
# Throw away any extra lines (headers) after the subject. # Throw away any extra lines (headers) after the subject.
if subject: if subject:
subject = subject.splitlines()[0] subject = subject.splitlines()[0]