mirror of
https://github.com/Llewellynvdm/conky.git
synced 2024-10-02 06:59:09 +00:00
696d841e97
This allows compilation under FreeBSD without errors. read_tcpip.cc: Avoid using of SOCK_NONBLOCK argument for socket(). SOCK_NONBLOCK is Linux-specific and its use was substituted with a fcntl() call after socket(). freebsd.cc: Change return type of update_*() to int, as required by common.h. Signed-off-by: Alexander Graf <agraf@znc.in> Signed-off-by: Pavel Labath <pavelo@centrum.sk>
221 lines
6.7 KiB
C++
221 lines
6.7 KiB
C++
/* -*- mode: c++; c-basic-offset: 4; tab-width: 4; indent-tabs-mode: t -*-
|
|
* vim: ts=4 sw=4 noet ai cindent syntax=cpp
|
|
*
|
|
* Conky, a system monitor, based on torsmo
|
|
*
|
|
* Any original torsmo code is licensed under the BSD license
|
|
*
|
|
* All code written since the fork of torsmo is licensed under the GPL
|
|
*
|
|
* Please see COPYING for details
|
|
*
|
|
* Copyright (c) 2004, Hannu Saransaari and Lauri Hakkarainen
|
|
* Copyright (c) 2005-2010 Brenden Matthews, Philip Kovacs, et. al.
|
|
* (see AUTHORS)
|
|
* All rights reserved.
|
|
*
|
|
* 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 3 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, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
#include "conky.h"
|
|
#include "logging.h"
|
|
#include "text_object.h"
|
|
#include <netdb.h>
|
|
#include <stdlib.h>
|
|
#include <string>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <netinet/in.h>
|
|
#include <sys/time.h>
|
|
#include <cinttypes>
|
|
|
|
struct read_tcpip_data {
|
|
char *host;
|
|
unsigned int port;
|
|
};
|
|
|
|
void parse_read_tcpip_arg(struct text_object *obj, const char *arg, void *free_at_crash)
|
|
{
|
|
struct read_tcpip_data *rtd;
|
|
|
|
rtd = (struct read_tcpip_data *) malloc(sizeof(struct read_tcpip_data));
|
|
memset(rtd, 0, sizeof(struct read_tcpip_data));
|
|
|
|
rtd->host = (char *) malloc(text_buffer_size);
|
|
sscanf(arg, "%s", rtd->host);
|
|
sscanf(arg+strlen(rtd->host), "%u", &(rtd->port));
|
|
if(rtd->port == 0) {
|
|
rtd->port = atoi(rtd->host);
|
|
strcpy(rtd->host,"localhost");
|
|
}
|
|
if(rtd->port < 1 || rtd->port > 65535)
|
|
CRIT_ERR(obj, free_at_crash, "read_tcp and read_udp need a port from 1 to 65535 as argument");
|
|
|
|
rtd->port = htons(rtd->port);
|
|
obj->data.opaque = rtd;
|
|
}
|
|
|
|
void parse_tcp_ping_arg(struct text_object *obj, const char *arg, void *free_at_crash)
|
|
{
|
|
#define DEFAULT_TCP_PING_PORT 80
|
|
struct sockaddr_in *addr;
|
|
char *hostname;
|
|
struct hostent* he;
|
|
|
|
addr = (struct sockaddr_in *) malloc(sizeof(struct sockaddr_in));
|
|
obj->data.opaque = addr;
|
|
memset(addr, 0, sizeof(struct sockaddr_in));
|
|
hostname = (char *) malloc(strlen(arg)+1);
|
|
switch( sscanf(arg, "%s %"SCNu16, hostname, &(addr->sin_port)) ) {
|
|
case 1:
|
|
addr->sin_port = DEFAULT_TCP_PING_PORT;
|
|
break;
|
|
case 2:
|
|
break;
|
|
default: //this point should never be reached
|
|
free(hostname);
|
|
CRIT_ERR(obj, free_at_crash, "tcp_ping: Reading arguments failed");
|
|
}
|
|
if(!(he = gethostbyname(hostname))) {
|
|
NORM_ERR("tcp_ping: Problem with resolving '%s', using 'localhost' instead", hostname);
|
|
if(!(he = gethostbyname("localhost"))) {
|
|
free(hostname);
|
|
CRIT_ERR(obj, free_at_crash, "tcp_ping: Resolving 'localhost' also failed");
|
|
}
|
|
}
|
|
free(hostname);
|
|
addr->sin_port = htons(addr->sin_port);
|
|
addr->sin_family = he->h_addrtype;
|
|
memcpy(&(addr->sin_addr), he->h_addr, he->h_length);
|
|
}
|
|
|
|
void print_tcp_ping(struct text_object *obj, char *p, int p_max_size)
|
|
{
|
|
struct timeval tv1, tv2, timeout;
|
|
struct sockaddr_in *addr = (struct sockaddr_in *) obj->data.opaque;
|
|
int addrlen = sizeof(struct sockaddr);
|
|
int sock = socket(addr->sin_family, SOCK_STREAM, IPPROTO_TCP);
|
|
unsigned long long usecdiff;
|
|
fd_set writefds;
|
|
|
|
if(sock != -1) {
|
|
fcntl(sock, F_SETFL, O_NONBLOCK | fcntl(sock, F_GETFL));
|
|
|
|
FD_ZERO(&writefds);
|
|
FD_SET(sock, &writefds);
|
|
#define TCP_PING_TIMEOUT 10
|
|
timeout.tv_sec = (int) TCP_PING_TIMEOUT;
|
|
timeout.tv_usec = (TCP_PING_TIMEOUT - timeout.tv_sec) * 1000000;
|
|
connect(sock, (struct sockaddr*) addr, addrlen); //this will "fail" because sock is non-blocking
|
|
if(errno == EINPROGRESS) { //but EINPROGRESS is only a "false fail"
|
|
gettimeofday(&tv1, 0);
|
|
if(select(sock+1, NULL, &writefds, NULL, &timeout) != -1) {
|
|
gettimeofday(&tv2, 0);
|
|
usecdiff = ((tv2.tv_sec - tv1.tv_sec) * 1000000) + tv2.tv_usec - tv1.tv_usec;
|
|
if(usecdiff <= TCP_PING_TIMEOUT * 1000000) {
|
|
snprintf(p, p_max_size, "%llu", usecdiff);
|
|
} else {
|
|
#define TCP_PING_FAILED "down"
|
|
snprintf(p, p_max_size, TCP_PING_FAILED);
|
|
}
|
|
} else {
|
|
NORM_ERR("tcp_ping: Couldn't wait on the 'pong'");
|
|
}
|
|
} else {
|
|
NORM_ERR("tcp_ping: Couldn't start connection");
|
|
}
|
|
close(sock);
|
|
} else {
|
|
NORM_ERR("tcp_ping: Couldn't create socket");
|
|
}
|
|
}
|
|
|
|
void print_read_tcpip(struct text_object *obj, char *p, int p_max_size, int protocol)
|
|
{
|
|
int sock, received;
|
|
struct sockaddr_in addr;
|
|
struct hostent* he;
|
|
fd_set readfds;
|
|
struct timeval tv;
|
|
struct read_tcpip_data *rtd = (struct read_tcpip_data *) obj->data.opaque;
|
|
ssize_t written; //only used to to suppress warning (gcc wants the returnvalue of write() in a var)
|
|
|
|
if (!rtd)
|
|
return;
|
|
|
|
if (!(he = gethostbyname(rtd->host))) {
|
|
NORM_ERR("%s: Problem with resolving the hostname", protocol == IPPROTO_TCP ? "read_tcp" : "read_udp");
|
|
return;
|
|
}
|
|
if ((sock = socket(he->h_addrtype, protocol == IPPROTO_TCP ? SOCK_STREAM : SOCK_DGRAM, protocol)) == -1) {
|
|
NORM_ERR("%s: Couldn't create a socket", protocol == IPPROTO_TCP ? "read_tcp" : "read_udp");
|
|
return;
|
|
}
|
|
memset(&addr, 0, sizeof(addr));
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_port = rtd->port;
|
|
memcpy(&addr.sin_addr, he->h_addr, he->h_length);
|
|
if (connect(sock, (struct sockaddr*)&addr, sizeof(struct sockaddr)) != 0) {
|
|
if(protocol == IPPROTO_TCP) {
|
|
NORM_ERR("read_tcp: Couldn't create a connection");
|
|
} else {
|
|
NORM_ERR("read_udp: Couldn't listen"); //other error because udp is connectionless
|
|
}
|
|
return;
|
|
}
|
|
if(protocol == IPPROTO_UDP)
|
|
written = write(sock, NULL, 0); //when using udp send a zero-length packet to let the other end know of our existence
|
|
FD_ZERO(&readfds);
|
|
FD_SET(sock, &readfds);
|
|
tv.tv_sec = 1;
|
|
tv.tv_usec = 0;
|
|
if(select(sock + 1, &readfds, NULL, NULL, &tv) > 0){
|
|
received = recv(sock, p, p_max_size, 0);
|
|
if(received != -1) p[received] = 0; else p[0] = 0;
|
|
}
|
|
close(sock);
|
|
}
|
|
|
|
void print_read_tcp(struct text_object *obj, char *p, int p_max_size)
|
|
{
|
|
print_read_tcpip(obj, p, p_max_size, IPPROTO_TCP);
|
|
}
|
|
|
|
void print_read_udp(struct text_object *obj, char *p, int p_max_size)
|
|
{
|
|
print_read_tcpip(obj, p, p_max_size, IPPROTO_UDP);
|
|
}
|
|
|
|
void free_read_tcpip(struct text_object *obj)
|
|
{
|
|
struct read_tcpip_data *rtd = (struct read_tcpip_data *) obj->data.opaque;
|
|
|
|
if (!rtd)
|
|
return;
|
|
|
|
free_and_zero(rtd->host);
|
|
free_and_zero(obj->data.opaque);
|
|
}
|
|
|
|
void free_tcp_ping(struct text_object *obj)
|
|
{
|
|
struct sockaddr_in *addr = (struct sockaddr_in *) obj->data.opaque;
|
|
|
|
if (!addr)
|
|
return;
|
|
|
|
free_and_zero(obj->data.opaque);
|
|
}
|