all_read.c 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. /* vi: set sw=4 ts=4: */
  2. /*
  3. * Utility routines.
  4. *
  5. * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
  6. *
  7. * This program is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU General Public License as published by
  9. * the Free Software Foundation; either version 2 of the License, or
  10. * (at your option) any later version.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU General Public License
  18. * along with this program; if not, write to the Free Software
  19. * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  20. */
  21. #include <stdio.h>
  22. #include <unistd.h>
  23. #include <errno.h>
  24. #include "libbb.h"
  25. extern void archive_xread_all(int fd, char *buf, size_t count)
  26. {
  27. ssize_t size;
  28. size = full_read(fd, buf, count);
  29. if (size != count) {
  30. perror_msg_and_die("Short read");
  31. }
  32. return;
  33. }
  34. /*
  35. * Read all of the supplied buffer from a file.
  36. * This does multiple reads as necessary.
  37. * Returns the amount read, or -1 on an error.
  38. * A short read is returned on an end of file.
  39. */
  40. ssize_t full_read(int fd, char *buf, int len)
  41. {
  42. ssize_t cc;
  43. ssize_t total;
  44. total = 0;
  45. while (len > 0) {
  46. cc = safe_read(fd, buf, len);
  47. if (cc < 0)
  48. return cc; /* read() returns -1 on failure. */
  49. if (cc == 0)
  50. break;
  51. buf = ((char *)buf) + cc;
  52. total += cc;
  53. len -= cc;
  54. }
  55. return total;
  56. }
  57. ssize_t safe_read(int fd, void *buf, size_t count)
  58. {
  59. ssize_t n;
  60. do {
  61. n = read(fd, buf, count);
  62. } while (n < 0 && errno == EINTR);
  63. return n;
  64. }
  65. /* END CODE */