edit-key is now complete
[gnupg.git] / g10 / build-packet.c
1 /* build-packet.c - assemble packets and write them
2  *      Copyright (C) 1998 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
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_key( IOBUF out, int ctb, PKT_public_key *pk );
40 static int do_secret_key( IOBUF out, int ctb, PKT_secret_key *pk );
41 static int do_symkey_enc( IOBUF out, int ctb, PKT_symkey_enc *enc );
42 static int do_pubkey_enc( IOBUF out, int ctb, PKT_pubkey_enc *enc );
43 static u32 calc_plaintext( PKT_plaintext *pt );
44 static int do_plaintext( IOBUF out, int ctb, PKT_plaintext *pt );
45 static int do_encrypted( IOBUF out, int ctb, PKT_encrypted *ed );
46 static int do_compressed( IOBUF out, int ctb, PKT_compressed *cd );
47 static int do_signature( IOBUF out, int ctb, PKT_signature *sig );
48 static int do_onepass_sig( IOBUF out, int ctb, PKT_onepass_sig *ops );
49
50 static int calc_header_length( u32 len );
51 static int write_16(IOBUF inp, u16 a);
52 static int write_32(IOBUF inp, u32 a);
53 static int write_header( IOBUF out, int ctb, u32 len );
54 static int write_sign_packet_header( IOBUF out, int ctb, u32 len );
55 static int write_header2( IOBUF out, int ctb, u32 len, int hdrlen, int blkmode );
56 static int write_new_header( IOBUF out, int ctb, u32 len, int hdrlen );
57 static int write_version( IOBUF out, int ctb );
58
59 /****************
60  * Build a packet and write it to INP
61  * Returns: 0 := okay
62  *         >0 := error
63  * Note: Caller must free the packet
64  */
65 int
66 build_packet( IOBUF out, PACKET *pkt )
67 {
68     int new_ctb=0, rc=0, ctb;
69
70     if( DBG_PACKET )
71         log_debug("build_packet() type=%d\n", pkt->pkttype );
72     assert( pkt->pkt.generic );
73
74     switch( pkt->pkttype ) {
75       case PKT_OLD_COMMENT: pkt->pkttype = PKT_COMMENT; break;
76       case PKT_PLAINTEXT: new_ctb = pkt->pkt.plaintext->new_ctb; break;
77       case PKT_ENCRYPTED: new_ctb = pkt->pkt.encrypted->new_ctb; break;
78       case PKT_COMPRESSED:new_ctb = pkt->pkt.compressed->new_ctb; break;
79       default: break;
80     }
81
82     if( new_ctb || pkt->pkttype > 15 ) /* new format */
83         ctb = 0xc0 | (pkt->pkttype & 0x3f);
84     else
85         ctb = 0x80 | ((pkt->pkttype & 15)<<2);
86     switch( pkt->pkttype ) {
87       case PKT_USER_ID:
88         rc = do_user_id( out, ctb, pkt->pkt.user_id );
89         break;
90       case PKT_COMMENT:
91         rc = do_comment( out, ctb, pkt->pkt.comment );
92         break;
93       case PKT_PUBLIC_SUBKEY:
94       case PKT_PUBLIC_KEY:
95         rc = do_public_key( out, ctb, pkt->pkt.public_key );
96         break;
97       case PKT_SECRET_SUBKEY:
98       case PKT_SECRET_KEY:
99         rc = do_secret_key( out, ctb, pkt->pkt.secret_key );
100         break;
101       case PKT_SYMKEY_ENC:
102         rc = do_symkey_enc( out, ctb, pkt->pkt.symkey_enc );
103         break;
104       case PKT_PUBKEY_ENC:
105         rc = do_pubkey_enc( out, ctb, pkt->pkt.pubkey_enc );
106         break;
107       case PKT_PLAINTEXT:
108         rc = do_plaintext( out, ctb, pkt->pkt.plaintext );
109         break;
110       case PKT_ENCRYPTED:
111         rc = do_encrypted( out, ctb, pkt->pkt.encrypted );
112         break;
113       case PKT_COMPRESSED:
114         rc = do_compressed( out, ctb, pkt->pkt.compressed );
115         break;
116       case PKT_SIGNATURE:
117         rc = do_signature( out, ctb, pkt->pkt.signature );
118         break;
119       case PKT_ONEPASS_SIG:
120         rc = do_onepass_sig( out, ctb, pkt->pkt.onepass_sig );
121         break;
122       case PKT_RING_TRUST:
123       default:
124         log_bug("invalid packet type in build_packet()\n");
125         break;
126     }
127
128     return rc;
129 }
130
131 /****************
132  * calculate the length of a packet described by PKT
133  */
134 u32
135 calc_packet_length( PACKET *pkt )
136 {
137     u32 n=0;
138
139     assert( pkt->pkt.generic );
140     switch( pkt->pkttype ) {
141       case PKT_PLAINTEXT:
142         n = calc_plaintext( pkt->pkt.plaintext );
143         break;
144       case PKT_USER_ID:
145       case PKT_COMMENT:
146       case PKT_PUBLIC_KEY:
147       case PKT_SECRET_KEY:
148       case PKT_SYMKEY_ENC:
149       case PKT_PUBKEY_ENC:
150       case PKT_ENCRYPTED:
151       case PKT_SIGNATURE:
152       case PKT_ONEPASS_SIG:
153       case PKT_RING_TRUST:
154       case PKT_COMPRESSED:
155       default:
156         log_bug("invalid packet type in calc_packet_length()");
157         break;
158     }
159     n += calc_header_length(n);
160     return n;
161 }
162
163 static void
164 write_fake_data( IOBUF out, MPI a )
165 {
166     byte *s;
167     u16 len;
168
169     if( a ) {
170         s = (byte*)a;
171         len = (s[0] << 8) | s[1];
172         iobuf_write( out, s+2, len );
173     }
174 }
175
176
177 static int
178 do_comment( IOBUF out, int ctb, PKT_comment *rem )
179 {
180     if( !opt.no_comment ) {
181         write_header(out, ctb, rem->len);
182         if( iobuf_write( out, rem->data, rem->len ) )
183             return G10ERR_WRITE_FILE;
184     }
185     return 0;
186 }
187
188 static int
189 do_user_id( IOBUF out, int ctb, PKT_user_id *uid )
190 {
191     write_header(out, ctb, uid->len);
192     if( iobuf_write( out, uid->name, uid->len ) )
193         return G10ERR_WRITE_FILE;
194     return 0;
195 }
196
197 static int
198 do_public_key( IOBUF out, int ctb, PKT_public_key *pk )
199 {
200     int rc = 0;
201     int n, i;
202     IOBUF a = iobuf_temp();
203
204     if( !pk->version )
205         iobuf_put( a, 3 );
206     else
207         iobuf_put( a, pk->version );
208     write_32(a, pk->timestamp );
209     if( pk->version < 4 )
210         write_16(a, pk->valid_days );
211     iobuf_put(a, pk->pubkey_algo );
212     n = pubkey_get_npkey( pk->pubkey_algo );
213     if( !n )
214         write_fake_data( a, pk->pkey[0] );
215     for(i=0; i < n; i++ )
216         mpi_write(a, pk->pkey[i] );
217
218     write_header2(out, ctb, iobuf_get_temp_length(a), pk->hdrbytes, 1 );
219     if( iobuf_write_temp( out, a ) )
220         rc = G10ERR_WRITE_FILE;
221
222     iobuf_close(a);
223     return rc;
224 }
225
226
227 /****************
228  * Make a hash value from the public key certificate
229  */
230 void
231 hash_public_key( MD_HANDLE md, PKT_public_key *pk )
232 {
233     PACKET pkt;
234     int rc = 0;
235     int c;
236     IOBUF a = iobuf_temp();
237   #if 0
238     FILE *fp = fopen("dump.pk", "a");
239     int i=0;
240
241     fprintf(fp, "\nHashing PK (v%d):\n", pk->version);
242   #endif
243
244     /* build the packet */
245     init_packet(&pkt);
246     pkt.pkttype = PKT_PUBLIC_KEY;
247     pkt.pkt.public_key = pk;
248     if( (rc = build_packet( a, &pkt )) )
249         log_fatal("build public_key for hashing failed: %s\n", g10_errstr(rc));
250     while( (c=iobuf_get(a)) != -1 ) {
251       #if 0
252         fprintf( fp," %02x", c );
253         if( (++i == 24) ) {
254             putc('\n', fp);
255             i=0;
256         }
257       #endif
258         md_putc( md, c );
259     }
260   #if 0
261     putc('\n', fp);
262     fclose(fp);
263   #endif
264     iobuf_cancel(a);
265 }
266
267
268 static int
269 do_secret_key( IOBUF out, int ctb, PKT_secret_key *sk )
270 {
271     int rc = 0;
272     int i, nskey, npkey;
273     IOBUF a = iobuf_temp();
274
275     if( !sk->version )
276         iobuf_put( a, 3 );
277     else
278         iobuf_put( a, sk->version );
279     write_32(a, sk->timestamp );
280     if( sk->version < 4 )
281         write_16(a, sk->valid_days );
282     iobuf_put(a, sk->pubkey_algo );
283     nskey = pubkey_get_nskey( sk->pubkey_algo );
284     npkey = pubkey_get_npkey( sk->pubkey_algo );
285     if( !npkey ) {
286         write_fake_data( a, sk->skey[0] );
287         goto leave;
288     }
289     assert( npkey < nskey );
290
291     for(i=0; i < npkey; i++ )
292         mpi_write(a, sk->skey[i] );
293     if( sk->is_protected ) {
294         if( is_RSA(sk->pubkey_algo) && sk->version < 4 ) {
295             iobuf_put(a, sk->protect.algo );
296             iobuf_write(a, sk->protect.iv, 8 );
297         }
298         else {
299             iobuf_put(a, 0xff );
300             iobuf_put(a, sk->protect.algo );
301             iobuf_put(a, sk->protect.s2k.mode );
302             iobuf_put(a, sk->protect.s2k.hash_algo );
303             if( sk->protect.s2k.mode == 1
304                 || sk->protect.s2k.mode == 4 )
305                 iobuf_write(a, sk->protect.s2k.salt, 8 );
306             if( sk->protect.s2k.mode == 4 )
307                 write_32(a, sk->protect.s2k.count );
308             iobuf_write(a, sk->protect.iv, 8 );
309         }
310     }
311     else
312         iobuf_put(a, 0 );
313     for(   ; i < nskey; i++ )
314         mpi_write(a, sk->skey[i] );
315     write_16(a, sk->csum );
316
317   leave:
318     write_header2(out, ctb, iobuf_get_temp_length(a), sk->hdrbytes, 1 );
319     if( iobuf_write_temp( out, a ) )
320         rc = G10ERR_WRITE_FILE;
321
322     iobuf_close(a);
323     return rc;
324 }
325
326 static int
327 do_symkey_enc( IOBUF out, int ctb, PKT_symkey_enc *enc )
328 {
329     int rc = 0;
330     IOBUF a = iobuf_temp();
331
332     assert( enc->version == 4 );
333     switch( enc->s2k.mode ) {
334       case 0: case 1: case 4: break;
335       default: log_bug("do_symkey_enc: s2k=%d\n", enc->s2k.mode );
336     }
337     iobuf_put( a, enc->version );
338     iobuf_put( a, enc->cipher_algo );
339     iobuf_put( a, enc->s2k.mode );
340     iobuf_put( a, enc->s2k.hash_algo );
341     if( enc->s2k.mode == 1 || enc->s2k.mode == 4 ) {
342         iobuf_write(a, enc->s2k.salt, 8 );
343         if( enc->s2k.mode == 4 )
344             write_32(a, enc->s2k.count);
345     }
346     if( enc->seskeylen )
347         iobuf_write(a, enc->seskey, enc->seskeylen );
348
349     write_header(out, ctb, iobuf_get_temp_length(a) );
350     if( iobuf_write_temp( out, a ) )
351         rc = G10ERR_WRITE_FILE;
352
353     iobuf_close(a);
354     return rc;
355 }
356
357
358
359
360 static int
361 do_pubkey_enc( IOBUF out, int ctb, PKT_pubkey_enc *enc )
362 {
363     int rc = 0;
364     int n, i;
365     IOBUF a = iobuf_temp();
366
367     write_version( a, ctb );
368     write_32(a, enc->keyid[0] );
369     write_32(a, enc->keyid[1] );
370     iobuf_put(a,enc->pubkey_algo );
371     n = pubkey_get_nenc( enc->pubkey_algo );
372     if( !n )
373         write_fake_data( a, enc->data[0] );
374     for(i=0; i < n; i++ )
375         mpi_write(a, enc->data[i] );
376
377     write_header(out, ctb, iobuf_get_temp_length(a) );
378     if( iobuf_write_temp( out, a ) )
379         rc = G10ERR_WRITE_FILE;
380
381     iobuf_close(a);
382     return rc;
383 }
384
385
386
387
388 static u32
389 calc_plaintext( PKT_plaintext *pt )
390 {
391     return pt->len? (1 + 1 + pt->namelen + 4 + pt->len) : 0;
392 }
393
394 static int
395 do_plaintext( IOBUF out, int ctb, PKT_plaintext *pt )
396 {
397     int i, rc = 0;
398     u32 n;
399     byte buf[1000]; /* FIXME: this buffer has the plaintext! */
400     int nbytes;
401
402     write_header(out, ctb, calc_plaintext( pt ) );
403     iobuf_put(out, pt->mode );
404     iobuf_put(out, pt->namelen );
405     for(i=0; i < pt->namelen; i++ )
406         iobuf_put(out, pt->name[i] );
407     if( write_32(out, pt->timestamp ) )
408         rc = G10ERR_WRITE_FILE;
409
410     n = 0;
411     while( (nbytes=iobuf_read(pt->buf, buf, 1000)) != -1 ) {
412         if( iobuf_write(out, buf, nbytes) == -1 ) {
413             rc = G10ERR_WRITE_FILE;
414             break;
415         }
416         n += nbytes;
417     }
418     memset(buf,0,1000); /* at least burn the buffer */
419     if( !pt->len )
420         iobuf_set_block_mode(out, 0 ); /* write end marker */
421     else if( n != pt->len )
422         log_error("do_plaintext(): wrote %lu bytes but expected %lu bytes\n",
423                         (ulong)n, (ulong)pt->len );
424
425     return rc;
426 }
427
428
429
430 static int
431 do_encrypted( IOBUF out, int ctb, PKT_encrypted *ed )
432 {
433     int rc = 0;
434     u32 n;
435
436     n = ed->len ? (ed->len + 10) : 0;
437     write_header(out, ctb, n );
438
439     /* This is all. The caller has to write the real data */
440
441     return rc;
442 }
443
444 static int
445 do_compressed( IOBUF out, int ctb, PKT_compressed *cd )
446 {
447     int rc = 0;
448
449     /* we must use the old convention and don't use blockmode */
450     write_header2(out, ctb, 0, 0, 0 );
451     iobuf_put(out, cd->algorithm );
452
453     /* This is all. The caller has to write the real data */
454
455     return rc;
456 }
457
458
459
460 /****************
461  * Find a subpacket of type REQTYPE in BUFFER and a return a pointer
462  * to the first byte of that subpacket data.
463  * And return the length of the packet in RET_N and the number of
464  * header bytes in RET_HLEN (length header and type byte).
465  */
466 byte *
467 find_subpkt( byte *buffer, sigsubpkttype_t reqtype,
468              size_t *ret_hlen, size_t *ret_n )
469 {
470     int buflen;
471     sigsubpkttype_t type;
472     byte *bufstart;
473     size_t n;
474
475     if( !buffer )
476         return NULL;
477     buflen = (*buffer << 8) | buffer[1];
478     buffer += 2;
479     for(;;) {
480         if( !buflen )
481             return NULL; /* end of packets; not found */
482         bufstart = buffer;
483         n = *buffer++; buflen--;
484         if( n == 255 ) {
485             if( buflen < 4 )
486                 break;
487             n = (buffer[0] << 24) | (buffer[1] << 16)
488                                   | (buffer[2] << 8) | buffer[3];
489             buffer += 4;
490             buflen -= 4;
491         }
492         else if( n >= 192 ) {
493             if( buflen < 2 )
494                 break;
495             n = (( n - 192 ) << 8) + *buffer + 192;
496             buflen--;
497         }
498         if( buflen < n )
499             break;
500         type = *buffer & 0x7f;
501         if( type == reqtype ) {
502             buffer++;
503             n--;
504             if( n > buflen )
505                 break;
506             if( ret_hlen )
507                 *ret_hlen = buffer - bufstart;
508             if( ret_n )
509                 *ret_n = n;
510             return buffer;
511         }
512         buffer += n; buflen -=n;
513     }
514
515     log_error("find_subpkt: buffer shorter than subpacket\n");
516     return NULL;
517 }
518
519
520 /****************
521  * Create or update a signature subpacket for SIG of TYPE.
522  * This functions know, where to put the data (hashed or unhashed).
523  * The function may move data from the unhased part to the hashed one.
524  * Note: All pointers into sig->[un]hashed are not valid after a call
525  * to this function.  The data to but into the subpaket should be
526  * in buffer with a length of buflen.
527  */
528 void
529 build_sig_subpkt( PKT_signature *sig, sigsubpkttype_t type,
530                   const byte *buffer, size_t buflen )
531 {
532
533     byte *data;
534     size_t hlen, dlen;
535     int found, hashed, realloced;
536     size_t n, n0;
537
538     if( (data = find_subpkt( sig->hashed_data, type, &hlen, &dlen )) )
539         found = 1;
540     else if( (data = find_subpkt( sig->unhashed_data, type, &hlen, &dlen )))
541         found = 2;
542     else
543         found = 0;
544
545     if( found )
546         log_bug("build_sig_packet: update nyi\n");
547     if( buflen+1 >= 192 )
548         log_bug("build_sig_packet: long subpackets are nyi\n");
549
550     switch( type ) {
551       case SIGSUBPKT_SIG_CREATED:
552       case SIGSUBPKT_PRIV_ADD_SIG:
553       case SIGSUBPKT_PREF_SYM:
554       case SIGSUBPKT_PREF_HASH:
555       case SIGSUBPKT_PREF_COMPR:
556       case SIGSUBPKT_KS_FLAGS:
557       case SIGSUBPKT_KEY_EXPIRE:
558                hashed = 1; break;
559       default: hashed = 0; break;
560     }
561
562     if( hashed ) {
563         n0 = sig->hashed_data ? ((*sig->hashed_data << 8)
564                                     | sig->hashed_data[1]) : 0;
565         n = n0 + 1 + 1 + buflen; /* length, type, buffer */
566         realloced = !!sig->hashed_data;
567         data = sig->hashed_data ? m_realloc( sig->hashed_data, n+2 )
568                                 : m_alloc( n+2 );
569     }
570     else {
571         n0 = sig->unhashed_data ? ((*sig->unhashed_data << 8)
572                                       | sig->unhashed_data[1]) : 0;
573         n = n0 + 1 + 1 + buflen; /* length, type, buffer */
574         realloced = !!sig->unhashed_data;
575         data = sig->unhashed_data ? m_realloc( sig->unhashed_data, n+2 )
576                                   : m_alloc( n+2 );
577     }
578
579     data[0] = (n >> 8) & 0xff;
580     data[1] = n & 0xff;
581     data[n0+2] = buflen+1;
582     data[n0+3] = type;
583     memcpy(data+n0+4, buffer, buflen );
584
585     if( hashed ) {
586         if( !realloced )
587             m_free(sig->hashed_data);
588         sig->hashed_data = data;
589     }
590     else {
591         if( !realloced )
592             m_free(sig->unhashed_data);
593         sig->unhashed_data = data;
594     }
595 }
596
597
598 /****************
599  * Put all the required stuff from SIG into subpackets of sig.
600  */
601 void
602 build_sig_subpkt_from_sig( PKT_signature *sig )
603 {
604     u32  u;
605     byte buf[8];
606
607     u = sig->keyid[0];
608     buf[0] = (u >> 24) & 0xff;
609     buf[1] = (u >> 16) & 0xff;
610     buf[2] = (u >>  8) & 0xff;
611     buf[3] = u & 0xff;
612     u = sig->keyid[1];
613     buf[4] = (u >> 24) & 0xff;
614     buf[5] = (u >> 16) & 0xff;
615     buf[6] = (u >>  8) & 0xff;
616     buf[7] = u & 0xff;
617     build_sig_subpkt( sig, SIGSUBPKT_ISSUER, buf, 8 );
618
619     u = sig->timestamp;
620     buf[0] = (u >> 24) & 0xff;
621     buf[1] = (u >> 16) & 0xff;
622     buf[2] = (u >>  8) & 0xff;
623     buf[3] = u & 0xff;
624     build_sig_subpkt( sig, SIGSUBPKT_SIG_CREATED, buf, 4 );
625 }
626
627
628 static int
629 do_signature( IOBUF out, int ctb, PKT_signature *sig )
630 {
631     int rc = 0;
632     int n, i;
633     IOBUF a = iobuf_temp();
634
635     if( !sig->version )
636         iobuf_put( a, 3 );
637     else
638         iobuf_put( a, sig->version );
639     if( sig->version < 4 )
640         iobuf_put(a, 5 ); /* constant */
641     iobuf_put(a, sig->sig_class );
642     if( sig->version < 4 ) {
643         write_32(a, sig->timestamp );
644         write_32(a, sig->keyid[0] );
645         write_32(a, sig->keyid[1] );
646     }
647     iobuf_put(a, sig->pubkey_algo );
648     iobuf_put(a, sig->digest_algo );
649     if( sig->version >= 4 ) {
650         size_t n;
651         /* timestamp and keyid must have been packed into the
652          * subpackets prior to the call of this function, because
653          * these subpackets are hashed */
654         n = sig->hashed_data?((sig->hashed_data[0]<<8)
655                               |sig->hashed_data[1])   :0;
656         write_16(a, n);
657         if( n )
658             iobuf_write( a, sig->hashed_data+2, n );
659         n = sig->unhashed_data?((sig->unhashed_data[0]<<8)
660                                 |sig->unhashed_data[1])   :0;
661         write_16(a, n);
662         if( n )
663             iobuf_write( a, sig->unhashed_data+2, n );
664     }
665     iobuf_put(a, sig->digest_start[0] );
666     iobuf_put(a, sig->digest_start[1] );
667     n = pubkey_get_nsig( sig->pubkey_algo );
668     if( !n )
669         write_fake_data( a, sig->data[0] );
670     for(i=0; i < n; i++ )
671         mpi_write(a, sig->data[i] );
672
673     if( is_RSA(sig->pubkey_algo) && sig->version < 4 )
674         write_sign_packet_header(out, ctb, iobuf_get_temp_length(a) );
675     else
676         write_header(out, ctb, iobuf_get_temp_length(a) );
677     if( iobuf_write_temp( out, a ) )
678         rc = G10ERR_WRITE_FILE;
679
680     iobuf_close(a);
681     return rc;
682 }
683
684
685 static int
686 do_onepass_sig( IOBUF out, int ctb, PKT_onepass_sig *ops )
687 {
688     int rc = 0;
689     IOBUF a = iobuf_temp();
690
691     write_version( a, ctb );
692     iobuf_put(a, ops->sig_class );
693     iobuf_put(a, ops->digest_algo );
694     iobuf_put(a, ops->pubkey_algo );
695     write_32(a, ops->keyid[0] );
696     write_32(a, ops->keyid[1] );
697     iobuf_put(a, ops->last );
698
699     write_header(out, ctb, iobuf_get_temp_length(a) );
700     if( iobuf_write_temp( out, a ) )
701         rc = G10ERR_WRITE_FILE;
702
703     iobuf_close(a);
704     return rc;
705 }
706
707
708 static int
709 write_16(IOBUF out, u16 a)
710 {
711     iobuf_put(out, a>>8);
712     if( iobuf_put(out,a) )
713         return -1;
714     return 0;
715 }
716
717 static int
718 write_32(IOBUF out, u32 a)
719 {
720     iobuf_put(out, a>> 24);
721     iobuf_put(out, a>> 16);
722     iobuf_put(out, a>> 8);
723     if( iobuf_put(out, a) )
724         return -1;
725     return 0;
726 }
727
728
729 /****************
730  * calculate the length of a header
731  */
732 static int
733 calc_header_length( u32 len )
734 {
735     if( !len )
736         return 1; /* only the ctb */
737     else if( len < 256 )
738         return 2;
739     else if( len < 65536 )
740         return 3;
741     else
742         return 5;
743 }
744
745 /****************
746  * Write the CTB and the packet length
747  */
748 static int
749 write_header( IOBUF out, int ctb, u32 len )
750 {
751     return write_header2( out, ctb, len, 0, 1 );
752 }
753
754
755 static int
756 write_sign_packet_header( IOBUF out, int ctb, u32 len )
757 {
758     /* work around a bug in the pgp read function for signature packets,
759      * which are not correctly coded and silently assume at some
760      * point 2 byte length headers.*/
761     iobuf_put(out, 0x89 );
762     iobuf_put(out, len >> 8 );
763     return iobuf_put(out, len ) == -1 ? -1:0;
764 }
765
766 /****************
767  * if HDRLEN is > 0, try to build a header of this length.
768  * we need this, so that we can hash packets without reading them again.
769  */
770 static int
771 write_header2( IOBUF out, int ctb, u32 len, int hdrlen, int blkmode )
772 {
773     if( ctb & 0x40 )
774         return write_new_header( out, ctb, len, hdrlen );
775
776     if( hdrlen ) {
777         if( !len )
778             ctb |= 3;
779         else if( hdrlen == 2 && len < 256 )
780             ;
781         else if( hdrlen == 3 && len < 65536 )
782             ctb |= 1;
783         else
784             ctb |= 2;
785     }
786     else {
787         if( !len )
788             ctb |= 3;
789         else if( len < 256 )
790             ;
791         else if( len < 65536 )
792             ctb |= 1;
793         else
794             ctb |= 2;
795     }
796     if( iobuf_put(out, ctb ) )
797         return -1;
798     if( !len ) {
799         if( blkmode )
800             iobuf_set_block_mode(out, 8196 );
801     }
802     else {
803         if( ctb & 2 ) {
804             iobuf_put(out, len >> 24 );
805             iobuf_put(out, len >> 16 );
806         }
807         if( ctb & 3 )
808             iobuf_put(out, len >> 8 );
809         if( iobuf_put(out, len ) )
810             return -1;
811     }
812     return 0;
813 }
814
815
816 static int
817 write_new_header( IOBUF out, int ctb, u32 len, int hdrlen )
818 {
819     if( hdrlen )
820         log_bug("can't cope with hdrlen yet\n");
821
822     if( iobuf_put(out, ctb ) )
823         return -1;
824     if( !len ) {
825         iobuf_set_partial_block_mode(out, 512 );
826     }
827     else {
828         if( len < 192 ) {
829             if( iobuf_put(out, len ) )
830                 return -1;
831         }
832         else if( len < 8384 ) {
833             len -= 192;
834             if( iobuf_put( out, (len / 256) + 192) )
835                 return -1;
836             if( iobuf_put( out, (len % 256) )  )
837                 return -1;
838         }
839         else {
840             if( iobuf_put( out, 0xff ) )
841                 return -1;
842             if( iobuf_put( out, (len >> 24)&0xff ) )
843                 return -1;
844             if( iobuf_put( out, (len >> 16)&0xff ) )
845                 return -1;
846             if( iobuf_put( out, (len >> 8)&0xff )  )
847                 return -1;
848             if( iobuf_put( out, len & 0xff ) )
849                 return -1;
850         }
851     }
852     return 0;
853 }
854
855 static int
856 write_version( IOBUF out, int ctb )
857 {
858     if( iobuf_put( out, 3 ) )
859         return -1;
860     return 0;
861 }
862