pngrio.c 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177
  1. /* pngrio.c - functions for data input
  2. *
  3. * Last changed in libpng 1.5.0 [January 6, 2011]
  4. * Copyright (c) 1998-2011 Glenn Randers-Pehrson
  5. * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
  6. * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
  7. *
  8. * This code is released under the libpng license.
  9. * For conditions of distribution and use, see the disclaimer
  10. * and license in png.h
  11. *
  12. * This file provides a location for all input. Users who need
  13. * special handling are expected to write a function that has the same
  14. * arguments as this and performs a similar function, but that possibly
  15. * has a different input method. Note that you shouldn't change this
  16. * function, but rather write a replacement function and then make
  17. * libpng use it at run time with png_set_read_fn(...).
  18. */
  19. #include "pngpriv.h"
  20. #ifdef PNG_READ_SUPPORTED
  21. /* Read the data from whatever input you are using. The default routine
  22. * reads from a file pointer. Note that this routine sometimes gets called
  23. * with very small lengths, so you should implement some kind of simple
  24. * buffering if you are using unbuffered reads. This should never be asked
  25. * to read more then 64K on a 16 bit machine.
  26. */
  27. void /* PRIVATE */
  28. png_read_data(png_structp png_ptr, png_bytep data, png_size_t length)
  29. {
  30. png_debug1(4, "reading %d bytes", (int)length);
  31. if (png_ptr->read_data_fn != NULL)
  32. (*(png_ptr->read_data_fn))(png_ptr, data, length);
  33. else
  34. png_error(png_ptr, "Call to NULL read function");
  35. }
  36. #ifdef PNG_STDIO_SUPPORTED
  37. /* This is the function that does the actual reading of data. If you are
  38. * not reading from a standard C stream, you should create a replacement
  39. * read_data function and use it at run time with png_set_read_fn(), rather
  40. * than changing the library.
  41. */
  42. # ifndef USE_FAR_KEYWORD
  43. void PNGCBAPI
  44. png_default_read_data(png_structp png_ptr, png_bytep data, png_size_t length)
  45. {
  46. png_size_t check;
  47. if (png_ptr == NULL)
  48. return;
  49. /* fread() returns 0 on error, so it is OK to store this in a png_size_t
  50. * instead of an int, which is what fread() actually returns.
  51. */
  52. check = fread(data, 1, length, (png_FILE_p)png_ptr->io_ptr);
  53. if (check != length)
  54. png_error(png_ptr, "Read Error");
  55. }
  56. # else
  57. /* This is the model-independent version. Since the standard I/O library
  58. can't handle far buffers in the medium and small models, we have to copy
  59. the data.
  60. */
  61. #define NEAR_BUF_SIZE 1024
  62. #define MIN(a,b) (a <= b ? a : b)
  63. static void PNGCBAPI
  64. png_default_read_data(png_structp png_ptr, png_bytep data, png_size_t length)
  65. {
  66. png_size_t check;
  67. png_byte *n_data;
  68. png_FILE_p io_ptr;
  69. if (png_ptr == NULL)
  70. return;
  71. /* Check if data really is near. If so, use usual code. */
  72. n_data = (png_byte *)CVT_PTR_NOCHECK(data);
  73. io_ptr = (png_FILE_p)CVT_PTR(png_ptr->io_ptr);
  74. if ((png_bytep)n_data == data)
  75. {
  76. check = fread(n_data, 1, length, io_ptr);
  77. }
  78. else
  79. {
  80. png_byte buf[NEAR_BUF_SIZE];
  81. png_size_t read, remaining, err;
  82. check = 0;
  83. remaining = length;
  84. do
  85. {
  86. read = MIN(NEAR_BUF_SIZE, remaining);
  87. err = fread(buf, 1, read, io_ptr);
  88. png_memcpy(data, buf, read); /* copy far buffer to near buffer */
  89. if (err != read)
  90. break;
  91. else
  92. check += err;
  93. data += read;
  94. remaining -= read;
  95. }
  96. while (remaining != 0);
  97. }
  98. if ((png_uint_32)check != (png_uint_32)length)
  99. png_error(png_ptr, "read Error");
  100. }
  101. # endif
  102. #endif
  103. /* This function allows the application to supply a new input function
  104. * for libpng if standard C streams aren't being used.
  105. *
  106. * This function takes as its arguments:
  107. *
  108. * png_ptr - pointer to a png input data structure
  109. *
  110. * io_ptr - pointer to user supplied structure containing info about
  111. * the input functions. May be NULL.
  112. *
  113. * read_data_fn - pointer to a new input function that takes as its
  114. * arguments a pointer to a png_struct, a pointer to
  115. * a location where input data can be stored, and a 32-bit
  116. * unsigned int that is the number of bytes to be read.
  117. * To exit and output any fatal error messages the new write
  118. * function should call png_error(png_ptr, "Error msg").
  119. * May be NULL, in which case libpng's default function will
  120. * be used.
  121. */
  122. void PNGAPI
  123. png_set_read_fn(png_structp png_ptr, png_voidp io_ptr,
  124. png_rw_ptr read_data_fn)
  125. {
  126. if (png_ptr == NULL)
  127. return;
  128. png_ptr->io_ptr = io_ptr;
  129. #ifdef PNG_STDIO_SUPPORTED
  130. if (read_data_fn != NULL)
  131. png_ptr->read_data_fn = read_data_fn;
  132. else
  133. png_ptr->read_data_fn = png_default_read_data;
  134. #else
  135. png_ptr->read_data_fn = read_data_fn;
  136. #endif
  137. /* It is an error to write to a read device */
  138. if (png_ptr->write_data_fn != NULL)
  139. {
  140. png_ptr->write_data_fn = NULL;
  141. png_warning(png_ptr,
  142. "Can't set both read_data_fn and write_data_fn in the"
  143. " same structure");
  144. }
  145. #ifdef PNG_WRITE_FLUSH_SUPPORTED
  146. png_ptr->output_flush_fn = NULL;
  147. #endif
  148. }
  149. #endif /* PNG_READ_SUPPORTED */