diff options
| author | kyab <[email protected]> | 2013-06-22 03:55:49 +0900 |
|---|---|---|
| committer | kyab <[email protected]> | 2013-06-22 04:02:55 +0900 |
| commit | 31fbfb51dbcb18f4f8893aa03195ba050a722a29 (patch) | |
| tree | 8b52787064bf640cce0ad207e0d42c574f415a33 /src/load.c | |
| parent | 3b751d6ccdccc27099b5b375a5cef4c93d1ded6a (diff) | |
| download | mruby-31fbfb51dbcb18f4f8893aa03195ba050a722a29.tar.gz mruby-31fbfb51dbcb18f4f8893aa03195ba050a722a29.zip | |
Add fallback for allocate CRC buffer
Diffstat (limited to 'src/load.c')
| -rw-r--r-- | src/load.c | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/load.c b/src/load.c index 62010e425..e832de628 100644 --- a/src/load.c +++ b/src/load.c @@ -539,7 +539,9 @@ mrb_read_irep_file(mrb_state *mrb, FILE* fp) size_t sirep; struct rite_section_header section_header; long fpos; - const size_t block_size = 1 << 14; + size_t block_size = 1 << 14; + const uint8_t block_fallback_count = 4; + int i; const size_t buf_size = sizeof(struct rite_binary_header); if ((mrb == NULL) || (fp == NULL)) { @@ -564,7 +566,10 @@ mrb_read_irep_file(mrb_state *mrb, FILE* fp) /* verify CRC */ fpos = ftell(fp); /* You don't need use SIZE_ERROR as block_size is enough small. */ - buf = mrb_malloc(mrb, block_size); + for (i = 0; i < block_fallback_count; i++,block_size >>= 1){ + buf = mrb_malloc(mrb, block_size); + if (buf) break; + } if (!buf) { return MRB_DUMP_GENERAL_FAILURE; } |