From 30484338092bfaad9f4ecff13b4219d94157fa06 Mon Sep 17 00:00:00 2001
From: Dave Hylands <dhylands@gmail.com>
Date: Wed, 25 Nov 2015 18:54:14 -0800
Subject: [PATCH] stmhal: Make stm.mem* support large integers.

With these you can now do things like:

stm.mem32[0x20000000] = 0x80000000

and read 32-bit values. You can also read all the way to the end
of memory using either stm.mem32[0xfffffffc] or stm.mem32[-4].

IRQs shouldn't use mem32 at all since they'd fail if the top 2 bits
weren't equal, so IRQs should be using 16-bit I/O.
---
 stmhal/modstm.c | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/stmhal/modstm.c b/stmhal/modstm.c
index f1f8ec390..6741a5c17 100644
--- a/stmhal/modstm.c
+++ b/stmhal/modstm.c
@@ -38,7 +38,10 @@
 // from an object, we must clear the MSB.
 
 STATIC uint32_t get_read_addr(mp_obj_t addr_o, uint align) {
-    uint32_t addr = mp_obj_get_int(addr_o) & 0x7fffffff;
+    uint32_t addr = mp_obj_get_int_truncated(addr_o);
+    if (MP_OBJ_IS_SMALL_INT(addr_o)) {
+        addr &= 0x7fffffff;
+    }
     /*
     if (addr < 0x10000000) {
         nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "cannot read from address %08x", addr));
@@ -51,7 +54,10 @@ STATIC uint32_t get_read_addr(mp_obj_t addr_o, uint align) {
 }
 
 STATIC uint32_t get_write_addr(mp_obj_t addr_o, uint align) {
-    uint32_t addr = mp_obj_get_int(addr_o) & 0x7fffffff;
+    uint32_t addr = mp_obj_get_int_truncated(addr_o);
+    if (MP_OBJ_IS_SMALL_INT(addr_o)) {
+        addr &= 0x7fffffff;
+    }
     if (addr < 0x10000000) {
         // Everything below 0x10000000 is either ROM or aliased to something higher, so we don't
         // lose anything by restricting writes to this area, and we gain some safety.
@@ -88,11 +94,11 @@ STATIC mp_obj_t stm_mem_subscr(mp_obj_t self_in, mp_obj_t index, mp_obj_t value)
             case 2: val = (*(uint16_t*)addr); break;
             default: val = (*(uint32_t*)addr); break;
         }
-        return mp_obj_new_int(val);
+        return mp_obj_new_int_from_uint(val);
     } else {
         // store
         uint32_t addr = get_write_addr(index, self->elem_size);
-        uint32_t val = mp_obj_get_int(value);
+        uint32_t val = mp_obj_get_int_truncated(value);
         switch (self->elem_size) {
             case 1: (*(uint8_t*)addr) = val; break;
             case 2: (*(uint16_t*)addr) = val; break;
-- 
GitLab