int _wait(void *socket, int fd, int rw)
{
startfdtask();
check(socket != NULL || fd >= 0, "Attempt to wait on a dead socket/fd: %p or %d", socket, fd);
int max = 0;
int hot_add = SuperPoll_active_hot(POLL) < SuperPoll_max_hot(POLL);
int was_registered = 0;
if(socket != NULL) {
taskstate(rw == 'r' ? "read handler" : "write handler");
} else {
was_registered = Register_fd_exists(fd) != NULL;
taskstate(rw == 'r' ? "read fd" : "write fd");
}
max = SuperPoll_add(POLL, (void *)taskrunning, socket, fd, rw, hot_add);
check(max != -1, "Error adding fd: %d or socket: %p to task wait list.", fd, socket);
taskswitch();
if(task_was_signaled()) {
debug("GOT SIGNAL %d AFTER WAIT", taskrunning->signal);
SuperPoll_del(POLL, socket, fd, hot_add);
return -1;
} if(was_registered && Register_fd_exists(fd) == NULL) {
debug("Socket %d was closed after a wait.", fd);
return -1;
} else {
return 0;
}
error:
return -1;
}
void Handler_task(void *v)
{
int rc = 0;
int i = 0;
Handler *handler = (Handler *)v;
HandlerParser *parser = NULL;
int max_targets = Setting_get_int("limits.handler_targets", 128);
log_info("MAX allowing limits.handler_targets=%d", max_targets);
parser = HandlerParser_create(max_targets);
check_mem(parser);
check(Handler_setup(handler) == 0, "Failed to initialize handler, exiting.");
while(handler->running && !task_was_signaled()) {
taskstate("delivering");
rc = handler_recv_parse(handler, parser);
if(task_was_signaled()) {
log_warn("Handler task signaled, exiting.");
break;
} else if( rc == -1 || parser->target_count <= 0) {
log_warn("Skipped invalid message from handler: %s", bdata(handler->send_spec));
taskdelay(100);
continue;
} else {
for(i = 0; i < (int)parser->target_count; i++) {
int id = (int)parser->targets[i];
int fd = Register_fd_for_id(id);
Connection *conn = fd == -1 ? NULL : Register_fd_exists(fd);
// don't bother calling process request if there's nothing to handle
if(conn && fd >= 0) {
handler_process_request(handler, id, fd, conn, parser->body);
} else {
// TODO: I believe we need to notify the connection that it is dead too
Handler_notify_leave(handler, id);
}
}
}
HandlerParser_reset(parser);
}
handler->running = 0;
handler->task = NULL;
HandlerParser_destroy(parser);
taskexit(0);
error:
log_err("HANDLER TASK DIED: %s", bdata(handler->send_spec));
handler->running = 0;
handler->task = NULL;
HandlerParser_destroy(parser);
taskexit(1);
}
void Handler_task(void *v)
{
int rc = 0;
int i = 0;
Handler *handler = (Handler *)v;
HandlerParser *parser = NULL;
int max_targets = Setting_get_int("limits.handler_targets", 128);
log_info("MAX allowing limits.handler_targets=%d", 128);
parser = HandlerParser_create(max_targets);
check_mem(parser);
check(Handler_setup(handler) == 0, "Failed to initialize handler, exiting.");
while(handler->running) {
taskstate("delivering");
rc = handler_recv_parse(handler, parser);
if(rc != -1 && parser->target_count > 0) {
for(i = 0; i < (int)parser->target_count; i++) {
int id = (int)parser->targets[i];
int fd = Register_fd_for_id(id);
Connection *conn = (Connection *)Register_fd_exists(fd);
handler_process_request(handler, id, fd, conn, parser->body);
}
} else {
debug("Skipped invalid message from handler: %s", bdata(handler->send_spec));
}
HandlerParser_reset(parser);
}
HandlerParser_destroy(parser);
debug("HANDLER EXITED.");
taskexit(0);
error:
HandlerParser_destroy(parser);
log_err("HANDLER TASK DIED");
taskexit(1);
}
char *test_IOBuf_send_operations()
{
Connection *conn = fake_conn("/dev/null", O_WRONLY);
mu_assert(conn != NULL, "Failed to allocate buf.");
IOBuf *buf = conn->iob;
mu_assert(Register_fd_exists(IOBuf_fd(buf)) != NULL, "Damn fd isn't registered.");
int rc = IOBuf_send(buf, "012345789", 10);
mu_assert(!IOBuf_closed(buf), "Should not be closed.");
mu_assert(rc == 10, "Should have sent 10 bytes.");
fdclose(IOBuf_fd(buf));
rc = IOBuf_send(buf, "012345789", 10);
mu_assert(IOBuf_closed(buf), "Should be closed.");
mu_assert(rc == -1, "Should send nothing.");
fake_conn_close(conn);
return NULL;
}