aboutsummaryrefslogtreecommitdiffstats
path: root/src/lua/lua_tcp.c
diff options
context:
space:
mode:
authorVsevolod Stakhov <vsevolod@rspamd.com>2023-07-26 10:49:23 +0100
committerVsevolod Stakhov <vsevolod@rspamd.com>2023-07-26 10:49:23 +0100
commit537a7180a0d5132c11636c4fd8b1450cd99d352c (patch)
treefb9f8c84955a411bdffbd6371ea32f2716fb3687 /src/lua/lua_tcp.c
parent5fd7a90fdaa33f52c59bdb0ca84451e5c1e22365 (diff)
downloadrspamd-537a7180a0d5132c11636c4fd8b1450cd99d352c.tar.gz
rspamd-537a7180a0d5132c11636c4fd8b1450cd99d352c.zip
[Rework] Use clang-format to unify formatting in all sources
No meaningful changes.
Diffstat (limited to 'src/lua/lua_tcp.c')
-rw-r--r--src/lua/lua_tcp.c1520
1 files changed, 761 insertions, 759 deletions
diff --git a/src/lua/lua_tcp.c b/src/lua/lua_tcp.c
index 2ea3affa6..45faa7964 100644
--- a/src/lua/lua_tcp.c
+++ b/src/lua/lua_tcp.c
@@ -94,7 +94,7 @@ rspamd_config:register_symbol({
})
*/
-LUA_FUNCTION_DEF (tcp, request);
+LUA_FUNCTION_DEF(tcp, request);
/***
* @function rspamd_tcp.connect_sync()
@@ -151,14 +151,14 @@ rspamd_config:register_symbol({
})
*
*/
-LUA_FUNCTION_DEF (tcp, connect_sync);
+LUA_FUNCTION_DEF(tcp, connect_sync);
/***
* @method tcp:close()
*
* Closes TCP connection
*/
-LUA_FUNCTION_DEF (tcp, close);
+LUA_FUNCTION_DEF(tcp, close);
/***
* @method tcp:add_read(callback, [pattern])
@@ -167,7 +167,7 @@ LUA_FUNCTION_DEF (tcp, close);
* @param {function} callback to be called when data is read
* @param {string} pattern optional stop pattern
*/
-LUA_FUNCTION_DEF (tcp, add_read);
+LUA_FUNCTION_DEF(tcp, add_read);
/***
* @method tcp:add_write(callback, data)
@@ -176,14 +176,14 @@ LUA_FUNCTION_DEF (tcp, add_read);
* @param {function} optional callback to be called when data is completely written
* @param {table/string/text} data to send to a remote server
*/
-LUA_FUNCTION_DEF (tcp, add_write);
+LUA_FUNCTION_DEF(tcp, add_write);
/***
* @method tcp:shift_callback()
*
* Shifts the current callback and go to the next one (if any)
*/
-LUA_FUNCTION_DEF (tcp, shift_callback);
+LUA_FUNCTION_DEF(tcp, shift_callback);
/***
* @method tcp:starttls([no_verify])
@@ -191,32 +191,30 @@ LUA_FUNCTION_DEF (tcp, shift_callback);
* Starts tls connection
* @param {boolean} no_verify used to skip ssl verification
*/
-LUA_FUNCTION_DEF (tcp, starttls);
+LUA_FUNCTION_DEF(tcp, starttls);
static const struct luaL_reg tcp_libf[] = {
- LUA_INTERFACE_DEF (tcp, request),
+ LUA_INTERFACE_DEF(tcp, request),
{"new", lua_tcp_request},
{"connect", lua_tcp_request},
{"connect_sync", lua_tcp_connect_sync},
- {NULL, NULL}
-};
+ {NULL, NULL}};
static const struct luaL_reg tcp_libm[] = {
- LUA_INTERFACE_DEF (tcp, close),
- LUA_INTERFACE_DEF (tcp, add_read),
- LUA_INTERFACE_DEF (tcp, add_write),
- LUA_INTERFACE_DEF (tcp, shift_callback),
- LUA_INTERFACE_DEF (tcp, starttls),
+ LUA_INTERFACE_DEF(tcp, close),
+ LUA_INTERFACE_DEF(tcp, add_read),
+ LUA_INTERFACE_DEF(tcp, add_write),
+ LUA_INTERFACE_DEF(tcp, shift_callback),
+ LUA_INTERFACE_DEF(tcp, starttls),
{"__tostring", rspamd_lua_class_tostring},
- {NULL, NULL}
-};
+ {NULL, NULL}};
/***
* @method tcp:close()
*
* Closes TCP connection
*/
-LUA_FUNCTION_DEF (tcp_sync, close);
+LUA_FUNCTION_DEF(tcp_sync, close);
/***
* @method read_once()
@@ -224,21 +222,21 @@ LUA_FUNCTION_DEF (tcp_sync, close);
* Performs one read operation. If syscall returned with EAGAIN/EINT,
* restarts the operation, so it always returns either data or error.
*/
-LUA_FUNCTION_DEF (tcp_sync, read_once);
+LUA_FUNCTION_DEF(tcp_sync, read_once);
/***
* @method eof()
*
* True if last IO operation ended with EOF, i.e. endpoint closed connection
*/
-LUA_FUNCTION_DEF (tcp_sync, eof);
+LUA_FUNCTION_DEF(tcp_sync, eof);
/***
* @method shutdown()
*
* Half-shutdown TCP connection
*/
-LUA_FUNCTION_DEF (tcp_sync, shutdown);
+LUA_FUNCTION_DEF(tcp_sync, shutdown);
/***
* @method write()
@@ -247,22 +245,21 @@ LUA_FUNCTION_DEF (tcp_sync, shutdown);
* restarts the operation. If performs write() until all the passed
* data is written completely.
*/
-LUA_FUNCTION_DEF (tcp_sync, write);
+LUA_FUNCTION_DEF(tcp_sync, write);
-LUA_FUNCTION_DEF (tcp_sync, gc);
+LUA_FUNCTION_DEF(tcp_sync, gc);
-static void lua_tcp_sync_session_dtor (gpointer ud);
+static void lua_tcp_sync_session_dtor(gpointer ud);
static const struct luaL_reg tcp_sync_libm[] = {
- LUA_INTERFACE_DEF (tcp_sync, close),
- LUA_INTERFACE_DEF (tcp_sync, read_once),
- LUA_INTERFACE_DEF (tcp_sync, write),
- LUA_INTERFACE_DEF (tcp_sync, eof),
- LUA_INTERFACE_DEF (tcp_sync, shutdown),
- {"__gc", lua_tcp_sync_gc},
+ LUA_INTERFACE_DEF(tcp_sync, close),
+ LUA_INTERFACE_DEF(tcp_sync, read_once),
+ LUA_INTERFACE_DEF(tcp_sync, write),
+ LUA_INTERFACE_DEF(tcp_sync, eof),
+ LUA_INTERFACE_DEF(tcp_sync, shutdown),
+ {"__gc", lua_tcp_sync_gc},
{"__tostring", rspamd_lua_class_tostring},
- {NULL, NULL}
-};
+ {NULL, NULL}};
struct lua_tcp_read_handler {
gchar *stop_pattern;
@@ -309,17 +306,19 @@ struct lua_tcp_dtor {
#undef TCP_DEBUG_REFS
#ifdef TCP_DEBUG_REFS
-#define TCP_RETAIN(x) do { \
- msg_info ("retain ref %p, refcount: %d", (x), (x)->ref.refcount); \
- REF_RETAIN(x); \
-} while (0)
-
-#define TCP_RELEASE(x) do { \
- msg_info ("release ref %p, refcount: %d", (x), (x)->ref.refcount); \
- REF_RELEASE(x); \
-} while (0)
+#define TCP_RETAIN(x) \
+ do { \
+ msg_info("retain ref %p, refcount: %d", (x), (x)->ref.refcount); \
+ REF_RETAIN(x); \
+ } while (0)
+
+#define TCP_RELEASE(x) \
+ do { \
+ msg_info("release ref %p, refcount: %d", (x), (x)->ref.refcount); \
+ REF_RELEASE(x); \
+ } while (0)
#else
-#define TCP_RETAIN(x) REF_RETAIN(x)
+#define TCP_RETAIN(x) REF_RETAIN(x)
#define TCP_RELEASE(x) REF_RELEASE(x)
#endif
@@ -350,26 +349,28 @@ struct lua_tcp_cbdata {
#define IS_SYNC(c) (((c)->flags & LUA_TCP_FLAG_SYNC) != 0)
-#define msg_debug_tcp(...) rspamd_conditional_debug_fast (NULL, cbd->addr, \
- rspamd_lua_tcp_log_id, "lua_tcp", cbd->tag, \
- G_STRFUNC, \
- __VA_ARGS__)
+#define msg_debug_tcp(...) rspamd_conditional_debug_fast(NULL, cbd->addr, \
+ rspamd_lua_tcp_log_id, "lua_tcp", cbd->tag, \
+ G_STRFUNC, \
+ __VA_ARGS__)
INIT_LOG_MODULE(lua_tcp)
-static void lua_tcp_handler (int fd, short what, gpointer ud);
-static void lua_tcp_plan_handler_event (struct lua_tcp_cbdata *cbd,
- gboolean can_read, gboolean can_write);
-static void lua_tcp_unregister_event (struct lua_tcp_cbdata *cbd);
+static void lua_tcp_handler(int fd, short what, gpointer ud);
+static void lua_tcp_plan_handler_event(struct lua_tcp_cbdata *cbd,
+ gboolean can_read, gboolean can_write);
+static void lua_tcp_unregister_event(struct lua_tcp_cbdata *cbd);
static void
-lua_tcp_void_finalyser (gpointer arg) {}
+lua_tcp_void_finalyser(gpointer arg)
+{
+}
static const gdouble default_tcp_timeout = 5.0;
static struct rspamd_dns_resolver *
-lua_tcp_global_resolver (struct ev_loop *ev_base,
- struct rspamd_config *cfg)
+lua_tcp_global_resolver(struct ev_loop *ev_base,
+ struct rspamd_config *cfg)
{
static struct rspamd_dns_resolver *global_resolver;
@@ -378,18 +379,18 @@ lua_tcp_global_resolver (struct ev_loop *ev_base,
}
if (global_resolver == NULL) {
- global_resolver = rspamd_dns_resolver_init (NULL, ev_base, cfg);
+ global_resolver = rspamd_dns_resolver_init(NULL, ev_base, cfg);
}
return global_resolver;
}
static gboolean
-lua_tcp_shift_handler (struct lua_tcp_cbdata *cbd)
+lua_tcp_shift_handler(struct lua_tcp_cbdata *cbd)
{
struct lua_tcp_handler *hdl;
- hdl = g_queue_pop_head (cbd->handlers);
+ hdl = g_queue_pop_head(cbd->handlers);
if (hdl == NULL) {
/* We are done */
@@ -397,142 +398,143 @@ lua_tcp_shift_handler (struct lua_tcp_cbdata *cbd)
}
if (hdl->type == LUA_WANT_READ) {
- msg_debug_tcp ("switch from read handler %d", hdl->h.r.cbref);
+ msg_debug_tcp("switch from read handler %d", hdl->h.r.cbref);
if (hdl->h.r.cbref && hdl->h.r.cbref != -1) {
- luaL_unref (cbd->cfg->lua_state, LUA_REGISTRYINDEX, hdl->h.r.cbref);
+ luaL_unref(cbd->cfg->lua_state, LUA_REGISTRYINDEX, hdl->h.r.cbref);
}
if (hdl->h.r.stop_pattern) {
- g_free (hdl->h.r.stop_pattern);
+ g_free(hdl->h.r.stop_pattern);
}
}
else if (hdl->type == LUA_WANT_WRITE) {
- msg_debug_tcp ("switch from write handler %d", hdl->h.r.cbref);
+ msg_debug_tcp("switch from write handler %d", hdl->h.r.cbref);
if (hdl->h.w.cbref && hdl->h.w.cbref != -1) {
- luaL_unref (cbd->cfg->lua_state, LUA_REGISTRYINDEX, hdl->h.w.cbref);
+ luaL_unref(cbd->cfg->lua_state, LUA_REGISTRYINDEX, hdl->h.w.cbref);
}
if (hdl->h.w.iov) {
- g_free (hdl->h.w.iov);
+ g_free(hdl->h.w.iov);
}
}
else {
- msg_debug_tcp ("removing connect handler");
+ msg_debug_tcp("removing connect handler");
/* LUA_WANT_CONNECT: it doesn't allocate anything, nothing to do here */
}
- g_free (hdl);
+ g_free(hdl);
return TRUE;
}
static void
-lua_tcp_fin (gpointer arg)
+lua_tcp_fin(gpointer arg)
{
- struct lua_tcp_cbdata *cbd = (struct lua_tcp_cbdata *)arg;
+ struct lua_tcp_cbdata *cbd = (struct lua_tcp_cbdata *) arg;
struct lua_tcp_dtor *dtor, *dttmp;
- if (IS_SYNC (cbd) && cbd->task) {
+ if (IS_SYNC(cbd) && cbd->task) {
/*
pointer is now becoming invalid, we should remove registered destructor,
all the necessary steps are done here
*/
- rspamd_mempool_replace_destructor (cbd->task->task_pool,
- lua_tcp_sync_session_dtor, cbd, NULL);
+ rspamd_mempool_replace_destructor(cbd->task->task_pool,
+ lua_tcp_sync_session_dtor, cbd, NULL);
}
- msg_debug_tcp ("finishing TCP %s connection", IS_SYNC (cbd) ? "sync" : "async");
+ msg_debug_tcp("finishing TCP %s connection", IS_SYNC(cbd) ? "sync" : "async");
if (cbd->connect_cb != -1) {
- luaL_unref (cbd->cfg->lua_state, LUA_REGISTRYINDEX, cbd->connect_cb);
+ luaL_unref(cbd->cfg->lua_state, LUA_REGISTRYINDEX, cbd->connect_cb);
}
if (cbd->ssl_conn) {
/* TODO: postpone close in case ssl is used ! */
- rspamd_ssl_connection_free (cbd->ssl_conn);
+ rspamd_ssl_connection_free(cbd->ssl_conn);
}
if (cbd->fd != -1) {
- rspamd_ev_watcher_stop (cbd->event_loop, &cbd->ev);
- close (cbd->fd);
+ rspamd_ev_watcher_stop(cbd->event_loop, &cbd->ev);
+ close(cbd->fd);
cbd->fd = -1;
}
if (cbd->addr) {
- rspamd_inet_address_free (cbd->addr);
+ rspamd_inet_address_free(cbd->addr);
}
if (cbd->up) {
rspamd_upstream_unref(cbd->up);
}
- while (lua_tcp_shift_handler (cbd)) {}
- g_queue_free (cbd->handlers);
+ while (lua_tcp_shift_handler(cbd)) {}
+ g_queue_free(cbd->handlers);
- LL_FOREACH_SAFE (cbd->dtors, dtor, dttmp) {
- dtor->dtor (dtor->data);
- g_free (dtor);
+ LL_FOREACH_SAFE(cbd->dtors, dtor, dttmp)
+ {
+ dtor->dtor(dtor->data);
+ g_free(dtor);
}
- g_byte_array_unref (cbd->in);
- g_free (cbd->hostname);
- g_free (cbd);
+ g_byte_array_unref(cbd->in);
+ g_free(cbd->hostname);
+ g_free(cbd);
}
static struct lua_tcp_cbdata *
-lua_check_tcp (lua_State *L, gint pos)
+lua_check_tcp(lua_State *L, gint pos)
{
- void *ud = rspamd_lua_check_udata (L, pos, "rspamd{tcp}");
- luaL_argcheck (L, ud != NULL, pos, "'tcp' expected");
- return ud ? *((struct lua_tcp_cbdata **)ud) : NULL;
+ void *ud = rspamd_lua_check_udata(L, pos, "rspamd{tcp}");
+ luaL_argcheck(L, ud != NULL, pos, "'tcp' expected");
+ return ud ? *((struct lua_tcp_cbdata **) ud) : NULL;
}
static void
-lua_tcp_maybe_free (struct lua_tcp_cbdata *cbd)
+lua_tcp_maybe_free(struct lua_tcp_cbdata *cbd)
{
- if (IS_SYNC (cbd)) {
+ if (IS_SYNC(cbd)) {
/*
* in this mode, we don't remove object, we only remove the event
* Object is owned by lua and will be destroyed on __gc()
*/
if (cbd->item) {
- rspamd_symcache_item_async_dec_check (cbd->task, cbd->item, M);
+ rspamd_symcache_item_async_dec_check(cbd->task, cbd->item, M);
cbd->item = NULL;
}
if (cbd->async_ev) {
- rspamd_session_remove_event (cbd->session, lua_tcp_void_finalyser, cbd);
+ rspamd_session_remove_event(cbd->session, lua_tcp_void_finalyser, cbd);
}
cbd->async_ev = NULL;
}
else {
if (cbd->item) {
- rspamd_symcache_item_async_dec_check (cbd->task, cbd->item, M);
+ rspamd_symcache_item_async_dec_check(cbd->task, cbd->item, M);
cbd->item = NULL;
}
if (cbd->async_ev) {
- rspamd_session_remove_event (cbd->session, lua_tcp_fin, cbd);
+ rspamd_session_remove_event(cbd->session, lua_tcp_fin, cbd);
}
else {
- lua_tcp_fin (cbd);
+ lua_tcp_fin(cbd);
}
}
}
#ifdef __GNUC__
static void
-lua_tcp_push_error (struct lua_tcp_cbdata *cbd, gboolean is_fatal,
- const char *err, ...) __attribute__ ((format(printf, 3, 4)));
+lua_tcp_push_error(struct lua_tcp_cbdata *cbd, gboolean is_fatal,
+ const char *err, ...) __attribute__((format(printf, 3, 4)));
#endif
-static void lua_tcp_resume_thread_error_argp (struct lua_tcp_cbdata *cbd, const gchar *error, va_list argp);
+static void lua_tcp_resume_thread_error_argp(struct lua_tcp_cbdata *cbd, const gchar *error, va_list argp);
static void
-lua_tcp_push_error (struct lua_tcp_cbdata *cbd, gboolean is_fatal,
- const char *err, ...)
+lua_tcp_push_error(struct lua_tcp_cbdata *cbd, gboolean is_fatal,
+ const char *err, ...)
{
va_list ap, ap_copy;
struct lua_tcp_cbdata **pcbd;
@@ -547,20 +549,20 @@ lua_tcp_push_error (struct lua_tcp_cbdata *cbd, gboolean is_fatal,
}
if (cbd->thread) {
- va_start (ap, err);
- lua_tcp_resume_thread_error_argp (cbd, err, ap);
- va_end (ap);
+ va_start(ap, err);
+ lua_tcp_resume_thread_error_argp(cbd, err, ap);
+ va_end(ap);
return;
}
- lua_thread_pool_prepare_callback (cbd->cfg->lua_thread_pool, &cbs);
+ lua_thread_pool_prepare_callback(cbd->cfg->lua_thread_pool, &cbs);
L = cbs.L;
- va_start (ap, err);
+ va_start(ap, err);
for (;;) {
- hdl = g_queue_peek_head (cbd->handlers);
+ hdl = g_queue_peek_head(cbd->handlers);
if (hdl == NULL) {
break;
@@ -574,38 +576,38 @@ lua_tcp_push_error (struct lua_tcp_cbdata *cbd, gboolean is_fatal,
}
if (cbref != -1) {
- top = lua_gettop (L);
- lua_rawgeti (L, LUA_REGISTRYINDEX, cbref);
+ top = lua_gettop(L);
+ lua_rawgeti(L, LUA_REGISTRYINDEX, cbref);
/* Error message */
- va_copy (ap_copy, ap);
- lua_pushvfstring (L, err, ap_copy);
- va_end (ap_copy);
+ va_copy(ap_copy, ap);
+ lua_pushvfstring(L, err, ap_copy);
+ va_end(ap_copy);
/* Body */
- lua_pushnil (L);
+ lua_pushnil(L);
/* Connection */
- pcbd = lua_newuserdata (L, sizeof (*pcbd));
+ pcbd = lua_newuserdata(L, sizeof(*pcbd));
*pcbd = cbd;
- rspamd_lua_setclass (L, "rspamd{tcp}", -1);
- TCP_RETAIN (cbd);
+ rspamd_lua_setclass(L, "rspamd{tcp}", -1);
+ TCP_RETAIN(cbd);
if (cbd->item) {
- rspamd_symcache_set_cur_item (cbd->task, cbd->item);
+ rspamd_symcache_set_cur_item(cbd->task, cbd->item);
}
- if (lua_pcall (L, 3, 0, 0) != 0) {
- msg_info ("callback call failed: %s", lua_tostring (L, -1));
+ if (lua_pcall(L, 3, 0, 0) != 0) {
+ msg_info("callback call failed: %s", lua_tostring(L, -1));
}
- lua_settop (L, top);
+ lua_settop(L, top);
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
- if ((cbd->flags & (LUA_TCP_FLAG_FINISHED|LUA_TCP_FLAG_CONNECTED)) ==
- (LUA_TCP_FLAG_FINISHED|LUA_TCP_FLAG_CONNECTED)) {
+ if ((cbd->flags & (LUA_TCP_FLAG_FINISHED | LUA_TCP_FLAG_CONNECTED)) ==
+ (LUA_TCP_FLAG_FINISHED | LUA_TCP_FLAG_CONNECTED)) {
/* A callback has called `close` method, so we need to release a refcount */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
callback_called = TRUE;
@@ -618,26 +620,26 @@ lua_tcp_push_error (struct lua_tcp_cbdata *cbd, gboolean is_fatal,
}
else {
/* Shift to another callback to inform about non fatal error */
- msg_debug_tcp ("non fatal error find matching callback");
- lua_tcp_shift_handler (cbd);
+ msg_debug_tcp("non fatal error find matching callback");
+ lua_tcp_shift_handler(cbd);
continue;
}
}
else {
- msg_debug_tcp ("fatal error rollback all handlers");
- lua_tcp_shift_handler (cbd);
+ msg_debug_tcp("fatal error rollback all handlers");
+ lua_tcp_shift_handler(cbd);
}
}
- va_end (ap);
+ va_end(ap);
- lua_thread_pool_restore_callback (&cbs);
+ lua_thread_pool_restore_callback(&cbs);
}
-static void lua_tcp_resume_thread (struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len);
+static void lua_tcp_resume_thread(struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len);
static void
-lua_tcp_push_data (struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len)
+lua_tcp_push_data(struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len)
{
struct rspamd_lua_text *t;
struct lua_tcp_cbdata **pcbd;
@@ -647,16 +649,16 @@ lua_tcp_push_data (struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len)
lua_State *L;
if (cbd->thread) {
- lua_tcp_resume_thread (cbd, str, len);
+ lua_tcp_resume_thread(cbd, str, len);
return;
}
- lua_thread_pool_prepare_callback (cbd->cfg->lua_thread_pool, &cbs);
+ lua_thread_pool_prepare_callback(cbd->cfg->lua_thread_pool, &cbs);
L = cbs.L;
- hdl = g_queue_peek_head (cbd->handlers);
+ hdl = g_queue_peek_head(cbd->handlers);
- g_assert (hdl != NULL);
+ g_assert(hdl != NULL);
if (hdl->type == LUA_WANT_READ) {
cbref = hdl->h.r.cbref;
@@ -666,16 +668,16 @@ lua_tcp_push_data (struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len)
}
if (cbref != -1) {
- top = lua_gettop (L);
- lua_rawgeti (L, LUA_REGISTRYINDEX, cbref);
+ top = lua_gettop(L);
+ lua_rawgeti(L, LUA_REGISTRYINDEX, cbref);
/* Error */
- lua_pushnil (L);
+ lua_pushnil(L);
/* Body */
if (hdl->type == LUA_WANT_READ) {
- t = lua_newuserdata (L, sizeof (*t));
- rspamd_lua_setclass (L, "rspamd{text}", -1);
- t->start = (const gchar *)str;
+ t = lua_newuserdata(L, sizeof(*t));
+ rspamd_lua_setclass(L, "rspamd{text}", -1);
+ t->start = (const gchar *) str;
t->len = len;
t->flags = 0;
arg_cnt = 3;
@@ -684,51 +686,51 @@ lua_tcp_push_data (struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len)
arg_cnt = 2;
}
/* Connection */
- pcbd = lua_newuserdata (L, sizeof (*pcbd));
+ pcbd = lua_newuserdata(L, sizeof(*pcbd));
*pcbd = cbd;
- rspamd_lua_setclass (L, "rspamd{tcp}", -1);
+ rspamd_lua_setclass(L, "rspamd{tcp}", -1);
- TCP_RETAIN (cbd);
+ TCP_RETAIN(cbd);
if (cbd->item) {
- rspamd_symcache_set_cur_item (cbd->task, cbd->item);
+ rspamd_symcache_set_cur_item(cbd->task, cbd->item);
}
- if (lua_pcall (L, arg_cnt, 0, 0) != 0) {
- msg_info ("callback call failed: %s", lua_tostring (L, -1));
+ if (lua_pcall(L, arg_cnt, 0, 0) != 0) {
+ msg_info("callback call failed: %s", lua_tostring(L, -1));
}
- lua_settop (L, top);
- TCP_RELEASE (cbd);
+ lua_settop(L, top);
+ TCP_RELEASE(cbd);
- if ((cbd->flags & (LUA_TCP_FLAG_FINISHED|LUA_TCP_FLAG_CONNECTED)) ==
- (LUA_TCP_FLAG_FINISHED|LUA_TCP_FLAG_CONNECTED)) {
+ if ((cbd->flags & (LUA_TCP_FLAG_FINISHED | LUA_TCP_FLAG_CONNECTED)) ==
+ (LUA_TCP_FLAG_FINISHED | LUA_TCP_FLAG_CONNECTED)) {
/* A callback has called `close` method, so we need to release a refcount */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
}
- lua_thread_pool_restore_callback (&cbs);
+ lua_thread_pool_restore_callback(&cbs);
}
static void
-lua_tcp_resume_thread_error_argp (struct lua_tcp_cbdata *cbd, const gchar *error, va_list argp)
+lua_tcp_resume_thread_error_argp(struct lua_tcp_cbdata *cbd, const gchar *error, va_list argp)
{
struct thread_entry *thread = cbd->thread;
lua_State *L = thread->lua_state;
- lua_pushboolean (L, FALSE);
- lua_pushvfstring (L, error, argp);
+ lua_pushboolean(L, FALSE);
+ lua_pushvfstring(L, error, argp);
- lua_tcp_shift_handler (cbd);
+ lua_tcp_shift_handler(cbd);
// lua_tcp_unregister_event (cbd);
- lua_thread_pool_set_running_entry (cbd->cfg->lua_thread_pool, cbd->thread);
- lua_thread_resume (thread, 2);
- TCP_RELEASE (cbd);
+ lua_thread_pool_set_running_entry(cbd->cfg->lua_thread_pool, cbd->thread);
+ lua_thread_resume(thread, 2);
+ TCP_RELEASE(cbd);
}
static void
-lua_tcp_resume_thread (struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len)
+lua_tcp_resume_thread(struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len)
{
/*
* typical call returns:
@@ -748,60 +750,60 @@ lua_tcp_resume_thread (struct lua_tcp_cbdata *cbd, const guint8 *str, gsize len)
lua_State *L = cbd->thread->lua_state;
struct lua_tcp_handler *hdl;
- hdl = g_queue_peek_head (cbd->handlers);
+ hdl = g_queue_peek_head(cbd->handlers);
- lua_pushboolean (L, TRUE);
+ lua_pushboolean(L, TRUE);
if (hdl->type == LUA_WANT_READ) {
- lua_pushlstring (L, str, len);
+ lua_pushlstring(L, str, len);
}
else {
- lua_pushnil (L);
+ lua_pushnil(L);
}
- lua_tcp_shift_handler (cbd);
- lua_thread_pool_set_running_entry (cbd->cfg->lua_thread_pool,
- cbd->thread);
+ lua_tcp_shift_handler(cbd);
+ lua_thread_pool_set_running_entry(cbd->cfg->lua_thread_pool,
+ cbd->thread);
if (cbd->item) {
- rspamd_symcache_set_cur_item (cbd->task, cbd->item);
+ rspamd_symcache_set_cur_item(cbd->task, cbd->item);
}
- lua_thread_resume (cbd->thread, 2);
+ lua_thread_resume(cbd->thread, 2);
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
static void
-lua_tcp_plan_read (struct lua_tcp_cbdata *cbd)
+lua_tcp_plan_read(struct lua_tcp_cbdata *cbd)
{
- rspamd_ev_watcher_reschedule (cbd->event_loop, &cbd->ev, EV_READ);
+ rspamd_ev_watcher_reschedule(cbd->event_loop, &cbd->ev, EV_READ);
}
static void
-lua_tcp_connect_helper (struct lua_tcp_cbdata *cbd)
+lua_tcp_connect_helper(struct lua_tcp_cbdata *cbd)
{
/* This is used for sync mode only */
lua_State *L = cbd->thread->lua_state;
struct lua_tcp_cbdata **pcbd;
- lua_pushboolean (L, TRUE);
+ lua_pushboolean(L, TRUE);
- lua_thread_pool_set_running_entry (cbd->cfg->lua_thread_pool, cbd->thread);
- pcbd = lua_newuserdata (L, sizeof (*pcbd));
+ lua_thread_pool_set_running_entry(cbd->cfg->lua_thread_pool, cbd->thread);
+ pcbd = lua_newuserdata(L, sizeof(*pcbd));
*pcbd = cbd;
- rspamd_lua_setclass (L, "rspamd{tcp_sync}", -1);
- msg_debug_tcp ("tcp connected");
+ rspamd_lua_setclass(L, "rspamd{tcp_sync}", -1);
+ msg_debug_tcp("tcp connected");
- lua_tcp_shift_handler (cbd);
+ lua_tcp_shift_handler(cbd);
// lua_tcp_unregister_event (cbd);
- lua_thread_resume (cbd->thread, 2);
- TCP_RELEASE (cbd);
+ lua_thread_resume(cbd->thread, 2);
+ TCP_RELEASE(cbd);
}
static void
-lua_tcp_write_helper (struct lua_tcp_cbdata *cbd)
+lua_tcp_write_helper(struct lua_tcp_cbdata *cbd)
{
struct iovec *start;
guint niov, i;
@@ -814,9 +816,9 @@ lua_tcp_write_helper (struct lua_tcp_cbdata *cbd)
struct lua_tcp_write_handler *wh;
struct msghdr msg;
- hdl = g_queue_peek_head (cbd->handlers);
+ hdl = g_queue_peek_head(cbd->handlers);
- g_assert (hdl != NULL && hdl->type == LUA_WANT_WRITE);
+ g_assert(hdl != NULL && hdl->type == LUA_WANT_WRITE);
wh = &hdl->h.w;
if (wh->pos == wh->total_bytes) {
@@ -829,14 +831,14 @@ lua_tcp_write_helper (struct lua_tcp_cbdata *cbd)
/* We know that niov is small enough for that */
if (niov < 1024) {
- cur_iov = g_alloca (niov * sizeof (struct iovec));
+ cur_iov = g_alloca(niov * sizeof(struct iovec));
}
else {
- cur_iov = g_malloc0 (niov * sizeof (struct iovec));
+ cur_iov = g_malloc0(niov * sizeof(struct iovec));
allocated_iov = true;
}
- memcpy (cur_iov, wh->iov, niov * sizeof (struct iovec));
+ memcpy(cur_iov, wh->iov, niov * sizeof(struct iovec));
for (i = 0; i < wh->iovlen && remain > 0; i++) {
/* Find out the first iov required */
@@ -847,48 +849,48 @@ lua_tcp_write_helper (struct lua_tcp_cbdata *cbd)
niov--;
}
else {
- start->iov_base = (void *)((char *)start->iov_base + remain);
+ start->iov_base = (void *) ((char *) start->iov_base + remain);
start->iov_len -= remain;
remain = 0;
}
}
- memset (&msg, 0, sizeof (msg));
+ memset(&msg, 0, sizeof(msg));
msg.msg_iov = start;
- msg.msg_iovlen = MIN (IOV_MAX, niov);
- g_assert (niov > 0);
+ msg.msg_iovlen = MIN(IOV_MAX, niov);
+ g_assert(niov > 0);
#ifdef MSG_NOSIGNAL
flags = MSG_NOSIGNAL;
#endif
- msg_debug_tcp ("want write %d io vectors of %d", (int)msg.msg_iovlen,
- (int)niov);
+ msg_debug_tcp("want write %d io vectors of %d", (int) msg.msg_iovlen,
+ (int) niov);
if (cbd->ssl_conn) {
- r = rspamd_ssl_writev (cbd->ssl_conn, msg.msg_iov, msg.msg_iovlen);
+ r = rspamd_ssl_writev(cbd->ssl_conn, msg.msg_iov, msg.msg_iovlen);
}
else {
- r = sendmsg (cbd->fd, &msg, flags);
+ r = sendmsg(cbd->fd, &msg, flags);
}
if (allocated_iov) {
- g_free (cur_iov);
+ g_free(cur_iov);
}
if (r == -1) {
if (!(cbd->ssl_conn)) {
if (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR) {
- msg_debug_tcp ("got temporary failure, retry write");
- lua_tcp_plan_handler_event (cbd, TRUE, TRUE);
+ msg_debug_tcp("got temporary failure, retry write");
+ lua_tcp_plan_handler_event(cbd, TRUE, TRUE);
return;
}
else {
- lua_tcp_push_error (cbd, TRUE,
- "IO write error while trying to write %d bytes: %s",
- (gint) remain, strerror (errno));
+ lua_tcp_push_error(cbd, TRUE,
+ "IO write error while trying to write %d bytes: %s",
+ (gint) remain, strerror(errno));
- msg_debug_tcp ("write error, terminate connection");
- TCP_RELEASE (cbd);
+ msg_debug_tcp("write error, terminate connection");
+ TCP_RELEASE(cbd);
}
}
@@ -898,8 +900,8 @@ lua_tcp_write_helper (struct lua_tcp_cbdata *cbd)
wh->pos += r;
}
- msg_debug_tcp ("written %z bytes: %z/%z", r,
- wh->pos, wh->total_bytes);
+ msg_debug_tcp("written %z bytes: %z/%z", r,
+ wh->pos, wh->total_bytes);
if (wh->pos >= wh->total_bytes) {
goto call_finish_handler;
@@ -912,7 +914,7 @@ lua_tcp_write_helper (struct lua_tcp_cbdata *cbd)
*
* Hence, we need to check if we can write more at some point...
*/
- lua_tcp_write_helper (cbd);
+ lua_tcp_write_helper(cbd);
}
}
@@ -920,11 +922,11 @@ lua_tcp_write_helper (struct lua_tcp_cbdata *cbd)
call_finish_handler:
- msg_debug_tcp ("finishing TCP write, calling TCP handler");
+ msg_debug_tcp("finishing TCP write, calling TCP handler");
if ((cbd->flags & LUA_TCP_FLAG_SHUTDOWN)) {
/* Half close the connection */
- shutdown (cbd->fd, SHUT_WR);
+ shutdown(cbd->fd, SHUT_WR);
cbd->flags &= ~LUA_TCP_FLAG_SHUTDOWN;
}
@@ -932,16 +934,16 @@ call_finish_handler:
rspamd_upstream_ok(cbd->up);
}
- lua_tcp_push_data (cbd, NULL, 0);
- if (!IS_SYNC (cbd)) {
- lua_tcp_shift_handler (cbd);
- lua_tcp_plan_handler_event (cbd, TRUE, TRUE);
+ lua_tcp_push_data(cbd, NULL, 0);
+ if (!IS_SYNC(cbd)) {
+ lua_tcp_shift_handler(cbd);
+ lua_tcp_plan_handler_event(cbd, TRUE, TRUE);
}
}
static gboolean
-lua_tcp_process_read_handler (struct lua_tcp_cbdata *cbd,
- struct lua_tcp_read_handler *rh, gboolean eof)
+lua_tcp_process_read_handler(struct lua_tcp_cbdata *cbd,
+ struct lua_tcp_read_handler *rh, gboolean eof)
{
guint slen;
goffset pos;
@@ -950,17 +952,17 @@ lua_tcp_process_read_handler (struct lua_tcp_cbdata *cbd,
slen = rh->plen;
if (cbd->in->len >= slen) {
- if ((pos = rspamd_substring_search (cbd->in->data, cbd->in->len,
- rh->stop_pattern, slen)) != -1) {
- msg_debug_tcp ("found TCP stop pattern");
- lua_tcp_push_data (cbd, cbd->in->data, pos);
+ if ((pos = rspamd_substring_search(cbd->in->data, cbd->in->len,
+ rh->stop_pattern, slen)) != -1) {
+ msg_debug_tcp("found TCP stop pattern");
+ lua_tcp_push_data(cbd, cbd->in->data, pos);
- if (!IS_SYNC (cbd)) {
- lua_tcp_shift_handler (cbd);
+ if (!IS_SYNC(cbd)) {
+ lua_tcp_shift_handler(cbd);
}
if (pos + slen < cbd->in->len) {
/* We have a leftover */
- memmove (cbd->in->data, cbd->in->data + pos + slen,
+ memmove(cbd->in->data, cbd->in->data + pos + slen,
cbd->in->len - (pos + slen));
cbd->in->len = cbd->in->len - (pos + slen);
}
@@ -972,28 +974,28 @@ lua_tcp_process_read_handler (struct lua_tcp_cbdata *cbd,
}
else {
/* Plan new read */
- msg_debug_tcp ("NOT found TCP stop pattern");
+ msg_debug_tcp("NOT found TCP stop pattern");
if (!cbd->eof) {
- lua_tcp_plan_read (cbd);
+ lua_tcp_plan_read(cbd);
}
else {
/* Got session finished but no stop pattern */
- lua_tcp_push_error (cbd, TRUE,
- "IO read error: connection terminated");
+ lua_tcp_push_error(cbd, TRUE,
+ "IO read error: connection terminated");
}
}
}
}
else {
- msg_debug_tcp ("read TCP partial data %d bytes", cbd->in->len);
+ msg_debug_tcp("read TCP partial data %d bytes", cbd->in->len);
slen = cbd->in->len;
/* we have eaten all the data, handler should not know that there is something */
cbd->in->len = 0;
- lua_tcp_push_data (cbd, cbd->in->data, slen);
- if (!IS_SYNC (cbd)) {
- lua_tcp_shift_handler (cbd);
+ lua_tcp_push_data(cbd, cbd->in->data, slen);
+ if (!IS_SYNC(cbd)) {
+ lua_tcp_shift_handler(cbd);
}
return TRUE;
@@ -1003,34 +1005,34 @@ lua_tcp_process_read_handler (struct lua_tcp_cbdata *cbd,
}
static void
-lua_tcp_process_read (struct lua_tcp_cbdata *cbd,
- guchar *in, gssize r)
+lua_tcp_process_read(struct lua_tcp_cbdata *cbd,
+ guchar *in, gssize r)
{
struct lua_tcp_handler *hdl;
struct lua_tcp_read_handler *rh;
- hdl = g_queue_peek_head (cbd->handlers);
+ hdl = g_queue_peek_head(cbd->handlers);
- g_assert (hdl != NULL && hdl->type == LUA_WANT_READ);
+ g_assert(hdl != NULL && hdl->type == LUA_WANT_READ);
rh = &hdl->h.r;
if (r > 0) {
if (cbd->flags & LUA_TCP_FLAG_PARTIAL) {
- lua_tcp_push_data (cbd, in, r);
+ lua_tcp_push_data(cbd, in, r);
/* Plan next event */
- lua_tcp_plan_read (cbd);
+ lua_tcp_plan_read(cbd);
}
else {
- g_byte_array_append (cbd->in, in, r);
+ g_byte_array_append(cbd->in, in, r);
- if (!lua_tcp_process_read_handler (cbd, rh, FALSE)) {
+ if (!lua_tcp_process_read_handler(cbd, rh, FALSE)) {
/* Plan more read */
- lua_tcp_plan_read (cbd);
+ lua_tcp_plan_read(cbd);
}
else {
/* Go towards the next handler */
- if (!IS_SYNC (cbd)) {
- lua_tcp_plan_handler_event (cbd, TRUE, TRUE);
+ if (!IS_SYNC(cbd)) {
+ lua_tcp_plan_handler_event(cbd, TRUE, TRUE);
}
}
}
@@ -1040,24 +1042,24 @@ lua_tcp_process_read (struct lua_tcp_cbdata *cbd,
cbd->eof = TRUE;
if (cbd->in->len > 0) {
/* We have some data to process */
- lua_tcp_process_read_handler (cbd, rh, TRUE);
+ lua_tcp_process_read_handler(cbd, rh, TRUE);
}
else {
- lua_tcp_push_error (cbd, TRUE, "IO read error: connection terminated");
+ lua_tcp_push_error(cbd, TRUE, "IO read error: connection terminated");
if ((cbd->flags & LUA_TCP_FLAG_FINISHED)) {
/* A callback has called `close` method, so we need to release a refcount */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
}
- lua_tcp_plan_handler_event (cbd, FALSE, FALSE);
+ lua_tcp_plan_handler_event(cbd, FALSE, FALSE);
}
else {
/* An error occurred */
if (errno == EAGAIN || errno == EINTR) {
/* Restart call */
- lua_tcp_plan_read (cbd);
+ lua_tcp_plan_read(cbd);
return;
}
@@ -1066,66 +1068,66 @@ lua_tcp_process_read (struct lua_tcp_cbdata *cbd,
cbd->eof = TRUE;
if (cbd->in->len > 0) {
/* We have some data to process */
- lua_tcp_process_read_handler (cbd, rh, TRUE);
+ lua_tcp_process_read_handler(cbd, rh, TRUE);
}
else {
- lua_tcp_push_error (cbd, TRUE,
- "IO read error while trying to read data: %s",
- strerror (errno));
+ lua_tcp_push_error(cbd, TRUE,
+ "IO read error while trying to read data: %s",
+ strerror(errno));
if ((cbd->flags & LUA_TCP_FLAG_FINISHED)) {
/* A callback has called `close` method, so we need to release a refcount */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
}
- lua_tcp_plan_handler_event (cbd, FALSE, FALSE);
+ lua_tcp_plan_handler_event(cbd, FALSE, FALSE);
}
}
static void
-lua_tcp_handler (int fd, short what, gpointer ud)
+lua_tcp_handler(int fd, short what, gpointer ud)
{
struct lua_tcp_cbdata *cbd = ud;
guchar inbuf[8192];
gssize r;
gint so_error = 0;
- socklen_t so_len = sizeof (so_error);
+ socklen_t so_len = sizeof(so_error);
struct lua_callback_state cbs;
lua_State *L;
enum lua_tcp_handler_type event_type;
- TCP_RETAIN (cbd);
+ TCP_RETAIN(cbd);
- msg_debug_tcp ("processed TCP event: %d", what);
+ msg_debug_tcp("processed TCP event: %d", what);
- struct lua_tcp_handler *rh = g_queue_peek_head (cbd->handlers);
+ struct lua_tcp_handler *rh = g_queue_peek_head(cbd->handlers);
event_type = rh->type;
- rspamd_ev_watcher_stop (cbd->event_loop, &cbd->ev);
+ rspamd_ev_watcher_stop(cbd->event_loop, &cbd->ev);
if (what == EV_READ) {
if (cbd->ssl_conn) {
- r = rspamd_ssl_read (cbd->ssl_conn, inbuf, sizeof (inbuf));
+ r = rspamd_ssl_read(cbd->ssl_conn, inbuf, sizeof(inbuf));
}
else {
- r = read (cbd->fd, inbuf, sizeof (inbuf));
+ r = read(cbd->fd, inbuf, sizeof(inbuf));
}
- lua_tcp_process_read (cbd, inbuf, r);
+ lua_tcp_process_read(cbd, inbuf, r);
}
else if (what == EV_WRITE) {
if (!(cbd->flags & LUA_TCP_FLAG_CONNECTED)) {
- if (getsockopt (fd, SOL_SOCKET, SO_ERROR, &so_error, &so_len) == -1) {
- lua_tcp_push_error (cbd, TRUE, "Cannot get socket error: %s",
- strerror (errno));
- TCP_RELEASE (cbd);
+ if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &so_error, &so_len) == -1) {
+ lua_tcp_push_error(cbd, TRUE, "Cannot get socket error: %s",
+ strerror(errno));
+ TCP_RELEASE(cbd);
goto out;
}
else if (so_error != 0) {
- lua_tcp_push_error (cbd, TRUE, "Socket error detected: %s",
- strerror (so_error));
- TCP_RELEASE (cbd);
+ lua_tcp_push_error(cbd, TRUE, "Socket error detected: %s",
+ strerror(so_error));
+ TCP_RELEASE(cbd);
goto out;
}
else {
@@ -1135,76 +1137,76 @@ lua_tcp_handler (int fd, short what, gpointer ud)
struct lua_tcp_cbdata **pcbd;
gint top;
- lua_thread_pool_prepare_callback (cbd->cfg->lua_thread_pool, &cbs);
+ lua_thread_pool_prepare_callback(cbd->cfg->lua_thread_pool, &cbs);
L = cbs.L;
- top = lua_gettop (L);
- lua_rawgeti (L, LUA_REGISTRYINDEX, cbd->connect_cb);
- pcbd = lua_newuserdata (L, sizeof (*pcbd));
+ top = lua_gettop(L);
+ lua_rawgeti(L, LUA_REGISTRYINDEX, cbd->connect_cb);
+ pcbd = lua_newuserdata(L, sizeof(*pcbd));
*pcbd = cbd;
- TCP_RETAIN (cbd);
- rspamd_lua_setclass (L, "rspamd{tcp}", -1);
+ TCP_RETAIN(cbd);
+ rspamd_lua_setclass(L, "rspamd{tcp}", -1);
if (cbd->item) {
- rspamd_symcache_set_cur_item (cbd->task, cbd->item);
+ rspamd_symcache_set_cur_item(cbd->task, cbd->item);
}
- if (lua_pcall (L, 1, 0, 0) != 0) {
- msg_info ("callback call failed: %s", lua_tostring (L, -1));
+ if (lua_pcall(L, 1, 0, 0) != 0) {
+ msg_info("callback call failed: %s", lua_tostring(L, -1));
}
- lua_settop (L, top);
- TCP_RELEASE (cbd);
- lua_thread_pool_restore_callback (&cbs);
+ lua_settop(L, top);
+ TCP_RELEASE(cbd);
+ lua_thread_pool_restore_callback(&cbs);
- if ((cbd->flags & (LUA_TCP_FLAG_FINISHED|LUA_TCP_FLAG_CONNECTED)) ==
- (LUA_TCP_FLAG_FINISHED|LUA_TCP_FLAG_CONNECTED)) {
+ if ((cbd->flags & (LUA_TCP_FLAG_FINISHED | LUA_TCP_FLAG_CONNECTED)) ==
+ (LUA_TCP_FLAG_FINISHED | LUA_TCP_FLAG_CONNECTED)) {
/* A callback has called `close` method, so we need to release a refcount */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
}
}
}
if (event_type == LUA_WANT_WRITE) {
- lua_tcp_write_helper (cbd);
+ lua_tcp_write_helper(cbd);
}
else if (event_type == LUA_WANT_CONNECT) {
- lua_tcp_connect_helper (cbd);
+ lua_tcp_connect_helper(cbd);
}
else {
- g_assert_not_reached ();
+ g_assert_not_reached();
}
- if ((cbd->flags & (LUA_TCP_FLAG_FINISHED|LUA_TCP_FLAG_CONNECTED)) ==
- (LUA_TCP_FLAG_FINISHED|LUA_TCP_FLAG_CONNECTED)) {
+ if ((cbd->flags & (LUA_TCP_FLAG_FINISHED | LUA_TCP_FLAG_CONNECTED)) ==
+ (LUA_TCP_FLAG_FINISHED | LUA_TCP_FLAG_CONNECTED)) {
/* A callback has called `close` method, so we need to release a refcount */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
}
else {
- lua_tcp_push_error (cbd, TRUE, "IO timeout");
- TCP_RELEASE (cbd);
+ lua_tcp_push_error(cbd, TRUE, "IO timeout");
+ TCP_RELEASE(cbd);
}
out:
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
static void
-lua_tcp_plan_handler_event (struct lua_tcp_cbdata *cbd, gboolean can_read,
- gboolean can_write)
+lua_tcp_plan_handler_event(struct lua_tcp_cbdata *cbd, gboolean can_read,
+ gboolean can_write)
{
struct lua_tcp_handler *hdl;
- hdl = g_queue_peek_head (cbd->handlers);
+ hdl = g_queue_peek_head(cbd->handlers);
if (hdl == NULL) {
if (!(cbd->flags & LUA_TCP_FLAG_FINISHED)) {
/* We are finished with a connection */
- msg_debug_tcp ("no handlers left, finish session");
+ msg_debug_tcp("no handlers left, finish session");
cbd->flags |= LUA_TCP_FLAG_FINISHED;
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
}
else {
@@ -1212,28 +1214,28 @@ lua_tcp_plan_handler_event (struct lua_tcp_cbdata *cbd, gboolean can_read,
/* We need to check if we have some leftover in the buffer */
if (cbd->in->len > 0) {
- msg_debug_tcp ("process read buffer leftover");
- if (lua_tcp_process_read_handler (cbd, &hdl->h.r, FALSE)) {
+ msg_debug_tcp("process read buffer leftover");
+ if (lua_tcp_process_read_handler(cbd, &hdl->h.r, FALSE)) {
if (!IS_SYNC(cbd)) {
/* We can go to the next handler */
- lua_tcp_plan_handler_event (cbd, can_read, can_write);
+ lua_tcp_plan_handler_event(cbd, can_read, can_write);
}
}
}
else {
if (can_read) {
/* We need to plan a new event */
- msg_debug_tcp ("plan new read");
- rspamd_ev_watcher_reschedule (cbd->event_loop, &cbd->ev,
- EV_READ);
+ msg_debug_tcp("plan new read");
+ rspamd_ev_watcher_reschedule(cbd->event_loop, &cbd->ev,
+ EV_READ);
}
else {
/* Cannot read more */
- msg_debug_tcp ("cannot read more");
- lua_tcp_push_error (cbd, FALSE, "EOF, cannot read more data");
- if (!IS_SYNC (cbd)) {
- lua_tcp_shift_handler (cbd);
- lua_tcp_plan_handler_event (cbd, can_read, can_write);
+ msg_debug_tcp("cannot read more");
+ lua_tcp_push_error(cbd, FALSE, "EOF, cannot read more data");
+ if (!IS_SYNC(cbd)) {
+ lua_tcp_shift_handler(cbd);
+ lua_tcp_plan_handler_event(cbd, can_read, can_write);
}
}
}
@@ -1245,45 +1247,45 @@ lua_tcp_plan_handler_event (struct lua_tcp_cbdata *cbd, gboolean can_read,
*/
if (hdl->h.w.pos < hdl->h.w.total_bytes) {
- msg_debug_tcp ("plan new write");
+ msg_debug_tcp("plan new write");
if (can_write) {
- rspamd_ev_watcher_reschedule (cbd->event_loop, &cbd->ev,
- EV_WRITE);
+ rspamd_ev_watcher_reschedule(cbd->event_loop, &cbd->ev,
+ EV_WRITE);
}
else {
/* Cannot write more */
- lua_tcp_push_error (cbd, FALSE, "EOF, cannot write more data");
+ lua_tcp_push_error(cbd, FALSE, "EOF, cannot write more data");
if (!IS_SYNC(cbd)) {
- lua_tcp_shift_handler (cbd);
- lua_tcp_plan_handler_event (cbd, can_read, can_write);
+ lua_tcp_shift_handler(cbd);
+ lua_tcp_plan_handler_event(cbd, can_read, can_write);
}
}
}
else {
/* We shouldn't have empty write handlers */
- g_assert_not_reached ();
+ g_assert_not_reached();
}
}
else { /* LUA_WANT_CONNECT */
- msg_debug_tcp ("plan new connect");
- rspamd_ev_watcher_reschedule (cbd->event_loop, &cbd->ev,
- EV_WRITE);
+ msg_debug_tcp("plan new connect");
+ rspamd_ev_watcher_reschedule(cbd->event_loop, &cbd->ev,
+ EV_WRITE);
}
}
}
static gboolean
-lua_tcp_register_event (struct lua_tcp_cbdata *cbd)
+lua_tcp_register_event(struct lua_tcp_cbdata *cbd)
{
if (cbd->session) {
- event_finalizer_t fin = IS_SYNC (cbd) ? lua_tcp_void_finalyser : lua_tcp_fin;
+ event_finalizer_t fin = IS_SYNC(cbd) ? lua_tcp_void_finalyser : lua_tcp_fin;
if (cbd->item) {
- cbd->async_ev = rspamd_session_add_event_full (cbd->session, fin, cbd, M,
- rspamd_symcache_dyn_item_name (cbd->task, cbd->item));
+ cbd->async_ev = rspamd_session_add_event_full(cbd->session, fin, cbd, M,
+ rspamd_symcache_dyn_item_name(cbd->task, cbd->item));
}
else {
- cbd->async_ev = rspamd_session_add_event (cbd->session, fin, cbd, M);
+ cbd->async_ev = rspamd_session_add_event(cbd->session, fin, cbd, M);
}
if (!cbd->async_ev) {
@@ -1295,49 +1297,49 @@ lua_tcp_register_event (struct lua_tcp_cbdata *cbd)
}
static void
-lua_tcp_register_watcher (struct lua_tcp_cbdata *cbd)
+lua_tcp_register_watcher(struct lua_tcp_cbdata *cbd)
{
if (cbd->item && cbd->task) {
- rspamd_symcache_item_async_inc (cbd->task, cbd->item, M);
+ rspamd_symcache_item_async_inc(cbd->task, cbd->item, M);
}
}
static void
-lua_tcp_ssl_on_error (gpointer ud, GError *err)
+lua_tcp_ssl_on_error(gpointer ud, GError *err)
{
- struct lua_tcp_cbdata *cbd = (struct lua_tcp_cbdata *)ud;
+ struct lua_tcp_cbdata *cbd = (struct lua_tcp_cbdata *) ud;
if (err) {
- lua_tcp_push_error (cbd, TRUE, "ssl error: %s", err->message);
+ lua_tcp_push_error(cbd, TRUE, "ssl error: %s", err->message);
}
else {
- lua_tcp_push_error (cbd, TRUE, "ssl error: unknown error");
+ lua_tcp_push_error(cbd, TRUE, "ssl error: unknown error");
}
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
}
static gboolean
-lua_tcp_make_connection (struct lua_tcp_cbdata *cbd)
+lua_tcp_make_connection(struct lua_tcp_cbdata *cbd)
{
int fd;
- rspamd_inet_address_set_port (cbd->addr, cbd->port);
- fd = rspamd_inet_address_connect (cbd->addr, SOCK_STREAM, TRUE);
+ rspamd_inet_address_set_port(cbd->addr, cbd->port);
+ fd = rspamd_inet_address_connect(cbd->addr, SOCK_STREAM, TRUE);
if (fd == -1) {
if (cbd->session) {
- rspamd_mempool_t *pool = rspamd_session_mempool (cbd->session);
- msg_info_pool ("cannot connect to %s (%s): %s",
- rspamd_inet_address_to_string (cbd->addr),
- cbd->hostname,
- strerror (errno));
+ rspamd_mempool_t *pool = rspamd_session_mempool(cbd->session);
+ msg_info_pool("cannot connect to %s (%s): %s",
+ rspamd_inet_address_to_string(cbd->addr),
+ cbd->hostname,
+ strerror(errno));
}
else {
- msg_info ("cannot connect to %s (%s): %s",
- rspamd_inet_address_to_string (cbd->addr),
- cbd->hostname,
- strerror (errno));
+ msg_info("cannot connect to %s (%s): %s",
+ rspamd_inet_address_to_string(cbd->addr),
+ cbd->hostname,
+ strerror(errno));
}
return FALSE;
@@ -1368,27 +1370,27 @@ lua_tcp_make_connection (struct lua_tcp_cbdata *cbd)
verify_peer = TRUE;
}
- cbd->ssl_conn = rspamd_ssl_connection_new (ssl_ctx,
- cbd->event_loop,
- verify_peer,
- cbd->tag);
+ cbd->ssl_conn = rspamd_ssl_connection_new(ssl_ctx,
+ cbd->event_loop,
+ verify_peer,
+ cbd->tag);
- if (!rspamd_ssl_connect_fd (cbd->ssl_conn, fd, cbd->hostname, &cbd->ev,
- cbd->ev.timeout, lua_tcp_handler, lua_tcp_ssl_on_error, cbd)) {
- lua_tcp_push_error (cbd, TRUE, "ssl connection failed: %s",
- strerror (errno));
+ if (!rspamd_ssl_connect_fd(cbd->ssl_conn, fd, cbd->hostname, &cbd->ev,
+ cbd->ev.timeout, lua_tcp_handler, lua_tcp_ssl_on_error, cbd)) {
+ lua_tcp_push_error(cbd, TRUE, "ssl connection failed: %s",
+ strerror(errno));
return FALSE;
}
else {
- lua_tcp_register_event (cbd);
+ lua_tcp_register_event(cbd);
}
}
else {
- rspamd_ev_watcher_init (&cbd->ev, cbd->fd, EV_WRITE,
- lua_tcp_handler, cbd);
- lua_tcp_register_event (cbd);
- lua_tcp_plan_handler_event (cbd, TRUE, TRUE);
+ rspamd_ev_watcher_init(&cbd->ev, cbd->fd, EV_WRITE,
+ lua_tcp_handler, cbd);
+ lua_tcp_register_event(cbd);
+ lua_tcp_plan_handler_event(cbd, TRUE, TRUE);
}
@@ -1396,16 +1398,16 @@ lua_tcp_make_connection (struct lua_tcp_cbdata *cbd)
}
static void
-lua_tcp_dns_handler (struct rdns_reply *reply, gpointer ud)
+lua_tcp_dns_handler(struct rdns_reply *reply, gpointer ud)
{
- struct lua_tcp_cbdata *cbd = (struct lua_tcp_cbdata *)ud;
+ struct lua_tcp_cbdata *cbd = (struct lua_tcp_cbdata *) ud;
const struct rdns_request_name *rn;
if (reply->code != RDNS_RC_NOERROR) {
- rn = rdns_request_get_name (reply->request, NULL);
- lua_tcp_push_error (cbd, TRUE, "unable to resolve host: %s",
- rn->name);
- TCP_RELEASE (cbd);
+ rn = rdns_request_get_name(reply->request, NULL);
+ lua_tcp_push_error(cbd, TRUE, "unable to resolve host: %s",
+ rn->name);
+ TCP_RELEASE(cbd);
}
else {
/*
@@ -1414,80 +1416,81 @@ lua_tcp_dns_handler (struct rdns_reply *reply, gpointer ud)
*/
struct rdns_reply_entry *entry;
- DL_FOREACH(reply->entries, entry) {
+ DL_FOREACH(reply->entries, entry)
+ {
if (entry->type == RDNS_REQUEST_A) {
cbd->addr = rspamd_inet_address_new(AF_INET,
- &entry->content.a.addr);
+ &entry->content.a.addr);
break;
}
else if (entry->type == RDNS_REQUEST_AAAA) {
cbd->addr = rspamd_inet_address_new(AF_INET6,
- &entry->content.aaa.addr);
+ &entry->content.aaa.addr);
break;
}
}
- if (cbd->addr == NULL){
- rn = rdns_request_get_name (reply->request, NULL);
- lua_tcp_push_error (cbd, TRUE, "unable to resolve host: %s; no records with this name",
- rn->name);
- TCP_RELEASE (cbd);
+ if (cbd->addr == NULL) {
+ rn = rdns_request_get_name(reply->request, NULL);
+ lua_tcp_push_error(cbd, TRUE, "unable to resolve host: %s; no records with this name",
+ rn->name);
+ TCP_RELEASE(cbd);
return;
}
cbd->flags |= LUA_TCP_FLAG_RESOLVED;
- rspamd_inet_address_set_port (cbd->addr, cbd->port);
+ rspamd_inet_address_set_port(cbd->addr, cbd->port);
- if (!lua_tcp_make_connection (cbd)) {
- lua_tcp_push_error (cbd, TRUE, "unable to make connection to the host %s",
- rspamd_inet_address_to_string (cbd->addr));
- TCP_RELEASE (cbd);
+ if (!lua_tcp_make_connection(cbd)) {
+ lua_tcp_push_error(cbd, TRUE, "unable to make connection to the host %s",
+ rspamd_inet_address_to_string(cbd->addr));
+ TCP_RELEASE(cbd);
}
}
}
static gboolean
-lua_tcp_arg_toiovec (lua_State *L, gint pos, struct lua_tcp_cbdata *cbd,
- struct iovec *vec)
+lua_tcp_arg_toiovec(lua_State *L, gint pos, struct lua_tcp_cbdata *cbd,
+ struct iovec *vec)
{
struct rspamd_lua_text *t;
gsize len;
const gchar *str;
struct lua_tcp_dtor *dtor;
- if (lua_type (L, pos) == LUA_TUSERDATA) {
- t = lua_check_text (L, pos);
+ if (lua_type(L, pos) == LUA_TUSERDATA) {
+ t = lua_check_text(L, pos);
if (t) {
- vec->iov_base = (void *)t->start;
+ vec->iov_base = (void *) t->start;
vec->iov_len = t->len;
if (t->flags & RSPAMD_TEXT_FLAG_OWN) {
/* Steal ownership */
t->flags = 0;
- dtor = g_malloc0 (sizeof (*dtor));
+ dtor = g_malloc0(sizeof(*dtor));
dtor->dtor = g_free;
- dtor->data = (void *)t->start;
- LL_PREPEND (cbd->dtors, dtor);
+ dtor->data = (void *) t->start;
+ LL_PREPEND(cbd->dtors, dtor);
}
}
else {
- msg_err ("bad userdata argument at position %d", pos);
+ msg_err("bad userdata argument at position %d", pos);
return FALSE;
}
}
- else if (lua_type (L, pos) == LUA_TSTRING) {
- str = luaL_checklstring (L, pos, &len);
- vec->iov_base = g_malloc (len);
- dtor = g_malloc0 (sizeof (*dtor));
+ else if (lua_type(L, pos) == LUA_TSTRING) {
+ str = luaL_checklstring(L, pos, &len);
+ vec->iov_base = g_malloc(len);
+ dtor = g_malloc0(sizeof(*dtor));
dtor->dtor = g_free;
dtor->data = vec->iov_base;
- LL_PREPEND (cbd->dtors, dtor);
- memcpy (vec->iov_base, str, len);
+ LL_PREPEND(cbd->dtors, dtor);
+ memcpy(vec->iov_base, str, len);
vec->iov_len = len;
}
else {
- msg_err ("bad argument at position %d", pos);
+ msg_err("bad argument at position %d", pos);
return FALSE;
}
@@ -1519,7 +1522,7 @@ lua_tcp_arg_toiovec (lua_State *L, gint pos, struct lua_tcp_cbdata *cbd,
* @return {boolean} true if request has been sent
*/
static gint
-lua_tcp_request (lua_State *L)
+lua_tcp_request(lua_State *L)
{
LUA_TRACE_POINT;
const gchar *host;
@@ -1539,175 +1542,175 @@ lua_tcp_request (lua_State *L)
guint64 h;
gdouble timeout = default_tcp_timeout;
gboolean partial = FALSE, do_shutdown = FALSE, do_read = TRUE,
- ssl = FALSE, ssl_noverify = FALSE;
+ ssl = FALSE, ssl_noverify = FALSE;
- if (lua_type (L, 1) == LUA_TTABLE) {
- lua_pushstring (L, "host");
- lua_gettable (L, -2);
- host = luaL_checkstring (L, -1);
- lua_pop (L, 1);
+ if (lua_type(L, 1) == LUA_TTABLE) {
+ lua_pushstring(L, "host");
+ lua_gettable(L, -2);
+ host = luaL_checkstring(L, -1);
+ lua_pop(L, 1);
- lua_pushstring (L, "port");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TNUMBER) {
- port = lua_tointeger (L, -1);
+ lua_pushstring(L, "port");
+ lua_gettable(L, -2);
+ if (lua_type(L, -1) == LUA_TNUMBER) {
+ port = lua_tointeger(L, -1);
}
else {
/* We assume that it is a unix socket */
port = 0;
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "callback");
- lua_gettable (L, -2);
- if (host == NULL || lua_type (L, -1) != LUA_TFUNCTION) {
- lua_pop (L, 1);
- msg_err ("tcp request has bad params");
- lua_pushboolean (L, FALSE);
+ lua_pushstring(L, "callback");
+ lua_gettable(L, -2);
+ if (host == NULL || lua_type(L, -1) != LUA_TFUNCTION) {
+ lua_pop(L, 1);
+ msg_err("tcp request has bad params");
+ lua_pushboolean(L, FALSE);
return 1;
}
- cbref = luaL_ref (L, LUA_REGISTRYINDEX);
+ cbref = luaL_ref(L, LUA_REGISTRYINDEX);
- cbd = g_malloc0 (sizeof (*cbd));
+ cbd = g_malloc0(sizeof(*cbd));
- lua_pushstring (L, "task");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TUSERDATA) {
- task = lua_check_task (L, -1);
+ lua_pushstring(L, "task");
+ lua_gettable(L, -2);
+ if (lua_type(L, -1) == LUA_TUSERDATA) {
+ task = lua_check_task(L, -1);
event_loop = task->event_loop;
resolver = task->resolver;
session = task->s;
cfg = task->cfg;
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
if (task == NULL) {
- lua_pushstring (L, "ev_base");
- lua_gettable (L, -2);
- if (rspamd_lua_check_udata_maybe (L, -1, "rspamd{ev_base}")) {
- event_loop = *(struct ev_loop **)lua_touserdata (L, -1);
+ lua_pushstring(L, "ev_base");
+ lua_gettable(L, -2);
+ if (rspamd_lua_check_udata_maybe(L, -1, "rspamd{ev_base}")) {
+ event_loop = *(struct ev_loop **) lua_touserdata(L, -1);
}
else {
- g_free (cbd);
+ g_free(cbd);
- return luaL_error (L, "event loop is required");
+ return luaL_error(L, "event loop is required");
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "session");
- lua_gettable (L, -2);
- if (rspamd_lua_check_udata_maybe (L, -1, "rspamd{session}")) {
- session = *(struct rspamd_async_session **)lua_touserdata (L, -1);
+ lua_pushstring(L, "session");
+ lua_gettable(L, -2);
+ if (rspamd_lua_check_udata_maybe(L, -1, "rspamd{session}")) {
+ session = *(struct rspamd_async_session **) lua_touserdata(L, -1);
}
else {
session = NULL;
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "config");
- lua_gettable (L, -2);
- if (rspamd_lua_check_udata_maybe (L, -1, "rspamd{config}")) {
- cfg = *(struct rspamd_config **)lua_touserdata (L, -1);
+ lua_pushstring(L, "config");
+ lua_gettable(L, -2);
+ if (rspamd_lua_check_udata_maybe(L, -1, "rspamd{config}")) {
+ cfg = *(struct rspamd_config **) lua_touserdata(L, -1);
}
else {
cfg = NULL;
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "resolver");
- lua_gettable (L, -2);
- if (rspamd_lua_check_udata_maybe (L, -1, "rspamd{resolver}")) {
- resolver = *(struct rspamd_dns_resolver **)lua_touserdata (L, -1);
+ lua_pushstring(L, "resolver");
+ lua_gettable(L, -2);
+ if (rspamd_lua_check_udata_maybe(L, -1, "rspamd{resolver}")) {
+ resolver = *(struct rspamd_dns_resolver **) lua_touserdata(L, -1);
}
else {
- resolver = lua_tcp_global_resolver (event_loop, cfg);
+ resolver = lua_tcp_global_resolver(event_loop, cfg);
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
}
- lua_pushstring (L, "timeout");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TNUMBER) {
- timeout = lua_tonumber (L, -1);
+ lua_pushstring(L, "timeout");
+ lua_gettable(L, -2);
+ if (lua_type(L, -1) == LUA_TNUMBER) {
+ timeout = lua_tonumber(L, -1);
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "stop_pattern");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TSTRING) {
+ lua_pushstring(L, "stop_pattern");
+ lua_gettable(L, -2);
+ if (lua_type(L, -1) == LUA_TSTRING) {
const gchar *p;
- p = lua_tolstring (L, -1, &plen);
+ p = lua_tolstring(L, -1, &plen);
if (p && plen > 0) {
- stop_pattern = g_malloc (plen);
- memcpy (stop_pattern, p, plen);
+ stop_pattern = g_malloc(plen);
+ memcpy(stop_pattern, p, plen);
}
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "partial");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- partial = lua_toboolean (L, -1);
+ lua_pushstring(L, "partial");
+ lua_gettable(L, -2);
+ if (lua_type(L, -1) == LUA_TBOOLEAN) {
+ partial = lua_toboolean(L, -1);
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "shutdown");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- do_shutdown = lua_toboolean (L, -1);
+ lua_pushstring(L, "shutdown");
+ lua_gettable(L, -2);
+ if (lua_type(L, -1) == LUA_TBOOLEAN) {
+ do_shutdown = lua_toboolean(L, -1);
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "read");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- do_read = lua_toboolean (L, -1);
+ lua_pushstring(L, "read");
+ lua_gettable(L, -2);
+ if (lua_type(L, -1) == LUA_TBOOLEAN) {
+ do_read = lua_toboolean(L, -1);
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "ssl");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- ssl = lua_toboolean (L, -1);
+ lua_pushstring(L, "ssl");
+ lua_gettable(L, -2);
+ if (lua_type(L, -1) == LUA_TBOOLEAN) {
+ ssl = lua_toboolean(L, -1);
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "ssl_noverify");
- lua_gettable (L, -2);
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- ssl_noverify = lua_toboolean (L, -1);
- lua_pop (L, 1);
+ lua_pushstring(L, "ssl_noverify");
+ lua_gettable(L, -2);
+ if (lua_type(L, -1) == LUA_TBOOLEAN) {
+ ssl_noverify = lua_toboolean(L, -1);
+ lua_pop(L, 1);
}
else {
- lua_pop (L, 1); /* Previous nil... */
+ lua_pop(L, 1); /* Previous nil... */
/* Similar to lua http, meh... */
- lua_pushstring (L, "no_ssl_verify");
- lua_gettable (L, -2);
+ lua_pushstring(L, "no_ssl_verify");
+ lua_gettable(L, -2);
- if (lua_type (L, -1) == LUA_TBOOLEAN) {
- ssl_noverify = lua_toboolean (L, -1);
+ if (lua_type(L, -1) == LUA_TBOOLEAN) {
+ ssl_noverify = lua_toboolean(L, -1);
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
}
- lua_pushstring (L, "on_connect");
- lua_gettable (L, -2);
+ lua_pushstring(L, "on_connect");
+ lua_gettable(L, -2);
- if (lua_type (L, -1) == LUA_TFUNCTION) {
- conn_cbref = luaL_ref (L, LUA_REGISTRYINDEX);
+ if (lua_type(L, -1) == LUA_TFUNCTION) {
+ conn_cbref = luaL_ref(L, LUA_REGISTRYINDEX);
}
else {
- lua_pop (L, 1);
+ lua_pop(L, 1);
}
- lua_pushstring (L, "upstream");
- lua_gettable (L, 1);
+ lua_pushstring(L, "upstream");
+ lua_gettable(L, 1);
- if (lua_type (L, -1) == LUA_TUSERDATA) {
+ if (lua_type(L, -1) == LUA_TUSERDATA) {
struct rspamd_lua_upstream *lup = lua_check_upstream(L, -1);
if (lup) {
@@ -1716,23 +1719,23 @@ lua_tcp_request (lua_State *L)
}
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
- lua_pushstring (L, "data");
- lua_gettable (L, -2);
+ lua_pushstring(L, "data");
+ lua_gettable(L, -2);
total_out = 0;
- tp = lua_type (L, -1);
+ tp = lua_type(L, -1);
if (tp == LUA_TSTRING || tp == LUA_TUSERDATA) {
- iov = g_malloc (sizeof (*iov));
+ iov = g_malloc(sizeof(*iov));
niov = 1;
- if (!lua_tcp_arg_toiovec (L, -1, cbd, iov)) {
- lua_pop (L, 1);
- msg_err ("tcp request has bad data argument");
- lua_pushboolean (L, FALSE);
- g_free (iov);
- g_free (cbd);
+ if (!lua_tcp_arg_toiovec(L, -1, cbd, iov)) {
+ lua_pop(L, 1);
+ msg_err("tcp request has bad data argument");
+ lua_pushboolean(L, FALSE);
+ g_free(iov);
+ g_free(cbd);
return 1;
}
@@ -1741,64 +1744,64 @@ lua_tcp_request (lua_State *L)
}
else if (tp == LUA_TTABLE) {
/* Count parts */
- lua_pushnil (L);
- while (lua_next (L, -2) != 0) {
- niov ++;
- lua_pop (L, 1);
+ lua_pushnil(L);
+ while (lua_next(L, -2) != 0) {
+ niov++;
+ lua_pop(L, 1);
}
- iov = g_malloc (sizeof (*iov) * niov);
- lua_pushnil (L);
+ iov = g_malloc(sizeof(*iov) * niov);
+ lua_pushnil(L);
niov = 0;
- while (lua_next (L, -2) != 0) {
- if (!lua_tcp_arg_toiovec (L, -1, cbd, &iov[niov])) {
- lua_pop (L, 2);
- msg_err ("tcp request has bad data argument at pos %d", niov);
- lua_pushboolean (L, FALSE);
- g_free (iov);
- g_free (cbd);
+ while (lua_next(L, -2) != 0) {
+ if (!lua_tcp_arg_toiovec(L, -1, cbd, &iov[niov])) {
+ lua_pop(L, 2);
+ msg_err("tcp request has bad data argument at pos %d", niov);
+ lua_pushboolean(L, FALSE);
+ g_free(iov);
+ g_free(cbd);
return 1;
}
total_out += iov[niov].iov_len;
- niov ++;
+ niov++;
- lua_pop (L, 1);
+ lua_pop(L, 1);
}
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
}
else {
- return luaL_error (L, "tcp request has bad params");
+ return luaL_error(L, "tcp request has bad params");
}
if (resolver == NULL && cfg == NULL && task == NULL) {
- g_free (cbd);
- g_free (iov);
+ g_free(cbd);
+ g_free(iov);
- return luaL_error (L, "tcp request has bad params: one of "
- "{resolver,task,config} should be set");
+ return luaL_error(L, "tcp request has bad params: one of "
+ "{resolver,task,config} should be set");
}
cbd->task = task;
if (task) {
- cbd->item = rspamd_symcache_get_cur_item (task);
+ cbd->item = rspamd_symcache_get_cur_item(task);
}
cbd->cfg = cfg;
- h = rspamd_random_uint64_fast ();
- rspamd_snprintf (cbd->tag, sizeof (cbd->tag), "%uxL", h);
- cbd->handlers = g_queue_new ();
- cbd->hostname = g_strdup (host);
+ h = rspamd_random_uint64_fast();
+ rspamd_snprintf(cbd->tag, sizeof(cbd->tag), "%uxL", h);
+ cbd->handlers = g_queue_new();
+ cbd->hostname = g_strdup(host);
if (total_out > 0) {
struct lua_tcp_handler *wh;
- wh = g_malloc0 (sizeof (*wh));
+ wh = g_malloc0(sizeof(*wh));
wh->type = LUA_WANT_WRITE;
wh->h.w.iov = iov;
wh->h.w.iovlen = niov;
@@ -1816,7 +1819,7 @@ lua_tcp_request (lua_State *L)
wh->h.w.cbref = -1;
}
- g_queue_push_tail (cbd->handlers, wh);
+ g_queue_push_tail(cbd->handlers, wh);
}
cbd->event_loop = event_loop;
@@ -1833,11 +1836,11 @@ lua_tcp_request (lua_State *L)
}
if (do_read) {
- cbd->in = g_byte_array_sized_new (8192);
+ cbd->in = g_byte_array_sized_new(8192);
}
else {
/* Save some space... */
- cbd->in = g_byte_array_new ();
+ cbd->in = g_byte_array_new();
}
if (partial) {
@@ -1851,16 +1854,16 @@ lua_tcp_request (lua_State *L)
if (do_read) {
struct lua_tcp_handler *rh;
- rh = g_malloc0 (sizeof (*rh));
+ rh = g_malloc0(sizeof(*rh));
rh->type = LUA_WANT_READ;
rh->h.r.cbref = cbref;
rh->h.r.stop_pattern = stop_pattern;
rh->h.r.plen = plen;
- g_queue_push_tail (cbd->handlers, rh);
+ g_queue_push_tail(cbd->handlers, rh);
}
cbd->connect_cb = conn_cbref;
- REF_INIT_RETAIN (cbd, lua_tcp_maybe_free);
+ REF_INIT_RETAIN(cbd, lua_tcp_maybe_free);
if (up) {
cbd->up = rspamd_upstream_ref(up);
@@ -1869,11 +1872,11 @@ lua_tcp_request (lua_State *L)
if (session) {
cbd->session = session;
- if (rspamd_session_blocked (session)) {
- lua_tcp_push_error (cbd, TRUE, "async session is the blocked state");
- TCP_RELEASE (cbd);
+ if (rspamd_session_blocked(session)) {
+ lua_tcp_push_error(cbd, TRUE, "async session is the blocked state");
+ TCP_RELEASE(cbd);
cbd->item = NULL; /* To avoid decrease with no watcher */
- lua_pushboolean (L, FALSE);
+ lua_pushboolean(L, FALSE);
return 1;
}
@@ -1884,69 +1887,69 @@ lua_tcp_request (lua_State *L)
cbd->addr = rspamd_inet_address_copy(rspamd_upstream_addr_next(cbd->up), NULL);
/* Host is numeric IP, no need to resolve */
- lua_tcp_register_watcher (cbd);
+ lua_tcp_register_watcher(cbd);
- if (!lua_tcp_make_connection (cbd)) {
- lua_tcp_push_error (cbd, TRUE, "cannot connect to the host: %s", host);
- lua_pushboolean (L, FALSE);
+ if (!lua_tcp_make_connection(cbd)) {
+ lua_tcp_push_error(cbd, TRUE, "cannot connect to the host: %s", host);
+ lua_pushboolean(L, FALSE);
rspamd_upstream_fail(cbd->up, true, "failed to connect");
/* No reset of the item as watcher has been registered */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
return 1;
}
}
- else if (rspamd_parse_inet_address (&cbd->addr,
- host, strlen (host), RSPAMD_INET_ADDRESS_PARSE_DEFAULT)) {
- rspamd_inet_address_set_port (cbd->addr, port);
+ else if (rspamd_parse_inet_address(&cbd->addr,
+ host, strlen(host), RSPAMD_INET_ADDRESS_PARSE_DEFAULT)) {
+ rspamd_inet_address_set_port(cbd->addr, port);
/* Host is numeric IP, no need to resolve */
- lua_tcp_register_watcher (cbd);
+ lua_tcp_register_watcher(cbd);
- if (!lua_tcp_make_connection (cbd)) {
- lua_tcp_push_error (cbd, TRUE, "cannot connect to the host: %s", host);
- lua_pushboolean (L, FALSE);
+ if (!lua_tcp_make_connection(cbd)) {
+ lua_tcp_push_error(cbd, TRUE, "cannot connect to the host: %s", host);
+ lua_pushboolean(L, FALSE);
/* No reset of the item as watcher has been registered */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
return 1;
}
}
else {
if (task == NULL) {
- if (!rspamd_dns_resolver_request (resolver, session, NULL, lua_tcp_dns_handler, cbd,
- RDNS_REQUEST_A, host)) {
- lua_tcp_push_error (cbd, TRUE, "cannot resolve host: %s", host);
- lua_pushboolean (L, FALSE);
+ if (!rspamd_dns_resolver_request(resolver, session, NULL, lua_tcp_dns_handler, cbd,
+ RDNS_REQUEST_A, host)) {
+ lua_tcp_push_error(cbd, TRUE, "cannot resolve host: %s", host);
+ lua_pushboolean(L, FALSE);
cbd->item = NULL; /* To avoid decrease with no watcher */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
return 1;
}
else {
- lua_tcp_register_watcher (cbd);
+ lua_tcp_register_watcher(cbd);
}
}
else {
- if (!rspamd_dns_resolver_request_task (task, lua_tcp_dns_handler, cbd,
- RDNS_REQUEST_A, host)) {
- lua_tcp_push_error (cbd, TRUE, "cannot resolve host: %s", host);
- lua_pushboolean (L, FALSE);
+ if (!rspamd_dns_resolver_request_task(task, lua_tcp_dns_handler, cbd,
+ RDNS_REQUEST_A, host)) {
+ lua_tcp_push_error(cbd, TRUE, "cannot resolve host: %s", host);
+ lua_pushboolean(L, FALSE);
cbd->item = NULL; /* To avoid decrease with no watcher */
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
return 1;
}
else {
- lua_tcp_register_watcher (cbd);
+ lua_tcp_register_watcher(cbd);
}
}
}
- lua_pushboolean (L, TRUE);
+ lua_pushboolean(L, TRUE);
return 1;
}
@@ -1965,7 +1968,7 @@ lua_tcp_request (lua_State *L)
* @return {boolean} true if request has been sent
*/
static gint
-lua_tcp_connect_sync (lua_State *L)
+lua_tcp_connect_sync(lua_State *L)
{
LUA_TRACE_POINT;
GError *err = NULL;
@@ -1984,38 +1987,38 @@ lua_tcp_connect_sync (lua_State *L)
struct lua_tcp_cbdata *cbd;
- int arguments_validated = rspamd_lua_parse_table_arguments (L, 1, &err,
- RSPAMD_LUA_PARSE_ARGUMENTS_DEFAULT,
- "task=U{task};session=U{session};resolver=U{resolver};ev_base=U{ev_base};"
- "*host=S;*port=I;timeout=D;config=U{config}",
- &task, &session, &resolver, &ev_base,
- &host, &port, &timeout, &cfg);
+ int arguments_validated = rspamd_lua_parse_table_arguments(L, 1, &err,
+ RSPAMD_LUA_PARSE_ARGUMENTS_DEFAULT,
+ "task=U{task};session=U{session};resolver=U{resolver};ev_base=U{ev_base};"
+ "*host=S;*port=I;timeout=D;config=U{config}",
+ &task, &session, &resolver, &ev_base,
+ &host, &port, &timeout, &cfg);
if (!arguments_validated) {
if (err) {
- ret = luaL_error (L, "invalid arguments: %s", err->message);
- g_error_free (err);
+ ret = luaL_error(L, "invalid arguments: %s", err->message);
+ g_error_free(err);
return ret;
}
- return luaL_error (L, "invalid arguments");
+ return luaL_error(L, "invalid arguments");
}
if (0 > port || port > 65535) {
- return luaL_error (L, "invalid port given (correct values: 1..65535)");
+ return luaL_error(L, "invalid port given (correct values: 1..65535)");
}
if (task == NULL && (cfg == NULL || ev_base == NULL || session == NULL)) {
- return luaL_error (L, "invalid arguments: either task or config+ev_base+session should be set");
+ return luaL_error(L, "invalid arguments: either task or config+ev_base+session should be set");
}
- if (isnan (timeout)) {
+ if (isnan(timeout)) {
/* rspamd_lua_parse_table_arguments() sets missing N field to zero */
timeout = default_tcp_timeout;
}
- cbd = g_new0 (struct lua_tcp_cbdata, 1);
+ cbd = g_new0(struct lua_tcp_cbdata, 1);
if (task) {
static const gchar hexdigests[16] = "0123456789abcdef";
@@ -2024,13 +2027,13 @@ lua_tcp_connect_sync (lua_State *L)
ev_base = task->event_loop;
session = task->s;
/* Make a readable tag */
- memcpy (cbd->tag, task->task_pool->tag.uid, sizeof (cbd->tag) - 2);
- cbd->tag[sizeof (cbd->tag) - 2] = hexdigests[GPOINTER_TO_INT (cbd) & 0xf];
- cbd->tag[sizeof (cbd->tag) - 1] = 0;
+ memcpy(cbd->tag, task->task_pool->tag.uid, sizeof(cbd->tag) - 2);
+ cbd->tag[sizeof(cbd->tag) - 2] = hexdigests[GPOINTER_TO_INT(cbd) & 0xf];
+ cbd->tag[sizeof(cbd->tag) - 1] = 0;
}
else {
- h = rspamd_random_uint64_fast ();
- rspamd_snprintf (cbd->tag, sizeof (cbd->tag), "%uxL", h);
+ h = rspamd_random_uint64_fast();
+ rspamd_snprintf(cbd->tag, sizeof(cbd->tag), "%uxL", h);
}
if (resolver == NULL) {
@@ -2038,126 +2041,126 @@ lua_tcp_connect_sync (lua_State *L)
resolver = task->resolver;
}
else {
- resolver = lua_tcp_global_resolver (ev_base, cfg);
+ resolver = lua_tcp_global_resolver(ev_base, cfg);
}
}
cbd->task = task;
cbd->cfg = cfg;
- cbd->thread = lua_thread_pool_get_running_entry (cfg->lua_thread_pool);
+ cbd->thread = lua_thread_pool_get_running_entry(cfg->lua_thread_pool);
- cbd->handlers = g_queue_new ();
+ cbd->handlers = g_queue_new();
cbd->event_loop = ev_base;
cbd->flags |= LUA_TCP_FLAG_SYNC;
cbd->fd = -1;
- cbd->port = (guint16)port;
+ cbd->port = (guint16) port;
- cbd->in = g_byte_array_new ();
+ cbd->in = g_byte_array_new();
cbd->connect_cb = -1;
- REF_INIT_RETAIN (cbd, lua_tcp_maybe_free);
+ REF_INIT_RETAIN(cbd, lua_tcp_maybe_free);
if (task) {
- rspamd_mempool_add_destructor (task->task_pool, lua_tcp_sync_session_dtor, cbd);
+ rspamd_mempool_add_destructor(task->task_pool, lua_tcp_sync_session_dtor, cbd);
}
struct lua_tcp_handler *wh;
- wh = g_malloc0 (sizeof (*wh));
+ wh = g_malloc0(sizeof(*wh));
wh->type = LUA_WANT_CONNECT;
- g_queue_push_tail (cbd->handlers, wh);
+ g_queue_push_tail(cbd->handlers, wh);
if (session) {
cbd->session = session;
- if (rspamd_session_blocked (session)) {
- TCP_RELEASE (cbd);
- lua_pushboolean (L, FALSE);
- lua_pushliteral (L, "Session is being destroyed, requests are not allowed");
+ if (rspamd_session_blocked(session)) {
+ TCP_RELEASE(cbd);
+ lua_pushboolean(L, FALSE);
+ lua_pushliteral(L, "Session is being destroyed, requests are not allowed");
return 2;
}
}
- if (rspamd_parse_inet_address (&cbd->addr,
- host, strlen (host), RSPAMD_INET_ADDRESS_PARSE_DEFAULT)) {
- rspamd_inet_address_set_port (cbd->addr, (guint16)port);
+ if (rspamd_parse_inet_address(&cbd->addr,
+ host, strlen(host), RSPAMD_INET_ADDRESS_PARSE_DEFAULT)) {
+ rspamd_inet_address_set_port(cbd->addr, (guint16) port);
/* Host is numeric IP, no need to resolve */
- if (!lua_tcp_make_connection (cbd)) {
- lua_pushboolean (L, FALSE);
- lua_pushliteral (L, "Failed to initiate connection");
+ if (!lua_tcp_make_connection(cbd)) {
+ lua_pushboolean(L, FALSE);
+ lua_pushliteral(L, "Failed to initiate connection");
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
return 2;
}
}
else {
if (task == NULL) {
- if (!rspamd_dns_resolver_request (resolver, session, NULL, lua_tcp_dns_handler, cbd,
- RDNS_REQUEST_A, host)) {
- lua_pushboolean (L, FALSE);
- lua_pushliteral (L, "Failed to initiate dns request");
+ if (!rspamd_dns_resolver_request(resolver, session, NULL, lua_tcp_dns_handler, cbd,
+ RDNS_REQUEST_A, host)) {
+ lua_pushboolean(L, FALSE);
+ lua_pushliteral(L, "Failed to initiate dns request");
- TCP_RELEASE (cbd);
+ TCP_RELEASE(cbd);
return 2;
}
else {
- lua_tcp_register_watcher (cbd);
+ lua_tcp_register_watcher(cbd);
}
}
else {
- cbd->item = rspamd_symcache_get_cur_item (task);
+ cbd->item = rspamd_symcache_get_cur_item(task);
- if (!rspamd_dns_resolver_request_task (task, lua_tcp_dns_handler, cbd,
- RDNS_REQUEST_A, host)) {
+ if (!rspamd_dns_resolver_request_task(task, lua_tcp_dns_handler, cbd,
+ RDNS_REQUEST_A, host)) {
cbd->item = NULL; /* We have not registered watcher */
- lua_pushboolean (L, FALSE);
- lua_pushliteral (L, "Failed to initiate dns request");
- TCP_RELEASE (cbd);
+ lua_pushboolean(L, FALSE);
+ lua_pushliteral(L, "Failed to initiate dns request");
+ TCP_RELEASE(cbd);
return 2;
}
else {
- lua_tcp_register_watcher (cbd);
+ lua_tcp_register_watcher(cbd);
}
}
}
- return lua_thread_yield (cbd->thread, 0);
+ return lua_thread_yield(cbd->thread, 0);
}
static gint
-lua_tcp_close (lua_State *L)
+lua_tcp_close(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_tcp(L, 1);
if (cbd == NULL) {
- return luaL_error (L, "invalid arguments");
+ return luaL_error(L, "invalid arguments");
}
cbd->flags |= LUA_TCP_FLAG_FINISHED;
if (cbd->ssl_conn) {
/* TODO: postpone close in case ssl is used ! */
- rspamd_ssl_connection_free (cbd->ssl_conn);
+ rspamd_ssl_connection_free(cbd->ssl_conn);
cbd->ssl_conn = NULL;
}
if (cbd->fd != -1) {
- rspamd_ev_watcher_stop (cbd->event_loop, &cbd->ev);
- close (cbd->fd);
+ rspamd_ev_watcher_stop(cbd->event_loop, &cbd->ev);
+ close(cbd->fd);
cbd->fd = -1;
}
if (cbd->addr) {
- rspamd_inet_address_free (cbd->addr);
+ rspamd_inet_address_free(cbd->addr);
cbd->addr = NULL;
}
@@ -2172,10 +2175,10 @@ lua_tcp_close (lua_State *L)
}
static gint
-lua_tcp_add_read (lua_State *L)
+lua_tcp_add_read(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_tcp(L, 1);
struct lua_tcp_handler *rh;
gchar *stop_pattern = NULL;
const gchar *p;
@@ -2183,63 +2186,63 @@ lua_tcp_add_read (lua_State *L)
gint cbref = -1;
if (cbd == NULL) {
- return luaL_error (L, "invalid arguments");
+ return luaL_error(L, "invalid arguments");
}
- if (lua_type (L, 2) == LUA_TFUNCTION) {
- lua_pushvalue (L, 2);
- cbref = luaL_ref (L, LUA_REGISTRYINDEX);
+ if (lua_type(L, 2) == LUA_TFUNCTION) {
+ lua_pushvalue(L, 2);
+ cbref = luaL_ref(L, LUA_REGISTRYINDEX);
}
- if (lua_type (L, 3) == LUA_TSTRING) {
- p = lua_tolstring (L, 3, &plen);
+ if (lua_type(L, 3) == LUA_TSTRING) {
+ p = lua_tolstring(L, 3, &plen);
if (p && plen > 0) {
- stop_pattern = g_malloc (plen);
- memcpy (stop_pattern, p, plen);
+ stop_pattern = g_malloc(plen);
+ memcpy(stop_pattern, p, plen);
}
}
- rh = g_malloc0 (sizeof (*rh));
+ rh = g_malloc0(sizeof(*rh));
rh->type = LUA_WANT_READ;
rh->h.r.cbref = cbref;
rh->h.r.stop_pattern = stop_pattern;
rh->h.r.plen = plen;
- msg_debug_tcp ("added read event, cbref: %d", cbref);
+ msg_debug_tcp("added read event, cbref: %d", cbref);
- g_queue_push_tail (cbd->handlers, rh);
+ g_queue_push_tail(cbd->handlers, rh);
return 0;
}
static gint
-lua_tcp_add_write (lua_State *L)
+lua_tcp_add_write(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_tcp(L, 1);
struct lua_tcp_handler *wh;
gint cbref = -1, tp;
struct iovec *iov = NULL;
guint niov = 0, total_out = 0;
if (cbd == NULL) {
- return luaL_error (L, "invalid arguments");
+ return luaL_error(L, "invalid arguments");
}
- if (lua_type (L, 2) == LUA_TFUNCTION) {
- lua_pushvalue (L, 2);
- cbref = luaL_ref (L, LUA_REGISTRYINDEX);
+ if (lua_type(L, 2) == LUA_TFUNCTION) {
+ lua_pushvalue(L, 2);
+ cbref = luaL_ref(L, LUA_REGISTRYINDEX);
}
- tp = lua_type (L, 3);
+ tp = lua_type(L, 3);
if (tp == LUA_TSTRING || tp == LUA_TUSERDATA) {
- iov = g_malloc (sizeof (*iov));
+ iov = g_malloc(sizeof(*iov));
niov = 1;
- if (!lua_tcp_arg_toiovec (L, 3, cbd, iov)) {
- msg_err ("tcp request has bad data argument");
- lua_pushboolean (L, FALSE);
- g_free (iov);
+ if (!lua_tcp_arg_toiovec(L, 3, cbd, iov)) {
+ msg_err("tcp request has bad data argument");
+ lua_pushboolean(L, FALSE);
+ g_free(iov);
return 1;
}
@@ -2248,39 +2251,39 @@ lua_tcp_add_write (lua_State *L)
}
else if (tp == LUA_TTABLE) {
/* Count parts */
- lua_pushvalue (L, 3);
+ lua_pushvalue(L, 3);
- lua_pushnil (L);
- while (lua_next (L, -2) != 0) {
- niov ++;
- lua_pop (L, 1);
+ lua_pushnil(L);
+ while (lua_next(L, -2) != 0) {
+ niov++;
+ lua_pop(L, 1);
}
- iov = g_malloc (sizeof (*iov) * niov);
- lua_pushnil (L);
+ iov = g_malloc(sizeof(*iov) * niov);
+ lua_pushnil(L);
niov = 0;
- while (lua_next (L, -2) != 0) {
- if (!lua_tcp_arg_toiovec (L, -1, cbd, &iov[niov])) {
- lua_pop (L, 2);
- msg_err ("tcp request has bad data argument at pos %d", niov);
- lua_pushboolean (L, FALSE);
- g_free (iov);
- g_free (cbd);
+ while (lua_next(L, -2) != 0) {
+ if (!lua_tcp_arg_toiovec(L, -1, cbd, &iov[niov])) {
+ lua_pop(L, 2);
+ msg_err("tcp request has bad data argument at pos %d", niov);
+ lua_pushboolean(L, FALSE);
+ g_free(iov);
+ g_free(cbd);
return 1;
}
total_out += iov[niov].iov_len;
- niov ++;
+ niov++;
- lua_pop (L, 1);
+ lua_pop(L, 1);
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
}
- wh = g_malloc0 (sizeof (*wh));
+ wh = g_malloc0(sizeof(*wh));
wh->type = LUA_WANT_WRITE;
wh->h.w.iov = iov;
wh->h.w.iovlen = niov;
@@ -2288,52 +2291,52 @@ lua_tcp_add_write (lua_State *L)
wh->h.w.pos = 0;
/* Cannot set write handler here */
wh->h.w.cbref = cbref;
- msg_debug_tcp ("added write event, cbref: %d", cbref);
+ msg_debug_tcp("added write event, cbref: %d", cbref);
- g_queue_push_tail (cbd->handlers, wh);
- lua_pushboolean (L, TRUE);
+ g_queue_push_tail(cbd->handlers, wh);
+ lua_pushboolean(L, TRUE);
return 1;
}
static gint
-lua_tcp_shift_callback (lua_State *L)
+lua_tcp_shift_callback(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_tcp(L, 1);
if (cbd == NULL) {
- return luaL_error (L, "invalid arguments");
+ return luaL_error(L, "invalid arguments");
}
- lua_tcp_shift_handler (cbd);
- lua_tcp_plan_handler_event (cbd, TRUE, TRUE);
+ lua_tcp_shift_handler(cbd);
+ lua_tcp_plan_handler_event(cbd, TRUE, TRUE);
return 0;
}
static struct lua_tcp_cbdata *
-lua_check_sync_tcp (lua_State *L, gint pos)
+lua_check_sync_tcp(lua_State *L, gint pos)
{
- void *ud = rspamd_lua_check_udata (L, pos, "rspamd{tcp_sync}");
- luaL_argcheck (L, ud != NULL, pos, "'tcp' expected");
- return ud ? *((struct lua_tcp_cbdata **)ud) : NULL;
+ void *ud = rspamd_lua_check_udata(L, pos, "rspamd{tcp_sync}");
+ luaL_argcheck(L, ud != NULL, pos, "'tcp' expected");
+ return ud ? *((struct lua_tcp_cbdata **) ud) : NULL;
}
static int
-lua_tcp_sync_close (lua_State *L)
+lua_tcp_sync_close(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_sync_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_sync_tcp(L, 1);
if (cbd == NULL) {
- return luaL_error (L, "invalid arguments [self is not rspamd{tcp_sync}]");
+ return luaL_error(L, "invalid arguments [self is not rspamd{tcp_sync}]");
}
cbd->flags |= LUA_TCP_FLAG_FINISHED;
if (cbd->fd != -1) {
- rspamd_ev_watcher_stop (cbd->event_loop, &cbd->ev);
- close (cbd->fd);
+ rspamd_ev_watcher_stop(cbd->event_loop, &cbd->ev);
+ close(cbd->fd);
cbd->fd = -1;
}
@@ -2341,15 +2344,15 @@ lua_tcp_sync_close (lua_State *L)
}
static void
-lua_tcp_sync_session_dtor (gpointer ud)
+lua_tcp_sync_session_dtor(gpointer ud)
{
struct lua_tcp_cbdata *cbd = ud;
cbd->flags |= LUA_TCP_FLAG_FINISHED;
if (cbd->fd != -1) {
- msg_debug ("closing sync TCP connection");
- rspamd_ev_watcher_stop (cbd->event_loop, &cbd->ev);
- close (cbd->fd);
+ msg_debug("closing sync TCP connection");
+ rspamd_ev_watcher_stop(cbd->event_loop, &cbd->ev);
+ close(cbd->fd);
cbd->fd = -1;
}
@@ -2361,37 +2364,37 @@ lua_tcp_sync_session_dtor (gpointer ud)
}
static int
-lua_tcp_sync_read_once (lua_State *L)
+lua_tcp_sync_read_once(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_sync_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_sync_tcp(L, 1);
struct lua_tcp_handler *rh;
if (cbd == NULL) {
- return luaL_error (L, "invalid arguments [self is not rspamd{tcp_sync}]");
+ return luaL_error(L, "invalid arguments [self is not rspamd{tcp_sync}]");
}
- struct thread_entry *thread = lua_thread_pool_get_running_entry (cbd->cfg->lua_thread_pool);
+ struct thread_entry *thread = lua_thread_pool_get_running_entry(cbd->cfg->lua_thread_pool);
- rh = g_malloc0 (sizeof (*rh));
+ rh = g_malloc0(sizeof(*rh));
rh->type = LUA_WANT_READ;
rh->h.r.cbref = -1;
- msg_debug_tcp ("added read sync event, thread: %p", thread);
+ msg_debug_tcp("added read sync event, thread: %p", thread);
- g_queue_push_tail (cbd->handlers, rh);
- lua_tcp_plan_handler_event (cbd, TRUE, TRUE);
+ g_queue_push_tail(cbd->handlers, rh);
+ lua_tcp_plan_handler_event(cbd, TRUE, TRUE);
- TCP_RETAIN (cbd);
+ TCP_RETAIN(cbd);
- return lua_thread_yield (thread, 0);
+ return lua_thread_yield(thread, 0);
}
static int
-lua_tcp_sync_write (lua_State *L)
+lua_tcp_sync_write(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_sync_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_sync_tcp(L, 1);
struct lua_tcp_handler *wh;
gint tp;
struct iovec *iov = NULL;
@@ -2399,82 +2402,82 @@ lua_tcp_sync_write (lua_State *L)
gsize total_out = 0;
if (cbd == NULL) {
- return luaL_error (L, "invalid arguments [self is not rspamd{tcp_sync}]");
+ return luaL_error(L, "invalid arguments [self is not rspamd{tcp_sync}]");
}
- struct thread_entry *thread = lua_thread_pool_get_running_entry (cbd->cfg->lua_thread_pool);
+ struct thread_entry *thread = lua_thread_pool_get_running_entry(cbd->cfg->lua_thread_pool);
- tp = lua_type (L, 2);
+ tp = lua_type(L, 2);
if (tp == LUA_TSTRING || tp == LUA_TUSERDATA) {
- iov = g_malloc (sizeof (*iov));
+ iov = g_malloc(sizeof(*iov));
niov = 1;
- if (!lua_tcp_arg_toiovec (L, 2, cbd, iov)) {
- msg_err ("tcp request has bad data argument");
- g_free (iov);
- g_free (cbd);
+ if (!lua_tcp_arg_toiovec(L, 2, cbd, iov)) {
+ msg_err("tcp request has bad data argument");
+ g_free(iov);
+ g_free(cbd);
- return luaL_error (L, "invalid arguments second parameter (data) is expected to be either string or rspamd{text}");
+ return luaL_error(L, "invalid arguments second parameter (data) is expected to be either string or rspamd{text}");
}
total_out = iov[0].iov_len;
}
else if (tp == LUA_TTABLE) {
/* Count parts */
- lua_pushvalue (L, 3);
+ lua_pushvalue(L, 3);
- lua_pushnil (L);
- while (lua_next (L, -2) != 0) {
- niov ++;
- lua_pop (L, 1);
+ lua_pushnil(L);
+ while (lua_next(L, -2) != 0) {
+ niov++;
+ lua_pop(L, 1);
}
- iov = g_malloc (sizeof (*iov) * niov);
- lua_pushnil (L);
+ iov = g_malloc(sizeof(*iov) * niov);
+ lua_pushnil(L);
niov = 0;
- while (lua_next (L, -2) != 0) {
- if (!lua_tcp_arg_toiovec (L, -1, cbd, &iov[niov])) {
- msg_err ("tcp request has bad data argument at pos %d", niov);
- g_free (iov);
- g_free (cbd);
+ while (lua_next(L, -2) != 0) {
+ if (!lua_tcp_arg_toiovec(L, -1, cbd, &iov[niov])) {
+ msg_err("tcp request has bad data argument at pos %d", niov);
+ g_free(iov);
+ g_free(cbd);
- return luaL_error (L, "invalid arguments second parameter (data) is expected to be either string or rspamd{text}");
+ return luaL_error(L, "invalid arguments second parameter (data) is expected to be either string or rspamd{text}");
}
total_out += iov[niov].iov_len;
- niov ++;
+ niov++;
- lua_pop (L, 1);
+ lua_pop(L, 1);
}
- lua_pop (L, 1);
+ lua_pop(L, 1);
}
- wh = g_malloc0 (sizeof (*wh));
+ wh = g_malloc0(sizeof(*wh));
wh->type = LUA_WANT_WRITE;
wh->h.w.iov = iov;
wh->h.w.iovlen = niov;
wh->h.w.total_bytes = total_out;
wh->h.w.pos = 0;
wh->h.w.cbref = -1;
- msg_debug_tcp ("added sync write event, thread: %p", thread);
+ msg_debug_tcp("added sync write event, thread: %p", thread);
- g_queue_push_tail (cbd->handlers, wh);
- lua_tcp_plan_handler_event (cbd, TRUE, TRUE);
+ g_queue_push_tail(cbd->handlers, wh);
+ lua_tcp_plan_handler_event(cbd, TRUE, TRUE);
- TCP_RETAIN (cbd);
+ TCP_RETAIN(cbd);
- return lua_thread_yield (thread, 0);
+ return lua_thread_yield(thread, 0);
}
static gint
lua_tcp_sync_eof(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_sync_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_sync_tcp(L, 1);
if (cbd == NULL) {
- return luaL_error (L, "invalid arguments [self is not rspamd{tcp_sync}]");
+ return luaL_error(L, "invalid arguments [self is not rspamd{tcp_sync}]");
}
lua_pushboolean(L, cbd->eof);
@@ -2483,29 +2486,29 @@ lua_tcp_sync_eof(lua_State *L)
}
static gint
-lua_tcp_sync_shutdown (lua_State *L)
+lua_tcp_sync_shutdown(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_sync_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_sync_tcp(L, 1);
if (cbd == NULL) {
- return luaL_error (L, "invalid arguments [self is not rspamd{tcp_sync}]");
+ return luaL_error(L, "invalid arguments [self is not rspamd{tcp_sync}]");
}
- shutdown (cbd->fd, SHUT_WR);
+ shutdown(cbd->fd, SHUT_WR);
return 0;
}
static gint
-lua_tcp_starttls (lua_State * L)
+lua_tcp_starttls(lua_State *L)
{
LUA_TRACE_POINT;
- struct lua_tcp_cbdata *cbd = lua_check_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_tcp(L, 1);
gpointer ssl_ctx;
gboolean verify_peer;
if (cbd == NULL || cbd->ssl_conn != NULL) {
- return luaL_error (L, "invalid arguments");
+ return luaL_error(L, "invalid arguments");
}
if (cbd->flags & LUA_TCP_FLAG_SSL_NOVERIFY) {
@@ -2517,48 +2520,47 @@ lua_tcp_starttls (lua_State * L)
verify_peer = TRUE;
}
- cbd->ssl_conn = rspamd_ssl_connection_new (ssl_ctx,
- cbd->event_loop,
- verify_peer,
- cbd->tag);
+ cbd->ssl_conn = rspamd_ssl_connection_new(ssl_ctx,
+ cbd->event_loop,
+ verify_peer,
+ cbd->tag);
- if (!rspamd_ssl_connect_fd (cbd->ssl_conn, cbd->fd, cbd->hostname, &cbd->ev,
- cbd->ev.timeout, lua_tcp_handler, lua_tcp_ssl_on_error, cbd)) {
- lua_tcp_push_error (cbd, TRUE, "ssl connection failed: %s",
- strerror (errno));
+ if (!rspamd_ssl_connect_fd(cbd->ssl_conn, cbd->fd, cbd->hostname, &cbd->ev,
+ cbd->ev.timeout, lua_tcp_handler, lua_tcp_ssl_on_error, cbd)) {
+ lua_tcp_push_error(cbd, TRUE, "ssl connection failed: %s",
+ strerror(errno));
}
return 0;
}
static gint
-lua_tcp_sync_gc (lua_State * L)
+lua_tcp_sync_gc(lua_State *L)
{
- struct lua_tcp_cbdata *cbd = lua_check_sync_tcp (L, 1);
+ struct lua_tcp_cbdata *cbd = lua_check_sync_tcp(L, 1);
if (!cbd) {
- return luaL_error (L, "invalid arguments [self is not rspamd{tcp_sync}]");
+ return luaL_error(L, "invalid arguments [self is not rspamd{tcp_sync}]");
}
lua_tcp_maybe_free(cbd);
- lua_tcp_fin (cbd);
+ lua_tcp_fin(cbd);
return 0;
}
static gint
-lua_load_tcp (lua_State * L)
+lua_load_tcp(lua_State *L)
{
- lua_newtable (L);
- luaL_register (L, NULL, tcp_libf);
+ lua_newtable(L);
+ luaL_register(L, NULL, tcp_libf);
return 1;
}
-void
-luaopen_tcp (lua_State * L)
+void luaopen_tcp(lua_State *L)
{
- rspamd_lua_add_preload (L, "rspamd_tcp", lua_load_tcp);
- rspamd_lua_new_class (L, "rspamd{tcp}", tcp_libm);
- rspamd_lua_new_class (L, "rspamd{tcp_sync}", tcp_sync_libm);
- lua_pop (L, 1);
+ rspamd_lua_add_preload(L, "rspamd_tcp", lua_load_tcp);
+ rspamd_lua_new_class(L, "rspamd{tcp}", tcp_libm);
+ rspamd_lua_new_class(L, "rspamd{tcp_sync}", tcp_sync_libm);
+ lua_pop(L, 1);
}