int WINAPI WinMain(HINSTANCE, HINSTANCE, LPSTR, int) {
// SHBrowseForFolder needs this, and says it's better to use OleInitialize than ComInitialize
VERIFY_HRESULT(OleInitialize(NULL));
app_check_kill_date();
app_init_message_loop();
app_init_sparkle();
app_init_win32_ui();
mainwnd_init();
os_init(); // must be called before app_init_logging
app_init_logging();
node_init();
mainwnd_show();
win_sparkle_init();
app_init_pings();
app_run_message_loop();
node_shutdown();
OleUninitialize();
return 0;
}
void tcp_gate( ITEM *listItem ) {
TCP_NODE *n = list_value( listItem );
switch( n->pipeline ) {
case NODE_SHUTDOWN:
node_shutdown( listItem );
break;
case NODE_READY:
tcp_rearm( listItem, TCP_INPUT );
break;
case NODE_SEND_INIT:
case NODE_SEND_DATA:
case NODE_SEND_STOP:
tcp_rearm( listItem, TCP_OUTPUT );
break;
default:
fail( "No shit" );
}
}
//-----------------------------------------------------------------------------
// When the SIGINT signal is received, we need to start shutting down the
// service. This means that it needs to:
// 1. Close the listening socket.
// 2. Send a CLOSING message to every connected node.
// 3. if there are any undelivered messages in the queues, we need to return them.
// 4. Shutdown the nodes if we can.
// 5. Shutdown the queues if we can.
// 6. Tell the stats system to close its event as soon as there are no more nodes connected.
void sigint_handler(evutil_socket_t fd, short what, void *arg)
{
system_data_t *sysdata = (system_data_t *) arg;
server_t *server;
stats_t *stats;
queue_t *q;
node_t *node;
controller_t *ct;
logger(sysdata->logging, 3, "SIGINT");
assert(sysdata);
assert(sysdata->servers);
// delete the sigint event, we dont need it anymore.
assert(sysdata->sigint_event);
event_free(sysdata->sigint_event);
sysdata->sigint_event = NULL;
// delete the sighup event, we dont need that either.
assert(sysdata->sighup_event);
event_free(sysdata->sighup_event);
sysdata->sighup_event = NULL;
// delete the sigusr1 event, we dont need that either.
assert(sysdata->sigusr1_event);
event_free(sysdata->sigusr1_event);
sysdata->sigusr1_event = NULL;
// delete the sigusr2 event, we dont need that either.
assert(sysdata->sigusr2_event);
event_free(sysdata->sigusr2_event);
sysdata->sigusr2_event = NULL;
logger(sysdata->logging, 2, "Shutting down servers.");
while ((server = ll_pop_head(sysdata->servers))) {
server_shutdown(server);
server_free(server);
free(server);
}
// All active nodes should be informed that we are shutting down.
logger(sysdata->logging, 2, "Shutting down nodes.");
assert(sysdata->nodelist);
ll_start(sysdata->nodelist);
while ((node = ll_next(sysdata->nodelist))) {
assert(node->handle > 0);
logger(sysdata->logging, 2, "Initiating shutdown of node %d.", node->handle);
node_shutdown(node);
}
ll_finish(sysdata->nodelist);
// Now attempt to shutdown all the queues. Basically, this just means that the queue will close down all the 'waiting' consumers.
logger(sysdata->logging, 2, "Initiating shutdown of queues.");
ll_start(sysdata->queues);
while ((q = ll_next(sysdata->queues))) {
assert(q->name != NULL);
assert(q->qid > 0);
logger(sysdata->logging, 2, "Initiating shutdown of queue %d ('%s').", q->qid, q->name);
queue_shutdown(q);
}
ll_finish(sysdata->queues);
// if we have controllers that are attempting to connect, we need to change their status so that they dont.
logger(sysdata->logging, 2, "Stopping controllers that are connecting.");
ll_start(sysdata->controllers);
while ((ct = ll_next(sysdata->controllers))) {
BIT_SET(ct->flags, FLAG_CONTROLLER_FAILED);
if (ct->connect_event) {
event_free(ct->connect_event);
ct->connect_event = NULL;
}
}
ll_finish(sysdata->controllers);
// Put stats event on notice that we are shutting down, so that as soon as there are no more nodes, it needs to stop its event.
assert(sysdata != NULL);
assert(sysdata->stats != NULL);
stats = sysdata->stats;
assert(stats->shutdown == 0);
stats->shutdown ++;
// Tell the logging system not to use events anymore.
log_direct(sysdata->logging);
}