admin/mruby-mruby
1
0
Fork 0
mirror of https://github.com/mruby/mruby synced 2026-06-08 16:11:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

mrb_str_pool use embed-string

Browse Source
This commit is contained in:
ksss
2014-03-23 16:03:32 +09:00
parent 81e0c30cc5
commit b0f10b5c3d
1 changed files with 29 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/state.c
+29 -12
View File
@@ -170,23 +170,40 @@ mrb_str_pool(mrb_state *mrb, mrb_value str)
ns->tt = MRB_TT_STRING;
ns->c = mrb->string_class;
if (s->flags & MRB_STR_EMBED)
len = (mrb_int)((s->flags & MRB_STR_EMBED_LEN_MASK) >> MRB_STR_EMBED_LEN_SHIFT);
else
len = s->as.heap.len;
ns->as.heap.len = len;
if (s->flags & MRB_STR_NOFREE) {
ns->as.heap.ptr = s->as.heap.ptr;
ns->flags = MRB_STR_NOFREE;
ns->as.heap.ptr = s->as.heap.ptr;
ns->as.heap.len = s->as.heap.len;
ns->as.heap.aux.capa = 0;
}
else {
ns->flags = 0;
ns->as.heap.ptr = (char *)mrb_malloc(mrb, (size_t)len+1);
ptr = (s->flags & MRB_STR_EMBED) ? s->as.ary : s->as.heap.ptr;
if (ptr) {
memcpy(ns->as.heap.ptr, ptr, len);
if (s->flags & MRB_STR_EMBED) {
ptr = s->as.ary;
len = (mrb_int)((s->flags & MRB_STR_EMBED_LEN_MASK) >> MRB_STR_EMBED_LEN_SHIFT);
}
else {
ptr = s->as.heap.ptr;
len = s->as.heap.len;
}
if (len < RSTRING_EMBED_LEN_MAX) {
ns->flags |= MRB_STR_EMBED;
ns->flags &= ~MRB_STR_EMBED_LEN_MASK;
ns->flags |= (size_t)len << MRB_STR_EMBED_LEN_SHIFT;
if (ptr) {
memcpy(ns->as.ary, ptr, len);
}
ns->as.ary[len] = '\0';
}
else {
ns->flags = 0;
ns->as.heap.ptr = (char *)mrb_malloc(mrb, (size_t)len+1);
ns->as.heap.len = len;
if (ptr) {
memcpy(ns->as.heap.ptr, ptr, len);
}
ns->as.heap.ptr[len] = '\0';
}
ns->as.heap.ptr[len] = '\0';
}
return mrb_obj_value(ns);
}