libbb/copyfd.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) 1999-2003 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
  22 #include <unistd.h>
  23 #include <string.h>
  24 #include <errno.h>
  25 #include "busybox.h"
  26
  27 #if BUFSIZ < 4096
  28 #undef BUFSIZ
  29 #define BUFSIZ 4096
  30 #endif
  31
  32 /* If chunksize is 0 copy until EOF */
  33 extern int bb_copyfd(int fd1, int fd2, const off_t chunksize)
  34 {
  35         ssize_t nread;
  36         size_t size;
  37         off_t remaining;
  38         RESERVE_CONFIG_BUFFER(buffer,BUFSIZ);
  39
  40         remaining = size = BUFSIZ;
  41         if (chunksize) {
  42                 remaining = chunksize;
  43         }
  44
  45         do {
  46                 if (size > remaining) {
  47                         size = remaining;
  48                 }
  49
  50                 if ((nread = safe_read(fd1, buffer, size)) > 0) {
  51                         if (bb_full_write(fd2, buffer, nread) < 0) {
  52                                 bb_perror_msg(bb_msg_write_error);      /* match Read error below */
  53                                 break;
  54                         }
  55                         if (chunksize && ((remaining -= nread) == 0)) {
  56                                 return 0;
  57                         }
  58                 } else if (!nread) {
  59                         if (chunksize) {
  60                                 bb_error_msg("Unable to read all data");
  61                                 break;
  62                         }
  63                         return 0;
  64                 } else {                                /* nread < 0 */
  65                         bb_perror_msg("Read error");    /* match bb_msg_write_error above */
  66                         break;
  67                 }
  68
  69         } while (1);
  70
  71         return -1;
  72 }