Use a switch instead of successive if (strcmp()) statments.
[oweals/busybox.git] / archival / libunarchive / get_header_tar.c
1 /*
2  *  This program is free software; you can redistribute it and/or modify
3  *  it under the terms of the GNU General Public License as published by
4  *  the Free Software Foundation; either version 2 of the License, or
5  *  (at your option) any later version.
6  *
7  *  This program is distributed in the hope that it will be useful,
8  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
9  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  *  GNU Library General Public License for more details.
11  *
12  *  You should have received a copy of the GNU General Public License
13  *  along with this program; if not, write to the Free Software
14  *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
15  */
16
17 #include <stdio.h>
18 #include <stdlib.h>
19 #include <string.h>
20 #include "unarchive.h"
21 #include "libbb.h"
22
23 extern char get_header_tar(archive_handle_t *archive_handle)
24 {
25         file_header_t *file_header = archive_handle->file_header;
26         union {
27                 unsigned char raw[512];
28                 struct {
29                         char name[100]; /*   0-99 */
30                         char mode[8];   /* 100-107 */
31                         char uid[8];    /* 108-115 */
32                         char gid[8];    /* 116-123 */
33                         char size[12];  /* 124-135 */
34                         char mtime[12]; /* 136-147 */
35                         char chksum[8]; /* 148-155 */
36                         char typeflag;  /* 156-156 */
37                         char linkname[100];     /* 157-256 */
38                         char magic[6];  /* 257-262 */
39                         char version[2];        /* 263-264 */
40                         char uname[32]; /* 265-296 */
41                         char gname[32]; /* 297-328 */
42                         char devmajor[8];       /* 329-336 */
43                         char devminor[8];       /* 337-344 */
44                         char prefix[155];       /* 345-499 */
45                         char padding[12];       /* 500-512 */
46                 } formated;
47         } tar;
48         long sum = 0;
49         long i;
50         char *tmp;
51
52         /* Align header */
53         data_align(archive_handle, 512);
54
55         if (archive_xread(archive_handle, tar.raw, 512) != 512) {
56                 /* Assume end of file */
57                 return(EXIT_FAILURE);
58         }
59         archive_handle->offset += 512;
60
61         /* If there is no filename its an empty header */
62         if (tar.formated.name[0] == 0) {
63                 return(EXIT_SUCCESS);
64         }
65
66         /* Check header has valid magic, "ustar" is for the proper tar
67          * 0's are for the old tar format
68          */
69         if (strncmp(tar.formated.magic, "ustar", 5) != 0) {
70 #ifdef CONFIG_FEATURE_TAR_OLD_FORMAT
71                 if (strncmp(tar.formated.magic, "\0\0\0\0\0", 5) != 0)
72 #endif
73                         error_msg_and_die("Invalid tar magic");
74         }
75         /* Do checksum on headers */
76         for (i =  0; i < 148 ; i++) {
77                 sum += tar.raw[i];
78         }
79         sum += ' ' * 8;
80         for (i =  156; i < 512 ; i++) {
81                 sum += tar.raw[i];
82         }
83         if (sum != strtol(tar.formated.chksum, NULL, 8)) {
84                 error_msg("Invalid tar header checksum");
85                 return(EXIT_FAILURE);
86         }
87
88         /* convert to type'ed variables */
89         if (tar.formated.prefix[0] == 0) {
90                 file_header->name = strdup(tar.formated.name);
91         } else {
92                 file_header->name = concat_path_file(tar.formated.prefix, tar.formated.name);
93         }
94         tmp = last_char_is(archive_handle->file_header->name, '/');
95         if (tmp) {
96                 *tmp = '\0';
97         }
98
99         file_header->mode = strtol(tar.formated.mode, NULL, 8);
100         file_header->uid = strtol(tar.formated.uid, NULL, 8);
101         file_header->gid = strtol(tar.formated.gid, NULL, 8);
102         file_header->size = strtol(tar.formated.size, NULL, 8);
103         file_header->mtime = strtol(tar.formated.mtime, NULL, 8);
104         file_header->link_name = (tar.formated.linkname[0] != '\0') ? 
105             xstrdup(tar.formated.linkname) : NULL;
106         file_header->device = (dev_t) ((strtol(tar.formated.devmajor, NULL, 8) << 8) +
107                                  strtol(tar.formated.devminor, NULL, 8));
108
109 #if defined CONFIG_FEATURE_TAR_OLD_FORMAT || defined CONFIG_FEATURE_GNUTAR_LONG_FILENAME
110         /* Fix mode, used by the old format */
111         switch (tar.formated.typeflag) {
112 # ifdef CONFIG_FEATURE_TAR_OLD_FORMAT
113         case 0:
114                 file_header->mode |= S_IFREG;
115                 break;
116         case 1:
117                 error_msg("Internal hard link not supported");
118                 break;
119         case 2:
120                 file_header->mode |= S_IFLNK;
121                 break;
122         case 3:
123                 file_header->mode |= S_IFCHR;
124                 break;
125         case 4:
126                 file_header->mode |= S_IFBLK;
127                 break;
128         case 5:
129                 file_header->mode |= S_IFDIR;
130                 break;
131         case 6:
132                 file_header->mode |= S_IFIFO;
133                 break;
134 # endif
135 # ifdef CONFIG_FEATURE_GNUTAR_LONG_FILENAME
136         case 'L': {
137                         char *longname;
138
139                         longname = xmalloc(file_header->size + 1);
140                         archive_xread_all(archive_handle, longname, file_header->size);
141                         longname[file_header->size] = '\0';
142                         archive_handle->offset += file_header->size;
143
144                         get_header_tar(archive_handle);
145                         file_header->name = longname;
146                         break;
147                 }
148         case 'K': {
149                         char *linkname;
150
151                         linkname = xmalloc(file_header->size + 1);
152                         archive_xread_all(archive_handle, linkname, file_header->size);
153                         linkname[file_header->size] = '\0';
154                         archive_handle->offset += file_header->size;
155
156                         get_header_tar(archive_handle);
157                         file_header->name = linkname;
158                         break;
159                 }
160 # endif
161         }
162 #endif
163         if (archive_handle->filter(archive_handle) == EXIT_SUCCESS) {
164                 archive_handle->action_header(archive_handle->file_header);
165                 archive_handle->flags |= ARCHIVE_EXTRACT_QUIET;
166                 archive_handle->action_data(archive_handle);
167                 archive_handle->passed = add_to_list(archive_handle->passed, archive_handle->file_header->name);
168         } else {
169                 data_skip(archive_handle);                      
170         }
171         archive_handle->offset += file_header->size;
172
173         return(EXIT_SUCCESS);
174 }
175