8d7b2a97ea88b3e0e2c9b4f3f6a1246c695d2641
[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 "i18n.h"
34
35
36 /****************
37  * Defer the last CR,LF
38  */
39 static void
40 special_md_putc( MD_HANDLE md, int c, int *state )
41 {
42     if( c == -1 ) { /* flush */
43         if( *state == 1 ) {
44             md_putc(md, '\r');
45         }
46         *state = 0;
47         return;
48     }
49   again:
50     switch( *state ) {
51       case 0:
52         if( c == '\r' )
53             *state = 1;
54         else
55             md_putc(md, c );
56         break;
57       case 1:
58         if( c == '\n' )
59             *state = 2;
60         else {
61             md_putc(md, '\r');
62             *state = 0;
63             goto again;
64         }
65         break;
66       case 2:
67         md_putc(md, '\r');
68         md_putc(md, '\n');
69         *state = 0;
70         goto again;
71       default: BUG();
72     }
73 }
74
75
76 /****************
77  * Handle a plaintext packet.  If MFX is not NULL, update the MDs
78  * Note: we should use the filter stuff here, but we have to add some
79  *       easy mimic to set a read limit, so we calculate only the
80  *       bytes from the plaintext.
81  */
82 int
83 handle_plaintext( PKT_plaintext *pt, md_filter_context_t *mfx,
84                   int nooutput, int clearsig )
85 {
86     char *fname = NULL;
87     FILE *fp = NULL;
88     int rc = 0;
89     int c;
90     int convert = pt->mode == 't';
91     int special_state = 0;
92
93     /* create the filename as C string */
94     if( nooutput )
95         ;
96     else if( opt.outfile ) {
97         fname = m_alloc( strlen( opt.outfile ) + 1);
98         strcpy(fname, opt.outfile );
99     }
100     else {
101         fname = m_alloc( pt->namelen +1 );
102         memcpy( fname, pt->name, pt->namelen );
103         fname[pt->namelen] = 0;
104     }
105
106     if( nooutput )
107         ;
108     else if( !*fname || (*fname=='-' && !fname[1])) {
109         /* no filename or "-" given; write to stdout */
110         fp = stdout;
111     }
112     else if( overwrite_filep( fname ) )
113         goto leave;
114
115     if( fp || nooutput )
116         ;
117     else if( !(fp = fopen(fname,"wb")) ) {
118         log_error("Error creating '%s': %s\n", fname, strerror(errno) );
119         rc = G10ERR_WRITE_FILE;
120         goto leave;
121     }
122
123     if( pt->len ) {
124         for( ; pt->len; pt->len-- ) {
125             if( (c = iobuf_get(pt->buf)) == -1 ) {
126                 log_error("Problem reading source (%u bytes remaining)\n",
127                                                          (unsigned)pt->len);
128                 rc = G10ERR_READ_FILE;
129                 goto leave;
130             }
131             if( mfx->md ) {
132                 if( convert && clearsig )
133                     special_md_putc(mfx->md, c, &special_state );
134                 else
135                     md_putc(mfx->md, c );
136             }
137             if( convert && c == '\r' )
138                 continue; /* fixme: this hack might be too simple */
139             if( fp ) {
140                 if( putc( c, fp ) == EOF ) {
141                     log_error("Error writing to '%s': %s\n",
142                                             fname, strerror(errno) );
143                     rc = G10ERR_WRITE_FILE;
144                     goto leave;
145                 }
146             }
147         }
148     }
149     else {
150         while( (c = iobuf_get(pt->buf)) != -1 ) {
151             if( mfx->md ) {
152                 if( convert && clearsig )
153                     special_md_putc(mfx->md, c, &special_state  );
154                 else
155                     md_putc(mfx->md, c );
156             }
157             if( convert && c == '\r' )
158                 continue; /* fixme: this hack might be too simple */
159             if( fp ) {
160                 if( putc( c, fp ) == EOF ) {
161                     log_error("Error writing to '%s': %s\n",
162                                                 fname, strerror(errno) );
163                     rc = G10ERR_WRITE_FILE;
164                     goto leave;
165                 }
166             }
167         }
168         iobuf_clear_eof(pt->buf);
169     }
170     if( mfx->md && convert && clearsig )
171         special_md_putc(mfx->md, -1, &special_state  ); /* flush */
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 = tty_get("Please enter name of data file: ");
208             tty_kill_prompt();
209             if( any && !*answer ) {
210                 rc = G10ERR_READ_FILE;
211                 goto leave;
212             }
213             fp = iobuf_open(answer);
214             if( !fp && errno == ENOENT ) {
215                 tty_printf("No such file, try again or hit enter to quit.\n");
216                 any++;
217             }
218             else if( !fp ) {
219                 log_error("can't open '%s': %s\n", answer, strerror(errno) );
220                 rc = G10ERR_READ_FILE;
221                 goto leave;
222             }
223         } while( !fp );
224     }
225
226     if( !fp ) {
227         if( opt.verbose )
228             log_info("reading stdin ...\n");
229         while( (c = getchar()) != EOF ) {
230             if( mfx->md )
231                 md_putc(mfx->md, c );
232         }
233     }
234     else {
235         while( (c = iobuf_get(fp)) != -1 ) {
236             if( mfx->md )
237                 md_putc(mfx->md, c );
238         }
239         iobuf_close(fp);
240     }
241
242   leave:
243     m_free(answer);
244     return rc;
245 }
246
247
248 /****************
249  * Hash the given files and append the hash to hash context md.
250  * If FILES is NULL, hash stdin.
251  */
252 int
253 hash_datafiles( MD_HANDLE md, STRLIST files, int textmode )
254 {
255     IOBUF fp;
256     STRLIST sl=NULL;
257     text_filter_context_t tfx;
258     int c;
259
260     if( !files )
261         add_to_strlist( &sl, "-");
262     else
263         sl = files;
264
265     for( ; sl; sl = sl->next ) {
266         fp = iobuf_open( sl->d );
267         if( !fp ) {
268             log_error(_("can't open signed data '%s'\n"),
269                                                 print_fname_stdin(sl->d));
270             if( !files )
271                 free_strlist(sl);
272             return G10ERR_OPEN_FILE;
273         }
274         if( textmode ) {
275             memset( &tfx, 0, sizeof tfx);
276             iobuf_push_filter( fp, text_filter, &tfx );
277         }
278         while( (c = iobuf_get(fp)) != -1 )
279             md_putc(md, c );
280         iobuf_close(fp);
281     }
282
283     if( !files )
284         free_strlist(sl);
285     return 0;
286 }
287
288