284b684b5ca4b50af850479b9f08a852560bb327
[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
34
35 /****************
36  * Handle a plaintext packet.  If MFX is not NULL, update the MDs
37  * Note: we should use the filter stuff here, but we have to add some
38  *       easy mimic to set a read limit, so we calculate only the
39  *       bytes from the plaintext.
40  */
41 int
42 handle_plaintext( PKT_plaintext *pt, md_filter_context_t *mfx )
43 {
44     char *fname;
45     FILE *fp = NULL;
46     int rc = 0;
47     int c;
48     int convert = pt->mode == 't';
49
50     /* create the filename as C string */
51     if( opt.outfile ) {
52         fname = m_alloc( strlen( opt.outfile ) + 1);
53         strcpy(fname, opt.outfile );
54     }
55     else {
56         fname = m_alloc( pt->namelen +1 );
57         memcpy( fname, pt->name, pt->namelen );
58         fname[pt->namelen] = 0;
59     }
60
61     if( !*fname ) { /* no filename given; write to stdout */
62         fp = stdout;
63     }
64     else if( overwrite_filep( fname ) )
65         goto leave;
66
67     if( fp )
68         ;
69     else if( !(fp = fopen(fname,"wb")) ) {
70         log_error("Error creating '%s': %s\n", fname, strerror(errno) );
71         rc = G10ERR_WRITE_FILE;
72         goto leave;
73     }
74
75     if( pt->len ) {
76         for( ; pt->len; pt->len-- ) {
77             if( (c = iobuf_get(pt->buf)) == -1 ) {
78                 log_error("Problem reading source\n");
79                 rc = G10ERR_READ_FILE;
80                 goto leave;
81             }
82             if( mfx->md )
83                 md_putc(mfx->md, c );
84             if( convert && c == '\r' )
85                 continue; /* FIXME: this hack is too simple */
86             if( putc( c, fp ) == EOF ) {
87                 log_error("Error writing to '%s': %s\n", fname, strerror(errno) );
88                 rc = G10ERR_WRITE_FILE;
89                 goto leave;
90             }
91         }
92     }
93     else {
94         while( (c = iobuf_get(pt->buf)) != -1 ) {
95             if( mfx->md )
96                 md_putc(mfx->md, c );
97             if( convert && c == '\r' )
98                 continue; /* FIXME: this hack is too simple */
99             if( putc( c, fp ) == EOF ) {
100                 log_error("Error writing to '%s': %s\n",
101                                             fname, strerror(errno) );
102                 rc = G10ERR_WRITE_FILE;
103                 goto leave;
104             }
105         }
106         iobuf_clear_eof(pt->buf);
107     }
108
109     if( fp && fp != stdout && fclose(fp) ) {
110         log_error("Error closing '%s': %s\n", fname, strerror(errno) );
111         fp = NULL;
112         rc = G10ERR_WRITE_FILE;
113         goto leave;
114     }
115     fp = NULL;
116
117   leave:
118     if( fp && fp != stdout )
119         fclose(fp);
120     m_free(fname);
121     return rc;
122 }
123
124
125 /****************
126  * Ask for the detached datafile and calculate the digest from it.
127  * INFILE is the name of the input file.
128  */
129 int
130 ask_for_detached_datafile( md_filter_context_t *mfx, const char *inname )
131 {
132     char *answer = NULL;
133     IOBUF fp;
134     int rc = 0;
135     int c;
136
137     fp = open_sigfile( inname ); /* open default file */
138     if( !fp && !opt.batch ) {
139         int any=0;
140         tty_printf("Detached signature.\n");
141         do {
142             m_free(answer);
143             answer = tty_get("Please enter name of data file: ");
144             tty_kill_prompt();
145             if( any && !*answer ) {
146                 rc = G10ERR_READ_FILE;
147                 goto leave;
148             }
149             fp = iobuf_open(answer);
150             if( !fp && errno == ENOENT ) {
151                 tty_printf("No such file, try again or hit enter to quit.\n");
152                 any++;
153             }
154             else if( !fp ) {
155                 log_error("can't open '%s': %s\n", answer, strerror(errno) );
156                 rc = G10ERR_READ_FILE;
157                 goto leave;
158             }
159         } while( !fp );
160     }
161
162     if( !fp ) {
163         if( opt.verbose )
164             log_info("reading stdin ...\n");
165         while( (c = getchar()) != EOF ) {
166             if( mfx->md )
167                 md_putc(mfx->md, c );
168         }
169     }
170     else {
171         while( (c = iobuf_get(fp)) != -1 ) {
172             if( mfx->md )
173                 md_putc(mfx->md, c );
174         }
175         iobuf_close(fp);
176     }
177
178   leave:
179     m_free(answer);
180     return rc;
181 }
182
183