diff options
| author | Yukihiro "Matz" Matsumoto <[email protected]> | 2014-04-09 21:38:29 +0900 |
|---|---|---|
| committer | Yukihiro "Matz" Matsumoto <[email protected]> | 2014-04-09 21:38:29 +0900 |
| commit | b45e99b0d5e6b1efb6aa6f38e9ab24ed93590fe2 (patch) | |
| tree | 55400ea5b679102add823d6ac3fccde8dfb1a94e /src/variable.c | |
| parent | ef9daf7ef37a4ad9469a260ea52aa014bdf6549e (diff) | |
| download | mruby-b45e99b0d5e6b1efb6aa6f38e9ab24ed93590fe2.tar.gz mruby-b45e99b0d5e6b1efb6aa6f38e9ab24ed93590fe2.zip | |
call kh_get before kh_put to avoid potential key lost; reported by @mirichi
Diffstat (limited to 'src/variable.c')
| -rw-r--r-- | src/variable.c | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/variable.c b/src/variable.c index 3dcdbf775..6503c8fb7 100644 --- a/src/variable.c +++ b/src/variable.c @@ -308,7 +308,10 @@ iv_put(mrb_state *mrb, iv_tbl *t, mrb_sym sym, mrb_value val) khash_t(iv) *h = &t->h; khiter_t k; - k = kh_put(iv, mrb, h, sym); + k = kh_get(mt, mrb, h, sym); + if (k == kh_end(h)) { + k = kh_put(mt, mrb, h, sym); + } kh_value(h, k) = val; } |