/*
* set up the listening socket (if applicable)
*/
int net_init(void)
{
int result, sockopt = 1;
struct sockaddr_in sain, inbound;
struct sockaddr *sa = (struct sockaddr *)&sain;
socklen_t addrlen = sizeof(inbound);
if (!num_nodes)
return 0;
sain.sin_addr.s_addr = default_addr;
sain.sin_port = htons(default_port);
sain.sin_family = AF_INET;
net_sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (net_sock < 0)
return -1;
merlin_set_socket_options(net_sock, 0);
/* if this fails we can do nothing but try anyway */
(void)setsockopt(net_sock, SOL_SOCKET, SO_REUSEADDR, &sockopt, sizeof(int));
result = bind(net_sock, sa, addrlen);
if (result < 0)
return -1;
result = listen(net_sock, SOMAXCONN);
if (result < 0)
return -1;
result = iobroker_register(nagios_iobs, net_sock, NULL, net_accept_one);
if (result < 0) {
lerr("IOB: Failed to register network socket with I/O broker: %s", iobroker_strerror(result));
return -1;
}
return 0;
}
void node_set_state(merlin_node *node, int state, const char *reason)
{
int prev_state, add;
if (!node)
return;
if (node->state == state)
return;
if (reason) {
linfo("NODESTATE: %s: %s -> %s: %s", node->name,
node_state_name(node->state), node_state_name(state), reason);
}
/*
* Keep track of active nodes. Setting 'add' to the proper
* value means we needn't bother with an insane chain of
* if()'s later.
* add = +1 if state changes TO 'STATE_CONNECTED'.
* add = -1 if state changes FROM 'STATE_CONNECTED'
* add remains zero for all other cases
*/
if (state == STATE_CONNECTED) {
add = 1;
node->connect_time = time(NULL);
} else if (node->state == STATE_CONNECTED) {
add = -1;
} else {
add = 0;
}
if (node->type == MODE_POLLER)
self.active_pollers += add;
else if (node->type == MODE_PEER)
self.active_peers += add;
else if (node->type == MODE_MASTER)
self.active_masters += add;
prev_state = node->state;
node->state = state;
if (node->state != STATE_CONNECTED && prev_state != STATE_CONNECTED)
return;
if (node->action)
node->action(node, prev_state);
if (node->state == STATE_CONNECTED && node->sock >= 0) {
int snd, rcv;
socklen_t size = sizeof(int);
/* mark this so we can disconnect nodes that never send data */
node->last_recv = time(NULL);
merlin_set_socket_options(node->sock, 224 * 1024);
getsockopt(node->sock, SOL_SOCKET, SO_SNDBUF, &snd, &size);
getsockopt(node->sock, SOL_SOCKET, SO_SNDBUF, &rcv, &size);
ldebug("send / receive buffers are %s / %s for node %s",
human_bytes(snd), human_bytes(rcv), node->name);
}
}
