diff options
| author | Tomoyuki Sahara <[email protected]> | 2012-09-10 18:44:30 +0900 |
|---|---|---|
| committer | Tomoyuki Sahara <[email protected]> | 2012-09-10 18:44:30 +0900 |
| commit | ebf88eec6b3e486bc127c43e613968f92626426f (patch) | |
| tree | 7b748ad3d975f82de14eed4e9759ad562812e450 | |
| parent | d416a5c3fe3682983cc365585aa6ac2b09763e4b (diff) | |
| download | mruby-ebf88eec6b3e486bc127c43e613968f92626426f.tar.gz mruby-ebf88eec6b3e486bc127c43e613968f92626426f.zip | |
a string may have NUL characters.
| -rw-r--r-- | src/dump.c | 24 |
1 files changed, 10 insertions, 14 deletions
diff --git a/src/dump.c b/src/dump.c index 190f16027..516374cd9 100644 --- a/src/dump.c +++ b/src/dump.c @@ -332,7 +332,7 @@ write_pool_block(mrb_state *mrb, mrb_irep *irep, char *buf, int type) char *buf_top = buf; char *char_buf; uint16_t buf_size =0; - int len; + uint16_t len =0; buf_size = MRB_DUMP_DEFAULT_STR_LEN; if ((char_buf = (char *)mrb_malloc(mrb, buf_size)) == 0) @@ -341,25 +341,23 @@ write_pool_block(mrb_state *mrb, mrb_irep *irep, char *buf, int type) buf += uint32_dump((uint32_t)irep->plen, buf, type); /* number of pool */ for (pool_no = 0; pool_no < irep->plen; pool_no++) { - uint16_t nlen =0; - buf += uint8_dump(mrb_type(irep->pool[pool_no]), buf, type); /* data type */ memset(char_buf, 0, buf_size); switch (mrb_type(irep->pool[pool_no])) { case MRB_TT_FIXNUM: - sprintf(char_buf, "%d", mrb_fixnum(irep->pool[pool_no])); + len = sprintf(char_buf, "%d", mrb_fixnum(irep->pool[pool_no])); break; case MRB_TT_FLOAT: - sprintf(char_buf, "%.16e", mrb_float(irep->pool[pool_no])); + len = sprintf(char_buf, "%.16e", mrb_float(irep->pool[pool_no])); break; case MRB_TT_STRING: str = mrb_string_value( mrb, &irep->pool[pool_no]); - nlen = str_dump_len(RSTRING_PTR(str), RSTRING_LEN(str), type); - if ( nlen > buf_size - 1) { - buf_size = nlen + 1; + len = str_dump_len(RSTRING_PTR(str), RSTRING_LEN(str), type); + if ( len > buf_size - 1) { + buf_size = len + 1; if ((char_buf = (char *)mrb_realloc(mrb, char_buf, buf_size)) == 0) goto error_exit; memset(char_buf, 0, buf_size); @@ -370,9 +368,9 @@ write_pool_block(mrb_state *mrb, mrb_irep *irep, char *buf, int type) #ifdef ENABLE_REGEXP case MRB_TT_REGEX: str = mrb_reg_to_s(mrb, irep->pool[pool_no]); - nlen = str_dump_len(RSTRING_PTR(str), RSTRING_LEN(str), type); - if ( nlen > buf_size - 1) { - buf_size = nlen + 1; + len = str_dump_len(RSTRING_PTR(str), RSTRING_LEN(str), type); + if ( len > buf_size - 1) { + buf_size = len + 1; if ((char_buf = mrb_realloc(mrb, char_buf, buf_size)) == 0) goto error_exit; memset(char_buf, 0, buf_size); @@ -386,9 +384,7 @@ write_pool_block(mrb_state *mrb, mrb_irep *irep, char *buf, int type) continue; } - len = strlen(char_buf); - - buf += uint16_dump((uint16_t)len, buf, type); /* data length */ + buf += uint16_dump(len, buf, type); /* data length */ memcpy(buf, char_buf, len); buf += len; |