See ChangeLog: Tue Aug 31 17:20:44 CEST 1999 Werner Koch
[gnupg.git] / g10 / plaintext.c
1 /* plaintext.c -  process an plaintext packet
2  *      Copyright (C) 1998, 1999 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 "memory.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 = m_alloc( 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 = G10ERR_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 = G10ERR_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 = G10ERR_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 = G10ERR_READ_FILE;
106                     goto leave;
107                 }
108                 if( mfx->md )
109                     md_putc(mfx->md, c );
110                 if( c == '\r' )
111                     continue; /* fixme: this hack might be too simple */
112                 if( fp ) {
113                     if( putc( c, fp ) == EOF ) {
114                         log_error("Error writing to `%s': %s\n",
115                                   fname, strerror(errno) );
116                         rc = G10ERR_WRITE_FILE;
117                         goto leave;
118                     }
119                 }
120             }
121         }
122         else { /* binary mode */
123             byte *buffer = m_alloc( 32768 );
124             while( pt->len ) {
125                 int len = pt->len > 32768 ? 32768 : pt->len;
126                 len = iobuf_read( pt->buf, buffer, len );
127                 if( len == -1 ) {
128                     log_error("Problem reading source (%u bytes remaining)\n",
129                               (unsigned)pt->len);
130                     rc = G10ERR_READ_FILE;
131                     m_free( buffer );
132                     goto leave;
133                 }
134                 if( mfx->md )
135                     md_write( mfx->md, buffer, len );
136                 if( fp ) {
137                     if( fwrite( buffer, 1, len, fp ) != len ) {
138                         log_error("Error writing to `%s': %s\n",
139                                   fname, strerror(errno) );
140                         rc = G10ERR_WRITE_FILE;
141                         m_free( buffer );
142                         goto leave;
143                     }
144                 }
145                 pt->len -= len;
146             }
147             m_free( buffer );
148         }
149     }
150     else if( !clearsig ) {
151         if( convert ) { /* text mode */
152             while( (c = iobuf_get(pt->buf)) != -1 ) {
153                 if( mfx->md )
154                     md_putc(mfx->md, c );
155                 if( convert && c == '\r' )
156                     continue; /* fixme: this hack might be too simple */
157                 if( fp ) {
158                     if( putc( c, fp ) == EOF ) {
159                         log_error("Error writing to `%s': %s\n",
160                                   fname, strerror(errno) );
161                         rc = G10ERR_WRITE_FILE;
162                         goto leave;
163                     }
164                 }
165             }
166         }
167         else { /* binary mode */
168             byte *buffer = m_alloc( 32768 );
169             for( ;; ) {
170                 int len = iobuf_read( pt->buf, buffer, 32768 );
171                 if( len == -1 )
172                     break;
173                 if( mfx->md )
174                     md_write( mfx->md, buffer, len );
175                 if( fp ) {
176                     if( fwrite( buffer, 1, len, fp ) != len ) {
177                         log_error("Error writing to `%s': %s\n",
178                                   fname, strerror(errno) );
179                         rc = G10ERR_WRITE_FILE;
180                         m_free( buffer );
181                         goto leave;
182                     }
183                 }
184             }
185             m_free( buffer );
186         }
187         pt->buf = NULL;
188     }
189     else {  /* clear text signature - don't hash the last cr,lf  */
190         int state = 0;
191
192         while( (c = iobuf_get(pt->buf)) != -1 ) {
193             if( fp ) {
194                 if( putc( c, fp ) == EOF ) {
195                     log_error("Error writing to `%s': %s\n",
196                                                 fname, strerror(errno) );
197                     rc = G10ERR_WRITE_FILE;
198                     goto leave;
199                 }
200             }
201             if( !mfx->md )
202                 continue;
203             if( state == 2 ) {
204                 md_putc(mfx->md, '\r' );
205                 md_putc(mfx->md, '\n' );
206                 state = 0;
207             }
208             if( !state ) {
209                 if( c == '\r'  )
210                     state = 1;
211                 else
212                     md_putc(mfx->md, c );
213             }
214             else if( state == 1 ) {
215                 if( c == '\n'  )
216                     state = 2;
217                 else {
218                     md_putc(mfx->md, '\r' );
219                     if( c == '\r'  )
220                         state = 1;
221                     else {
222                         state = 0;
223                         md_putc(mfx->md, c );
224                     }
225                 }
226             }
227         }
228         pt->buf = NULL;
229     }
230
231     if( fp && fp != stdout && fclose(fp) ) {
232         log_error("Error closing `%s': %s\n", fname, strerror(errno) );
233         fp = NULL;
234         rc = G10ERR_WRITE_FILE;
235         goto leave;
236     }
237     fp = NULL;
238
239   leave:
240     if( fp && fp != stdout )
241         fclose(fp);
242     m_free(fname);
243     return rc;
244 }
245
246
247 /****************
248  * Ask for the detached datafile and calculate the digest from it.
249  * INFILE is the name of the input file.
250  */
251 int
252 ask_for_detached_datafile( MD_HANDLE md, MD_HANDLE md2,
253                            const char *inname, int textmode )
254 {
255     char *answer = NULL;
256     IOBUF fp;
257     int rc = 0;
258     int c;
259
260     fp = open_sigfile( inname ); /* open default file */
261     if( !fp && !opt.batch ) {
262         int any=0;
263         tty_printf("Detached signature.\n");
264         do {
265             m_free(answer);
266             answer = cpr_get("detached_signature.filename",
267                            _("Please enter name of data file: "));
268             cpr_kill_prompt();
269             if( any && !*answer ) {
270                 rc = G10ERR_READ_FILE;
271                 goto leave;
272             }
273             fp = iobuf_open(answer);
274             if( !fp && errno == ENOENT ) {
275                 tty_printf("No such file, try again or hit enter to quit.\n");
276                 any++;
277             }
278             else if( !fp ) {
279                 log_error("can't open `%s': %s\n", answer, strerror(errno) );
280                 rc = G10ERR_READ_FILE;
281                 goto leave;
282             }
283         } while( !fp );
284     }
285
286     if( !fp ) {
287         if( opt.verbose )
288             log_info(_("reading stdin ...\n"));
289         fp = iobuf_open( NULL );
290         assert(fp);
291     }
292     do_hash( md, md2, fp, textmode );
293     iobuf_close(fp);
294
295
296   leave:
297     m_free(answer);
298     return rc;
299 }
300
301
302 static void
303 do_hash( MD_HANDLE md, MD_HANDLE md2, IOBUF fp, int textmode )
304 {
305     text_filter_context_t tfx;
306     int c;
307
308     if( textmode ) {
309         memset( &tfx, 0, sizeof tfx);
310         iobuf_push_filter( fp, text_filter, &tfx );
311     }
312     if( md2 ) { /* work around a strange behaviour in pgp2 */
313         /* It seems that at least PGP5 converts a single CR to a CR,LF too */
314         int lc = -1;
315         while( (c = iobuf_get(fp)) != -1 ) {
316             if( c == '\n' && lc == '\r' )
317                 md_putc(md2, c);
318             else if( c == '\n' ) {
319                 md_putc(md2, '\r');
320                 md_putc(md2, c);
321             }
322             else if( c != '\n' && lc == '\r' ) {
323                 md_putc(md2, '\n');
324                 md_putc(md2, c);
325             }
326             else
327                 md_putc(md2, c);
328
329             if( md )
330                 md_putc(md, c );
331             lc = c;
332         }
333     }
334     else {
335         while( (c = iobuf_get(fp)) != -1 ) {
336             if( md )
337                 md_putc(md, c );
338         }
339     }
340 }
341
342
343 /****************
344  * Hash the given files and append the hash to hash context md.
345  * If FILES is NULL, hash stdin.
346  */
347 int
348 hash_datafiles( MD_HANDLE md, MD_HANDLE md2, STRLIST files,
349                 const char *sigfilename, int textmode )
350 {
351     IOBUF fp;
352     STRLIST sl=NULL;
353
354     if( !files ) {
355         /* check whether we can open the signed material */
356         fp = open_sigfile( sigfilename );
357         if( fp ) {
358             do_hash( md, md2, fp, textmode );
359             iobuf_close(fp);
360             return 0;
361         }
362         /* no we can't (no sigfile) - read signed stuff from stdin */
363         add_to_strlist( &sl, "-");
364     }
365     else
366         sl = files;
367
368     for( ; sl; sl = sl->next ) {
369         fp = iobuf_open( sl->d );
370         if( !fp ) {
371             log_error(_("can't open signed data `%s'\n"),
372                                                 print_fname_stdin(sl->d));
373             if( !files )
374                 free_strlist(sl);
375             return G10ERR_OPEN_FILE;
376         }
377         do_hash( md, md2, fp, textmode );
378         iobuf_close(fp);
379     }
380
381     if( !files )
382         free_strlist(sl);
383     return 0;
384 }
385
386