1 /*
   2  * Copyright (c) 2005, 2012, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 #include "splashscreen_impl.h"
  27 
  28 #include "jpeglib.h"
  29 #include "jerror.h"
  30 
  31 #include <setjmp.h>
  32 
  33 #ifdef __APPLE__
  34 /* use setjmp/longjmp versions that do not save/restore the signal mask */
  35 #define setjmp _setjmp
  36 #define longjmp _longjmp
  37 #endif
  38 
  39 /* stream input handling */
  40 
  41 typedef struct
  42 {
  43     struct jpeg_source_mgr pub; /* public fields */
  44     SplashStream * stream;      /* source stream */
  45     JOCTET *buffer;             /* start of buffer */
  46     boolean start_of_file;      /* have we gotten any data yet? */
  47 } stream_source_mgr;
  48 
  49 typedef stream_source_mgr *stream_src_ptr;
  50 
  51 #define INPUT_BUF_SIZE  4096    /* choose an efficiently fread'able size */
  52 
  53 METHODDEF(void)
  54 stream_init_source(j_decompress_ptr cinfo)
  55 {
  56     stream_src_ptr src = (stream_src_ptr) cinfo->src;
  57 
  58     src->start_of_file = TRUE;
  59 }
  60 
  61 METHODDEF(boolean)
  62 stream_fill_input_buffer(j_decompress_ptr cinfo)
  63 {
  64     stream_src_ptr src = (stream_src_ptr) cinfo->src;
  65     size_t nbytes;
  66 
  67 
  68     nbytes = src->stream->read(src->stream, src->buffer, INPUT_BUF_SIZE);
  69 
  70     if (nbytes <= 0) {
  71         if (src->start_of_file) /* Treat empty input file as fatal error */
  72             ERREXIT(cinfo, JERR_INPUT_EMPTY);
  73         WARNMS(cinfo, JWRN_JPEG_EOF);
  74         /* Insert a fake EOI marker */
  75         src->buffer[0] = (JOCTET) 0xFF;
  76         src->buffer[1] = (JOCTET) JPEG_EOI;
  77         nbytes = 2;
  78     }
  79 
  80     src->pub.next_input_byte = src->buffer;
  81     src->pub.bytes_in_buffer = nbytes;
  82     src->start_of_file = FALSE;
  83 
  84     return TRUE;
  85 }
  86 
  87 METHODDEF(void)
  88     stream_skip_input_data(j_decompress_ptr cinfo, long num_bytes)
  89 {
  90     stream_src_ptr src = (stream_src_ptr) cinfo->src;
  91 
  92     if (num_bytes > 0) {
  93         while (num_bytes > (long) src->pub.bytes_in_buffer) {
  94             num_bytes -= (long) src->pub.bytes_in_buffer;
  95             (void) stream_fill_input_buffer(cinfo);
  96         }
  97         src->pub.next_input_byte += (size_t) num_bytes;
  98         src->pub.bytes_in_buffer -= (size_t) num_bytes;
  99     }
 100 }
 101 
 102 METHODDEF(void)
 103 stream_term_source(j_decompress_ptr cinfo)
 104 {
 105 }
 106 
 107 static void
 108 set_stream_src(j_decompress_ptr cinfo, SplashStream * stream)
 109 {
 110     stream_src_ptr src;
 111 
 112     if (cinfo->src == NULL) {   /* first time for this JPEG object? */
 113         cinfo->src = (struct jpeg_source_mgr *)
 114             (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo,
 115             JPOOL_PERMANENT, sizeof(stream_source_mgr));
 116         src = (stream_src_ptr) cinfo->src;
 117         src->buffer = (JOCTET *)
 118             (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo,
 119             JPOOL_PERMANENT, INPUT_BUF_SIZE * sizeof(JOCTET));
 120     }
 121 
 122     src = (stream_src_ptr) cinfo->src;
 123     src->pub.init_source = stream_init_source;
 124     src->pub.fill_input_buffer = stream_fill_input_buffer;
 125     src->pub.skip_input_data = stream_skip_input_data;
 126     src->pub.resync_to_restart = jpeg_resync_to_restart;        /* use default method */
 127     src->pub.term_source = stream_term_source;
 128     src->stream = stream;
 129     src->pub.bytes_in_buffer = 0;       /* forces fill_input_buffer on first read */
 130     src->pub.next_input_byte = NULL;    /* until buffer loaded */
 131 }
 132 
 133 int
 134 SplashDecodeJpeg(Splash * splash, struct jpeg_decompress_struct *cinfo)
 135 {
 136     int rowStride, stride;
 137     JSAMPARRAY buffer;
 138     ImageFormat srcFormat;
 139 
 140     jpeg_read_header(cinfo, TRUE);
 141 
 142     // SplashScreen jpeg converter expects data in RGB format only
 143     cinfo->out_color_space = JCS_RGB;
 144 
 145     jpeg_start_decompress(cinfo);
 146 
 147     SplashCleanup(splash);
 148 
 149     splash->width = cinfo->output_width;
 150     splash->height = cinfo->output_height;
 151 
 152     if (!SAFE_TO_ALLOC(splash->imageFormat.depthBytes, splash->width)) {
 153         return 0;
 154     }
 155     stride = splash->width * splash->imageFormat.depthBytes;
 156 
 157     if (!SAFE_TO_ALLOC(stride, splash->height)) {
 158         return 0;
 159     }
 160     if (!SAFE_TO_ALLOC(cinfo->output_width, cinfo->output_components)) {
 161         return 0;
 162     }
 163 
 164     splash->frameCount = 1;
 165     splash->frames = (SplashImage *) malloc(sizeof(SplashImage) *
 166         splash->frameCount);
 167     if (splash->frames == NULL) {
 168         return 0;
 169     }
 170     memset(splash->frames, 0, sizeof(SplashImage) *
 171         splash->frameCount);
 172 
 173     splash->loopCount = 1;
 174     splash->frames[0].delay = 0;
 175     splash->frames[0].bitmapBits = malloc(stride * splash->height);
 176     if (splash->frames[0].bitmapBits == NULL) {
 177         free(splash->frames);
 178         return 0;
 179     }
 180 
 181     rowStride = cinfo->output_width * cinfo->output_components;
 182 
 183     buffer = (*cinfo->mem->alloc_sarray)
 184         ((j_common_ptr) cinfo, JPOOL_IMAGE, rowStride, 1);
 185     if (buffer == NULL) {
 186         free(splash->frames[0].bitmapBits);
 187         free(splash->frames);
 188         return 0;
 189     }
 190 
 191     initFormat(&srcFormat, 0x00FF0000, 0x0000FF00, 0x000000FF, 0x00000000);
 192     srcFormat.byteOrder = BYTE_ORDER_LSBFIRST;
 193     srcFormat.depthBytes = 3;
 194     srcFormat.fixedBits = 0xFF000000;
 195 
 196     splash->maskRequired = 0;   // reset maskRequired as JPEG can't be transparent
 197 
 198     while (cinfo->output_scanline < cinfo->output_height) {
 199         rgbquad_t *out =
 200             (rgbquad_t *) ((byte_t *) splash->frames[0].bitmapBits +
 201                 cinfo->output_scanline * stride);
 202 
 203         jpeg_read_scanlines(cinfo, buffer, 1);
 204         convertLine(buffer[0], sizeof(JSAMPLE) * 3, out,
 205             splash->imageFormat.depthBytes, cinfo->output_width, &srcFormat,
 206             &splash->imageFormat, CVT_COPY, NULL, 0, NULL,
 207             cinfo->output_scanline, 0);
 208     }
 209     jpeg_finish_decompress(cinfo);
 210 
 211     return 1;
 212 }
 213 
 214 struct my_error_mgr
 215 {
 216     struct jpeg_error_mgr pub;  /* "public" fields */
 217     jmp_buf setjmp_buffer;      /* for return to caller */
 218 };
 219 
 220 typedef struct my_error_mgr *my_error_ptr;
 221 
 222 static void
 223 my_error_exit(j_common_ptr cinfo)
 224 {
 225     /* cinfo->err really points to a my_error_mgr struct, so coerce pointer */
 226     my_error_ptr myerr = (my_error_ptr) cinfo->err;
 227 
 228     /* Always display the message. */
 229     /* We could postpone this until after returning, if we chose. */
 230     (*cinfo->err->output_message) (cinfo);
 231 
 232     /* Return control to the setjmp point */
 233     longjmp(myerr->setjmp_buffer, 1);
 234 }
 235 
 236 int
 237 SplashDecodeJpegStream(Splash * splash, SplashStream * stream)
 238 {
 239     struct jpeg_decompress_struct cinfo;
 240     int success;
 241     struct my_error_mgr jerr;
 242 
 243     cinfo.err = jpeg_std_error(&jerr.pub);
 244     jerr.pub.error_exit = my_error_exit;
 245 
 246     if (setjmp(jerr.setjmp_buffer)) {
 247         success = 0;
 248         goto done;
 249     }
 250     jpeg_create_decompress(&cinfo);
 251     set_stream_src(&cinfo, stream);
 252     success = SplashDecodeJpeg(splash, &cinfo);
 253 
 254   done:
 255     jpeg_destroy_decompress(&cinfo);
 256     return success;
 257 }
--- EOF ---