From 05da7503e1606bb7300cea15fcdc3bd36cf79c67 Mon Sep 17 00:00:00 2001 From: guy Decoux Date: Sat, 28 Feb 2009 20:04:27 +0100 Subject: mmap-0.1.6 --- Changes | 8 +++++++- mmap.c | 40 ++++++++++++++++++++++++++++++---------- 2 files changed, 37 insertions(+), 11 deletions(-) diff --git a/Changes b/Changes index f6c66fa..28e12e6 100644 --- a/Changes +++ b/Changes @@ -23,4 +23,10 @@ * added str[/a(a)/, 1] for 1.7.1 * corrected mm_aset (mm_update(str, ...)) * corrected ==, ===, eql? -* corrected mm_sub_bang, mm_gsub_bang (to_str) \ No newline at end of file +* corrected mm_sub_bang, mm_gsub_bang (to_str) + +-- 0.1.6 + +* adapted for 1.7.2 (mm_str) +* corrected real for mm_sub_bang +* protected against old class diff --git a/mmap.c b/mmap.c index 5d33fba..6a68bcb 100644 --- a/mmap.c +++ b/mmap.c @@ -120,6 +120,11 @@ mm_str(obj, modify) VALUE ret; GetMmap(obj, t_mm, modify & ~MM_ORIGIN); + if (modify & MM_MODIFY) { + if (t_mm->frozen & MM_FROZEN) rb_error_frozen("mmap"); + if (!OBJ_TAINTED(ret) && rb_safe_level() >= 4) + rb_raise(rb_eSecurityError, "Insecure: can't modify mmap"); + } #if RUBY_VERSION_CODE >= 171 ret = rb_obj_alloc(rb_cString); if (rb_obj_tainted(obj)) { @@ -132,20 +137,21 @@ mm_str(obj, modify) else { ret = rb_str_new2(""); } -#endif - if (modify & MM_MODIFY) { - if (t_mm->frozen & MM_FROZEN) rb_error_frozen("mmap"); - if (!OBJ_TAINTED(ret) && rb_safe_level() >= 4) - rb_raise(rb_eSecurityError, "Insecure: can't modify mmap"); - } - if (t_mm->frozen & MM_FROZEN) ret = rb_obj_freeze(ret); -#if RUBY_VERSION_CODE < 171 free(RSTRING(ret)->ptr); #endif RSTRING(ret)->ptr = t_mm->addr; RSTRING(ret)->len = t_mm->real; - if (modify & MM_ORIGIN) + if (modify & MM_ORIGIN) { +#if RUBY_VERSION_CODE >= 172 + RSTRING(ret)->aux.shared = ret; + FL_SET(ret, ELTS_SHARED); +#else RSTRING(ret)->orig = ret; +#endif + } + if (t_mm->frozen & MM_FROZEN) { + ret = rb_obj_freeze(ret); + } return ret; } @@ -302,9 +308,18 @@ mm_s_new(argc, argv, obj) else { OBJ_TAINT(res); } + rb_obj_call_init(res, argc, argv); return res; } +static VALUE +mm_init(argc, argv, obj) + int argc; + VALUE *argv, obj; +{ + return obj; +} + static void mm_expandf(t_mm, len) mm_mmap *t_mm; @@ -630,7 +645,6 @@ mm_sub_bang(argc, argv, obj) plen = END(0); if (RSTRING(repl)->len > plen) { mm_realloc(t_mm, RSTRING(str)->len + RSTRING(repl)->len - plen); - t_mm->real = t_mm->len; RSTRING(str)->ptr = t_mm->addr; } if (RSTRING(repl)->len != plen) { @@ -1497,6 +1511,9 @@ mm_munlock(obj) void Init_mmap() { + if (rb_const_defined_at(rb_cObject, rb_intern("Mmap"))) { + rb_raise(rb_eNameError, "class already defined"); + } mm_cMap = rb_define_class("Mmap", rb_cObject); rb_define_const(mm_cMap, "MS_SYNC", INT2FIX(MS_SYNC)); rb_define_const(mm_cMap, "MS_ASYNC", INT2FIX(MS_ASYNC)); @@ -1548,6 +1565,8 @@ Init_mmap() rb_define_singleton_method(mm_cMap, "munlockall", mm_munlockall, 0); rb_define_singleton_method(mm_cMap, "unlockall", mm_munlockall, 0); + rb_define_method(mm_cMap, "initialize", mm_init, -1); + rb_define_method(mm_cMap, "unmap", mm_unmap, 0); rb_define_method(mm_cMap, "munmap", mm_unmap, 0); rb_define_method(mm_cMap, "msync", mm_msync, -1); @@ -1656,6 +1675,7 @@ Init_mmap() rb_define_method(mm_cMap, "delete", mm_undefined, -1); rb_define_method(mm_cMap, "squeeze", mm_undefined, -1); rb_define_method(mm_cMap, "count", mm_count, -1); + rb_define_method(mm_cMap, "tr!", mm_tr_bang, 2); rb_define_method(mm_cMap, "tr_s!", mm_tr_s_bang, 2); rb_define_method(mm_cMap, "delete!", mm_delete_bang, -1); -- cgit v1.1-2-g2b99