diff options
Diffstat (limited to 'content/handlers/image/png.c')
-rw-r--r-- | content/handlers/image/png.c | 614 |
1 files changed, 614 insertions, 0 deletions
diff --git a/content/handlers/image/png.c b/content/handlers/image/png.c new file mode 100644 index 000000000..4f6f17d58 --- /dev/null +++ b/content/handlers/image/png.c @@ -0,0 +1,614 @@ +/* + * Copyright 2004 James Bursa <bursa@users.sourceforge.net> + * Copyright 2004 Richard Wilson <not_ginger_matt@hotmail.com> + * Copyright 2008 Daniel Silverstone <dsilvers@netsurf-browser.org> + * + * This file is part of NetSurf, http://www.netsurf-browser.org/ + * + * NetSurf is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; version 2 of the License. + * + * NetSurf is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +#include <stdbool.h> +#include <stdlib.h> +#include <png.h> + +#include "utils/utils.h" +#include "utils/log.h" +#include "utils/messages.h" +#include "content/content_protected.h" +#include "desktop/gui_internal.h" + +#include "image_cache.h" +#include "bitmap.h" +#include "png.h" + +/* accommodate for old versions of libpng (beware security holes!) */ + +#ifndef png_jmpbuf +#warning you have an antique libpng +#define png_jmpbuf(png_ptr) ((png_ptr)->jmpbuf) +#endif + +#if PNG_LIBPNG_VER < 10209 +#define png_set_expand_gray_1_2_4_to_8(png) png_set_gray_1_2_4_to_8(png) +#endif + +typedef struct nspng_content { + struct content base; /**< base content type */ + + bool no_process_data; /**< Do not continue to process data as it arrives */ + png_structp png; + png_infop info; + int interlace; + struct bitmap *bitmap; /**< Created NetSurf bitmap */ + size_t rowstride, bpp; /**< Bitmap rowstride and bpp */ + size_t rowbytes; /**< Number of bytes per row */ +} nspng_content; + +static unsigned int interlace_start[8] = {0, 16, 0, 8, 0, 4, 0}; +static unsigned int interlace_step[8] = {28, 28, 12, 12, 4, 4, 0}; +static unsigned int interlace_row_start[8] = {0, 0, 4, 0, 2, 0, 1}; +static unsigned int interlace_row_step[8] = {8, 8, 8, 4, 4, 2, 2}; + +/** Callbak error numbers*/ +enum nspng_cberr { + CBERR_NONE = 0, /* no error */ + CBERR_LIBPNG, /* error from png library */ + CBERR_NOPRE, /* no pre-conversion performed */ +}; + +/** + * nspng_warning -- callback for libpng warnings + */ +static void nspng_warning(png_structp png_ptr, png_const_charp warning_message) +{ + LOG("%s", warning_message); +} + +/** + * nspng_error -- callback for libpng errors + */ +static void nspng_error(png_structp png_ptr, png_const_charp error_message) +{ + LOG("%s", error_message); + longjmp(png_jmpbuf(png_ptr), CBERR_LIBPNG); +} + +static void nspng_setup_transforms(png_structp png_ptr, png_infop info_ptr) +{ + int bit_depth, color_type, intent; + double gamma; + + bit_depth = png_get_bit_depth(png_ptr, info_ptr); + color_type = png_get_color_type(png_ptr, info_ptr); + + /* Set up our transformations */ + if (color_type == PNG_COLOR_TYPE_PALETTE) { + png_set_palette_to_rgb(png_ptr); + } + + if ((color_type == PNG_COLOR_TYPE_GRAY) && (bit_depth < 8)) { + png_set_expand_gray_1_2_4_to_8(png_ptr); + } + + if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS)) { + png_set_tRNS_to_alpha(png_ptr); + } + + if (bit_depth == 16) { + png_set_strip_16(png_ptr); + } + + if (color_type == PNG_COLOR_TYPE_GRAY || + color_type == PNG_COLOR_TYPE_GRAY_ALPHA) { + png_set_gray_to_rgb(png_ptr); + } + + if (!(color_type & PNG_COLOR_MASK_ALPHA)) { + png_set_filler(png_ptr, 0xff, PNG_FILLER_AFTER); + } + + /* gamma correction - we use 2.2 as our screen gamma + * this appears to be correct (at least in respect to !Browse) + * see http://www.w3.org/Graphics/PNG/all_seven.html for a test case + */ + if (png_get_sRGB(png_ptr, info_ptr, &intent)) { + png_set_gamma(png_ptr, 2.2, 0.45455); + } else { + if (png_get_gAMA(png_ptr, info_ptr, &gamma)) { + png_set_gamma(png_ptr, 2.2, gamma); + } else { + png_set_gamma(png_ptr, 2.2, 0.45455); + } + } + + png_read_update_info(png_ptr, info_ptr); +} + +/** + * info_callback -- PNG header has been completely received, prepare to process + * image data + */ +static void info_callback(png_structp png_s, png_infop info) +{ + int interlace; + png_uint_32 width, height; + nspng_content *png_c = png_get_progressive_ptr(png_s); + + width = png_get_image_width(png_s, info); + height = png_get_image_height(png_s, info); + interlace = png_get_interlace_type(png_s, info); + + png_c->base.width = width; + png_c->base.height = height; + png_c->base.size += width * height * 4; + + /* see if progressive-conversion should continue */ + if (image_cache_speculate((struct content *)png_c) == false) { + longjmp(png_jmpbuf(png_s), CBERR_NOPRE); + } + + /* Claim the required memory for the converted PNG */ + png_c->bitmap = guit->bitmap->create(width, height, BITMAP_NEW); + if (png_c->bitmap == NULL) { + /* Failed to create bitmap skip pre-conversion */ + longjmp(png_jmpbuf(png_s), CBERR_NOPRE); + } + + png_c->rowstride = guit->bitmap->get_rowstride(png_c->bitmap); + png_c->bpp = guit->bitmap->get_bpp(png_c->bitmap); + + nspng_setup_transforms(png_s, info); + + png_c->rowbytes = png_get_rowbytes(png_s, info); + png_c->interlace = (interlace == PNG_INTERLACE_ADAM7); + + LOG("size %li * %li, rowbytes %" PRIsizet, + (unsigned long)width, + (unsigned long)height, + png_c->rowbytes); +} + +static void row_callback(png_structp png_s, png_bytep new_row, + png_uint_32 row_num, int pass) +{ + nspng_content *png_c = png_get_progressive_ptr(png_s); + unsigned long rowbytes = png_c->rowbytes; + unsigned char *buffer, *row; + + /* Give up if there's no bitmap */ + if (png_c->bitmap == NULL) + return; + + /* Abort if we've not got any data */ + if (new_row == NULL) + return; + + /* Get bitmap buffer */ + buffer = guit->bitmap->get_buffer(png_c->bitmap); + if (buffer == NULL) { + /* No buffer, bail out */ + longjmp(png_jmpbuf(png_s), 1); + } + + /* Calculate address of row start */ + row = buffer + (png_c->rowstride * row_num); + + /* Handle interlaced sprites using the Adam7 algorithm */ + if (png_c->interlace) { + unsigned long dst_off; + unsigned long src_off = 0; + unsigned int start, step; + + start = interlace_start[pass]; + step = interlace_step[pass]; + row_num = interlace_row_start[pass] + + interlace_row_step[pass] * row_num; + + /* Copy the data to our current row taking interlacing + * into consideration */ + row = buffer + (png_c->rowstride * row_num); + + for (dst_off = start; dst_off < rowbytes; dst_off += step) { + row[dst_off++] = new_row[src_off++]; + row[dst_off++] = new_row[src_off++]; + row[dst_off++] = new_row[src_off++]; + row[dst_off++] = new_row[src_off++]; + } + } else { + /* Do a fast memcpy of the row data */ + memcpy(row, new_row, rowbytes); + } +} + + +static void end_callback(png_structp png_s, png_infop info) +{ +} + +static nserror nspng_create_png_data(nspng_content *png_c) +{ + union content_msg_data msg_data; + + png_c->bitmap = NULL; + + png_c->png = png_create_read_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0); + if (png_c->png == NULL) { + msg_data.error = messages_get("NoMemory"); + content_broadcast(&png_c->base, CONTENT_MSG_ERROR, msg_data); + return NSERROR_NOMEM; + } + + png_set_error_fn(png_c->png, NULL, nspng_error, nspng_warning); + + png_c->info = png_create_info_struct(png_c->png); + if (png_c->info == NULL) { + png_destroy_read_struct(&png_c->png, &png_c->info, 0); + + msg_data.error = messages_get("NoMemory"); + content_broadcast(&png_c->base, CONTENT_MSG_ERROR, msg_data); + return NSERROR_NOMEM; + } + + if (setjmp(png_jmpbuf(png_c->png))) { + png_destroy_read_struct(&png_c->png, &png_c->info, 0); + LOG("Failed to set callbacks"); + png_c->png = NULL; + png_c->info = NULL; + + msg_data.error = messages_get("PNGError"); + content_broadcast(&png_c->base, CONTENT_MSG_ERROR, msg_data); + return NSERROR_NOMEM; + } + + png_set_progressive_read_fn(png_c->png, png_c, + info_callback, row_callback, end_callback); + + return NSERROR_OK; +} + +static nserror nspng_create(const content_handler *handler, + lwc_string *imime_type, const struct http_parameter *params, + llcache_handle *llcache, const char *fallback_charset, + bool quirks, struct content **c) +{ + nspng_content *png_c; + nserror error; + + png_c = calloc(1, sizeof(nspng_content)); + if (png_c == NULL) + return NSERROR_NOMEM; + + error = content__init(&png_c->base, + handler, + imime_type, + params, + llcache, + fallback_charset, + quirks); + if (error != NSERROR_OK) { + free(png_c); + return error; + } + + error = nspng_create_png_data(png_c); + if (error != NSERROR_OK) { + free(png_c); + return error; + } + + *c = (struct content *)png_c; + + return NSERROR_OK; +} + + +static bool nspng_process_data(struct content *c, const char *data, + unsigned int size) +{ + nspng_content *png_c = (nspng_content *)c; + union content_msg_data msg_data; + volatile bool ret = true; + + if (png_c->no_process_data) { + return ret; + } + + switch (setjmp(png_jmpbuf(png_c->png))) { + case CBERR_NONE: /* direct return */ + png_process_data(png_c->png, png_c->info, (uint8_t *)data, size); + break; + + case CBERR_NOPRE: /* not going to progressive convert */ + png_c->no_process_data = true; + break; + + default: /* fatal error from library processing png */ + if (png_c->bitmap != NULL) { + /* A bitmap managed to get created so + * operation is past header and possibly some + * conversion happened before faliure. + * + * In this case keep the partial + * conversion. This is usually seen if a png + * has been truncated (often jsut lost its + * last byte and hence end of image marker) + */ + png_c->no_process_data = true; + } else { + /* not managed to progress past header, clean + * up png conversion and signal the content + * error + */ + LOG("Fatal PNG error during header, error content"); + + png_destroy_read_struct(&png_c->png, &png_c->info, 0); + png_c->png = NULL; + png_c->info = NULL; + + msg_data.error = messages_get("PNGError"); + content_broadcast(c, CONTENT_MSG_ERROR, msg_data); + + ret = false; + + } + break; + } + + return ret; +} + +struct png_cache_read_data_s { + const char *data; + unsigned long size; +}; + +/** PNG library read fucntion to read data from a memory array + */ +static void +png_cache_read_fn(png_structp png_ptr, png_bytep data, png_size_t length) +{ + struct png_cache_read_data_s *png_cache_read_data; + png_cache_read_data = png_get_io_ptr(png_ptr); + + if (length > png_cache_read_data->size) { + length = png_cache_read_data->size; + } + + if (length == 0) { + png_error(png_ptr, "Read Error"); + } + + memcpy(data, png_cache_read_data->data, length); + + png_cache_read_data->data += length; + png_cache_read_data->size -= length; +} + +/** calculate an array of row pointers into a bitmap data area + */ +static png_bytep *calc_row_pointers(struct bitmap *bitmap) +{ + int height = guit->bitmap->get_height(bitmap); + unsigned char *buffer= guit->bitmap->get_buffer(bitmap); + size_t rowstride = guit->bitmap->get_rowstride(bitmap); + png_bytep *row_ptrs; + int hloop; + + /* The buffer allocation may occour when the buffer is aquired + * and therefore may fail. + */ + if (buffer == NULL) { + return NULL; + } + + row_ptrs = malloc(sizeof(png_bytep) * height); + + if (row_ptrs != NULL) { + for (hloop = 0; hloop < height; hloop++) { + row_ptrs[hloop] = buffer + (rowstride * hloop); + } + } + + return row_ptrs; +} + +/** PNG content to bitmap conversion. + * + * This routine generates a bitmap object from a PNG image content + */ +static struct bitmap * +png_cache_convert(struct content *c) +{ + png_structp png_ptr; + png_infop info_ptr; + png_infop end_info_ptr; + volatile struct bitmap * volatile bitmap = NULL; + struct png_cache_read_data_s png_cache_read_data; + png_uint_32 width, height; + volatile png_bytep * volatile row_pointers = NULL; + + png_cache_read_data.data = + content__get_source_data(c, &png_cache_read_data.size); + + if ((png_cache_read_data.data == NULL) || + (png_cache_read_data.size <= 8)) { + return NULL; + } + + png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, + nspng_error, nspng_warning); + if (png_ptr == NULL) { + return NULL; + } + + info_ptr = png_create_info_struct(png_ptr); + if (info_ptr == NULL) { + png_destroy_read_struct(&png_ptr, NULL, NULL); + return NULL; + } + + end_info_ptr = png_create_info_struct(png_ptr); + if (end_info_ptr == NULL) { + png_destroy_read_struct(&png_ptr, &info_ptr, NULL); + return NULL; + } + + /* setup error exit path */ + if (setjmp(png_jmpbuf(png_ptr))) { + /* cleanup and bail */ + goto png_cache_convert_error; + } + + /* read from a buffer instead of stdio */ + png_set_read_fn(png_ptr, &png_cache_read_data, png_cache_read_fn); + + /* ensure the png info structure is populated */ + png_read_info(png_ptr, info_ptr); + + /* setup output transforms */ + nspng_setup_transforms(png_ptr, info_ptr); + + width = png_get_image_width(png_ptr, info_ptr); + height = png_get_image_height(png_ptr, info_ptr); + + /* Claim the required memory for the converted PNG */ + bitmap = guit->bitmap->create(width, height, BITMAP_NEW); + if (bitmap == NULL) { + /* cleanup and bail */ + goto png_cache_convert_error; + } + + row_pointers = calc_row_pointers((struct bitmap *) bitmap); + + if (row_pointers != NULL) { + png_read_image(png_ptr, (png_bytep *) row_pointers); + } else { + guit->bitmap->destroy((struct bitmap *)bitmap); + bitmap = NULL; + } + +png_cache_convert_error: + + /* cleanup png read */ + png_destroy_read_struct(&png_ptr, &info_ptr, &end_info_ptr); + + if (row_pointers != NULL) { + free((png_bytep *) row_pointers); + } + + if (bitmap != NULL) { + guit->bitmap->modified((struct bitmap *)bitmap); + } + + return (struct bitmap *)bitmap; +} + +static bool nspng_convert(struct content *c) +{ + nspng_content *png_c = (nspng_content *) c; + char *title; + + assert(png_c->png != NULL); + assert(png_c->info != NULL); + + /* clean up png structures */ + png_destroy_read_struct(&png_c->png, &png_c->info, 0); + + /* set title text */ + title = messages_get_buff("PNGTitle", + nsurl_access_leaf(llcache_handle_get_url(c->llcache)), + c->width, c->height); + if (title != NULL) { + content__set_title(c, title); + free(title); + } + + if (png_c->bitmap != NULL) { + guit->bitmap->set_opaque(png_c->bitmap, guit->bitmap->test_opaque(png_c->bitmap)); + guit->bitmap->modified(png_c->bitmap); + } + + image_cache_add(c, png_c->bitmap, png_cache_convert); + + content_set_ready(c); + content_set_done(c); + content_set_status(c, ""); + + return true; +} + + +static nserror nspng_clone(const struct content *old_c, struct content **new_c) +{ + nspng_content *clone_png_c; + nserror error; + const char *data; + unsigned long size; + + clone_png_c = calloc(1, sizeof(nspng_content)); + if (clone_png_c == NULL) + return NSERROR_NOMEM; + + error = content__clone(old_c, &clone_png_c->base); + if (error != NSERROR_OK) { + content_destroy(&clone_png_c->base); + return error; + } + + /* Simply replay create/process/convert */ + error = nspng_create_png_data(clone_png_c); + if (error != NSERROR_OK) { + content_destroy(&clone_png_c->base); + return error; + } + + data = content__get_source_data(&clone_png_c->base, &size); + if (size > 0) { + if (nspng_process_data(&clone_png_c->base, data, size) == false) { + content_destroy(&clone_png_c->base); + return NSERROR_NOMEM; + } + } + + if ((old_c->status == CONTENT_STATUS_READY) || + (old_c->status == CONTENT_STATUS_DONE)) { + if (nspng_convert(&clone_png_c->base) == false) { + content_destroy(&clone_png_c->base); + return NSERROR_CLONE_FAILED; + } + } + + *new_c = (struct content *)clone_png_c; + + return NSERROR_OK; +} + +static const content_handler nspng_content_handler = { + .create = nspng_create, + .process_data = nspng_process_data, + .data_complete = nspng_convert, + .clone = nspng_clone, + .destroy = image_cache_destroy, + .redraw = image_cache_redraw, + .get_internal = image_cache_get_internal, + .type = image_cache_content_type, + .no_share = false, +}; + +static const char *nspng_types[] = { + "image/png", + "image/x-png" +}; + +CONTENT_FACTORY_REGISTER_TYPES(nspng, nspng_types, nspng_content_handler); |