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