1f8c5759f060b0cec4a26cbbbdbcfd5d3d100b1c
[gnupg.git] / g10 / plaintext.c
1 /* plaintext.c -  process plaintext packets
2  * Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003, 2004,
3  *               2005, 2006 Free Software Foundation, Inc.
4  *
5  * This file is part of GnuPG.
6  *
7  * GnuPG 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  * GnuPG 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
15  * GNU 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
20  * USA.
21  */
22
23 #include <config.h>
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <errno.h>
28 #include <assert.h>
29 #include <sys/types.h>
30 #ifdef HAVE_DOSISH_SYSTEM
31 #include <fcntl.h> /* for setmode() */
32 #endif
33
34 #include "gpg.h"
35 #include "util.h"
36 #include "options.h"
37 #include "packet.h"
38 #include "ttyio.h"
39 #include "filter.h"
40 #include "main.h"
41 #include "status.h"
42 #include "i18n.h"
43
44
45 /****************
46  * Handle a plaintext packet.  If MFX is not NULL, update the MDs
47  * Note: we should use the filter stuff here, but we have to add some
48  *       easy mimic to set a read limit, so we calculate only the
49  *       bytes from the plaintext.
50  */
51 int
52 handle_plaintext( PKT_plaintext *pt, md_filter_context_t *mfx,
53                   int nooutput, int clearsig )
54 {
55     char *fname = NULL;
56     FILE *fp = NULL;
57     static off_t count=0;
58     int rc = 0;
59     int c;
60     int convert = (pt->mode == 't' || pt->mode == 'u');
61 #ifdef __riscos__
62     int filetype = 0xfff;
63 #endif
64
65     /* Let people know what the plaintext info is. This allows the
66        receiving program to try and do something different based on
67        the format code (say, recode UTF-8 to local). */
68     if(!nooutput && is_status_enabled())
69       {
70         char status[50];
71
72         sprintf(status,"%X %lu ",(byte)pt->mode,(ulong)pt->timestamp);
73         write_status_text_and_buffer(STATUS_PLAINTEXT,
74                                      status,pt->name,pt->namelen,0);
75
76         if(!pt->is_partial)
77           {
78             sprintf(status,"%lu",(ulong)pt->len);
79             write_status_text(STATUS_PLAINTEXT_LENGTH,status);
80           }
81       }
82
83     /* create the filename as C string */
84     if( nooutput )
85         ;
86     else if( opt.outfile ) {
87         fname = xmalloc( strlen( opt.outfile ) + 1);
88         strcpy(fname, opt.outfile );
89     }
90     else if( pt->namelen == 8 && !memcmp( pt->name, "_CONSOLE", 8 ) ) {
91         log_info(_("data not saved; use option \"--output\" to save it\n"));
92         nooutput = 1;
93     }
94     else if( !opt.use_embedded_filename ) {
95         fname = make_outfile_name( iobuf_get_real_fname(pt->buf) );
96         if( !fname )
97             fname = ask_outfile_name( pt->name, pt->namelen );
98         if( !fname ) {
99              rc = gpg_error (GPG_ERR_GENERAL); /* Can't create file. */
100              goto leave;
101         }
102     }
103     else {
104         fname = make_printable_string( pt->name, pt->namelen, 0 );
105     }
106
107     if( nooutput )
108         ;
109     else if ( iobuf_is_pipe_filename (fname) || !*fname)
110       {
111         /* No filename or "-" given; write to stdout. */
112         fp = stdout;
113 #ifdef HAVE_DOSISH_SYSTEM
114         setmode ( fileno(fp) , O_BINARY );
115 #endif
116       }
117     else {
118         while( !overwrite_filep (fname) ) {
119             char *tmp = ask_outfile_name (NULL, 0);
120             if ( !tmp || !*tmp ) {
121                 xfree (tmp);
122                 rc = gpg_error (GPG_ERR_GENERAL); /* G10ERR_CREATE_FILE*/
123                 goto leave;
124             }
125             xfree (fname);
126             fname = tmp;
127         }
128     }
129
130 #ifndef __riscos__
131     if( fp || nooutput )
132         ;
133     else if (is_secured_filename (fname))
134       {
135         errno = EPERM;
136         log_error(_("error creating `%s': %s\n"), fname, strerror(errno) );
137         rc = G10ERR_CREATE_FILE;
138         goto leave;
139       }
140     else if( !(fp = fopen(fname,"wb")) ) {
141         log_error(_("error creating `%s': %s\n"), fname, strerror(errno) );
142         rc = G10ERR_CREATE_FILE;
143         goto leave;
144     }
145 #else /* __riscos__ */
146     /* If no output filename was given, i.e. we constructed it,
147        convert all '.' in fname to '/' but not vice versa as
148        we don't create directories! */
149     if( !opt.outfile )
150         for( c=0; fname[c]; ++c )
151             if( fname[c] == '.' )
152                 fname[c] = '/';
153
154     if( fp || nooutput )
155         ;
156     else {
157         fp = fopen(fname,"wb");
158         if( !fp ) {
159             log_error(_("error creating `%s': %s\n"), fname, strerror(errno) );
160             rc = G10ERR_CREATE_FILE;
161             if (errno == 106)
162                 log_info("Do output file and input file have the same name?\n");
163             goto leave;
164         }
165
166         /* If there's a ,xxx extension in the embedded filename,
167            use that, else check whether the user input (in fname)
168            has a ,xxx appended, then use that in preference */
169         if( (c = riscos_get_filetype_from_string( pt->name,
170                                                   pt->namelen )) != -1 )
171             filetype = c;
172         if( (c = riscos_get_filetype_from_string( fname,
173                                                   strlen(fname) )) != -1 )
174             filetype = c;
175         riscos_set_filetype_by_number(fname, filetype);
176     }
177 #endif /* __riscos__ */
178
179     if( !pt->is_partial ) {
180         /* We have an actual length (which might be zero). */
181
182         if (clearsig) {
183             log_error ("clearsig encountered while not expected\n");
184             rc = G10ERR_UNEXPECTED;
185             goto leave;
186         }
187
188         if( convert ) { /* text mode */
189             for( ; pt->len; pt->len-- ) {
190                 if( (c = iobuf_get(pt->buf)) == -1 ) {
191                     rc = gpg_error_from_errno (errno);
192                     log_error ("problem reading source (%u bytes remaining)\n",
193                                (unsigned)pt->len);
194                     goto leave;
195                 }
196                 if( mfx->md )
197                     gcry_md_putc (mfx->md, c );
198 #ifndef HAVE_DOSISH_SYSTEM
199                 if( c == '\r' )  /* convert to native line ending */
200                     continue;    /* fixme: this hack might be too simple */
201 #endif
202                 if( fp )
203                   {
204                     if(opt.max_output && (++count)>opt.max_output)
205                       {
206                         log_error ("error writing to `%s': %s\n",
207                                    fname,"exceeded --max-output limit\n");
208                         rc = gpg_error (GPG_ERR_GENERAL);
209                         goto leave;
210                       }
211                     else if( putc( c, fp ) == EOF )
212                       {
213                         if (ferror (fp))
214                           rc = gpg_error_from_errno (errno);
215                         else
216                           rc = gpg_error (GPG_ERR_EOF);
217                         log_error ("error writing to `%s': %s\n",
218                                    fname, strerror(errno) );
219                         goto leave;
220                       }
221                   }
222             }
223         }
224         else { /* binary mode */
225             byte *buffer = xmalloc( 32768 );
226             while( pt->len ) {
227                 int len = pt->len > 32768 ? 32768 : pt->len;
228                 len = iobuf_read( pt->buf, buffer, len );
229                 if( len == -1 ) {
230                     rc = gpg_error_from_errno (errno);
231                     log_error ("problem reading source (%u bytes remaining)\n",
232                                (unsigned)pt->len);
233                     xfree( buffer );
234                     goto leave;
235                 }
236                 if( mfx->md )
237                     gcry_md_write ( mfx->md, buffer, len );
238                 if( fp )
239                   {
240                     if(opt.max_output && (count+=len)>opt.max_output)
241                       {
242                         log_error("Error writing to `%s': %s\n",
243                                   fname,"exceeded --max-output limit\n");
244                         rc = G10ERR_WRITE_FILE;
245                         xfree( buffer );
246                         goto leave;
247                       }
248                     else if( fwrite( buffer, 1, len, fp ) != len )
249                       {
250                         log_error("Error writing to `%s': %s\n",
251                                   fname, strerror(errno) );
252                         rc = G10ERR_WRITE_FILE;
253                         xfree( buffer );
254                         goto leave;
255                       }
256                   }
257                 pt->len -= len;
258             }
259             xfree( buffer );
260         }
261     }
262     else if( !clearsig ) {
263         if( convert ) { /* text mode */
264             while( (c = iobuf_get(pt->buf)) != -1 ) {
265                 if( mfx->md )
266                     gcry_md_putc (mfx->md, c );
267 #ifndef HAVE_DOSISH_SYSTEM
268                 if( convert && c == '\r' )
269                     continue; /* fixme: this hack might be too simple */
270 #endif
271                 if( fp )
272                   {
273                     if(opt.max_output && (++count)>opt.max_output)
274                       {
275                         log_error("Error writing to `%s': %s\n",
276                                   fname,"exceeded --max-output limit\n");
277                         rc = G10ERR_WRITE_FILE;
278                         goto leave;
279                       }
280                     else if( putc( c, fp ) == EOF )
281                       {
282                         log_error("Error writing to `%s': %s\n",
283                                   fname, strerror(errno) );
284                         rc = G10ERR_WRITE_FILE;
285                         goto leave;
286                       }
287                   }
288             }
289         }
290         else { /* binary mode */
291             byte *buffer = xmalloc( 32768 );
292             int eof;
293             for( eof=0; !eof; ) {
294                 /* Why do we check for len < 32768:
295                  * If we won't, we would practically read 2 EOFs but
296                  * the first one has already popped the block_filter
297                  * off and therefore we don't catch the boundary.
298                  * So, always assume EOF if iobuf_read returns less bytes
299                  * then requested */
300                 int len = iobuf_read( pt->buf, buffer, 32768 );
301                 if( len == -1 )
302                     break;
303                 if( len < 32768 )
304                     eof = 1;
305                 if( mfx->md )
306                     gcry_md_write ( mfx->md, buffer, len );
307                 if( fp )
308                   {
309                     if(opt.max_output && (count+=len)>opt.max_output)
310                       {
311                         log_error("Error writing to `%s': %s\n",
312                                   fname,"exceeded --max-output limit\n");
313                         rc = G10ERR_WRITE_FILE;
314                         xfree( buffer );
315                         goto leave;
316                       }
317                     else if( fwrite( buffer, 1, len, fp ) != len ) {
318                       log_error("Error writing to `%s': %s\n",
319                                 fname, strerror(errno) );
320                       rc = G10ERR_WRITE_FILE;
321                       xfree( buffer );
322                       goto leave;
323                     }
324                   }
325             }
326             xfree( buffer );
327         }
328         pt->buf = NULL;
329     }
330     else {  /* clear text signature - don't hash the last cr,lf  */
331         int state = 0;
332
333         while( (c = iobuf_get(pt->buf)) != -1 ) {
334             if( fp )
335               {
336                 if(opt.max_output && (++count)>opt.max_output)
337                   {
338                     log_error("Error writing to `%s': %s\n",
339                               fname,"exceeded --max-output limit\n");
340                     rc = G10ERR_WRITE_FILE;
341                     goto leave;
342                   }
343                 else if( putc( c, fp ) == EOF )
344                   {
345                     log_error("Error writing to `%s': %s\n",
346                               fname, strerror(errno) );
347                     rc = G10ERR_WRITE_FILE;
348                     goto leave;
349                   }
350               }
351             if( !mfx->md )
352                 continue;
353             if( state == 2 ) {
354                 md_putc(mfx->md, '\r' );
355                 md_putc(mfx->md, '\n' );
356                 state = 0;
357             }
358             if( !state ) {
359                 if( c == '\r'  )
360                     state = 1;
361                 else if( c == '\n'  )
362                     state = 2;
363                 else
364                     md_putc(mfx->md, c );
365             }
366             else if( state == 1 ) {
367                 if( c == '\n'  )
368                     state = 2;
369                 else {
370                     md_putc(mfx->md, '\r' );
371                     if( c == '\r'  )
372                         state = 1;
373                     else {
374                         state = 0;
375                         md_putc(mfx->md, c );
376                     }
377                 }
378             }
379         }
380         pt->buf = NULL;
381     }
382
383     if( fp && fp != stdout && fclose(fp) ) {
384         log_error("Error closing `%s': %s\n", fname, strerror(errno) );
385         fp = NULL;
386         rc = G10ERR_WRITE_FILE;
387         goto leave;
388     }
389     fp = NULL;
390
391   leave:
392     if( fp && fp != stdout )
393         fclose(fp);
394     xfree(fname);
395     return rc;
396 }
397
398 static void
399 do_hash( gcry_md_hd_t md, gcry_md_hd_t md2, IOBUF fp, int textmode )
400 {
401     text_filter_context_t tfx;
402     int c;
403
404     if( textmode ) {
405         memset( &tfx, 0, sizeof tfx);
406         iobuf_push_filter( fp, text_filter, &tfx );
407     }
408     if( md2 ) { /* work around a strange behaviour in pgp2 */
409         /* It seems that at least PGP5 converts a single CR to a CR,LF too */
410         int lc = -1;
411         while( (c = iobuf_get(fp)) != -1 ) {
412             if( c == '\n' && lc == '\r' )
413                 gcry_md_putc (md2, c);
414             else if( c == '\n' ) {
415                 gcry_md_putc (md2, '\r');
416                 gcry_md_putc (md2, c);
417             }
418             else if( c != '\n' && lc == '\r' ) {
419                 gcry_md_putc (md2, '\n');
420                 gcry_md_putc (md2, c);
421             }
422             else
423                 gcry_md_putc (md2, c);
424
425             if( md )
426                 gcry_md_putc (md, c );
427             lc = c;
428         }
429     }
430     else {
431         while( (c = iobuf_get(fp)) != -1 ) {
432             if( md )
433                 gcry_md_putc (md, c );
434         }
435     }
436 }
437
438
439 /****************
440  * Ask for the detached datafile and calculate the digest from it.
441  * INFILE is the name of the input file.
442  */
443 int
444 ask_for_detached_datafile (gcry_md_hd_t md, gcry_md_hd_t md2,
445                            const char *inname, int textmode )
446 {
447     progress_filter_context_t pfx;
448     char *answer = NULL;
449     IOBUF fp;
450     int rc = 0;
451
452     fp = open_sigfile( inname, &pfx ); /* open default file */
453
454     if( !fp && !opt.batch ) {
455         int any=0;
456         tty_printf(_("Detached signature.\n"));
457         do {
458             char *name;
459             xfree(answer);
460             tty_enable_completion(NULL);
461             name = cpr_get("detached_signature.filename",
462                            _("Please enter name of data file: "));
463             tty_disable_completion();
464             cpr_kill_prompt();
465             answer=make_filename(name,(void *)NULL);
466             xfree(name);
467
468             if( any && !*answer ) {
469                 rc = gpg_error (GPG_ERR_GENERAL); /*G10ERR_READ_FILE*/
470                 goto leave;
471             }
472             fp = iobuf_open(answer);
473             if (fp && is_secured_file (iobuf_get_fd (fp)))
474               {
475                 iobuf_close (fp);
476                 fp = NULL;
477                 errno = EPERM;
478               }
479             if( !fp && errno == ENOENT ) {
480                 tty_printf("No such file, try again or hit enter to quit.\n");
481                 any++;
482             }
483             else if( !fp )
484               {
485                 rc = gpg_error_from_errno (errno);
486                 log_error(_("can't open `%s': %s\n"), answer, strerror(errno));
487                 goto leave;
488               }
489         } while( !fp );
490     }
491
492     if( !fp ) {
493         if( opt.verbose )
494             log_info(_("reading stdin ...\n"));
495         fp = iobuf_open( NULL );
496         assert(fp);
497     }
498     do_hash( md, md2, fp, textmode );
499     iobuf_close(fp);
500
501   leave:
502     xfree(answer);
503     return rc;
504 }
505
506
507
508 /****************
509  * Hash the given files and append the hash to hash context md.
510  * If FILES is NULL, hash stdin.
511  */
512 int
513 hash_datafiles( gcry_md_hd_t md, gcry_md_hd_t md2, STRLIST files,
514                 const char *sigfilename, int textmode )
515 {
516     progress_filter_context_t pfx;
517     IOBUF fp;
518     STRLIST sl;
519
520     if( !files ) {
521         /* check whether we can open the signed material */
522         fp = open_sigfile( sigfilename, &pfx );
523         if( fp ) {
524             do_hash( md, md2, fp, textmode );
525             iobuf_close(fp);
526             return 0;
527         }
528         log_error (_("no signed data\n"));
529         return gpg_error (GPG_ERR_NO_DATA);
530     }
531
532
533     for (sl=files; sl; sl = sl->next ) {
534         fp = iobuf_open( sl->d );
535         if (fp && is_secured_file (iobuf_get_fd (fp)))
536           {
537             iobuf_close (fp);
538             fp = NULL;
539             errno = EPERM;
540           }
541         if( !fp ) {
542             int rc = gpg_error_from_errno (errno);
543             log_error(_("can't open signed data `%s'\n"),
544                                                 print_fname_stdin(sl->d));
545             return rc;
546         }
547         handle_progress (&pfx, fp, sl->d);
548         do_hash( md, md2, fp, textmode );
549         iobuf_close(fp);
550     }
551
552     return 0;
553 }