cleaned up RISC OS code and removed lots of unnecessary stuff
[gnupg.git] / g10 / plaintext.c
1 /* plaintext.c -  process plaintext packets
2  * Copyright (C) 1998, 1999, 2000, 2001, 2002 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 #ifdef HAVE_DOSISH_SYSTEM
28 #include <fcntl.h> /* for setmode() */
29 #endif
30
31 #include "util.h"
32 #include "memory.h"
33 #include "options.h"
34 #include "packet.h"
35 #include "ttyio.h"
36 #include "filter.h"
37 #include "main.h"
38 #include "status.h"
39 #include "i18n.h"
40
41
42
43 /****************
44  * Handle a plaintext packet.  If MFX is not NULL, update the MDs
45  * Note: we should use the filter stuff here, but we have to add some
46  *       easy mimic to set a read limit, so we calculate only the
47  *       bytes from the plaintext.
48  */
49 int
50 handle_plaintext( PKT_plaintext *pt, md_filter_context_t *mfx,
51                   int nooutput, int clearsig )
52 {
53     char *fname = NULL;
54     FILE *fp = NULL;
55     int rc = 0;
56     int c;
57     int convert = pt->mode == 't';
58 #ifdef __riscos__
59     int filetype = 0xfff;
60 #endif
61
62     /* create the filename as C string */
63     if( nooutput )
64         ;
65     else if( opt.outfile ) {
66         fname = m_alloc( strlen( opt.outfile ) + 1);
67         strcpy(fname, opt.outfile );
68     }
69     else if( pt->namelen == 8 && !memcmp( pt->name, "_CONSOLE", 8 ) ) {
70         log_info(_("data not saved; use option \"--output\" to save it\n"));
71         nooutput = 1;
72     }
73     else if( !opt.use_embedded_filename ) {
74         fname = make_outfile_name( iobuf_get_real_fname(pt->buf) );
75         if( !fname )
76             fname = ask_outfile_name( pt->name, pt->namelen );
77         if( !fname ) {
78             rc = G10ERR_CREATE_FILE;
79             goto leave;
80         }
81     }
82     else {
83         fname = make_printable_string( pt->name, pt->namelen, 0 );
84     }
85
86     if( nooutput )
87         ;
88     else if( !*fname || (*fname=='-' && !fname[1])) {
89         /* no filename or "-" given; write to stdout */
90         fp = stdout;
91 #ifdef HAVE_DOSISH_SYSTEM
92         setmode ( fileno(fp) , O_BINARY );
93 #endif
94     }
95     else {
96         while( !overwrite_filep (fname) ) {
97             char *tmp = ask_outfile_name (NULL, 0);
98             if ( !tmp || !*tmp ) {
99                 m_free (tmp);
100                 rc = G10ERR_CREATE_FILE;
101                 goto leave;
102             }
103             m_free (fname);
104             fname = tmp;
105         }
106     }
107
108 #ifndef __riscos__
109     if( fp || nooutput )
110         ;
111     else if( !(fp = fopen(fname,"wb")) ) {
112         log_error(_("error creating `%s': %s\n"), fname, strerror(errno) );
113         rc = G10ERR_CREATE_FILE;
114         goto leave;
115     }
116 #else /* __riscos__ */
117     /* If no output filename was given, i.e. we constructed it,
118        convert all '.' in fname to '/' but not vice versa as
119        we don't create directories! */
120     if( !opt.outfile )
121         for( c=0; fname[c]; ++c )
122             if( fname[c] == '.' )
123                 fname[c] = '/';
124
125     if( fp || nooutput )
126         ;
127     else {
128         fp = fopen(fname,"wb");
129         if( !fp ) {
130             log_error(_("error creating `%s': %s\n"), fname, strerror(errno) );
131             rc = G10ERR_CREATE_FILE;
132             if (errno == 106)
133                 log_info("Do output file and input file have the same name?\n");
134             goto leave;
135         }
136
137         /* If there's a ,xxx extension in the embedded filename,
138            use that, else check whether the user input (in fname)
139            has a ,xxx appended, then use that in preference */
140         if( (c = riscos_get_filetype_from_string( pt->name,
141                                                   pt->namelen )) != -1 )
142             filetype = c;
143         if( (c = riscos_get_filetype_from_string( fname,
144                                                   strlen(fname) )) != -1 )
145             filetype = c;
146         riscos_set_filetype_by_number(fname, filetype);
147     }
148 #endif /* __riscos__ */
149
150     if( !pt->is_partial ) {
151         /* we have an actual length (which might be zero). */
152         assert( !clearsig );
153         if( convert ) { /* text mode */
154             for( ; pt->len; pt->len-- ) {
155                 if( (c = iobuf_get(pt->buf)) == -1 ) {
156                     log_error("Problem reading source (%u bytes remaining)\n",
157                               (unsigned)pt->len);
158                     rc = G10ERR_READ_FILE;
159                     goto leave;
160                 }
161                 if( mfx->md )
162                     md_putc(mfx->md, c );
163 #ifndef HAVE_DOSISH_SYSTEM
164                 if( c == '\r' )  /* convert to native line ending */
165                     continue;    /* fixme: this hack might be too simple */
166 #endif
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         }
177         else { /* binary mode */
178             byte *buffer = m_alloc( 32768 );
179             while( pt->len ) {
180                 int len = pt->len > 32768 ? 32768 : pt->len;
181                 len = iobuf_read( pt->buf, buffer, len );
182                 if( len == -1 ) {
183                     log_error("Problem reading source (%u bytes remaining)\n",
184                               (unsigned)pt->len);
185                     rc = G10ERR_READ_FILE;
186                     m_free( buffer );
187                     goto leave;
188                 }
189                 if( mfx->md )
190                     md_write( mfx->md, buffer, len );
191                 if( fp ) {
192                     if( fwrite( buffer, 1, len, fp ) != len ) {
193                         log_error("Error writing to `%s': %s\n",
194                                   fname, strerror(errno) );
195                         rc = G10ERR_WRITE_FILE;
196                         m_free( buffer );
197                         goto leave;
198                     }
199                 }
200                 pt->len -= len;
201             }
202             m_free( buffer );
203         }
204     }
205     else if( !clearsig ) {
206         if( convert ) { /* text mode */
207             while( (c = iobuf_get(pt->buf)) != -1 ) {
208                 if( mfx->md )
209                     md_putc(mfx->md, c );
210 #ifndef HAVE_DOSISH_SYSTEM
211                 if( convert && c == '\r' )
212                     continue; /* fixme: this hack might be too simple */
213 #endif
214                 if( fp ) {
215                     if( putc( c, fp ) == EOF ) {
216                         log_error("Error writing to `%s': %s\n",
217                                   fname, strerror(errno) );
218                         rc = G10ERR_WRITE_FILE;
219                         goto leave;
220                     }
221                 }
222             }
223         }
224         else { /* binary mode */
225             byte *buffer = m_alloc( 32768 );
226             int eof;
227             for( eof=0; !eof; ) {
228                 /* Why do we check for len < 32768:
229                  * If we won't, we would practically read 2 EOFs but
230                  * the first one has already popped the block_filter
231                  * off and therefore we don't catch the boundary.
232                  * So, always assume EOF if iobuf_read returns less bytes
233                  * then requested */
234                 int len = iobuf_read( pt->buf, buffer, 32768 );
235                 if( len == -1 )
236                     break;
237                 if( len < 32768 )
238                     eof = 1;
239                 if( mfx->md )
240                     md_write( mfx->md, buffer, len );
241                 if( fp ) {
242                     if( fwrite( buffer, 1, len, fp ) != len ) {
243                         log_error("Error writing to `%s': %s\n",
244                                   fname, strerror(errno) );
245                         rc = G10ERR_WRITE_FILE;
246                         m_free( buffer );
247                         goto leave;
248                     }
249                 }
250             }
251             m_free( buffer );
252         }
253         pt->buf = NULL;
254     }
255     else {  /* clear text signature - don't hash the last cr,lf  */
256         int state = 0;
257
258         while( (c = iobuf_get(pt->buf)) != -1 ) {
259             if( fp ) {
260                 if( putc( c, fp ) == EOF ) {
261                     log_error("Error writing to `%s': %s\n",
262                                                 fname, strerror(errno) );
263                     rc = G10ERR_WRITE_FILE;
264                     goto leave;
265                 }
266             }
267             if( !mfx->md )
268                 continue;
269             if( state == 2 ) {
270                 md_putc(mfx->md, '\r' );
271                 md_putc(mfx->md, '\n' );
272                 state = 0;
273             }
274             if( !state ) {
275                 if( c == '\r'  )
276                     state = 1;
277                 else if( c == '\n'  )
278                     state = 2;
279                 else
280                     md_putc(mfx->md, c );
281             }
282             else if( state == 1 ) {
283                 if( c == '\n'  )
284                     state = 2;
285                 else {
286                     md_putc(mfx->md, '\r' );
287                     if( c == '\r'  )
288                         state = 1;
289                     else {
290                         state = 0;
291                         md_putc(mfx->md, c );
292                     }
293                 }
294             }
295         }
296         pt->buf = NULL;
297     }
298
299     if( fp && fp != stdout && fclose(fp) ) {
300         log_error("Error closing `%s': %s\n", fname, strerror(errno) );
301         fp = NULL;
302         rc = G10ERR_WRITE_FILE;
303         goto leave;
304     }
305     fp = NULL;
306
307   leave:
308     if( fp && fp != stdout )
309         fclose(fp);
310     m_free(fname);
311     return rc;
312 }
313
314 static void
315 do_hash( MD_HANDLE md, MD_HANDLE md2, IOBUF fp, int textmode )
316 {
317     text_filter_context_t tfx;
318     int c;
319
320     if( textmode ) {
321         memset( &tfx, 0, sizeof tfx);
322         iobuf_push_filter( fp, text_filter, &tfx );
323     }
324     if( md2 ) { /* work around a strange behaviour in pgp2 */
325         /* It seems that at least PGP5 converts a single CR to a CR,LF too */
326         int lc = -1;
327         while( (c = iobuf_get(fp)) != -1 ) {
328             if( c == '\n' && lc == '\r' )
329                 md_putc(md2, c);
330             else if( c == '\n' ) {
331                 md_putc(md2, '\r');
332                 md_putc(md2, c);
333             }
334             else if( c != '\n' && lc == '\r' ) {
335                 md_putc(md2, '\n');
336                 md_putc(md2, c);
337             }
338             else
339                 md_putc(md2, c);
340
341             if( md )
342                 md_putc(md, c );
343             lc = c;
344         }
345     }
346     else {
347         while( (c = iobuf_get(fp)) != -1 ) {
348             if( md )
349                 md_putc(md, c );
350         }
351     }
352 }
353
354
355 /****************
356  * Ask for the detached datafile and calculate the digest from it.
357  * INFILE is the name of the input file.
358  */
359 int
360 ask_for_detached_datafile( MD_HANDLE md, MD_HANDLE md2,
361                            const char *inname, int textmode )
362 {
363     progress_filter_context_t pfx;
364     char *answer = NULL;
365     IOBUF fp;
366     int rc = 0;
367
368     fp = open_sigfile( inname, &pfx ); /* open default file */
369
370     if( !fp && !opt.batch ) {
371         int any=0;
372         tty_printf(_("Detached signature.\n"));
373         do {
374             m_free(answer);
375             answer = cpr_get("detached_signature.filename",
376                            _("Please enter name of data file: "));
377             cpr_kill_prompt();
378             if( any && !*answer ) {
379                 rc = G10ERR_READ_FILE;
380                 goto leave;
381             }
382             fp = iobuf_open(answer);
383             if( !fp && errno == ENOENT ) {
384                 tty_printf("No such file, try again or hit enter to quit.\n");
385                 any++;
386             }
387             else if( !fp ) {
388                 log_error("can't open `%s': %s\n", answer, strerror(errno) );
389                 rc = G10ERR_READ_FILE;
390                 goto leave;
391             }
392         } while( !fp );
393     }
394
395     if( !fp ) {
396         if( opt.verbose )
397             log_info(_("reading stdin ...\n"));
398         fp = iobuf_open( NULL );
399         assert(fp);
400     }
401     do_hash( md, md2, fp, textmode );
402     iobuf_close(fp);
403
404   leave:
405     m_free(answer);
406     return rc;
407 }
408
409
410
411 /****************
412  * Hash the given files and append the hash to hash context md.
413  * If FILES is NULL, hash stdin.
414  */
415 int
416 hash_datafiles( MD_HANDLE md, MD_HANDLE md2, STRLIST files,
417                 const char *sigfilename, int textmode )
418 {
419     progress_filter_context_t pfx;
420     IOBUF fp;
421     STRLIST sl;
422
423     if( !files ) {
424         /* check whether we can open the signed material */
425         fp = open_sigfile( sigfilename, &pfx );
426         if( fp ) {
427             do_hash( md, md2, fp, textmode );
428             iobuf_close(fp);
429             return 0;
430         }
431         log_error (_("no signed data\n"));
432         return G10ERR_OPEN_FILE;
433     }
434
435
436     for (sl=files; sl; sl = sl->next ) {
437         fp = iobuf_open( sl->d );
438         if( !fp ) {
439             log_error(_("can't open signed data `%s'\n"),
440                                                 print_fname_stdin(sl->d));
441             return G10ERR_OPEN_FILE;
442         }
443         handle_progress (&pfx, fp, sl->d);
444         do_hash( md, md2, fp, textmode );
445         iobuf_close(fp);
446     }
447
448     return 0;
449 }