move soures to src subdir

1 files changed, 157 insertions, 0 deletions

diff --git a/src/pingu_ping.c b/src/pingu_ping.c
new file mode 100644
index 0000000..186673e
--- /dev/null
+++ b/src/pingu_ping.c
@@ -0,0 +1,157 @@
+#include <arpa/inet.h>
+#include <net/if.h>
+#include <netinet/in.h>
+#include <netinet/ip.h>
+#include <netinet/ip_icmp.h>
+#include <sys/socket.h>
+#include <sys/types.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include <ev.h>
+
+#include "icmp.h"
+#include "list.h"
+#include "log.h"
+#include "pingu_burst.h"
+#include "pingu_host.h"
+#include "pingu_iface.h"
+#include "pingu_ping.h"
+#include "sockaddr_util.h"
+
+#define PING_SEQ_MAX 32000
+
+static int pingu_ping_get_seq(void)
+{
+	static int seq = 0;
+	seq = (seq + 1) % PING_SEQ_MAX;
+	return seq;
+}
+
+static void pingu_ping_free(struct ev_loop *loop, struct pingu_ping *ping)
+{
+	list_del(&ping->ping_list_entry);
+	ev_timer_stop(loop, &ping->timeout_watcher);
+	free(ping);
+}
+
+static void pingu_ping_verify_and_free(struct ev_loop *loop, struct pingu_ping *ping)
+{
+	pingu_host_verify_status(loop, ping->host);
+	pingu_ping_free(loop, ping);
+}
+
+static void pingu_ping_timeout_cb(struct ev_loop *loop, ev_timer *w,
+				  int revents)
+{
+	struct pingu_ping *ping = container_of(w, struct pingu_ping, timeout_watcher);
+	log_debug("%s: seq %i (%i/%i) timed out", ping->host->label, ping->seq,
+		ping->host->burst.pings_sent, ping->host->max_retries);
+	pingu_ping_verify_and_free(loop, ping);
+}
+
+static struct pingu_ping *pingu_ping_add(struct ev_loop *loop,
+					 struct pingu_host *host, int seq)
+{
+	struct pingu_ping *ping = calloc(1, sizeof(struct pingu_ping));
+	if (ping == NULL)
+		return NULL;
+	ping->seq = seq;
+	ping->host = host;
+	ping->host->burst.pings_sent++;
+	ev_timer_init(&ping->timeout_watcher, pingu_ping_timeout_cb,
+		      host->timeout, 0);
+	ev_timer_start(loop, &ping->timeout_watcher);
+	list_add(&ping->ping_list_entry, &host->iface->ping_list);
+	return ping;
+}
+
+static struct pingu_ping *pingu_ping_find(struct icmphdr *icp,
+					  union sockaddr_any *from,
+					  struct list_head *ping_list)
+{
+	struct pingu_ping *ping;
+	if (icp->type != ICMP_ECHOREPLY || icp->un.echo.id != getpid())
+		return NULL;
+
+	list_for_each_entry(ping, ping_list, ping_list_entry) {
+		if (sockaddr_cmp(&ping->host->burst.saddr, from) == 0
+		    && ping->seq == ntohs(icp->un.echo.sequence))
+			return ping;
+	}
+	return NULL;
+}
+
+static void pingu_ping_handle_reply(struct ev_loop *loop,
+				    struct pingu_ping *ping)
+{
+	ping->host->burst.pings_replied++;
+	log_debug("%s: got seq %i (%i/%i)", ping->host->label, ping->seq,
+		  ping->host->burst.pings_replied,
+		  ping->host->required_replies);
+	pingu_ping_verify_and_free(loop, ping);
+}
+
+int pingu_ping_send(struct ev_loop *loop, struct pingu_host *host,
+		    int set_status_on_failure)
+{
+	int packetlen = sizeof(struct iphdr) + sizeof(struct icmphdr);
+	struct pingu_ping *ping;
+	int seq, r;
+
+	if (!pingu_iface_usable(host->iface))
+		return pingu_host_set_status(host, PINGU_HOST_STATUS_OFFLINE) - 1;
+
+	seq = pingu_ping_get_seq();
+	r = icmp_send_ping(host->iface->fd, &host->burst.saddr.sa,
+			       sizeof(host->burst.saddr), seq, packetlen);
+	if (r < 0) {
+		if (set_status_on_failure)
+			pingu_host_set_status(host, PINGU_HOST_STATUS_OFFLINE);
+		return -1;
+	}
+
+	ping = pingu_ping_add(loop, host, seq);
+	return ping == NULL ? -1 : 0;
+}
+
+void pingu_ping_read_reply(struct ev_loop *loop, struct pingu_iface *iface)
+{
+	union sockaddr_any from;
+	unsigned char buf[1500];
+	struct iphdr *ip = (struct iphdr *) buf;
+	struct pingu_ping *ping;
+
+	int len = icmp_read_reply(iface->fd, &from.sa, sizeof(from), buf,
+				  sizeof(buf));
+	if (len <= 0)
+		return;
+	ping = pingu_ping_find((struct icmphdr *) &buf[ip->ihl * 4], &from,
+				  &iface->ping_list);
+	if (ping == NULL)
+		return;
+
+	pingu_ping_handle_reply(loop, ping);
+}
+
+void pingu_ping_cleanup(struct ev_loop *loop, struct list_head *ping_list)
+{
+	struct pingu_ping *ping, *n;
+	list_for_each_entry_safe(ping, n, ping_list, ping_list_entry) {
+		pingu_ping_free(loop, ping);
+	}
+}
+
+void pingu_ping_dump(int fd, struct list_head *ping_list, const char *prefix)
+{
+	struct pingu_ping *ping;
+	char str[IF_NAMESIZE + 80];
+	list_for_each_entry(ping, ping_list, ping_list_entry) {
+		snprintf(str, sizeof(str), "%s %s %i\n",
+			 prefix, ping->host->host, ping->seq);
+		write(fd, str, strlen(str));
+	}
+}