214 lines
5.4 KiB
C
214 lines
5.4 KiB
C
/*
|
|
* Copyright (c) 2015 Carsten Larsen
|
|
* 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 ``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 THE AUTHOR 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.
|
|
*
|
|
*/
|
|
|
|
#include "ntimed_platform.h"
|
|
#include "ntimed.h"
|
|
#include "ntp.h"
|
|
#include "udp.h"
|
|
|
|
#define ARGSFORMAT "D=DURATION/N,M=MONITOR/K,T=TRACEFILE/K,SERVERS/M/A,SHOW/S,QUIET/S"
|
|
const char *vers = "\0$VER: ntimed-poll" AMIGA_VERSION;
|
|
|
|
struct RDArgs *rdargs = NULL;
|
|
struct udp_socket *usc = NULL;
|
|
struct ntp_peerset *npl = NULL;
|
|
struct todolist *tdl = NULL;
|
|
|
|
struct ntimedargs {
|
|
long *duration;
|
|
char *monitor;
|
|
char *tracefile;
|
|
char **servers;
|
|
long show;
|
|
long quiet;
|
|
};
|
|
|
|
static void init_logic(struct ntimedargs *args);
|
|
static void clean_exit();
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int result = 0;
|
|
struct ntimedargs args = { NULL, NULL, NULL, NULL, FALSE, FALSE };
|
|
|
|
atexit(clean_exit);
|
|
if((result = amiga_open_libs()) != 0) {
|
|
exit(result);
|
|
}
|
|
|
|
rdargs = ReadArgs((STRPTR)ARGSFORMAT, (APTR)&args, NULL);
|
|
if (!rdargs)
|
|
{
|
|
PrintFault(IoErr(), (STRPTR)argv[0]);
|
|
exit(5);
|
|
}
|
|
|
|
if (!args.quiet) {
|
|
EnableDebug();
|
|
|
|
if (args.show) {
|
|
EnableTraceDebug();
|
|
}
|
|
}
|
|
|
|
if (args.tracefile) {
|
|
ArgTracefile(args.tracefile);
|
|
}
|
|
|
|
init_logic(&args);
|
|
|
|
NTP_PeerSet_Poll(NULL, npl, usc, tdl);
|
|
(void)TODO_Run(NULL, tdl);
|
|
|
|
return result;
|
|
}
|
|
|
|
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);
|
|
}
|
|
|
|
static void init_logic(struct ntimedargs *args)
|
|
{
|
|
struct ntp_peer *mon;
|
|
struct ntp_peer *np;
|
|
int npeer;
|
|
int i;
|
|
long duration;
|
|
mon = NULL;
|
|
npeer = 0;
|
|
duration = 1800;
|
|
|
|
if (args->duration) {
|
|
duration = *(args->duration);
|
|
if(duration < 1) {
|
|
Put(NULL, OCX_DEBUG, "Invalid DURATION value.\n");
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
if(args->monitor) {
|
|
mon = NTP_Peer_NewLookup(NULL, args->monitor);
|
|
if (mon == NULL) {
|
|
Put(NULL, OCX_DEBUG, "MONITOR did not resolve.\n");
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
tdl = TODO_NewList();
|
|
Time_Amiga_Passive();
|
|
|
|
npl = NTP_PeerSet_New(NULL);
|
|
AN(npl);
|
|
|
|
for (i = 0; args->servers[i]; i++)
|
|
npeer += NTP_PeerSet_Add(NULL, npl, args->servers[i]);
|
|
|
|
Put(NULL, OCX_TRACE, "# NTIMED Format poll-server 1.0\n");
|
|
Put(NULL, OCX_TRACE, "# Found %d peers\n", npeer);
|
|
|
|
if (npeer == 0) {
|
|
Put(NULL, OCX_DEBUG, "No NTP peers found\n");
|
|
exit(1);
|
|
}
|
|
|
|
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);
|
|
if (usc == NULL) {
|
|
Put(NULL, OCX_DEBUG, "Could not open UDP socket\n");
|
|
exit(1);
|
|
}
|
|
|
|
Put(NULL, OCX_TRACE, "# Duration of run set to %d seconds\n", duration);
|
|
|
|
TODO_ScheduleRel(tdl, mps_end, NULL, duration, 0, "End task");
|
|
|
|
if (mon != NULL)
|
|
TODO_ScheduleRel(tdl, mps_mon, mon, 0, 32, "Monitor");
|
|
|
|
amiga_init_offset();
|
|
}
|
|
|
|
static void clean_exit()
|
|
{
|
|
ArgTracefile(NULL);
|
|
|
|
if(usc) {
|
|
UDP_Socket_Destroy(usc);
|
|
usc = NULL;
|
|
}
|
|
|
|
if (rdargs) {
|
|
FreeArgs(rdargs);
|
|
rdargs = NULL;
|
|
}
|
|
|
|
amiga_close_libs();
|
|
freeall();
|
|
}
|