--- /dev/null
+/* Copyright (c) 2010, Vsevolod Stakhov
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ * * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * * Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY Rambler media ''AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL Rambler BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "config.h"
+#include "images.h"
+#include "main.h"
+#include "message.h"
+
+static const guint8 png_signature[] = {137, 80, 78, 71, 13, 10, 26, 10};
+static const guint8 jpg_sig1[] = {0xff, 0xd8};
+static const guint8 jpg_sig2[] = {'J', 'F', 'I', 'F'};
+static const guint8 gif_signature[] = {'G', 'I', 'F', '8'};
+static const guint8 bmp_signature[] = {'B', 'M'};
+
+static void process_image (struct worker_task *task, struct mime_part *part);
+
+
+void
+process_images (struct worker_task *task)
+{
+ GList *cur;
+ struct mime_part *part;
+
+ cur = task->parts;
+ while (cur) {
+ part = cur->data;
+ if (g_mime_content_type_is_type (part->type, "image", "*") && part->content->len > 0) {
+ process_image (task, part);
+ }
+ cur = g_list_next (cur);
+ }
+
+}
+
+static enum known_image_types
+detect_image_type (GByteArray *data)
+{
+ if (data->len > sizeof (png_signature) / sizeof (png_signature[0])) {
+ if (memcmp (data->data, png_signature, sizeof (png_signature)) == 0) {
+ return IMAGE_TYPE_PNG;
+ }
+ }
+ if (data->len > 10) {
+ if (memcmp (data->data, jpg_sig1, sizeof (jpg_sig1)) == 0) {
+ if (memcmp (data->data + 6, jpg_sig2, sizeof (jpg_sig2)) == 0) {
+ return IMAGE_TYPE_JPG;
+ }
+ }
+ }
+ if (data->len > sizeof (gif_signature) / sizeof (gif_signature[0])) {
+ if (memcmp (data->data, gif_signature, sizeof (gif_signature)) == 0) {
+ return IMAGE_TYPE_GIF;
+ }
+ }
+ if (data->len > sizeof (bmp_signature) / sizeof (bmp_signature[0])) {
+ if (memcmp (data->data, bmp_signature, sizeof (bmp_signature)) == 0) {
+ return IMAGE_TYPE_BMP;
+ }
+ }
+
+ return IMAGE_TYPE_UNKNOWN;
+}
+
+
+static struct rspamd_image *
+process_png_image (struct worker_task *task, GByteArray *data)
+{
+ struct rspamd_image *img;
+ guint32 t;
+ guint8 *p;
+
+ if (data->len < 24) {
+ msg_info ("bad png detected (maybe striped): <%s>", task->message_id);
+ return NULL;
+ }
+
+ /* In png we should find iHDR section and get data from it */
+ /* Skip signature and read header section */
+ p = data->data + 12;
+ if (memcmp (p, "IHDR", 4) != 0) {
+ msg_info ("png doesn't begins with IHDR section", task->message_id);
+ return NULL;
+ }
+
+ img = memory_pool_alloc (task->task_pool, sizeof (struct rspamd_image));
+ img->type = IMAGE_TYPE_PNG;
+ img->data = data;
+
+ p += 4;
+ memcpy (&t, p, sizeof (guint32));
+ img->width = ntohl (t);
+ p += 4;
+ memcpy (&t, p, sizeof (guint32));
+ img->height = ntohl (t);
+
+ return img;
+}
+
+static struct rspamd_image *
+process_jpg_image (struct worker_task *task, GByteArray *data)
+{
+ guint8 *p;
+ guint16 t;
+ gsize remain;
+ struct rspamd_image *img;
+
+ img = memory_pool_alloc (task->task_pool, sizeof (struct rspamd_image));
+ img->type = IMAGE_TYPE_JPG;
+ img->data = data;
+
+ p = data->data;
+ remain = data->len;
+ /* In jpeg we should find any data stream (ff c0 .. ff c3) and extract its height and width */
+ while (remain --) {
+ if (*p == 0xFF && remain > 8 && (*(p + 1) >= 0xC0 && *(p + 1) <= 0xC3)) {
+ memcpy (&t, p + 5, sizeof (guint16));
+ img->height = ntohs (t);
+ memcpy (&t, p + 7, sizeof (guint16));
+ img->width = ntohs (t);
+ return img;
+ }
+ p ++;
+ }
+
+ return NULL;
+}
+
+static struct rspamd_image *
+process_gif_image (struct worker_task *task, GByteArray *data)
+{
+ struct rspamd_image *img;
+ guint8 *p;
+ guint16 t;
+
+ if (data->len < 10) {
+ msg_info ("bad gif detected (maybe striped): <%s>", task->message_id);
+ return NULL;
+ }
+
+ img = memory_pool_alloc (task->task_pool, sizeof (struct rspamd_image));
+ img->type = IMAGE_TYPE_GIF;
+ img->data = data;
+
+ p = data->data + 6;
+ memcpy (&t, p, sizeof (guint16));
+ img->width = GUINT16_FROM_LE (t);
+ memcpy (&t, p + 2, sizeof (guint16));
+ img->height = GUINT16_FROM_LE (t);
+
+ return img;
+}
+
+static struct rspamd_image *
+process_bmp_image (struct worker_task *task, GByteArray *data)
+{
+ struct rspamd_image *img;
+ gint32 t;
+ guint8 *p;
+
+
+
+ if (data->len < 28) {
+ msg_info ("bad bmp detected (maybe striped): <%s>", task->message_id);
+ return NULL;
+ }
+
+ img = memory_pool_alloc (task->task_pool, sizeof (struct rspamd_image));
+ img->type = IMAGE_TYPE_BMP;
+ img->data = data;
+ p = data->data + 18;
+ memcpy (&t, p, sizeof (gint32));
+ img->width = abs (GINT32_FROM_LE (t));
+ memcpy (&t, p + 4, sizeof (gint32));
+ img->height = abs (GINT32_FROM_LE (t));
+
+ return img;
+}
+
+static void
+process_image (struct worker_task *task, struct mime_part *part)
+{
+ enum known_image_types type;
+ struct rspamd_image *img = NULL;
+ if ((type = detect_image_type (part->content)) != IMAGE_TYPE_UNKNOWN) {
+ switch (type) {
+ case IMAGE_TYPE_PNG:
+ img = process_png_image (task, part->content);
+ break;
+ case IMAGE_TYPE_JPG:
+ img = process_jpg_image (task, part->content);
+ break;
+ case IMAGE_TYPE_GIF:
+ img = process_gif_image (task, part->content);
+ break;
+ case IMAGE_TYPE_BMP:
+ img = process_bmp_image (task, part->content);
+ break;
+ default:
+ img = NULL;
+ break;
+ }
+ }
+
+ if (img != NULL) {
+ msg_info ("detected %s image of size %ud x %ud in message <%s>",
+ image_type_str (img->type),
+ img->width, img->height,
+ task->message_id);
+ task->images = g_list_prepend (task->images, img);
+ }
+}
+
+const char *
+image_type_str (enum known_image_types type)
+{
+ switch (type) {
+ case IMAGE_TYPE_PNG:
+ return "PNG";
+ break;
+ case IMAGE_TYPE_JPG:
+ return "JPEG";
+ break;
+ case IMAGE_TYPE_GIF:
+ return "GIF";
+ break;
+ case IMAGE_TYPE_BMP:
+ return "BMP";
+ break;
+ default:
+ return "unknown";
+ }
+
+ return "unknown";
+}
#include "../message.h"
#include "../expressions.h"
#include "../dns.h"
+#include "../images.h"
/* Task methods */
LUA_FUNCTION_DEF (task, get_message);
LUA_FUNCTION_DEF (task, get_from_ip_num);
LUA_FUNCTION_DEF (task, get_client_ip_num);
LUA_FUNCTION_DEF (task, get_helo);
+LUA_FUNCTION_DEF (task, get_images);
static const struct luaL_reg tasklib_m[] = {
LUA_INTERFACE_DEF (task, get_message),
LUA_INTERFACE_DEF (task, get_from_ip_num),
LUA_INTERFACE_DEF (task, get_client_ip_num),
LUA_INTERFACE_DEF (task, get_helo),
+ LUA_INTERFACE_DEF (task, get_images),
{"__tostring", lua_class_tostring},
{NULL, NULL}
};
{NULL, NULL}
};
+/* Image methods */
+LUA_FUNCTION_DEF (image, get_width);
+LUA_FUNCTION_DEF (image, get_height);
+LUA_FUNCTION_DEF (image, get_type);
+
+static const struct luaL_reg imagelib_m[] = {
+ LUA_INTERFACE_DEF (image, get_width),
+ LUA_INTERFACE_DEF (image, get_height),
+ LUA_INTERFACE_DEF (image, get_type),
+ {"__tostring", lua_class_tostring},
+ {NULL, NULL}
+};
+
/* Utility functions */
static struct worker_task *
lua_check_task (lua_State * L)
return *((struct mime_text_part **)ud);
}
+static struct rspamd_image *
+lua_check_image (lua_State * L)
+{
+ void *ud = luaL_checkudata (L, 1, "rspamd{image}");
+ luaL_argcheck (L, ud != NULL, 1, "'image' expected");
+ return *((struct rspamd_image **)ud);
+}
+
/*** Task interface ***/
static int
lua_task_get_message (lua_State * L)
return 1;
}
+static int
+lua_task_get_images (lua_State *L)
+{
+ struct worker_task *task = lua_check_task (L);
+ int i = 1;
+ GList *cur;
+ struct rspamd_image **pimg;
+
+ if (task) {
+ cur = task->images;
+ if (cur != NULL) {
+ lua_newtable (L);
+ while (cur) {
+ pimg = lua_newuserdata (L, sizeof (struct rspamd_image *));
+ lua_setclass (L, "rspamd{image}", -1);
+ *pimg = cur->data;
+ lua_rawseti (L, -2, i++);
+ cur = g_list_next (cur);
+ }
+ return 1;
+ }
+ }
+
+ lua_pushnil (L);
+ return 1;
+}
+
/**** Textpart implementation *****/
return 1;
}
+/* Image functions */
+static int
+lua_image_get_width (lua_State *L)
+{
+ struct rspamd_image *img = lua_check_image (L);
+
+ if (img != NULL) {
+ lua_pushnumber (L, img->width);
+ }
+ else {
+ lua_pushnumber (L, 0);
+ }
+ return 1;
+}
+
+static int
+lua_image_get_height (lua_State *L)
+{
+ struct rspamd_image *img = lua_check_image (L);
+
+ if (img != NULL) {
+ lua_pushnumber (L, img->height);
+ }
+ else {
+ lua_pushnumber (L, 0);
+ }
+
+ return 1;
+}
+
+static int
+lua_image_get_type (lua_State *L)
+{
+ struct rspamd_image *img = lua_check_image (L);
+
+ if (img != NULL) {
+ lua_pushstring (L, image_type_str (img->type));
+ }
+ else {
+ lua_pushnil (L);
+ }
+
+ return 1;
+}
/* Init part */
int
return 1;
}
+
+int
+luaopen_image (lua_State * L)
+{
+ lua_newclass (L, "rspamd{image}", imagelib_m);
+ luaL_openlib (L, "rspamd_image", null_reg, 0);
+
+ return 1;
+}
projects / rspamd.git / commitdiff