/*
 * Copyright (C) 2012 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"
#include "smatch_slist.h"

#define NOBUF -2

static int my_id;

static struct expression *get_returned_expr(struct expression *expr)
{
	struct statement *stmt;

	stmt = last_ptr_list((struct ptr_list *)big_statement_stack);
	if (!stmt || stmt->type != STMT_EXPRESSION || !stmt->expression)
		return NULL;
	if (stmt->expression->type != EXPR_ASSIGNMENT)
		return NULL;
	if (stmt->expression->right != expr)
		return NULL;
	return stmt->expression->left;
}

static struct expression *remove_dereference(struct expression *expr)
{
	if (!expr || expr->type != EXPR_PREOP || expr->op != '*')
		return expr;
	expr = expr->unop;
	if (!expr || expr->type != EXPR_PREOP || expr->op != '*')
		return expr;
	return expr->unop;
}

static int get_buf_number(struct expression *call, struct expression *size_arg)
{
	struct expression *arg;
	int idx = -1;

	size_arg = strip_expr(size_arg->cast_expression);
	size_arg = remove_dereference(size_arg);

	arg = get_returned_expr(call);
	if (arg && expr_equiv(arg, size_arg))
		return idx;

	FOR_EACH_PTR(call->args, arg) {
		idx++;
		if (expr_equiv(arg, size_arg))
			return idx;
	} END_FOR_EACH_PTR(arg);

	return NOBUF;
}

static void match_call(struct expression *call)
{
	struct expression *arg;
	char *name;
	int buf_nr;
	int i = -1;

	if (call->fn->type != EXPR_SYMBOL)
		return;

	name = expr_to_var(call->fn);
	FOR_EACH_PTR(call->args, arg) {
		i++;
		if (arg->type != EXPR_SIZEOF)
			continue;
		buf_nr = get_buf_number(call, arg);
		if (buf_nr == NOBUF)
			sm_msg("info: sizeof_param '%s' %d", name, i);
		else
			sm_msg("info: sizeof_param '%s' %d %d", name, i, buf_nr);
	} END_FOR_EACH_PTR(arg);
	free_string(name);
}

void check_passes_sizeof(int id)
{
	if (!option_info)
		return;

	my_id = id;
	add_hook(&match_call, FUNCTION_CALL_HOOK);
}