1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/*
** Zabbix
** Copyright (C) 2001-2019 Zabbix SIA
**
** 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, write to the Free Software
** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
**/
#include "checks_db.h"
#ifdef HAVE_UNIXODBC
#include "log.h"
#include "../odbc/odbc.h"
/******************************************************************************
* *
* Function: get_value_db *
* *
* Purpose: retrieve data from database *
* *
* Parameters: item - [IN] item we are interested in *
* result - [OUT] check result *
* *
* Return value: SUCCEED - data successfully retrieved and stored in result *
* NOTSUPPORTED - requested item is not supported *
* *
* Author: Eugene Grigorjev *
* *
******************************************************************************/
int get_value_db(DC_ITEM *item, AGENT_RESULT *result)
{
AGENT_REQUEST request;
const char *dsn;
zbx_odbc_data_source_t *data_source;
zbx_odbc_query_result_t *query_result;
char *error = NULL;
int (*query_result_to_text)(zbx_odbc_query_result_t *query_result, char **text, char **error),
ret = NOTSUPPORTED;
zabbix_log(LOG_LEVEL_DEBUG, "In %s() key_orig:'%s' query:'%s'", __func__, item->key_orig, item->params);
init_request(&request);
if (SUCCEED != parse_item_key(item->key, &request))
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid item key format."));
goto out;
}
if (0 == strcmp(request.key, "db.odbc.select"))
{
query_result_to_text = zbx_odbc_query_result_to_string;
}
else if (0 == strcmp(request.key, "db.odbc.discovery"))
{
query_result_to_text = zbx_odbc_query_result_to_lld_json;
}
else if (0 == strcmp(request.key, "db.odbc.get"))
{
query_result_to_text = zbx_odbc_query_result_to_json;
}
else
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Unsupported item key for this item type."));
goto out;
}
if (2 != request.nparam)
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid number of parameters."));
goto out;
}
/* request.params[0] is ignored and is only needed to distinguish queries of same DSN */
dsn = request.params[1];
if (NULL == dsn || '\0' == *dsn)
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid second parameter."));
goto out;
}
if (NULL != (data_source = zbx_odbc_connect(dsn, item->username, item->password, CONFIG_TIMEOUT, &error)))
{
if (NULL != (query_result = zbx_odbc_select(data_source, item->params, &error)))
{
char *text = NULL;
if (SUCCEED == query_result_to_text(query_result, &text, &error))
{
SET_TEXT_RESULT(result, text);
ret = SUCCEED;
}
zbx_odbc_query_result_free(query_result);
}
zbx_odbc_data_source_free(data_source);
}
if (SUCCEED != ret)
SET_MSG_RESULT(result, error);
out:
free_request(&request);
zabbix_log(LOG_LEVEL_DEBUG, "End of %s():%s", __func__, zbx_result_string(ret));
return ret;
}
#endif /* HAVE_UNIXODBC */
|