buffer.c 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. /*
  2. * Copyright (c) Christos Zoulas 2017.
  3. * All Rights Reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions
  7. * are met:
  8. * 1. Redistributions of source code must retain the above copyright
  9. * notice immediately at the beginning of the file, without modification,
  10. * this list of conditions, and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  16. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18. * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR
  19. * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  21. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  22. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  23. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  24. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  25. * SUCH DAMAGE.
  26. */
  27. #include "file.h"
  28. #ifndef lint
  29. FILE_RCSID("@(#)$File: buffer.c,v 1.8 2020/02/16 15:52:49 christos Exp $")
  30. #endif /* lint */
  31. #include "magic.h"
  32. #ifdef PHP_WIN32
  33. #include "win32/unistd.h"
  34. #else
  35. #include <unistd.h>
  36. #endif
  37. #include <string.h>
  38. #include <stdlib.h>
  39. #include <sys/stat.h>
  40. void
  41. buffer_init(struct buffer *b, int fd, const zend_stat_t *st, const void *data,
  42. size_t len)
  43. {
  44. b->fd = fd;
  45. if (st)
  46. memcpy(&b->st, st, sizeof(b->st));
  47. else if (b->fd == -1 || zend_fstat(b->fd, &b->st) == -1)
  48. memset(&b->st, 0, sizeof(b->st));
  49. b->fbuf = data;
  50. b->flen = len;
  51. b->eoff = 0;
  52. b->ebuf = NULL;
  53. b->elen = 0;
  54. }
  55. void
  56. buffer_fini(struct buffer *b)
  57. {
  58. efree(b->ebuf);
  59. }
  60. int
  61. buffer_fill(const struct buffer *bb)
  62. {
  63. struct buffer *b = CCAST(struct buffer *, bb);
  64. if (b->elen != 0)
  65. return b->elen == FILE_BADSIZE ? -1 : 0;
  66. if (!S_ISREG(b->st.st_mode))
  67. goto out;
  68. b->elen = CAST(size_t, b->st.st_size) < b->flen ?
  69. CAST(size_t, b->st.st_size) : b->flen;
  70. if ((b->ebuf = emalloc(b->elen)) == NULL)
  71. goto out;
  72. b->eoff = b->st.st_size - b->elen;
  73. if (FINFO_LSEEK_FUNC(b->fd, b->eoff, SEEK_SET) == (zend_off_t)-1 ||
  74. FINFO_READ_FUNC(b->fd, b->ebuf, b->elen) != (ssize_t)b->elen)
  75. {
  76. efree(b->ebuf);
  77. b->ebuf = NULL;
  78. goto out;
  79. }
  80. return 0;
  81. out:
  82. b->elen = FILE_BADSIZE;
  83. return -1;
  84. }