release 0.2.3
[gnupg.git] / g10 / plaintext.c
1 /* plaintext.c -  process an plaintext packet
2  *      Copyright (c) 1997 by Werner Koch (dd9jn)
3  *
4  * This file is part of G10.
5  *
6  * G10 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  * G10 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 static FILE *abc;
49 if( !abc )
50     abc=fopen("plaintext.out", "wb");
51 if( !abc ) BUG();
52
53     /* create the filename as C string */
54     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( !*fname ) { /* no filename given; write to stdout */
65         fp = stdout;
66     }
67     else if( overwrite_filep( fname ) )
68         goto leave;
69
70     if( fp )
71         ;
72     else if( !(fp = fopen(fname,"wb")) ) {
73         log_error("Error creating '%s': %s\n", fname, strerror(errno) );
74         rc = G10ERR_WRITE_FILE;
75         goto leave;
76     }
77
78     if( pt->len ) {
79         for( ; pt->len; pt->len-- ) {
80             if( (c = iobuf_get(pt->buf)) == -1 ) {
81                 log_error("Problem reading source\n");
82                 rc = G10ERR_READ_FILE;
83                 goto leave;
84             }
85             putc( c, abc );
86             if( mfx->md )
87                 md_putc(mfx->md, c );
88             if( putc( c, fp ) == EOF ) {
89                 log_error("Error writing to '%s': %s\n", fname, strerror(errno) );
90                 rc = G10ERR_WRITE_FILE;
91                 goto leave;
92             }
93         }
94     }
95     else {
96         while( (c = iobuf_get(pt->buf)) != -1 ) {
97             putc( c, abc );
98             if( mfx->md )
99                 md_putc(mfx->md, c );
100             if( putc( c, fp ) == EOF ) {
101                 log_error("Error writing to '%s': %s\n",
102                                             fname, strerror(errno) );
103                 rc = G10ERR_WRITE_FILE;
104                 goto leave;
105             }
106         }
107         iobuf_clear_eof(pt->buf);
108     }
109
110     if( fp && fp != stdout && fclose(fp) ) {
111         log_error("Error closing '%s': %s\n", fname, strerror(errno) );
112         fp = NULL;
113         rc = G10ERR_WRITE_FILE;
114         goto leave;
115     }
116     fp = NULL;
117
118   leave:
119     if( fp && fp != stdout )
120         fclose(fp);
121     m_free(fname);
122     return rc;
123 }
124
125
126 /****************
127  * Ask for the detached datafile and calculate the digest from it.
128  * INFILE is the name of the input file.
129  */
130 int
131 ask_for_detached_datafile( md_filter_context_t *mfx, const char *inname )
132 {
133     char *answer = NULL;
134     IOBUF fp;
135     int rc = 0;
136     int c;
137
138     fp = open_sigfile( inname ); /* open default file */
139     if( !fp ) {
140         int any=0;
141         tty_printf("Detached signature.\n");
142         do {
143             m_free(answer);
144             answer = tty_get("Please enter name of data file: ");
145             tty_kill_prompt();
146             if( any && !*answer ) {
147                 rc = G10ERR_READ_FILE;
148                 goto leave;
149             }
150             fp = iobuf_open(answer);
151             if( !fp && errno == ENOENT ) {
152                 tty_printf("No such file, try again or hit enter to quit.\n");
153                 any++;
154             }
155             else if( !fp ) {
156                 log_error("can't open '%s': %s\n", answer, strerror(errno) );
157                 rc = G10ERR_READ_FILE;
158                 goto leave;
159             }
160         } while( !fp );
161     }
162
163     while( (c = iobuf_get(fp)) != -1 ) {
164         if( mfx->md )
165             md_putc(mfx->md, c );
166     }
167     iobuf_close(fp);
168
169   leave:
170     m_free(answer);
171     return rc;
172 }
173
174