1997-11-18 14:06:00 +00:00
|
|
|
/* iobuf.c - file handling
|
1998-02-24 18:50:46 +00:00
|
|
|
* Copyright (C) 1998 Free Software Foundation, Inc.
|
1997-11-18 14:06:00 +00:00
|
|
|
*
|
1998-02-24 18:50:46 +00:00
|
|
|
* This file is part of GNUPG.
|
1997-11-18 14:06:00 +00:00
|
|
|
*
|
1998-02-24 18:50:46 +00:00
|
|
|
* GNUPG is free software; you can redistribute it and/or modify
|
1997-11-18 14:06:00 +00:00
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
1998-02-24 18:50:46 +00:00
|
|
|
* GNUPG is distributed in the hope that it will be useful,
|
1997-11-18 14:06:00 +00:00
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <config.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <errno.h>
|
|
|
|
#include <assert.h>
|
1998-05-13 17:53:36 +00:00
|
|
|
#include <sys/types.h>
|
1997-11-18 14:06:00 +00:00
|
|
|
#include <sys/stat.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
|
|
|
|
#include "memory.h"
|
|
|
|
#include "util.h"
|
|
|
|
#include "iobuf.h"
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
FILE *fp; /* open file handle */
|
1998-09-11 05:47:32 +00:00
|
|
|
int print_only_name; /* flags indicating that fname is not a real file*/
|
1997-11-18 14:06:00 +00:00
|
|
|
char fname[1]; /* name of the file */
|
|
|
|
} file_filter_ctx_t ;
|
|
|
|
|
1998-07-02 19:31:46 +00:00
|
|
|
/* The first partial length header block must be of size 512
|
|
|
|
* to make it easier (and efficienter) we use a min. block size of 512
|
|
|
|
* for all chznks (but the last one) */
|
|
|
|
#define OP_MIN_PARTIAL_CHUNK 512
|
|
|
|
#define OP_MIN_PARTIAL_CHUNK_2POW 9
|
|
|
|
|
1997-11-18 14:06:00 +00:00
|
|
|
typedef struct {
|
|
|
|
int usage;
|
|
|
|
size_t size;
|
|
|
|
size_t count;
|
1998-01-16 21:15:24 +00:00
|
|
|
int partial; /* 1 = partial header, 2 in last partial packet */
|
1998-07-02 19:31:46 +00:00
|
|
|
char *buffer; /* used for partial header */
|
|
|
|
size_t buflen; /* used size of buffer */
|
|
|
|
int first_c; /* of partial header (which is > 0)*/
|
1997-11-18 14:06:00 +00:00
|
|
|
int eof;
|
|
|
|
} block_filter_ctx_t;
|
|
|
|
|
1998-07-02 19:31:46 +00:00
|
|
|
|
1997-11-18 14:06:00 +00:00
|
|
|
static int underflow(IOBUF a);
|
1998-09-11 05:47:32 +00:00
|
|
|
static const char *get_real_fname( IOBUF a );
|
1997-11-18 14:06:00 +00:00
|
|
|
|
|
|
|
/****************
|
|
|
|
* Read data from a file into buf which has an allocated length of *LEN.
|
|
|
|
* return the number of read bytes in *LEN. OPAQUE is the FILE * of
|
|
|
|
* the stream. A is not used.
|
1998-04-14 17:51:16 +00:00
|
|
|
* control may be:
|
1997-11-18 14:06:00 +00:00
|
|
|
* IOBUFCTRL_INIT: called just before the function is linked into the
|
|
|
|
* list of function. This can be used to prepare internal
|
|
|
|
* data structures of the function.
|
|
|
|
* IOBUFCTRL_FREE: called just before the function is removed from the
|
|
|
|
* list of functions and can be used to release internal
|
|
|
|
* data structures or close a file etc.
|
|
|
|
* IOBUFCTRL_UNDERFLOW: called by iobuf_underflow to fill the buffer
|
|
|
|
* with new stuff. *RET_LEN is the available size of the
|
|
|
|
* buffer, and should be set to the number of bytes
|
|
|
|
* which were put into the buffer. The function
|
|
|
|
* returns 0 to indicate success, -1 on EOF and
|
|
|
|
* G10ERR_xxxxx for other errors.
|
|
|
|
*
|
|
|
|
* IOBUFCTRL_FLUSH: called by iobuf_flush() to write out the collected stuff.
|
|
|
|
* *RET_LAN is the number of bytes in BUF.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
static int
|
|
|
|
file_filter(void *opaque, int control, IOBUF chain, byte *buf, size_t *ret_len)
|
|
|
|
{
|
|
|
|
file_filter_ctx_t *a = opaque;
|
|
|
|
FILE *fp = a->fp;
|
|
|
|
size_t size = *ret_len;
|
|
|
|
size_t nbytes = 0;
|
|
|
|
int c, rc = 0;
|
|
|
|
char *p;
|
|
|
|
|
|
|
|
if( control == IOBUFCTRL_UNDERFLOW ) {
|
|
|
|
assert( size ); /* need a buffer */
|
|
|
|
for(; size; size-- ) {
|
|
|
|
if( (c=getc(fp)) == EOF ) {
|
|
|
|
if( ferror(fp) ) {
|
|
|
|
log_error("%s: read error: %s\n",
|
|
|
|
a->fname, strerror(errno));
|
|
|
|
rc = G10ERR_READ_FILE;
|
|
|
|
}
|
|
|
|
else if( !nbytes )
|
|
|
|
rc = -1; /* okay: we can return EOF now. */
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
buf[nbytes++] = c & 0xff;
|
|
|
|
}
|
|
|
|
*ret_len = nbytes;
|
|
|
|
}
|
|
|
|
else if( control == IOBUFCTRL_FLUSH ) {
|
|
|
|
for(p=buf; nbytes < size; nbytes++, p++ ) {
|
|
|
|
if( putc(*p, fp) == EOF ) {
|
|
|
|
log_error("%s: write error: %s\n",
|
|
|
|
a->fname, strerror(errno));
|
|
|
|
rc = G10ERR_WRITE_FILE;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
*ret_len = nbytes;
|
|
|
|
}
|
|
|
|
else if( control == IOBUFCTRL_INIT ) {
|
|
|
|
}
|
|
|
|
else if( control == IOBUFCTRL_DESC ) {
|
|
|
|
*(char**)buf = "file_filter";
|
|
|
|
}
|
|
|
|
else if( control == IOBUFCTRL_FREE ) {
|
|
|
|
if( fp != stdin && fp != stdout )
|
|
|
|
fclose(fp);
|
|
|
|
fp = NULL;
|
|
|
|
m_free(a); /* we can free our context now */
|
|
|
|
}
|
|
|
|
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* This is used to implement the block write mode.
|
|
|
|
* Block reading is done on a byte by byte basis in readbyte(),
|
|
|
|
* without a filter
|
|
|
|
*/
|
|
|
|
static int
|
|
|
|
block_filter(void *opaque, int control, IOBUF chain, byte *buf, size_t *ret_len)
|
|
|
|
{
|
|
|
|
block_filter_ctx_t *a = opaque;
|
|
|
|
size_t size = *ret_len;
|
|
|
|
int c, rc = 0;
|
|
|
|
char *p;
|
|
|
|
|
|
|
|
if( control == IOBUFCTRL_UNDERFLOW ) {
|
|
|
|
size_t n=0;
|
|
|
|
|
|
|
|
p = buf;
|
|
|
|
assert( size ); /* need a buffer */
|
|
|
|
if( a->eof ) /* don't read any further */
|
|
|
|
rc = -1;
|
|
|
|
while( !rc && size ) {
|
|
|
|
if( !a->size ) { /* get the length bytes */
|
1998-01-16 21:15:24 +00:00
|
|
|
if( a->partial == 2 ) {
|
1997-11-18 14:06:00 +00:00
|
|
|
a->eof = 1;
|
|
|
|
if( !n )
|
|
|
|
rc = -1;
|
|
|
|
break;
|
|
|
|
}
|
1998-01-16 21:15:24 +00:00
|
|
|
else if( a->partial ) {
|
1998-07-02 19:31:46 +00:00
|
|
|
/* These OpenPGP introduced huffman encoded length
|
|
|
|
* bytes are really a mess :-( */
|
|
|
|
if( a->first_c ) {
|
|
|
|
c = a->first_c;
|
|
|
|
a->first_c = 0;
|
|
|
|
assert( c >= 224 && c < 255 );
|
|
|
|
}
|
|
|
|
else if( (c = iobuf_get(chain)) == -1 ) {
|
1998-01-16 21:15:24 +00:00
|
|
|
log_error("block_filter: 1st length byte missing\n");
|
|
|
|
rc = G10ERR_READ_FILE;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if( c < 192 ) {
|
|
|
|
a->size = c;
|
|
|
|
a->partial = 2;
|
|
|
|
if( !a->size ) {
|
|
|
|
a->eof = 1;
|
|
|
|
if( !n )
|
|
|
|
rc = -1;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else if( c < 224 ) {
|
|
|
|
a->size = (c - 192) * 256;
|
|
|
|
if( (c = iobuf_get(chain)) == -1 ) {
|
|
|
|
log_error("block_filter: 2nd length byte missing\n");
|
|
|
|
rc = G10ERR_READ_FILE;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
a->size += c + 192;
|
|
|
|
a->partial = 2;
|
|
|
|
if( !a->size ) {
|
|
|
|
a->eof = 1;
|
|
|
|
if( !n )
|
|
|
|
rc = -1;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
1998-07-02 19:31:46 +00:00
|
|
|
else if( c == 255 ) {
|
|
|
|
a->size = iobuf_get(chain) << 24;
|
|
|
|
a->size |= iobuf_get(chain) << 16;
|
|
|
|
a->size |= iobuf_get(chain) << 8;
|
|
|
|
if( (c = iobuf_get(chain)) == -1 ) {
|
|
|
|
log_error("block_filter: invalid 4 byte length\n");
|
|
|
|
rc = G10ERR_READ_FILE;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
a->size |= c;
|
|
|
|
}
|
1998-01-16 21:15:24 +00:00
|
|
|
else { /* next partial body length */
|
|
|
|
a->size = 1 << (c & 0x1f);
|
|
|
|
}
|
|
|
|
}
|
1998-07-02 19:31:46 +00:00
|
|
|
else { /* the gnupg partial length scheme - much better :-) */
|
1998-01-16 21:15:24 +00:00
|
|
|
c = iobuf_get(chain);
|
|
|
|
a->size = c << 8;
|
|
|
|
c = iobuf_get(chain);
|
|
|
|
a->size |= c;
|
|
|
|
if( c == -1 ) {
|
|
|
|
log_error("block_filter: error reading length info\n");
|
|
|
|
rc = G10ERR_READ_FILE;
|
|
|
|
}
|
|
|
|
if( !a->size ) {
|
|
|
|
a->eof = 1;
|
|
|
|
if( !n )
|
|
|
|
rc = -1;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
for(; !rc && size && a->size; size--, a->size-- ) {
|
|
|
|
if( (c=iobuf_get(chain)) == -1 ) {
|
|
|
|
log_error("block_filter %p: read error (size=%lu,a->size=%lu)\n",
|
|
|
|
a, (ulong)size, (ulong)a->size);
|
|
|
|
rc = G10ERR_READ_FILE;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
*p++ = c;
|
|
|
|
n++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
*ret_len = n;
|
|
|
|
}
|
|
|
|
else if( control == IOBUFCTRL_FLUSH ) {
|
1998-07-02 19:31:46 +00:00
|
|
|
if( a->partial ) { /* the complicated openpgp scheme */
|
|
|
|
size_t blen, n, nbytes = size + a->buflen;
|
|
|
|
|
|
|
|
assert( a->buflen <= OP_MIN_PARTIAL_CHUNK );
|
|
|
|
if( nbytes < OP_MIN_PARTIAL_CHUNK ) {
|
|
|
|
/* not enough to write a partial block out , so we store it*/
|
|
|
|
if( !a->buffer )
|
|
|
|
a->buffer = m_alloc( OP_MIN_PARTIAL_CHUNK );
|
|
|
|
memcpy( a->buffer + a->buflen, buf, size );
|
|
|
|
a->buflen += size;
|
|
|
|
}
|
|
|
|
else { /* okay, we can write out something */
|
|
|
|
/* do this in a loop to use the most efficient block lengths */
|
|
|
|
p = buf;
|
|
|
|
do {
|
|
|
|
/* find the best matching block length - this is limited
|
|
|
|
* by the size of the internal buffering */
|
|
|
|
for( blen=OP_MIN_PARTIAL_CHUNK*2,
|
|
|
|
c=OP_MIN_PARTIAL_CHUNK_2POW+1; blen < nbytes;
|
|
|
|
blen *=2, c++ )
|
|
|
|
;
|
|
|
|
blen /= 2; c--;
|
|
|
|
/* write the partial length header */
|
|
|
|
assert( c <= 0x1f ); /*;-)*/
|
|
|
|
c |= 0xe0;
|
|
|
|
iobuf_put( chain, c );
|
|
|
|
if( (n=a->buflen) ) { /* write stuff from the buffer */
|
|
|
|
assert( n == OP_MIN_PARTIAL_CHUNK);
|
|
|
|
if( iobuf_write(chain, a->buffer, n ) )
|
|
|
|
rc = G10ERR_WRITE_FILE;
|
|
|
|
a->buflen = 0;
|
|
|
|
nbytes -= n;
|
|
|
|
}
|
|
|
|
if( (n = nbytes) > blen )
|
|
|
|
n = blen;
|
|
|
|
if( n && iobuf_write(chain, p, n ) )
|
|
|
|
rc = G10ERR_WRITE_FILE;
|
|
|
|
p += n;
|
|
|
|
nbytes -= n;
|
|
|
|
} while( !rc && nbytes >= OP_MIN_PARTIAL_CHUNK );
|
|
|
|
/* store the rest in the buffer */
|
|
|
|
if( !rc && nbytes ) {
|
|
|
|
assert( !a->buflen );
|
|
|
|
assert( nbytes < OP_MIN_PARTIAL_CHUNK );
|
|
|
|
if( !a->buffer )
|
|
|
|
a->buffer = m_alloc( OP_MIN_PARTIAL_CHUNK );
|
|
|
|
memcpy( a->buffer, p, nbytes );
|
|
|
|
a->buflen = nbytes;
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
1998-07-02 19:31:46 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else { /* the gnupg scheme */
|
|
|
|
size_t avail, n;
|
|
|
|
|
|
|
|
for(p=buf; !rc && size; ) {
|
|
|
|
n = size;
|
|
|
|
avail = a->size - a->count;
|
|
|
|
if( !avail ) {
|
|
|
|
if( n > a->size ) {
|
|
|
|
iobuf_put( chain, (a->size >> 8) & 0xff );
|
|
|
|
iobuf_put( chain, a->size & 0xff );
|
|
|
|
avail = a->size;
|
|
|
|
a->count = 0;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
iobuf_put( chain, (n >> 8) & 0xff );
|
|
|
|
iobuf_put( chain, n & 0xff );
|
|
|
|
avail = n;
|
|
|
|
a->count = a->size - n;
|
|
|
|
}
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
1998-07-02 19:31:46 +00:00
|
|
|
if( n > avail )
|
|
|
|
n = avail;
|
|
|
|
if( iobuf_write(chain, p, n ) )
|
|
|
|
rc = G10ERR_WRITE_FILE;
|
|
|
|
a->count += n;
|
|
|
|
p += n;
|
|
|
|
size -= n;
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else if( control == IOBUFCTRL_INIT ) {
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("init block_filter %p\n", a );
|
1998-01-16 21:15:24 +00:00
|
|
|
if( a->partial )
|
|
|
|
a->count = 0;
|
|
|
|
else if( a->usage == 1 )
|
1997-11-18 14:06:00 +00:00
|
|
|
a->count = a->size = 0;
|
|
|
|
else
|
|
|
|
a->count = a->size; /* force first length bytes */
|
|
|
|
a->eof = 0;
|
1998-07-02 19:31:46 +00:00
|
|
|
a->buffer = NULL;
|
|
|
|
a->buflen = 0;
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
|
|
|
else if( control == IOBUFCTRL_DESC ) {
|
|
|
|
*(char**)buf = "block_filter";
|
|
|
|
}
|
|
|
|
else if( control == IOBUFCTRL_FREE ) {
|
|
|
|
if( a->usage == 2 ) { /* write the end markers */
|
1998-01-16 21:15:24 +00:00
|
|
|
if( a->partial ) {
|
1998-07-02 19:31:46 +00:00
|
|
|
u32 len;
|
|
|
|
/* write out the remaining bytes without a partial header
|
|
|
|
* the length of this header may be 0 - but if it is
|
|
|
|
* the first block we are not allowed to use a partial header
|
|
|
|
* and frankly we can't do so, because this length must be
|
|
|
|
* a power of 2. This is _really_ complicated because we
|
|
|
|
* have to check the possible length of a packet prior
|
|
|
|
* to it's creation: a chein of filters becomes complicated
|
|
|
|
* and we need a lot of code to handle compressed packets etc.
|
|
|
|
* :-(((((((
|
|
|
|
*/
|
|
|
|
/* construct header */
|
|
|
|
len = a->buflen;
|
|
|
|
if( len < 192 )
|
|
|
|
rc = iobuf_put(chain, len );
|
|
|
|
else if( len < 8384 ) {
|
|
|
|
if( !(rc=iobuf_put( chain, ((len-192) / 256) + 192)) )
|
|
|
|
rc = iobuf_put( chain, ((len-192) % 256));
|
|
|
|
}
|
|
|
|
else { /* use a 4 byte header */
|
|
|
|
if( !(rc=iobuf_put( chain, 0xff )) )
|
|
|
|
if( !(rc=iobuf_put( chain, (len >> 24)&0xff )) )
|
|
|
|
if( !(rc=iobuf_put( chain, (len >> 16)&0xff )) )
|
|
|
|
if( !(rc=iobuf_put( chain, (len >> 8)&0xff )))
|
|
|
|
rc=iobuf_put( chain, len & 0xff );
|
|
|
|
}
|
|
|
|
if( !rc && len )
|
|
|
|
rc = iobuf_write(chain, a->buffer, len );
|
|
|
|
if( rc ) {
|
|
|
|
log_error("block_filter: write error: %s\n",strerror(errno));
|
|
|
|
rc = G10ERR_WRITE_FILE;
|
|
|
|
}
|
|
|
|
m_free( a->buffer ); a->buffer = NULL; a->buflen = 0;
|
1998-01-16 21:15:24 +00:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
iobuf_writebyte(chain, 0);
|
|
|
|
iobuf_writebyte(chain, 0);
|
|
|
|
}
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
|
|
|
else if( a->size ) {
|
|
|
|
log_error("block_filter: pending bytes!\n");
|
|
|
|
}
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("free block_filter %p\n", a );
|
|
|
|
m_free(a); /* we can free our context now */
|
|
|
|
}
|
|
|
|
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* Allocate a new io buffer, with no function assigned.
|
|
|
|
* Usage is the desired usage: 1 for input, 2 for output, 3 for temp buffer
|
|
|
|
* BUFSIZE is a suggested buffer size.
|
|
|
|
*/
|
|
|
|
IOBUF
|
|
|
|
iobuf_alloc(int usage, size_t bufsize)
|
|
|
|
{
|
|
|
|
IOBUF a;
|
|
|
|
static int number=0;
|
|
|
|
|
|
|
|
a = m_alloc_clear(sizeof *a);
|
|
|
|
a->usage = usage;
|
|
|
|
a->d.buf = m_alloc( bufsize );
|
|
|
|
a->d.size = bufsize;
|
|
|
|
a->no = ++number;
|
|
|
|
a->subno = 0;
|
1997-11-21 14:53:57 +00:00
|
|
|
a->opaque = NULL;
|
1997-11-18 14:06:00 +00:00
|
|
|
return a;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
iobuf_close( IOBUF a )
|
|
|
|
{
|
|
|
|
IOBUF a2;
|
|
|
|
size_t dummy_len;
|
|
|
|
int rc=0;
|
|
|
|
|
1998-02-11 03:25:44 +00:00
|
|
|
for( ; a && !rc ; a = a2 ) {
|
1997-11-18 14:06:00 +00:00
|
|
|
a2 = a->chain;
|
|
|
|
if( a->usage == 2 && (rc=iobuf_flush(a)) )
|
|
|
|
log_error("iobuf_flush failed on close: %s\n", g10_errstr(rc));
|
|
|
|
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("iobuf-%d.%d: close '%s'\n", a->no, a->subno, a->desc );
|
|
|
|
if( a->filter && (rc = a->filter(a->filter_ov, IOBUFCTRL_FREE,
|
|
|
|
a->chain, NULL, &dummy_len)) )
|
|
|
|
log_error("IOBUFCTRL_FREE failed on close: %s\n", g10_errstr(rc) );
|
|
|
|
m_free(a->d.buf);
|
|
|
|
m_free(a);
|
|
|
|
}
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
int
|
|
|
|
iobuf_cancel( IOBUF a )
|
|
|
|
{
|
1997-11-21 14:53:57 +00:00
|
|
|
const char *s;
|
|
|
|
|
1997-12-02 19:36:53 +00:00
|
|
|
if( a && a->usage == 2 ) {
|
1998-09-11 05:47:32 +00:00
|
|
|
s = get_real_fname(a);
|
1997-11-21 14:53:57 +00:00
|
|
|
if( s && *s )
|
|
|
|
remove(s); /* remove the file. Fixme: this will fail for MSDOZE*/
|
|
|
|
} /* because the file is still open */
|
1997-11-18 14:06:00 +00:00
|
|
|
return iobuf_close(a);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* create a temporary iobuf, which can be used to collect stuff
|
|
|
|
* in an iobuf and later be written by iobuf_write_temp() to another
|
|
|
|
* iobuf.
|
|
|
|
*/
|
|
|
|
IOBUF
|
|
|
|
iobuf_temp()
|
|
|
|
{
|
|
|
|
IOBUF a;
|
|
|
|
|
|
|
|
a = iobuf_alloc(3, 8192 );
|
|
|
|
|
|
|
|
return a;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* Create a head iobuf for reading from a file
|
|
|
|
* returns: NULL if an error occures and sets errno
|
|
|
|
*/
|
|
|
|
IOBUF
|
|
|
|
iobuf_open( const char *fname )
|
|
|
|
{
|
|
|
|
IOBUF a;
|
|
|
|
FILE *fp;
|
|
|
|
file_filter_ctx_t *fcx;
|
|
|
|
size_t len;
|
1998-09-11 05:47:32 +00:00
|
|
|
int print_only = 0;
|
1997-11-18 14:06:00 +00:00
|
|
|
|
1998-03-09 21:44:06 +00:00
|
|
|
if( !fname || (*fname=='-' && !fname[1]) ) {
|
1997-11-18 14:06:00 +00:00
|
|
|
fp = stdin; /* fixme: set binary mode for msdoze */
|
|
|
|
fname = "[stdin]";
|
1998-09-11 05:47:32 +00:00
|
|
|
print_only = 1;
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
|
|
|
else if( !(fp = fopen(fname, "rb")) )
|
|
|
|
return NULL;
|
|
|
|
a = iobuf_alloc(1, 8192 );
|
|
|
|
fcx = m_alloc( sizeof *fcx + strlen(fname) );
|
|
|
|
fcx->fp = fp;
|
1998-09-11 05:47:32 +00:00
|
|
|
fcx->print_only_name = print_only;
|
1997-11-18 14:06:00 +00:00
|
|
|
strcpy(fcx->fname, fname );
|
|
|
|
a->filter = file_filter;
|
|
|
|
a->filter_ov = fcx;
|
|
|
|
file_filter( fcx, IOBUFCTRL_DESC, NULL, (byte*)&a->desc, &len );
|
|
|
|
file_filter( fcx, IOBUFCTRL_INIT, NULL, NULL, &len );
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("iobuf-%d.%d: open '%s'\n", a->no, a->subno, fname );
|
|
|
|
|
|
|
|
return a;
|
|
|
|
}
|
|
|
|
|
|
|
|
/****************
|
1998-04-14 17:51:16 +00:00
|
|
|
* create an iobuf for writing to a file; the file will be created.
|
1997-11-18 14:06:00 +00:00
|
|
|
*/
|
|
|
|
IOBUF
|
|
|
|
iobuf_create( const char *fname )
|
|
|
|
{
|
|
|
|
IOBUF a;
|
|
|
|
FILE *fp;
|
|
|
|
file_filter_ctx_t *fcx;
|
|
|
|
size_t len;
|
1998-09-11 05:47:32 +00:00
|
|
|
int print_only = 0;
|
1997-11-18 14:06:00 +00:00
|
|
|
|
1998-01-07 20:47:46 +00:00
|
|
|
if( !fname || (*fname=='-' && !fname[1]) ) {
|
1997-11-18 14:06:00 +00:00
|
|
|
fp = stdout;
|
|
|
|
fname = "[stdout]";
|
1998-09-11 05:47:32 +00:00
|
|
|
print_only = 1;
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
|
|
|
else if( !(fp = fopen(fname, "wb")) )
|
|
|
|
return NULL;
|
|
|
|
a = iobuf_alloc(2, 8192 );
|
|
|
|
fcx = m_alloc( sizeof *fcx + strlen(fname) );
|
|
|
|
fcx->fp = fp;
|
1998-09-11 05:47:32 +00:00
|
|
|
fcx->print_only_name = print_only;
|
1997-11-18 14:06:00 +00:00
|
|
|
strcpy(fcx->fname, fname );
|
|
|
|
a->filter = file_filter;
|
|
|
|
a->filter_ov = fcx;
|
|
|
|
file_filter( fcx, IOBUFCTRL_DESC, NULL, (byte*)&a->desc, &len );
|
|
|
|
file_filter( fcx, IOBUFCTRL_INIT, NULL, NULL, &len );
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("iobuf-%d.%d: create '%s'\n", a->no, a->subno, a->desc );
|
|
|
|
|
|
|
|
return a;
|
|
|
|
}
|
|
|
|
|
1997-12-16 19:15:09 +00:00
|
|
|
/****************
|
1998-04-14 17:51:16 +00:00
|
|
|
* append to an iobuf; if the file does not exist, create it.
|
|
|
|
* cannot be used for stdout.
|
1997-12-16 19:15:09 +00:00
|
|
|
*/
|
|
|
|
IOBUF
|
|
|
|
iobuf_append( const char *fname )
|
|
|
|
{
|
|
|
|
IOBUF a;
|
|
|
|
FILE *fp;
|
|
|
|
file_filter_ctx_t *fcx;
|
|
|
|
size_t len;
|
|
|
|
|
|
|
|
if( !fname )
|
|
|
|
return NULL;
|
|
|
|
else if( !(fp = fopen(fname, "ab")) )
|
|
|
|
return NULL;
|
|
|
|
a = iobuf_alloc(2, 8192 );
|
|
|
|
fcx = m_alloc( sizeof *fcx + strlen(fname) );
|
|
|
|
fcx->fp = fp;
|
|
|
|
strcpy(fcx->fname, fname );
|
|
|
|
a->filter = file_filter;
|
|
|
|
a->filter_ov = fcx;
|
|
|
|
file_filter( fcx, IOBUFCTRL_DESC, NULL, (byte*)&a->desc, &len );
|
|
|
|
file_filter( fcx, IOBUFCTRL_INIT, NULL, NULL, &len );
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("iobuf-%d.%d: append '%s'\n", a->no, a->subno, a->desc );
|
|
|
|
|
|
|
|
return a;
|
|
|
|
}
|
|
|
|
|
1997-12-19 11:41:47 +00:00
|
|
|
IOBUF
|
|
|
|
iobuf_openrw( const char *fname )
|
|
|
|
{
|
|
|
|
IOBUF a;
|
|
|
|
FILE *fp;
|
|
|
|
file_filter_ctx_t *fcx;
|
|
|
|
size_t len;
|
|
|
|
|
|
|
|
if( !fname )
|
|
|
|
return NULL;
|
|
|
|
else if( !(fp = fopen(fname, "r+b")) )
|
|
|
|
return NULL;
|
|
|
|
a = iobuf_alloc(2, 8192 );
|
|
|
|
fcx = m_alloc( sizeof *fcx + strlen(fname) );
|
|
|
|
fcx->fp = fp;
|
|
|
|
strcpy(fcx->fname, fname );
|
|
|
|
a->filter = file_filter;
|
|
|
|
a->filter_ov = fcx;
|
|
|
|
file_filter( fcx, IOBUFCTRL_DESC, NULL, (byte*)&a->desc, &len );
|
|
|
|
file_filter( fcx, IOBUFCTRL_INIT, NULL, NULL, &len );
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("iobuf-%d.%d: openrw '%s'\n", a->no, a->subno, a->desc );
|
|
|
|
|
|
|
|
return a;
|
|
|
|
}
|
|
|
|
|
1997-11-18 14:06:00 +00:00
|
|
|
/****************
|
|
|
|
* Register an i/o filter.
|
|
|
|
*/
|
|
|
|
int
|
|
|
|
iobuf_push_filter( IOBUF a,
|
|
|
|
int (*f)(void *opaque, int control,
|
|
|
|
IOBUF chain, byte *buf, size_t *len), void *ov )
|
|
|
|
{
|
|
|
|
IOBUF b;
|
|
|
|
size_t dummy_len=0;
|
|
|
|
int rc=0;
|
|
|
|
|
|
|
|
if( a->usage == 2 && (rc=iobuf_flush(a)) )
|
|
|
|
return rc;
|
|
|
|
/* make a copy of the current stream, so that
|
|
|
|
* A is the new stream and B the original one.
|
|
|
|
* The contents of the buffers are transferred to the
|
|
|
|
* new stream.
|
|
|
|
*/
|
|
|
|
b = m_alloc(sizeof *b);
|
|
|
|
memcpy(b, a, sizeof *b );
|
|
|
|
/* remove the filter stuff from the new stream */
|
|
|
|
a->filter = NULL;
|
|
|
|
a->filter_ov = NULL;
|
1998-01-02 20:40:10 +00:00
|
|
|
a->filter_eof = 0;
|
1997-11-18 14:06:00 +00:00
|
|
|
if( a->usage == 2 ) { /* allocate a fresh buffer for the original stream */
|
|
|
|
b->d.buf = m_alloc( a->d.size );
|
|
|
|
b->d.len = 0;
|
|
|
|
b->d.start = 0;
|
|
|
|
}
|
|
|
|
else { /* allocate a fresh buffer for the new stream */
|
|
|
|
a->d.buf = m_alloc( a->d.size );
|
|
|
|
a->d.len = 0;
|
|
|
|
a->d.start = 0;
|
|
|
|
}
|
|
|
|
/* disable nlimit for the new stream */
|
1997-12-12 12:03:58 +00:00
|
|
|
a->ntotal = b->ntotal + b->nbytes;
|
1997-11-18 14:06:00 +00:00
|
|
|
a->nlimit = a->nbytes = 0;
|
|
|
|
/* make a link from the new stream to the original stream */
|
|
|
|
a->chain = b;
|
1997-11-21 14:53:57 +00:00
|
|
|
a->opaque = b->opaque;
|
1997-11-18 14:06:00 +00:00
|
|
|
|
|
|
|
/* setup the function on the new stream */
|
|
|
|
a->filter = f;
|
|
|
|
a->filter_ov = ov;
|
|
|
|
|
|
|
|
a->subno = b->subno + 1;
|
|
|
|
f( ov, IOBUFCTRL_DESC, NULL, (byte*)&a->desc, &dummy_len );
|
|
|
|
|
|
|
|
if( DBG_IOBUF ) {
|
|
|
|
log_debug("iobuf-%d.%d: push '%s'\n", a->no, a->subno, a->desc );
|
|
|
|
for(b=a; b; b = b->chain )
|
|
|
|
log_debug("\tchain: %d.%d '%s'\n", b->no, b->subno, b->desc );
|
|
|
|
}
|
|
|
|
|
|
|
|
/* now we can initialize the new function if we have one */
|
|
|
|
if( a->filter && (rc = a->filter(a->filter_ov, IOBUFCTRL_INIT, a->chain,
|
|
|
|
NULL, &dummy_len)) )
|
|
|
|
log_error("IOBUFCTRL_INIT failed: %s\n", g10_errstr(rc) );
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* Remove an i/o filter.
|
|
|
|
*/
|
|
|
|
int
|
|
|
|
iobuf_pop_filter( IOBUF a, int (*f)(void *opaque, int control,
|
|
|
|
IOBUF chain, byte *buf, size_t *len), void *ov )
|
|
|
|
{
|
|
|
|
IOBUF b;
|
|
|
|
size_t dummy_len=0;
|
|
|
|
int rc=0;
|
|
|
|
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("iobuf-%d.%d: pop '%s'\n", a->no, a->subno, a->desc );
|
|
|
|
if( !a->filter ) { /* this is simple */
|
|
|
|
b = a->chain;
|
|
|
|
assert(b);
|
|
|
|
m_free(a->d.buf);
|
|
|
|
memcpy(a,b, sizeof *a);
|
|
|
|
m_free(b);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
for(b=a ; b; b = b->chain )
|
|
|
|
if( b->filter == f && (!ov || b->filter_ov == ov) )
|
|
|
|
break;
|
|
|
|
if( !b )
|
|
|
|
log_bug("iobuf_pop_filter(): filter function not found\n");
|
|
|
|
|
|
|
|
/* flush this stream if it is an output stream */
|
|
|
|
if( a->usage == 2 && (rc=iobuf_flush(b)) ) {
|
|
|
|
log_error("iobuf_flush failed in pop_filter: %s\n", g10_errstr(rc));
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
/* and tell the filter to free it self */
|
|
|
|
if( (rc = b->filter(b->filter_ov, IOBUFCTRL_FREE, b->chain,
|
|
|
|
NULL, &dummy_len)) ) {
|
|
|
|
log_error("IOBUFCTRL_FREE failed: %s\n", g10_errstr(rc) );
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
1998-04-14 17:51:16 +00:00
|
|
|
/* and see how to remove it */
|
1997-11-18 14:06:00 +00:00
|
|
|
if( a == b && !b->chain )
|
|
|
|
log_bug("can't remove the last filter from the chain\n");
|
|
|
|
else if( a == b ) { /* remove the first iobuf from the chain */
|
|
|
|
/* everything from b is copied to a. This is save because
|
|
|
|
* a flush has been done on the to be removed entry
|
|
|
|
*/
|
|
|
|
b = a->chain;
|
|
|
|
m_free(a->d.buf);
|
|
|
|
memcpy(a,b, sizeof *a);
|
|
|
|
m_free(b);
|
1998-08-05 16:51:59 +00:00
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("iobuf-%d.%d: popped filter\n", a->no, a->subno );
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
|
|
|
else if( !b->chain ) { /* remove the last iobuf from the chain */
|
1998-01-02 20:40:10 +00:00
|
|
|
log_bug("Ohh jeee, trying to remove a head filter\n");
|
1997-11-18 14:06:00 +00:00
|
|
|
}
|
|
|
|
else { /* remove an intermediate iobuf from the chain */
|
|
|
|
log_bug("Ohh jeee, trying to remove an intermediate filter\n");
|
|
|
|
}
|
|
|
|
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* read underflow: read more bytes into the buffer and return
|
|
|
|
* the first byte or -1 on EOF.
|
|
|
|
*/
|
|
|
|
static int
|
|
|
|
underflow(IOBUF a)
|
|
|
|
{
|
|
|
|
size_t len;
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
/*log_debug("iobuf-%d.%d: underflow: start=%lu len=%lu\n",
|
|
|
|
a->no, a->subno, (ulong)a->d.start, (ulong)a->d.len );*/
|
|
|
|
assert( a->d.start == a->d.len );
|
|
|
|
if( a->usage == 3 )
|
|
|
|
return -1; /* EOF because a temp buffer can't do an underflow */
|
|
|
|
if( a->filter_eof ) {
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("iobuf-%d.%d: filter eof\n", a->no, a->subno );
|
|
|
|
return -1;
|
|
|
|
}
|
1998-07-06 10:23:57 +00:00
|
|
|
if( a->error ) {
|
|
|
|
if( DBG_IOBUF )
|
|
|
|
log_debug("iobuf-%d.%d: error\n", a->no, a->subno );
|
|
|
|
return -1;
|
|
|
|
}
|
1997-11-18 14:06:00 +00:00
|
|
|
|
|
|
|
if( a->filter ) {
|
|
|
|
len = a->d.size;
|
|
|
|
rc = a->filter( a->filter_ov, IOBUFCTRL_UNDERFLOW, a->chain,
|
|
|
|
a->d.buf, &len );
|
|
|
|
if( a->usage == 1 && rc == -1 ) { /* EOF: we can remove the filter */
|
|
|
|
size_t dummy_len;
|
|
|
|
|
|
|
|
/* and tell the filter to free it self */
|
1997-12-23 17:30:18 +00:00
|
|
|
if( a->filter != file_filter ) {
|
|
|
|
if( (rc = a->filter(a->filter_ov, IOBUFCTRL_FREE, a->chain,
|
|
|
|
NULL, &dummy_len)) )
|
|
|
|
log_error("IOBUFCTRL_FREE failed: %s\n", g10_errstr(rc) );
|
|
|
|
a->filter = NULL;
|
|
|
|
a->desc = NULL;
|
|
|
|
a->filter_ov = NULL;
|
|
|
|
}
|
1997-11-18 14:06:00 +00:00
|
|
|
a->filter_eof = 1;
|
|
|
|
}
|
1998-07-06 10:23:57 +00:00
|
|
|
else if( rc )
|
|
|
|
a->error = 1;
|
1997-11-18 14:06:00 +00:00
|
|
|
|
|
|
|
if( !len )
|
|
|
|
return -1;
|
|
|
|
a->d.len = len;
|
|
|
|
a->d.start = 0;
|
|
|
|
return a->d.buf[a->d.start++];
|
|
|
|
}
|
|
|
|
else
|
|
|
|
return -1; /* no filter; return EOF */
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
iobuf_clear_eof(IOBUF a)
|
|
|
|
{
|
|
|
|
assert(a->usage == 1);
|
|
|
|
|
|
|
|
if( a->filter )
|
|
|
|
log_info("iobuf-%d.%d: clear_eof '%s' with enabled filter\n", a->no, a->subno, a->desc );
|
|
|
|
if( !a->filter_eof )
|
|
|
|
log_info("iobuf-%d.%d: clear_eof '%s' with no EOF pending\n", a->no, a->subno, a->desc );
|
|
|
|
iobuf_pop_filter(a, NULL, NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
iobuf_flush(IOBUF a)
|
|
|
|
{
|
|
|
|
size_t len;
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
/*log_debug("iobuf-%d.%d: flush\n", a->no, a->subno );*/
|
|
|
|
if( a->usage == 3 )
|
|
|
|
log_bug("temp buffer too short\n");
|
|
|
|
else if( a->usage != 2 )
|
|
|
|
log_bug("flush on non-output iobuf\n");
|
|
|
|
else if( !a->filter )
|
|
|
|
log_bug("iobuf_flush: no filter\n");
|
|
|
|
len = a->d.len;
|
|
|
|
rc = a->filter( a->filter_ov, IOBUFCTRL_FLUSH, a->chain, a->d.buf, &len );
|
|
|
|
if( !rc && len != a->d.len ) {
|
|
|
|
log_info("iobuf_flush did not write all!\n");
|
|
|
|
rc = G10ERR_WRITE_FILE;
|
|
|
|
}
|
1998-07-06 10:23:57 +00:00
|
|
|
else if( rc )
|
|
|
|
a->error = 1;
|
1997-11-18 14:06:00 +00:00
|
|
|
a->d.len = 0;
|
|
|
|
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* Read a byte from the iobuf; returns -1 on EOF
|
|
|
|
*/
|
|
|
|
int
|
|
|
|
iobuf_readbyte(IOBUF a)
|
|
|
|
{
|
|
|
|
int c;
|
|
|
|
|
1998-02-04 18:54:31 +00:00
|
|
|
/* nlimit does not work together with unget */
|
|
|
|
/* nbytes is also not valid! */
|
|
|
|
if( a->unget.buf ) {
|
|
|
|
if( a->unget.start < a->unget.len )
|
|
|
|
return a->unget.buf[a->unget.start++];
|
|
|
|
m_free(a->unget.buf);
|
|
|
|
a->unget.buf = NULL;
|
|
|
|
}
|
|
|
|
|
1997-11-18 14:06:00 +00:00
|
|
|
if( a->nlimit && a->nbytes >= a->nlimit )
|
|
|
|
return -1; /* forced EOF */
|
|
|
|
|
|
|
|
if( a->d.start < a->d.len ) {
|
|
|
|
c = a->d.buf[a->d.start++];
|
|
|
|
}
|
|
|
|
else if( (c=underflow(a)) == -1 )
|
|
|
|
return -1; /* EOF */
|
|
|
|
|
|
|
|
a->nbytes++;
|
|
|
|
return c;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
1998-02-11 03:25:44 +00:00
|
|
|
int
|
|
|
|
iobuf_read(IOBUF a, byte *buf, unsigned buflen )
|
|
|
|
{
|
|
|
|
int c, n;
|
|
|
|
|
1998-05-04 18:49:26 +00:00
|
|
|
if( a->unget.buf || a->nlimit ) {
|
|
|
|
/* handle special cases */
|
|
|
|
for(n=0 ; n < buflen; n++, buf++ ) {
|
|
|
|
if( (c = iobuf_readbyte(a)) == -1 ) {
|
|
|
|
if( !n )
|
|
|
|
return -1; /* eof */
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
*buf = c;
|
1998-02-11 03:25:44 +00:00
|
|
|
}
|
1998-05-04 18:49:26 +00:00
|
|
|
return n;
|
1998-02-11 03:25:44 +00:00
|
|
|
}
|
1998-05-04 18:49:26 +00:00
|
|
|
|
|
|
|
n = 0;
|
|
|
|
do {
|
|
|
|
for( ; n < buflen && a->d.start < a->d.len; n++ )
|
|
|
|
*buf++ = a->d.buf[a->d.start++];
|
|
|
|
if( n < buflen ) {
|
|
|
|
if( (c=underflow(a)) == -1 ) {
|
|
|
|
a->nbytes += n;
|
|
|
|
return n? n : -1/*EOF*/;
|
|
|
|
}
|
|
|
|
*buf++ = c; n++;
|
|
|
|
}
|
|
|
|
} while( n < buflen );
|
|
|
|
a->nbytes += n;
|
1998-02-11 03:25:44 +00:00
|
|
|
return n;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
1998-02-16 20:05:02 +00:00
|
|
|
/****************
|
|
|
|
* Have a look at the iobuf.
|
1998-04-14 17:51:16 +00:00
|
|
|
* NOTE: This only works in special cases.
|
1998-02-16 20:05:02 +00:00
|
|
|
*/
|
|
|
|
int
|
|
|
|
iobuf_peek(IOBUF a, byte *buf, unsigned buflen )
|
|
|
|
{
|
|
|
|
int n=0;
|
|
|
|
|
1998-07-08 09:29:43 +00:00
|
|
|
if( a->filter_eof )
|
|
|
|
return -1;
|
1998-02-16 20:05:02 +00:00
|
|
|
if( !(a->d.start < a->d.len) ) {
|
|
|
|
if( underflow(a) == -1 )
|
|
|
|
return -1;
|
|
|
|
/* and unget this character */
|
|
|
|
assert(a->d.start == 1);
|
|
|
|
a->d.start = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
for(n=0 ; n < buflen && (a->d.start+n) < a->d.len ; n++, buf++ )
|
|
|
|
*buf = a->d.buf[n];
|
|
|
|
return n;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
1998-02-11 03:25:44 +00:00
|
|
|
|
|
|
|
|
1997-11-18 14:06:00 +00:00
|
|
|
int
|
|
|
|
iobuf_writebyte(IOBUF a, unsigned c)
|
|
|
|
{
|
|
|
|
if( a->d.len == a->d.size )
|
|
|
|
if( iobuf_flush(a) )
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
assert( a->d.len < a->d.size );
|
|
|
|
a->d.buf[a->d.len++] = c;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
iobuf_write(IOBUF a, byte *buf, unsigned buflen )
|
|
|
|
{
|
1998-05-04 18:49:26 +00:00
|
|
|
do {
|
|
|
|
for( ; buflen && a->d.len < a->d.size; buflen--, buf++ )
|
|
|
|
a->d.buf[a->d.len++] = *buf;
|
|
|
|
if( buflen ) {
|
|
|
|
if( iobuf_flush(a) )
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
} while( buflen );
|
1997-11-18 14:06:00 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1998-05-04 18:49:26 +00:00
|
|
|
|
1997-11-19 13:12:23 +00:00
|
|
|
int
|
|
|
|
iobuf_writestr(IOBUF a, const char *buf )
|
|
|
|
{
|
|
|
|
for( ; *buf; buf++ )
|
|
|
|
if( iobuf_writebyte(a, *buf) )
|
|
|
|
return -1;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1997-11-18 14:06:00 +00:00
|
|
|
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* copy the contents of TEMP to A.
|
|
|
|
*/
|
|
|
|
int
|
|
|
|
iobuf_write_temp( IOBUF a, IOBUF temp )
|
|
|
|
{
|
|
|
|
return iobuf_write(a, temp->d.buf, temp->d.len );
|
|
|
|
}
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* copy the contents of the temp io stream to BUFFER.
|
|
|
|
*/
|
|
|
|
size_t
|
|
|
|
iobuf_temp_to_buffer( IOBUF a, byte *buffer, size_t buflen )
|
|
|
|
{
|
|
|
|
size_t n = a->d.len;
|
|
|
|
|
|
|
|
if( n > buflen )
|
|
|
|
n = buflen;
|
|
|
|
memcpy( buffer, a->d.buf, n );
|
|
|
|
return n;
|
|
|
|
}
|
|
|
|
|
1998-02-04 18:54:31 +00:00
|
|
|
/****************
|
|
|
|
* unget the contents of the temp io stream to A and close temp
|
|
|
|
* Could be optimized!!
|
|
|
|
*/
|
|
|
|
void
|
|
|
|
iobuf_unget_and_close_temp( IOBUF a, IOBUF temp )
|
|
|
|
{
|
|
|
|
if( a->unget.buf ) {
|
|
|
|
if( a->unget.start < a->unget.len )
|
|
|
|
log_fatal("cannot do any more ungets on this buffer\n");
|
|
|
|
/* not yet cleaned up; do it now */
|
|
|
|
m_free(a->unget.buf);
|
|
|
|
a->unget.buf = NULL;
|
|
|
|
}
|
|
|
|
a->unget.size = temp->d.len;
|
|
|
|
a->unget.buf = m_alloc( a->unget.size );
|
|
|
|
a->unget.len = temp->d.len;
|
|
|
|
memcpy( a->unget.buf, temp->d.buf, a->unget.len );
|
|
|
|
iobuf_close(temp);
|
|
|
|
}
|
|
|
|
|
1997-11-18 14:06:00 +00:00
|
|
|
|
|
|
|
/****************
|
1998-04-14 17:51:16 +00:00
|
|
|
* Set a limit on how many bytes may be read from the input stream A.
|
1997-11-18 14:06:00 +00:00
|
|
|
* Setting the limit to 0 disables this feature.
|
|
|
|
*/
|
|
|
|
void
|
|
|
|
iobuf_set_limit( IOBUF a, unsigned long nlimit )
|
|
|
|
{
|
|
|
|
a->nlimit = nlimit;
|
1997-12-12 12:03:58 +00:00
|
|
|
a->ntotal += a->nbytes;
|
1997-11-18 14:06:00 +00:00
|
|
|
a->nbytes = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/****************
|
|
|
|
* Return the length of an open file
|
|
|
|
*/
|
|
|
|
u32
|
|
|
|
iobuf_get_filelength( IOBUF a )
|
|
|
|
{
|
|
|
|
struct stat st;
|
|
|
|
|
|
|
|
for( ; a; a = a->chain )
|
|
|
|
if( !a->chain && a->filter == file_filter ) {
|
|
|
|
file_filter_ctx_t *b = a->filter_ov;
|
|
|
|
FILE *fp = b->fp;
|
|
|
|
|
|
|
|
if( !fstat(fileno(fp), &st) )
|
|
|
|
return st.st_size;
|
|
|
|
log_error("fstat() failed: %s\n", strerror(errno) );
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
1997-12-12 12:03:58 +00:00
|
|
|
/****************
|
|
|
|
* Tell the file position, where the next read will take place
|
|
|
|
*/
|
|
|
|
ulong
|
|
|
|
iobuf_tell( IOBUF a )
|
|
|
|
{
|
|
|
|
return a->ntotal + a->nbytes;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
1998-02-13 20:58:50 +00:00
|
|
|
|
|
|
|
/****************
|
|
|
|
* This is a very limited implementation. It simply discards all internal
|
1998-04-14 17:51:16 +00:00
|
|
|
* buffering and removes all filters but the first one.
|
1998-02-13 20:58:50 +00:00
|
|
|
*/
|
1997-12-12 12:03:58 +00:00
|
|
|
int
|
|
|
|
iobuf_seek( IOBUF a, ulong newpos )
|
|
|
|
{
|
1997-12-16 19:15:09 +00:00
|
|
|
file_filter_ctx_t *b = NULL;
|
1997-12-12 12:03:58 +00:00
|
|
|
|
1997-12-16 19:15:09 +00:00
|
|
|
for( ; a; a = a->chain ) {
|
|
|
|
if( !a->chain && a->filter == file_filter ) {
|
|
|
|
b = a->filter_ov;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if( !a )
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
if( fseek( b->fp, newpos, SEEK_SET ) ) {
|
|
|
|
log_error("can't seek to %lu: %s\n", newpos, strerror(errno) );
|
|
|
|
return -1;
|
|
|
|
}
|
1998-02-13 20:58:50 +00:00
|
|
|
a->d.len = 0; /* discard buffer */
|
|
|
|
a->d.start = 0;
|
|
|
|
a->nbytes = 0;
|
|
|
|
a->nlimit = 0;
|
1997-12-19 11:41:47 +00:00
|
|
|
a->ntotal = newpos;
|
1998-07-06 10:23:57 +00:00
|
|
|
a->error = 0;
|
1998-01-16 21:15:24 +00:00
|
|
|
/* remove filters, but the last */
|
|
|
|
while( a->chain )
|
|
|
|
iobuf_pop_filter( a, a->filter, NULL );
|
|
|
|
|
1997-12-16 19:15:09 +00:00
|
|
|
|
|
|
|
return 0;
|
1997-12-12 12:03:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1998-09-11 05:47:32 +00:00
|
|
|
/****************
|
|
|
|
* Retrieve the real filename
|
|
|
|
*/
|
|
|
|
static const char *
|
|
|
|
get_real_fname( IOBUF a )
|
|
|
|
{
|
|
|
|
for( ; a; a = a->chain )
|
|
|
|
if( !a->chain && a->filter == file_filter ) {
|
|
|
|
file_filter_ctx_t *b = a->filter_ov;
|
|
|
|
return b->print_only_name? NULL : b->fname;
|
|
|
|
}
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
1997-11-21 14:53:57 +00:00
|
|
|
/****************
|
|
|
|
* Retrieve the filename
|
|
|
|
*/
|
|
|
|
const char *
|
|
|
|
iobuf_get_fname( IOBUF a )
|
|
|
|
{
|
|
|
|
for( ; a; a = a->chain )
|
|
|
|
if( !a->chain && a->filter == file_filter ) {
|
|
|
|
file_filter_ctx_t *b = a->filter_ov;
|
|
|
|
return b->fname;
|
|
|
|
}
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
1997-11-18 14:06:00 +00:00
|
|
|
/****************
|
|
|
|
* Start the block write mode, see rfc1991.new for details.
|
|
|
|
* A value of 0 for N stops this mode (flushes and writes
|
|
|
|
* the end marker)
|
|
|
|
*/
|
|
|
|
void
|
|
|
|
iobuf_set_block_mode( IOBUF a, size_t n )
|
|
|
|
{
|
|
|
|
block_filter_ctx_t *ctx = m_alloc_clear( sizeof *ctx );
|
|
|
|
|
|
|
|
assert( a->usage == 1 || a->usage == 2 );
|
|
|
|
ctx->usage = a->usage;
|
|
|
|
if( !n ) {
|
|
|
|
iobuf_pop_filter(a, block_filter, NULL );
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
ctx->size = n; /* only needed for usage 2 */
|
|
|
|
iobuf_push_filter(a, block_filter, ctx );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
1998-01-16 21:15:24 +00:00
|
|
|
/****************
|
1998-04-14 17:51:16 +00:00
|
|
|
* enable partial block mode as described in the OpenPGP draft.
|
1998-07-02 19:31:46 +00:00
|
|
|
* LEN is the first length byte on read, but ignored on writes.
|
1998-01-16 21:15:24 +00:00
|
|
|
*/
|
|
|
|
void
|
|
|
|
iobuf_set_partial_block_mode( IOBUF a, size_t len )
|
|
|
|
{
|
|
|
|
block_filter_ctx_t *ctx = m_alloc_clear( sizeof *ctx );
|
|
|
|
|
|
|
|
assert( a->usage == 1 || a->usage == 2 );
|
|
|
|
ctx->usage = a->usage;
|
|
|
|
if( !len ) {
|
|
|
|
iobuf_pop_filter(a, block_filter, NULL );
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
ctx->partial = 1;
|
1998-07-02 19:31:46 +00:00
|
|
|
ctx->size = 0;
|
|
|
|
ctx->first_c = len;
|
1998-01-16 21:15:24 +00:00
|
|
|
iobuf_push_filter(a, block_filter, ctx );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
1997-11-18 14:06:00 +00:00
|
|
|
|
|
|
|
/****************
|
1998-04-14 17:51:16 +00:00
|
|
|
* Checks whether the stream is in block mode
|
1998-01-02 20:40:10 +00:00
|
|
|
* Note: This does not work if other filters are pushed on the stream.
|
1997-11-18 14:06:00 +00:00
|
|
|
*/
|
|
|
|
int
|
|
|
|
iobuf_in_block_mode( IOBUF a )
|
|
|
|
{
|
1998-01-02 20:40:10 +00:00
|
|
|
if( a && a->filter == block_filter )
|
1998-01-16 21:15:24 +00:00
|
|
|
return 1; /* yes */
|
1997-11-18 14:06:00 +00:00
|
|
|
return 0; /* no */
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|