main_poll_server.c

/*-
 * Copyright (c) 2014 Poul-Henning Kamp
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * poll-server
 *	[-d duration]	When to stop
 *	[-m monitor]	Poll this monitor every 32 seconds
 *	[-t tracefile]	Where to save the output (if not stdout)
 *	server ...	What servers to poll
 *
 */

#include <stdio.h>
#include <stdlib.h>

#include <sys/socket.h>

#include "ntimed.h"
#include "ntp.h"
#include "udp.h"

static struct udp_socket *usc;

static void
mps_filter(struct ocx *ocx, const struct ntp_peer *np)
{
	char buf[256];

	NTP_Tool_Format(buf, sizeof buf, np->rx_pkt);
	Put(ocx, OCX_TRACE, "Poll %s %s %s\n", np->hostname, np->ip, buf);
}

static enum todo_e __match_proto__(todo_f)
mps_mon(struct ocx *ocx, struct todolist *tdl, void *priv)
{
	char buf[256];
	struct ntp_peer *np;
	int i;

	(void)ocx;
	(void)tdl;
	CAST_OBJ_NOTNULL(np, priv, NTP_PEER_MAGIC);
	i = NTP_Peer_Poll(ocx, usc, np, 0.2);
	if (i == 1) {
		NTP_Tool_Format(buf, sizeof buf, np->rx_pkt);
		Put(ocx, OCX_TRACE,
		    "Monitor %s %s %s\n", np->hostname, np->ip, buf);
	} else {
		Put(ocx, OCX_TRACE,
		    "Monitor_err %s %s %d\n", np->hostname, np->ip, i);
	}
	return(TODO_OK);
}

static enum todo_e __match_proto__(todo_f)
mps_end(struct ocx *ocx, struct todolist *tdl, void *priv)
{
	(void)tdl;
	(void)priv;
	Put(ocx, OCX_TRACE, "# Run completed\n");
	return(TODO_FAIL);
}

int
main_poll_server(int argc, char *const *argv)
{
	int ch;
	int npeer = 0;
	char *p;
	struct ntp_peerset *npl;
	struct ntp_peer *mon = NULL;
	struct ntp_peer *np;
	struct todolist *tdl;
	double duration = 1800;

	setbuf(stdout, NULL);
	setbuf(stderr, NULL);

	ArgTracefile("-");

	tdl = TODO_NewList();
	Time_Unix_Passive();

	npl = NTP_PeerSet_New(NULL);
	AN(npl);

	while ((ch = getopt(argc, argv, "d:m:t:")) != -1) {
		switch(ch) {
		case 'd':
			duration = strtod(optarg, &p);
			if (*p != '\0' || duration < 1.0)
				Fail(NULL, 0, "Invalid -d argument");
			break;
		case 'm':
			mon = NTP_Peer_NewLookup(NULL, optarg);
			if (mon == NULL)
				Fail(NULL, 0, "Monitor (-m) didn't resolve.");
			break;
		case 't':
			ArgTracefile(optarg);
			break;
		default:
			Fail(NULL, 0,
			    "Usage %s [-d duration] [-m monitor] "
			    "[-t tracefile] server...", argv[0]);
			break;
		}
	}
	argc -= optind;
	argv += optind;

	for (ch = 0; ch < argc; ch++)
		npeer += NTP_PeerSet_Add(NULL, npl, argv[ch]);
	Put(NULL, OCX_TRACE, "# NTIMED Format poll-server 1.0\n");
	Put(NULL, OCX_TRACE, "# Found %d peers\n", npeer);
	if (npeer == 0)
		Fail(NULL, 0, "No peers found");

	NTP_PeerSet_Foreach(np, npl) {
		Put(NULL, OCX_TRACE, "# Peer %s %s\n", np->hostname, np->ip);
		np->filter_func = mps_filter;
	}

	if (mon != NULL)
		Put(NULL, OCX_TRACE,
		    "# Monitor %s %s\n", mon->hostname, mon->ip);

	usc = UdpTimedSocket(NULL);
	assert(usc != NULL);

	TODO_ScheduleRel(tdl, mps_end, NULL, duration, 0, "End task");

	if (mon != NULL)
		TODO_ScheduleRel(tdl, mps_mon, mon, 0, 32, "Monitor");

	NTP_PeerSet_Poll(NULL, npl, usc, tdl);
	(void)TODO_Run(NULL, tdl);
	return (0);
}