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 = 0;
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 goto done;
248 }
249 jpeg_create_decompress(&cinfo);
250 set_stream_src(&cinfo, stream);
251 success = SplashDecodeJpeg(splash, &cinfo);
252
253 done:
254 jpeg_destroy_decompress(&cinfo);
255 return success;
256 }