1/*
2 * Copyright (C) 2018 Oracle.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version 2
7 * of the License, or (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, see http://www.gnu.org/copyleft/gpl.txt
16 */
17
18#include <stdlib.h>
19#include "parse.h"
20#include "smatch.h"
21#include "smatch_slist.h"
22#include "smatch_extra.h"
23
24static int my_id;
25
26static struct smatch_state *merge_states(struct smatch_state *s1, struct smatch_state *s2)
27{
28	int left, right, min;
29
30	left = PTR_INT(s1->data);
31	right = PTR_INT(s2->data);
32
33	min = left;
34	if (right < min)
35		min = right;
36	return alloc_state_num(min);
37}
38
39long get_stmt_cnt(void)
40{
41	struct smatch_state *state;
42
43	state = get_state(my_id, "stmts", NULL);
44	if (!state)
45		return 0;
46	return (long)state->data;
47}
48
49static void match_statement(struct statement *stmt)
50{
51	int cnt;
52
53	cnt = get_stmt_cnt();
54	cnt++;
55	set_state(my_id, "stmts", NULL, alloc_state_num(cnt));
56}
57
58static void insert_return_info(int return_id, char *return_ranges, struct expression *expr)
59{
60	char buf[32];
61	int cnt;
62
63	cnt = get_stmt_cnt();
64	snprintf(buf, sizeof(buf), "%d", cnt);
65	sql_insert_return_states(return_id, return_ranges, STMT_CNT, -1, "", buf);
66}
67
68static void select_return_info(struct expression *expr, int param, char *key, char *value)
69{
70	int cnt, add;
71
72	cnt = get_stmt_cnt();
73	add = atoi(value);
74
75	set_state(my_id, "stmts", NULL, alloc_state_num(cnt + add));
76}
77
78void register_statement_count(int id)
79{
80	my_id = id;
81
82	set_dynamic_states(my_id);
83	add_hook(match_statement, STMT_HOOK);
84	add_merge_hook(my_id, &merge_states);
85
86	add_split_return_callback(&insert_return_info);
87	select_return_states_hook(STMT_CNT, &select_return_info);
88}
89
90