/* Copyright (c) 2020 Contrast Security, Inc.  See
 * https://www.contrastsecurity.com/enduser-terms-0317a for more details. */

#include "cs__assess_string_interpolation26.h"
#include "../cs__common/cs__common.h"
#include <funchook.h>
#include <ruby.h>

static VALUE rb_str_concat_literals_hook(size_t num, VALUE *strary) {
    VALUE result = rb_str_concat_literals_original(num, strary);
    VALUE rb_params = rb_ary_new_from_values((int)num, strary);
    rb_funcall(string_class, track_interpolation, 2, rb_params, result);
    return result;
}

static int install_hooks() {
    funchook_t *funchook = funchook_create();

    rb_str_concat_literals_original = rb_str_concat_literals;
    funchook_prepare(funchook, (void **)&rb_str_concat_literals_original,
                     rb_str_concat_literals_hook);

    funchook_install(funchook, 0);
    return 0;
}

void Init_cs__assess_string_interpolation26(void) {
    string_class = rb_define_class_under(core_assess, "StringPropagator", rb_cObject);
    track_interpolation = rb_intern("track_interpolation");
    install_hooks();
}