commit 271d18eb08ec488ee45f8e6cd852e8236074f082
author Damien George <damien.p.george@gmail.com> Sat Apr 25 23:16:39 2015 +0100
committer Damien George <damien.p.george@gmail.com> Sat Apr 25 23:16:39 2015 +0100
tree f5ba5d1b28cb48e72334fb079a5d6ef148083970
parent 7c8b4c1a8b8278d864649c127857e34a6bd29504

py: Support conversion of bignum to bytes.

This gets int.to_bytes working for bignum, and also struct.pack with 'q'
and 'Q' args on 32-bit machines.

Addresses issue #1155.

py/binary.c

diff --git a/py/binary.c b/py/binary.c
index 927a426..8b5c05a 100644
--- a/py/binary.c
+++ b/py/binary.c
@@ -32,6 +32,7 @@
 
 #include "py/binary.h"
 #include "py/smallint.h"
+#include "py/objint.h"
 
 // Helpers to work with binary-encoded data
 
@@ -282,10 +283,13 @@
         }
 #endif
         default:
-            // we handle large ints here by calling the truncated accessor
+            #if MICROPY_LONGINT_IMPL != MICROPY_LONGINT_IMPL_NONE
             if (MP_OBJ_IS_TYPE(val_in, &mp_type_int)) {
-                val = mp_obj_int_get_truncated(val_in);
-            } else {
+                mp_obj_int_to_bytes_impl(val_in, struct_type == '>', size, p);
+                return;
+            } else
+            #endif
+            {
                 val = mp_obj_get_int(val_in);
             }
     }