945758e984e3fc885076bce911feb8aaf3539928
[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_pubkey_cert( IOBUF out, int ctb, PKT_pubkey_cert *pk );
40 static int do_seckey_cert( IOBUF out, int ctb, PKT_seckey_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_encr_data( IOBUF out, int ctb, PKT_encr_data *ed );
45 static int do_compressed( IOBUF out, int ctb, PKT_compressed *cd );
46
47 static int calc_header_length( u32 len );
48 static int write_16(IOBUF inp, u16 a);
49 static int write_32(IOBUF inp, u32 a);
50 static int write_header( IOBUF out, int ctb, u32 len );
51 static int write_header2( IOBUF out, int ctb, u32 len, int blkmode );
52 static int write_version( IOBUF out, int ctb );
53
54 /****************
55  * Build a packet and write it to INP
56  * Returns: 0 := okay
57  *         >0 := error
58  * Note: Caller must free the packet
59  */
60 int
61 build_packet( IOBUF out, PACKET *pkt )
62 {
63     int rc=0, ctb;
64
65     if( DBG_PACKET )
66         log_debug("build_packet() type=%d\n", pkt->pkttype );
67     assert( pkt->pkt.generic );
68     ctb = 0x80 | ((pkt->pkttype & 15)<<2);
69     switch( pkt->pkttype ) {
70       case PKT_USER_ID:
71         rc = do_user_id( out, ctb, pkt->pkt.user_id );
72         break;
73       case PKT_COMMENT:
74         rc = do_comment( out, ctb, pkt->pkt.comment );
75         break;
76       case PKT_PUBKEY_CERT:
77         rc = do_pubkey_cert( out, ctb, pkt->pkt.pubkey_cert );
78         break;
79       case PKT_SECKEY_CERT:
80         rc = do_seckey_cert( out, ctb, pkt->pkt.seckey_cert );
81         break;
82       case PKT_PUBKEY_ENC:
83         rc = do_pubkey_enc( out, ctb, pkt->pkt.pubkey_enc );
84         break;
85       case PKT_PLAINTEXT:
86         rc = do_plaintext( out, ctb, pkt->pkt.plaintext );
87         break;
88       case PKT_ENCR_DATA:
89         rc = do_encr_data( out, ctb, pkt->pkt.encr_data );
90         break;
91       case PKT_COMPR_DATA:
92         rc = do_compressed( out, ctb, pkt->pkt.compressed );
93         break;
94       case PKT_SIGNATURE:
95       case PKT_RING_TRUST:
96       default:
97         log_bug("invalid packet type in build_packet()");
98         break;
99     }
100
101     return rc;
102 }
103
104 /****************
105  * calculate the length of a packet described by PKT
106  */
107 u32
108 calc_packet_length( PACKET *pkt )
109 {
110     u32 n=0;
111
112     assert( pkt->pkt.generic );
113     switch( pkt->pkttype ) {
114       case PKT_PLAINTEXT:
115         n = calc_plaintext( pkt->pkt.plaintext );
116         break;
117       case PKT_USER_ID:
118       case PKT_COMMENT:
119       case PKT_PUBKEY_CERT:
120       case PKT_SECKEY_CERT:
121       case PKT_PUBKEY_ENC:
122       case PKT_ENCR_DATA:
123       case PKT_SIGNATURE:
124       case PKT_RING_TRUST:
125       case PKT_COMPR_DATA:
126       default:
127         log_bug("invalid packet type in calc_packet_length()");
128         break;
129     }
130     n += calc_header_length(n);
131     return n;
132 }
133
134
135 static int
136 do_comment( IOBUF out, int ctb, PKT_comment *rem )
137 {
138     write_header(out, ctb, rem->len);
139     if( iobuf_write( out, rem->data, rem->len ) )
140         return G10ERR_WRITE_FILE;
141     return 0;
142 }
143
144 static int
145 do_user_id( IOBUF out, int ctb, PKT_user_id *uid )
146 {
147     write_header(out, ctb, uid->len);
148     if( iobuf_write( out, uid->name, uid->len ) )
149         return G10ERR_WRITE_FILE;
150     return 0;
151 }
152
153 static int
154 do_pubkey_cert( IOBUF out, int ctb, PKT_pubkey_cert *pkc )
155 {
156     int rc = 0;
157     IOBUF a = iobuf_temp();
158
159     write_version( a, ctb );
160     write_32(a, pkc->timestamp );
161     write_16(a, pkc->valid_days );
162     iobuf_put(a, pkc->pubkey_algo );
163     if( pkc->pubkey_algo == PUBKEY_ALGO_RSA ) {
164         mpi_encode(a, pkc->d.rsa.rsa_n );
165         mpi_encode(a, pkc->d.rsa.rsa_e );
166     }
167     else {
168         rc = G10ERR_PUBKEY_ALGO;
169         goto leave;
170     }
171
172     write_header(out, ctb, iobuf_get_temp_length(a) );
173     if( iobuf_write_temp( out, a ) )
174         rc = G10ERR_WRITE_FILE;
175
176   leave:
177     iobuf_close(a);
178     return rc;
179 }
180
181 static int
182 do_seckey_cert( IOBUF out, int ctb, PKT_seckey_cert *skc )
183 {
184     int rc = 0;
185     IOBUF a = iobuf_temp();
186
187     write_version( a, ctb );
188     write_32(a, skc->timestamp );
189     write_16(a, skc->valid_days );
190     iobuf_put(a, skc->pubkey_algo );
191     if( skc->pubkey_algo == PUBKEY_ALGO_RSA ) {
192         mpi_encode(a, skc->d.rsa.rsa_n );
193         mpi_encode(a, skc->d.rsa.rsa_e );
194         iobuf_put(a, skc->d.rsa.protect_algo );
195         skc->d.rsa.calc_csum = 0;
196         if( skc->d.rsa.protect_algo ) {
197             assert( skc->d.rsa.is_protected == 1 );
198             assert( skc->d.rsa.protect_algo == CIPHER_ALGO_BLOWFISH );
199             iobuf_write(a, skc->d.rsa.protect.blowfish.iv, 8 );
200
201             mpi_write_csum(a, (byte*)skc->d.rsa.rsa_d, &skc->d.rsa.calc_csum );
202             mpi_write_csum(a, (byte*)skc->d.rsa.rsa_p, &skc->d.rsa.calc_csum );
203             mpi_write_csum(a, (byte*)skc->d.rsa.rsa_q, &skc->d.rsa.calc_csum );
204             mpi_write_csum(a, (byte*)skc->d.rsa.rsa_u, &skc->d.rsa.calc_csum );
205         }
206         else {  /* Not protected: You fool you! */
207             assert( !skc->d.rsa.is_protected );
208             mpi_encode_csum(a, skc->d.rsa.rsa_d, &skc->d.rsa.calc_csum );
209             mpi_encode_csum(a, skc->d.rsa.rsa_p, &skc->d.rsa.calc_csum );
210             mpi_encode_csum(a, skc->d.rsa.rsa_q, &skc->d.rsa.calc_csum );
211             mpi_encode_csum(a, skc->d.rsa.rsa_u, &skc->d.rsa.calc_csum );
212         }
213
214         write_16(a, skc->d.rsa.calc_csum );
215     }
216     else {
217         rc = G10ERR_PUBKEY_ALGO;
218         goto leave;
219     }
220
221     write_header(out, ctb, iobuf_get_temp_length(a) );
222     if( iobuf_write_temp( out, a ) )
223         rc = G10ERR_WRITE_FILE;
224
225   leave:
226     iobuf_close(a);
227     return rc;
228 }
229
230 static int
231 do_pubkey_enc( IOBUF out, int ctb, PKT_pubkey_enc *enc )
232 {
233     int rc = 0;
234     IOBUF a = iobuf_temp();
235
236     write_version( a, ctb );
237     write_32(a, enc->keyid[0] );
238     write_32(a, enc->keyid[1] );
239     iobuf_put(a,enc->pubkey_algo );
240     if( enc->pubkey_algo == PUBKEY_ALGO_RSA ) {
241         mpi_encode(a, enc->d.rsa.rsa_integer );
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
258 static u32
259 calc_plaintext( PKT_plaintext *pt )
260 {
261     return pt->len? (1 + 1 + pt->namelen + 4 + pt->len) : 0;
262 }
263
264 static int
265 do_plaintext( IOBUF out, int ctb, PKT_plaintext *pt )
266 {
267     int c, i, rc = 0;
268     u32 n;
269
270     write_header(out, ctb, calc_plaintext( pt ) );
271     iobuf_put(out, pt->mode );
272     iobuf_put(out, pt->namelen );
273     for(i=0; i < pt->namelen; i++ )
274         iobuf_put(out, pt->name[i] );
275     if( write_32(out, pt->timestamp ) )
276         rc = G10ERR_WRITE_FILE;
277
278     n = 0;
279     while( (c=iobuf_get(pt->buf)) != -1 ) {
280         if( iobuf_put(out, c) ) {
281             rc = G10ERR_WRITE_FILE;
282             break;
283         }
284         n++;
285     }
286     if( !pt->len )
287         iobuf_set_block_mode(out, 0 ); /* write end marker */
288     else if( n != pt->len )
289         log_error("do_plaintext(): wrote %lu bytes but expected %lu bytes\n",
290                         (ulong)n, (ulong)pt->len );
291
292     return rc;
293 }
294
295
296
297 static int
298 do_encr_data( IOBUF out, int ctb, PKT_encr_data *ed )
299 {
300     int rc = 0;
301     u32 n;
302
303     n = ed->len ? (ed->len + 10) : 0;
304     write_header(out, ctb, n );
305
306     /* This is all. The caller has to write the real data */
307
308     return rc;
309 }
310
311 static int
312 do_compressed( IOBUF out, int ctb, PKT_compressed *cd )
313 {
314     int rc = 0;
315
316     /* we must use the old convention and don't use blockmode */
317     write_header2(out, ctb, 0, 0 );
318     iobuf_put(out, cd->algorithm );
319
320     /* This is all. The caller has to write the real data */
321
322     return rc;
323 }
324
325
326
327
328 static int
329 write_16(IOBUF out, u16 a)
330 {
331     iobuf_put(out, a>>8);
332     if( iobuf_put(out,a) )
333         return -1;
334     return 0;
335 }
336
337 static int
338 write_32(IOBUF out, u32 a)
339 {
340     iobuf_put(out, a>> 24);
341     iobuf_put(out, a>> 16);
342     iobuf_put(out, a>> 8);
343     if( iobuf_put(out, a) )
344         return -1;
345     return 0;
346 }
347
348
349 /****************
350  * calculate the length of a header
351  */
352 static int
353 calc_header_length( u32 len )
354 {
355     if( !len )
356         return 1; /* only the ctb */
357     else if( len < 256 )
358         return 2;
359     else if( len < 65536 )
360         return 3;
361     else
362         return 5;
363 }
364
365 /****************
366  * Write the CTB and the packet length
367  */
368 static int
369 write_header( IOBUF out, int ctb, u32 len )
370 {
371     return write_header2( out, ctb, len, 1 );
372 }
373
374 static int
375 write_header2( IOBUF out, int ctb, u32 len, int blkmode )
376 {
377     if( !len )
378         ctb |= 3;
379     else if( len < 256 )
380         ;
381     else if( len < 65536 )
382         ctb |= 1;
383     else
384         ctb |= 2;
385     if( iobuf_put(out, ctb ) )
386         return -1;
387     if( !len ) {
388         if( blkmode )
389             iobuf_set_block_mode(out, 8196 );
390     }
391     else {
392         if( ctb & 2 ) {
393             iobuf_put(out, len >> 24 );
394             iobuf_put(out, len >> 16 );
395         }
396         if( ctb & 3 )
397             iobuf_put(out, len >> 8 );
398         if( iobuf_put(out, len ) )
399             return -1;
400     }
401     return 0;
402 }
403
404 static int
405 write_version( IOBUF out, int ctb )
406 {
407     if( iobuf_put( out, 3 ) )
408         return -1;
409     return 0;
410 }
411