| 1 |
+ |
/* |
| 2 |
+ |
* i-scream central monitoring system |
| 3 |
+ |
* Copyright (C) 2000-2002 i-scream |
| 4 |
+ |
* |
| 5 |
+ |
* This program is free software; you can redistribute it and/or |
| 6 |
+ |
* modify it under the terms of the GNU General Public License |
| 7 |
+ |
* as published by the Free Software Foundation; either version 2 |
| 8 |
+ |
* of the License, or (at your option) any later version. |
| 9 |
+ |
* |
| 10 |
+ |
* This program is distributed in the hope that it will be useful, |
| 11 |
+ |
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 12 |
+ |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 13 |
+ |
* GNU General Public License for more details. |
| 14 |
+ |
* |
| 15 |
+ |
* You should have received a copy of the GNU General Public License |
| 16 |
+ |
* along with this program; if not, write to the Free Software |
| 17 |
+ |
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. |
| 18 |
+ |
*/ |
| 19 |
+ |
|
| 20 |
|
#include <stdio.h> |
| 2 |
– |
#include <sys/socket.h> |
| 21 |
|
#include <stdlib.h> |
| 22 |
|
#include <unistd.h> |
| 23 |
|
#include <syslog.h> |
| 24 |
|
#include <netinet/in.h> |
| 25 |
|
#include "ukcprog.h" |
| 26 |
|
#include <netdb.h> |
| 27 |
< |
#include <strings.h> |
| 27 |
> |
#include <string.h> |
| 28 |
|
#include "statgrab.h" |
| 29 |
+ |
#include <time.h> |
| 30 |
+ |
#include <sys/socket.h> |
| 31 |
+ |
#include <netinet/in.h> |
| 32 |
+ |
#include <arpa/inet.h> |
| 33 |
|
|
| 34 |
|
#define RECONFIGURE_RETURN_CODE 2 |
| 35 |
|
#define UDP_MAX_PACKET_SIZE 8192 |
| 37 |
|
typedef struct{ |
| 38 |
|
int fm_port; |
| 39 |
|
char *fm_host; |
| 40 |
< |
|
| 40 |
> |
|
| 41 |
> |
char *my_ip; |
| 42 |
|
char *my_fqdn; |
| 43 |
|
char *server_fqdn; |
| 44 |
|
int server_udp_port; |
| 63 |
|
int ihost_configure(ihost_state_t *ihost_state){ |
| 64 |
|
struct sockaddr_in addr; |
| 65 |
|
struct in_addr haddr; |
| 66 |
+ |
struct sockaddr ip; |
| 67 |
+ |
int ip_len; |
| 68 |
|
int sd; |
| 69 |
|
FILE *fm_fd_r, *fm_fd_w; |
| 70 |
|
char *reply; |
| 106 |
|
errf("Failed to open write stream (%m)"); |
| 107 |
|
return -1; |
| 108 |
|
} |
| 109 |
+ |
ip_len=sizeof ip; |
| 110 |
+ |
memset(&ip, 0, ip_len); |
| 111 |
+ |
if((getsockname(sd, &ip, &ip_len)) != 0){ |
| 112 |
+ |
errf("Failed to get IP address (%m)"); |
| 113 |
+ |
return -1; |
| 114 |
+ |
} |
| 115 |
+ |
if (ip.sa_family!=AF_INET){ |
| 116 |
+ |
errf("sa family is wrong type"); |
| 117 |
+ |
return -1; |
| 118 |
+ |
} |
| 119 |
+ |
|
| 120 |
+ |
if((ihost_state->my_ip=inet_ntoa(((struct sockaddr_in *)&ip)->sin_addr))==NULL){ |
| 121 |
+ |
errf("Failed to get IP (%m)"); |
| 122 |
+ |
return -1; |
| 123 |
+ |
} |
| 124 |
|
|
| 125 |
|
reply=sock_comm(fm_fd_r, fm_fd_w, "STARTCONFIG"); |
| 126 |
|
if ((reply==NULL) || (strncasecmp(reply, "OK", 2) != 0) ) { |
| 314 |
|
return -1; |
| 315 |
|
} |
| 316 |
|
|
| 317 |
+ |
fflush(fm_fd_r); |
| 318 |
+ |
fflush(fm_fd_w); |
| 319 |
+ |
|
| 320 |
|
if(fclose(fm_fd_r) !=0){ |
| 321 |
|
errf("Failed to close read FD (%m)"); |
| 322 |
|
return -1; |
| 366 |
|
free(stats[x]); |
| 367 |
|
} |
| 368 |
|
xml_data_p=xml_data; |
| 369 |
< |
xml_data=strf("<packet seq_no=\"%d\" machine_name=\"%s\" date=\"%ld\" type=\"data\" ip=\"%s\" key=\"%s\">%s</packet>", counter, ihost_state->my_fqdn, time(NULL), "127.0.0.1", ihost_state->key, xml_data); |
| 369 |
> |
xml_data=strf("<packet seq_no=\"%d\" machine_name=\"%s\" date=\"%ld\" type=\"data\" ip=\"%s\" key=\"%s\">%s</packet>", counter, ihost_state->my_fqdn, time(NULL), ihost_state->my_ip, ihost_state->key, xml_data); |
| 370 |
|
free(xml_data_p); |
| 371 |
|
|
| 372 |
|
return xml_data; |
| 405 |
|
return -1; |
| 406 |
|
} |
| 407 |
|
|
| 408 |
+ |
close(sd); |
| 409 |
+ |
|
| 410 |
|
return 0; |
| 411 |
|
} |
| 412 |
|
|
| 414 |
|
ihost_state_t ihost_state; |
| 415 |
|
int heartbeat_exit; |
| 416 |
|
int counter=0; |
| 417 |
< |
long udp_time=0, tcp_time=0, cur_time=0; |
| 417 |
> |
long udp_time=0, tcp_time=0, stat_grab_time=0, cur_time=0; |
| 418 |
|
int sleep_delay=0; |
| 419 |
|
char *xml_stats; |
| 420 |
|
|
| 442 |
|
exit(1); |
| 443 |
|
} |
| 444 |
|
|
| 400 |
– |
/* get starting stats */ |
| 401 |
– |
if((heartbeat(&ihost_state)) == -1){ |
| 402 |
– |
errf("inital heartbeat failed (%m)"); |
| 403 |
– |
exit(1); |
| 404 |
– |
} |
| 405 |
– |
|
| 406 |
– |
|
| 407 |
– |
if((xml_stats=stat_grab(&ihost_state, counter)) == NULL){ |
| 408 |
– |
errf("Failed to get stats (%m)"); |
| 409 |
– |
exit(1); |
| 410 |
– |
} |
| 411 |
– |
|
| 412 |
– |
|
| 445 |
|
for(;;){ |
| 446 |
|
cur_time=time(NULL); |
| 447 |
|
if(cur_time>=tcp_time){ |
| 448 |
+ |
/*printf("sending TCP\n");*/ |
| 449 |
|
heartbeat_exit=heartbeat(&ihost_state); |
| 450 |
|
if(heartbeat_exit==RECONFIGURE_RETURN_CODE){ |
| 451 |
< |
errf("heartbeat needs to be reconfigured"); |
| 451 |
> |
/*errf("heartbeat needs to be reconfigured");*/ |
| 452 |
|
ihost_configure(&ihost_state); |
| 453 |
|
/* So udp doesn't wait til next sending before updating */ |
| 454 |
|
udp_time=0; |
| 459 |
|
} |
| 460 |
|
tcp_time=time(NULL)+ihost_state.tcp_update_time; |
| 461 |
|
} |
| 462 |
+ |
|
| 463 |
|
if(cur_time>=udp_time){ |
| 464 |
< |
send_stats(&ihost_state, xml_stats); |
| 465 |
< |
free(xml_stats); |
| 466 |
< |
if((xml_stats=stat_grab(&ihost_state, counter)) == NULL){ |
| 464 |
> |
/*printf("sending UDP\n");*/ |
| 465 |
> |
stat_grab_time=time(NULL); |
| 466 |
> |
if((xml_stats=stat_grab(&ihost_state, counter++)) == NULL){ |
| 467 |
|
errf("Failed to get stats (%m)"); |
| 468 |
|
exit(1); |
| 469 |
|
} |
| 470 |
< |
udp_time=time(NULL)+ihost_state.udp_update_time; |
| 470 |
> |
stat_grab_time=time(NULL)-stat_grab_time; |
| 471 |
> |
send_stats(&ihost_state, xml_stats); |
| 472 |
> |
free(xml_stats); |
| 473 |
> |
udp_time=time(NULL)+ihost_state.udp_update_time-stat_grab_time; |
| 474 |
|
} |
| 475 |
< |
if(tcp_time>udp_time){ |
| 476 |
< |
sleep_delay=time(NULL)-tcp_time; |
| 475 |
> |
|
| 476 |
> |
if(tcp_time<udp_time){ |
| 477 |
> |
sleep_delay=tcp_time-time(NULL); |
| 478 |
|
}else{ |
| 479 |
< |
sleep_delay=time(NULL)-udp_time; |
| 479 |
> |
sleep_delay=udp_time-time(NULL); |
| 480 |
|
} |
| 481 |
< |
sleep(sleep_delay); |
| 481 |
> |
|
| 482 |
> |
/*printf("tcp epoc: %ld \t udp epoc: %ld\ntime:%ld \tsleeping: %d\n", tcp_time, udp_time, time(NULL), sleep_delay);*/ |
| 483 |
> |
if(sleep_delay>0) sleep(sleep_delay); |
| 484 |
|
} |
| 485 |
|
return 0; |
| 486 |
|
} |