the upcoming 2.0 version
https://redmine.lighttpd.net/projects/lighttpd2
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
712 lines
21 KiB
712 lines
21 KiB
|
|
#include <lighttpd/utils.h> |
|
#include <lighttpd/angel_data.h> |
|
#include <lighttpd/angel_connection.h> |
|
|
|
#define ANGEL_MAGIC ((gint32) 0x8a930a9f) |
|
|
|
static void close_fd_array(GArray *fds); |
|
|
|
typedef enum { |
|
ANGEL_CALL_SEND_SIMPLE = 1, |
|
ANGEL_CALL_SEND_CALL = 2, |
|
ANGEL_CALL_SEND_RESULT = 3 |
|
} angel_call_send_t; |
|
|
|
typedef struct { |
|
enum { ANGEL_CONNECTION_ITEM_GSTRING, ANGEL_CONNECTION_ITEM_FDS } type; |
|
union { |
|
struct { |
|
GString *buf; |
|
guint pos; |
|
} string; |
|
struct { |
|
GArray *fds; |
|
guint pos; |
|
} fds; |
|
} value; |
|
} angel_connection_send_item_t; |
|
|
|
static void send_queue_push_string(GQueue *queue, GString *buf) { |
|
angel_connection_send_item_t *i; |
|
if (!buf || !buf->len) return; |
|
|
|
i = g_slice_new0(angel_connection_send_item_t); |
|
i->type = ANGEL_CONNECTION_ITEM_GSTRING; |
|
i->value.string.buf = buf; |
|
i->value.string.pos = 0; |
|
|
|
g_queue_push_tail(queue, i); |
|
} |
|
|
|
static void send_queue_push_fds(GQueue *queue, GArray *fds) { |
|
angel_connection_send_item_t *i; |
|
if (!fds || !fds->len) return; |
|
|
|
i = g_slice_new0(angel_connection_send_item_t); |
|
i->type = ANGEL_CONNECTION_ITEM_FDS; |
|
i->value.fds.fds = fds; |
|
i->value.fds.pos = 0; |
|
|
|
g_queue_push_tail(queue, i); |
|
} |
|
|
|
static void send_queue_item_free(angel_connection_send_item_t *i) { |
|
if (!i) return; |
|
switch (i->type) { |
|
case ANGEL_CONNECTION_ITEM_GSTRING: |
|
g_string_free(i->value.string.buf, TRUE); |
|
break; |
|
case ANGEL_CONNECTION_ITEM_FDS: |
|
close_fd_array(i->value.fds.fds); |
|
g_array_free(i->value.fds.fds, TRUE); |
|
break; |
|
} |
|
g_slice_free(angel_connection_send_item_t, i); |
|
} |
|
|
|
static void send_queue_clean(GQueue *queue) { |
|
angel_connection_send_item_t *i; |
|
while (NULL != (i = g_queue_peek_head(queue))) { |
|
switch (i->type) { |
|
case ANGEL_CONNECTION_ITEM_GSTRING: |
|
if (i->value.string.pos < i->value.string.buf->len) return; |
|
g_string_free(i->value.string.buf, TRUE); |
|
break; |
|
case ANGEL_CONNECTION_ITEM_FDS: |
|
if (i->value.fds.pos < i->value.fds.fds->len) return; |
|
close_fd_array(i->value.fds.fds); |
|
g_array_free(i->value.fds.fds, TRUE); |
|
break; |
|
} |
|
g_queue_peek_head(queue); |
|
g_slice_free(angel_connection_send_item_t, i); |
|
} |
|
} |
|
|
|
GQuark li_angel_call_error_quark() { |
|
return g_quark_from_static_string("angel-call-error-quark"); |
|
} |
|
GQuark li_angel_connection_error_quark() { |
|
return g_quark_from_static_string("angel-connection-error-quark"); |
|
} |
|
|
|
static gboolean angel_fill_buffer(liAngelConnection *acon, guint need, GError **err) { |
|
gsize old_len; |
|
ssize_t want, r; |
|
if (acon->in.pos > 0) { |
|
g_string_erase(acon->in.data, 0, acon->in.pos); |
|
acon->in.pos = 0; |
|
} |
|
if (acon->in.data->len >= need) return TRUE; |
|
|
|
want = need - acon->in.data->len; /* always > 0 */ |
|
old_len = acon->in.data->len; |
|
g_string_set_size(acon->in.data, need); |
|
for ( ; want > 0; ) { |
|
r = read(acon->fd, acon->in.data->str + old_len, want); |
|
if (r < 0) { |
|
switch (errno) { |
|
case EINTR: |
|
continue; |
|
case EAGAIN: |
|
#if EWOULDBLOCK != EAGAIN |
|
case EWOULDBLOCK: |
|
#endif |
|
g_string_set_size(acon->in.data, old_len); |
|
return TRUE; |
|
case ECONNRESET: |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_RESET, "eof"); |
|
g_string_set_size(acon->in.data, old_len); |
|
return FALSE; |
|
default: |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_CLOSED, |
|
"read error: %s", g_strerror(errno)); |
|
g_string_set_size(acon->in.data, old_len); |
|
return FALSE; |
|
} |
|
} else if (r == 0) { /* eof */ |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_RESET, "eof"); |
|
g_string_set_size(acon->in.data, old_len); |
|
return FALSE; |
|
} else { |
|
want -= r; |
|
old_len += r; |
|
} |
|
} |
|
|
|
g_string_set_size(acon->in.data, old_len); |
|
return TRUE; |
|
} |
|
|
|
static void close_fd_array(GArray *fds) { |
|
guint i; |
|
for (i = 0; i < fds->len; i++) { |
|
close(g_array_index(fds, int, i)); |
|
} |
|
g_array_set_size(fds, 0); |
|
} |
|
|
|
static gboolean angel_dispatch(liAngelConnection *acon, GError **err) { |
|
gint32 id = acon->parse.id, type = acon->parse.type; |
|
liAngelCall *call = NULL; |
|
liAngelCallCB cb = NULL; |
|
gpointer ctx = NULL; |
|
|
|
switch (type) { |
|
case ANGEL_CALL_SEND_SIMPLE: |
|
if (-1 != id) { |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_INVALID_DATA, |
|
"Invalid id: %i, should be -1 for simple call", (gint) id); |
|
close_fd_array(acon->parse.fds); |
|
return FALSE; |
|
} |
|
|
|
if (acon->parse.error->len > 0 || acon->parse.fds->len > 0) { |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_INVALID_DATA, |
|
"Wrong data in call"); |
|
close_fd_array(acon->parse.fds); |
|
return FALSE; |
|
} |
|
acon->recv_call(acon, GSTR_LEN(acon->parse.mod), GSTR_LEN(acon->parse.action), |
|
id, acon->parse.data); |
|
break; |
|
case ANGEL_CALL_SEND_CALL: |
|
if (-1 == id) { |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_INVALID_DATA, |
|
"Invalid id: -1, should be >= 0 for call"); |
|
close_fd_array(acon->parse.fds); |
|
return FALSE; |
|
} |
|
|
|
if (acon->parse.error->len > 0 || acon->parse.fds->len > 0) { |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_INVALID_DATA, |
|
"Wrong data in call"); |
|
close_fd_array(acon->parse.fds); |
|
return FALSE; |
|
} |
|
acon->recv_call(acon, GSTR_LEN(acon->parse.mod), GSTR_LEN(acon->parse.action), |
|
id, acon->parse.data); |
|
break; |
|
case ANGEL_CALL_SEND_RESULT: |
|
g_mutex_lock(acon->mutex); |
|
if (!li_idlist_is_used(acon->call_id_list, id)) { |
|
g_mutex_unlock(acon->mutex); |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_INVALID_DATA, |
|
"Invalid id: %i", (gint) id); |
|
close_fd_array(acon->parse.fds); |
|
return FALSE; |
|
} |
|
li_idlist_put(acon->call_id_list, id); |
|
if (type == ANGEL_CALL_SEND_RESULT && (guint) id < acon->call_table->len) { |
|
call = (liAngelCall*) g_ptr_array_index(acon->call_table, id); |
|
g_ptr_array_index(acon->call_table, id) = NULL; |
|
if (call) { |
|
call->id = -1; |
|
ev_timer_stop(acon->loop, &call->timeout_watcher); |
|
ctx = call->context; |
|
if (NULL == (cb = call->callback)) { |
|
g_slice_free(liAngelCall, call); |
|
} |
|
} |
|
} |
|
g_mutex_unlock(acon->mutex); |
|
|
|
if (cb) { |
|
cb(call, ctx, FALSE, acon->parse.error, acon->parse.data, acon->parse.fds); |
|
} |
|
close_fd_array(acon->parse.fds); |
|
break; |
|
default: |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_INVALID_DATA, |
|
"Invalid type: %i", (gint) type); |
|
close_fd_array(acon->parse.fds); |
|
return FALSE; |
|
} |
|
|
|
return TRUE; |
|
} |
|
|
|
static gboolean angel_connection_read(liAngelConnection *acon, GError **err) { |
|
for ( ;; ) { |
|
if (!acon->parse.have_header) { |
|
gint32 magic; |
|
if (!angel_fill_buffer(acon, 8*4, err)) return FALSE; |
|
if (acon->in.data->len - acon->in.pos < 8*4) return TRUE; /* need more data */ |
|
|
|
if (!li_angel_data_read_int32(&acon->in, &magic, err)) return FALSE; |
|
if (!li_angel_data_read_int32(&acon->in, &acon->parse.type, err)) return FALSE; |
|
if (!li_angel_data_read_int32(&acon->in, &acon->parse.id, err)) return FALSE; |
|
if (!li_angel_data_read_int32(&acon->in, &acon->parse.mod_len, err)) return FALSE; |
|
if (!li_angel_data_read_int32(&acon->in, &acon->parse.action_len, err)) return FALSE; |
|
if (!li_angel_data_read_int32(&acon->in, &acon->parse.error_len, err)) return FALSE; |
|
if (!li_angel_data_read_int32(&acon->in, &acon->parse.data_len, err)) return FALSE; |
|
if (!li_angel_data_read_int32(&acon->in, &acon->parse.missing_fds, err)) return FALSE; |
|
|
|
if (ANGEL_MAGIC != magic) { |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_INVALID_DATA, |
|
"Invalid magic: 0x%x (should be 0x%x)", (gint) magic, (gint) ANGEL_MAGIC); |
|
return FALSE; |
|
} |
|
|
|
acon->parse.body_size = acon->parse.mod_len + acon->parse.action_len + acon->parse.error_len + acon->parse.data_len; |
|
acon->parse.have_header = TRUE; |
|
} |
|
|
|
if (!angel_fill_buffer(acon, acon->parse.body_size, err)) return FALSE; |
|
if (acon->in.data->len - acon->in.pos < acon->parse.body_size) return TRUE; /* need more data */ |
|
while (acon->parse.missing_fds > 0) { |
|
int fd = -1; |
|
switch (li_receive_fd(acon->fd, &fd)) { |
|
case 0: |
|
g_array_append_val(acon->parse.fds, fd); |
|
acon->parse.missing_fds--; |
|
break; |
|
case -1: |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_CLOSED, |
|
"receive fd error: %s", g_strerror(errno)); |
|
return FALSE; |
|
case -2: |
|
return TRUE; /* need more data */ |
|
} |
|
} |
|
|
|
acon->parse.have_header = FALSE; |
|
if (!li_angel_data_read_mem(&acon->in, &acon->parse.mod, acon->parse.mod_len, err)) return FALSE; |
|
if (!li_angel_data_read_mem(&acon->in, &acon->parse.action, acon->parse.action_len, err)) return FALSE; |
|
if (!li_angel_data_read_mem(&acon->in, &acon->parse.error, acon->parse.error_len, err)) return FALSE; |
|
if (!li_angel_data_read_mem(&acon->in, &acon->parse.data, acon->parse.data_len, err)) return FALSE; |
|
|
|
if (!angel_dispatch(acon, err)) return FALSE; |
|
|
|
g_string_truncate(acon->parse.error, 0); |
|
g_string_truncate(acon->parse.data, 0); |
|
g_array_set_size(acon->parse.fds, 0); |
|
} |
|
} |
|
|
|
static void angel_connection_io_cb(struct ev_loop *loop, ev_io *w, int revents) { |
|
liAngelConnection *acon = (liAngelConnection*) w->data; |
|
|
|
if (revents | EV_WRITE) { |
|
GString *out_str; |
|
int i; |
|
ssize_t written, len; |
|
gchar *data; |
|
gboolean out_queue_empty; |
|
angel_connection_send_item_t *send_item; |
|
|
|
g_mutex_lock(acon->mutex); |
|
send_item = g_queue_peek_head(acon->out); |
|
g_mutex_unlock(acon->mutex); |
|
|
|
for (i = 0; send_item && (i < 10); i++) { /* don't send more than 10 chunks */ |
|
switch (send_item->type) { |
|
case ANGEL_CONNECTION_ITEM_GSTRING: |
|
out_str = send_item->value.string.buf; |
|
written = send_item->value.string.pos; |
|
data = out_str->str + written; |
|
len = out_str->len - written; |
|
while (len > 0) { |
|
written = write(w->fd, data, len); |
|
if (written < 0) { |
|
switch (errno) { |
|
case EINTR: |
|
continue; |
|
case EAGAIN: |
|
#if EWOULDBLOCK != EAGAIN |
|
case EWOULDBLOCK: |
|
#endif |
|
goto write_eagain; |
|
default: /* Fatal error, connection has to be closed */ |
|
li_ev_safe_ref_and_stop(ev_async_stop, loop, &acon->out_notify_watcher); |
|
li_ev_safe_ref_and_stop(ev_io_stop, loop, &acon->fd_watcher); |
|
acon->close_cb(acon, NULL); /* TODO: set err */ |
|
return; |
|
} |
|
} else { |
|
data += written; |
|
len -= written; |
|
send_item->value.string.pos += written; |
|
} |
|
} |
|
break; |
|
|
|
case ANGEL_CONNECTION_ITEM_FDS: |
|
while (send_item->value.fds.pos < send_item->value.fds.fds->len) { |
|
switch (li_send_fd(w->fd, g_array_index(send_item->value.fds.fds, int, send_item->value.fds.pos))) { |
|
case 0: |
|
send_item->value.fds.pos++; |
|
continue; |
|
case -1: /* Fatal error, connection has to be closed */ |
|
li_ev_safe_ref_and_stop(ev_async_stop, loop, &acon->out_notify_watcher); |
|
li_ev_safe_ref_and_stop(ev_io_stop, loop, &acon->fd_watcher); |
|
acon->close_cb(acon, NULL); /* TODO: set err */ |
|
return; |
|
case -2: goto write_eagain; |
|
} |
|
} |
|
break; |
|
} |
|
|
|
send_queue_item_free(send_item); |
|
|
|
g_mutex_lock(acon->mutex); |
|
g_queue_pop_head(acon->out); |
|
send_item = g_queue_peek_head(acon->out); |
|
g_mutex_unlock(acon->mutex); |
|
} |
|
|
|
write_eagain: |
|
g_mutex_lock(acon->mutex); |
|
send_queue_clean(acon->out); |
|
out_queue_empty = (0 == acon->out->length); |
|
g_mutex_unlock(acon->mutex); |
|
|
|
if (out_queue_empty) li_ev_io_rem_events(loop, w, EV_WRITE); |
|
} |
|
|
|
if (revents | EV_READ) { |
|
GError *err = NULL; |
|
if (!angel_connection_read(acon, &err)) { |
|
li_ev_safe_ref_and_stop(ev_async_stop, loop, &acon->out_notify_watcher); |
|
li_ev_safe_ref_and_stop(ev_io_stop, loop, &acon->fd_watcher); |
|
acon->close_cb(acon, err); |
|
} |
|
} |
|
} |
|
|
|
static void angel_connection_out_notify_cb(struct ev_loop *loop, ev_async *w, int revents) { |
|
liAngelConnection *acon = (liAngelConnection*) w->data; |
|
UNUSED(revents); |
|
li_ev_io_add_events(loop, &acon->fd_watcher, EV_WRITE); |
|
} |
|
|
|
/* create connection */ |
|
liAngelConnection* li_angel_connection_new(struct ev_loop *loop, int fd, gpointer data, |
|
liAngelReceiveCallCB recv_call, liAngelCloseCB close_cb) { |
|
liAngelConnection *acon = g_slice_new0(liAngelConnection); |
|
|
|
acon->data = data; |
|
acon->mutex = g_mutex_new(); |
|
acon->loop = loop; |
|
acon->fd = fd; |
|
acon->call_id_list = li_idlist_new(65535); |
|
acon->call_table = g_ptr_array_new(); |
|
|
|
ev_io_init(&acon->fd_watcher, angel_connection_io_cb, fd, EV_READ); |
|
ev_io_start(acon->loop, &acon->fd_watcher); |
|
acon->fd_watcher.data = acon; |
|
ev_unref(acon->loop); /* this watcher shouldn't keep the loop alive */ |
|
|
|
ev_async_init(&acon->out_notify_watcher, angel_connection_out_notify_cb); |
|
ev_async_start(acon->loop, &acon->out_notify_watcher); |
|
acon->out_notify_watcher.data = acon; |
|
ev_unref(acon->loop); /* this watcher shouldn't keep the loop alive */ |
|
|
|
acon->out = g_queue_new(); |
|
acon->in.data = g_string_sized_new(1024); |
|
acon->in.pos = 0; |
|
|
|
acon->parse.mod = g_string_sized_new(0); |
|
acon->parse.action = g_string_sized_new(0); |
|
acon->parse.error = g_string_sized_new(0); |
|
acon->parse.data = g_string_sized_new(0); |
|
acon->parse.fds = g_array_new(FALSE, FALSE, sizeof(int)); |
|
|
|
acon->recv_call = recv_call; |
|
acon->close_cb = close_cb; |
|
|
|
return acon; |
|
} |
|
|
|
void li_angel_connection_free(liAngelConnection *acon) { |
|
angel_connection_send_item_t *send_item; |
|
guint i; |
|
|
|
if (!acon) return; |
|
|
|
close(acon->fd); |
|
acon->fd = -1; |
|
|
|
for (i = 0; i < acon->call_table->len; i++) { |
|
liAngelCall *acall = g_ptr_array_index(acon->call_table, i); |
|
liAngelCallCB cb; |
|
if (!acall) continue; |
|
g_ptr_array_index(acon->call_table, i) = NULL; |
|
|
|
cb = acall->callback; |
|
ev_timer_stop(acon->loop, &acall->timeout_watcher); |
|
if (cb) { |
|
cb(acall, acall->context, TRUE, NULL, NULL, NULL); |
|
} else { |
|
g_slice_free(liAngelCall, acall); |
|
} |
|
} |
|
g_ptr_array_free(acon->call_table, TRUE); |
|
|
|
g_mutex_free(acon->mutex); |
|
acon->mutex = NULL; |
|
|
|
li_ev_safe_ref_and_stop(ev_async_stop, acon->loop, &acon->out_notify_watcher); |
|
li_ev_safe_ref_and_stop(ev_io_stop, acon->loop, &acon->fd_watcher); |
|
|
|
li_idlist_free(acon->call_id_list); |
|
while (NULL != (send_item = g_queue_pop_head(acon->out))) { |
|
send_queue_item_free(send_item); |
|
} |
|
g_queue_free(acon->out); |
|
g_string_free(acon->in.data, TRUE); |
|
|
|
g_string_free(acon->parse.mod, TRUE); |
|
g_string_free(acon->parse.action, TRUE); |
|
g_string_free(acon->parse.error, TRUE); |
|
g_string_free(acon->parse.data, TRUE); |
|
close_fd_array(acon->parse.fds); |
|
g_array_free(acon->parse.fds, TRUE); |
|
/* TODO */ |
|
|
|
g_slice_free(liAngelConnection, acon); |
|
} |
|
|
|
static void angel_call_timeout_cb(struct ev_loop *loop, ev_timer *w, int revents) { |
|
liAngelCall* call = (liAngelCall*) w->data; |
|
liAngelConnection *acon = call->acon; |
|
liAngelCallCB cb = NULL; |
|
gpointer ctx; |
|
UNUSED(loop); UNUSED(revents); |
|
|
|
g_mutex_lock(acon->mutex); |
|
g_ptr_array_index(acon->call_table, call->id) = NULL; |
|
if (NULL == (cb = call->callback)) { |
|
g_slice_free(liAngelCall, call); |
|
} |
|
ctx = call->context; |
|
g_mutex_unlock(acon->mutex); |
|
|
|
if (cb) cb(call, ctx, TRUE, NULL, NULL, NULL); |
|
} |
|
|
|
liAngelCall *li_angel_call_new(liAngelCallCB callback, ev_tstamp timeout) { |
|
liAngelCall* call = g_slice_new0(liAngelCall); |
|
|
|
g_assert(NULL != callback); |
|
call->callback = callback; |
|
ev_timer_init(&call->timeout_watcher, angel_call_timeout_cb, timeout, 0); |
|
call->timeout_watcher.data = call; |
|
call->id = -1; |
|
|
|
return call; |
|
} |
|
|
|
/* returns TRUE if a call was cancelled */ |
|
gboolean li_angel_call_free(liAngelCall *call) { |
|
gboolean r = FALSE; |
|
|
|
if (call->acon) { |
|
liAngelConnection *acon = call->acon; |
|
g_mutex_lock(acon->mutex); |
|
if (-1 != call->id) { |
|
r = TRUE; |
|
call->callback = NULL; |
|
} else { |
|
g_slice_free(liAngelCall, call); |
|
} |
|
g_mutex_unlock(acon->mutex); |
|
} else { |
|
g_slice_free(liAngelCall, call); |
|
} |
|
|
|
return r; |
|
} |
|
|
|
static gboolean prepare_call_header(GString **pbuf, |
|
gint32 type, gint32 id, |
|
const gchar *mod, gsize mod_len, const gchar *action, gsize action_len, |
|
gsize error_len, gsize data_len, gsize fd_count, GError **err) { |
|
GString *buf; |
|
buf = g_string_sized_new(8*4 + mod_len + action_len); |
|
*pbuf = buf; |
|
|
|
if (!li_angel_data_write_int32(buf, ANGEL_MAGIC, err)) return FALSE; |
|
if (!li_angel_data_write_int32(buf, type, err)) return FALSE; |
|
if (!li_angel_data_write_int32(buf, id, err)) return FALSE; |
|
if (type != ANGEL_CALL_SEND_RESULT) { |
|
if (!li_angel_data_write_int32(buf, mod_len, err)) return FALSE; |
|
if (!li_angel_data_write_int32(buf, action_len, err)) return FALSE; |
|
} else { |
|
if (!li_angel_data_write_int32(buf, 0, err)) return FALSE; |
|
if (!li_angel_data_write_int32(buf, 0, err)) return FALSE; |
|
} |
|
if (!li_angel_data_write_int32(buf, error_len, err)) return FALSE; |
|
if (!li_angel_data_write_int32(buf, data_len, err)) return FALSE; |
|
if (!li_angel_data_write_int32(buf, fd_count, err)) return FALSE; |
|
|
|
if (type != ANGEL_CALL_SEND_RESULT) { |
|
g_string_append_len(buf, mod, mod_len); |
|
g_string_append_len(buf, action, action_len); |
|
} |
|
|
|
return TRUE; |
|
} |
|
|
|
gboolean li_angel_send_simple_call( |
|
liAngelConnection *acon, |
|
const gchar *mod, gsize mod_len, const gchar *action, gsize action_len, |
|
GString *data, |
|
GError **err) { |
|
GString *buf = NULL; |
|
gboolean queue_was_empty; |
|
|
|
if (err && *err) goto error; |
|
|
|
if (-1 == acon->fd) { |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_CLOSED, "connection already closed"); |
|
goto error; |
|
} |
|
|
|
if (data && data->len > ANGEL_CALL_MAX_STR_LEN) { |
|
g_set_error(err, LI_ANGEL_CALL_ERROR, LI_ANGEL_CALL_INVALID, "data too lang for angel call: %" G_GSIZE_FORMAT " > %i", data->len, ANGEL_CALL_MAX_STR_LEN); |
|
goto error; |
|
} |
|
|
|
if (!prepare_call_header(&buf, ANGEL_CALL_SEND_SIMPLE, -1, mod, mod_len, action, action_len, 0, data ? data->len : 0, 0, err)) goto error; |
|
|
|
g_mutex_lock(acon->mutex); |
|
queue_was_empty = (0 == acon->out->length); |
|
send_queue_push_string(acon->out, buf); |
|
if (data) send_queue_push_string(acon->out, data); |
|
g_mutex_unlock(acon->mutex); |
|
|
|
if (queue_was_empty) |
|
ev_async_send(acon->loop, &acon->out_notify_watcher); |
|
|
|
return TRUE; |
|
|
|
error: |
|
if (data) g_string_free(data, TRUE); |
|
if (buf) g_string_free(buf, TRUE); |
|
return FALSE; |
|
} |
|
|
|
gboolean li_angel_send_call( |
|
liAngelConnection *acon, |
|
const gchar *mod, gsize mod_len, const gchar *action, gsize action_len, |
|
liAngelCall *call, |
|
GString *data, |
|
GError **err) { |
|
GString *buf = NULL; |
|
gboolean queue_was_empty; |
|
|
|
if (err && *err) goto error; |
|
|
|
if (-1 == acon->fd) { |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_CLOSED, "connection already closed"); |
|
goto error; |
|
} |
|
|
|
g_mutex_lock(acon->mutex); |
|
if (-1 != call->id) { |
|
g_mutex_unlock(acon->mutex); |
|
g_set_error(err, LI_ANGEL_CALL_ERROR, LI_ANGEL_CALL_ALREADY_RUNNING, "call already running"); |
|
goto error_before_new_id; |
|
} |
|
|
|
if (-1 == (call->id = li_idlist_get(acon->call_id_list))) { |
|
g_mutex_unlock(acon->mutex); |
|
g_set_error(err, LI_ANGEL_CALL_ERROR, LI_ANGEL_CALL_OUT_OF_CALL_IDS, "out of call ids"); |
|
goto error; |
|
} |
|
call->acon = acon; |
|
|
|
if ((guint) call->id >= acon->call_table->len) { |
|
g_ptr_array_set_size(acon->call_table, call->id + 1); |
|
} |
|
g_ptr_array_index(acon->call_table, call->id) = call; |
|
g_mutex_unlock(acon->mutex); |
|
|
|
if (data && data->len > ANGEL_CALL_MAX_STR_LEN) { |
|
g_set_error(err, LI_ANGEL_CALL_ERROR, LI_ANGEL_CALL_INVALID, "data too lang for angel call: %" G_GSIZE_FORMAT " > %i", data->len, ANGEL_CALL_MAX_STR_LEN); |
|
goto error; |
|
} |
|
|
|
if (!prepare_call_header(&buf, ANGEL_CALL_SEND_CALL, call->id, mod, mod_len, action, action_len, 0, data ? data->len : 0, 0, err)) goto error; |
|
|
|
ev_timer_start(acon->loop, &call->timeout_watcher); |
|
|
|
g_mutex_lock(acon->mutex); |
|
queue_was_empty = (0 == acon->out->length); |
|
send_queue_push_string(acon->out, buf); |
|
if (data) send_queue_push_string(acon->out, data); |
|
g_mutex_unlock(acon->mutex); |
|
|
|
if (queue_was_empty) |
|
ev_async_send(acon->loop, &acon->out_notify_watcher); |
|
|
|
return TRUE; |
|
|
|
error: |
|
if (-1 != call->id) { |
|
li_idlist_put(acon->call_id_list, call->id); |
|
call->id = -1; |
|
call->acon = NULL; |
|
} |
|
error_before_new_id: |
|
if (data) g_string_free(data, TRUE); |
|
if (buf) g_string_free(buf, TRUE); |
|
return FALSE; |
|
} |
|
|
|
gboolean li_angel_send_result( |
|
liAngelConnection *acon, |
|
gint32 id, |
|
GString *error, GString *data, GArray *fds, |
|
GError **err) { |
|
GString *buf = NULL; |
|
gboolean queue_was_empty; |
|
|
|
if (err && *err) goto error; |
|
|
|
if (-1 == acon->fd) { |
|
g_set_error(err, LI_ANGEL_CONNECTION_ERROR, LI_ANGEL_CONNECTION_CLOSED, "connection already closed"); |
|
goto error; |
|
} |
|
|
|
if (data && data->len > ANGEL_CALL_MAX_STR_LEN) { |
|
g_set_error(err, LI_ANGEL_CALL_ERROR, LI_ANGEL_CALL_INVALID, "data too lang for angel call: %" G_GSIZE_FORMAT " > %i", data->len, ANGEL_CALL_MAX_STR_LEN); |
|
goto error; |
|
} |
|
|
|
if (!prepare_call_header(&buf, ANGEL_CALL_SEND_RESULT, id, NULL, 0, NULL, 0, error ? error->len : 0, data ? data->len : 0, fds ? fds->len : 0, err)) goto error; |
|
|
|
g_mutex_lock(acon->mutex); |
|
queue_was_empty = (0 == acon->out->length); |
|
send_queue_push_string(acon->out, buf); |
|
send_queue_push_string(acon->out, error); |
|
if (data) send_queue_push_string(acon->out, data); |
|
send_queue_push_fds(acon->out, fds); |
|
g_mutex_unlock(acon->mutex); |
|
|
|
if (queue_was_empty) |
|
ev_async_send(acon->loop, &acon->out_notify_watcher); |
|
|
|
return TRUE; |
|
|
|
error: |
|
if (data) g_string_free(data, TRUE); |
|
if (buf) g_string_free(buf, TRUE); |
|
if (error) g_string_free(error, TRUE); |
|
if (fds) close_fd_array(fds); |
|
return FALSE; |
|
|
|
return FALSE; |
|
} |
|
|
|
/* free temporary needed memroy; call this once in while after some activity */ |
|
void li_angel_cleanup_tables(liAngelConnection *acon) { |
|
UNUSED(acon); |
|
/* TODO |
|
guint max_used_id = idlist_cleanup(acon->call_id_list); |
|
g_ptr_array_set_size(acon->call_id_list, max_used_id); |
|
*/ |
|
}
|
|
|