See ChangeLog: Fri Jul 14 19:38:23 CEST 2000 Werner Koch
[gnupg.git] / g10 / plaintext.c
1 /* plaintext.c -  process an plaintext packet
2  *      Copyright (C) 1998, 1999, 2000 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 2 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, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #include <config.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <errno.h>
26 #include <assert.h>
27 #include "util.h"
28 #include <gcrypt.h>
29 #include "options.h"
30 #include "packet.h"
31 #include "ttyio.h"
32 #include "filter.h"
33 #include "main.h"
34 #include "status.h"
35 #include "i18n.h"
36
37
38
39 /****************
40  * Handle a plaintext packet.  If MFX is not NULL, update the MDs
41  * Note: we should use the filter stuff here, but we have to add some
42  *       easy mimic to set a read limit, so we calculate only the
43  *       bytes from the plaintext.
44  */
45 int
46 handle_plaintext( PKT_plaintext *pt, md_filter_context_t *mfx,
47                   int nooutput, int clearsig )
48 {
49     char *fname = NULL;
50     FILE *fp = NULL;
51     int rc = 0;
52     int c;
53     int convert = pt->mode == 't';
54
55     /* create the filename as C string */
56     if( nooutput )
57         ;
58     else if( opt.outfile ) {
59         fname = gcry_xmalloc( strlen( opt.outfile ) + 1);
60         strcpy(fname, opt.outfile );
61     }
62     else if( pt->namelen == 8 && !memcmp( pt->name, "_CONSOLE", 8 ) ) {
63         log_info(_("data not saved; use option \"--output\" to save it\n"));
64         nooutput = 1;
65     }
66     else if( !opt.use_embedded_filename ) {
67         fname = make_outfile_name( iobuf_get_real_fname(pt->buf) );
68         if( !fname )
69             fname = ask_outfile_name( pt->name, pt->namelen );
70         if( !fname ) {
71             rc = GPGERR_CREATE_FILE;
72             goto leave;
73         }
74     }
75     else {
76         fname = make_printable_string( pt->name, pt->namelen, 0 );
77     }
78
79     if( nooutput )
80         ;
81     else if( !*fname || (*fname=='-' && !fname[1])) {
82         /* no filename or "-" given; write to stdout */
83         fp = stdout;
84     }
85     else if( !overwrite_filep( fname ) ) {
86         rc = GPGERR_CREATE_FILE;
87         goto leave;
88     }
89
90     if( fp || nooutput )
91         ;
92     else if( !(fp = fopen(fname,"wb")) ) {
93         log_error("Error creating `%s': %s\n", fname, strerror(errno) );
94         rc = GPGERR_CREATE_FILE;
95         goto leave;
96     }
97
98     if( pt->len ) {
99         assert( !clearsig );
100         if( convert ) { /* text mode */
101             for( ; pt->len; pt->len-- ) {
102                 if( (c = iobuf_get(pt->buf)) == -1 ) {
103                     log_error("Problem reading source (%u bytes remaining)\n",
104                               (unsigned)pt->len);
105                     rc = GPGERR_READ_FILE;
106                     goto leave;
107                 }
108                 if( mfx->md )
109                     gcry_md_putc(mfx->md, c );
110               #ifndef HAVE_DOSISH_SYSTEM
111                 if( c == '\r' )  /* convert to native line ending */
112                     continue;    /* fixme: this hack might be too simple */
113               #endif
114                 if( fp ) {
115                     if( putc( c, fp ) == EOF ) {
116                         log_error("Error writing to `%s': %s\n",
117                                   fname, strerror(errno) );
118                         rc = GPGERR_WRITE_FILE;
119                         goto leave;
120                     }
121                 }
122             }
123         }
124         else { /* binary mode */
125             byte *buffer = gcry_xmalloc( 32768 );
126             while( pt->len ) {
127                 int len = pt->len > 32768 ? 32768 : pt->len;
128                 len = iobuf_read( pt->buf, buffer, len );
129                 if( len == -1 ) {
130                     log_error("Problem reading source (%u bytes remaining)\n",
131                               (unsigned)pt->len);
132                     rc = GPGERR_READ_FILE;
133                     gcry_free( buffer );
134                     goto leave;
135                 }
136                 if( mfx->md )
137                     gcry_md_write( mfx->md, buffer, len );
138                 if( fp ) {
139                     if( fwrite( buffer, 1, len, fp ) != len ) {
140                         log_error("Error writing to `%s': %s\n",
141                                   fname, strerror(errno) );
142                         rc = GPGERR_WRITE_FILE;
143                         gcry_free( buffer );
144                         goto leave;
145                     }
146                 }
147                 pt->len -= len;
148             }
149             gcry_free( buffer );
150         }
151     }
152     else if( !clearsig ) {
153         if( convert ) { /* text mode */
154             while( (c = iobuf_get(pt->buf)) != -1 ) {
155                 if( mfx->md )
156                     gcry_md_putc(mfx->md, c );
157               #ifndef HAVE_DOSISH_SYSTEM
158                 if( convert && c == '\r' )
159                     continue; /* fixme: this hack might be too simple */
160               #endif
161                 if( fp ) {
162                     if( putc( c, fp ) == EOF ) {
163                         log_error("Error writing to `%s': %s\n",
164                                   fname, strerror(errno) );
165                         rc = GPGERR_WRITE_FILE;
166                         goto leave;
167                     }
168                 }
169             }
170         }
171         else { /* binary mode */
172             byte *buffer = gcry_xmalloc( 32768 );
173             int eof;
174             for( eof=0; !eof; ) {
175                 /* Why do we check for len < 32768:
176                  * If we won't, we would practically read 2 EOFs but
177                  * the first one has already popped the block_filter
178                  * off and therefore we don't catch the boundary.
179                  * So, always assume EOF if iobuf_read returns less bytes
180                  * then requested */
181                 int len = iobuf_read( pt->buf, buffer, 32768 );
182                 if( len == -1 )
183                     break;
184                 if( len < 32768 )
185                     eof = 1;
186                 if( mfx->md )
187                     gcry_md_write( mfx->md, buffer, len );
188                 if( fp ) {
189                     if( fwrite( buffer, 1, len, fp ) != len ) {
190                         log_error("Error writing to `%s': %s\n",
191                                   fname, strerror(errno) );
192                         rc = GPGERR_WRITE_FILE;
193                         gcry_free( buffer );
194                         goto leave;
195                     }
196                 }
197             }
198             gcry_free( buffer );
199         }
200         pt->buf = NULL;
201     }
202     else {  /* clear text signature - don't hash the last cr,lf  */
203         int state = 0;
204
205         while( (c = iobuf_get(pt->buf)) != -1 ) {
206             if( fp ) {
207                 if( putc( c, fp ) == EOF ) {
208                     log_error("Error writing to `%s': %s\n",
209                                                 fname, strerror(errno) );
210                     rc = GPGERR_WRITE_FILE;
211                     goto leave;
212                 }
213             }
214             if( !mfx->md )
215                 continue;
216             if( state == 2 ) {
217                 gcry_md_putc(mfx->md, '\r' );
218                 gcry_md_putc(mfx->md, '\n' );
219                 state = 0;
220             }
221             if( !state ) {
222                 if( c == '\r'  )
223                     state = 1;
224                 else if( c == '\n'  )
225                     state = 2;
226                 else
227                     gcry_md_putc(mfx->md, c );
228             }
229             else if( state == 1 ) {
230                 if( c == '\n'  )
231                     state = 2;
232                 else {
233                     gcry_md_putc(mfx->md, '\r' );
234                     if( c == '\r'  )
235                         state = 1;
236                     else {
237                         state = 0;
238                         gcry_md_putc(mfx->md, c );
239                     }
240                 }
241             }
242         }
243         pt->buf = NULL;
244     }
245
246     if( fp && fp != stdout && fclose(fp) ) {
247         log_error("Error closing `%s': %s\n", fname, strerror(errno) );
248         fp = NULL;
249         rc = GPGERR_WRITE_FILE;
250         goto leave;
251     }
252     fp = NULL;
253
254   leave:
255     if( fp && fp != stdout )
256         fclose(fp);
257     gcry_free(fname);
258     return rc;
259 }
260
261 static void
262 do_hash( GCRY_MD_HD md, GCRY_MD_HD md2, IOBUF fp, int textmode )
263 {
264     text_filter_context_t tfx;
265     int c;
266
267     if( textmode ) {
268         memset( &tfx, 0, sizeof tfx);
269         iobuf_push_filter( fp, text_filter, &tfx );
270     }
271     if( md2 ) { /* work around a strange behaviour in pgp2 */
272         /* It seems that at least PGP5 converts a single CR to a CR,LF too */
273         int lc = -1;
274         while( (c = iobuf_get(fp)) != -1 ) {
275             if( c == '\n' && lc == '\r' )
276                 gcry_md_putc(md2, c);
277             else if( c == '\n' ) {
278                 gcry_md_putc(md2, '\r');
279                 gcry_md_putc(md2, c);
280             }
281             else if( c != '\n' && lc == '\r' ) {
282                 gcry_md_putc(md2, '\n');
283                 gcry_md_putc(md2, c);
284             }
285             else
286                 gcry_md_putc(md2, c);
287
288             if( md )
289                 gcry_md_putc(md, c );
290             lc = c;
291         }
292     }
293     else {
294         while( (c = iobuf_get(fp)) != -1 ) {
295             if( md )
296                 gcry_md_putc(md, c );
297         }
298     }
299 }
300
301
302 /****************
303  * Ask for the detached datafile and calculate the digest from it.
304  * INFILE is the name of the input file.
305  */
306 int
307 ask_for_detached_datafile( GCRY_MD_HD md, GCRY_MD_HD md2,
308                            const char *inname, int textmode )
309 {
310     char *answer = NULL;
311     IOBUF fp;
312     int rc = 0;
313
314     fp = open_sigfile( inname ); /* open default file */
315     if( !fp && !opt.batch ) {
316         int any=0;
317         tty_printf(_("Detached signature.\n"));
318         do {
319             gcry_free(answer);
320             answer = cpr_get("detached_signature.filename",
321                            _("Please enter name of data file: "));
322             cpr_kill_prompt();
323             if( any && !*answer ) {
324                 rc = GPGERR_READ_FILE;
325                 goto leave;
326             }
327             fp = iobuf_open(answer);
328             if( !fp && errno == ENOENT ) {
329                 tty_printf("No such file, try again or hit enter to quit.\n");
330                 any++;
331             }
332             else if( !fp ) {
333                 log_error("can't open `%s': %s\n", answer, strerror(errno) );
334                 rc = GPGERR_READ_FILE;
335                 goto leave;
336             }
337         } while( !fp );
338     }
339
340     if( !fp ) {
341         if( opt.verbose )
342             log_info(_("reading stdin ...\n"));
343         fp = iobuf_open( NULL );
344         assert(fp);
345     }
346     do_hash( md, md2, fp, textmode );
347     iobuf_close(fp);
348
349
350   leave:
351     gcry_free(answer);
352     return rc;
353 }
354
355
356
357 /****************
358  * Hash the given files and append the hash to hash context md.
359  * If FILES is NULL, hash stdin.
360  */
361 int
362 hash_datafiles( GCRY_MD_HD md, GCRY_MD_HD md2, STRLIST files,
363                 const char *sigfilename, int textmode )
364 {
365     IOBUF fp;
366     STRLIST sl=NULL;
367
368     if( !files ) {
369         /* check whether we can open the signed material */
370         fp = open_sigfile( sigfilename );
371         if( fp ) {
372             do_hash( md, md2, fp, textmode );
373             iobuf_close(fp);
374             return 0;
375         }
376         /* no we can't (no sigfile) - read signed stuff from stdin */
377         add_to_strlist( &sl, "-");
378     }
379     else
380         sl = files;
381
382     for( ; sl; sl = sl->next ) {
383         fp = iobuf_open( sl->d );
384         if( !fp ) {
385             log_error(_("can't open signed data `%s'\n"),
386                                                 print_fname_stdin(sl->d));
387             if( !files )
388                 free_strlist(sl);
389             return GPGERR_OPEN_FILE;
390         }
391         do_hash( md, md2, fp, textmode );
392         iobuf_close(fp);
393     }
394
395     if( !files )
396         free_strlist(sl);
397     return 0;
398 }
399
400