void * server_io_loop(void * arg) {
int listenfd = -1, connfd = -1;
struct sockaddr_in serv_addr;
//Network socket, TCP streaming protocol, default options
listenfd = socket(AF_INET, SOCK_STREAM, 0);
LOGDEBUG(1, "log_servio listenfd=[%i]\n", listenfd);
//memset(&serv_addr, '0', sizeof(serv_addr));
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
serv_addr.sin_port = htons(LOG_PORT);
bind(listenfd, (struct sockaddr*)&serv_addr, sizeof(serv_addr));
//We don't want to crash the server if the pipe dies...
//signal(SIGPIPE, SIG_IGN);
//Accepting connections on this socket, backqueue of size 1.
listen(listenfd, 1);
LOGDEBUG(2, "log_serverio listening...port=%i\n", LOG_PORT);
nbsf::flags[nbsf::serv_connected] = false;
for (;;) {
connfd = block_accept(listenfd);
log_object_t * obj = NULL;
LOGDEBUG(3, "log_serverio FOUND CLIENT [%i]\n", connfd);
memcpy(nbsf::cio_start, nbsf::total, sizeof(nbsf::io_state_t) * NUM_LOG_BUFFERS);
nbsf::cio_upstart = time(NULL);
nbsf::flags[nbsf::serv_connected] = true;
uint32_t version = htonl(LOG_VERSION);
uint32_t seq_num = 0;
uint32_t recvd;
CHECK_RET(logio::send_exact(connfd, 4, (uint8_t *) &seq_num))
CHECK_RET(logio::recv_exact(connfd, 4, (uint8_t *) &recvd, IO_SEC_TO_BREAK));
if (recvd != 0) {
LOGDEBUG(1, "log_serverio got bad ping initiation: %u\n", ntohl(recvd));
goto connection_died;
}
CHECK_RET(logio::send_exact(connfd, 4, (uint8_t *) &version))
CHECK_RET(logio::recv_exact(connfd, 4, (uint8_t *) &recvd, IO_SEC_TO_BREAK));
LOGDEBUG(1, "log_serverio starting connection; server version: %u, client version: %u\n", ntohl(version), ntohl(recvd));
++seq_num;
LOGDEBUG(1, "log_servio client connection ready...\n");
for (;;) {
if (!(nbsf::flags[nbsf::servio])) {
uint32_t ping = 0;
CHECK_RET(logio::send_exact(connfd, 4, (uint8_t *) &ping))
CHECK_RET(logio::recv_exact(connfd, 4, (uint8_t *) &recvd, IO_SEC_TO_BREAK));
if (recvd != 0) {
LOGDEBUG(1, "log_serverio got bad ping while waiting: %u\n", ntohl(recvd));
goto connection_died;
}
usleep(SERVER_USLEEP_WAITING);
continue;
}
//we're writing.
uint8_t ws = false;
for (int i = 0; i < NUM_LOG_BUFFERS; ++i) {
for (int r = 0; r < LOG_RATIO[i]; ++r) {
if (!(nbsf::flags[nbsf::servio])) {
break;
}
obj = acquire(i, &(log_main->buffers[i]->servio_nextr));
if (obj) {
ws = true;
uint32_t msg_seq_n = htonl(seq_num++);
CHECK_RET(logio::send_exact(connfd, 4, (uint8_t *) &msg_seq_n));
CHECK_RET(logio::send_log(connfd, &(obj->log)));
release(obj, true);
} else {
break;
}
}
}
if (!ws) {
usleep(SERVER_USLEEP_WAITING);
}
}
//Acts like a c exception.
connection_died:
close(connfd);
LOGDEBUG(1, "log_serverio loop broken, connection closed.\n");
nbsf::flags[nbsf::serv_connected] = false;
if (obj)
release(obj, true);
}
return NULL;
}
void * server_io_loop(void * arg) {
int listenfd = -1, connfd = -1;
struct sockaddr_in serv_addr;
//Network socket, TCP streaming protocol, default options
listenfd = socket(AF_INET, SOCK_STREAM, 0);
NBDEBUG( "log_streamio listenfd=[%i]\n", listenfd);
//memset(&serv_addr, '0', sizeof(serv_addr));
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
serv_addr.sin_port = htons(STREAM_PORT);
bind(listenfd, (struct sockaddr*)&serv_addr, sizeof(serv_addr));
//Accepting connections on this socket, no backqueue
listen(listenfd, 0);
NBDEBUG("log_streamio listening...port=%i\n", STREAM_PORT);
control::flags[control::serv_connected] = false;
for (;;) {
connfd = block_accept(listenfd, STREAM_USLEEP_WAITING);
Log * obj = NULL;
int obj_bi = -1;
NBDEBUG("log_streamio FOUND CLIENT [%i]\n", connfd);
memcpy(cio_start, total, sizeof(io_state_t) * NUM_LOG_BUFFERS);
cio_upstart = time(NULL);
control::flags[control::serv_connected] = true;
uint32_t version = htonl(LOG_VERSION);
uint32_t seq_num = 0;
uint32_t recvd;
int ourVersion, thmVersion;
CHECK_RET(send_exact(connfd, 4, (uint8_t *) &seq_num))
CHECK_RET(recv_exact(connfd, 4, (uint8_t *) &recvd, IO_SEC_TO_BREAK));
if (recvd != 0) {
NBDEBUG( "log_streamio got bad ping initiation: %u\n", ntohl(recvd));
goto connection_died;
}
CHECK_RET(send_exact(connfd, 4, (uint8_t *) &version))
CHECK_RET(recv_exact(connfd, 4, (uint8_t *) &recvd, IO_SEC_TO_BREAK));
ourVersion = ntohl(version);
thmVersion = ntohl(recvd);
/*
NBDEBUG( "log_streamio starting connection; server version: %u, client version: %u\n", ourVersion, thmVersion); */
NBDEBUG( "log_streamio starting connection; server version: %u, client version: %u\n", ntohl(version), ntohl(recvd));
++seq_num;
for (;;) {
//we're writing.
uint8_t ws = false;
for (int i = 0; i < NUM_LOG_BUFFERS; ++i) {
obj_bi = i;
for (int r = 0; r < LOG_RATIO[i]; ++r) {
obj = acquire(i, &(log_main.buffers[i].servio_nextr));
if (obj) {
ws = true;
uint32_t msg_seq_n = htonl(seq_num++);
CHECK_RET(send_exact(connfd, 4, (uint8_t *) &msg_seq_n));
CHECK_RET(!obj->send(connfd));
releaseWrapper(i, obj, true);
} else {
break;
}
obj = NULL;
}
}
if (!ws) {
int32_t zero = 0, resp;
CHECK_RET( send_exact(connfd, 4, &zero) );
CHECK_RET( recv_exact(connfd, 4, &resp, IO_SEC_TO_BREAK) );
if (zero != resp) {
goto connection_died;
}
usleep(STREAM_USLEEP_WAITING);
}
}
//Acts like a c exception.
connection_died:
close(connfd);
NBDEBUG( "log_streamio loop broken, connection closed.\n");
control::flags[control::serv_connected] = false;
if (obj)
releaseWrapper(obj_bi, obj, true);
obj = NULL;
}
return NULL;
}
