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