util/fuzz-tests/jpeg-test.c

   1 /* SPDX-License-Identifier: GPL-2.0-only */
   2
   3 #include <stdlib.h>
   4 #include <stdio.h>
   5 #include "jpeg.h"
   6
   7 const int depth = 16;
   8
   9 int main(int argc, char **argv)
  10 {
  11         FILE *f = fopen(argv[1], "rb");
  12         unsigned long len;
  13
  14         if (!f)
  15                 return 1;
  16         if (fseek(f, 0, SEEK_END) != 0)
  17                 return 1;
  18         len = ftell(f);
  19         if (fseek(f, 0, SEEK_SET) != 0)
  20                 return 1;
  21
  22         unsigned char *buf = malloc(len);
  23         if (fread(buf, len, 1, f) != 1)
  24                 return 1;
  25         fclose(f);
  26
  27         unsigned int width;
  28         unsigned int height;
  29         if (jpeg_fetch_size(buf, len, &width, &height) != 0) {
  30                 return 1;
  31         }
  32         if ((width > 6000) || (height > 6000)) {
  33                 // infeasible data set
  34                 return 1;
  35         }
  36         //printf("width: %d, height: %d\n", width, height);
  37         unsigned char *pic = malloc(depth / 8 * width * height);
  38         int ret = jpeg_decode(buf, len, pic, width, height, width * depth / 8, depth);
  39         //printf("ret: %x\n", ret);
  40         return ret;
  41 }