See ChangeLog: Tue Dec 29 14:41:47 CET 1998 Werner Koch
[gnupg.git] / g10 / plaintext.c
1 /* plaintext.c -  process an plaintext packet
2  *      Copyright (C) 1998 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 "util.h"
27 #include "memory.h"
28 #include "options.h"
29 #include "packet.h"
30 #include "ttyio.h"
31 #include "filter.h"
32 #include "main.h"
33 #include "status.h"
34 #include "i18n.h"
35
36
37 /****************
38  * Defer the last CR,LF
39  */
40 static void
41 special_md_putc( MD_HANDLE md, int c, int *state )
42 {
43     if( c == -1 ) { /* flush */
44         if( *state == 1 ) {
45             md_putc(md, '\r');
46         }
47
48         *state = 0;
49         return;
50     }
51   again:
52     switch( *state ) {
53       case 0:
54         if( c == '\r' )
55             *state = 1;
56         else
57             md_putc(md, c );
58         break;
59       case 1:
60         if( c == '\n' )
61             *state = 2;
62         else {
63             md_putc(md, '\r');
64             *state = 0;
65             goto again;
66         }
67         break;
68       case 2:
69         md_putc(md, '\r');
70         md_putc(md, '\n');
71         *state = 0;
72         goto again;
73       default: BUG();
74     }
75 }
76
77
78 /****************
79  * Handle a plaintext packet.  If MFX is not NULL, update the MDs
80  * Note: we should use the filter stuff here, but we have to add some
81  *       easy mimic to set a read limit, so we calculate only the
82  *       bytes from the plaintext.
83  */
84 int
85 handle_plaintext( PKT_plaintext *pt, md_filter_context_t *mfx,
86                   int nooutput, int clearsig )
87 {
88     char *fname = NULL;
89     FILE *fp = NULL;
90     int rc = 0;
91     int c;
92     int convert = pt->mode == 't';
93     int special_state = 0;
94
95     /* create the filename as C string */
96     if( nooutput )
97         ;
98     else if( opt.outfile ) {
99         fname = m_alloc( strlen( opt.outfile ) + 1);
100         strcpy(fname, opt.outfile );
101     }
102     else if( pt->namelen == 8 && !memcmp( pt->name, "_CONSOLE", 8 ) ) {
103         log_info(_("data not saved; use option \"--output\" to save it\n"));
104         nooutput = 1;
105     }
106     else {
107         fname = m_alloc( pt->namelen +1 );
108         memcpy( fname, pt->name, pt->namelen );
109         fname[pt->namelen] = 0;
110     }
111
112     if( nooutput )
113         ;
114     else if( !*fname || (*fname=='-' && !fname[1])) {
115         /* no filename or "-" given; write to stdout */
116         fp = stdout;
117     }
118     else if( !overwrite_filep( fname ) ) {
119         rc = G10ERR_CREATE_FILE;
120         goto leave;
121     }
122
123     if( fp || nooutput )
124         ;
125     else if( !(fp = fopen(fname,"wb")) ) {
126         log_error("Error creating `%s': %s\n", fname, strerror(errno) );
127         rc = G10ERR_CREATE_FILE;
128         goto leave;
129     }
130
131     if( pt->len ) {
132         for( ; pt->len; pt->len-- ) {
133             if( (c = iobuf_get(pt->buf)) == -1 ) {
134                 log_error("Problem reading source (%u bytes remaining)\n",
135                                                          (unsigned)pt->len);
136                 rc = G10ERR_READ_FILE;
137                 goto leave;
138             }
139             if( mfx->md ) {
140                 if( convert && clearsig )
141                     special_md_putc(mfx->md, c, &special_state );
142                 else
143                     md_putc(mfx->md, c );
144             }
145             if( convert && !clearsig && c == '\r' )
146                 continue; /* fixme: this hack might be too simple */
147             if( fp ) {
148                 if( putc( c, fp ) == EOF ) {
149                     log_error("Error writing to `%s': %s\n",
150                                             fname, strerror(errno) );
151                     rc = G10ERR_WRITE_FILE;
152                     goto leave;
153                 }
154             }
155         }
156     }
157     else {
158         while( (c = iobuf_get(pt->buf)) != -1 ) {
159             if( mfx->md ) {
160                 if( convert && clearsig )
161                     special_md_putc(mfx->md, c, &special_state  );
162                 else
163                     md_putc(mfx->md, c );
164             }
165             if( convert && !clearsig && c == '\r' )
166                 continue; /* fixme: this hack might be too simple */
167             if( fp ) {
168                 if( putc( c, fp ) == EOF ) {
169                     log_error("Error writing to `%s': %s\n",
170                                                 fname, strerror(errno) );
171                     rc = G10ERR_WRITE_FILE;
172                     goto leave;
173                 }
174             }
175         }
176         iobuf_clear_eof(pt->buf);
177     }
178     if( mfx->md && convert && clearsig )
179         special_md_putc(mfx->md, -1, &special_state  ); /* flush */
180
181     if( fp && fp != stdout && fclose(fp) ) {
182         log_error("Error closing `%s': %s\n", fname, strerror(errno) );
183         fp = NULL;
184         rc = G10ERR_WRITE_FILE;
185         goto leave;
186     }
187     fp = NULL;
188
189   leave:
190     if( fp && fp != stdout )
191         fclose(fp);
192     m_free(fname);
193     return rc;
194 }
195
196
197 /****************
198  * Ask for the detached datafile and calculate the digest from it.
199  * INFILE is the name of the input file.
200  */
201 int
202 ask_for_detached_datafile( md_filter_context_t *mfx, const char *inname )
203 {
204     char *answer = NULL;
205     IOBUF fp;
206     int rc = 0;
207     int c;
208
209     fp = open_sigfile( inname ); /* open default file */
210     if( !fp && !opt.batch ) {
211         int any=0;
212         tty_printf("Detached signature.\n");
213         do {
214             m_free(answer);
215             answer = cpr_get("detached_signature.filename",
216                            _("Please enter name of data file: "));
217             cpr_kill_prompt();
218             if( any && !*answer ) {
219                 rc = G10ERR_READ_FILE;
220                 goto leave;
221             }
222             fp = iobuf_open(answer);
223             if( !fp && errno == ENOENT ) {
224                 tty_printf("No such file, try again or hit enter to quit.\n");
225                 any++;
226             }
227             else if( !fp ) {
228                 log_error("can't open `%s': %s\n", answer, strerror(errno) );
229                 rc = G10ERR_READ_FILE;
230                 goto leave;
231             }
232         } while( !fp );
233     }
234
235     if( !fp ) {
236         if( opt.verbose )
237             log_info(_("reading stdin ...\n"));
238         while( (c = getchar()) != EOF ) {
239             if( mfx->md )
240                 md_putc(mfx->md, c );
241         }
242     }
243     else {
244         while( (c = iobuf_get(fp)) != -1 ) {
245             if( mfx->md )
246                 md_putc(mfx->md, c );
247         }
248         iobuf_close(fp);
249     }
250
251   leave:
252     m_free(answer);
253     return rc;
254 }
255
256
257 static void
258 do_hash( MD_HANDLE md, IOBUF fp, int textmode )
259 {
260     text_filter_context_t tfx;
261     int c;
262
263     if( textmode ) {
264         memset( &tfx, 0, sizeof tfx);
265         iobuf_push_filter( fp, text_filter, &tfx );
266     }
267     while( (c = iobuf_get(fp)) != -1 )
268         md_putc(md, c );
269 }
270
271
272 /****************
273  * Hash the given files and append the hash to hash context md.
274  * If FILES is NULL, hash stdin.
275  */
276 int
277 hash_datafiles( MD_HANDLE md, STRLIST files,
278                 const char *sigfilename, int textmode )
279 {
280     IOBUF fp;
281     STRLIST sl=NULL;
282
283     if( !files ) {
284         /* check whether we can opne the signed material */
285         fp = open_sigfile( sigfilename );
286         if( fp ) {
287             do_hash( md, fp, textmode );
288             iobuf_close(fp);
289             return 0;
290         }
291         /* no we can't (no sigfile) - read signed stuff from stdin */
292         add_to_strlist( &sl, "-");
293     }
294     else
295         sl = files;
296
297     for( ; sl; sl = sl->next ) {
298         fp = iobuf_open( sl->d );
299         if( !fp ) {
300             log_error(_("can't open signed data `%s'\n"),
301                                                 print_fname_stdin(sl->d));
302             if( !files )
303                 free_strlist(sl);
304             return G10ERR_OPEN_FILE;
305         }
306         do_hash( md, fp, textmode );
307         iobuf_close(fp);
308     }
309
310     if( !files )
311         free_strlist(sl);
312     return 0;
313 }
314
315