Commit e269cabe authored by Damien George's avatar Damien George

py/objint: In to_bytes(), allow length arg to be any int and check sign.

parent 8c5632a8
...@@ -435,7 +435,10 @@ STATIC mp_obj_t int_to_bytes(size_t n_args, const mp_obj_t *args) { ...@@ -435,7 +435,10 @@ STATIC mp_obj_t int_to_bytes(size_t n_args, const mp_obj_t *args) {
// TODO: Support signed param (assumes signed=False) // TODO: Support signed param (assumes signed=False)
(void)n_args; (void)n_args;
mp_uint_t len = MP_OBJ_SMALL_INT_VALUE(args[1]); mp_int_t len = mp_obj_get_int(args[1]);
if (len < 0) {
mp_raise_ValueError(NULL);
}
bool big_endian = args[2] != MP_OBJ_NEW_QSTR(MP_QSTR_little); bool big_endian = args[2] != MP_OBJ_NEW_QSTR(MP_QSTR_little);
vstr_t vstr; vstr_t vstr;
......
...@@ -14,3 +14,9 @@ print((10).to_bytes(1, "big")) ...@@ -14,3 +14,9 @@ print((10).to_bytes(1, "big"))
print((100).to_bytes(10, "big")) print((100).to_bytes(10, "big"))
print(int.from_bytes(b"\0\0\0\0\0\0\0\0\0\x01", "big")) print(int.from_bytes(b"\0\0\0\0\0\0\0\0\0\x01", "big"))
print(int.from_bytes(b"\x01\0", "big")) print(int.from_bytes(b"\x01\0", "big"))
# negative number of bytes should raise an error
try:
(1).to_bytes(-1, "little")
except ValueError:
print("ValueError")
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