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