/*
 * Copyright (C) 2013 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_slist.h"
#include "smatch_extra.h"

static int my_id;

static char *get_source_parameter(struct expression *expr)
{
	struct expression *tmp;
	const char *param_name;
	struct symbol *sym;
	char *name;
	int param;
	char *ret = NULL;
	char buf[32];
	int cnt = 0;
	bool modified = false;

	tmp = expr;
	while ((tmp = get_assigned_expr(tmp))) {
		expr = tmp;
		if (cnt++ > 3)
			break;
	}

	expr = strip_expr(expr);
	if (expr->type != EXPR_SYMBOL)
		return NULL;

	name = expr_to_var_sym(expr, &sym);
	if (!name || !sym)
		goto free;
	param = get_param_num_from_sym(sym);
	if (param < 0)
		goto free;
	param_name = get_param_name_var_sym(name, sym);
	if (!param_name)
		goto free;
	if (param_was_set_var_sym(name, sym))
		modified = true;

	snprintf(buf, sizeof(buf), "$%d%s%s", param, param_name + 1,
		 modified ? " [m]" : "");
	ret = alloc_string(buf);

free:
	free_string(name);
	return ret;
}

static char *get_source_assignment(struct expression *expr)
{
	struct expression *right;
	char *name;
	char buf[64];
	char *ret;

	right = get_assigned_expr(expr);
	right = strip_expr(right);
	if (!right)
		return NULL;
	if (right->type != EXPR_CALL || right->fn->type != EXPR_SYMBOL)
		return NULL;
	if (is_fake_call(right))
		return NULL;
	name = expr_to_str(right->fn);
	if (!name)
		return NULL;
	snprintf(buf, sizeof(buf), "r %s", name);
	ret = alloc_string(buf);
	free_string(name);
	return ret;
}

static char *get_source_str(struct expression *arg)
{
	char *source;

	source = get_source_parameter(arg);
	if (source)
		return source;
	return get_source_assignment(arg);
}

static void match_caller_info(struct expression *expr)
{
	struct expression *arg;
	char *source;
	int i;

	i = -1;
	FOR_EACH_PTR(expr->args, arg) {
		i++;
		source = get_source_str(arg);
		if (!source)
			continue;
		sql_insert_caller_info(expr, DATA_SOURCE, i, "$", source);
		free_string(source);
	} END_FOR_EACH_PTR(arg);
}

void register_data_source(int id)
{
//	if (!option_info)
//		return;
	my_id = id;
	add_hook(&match_caller_info, FUNCTION_CALL_HOOK);
}