nearly ready for 0.3.0
[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  * Handle a plaintext packet.  If MFX is not NULL, update the MDs
38  * Note: we should use the filter stuff here, but we have to add some
39  *       easy mimic to set a read limit, so we calculate only the
40  *       bytes from the plaintext.
41  */
42 int
43 handle_plaintext( PKT_plaintext *pt, md_filter_context_t *mfx, int nooutput )
44 {
45     char *fname = NULL;
46     FILE *fp = NULL;
47     int rc = 0;
48     int c;
49     int convert = pt->mode == 't';
50
51     /* create the filename as C string */
52     if( nooutput )
53         ;
54     else if( opt.outfile ) {
55         fname = m_alloc( strlen( opt.outfile ) + 1);
56         strcpy(fname, opt.outfile );
57     }
58     else {
59         fname = m_alloc( pt->namelen +1 );
60         memcpy( fname, pt->name, pt->namelen );
61         fname[pt->namelen] = 0;
62     }
63
64     if( nooutput )
65         ;
66     else if( !*fname || (*fname=='-' && !fname[1])) {
67         /* no filename or "-" given; write to stdout */
68         fp = stdout;
69     }
70     else if( overwrite_filep( fname ) )
71         goto leave;
72
73     if( fp || nooutput )
74         ;
75     else if( !(fp = fopen(fname,"wb")) ) {
76         log_error("Error creating '%s': %s\n", fname, strerror(errno) );
77         rc = G10ERR_WRITE_FILE;
78         goto leave;
79     }
80
81     if( pt->len ) {
82         for( ; pt->len; pt->len-- ) {
83             if( (c = iobuf_get(pt->buf)) == -1 ) {
84                 log_error("Problem reading source (%u bytes remaining)\n",
85                                                          (unsigned)pt->len);
86                 rc = G10ERR_READ_FILE;
87                 goto leave;
88             }
89             if( mfx->md )
90                 md_putc(mfx->md, c );
91             if( convert && c == '\r' )
92                 continue; /* FIXME: this hack is too simple */
93             if( fp ) {
94                 if( putc( c, fp ) == EOF ) {
95                     log_error("Error writing to '%s': %s\n",
96                                             fname, strerror(errno) );
97                     rc = G10ERR_WRITE_FILE;
98                     goto leave;
99                 }
100             }
101         }
102     }
103     else {
104         while( (c = iobuf_get(pt->buf)) != -1 ) {
105             if( mfx->md )
106                 md_putc(mfx->md, c );
107             if( convert && c == '\r' )
108                 continue; /* FIXME: this hack is too simple */
109             if( fp ) {
110                 if( putc( c, fp ) == EOF ) {
111                     log_error("Error writing to '%s': %s\n",
112                                                 fname, strerror(errno) );
113                     rc = G10ERR_WRITE_FILE;
114                     goto leave;
115                 }
116             }
117         }
118         iobuf_clear_eof(pt->buf);
119     }
120
121     if( fp && fp != stdout && fclose(fp) ) {
122         log_error("Error closing '%s': %s\n", fname, strerror(errno) );
123         fp = NULL;
124         rc = G10ERR_WRITE_FILE;
125         goto leave;
126     }
127     fp = NULL;
128
129   leave:
130     if( fp && fp != stdout )
131         fclose(fp);
132     m_free(fname);
133     return rc;
134 }
135
136
137 /****************
138  * Ask for the detached datafile and calculate the digest from it.
139  * INFILE is the name of the input file.
140  */
141 int
142 ask_for_detached_datafile( md_filter_context_t *mfx, const char *inname )
143 {
144     char *answer = NULL;
145     IOBUF fp;
146     int rc = 0;
147     int c;
148
149     fp = open_sigfile( inname ); /* open default file */
150     if( !fp && !opt.batch ) {
151         int any=0;
152         tty_printf("Detached signature.\n");
153         do {
154             m_free(answer);
155             answer = tty_get("Please enter name of data file: ");
156             tty_kill_prompt();
157             if( any && !*answer ) {
158                 rc = G10ERR_READ_FILE;
159                 goto leave;
160             }
161             fp = iobuf_open(answer);
162             if( !fp && errno == ENOENT ) {
163                 tty_printf("No such file, try again or hit enter to quit.\n");
164                 any++;
165             }
166             else if( !fp ) {
167                 log_error("can't open '%s': %s\n", answer, strerror(errno) );
168                 rc = G10ERR_READ_FILE;
169                 goto leave;
170             }
171         } while( !fp );
172     }
173
174     if( !fp ) {
175         if( opt.verbose )
176             log_info("reading stdin ...\n");
177         while( (c = getchar()) != EOF ) {
178             if( mfx->md )
179                 md_putc(mfx->md, c );
180         }
181     }
182     else {
183         while( (c = iobuf_get(fp)) != -1 ) {
184             if( mfx->md )
185                 md_putc(mfx->md, c );
186         }
187         iobuf_close(fp);
188     }
189
190   leave:
191     m_free(answer);
192     return rc;
193 }
194
195
196 /****************
197  * Hash the given files and append the hash to hash context md.
198  * If FILES is NULL, hash stdin.
199  */
200 int
201 hash_datafiles( MD_HANDLE md, STRLIST files, int textmode )
202 {
203     IOBUF fp;
204     STRLIST sl=NULL;
205     text_filter_context_t tfx;
206     int c;
207
208     if( !files )
209         add_to_strlist( &sl, "-");
210     else
211         sl = files;
212
213     for( ; sl; sl = sl->next ) {
214         fp = iobuf_open( sl->d );
215         if( !fp ) {
216             log_error(_("can't open signed data '%s'\n"),
217                                                 print_fname_stdin(sl->d));
218             if( !files )
219                 free_strlist(sl);
220             return G10ERR_OPEN_FILE;
221         }
222         if( textmode ) {
223             memset( &tfx, 0, sizeof tfx);
224             iobuf_push_filter( fp, text_filter, &tfx );
225         }
226         while( (c = iobuf_get(fp)) != -1 )
227             md_putc(md, c );
228         iobuf_close(fp);
229     }
230
231     if( !files )
232         free_strlist(sl);
233     return 0;
234 }
235
236