/* 
** ZABBIX
** Copyright (C) 2000-2005 SIA Zabbix
**
** 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., 675 Mass Ave, Cambridge, MA 02139, USA.
**/
#include "common.h"
#include "comms.h"
#include "log.h"

#include "checks_agent.h"

/******************************************************************************
 *                                                                            *
 * Function: get_value_agent                                                  *
 *                                                                            *
 * Purpose: retrieve data from ZABBIX agent                                   *
 *                                                                            *
 * Parameters: item - item we are interested in                               *
 *                                                                            *
 * Return value: SUCCEED - data succesfully retrieved and stored in result    *
 *                         and result_str (as string)                         *
 *               NETWORK_ERROR - network related error occured                *
 *               NOTSUPPORTED - item not supported by the agent               *
 *               AGENT_ERROR - uncritical error on agent side occured         *
 *               FAIL - otherwise                                             *
 *                                                                            *
 * Author: Alexei Vladishev                                                   *
 *                                                                            *
 * Comments: error will contain error message                                 *
 *                                                                            *
 ******************************************************************************/
int	get_value_agent(DB_ITEM *item, AGENT_RESULT *result)
{
	zbx_sock_t	s;

	char	*addr,
		*buf,
		packet[MAX_STRING_LEN],
		error[MAX_STRING_LEN];

	int	ret = SUCCEED;

	init_result(result);

	addr = (item->useip == 1) ? item->host_ip : item->host_dns;
	zabbix_log( LOG_LEVEL_DEBUG, "In get_value_agent(host:%s,addr:%s,key:%s)",
			item->host_name,
			addr,
			item->key);

	if (SUCCEED == (ret = zbx_tcp_connect(&s, CONFIG_SOURCE_IP, addr, item->port, 0))) {
		zbx_snprintf(packet, sizeof(packet), "%s\n",item->key);
		zabbix_log(LOG_LEVEL_DEBUG, "Sending [%s]", packet);

		/* Send requests using old protocol */
		if( SUCCEED == (ret = zbx_tcp_send_raw(&s, packet)) )
		{
			ret = zbx_tcp_recv_ext(&s, &buf, ZBX_TCP_READ_UNTIL_CLOSE);
		}
	}

	if( SUCCEED == ret )
	{
		zbx_rtrim(buf, " \r\n\0");
		zbx_ltrim(buf, " ");

		if( strcmp(buf,"ZBX_NOTSUPPORTED") == 0)
		{
			zbx_snprintf(error,sizeof(error),"Not supported by ZABBIX agent");
			SET_MSG_RESULT(result, strdup(error));
			ret = NOTSUPPORTED;
		}
		else if( strcmp(buf,"ZBX_ERROR") == 0)
		{
			zbx_snprintf(error,sizeof(error),"ZABBIX agent non-critical error");
			SET_MSG_RESULT(result, strdup(error));
			ret = AGENT_ERROR;
		}
		/* The section should be improved */
		else if(buf[0]==0)
		{
			zbx_snprintf(error,sizeof(error),"Got empty string from [%s] IP [%s] Parameter [%s]",
				item->host_name,
				item->host_ip,
				item->key);
			zabbix_log( LOG_LEVEL_WARNING, "%s",
				error);
			zabbix_log( LOG_LEVEL_WARNING, "Assuming that agent dropped connection because of access permissions");
			SET_MSG_RESULT(result, strdup(error));
			ret = NETWORK_ERROR;
		}
		else if(set_result_type(result, item->value_type, buf) == FAIL)
		{
			zbx_snprintf(error,sizeof(error), "Type of received value [%s] is not sutable for [%s@%s] having type [%d]",
				buf,
				item->key,
				item->host_name,
				item->value_type);
			zabbix_log( LOG_LEVEL_WARNING, "%s",
				error);
			zabbix_log( LOG_LEVEL_WARNING, "Returning NOTSUPPORTED");
			result->msg=strdup(error);
			ret = NOTSUPPORTED;
		}

		zabbix_log( LOG_LEVEL_DEBUG, "End get_value_agent(result:%s)",
			buf);

	}
	else
	{
		zabbix_log(LOG_LEVEL_WARNING, "Get value from agent failed. Error: %s", zbx_tcp_strerror());
		SET_MSG_RESULT(result, strdup(zbx_tcp_strerror()));
		ret = NETWORK_ERROR;

	}	
	zbx_tcp_close(&s);

	return ret;
}