adjust string length after String#chomp!

This commit is contained in:
Akira Yumiyama
2012-11-08 15:10:18 +09:00
parent b777850c78
commit d090f3eedb
+2 -1
View File
@@ -939,6 +939,7 @@ mrb_str_chomp_bang(mrb_state *mrb, mrb_value str)
else {
return mrb_nil_value();
}
s->ptr[s->len] = '\0';
return str;
}
@@ -970,7 +971,7 @@ mrb_str_chomp_bang(mrb_state *mrb, mrb_value str)
(rslen <= 1 ||
memcmp(RSTRING_PTR(rs), pp, rslen) == 0)) {
s->len = len - rslen;
p[len] = '\0';
p[s->len] = '\0';
return str;
}
return mrb_nil_value();