1 files changed, 76 insertions, 0 deletions
diff --git a/plugins/check_ntp.c b/plugins/check_ntp.c
index 2954aa88..86709a1f 100644
--- a/plugins/check_ntp.c
+++ b/plugins/check_ntp.c
@@ -295,6 +295,82 @@ double offset_request(const char *host){
 	return avg_offset;
 }
 
+
+/* this should behave more like ntpdate, but needs optomisations... */
+double offset_request_ntpdate(const char *host){
+	int i=0, j=0, ga_result=0, num_hosts=0, *socklist=NULL;
+	ntp_message req;
+	double offset=0., avg_offset=0.;
+	struct timeval recv_time;
+	struct addrinfo *ai=NULL, *ai_tmp=NULL, hints;
+
+	/* setup hints to only return results from getaddrinfo that we'd like */
+	memset(&hints, 0, sizeof(struct addrinfo));
+	hints.ai_family = address_family;
+	hints.ai_protocol = IPPROTO_UDP;
+	hints.ai_socktype = SOCK_DGRAM;
+
+	/* XXX better error handling here... */
+	ga_result = getaddrinfo(host, "123", &hints, &ai);
+	if(ga_result!=0){
+		fprintf(stderr, "error getting address for %s: %s\n",
+				host, gai_strerror(ga_result));
+		return -1.0;
+	}
+
+	/* count te number of returned hosts, and allocate an array of sockets */
+	ai_tmp=ai;
+	while(ai_tmp){
+		ai_tmp = ai_tmp->ai_next;
+		num_hosts++;
+	}
+	socklist=(int*)malloc(sizeof(int)*num_hosts);
+	if(socklist==NULL) die(STATE_UNKNOWN, "can not allocate socket array");
+
+	/* setup each socket for writing */
+	ai_tmp=ai;
+	for(i=0;ai_tmp;i++){
+		socklist[i]=socket(ai_tmp->ai_family, SOCK_DGRAM, IPPROTO_UDP);
+		if(socklist[i] == -1) {
+			perror(NULL);
+			die(STATE_UNKNOWN, "can not create new socket");
+		}
+		if(connect(socklist[i], ai_tmp->ai_addr, ai_tmp->ai_addrlen)){
+			die(STATE_UNKNOWN, "can't create socket connection");
+		}
+		ai_tmp = ai_tmp->ai_next;
+	}
+
+	/* now do AVG_NUM checks to each host. this needs to be optimized
+	 * two ways:
+	 *  - use some parellization w/poll for much faster results.  currently
+	 *    we do send/recv, send/recv, etc, whereas we could use poll(), to
+	 *    determine when to read and just do a bunch of writing when we
+	 *    have free time.
+	 *  - behave like ntpdate and only take the 5 best responses.
+	 */
+	for(i=0; i<AVG_NUM; i++){
+		if(verbose) printf("offset calculation run %d/%d\n", i+1, AVG_NUM);
+		for(j=0; j<num_hosts; j++){
+			if(verbose) printf("peer %d: ", j);
+			setup_request(&req);
+			write(socklist[j], &req, sizeof(ntp_message));
+			read(socklist[j], &req, sizeof(ntp_message));
+			gettimeofday(&recv_time, NULL);
+			offset=calc_offset(&req, &recv_time);
+			if(verbose) printf("offset: %g\n", offset);
+			avg_offset+=offset;
+		}
+		avg_offset/=num_hosts;
+	}
+	avg_offset/=AVG_NUM;
+	if(verbose) printf("overall average offset: %g\n", avg_offset);
+
+	for(j=0; j<num_hosts; j++){ close(socklist[j]); }
+	freeaddrinfo(ai);
+	return avg_offset;
+}
+
 void
 setup_control_request(ntp_control_message *p, uint8_t opcode, uint16_t seq){
 	memset(p, 0, sizeof(ntp_control_message));