List and check sigs works
[gnupg.git] / g10 / build-packet.c
1 /* build-packet.c - assemble packets and write them
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 <assert.h>
26
27 #include "packet.h"
28 #include "errors.h"
29 #include "iobuf.h"
30 #include "mpi.h"
31 #include "util.h"
32 #include "cipher.h"
33 #include "memory.h"
34 #include "options.h"
35
36
37 static int do_comment( IOBUF out, int ctb, PKT_comment *rem );
38 static int do_user_id( IOBUF out, int ctb, PKT_user_id *uid );
39 static int do_public_cert( IOBUF out, int ctb, PKT_public_cert *pk );
40 static int do_secret_cert( IOBUF out, int ctb, PKT_secret_cert *pk );
41 static int do_pubkey_enc( IOBUF out, int ctb, PKT_pubkey_enc *enc );
42 static u32 calc_plaintext( PKT_plaintext *pt );
43 static int do_plaintext( IOBUF out, int ctb, PKT_plaintext *pt );
44 static int do_encrypted( IOBUF out, int ctb, PKT_encrypted *ed );
45 static int do_compressed( IOBUF out, int ctb, PKT_compressed *cd );
46 static int do_signature( IOBUF out, int ctb, PKT_signature *sig );
47
48 static int calc_header_length( u32 len );
49 static int write_16(IOBUF inp, u16 a);
50 static int write_32(IOBUF inp, u32 a);
51 static int write_header( IOBUF out, int ctb, u32 len );
52 static int write_header2( IOBUF out, int ctb, u32 len, int blkmode );
53 static int write_version( IOBUF out, int ctb );
54
55 /****************
56  * Build a packet and write it to INP
57  * Returns: 0 := okay
58  *         >0 := error
59  * Note: Caller must free the packet
60  */
61 int
62 build_packet( IOBUF out, PACKET *pkt )
63 {
64     int rc=0, ctb;
65
66     if( DBG_PACKET )
67         log_debug("build_packet() type=%d\n", pkt->pkttype );
68     assert( pkt->pkt.generic );
69     ctb = 0x80 | ((pkt->pkttype & 15)<<2);
70     switch( pkt->pkttype ) {
71       case PKT_USER_ID:
72         rc = do_user_id( out, ctb, pkt->pkt.user_id );
73         break;
74       case PKT_COMMENT:
75         rc = do_comment( out, ctb, pkt->pkt.comment );
76         break;
77       case PKT_PUBLIC_CERT:
78         rc = do_public_cert( out, ctb, pkt->pkt.public_cert );
79         break;
80       case PKT_SECRET_CERT:
81         rc = do_secret_cert( out, ctb, pkt->pkt.secret_cert );
82         break;
83       case PKT_PUBKEY_ENC:
84         rc = do_pubkey_enc( out, ctb, pkt->pkt.pubkey_enc );
85         break;
86       case PKT_PLAINTEXT:
87         rc = do_plaintext( out, ctb, pkt->pkt.plaintext );
88         break;
89       case PKT_ENCRYPTED:
90         rc = do_encrypted( out, ctb, pkt->pkt.encrypted );
91         break;
92       case PKT_COMPRESSED:
93         rc = do_compressed( out, ctb, pkt->pkt.compressed );
94         break;
95       case PKT_SIGNATURE:
96         rc = do_signature( out, ctb, pkt->pkt.signature );
97         break;
98       case PKT_RING_TRUST:
99       default:
100         log_bug("invalid packet type in build_packet()");
101         break;
102     }
103
104     return rc;
105 }
106
107 /****************
108  * calculate the length of a packet described by PKT
109  */
110 u32
111 calc_packet_length( PACKET *pkt )
112 {
113     u32 n=0;
114
115     assert( pkt->pkt.generic );
116     switch( pkt->pkttype ) {
117       case PKT_PLAINTEXT:
118         n = calc_plaintext( pkt->pkt.plaintext );
119         break;
120       case PKT_USER_ID:
121       case PKT_COMMENT:
122       case PKT_PUBLIC_CERT:
123       case PKT_SECRET_CERT:
124       case PKT_PUBKEY_ENC:
125       case PKT_ENCRYPTED:
126       case PKT_SIGNATURE:
127       case PKT_RING_TRUST:
128       case PKT_COMPRESSED:
129       default:
130         log_bug("invalid packet type in calc_packet_length()");
131         break;
132     }
133     n += calc_header_length(n);
134     return n;
135 }
136
137
138 static int
139 do_comment( IOBUF out, int ctb, PKT_comment *rem )
140 {
141     write_header(out, ctb, rem->len);
142     if( iobuf_write( out, rem->data, rem->len ) )
143         return G10ERR_WRITE_FILE;
144     return 0;
145 }
146
147 static int
148 do_user_id( IOBUF out, int ctb, PKT_user_id *uid )
149 {
150     write_header(out, ctb, uid->len);
151     if( iobuf_write( out, uid->name, uid->len ) )
152         return G10ERR_WRITE_FILE;
153     return 0;
154 }
155
156 static int
157 do_public_cert( IOBUF out, int ctb, PKT_public_cert *pkc )
158 {
159     int rc = 0;
160     IOBUF a = iobuf_temp();
161
162     write_version( a, ctb );
163     write_32(a, pkc->timestamp );
164     write_16(a, pkc->valid_days );
165     iobuf_put(a, pkc->pubkey_algo );
166     if( pkc->pubkey_algo == PUBKEY_ALGO_ELGAMAL ) {
167         mpi_encode(a, pkc->d.elg.p );
168         mpi_encode(a, pkc->d.elg.g );
169         mpi_encode(a, pkc->d.elg.y );
170     }
171     else if( pkc->pubkey_algo == PUBKEY_ALGO_RSA ) {
172         mpi_encode(a, pkc->d.rsa.rsa_n );
173         mpi_encode(a, pkc->d.rsa.rsa_e );
174     }
175     else {
176         rc = G10ERR_PUBKEY_ALGO;
177         goto leave;
178     }
179
180     write_header(out, ctb, iobuf_get_temp_length(a) );
181     if( iobuf_write_temp( out, a ) )
182         rc = G10ERR_WRITE_FILE;
183
184   leave:
185     iobuf_close(a);
186     return rc;
187 }
188
189 static int
190 do_secret_cert( IOBUF out, int ctb, PKT_secret_cert *skc )
191 {
192     int rc = 0;
193     IOBUF a = iobuf_temp();
194
195     write_version( a, ctb );
196     write_32(a, skc->timestamp );
197     write_16(a, skc->valid_days );
198     iobuf_put(a, skc->pubkey_algo );
199     if( skc->pubkey_algo == PUBKEY_ALGO_ELGAMAL ) {
200         mpi_encode(a, skc->d.elg.p );
201         mpi_encode(a, skc->d.elg.g );
202         mpi_encode(a, skc->d.elg.y );
203         iobuf_put(a, skc->d.elg.protect_algo );
204         skc->d.elg.calc_csum = 0;
205         if( skc->d.elg.protect_algo ) {
206             assert( skc->d.elg.is_protected == 1 );
207             assert( skc->d.elg.protect_algo == CIPHER_ALGO_BLOWFISH );
208             iobuf_write(a, skc->d.elg.protect.blowfish.iv, 8 );
209             mpi_write_csum(a, (byte*)skc->d.elg.x, &skc->d.elg.calc_csum );
210         }
211         else {  /* not protected */
212             assert( !skc->d.elg.is_protected );
213             mpi_encode_csum(a, skc->d.elg.x, &skc->d.elg.calc_csum );
214         }
215
216         write_16(a, skc->d.elg.calc_csum );
217     }
218     else if( skc->pubkey_algo == PUBKEY_ALGO_RSA ) {
219         mpi_encode(a, skc->d.rsa.rsa_n );
220         mpi_encode(a, skc->d.rsa.rsa_e );
221         iobuf_put(a, skc->d.rsa.protect_algo );
222         skc->d.rsa.calc_csum = 0;
223         if( skc->d.rsa.protect_algo ) {
224             assert( skc->d.rsa.is_protected == 1 );
225             assert( skc->d.rsa.protect_algo == CIPHER_ALGO_BLOWFISH );
226             iobuf_write(a, skc->d.rsa.protect.blowfish.iv, 8 );
227
228             mpi_write_csum(a, (byte*)skc->d.rsa.rsa_d, &skc->d.rsa.calc_csum );
229             mpi_write_csum(a, (byte*)skc->d.rsa.rsa_p, &skc->d.rsa.calc_csum );
230             mpi_write_csum(a, (byte*)skc->d.rsa.rsa_q, &skc->d.rsa.calc_csum );
231             mpi_write_csum(a, (byte*)skc->d.rsa.rsa_u, &skc->d.rsa.calc_csum );
232         }
233         else {  /* Not protected: You fool you! */
234             assert( !skc->d.rsa.is_protected );
235             mpi_encode_csum(a, skc->d.rsa.rsa_d, &skc->d.rsa.calc_csum );
236             mpi_encode_csum(a, skc->d.rsa.rsa_p, &skc->d.rsa.calc_csum );
237             mpi_encode_csum(a, skc->d.rsa.rsa_q, &skc->d.rsa.calc_csum );
238             mpi_encode_csum(a, skc->d.rsa.rsa_u, &skc->d.rsa.calc_csum );
239         }
240
241         write_16(a, skc->d.rsa.calc_csum );
242     }
243     else {
244         rc = G10ERR_PUBKEY_ALGO;
245         goto leave;
246     }
247
248     write_header(out, ctb, iobuf_get_temp_length(a) );
249     if( iobuf_write_temp( out, a ) )
250         rc = G10ERR_WRITE_FILE;
251
252   leave:
253     iobuf_close(a);
254     return rc;
255 }
256
257 static int
258 do_pubkey_enc( IOBUF out, int ctb, PKT_pubkey_enc *enc )
259 {
260     int rc = 0;
261     IOBUF a = iobuf_temp();
262
263     write_version( a, ctb );
264     write_32(a, enc->keyid[0] );
265     write_32(a, enc->keyid[1] );
266     iobuf_put(a,enc->pubkey_algo );
267     if( enc->pubkey_algo == PUBKEY_ALGO_ELGAMAL ) {
268         mpi_encode(a, enc->d.elg.a );
269         mpi_encode(a, enc->d.elg.b );
270     }
271     else if( enc->pubkey_algo == PUBKEY_ALGO_RSA ) {
272         mpi_encode(a, enc->d.rsa.rsa_integer );
273     }
274     else {
275         rc = G10ERR_PUBKEY_ALGO;
276         goto leave;
277     }
278
279     write_header(out, ctb, iobuf_get_temp_length(a) );
280     if( iobuf_write_temp( out, a ) )
281         rc = G10ERR_WRITE_FILE;
282
283   leave:
284     iobuf_close(a);
285     return rc;
286 }
287
288
289
290
291
292 static u32
293 calc_plaintext( PKT_plaintext *pt )
294 {
295     return pt->len? (1 + 1 + pt->namelen + 4 + pt->len) : 0;
296 }
297
298 static int
299 do_plaintext( IOBUF out, int ctb, PKT_plaintext *pt )
300 {
301     int c, i, rc = 0;
302     u32 n;
303
304     write_header(out, ctb, calc_plaintext( pt ) );
305     iobuf_put(out, pt->mode );
306     iobuf_put(out, pt->namelen );
307     for(i=0; i < pt->namelen; i++ )
308         iobuf_put(out, pt->name[i] );
309     if( write_32(out, pt->timestamp ) )
310         rc = G10ERR_WRITE_FILE;
311
312     n = 0;
313     while( (c=iobuf_get(pt->buf)) != -1 ) {
314         if( iobuf_put(out, c) ) {
315             rc = G10ERR_WRITE_FILE;
316             break;
317         }
318         n++;
319     }
320     if( !pt->len )
321         iobuf_set_block_mode(out, 0 ); /* write end marker */
322     else if( n != pt->len )
323         log_error("do_plaintext(): wrote %lu bytes but expected %lu bytes\n",
324                         (ulong)n, (ulong)pt->len );
325
326     return rc;
327 }
328
329
330
331 static int
332 do_encrypted( IOBUF out, int ctb, PKT_encrypted *ed )
333 {
334     int rc = 0;
335     u32 n;
336
337     n = ed->len ? (ed->len + 10) : 0;
338     write_header(out, ctb, n );
339
340     /* This is all. The caller has to write the real data */
341
342     return rc;
343 }
344
345 static int
346 do_compressed( IOBUF out, int ctb, PKT_compressed *cd )
347 {
348     int rc = 0;
349
350     /* we must use the old convention and don't use blockmode */
351     write_header2(out, ctb, 0, 0 );
352     iobuf_put(out, cd->algorithm );
353
354     /* This is all. The caller has to write the real data */
355
356     return rc;
357 }
358
359
360 static int
361 do_signature( IOBUF out, int ctb, PKT_signature *sig )
362 {
363     int rc = 0;
364     IOBUF a = iobuf_temp();
365
366     write_version( a, ctb );
367     iobuf_put(a, 5 ); /* constant */
368     iobuf_put(a, sig->sig_class );
369     write_32(a, sig->timestamp );
370     write_32(a, sig->keyid[0] );
371     write_32(a, sig->keyid[1] );
372     iobuf_put(a, sig->pubkey_algo );
373     if( sig->pubkey_algo == PUBKEY_ALGO_ELGAMAL ) {
374         iobuf_put(a, sig->d.elg.digest_algo );
375         iobuf_put(a, sig->d.elg.digest_start[0] );
376         iobuf_put(a, sig->d.elg.digest_start[1] );
377         mpi_encode(a, sig->d.elg.a );
378         mpi_encode(a, sig->d.elg.b );
379     }
380     else if( sig->pubkey_algo == PUBKEY_ALGO_RSA ) {
381         iobuf_put(a, sig->d.rsa.digest_algo );
382         iobuf_put(a, sig->d.rsa.digest_start[0] );
383         iobuf_put(a, sig->d.rsa.digest_start[1] );
384         mpi_encode(a, sig->d.rsa.rsa_integer );
385     }
386     else {
387         rc = G10ERR_PUBKEY_ALGO;
388         goto leave;
389     }
390
391     write_header(out, ctb, iobuf_get_temp_length(a) );
392     if( iobuf_write_temp( out, a ) )
393         rc = G10ERR_WRITE_FILE;
394
395   leave:
396     iobuf_close(a);
397     return rc;
398 }
399
400
401 static int
402 write_16(IOBUF out, u16 a)
403 {
404     iobuf_put(out, a>>8);
405     if( iobuf_put(out,a) )
406         return -1;
407     return 0;
408 }
409
410 static int
411 write_32(IOBUF out, u32 a)
412 {
413     iobuf_put(out, a>> 24);
414     iobuf_put(out, a>> 16);
415     iobuf_put(out, a>> 8);
416     if( iobuf_put(out, a) )
417         return -1;
418     return 0;
419 }
420
421
422 /****************
423  * calculate the length of a header
424  */
425 static int
426 calc_header_length( u32 len )
427 {
428     if( !len )
429         return 1; /* only the ctb */
430     else if( len < 256 )
431         return 2;
432     else if( len < 65536 )
433         return 3;
434     else
435         return 5;
436 }
437
438 /****************
439  * Write the CTB and the packet length
440  */
441 static int
442 write_header( IOBUF out, int ctb, u32 len )
443 {
444     return write_header2( out, ctb, len, 1 );
445 }
446
447 static int
448 write_header2( IOBUF out, int ctb, u32 len, int blkmode )
449 {
450     if( !len )
451         ctb |= 3;
452     else if( len < 256 )
453         ;
454     else if( len < 65536 )
455         ctb |= 1;
456     else
457         ctb |= 2;
458     if( iobuf_put(out, ctb ) )
459         return -1;
460     if( !len ) {
461         if( blkmode )
462             iobuf_set_block_mode(out, 8196 );
463     }
464     else {
465         if( ctb & 2 ) {
466             iobuf_put(out, len >> 24 );
467             iobuf_put(out, len >> 16 );
468         }
469         if( ctb & 3 )
470             iobuf_put(out, len >> 8 );
471         if( iobuf_put(out, len ) )
472             return -1;
473     }
474     return 0;
475 }
476
477 static int
478 write_version( IOBUF out, int ctb )
479 {
480     if( iobuf_put( out, 3 ) )
481         return -1;
482     return 0;
483 }
484