fix null in .mpy

dpgeorge · dpgeorge · commit 086a1fc2c974 · 2022-02-17T12:35:13.000+11:00
diff --git a/py/persistentcode.c b/py/persistentcode.c
@@ -228,8 +228,8 @@ STATIC mp_obj_t load_obj(mp_reader_t *reader) {
         vstr_t vstr;
         vstr_init_len(&vstr, len);
         read_bytes(reader, (byte *)vstr.buf, len);
-        read_byte(reader); // skip null terminator
         if (obj_type == 's' || obj_type == 'b') {
+            read_byte(reader); // skip null terminator
             return mp_obj_new_str_from_vstr(obj_type == 's' ? &mp_type_str : &mp_type_bytes, &vstr);
         } else if (obj_type == 'i') {
             return mp_parse_num_integer(vstr.buf, vstr.len, 10, NULL);
@@ -532,7 +532,7 @@ STATIC void save_obj(mp_print_t *print, mp_obj_t o) {
         const char *str = mp_obj_str_get_data(o, &len);
         mp_print_bytes(print, &obj_type, 1);
         mp_print_uint(print, len);
-        mp_print_bytes(print, (const byte *)str, len + 1);
+        mp_print_bytes(print, (const byte *)str, len + 1); // +1 to store null terminator
     } else if (MP_OBJ_TO_PTR(o) == &mp_const_ellipsis_obj) {
         byte obj_type = 'e';
         mp_print_bytes(print, &obj_type, 1);
