py: Allow to pass in read-only buffers to viper and inline-asm funcs.

Fixes #4936.
pull/4967/head
Damien George 2019-08-05 16:32:10 +10:00
rodzic 2d3d4f7483
commit cd35dd9d9a
6 zmienionych plików z 18 dodań i 2 usunięć

Wyświetl plik

@ -65,7 +65,7 @@ mp_uint_t mp_native_from_obj(mp_obj_t obj, mp_uint_t type) {
case MP_NATIVE_TYPE_UINT: return mp_obj_get_int_truncated(obj);
default: { // cast obj to a pointer
mp_buffer_info_t bufinfo;
if (mp_get_buffer(obj, &bufinfo, MP_BUFFER_RW)) {
if (mp_get_buffer(obj, &bufinfo, MP_BUFFER_READ)) {
return (mp_uint_t)bufinfo.buf;
} else {
// assume obj is an integer that represents an address

Wyświetl plik

@ -472,7 +472,7 @@ STATIC mp_uint_t convert_obj_for_inline_asm(mp_obj_t obj) {
return (mp_uint_t)items;
} else {
mp_buffer_info_t bufinfo;
if (mp_get_buffer(obj, &bufinfo, MP_BUFFER_WRITE)) {
if (mp_get_buffer(obj, &bufinfo, MP_BUFFER_READ)) {
// supports the buffer protocol, return a pointer to the data
return (mp_uint_t)bufinfo.buf;
} else {

Wyświetl plik

@ -55,3 +55,7 @@ print(b, n)
b = array.array('b', (10, 20, 30, 40, 50, 60, 70, 80))
n = asm_sum_bytes(len(b), b)
print(b, n)
b = b'\x01\x02\x03\x04'
n = asm_sum_bytes(len(b), b)
print(b, n)

Wyświetl plik

@ -1,2 +1,3 @@
array('l', [100, 200, 300, 400]) 1000
array('b', [10, 20, 30, 40, 50, 60, 70, 80]) 360
b'\x01\x02\x03\x04' 10

Wyświetl plik

@ -9,6 +9,13 @@ def memset(dest:ptr8, c:int, n:int):
for i in range(n):
dest[i] = c
@micropython.viper
def memsum(src:ptr8, n:int) -> int:
s = 0
for i in range(n):
s += src[i]
return s
# create array and get its address
ar = bytearray('0000')
addr = get_addr(ar)
@ -27,3 +34,6 @@ print(ar)
# pass direct pointer to array buffer, with offset
memset(addr + 2, ord('3'), len(ar) - 2)
print(ar)
# pass a read-only bytes object in
print(memsum(b'\x01\x02\x03\x04', 4))

Wyświetl plik

@ -4,3 +4,4 @@ bytearray(b'0000')
bytearray(b'1111')
bytearray(b'2222')
bytearray(b'2233')
10