filetype support for RISC OS
[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 #ifdef __riscos__
82         /* If there's a ,xxx extension in the embedded filename,
83            get filetype from it and use it later on */
84         filetype = riscos_get_filetype_from_string( pt->name, pt->namelen );
85         c = riscos_get_filetype_from_string( fname, strlen(fname) );
86         if( c != 0xfff && filetype == 0xfff)
87             filetype = c;
88 #endif
89     }
90     else {
91         fname = make_printable_string( pt->name, pt->namelen, 0 );
92 #ifdef __riscos__
93         /* If there's a ,xxx extension in the embedded filename,
94            get filetype from it and use it later on, remove ,xxx from
95            actual filename */
96         if( fname[strlen(fname) - 4] == ',' ) {
97             filetype = riscos_get_filetype_from_string( pt->name, pt->namelen );
98             fname[strlen(fname) - 4] = 0;
99         }
100         for( c=0; fname[c]; ++c)
101             if( fname[c] == '.' )
102                 fname[c] = '/';
103 #endif
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 #ifdef HAVE_DOSISH_SYSTEM
112         setmode ( fileno(fp) , O_BINARY );
113 #endif
114     }
115     else {
116         while( !overwrite_filep (fname) ) {
117             char *tmp = ask_outfile_name (NULL, 0);
118             if ( !tmp || !*tmp ) {
119                 m_free (tmp);
120                 rc = G10ERR_CREATE_FILE;
121                 goto leave;
122             }
123             m_free (fname);
124             fname = tmp;
125         }
126     }
127
128     if( fp || nooutput )
129         ;
130     else if( !(fp = fopen(fname,"wb")) ) {
131         log_error(_("error creating `%s': %s\n"), fname, strerror(errno) );
132         rc = G10ERR_CREATE_FILE;
133 #ifdef __riscos__
134         if (errno == 106)
135             log_info("perhaps the output file has the same name as the input file?\n");
136 #endif /* __riscos__ */
137         goto leave;
138     }
139 #ifdef __riscos__
140     riscos_set_filetype_by_number(fname, filetype);
141 #endif
142
143     if( !pt->is_partial ) {
144         /* we have an actual length (which might be zero). */
145         assert( !clearsig );
146         if( convert ) { /* text mode */
147             for( ; pt->len; pt->len-- ) {
148                 if( (c = iobuf_get(pt->buf)) == -1 ) {
149                     log_error("Problem reading source (%u bytes remaining)\n",
150                               (unsigned)pt->len);
151                     rc = G10ERR_READ_FILE;
152                     goto leave;
153                 }
154                 if( mfx->md )
155                     md_putc(mfx->md, c );
156 #ifndef HAVE_DOSISH_SYSTEM
157                 if( c == '\r' )  /* convert to native line ending */
158                     continue;    /* fixme: this hack might be too simple */
159 #endif
160                 if( fp ) {
161                     if( putc( c, fp ) == EOF ) {
162                         log_error("Error writing to `%s': %s\n",
163                                   fname, strerror(errno) );
164                         rc = G10ERR_WRITE_FILE;
165                         goto leave;
166                     }
167                 }
168             }
169         }
170         else { /* binary mode */
171             byte *buffer = m_alloc( 32768 );
172             while( pt->len ) {
173                 int len = pt->len > 32768 ? 32768 : pt->len;
174                 len = iobuf_read( pt->buf, buffer, len );
175                 if( len == -1 ) {
176                     log_error("Problem reading source (%u bytes remaining)\n",
177                               (unsigned)pt->len);
178                     rc = G10ERR_READ_FILE;
179                     m_free( buffer );
180                     goto leave;
181                 }
182                 if( mfx->md )
183                     md_write( mfx->md, buffer, len );
184                 if( fp ) {
185                     if( fwrite( buffer, 1, len, fp ) != len ) {
186                         log_error("Error writing to `%s': %s\n",
187                                   fname, strerror(errno) );
188                         rc = G10ERR_WRITE_FILE;
189                         m_free( buffer );
190                         goto leave;
191                     }
192                 }
193                 pt->len -= len;
194             }
195             m_free( buffer );
196         }
197     }
198     else if( !clearsig ) {
199         if( convert ) { /* text mode */
200             while( (c = iobuf_get(pt->buf)) != -1 ) {
201                 if( mfx->md )
202                     md_putc(mfx->md, c );
203 #ifndef HAVE_DOSISH_SYSTEM
204                 if( convert && c == '\r' )
205                     continue; /* fixme: this hack might be too simple */
206 #endif
207                 if( fp ) {
208                     if( putc( c, fp ) == EOF ) {
209                         log_error("Error writing to `%s': %s\n",
210                                   fname, strerror(errno) );
211                         rc = G10ERR_WRITE_FILE;
212                         goto leave;
213                     }
214                 }
215             }
216         }
217         else { /* binary mode */
218             byte *buffer = m_alloc( 32768 );
219             int eof;
220             for( eof=0; !eof; ) {
221                 /* Why do we check for len < 32768:
222                  * If we won't, we would practically read 2 EOFs but
223                  * the first one has already popped the block_filter
224                  * off and therefore we don't catch the boundary.
225                  * So, always assume EOF if iobuf_read returns less bytes
226                  * then requested */
227                 int len = iobuf_read( pt->buf, buffer, 32768 );
228                 if( len == -1 )
229                     break;
230                 if( len < 32768 )
231                     eof = 1;
232                 if( mfx->md )
233                     md_write( mfx->md, buffer, len );
234                 if( fp ) {
235                     if( fwrite( buffer, 1, len, fp ) != len ) {
236                         log_error("Error writing to `%s': %s\n",
237                                   fname, strerror(errno) );
238                         rc = G10ERR_WRITE_FILE;
239                         m_free( buffer );
240                         goto leave;
241                     }
242                 }
243             }
244             m_free( buffer );
245         }
246         pt->buf = NULL;
247     }
248     else {  /* clear text signature - don't hash the last cr,lf  */
249         int state = 0;
250
251         while( (c = iobuf_get(pt->buf)) != -1 ) {
252             if( fp ) {
253                 if( putc( c, fp ) == EOF ) {
254                     log_error("Error writing to `%s': %s\n",
255                                                 fname, strerror(errno) );
256                     rc = G10ERR_WRITE_FILE;
257                     goto leave;
258                 }
259             }
260             if( !mfx->md )
261                 continue;
262             if( state == 2 ) {
263                 md_putc(mfx->md, '\r' );
264                 md_putc(mfx->md, '\n' );
265                 state = 0;
266             }
267             if( !state ) {
268                 if( c == '\r'  )
269                     state = 1;
270                 else if( c == '\n'  )
271                     state = 2;
272                 else
273                     md_putc(mfx->md, c );
274             }
275             else if( state == 1 ) {
276                 if( c == '\n'  )
277                     state = 2;
278                 else {
279                     md_putc(mfx->md, '\r' );
280                     if( c == '\r'  )
281                         state = 1;
282                     else {
283                         state = 0;
284                         md_putc(mfx->md, c );
285                     }
286                 }
287             }
288         }
289         pt->buf = NULL;
290     }
291
292     if( fp && fp != stdout && fclose(fp) ) {
293         log_error("Error closing `%s': %s\n", fname, strerror(errno) );
294         fp = NULL;
295         rc = G10ERR_WRITE_FILE;
296         goto leave;
297     }
298     fp = NULL;
299
300   leave:
301     if( fp && fp != stdout )
302         fclose(fp);
303     m_free(fname);
304     return rc;
305 }
306
307 static void
308 do_hash( MD_HANDLE md, MD_HANDLE md2, IOBUF fp, int textmode )
309 {
310     text_filter_context_t tfx;
311     int c;
312
313     if( textmode ) {
314         memset( &tfx, 0, sizeof tfx);
315         iobuf_push_filter( fp, text_filter, &tfx );
316     }
317     if( md2 ) { /* work around a strange behaviour in pgp2 */
318         /* It seems that at least PGP5 converts a single CR to a CR,LF too */
319         int lc = -1;
320         while( (c = iobuf_get(fp)) != -1 ) {
321             if( c == '\n' && lc == '\r' )
322                 md_putc(md2, c);
323             else if( c == '\n' ) {
324                 md_putc(md2, '\r');
325                 md_putc(md2, c);
326             }
327             else if( c != '\n' && lc == '\r' ) {
328                 md_putc(md2, '\n');
329                 md_putc(md2, c);
330             }
331             else
332                 md_putc(md2, c);
333
334             if( md )
335                 md_putc(md, c );
336             lc = c;
337         }
338     }
339     else {
340         while( (c = iobuf_get(fp)) != -1 ) {
341             if( md )
342                 md_putc(md, c );
343         }
344     }
345 }
346
347
348 /****************
349  * Ask for the detached datafile and calculate the digest from it.
350  * INFILE is the name of the input file.
351  */
352 int
353 ask_for_detached_datafile( MD_HANDLE md, MD_HANDLE md2,
354                            const char *inname, int textmode )
355 {
356     char *answer = NULL;
357     IOBUF fp;
358     int rc = 0;
359
360     fp = open_sigfile( inname ); /* open default file */
361     if( !fp && !opt.batch ) {
362         int any=0;
363         tty_printf(_("Detached signature.\n"));
364         do {
365             m_free(answer);
366             answer = cpr_get("detached_signature.filename",
367                            _("Please enter name of data file: "));
368             cpr_kill_prompt();
369             if( any && !*answer ) {
370                 rc = G10ERR_READ_FILE;
371                 goto leave;
372             }
373             fp = iobuf_open(answer);
374             if( !fp && errno == ENOENT ) {
375                 tty_printf("No such file, try again or hit enter to quit.\n");
376                 any++;
377             }
378             else if( !fp ) {
379                 log_error("can't open `%s': %s\n", answer, strerror(errno) );
380                 rc = G10ERR_READ_FILE;
381                 goto leave;
382             }
383         } while( !fp );
384     }
385
386     if( !fp ) {
387         if( opt.verbose )
388             log_info(_("reading stdin ...\n"));
389         fp = iobuf_open( NULL );
390         assert(fp);
391     }
392     do_hash( md, md2, fp, textmode );
393     iobuf_close(fp);
394
395
396   leave:
397     m_free(answer);
398     return rc;
399 }
400
401
402
403 /****************
404  * Hash the given files and append the hash to hash context md.
405  * If FILES is NULL, hash stdin.
406  */
407 int
408 hash_datafiles( MD_HANDLE md, MD_HANDLE md2, STRLIST files,
409                 const char *sigfilename, int textmode )
410 {
411     IOBUF fp;
412     STRLIST sl;
413
414     if( !files ) {
415         /* check whether we can open the signed material */
416         fp = open_sigfile( sigfilename );
417         if( fp ) {
418             do_hash( md, md2, fp, textmode );
419             iobuf_close(fp);
420             return 0;
421         }
422         log_error (_("no signed data\n"));
423         return G10ERR_OPEN_FILE;
424     }
425
426
427     for (sl=files; sl; sl = sl->next ) {
428         fp = iobuf_open( sl->d );
429         if( !fp ) {
430             log_error(_("can't open signed data `%s'\n"),
431                                                 print_fname_stdin(sl->d));
432             return G10ERR_OPEN_FILE;
433         }
434         do_hash( md, md2, fp, textmode );
435         iobuf_close(fp);
436     }
437
438     return 0;
439 }
440
441
442
443
444