See ChangeLog: Wed Jan 20 18:59:49 CET 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         for( ; pt->len; pt->len-- ) {
94             if( (c = iobuf_get(pt->buf)) == -1 ) {
95                 log_error("Problem reading source (%u bytes remaining)\n",
96                                                          (unsigned)pt->len);
97                 rc = G10ERR_READ_FILE;
98                 goto leave;
99             }
100             if( mfx->md )
101                 md_putc(mfx->md, c );
102             if( convert && c == '\r' )
103                 continue; /* fixme: this hack might be too simple */
104             if( fp ) {
105                 if( putc( c, fp ) == EOF ) {
106                     log_error("Error writing to `%s': %s\n",
107                                             fname, strerror(errno) );
108                     rc = G10ERR_WRITE_FILE;
109                     goto leave;
110                 }
111             }
112         }
113     }
114     else if( !clearsig ) {
115         while( (c = iobuf_get(pt->buf)) != -1 ) {
116             if( mfx->md )
117                 md_putc(mfx->md, c );
118             if( convert && c == '\r' )
119                 continue; /* fixme: this hack might be too simple */
120             if( fp ) {
121                 if( putc( c, fp ) == EOF ) {
122                     log_error("Error writing to `%s': %s\n",
123                                                 fname, strerror(errno) );
124                     rc = G10ERR_WRITE_FILE;
125                     goto leave;
126                 }
127             }
128         }
129         pt->buf = NULL;
130     }
131     else {  /* clear text signature - don't hash the last cr,lf  */
132         int state = 0;
133
134         while( (c = iobuf_get(pt->buf)) != -1 ) {
135             if( fp ) {
136                 if( putc( c, fp ) == EOF ) {
137                     log_error("Error writing to `%s': %s\n",
138                                                 fname, strerror(errno) );
139                     rc = G10ERR_WRITE_FILE;
140                     goto leave;
141                 }
142             }
143             if( !mfx->md )
144                 continue;
145             if( state == 2 ) {
146                 md_putc(mfx->md, '\r' );
147                 md_putc(mfx->md, '\n' );
148                 state = 0;
149             }
150             if( !state ) {
151                 if( c == '\r'  )
152                     state = 1;
153                 else
154                     md_putc(mfx->md, c );
155             }
156             else if( state == 1 ) {
157                 if( c == '\n'  )
158                     state = 2;
159                 else {
160                     md_putc(mfx->md, '\r' );
161                     if( c == '\r'  )
162                         state = 1;
163                     else {
164                         state = 0;
165                         md_putc(mfx->md, c );
166                     }
167                 }
168             }
169         }
170         pt->buf = NULL;
171     }
172
173     if( fp && fp != stdout && fclose(fp) ) {
174         log_error("Error closing `%s': %s\n", fname, strerror(errno) );
175         fp = NULL;
176         rc = G10ERR_WRITE_FILE;
177         goto leave;
178     }
179     fp = NULL;
180
181   leave:
182     if( fp && fp != stdout )
183         fclose(fp);
184     m_free(fname);
185     return rc;
186 }
187
188
189 /****************
190  * Ask for the detached datafile and calculate the digest from it.
191  * INFILE is the name of the input file.
192  */
193 int
194 ask_for_detached_datafile( md_filter_context_t *mfx, const char *inname )
195 {
196     char *answer = NULL;
197     IOBUF fp;
198     int rc = 0;
199     int c;
200
201     fp = open_sigfile( inname ); /* open default file */
202     if( !fp && !opt.batch ) {
203         int any=0;
204         tty_printf("Detached signature.\n");
205         do {
206             m_free(answer);
207             answer = cpr_get("detached_signature.filename",
208                            _("Please enter name of data file: "));
209             cpr_kill_prompt();
210             if( any && !*answer ) {
211                 rc = G10ERR_READ_FILE;
212                 goto leave;
213             }
214             fp = iobuf_open(answer);
215             if( !fp && errno == ENOENT ) {
216                 tty_printf("No such file, try again or hit enter to quit.\n");
217                 any++;
218             }
219             else if( !fp ) {
220                 log_error("can't open `%s': %s\n", answer, strerror(errno) );
221                 rc = G10ERR_READ_FILE;
222                 goto leave;
223             }
224         } while( !fp );
225     }
226
227     if( !fp ) {
228         if( opt.verbose )
229             log_info(_("reading stdin ...\n"));
230         while( (c = getchar()) != EOF ) {
231             if( mfx->md )
232                 md_putc(mfx->md, c );
233         }
234     }
235     else {
236         while( (c = iobuf_get(fp)) != -1 ) {
237             if( mfx->md )
238                 md_putc(mfx->md, c );
239         }
240         iobuf_close(fp);
241     }
242
243   leave:
244     m_free(answer);
245     return rc;
246 }
247
248
249 static void
250 do_hash( MD_HANDLE md, IOBUF fp, int textmode )
251 {
252     text_filter_context_t tfx;
253     int c;
254
255     if( textmode ) {
256         memset( &tfx, 0, sizeof tfx);
257         iobuf_push_filter( fp, text_filter, &tfx );
258     }
259     while( (c = iobuf_get(fp)) != -1 )
260         md_putc(md, c );
261 }
262
263
264 /****************
265  * Hash the given files and append the hash to hash context md.
266  * If FILES is NULL, hash stdin.
267  */
268 int
269 hash_datafiles( MD_HANDLE md, STRLIST files,
270                 const char *sigfilename, int textmode )
271 {
272     IOBUF fp;
273     STRLIST sl=NULL;
274
275     if( !files ) {
276         /* check whether we can open the signed material */
277         fp = open_sigfile( sigfilename );
278         if( fp ) {
279             do_hash( md, fp, textmode );
280             iobuf_close(fp);
281             return 0;
282         }
283         /* no we can't (no sigfile) - read signed stuff from stdin */
284         add_to_strlist( &sl, "-");
285     }
286     else
287         sl = files;
288
289     for( ; sl; sl = sl->next ) {
290         fp = iobuf_open( sl->d );
291         if( !fp ) {
292             log_error(_("can't open signed data `%s'\n"),
293                                                 print_fname_stdin(sl->d));
294             if( !files )
295                 free_strlist(sl);
296             return G10ERR_OPEN_FILE;
297         }
298         do_hash( md, fp, textmode );
299         iobuf_close(fp);
300     }
301
302     if( !files )
303         free_strlist(sl);
304     return 0;
305 }
306
307