Use busybox error handling functions wherever possible.
[oweals/busybox.git] / rpmunpack.c
1 /*
2  * rpmunpack for busybox
3  *
4  * rpmunpack.c  -  Utility program to unpack an RPM archive
5  *
6  * Gero Kuhlmann <gero@gkminix.han.de> 1998
7  *
8  *  This program is public domain software; you can do whatever you like
9  *  with this source, including modifying and redistributing it.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
14  */
15  
16 #include "busybox.h" 
17 #include <fcntl.h>
18
19 /*
20  * Some general definitions
21  */
22 #define BUFSIZE         512
23 #define RPM_MAGIC       "\355\253\356\333"
24 #define GZ_MAGIC_1      '\037'
25 #define GZ_MAGIC_2      '\213'
26
27 /*
28  * Global variables
29  */
30 static char buffer[BUFSIZE];
31 static char *progname;
32 static int infile, outfile;
33
34 /*
35  * Read a specified number of bytes from input file
36  */
37 static void myread(int num)
38 {
39   int err;
40
41   if ((err = read(infile, buffer, num)) != num) {
42         if (err < 0)
43                 perror_msg_and_die(progname);
44         else
45                 error_msg_and_die("Unexpected end of input file!\n");
46   }
47 }
48
49 /*
50  * Main program
51  */
52 int rpmunpack_main(int argc, char **argv)
53 {
54   int len, status = 0;
55
56   /* Get our own program name */
57   if ((progname = strrchr(argv[0], '/')) == NULL)
58         progname = argv[0];
59   else
60         progname++;
61
62   /* Check for command line parameters */
63         if (argc>=2 && *argv[1]=='-') {
64            usage(rpmunpack_usage);
65         }
66
67   /* Open input file */
68   if (argc == 1)
69         infile = STDIN_FILENO;
70   else if ((infile = open(argv[1], O_RDONLY)) < 0)
71         perror_msg_and_die("%s", argv[1]);
72
73   /* Read magic ID and output filename */
74   myread(4);
75   if (strncmp(buffer, RPM_MAGIC, 4)) {
76         fprintf(stderr, "Input file is not in RPM format!\n");
77         exit(1);
78   }
79   myread(6);            /* Skip flags */
80   myread(64);
81   buffer[64] = '\0';
82
83   /* Open output file */
84   strcat(buffer, ".cpio.gz");
85   if (infile == STDIN_FILENO)
86         outfile = STDOUT_FILENO;
87   else if ((outfile = open(buffer, O_WRONLY | O_CREAT | O_TRUNC, 0644)) < 0)
88           perror_msg_and_die("%s", buffer);
89
90   /*
91    * Now search for the GZIP signature. This is rather awkward, but I don't
92    * know any other way how to find out the exact starting position of the
93    * archive within the input file. There are a couple of data structures
94    * and texts (obviously descriptions, installation shell scripts etc.)
95    * coming before the archive, but even they start at different offsets
96    * with different RPM files. However, it looks like the GZIP signature
97    * never appears before offset 0x200, so we skip these first couple of
98    * bytes to make the signature scan a little more reliable.
99    */
100   myread(0x200 - 74);
101   while (status < 2) {
102         myread(1);
103         if (status == 0 && buffer[0] == GZ_MAGIC_1)
104                 status++;
105         else if (status == 1 && buffer[0] == GZ_MAGIC_2)
106                 status++;
107         else
108                 status = 0;
109   }
110   buffer[0] = GZ_MAGIC_1;
111   buffer[1] = GZ_MAGIC_2;
112   if (write(outfile, buffer, 2) < 0)
113         perror_msg_and_die("write");
114
115   /* Now simply copy the GZIP archive into the output file */
116   while ((len = read(infile, buffer, BUFSIZE)) > 0) {
117         if (write(outfile, buffer, len) < 0)
118                 perror_msg_and_die("write");
119   }
120   if (len < 0)
121     perror_msg_and_die("read");
122   return EXIT_SUCCESS;
123 }