From 5f6ae74dad03cad6f61f517a67bb19e33ddfc040 Mon Sep 17 00:00:00 2001
From: Neradoc <neraOnGit@ri1.fr>
Date: Wed, 29 Mar 2023 14:31:54 +0200
Subject: [PATCH] Add big int format to msgpack.pack

---
 shared-module/msgpack/__init__.c | 49 ++++++++++++++++++++++----------
 1 file changed, 34 insertions(+), 15 deletions(-)

diff --git a/shared-module/msgpack/__init__.c b/shared-module/msgpack/__init__.c
index 3a34731d77..20e0e499c0 100644
--- a/shared-module/msgpack/__init__.c
+++ b/shared-module/msgpack/__init__.c
@@ -30,6 +30,7 @@
 #include "py/obj.h"
 #include "py/binary.h"
 #include "py/objarray.h"
+#include "py/objint.h"
 #include "py/objlist.h"
 #include "py/objstringio.h"
 #include "py/parsenum.h"
@@ -195,19 +196,34 @@ STATIC mp_map_elem_t *dict_iter_next(mp_obj_dict_t *dict, size_t *cur) {
     return NULL;
 }
 
-STATIC void pack_int(msgpack_stream_t *s, int32_t x) {
-    if (x > -32 && x < 128) {
-        write1(s, x);
-    } else if ((int8_t)x == x) {
-        write1(s, 0xd0);
-        write1(s, x);
-    } else if ((int16_t)x == x) {
-        write1(s, 0xd1);
-        write2(s, x);
+STATIC void pack_int(msgpack_stream_t *s, mp_obj_t obj, bool _signed) {
+    byte buffer[9];
+    byte *buf = (buffer + 1);
+    byte *type = buffer;
+    size_t len = 0;
+    if (mp_obj_is_small_int(obj)) {
+        int32_t x = MP_OBJ_SMALL_INT_VALUE(obj);
+        if (x > -32 && x < 128) {
+            write1(s, x);
+            return;
+        } else if (-0x80 <= x && x <= 0xff) {
+            *type = _signed ? 0xd0 : 0xcc;
+            len = 1;
+        } else if (-0x8000 <= x && x <= 0xffff) {
+            *type = _signed ? 0xd1 : 0xcd;
+            len = 2;
+        } else {
+            *type = _signed ? 0xd2 : 0xce;
+            len = 4;
+        }
+        mp_binary_set_int(len, true, buf, x);
     } else {
-        write1(s, 0xd2);
-        write4(s, x);
+        // todo: encode remaining 32 bit values as 0xd2/0xce ?
+        *type = _signed ? 0xd3 : 0xcf;
+        len = 8;
+        mp_obj_int_to_bytes_impl(obj, true, len, buf);
     }
+    write(s, buffer, len + 1);
 }
 
 STATIC void pack_bin(msgpack_stream_t *s, const uint8_t *data, size_t len) {
@@ -275,10 +291,13 @@ STATIC void pack_dict(msgpack_stream_t *s, size_t len) {
 }
 
 STATIC void pack(mp_obj_t obj, msgpack_stream_t *s, mp_obj_t default_handler) {
-    if (mp_obj_is_small_int(obj)) {
-        // int
-        int32_t x = MP_OBJ_SMALL_INT_VALUE(obj);
-        pack_int(s, x);
+    if (mp_obj_is_int(obj)) {
+        // big int
+        // encode signed or unsigned
+        bool _signed = mp_obj_int_sign(obj) < 0;
+        // raise if overflow
+        mp_obj_int_buffer_overflow_check(obj, 8, _signed);
+        pack_int(s, obj, _signed);
     } else if (mp_obj_is_str(obj)) {
         // string
         size_t len;