void zmq::socks_connecter_t::initiate_connect ()
{
// Open the connecting socket.
const int rc = connect_to_proxy ();
// Connect may succeed in synchronous manner.
if (rc == 0) {
handle = add_fd (s);
set_pollout (handle);
status = sending_greeting;
}
// Connection establishment may be delayed. Poll for its completion.
else
if (errno == EINPROGRESS) {
handle = add_fd (s);
set_pollout (handle);
status = waiting_for_proxy_connection;
socket->event_connect_delayed (endpoint, zmq_errno ());
}
// Handle any other error condition by eventual reconnect.
else {
if (s != retired_fd)
close ();
start_timer ();
}
}
/******************************************************************************
* *
* Function: get_data_from_proxy *
* *
* Purpose: get historical data from proxy *
* *
* Parameters: *
* *
* Return value: SUCCESS - processed successfully *
* FAIL - an error occurred *
* *
* Author: Alexander Vladishev *
* *
* Comments: *
* *
******************************************************************************/
static int get_data_from_proxy(DC_PROXY *proxy, const char *request, char **data)
{
const char *__function_name = "get_data_from_proxy";
zbx_socket_t s;
struct zbx_json j;
int ret = FAIL;
zabbix_log(LOG_LEVEL_DEBUG, "In %s() request:'%s'", __function_name, request);
zbx_json_init(&j, ZBX_JSON_STAT_BUF_LEN);
zbx_json_addstring(&j, "request", request, ZBX_JSON_TYPE_STRING);
if (SUCCEED == (ret = connect_to_proxy(proxy, &s, CONFIG_TRAPPER_TIMEOUT)))
{
if (SUCCEED == (ret = send_data_to_proxy(proxy, &s, j.buffer)))
if (SUCCEED == (ret = recv_data_from_proxy(proxy, &s)))
if (SUCCEED == (ret = zbx_send_response(&s, SUCCEED, NULL, 0)))
*data = zbx_strdup(*data, s.buffer);
disconnect_proxy(&s);
}
zbx_json_free(&j);
zabbix_log(LOG_LEVEL_DEBUG, "End of %s():%s", __function_name, zbx_result_string(ret));
return ret;
}
/******************************************************************************
* *
* Function: process_proxy *
* *
* Purpose: retrieve values of metrics from monitored hosts *
* *
* Parameters: *
* *
* Return value: *
* *
* Author: Alexei Vladishev *
* *
* Comments: *
* *
******************************************************************************/
static int process_proxy(void)
{
const char *__function_name = "process_proxy";
DC_PROXY proxy;
int num, i, ret;
struct zbx_json j;
struct zbx_json_parse jp, jp_data;
zbx_socket_t s;
char *answer = NULL, *port = NULL;
time_t now;
unsigned char update_nextcheck;
zabbix_log(LOG_LEVEL_DEBUG, "In %s()", __function_name);
if (0 == (num = DCconfig_get_proxypoller_hosts(&proxy, 1)))
goto exit;
now = time(NULL);
zbx_json_init(&j, 512 * 1024);
for (i = 0; i < num; i++)
{
update_nextcheck = 0;
if (proxy.proxy_config_nextcheck <= now)
update_nextcheck |= 0x01;
if (proxy.proxy_data_nextcheck <= now)
update_nextcheck |= 0x02;
proxy.addr = proxy.addr_orig;
port = zbx_strdup(port, proxy.port_orig);
substitute_simple_macros(NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
&port, MACRO_TYPE_COMMON, NULL, 0);
if (FAIL == is_ushort(port, &proxy.port))
{
zabbix_log(LOG_LEVEL_ERR, "invalid proxy \"%s\" port: \"%s\"", proxy.host, port);
goto network_error;
}
if (proxy.proxy_config_nextcheck <= now)
{
char *error = NULL;
zbx_json_clean(&j);
zbx_json_addstring(&j, ZBX_PROTO_TAG_REQUEST,
ZBX_PROTO_VALUE_PROXY_CONFIG, ZBX_JSON_TYPE_STRING);
zbx_json_addobject(&j, ZBX_PROTO_TAG_DATA);
if (SUCCEED != (ret = get_proxyconfig_data(proxy.hostid, &j, &error)))
{
zabbix_log(LOG_LEVEL_ERR, "cannot collect configuration data for proxy \"%s\": %s",
proxy.host, error);
zbx_free(error);
goto network_error;
}
if (SUCCEED == (ret = connect_to_proxy(&proxy, &s, CONFIG_TRAPPER_TIMEOUT)))
{
zabbix_log(LOG_LEVEL_WARNING, "sending configuration data to proxy \"%s\" at \"%s\","
" datalen " ZBX_FS_SIZE_T,
proxy.host, get_ip_by_socket(&s), (zbx_fs_size_t)j.buffer_size);
if (SUCCEED == (ret = send_data_to_proxy(&proxy, &s, j.buffer)))
{
char *error = NULL;
if (SUCCEED != (ret = zbx_recv_response(&s, 0, &error)))
{
zabbix_log(LOG_LEVEL_WARNING, "cannot send configuration data to proxy"
" \"%s\" at \"%s\": %s",
proxy.host, get_ip_by_socket(&s), error);
}
zbx_free(error);
}
disconnect_proxy(&s);
}
if (SUCCEED != ret)
goto network_error;
}
if (proxy.proxy_data_nextcheck <= now)
{
if (SUCCEED == get_data_from_proxy(&proxy,
ZBX_PROTO_VALUE_HOST_AVAILABILITY, &answer))
{
if (SUCCEED == zbx_json_open(answer, &jp))
process_host_availability(&jp);
zbx_free(answer);
}
else
goto network_error;
retry_history:
if (SUCCEED == get_data_from_proxy(&proxy,
ZBX_PROTO_VALUE_HISTORY_DATA, &answer))
{
if (SUCCEED == zbx_json_open(answer, &jp))
{
process_hist_data(NULL, &jp, proxy.hostid, NULL, 0);
if (SUCCEED == zbx_json_brackets_by_name(&jp, ZBX_PROTO_TAG_DATA, &jp_data))
{
if (ZBX_MAX_HRECORDS <= zbx_json_count(&jp_data))
{
zbx_free(answer);
goto retry_history;
}
}
}
zbx_free(answer);
}
else
goto network_error;
retry_dhistory:
if (SUCCEED == get_data_from_proxy(&proxy,
ZBX_PROTO_VALUE_DISCOVERY_DATA, &answer))
{
if (SUCCEED == zbx_json_open(answer, &jp))
{
process_dhis_data(&jp);
if (SUCCEED == zbx_json_brackets_by_name(&jp, ZBX_PROTO_TAG_DATA, &jp_data))
{
if (ZBX_MAX_HRECORDS <= zbx_json_count(&jp_data))
{
zbx_free(answer);
goto retry_dhistory;
}
}
}
zbx_free(answer);
}
else
goto network_error;
retry_autoreg_host:
if (SUCCEED == get_data_from_proxy(&proxy,
ZBX_PROTO_VALUE_AUTO_REGISTRATION_DATA, &answer))
{
if (SUCCEED == zbx_json_open(answer, &jp))
{
process_areg_data(&jp, proxy.hostid);
if (SUCCEED == zbx_json_brackets_by_name(&jp, ZBX_PROTO_TAG_DATA, &jp_data))
{
if (ZBX_MAX_HRECORDS <= zbx_json_count(&jp_data))
{
zbx_free(answer);
goto retry_autoreg_host;
}
}
}
zbx_free(answer);
}
else
goto network_error;
}
DBbegin();
update_proxy_lastaccess(proxy.hostid);
DBcommit();
network_error:
DCrequeue_proxy(proxy.hostid, update_nextcheck);
}
zbx_free(port);
zbx_json_free(&j);
exit:
zabbix_log(LOG_LEVEL_DEBUG, "End of %s()", __function_name);
return num;
}
