Fix gpg-preset-passphrase bug.
[gnupg.git] / common / iobuf.c
1 /* iobuf.c  -  File Handling for OpenPGP.
2  * Copyright (C) 1998, 1999, 2000, 2001, 2003, 2004, 2006,
3  *               2007, 2008  Free Software Foundation, Inc.
4  *
5  * This file is part of GnuPG.
6  *
7  * GnuPG is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * GnuPG is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #include <config.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <errno.h>
26 #include <ctype.h>
27 #include <assert.h>
28 #include <sys/types.h>
29 #include <sys/stat.h>
30 #include <fcntl.h>
31 #include <unistd.h>
32 #ifdef HAVE_W32_SYSTEM
33 # include <windows.h>
34 #endif
35 #ifdef __riscos__
36 # include <kernel.h>
37 # include <swis.h>
38 #endif /* __riscos__ */
39
40 #include "util.h"
41 #include "sysutils.h"
42 #include "iobuf.h"
43
44 /*-- Begin configurable part.  --*/
45
46 /* The size of the internal buffers. 
47    NOTE: If you change this value you MUST also adjust the regression
48    test "armored_key_8192" in armor.test! */
49 #define IOBUF_BUFFER_SIZE  8192
50
51 /* We don't want to use the STDIO based backend.  */
52 #undef FILE_FILTER_USES_STDIO
53
54 /*-- End configurable part.  --*/
55
56
57 /* Under W32 the default is to use the setmode call.  Define a macro
58    which allows us to enable this call.  */
59 #ifdef HAVE_W32_SYSTEM
60 # define USE_SETMODE 1
61 #endif /*HAVE_W32_SYSTEM*/
62
63
64 /* Definition of constants and macros used by our file filter
65    implementation.  What we define here are 3 macros to make the
66    appropriate calls:
67
68    my_fileno 
69      Is expanded to fileno(a) if using a stdion backend and to a if we
70      are using the low-level backend.
71
72    my_fopen 
73      Is defined to fopen for the stdio backend and to direct_open if
74      we are using the low-evel backend.
75
76    my_fopen_ro 
77      Is defined to fopen for the stdio backend and to fd_cache_open if
78      we are using the low-evel backend.
79
80    fp_or_fd_t
81      Is the type we use for the backend stream or file descriptor.
82
83    INVALID_FP, FILEP_OR_FD_FOR_STDIN, FILEP_OR_FD_FOR_STDOUT
84      Are macros defined depending on the used backend.
85
86 */
87 #ifdef FILE_FILTER_USES_STDIO
88 # define my_fileno(a)     fileno ((a))
89 # define my_fopen_ro(a,b) fopen ((a),(b))
90 # define my_fopen(a,b)    fopen ((a),(b))
91   typedef FILE *fp_or_fd_t;
92 # define INVALID_FP              NULL
93 # define FILEP_OR_FD_FOR_STDIN   (stdin)
94 # define FILEP_OR_FD_FOR_STDOUT  (stdout)
95 #else /*!FILE_FILTER_USES_STDIO*/
96 # define my_fopen_ro(a,b) fd_cache_open ((a),(b))
97 # define my_fopen(a,b)    direct_open ((a),(b))
98 # ifdef HAVE_W32_SYSTEM
99    /* (We assume that a HANDLE first into an int.)  */
100 #  define my_fileno(a)  ((int)(a))
101    typedef HANDLE fp_or_fd_t;
102 #  define INVALID_FP             ((HANDLE)-1)
103 #  define FILEP_OR_FD_FOR_STDIN  (GetStdHandle (STD_INPUT_HANDLE))
104 #  define FILEP_OR_FD_FOR_STDOUT (GetStdHandle (STD_OUTPUT_HANDLE))
105 #  undef USE_SETMODE
106 # else /*!HAVE_W32_SYSTEM*/
107 #  define my_fileno(a)  (a)
108    typedef int fp_or_fd_t;
109 #  define INVALID_FP             (-1)
110 #  define FILEP_OR_FD_FOR_STDIN  (0)
111 #  define FILEP_OR_FD_FOR_STDOUT (1)
112 # endif /*!HAVE_W32_SYSTEM*/
113 #endif /*!FILE_FILTER_USES_STDIO*/
114
115 /* The context used by the file filter.  */
116 typedef struct
117 {
118   fp_or_fd_t fp;       /* Open file pointer or handle.  */
119   int keep_open; 
120   int no_cache;
121   int eof_seen;
122   int print_only_name; /* Flags indicating that fname is not a real file.  */
123   char fname[1];       /* Name of the file.  */
124 }
125 file_filter_ctx_t;
126
127
128 /* If we are not using stdio as the backend we make use of a "close
129    cache".  */
130 #ifndef FILE_FILTER_USES_STDIO
131 struct close_cache_s
132 {
133   struct close_cache_s *next;
134   fp_or_fd_t fp;
135   char fname[1];
136 };
137 typedef struct close_cache_s *close_cache_t;
138 static close_cache_t close_cache;
139 #endif /*!FILE_FILTER_USES_STDIO*/
140
141
142
143 #ifdef HAVE_W32_SYSTEM
144 typedef struct
145 {
146   int sock;
147   int keep_open;
148   int no_cache;
149   int eof_seen;
150   int print_only_name;  /* Flag indicating that fname is not a real file.  */
151   char fname[1];        /* Name of the file */
152 }
153 sock_filter_ctx_t;
154 #endif /*HAVE_W32_SYSTEM*/
155
156 /* The first partial length header block must be of size 512
157  * to make it easier (and efficienter) we use a min. block size of 512
158  * for all chunks (but the last one) */
159 #define OP_MIN_PARTIAL_CHUNK      512
160 #define OP_MIN_PARTIAL_CHUNK_2POW 9
161
162 /* The context we use for the block filter (used to handle OpenPGP
163    length information header).  */
164 typedef struct
165 {
166   int use;
167   size_t size;
168   size_t count;
169   int partial;     /* 1 = partial header, 2 in last partial packet.  */
170   char *buffer;    /* Used for partial header.  */
171   size_t buflen;   /* Used size of buffer.  */
172   int first_c;     /* First character of a partial header (which is > 0).  */
173   int eof;
174 }
175 block_filter_ctx_t;
176
177
178 /* Global flag to tell whether special file names are enabled.  See
179    gpg.c for an explanation of these file names.  FIXME: it does not
180    belong into the iobuf subsystem. */
181 static int special_names_enabled;
182
183 /* Local prototypes.  */
184 static int underflow (iobuf_t a);
185 static int translate_file_handle (int fd, int for_write);
186
187
188 \f
189 #ifndef FILE_FILTER_USES_STDIO
190 /*
191  * Invalidate (i.e. close) a cached iobuf
192  */
193 static void
194 fd_cache_invalidate (const char *fname)
195 {
196   close_cache_t cc;
197
198   assert (fname);
199   if (DBG_IOBUF)
200     log_debug ("fd_cache_invalidate (%s)\n", fname);
201
202   for (cc = close_cache; cc; cc = cc->next)
203     {
204       if (cc->fp != INVALID_FP && !strcmp (cc->fname, fname))
205         {
206           if (DBG_IOBUF)
207             log_debug ("                did (%s)\n", cc->fname);
208 #ifdef HAVE_W32_SYSTEM
209           CloseHandle (cc->fp);
210 #else
211           close (cc->fp);
212 #endif
213           cc->fp = INVALID_FP;
214         }
215     }
216 }
217
218
219 static fp_or_fd_t
220 direct_open (const char *fname, const char *mode)
221 {
222 #ifdef HAVE_W32_SYSTEM
223   unsigned long da, cd, sm;
224   HANDLE hfile;
225
226   /* Note, that we do not handle all mode combinations */
227
228   /* According to the ReactOS source it seems that open() of the
229    * standard MSW32 crt does open the file in share mode which is
230    * something new for MS applications ;-)
231    */
232   if (strchr (mode, '+'))
233     {
234       fd_cache_invalidate (fname);
235       da = GENERIC_READ | GENERIC_WRITE;
236       cd = OPEN_EXISTING;
237       sm = FILE_SHARE_READ | FILE_SHARE_WRITE;
238     }
239   else if (strchr (mode, 'w'))
240     {
241       fd_cache_invalidate (fname);
242       da = GENERIC_WRITE;
243       cd = CREATE_ALWAYS;
244       sm = FILE_SHARE_WRITE;
245     }
246   else
247     {
248       da = GENERIC_READ;
249       cd = OPEN_EXISTING;
250       sm = FILE_SHARE_READ;
251     }
252
253   hfile = CreateFile (fname, da, sm, NULL, cd, FILE_ATTRIBUTE_NORMAL, NULL);
254   return hfile;
255 #else /*!HAVE_W32_SYSTEM*/
256   int oflag;
257   int cflag = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH;
258
259   /* Note, that we do not handle all mode combinations */
260   if (strchr (mode, '+'))
261     {
262       fd_cache_invalidate (fname);
263       oflag = O_RDWR;
264     }
265   else if (strchr (mode, 'w'))
266     {
267       fd_cache_invalidate (fname);
268       oflag = O_WRONLY | O_CREAT | O_TRUNC;
269     }
270   else
271     {
272       oflag = O_RDONLY;
273     }
274 #ifdef O_BINARY
275   if (strchr (mode, 'b'))
276     oflag |= O_BINARY;
277 #endif
278   /* No we need to distinguish between POSIX and RISC OS.  */
279 #ifndef __riscos__
280   return open (fname, oflag, cflag);
281 #else
282   {
283     struct stat buf;
284     int rc = stat (fname, &buf);
285
286     /* Don't allow iobufs on directories */
287     if (!rc && S_ISDIR (buf.st_mode) && !S_ISREG (buf.st_mode))
288       return __set_errno (EISDIR);
289     else
290       return open (fname, oflag, cflag);
291   }
292 #endif
293 #endif /*!HAVE_W32_SYSTEM*/
294 }
295
296
297 /*
298  * Instead of closing an FD we keep it open and cache it for later reuse 
299  * Note that this caching strategy only works if the process does not chdir.
300  */
301 static void
302 fd_cache_close (const char *fname, fp_or_fd_t fp)
303 {
304   close_cache_t cc;
305
306   assert (fp);
307   if (!fname || !*fname)
308     {
309 #ifdef HAVE_W32_SYSTEM
310       CloseHandle (fp);
311 #else
312       close (fp);
313 #endif
314       if (DBG_IOBUF)
315         log_debug ("fd_cache_close (%d) real\n", (int)fp);
316       return;
317     }
318   /* try to reuse a slot */
319   for (cc = close_cache; cc; cc = cc->next)
320     {
321       if (cc->fp == INVALID_FP && !strcmp (cc->fname, fname))
322         {
323           cc->fp = fp;
324           if (DBG_IOBUF)
325             log_debug ("fd_cache_close (%s) used existing slot\n", fname);
326           return;
327         }
328     }
329   /* add a new one */
330   if (DBG_IOBUF)
331     log_debug ("fd_cache_close (%s) new slot created\n", fname);
332   cc = xcalloc (1, sizeof *cc + strlen (fname));
333   strcpy (cc->fname, fname);
334   cc->fp = fp;
335   cc->next = close_cache;
336   close_cache = cc;
337 }
338
339 /*
340  * Do an direct_open on FNAME but first try to reuse one from the fd_cache
341  */
342 static fp_or_fd_t
343 fd_cache_open (const char *fname, const char *mode)
344 {
345   close_cache_t cc;
346
347   assert (fname);
348   for (cc = close_cache; cc; cc = cc->next)
349     {
350       if (cc->fp != INVALID_FP && !strcmp (cc->fname, fname))
351         {
352           fp_or_fd_t fp = cc->fp;
353           cc->fp = INVALID_FP;
354           if (DBG_IOBUF)
355             log_debug ("fd_cache_open (%s) using cached fp\n", fname);
356 #ifdef HAVE_W32_SYSTEM
357           if (SetFilePointer (fp, 0, NULL, FILE_BEGIN) == 0xffffffff)
358             {
359               log_error ("rewind file failed on handle %p: ec=%d\n",
360                          fp, (int) GetLastError ());
361               fp = INVALID_FP;
362             }
363 #else
364           if (lseek (fp, 0, SEEK_SET) == (off_t) - 1)
365             {
366               log_error ("can't rewind fd %d: %s\n", fp, strerror (errno));
367               fp = INVALID_FP;
368             }
369 #endif
370           return fp;
371         }
372     }
373   if (DBG_IOBUF)
374     log_debug ("fd_cache_open (%s) not cached\n", fname);
375   return direct_open (fname, mode);
376 }
377
378 #endif /*FILE_FILTER_USES_STDIO */
379
380
381 /****************
382  * Read data from a file into buf which has an allocated length of *LEN.
383  * return the number of read bytes in *LEN. OPAQUE is the FILE * of
384  * the stream. A is not used.
385  * control may be:
386  * IOBUFCTRL_INIT: called just before the function is linked into the
387  *                 list of function. This can be used to prepare internal
388  *                 data structures of the function.
389  * IOBUFCTRL_FREE: called just before the function is removed from the
390  *                  list of functions and can be used to release internal
391  *                  data structures or close a file etc.
392  * IOBUFCTRL_UNDERFLOW: called by iobuf_underflow to fill the buffer
393  *                  with new stuff. *RET_LEN is the available size of the
394  *                  buffer, and should be set to the number of bytes
395  *                  which were put into the buffer. The function
396  *                  returns 0 to indicate success, -1 on EOF and
397  *                  GPG_ERR_xxxxx for other errors.
398  *
399  * IOBUFCTRL_FLUSH: called by iobuf_flush() to write out the collected stuff.
400  *                  *RET_LAN is the number of bytes in BUF.
401  *
402  * IOBUFCTRL_CANCEL: send to all filters on behalf of iobuf_cancel.  The
403  *                  filter may take appropriate action on this message.
404  */
405 static int
406 file_filter (void *opaque, int control, iobuf_t chain, byte * buf,
407              size_t * ret_len)
408 {
409   file_filter_ctx_t *a = opaque;
410   fp_or_fd_t f = a->fp;
411   size_t size = *ret_len;
412   size_t nbytes = 0;
413   int rc = 0;
414
415 #ifdef FILE_FILTER_USES_STDIO
416   if (control == IOBUFCTRL_UNDERFLOW)
417     {
418       assert (size);            /* need a buffer */
419       if (feof (f))
420         {                       /* On terminals you could easiely read as many EOFs as you call         */
421           rc = -1;              /* fread() or fgetc() repeatly. Every call will block until you press   */
422           *ret_len = 0;         /* CTRL-D. So we catch this case before we call fread() again.          */
423         }
424       else
425         {
426           clearerr (f);
427           nbytes = fread (buf, 1, size, f);
428           if (feof (f) && !nbytes)
429             {
430               rc = -1;          /* okay: we can return EOF now. */
431             }
432           else if (ferror (f) && errno != EPIPE)
433             {
434               rc = gpg_error_from_syserror ();
435               log_error ("%s: read error: %s\n", a->fname, strerror (errno));
436             }
437           *ret_len = nbytes;
438         }
439     }
440   else if (control == IOBUFCTRL_FLUSH)
441     {
442       if (size)
443         {
444           clearerr (f);
445           nbytes = fwrite (buf, 1, size, f);
446           if (ferror (f))
447             {
448               rc = gpg_error_from_syserror ();
449               log_error ("%s: write error: %s\n", a->fname, strerror (errno));
450             }
451         }
452       *ret_len = nbytes;
453     }
454   else if (control == IOBUFCTRL_INIT)
455     {
456       a->keep_open = a->no_cache = 0;
457     }
458   else if (control == IOBUFCTRL_DESC)
459     {
460       *(char **) buf = "file_filter";
461     }
462   else if (control == IOBUFCTRL_FREE)
463     {
464       if (f != stdin && f != stdout)
465         {
466           if (DBG_IOBUF)
467             log_debug ("%s: close fd %d\n", a->fname, fileno (f));
468           if (!a->keep_open)
469             fclose (f);
470         }
471       f = NULL;
472       xfree (a);                /* we can free our context now */
473     }
474 #else /* !stdio implementation */
475
476   if (control == IOBUFCTRL_UNDERFLOW)
477     {
478       assert (size);            /* need a buffer */
479       if (a->eof_seen)
480         {
481           rc = -1;
482           *ret_len = 0;
483         }
484       else
485         {
486 #ifdef HAVE_W32_SYSTEM
487           unsigned long nread;
488
489           nbytes = 0;
490           if (!ReadFile (f, buf, size, &nread, NULL))
491             {
492               int ec = (int) GetLastError ();
493               if (ec != ERROR_BROKEN_PIPE)
494                 {
495                   rc = gpg_error_from_errno (ec);
496                   log_error ("%s: read error: ec=%d\n", a->fname, ec);
497                 }
498             }
499           else if (!nread)
500             {
501               a->eof_seen = 1;
502               rc = -1;
503             }
504           else
505             {
506               nbytes = nread;
507             }
508
509 #else
510
511           int n;
512
513           nbytes = 0;
514           do
515             {
516               n = read (f, buf, size);
517             }
518           while (n == -1 && errno == EINTR);
519           if (n == -1)
520             {                   /* error */
521               if (errno != EPIPE)
522                 {
523                   rc = gpg_error_from_syserror ();
524                   log_error ("%s: read error: %s\n",
525                              a->fname, strerror (errno));
526                 }
527             }
528           else if (!n)
529             {                   /* eof */
530               a->eof_seen = 1;
531               rc = -1;
532             }
533           else
534             {
535               nbytes = n;
536             }
537 #endif
538           *ret_len = nbytes;
539         }
540     }
541   else if (control == IOBUFCTRL_FLUSH)
542     {
543       if (size)
544         {
545 #ifdef HAVE_W32_SYSTEM
546           byte *p = buf;
547           unsigned long n;
548
549           nbytes = size;
550           do
551             {
552               if (size && !WriteFile (f, p, nbytes, &n, NULL))
553                 {
554                   int ec = (int) GetLastError ();
555                   rc = gpg_error_from_errno (ec);
556                   log_error ("%s: write error: ec=%d\n", a->fname, ec);
557                   break;
558                 }
559               p += n;
560               nbytes -= n;
561             }
562           while (nbytes);
563           nbytes = p - buf;
564 #else
565           byte *p = buf;
566           int n;
567
568           nbytes = size;
569           do
570             {
571               do
572                 {
573                   n = write (f, p, nbytes);
574                 }
575               while (n == -1 && errno == EINTR);
576               if (n > 0)
577                 {
578                   p += n;
579                   nbytes -= n;
580                 }
581             }
582           while (n != -1 && nbytes);
583           if (n == -1)
584             {
585               rc = gpg_error_from_syserror ();
586               log_error ("%s: write error: %s\n", a->fname, strerror (errno));
587             }
588           nbytes = p - buf;
589 #endif
590         }
591       *ret_len = nbytes;
592     }
593   else if (control == IOBUFCTRL_INIT)
594     {
595       a->eof_seen = 0;
596       a->keep_open = 0;
597       a->no_cache = 0;
598     }
599   else if (control == IOBUFCTRL_DESC)
600     {
601       *(char **) buf = "file_filter(fd)";
602     }
603   else if (control == IOBUFCTRL_FREE)
604     {
605 #ifdef HAVE_W32_SYSTEM
606       if (f != FILEP_OR_FD_FOR_STDIN && f != FILEP_OR_FD_FOR_STDOUT)
607         {
608           if (DBG_IOBUF)
609             log_debug ("%s: close handle %p\n", a->fname, f);
610           if (!a->keep_open)
611             fd_cache_close (a->no_cache ? NULL : a->fname, f);
612         }
613 #else
614       if ((int) f != 0 && (int) f != 1)
615         {
616           if (DBG_IOBUF)
617             log_debug ("%s: close fd %d\n", a->fname, f);
618           if (!a->keep_open)
619             fd_cache_close (a->no_cache ? NULL : a->fname, f);
620         }
621       f = INVALID_FP;
622 #endif
623       xfree (a);                /* we can free our context now */
624     }
625 #endif /* !stdio implementation */
626   return rc;
627 }
628
629
630 #ifdef HAVE_W32_SYSTEM
631 /* Because network sockets are special objects under Lose32 we have to
632    use a dedicated filter for them. */
633 static int
634 sock_filter (void *opaque, int control, iobuf_t chain, byte * buf,
635              size_t * ret_len)
636 {
637   sock_filter_ctx_t *a = opaque;
638   size_t size = *ret_len;
639   size_t nbytes = 0;
640   int rc = 0;
641
642   if (control == IOBUFCTRL_UNDERFLOW)
643     {
644       assert (size);            /* need a buffer */
645       if (a->eof_seen)
646         {
647           rc = -1;
648           *ret_len = 0;
649         }
650       else
651         {
652           int nread;
653
654           nread = recv (a->sock, buf, size, 0);
655           if (nread == SOCKET_ERROR)
656             {
657               int ec = (int) WSAGetLastError ();
658               rc = gpg_error_from_errno (ec);
659               log_error ("socket read error: ec=%d\n", ec);
660             }
661           else if (!nread)
662             {
663               a->eof_seen = 1;
664               rc = -1;
665             }
666           else
667             {
668               nbytes = nread;
669             }
670           *ret_len = nbytes;
671         }
672     }
673   else if (control == IOBUFCTRL_FLUSH)
674     {
675       if (size)
676         {
677           byte *p = buf;
678           int n;
679
680           nbytes = size;
681           do
682             {
683               n = send (a->sock, p, nbytes, 0);
684               if (n == SOCKET_ERROR)
685                 {
686                   int ec = (int) WSAGetLastError ();
687                   rc = gpg_error_from_errno (ec);
688                   log_error ("socket write error: ec=%d\n", ec);
689                   break;
690                 }
691               p += n;
692               nbytes -= n;
693             }
694           while (nbytes);
695           nbytes = p - buf;
696         }
697       *ret_len = nbytes;
698     }
699   else if (control == IOBUFCTRL_INIT)
700     {
701       a->eof_seen = 0;
702       a->keep_open = 0;
703       a->no_cache = 0;
704     }
705   else if (control == IOBUFCTRL_DESC)
706     {
707       *(char **) buf = "sock_filter";
708     }
709   else if (control == IOBUFCTRL_FREE)
710     {
711       if (!a->keep_open)
712         closesocket (a->sock);
713       xfree (a);                /* we can free our context now */
714     }
715   return rc;
716 }
717 #endif /*HAVE_W32_SYSTEM*/
718
719 /****************
720  * This is used to implement the block write mode.
721  * Block reading is done on a byte by byte basis in readbyte(),
722  * without a filter
723  */
724 static int
725 block_filter (void *opaque, int control, iobuf_t chain, byte * buffer,
726               size_t * ret_len)
727 {
728   block_filter_ctx_t *a = opaque;
729   char *buf = (char *)buffer;
730   size_t size = *ret_len;
731   int c, needed, rc = 0;
732   char *p;
733
734   if (control == IOBUFCTRL_UNDERFLOW)
735     {
736       size_t n = 0;
737
738       p = buf;
739       assert (size);            /* need a buffer */
740       if (a->eof)               /* don't read any further */
741         rc = -1;
742       while (!rc && size)
743         {
744           if (!a->size)
745             {                   /* get the length bytes */
746               if (a->partial == 2)
747                 {
748                   a->eof = 1;
749                   if (!n)
750                     rc = -1;
751                   break;
752                 }
753               else if (a->partial)
754                 {
755                   /* These OpenPGP introduced huffman like encoded length
756                    * bytes are really a mess :-( */
757                   if (a->first_c)
758                     {
759                       c = a->first_c;
760                       a->first_c = 0;
761                     }
762                   else if ((c = iobuf_get (chain)) == -1)
763                     {
764                       log_error ("block_filter: 1st length byte missing\n");
765                       rc = GPG_ERR_BAD_DATA;
766                       break;
767                     }
768                   if (c < 192)
769                     {
770                       a->size = c;
771                       a->partial = 2;
772                       if (!a->size)
773                         {
774                           a->eof = 1;
775                           if (!n)
776                             rc = -1;
777                           break;
778                         }
779                     }
780                   else if (c < 224)
781                     {
782                       a->size = (c - 192) * 256;
783                       if ((c = iobuf_get (chain)) == -1)
784                         {
785                           log_error
786                             ("block_filter: 2nd length byte missing\n");
787                           rc = GPG_ERR_BAD_DATA;
788                           break;
789                         }
790                       a->size += c + 192;
791                       a->partial = 2;
792                       if (!a->size)
793                         {
794                           a->eof = 1;
795                           if (!n)
796                             rc = -1;
797                           break;
798                         }
799                     }
800                   else if (c == 255)
801                     {
802                       a->size = iobuf_get (chain) << 24;
803                       a->size |= iobuf_get (chain) << 16;
804                       a->size |= iobuf_get (chain) << 8;
805                       if ((c = iobuf_get (chain)) == -1)
806                         {
807                           log_error ("block_filter: invalid 4 byte length\n");
808                           rc = GPG_ERR_BAD_DATA;
809                           break;
810                         }
811                       a->size |= c;
812                       a->partial = 2;
813                       if (!a->size)
814                         {
815                           a->eof = 1;
816                           if (!n)
817                             rc = -1;
818                           break;
819                         }
820                     }
821                   else
822                     { /* Next partial body length. */
823                       a->size = 1 << (c & 0x1f);
824                     }
825                   /*  log_debug("partial: ctx=%p c=%02x size=%u\n", a, c, a->size); */
826                 }
827               else
828                 BUG (); 
829             }
830
831           while (!rc && size && a->size)
832             {
833               needed = size < a->size ? size : a->size;
834               c = iobuf_read (chain, p, needed);
835               if (c < needed)
836                 {
837                   if (c == -1)
838                     c = 0;
839                   log_error
840                     ("block_filter %p: read error (size=%lu,a->size=%lu)\n",
841                      a, (ulong) size + c, (ulong) a->size + c);
842                   rc = GPG_ERR_BAD_DATA;
843                 }
844               else
845                 {
846                   size -= c;
847                   a->size -= c;
848                   p += c;
849                   n += c;
850                 }
851             }
852         }
853       *ret_len = n;
854     }
855   else if (control == IOBUFCTRL_FLUSH)
856     {
857       if (a->partial)
858         {                       /* the complicated openpgp scheme */
859           size_t blen, n, nbytes = size + a->buflen;
860
861           assert (a->buflen <= OP_MIN_PARTIAL_CHUNK);
862           if (nbytes < OP_MIN_PARTIAL_CHUNK)
863             {
864               /* not enough to write a partial block out; so we store it */
865               if (!a->buffer)
866                 a->buffer = xmalloc (OP_MIN_PARTIAL_CHUNK);
867               memcpy (a->buffer + a->buflen, buf, size);
868               a->buflen += size;
869             }
870           else
871             {                   /* okay, we can write out something */
872               /* do this in a loop to use the most efficient block lengths */
873               p = buf;
874               do
875                 {
876                   /* find the best matching block length - this is limited
877                    * by the size of the internal buffering */
878                   for (blen = OP_MIN_PARTIAL_CHUNK * 2,
879                        c = OP_MIN_PARTIAL_CHUNK_2POW + 1; blen <= nbytes;
880                        blen *= 2, c++)
881                     ;
882                   blen /= 2;
883                   c--;
884                   /* write the partial length header */
885                   assert (c <= 0x1f);   /*;-) */
886                   c |= 0xe0;
887                   iobuf_put (chain, c);
888                   if ((n = a->buflen))
889                     {           /* write stuff from the buffer */
890                       assert (n == OP_MIN_PARTIAL_CHUNK);
891                       if (iobuf_write (chain, a->buffer, n))
892                         rc = gpg_error_from_syserror ();
893                       a->buflen = 0;
894                       nbytes -= n;
895                     }
896                   if ((n = nbytes) > blen)
897                     n = blen;
898                   if (n && iobuf_write (chain, p, n))
899                     rc = gpg_error_from_syserror ();
900                   p += n;
901                   nbytes -= n;
902                 }
903               while (!rc && nbytes >= OP_MIN_PARTIAL_CHUNK);
904               /* store the rest in the buffer */
905               if (!rc && nbytes)
906                 {
907                   assert (!a->buflen);
908                   assert (nbytes < OP_MIN_PARTIAL_CHUNK);
909                   if (!a->buffer)
910                     a->buffer = xmalloc (OP_MIN_PARTIAL_CHUNK);
911                   memcpy (a->buffer, p, nbytes);
912                   a->buflen = nbytes;
913                 }
914             }
915         }
916       else
917         BUG ();
918     }
919   else if (control == IOBUFCTRL_INIT)
920     {
921       if (DBG_IOBUF)
922         log_debug ("init block_filter %p\n", a);
923       if (a->partial)
924         a->count = 0;
925       else if (a->use == 1)
926         a->count = a->size = 0;
927       else
928         a->count = a->size;     /* force first length bytes */
929       a->eof = 0;
930       a->buffer = NULL;
931       a->buflen = 0;
932     }
933   else if (control == IOBUFCTRL_DESC)
934     {
935       *(char **) buf = "block_filter";
936     }
937   else if (control == IOBUFCTRL_FREE)
938     {
939       if (a->use == 2)
940         {                       /* write the end markers */
941           if (a->partial)
942             {
943               u32 len;
944               /* write out the remaining bytes without a partial header
945                * the length of this header may be 0 - but if it is
946                * the first block we are not allowed to use a partial header
947                * and frankly we can't do so, because this length must be
948                * a power of 2. This is _really_ complicated because we
949                * have to check the possible length of a packet prior
950                * to it's creation: a chain of filters becomes complicated
951                * and we need a lot of code to handle compressed packets etc.
952                *   :-(((((((
953                */
954               /* construct header */
955               len = a->buflen;
956               /*log_debug("partial: remaining length=%u\n", len ); */
957               if (len < 192)
958                 rc = iobuf_put (chain, len);
959               else if (len < 8384)
960                 {
961                   if (!(rc = iobuf_put (chain, ((len - 192) / 256) + 192)))
962                     rc = iobuf_put (chain, ((len - 192) % 256));
963                 }
964               else
965                 {               /* use a 4 byte header */
966                   if (!(rc = iobuf_put (chain, 0xff)))
967                     if (!(rc = iobuf_put (chain, (len >> 24) & 0xff)))
968                       if (!(rc = iobuf_put (chain, (len >> 16) & 0xff)))
969                         if (!(rc = iobuf_put (chain, (len >> 8) & 0xff)))
970                           rc = iobuf_put (chain, len & 0xff);
971                 }
972               if (!rc && len)
973                 rc = iobuf_write (chain, a->buffer, len);
974               if (rc)
975                 {
976                   log_error ("block_filter: write error: %s\n",
977                              strerror (errno));
978                   rc = gpg_error_from_syserror ();
979                 }
980               xfree (a->buffer);
981               a->buffer = NULL;
982               a->buflen = 0;
983             }
984           else
985             BUG ();
986         }
987       else if (a->size)
988         {
989           log_error ("block_filter: pending bytes!\n");
990         }
991       if (DBG_IOBUF)
992         log_debug ("free block_filter %p\n", a);
993       xfree (a);                /* we can free our context now */
994     }
995
996   return rc;
997 }
998
999
1000 static void
1001 print_chain (iobuf_t a)
1002 {
1003   if (!DBG_IOBUF)
1004     return;
1005   for (; a; a = a->chain)
1006     {
1007       size_t dummy_len = 0;
1008       const char *desc = "[none]";
1009
1010       if (a->filter)
1011         a->filter (a->filter_ov, IOBUFCTRL_DESC, NULL,
1012                    (byte *) & desc, &dummy_len);
1013
1014       log_debug ("iobuf chain: %d.%d `%s' filter_eof=%d start=%d len=%d\n",
1015                  a->no, a->subno, desc?desc:"?", a->filter_eof,
1016                  (int) a->d.start, (int) a->d.len);
1017     }
1018 }
1019
1020 int
1021 iobuf_print_chain (iobuf_t a)
1022 {
1023   print_chain (a);
1024   return 0;
1025 }
1026
1027 /****************
1028  * Allocate a new io buffer, with no function assigned.
1029  * Use is the desired usage: 1 for input, 2 for output, 3 for temp buffer
1030  * BUFSIZE is a suggested buffer size.
1031  */
1032 iobuf_t
1033 iobuf_alloc (int use, size_t bufsize)
1034 {
1035   iobuf_t a;
1036   static int number = 0;
1037
1038   a = xcalloc (1, sizeof *a);
1039   a->use = use;
1040   a->d.buf = xmalloc (bufsize);
1041   a->d.size = bufsize;
1042   a->no = ++number;
1043   a->subno = 0;
1044   a->opaque = NULL;
1045   a->real_fname = NULL;
1046   return a;
1047 }
1048
1049 int
1050 iobuf_close (iobuf_t a)
1051 {
1052   iobuf_t a2;
1053   size_t dummy_len = 0;
1054   int rc = 0;
1055
1056   if (a && a->directfp)
1057     {
1058       fclose (a->directfp);
1059       xfree (a->real_fname);
1060       if (DBG_IOBUF)
1061         log_debug ("iobuf_close -> %p\n", a->directfp);
1062       return 0;
1063     }
1064
1065   for (; a && !rc; a = a2)
1066     {
1067       a2 = a->chain;
1068       if (a->use == 2 && (rc = iobuf_flush (a)))
1069         log_error ("iobuf_flush failed on close: %s\n", gpg_strerror (rc));
1070
1071       if (DBG_IOBUF)
1072         log_debug ("iobuf-%d.%d: close `%s'\n", a->no, a->subno,
1073                    a->desc?a->desc:"?");
1074       if (a->filter && (rc = a->filter (a->filter_ov, IOBUFCTRL_FREE,
1075                                         a->chain, NULL, &dummy_len)))
1076         log_error ("IOBUFCTRL_FREE failed on close: %s\n", gpg_strerror (rc));
1077       xfree (a->real_fname);
1078       if (a->d.buf)
1079         {
1080           memset (a->d.buf, 0, a->d.size);      /* erase the buffer */
1081           xfree (a->d.buf);
1082         }
1083       xfree (a);
1084     }
1085   return rc;
1086 }
1087
1088 int
1089 iobuf_cancel (iobuf_t a)
1090 {
1091   const char *s;
1092   iobuf_t a2;
1093   int rc;
1094 #if defined(HAVE_W32_SYSTEM) || defined(__riscos__)
1095   char *remove_name = NULL;
1096 #endif
1097
1098   if (a && a->use == 2)
1099     {
1100       s = iobuf_get_real_fname (a);
1101       if (s && *s)
1102         {
1103 #if defined(HAVE_W32_SYSTEM) || defined(__riscos__)
1104           remove_name = xstrdup (s);
1105 #else
1106           remove (s);
1107 #endif
1108         }
1109     }
1110
1111   /* send a cancel message to all filters */
1112   for (a2 = a; a2; a2 = a2->chain)
1113     {
1114       size_t dummy;
1115       if (a2->filter)
1116         a2->filter (a2->filter_ov, IOBUFCTRL_CANCEL, a2->chain, NULL, &dummy);
1117     }
1118
1119   rc = iobuf_close (a);
1120 #if defined(HAVE_W32_SYSTEM) || defined(__riscos__)
1121   if (remove_name)
1122     {
1123       /* Argg, MSDOS does not allow to remove open files.  So
1124        * we have to do it here */
1125       remove (remove_name);
1126       xfree (remove_name);
1127     }
1128 #endif
1129   return rc;
1130 }
1131
1132
1133 /****************
1134  * create a temporary iobuf, which can be used to collect stuff
1135  * in an iobuf and later be written by iobuf_write_temp() to another
1136  * iobuf.
1137  */
1138 iobuf_t
1139 iobuf_temp ()
1140 {
1141   iobuf_t a;
1142
1143   a = iobuf_alloc (3, IOBUF_BUFFER_SIZE);
1144
1145   return a;
1146 }
1147
1148 iobuf_t
1149 iobuf_temp_with_content (const char *buffer, size_t length)
1150 {
1151   iobuf_t a;
1152
1153   a = iobuf_alloc (3, length);
1154   memcpy (a->d.buf, buffer, length);
1155   a->d.len = length;
1156
1157   return a;
1158 }
1159
1160 void
1161 iobuf_enable_special_filenames (int yes)
1162 {
1163   special_names_enabled = yes;
1164 }
1165
1166
1167 /* See whether the filename has the form "-&nnnn", where n is a
1168    non-zero number.  Returns this number or -1 if it is not the
1169    case.  */
1170 static int
1171 check_special_filename (const char *fname)
1172 {
1173   if (special_names_enabled && fname && *fname == '-' && fname[1] == '&')
1174     {
1175       int i;
1176
1177       fname += 2;
1178       for (i = 0; digitp (fname+i); i++)
1179         ;
1180       if (!fname[i])
1181         return atoi (fname);
1182     }
1183   return -1;
1184 }
1185
1186
1187 /* This fucntion returns true if FNAME indicates a PIPE (stdout or
1188    stderr) or a special file name if those are enabled. */
1189 int
1190 iobuf_is_pipe_filename (const char *fname)
1191 {
1192   if (!fname || (*fname=='-' && !fname[1]) )
1193     return 1;
1194   return check_special_filename (fname) != -1;
1195 }
1196
1197 /****************
1198  * Create a head iobuf for reading from a file
1199  * returns: NULL if an error occures and sets errno
1200  */
1201 iobuf_t
1202 iobuf_open (const char *fname)
1203 {
1204   iobuf_t a;
1205   fp_or_fd_t fp;
1206   file_filter_ctx_t *fcx;
1207   size_t len;
1208   int print_only = 0;
1209   int fd;
1210
1211   if (!fname || (*fname == '-' && !fname[1]))
1212     {
1213       fp = FILEP_OR_FD_FOR_STDIN;
1214 #ifdef USE_SETMODE
1215       setmode (my_fileno (fp), O_BINARY);
1216 #endif
1217       fname = "[stdin]";
1218       print_only = 1;
1219     }
1220   else if ((fd = check_special_filename (fname)) != -1)
1221     return iobuf_fdopen (translate_file_handle (fd, 0), "rb");
1222   else if ((fp = my_fopen_ro (fname, "rb")) == INVALID_FP)
1223     return NULL;
1224   a = iobuf_alloc (1, IOBUF_BUFFER_SIZE);
1225   fcx = xmalloc (sizeof *fcx + strlen (fname));
1226   fcx->fp = fp;
1227   fcx->print_only_name = print_only;
1228   strcpy (fcx->fname, fname);
1229   if (!print_only)
1230     a->real_fname = xstrdup (fname);
1231   a->filter = file_filter;
1232   a->filter_ov = fcx;
1233   file_filter (fcx, IOBUFCTRL_DESC, NULL, (byte *) & a->desc, &len);
1234   file_filter (fcx, IOBUFCTRL_INIT, NULL, NULL, &len);
1235   if (DBG_IOBUF)
1236     log_debug ("iobuf-%d.%d: open `%s' fd=%d\n",
1237                a->no, a->subno, fname, (int) my_fileno (fcx->fp));
1238
1239   return a;
1240 }
1241
1242 /****************
1243  * Create a head iobuf for reading from a file
1244  * returns: NULL if an error occures and sets errno
1245  */
1246 iobuf_t
1247 iobuf_fdopen (int fd, const char *mode)
1248 {
1249   iobuf_t a;
1250   fp_or_fd_t fp;
1251   file_filter_ctx_t *fcx;
1252   size_t len;
1253
1254 #ifdef FILE_FILTER_USES_STDIO
1255   if (!(fp = fdopen (fd, mode)))
1256     return NULL;
1257 #else
1258   fp = (fp_or_fd_t) fd;
1259 #endif
1260   a = iobuf_alloc (strchr (mode, 'w') ? 2 : 1, IOBUF_BUFFER_SIZE);
1261   fcx = xmalloc (sizeof *fcx + 20);
1262   fcx->fp = fp;
1263   fcx->print_only_name = 1;
1264   sprintf (fcx->fname, "[fd %d]", fd);
1265   a->filter = file_filter;
1266   a->filter_ov = fcx;
1267   file_filter (fcx, IOBUFCTRL_DESC, NULL, (byte *) & a->desc, &len);
1268   file_filter (fcx, IOBUFCTRL_INIT, NULL, NULL, &len);
1269   if (DBG_IOBUF)
1270     log_debug ("iobuf-%d.%d: fdopen `%s'\n", a->no, a->subno, fcx->fname);
1271   iobuf_ioctl (a, 3, 1, NULL);  /* disable fd caching */
1272   return a;
1273 }
1274
1275
1276 iobuf_t
1277 iobuf_sockopen (int fd, const char *mode)
1278 {
1279   iobuf_t a;
1280 #ifdef HAVE_W32_SYSTEM
1281   sock_filter_ctx_t *scx;
1282   size_t len;
1283
1284   a = iobuf_alloc (strchr (mode, 'w') ? 2 : 1, IOBUF_BUFFER_SIZE);
1285   scx = xmalloc (sizeof *scx + 25);
1286   scx->sock = fd;
1287   scx->print_only_name = 1;
1288   sprintf (scx->fname, "[sock %d]", fd);
1289   a->filter = sock_filter;
1290   a->filter_ov = scx;
1291   sock_filter (scx, IOBUFCTRL_DESC, NULL, (byte *) & a->desc, &len);
1292   sock_filter (scx, IOBUFCTRL_INIT, NULL, NULL, &len);
1293   if (DBG_IOBUF)
1294     log_debug ("iobuf-%d.%d: sockopen `%s'\n", a->no, a->subno, scx->fname);
1295   iobuf_ioctl (a, 3, 1, NULL);  /* disable fd caching */
1296 #else
1297   a = iobuf_fdopen (fd, mode);
1298 #endif
1299   return a;
1300 }
1301
1302 /****************
1303  * create an iobuf for writing to a file; the file will be created.
1304  */
1305 iobuf_t
1306 iobuf_create (const char *fname)
1307 {
1308   iobuf_t a;
1309   fp_or_fd_t fp;
1310   file_filter_ctx_t *fcx;
1311   size_t len;
1312   int print_only = 0;
1313   int fd;
1314
1315   if (!fname || (*fname == '-' && !fname[1]))
1316     {
1317       fp = FILEP_OR_FD_FOR_STDOUT;
1318 #ifdef USE_SETMODE
1319       setmode (my_fileno (fp), O_BINARY);
1320 #endif
1321       fname = "[stdout]";
1322       print_only = 1;
1323     }
1324   else if ((fd = check_special_filename (fname)) != -1)
1325     return iobuf_fdopen (translate_file_handle (fd, 1), "wb");
1326   else if ((fp = my_fopen (fname, "wb")) == INVALID_FP)
1327     return NULL;
1328   a = iobuf_alloc (2, IOBUF_BUFFER_SIZE);
1329   fcx = xmalloc (sizeof *fcx + strlen (fname));
1330   fcx->fp = fp;
1331   fcx->print_only_name = print_only;
1332   strcpy (fcx->fname, fname);
1333   if (!print_only)
1334     a->real_fname = xstrdup (fname);
1335   a->filter = file_filter;
1336   a->filter_ov = fcx;
1337   file_filter (fcx, IOBUFCTRL_DESC, NULL, (byte *) & a->desc, &len);
1338   file_filter (fcx, IOBUFCTRL_INIT, NULL, NULL, &len);
1339   if (DBG_IOBUF)
1340     log_debug ("iobuf-%d.%d: create `%s'\n", a->no, a->subno,
1341                a->desc?a->desc:"?");
1342
1343   return a;
1344 }
1345
1346 /****************
1347  * append to an iobuf; if the file does not exist, create it.
1348  * cannot be used for stdout.
1349  * Note: This is not used.
1350  */
1351 #if 0                           /* not used */
1352 iobuf_t
1353 iobuf_append (const char *fname)
1354 {
1355   iobuf_t a;
1356   FILE *fp;
1357   file_filter_ctx_t *fcx;
1358   size_t len;
1359
1360   if (!fname)
1361     return NULL;
1362   else if (!(fp = my_fopen (fname, "ab")))
1363     return NULL;
1364   a = iobuf_alloc (2, IOBUF_BUFFER_SIZE);
1365   fcx = m_alloc (sizeof *fcx + strlen (fname));
1366   fcx->fp = fp;
1367   strcpy (fcx->fname, fname);
1368   a->real_fname = m_strdup (fname);
1369   a->filter = file_filter;
1370   a->filter_ov = fcx;
1371   file_filter (fcx, IOBUFCTRL_DESC, NULL, (byte *) & a->desc, &len);
1372   file_filter (fcx, IOBUFCTRL_INIT, NULL, NULL, &len);
1373   if (DBG_IOBUF)
1374     log_debug ("iobuf-%d.%d: append `%s'\n", a->no, a->subno,
1375                a->desc?a->desc:"?");
1376
1377   return a;
1378 }
1379 #endif
1380
1381 iobuf_t
1382 iobuf_openrw (const char *fname)
1383 {
1384   iobuf_t a;
1385   fp_or_fd_t fp;
1386   file_filter_ctx_t *fcx;
1387   size_t len;
1388
1389   if (!fname)
1390     return NULL;
1391   else if ((fp = my_fopen (fname, "r+b")) == INVALID_FP)
1392     return NULL;
1393   a = iobuf_alloc (2, IOBUF_BUFFER_SIZE);
1394   fcx = xmalloc (sizeof *fcx + strlen (fname));
1395   fcx->fp = fp;
1396   strcpy (fcx->fname, fname);
1397   a->real_fname = xstrdup (fname);
1398   a->filter = file_filter;
1399   a->filter_ov = fcx;
1400   file_filter (fcx, IOBUFCTRL_DESC, NULL, (byte *) & a->desc, &len);
1401   file_filter (fcx, IOBUFCTRL_INIT, NULL, NULL, &len);
1402   if (DBG_IOBUF)
1403     log_debug ("iobuf-%d.%d: openrw `%s'\n", a->no, a->subno,
1404                a->desc?a->desc:"?");
1405
1406   return a;
1407 }
1408
1409
1410 int
1411 iobuf_ioctl (iobuf_t a, int cmd, int intval, void *ptrval)
1412 {
1413   if (cmd == 1)
1414     {                           /* keep system filepointer/descriptor open */
1415       if (DBG_IOBUF)
1416         log_debug ("iobuf-%d.%d: ioctl `%s' keep=%d\n",
1417                    a ? a->no : -1, a ? a->subno : -1, 
1418                    a && a->desc ? a->desc : "?",
1419                    intval);
1420       for (; a; a = a->chain)
1421         if (!a->chain && a->filter == file_filter)
1422           {
1423             file_filter_ctx_t *b = a->filter_ov;
1424             b->keep_open = intval;
1425             return 0;
1426           }
1427 #ifdef HAVE_W32_SYSTEM
1428         else if (!a->chain && a->filter == sock_filter)
1429           {
1430             sock_filter_ctx_t *b = a->filter_ov;
1431             b->keep_open = intval;
1432             return 0;
1433           }
1434 #endif
1435     }
1436   else if (cmd == 2)
1437     {                           /* invalidate cache */
1438       if (DBG_IOBUF)
1439         log_debug ("iobuf-*.*: ioctl `%s' invalidate\n",
1440                    ptrval ? (char *) ptrval : "?");
1441       if (!a && !intval && ptrval)
1442         {
1443 #ifndef FILE_FILTER_USES_STDIO
1444           fd_cache_invalidate (ptrval);
1445 #endif
1446           return 0;
1447         }
1448     }
1449   else if (cmd == 3)
1450     {                           /* disallow/allow caching */
1451       if (DBG_IOBUF)
1452         log_debug ("iobuf-%d.%d: ioctl `%s' no_cache=%d\n",
1453                    a ? a->no : -1, a ? a->subno : -1, 
1454                    a && a->desc? a->desc : "?",
1455                    intval);
1456       for (; a; a = a->chain)
1457         if (!a->chain && a->filter == file_filter)
1458           {
1459             file_filter_ctx_t *b = a->filter_ov;
1460             b->no_cache = intval;
1461             return 0;
1462           }
1463 #ifdef HAVE_W32_SYSTEM
1464         else if (!a->chain && a->filter == sock_filter)
1465           {
1466             sock_filter_ctx_t *b = a->filter_ov;
1467             b->no_cache = intval;
1468             return 0;
1469           }
1470 #endif
1471     }
1472
1473   return -1;
1474 }
1475
1476
1477 /****************
1478  * Register an i/o filter.
1479  */
1480 int
1481 iobuf_push_filter (iobuf_t a,
1482                    int (*f) (void *opaque, int control,
1483                              iobuf_t chain, byte * buf, size_t * len),
1484                    void *ov)
1485 {
1486   return iobuf_push_filter2 (a, f, ov, 0);
1487 }
1488
1489 int
1490 iobuf_push_filter2 (iobuf_t a,
1491                     int (*f) (void *opaque, int control,
1492                               iobuf_t chain, byte * buf, size_t * len),
1493                     void *ov, int rel_ov)
1494 {
1495   iobuf_t b;
1496   size_t dummy_len = 0;
1497   int rc = 0;
1498
1499   if (a->directfp)
1500     BUG ();
1501
1502   if (a->use == 2 && (rc = iobuf_flush (a)))
1503     return rc;
1504   /* make a copy of the current stream, so that
1505    * A is the new stream and B the original one.
1506    * The contents of the buffers are transferred to the
1507    * new stream.
1508    */
1509   b = xmalloc (sizeof *b);
1510   memcpy (b, a, sizeof *b);
1511   /* fixme: it is stupid to keep a copy of the name at every level
1512    * but we need the name somewhere because the name known by file_filter
1513    * may have been released when we need the name of the file */
1514   b->real_fname = a->real_fname ? xstrdup (a->real_fname) : NULL;
1515   /* remove the filter stuff from the new stream */
1516   a->filter = NULL;
1517   a->filter_ov = NULL;
1518   a->filter_ov_owner = 0;
1519   a->filter_eof = 0;
1520   if (a->use == 3)
1521     a->use = 2;                 /* make a write stream from a temp stream */
1522
1523   if (a->use == 2)
1524     {                           /* allocate a fresh buffer for the
1525                                    original stream */
1526       b->d.buf = xmalloc (a->d.size);
1527       b->d.len = 0;
1528       b->d.start = 0;
1529     }
1530   else
1531     {                           /* allocate a fresh buffer for the new
1532                                    stream */
1533       a->d.buf = xmalloc (a->d.size);
1534       a->d.len = 0;
1535       a->d.start = 0;
1536     }
1537   /* disable nlimit for the new stream */
1538   a->ntotal = b->ntotal + b->nbytes;
1539   a->nlimit = a->nbytes = 0;
1540   a->nofast &= ~1;
1541   /* make a link from the new stream to the original stream */
1542   a->chain = b;
1543   a->opaque = b->opaque;
1544
1545   /* setup the function on the new stream */
1546   a->filter = f;
1547   a->filter_ov = ov;
1548   a->filter_ov_owner = rel_ov;
1549
1550   a->subno = b->subno + 1;
1551   f (ov, IOBUFCTRL_DESC, NULL, (byte *) & a->desc, &dummy_len);
1552
1553   if (DBG_IOBUF)
1554     {
1555       log_debug ("iobuf-%d.%d: push `%s'\n", a->no, a->subno, 
1556                  a->desc?a->desc:"?");
1557       print_chain (a);
1558     }
1559
1560   /* now we can initialize the new function if we have one */
1561   if (a->filter && (rc = a->filter (a->filter_ov, IOBUFCTRL_INIT, a->chain,
1562                                     NULL, &dummy_len)))
1563     log_error ("IOBUFCTRL_INIT failed: %s\n", gpg_strerror (rc));
1564   return rc;
1565 }
1566
1567 /****************
1568  * Remove an i/o filter.
1569  */
1570 static int
1571 pop_filter (iobuf_t a, int (*f) (void *opaque, int control,
1572                                iobuf_t chain, byte * buf, size_t * len),
1573             void *ov)
1574 {
1575   iobuf_t b;
1576   size_t dummy_len = 0;
1577   int rc = 0;
1578
1579   if (a->directfp)
1580     BUG ();
1581
1582   if (DBG_IOBUF)
1583     log_debug ("iobuf-%d.%d: pop `%s'\n", a->no, a->subno,
1584                a->desc?a->desc:"?");
1585   if (!a->filter)
1586     {                           /* this is simple */
1587       b = a->chain;
1588       assert (b);
1589       xfree (a->d.buf);
1590       xfree (a->real_fname);
1591       memcpy (a, b, sizeof *a);
1592       xfree (b);
1593       return 0;
1594     }
1595   for (b = a; b; b = b->chain)
1596     if (b->filter == f && (!ov || b->filter_ov == ov))
1597       break;
1598   if (!b)
1599     log_bug ("pop_filter(): filter function not found\n");
1600
1601   /* flush this stream if it is an output stream */
1602   if (a->use == 2 && (rc = iobuf_flush (b)))
1603     {
1604       log_error ("iobuf_flush failed in pop_filter: %s\n", gpg_strerror (rc));
1605       return rc;
1606     }
1607   /* and tell the filter to free it self */
1608   if (b->filter && (rc = b->filter (b->filter_ov, IOBUFCTRL_FREE, b->chain,
1609                                     NULL, &dummy_len)))
1610     {
1611       log_error ("IOBUFCTRL_FREE failed: %s\n", gpg_strerror (rc));
1612       return rc;
1613     }
1614   if (b->filter_ov && b->filter_ov_owner)
1615     {
1616       xfree (b->filter_ov);
1617       b->filter_ov = NULL;
1618     }
1619
1620
1621   /* and see how to remove it */
1622   if (a == b && !b->chain)
1623     log_bug ("can't remove the last filter from the chain\n");
1624   else if (a == b)
1625     {                           /* remove the first iobuf from the chain */
1626       /* everything from b is copied to a. This is save because
1627        * a flush has been done on the to be removed entry
1628        */
1629       b = a->chain;
1630       xfree (a->d.buf);
1631       xfree (a->real_fname);
1632       memcpy (a, b, sizeof *a);
1633       xfree (b);
1634       if (DBG_IOBUF)
1635         log_debug ("iobuf-%d.%d: popped filter\n", a->no, a->subno);
1636     }
1637   else if (!b->chain)
1638     {                           /* remove the last iobuf from the chain */
1639       log_bug ("Ohh jeee, trying to remove a head filter\n");
1640     }
1641   else
1642     {                           /* remove an intermediate iobuf from the chain */
1643       log_bug ("Ohh jeee, trying to remove an intermediate filter\n");
1644     }
1645
1646   return rc;
1647 }
1648
1649
1650 /****************
1651  * read underflow: read more bytes into the buffer and return
1652  * the first byte or -1 on EOF.
1653  */
1654 static int
1655 underflow (iobuf_t a)
1656 {
1657   size_t len;
1658   int rc;
1659
1660   assert (a->d.start == a->d.len);
1661   if (a->use == 3)
1662     return -1;                  /* EOF because a temp buffer can't do an underflow */
1663
1664   if (a->filter_eof)
1665     {
1666       if (a->chain)
1667         {
1668           iobuf_t b = a->chain;
1669           if (DBG_IOBUF)
1670             log_debug ("iobuf-%d.%d: pop `%s' in underflow\n",
1671                        a->no, a->subno, a->desc?a->desc:"?");
1672           xfree (a->d.buf);
1673           xfree (a->real_fname);
1674           memcpy (a, b, sizeof *a);
1675           xfree (b);
1676           print_chain (a);
1677         }
1678       else
1679         a->filter_eof = 0;      /* for the top level filter */
1680       if (DBG_IOBUF)
1681         log_debug ("iobuf-%d.%d: underflow: eof (due to filter eof)\n",
1682                    a->no, a->subno);
1683       return -1;                /* return one(!) EOF */
1684     }
1685   if (a->error)
1686     {
1687       if (DBG_IOBUF)
1688         log_debug ("iobuf-%d.%d: error\n", a->no, a->subno);
1689       return -1;
1690     }
1691
1692   if (a->directfp)
1693     {
1694       FILE *fp = a->directfp;
1695
1696       len = fread (a->d.buf, 1, a->d.size, fp);
1697       if (len < a->d.size)
1698         {
1699           if (ferror (fp))
1700             a->error = gpg_error_from_syserror ();
1701         }
1702       a->d.len = len;
1703       a->d.start = 0;
1704       return len ? a->d.buf[a->d.start++] : -1;
1705     }
1706
1707
1708   if (a->filter)
1709     {
1710       len = a->d.size;
1711       if (DBG_IOBUF)
1712         log_debug ("iobuf-%d.%d: underflow: req=%lu\n",
1713                    a->no, a->subno, (ulong) len);
1714       rc = a->filter (a->filter_ov, IOBUFCTRL_UNDERFLOW, a->chain,
1715                       a->d.buf, &len);
1716       if (DBG_IOBUF)
1717         {
1718           log_debug ("iobuf-%d.%d: underflow: got=%lu rc=%d\n",
1719                      a->no, a->subno, (ulong) len, rc);
1720 /*          if( a->no == 1 ) */
1721 /*                   log_hexdump ("     data:", a->d.buf, len); */
1722         }
1723       if (a->use == 1 && rc == -1)
1724         {                       /* EOF: we can remove the filter */
1725           size_t dummy_len = 0;
1726
1727           /* and tell the filter to free itself */
1728           if ((rc = a->filter (a->filter_ov, IOBUFCTRL_FREE, a->chain,
1729                                NULL, &dummy_len)))
1730             log_error ("IOBUFCTRL_FREE failed: %s\n", gpg_strerror (rc));
1731           if (a->filter_ov && a->filter_ov_owner)
1732             {
1733               xfree (a->filter_ov);
1734               a->filter_ov = NULL;
1735             }
1736           a->filter = NULL;
1737           a->desc = NULL;
1738           a->filter_ov = NULL;
1739           a->filter_eof = 1;
1740           if (!len && a->chain)
1741             {
1742               iobuf_t b = a->chain;
1743               if (DBG_IOBUF)
1744                 log_debug ("iobuf-%d.%d: pop in underflow (!len)\n",
1745                            a->no, a->subno);
1746               xfree (a->d.buf);
1747               xfree (a->real_fname);
1748               memcpy (a, b, sizeof *a);
1749               xfree (b);
1750               print_chain (a);
1751             }
1752         }
1753       else if (rc)
1754         a->error = rc;
1755
1756       if (!len)
1757         {
1758           if (DBG_IOBUF)
1759             log_debug ("iobuf-%d.%d: underflow: eof\n", a->no, a->subno);
1760           return -1;
1761         }
1762       a->d.len = len;
1763       a->d.start = 0;
1764       return a->d.buf[a->d.start++];
1765     }
1766   else
1767     {
1768       if (DBG_IOBUF)
1769         log_debug ("iobuf-%d.%d: underflow: eof (no filter)\n",
1770                    a->no, a->subno);
1771       return -1;                /* no filter; return EOF */
1772     }
1773 }
1774
1775
1776 int
1777 iobuf_flush (iobuf_t a)
1778 {
1779   size_t len;
1780   int rc;
1781
1782   if (a->directfp)
1783     return 0;
1784
1785   if (a->use == 3)
1786     {                           /* increase the temp buffer */
1787       unsigned char *newbuf;
1788       size_t newsize = a->d.size + IOBUF_BUFFER_SIZE;
1789
1790       if (DBG_IOBUF)
1791         log_debug ("increasing temp iobuf from %lu to %lu\n",
1792                    (ulong) a->d.size, (ulong) newsize);
1793       newbuf = xmalloc (newsize);
1794       memcpy (newbuf, a->d.buf, a->d.len);
1795       xfree (a->d.buf);
1796       a->d.buf = newbuf;
1797       a->d.size = newsize;
1798       return 0;
1799     }
1800   else if (a->use != 2)
1801     log_bug ("flush on non-output iobuf\n");
1802   else if (!a->filter)
1803     log_bug ("iobuf_flush: no filter\n");
1804   len = a->d.len;
1805   rc = a->filter (a->filter_ov, IOBUFCTRL_FLUSH, a->chain, a->d.buf, &len);
1806   if (!rc && len != a->d.len)
1807     {
1808       log_info ("iobuf_flush did not write all!\n");
1809       rc = GPG_ERR_INTERNAL;
1810     }
1811   else if (rc)
1812     a->error = rc;
1813   a->d.len = 0;
1814
1815   return rc;
1816 }
1817
1818
1819 /****************
1820  * Read a byte from the iobuf; returns -1 on EOF
1821  */
1822 int
1823 iobuf_readbyte (iobuf_t a)
1824 {
1825   int c;
1826
1827   if (a->nlimit && a->nbytes >= a->nlimit)
1828     return -1;                  /* forced EOF */
1829
1830   if (a->d.start < a->d.len)
1831     {
1832       c = a->d.buf[a->d.start++];
1833     }
1834   else if ((c = underflow (a)) == -1)
1835     return -1;                  /* EOF */
1836
1837   a->nbytes++;
1838   return c;
1839 }
1840
1841
1842 int
1843 iobuf_read (iobuf_t a, void *buffer, unsigned int buflen)
1844 {
1845   unsigned char *buf = (unsigned char *)buffer;
1846   int c, n;
1847
1848   if (a->nlimit)
1849     {
1850       /* Handle special cases. */
1851       for (n = 0; n < buflen; n++)
1852         {
1853           if ((c = iobuf_readbyte (a)) == -1)
1854             {
1855               if (!n)
1856                 return -1;      /* eof */
1857               break;
1858             }
1859           else if (buf)
1860             *buf = c;
1861           if (buf)
1862             buf++;
1863         }
1864       return n;
1865     }
1866
1867   n = 0;
1868   do
1869     {
1870       if (n < buflen && a->d.start < a->d.len)
1871         {
1872           unsigned size = a->d.len - a->d.start;
1873           if (size > buflen - n)
1874             size = buflen - n;
1875           if (buf)
1876             memcpy (buf, a->d.buf + a->d.start, size);
1877           n += size;
1878           a->d.start += size;
1879           if (buf)
1880             buf += size;
1881         }
1882       if (n < buflen)
1883         {
1884           if ((c = underflow (a)) == -1)
1885             {
1886               a->nbytes += n;
1887               return n ? n : -1 /*EOF*/;
1888             }
1889           if (buf)
1890             *buf++ = c;
1891           n++;
1892         }
1893     }
1894   while (n < buflen);
1895   a->nbytes += n;
1896   return n;
1897 }
1898
1899
1900
1901 /****************
1902  * Have a look at the iobuf.
1903  * NOTE: This only works in special cases.
1904  */
1905 int
1906 iobuf_peek (iobuf_t a, byte * buf, unsigned buflen)
1907 {
1908   int n = 0;
1909
1910   if (a->filter_eof)
1911     return -1;
1912
1913   if (!(a->d.start < a->d.len))
1914     {
1915       if (underflow (a) == -1)
1916         return -1;
1917       /* And unget this character. */
1918       assert (a->d.start == 1);
1919       a->d.start = 0;
1920     }
1921
1922   for (n = 0; n < buflen && (a->d.start + n) < a->d.len; n++, buf++)
1923     *buf = a->d.buf[n];
1924   return n;
1925 }
1926
1927
1928
1929
1930 int
1931 iobuf_writebyte (iobuf_t a, unsigned int c)
1932 {
1933   int rc;
1934
1935   if (a->directfp)
1936     BUG ();
1937
1938   if (a->d.len == a->d.size)
1939     if ((rc=iobuf_flush (a)))
1940       return rc;
1941
1942   assert (a->d.len < a->d.size);
1943   a->d.buf[a->d.len++] = c;
1944   return 0;
1945 }
1946
1947
1948 int
1949 iobuf_write (iobuf_t a, const void *buffer, unsigned int buflen)
1950 {
1951   const unsigned char *buf = (const unsigned char *)buffer;
1952   int rc;
1953
1954   if (a->directfp)
1955     BUG ();
1956
1957   do
1958     {
1959       if (buflen && a->d.len < a->d.size)
1960         {
1961           unsigned size = a->d.size - a->d.len;
1962           if (size > buflen)
1963             size = buflen;
1964           memcpy (a->d.buf + a->d.len, buf, size);
1965           buflen -= size;
1966           buf += size;
1967           a->d.len += size;
1968         }
1969       if (buflen)
1970         {
1971           rc = iobuf_flush (a);
1972           if (rc)
1973             return rc;
1974         }
1975     }
1976   while (buflen);
1977   return 0;
1978 }
1979
1980
1981 int
1982 iobuf_writestr (iobuf_t a, const char *buf)
1983 {
1984   int rc;
1985
1986   for (; *buf; buf++)
1987     if ((rc=iobuf_writebyte (a, *buf)))
1988       return rc;
1989   return 0;
1990 }
1991
1992
1993
1994 /****************
1995  * copy the contents of TEMP to A.
1996  */
1997 int
1998 iobuf_write_temp (iobuf_t a, iobuf_t temp)
1999 {
2000   while (temp->chain)
2001     pop_filter (temp, temp->filter, NULL);
2002   return iobuf_write (a, temp->d.buf, temp->d.len);
2003 }
2004
2005 /****************
2006  * copy the contents of the temp io stream to BUFFER.
2007  */
2008 size_t
2009 iobuf_temp_to_buffer (iobuf_t a, byte * buffer, size_t buflen)
2010 {
2011   size_t n = a->d.len;
2012
2013   if (n > buflen)
2014     n = buflen;
2015   memcpy (buffer, a->d.buf, n);
2016   return n;
2017 }
2018
2019
2020 /****************
2021  * Call this function to terminate processing of the temp stream
2022  * without closing it.  This removes all filters from the stream
2023  * makes sure that iobuf_get_temp_{buffer,length}() returns correct
2024  * values.
2025  */
2026 void
2027 iobuf_flush_temp (iobuf_t temp)
2028 {
2029   while (temp->chain)
2030     pop_filter (temp, temp->filter, NULL);
2031 }
2032
2033
2034 /****************
2035  * Set a limit on how many bytes may be read from the input stream A.
2036  * Setting the limit to 0 disables this feature.
2037  */
2038 void
2039 iobuf_set_limit (iobuf_t a, off_t nlimit)
2040 {
2041   if (nlimit)
2042     a->nofast |= 1;
2043   else
2044     a->nofast &= ~1;
2045   a->nlimit = nlimit;
2046   a->ntotal += a->nbytes;
2047   a->nbytes = 0;
2048 }
2049
2050
2051
2052 /* Return the length of an open file A.  IF OVERFLOW is not NULL it
2053    will be set to true if the file is larger than what off_t can cope
2054    with.  The function return 0 on error or on overflow condition.  */
2055 off_t
2056 iobuf_get_filelength (iobuf_t a, int *overflow)
2057 {
2058   struct stat st;
2059
2060   if (overflow)
2061     *overflow = 0;
2062   
2063   if ( a->directfp )  
2064     {
2065       FILE *fp = a->directfp;
2066       
2067       if ( !fstat(fileno(fp), &st) )
2068         return st.st_size;
2069       log_error("fstat() failed: %s\n", strerror(errno) );
2070       return 0;
2071     }
2072   
2073   /* Hmmm: file_filter may have already been removed */
2074   for ( ; a; a = a->chain )
2075     if ( !a->chain && a->filter == file_filter )
2076       {
2077         file_filter_ctx_t *b = a->filter_ov;
2078         fp_or_fd_t fp = b->fp;
2079         
2080 #if defined(HAVE_W32_SYSTEM) && !defined(FILE_FILTER_USES_STDIO)
2081         ulong size;
2082         static int (* __stdcall get_file_size_ex) (void *handle,
2083                                                    LARGE_INTEGER *r_size);
2084         static int get_file_size_ex_initialized;
2085
2086         if (!get_file_size_ex_initialized)
2087           {
2088             void *handle;
2089             
2090             handle = dlopen ("kernel32.dll", RTLD_LAZY);
2091             if (handle)
2092               {
2093                 get_file_size_ex = dlsym (handle, "GetFileSizeEx");
2094                 if (!get_file_size_ex)
2095                   dlclose (handle);
2096               }
2097             get_file_size_ex_initialized = 1;
2098           }
2099         
2100         if (get_file_size_ex)
2101           {
2102             /* This is a newer system with GetFileSizeEx; we use this
2103                then because it seem that GetFileSize won't return a
2104                proper error in case a file is larger than 4GB. */
2105             LARGE_INTEGER exsize;
2106             
2107             if (get_file_size_ex (fp, &exsize))
2108               {
2109                 if (!exsize.u.HighPart)
2110                   return exsize.u.LowPart;
2111                 if (overflow)
2112                   *overflow = 1;
2113                 return 0; 
2114               }
2115           }
2116         else
2117           {
2118             if ((size=GetFileSize (fp, NULL)) != 0xffffffff)
2119               return size;
2120           }
2121         log_error ("GetFileSize for handle %p failed: %s\n",
2122                    fp, w32_strerror (0));
2123 #else
2124         if ( !fstat(my_fileno(fp), &st) )
2125           return st.st_size;
2126         log_error("fstat() failed: %s\n", strerror(errno) );
2127 #endif
2128         break/*the for loop*/;
2129       }
2130   
2131     return 0;
2132 }
2133
2134
2135 /* Return the file descriptor of the underlying file or -1 if it is
2136    not available.  */
2137 int 
2138 iobuf_get_fd (iobuf_t a)
2139 {
2140   if (a->directfp)
2141     return fileno ( (FILE*)a->directfp );
2142
2143   for ( ; a; a = a->chain )
2144     if (!a->chain && a->filter == file_filter)
2145       {
2146         file_filter_ctx_t *b = a->filter_ov;
2147         fp_or_fd_t fp = b->fp;
2148
2149         return my_fileno (fp);
2150       }
2151
2152   return -1;
2153 }
2154
2155
2156
2157 /****************
2158  * Tell the file position, where the next read will take place
2159  */
2160 off_t
2161 iobuf_tell (iobuf_t a)
2162 {
2163   return a->ntotal + a->nbytes;
2164 }
2165
2166
2167 #if !defined(HAVE_FSEEKO) && !defined(fseeko)
2168
2169 #ifdef HAVE_LIMITS_H
2170 # include <limits.h>
2171 #endif
2172 #ifndef LONG_MAX
2173 # define LONG_MAX ((long) ((unsigned long) -1 >> 1))
2174 #endif
2175 #ifndef LONG_MIN
2176 # define LONG_MIN (-1 - LONG_MAX)
2177 #endif
2178
2179 /****************
2180  * A substitute for fseeko, for hosts that don't have it.
2181  */
2182 static int
2183 fseeko (FILE * stream, off_t newpos, int whence)
2184 {
2185   while (newpos != (long) newpos)
2186     {
2187       long pos = newpos < 0 ? LONG_MIN : LONG_MAX;
2188       if (fseek (stream, pos, whence) != 0)
2189         return -1;
2190       newpos -= pos;
2191       whence = SEEK_CUR;
2192     }
2193   return fseek (stream, (long) newpos, whence);
2194 }
2195 #endif
2196
2197 /****************
2198  * This is a very limited implementation. It simply discards all internal
2199  * buffering and removes all filters but the first one.
2200  */
2201 int
2202 iobuf_seek (iobuf_t a, off_t newpos)
2203 {
2204   file_filter_ctx_t *b = NULL;
2205
2206   if (a->directfp)
2207     {
2208       FILE *fp = a->directfp;
2209       if (fseeko (fp, newpos, SEEK_SET))
2210         {
2211           log_error ("can't seek: %s\n", strerror (errno));
2212           return -1;
2213         }
2214       clearerr (fp);
2215     }
2216   else
2217     {
2218       for (; a; a = a->chain)
2219         {
2220           if (!a->chain && a->filter == file_filter)
2221             {
2222               b = a->filter_ov;
2223               break;
2224             }
2225         }
2226       if (!a)
2227         return -1;
2228 #ifdef FILE_FILTER_USES_STDIO
2229       if (fseeko (b->fp, newpos, SEEK_SET))
2230         {
2231           log_error ("can't fseek: %s\n", strerror (errno));
2232           return -1;
2233         }
2234 #else
2235 #ifdef HAVE_W32_SYSTEM
2236       if (SetFilePointer (b->fp, newpos, NULL, FILE_BEGIN) == 0xffffffff)
2237         {
2238           log_error ("SetFilePointer failed on handle %p: ec=%d\n",
2239                      b->fp, (int) GetLastError ());
2240           return -1;
2241         }
2242 #else
2243       if (lseek (b->fp, newpos, SEEK_SET) == (off_t) - 1)
2244         {
2245           log_error ("can't lseek: %s\n", strerror (errno));
2246           return -1;
2247         }
2248 #endif
2249 #endif
2250     }
2251   a->d.len = 0;                 /* discard buffer */
2252   a->d.start = 0;
2253   a->nbytes = 0;
2254   a->nlimit = 0;
2255   a->nofast &= ~1;
2256   a->ntotal = newpos;
2257   a->error = 0;
2258   /* remove filters, but the last */
2259   if (a->chain)
2260     log_debug ("pop_filter called in iobuf_seek - please report\n");
2261   while (a->chain)
2262     pop_filter (a, a->filter, NULL);
2263
2264   return 0;
2265 }
2266
2267
2268
2269
2270
2271
2272 /****************
2273  * Retrieve the real filename
2274  */
2275 const char *
2276 iobuf_get_real_fname (iobuf_t a)
2277 {
2278   if (a->real_fname)
2279     return a->real_fname;
2280
2281   /* the old solution */
2282   for (; a; a = a->chain)
2283     if (!a->chain && a->filter == file_filter)
2284       {
2285         file_filter_ctx_t *b = a->filter_ov;
2286         return b->print_only_name ? NULL : b->fname;
2287       }
2288
2289   return NULL;
2290 }
2291
2292
2293 /****************
2294  * Retrieve the filename
2295  */
2296 const char *
2297 iobuf_get_fname (iobuf_t a)
2298 {
2299   for (; a; a = a->chain)
2300     if (!a->chain && a->filter == file_filter)
2301       {
2302         file_filter_ctx_t *b = a->filter_ov;
2303         return b->fname;
2304       }
2305
2306   return NULL;
2307 }
2308
2309
2310 /****************
2311  * enable partial block mode as described in the OpenPGP draft.
2312  * LEN is the first length byte on read, but ignored on writes.
2313  */
2314 void
2315 iobuf_set_partial_block_mode (iobuf_t a, size_t len)
2316 {
2317   block_filter_ctx_t *ctx = xcalloc (1, sizeof *ctx);
2318
2319   assert (a->use == 1 || a->use == 2);
2320   ctx->use = a->use;
2321   if (!len)
2322     {
2323       if (a->use == 1)
2324         log_debug ("pop_filter called in set_partial_block_mode"
2325                    " - please report\n");
2326       pop_filter (a, block_filter, NULL);
2327     }
2328   else
2329     {
2330       ctx->partial = 1;
2331       ctx->size = 0;
2332       ctx->first_c = len;
2333       iobuf_push_filter (a, block_filter, ctx);
2334     }
2335 }
2336
2337
2338
2339 /****************
2340  * Same as fgets() but if the buffer is too short a larger one will
2341  * be allocated up to some limit *max_length.
2342  * A line is considered a byte stream ending in a LF.
2343  * Returns the length of the line. EOF is indicated by a line of
2344  * length zero. The last LF may be missing due to an EOF.
2345  * is max_length is zero on return, the line has been truncated.
2346  *
2347  * Note: The buffer is allocated with enough space to append a CR,LF,EOL
2348  */
2349 unsigned int
2350 iobuf_read_line (iobuf_t a, byte ** addr_of_buffer,
2351                  unsigned *length_of_buffer, unsigned *max_length)
2352 {
2353   int c;
2354   char *buffer = (char *)*addr_of_buffer;
2355   unsigned length = *length_of_buffer;
2356   unsigned nbytes = 0;
2357   unsigned maxlen = *max_length;
2358   char *p;
2359
2360   if (!buffer)
2361     {                           /* must allocate a new buffer */
2362       length = 256;
2363       buffer = xmalloc (length);
2364       *addr_of_buffer = (unsigned char *)buffer;
2365       *length_of_buffer = length;
2366     }
2367
2368   length -= 3;                  /* reserve 3 bytes (cr,lf,eol) */
2369   p = buffer;
2370   while ((c = iobuf_get (a)) != -1)
2371     {
2372       if (nbytes == length)
2373         {                       /* increase the buffer */
2374           if (length > maxlen)
2375             {                   /* this is out limit */
2376               /* skip the rest of the line */
2377               while (c != '\n' && (c = iobuf_get (a)) != -1)
2378                 ;
2379               *p++ = '\n';      /* always append a LF (we have reserved space) */
2380               nbytes++;
2381               *max_length = 0;  /* indicate truncation */
2382               break;
2383             }
2384           length += 3;          /* correct for the reserved byte */
2385           length += length < 1024 ? 256 : 1024;
2386           buffer = xrealloc (buffer, length);
2387           *addr_of_buffer = (unsigned char *)buffer;
2388           *length_of_buffer = length;
2389           length -= 3;          /* and reserve again */
2390           p = buffer + nbytes;
2391         }
2392       *p++ = c;
2393       nbytes++;
2394       if (c == '\n')
2395         break;
2396     }
2397   *p = 0;                       /* make sure the line is a string */
2398
2399   return nbytes;
2400 }
2401
2402 static int
2403 translate_file_handle (int fd, int for_write)
2404 {
2405 #ifdef HAVE_W32_SYSTEM
2406 # ifdef FILE_FILTER_USES_STDIO
2407   fd = translate_sys2libc_fd (fd, for_write);
2408 # else
2409   {
2410     int x;
2411
2412     if (fd == 0)
2413       x = (int) GetStdHandle (STD_INPUT_HANDLE);
2414     else if (fd == 1)
2415       x = (int) GetStdHandle (STD_OUTPUT_HANDLE);
2416     else if (fd == 2)
2417       x = (int) GetStdHandle (STD_ERROR_HANDLE);
2418     else
2419       x = fd;
2420
2421     if (x == -1)
2422       log_debug ("GetStdHandle(%d) failed: ec=%d\n",
2423                  fd, (int) GetLastError ());
2424
2425     fd = x;
2426   }
2427 # endif
2428 #endif
2429   return fd;
2430 }
2431
2432
2433 void
2434 iobuf_skip_rest (iobuf_t a, unsigned long n, int partial)
2435 {
2436   if ( partial )
2437     {
2438       for (;;)
2439         {
2440           if (a->nofast || a->d.start >= a->d.len) 
2441             {
2442               if (iobuf_readbyte (a) == -1)
2443                 {
2444                   break;
2445                 }
2446             } 
2447           else
2448             {
2449               unsigned long count = a->d.len - a->d.start;
2450               a->nbytes += count;
2451               a->d.start = a->d.len;
2452             }
2453         }
2454     } 
2455   else
2456     {
2457       unsigned long remaining = n;
2458       while (remaining > 0) 
2459         {
2460           if (a->nofast || a->d.start >= a->d.len)
2461             {
2462               if (iobuf_readbyte (a) == -1)
2463                 {
2464                   break;
2465                 }
2466               --remaining;
2467             } 
2468           else 
2469             {
2470               unsigned long count = a->d.len - a->d.start;
2471               if (count > remaining) 
2472                 {
2473                   count = remaining;
2474                 }
2475               a->nbytes += count;
2476               a->d.start += count;
2477               remaining -= count;
2478             }
2479         }
2480     }
2481 }