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