diff options
Diffstat (limited to 'src/dump.c')
| -rw-r--r-- | src/dump.c | 127 |
1 files changed, 120 insertions, 7 deletions
diff --git a/src/dump.c b/src/dump.c index fdb20f68a..d3f6ee62b 100644 --- a/src/dump.c +++ b/src/dump.c @@ -334,6 +334,97 @@ mrb_write_section_irep(mrb_state *mrb, size_t start_index, uint8_t *bin) } static int +mrb_write_section_lineno_header(mrb_state *mrb, uint32_t section_size, uint16_t nirep, uint16_t sirep, uint8_t *bin) +{ + struct rite_section_lineno_header *header = (struct rite_section_lineno_header*)bin; + + // TODO + memcpy(header->section_identify, RITE_SECTION_LIENO_IDENTIFIER, sizeof(header->section_identify)); + uint32_to_bin(section_size, header->section_size); + uint16_to_bin(nirep, header->nirep); + uint16_to_bin(sirep, header->sirep); + + return MRB_DUMP_OK; +} + +static size_t +get_debug_record_size(mrb_state *mrb, mrb_irep *irep) +{ + size_t size = 0; + + size += sizeof(uint32_t); // record size + size += sizeof(uint16_t); // filename size + if(irep->filename) { + size += strlen(irep->filename); // filename + } + size += sizeof(uint32_t); // niseq + if(irep->lines) { + size += sizeof(uint16_t) * irep->ilen; // lineno + } + + return size; +} + +static int +write_lineno_record(mrb_state *mrb, mrb_irep *irep, uint8_t* bin) +{ + uint8_t *cur = bin; + size_t filename_len = 0; + int iseq_no; + + cur += sizeof(uint32_t); /* record size */ + + if(irep->filename) { + filename_len = strlen(irep->filename); + } + cur += uint16_to_bin(filename_len, cur); /* filename size */ + + if(filename_len) { + memcpy(cur, irep->filename, filename_len); + cur += filename_len; /* filename */ + } + + if(irep->lines) { + cur += uint32_to_bin(irep->ilen, cur); /* niseq */ + for (iseq_no = 0; iseq_no < irep->ilen; iseq_no++) { + cur += uint16_to_bin(irep->lines[iseq_no], cur); /* opcode */ + } + } + else { + cur += uint32_to_bin(0, cur); /* niseq */ + } + + uint32_to_bin(cur - bin, bin); /* record size */ + + return (cur - bin); +} + +static int +mrb_write_section_lineno(mrb_state *mrb, int start_index, uint8_t *bin) +{ + int irep_no; + uint32_t section_size = 0, rlen = 0; /* size of irep record */ + uint8_t *cur = bin; + + if (mrb == NULL || start_index < 0 || start_index >= mrb->irep_len || bin == NULL) { + return MRB_DUMP_INVALID_ARGUMENT; + } + + cur += sizeof(struct rite_section_lineno_header); + section_size += sizeof(struct rite_section_lineno_header); + + for (irep_no = start_index; irep_no < mrb->irep_len; irep_no++) { + rlen = write_lineno_record(mrb, mrb->irep[irep_no], cur); + cur += rlen; + section_size += rlen; + } + + mrb_write_section_lineno_header(mrb, section_size, mrb->irep_len - start_index, start_index, bin); + + return MRB_DUMP_OK; +} + +static int write_rite_binary_header(mrb_state *mrb, size_t binary_size, uint8_t* bin) { struct rite_binary_header *header = (struct rite_binary_header*)bin; @@ -354,10 +445,12 @@ write_rite_binary_header(mrb_state *mrb, size_t binary_size, uint8_t* bin) } static int -mrb_dump_irep(mrb_state *mrb, size_t start_index, uint8_t **bin, size_t *bin_size) +mrb_dump_irep(mrb_state *mrb, size_t start_index, int debug_info, uint8_t **bin, size_t *bin_size) { int result = MRB_DUMP_GENERAL_FAILURE; + size_t section_size = 0; size_t section_irep_size; + size_t section_lineno_size = 0; size_t irep_no; uint8_t *cur = NULL; @@ -370,8 +463,18 @@ mrb_dump_irep(mrb_state *mrb, size_t start_index, uint8_t **bin, size_t *bin_siz for (irep_no = start_index; irep_no < mrb->irep_len; irep_no++) { section_irep_size += get_irep_record_size(mrb, mrb->irep[irep_no]); } + section_size += section_irep_size; + + /* DEBUG section size */ + if(debug_info) { + section_lineno_size += sizeof(struct rite_section_lineno_header); + for (irep_no = start_index; irep_no < mrb->irep_len; irep_no++) { + section_lineno_size += get_debug_record_size(mrb, mrb->irep[irep_no]); + } + section_size += section_lineno_size; + } - *bin_size += sizeof(struct rite_binary_header) + section_irep_size + sizeof(struct rite_binary_footer); + *bin_size += sizeof(struct rite_binary_header) + section_size + sizeof(struct rite_binary_footer); cur = *bin = (uint8_t *)mrb_malloc(mrb, *bin_size); if(cur == NULL) { goto error_exit; @@ -383,8 +486,18 @@ mrb_dump_irep(mrb_state *mrb, size_t start_index, uint8_t **bin, size_t *bin_siz if (result != MRB_DUMP_OK) { goto error_exit; } - + cur += section_irep_size; + + /* write DEBUG section */ + if(debug_info) { + result = mrb_write_section_lineno(mrb, start_index, cur); + if (result != MRB_DUMP_OK) { + goto error_exit; + } + cur += section_lineno_size; + } + mrb_write_eof(mrb, cur); result = write_rite_binary_header(mrb, *bin_size, *bin); @@ -401,7 +514,7 @@ error_exit: #ifdef ENABLE_STDIO int -mrb_dump_irep_binary(mrb_state *mrb, size_t start_index, FILE* fp) +mrb_dump_irep_binary(mrb_state *mrb, size_t start_index, int debug_info, FILE* fp) { uint8_t *bin = NULL; size_t bin_size = 0; @@ -411,7 +524,7 @@ mrb_dump_irep_binary(mrb_state *mrb, size_t start_index, FILE* fp) return MRB_DUMP_INVALID_ARGUMENT; } - result = mrb_dump_irep(mrb, start_index, &bin, &bin_size); + result = mrb_dump_irep(mrb, start_index, debug_info, &bin, &bin_size); if (result == MRB_DUMP_OK) { fwrite(bin, bin_size, 1, fp); } @@ -421,7 +534,7 @@ mrb_dump_irep_binary(mrb_state *mrb, size_t start_index, FILE* fp) } int -mrb_dump_irep_cfunc(mrb_state *mrb, size_t start_index, FILE *fp, const char *initname) +mrb_dump_irep_cfunc(mrb_state *mrb, size_t start_index, int debug_info, FILE *fp, const char *initname) { uint8_t *bin = NULL; size_t bin_size = 0, bin_idx = 0; @@ -431,7 +544,7 @@ mrb_dump_irep_cfunc(mrb_state *mrb, size_t start_index, FILE *fp, const char *in return MRB_DUMP_INVALID_ARGUMENT; } - result = mrb_dump_irep(mrb, start_index, &bin, &bin_size); + result = mrb_dump_irep(mrb, start_index, debug_info, &bin, &bin_size); if (result == MRB_DUMP_OK) { fprintf(fp, "const uint8_t %s[] = {", initname); while (bin_idx < bin_size) { |