libbb/gz_open.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) many different people.  If you wrote this, please
   6  * acknowledge your work.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful, but
  14  * WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16  * General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  21  * USA
  22  */
  23
  24 #include <sys/types.h>
  25 #include <sys/wait.h>
  26 #include <signal.h>
  27 #include <stdlib.h>
  28 #include <string.h>
  29 #include <unistd.h>
  30 #include "libbb.h"
  31
  32 extern FILE *gz_open(FILE *compressed_file, int *pid)
  33 {
  34         int unzip_pipe[2];
  35
  36         if (pipe(unzip_pipe)!=0) {
  37                 error_msg("pipe error");
  38                 return(NULL);
  39         }
  40         if ((*pid = fork()) == -1) {
  41                 error_msg("fork failed");
  42                 return(NULL);
  43         }
  44         if (*pid==0) {
  45                 /* child process */
  46                 close(unzip_pipe[0]);
  47                 unzip(compressed_file, fdopen(unzip_pipe[1], "w"));
  48                 fflush(NULL);
  49                 fclose(compressed_file);
  50                 close(unzip_pipe[1]);
  51                 exit(EXIT_SUCCESS);
  52         }
  53         close(unzip_pipe[1]);
  54         if (unzip_pipe[0] == -1) {
  55                 error_msg("gzip stream init failed");
  56         }
  57         return(fdopen(unzip_pipe[0], "r"));
  58 }
  59
  60 /* gz_open implementation using gunzip and a vfork/exec -- dodges OOM killer */
  61 extern FILE *gzvopen(FILE *compressed_file, int *pid)
  62 {
  63         int unzip_pipe[2];
  64         off_t floc;
  65         int cfile;
  66
  67         /* create a new file descriptor for the input stream
  68          * (it *must* be associated with a file)
  69          * and seek to the same position in that fd as the stream.
  70          */
  71         cfile = dup(fileno(compressed_file));
  72         floc = ftello(compressed_file);
  73         lseek(cfile, floc, SEEK_SET);
  74
  75         /* create the pipe */
  76         if (pipe(unzip_pipe)!=0) {
  77                 error_msg("gzvopen(): pipe error");
  78                 return(NULL);
  79         }
  80
  81         *pid = vfork();
  82
  83         if (*pid < 0) {
  84                 error_msg("gzvopen(): fork failed");
  85                 return(NULL);
  86         }
  87
  88         if (*pid==0) {
  89                 /* child process - reads STDIN, writes to pipe */
  90
  91                 /* close unused read end of pipe */
  92                 close(unzip_pipe[0]);
  93
  94                 /* connect child's stdout to the pipe write end */
  95                 dup2(unzip_pipe[1], 1);
  96
  97                 /* connect child's stdin to the fd passed in to us */
  98                 dup2(cfile, 0);
  99
 100                 /* execute the gunzip utility */
 101                 execlp("gunzip","gunzip",NULL);
 102
 103                 /* if we get here, we had a failure - since we are
 104                  * using vfork(), we cannot call exit(), must call _exit().
 105                  */
 106                 _exit(-1);
 107         }
 108
 109         /* Parent process is executing here */
 110
 111         /* we have no more need of the duplicate fd */
 112         close(cfile);
 113
 114         /* close the write end of the pipe */
 115         close(unzip_pipe[1]);
 116
 117         /* return the read end of the pipe as a FILE */
 118         return(fdopen(unzip_pipe[0], "r"));
 119 }
 120
 121 extern void gzvclose(int gunzip_pid)
 122 {
 123         if (kill(gunzip_pid, SIGTERM) == -1) {
 124                 perror("gzvclose()");
 125                 fprintf(stderr,"%s: unable to kill gunzip pid.\n",
 126                         __FUNCTION__);
 127         }
 128
 129         if (waitpid(gunzip_pid, NULL, 0) == -1) {
 130                 fprintf(stderr,"%s: unable to wait on gunzip pid.\n",
 131                         __FUNCTION__);
 132         }
 133 }