StackRduino
diff --git a/‎py/mpz.c‎
Lines changed: 4 additions & 1 deletion b/‎py/mpz.c‎
Lines changed: 4 additions & 1 deletion
diff --git a/‎py/runtime.c‎
Lines changed: 18 additions & 2 deletions b/‎py/runtime.c‎
Lines changed: 18 additions & 2 deletions
@@ -993,8 +993,11 @@ void mpz_pow_inpl(mpz_t *dest, const mpz_t *lhs, const mpz_t *rhs) {
         if (mpz_is_odd(n)) {
             mpz_mul_inpl(dest, dest, x);
         }
-        mpz_mul_inpl(x, x, x);
         n->len = mpn_shr(n->dig, n->dig, n->len, 1);
+        if (n->len == 0) {
+            break;
+        }
+        mpz_mul_inpl(x, x, x);
     }
 
     mpz_free(x);
 
@@ -367,18 +367,34 @@ mp_obj_t mp_binary_op(int op, mp_obj_t lhs, mp_obj_t rhs) {
                         nlr_jump(mp_obj_new_exception_msg(&mp_type_ValueError, "negative power with no float support"));
                         #endif
                     } else {
-                        // TODO check for overflow
                         machine_int_t ans = 1;
                         while (rhs_val > 0) {
                             if (rhs_val & 1) {
+                                machine_int_t old = ans;
                                 ans *= lhs_val;
+                                if (ans < old) {
+                                    goto power_overflow;
+                                }
+                            }
+                            if (rhs_val == 1) {
+                                break;
                             }
-                            lhs_val *= lhs_val;
                             rhs_val /= 2;
+                            machine_int_t old = lhs_val;
+                            lhs_val *= lhs_val;
+                            if (lhs_val < old) {
+                                goto power_overflow;
+                            }
                         }
                         lhs_val = ans;
                     }
                     break;
+
+                power_overflow:
+                    // use higher precision
+                    lhs = mp_obj_new_int_from_ll(MP_OBJ_SMALL_INT_VALUE(lhs));
+                    goto generic_binary_op;
+
                 case MP_BINARY_OP_LESS: return MP_BOOL(lhs_val < rhs_val); break;
                 case MP_BINARY_OP_MORE: return MP_BOOL(lhs_val > rhs_val); break;
                 case MP_BINARY_OP_LESS_EQUAL: return MP_BOOL(lhs_val <= rhs_val); break;
Original file line number	Diff line number	Diff line change
`@@ -993,8 +993,11 @@ void mpz_pow_inpl(mpz_t dest, const mpz_t lhs, const mpz_t *rhs) {`
`993`	`993`	`if (mpz_is_odd(n)) {`
`994`	`994`	`mpz_mul_inpl(dest, dest, x);`
`995`	`995`	`}`
`996`		`- mpz_mul_inpl(x, x, x);`
`997`	`996`	`n->len = mpn_shr(n->dig, n->dig, n->len, 1);`
	`997`	`+ if (n->len == 0) {`
	`998`	`+ break;`
	`999`	`+ }`
	`1000`	`+ mpz_mul_inpl(x, x, x);`
`998`	`1001`	`}`
`999`	`1002`
`1000`	`1003`	`mpz_free(x);`