summaryrefslogtreecommitdiff
path: root/python/transform.py
diff options
context:
space:
mode:
authorJosh Watson <josh@trailofbits.com>2019-06-21 14:15:26 -0700
committerJosh Watson <josh@trailofbits.com>2019-06-21 14:15:26 -0700
commit2b52bef755671a22cd3533f5a8303678c740b385 (patch)
tree92173199e8f62224da497bfd9c18a20fe2d81adf /python/transform.py
parenta37745be0353874990a953c956059b9d8953f7af (diff)
Fix custom Transforms for python3
Diffstat (limited to 'python/transform.py')
-rw-r--r--python/transform.py12
1 files changed, 6 insertions, 6 deletions
diff --git a/python/transform.py b/python/transform.py
index 3afaaa59..95aeac87 100644
--- a/python/transform.py
+++ b/python/transform.py
@@ -196,11 +196,11 @@ class Transform(with_metaclass(_TransformMetaClass, object)):
param_map = {}
for i in range(0, count):
data = databuffer.DataBuffer(handle = core.BNDuplicateDataBuffer(params[i].value))
- param_map[params[i].name] = str(data)
- result = self.perform_decode(str(input_obj), param_map)
+ param_map[params[i].name] = bytes(data)
+ result = self.perform_decode(bytes(input_obj), param_map)
if result is None:
return False
- result = str(result)
+ result = bytes(result)
core.BNSetDataBufferContents(output_buf, result, len(result))
return True
except:
@@ -213,11 +213,11 @@ class Transform(with_metaclass(_TransformMetaClass, object)):
param_map = {}
for i in range(0, count):
data = databuffer.DataBuffer(handle = core.BNDuplicateDataBuffer(params[i].value))
- param_map[params[i].name] = str(data)
- result = self.perform_encode(str(input_obj), param_map)
+ param_map[params[i].name] = bytes(data)
+ result = self.perform_encode(bytes(input_obj), param_map)
if result is None:
return False
- result = str(result)
+ result = bytes(result)
core.BNSetDataBufferContents(output_buf, result, len(result))
return True
except: