Commit 3b1563eb authored by Radostin Stoyanov's avatar Radostin Stoyanov Committed by Andrei Vagin

crit-images: Don't overwrite 'pb' with local var

Rename the local variables 'pb' to 'pbuff' to avoid conflict with the
imported 'pb' module.
Signed-off-by: 's avatarRadostin Stoyanov <rstoyanov1@gmail.com>
Signed-off-by: 's avatarAndrei Vagin <avagin@gmail.com>
parent ac3296c3
...@@ -91,13 +91,13 @@ class entry_handler: ...@@ -91,13 +91,13 @@ class entry_handler:
entry = {} entry = {}
# Read payload # Read payload
pb = self.payload() pbuff = self.payload()
buf = f.read(4) buf = f.read(4)
if buf == b'': if buf == b'':
break break
size, = struct.unpack('i', buf) size, = struct.unpack('i', buf)
pb.ParseFromString(f.read(size)) pbuff.ParseFromString(f.read(size))
entry = pb2dict.pb2dict(pb, pretty) entry = pb2dict.pb2dict(pbuff, pretty)
# Read extra # Read extra
if self.extra_handler: if self.extra_handler:
...@@ -112,10 +112,10 @@ class entry_handler: ...@@ -112,10 +112,10 @@ class entry_handler:
num /= 1024.0 num /= 1024.0
return "%.1fYB" % num return "%.1fYB" % num
pl_size = self.extra_handler.skip(f, pb) pl_size = self.extra_handler.skip(f, pbuff)
entry['extra'] = '... <%s>' % human_readable(pl_size) entry['extra'] = '... <%s>' % human_readable(pl_size)
else: else:
entry['extra'] = self.extra_handler.load(f, pb) entry['extra'] = self.extra_handler.load(f, pbuff)
entries.append(entry) entries.append(entry)
...@@ -138,16 +138,16 @@ class entry_handler: ...@@ -138,16 +138,16 @@ class entry_handler:
extra = entry.pop('extra', None) extra = entry.pop('extra', None)
# Write payload # Write payload
pb = self.payload() pbuff = self.payload()
pb2dict.dict2pb(entry, pb) pb2dict.dict2pb(entry, pbuff)
pb_str = pb.SerializeToString() pb_str = pbuff.SerializeToString()
size = len(pb_str) size = len(pb_str)
f.write(struct.pack('i', size)) f.write(struct.pack('i', size))
f.write(pb_str) f.write(pb_str)
# Write extra # Write extra
if self.extra_handler and extra: if self.extra_handler and extra:
self.extra_handler.dump(extra, f, pb) self.extra_handler.dump(extra, f, pbuff)
def dumps(self, entries): def dumps(self, entries):
""" """
...@@ -312,7 +312,7 @@ class sk_queues_extra_handler: ...@@ -312,7 +312,7 @@ class sk_queues_extra_handler:
data = f.read(size) data = f.read(size)
return base64.encodebytes(data) return base64.encodebytes(data)
def dump(self, extra, f, pb): def dump(self, extra, f, _unused):
data = base64.decodebytes(extra) data = base64.decodebytes(extra)
f.write(data) f.write(data)
...@@ -322,31 +322,31 @@ class sk_queues_extra_handler: ...@@ -322,31 +322,31 @@ class sk_queues_extra_handler:
class tcp_stream_extra_handler: class tcp_stream_extra_handler:
def load(self, f, pb): def load(self, f, pbuff):
d = {} d = {}
inq = f.read(pb.inq_len) inq = f.read(pbuff.inq_len)
outq = f.read(pb.outq_len) outq = f.read(pbuff.outq_len)
d['inq'] = base64.encodebytes(inq) d['inq'] = base64.encodebytes(inq)
d['outq'] = base64.encodebytes(outq) d['outq'] = base64.encodebytes(outq)
return d return d
def dump(self, extra, f, pb): def dump(self, extra, f, _unused):
inq = base64.decodebytes(extra['inq']) inq = base64.decodebytes(extra['inq'])
outq = base64.decodebytes(extra['outq']) outq = base64.decodebytes(extra['outq'])
f.write(inq) f.write(inq)
f.write(outq) f.write(outq)
def skip(self, f, pb): def skip(self, f, pbuff):
f.seek(0, os.SEEK_END) f.seek(0, os.SEEK_END)
return pb.inq_len + pb.outq_len return pbuff.inq_len + pbuff.outq_len
class ipc_sem_set_handler: class ipc_sem_set_handler:
def load(self, f, pb): def load(self, f, pbuff):
entry = pb2dict.pb2dict(pb) entry = pb2dict.pb2dict(pbuff)
size = sizeof_u16 * entry['nsems'] size = sizeof_u16 * entry['nsems']
rounded = round_up(size, sizeof_u64) rounded = round_up(size, sizeof_u64)
s = array.array('H') s = array.array('H')
...@@ -356,8 +356,8 @@ class ipc_sem_set_handler: ...@@ -356,8 +356,8 @@ class ipc_sem_set_handler:
f.seek(rounded - size, 1) f.seek(rounded - size, 1)
return s.tolist() return s.tolist()
def dump(self, extra, f, pb): def dump(self, extra, f, pbuff):
entry = pb2dict.pb2dict(pb) entry = pb2dict.pb2dict(pbuff)
size = sizeof_u16 * entry['nsems'] size = sizeof_u16 * entry['nsems']
rounded = round_up(size, sizeof_u64) rounded = round_up(size, sizeof_u64)
s = array.array('H') s = array.array('H')
...@@ -369,22 +369,22 @@ class ipc_sem_set_handler: ...@@ -369,22 +369,22 @@ class ipc_sem_set_handler:
f.write(s.tostring()) f.write(s.tostring())
f.write('\0' * (rounded - size)) f.write('\0' * (rounded - size))
def skip(self, f, pb): def skip(self, f, pbuff):
entry = pb2dict.pb2dict(pb) entry = pb2dict.pb2dict(pbuff)
size = sizeof_u16 * entry['nsems'] size = sizeof_u16 * entry['nsems']
f.seek(round_up(size, sizeof_u64), os.SEEK_CUR) f.seek(round_up(size, sizeof_u64), os.SEEK_CUR)
return size return size
class ipc_msg_queue_handler: class ipc_msg_queue_handler:
def load(self, f, pb): def load(self, f, pbuff):
entry = pb2dict.pb2dict(pb) entry = pb2dict.pb2dict(pbuff)
messages = [] messages = []
for x in range (0, entry['qnum']): for x in range (0, entry['qnum']):
buf = f.read(4) buf = f.read(4)
if buf == '': if buf == '':
break break
size, = struct.unpack('i', buf) size, = struct.unpack('i', buf)
msg = pb.ipc_msg() msg = pbuff.ipc_msg()
msg.ParseFromString(f.read(size)) msg.ParseFromString(f.read(size))
rounded = round_up(msg.msize, sizeof_u64) rounded = round_up(msg.msize, sizeof_u64)
data = f.read(msg.msize) data = f.read(msg.msize)
...@@ -393,10 +393,10 @@ class ipc_msg_queue_handler: ...@@ -393,10 +393,10 @@ class ipc_msg_queue_handler:
messages.append(base64.encodebytes(data)) messages.append(base64.encodebytes(data))
return messages return messages
def dump(self, extra, f, pb): def dump(self, extra, f, pbuff):
entry = pb2dict.pb2dict(pb) entry = pb2dict.pb2dict(pbuff)
for i in range (0, len(extra), 2): for i in range (0, len(extra), 2):
msg = pb.ipc_msg() msg = pbuff.ipc_msg()
pb2dict.dict2pb(extra[i], msg) pb2dict.dict2pb(extra[i], msg)
msg_str = msg.SerializeToString() msg_str = msg.SerializeToString()
size = len(msg_str) size = len(msg_str)
...@@ -407,15 +407,15 @@ class ipc_msg_queue_handler: ...@@ -407,15 +407,15 @@ class ipc_msg_queue_handler:
f.write(data[:msg.msize]) f.write(data[:msg.msize])
f.write('\0' * (rounded - msg.msize)) f.write('\0' * (rounded - msg.msize))
def skip(self, f, pb): def skip(self, f, pbuff):
entry = pb2dict.pb2dict(pb) entry = pb2dict.pb2dict(pbuff)
pl_len = 0 pl_len = 0
for x in range (0, entry['qnum']): for x in range (0, entry['qnum']):
buf = f.read(4) buf = f.read(4)
if buf == '': if buf == '':
break break
size, = struct.unpack('i', buf) size, = struct.unpack('i', buf)
msg = pb.ipc_msg() msg = pbuff.ipc_msg()
msg.ParseFromString(f.read(size)) msg.ParseFromString(f.read(size))
rounded = round_up(msg.msize, sizeof_u64) rounded = round_up(msg.msize, sizeof_u64)
f.seek(rounded, os.SEEK_CUR) f.seek(rounded, os.SEEK_CUR)
...@@ -424,24 +424,24 @@ class ipc_msg_queue_handler: ...@@ -424,24 +424,24 @@ class ipc_msg_queue_handler:
return pl_len return pl_len
class ipc_shm_handler: class ipc_shm_handler:
def load(self, f, pb): def load(self, f, pbuff):
entry = pb2dict.pb2dict(pb) entry = pb2dict.pb2dict(pbuff)
size = entry['size'] size = entry['size']
data = f.read(size) data = f.read(size)
rounded = round_up(size, sizeof_u32) rounded = round_up(size, sizeof_u32)
f.seek(rounded - size, 1) f.seek(rounded - size, 1)
return base64.encodebytes(data) return base64.encodebytes(data)
def dump(self, extra, f, pb): def dump(self, extra, f, pbuff):
entry = pb2dict.pb2dict(pb) entry = pb2dict.pb2dict(pbuff)
size = entry['size'] size = entry['size']
data = base64.decodebytes(extra) data = base64.decodebytes(extra)
rounded = round_up(size, sizeof_u32) rounded = round_up(size, sizeof_u32)
f.write(data[:size]) f.write(data[:size])
f.write('\0' * (rounded - size)) f.write('\0' * (rounded - size))
def skip(self, f, pb): def skip(self, f, pbuff):
entry = pb2dict.pb2dict(pb) entry = pb2dict.pb2dict(pbuff)
size = entry['size'] size = entry['size']
rounded = round_up(size, sizeof_u32) rounded = round_up(size, sizeof_u32)
f.seek(rounded, os.SEEK_CUR) f.seek(rounded, os.SEEK_CUR)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment