/*
 * Copyright (C) 2010 Dan Carpenter.
 *
 * 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 <stdlib.h>
#include <errno.h>
#include "parse.h"
#include "smatch.h"
#include "smatch_slist.h"
#include "smatch_extra.h"
#include "smatch_function_hashtable.h"

#define UNKNOWN_SIZE -1

static int my_size_id;

static DEFINE_HASHTABLE_INSERT(insert_func, char, int);
static DEFINE_HASHTABLE_SEARCH(search_func, char, int);
static struct hashtable *allocation_funcs;

static char *get_fn_name(struct expression *expr)
{
	if (expr->type != EXPR_CALL)
		return NULL;
	if (expr->fn->type != EXPR_SYMBOL)
		return NULL;
	return expr_to_var(expr->fn);
}

static int is_allocation_function(struct expression *expr)
{
	char *func;
	int ret = 0;

	func = get_fn_name(expr);
	if (!func)
		return 0;
	if (search_func(allocation_funcs, func))
		ret = 1;
	free_string(func);
	return ret;
}

static void add_allocation_function(const char *func, void *call_back, int param)
{
	insert_func(allocation_funcs, (char *)func, (int *)1);
	add_function_assign_hook(func, call_back, INT_PTR(param));
}

static int estate_to_size(struct smatch_state *state)
{
	sval_t sval;

	if (!state || !estate_rl(state))
		return 0;
	sval = estate_max(state);
	return sval.value;
}

static struct smatch_state *size_to_estate(int size)
{
	sval_t sval;

	sval.type = &int_ctype;
	sval.value = size;

	return alloc_estate_sval(sval);
}

static struct range_list *size_to_rl(int size)
{
	sval_t sval;

	sval.type = &int_ctype;
	sval.value = size;

	return alloc_rl(sval, sval);
}

static struct smatch_state *unmatched_size_state(struct sm_state *sm)
{
	return size_to_estate(UNKNOWN_SIZE);
}

static void set_size_undefined(struct sm_state *sm, struct expression *mod_expr)
{
	set_state(sm->owner, sm->name, sm->sym, size_to_estate(UNKNOWN_SIZE));
}

static struct smatch_state *merge_size_func(struct smatch_state *s1, struct smatch_state *s2)
{
	return merge_estates(s1, s2);
}

void set_param_buf_size(const char *name, struct symbol *sym, char *key, char *value)
{
	struct range_list *rl = NULL;
	struct smatch_state *state;
	char fullname[256];

	if (strncmp(key, "$", 1) != 0)
		return;

	snprintf(fullname, 256, "%s%s", name, key + 1);

	str_to_rl(&int_ctype, value, &rl);
	if (!rl || is_whole_rl(rl))
		return;
	state = alloc_estate_rl(rl);
	set_state(my_size_id, fullname, sym, state);
}

static int bytes_per_element(struct expression *expr)
{
	struct symbol *type;

	if (!expr)
		return 0;
	if (expr->type == EXPR_STRING)
		return 1;
	if (expr->type == EXPR_PREOP && expr->op == '&') {
		type = get_type(expr->unop);
		if (type && type->type == SYM_ARRAY)
			expr = expr->unop;
	}
	type = get_type(expr);
	if (!type)
		return 0;

	if (type->type != SYM_PTR && type->type != SYM_ARRAY)
		return 0;

	type = get_base_type(type);
	return type_bytes(type);
}

static int bytes_to_elements(struct expression *expr, int bytes)
{
	int bpe;

	bpe = bytes_per_element(expr);
	if (bpe == 0)
		return 0;
	return bytes / bpe;
}

static int elements_to_bytes(struct expression *expr, int elements)
{
	int bpe;

	bpe = bytes_per_element(expr);
	return elements * bpe;
}

static int get_initializer_size(struct expression *expr)
{
	switch (expr->type) {
	case EXPR_STRING:
		return expr->string->length;
	case EXPR_INITIALIZER: {
		struct expression *tmp;
		int i = 0;

		FOR_EACH_PTR(expr->expr_list, tmp) {
			if (tmp->type == EXPR_INDEX) {
				if (tmp->idx_to >= i)
					i = tmp->idx_to;
				else
					continue;
			}

			i++;
		} END_FOR_EACH_PTR(tmp);
		return i;
	}
	case EXPR_SYMBOL:
		return get_array_size(expr);
	}
	return 0;
}

static struct range_list *db_size_rl;
static int db_size_callback(void *unused, int argc, char **argv, char **azColName)
{
	struct range_list *tmp = NULL;

	if (!db_size_rl) {
		str_to_rl(&int_ctype, argv[0], &db_size_rl);
	} else {
		str_to_rl(&int_ctype, argv[0], &tmp);
		db_size_rl = rl_union(db_size_rl, tmp);
	}
	return 0;
}

static struct range_list *size_from_db_type(struct expression *expr)
{
	int this_file_only = 0;
	char *name;

	name = get_member_name(expr);
	if (!name && is_static(expr)) {
		name = expr_to_var(expr);
		this_file_only = 1;
	}
	if (!name)
		return NULL;

	if (this_file_only) {
		db_size_rl = NULL;
		run_sql(db_size_callback, NULL,
			"select size from function_type_size where type = '%s' and file = '%s';",
			name, get_filename());
		if (db_size_rl)
			return db_size_rl;
		return NULL;
	}

	db_size_rl = NULL;
	run_sql(db_size_callback, NULL,
		"select size from type_size where type = '%s';",
		name);
	return db_size_rl;
}

static struct range_list *size_from_db_symbol(struct expression *expr)
{
	struct symbol *sym;

	if (expr->type != EXPR_SYMBOL)
		return NULL;
	sym = expr->symbol;
	if (!sym || !sym->ident ||
	    !(sym->ctype.modifiers & MOD_TOPLEVEL) ||
	    sym->ctype.modifiers & MOD_STATIC)
		return NULL;

	db_size_rl = NULL;
	run_sql(db_size_callback, NULL,
		"select value from data_info where file = 'extern' and data = '%s' and type = %d;",
		sym->ident->name, BUF_SIZE);
	return db_size_rl;
}

static struct range_list *size_from_db(struct expression *expr)
{
	struct range_list *rl;

	rl = size_from_db_symbol(expr);
	if (rl)
		return rl;
	return size_from_db_type(expr);
}

static void db_returns_buf_size(struct expression *expr, int param, char *unused, char *math)
{
	struct expression *call;
	struct range_list *rl;
	sval_t sval;

	if (expr->type != EXPR_ASSIGNMENT)
		return;
	call = strip_expr(expr->right);

	call_results_to_rl(call, &int_ctype, math, &rl);
	rl = cast_rl(&int_ctype, rl);
	if (rl_to_sval(rl, &sval) && sval.value == 0)
		return;
	set_state_expr(my_size_id, expr->left, alloc_estate_rl(rl));
}

static int get_real_array_size_from_type(struct symbol *type)
{
	sval_t sval;

	if (!type)
		return 0;
	if (!type || type->type != SYM_ARRAY)
		return 0;

	if (!get_implied_value(type->array_size, &sval))
		return 0;

	return sval.value;
}

int get_real_array_size(struct expression *expr)
{
	if (!expr)
		return 0;
	if (expr->type == EXPR_PREOP && expr->op == '&')
		expr = expr->unop;
	if (expr->type == EXPR_BINOP) /* array elements foo[5] */
		return 0;
	return get_real_array_size_from_type(get_type(expr));
}

static int get_size_from_initializer(struct expression *expr)
{
	if (expr->type != EXPR_SYMBOL || !expr->symbol || !expr->symbol->initializer)
		return 0;
	if (expr->symbol->initializer == expr) /* int a = a; */
		return 0;
	return get_initializer_size(expr->symbol->initializer);
}

static struct range_list *get_stored_size_bytes(struct expression *expr)
{
	struct smatch_state *state;

	state = get_state_expr(my_size_id, expr);
	if (!state)
		return NULL;
	return estate_rl(state);
}

static int get_bytes_from_address(struct expression *expr)
{
	struct symbol *type;
	int ret;

	if (expr->type != EXPR_PREOP || expr->op != '&')
		return 0;
	type = get_type(expr);
	if (!type)
		return 0;

	if (type->type == SYM_PTR)
		type = get_base_type(type);

	ret = type_bytes(type);
	if (ret == 1)
		return 0;  /* ignore char pointers */

	return ret;
}

static struct expression *remove_addr_fluff(struct expression *expr)
{
	struct expression *tmp;
	sval_t sval;

	expr = strip_expr(expr);

	/* remove '&' and '*' operations that cancel */
	while (expr && expr->type == EXPR_PREOP && expr->op == '&') {
		tmp = strip_expr(expr->unop);
		if (tmp->type != EXPR_PREOP)
			break;
		if (tmp->op != '*')
			break;
		expr = strip_expr(tmp->unop);
	}

	if (!expr)
		return NULL;

	/* "foo + 0" is just "foo" */
	if (expr->type == EXPR_BINOP && expr->op == '+' &&
	    get_value(expr->right, &sval) && sval.value == 0)
		return expr->left;

	return expr;
}

static int is_last_member_of_struct(struct symbol *sym, struct ident *member)
{
	struct symbol *tmp;
	int i;

	i = 0;
	FOR_EACH_PTR_REVERSE(sym->symbol_list, tmp) {
		if (i++ || !tmp->ident)
			return 0;
		if (tmp->ident == member)
			return 1;
		return 0;
	} END_FOR_EACH_PTR_REVERSE(tmp);

	return 0;
}

int last_member_is_resizable(struct symbol *sym)
{
	struct symbol *last_member;
	struct symbol *type;
	sval_t sval;

	if (!sym || sym->type != SYM_STRUCT)
		return 0;

	last_member = last_ptr_list((struct ptr_list *)sym->symbol_list);
	if (!last_member || !last_member->ident)
		return 0;

	type = get_real_base_type(last_member);
	if (type->type == SYM_STRUCT)
		return last_member_is_resizable(type);
	if (type->type != SYM_ARRAY)
		return 0;

	if (!get_implied_value(type->array_size, &sval))
		return 0;

	if (sval.value != 0 && sval.value != 1)
		return 0;

	return 1;
}

static int get_stored_size_end_struct_bytes(struct expression *expr)
{
	struct symbol *sym;
	struct symbol *base_sym;
	struct smatch_state *state;

	if (expr->type == EXPR_BINOP) /* array elements foo[5] */
		return 0;

	if (expr->type == EXPR_PREOP && expr->op == '&')
		expr = strip_parens(expr->unop);

	sym = expr_to_sym(expr);
	if (!sym || !sym->ident)
		return 0;
	if (!type_bytes(sym))
		return 0;
	if (sym->type != SYM_NODE)
		return 0;

	base_sym = get_real_base_type(sym);
	if (!base_sym || base_sym->type != SYM_PTR)
		return 0;
	base_sym = get_real_base_type(base_sym);
	if (!base_sym || base_sym->type != SYM_STRUCT)
		return 0;

	if (!is_last_member_of_struct(base_sym, expr->member))
		return 0;
	if (!last_member_is_resizable(base_sym))
		return 0;

	state = get_state(my_size_id, sym->ident->name, sym);
	if (!estate_to_size(state) || estate_to_size(state) == -1)
		return 0;

	return estate_to_size(state) - type_bytes(base_sym) + type_bytes(get_type(expr));
}

static struct range_list *alloc_int_rl(int value)
{
	sval_t sval = {
		.type = &int_ctype,
		{.value = value},
	};

	return alloc_rl(sval, sval);
}

struct range_list *get_array_size_bytes_rl(struct expression *expr)
{
	struct range_list *ret = NULL;
	sval_t sval;
	int size;

	expr = remove_addr_fluff(expr);
	if (!expr)
		return NULL;

	/* "BAR" */
	if (expr->type == EXPR_STRING)
		return alloc_int_rl(expr->string->length);

	if (expr->type == EXPR_BINOP && expr->op == '+') {
		sval_t offset;
		struct symbol *type;
		int bytes;

		if (!get_implied_value(expr->right, &offset))
			return NULL;
		type = get_type(expr->left);
		if (!type)
			return NULL;
		if (type->type != SYM_ARRAY && type->type != SYM_PTR)
			return NULL;
		type = get_real_base_type(type);
		bytes = type_bytes(type);
		if (bytes == 0)
			return NULL;
		offset.value *= bytes;
		size = get_array_size_bytes(expr->left);
		if (size <= 0)
			return NULL;
		return alloc_int_rl(size - offset.value);
	}

	/* buf = malloc(1024); */
	ret = get_stored_size_bytes(expr);
	if (ret)
		return ret;

	size = get_stored_size_end_struct_bytes(expr);
	if (size)
		return alloc_int_rl(size);

	/* buf[4] */
	size = get_real_array_size(expr);
	if (size)
		return alloc_int_rl(elements_to_bytes(expr, size));

	/* char *foo = "BAR" */
	size = get_size_from_initializer(expr);
	if (size)
		return alloc_int_rl(elements_to_bytes(expr, size));

	size = get_bytes_from_address(expr);
	if (size)
		return alloc_int_rl(size);

	ret = size_from_db(expr);
	if (rl_to_sval(ret, &sval) && sval.value == -1)
		return NULL;
	if (ret)
		return ret;

	return NULL;
}

int get_array_size_bytes(struct expression *expr)
{
	struct range_list *rl;
	sval_t sval;

	rl = get_array_size_bytes_rl(expr);
	if (!rl_to_sval(rl, &sval))
		return 0;
	if (sval.uvalue >= INT_MAX)
		return 0;
	return sval.value;
}

int get_array_size_bytes_max(struct expression *expr)
{
	struct range_list *rl;
	sval_t bytes;

	rl = get_array_size_bytes_rl(expr);
	if (!rl)
		return 0;
	bytes = rl_min(rl);
	if (bytes.value < 0)
		return 0;
	bytes = rl_max(rl);
	if (bytes.uvalue >= INT_MAX)
		return 0;
	return bytes.value;
}

int get_array_size_bytes_min(struct expression *expr)
{
	struct range_list *rl;
	struct data_range *range;

	rl = get_array_size_bytes_rl(expr);
	if (!rl)
		return 0;

	FOR_EACH_PTR(rl, range) {
		if (range->min.value <= 0)
			return 0;
		if (range->max.value <= 0)
			return 0;
		if (range->min.uvalue >= INT_MAX)
			return 0;
		return range->min.value;
	} END_FOR_EACH_PTR(range);

	return 0;
}

int get_array_size(struct expression *expr)
{
	if (!expr)
		return 0;
	return bytes_to_elements(expr, get_array_size_bytes_max(expr));
}

static struct expression *strip_ampersands(struct expression *expr)
{
	struct symbol *type;

	if (expr->type != EXPR_PREOP)
		return expr;
	if (expr->op != '&')
		return expr;
	type = get_type(expr->unop);
	if (!type || type->type != SYM_ARRAY)
		return expr;
	return expr->unop;
}

static void info_record_alloction(struct expression *buffer, struct range_list *rl)
{
	char *name;

	if (!option_info)
		return;

	name = get_member_name(buffer);
	if (!name && is_static(buffer))
		name = expr_to_var(buffer);
	if (!name)
		return;
	if (rl && !is_whole_rl(rl))
		sql_insert_function_type_size(name, show_rl(rl));
	else
		sql_insert_function_type_size(name, "(-1)");

	free_string(name);
}

static void store_alloc(struct expression *expr, struct range_list *rl)
{
	struct symbol *type;

	rl = clone_rl(rl); // FIXME!!!
	if (!rl)
		rl = size_to_rl(UNKNOWN_SIZE);

	if (rl_min(rl).value != UNKNOWN_SIZE ||
	    rl_max(rl).value != UNKNOWN_SIZE ||
	    get_state_expr(my_size_id, expr))
		set_state_expr(my_size_id, expr, alloc_estate_rl(rl));

	type = get_type(expr);
	if (!type)
		return;
	if (type->type != SYM_PTR)
		return;
	type = get_real_base_type(type);
	if (!type)
		return;
	if (type == &void_ctype)
		return;
	if (type->type != SYM_BASETYPE && type->type != SYM_PTR)
		return;

	info_record_alloction(expr, rl);
}

static bool is_array_base(struct expression *expr)
{
	struct symbol *type;

	type = get_type(expr);
	if (type && type->type == SYM_ARRAY)
		return true;
	return false;
}

static void match_array_assignment(struct expression *expr)
{
	struct expression *left;
	struct expression *right;
	char *left_member, *right_member;
	struct range_list *rl;
	sval_t sval;

	if (expr->op != '=')
		return;

	left = strip_expr(expr->left);
	right = strip_expr(expr->right);
	right = strip_ampersands(right);

	if (!is_pointer(left))
		return;
	/* char buf[24] = "str"; */
	if (is_array_base(left))
		return;
	if (is_allocation_function(right))
		return;

	left_member = get_member_name(left);
	right_member = get_member_name(right);
	if (left_member && right_member && strcmp(left_member, right_member) == 0) {
		free_string(left_member);
		free_string(right_member);
		return;
	}
	free_string(left_member);
	free_string(right_member);

	if (get_implied_value(right, &sval) && sval.value == 0) {
		rl = alloc_int_rl(0);
		goto store;
	}

	rl = get_array_size_bytes_rl(right);
	if (!rl && __in_fake_assign)
		return;

store:
	store_alloc(left, rl);
}

static void match_alloc(const char *fn, struct expression *expr, void *_size_arg)
{
	int size_arg = PTR_INT(_size_arg);
	struct expression *right;
	struct expression *arg;
	struct range_list *rl;

	right = strip_expr(expr->right);
	arg = get_argument_from_call_expr(right->args, size_arg);
	get_absolute_rl(arg, &rl);
	rl = cast_rl(&int_ctype, rl);
	store_alloc(expr->left, rl);
}

static void match_calloc(const char *fn, struct expression *expr, void *_param)
{
	struct expression *right;
	struct expression *size, *nr, *mult;
	struct range_list *rl;
	int param = PTR_INT(_param);

	right = strip_expr(expr->right);
	nr = get_argument_from_call_expr(right->args, param);
	size = get_argument_from_call_expr(right->args, param + 1);
	mult = binop_expression(nr, '*', size);
	if (get_implied_rl(mult, &rl))
		store_alloc(expr->left, rl);
	else
		store_alloc(expr->left, size_to_rl(UNKNOWN_SIZE));
}

static void match_page(const char *fn, struct expression *expr, void *_unused)
{
	sval_t page_size = {
		.type = &int_ctype,
		{.value = 4096},
	};

	store_alloc(expr->left, alloc_rl(page_size, page_size));
}

static void match_strndup(const char *fn, struct expression *expr, void *unused)
{
	struct expression *fn_expr;
	struct expression *size_expr;
	sval_t size;

	fn_expr = strip_expr(expr->right);
	size_expr = get_argument_from_call_expr(fn_expr->args, 1);
	if (get_implied_max(size_expr, &size)) {
		size.value++;
		store_alloc(expr->left, size_to_rl(size.value));
	} else {
		store_alloc(expr->left, size_to_rl(UNKNOWN_SIZE));
	}

}

static void match_alloc_pages(const char *fn, struct expression *expr, void *_order_arg)
{
	int order_arg = PTR_INT(_order_arg);
	struct expression *right;
	struct expression *arg;
	sval_t sval;

	right = strip_expr(expr->right);
	arg = get_argument_from_call_expr(right->args, order_arg);
	if (!get_implied_value(arg, &sval))
		return;
	if (sval.value < 0 || sval.value > 10)
		return;

	sval.type = &int_ctype;
	sval.value = 1 << sval.value;
	sval.value *= 4096;

	store_alloc(expr->left, alloc_rl(sval, sval));
}

static int is_type_bytes(struct range_list *rl, struct expression *arg)
{
	struct symbol *type;
	sval_t sval;

	if (!rl_to_sval(rl, &sval))
		return 0;

	type = get_type(arg);
	if (!type)
		return 0;
	if (type->type != SYM_PTR)
		return 0;
	type = get_real_base_type(type);
	if (type->type != SYM_STRUCT &&
	    type->type != SYM_UNION)
		return 0;
	if (sval.value != type_bytes(type))
		return 0;
	return 1;
}

static void match_call(struct expression *expr)
{
	struct expression *arg;
	struct symbol *type;
	struct range_list *rl;
	int i;

	i = -1;
	FOR_EACH_PTR(expr->args, arg) {
		i++;
		type = get_type(arg);
		if (!type || (type->type != SYM_PTR && type->type != SYM_ARRAY))
			continue;
		rl = get_array_size_bytes_rl(arg);
		if (!rl)
			continue;
		if (rl_min(rl).value == UNKNOWN_SIZE &&
		    rl_max(rl).value == UNKNOWN_SIZE)
			continue;
		if (is_whole_rl(rl))
			continue;
		if (is_type_bytes(rl, arg))
			continue;
		sql_insert_caller_info(expr, BUF_SIZE, i, "$", show_rl(rl));
	} END_FOR_EACH_PTR(arg);
}

static void struct_member_callback(struct expression *call, int param, char *printed_name, struct sm_state *sm)
{
	sval_t sval;

	if (!estate_rl(sm->state) ||
	    (estate_get_single_value(sm->state, &sval) &&
	     (sval.value == -1 || sval.value == 0)))
		return;

	sql_insert_caller_info(call, BUF_SIZE, param, printed_name, sm->state->name);
}

/*
 * This is slightly (very) weird because half of this stuff is handled in
 * smatch_parse_call_math.c which is poorly named.  But anyway, add some buf
 * sizes here.
 *
 */
static void print_returned_allocations(int return_id, char *return_ranges, struct expression *expr)
{
	const char *param_math;
	struct range_list *rl;
	char buf[64];
	sval_t sval;

	rl = get_array_size_bytes_rl(expr);
	param_math = get_allocation_math(expr);
	if (!rl && !param_math)
		return;

	if (!param_math &&
	    rl_to_sval(rl, &sval) &&
	    (sval.value == -1 || sval.value == 0))
		return;

	if (param_math)
		snprintf(buf, sizeof(buf), "%s[%s]", show_rl(rl), param_math);
	else
		snprintf(buf, sizeof(buf), "%s", show_rl(rl));

	// FIXME: don't store if you can guess the size from the type
	// FIXME: return if we allocate a parameter $0->bar
	sql_insert_return_states(return_id, return_ranges, BUF_SIZE, -1, "", buf);
}

static void record_global_size(struct symbol *sym)
{
	int bytes;
	char buf[16];

	if (!sym->ident)
		return;

	if (!(sym->ctype.modifiers & MOD_TOPLEVEL) ||
	    sym->ctype.modifiers & MOD_STATIC)
		return;

	bytes = get_array_size_bytes(symbol_expression(sym));
	if (bytes <= 1)
		return;

	snprintf(buf, sizeof(buf), "%d", bytes);
	sql_insert_data_info_var_sym(sym->ident->name, sym, BUF_SIZE, buf);
}

void register_buf_size(int id)
{
	my_size_id = id;

	set_dynamic_states(my_size_id);

	add_unmatched_state_hook(my_size_id, &unmatched_size_state);
	add_merge_hook(my_size_id, &merge_estates);

	select_caller_info_hook(set_param_buf_size, BUF_SIZE);
	select_return_states_hook(BUF_SIZE, &db_returns_buf_size);
	add_split_return_callback(print_returned_allocations);

	allocation_funcs = create_function_hashtable(100);
	add_allocation_function("malloc", &match_alloc, 0);
	add_allocation_function("calloc", &match_calloc, 0);
	add_allocation_function("memdup", &match_alloc, 1);
	add_allocation_function("realloc", &match_alloc, 1);
	if (option_project == PROJ_KERNEL) {
		add_allocation_function("kmalloc", &match_alloc, 0);
		add_allocation_function("kmalloc_node", &match_alloc, 0);
		add_allocation_function("kzalloc", &match_alloc, 0);
		add_allocation_function("kzalloc_node", &match_alloc, 0);
		add_allocation_function("vmalloc", &match_alloc, 0);
		add_allocation_function("vzalloc", &match_alloc, 0);
		add_allocation_function("__vmalloc", &match_alloc, 0);
		add_allocation_function("kvmalloc", &match_alloc, 0);
		add_allocation_function("kcalloc", &match_calloc, 0);
		add_allocation_function("kmalloc_array", &match_calloc, 0);
		add_allocation_function("devm_kmalloc_array", &match_calloc, 1);
		add_allocation_function("sock_kmalloc", &match_alloc, 1);
		add_allocation_function("kmemdup", &match_alloc, 1);
		add_allocation_function("kmemdup_user", &match_alloc, 1);
		add_allocation_function("dma_alloc_attrs", &match_alloc, 1);
		add_allocation_function("pci_alloc_consistent", &match_alloc, 1);
		add_allocation_function("pci_alloc_coherent", &match_alloc, 1);
		add_allocation_function("devm_kmalloc", &match_alloc, 1);
		add_allocation_function("devm_kzalloc", &match_alloc, 1);
		add_allocation_function("krealloc", &match_alloc, 1);
		add_allocation_function("__alloc_bootmem", &match_alloc, 0);
		add_allocation_function("alloc_bootmem", &match_alloc, 0);
		add_allocation_function("kmap", &match_page, 0);
		add_allocation_function("kmap_atomic", &match_page, 0);
		add_allocation_function("get_zeroed_page", &match_page, 0);
		add_allocation_function("alloc_page", &match_page, 0);
		add_allocation_function("alloc_pages", &match_alloc_pages, 1);
		add_allocation_function("alloc_pages_current", &match_alloc_pages, 1);
		add_allocation_function("__get_free_pages", &match_alloc_pages, 1);
		add_allocation_function("dma_alloc_contiguous", &match_alloc, 1);
		add_allocation_function("dma_alloc_coherent", &match_alloc, 1);
	}

	add_allocation_function("strndup", match_strndup, 0);
	if (option_project == PROJ_KERNEL)
		add_allocation_function("kstrndup", match_strndup, 0);

	add_modification_hook(my_size_id, &set_size_undefined);

	add_merge_hook(my_size_id, &merge_size_func);

	if (option_info)
		add_hook(record_global_size, BASE_HOOK);
}

void register_buf_size_late(int id)
{
	/* has to happen after match_alloc() */
	add_hook(&match_array_assignment, ASSIGNMENT_HOOK);

	add_hook(&match_call, FUNCTION_CALL_HOOK);
	add_member_info_callback(my_size_id, struct_member_callback);
}