1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

file.c: fix possible alignment bugs in r60386

* file.c (struct apply_filename): split out from struct apply_arg
* file.c (apply2files): use offsetof for flex array size calculation
* file.c (apply2files): avoid redundant marking with ALLOCV
  [ruby-core:83535]

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@60408 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
normal 2017-10-24 23:35:52 +00:00
parent 3fc0c18193
commit 95e1c87014

31
file.c
View file

@ -349,16 +349,18 @@ ignored_char_p(const char *p, const char *e, rb_encoding *enc)
#define apply2args(n) (rb_check_arity(argc, n, UNLIMITED_ARGUMENTS), argc-=n)
struct apply_filename {
const char *ptr;
VALUE path;
};
struct apply_arg {
int i;
int argc;
int errnum;
int (*func)(const char *, void *);
void *arg;
struct {
const char *ptr;
VALUE path;
} fn[1]; /* flexible array */
struct apply_filename fn[1]; /* flexible array */
};
static void *
@ -384,9 +386,8 @@ static VALUE
apply2files(int (*func)(const char *, void *), int argc, VALUE *argv, void *arg)
{
VALUE v;
VALUE tmp = Qfalse;
size_t size = sizeof(const char *) + sizeof(VALUE);
const long len = (long)(sizeof(struct apply_arg) + (size * argc) - size);
const size_t size = sizeof(struct apply_filename);
const long len = (long)(offsetof(struct apply_arg, fn) + (size * argc));
struct apply_arg *aa = ALLOCV(v, len);
aa->errnum = 0;
@ -394,24 +395,12 @@ apply2files(int (*func)(const char *, void *), int argc, VALUE *argv, void *arg)
aa->arg = arg;
aa->func = func;
/*
* aa is on-stack for small argc, we must ensure paths are marked
* for large argv if aa is on the heap.
*/
if (v) {
tmp = rb_ary_tmp_new(argc);
}
for (aa->i = 0; aa->i < argc; aa->i++) {
VALUE path = rb_get_path(argv[aa->i]);
path = rb_str_encode_ospath(path);
aa->fn[aa->i].ptr = RSTRING_PTR(path);
aa->fn[aa->i].path = path;
if (tmp != Qfalse) {
rb_ary_push(tmp, path);
}
}
rb_thread_call_without_gvl(no_gvl_apply2files, aa, RUBY_UBF_IO, 0);
@ -426,10 +415,6 @@ apply2files(int (*func)(const char *, void *), int argc, VALUE *argv, void *arg)
if (v) {
ALLOCV_END(v);
}
if (tmp != Qfalse) {
rb_ary_clear(tmp);
rb_gc_force_recycle(tmp);
}
return LONG2FIX(argc);
}