Skip to content

Commit eafcfeb

Browse files
committed
Safe int handling
1 parent 3f288c6 commit eafcfeb

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

pyfaasm/pyfaasm/core.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,13 +74,16 @@ def read_state_size(key):
7474

7575

7676
def read_state(key, state_len):
77+
state_len = int(state_len)
7778
buff = ctypes.create_string_buffer(state_len)
7879
_host_interface.__faasm_read_state(bytes(key, "utf-8"), buff, state_len)
7980

8081
return bytes(buff)
8182

8283

8384
def read_state_offset(key, total_len, offset, offset_len):
85+
total_len = int(total_len)
86+
offset_len = int(offset_len)
8487
buff = ctypes.create_string_buffer(offset_len)
8588
_host_interface.__faasm_read_state_offset(
8689
bytes(key, "utf-8"), total_len, offset, buff, offset_len
@@ -94,6 +97,9 @@ def write_state(key, value):
9497

9598

9699
def write_state_offset(key, total_len, offset, value):
100+
offset = int(offset)
101+
total_len = int(total_len)
102+
97103
_host_interface.__faasm_write_state_offset(
98104
bytes(key, "utf-8"), total_len, offset, value, len(value)
99105
)
@@ -108,6 +114,7 @@ def push_state_partial(key):
108114

109115

110116
def pull_state(key, state_len):
117+
state_len = int(state_len)
111118
_host_interface.__faasm_pull_state(bytes(key, "utf-8"), state_len)
112119

113120

0 commit comments

Comments
 (0)