/*
 * Copyright (C) 2017 Oracle.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see http://www.gnu.org/copyleft/gpl.txt
 */

#include "smatch.h"
#include "smatch_extra.h"

static int find_param_eq(struct expression *expr, int size)
{
        struct expression *arg;
        sval_t val;
        int i;

        i = -1;
        FOR_EACH_PTR(expr->args, arg) {
                i++;
                if (!get_implied_value(arg, &val))
                        continue;
                if (val.value == size)
                        return i;
        } END_FOR_EACH_PTR(arg);

        return -1;
}

static void match_call(struct expression *expr)
{
        struct expression *arg;
        struct symbol *type, *arg_type;
        int size, bytes;
        int i, nr;
        char buf[16];
        char elem_count[8];
        char byte_count[8];

        snprintf(elem_count, sizeof(elem_count), "%d", ELEM_COUNT);
        snprintf(byte_count, sizeof(byte_count), "%d", BYTE_COUNT);

        i = -1;
        FOR_EACH_PTR(expr->args, arg) {
                i++;
                type = get_type(arg);
                if (!type || (type->type != SYM_PTR && type->type != SYM_ARRAY))
                        continue;
                arg_type = get_arg_type(expr->fn, i);
                if (arg_type != type)
                        continue;

                size = get_array_size(arg);
                if (size > 0) {
                        nr = find_param_eq(expr, size);
                        if (nr >= 0) {
                                snprintf(buf, sizeof(buf), "==$%d", nr);
                                sql_insert_caller_info(expr, ELEM_COUNT, i, buf, elem_count);
                                continue;
                        }
                }
                bytes = get_array_size_bytes(arg);
                if (bytes > 0) {
                        nr = find_param_eq(expr, bytes);
                        if (nr >= 0) {
                                snprintf(buf, sizeof(buf), "==$%d", nr);
                                sql_insert_caller_info(expr, BYTE_COUNT, i, buf, byte_count);
                                continue;
                        }
                }
        } END_FOR_EACH_PTR(arg);
}

void register_passes_array_size(int id)
{
        add_hook(&match_call, FUNCTION_CALL_HOOK);
}