From be47d646f01c217048e5970335c5797f60de8bd6 Mon Sep 17 00:00:00 2001
From: Roman Artiukhin <bahusdrive@gmail.com>
Date: Wed, 16 Apr 2025 12:25:08 +0300
Subject: [PATCH] jpeg: Add support for FORMAT_RETURN_SIZE flag

Change-Id: I01f883400d775ffee5cdaa107fc3d6fb89b02573
---
 apps/recorder/jpeg_load.c | 39 +++++++++++++++++++++++++++++----------
 1 file changed, 29 insertions(+), 10 deletions(-)

diff --git a/apps/recorder/jpeg_load.c b/apps/recorder/jpeg_load.c
index 52d426fdec..815423bb1d 100644
--- a/apps/recorder/jpeg_load.c
+++ b/apps/recorder/jpeg_load.c
@@ -2210,21 +2210,22 @@ int clip_jpeg_fd(int fd, int flags,
         bm_size = cformat->get_size(bm);
     else
         bm_size = BM_SIZE(bm->width,bm->height,FORMAT_NATIVE,false);
-    if (bm_size > maxsize)
-        return -1;
+
     char *buf_start = (char *)bm->data + bm_size;
     char *buf_end = (char *)bm->data + maxsize;
-    maxsize = buf_end - buf_start;
+    bool return_size = format & FORMAT_RETURN_SIZE;
 #ifndef JPEG_FROM_MEM
     ALIGN_BUFFER(buf_start, maxsize, sizeof(long));
-    if (maxsize < (int)sizeof(struct jpeg))
-        return -1;
-    memmove(buf_start, p_jpeg, sizeof(struct jpeg));
-    p_jpeg = (struct jpeg *)buf_start;
+    if (!return_size)
+    {
+        if (maxsize < (int)sizeof(struct jpeg))
+            return -1;
+        memmove(buf_start, p_jpeg, sizeof(struct jpeg));
+        p_jpeg = (struct jpeg *)buf_start;
+    }
     buf_start += sizeof(struct jpeg);
-    maxsize = buf_end - buf_start;
 #endif
-    fix_huff_tables(p_jpeg);
+    maxsize = buf_end - buf_start;
 #ifdef HAVE_LCD_COLOR
     int decode_buf_size = (p_jpeg->x_mbl << p_jpeg->h_scale[1])
         << p_jpeg->v_scale[1];
@@ -2236,9 +2237,27 @@ int clip_jpeg_fd(int fd, int flags,
 #endif
     decode_buf_size *= JPEG_PIX_SZ;
     JDEBUGF("decode buffer size: %d\n", decode_buf_size);
-    p_jpeg->img_buf = (jpeg_pix_t *)buf_start;
+    if (return_size)
+    {
+        return (buf_start - (char *) bm->data) + decode_buf_size
+               + (resize
+                      ?
+                      /* buffer for 1 line + 2 spare lines */
+#ifdef HAVE_LCD_COLOR
+                      sizeof(struct uint32_argb)
+#else
+                      sizeof(uint32_t)
+#endif
+                      * 3 * bm->width
+                      : 0);
+    }
+
     if (buf_end - buf_start < decode_buf_size)
         return -1;
+
+    fix_huff_tables(p_jpeg);
+
+    p_jpeg->img_buf = (jpeg_pix_t *)buf_start;
     buf_start += decode_buf_size;
     maxsize = buf_end - buf_start;
     memset(p_jpeg->img_buf, 0, decode_buf_size);