int NET_TCP_PORT(const char *cmd, const char *param, unsigned flags, AGENT_RESULT *result)
{
unsigned short port;
int value_int, ret;
char ip[64], port_str[8];
if (2 < num_param(param))
return SYSINFO_RET_FAIL;
if (0 != get_param(param, 1, ip, sizeof(ip)))
*ip = '\0';
if ('\0' == *ip)
strscpy(ip, "127.0.0.1");
if (0 != get_param(param, 2, port_str, sizeof(port_str)))
*port_str = '\0';
if (SUCCEED != is_ushort(port_str, &port))
return SYSINFO_RET_FAIL;
if (SYSINFO_RET_OK == (ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, NULL, NULL, &value_int)))
SET_UI64_RESULT(result, value_int);
return ret;
}
int NET_TCP_PORT(AGENT_REQUEST *request, AGENT_RESULT *result)
{
unsigned short port;
int value_int, ret;
char *ip_str, ip[MAX_ZBX_DNSNAME_LEN + 1], *port_str;
if (2 < request->nparam)
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Too many parameters."));
return SYSINFO_RET_FAIL;
}
ip_str = get_rparam(request, 0);
port_str = get_rparam(request, 1);
if (NULL == ip_str || '\0' == *ip_str)
strscpy(ip, "127.0.0.1");
else
strscpy(ip, ip_str);
if (NULL == port_str || SUCCEED != is_ushort(port_str, &port))
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid second parameter."));
return SYSINFO_RET_FAIL;
}
if (SYSINFO_RET_OK == (ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, NULL, NULL, &value_int)))
SET_UI64_RESULT(result, value_int);
return ret;
}
int CHECK_PORT(const char *cmd, const char *param, unsigned flags, AGENT_RESULT *result)
{
short port=0;
int value_int;
int ret;
char ip[MAX_STRING_LEN];
char port_str[MAX_STRING_LEN];
assert(result);
init_result(result);
if(num_param(param) > 2)
{
return SYSINFO_RET_FAIL;
}
if(get_param(param, 1, ip, MAX_STRING_LEN) != 0)
{
ip[0] = '\0';
}
if(ip[0] == '\0')
{
strscpy(ip, "127.0.0.1");
}
if(get_param(param, 2, port_str, MAX_STRING_LEN) != 0)
{
port_str[0] = '\0';
}
if(port_str[0] == '\0')
{
return SYSINFO_RET_FAIL;
}
port=atoi(port_str);
ret = tcp_expect(ip,port,NULL,NULL,"",&value_int);
if(ret == SYSINFO_RET_OK)
{
SET_UI64_RESULT(result, value_int);
}
return ret;
}
int check_service(AGENT_REQUEST *request, const char *default_addr, AGENT_RESULT *result, int perf)
{
unsigned short port = 0;
char *service, *ip_str, ip[MAX_ZBX_DNSNAME_LEN + 1], *port_str;
int value_int, ret = SYSINFO_RET_FAIL;
double check_time;
check_time = zbx_time();
if (3 < request->nparam)
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Too many parameters."));
return SYSINFO_RET_FAIL;
}
service = get_rparam(request, 0);
ip_str = get_rparam(request, 1);
port_str = get_rparam(request, 2);
if (NULL == service || '\0' == *service)
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid first parameter."));
return SYSINFO_RET_FAIL;
}
if (NULL == ip_str || '\0' == *ip_str)
strscpy(ip, default_addr);
else
strscpy(ip, ip_str);
if (NULL != port_str && '\0' != *port_str && SUCCEED != is_ushort(port_str, &port))
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid third parameter."));
return SYSINFO_RET_FAIL;
}
if (0 == strncmp("net.tcp.service", get_rkey(request), 15))
{
if (0 == strcmp(service, "ssh"))
{
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_SSH_PORT;
ret = check_ssh(ip, port, CONFIG_TIMEOUT, &value_int);
}
else if (0 == strcmp(service, "ldap"))
{
#ifdef HAVE_LDAP
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_LDAP_PORT;
ret = check_ldap(ip, port, CONFIG_TIMEOUT, &value_int);
#else
SET_MSG_RESULT(result, zbx_strdup(NULL, "Support for LDAP check was not compiled in."));
#endif
}
else if (0 == strcmp(service, "smtp"))
{
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_SMTP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, validate_smtp, "QUIT\r\n", &value_int);
}
else if (0 == strcmp(service, "ftp"))
{
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_FTP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, validate_ftp, "QUIT\r\n", &value_int);
}
else if (0 == strcmp(service, "http"))
{
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_HTTP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, NULL, NULL, &value_int);
}
else if (0 == strcmp(service, "pop"))
{
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_POP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, validate_pop, "QUIT\r\n", &value_int);
}
else if (0 == strcmp(service, "nntp"))
{
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_NNTP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, validate_nntp, "QUIT\r\n", &value_int);
}
else if (0 == strcmp(service, "imap"))
{
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_IMAP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, validate_imap, "a1 LOGOUT\r\n", &value_int);
}
else if (0 == strcmp(service, "tcp"))
{
if (NULL == port_str || '\0' == *port_str)
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid third parameter."));
return SYSINFO_RET_FAIL;
}
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, NULL, NULL, &value_int);
}
else if (0 == strcmp(service, "https"))
{
#ifdef HAVE_LIBCURL
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_HTTPS_PORT;
ret = check_https(ip, port, CONFIG_TIMEOUT, &value_int);
#else
SET_MSG_RESULT(result, zbx_strdup(NULL, "Support for HTTPS check was not compiled in."));
#endif
}
else if (0 == strcmp(service, "telnet"))
{
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_TELNET_PORT;
ret = check_telnet(ip, port, CONFIG_TIMEOUT, &value_int);
}
else
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid first parameter."));
return ret;
}
}
else /* net.udp.service */
{
if (0 == strcmp(service, "ntp"))
{
if (NULL == port_str || '\0' == *port_str)
port = ZBX_DEFAULT_NTP_PORT;
ret = check_ntp(ip, port, CONFIG_TIMEOUT, &value_int);
}
else
{
SET_MSG_RESULT(result, zbx_strdup(NULL, "Invalid first parameter."));
return ret;
}
}
if (SYSINFO_RET_OK == ret)
{
if (0 != perf)
{
if (0 != value_int)
{
check_time = zbx_time() - check_time;
if (ZBX_FLOAT_PRECISION > check_time)
check_time = ZBX_FLOAT_PRECISION;
SET_DBL_RESULT(result, check_time);
}
else
SET_DBL_RESULT(result, 0.0);
}
else
SET_UI64_RESULT(result, value_int);
}
return ret;
}
/* check_service[ssh,127.0.0.1,ssh] */
int CHECK_SERVICE(const char *cmd, const char *param, unsigned flags, AGENT_RESULT *result)
{
unsigned short port=0;
char service[MAX_STRING_LEN];
char ip[MAX_STRING_LEN];
char str_port[MAX_STRING_LEN];
int ret;
int value_int = 0;
assert(result);
init_result(result);
if(num_param(param) > 3)
{
return SYSINFO_RET_FAIL;
}
if(get_param(param, 1, service, MAX_STRING_LEN) != 0)
{
return SYSINFO_RET_FAIL;
}
if(get_param(param, 2, ip, MAX_STRING_LEN) != 0)
{
ip[0] = '\0';
}
if(ip[0] == '\0')
{
strscpy(ip, "127.0.0.1");
}
if(get_param(param, 3, str_port, MAX_STRING_LEN) != 0)
{
str_port[0] = '\0';
}
if(str_port[0] != '\0')
{
port = atoi(str_port);
}
else
{
port = 0;
}
/* printf("IP:[%s]",ip);
printf("Service:[%s]",service);
printf("Port:[%d]",port);*/
if(strcmp(service,"ssh") == 0)
{
if(port == 0) port=22;
ret=check_ssh(ip,port,&value_int);
}
else if(strcmp(service,"service.ntp") == 0)
{
if(port == 0) port=123;
ret=check_ntp(ip,port,&value_int);
}
#ifdef HAVE_LDAP
else if(strcmp(service,"ldap") == 0)
{
if(port == 0) port=389;
ret=check_ldap(ip,port,&value_int);
}
#endif
else if(strcmp(service,"smtp") == 0)
{
if(port == 0) port=25;
ret=tcp_expect(ip,port,NULL,"220","QUIT\n",&value_int);
}
else if(strcmp(service,"ftp") == 0)
{
if(port == 0) port=21;
ret=tcp_expect(ip,port,NULL,"220","",&value_int);
}
else if(strcmp(service,"http") == 0)
{
if(port == 0) port=80;
ret=tcp_expect(ip,port,NULL,NULL,"",&value_int);
}
else if(strcmp(service,"pop") == 0)
{
if(port == 0) port=110;
ret=tcp_expect(ip,port,NULL,"+OK","",&value_int);
}
else if(strcmp(service,"nntp") == 0)
{
if(port == 0) port=119;
/* 220 is incorrect */
/* ret=tcp_expect(ip,port,"220","");*/
ret=tcp_expect(ip,port,NULL,"200","",&value_int);
}
else if(strcmp(service,"imap") == 0)
{
if(port == 0) port=143;
ret=tcp_expect(ip,port,NULL,"* OK","a1 LOGOUT\n",&value_int);
}
else if(strcmp(service,"tcp") == 0)
{
if(port == 0) port=80;
ret=tcp_expect(ip,port,NULL,NULL,"",&value_int);
}
else
{
return SYSINFO_RET_FAIL;
}
if(SYSINFO_RET_OK == ret)
{
SET_UI64_RESULT(result, value_int);
}
return ret;
}
int check_service(const char *params, const char *default_addr, AGENT_RESULT *result, int perf)
{
unsigned short port = 0;
char service[16], ip[64], str_port[8];
int value_int, ret = SYSINFO_RET_FAIL;
double check_time;
check_time = zbx_time();
if (3 < num_param(params))
return ret;
if (0 != get_param(params, 1, service, sizeof(service)))
return ret;
if (0 != get_param(params, 2, ip, sizeof(ip)) || '\0' == *ip)
strscpy(ip, default_addr);
if (0 != get_param(params, 3, str_port, sizeof(str_port)))
*str_port = '\0';
if ('\0' != *str_port && FAIL == is_ushort(str_port, &port))
{
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Invalid \"port\" parameter"));
return ret;
}
if (0 == strcmp(service, "ssh"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_SSH_PORT;
ret = check_ssh(ip, port, CONFIG_TIMEOUT, &value_int);
}
else if (0 == strcmp(service, "ntp") || 0 == strcmp(service, "service.ntp" /* deprecated */))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_NTP_PORT;
ret = check_ntp(ip, port, CONFIG_TIMEOUT, &value_int);
}
#ifdef HAVE_LDAP
else if (0 == strcmp(service, "ldap"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_LDAP_PORT;
ret = check_ldap(ip, port, CONFIG_TIMEOUT, &value_int);
}
#endif
else if (0 == strcmp(service, "smtp"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_SMTP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, "220", "QUIT\r\n", &value_int);
}
else if (0 == strcmp(service, "ftp"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_FTP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, "220", "QUIT\n", &value_int);
}
else if (0 == strcmp(service, "http"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_HTTP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, NULL, NULL, &value_int);
}
else if (0 == strcmp(service, "pop"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_POP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, "+OK", "QUIT\n", &value_int);
}
else if (0 == strcmp(service, "nntp"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_NNTP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, "200", "QUIT\n", &value_int);
}
else if (0 == strcmp(service, "imap"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_IMAP_PORT;
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, "* OK", "a1 LOGOUT\n", &value_int);
}
else if (0 == strcmp(service, "tcp"))
{
if ('\0' == *str_port)
{
SET_MSG_RESULT(result, zbx_dsprintf(NULL, "Required \"port\" parameter missing"));
return ret;
}
ret = tcp_expect(ip, port, CONFIG_TIMEOUT, NULL, NULL, NULL, &value_int);
}
#ifdef HAVE_LIBCURL
else if (0 == strcmp(service, "https"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_HTTPS_PORT;
ret = check_https(ip, port, CONFIG_TIMEOUT, &value_int);
}
#endif
else if (0 == strcmp(service, "telnet"))
{
if ('\0' == *str_port)
port = ZBX_DEFAULT_TELNET_PORT;
ret = check_telnet(ip, port, CONFIG_TIMEOUT, &value_int);
}
else
return ret;
if (SYSINFO_RET_OK == ret)
{
if (0 != perf)
{
if (0 != value_int)
{
check_time = zbx_time() - check_time;
check_time = MAX(check_time, 0.0001);
SET_DBL_RESULT(result, check_time);
}
else
SET_DBL_RESULT(result, 0.0);
}
else
SET_UI64_RESULT(result, value_int);
}
return ret;
}
