wks: Avoid long trustdb checks.
[gnupg.git] / tools / gpgtar-extract.c
1 /* gpgtar-extract.c - Extract from a TAR archive
2  * Copyright (C) 2010 Free Software Foundation, Inc.
3  *
4  * This file is part of GnuPG.
5  *
6  * GnuPG is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 3 of the License, or
9  * (at your option) any later version.
10  *
11  * GnuPG 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.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include <config.h>
21 #include <errno.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <sys/types.h>
26 #include <sys/stat.h>
27 #include <unistd.h>
28 #include <assert.h>
29
30 #include "i18n.h"
31 #include "../common/exectool.h"
32 #include "../common/sysutils.h"
33 #include "../common/ccparray.h"
34 #include "gpgtar.h"
35
36
37 static gpg_error_t
38 extract_regular (estream_t stream, const char *dirname,
39                  tar_header_t hdr)
40 {
41   gpg_error_t err;
42   char record[RECORDSIZE];
43   size_t n, nbytes, nwritten;
44   char *fname;
45   estream_t outfp = NULL;
46
47   fname = strconcat (dirname, "/", hdr->name, NULL);
48   if (!fname)
49     {
50       err = gpg_error_from_syserror ();
51       log_error ("error creating filename: %s\n", gpg_strerror (err));
52       goto leave;
53     }
54   else
55     err = 0;
56
57   if (opt.dry_run)
58     outfp = es_fopenmem (0, "wb");
59   else
60     outfp = es_fopen (fname, "wb");
61   if (!outfp)
62     {
63       err = gpg_error_from_syserror ();
64       log_error ("error creating '%s': %s\n", fname, gpg_strerror (err));
65       goto leave;
66     }
67
68   for (n=0; n < hdr->nrecords;)
69     {
70       err = read_record (stream, record);
71       if (err)
72         goto leave;
73       n++;
74       if (n < hdr->nrecords || (hdr->size && !(hdr->size % RECORDSIZE)))
75         nbytes = RECORDSIZE;
76       else
77         nbytes = (hdr->size % RECORDSIZE);
78
79       nwritten = es_fwrite (record, 1, nbytes, outfp);
80       if (nwritten != nbytes)
81         {
82           err = gpg_error_from_syserror ();
83           log_error ("error writing '%s': %s\n", fname, gpg_strerror (err));
84           goto leave;
85         }
86     }
87   /* Fixme: Set permissions etc.  */
88
89  leave:
90   if (!err && opt.verbose)
91     log_info ("extracted '%s'\n", fname);
92   es_fclose (outfp);
93   if (err && fname && outfp)
94     {
95       if (gnupg_remove (fname))
96         log_error ("error removing incomplete file '%s': %s\n",
97                    fname, gpg_strerror (gpg_error_from_syserror ()));
98     }
99   xfree (fname);
100   return err;
101 }
102
103
104 static gpg_error_t
105 extract_directory (const char *dirname, tar_header_t hdr)
106 {
107   gpg_error_t err;
108   char *fname;
109   size_t prefixlen;
110
111   prefixlen = strlen (dirname) + 1;
112   fname = strconcat (dirname, "/", hdr->name, NULL);
113   if (!fname)
114     {
115       err = gpg_error_from_syserror ();
116       log_error ("error creating filename: %s\n", gpg_strerror (err));
117       goto leave;
118     }
119   else
120     err = 0;
121
122   if (fname[strlen (fname)-1] == '/')
123     fname[strlen (fname)-1] = 0;
124
125  /* Note that we don't need to care about EEXIST because we always
126      extract into a new hierarchy.  */
127   if (! opt.dry_run && gnupg_mkdir (fname, "-rwx------"))
128     {
129       err = gpg_error_from_syserror ();
130       if (gpg_err_code (err) == GPG_ERR_ENOENT)
131         {
132           /* Try to create the directory with parents but keep the
133              original error code in case of a failure.  */
134           char *p;
135           int rc = 0;
136
137           for (p = fname+prefixlen; (p = strchr (p, '/')); p++)
138             {
139               *p = 0;
140               rc = gnupg_mkdir (fname, "-rwx------");
141               *p = '/';
142               if (rc)
143                 break;
144             }
145           if (!rc && !gnupg_mkdir (fname, "-rwx------"))
146             err = 0;
147         }
148       if (err)
149         log_error ("error creating directory '%s': %s\n",
150                    fname, gpg_strerror (err));
151     }
152
153  leave:
154   if (!err && opt.verbose)
155     log_info ("created   '%s/'\n", fname);
156   xfree (fname);
157   return err;
158 }
159
160
161 static gpg_error_t
162 extract (estream_t stream, const char *dirname, tar_header_t hdr)
163 {
164   gpg_error_t err;
165   size_t n;
166
167   n = strlen (hdr->name);
168 #ifdef HAVE_DOSISH_SYSTEM
169   if (strchr (hdr->name, '\\'))
170     {
171       log_error ("filename '%s' contains a backslash - "
172                  "can't extract on this system\n", hdr->name);
173       return gpg_error (GPG_ERR_INV_NAME);
174     }
175 #endif /*HAVE_DOSISH_SYSTEM*/
176
177   if (!n
178       || strstr (hdr->name, "//")
179       || strstr (hdr->name, "/../")
180       || !strncmp (hdr->name, "../", 3)
181       || (n >= 3 && !strcmp (hdr->name+n-3, "/.." )))
182     {
183       log_error ("filename '%s' as suspicious parts - not extracting\n",
184                  hdr->name);
185       return gpg_error (GPG_ERR_INV_NAME);
186     }
187
188   if (hdr->typeflag == TF_REGULAR || hdr->typeflag == TF_UNKNOWN)
189     err = extract_regular (stream, dirname, hdr);
190   else if (hdr->typeflag == TF_DIRECTORY)
191     err = extract_directory (dirname, hdr);
192   else
193     {
194       char record[RECORDSIZE];
195
196       log_info ("unsupported file type %d for '%s' - skipped\n",
197                 (int)hdr->typeflag, hdr->name);
198       for (err = 0, n=0; !err && n < hdr->nrecords; n++)
199         err = read_record (stream, record);
200     }
201   return err;
202 }
203
204
205 /* Create a new directory to be used for extracting the tarball.
206    Returns the name of the directory which must be freed by the
207    caller.  In case of an error a diagnostic is printed and NULL
208    returned.  */
209 static char *
210 create_directory (const char *dirprefix)
211 {
212   gpg_error_t err = 0;
213   char *prefix_buffer = NULL;
214   char *dirname = NULL;
215   size_t n;
216   int idx;
217
218   /* Remove common suffixes.  */
219   n = strlen (dirprefix);
220   if (n > 4 && (!compare_filenames    (dirprefix + n - 4, EXTSEP_S GPGEXT_GPG)
221                 || !compare_filenames (dirprefix + n - 4, EXTSEP_S "pgp")
222                 || !compare_filenames (dirprefix + n - 4, EXTSEP_S "asc")
223                 || !compare_filenames (dirprefix + n - 4, EXTSEP_S "pem")
224                 || !compare_filenames (dirprefix + n - 4, EXTSEP_S "p7m")
225                 || !compare_filenames (dirprefix + n - 4, EXTSEP_S "p7e")))
226     {
227       prefix_buffer = xtrystrdup (dirprefix);
228       if (!prefix_buffer)
229         {
230           err = gpg_error_from_syserror ();
231           goto leave;
232         }
233       prefix_buffer[n-4] = 0;
234       dirprefix = prefix_buffer;
235     }
236
237
238
239   for (idx=1; idx < 5000; idx++)
240     {
241       xfree (dirname);
242       dirname = xtryasprintf ("%s_%d_", dirprefix, idx);
243       if (!dirname)
244         {
245           err = gpg_error_from_syserror ();
246           goto leave;
247         }
248       if (!gnupg_mkdir (dirname, "-rwx------"))
249         goto leave; /* Ready.  */
250       if (errno != EEXIST && errno != ENOTDIR)
251         {
252           err = gpg_error_from_syserror ();
253           goto leave;
254         }
255     }
256   err = gpg_error (GPG_ERR_LIMIT_REACHED);
257
258  leave:
259   if (err)
260     {
261       log_error ("error creating an extract directory: %s\n",
262                  gpg_strerror (err));
263       xfree (dirname);
264       dirname = NULL;
265     }
266   xfree (prefix_buffer);
267   return dirname;
268 }
269
270
271 \f
272 gpg_error_t
273 gpgtar_extract (const char *filename, int decrypt)
274 {
275   gpg_error_t err;
276   estream_t stream;
277   estream_t cipher_stream = NULL;
278   tar_header_t header = NULL;
279   const char *dirprefix = NULL;
280   char *dirname = NULL;
281
282   if (filename)
283     {
284       if (!strcmp (filename, "-"))
285         stream = es_stdin;
286       else
287         stream = es_fopen (filename, "rb");
288       if (!stream)
289         {
290           err = gpg_error_from_syserror ();
291           log_error ("error opening '%s': %s\n", filename, gpg_strerror (err));
292           return err;
293         }
294     }
295   else
296     stream = es_stdin;
297
298   if (stream == es_stdin)
299     es_set_binary (es_stdin);
300
301   if (decrypt)
302     {
303       strlist_t arg;
304       ccparray_t ccp;
305       const char **argv;
306
307       cipher_stream = stream;
308       stream = es_fopenmem (0, "rwb");
309       if (! stream)
310         {
311           err = gpg_error_from_syserror ();
312           goto leave;
313         }
314
315       ccparray_init (&ccp, 0);
316
317       ccparray_put (&ccp, "--decrypt");
318       for (arg = opt.gpg_arguments; arg; arg = arg->next)
319         ccparray_put (&ccp, arg->d);
320
321       ccparray_put (&ccp, NULL);
322       argv = ccparray_get (&ccp, NULL);
323       if (!argv)
324         {
325           err = gpg_error_from_syserror ();
326           goto leave;
327         }
328
329       err = gnupg_exec_tool_stream (opt.gpg_program, argv,
330                                     cipher_stream, NULL, stream, NULL, NULL);
331       xfree (argv);
332       if (err)
333         goto leave;
334
335       err = es_fseek (stream, 0, SEEK_SET);
336       if (err)
337         goto leave;
338     }
339
340   if (opt.directory)
341     dirname = xtrystrdup (opt.directory);
342   else
343     {
344       if (filename)
345         {
346           dirprefix = strrchr (filename, '/');
347           if (dirprefix)
348             dirprefix++;
349           else
350             dirprefix = filename;
351         }
352       else if (opt.filename)
353         {
354           dirprefix = strrchr (opt.filename, '/');
355           if (dirprefix)
356             dirprefix++;
357           else
358             dirprefix = opt.filename;
359         }
360
361       if (!dirprefix || !*dirprefix)
362         dirprefix = "GPGARCH";
363
364       dirname = create_directory (dirprefix);
365       if (!dirname)
366         {
367           err = gpg_error (GPG_ERR_GENERAL);
368           goto leave;
369         }
370     }
371
372   if (opt.verbose)
373     log_info ("extracting to '%s/'\n", dirname);
374
375   for (;;)
376     {
377       err = gpgtar_read_header (stream, &header);
378       if (err || header == NULL)
379         goto leave;
380
381       err = extract (stream, dirname, header);
382       if (err)
383         goto leave;
384       xfree (header);
385       header = NULL;
386     }
387
388
389  leave:
390   xfree (header);
391   xfree (dirname);
392   if (stream != es_stdin)
393     es_fclose (stream);
394   if (stream != cipher_stream)
395     es_fclose (cipher_stream);
396   return err;
397 }