int connection_msg_to_handler(Connection *conn)
{
Handler *handler = Request_get_action(conn->req, handler);
int rc = 0;
int header_len = Request_header_length(conn->req);
// body_len will include \0
int body_len = Request_content_length(conn->req);
check(handler, "JSON request doesn't match any handler: %s",
bdata(Request_path(conn->req)));
if(pattern_match(IOBuf_start(conn->iob), header_len + body_len, bdata(&PING_PATTERN))) {
Register_ping(IOBuf_fd(conn->iob));
} else {
check(body_len >= 0, "Parsing error, body length ended up being: %d", body_len);
bstring payload = Request_to_payload(conn->req, handler->send_ident,
IOBuf_fd(conn->iob), IOBuf_start(conn->iob) + header_len,
body_len - 1); // drop \0 on payloads
rc = Handler_deliver(handler->send_socket, bdata(payload), blength(payload));
bdestroy(payload);
check(rc == 0, "Failed to deliver to handler: %s", bdata(Request_path(conn->req)));
}
// consumes \0 from body_len
IOBuf_read_commit(conn->iob, header_len + body_len);
return REQ_SENT;
error:
return CLOSE;
}
void Connection_task(void *v)
{
Connection *conn = (Connection *)v;
int i = 0;
int next = OPEN;
State_init(&conn->state, &CONN_ACTIONS);
while(1) {
if(Filter_activated()) {
next = Filter_run(next, conn);
check(next >= CLOSE && next < EVENT_END,
"!!! Invalid next event[%d]: %d from filter!", i, next);
}
if(next == CLOSE) break;
next = State_exec(&conn->state, next, (void *)conn);
check(next >= CLOSE && next < EVENT_END,
"!!! Invalid next event[%d]: %d, Tell ZED!", i, next);
if(conn->iob && !conn->iob->closed) {
Register_ping(IOBuf_fd(conn->iob));
}
i++;
}
error: // fallthrough
State_exec(&conn->state, CLOSE, (void *)conn);
Connection_destroy(conn);
taskexit(0);
}
static int handler_process_control_request(Connection *conn, tns_value_t *data)
{
tns_value_t *args = darray_get(data->value.list, 1);
check(args->type==tns_tag_dict, "Invalid control response: not a dict.");
hnode_t *n = hash_lookup(args->value.dict, &KEEP_ALIVE);
if(n != NULL) {
Register_ping(IOBuf_fd(conn->iob));
}
n = hash_lookup(args->value.dict, &CREDITS);
if(n != NULL) {
tns_value_t *credits = (tns_value_t *)hnode_get(n);
conn->sendCredits += credits->value.number;
taskwakeup(&conn->uploadRendez);
}
n = hash_lookup(args->value.dict, &CANCEL);
if(n != NULL && !conn->closing) {
Register_disconnect(IOBuf_fd(conn->iob));
taskwakeup(&conn->uploadRendez);
}
tns_value_destroy(data);
return 0;
error:
return -1;
}
int connection_msg_to_handler(Connection *conn)
{
Handler *handler = Request_get_action(conn->req, handler);
int rc = 0;
int header_len = Request_header_length(conn->req);
// body_len will include \0
int body_len = Request_content_length(conn->req);
check(handler, "JSON request doesn't match any handler: %s",
bdata(Request_path(conn->req)));
if(pattern_match(IOBuf_start(conn->iob), header_len + body_len, bdata(&PING_PATTERN))) {
Register_ping(IOBuf_fd(conn->iob));
} else {
check(body_len >= 0, "Parsing error, body length ended up being: %d", body_len);
bstring payload = NULL;
if(handler->protocol == HANDLER_PROTO_TNET) {
payload = Request_to_tnetstring(conn->req, handler->send_ident,
IOBuf_fd(conn->iob), IOBuf_start(conn->iob) + header_len,
body_len - 1); // drop \0 on payloads
} else if(handler->protocol == HANDLER_PROTO_JSON) {
payload = Request_to_payload(conn->req, handler->send_ident,
IOBuf_fd(conn->iob), IOBuf_start(conn->iob) + header_len,
body_len - 1); // drop \0 on payloads
} else {
sentinel("Invalid protocol type: %d", handler->protocol);
}
debug("SENT: %s", bdata(payload));
check(payload != NULL, "Failed to generate payload.");
check(handler->send_socket != NULL, "Handler socket is NULL, tell Zed.");
rc = Handler_deliver(handler->send_socket, bdata(payload), blength(payload));
free(payload);
check(rc == 0, "Failed to deliver to handler: %s", bdata(Request_path(conn->req)));
}
// consumes \0 from body_len
check(IOBuf_read_commit(conn->iob, header_len + body_len) != -1, "Final commit failed.");
return REQ_SENT;
error:
return CLOSE;
}
