Minor fixes
[gnupg.git] / g10 / encr-data.c
1 /* encr-data.c -  process an encrypted data packet
2  * Copyright (C) 1998, 1999, 2000, 2001 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 <assert.h>
26 #include "util.h"
27 #include "memory.h"
28 #include "packet.h"
29 #include "mpi.h"
30 #include "cipher.h"
31 #include "options.h"
32 #include "i18n.h"
33
34
35 static int mdc_decode_filter( void *opaque, int control, IOBUF a,
36                                               byte *buf, size_t *ret_len);
37 static int decode_filter( void *opaque, int control, IOBUF a,
38                                         byte *buf, size_t *ret_len);
39
40 typedef struct {
41     CIPHER_HANDLE cipher_hd;
42     MD_HANDLE mdc_hash;
43     char defer[20];
44     int  defer_filled;
45     int  eof_seen;
46 } decode_filter_ctx_t;
47
48
49 /****************
50  * Decrypt the data, specified by ED with the key DEK.
51  */
52 int
53 decrypt_data( void *procctx, PKT_encrypted *ed, DEK *dek )
54 {
55     decode_filter_ctx_t dfx;
56     byte *p;
57     int rc=0, c, i;
58     byte temp[32];
59     unsigned blocksize;
60     unsigned nprefix;
61
62     memset( &dfx, 0, sizeof dfx );
63     if( opt.verbose && !dek->algo_info_printed ) {
64         const char *s = cipher_algo_to_string( dek->algo );
65         if( s )
66             log_info(_("%s encrypted data\n"), s );
67         else
68             log_info(_("encrypted with unknown algorithm %d\n"), dek->algo );
69         dek->algo_info_printed = 1;
70     }
71     if( (rc=check_cipher_algo(dek->algo)) )
72         goto leave;
73     blocksize = cipher_get_blocksize(dek->algo);
74     if( !blocksize || blocksize > 16 )
75         log_fatal("unsupported blocksize %u\n", blocksize );
76     nprefix = blocksize;
77     if( ed->len && ed->len < (nprefix+2) )
78         BUG();
79
80     if( ed->mdc_method ) {
81         dfx.mdc_hash = md_open( ed->mdc_method, 0 );
82         if ( DBG_HASHING )
83             md_start_debug(dfx.mdc_hash, "checkmdc");
84     }
85     dfx.cipher_hd = cipher_open( dek->algo,
86                                  ed->mdc_method? CIPHER_MODE_CFB
87                                                : CIPHER_MODE_AUTO_CFB, 1 );
88     /* log_hexdump( "thekey", dek->key, dek->keylen );*/
89     rc = cipher_setkey( dfx.cipher_hd, dek->key, dek->keylen );
90     if( rc == G10ERR_WEAK_KEY )
91       {
92         log_info(_("WARNING: message was encrypted with"
93                    " a weak key in the symmetric cipher.\n"));
94         rc=0;
95       }
96     else if( rc )
97       {
98         log_error("key setup failed: %s\n", g10_errstr(rc) );
99         goto leave;
100       
101 }
102     if (!ed->buf) {
103         log_error(_("problem handling encrypted packet\n"));
104         goto leave;
105     }
106
107     cipher_setiv( dfx.cipher_hd, NULL, 0 );
108
109     if( ed->len ) {
110         for(i=0; i < (nprefix+2) && ed->len; i++, ed->len-- ) {
111             if( (c=iobuf_get(ed->buf)) == -1 )
112                 break;
113             else
114                 temp[i] = c;
115         }
116     }
117     else {
118         for(i=0; i < (nprefix+2); i++ )
119             if( (c=iobuf_get(ed->buf)) == -1 )
120                 break;
121             else
122                 temp[i] = c;
123     }
124     cipher_decrypt( dfx.cipher_hd, temp, temp, nprefix+2);
125     cipher_sync( dfx.cipher_hd );
126     p = temp;
127 /* log_hexdump( "prefix", temp, nprefix+2 ); */
128     if( p[nprefix-2] != p[nprefix] || p[nprefix-1] != p[nprefix+1] ) {
129         rc = G10ERR_BAD_KEY;
130         goto leave;
131     }
132
133     if( dfx.mdc_hash )
134         md_write( dfx.mdc_hash, temp, nprefix+2 );
135
136     if( ed->mdc_method )
137         iobuf_push_filter( ed->buf, mdc_decode_filter, &dfx );
138     else
139         iobuf_push_filter( ed->buf, decode_filter, &dfx );
140
141     proc_packets( procctx, ed->buf );
142     ed->buf = NULL;
143     if( ed->mdc_method && dfx.eof_seen == 2 )
144         rc = G10ERR_INVALID_PACKET;
145     else if( ed->mdc_method ) { /* check the mdc */
146         int datalen = md_digest_length( ed->mdc_method );
147
148         cipher_decrypt( dfx.cipher_hd, dfx.defer, dfx.defer, 20);
149         md_final( dfx.mdc_hash );
150         if( datalen != 20
151             || memcmp(md_read( dfx.mdc_hash, 0 ), dfx.defer, datalen) )
152             rc = G10ERR_BAD_SIGN;
153         /*log_hexdump("MDC calculated:", md_read( dfx.mdc_hash, 0), datalen);*/
154         /*log_hexdump("MDC message   :", dfx.defer, 20);*/
155     }
156     
157
158   leave:
159     cipher_close(dfx.cipher_hd);
160     md_close( dfx.mdc_hash );
161     return rc;
162 }
163
164
165
166 /* I think we should merge this with cipher_filter */
167 static int
168 mdc_decode_filter( void *opaque, int control, IOBUF a,
169                                               byte *buf, size_t *ret_len)
170 {
171     decode_filter_ctx_t *dfx = opaque;
172     size_t n, size = *ret_len;
173     int rc = 0;
174     int c;
175
176     if( control == IOBUFCTRL_UNDERFLOW && dfx->eof_seen ) {
177         *ret_len = 0;
178         rc = -1;
179     }
180     else if( control == IOBUFCTRL_UNDERFLOW ) {
181         assert(a);
182         assert( size > 40 );
183
184         /* get at least 20 bytes and put it somewhere ahead in the buffer */
185         for(n=20; n < 40 ; n++ ) {
186             if( (c = iobuf_get(a)) == -1 )
187                 break;
188             buf[n] = c;
189         }
190         if( n == 40 ) {
191             /* we have enough stuff - flush the deferred stuff */
192             /* (we have asserted that the buffer is large enough) */
193             if( !dfx->defer_filled ) { /* the first time */
194                 memcpy(buf, buf+20, 20 );
195                 n = 20;
196             }
197             else {
198                 memcpy(buf, dfx->defer, 20 );
199             }
200             /* now fill up */
201             for(; n < size; n++ ) {
202                 if( (c = iobuf_get(a)) == -1 )
203                     break;
204                 buf[n] = c;
205             }
206             /* move the last 20 bytes back to the defer buffer */
207             /* (okay, we are wasting 20 bytes of supplied buffer) */
208             n -= 20;
209             memcpy( dfx->defer, buf+n, 20 );
210             dfx->defer_filled = 1;
211         }
212         else if( !dfx->defer_filled ) { /* eof seen buf empty defer */
213             /* this is bad because there is an incomplete hash */
214             n -= 20;
215             memcpy(buf, buf+20, n );
216             dfx->eof_seen = 2; /* eof with incomplete hash */
217         }
218         else { /* eof seen */
219             memcpy(buf, dfx->defer, 20 );
220             n -= 20;
221             memcpy( dfx->defer, buf+n, 20 );
222             dfx->eof_seen = 1; /* normal eof */
223         }
224
225         if( n ) {
226             cipher_decrypt( dfx->cipher_hd, buf, buf, n);
227             md_write( dfx->mdc_hash, buf, n );
228         }
229         else {
230             assert( dfx->eof_seen );
231             rc = -1; /* eof */
232         }
233         *ret_len = n;
234     }
235     else if( control == IOBUFCTRL_DESC ) {
236         *(char**)buf = "mdc_decode_filter";
237     }
238     return rc;
239 }
240
241 static int
242 decode_filter( void *opaque, int control, IOBUF a, byte *buf, size_t *ret_len)
243 {
244     decode_filter_ctx_t *fc = opaque;
245     size_t n, size = *ret_len;
246     int rc = 0;
247
248     if( control == IOBUFCTRL_UNDERFLOW ) {
249         assert(a);
250         n = iobuf_read( a, buf, size );
251         if( n == -1 ) n = 0;
252         if( n )
253             cipher_decrypt( fc->cipher_hd, buf, buf, n);
254         else
255             rc = -1; /* eof */
256         *ret_len = n;
257     }
258     else if( control == IOBUFCTRL_DESC ) {
259         *(char**)buf = "decode_filter";
260     }
261     return rc;
262 }
263