From 1ac9efbac6cd81dd7afb8bfb5672d5d7592af26d Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov <vsevolod@highsecure.ru> Date: Thu, 6 Aug 2020 14:18:47 +0100 Subject: [Minor] Lua_tensor: Implement serialisation --- src/lua/lua_tensor.c | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/src/lua/lua_tensor.c b/src/lua/lua_tensor.c index 85aaa2e95..9b85779d7 100644 --- a/src/lua/lua_tensor.c +++ b/src/lua/lua_tensor.c @@ -236,7 +236,20 @@ lua_tensor_save (lua_State *L) struct rspamd_lua_tensor *t = lua_check_tensor (L, 1); if (t) { + gsize sz = sizeof (gint) * 4 + t->size * sizeof (rspamd_tensor_num_t); + guchar *data; + struct rspamd_lua_text *out = lua_new_text (L, NULL, 0, TRUE); + + data = g_malloc (sz); + memcpy (data, &t->ndims, sizeof (int)); + memcpy (data + sizeof (int), &t->size, sizeof (int)); + memcpy (data + 2 * sizeof (int), t->dim, sizeof (int) * 2); + memcpy (data + 4 * sizeof (int), t->data, + t->size * sizeof (rspamd_tensor_num_t)); + + out->start = (const gchar *)data; + out->len = sz; } else { return luaL_error (L, "invalid arguments"); -- cgit v1.2.3