2006-08-14 16:40:07 +02:00
|
|
|
|
/* estream.c - Extended Stream I/O Library
|
2007-05-15 18:10:48 +02:00
|
|
|
|
* Copyright (C) 2004, 2005, 2006, 2007 g10 Code GmbH
|
2006-06-20 19:21:37 +02:00
|
|
|
|
*
|
|
|
|
|
* This file is part of Libestream.
|
|
|
|
|
*
|
|
|
|
|
* Libestream is free software; you can redistribute it and/or modify
|
|
|
|
|
* it under the terms of the GNU General Public License as published
|
2007-08-22 12:55:07 +02:00
|
|
|
|
* by the Free Software Foundation; either version 2 of the License,
|
2006-06-20 19:21:37 +02:00
|
|
|
|
* or (at your option) any later version.
|
|
|
|
|
*
|
|
|
|
|
* Libestream is distributed in the hope that it will be useful, but
|
|
|
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
2007-05-15 18:10:48 +02:00
|
|
|
|
* General Public License for more details.
|
2006-06-20 19:21:37 +02:00
|
|
|
|
*
|
|
|
|
|
* You should have received a copy of the GNU General Public License
|
2007-07-04 21:49:40 +02:00
|
|
|
|
* along with Libestream; if not, see <http://www.gnu.org/licenses/>.
|
2006-06-20 19:21:37 +02:00
|
|
|
|
*/
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
#ifdef USE_ESTREAM_SUPPORT_H
|
|
|
|
|
# include <estream-support.h>
|
|
|
|
|
#endif
|
|
|
|
|
|
2005-06-03 15:57:24 +02:00
|
|
|
|
#ifdef HAVE_CONFIG_H
|
2005-01-26 23:24:59 +01:00
|
|
|
|
# include <config.h>
|
|
|
|
|
#endif
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
#if defined(_WIN32) && !defined(HAVE_W32_SYSTEM)
|
|
|
|
|
# define HAVE_W32_SYSTEM 1
|
|
|
|
|
#endif
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
#include <sys/types.h>
|
|
|
|
|
#include <sys/file.h>
|
|
|
|
|
#include <sys/stat.h>
|
2006-10-17 16:34:42 +02:00
|
|
|
|
#include <stdio.h>
|
2005-01-26 23:24:59 +01:00
|
|
|
|
#include <stdlib.h>
|
|
|
|
|
#include <string.h>
|
|
|
|
|
#include <unistd.h>
|
|
|
|
|
#include <stdarg.h>
|
|
|
|
|
#include <fcntl.h>
|
|
|
|
|
#include <errno.h>
|
|
|
|
|
#include <stddef.h>
|
|
|
|
|
#include <assert.h>
|
2007-08-22 12:55:07 +02:00
|
|
|
|
#ifdef HAVE_W32_SYSTEM
|
|
|
|
|
# include <windows.h>
|
|
|
|
|
#endif
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2006-10-17 16:34:42 +02:00
|
|
|
|
#ifdef WITHOUT_GNU_PTH /* Give the Makefile a chance to build without Pth. */
|
2007-08-22 12:55:07 +02:00
|
|
|
|
# undef HAVE_PTH
|
|
|
|
|
# undef USE_GNU_PTH
|
2006-10-17 16:34:42 +02:00
|
|
|
|
#endif
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
#ifdef HAVE_PTH
|
|
|
|
|
# include <pth.h>
|
|
|
|
|
#endif
|
|
|
|
|
|
2007-05-15 18:10:48 +02:00
|
|
|
|
/* This is for the special hack to use estream.c in GnuPG. */
|
2007-03-19 15:35:04 +01:00
|
|
|
|
#ifdef GNUPG_MAJOR_VERSION
|
2007-08-22 12:55:07 +02:00
|
|
|
|
# include "../common/util.h"
|
2007-03-19 15:35:04 +01:00
|
|
|
|
#endif
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
#ifndef HAVE_MKSTEMP
|
|
|
|
|
int mkstemp (char *template);
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
#ifndef HAVE_MEMRCHR
|
|
|
|
|
void *memrchr (const void *block, int c, size_t size);
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
#include <estream.h>
|
2007-05-15 18:10:48 +02:00
|
|
|
|
#include <estream-printf.h>
|
2006-11-20 17:49:41 +01:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
#ifndef O_BINARY
|
|
|
|
|
#define O_BINARY 0
|
|
|
|
|
#endif
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/* Generally used types. */
|
|
|
|
|
|
|
|
|
|
typedef void *(*func_realloc_t) (void *mem, size_t size);
|
|
|
|
|
typedef void (*func_free_t) (void *mem);
|
|
|
|
|
|
2006-11-20 17:49:41 +01:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Buffer management layer. */
|
|
|
|
|
|
|
|
|
|
#define BUFFER_BLOCK_SIZE BUFSIZ
|
|
|
|
|
#define BUFFER_UNREAD_SIZE 16
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Macros. */
|
|
|
|
|
|
|
|
|
|
#define BUFFER_ROUND_TO_BLOCK(size, block_size) \
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Locking. */
|
|
|
|
|
|
|
|
|
|
#ifdef HAVE_PTH
|
|
|
|
|
|
|
|
|
|
typedef pth_mutex_t estream_mutex_t;
|
|
|
|
|
# define ESTREAM_MUTEX_INITIALIZER PTH_MUTEX_INIT
|
|
|
|
|
# define ESTREAM_MUTEX_LOCK(mutex) \
|
|
|
|
|
pth_mutex_acquire (&(mutex), 0, NULL)
|
|
|
|
|
# define ESTREAM_MUTEX_UNLOCK(mutex) \
|
|
|
|
|
pth_mutex_release (&(mutex))
|
|
|
|
|
# define ESTREAM_MUTEX_TRYLOCK(mutex) \
|
|
|
|
|
((pth_mutex_acquire (&(mutex), 1, NULL) == TRUE) ? 0 : -1)
|
|
|
|
|
# define ESTREAM_MUTEX_INITIALIZE(mutex) \
|
|
|
|
|
pth_mutex_init (&(mutex))
|
|
|
|
|
#else
|
|
|
|
|
|
|
|
|
|
typedef void *estream_mutex_t;
|
2008-10-20 15:53:23 +02:00
|
|
|
|
|
|
|
|
|
static inline void
|
|
|
|
|
dummy_mutex_call_void (estream_mutex_t mutex)
|
|
|
|
|
{
|
|
|
|
|
(void)mutex;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static inline int
|
|
|
|
|
dummy_mutex_call_int (estream_mutex_t mutex)
|
|
|
|
|
{
|
|
|
|
|
(void)mutex;
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
# define ESTREAM_MUTEX_INITIALIZER NULL
|
2008-10-20 15:53:23 +02:00
|
|
|
|
# define ESTREAM_MUTEX_LOCK(mutex) dummy_mutex_call_void ((mutex))
|
|
|
|
|
# define ESTREAM_MUTEX_UNLOCK(mutex) dummy_mutex_call_void ((mutex))
|
|
|
|
|
# define ESTREAM_MUTEX_TRYLOCK(mutex) dummy_mutex_call_int ((mutex))
|
|
|
|
|
# define ESTREAM_MUTEX_INITIALIZE(mutex) dummy_mutex_call_void ((mutex))
|
2005-01-26 23:24:59 +01:00
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
/* Primitive system I/O. */
|
|
|
|
|
|
|
|
|
|
#ifdef HAVE_PTH
|
|
|
|
|
# define ESTREAM_SYS_READ pth_read
|
|
|
|
|
# define ESTREAM_SYS_WRITE pth_write
|
|
|
|
|
#else
|
|
|
|
|
# define ESTREAM_SYS_READ read
|
|
|
|
|
# define ESTREAM_SYS_WRITE write
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
/* Misc definitions. */
|
|
|
|
|
|
|
|
|
|
#define ES_DEFAULT_OPEN_MODE (S_IRUSR | S_IWUSR)
|
|
|
|
|
|
|
|
|
|
/* An internal stream object. */
|
|
|
|
|
|
|
|
|
|
struct estream_internal
|
|
|
|
|
{
|
|
|
|
|
unsigned char buffer[BUFFER_BLOCK_SIZE];
|
|
|
|
|
unsigned char unread_buffer[BUFFER_UNREAD_SIZE];
|
|
|
|
|
estream_mutex_t lock; /* Lock. */
|
2007-06-25 13:54:43 +02:00
|
|
|
|
void *cookie; /* Cookie. */
|
|
|
|
|
void *opaque; /* Opaque data. */
|
|
|
|
|
unsigned int modeflags; /* Flags for the backend. */
|
2005-01-26 23:24:59 +01:00
|
|
|
|
off_t offset;
|
|
|
|
|
es_cookie_read_function_t func_read;
|
|
|
|
|
es_cookie_write_function_t func_write;
|
|
|
|
|
es_cookie_seek_function_t func_seek;
|
|
|
|
|
es_cookie_close_function_t func_close;
|
|
|
|
|
int strategy;
|
|
|
|
|
int fd;
|
|
|
|
|
struct
|
|
|
|
|
{
|
|
|
|
|
unsigned int err: 1;
|
|
|
|
|
unsigned int eof: 1;
|
|
|
|
|
} indicators;
|
|
|
|
|
unsigned int deallocate_buffer: 1;
|
2006-10-17 16:34:42 +02:00
|
|
|
|
unsigned int print_err: 1; /* Error in print_fun_writer. */
|
|
|
|
|
int print_errno; /* Errno from print_fun_writer. */
|
|
|
|
|
size_t print_ntotal; /* Bytes written from in print_fun_writer. */
|
|
|
|
|
FILE *print_fp; /* Stdio stream used by print_fun_writer. */
|
2005-01-26 23:24:59 +01:00
|
|
|
|
};
|
|
|
|
|
|
2006-10-17 16:34:42 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
typedef struct estream_internal *estream_internal_t;
|
|
|
|
|
|
|
|
|
|
#define ESTREAM_LOCK(stream) ESTREAM_MUTEX_LOCK (stream->intern->lock)
|
|
|
|
|
#define ESTREAM_UNLOCK(stream) ESTREAM_MUTEX_UNLOCK (stream->intern->lock)
|
|
|
|
|
#define ESTREAM_TRYLOCK(stream) ESTREAM_MUTEX_TRYLOCK (stream->intern->lock)
|
|
|
|
|
|
|
|
|
|
/* Stream list. */
|
|
|
|
|
|
|
|
|
|
typedef struct estream_list *estream_list_t;
|
|
|
|
|
|
|
|
|
|
struct estream_list
|
|
|
|
|
{
|
|
|
|
|
estream_t car;
|
|
|
|
|
estream_list_t cdr;
|
|
|
|
|
estream_list_t *prev_cdr;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
static estream_list_t estream_list;
|
2007-06-14 19:05:07 +02:00
|
|
|
|
static estream_mutex_t estream_list_lock;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
#define ESTREAM_LIST_LOCK ESTREAM_MUTEX_LOCK (estream_list_lock)
|
|
|
|
|
#define ESTREAM_LIST_UNLOCK ESTREAM_MUTEX_UNLOCK (estream_list_lock)
|
|
|
|
|
|
|
|
|
|
#ifndef EOPNOTSUPP
|
|
|
|
|
# define EOPNOTSUPP ENOSYS
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Macros. */
|
|
|
|
|
|
|
|
|
|
/* Calculate array dimension. */
|
2007-03-19 15:35:04 +01:00
|
|
|
|
#ifndef DIM
|
2005-01-26 23:24:59 +01:00
|
|
|
|
#define DIM(array) (sizeof (array) / sizeof (*array))
|
2007-03-19 15:35:04 +01:00
|
|
|
|
#endif
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
#define tohex(n) ((n) < 10 ? ((n) + '0') : (((n) - 10) + 'A'))
|
|
|
|
|
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/* Evaluate EXPRESSION, setting VARIABLE to the return code, if
|
|
|
|
|
VARIABLE is zero. */
|
|
|
|
|
#define SET_UNLESS_NONZERO(variable, tmp_variable, expression) \
|
|
|
|
|
do \
|
|
|
|
|
{ \
|
|
|
|
|
tmp_variable = expression; \
|
|
|
|
|
if ((! variable) && tmp_variable) \
|
|
|
|
|
variable = tmp_variable; \
|
|
|
|
|
} \
|
|
|
|
|
while (0)
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
|
|
|
|
|
/* Malloc wrappers to overcvome problems on some older OSes. */
|
|
|
|
|
static void *
|
|
|
|
|
mem_alloc (size_t n)
|
|
|
|
|
{
|
|
|
|
|
if (!n)
|
|
|
|
|
n++;
|
|
|
|
|
return malloc (n);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void *
|
|
|
|
|
mem_realloc (void *p, size_t n)
|
|
|
|
|
{
|
|
|
|
|
if (!p)
|
|
|
|
|
return mem_alloc (n);
|
|
|
|
|
return realloc (p, n);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void
|
|
|
|
|
mem_free (void *p)
|
|
|
|
|
{
|
|
|
|
|
if (p)
|
|
|
|
|
free (p);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/*
|
|
|
|
|
* List manipulation.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
/* Add STREAM to the list of registered stream objects. */
|
|
|
|
|
static int
|
|
|
|
|
es_list_add (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
estream_list_t list_obj;
|
|
|
|
|
int ret;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
list_obj = mem_alloc (sizeof (*list_obj));
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (! list_obj)
|
|
|
|
|
ret = -1;
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LIST_LOCK;
|
|
|
|
|
list_obj->car = stream;
|
|
|
|
|
list_obj->cdr = estream_list;
|
|
|
|
|
list_obj->prev_cdr = &estream_list;
|
|
|
|
|
if (estream_list)
|
|
|
|
|
estream_list->prev_cdr = &list_obj->cdr;
|
|
|
|
|
estream_list = list_obj;
|
|
|
|
|
ESTREAM_LIST_UNLOCK;
|
|
|
|
|
ret = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Remove STREAM from the list of registered stream objects. */
|
|
|
|
|
static void
|
|
|
|
|
es_list_remove (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
estream_list_t list_obj;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LIST_LOCK;
|
|
|
|
|
for (list_obj = estream_list; list_obj; list_obj = list_obj->cdr)
|
|
|
|
|
if (list_obj->car == stream)
|
|
|
|
|
{
|
|
|
|
|
*list_obj->prev_cdr = list_obj->cdr;
|
|
|
|
|
if (list_obj->cdr)
|
|
|
|
|
list_obj->cdr->prev_cdr = list_obj->prev_cdr;
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (list_obj);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
ESTREAM_LIST_UNLOCK;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Type of an stream-iterator-function. */
|
|
|
|
|
typedef int (*estream_iterator_t) (estream_t stream);
|
|
|
|
|
|
|
|
|
|
/* Iterate over list of registered streams, calling ITERATOR for each
|
|
|
|
|
of them. */
|
|
|
|
|
static int
|
|
|
|
|
es_list_iterate (estream_iterator_t iterator)
|
|
|
|
|
{
|
|
|
|
|
estream_list_t list_obj;
|
|
|
|
|
int ret = 0;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LIST_LOCK;
|
|
|
|
|
for (list_obj = estream_list; list_obj; list_obj = list_obj->cdr)
|
|
|
|
|
ret |= (*iterator) (list_obj->car);
|
|
|
|
|
ESTREAM_LIST_UNLOCK;
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* Initialization.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
|
es_init_do (void)
|
|
|
|
|
{
|
2007-06-14 19:05:07 +02:00
|
|
|
|
#ifdef HAVE_PTH
|
|
|
|
|
static int initialized;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-06-14 19:05:07 +02:00
|
|
|
|
if (!initialized)
|
|
|
|
|
{
|
2007-07-04 21:49:40 +02:00
|
|
|
|
if (!pth_init () && errno != EPERM )
|
2007-06-14 19:05:07 +02:00
|
|
|
|
return -1;
|
|
|
|
|
if (pth_mutex_init (&estream_list_lock))
|
|
|
|
|
initialized = 1;
|
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
return 0;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* I/O methods.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
/* Implementation of Memory I/O. */
|
|
|
|
|
|
|
|
|
|
/* Cookie for memory objects. */
|
|
|
|
|
typedef struct estream_cookie_mem
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
unsigned int modeflags; /* Open flags. */
|
2007-08-22 12:55:07 +02:00
|
|
|
|
unsigned char *memory; /* Allocated data buffer. */
|
|
|
|
|
size_t memory_size; /* Allocated size of memory. */
|
|
|
|
|
size_t memory_limit; /* Maximum allowed allocation size or
|
|
|
|
|
0 for no limit. */
|
2005-01-26 23:24:59 +01:00
|
|
|
|
size_t offset; /* Current offset in MEMORY. */
|
|
|
|
|
size_t data_len; /* Length of data in MEMORY. */
|
|
|
|
|
size_t block_size; /* Block size. */
|
2007-08-22 12:55:07 +02:00
|
|
|
|
struct {
|
|
|
|
|
unsigned int grow: 1; /* MEMORY is allowed to grow. */
|
|
|
|
|
} flags;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
func_realloc_t func_realloc;
|
|
|
|
|
func_free_t func_free;
|
|
|
|
|
} *estream_cookie_mem_t;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/* Create function for memory objects. */
|
|
|
|
|
static int
|
|
|
|
|
es_func_mem_create (void *ES__RESTRICT *ES__RESTRICT cookie,
|
|
|
|
|
unsigned char *ES__RESTRICT data, size_t data_n,
|
|
|
|
|
size_t data_len,
|
|
|
|
|
size_t block_size, unsigned int grow,
|
|
|
|
|
func_realloc_t func_realloc, func_free_t func_free,
|
2007-08-22 12:55:07 +02:00
|
|
|
|
unsigned int modeflags,
|
|
|
|
|
size_t memory_limit)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
estream_cookie_mem_t mem_cookie;
|
|
|
|
|
int err;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_cookie = mem_alloc (sizeof (*mem_cookie));
|
|
|
|
|
if (!mem_cookie)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
err = -1;
|
|
|
|
|
else
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
mem_cookie->modeflags = modeflags;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
mem_cookie->memory = data;
|
|
|
|
|
mem_cookie->memory_size = data_n;
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_cookie->memory_limit = memory_limit;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
mem_cookie->offset = 0;
|
|
|
|
|
mem_cookie->data_len = data_len;
|
|
|
|
|
mem_cookie->block_size = block_size;
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_cookie->flags.grow = !!grow;
|
|
|
|
|
mem_cookie->func_realloc = func_realloc ? func_realloc : mem_realloc;
|
|
|
|
|
mem_cookie->func_free = func_free ? func_free : mem_free;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
*cookie = mem_cookie;
|
|
|
|
|
err = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/* Read function for memory objects. */
|
|
|
|
|
static ssize_t
|
2005-06-16 10:12:03 +02:00
|
|
|
|
es_func_mem_read (void *cookie, void *buffer, size_t size)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
estream_cookie_mem_t mem_cookie = cookie;
|
|
|
|
|
ssize_t ret;
|
|
|
|
|
|
|
|
|
|
if (size > mem_cookie->data_len - mem_cookie->offset)
|
|
|
|
|
size = mem_cookie->data_len - mem_cookie->offset;
|
|
|
|
|
|
|
|
|
|
if (size)
|
|
|
|
|
{
|
|
|
|
|
memcpy (buffer, mem_cookie->memory + mem_cookie->offset, size);
|
|
|
|
|
mem_cookie->offset += size;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ret = size;
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/* Write function for memory objects. */
|
|
|
|
|
static ssize_t
|
2005-06-16 10:12:03 +02:00
|
|
|
|
es_func_mem_write (void *cookie, const void *buffer, size_t size)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
estream_cookie_mem_t mem_cookie = cookie;
|
|
|
|
|
ssize_t ret;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
if (!size)
|
|
|
|
|
return 0; /* A flush is a NOP for memory objects. */
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
if (mem_cookie->modeflags & O_APPEND)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
/* Append to data. */
|
|
|
|
|
mem_cookie->offset = mem_cookie->data_len;
|
|
|
|
|
}
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
if (!mem_cookie->flags.grow)
|
|
|
|
|
{
|
|
|
|
|
/* We are not alloew to grow, thus limit the size to the left
|
|
|
|
|
space. FIXME: Does the grow flag an its semtics make sense
|
|
|
|
|
at all? */
|
|
|
|
|
if (size > mem_cookie->memory_size - mem_cookie->offset)
|
|
|
|
|
size = mem_cookie->memory_size - mem_cookie->offset;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
if (size > (mem_cookie->memory_size - mem_cookie->offset))
|
|
|
|
|
{
|
|
|
|
|
unsigned char *newbuf;
|
|
|
|
|
size_t newsize;
|
|
|
|
|
|
|
|
|
|
newsize = mem_cookie->memory_size + mem_cookie->block_size;
|
|
|
|
|
|
|
|
|
|
newsize = mem_cookie->offset + size;
|
|
|
|
|
if (newsize < mem_cookie->offset)
|
|
|
|
|
{
|
|
|
|
|
errno = EINVAL;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
newsize += mem_cookie->block_size - 1;
|
|
|
|
|
if (newsize < mem_cookie->offset)
|
|
|
|
|
{
|
|
|
|
|
errno = EINVAL;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
newsize /= mem_cookie->block_size;
|
|
|
|
|
newsize *= mem_cookie->block_size;
|
|
|
|
|
|
|
|
|
|
if (mem_cookie->memory_limit && newsize > mem_cookie->memory_limit)
|
|
|
|
|
{
|
|
|
|
|
errno = ENOSPC;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
newbuf = mem_cookie->func_realloc (mem_cookie->memory, newsize);
|
|
|
|
|
if (!newbuf)
|
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
|
mem_cookie->memory = newbuf;
|
|
|
|
|
mem_cookie->memory_size = newsize;
|
|
|
|
|
|
|
|
|
|
assert (!(size > (mem_cookie->memory_size - mem_cookie->offset)));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
memcpy (mem_cookie->memory + mem_cookie->offset, buffer, size);
|
|
|
|
|
if (mem_cookie->offset + size > mem_cookie->data_len)
|
|
|
|
|
mem_cookie->data_len = mem_cookie->offset + size;
|
|
|
|
|
mem_cookie->offset += size;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
ret = size;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/* Seek function for memory objects. */
|
|
|
|
|
static int
|
|
|
|
|
es_func_mem_seek (void *cookie, off_t *offset, int whence)
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_mem_t mem_cookie = cookie;
|
|
|
|
|
off_t pos_new;
|
|
|
|
|
|
|
|
|
|
switch (whence)
|
|
|
|
|
{
|
|
|
|
|
case SEEK_SET:
|
|
|
|
|
pos_new = *offset;
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
case SEEK_CUR:
|
|
|
|
|
pos_new = mem_cookie->offset += *offset;
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
case SEEK_END:
|
|
|
|
|
pos_new = mem_cookie->data_len += *offset;
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
default:
|
2007-08-22 12:55:07 +02:00
|
|
|
|
errno = EINVAL;
|
|
|
|
|
return -1;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (pos_new > mem_cookie->memory_size)
|
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
size_t newsize;
|
|
|
|
|
void *newbuf;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
if (!mem_cookie->flags.grow)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
errno = ENOSPC;
|
|
|
|
|
return -1;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
newsize = pos_new + mem_cookie->block_size - 1;
|
|
|
|
|
if (newsize < pos_new)
|
|
|
|
|
{
|
|
|
|
|
errno = EINVAL;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
newsize /= mem_cookie->block_size;
|
|
|
|
|
newsize *= mem_cookie->block_size;
|
|
|
|
|
if (mem_cookie->memory_limit && newsize > mem_cookie->memory_limit)
|
|
|
|
|
{
|
|
|
|
|
errno = ENOSPC;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
newbuf = mem_cookie->func_realloc (mem_cookie->memory, newsize);
|
|
|
|
|
if (!newbuf)
|
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
|
mem_cookie->memory = newbuf;
|
|
|
|
|
mem_cookie->memory_size = newsize;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (pos_new > mem_cookie->data_len)
|
2007-08-22 12:55:07 +02:00
|
|
|
|
{
|
|
|
|
|
/* Fill spare space with zeroes. */
|
|
|
|
|
memset (mem_cookie->memory + mem_cookie->data_len,
|
|
|
|
|
0, pos_new - mem_cookie->data_len);
|
|
|
|
|
mem_cookie->data_len = pos_new;
|
|
|
|
|
}
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
mem_cookie->offset = pos_new;
|
|
|
|
|
*offset = pos_new;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
return 0;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/* Destroy function for memory objects. */
|
|
|
|
|
static int
|
|
|
|
|
es_func_mem_destroy (void *cookie)
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_mem_t mem_cookie = cookie;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
if (cookie)
|
|
|
|
|
{
|
|
|
|
|
mem_cookie->func_free (mem_cookie->memory);
|
|
|
|
|
mem_free (mem_cookie);
|
|
|
|
|
}
|
2005-01-26 23:24:59 +01:00
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
static es_cookie_io_functions_t estream_functions_mem =
|
|
|
|
|
{
|
|
|
|
|
es_func_mem_read,
|
|
|
|
|
es_func_mem_write,
|
|
|
|
|
es_func_mem_seek,
|
2006-08-11 13:04:38 +02:00
|
|
|
|
es_func_mem_destroy
|
2005-01-26 23:24:59 +01:00
|
|
|
|
};
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/* Implementation of fd I/O. */
|
|
|
|
|
|
|
|
|
|
/* Cookie for fd objects. */
|
|
|
|
|
typedef struct estream_cookie_fd
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
int fd; /* The file descriptor we are using for actual output. */
|
|
|
|
|
int no_close; /* If set we won't close the file descriptor. */
|
2005-01-26 23:24:59 +01:00
|
|
|
|
} *estream_cookie_fd_t;
|
|
|
|
|
|
|
|
|
|
/* Create function for fd objects. */
|
|
|
|
|
static int
|
2007-06-25 13:54:43 +02:00
|
|
|
|
es_func_fd_create (void **cookie, int fd, unsigned int modeflags, int no_close)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
estream_cookie_fd_t fd_cookie;
|
|
|
|
|
int err;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
fd_cookie = mem_alloc (sizeof (*fd_cookie));
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (! fd_cookie)
|
|
|
|
|
err = -1;
|
|
|
|
|
else
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
#ifdef HAVE_DOSISH_SYSTEM
|
|
|
|
|
/* Make sure it is in binary mode if requested. */
|
|
|
|
|
if ( (modeflags & O_BINARY) )
|
|
|
|
|
setmode (fd, O_BINARY);
|
2008-10-20 15:53:23 +02:00
|
|
|
|
#else
|
|
|
|
|
(void)modeflags;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
#endif
|
2005-01-26 23:24:59 +01:00
|
|
|
|
fd_cookie->fd = fd;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
fd_cookie->no_close = no_close;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
*cookie = fd_cookie;
|
|
|
|
|
err = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Read function for fd objects. */
|
|
|
|
|
static ssize_t
|
2005-06-16 10:12:03 +02:00
|
|
|
|
es_func_fd_read (void *cookie, void *buffer, size_t size)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_fd_t file_cookie = cookie;
|
|
|
|
|
ssize_t bytes_read;
|
|
|
|
|
|
2005-06-03 15:57:24 +02:00
|
|
|
|
do
|
|
|
|
|
bytes_read = ESTREAM_SYS_READ (file_cookie->fd, buffer, size);
|
|
|
|
|
while (bytes_read == -1 && errno == EINTR);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
return bytes_read;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Write function for fd objects. */
|
|
|
|
|
static ssize_t
|
2005-06-16 10:12:03 +02:00
|
|
|
|
es_func_fd_write (void *cookie, const void *buffer, size_t size)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_fd_t file_cookie = cookie;
|
|
|
|
|
ssize_t bytes_written;
|
|
|
|
|
|
2005-06-03 15:57:24 +02:00
|
|
|
|
do
|
|
|
|
|
bytes_written = ESTREAM_SYS_WRITE (file_cookie->fd, buffer, size);
|
|
|
|
|
while (bytes_written == -1 && errno == EINTR);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
return bytes_written;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Seek function for fd objects. */
|
|
|
|
|
static int
|
|
|
|
|
es_func_fd_seek (void *cookie, off_t *offset, int whence)
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_fd_t file_cookie = cookie;
|
|
|
|
|
off_t offset_new;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
offset_new = lseek (file_cookie->fd, *offset, whence);
|
|
|
|
|
if (offset_new == -1)
|
|
|
|
|
err = -1;
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
*offset = offset_new;
|
|
|
|
|
err = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Destroy function for fd objects. */
|
|
|
|
|
static int
|
|
|
|
|
es_func_fd_destroy (void *cookie)
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_fd_t fd_cookie = cookie;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (fd_cookie)
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = fd_cookie->no_close? 0 : close (fd_cookie->fd);
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (fd_cookie);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
static es_cookie_io_functions_t estream_functions_fd =
|
|
|
|
|
{
|
|
|
|
|
es_func_fd_read,
|
|
|
|
|
es_func_fd_write,
|
|
|
|
|
es_func_fd_seek,
|
|
|
|
|
es_func_fd_destroy
|
|
|
|
|
};
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Implementation of FILE* I/O. */
|
|
|
|
|
|
|
|
|
|
/* Cookie for fp objects. */
|
|
|
|
|
typedef struct estream_cookie_fp
|
|
|
|
|
{
|
|
|
|
|
FILE *fp; /* The file pointer we are using for actual output. */
|
|
|
|
|
int no_close; /* If set we won't close the file pointer. */
|
|
|
|
|
} *estream_cookie_fp_t;
|
|
|
|
|
|
|
|
|
|
/* Create function for fd objects. */
|
|
|
|
|
static int
|
2008-10-20 15:53:23 +02:00
|
|
|
|
es_func_fp_create (void **cookie, FILE *fp,
|
|
|
|
|
unsigned int modeflags, int no_close)
|
2007-06-25 13:54:43 +02:00
|
|
|
|
{
|
|
|
|
|
estream_cookie_fp_t fp_cookie;
|
|
|
|
|
int err;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
fp_cookie = mem_alloc (sizeof *fp_cookie);
|
2007-06-25 13:54:43 +02:00
|
|
|
|
if (!fp_cookie)
|
|
|
|
|
err = -1;
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
#ifdef HAVE_DOSISH_SYSTEM
|
|
|
|
|
/* Make sure it is in binary mode if requested. */
|
|
|
|
|
if ( (modeflags & O_BINARY) )
|
|
|
|
|
setmode (fileno (fp), O_BINARY);
|
2008-10-20 15:53:23 +02:00
|
|
|
|
#else
|
|
|
|
|
(void)modeflags;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
#endif
|
|
|
|
|
fp_cookie->fp = fp;
|
|
|
|
|
fp_cookie->no_close = no_close;
|
|
|
|
|
*cookie = fp_cookie;
|
|
|
|
|
err = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Read function for FILE* objects. */
|
|
|
|
|
static ssize_t
|
|
|
|
|
es_func_fp_read (void *cookie, void *buffer, size_t size)
|
|
|
|
|
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_fp_t file_cookie = cookie;
|
|
|
|
|
ssize_t bytes_read;
|
|
|
|
|
|
|
|
|
|
bytes_read = fread (buffer, 1, size, file_cookie->fp);
|
|
|
|
|
if (!bytes_read && ferror (file_cookie->fp))
|
|
|
|
|
return -1;
|
|
|
|
|
return bytes_read;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Write function for FILE* objects. */
|
|
|
|
|
static ssize_t
|
|
|
|
|
es_func_fp_write (void *cookie, const void *buffer, size_t size)
|
|
|
|
|
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_fp_t file_cookie = cookie;
|
|
|
|
|
size_t bytes_written;
|
|
|
|
|
|
|
|
|
|
bytes_written = fwrite (buffer, 1, size, file_cookie->fp);
|
|
|
|
|
if (bytes_written != size)
|
|
|
|
|
return -1;
|
|
|
|
|
return bytes_written;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Seek function for FILE* objects. */
|
|
|
|
|
static int
|
|
|
|
|
es_func_fp_seek (void *cookie, off_t *offset, int whence)
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_fp_t file_cookie = cookie;
|
|
|
|
|
long int offset_new;
|
|
|
|
|
|
|
|
|
|
if ( fseek (file_cookie->fp, (long int)*offset, whence) )
|
|
|
|
|
{
|
|
|
|
|
fprintf (stderr, "\nfseek failed: errno=%d (%s)\n", errno,strerror (errno));
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
offset_new = ftell (file_cookie->fp);
|
|
|
|
|
if (offset_new == -1)
|
|
|
|
|
{
|
|
|
|
|
fprintf (stderr, "\nftell failed: errno=%d (%s)\n", errno,strerror (errno));
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
*offset = offset_new;
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Destroy function for fd objects. */
|
|
|
|
|
static int
|
|
|
|
|
es_func_fp_destroy (void *cookie)
|
|
|
|
|
{
|
|
|
|
|
estream_cookie_fp_t fp_cookie = cookie;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (fp_cookie)
|
|
|
|
|
{
|
|
|
|
|
fflush (fp_cookie->fp);
|
|
|
|
|
err = fp_cookie->no_close? 0 : fclose (fp_cookie->fp);
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (fp_cookie);
|
2007-06-25 13:54:43 +02:00
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static es_cookie_io_functions_t estream_functions_fp =
|
|
|
|
|
{
|
|
|
|
|
es_func_fp_read,
|
|
|
|
|
es_func_fp_write,
|
|
|
|
|
es_func_fp_seek,
|
|
|
|
|
es_func_fp_destroy
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
/* Implementation of file I/O. */
|
|
|
|
|
|
|
|
|
|
/* Create function for file objects. */
|
|
|
|
|
static int
|
|
|
|
|
es_func_file_create (void **cookie, int *filedes,
|
2007-06-25 13:54:43 +02:00
|
|
|
|
const char *path, unsigned int modeflags)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
estream_cookie_fd_t file_cookie;
|
|
|
|
|
int err;
|
|
|
|
|
int fd;
|
|
|
|
|
|
|
|
|
|
err = 0;
|
|
|
|
|
fd = -1;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
file_cookie = mem_alloc (sizeof (*file_cookie));
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (! file_cookie)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
fd = open (path, modeflags, ES_DEFAULT_OPEN_MODE);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (fd == -1)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
2007-06-25 13:54:43 +02:00
|
|
|
|
#ifdef HAVE_DOSISH_SYSTEM
|
|
|
|
|
/* Make sure it is in binary mode if requested. */
|
|
|
|
|
if ( (modeflags & O_BINARY) )
|
|
|
|
|
setmode (fd, O_BINARY);
|
|
|
|
|
#endif
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
file_cookie->fd = fd;
|
2007-07-16 11:53:47 +02:00
|
|
|
|
file_cookie->no_close = 0;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
*cookie = file_cookie;
|
|
|
|
|
*filedes = fd;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (err)
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (file_cookie);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static es_cookie_io_functions_t estream_functions_file =
|
|
|
|
|
{
|
|
|
|
|
es_func_fd_read,
|
|
|
|
|
es_func_fd_write,
|
|
|
|
|
es_func_fd_seek,
|
|
|
|
|
es_func_fd_destroy
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Stream primitives. */
|
|
|
|
|
|
|
|
|
|
static int
|
2007-06-25 13:54:43 +02:00
|
|
|
|
es_convert_mode (const char *mode, unsigned int *modeflags)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
2007-03-19 15:35:04 +01:00
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
/* FIXME: We need to allow all mode flags permutations. */
|
2005-01-26 23:24:59 +01:00
|
|
|
|
struct
|
|
|
|
|
{
|
|
|
|
|
const char *mode;
|
|
|
|
|
unsigned int flags;
|
|
|
|
|
} mode_flags[] = { { "r",
|
|
|
|
|
O_RDONLY },
|
|
|
|
|
{ "rb",
|
2007-06-25 13:54:43 +02:00
|
|
|
|
O_RDONLY | O_BINARY },
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{ "w",
|
|
|
|
|
O_WRONLY | O_TRUNC | O_CREAT },
|
|
|
|
|
{ "wb",
|
2007-06-25 13:54:43 +02:00
|
|
|
|
O_WRONLY | O_TRUNC | O_CREAT | O_BINARY },
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{ "a",
|
|
|
|
|
O_WRONLY | O_APPEND | O_CREAT },
|
|
|
|
|
{ "ab",
|
2007-06-25 13:54:43 +02:00
|
|
|
|
O_WRONLY | O_APPEND | O_CREAT | O_BINARY },
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{ "r+",
|
|
|
|
|
O_RDWR },
|
|
|
|
|
{ "rb+",
|
2007-06-25 13:54:43 +02:00
|
|
|
|
O_RDWR | O_BINARY },
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{ "r+b",
|
2007-06-25 13:54:43 +02:00
|
|
|
|
O_RDONLY | O_WRONLY | O_BINARY },
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{ "w+",
|
|
|
|
|
O_RDWR | O_TRUNC | O_CREAT },
|
|
|
|
|
{ "wb+",
|
2007-06-25 13:54:43 +02:00
|
|
|
|
O_RDWR | O_TRUNC | O_CREAT | O_BINARY },
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{ "w+b",
|
2007-06-25 13:54:43 +02:00
|
|
|
|
O_RDWR | O_TRUNC | O_CREAT | O_BINARY },
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{ "a+",
|
|
|
|
|
O_RDWR | O_CREAT | O_APPEND },
|
|
|
|
|
{ "ab+",
|
2007-06-25 13:54:43 +02:00
|
|
|
|
O_RDWR | O_CREAT | O_APPEND | O_BINARY },
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{ "a+b",
|
2007-06-25 13:54:43 +02:00
|
|
|
|
O_RDWR | O_CREAT | O_APPEND | O_BINARY }
|
|
|
|
|
};
|
2005-01-26 23:24:59 +01:00
|
|
|
|
unsigned int i;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
for (i = 0; i < DIM (mode_flags); i++)
|
|
|
|
|
if (! strcmp (mode_flags[i].mode, mode))
|
|
|
|
|
break;
|
|
|
|
|
if (i == DIM (mode_flags))
|
|
|
|
|
{
|
|
|
|
|
errno = EINVAL;
|
|
|
|
|
err = -1;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
err = 0;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
*modeflags = mode_flags[i].flags;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* Low level stream functionality.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
|
es_fill (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
size_t bytes_read = 0;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (!stream->intern->func_read)
|
|
|
|
|
{
|
|
|
|
|
errno = EOPNOTSUPP;
|
|
|
|
|
err = -1;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
es_cookie_read_function_t func_read = stream->intern->func_read;
|
|
|
|
|
ssize_t ret;
|
|
|
|
|
|
|
|
|
|
ret = (*func_read) (stream->intern->cookie,
|
|
|
|
|
stream->buffer, stream->buffer_size);
|
|
|
|
|
if (ret == -1)
|
|
|
|
|
{
|
|
|
|
|
bytes_read = 0;
|
|
|
|
|
err = -1;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
bytes_read = ret;
|
|
|
|
|
err = 0;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (err)
|
|
|
|
|
stream->intern->indicators.err = 1;
|
|
|
|
|
else if (!bytes_read)
|
|
|
|
|
stream->intern->indicators.eof = 1;
|
|
|
|
|
|
|
|
|
|
stream->intern->offset += stream->data_len;
|
|
|
|
|
stream->data_len = bytes_read;
|
|
|
|
|
stream->data_offset = 0;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
|
es_flush (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
es_cookie_write_function_t func_write = stream->intern->func_write;
|
|
|
|
|
int err;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
assert (stream->flags.writing);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
if (stream->data_offset)
|
|
|
|
|
{
|
|
|
|
|
size_t bytes_written;
|
|
|
|
|
size_t data_flushed;
|
|
|
|
|
ssize_t ret;
|
|
|
|
|
|
|
|
|
|
if (! func_write)
|
|
|
|
|
{
|
|
|
|
|
err = EOPNOTSUPP;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Note: to prevent an endless loop caused by user-provided
|
|
|
|
|
write-functions that pretend to have written more bytes than
|
|
|
|
|
they were asked to write, we have to check for
|
|
|
|
|
"(stream->data_offset - data_flushed) > 0" instead of
|
|
|
|
|
"stream->data_offset - data_flushed". */
|
|
|
|
|
|
|
|
|
|
data_flushed = 0;
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
while ((((ssize_t) (stream->data_offset - data_flushed)) > 0) && (! err))
|
|
|
|
|
{
|
|
|
|
|
ret = (*func_write) (stream->intern->cookie,
|
|
|
|
|
stream->buffer + data_flushed,
|
|
|
|
|
stream->data_offset - data_flushed);
|
|
|
|
|
if (ret == -1)
|
|
|
|
|
{
|
|
|
|
|
bytes_written = 0;
|
|
|
|
|
err = -1;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
bytes_written = ret;
|
|
|
|
|
|
|
|
|
|
data_flushed += bytes_written;
|
|
|
|
|
if (err)
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
stream->data_flushed += data_flushed;
|
|
|
|
|
if (stream->data_offset == data_flushed)
|
|
|
|
|
{
|
|
|
|
|
stream->intern->offset += stream->data_offset;
|
|
|
|
|
stream->data_offset = 0;
|
|
|
|
|
stream->data_flushed = 0;
|
|
|
|
|
|
|
|
|
|
/* Propagate flush event. */
|
|
|
|
|
(*func_write) (stream->intern->cookie, NULL, 0);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (err)
|
|
|
|
|
stream->intern->indicators.err = 1;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Discard buffered data for STREAM. */
|
|
|
|
|
static void
|
|
|
|
|
es_empty (estream_t stream)
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
assert (!stream->flags.writing);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
stream->data_len = 0;
|
|
|
|
|
stream->data_offset = 0;
|
|
|
|
|
stream->unread_data_len = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Initialize STREAM. */
|
|
|
|
|
static void
|
|
|
|
|
es_initialize (estream_t stream,
|
2007-06-25 13:54:43 +02:00
|
|
|
|
void *cookie, int fd, es_cookie_io_functions_t functions,
|
|
|
|
|
unsigned int modeflags)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
stream->intern->cookie = cookie;
|
|
|
|
|
stream->intern->opaque = NULL;
|
|
|
|
|
stream->intern->offset = 0;
|
|
|
|
|
stream->intern->func_read = functions.func_read;
|
|
|
|
|
stream->intern->func_write = functions.func_write;
|
|
|
|
|
stream->intern->func_seek = functions.func_seek;
|
|
|
|
|
stream->intern->func_close = functions.func_close;
|
|
|
|
|
stream->intern->strategy = _IOFBF;
|
|
|
|
|
stream->intern->fd = fd;
|
2006-10-17 16:34:42 +02:00
|
|
|
|
stream->intern->print_err = 0;
|
|
|
|
|
stream->intern->print_errno = 0;
|
|
|
|
|
stream->intern->print_ntotal = 0;
|
|
|
|
|
stream->intern->print_fp = NULL;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
stream->intern->indicators.err = 0;
|
|
|
|
|
stream->intern->indicators.eof = 0;
|
|
|
|
|
stream->intern->deallocate_buffer = 0;
|
|
|
|
|
|
|
|
|
|
stream->data_len = 0;
|
|
|
|
|
stream->data_offset = 0;
|
|
|
|
|
stream->data_flushed = 0;
|
|
|
|
|
stream->unread_data_len = 0;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
/* Depending on the modeflags we set whether we start in writing or
|
|
|
|
|
reading mode. This is required in case we are working on a
|
|
|
|
|
wronly stream which is not seeekable (like stdout). Without this
|
|
|
|
|
pre-initialization we would do a seek at the first write call and
|
|
|
|
|
as this will fail no utput will be delivered. */
|
|
|
|
|
if ((modeflags & O_WRONLY) || (modeflags & O_RDWR) )
|
|
|
|
|
stream->flags.writing = 1;
|
|
|
|
|
else
|
|
|
|
|
stream->flags.writing = 0;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Deinitialize STREAM. */
|
|
|
|
|
static int
|
|
|
|
|
es_deinitialize (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
es_cookie_close_function_t func_close;
|
|
|
|
|
int err, tmp_err;
|
|
|
|
|
|
2006-10-17 16:34:42 +02:00
|
|
|
|
if (stream->intern->print_fp)
|
|
|
|
|
{
|
|
|
|
|
int save_errno = errno;
|
|
|
|
|
fclose (stream->intern->print_fp);
|
|
|
|
|
stream->intern->print_fp = NULL;
|
|
|
|
|
errno = save_errno;
|
|
|
|
|
}
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
func_close = stream->intern->func_close;
|
|
|
|
|
|
|
|
|
|
err = 0;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
if (stream->flags.writing)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
SET_UNLESS_NONZERO (err, tmp_err, es_flush (stream));
|
|
|
|
|
if (func_close)
|
|
|
|
|
SET_UNLESS_NONZERO (err, tmp_err, (*func_close) (stream->intern->cookie));
|
2006-10-17 16:34:42 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Create a new stream object, initialize it. */
|
|
|
|
|
static int
|
|
|
|
|
es_create (estream_t *stream, void *cookie, int fd,
|
2007-06-25 13:54:43 +02:00
|
|
|
|
es_cookie_io_functions_t functions, unsigned int modeflags)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
estream_internal_t stream_internal_new;
|
|
|
|
|
estream_t stream_new;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
stream_new = NULL;
|
|
|
|
|
stream_internal_new = NULL;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
stream_new = mem_alloc (sizeof (*stream_new));
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (! stream_new)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
stream_internal_new = mem_alloc (sizeof (*stream_internal_new));
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (! stream_internal_new)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
stream_new->buffer = stream_internal_new->buffer;
|
|
|
|
|
stream_new->buffer_size = sizeof (stream_internal_new->buffer);
|
|
|
|
|
stream_new->unread_buffer = stream_internal_new->unread_buffer;
|
|
|
|
|
stream_new->unread_buffer_size = sizeof (stream_internal_new->unread_buffer);
|
|
|
|
|
stream_new->intern = stream_internal_new;
|
|
|
|
|
|
|
|
|
|
ESTREAM_MUTEX_INITIALIZE (stream_new->intern->lock);
|
2007-06-25 13:54:43 +02:00
|
|
|
|
es_initialize (stream_new, cookie, fd, functions, modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
err = es_list_add (stream_new);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
*stream = stream_new;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (err)
|
|
|
|
|
{
|
|
|
|
|
if (stream_new)
|
|
|
|
|
{
|
|
|
|
|
es_deinitialize (stream_new);
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (stream_new);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Deinitialize a stream object and destroy it. */
|
|
|
|
|
static int
|
|
|
|
|
es_destroy (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
int err = 0;
|
|
|
|
|
|
|
|
|
|
if (stream)
|
|
|
|
|
{
|
|
|
|
|
es_list_remove (stream);
|
|
|
|
|
err = es_deinitialize (stream);
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (stream->intern);
|
|
|
|
|
mem_free (stream);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Try to read BYTES_TO_READ bytes FROM STREAM into BUFFER in
|
|
|
|
|
unbuffered-mode, storing the amount of bytes read in
|
|
|
|
|
*BYTES_READ. */
|
|
|
|
|
static int
|
|
|
|
|
es_read_nbf (estream_t ES__RESTRICT stream,
|
|
|
|
|
unsigned char *ES__RESTRICT buffer,
|
|
|
|
|
size_t bytes_to_read, size_t *ES__RESTRICT bytes_read)
|
|
|
|
|
{
|
|
|
|
|
es_cookie_read_function_t func_read = stream->intern->func_read;
|
|
|
|
|
size_t data_read;
|
|
|
|
|
ssize_t ret;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
data_read = 0;
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
while (bytes_to_read - data_read)
|
|
|
|
|
{
|
|
|
|
|
ret = (*func_read) (stream->intern->cookie,
|
|
|
|
|
buffer + data_read, bytes_to_read - data_read);
|
|
|
|
|
if (ret == -1)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
else if (ret)
|
|
|
|
|
data_read += ret;
|
|
|
|
|
else
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
stream->intern->offset += data_read;
|
|
|
|
|
*bytes_read = data_read;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Try to read BYTES_TO_READ bytes FROM STREAM into BUFFER in
|
|
|
|
|
fully-buffered-mode, storing the amount of bytes read in
|
|
|
|
|
*BYTES_READ. */
|
|
|
|
|
static int
|
|
|
|
|
es_read_fbf (estream_t ES__RESTRICT stream,
|
|
|
|
|
unsigned char *ES__RESTRICT buffer,
|
|
|
|
|
size_t bytes_to_read, size_t *ES__RESTRICT bytes_read)
|
|
|
|
|
{
|
|
|
|
|
size_t data_available;
|
|
|
|
|
size_t data_to_read;
|
|
|
|
|
size_t data_read;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
data_read = 0;
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
while ((bytes_to_read - data_read) && (! err))
|
|
|
|
|
{
|
|
|
|
|
if (stream->data_offset == stream->data_len)
|
|
|
|
|
{
|
|
|
|
|
/* Nothing more to read in current container, try to
|
|
|
|
|
fill container with new data. */
|
|
|
|
|
err = es_fill (stream);
|
|
|
|
|
if (! err)
|
|
|
|
|
if (! stream->data_len)
|
|
|
|
|
/* Filling did not result in any data read. */
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (! err)
|
|
|
|
|
{
|
|
|
|
|
/* Filling resulted in some new data. */
|
|
|
|
|
|
|
|
|
|
data_to_read = bytes_to_read - data_read;
|
|
|
|
|
data_available = stream->data_len - stream->data_offset;
|
|
|
|
|
if (data_to_read > data_available)
|
|
|
|
|
data_to_read = data_available;
|
|
|
|
|
|
|
|
|
|
memcpy (buffer + data_read,
|
|
|
|
|
stream->buffer + stream->data_offset, data_to_read);
|
|
|
|
|
stream->data_offset += data_to_read;
|
|
|
|
|
data_read += data_to_read;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
*bytes_read = data_read;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Try to read BYTES_TO_READ bytes FROM STREAM into BUFFER in
|
|
|
|
|
line-buffered-mode, storing the amount of bytes read in
|
|
|
|
|
*BYTES_READ. */
|
|
|
|
|
static int
|
|
|
|
|
es_read_lbf (estream_t ES__RESTRICT stream,
|
|
|
|
|
unsigned char *ES__RESTRICT buffer,
|
|
|
|
|
size_t bytes_to_read, size_t *ES__RESTRICT bytes_read)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
err = es_read_fbf (stream, buffer, bytes_to_read, bytes_read);
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Try to read BYTES_TO_READ bytes FROM STREAM into BUFFER, storing
|
|
|
|
|
*the amount of bytes read in BYTES_READ. */
|
|
|
|
|
static int
|
|
|
|
|
es_readn (estream_t ES__RESTRICT stream,
|
2005-06-16 10:12:03 +02:00
|
|
|
|
void *ES__RESTRICT buffer_arg,
|
2005-01-26 23:24:59 +01:00
|
|
|
|
size_t bytes_to_read, size_t *ES__RESTRICT bytes_read)
|
|
|
|
|
{
|
2005-06-16 10:12:03 +02:00
|
|
|
|
unsigned char *buffer = (unsigned char *)buffer_arg;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
size_t data_read_unread, data_read;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
data_read_unread = 0;
|
|
|
|
|
data_read = 0;
|
|
|
|
|
err = 0;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
if (stream->flags.writing)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
/* Switching to reading mode -> flush output. */
|
|
|
|
|
err = es_flush (stream);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
stream->flags.writing = 0;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Read unread data first. */
|
|
|
|
|
while ((bytes_to_read - data_read_unread) && stream->unread_data_len)
|
|
|
|
|
{
|
|
|
|
|
buffer[data_read_unread]
|
|
|
|
|
= stream->unread_buffer[stream->unread_data_len - 1];
|
|
|
|
|
stream->unread_data_len--;
|
|
|
|
|
data_read_unread++;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
switch (stream->intern->strategy)
|
|
|
|
|
{
|
|
|
|
|
case _IONBF:
|
|
|
|
|
err = es_read_nbf (stream,
|
|
|
|
|
buffer + data_read_unread,
|
|
|
|
|
bytes_to_read - data_read_unread, &data_read);
|
|
|
|
|
break;
|
|
|
|
|
case _IOLBF:
|
|
|
|
|
err = es_read_lbf (stream,
|
|
|
|
|
buffer + data_read_unread,
|
|
|
|
|
bytes_to_read - data_read_unread, &data_read);
|
|
|
|
|
break;
|
|
|
|
|
case _IOFBF:
|
|
|
|
|
err = es_read_fbf (stream,
|
|
|
|
|
buffer + data_read_unread,
|
|
|
|
|
bytes_to_read - data_read_unread, &data_read);
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (bytes_read)
|
|
|
|
|
*bytes_read = data_read_unread + data_read;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Try to unread DATA_N bytes from DATA into STREAM, storing the
|
|
|
|
|
amount of bytes succesfully unread in *BYTES_UNREAD. */
|
|
|
|
|
static void
|
|
|
|
|
es_unreadn (estream_t ES__RESTRICT stream,
|
|
|
|
|
const unsigned char *ES__RESTRICT data, size_t data_n,
|
|
|
|
|
size_t *ES__RESTRICT bytes_unread)
|
|
|
|
|
{
|
|
|
|
|
size_t space_left;
|
|
|
|
|
|
|
|
|
|
space_left = stream->unread_buffer_size - stream->unread_data_len;
|
|
|
|
|
|
|
|
|
|
if (data_n > space_left)
|
|
|
|
|
data_n = space_left;
|
|
|
|
|
|
|
|
|
|
if (! data_n)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
memcpy (stream->unread_buffer + stream->unread_data_len, data, data_n);
|
|
|
|
|
stream->unread_data_len += data_n;
|
|
|
|
|
stream->intern->indicators.eof = 0;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (bytes_unread)
|
|
|
|
|
*bytes_unread = data_n;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Seek in STREAM. */
|
|
|
|
|
static int
|
|
|
|
|
es_seek (estream_t ES__RESTRICT stream, off_t offset, int whence,
|
|
|
|
|
off_t *ES__RESTRICT offset_new)
|
|
|
|
|
{
|
|
|
|
|
es_cookie_seek_function_t func_seek = stream->intern->func_seek;
|
|
|
|
|
int err, ret;
|
|
|
|
|
off_t off;
|
|
|
|
|
|
|
|
|
|
if (! func_seek)
|
|
|
|
|
{
|
|
|
|
|
errno = EOPNOTSUPP;
|
|
|
|
|
err = -1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
if (stream->flags.writing)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
/* Flush data first in order to prevent flushing it to the wrong
|
|
|
|
|
offset. */
|
|
|
|
|
err = es_flush (stream);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
stream->flags.writing = 0;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
off = offset;
|
|
|
|
|
if (whence == SEEK_CUR)
|
|
|
|
|
{
|
|
|
|
|
off = off - stream->data_len + stream->data_offset;
|
|
|
|
|
off -= stream->unread_data_len;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ret = (*func_seek) (stream->intern->cookie, &off, whence);
|
|
|
|
|
if (ret == -1)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
err = 0;
|
|
|
|
|
es_empty (stream);
|
|
|
|
|
|
|
|
|
|
if (offset_new)
|
|
|
|
|
*offset_new = off;
|
|
|
|
|
|
|
|
|
|
stream->intern->indicators.eof = 0;
|
|
|
|
|
stream->intern->offset = off;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (err)
|
|
|
|
|
stream->intern->indicators.err = 1;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Write BYTES_TO_WRITE bytes from BUFFER into STREAM in
|
|
|
|
|
unbuffered-mode, storing the amount of bytes written in
|
|
|
|
|
*BYTES_WRITTEN. */
|
|
|
|
|
static int
|
|
|
|
|
es_write_nbf (estream_t ES__RESTRICT stream,
|
|
|
|
|
const unsigned char *ES__RESTRICT buffer,
|
|
|
|
|
size_t bytes_to_write, size_t *ES__RESTRICT bytes_written)
|
|
|
|
|
{
|
|
|
|
|
es_cookie_write_function_t func_write = stream->intern->func_write;
|
|
|
|
|
size_t data_written;
|
|
|
|
|
ssize_t ret;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (bytes_to_write && (! func_write))
|
|
|
|
|
{
|
|
|
|
|
err = EOPNOTSUPP;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
data_written = 0;
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
while (bytes_to_write - data_written)
|
|
|
|
|
{
|
|
|
|
|
ret = (*func_write) (stream->intern->cookie,
|
|
|
|
|
buffer + data_written,
|
|
|
|
|
bytes_to_write - data_written);
|
|
|
|
|
if (ret == -1)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
data_written += ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
stream->intern->offset += data_written;
|
|
|
|
|
*bytes_written = data_written;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Write BYTES_TO_WRITE bytes from BUFFER into STREAM in
|
|
|
|
|
fully-buffered-mode, storing the amount of bytes written in
|
|
|
|
|
*BYTES_WRITTEN. */
|
|
|
|
|
static int
|
|
|
|
|
es_write_fbf (estream_t ES__RESTRICT stream,
|
|
|
|
|
const unsigned char *ES__RESTRICT buffer,
|
|
|
|
|
size_t bytes_to_write, size_t *ES__RESTRICT bytes_written)
|
|
|
|
|
{
|
|
|
|
|
size_t space_available;
|
|
|
|
|
size_t data_to_write;
|
|
|
|
|
size_t data_written;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
data_written = 0;
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
while ((bytes_to_write - data_written) && (! err))
|
|
|
|
|
{
|
|
|
|
|
if (stream->data_offset == stream->buffer_size)
|
|
|
|
|
/* Container full, flush buffer. */
|
|
|
|
|
err = es_flush (stream);
|
|
|
|
|
|
|
|
|
|
if (! err)
|
|
|
|
|
{
|
|
|
|
|
/* Flushing resulted in empty container. */
|
|
|
|
|
|
|
|
|
|
data_to_write = bytes_to_write - data_written;
|
|
|
|
|
space_available = stream->buffer_size - stream->data_offset;
|
|
|
|
|
if (data_to_write > space_available)
|
|
|
|
|
data_to_write = space_available;
|
|
|
|
|
|
|
|
|
|
memcpy (stream->buffer + stream->data_offset,
|
|
|
|
|
buffer + data_written, data_to_write);
|
|
|
|
|
stream->data_offset += data_to_write;
|
|
|
|
|
data_written += data_to_write;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
*bytes_written = data_written;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Write BYTES_TO_WRITE bytes from BUFFER into STREAM in
|
|
|
|
|
line-buffered-mode, storing the amount of bytes written in
|
|
|
|
|
*BYTES_WRITTEN. */
|
|
|
|
|
static int
|
|
|
|
|
es_write_lbf (estream_t ES__RESTRICT stream,
|
|
|
|
|
const unsigned char *ES__RESTRICT buffer,
|
|
|
|
|
size_t bytes_to_write, size_t *ES__RESTRICT bytes_written)
|
|
|
|
|
{
|
|
|
|
|
size_t data_flushed = 0;
|
|
|
|
|
size_t data_buffered = 0;
|
|
|
|
|
unsigned char *nlp;
|
|
|
|
|
int err = 0;
|
|
|
|
|
|
|
|
|
|
nlp = memrchr (buffer, '\n', bytes_to_write);
|
|
|
|
|
if (nlp)
|
|
|
|
|
{
|
|
|
|
|
/* Found a newline, directly write up to (including) this
|
|
|
|
|
character. */
|
|
|
|
|
err = es_flush (stream);
|
|
|
|
|
if (!err)
|
|
|
|
|
err = es_write_nbf (stream, buffer, nlp - buffer + 1, &data_flushed);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!err)
|
|
|
|
|
{
|
|
|
|
|
/* Write remaining data fully buffered. */
|
|
|
|
|
err = es_write_fbf (stream, buffer + data_flushed,
|
|
|
|
|
bytes_to_write - data_flushed, &data_buffered);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
*bytes_written = data_flushed + data_buffered;
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Write BYTES_TO_WRITE bytes from BUFFER into STREAM in, storing the
|
|
|
|
|
amount of bytes written in BYTES_WRITTEN. */
|
|
|
|
|
static int
|
|
|
|
|
es_writen (estream_t ES__RESTRICT stream,
|
2005-06-16 10:12:03 +02:00
|
|
|
|
const void *ES__RESTRICT buffer,
|
2005-01-26 23:24:59 +01:00
|
|
|
|
size_t bytes_to_write, size_t *ES__RESTRICT bytes_written)
|
|
|
|
|
{
|
|
|
|
|
size_t data_written;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
data_written = 0;
|
|
|
|
|
err = 0;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
if (!stream->flags.writing)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
/* Switching to writing mode -> discard input data and seek to
|
2006-08-11 13:04:38 +02:00
|
|
|
|
position at which reading has stopped. We can do this only
|
|
|
|
|
if a seek function has been registered. */
|
|
|
|
|
if (stream->intern->func_seek)
|
|
|
|
|
{
|
|
|
|
|
err = es_seek (stream, 0, SEEK_CUR, NULL);
|
|
|
|
|
if (err)
|
|
|
|
|
{
|
|
|
|
|
if (errno == ESPIPE)
|
|
|
|
|
err = 0;
|
|
|
|
|
else
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
}
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
switch (stream->intern->strategy)
|
|
|
|
|
{
|
|
|
|
|
case _IONBF:
|
|
|
|
|
err = es_write_nbf (stream, buffer, bytes_to_write, &data_written);
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
case _IOLBF:
|
|
|
|
|
err = es_write_lbf (stream, buffer, bytes_to_write, &data_written);
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
case _IOFBF:
|
|
|
|
|
err = es_write_fbf (stream, buffer, bytes_to_write, &data_written);
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (bytes_written)
|
|
|
|
|
*bytes_written = data_written;
|
|
|
|
|
if (data_written)
|
2007-06-25 13:54:43 +02:00
|
|
|
|
if (!stream->flags.writing)
|
|
|
|
|
stream->flags.writing = 1;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
|
es_peek (estream_t ES__RESTRICT stream, unsigned char **ES__RESTRICT data,
|
|
|
|
|
size_t *ES__RESTRICT data_len)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
if (stream->flags.writing)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
/* Switching to reading mode -> flush output. */
|
|
|
|
|
err = es_flush (stream);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
stream->flags.writing = 0;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (stream->data_offset == stream->data_len)
|
|
|
|
|
{
|
|
|
|
|
/* Refill container. */
|
|
|
|
|
err = es_fill (stream);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (data)
|
|
|
|
|
*data = stream->buffer + stream->data_offset;
|
|
|
|
|
if (data_len)
|
|
|
|
|
*data_len = stream->data_len - stream->data_offset;
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Skip SIZE bytes of input data contained in buffer. */
|
|
|
|
|
static int
|
|
|
|
|
es_skip (estream_t stream, size_t size)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (stream->data_offset + size > stream->data_len)
|
|
|
|
|
{
|
|
|
|
|
errno = EINVAL;
|
|
|
|
|
err = -1;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
stream->data_offset += size;
|
|
|
|
|
err = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static int
|
2006-08-14 16:40:07 +02:00
|
|
|
|
doreadline (estream_t ES__RESTRICT stream, size_t max_length,
|
2007-08-22 12:55:07 +02:00
|
|
|
|
char *ES__RESTRICT *ES__RESTRICT line,
|
|
|
|
|
size_t *ES__RESTRICT line_length)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
size_t space_left;
|
|
|
|
|
size_t line_size;
|
|
|
|
|
estream_t line_stream;
|
|
|
|
|
char *line_new;
|
|
|
|
|
void *line_stream_cookie;
|
|
|
|
|
char *newline;
|
|
|
|
|
unsigned char *data;
|
|
|
|
|
size_t data_len;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
line_new = NULL;
|
|
|
|
|
line_stream = NULL;
|
|
|
|
|
line_stream_cookie = NULL;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
err = es_func_mem_create (&line_stream_cookie, NULL, 0, 0,
|
|
|
|
|
BUFFER_BLOCK_SIZE, 1,
|
|
|
|
|
mem_realloc, mem_free,
|
|
|
|
|
O_RDWR,
|
|
|
|
|
0);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
err = es_create (&line_stream, line_stream_cookie, -1,
|
2007-06-25 13:54:43 +02:00
|
|
|
|
estream_functions_mem, O_RDWR);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
space_left = max_length;
|
|
|
|
|
line_size = 0;
|
|
|
|
|
while (1)
|
|
|
|
|
{
|
|
|
|
|
if (max_length && (space_left == 1))
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
err = es_peek (stream, &data, &data_len);
|
|
|
|
|
if (err || (! data_len))
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
if (data_len > (space_left - 1))
|
|
|
|
|
data_len = space_left - 1;
|
|
|
|
|
|
|
|
|
|
newline = memchr (data, '\n', data_len);
|
|
|
|
|
if (newline)
|
|
|
|
|
{
|
|
|
|
|
data_len = (newline - (char *) data) + 1;
|
|
|
|
|
err = es_write (line_stream, data, data_len, NULL);
|
|
|
|
|
if (! err)
|
|
|
|
|
{
|
|
|
|
|
space_left -= data_len;
|
|
|
|
|
line_size += data_len;
|
|
|
|
|
es_skip (stream, data_len);
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
err = es_write (line_stream, data, data_len, NULL);
|
|
|
|
|
if (! err)
|
|
|
|
|
{
|
|
|
|
|
space_left -= data_len;
|
|
|
|
|
line_size += data_len;
|
|
|
|
|
es_skip (stream, data_len);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (err)
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
/* Complete line has been written to line_stream. */
|
|
|
|
|
|
|
|
|
|
if ((max_length > 1) && (! line_size))
|
|
|
|
|
{
|
|
|
|
|
stream->intern->indicators.eof = 1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
err = es_seek (line_stream, 0, SEEK_SET, NULL);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
if (! *line)
|
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
line_new = mem_alloc (line_size + 1);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (! line_new)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
line_new = *line;
|
|
|
|
|
|
|
|
|
|
err = es_read (line_stream, line_new, line_size, NULL);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
line_new[line_size] = '\0';
|
|
|
|
|
|
|
|
|
|
if (! *line)
|
|
|
|
|
*line = line_new;
|
|
|
|
|
if (line_length)
|
|
|
|
|
*line_length = line_size;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (line_stream)
|
|
|
|
|
es_destroy (line_stream);
|
|
|
|
|
else if (line_stream_cookie)
|
|
|
|
|
es_func_mem_destroy (line_stream_cookie);
|
|
|
|
|
|
|
|
|
|
if (err)
|
|
|
|
|
{
|
|
|
|
|
if (! *line)
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (line_new);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
stream->intern->indicators.err = 1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2007-05-15 18:10:48 +02:00
|
|
|
|
/* Output fucntion used for estream_format. */
|
|
|
|
|
static int
|
|
|
|
|
print_writer (void *outfncarg, const char *buf, size_t buflen)
|
2006-10-17 16:34:42 +02:00
|
|
|
|
{
|
2007-05-15 18:10:48 +02:00
|
|
|
|
estream_t stream = outfncarg;
|
2006-10-17 16:34:42 +02:00
|
|
|
|
size_t nwritten;
|
2007-05-15 18:10:48 +02:00
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
|
|
nwritten = 0;
|
|
|
|
|
rc = es_writen (stream, buf, buflen, &nwritten);
|
|
|
|
|
stream->intern->print_ntotal += nwritten;
|
|
|
|
|
return rc;
|
2006-10-17 16:34:42 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* The core of our printf function. This is called in locked state. */
|
2005-01-26 23:24:59 +01:00
|
|
|
|
static int
|
|
|
|
|
es_print (estream_t ES__RESTRICT stream,
|
|
|
|
|
const char *ES__RESTRICT format, va_list ap)
|
|
|
|
|
{
|
2007-05-15 18:10:48 +02:00
|
|
|
|
int rc;
|
2006-10-17 16:34:42 +02:00
|
|
|
|
|
|
|
|
|
stream->intern->print_ntotal = 0;
|
2007-05-15 18:10:48 +02:00
|
|
|
|
rc = estream_format (print_writer, stream, format, ap);
|
|
|
|
|
if (rc)
|
|
|
|
|
return -1;
|
2006-10-17 16:34:42 +02:00
|
|
|
|
return (int)stream->intern->print_ntotal;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static void
|
|
|
|
|
es_set_indicators (estream_t stream, int ind_err, int ind_eof)
|
|
|
|
|
{
|
|
|
|
|
if (ind_err != -1)
|
|
|
|
|
stream->intern->indicators.err = ind_err ? 1 : 0;
|
|
|
|
|
if (ind_eof != -1)
|
|
|
|
|
stream->intern->indicators.eof = ind_eof ? 1 : 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
|
es_get_indicator (estream_t stream, int ind_err, int ind_eof)
|
|
|
|
|
{
|
|
|
|
|
int ret = 0;
|
|
|
|
|
|
|
|
|
|
if (ind_err)
|
|
|
|
|
ret = stream->intern->indicators.err;
|
|
|
|
|
else if (ind_eof)
|
|
|
|
|
ret = stream->intern->indicators.eof;
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
|
es_set_buffering (estream_t ES__RESTRICT stream,
|
|
|
|
|
char *ES__RESTRICT buffer, int mode, size_t size)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
/* Flush or empty buffer depending on mode. */
|
2007-06-25 13:54:43 +02:00
|
|
|
|
if (stream->flags.writing)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
|
|
|
|
err = es_flush (stream);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
es_empty (stream);
|
|
|
|
|
|
|
|
|
|
es_set_indicators (stream, -1, 0);
|
|
|
|
|
|
|
|
|
|
/* Free old buffer in case that was allocated by this function. */
|
|
|
|
|
if (stream->intern->deallocate_buffer)
|
|
|
|
|
{
|
|
|
|
|
stream->intern->deallocate_buffer = 0;
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (stream->buffer);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
stream->buffer = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (mode == _IONBF)
|
|
|
|
|
stream->buffer_size = 0;
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
void *buffer_new;
|
|
|
|
|
|
|
|
|
|
if (buffer)
|
|
|
|
|
buffer_new = buffer;
|
|
|
|
|
else
|
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
buffer_new = mem_alloc (size);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (! buffer_new)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
stream->buffer = buffer_new;
|
|
|
|
|
stream->buffer_size = size;
|
|
|
|
|
if (! buffer)
|
|
|
|
|
stream->intern->deallocate_buffer = 1;
|
|
|
|
|
}
|
|
|
|
|
stream->intern->strategy = mode;
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static off_t
|
|
|
|
|
es_offset_calculate (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
off_t offset;
|
|
|
|
|
|
|
|
|
|
offset = stream->intern->offset + stream->data_offset;
|
|
|
|
|
if (offset < stream->unread_data_len)
|
|
|
|
|
/* Offset undefined. */
|
|
|
|
|
offset = 0;
|
|
|
|
|
else
|
|
|
|
|
offset -= stream->unread_data_len;
|
|
|
|
|
|
|
|
|
|
return offset;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static void
|
|
|
|
|
es_opaque_ctrl (estream_t ES__RESTRICT stream, void *ES__RESTRICT opaque_new,
|
|
|
|
|
void **ES__RESTRICT opaque_old)
|
|
|
|
|
{
|
|
|
|
|
if (opaque_old)
|
|
|
|
|
*opaque_old = stream->intern->opaque;
|
|
|
|
|
if (opaque_new)
|
|
|
|
|
stream->intern->opaque = opaque_new;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
|
es_get_fd (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
return stream->intern->fd;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* API. */
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_init (void)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
err = es_init_do ();
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
estream_t
|
|
|
|
|
es_fopen (const char *ES__RESTRICT path, const char *ES__RESTRICT mode)
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
unsigned int modeflags;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
int create_called;
|
|
|
|
|
estream_t stream;
|
|
|
|
|
void *cookie;
|
|
|
|
|
int err;
|
|
|
|
|
int fd;
|
|
|
|
|
|
|
|
|
|
stream = NULL;
|
|
|
|
|
cookie = NULL;
|
|
|
|
|
create_called = 0;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_convert_mode (mode, &modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_func_file_create (&cookie, &fd, path, modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
create_called = 1;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_create (&stream, cookie, fd, estream_functions_file, modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (err && create_called)
|
|
|
|
|
(*estream_functions_file.func_close) (cookie);
|
|
|
|
|
|
|
|
|
|
return stream;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
estream_t
|
|
|
|
|
es_mopen (unsigned char *ES__RESTRICT data, size_t data_n, size_t data_len,
|
|
|
|
|
unsigned int grow,
|
|
|
|
|
func_realloc_t func_realloc, func_free_t func_free,
|
|
|
|
|
const char *ES__RESTRICT mode)
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
unsigned int modeflags;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
int create_called;
|
|
|
|
|
estream_t stream;
|
|
|
|
|
void *cookie;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
cookie = 0;
|
|
|
|
|
stream = NULL;
|
|
|
|
|
create_called = 0;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_convert_mode (mode, &modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
err = es_func_mem_create (&cookie, data, data_n, data_len,
|
2007-08-22 12:55:07 +02:00
|
|
|
|
BUFFER_BLOCK_SIZE, grow,
|
|
|
|
|
func_realloc, func_free, modeflags, 0);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
create_called = 1;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_create (&stream, cookie, -1, estream_functions_mem, modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (err && create_called)
|
|
|
|
|
(*estream_functions_mem.func_close) (cookie);
|
|
|
|
|
|
|
|
|
|
return stream;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
estream_t
|
2007-08-22 12:55:07 +02:00
|
|
|
|
es_fopenmem (size_t memlimit, const char *ES__RESTRICT mode)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
unsigned int modeflags;
|
2007-08-22 12:55:07 +02:00
|
|
|
|
estream_t stream = NULL;
|
|
|
|
|
void *cookie = NULL;
|
|
|
|
|
|
|
|
|
|
/* Memory streams are always read/write. We use MODE only to get
|
|
|
|
|
the append flag. */
|
|
|
|
|
if (es_convert_mode (mode, &modeflags))
|
|
|
|
|
return NULL;
|
|
|
|
|
modeflags |= O_RDWR;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
if (es_func_mem_create (&cookie, NULL, 0, 0,
|
|
|
|
|
BUFFER_BLOCK_SIZE, 1,
|
|
|
|
|
mem_realloc, mem_free, modeflags,
|
|
|
|
|
memlimit))
|
|
|
|
|
return NULL;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
if (es_create (&stream, cookie, -1, estream_functions_mem, modeflags))
|
2005-01-26 23:24:59 +01:00
|
|
|
|
(*estream_functions_mem.func_close) (cookie);
|
|
|
|
|
|
|
|
|
|
return stream;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
estream_t
|
|
|
|
|
es_fopencookie (void *ES__RESTRICT cookie,
|
|
|
|
|
const char *ES__RESTRICT mode,
|
|
|
|
|
es_cookie_io_functions_t functions)
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
unsigned int modeflags;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
estream_t stream;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
stream = NULL;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
modeflags = 0;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_convert_mode (mode, &modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_create (&stream, cookie, -1, functions, modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
return stream;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
estream_t
|
2007-06-25 13:54:43 +02:00
|
|
|
|
do_fdopen (int filedes, const char *mode, int no_close)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
unsigned int modeflags;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
int create_called;
|
|
|
|
|
estream_t stream;
|
|
|
|
|
void *cookie;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
stream = NULL;
|
|
|
|
|
cookie = NULL;
|
|
|
|
|
create_called = 0;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_convert_mode (mode, &modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_func_fd_create (&cookie, filedes, modeflags, no_close);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
create_called = 1;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_create (&stream, cookie, filedes, estream_functions_fd, modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (err && create_called)
|
|
|
|
|
(*estream_functions_fd.func_close) (cookie);
|
|
|
|
|
|
|
|
|
|
return stream;
|
|
|
|
|
}
|
2007-06-25 13:54:43 +02:00
|
|
|
|
|
|
|
|
|
estream_t
|
|
|
|
|
es_fdopen (int filedes, const char *mode)
|
|
|
|
|
{
|
|
|
|
|
return do_fdopen (filedes, mode, 0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* A variant of es_fdopen which does not close FILEDES at the end. */
|
|
|
|
|
estream_t
|
|
|
|
|
es_fdopen_nc (int filedes, const char *mode)
|
|
|
|
|
{
|
|
|
|
|
return do_fdopen (filedes, mode, 1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
estream_t
|
|
|
|
|
do_fpopen (FILE *fp, const char *mode, int no_close)
|
|
|
|
|
{
|
|
|
|
|
unsigned int modeflags;
|
|
|
|
|
int create_called;
|
|
|
|
|
estream_t stream;
|
|
|
|
|
void *cookie;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
stream = NULL;
|
|
|
|
|
cookie = NULL;
|
|
|
|
|
create_called = 0;
|
|
|
|
|
|
|
|
|
|
err = es_convert_mode (mode, &modeflags);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
fflush (fp);
|
|
|
|
|
err = es_func_fp_create (&cookie, fp, modeflags, no_close);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
create_called = 1;
|
|
|
|
|
err = es_create (&stream, cookie, fileno (fp), estream_functions_fp,
|
|
|
|
|
modeflags);
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (err && create_called)
|
|
|
|
|
(*estream_functions_fp.func_close) (cookie);
|
|
|
|
|
|
|
|
|
|
return stream;
|
|
|
|
|
}
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
/* Create an estream from the stdio stream FP. This mechanism is
|
|
|
|
|
useful in case the stdio streams have special properties and may
|
|
|
|
|
not be mixed with fd based functions. This is for example the case
|
|
|
|
|
under Windows where the 3 standard streams are associated with the
|
|
|
|
|
console whereas a duped and fd-opened stream of one of this stream
|
|
|
|
|
won't be associated with the console. As this messes things up it
|
|
|
|
|
is easier to keep on using the standard I/O stream as a backend for
|
|
|
|
|
estream. */
|
|
|
|
|
estream_t
|
|
|
|
|
es_fpopen (FILE *fp, const char *mode)
|
|
|
|
|
{
|
|
|
|
|
return do_fpopen (fp, mode, 0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Same as es_fpopen but does not close FP at the end. */
|
|
|
|
|
estream_t
|
|
|
|
|
es_fpopen_nc (FILE *fp, const char *mode)
|
|
|
|
|
{
|
|
|
|
|
return do_fpopen (fp, mode, 1);
|
|
|
|
|
}
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
estream_t
|
|
|
|
|
es_freopen (const char *ES__RESTRICT path, const char *ES__RESTRICT mode,
|
|
|
|
|
estream_t ES__RESTRICT stream)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (path)
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
unsigned int modeflags;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
int create_called;
|
|
|
|
|
void *cookie;
|
|
|
|
|
int fd;
|
|
|
|
|
|
|
|
|
|
cookie = NULL;
|
|
|
|
|
create_called = 0;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
|
|
|
|
|
es_deinitialize (stream);
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_convert_mode (mode, &modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto leave;
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_func_file_create (&cookie, &fd, path, modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto leave;
|
|
|
|
|
|
|
|
|
|
create_called = 1;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
es_initialize (stream, cookie, fd, estream_functions_file, modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
leave:
|
|
|
|
|
|
|
|
|
|
if (err)
|
|
|
|
|
{
|
|
|
|
|
if (create_called)
|
|
|
|
|
es_func_fd_destroy (cookie);
|
|
|
|
|
|
|
|
|
|
es_destroy (stream);
|
|
|
|
|
stream = NULL;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
/* FIXME? We don't support re-opening at the moment. */
|
|
|
|
|
errno = EINVAL;
|
|
|
|
|
es_deinitialize (stream);
|
|
|
|
|
es_destroy (stream);
|
|
|
|
|
stream = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return stream;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_fclose (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
err = es_destroy (stream);
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_fileno_unlocked (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
return es_get_fd (stream);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
es_flockfile (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_ftrylockfile (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
return ESTREAM_TRYLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
es_funlockfile (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_fileno (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
ret = es_fileno_unlocked (stream);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_feof_unlocked (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
return es_get_indicator (stream, 0, 1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_feof (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
ret = es_feof_unlocked (stream);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_ferror_unlocked (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
return es_get_indicator (stream, 1, 0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_ferror (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
ret = es_ferror_unlocked (stream);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
es_clearerr_unlocked (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
es_set_indicators (stream, 0, 0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
es_clearerr (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
es_clearerr_unlocked (stream);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_fflush (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (stream)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
2007-06-25 13:54:43 +02:00
|
|
|
|
if (stream->flags.writing)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
err = es_flush (stream);
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
es_empty (stream);
|
|
|
|
|
err = 0;
|
|
|
|
|
}
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
err = es_list_iterate (es_fflush);
|
|
|
|
|
|
|
|
|
|
return err ? EOF : 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_fseek (estream_t stream, long int offset, int whence)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
err = es_seek (stream, offset, whence, NULL);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_fseeko (estream_t stream, off_t offset, int whence)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
err = es_seek (stream, offset, whence, NULL);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
long int
|
|
|
|
|
es_ftell (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
long int ret;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
ret = es_offset_calculate (stream);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
off_t
|
|
|
|
|
es_ftello (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
off_t ret = -1;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
ret = es_offset_calculate (stream);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
es_rewind (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
es_seek (stream, 0L, SEEK_SET, NULL);
|
|
|
|
|
es_set_indicators (stream, 0, -1);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
_es_getc_underflow (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
unsigned char c;
|
|
|
|
|
size_t bytes_read;
|
|
|
|
|
|
|
|
|
|
err = es_readn (stream, &c, 1, &bytes_read);
|
|
|
|
|
|
|
|
|
|
return (err || (! bytes_read)) ? EOF : c;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
_es_putc_overflow (int c, estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
unsigned char d = c;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
err = es_writen (stream, &d, 1, NULL);
|
|
|
|
|
|
|
|
|
|
return err ? EOF : c;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_fgetc (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
ret = es_getc_unlocked (stream);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_fputc (int c, estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
ret = es_putc_unlocked (c, stream);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_ungetc (int c, estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
unsigned char data = (unsigned char) c;
|
|
|
|
|
size_t data_unread;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
es_unreadn (stream, &data, 1, &data_unread);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return data_unread ? c : EOF;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_read (estream_t ES__RESTRICT stream,
|
2005-06-16 10:12:03 +02:00
|
|
|
|
void *ES__RESTRICT buffer, size_t bytes_to_read,
|
2005-01-26 23:24:59 +01:00
|
|
|
|
size_t *ES__RESTRICT bytes_read)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (bytes_to_read)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
err = es_readn (stream, buffer, bytes_to_read, bytes_read);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_write (estream_t ES__RESTRICT stream,
|
2005-06-16 10:12:03 +02:00
|
|
|
|
const void *ES__RESTRICT buffer, size_t bytes_to_write,
|
2005-01-26 23:24:59 +01:00
|
|
|
|
size_t *ES__RESTRICT bytes_written)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (bytes_to_write)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
err = es_writen (stream, buffer, bytes_to_write, bytes_written);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
err = 0;
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
size_t
|
|
|
|
|
es_fread (void *ES__RESTRICT ptr, size_t size, size_t nitems,
|
|
|
|
|
estream_t ES__RESTRICT stream)
|
|
|
|
|
{
|
|
|
|
|
size_t ret, bytes;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (size * nitems)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
err = es_readn (stream, ptr, size * nitems, &bytes);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
ret = bytes / size;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
ret = 0;
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
size_t
|
|
|
|
|
es_fwrite (const void *ES__RESTRICT ptr, size_t size, size_t nitems,
|
|
|
|
|
estream_t ES__RESTRICT stream)
|
|
|
|
|
{
|
|
|
|
|
size_t ret, bytes;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (size * nitems)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
err = es_writen (stream, ptr, size * nitems, &bytes);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
ret = bytes / size;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
ret = 0;
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
char *
|
2007-08-22 12:55:07 +02:00
|
|
|
|
es_fgets (char *ES__RESTRICT buffer, int length, estream_t ES__RESTRICT stream)
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
unsigned char *s = (unsigned char*)buffer;
|
|
|
|
|
int c;
|
|
|
|
|
|
|
|
|
|
if (!length)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
c = EOF;
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
while (length > 1 && (c = es_getc_unlocked (stream)) != EOF && c != '\n')
|
2005-01-26 23:24:59 +01:00
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
*s++ = c;
|
|
|
|
|
length--;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
2007-08-22 12:55:07 +02:00
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
if (c == EOF && s == (unsigned char*)buffer)
|
|
|
|
|
return NULL; /* Nothing read. */
|
|
|
|
|
|
|
|
|
|
if (c != EOF && length > 1)
|
|
|
|
|
*s++ = c;
|
|
|
|
|
|
|
|
|
|
*s = 0;
|
|
|
|
|
return buffer;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_fputs (const char *ES__RESTRICT s, estream_t ES__RESTRICT stream)
|
|
|
|
|
{
|
|
|
|
|
size_t length;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
length = strlen (s);
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
err = es_writen (stream, s, length, NULL);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return err ? EOF : 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ssize_t
|
|
|
|
|
es_getline (char *ES__RESTRICT *ES__RESTRICT lineptr, size_t *ES__RESTRICT n,
|
|
|
|
|
estream_t ES__RESTRICT stream)
|
|
|
|
|
{
|
|
|
|
|
char *line = NULL;
|
|
|
|
|
size_t line_n = 0;
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
2006-08-14 16:40:07 +02:00
|
|
|
|
err = doreadline (stream, 0, &line, &line_n);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
if (*n)
|
|
|
|
|
{
|
|
|
|
|
/* Caller wants us to use his buffer. */
|
|
|
|
|
|
|
|
|
|
if (*n < (line_n + 1))
|
|
|
|
|
{
|
|
|
|
|
/* Provided buffer is too small -> resize. */
|
|
|
|
|
|
|
|
|
|
void *p;
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
p = mem_realloc (*lineptr, line_n + 1);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (! p)
|
|
|
|
|
err = -1;
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
if (*lineptr != p)
|
|
|
|
|
*lineptr = p;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (! err)
|
|
|
|
|
{
|
|
|
|
|
memcpy (*lineptr, line, line_n + 1);
|
|
|
|
|
if (*n != line_n)
|
|
|
|
|
*n = line_n;
|
|
|
|
|
}
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (line);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
/* Caller wants new buffers. */
|
|
|
|
|
*lineptr = line;
|
|
|
|
|
*n = line_n;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
return err ? err : line_n;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2006-08-14 16:40:07 +02:00
|
|
|
|
|
|
|
|
|
/* Same as fgets() but if the provided buffer is too short a larger
|
|
|
|
|
one will be allocated. This is similar to getline. A line is
|
|
|
|
|
considered a byte stream ending in a LF.
|
|
|
|
|
|
|
|
|
|
If MAX_LENGTH is not NULL, it shall point to a value with the
|
|
|
|
|
maximum allowed allocation.
|
|
|
|
|
|
|
|
|
|
Returns the length of the line. EOF is indicated by a line of
|
|
|
|
|
length zero. A truncated line is indicated my setting the value at
|
|
|
|
|
MAX_LENGTH to 0. If the returned value is less then 0 not enough
|
|
|
|
|
memory was enable or another error occurred; ERRNO is then set
|
|
|
|
|
accordingly.
|
|
|
|
|
|
|
|
|
|
If a line has been truncated, the file pointer is moved forward to
|
|
|
|
|
the end of the line so that the next read starts with the next
|
|
|
|
|
line. Note that MAX_LENGTH must be re-initialzied in this case.
|
|
|
|
|
|
|
|
|
|
The caller initially needs to provide the address of a variable,
|
|
|
|
|
initialized to NULL, at ADDR_OF_BUFFER and don't change this value
|
|
|
|
|
anymore with the following invocations. LENGTH_OF_BUFFER should be
|
|
|
|
|
the address of a variable, initialized to 0, which is also
|
|
|
|
|
maintained by this function. Thus, both paramaters should be
|
|
|
|
|
considered the state of this function.
|
|
|
|
|
|
|
|
|
|
Note: The returned buffer is allocated with enough extra space to
|
|
|
|
|
allow the caller to append a CR,LF,Nul. The buffer should be
|
|
|
|
|
released using es_free.
|
|
|
|
|
*/
|
|
|
|
|
ssize_t
|
|
|
|
|
es_read_line (estream_t stream,
|
|
|
|
|
char **addr_of_buffer, size_t *length_of_buffer,
|
|
|
|
|
size_t *max_length)
|
|
|
|
|
{
|
|
|
|
|
int c;
|
|
|
|
|
char *buffer = *addr_of_buffer;
|
|
|
|
|
size_t length = *length_of_buffer;
|
|
|
|
|
size_t nbytes = 0;
|
|
|
|
|
size_t maxlen = max_length? *max_length : 0;
|
|
|
|
|
char *p;
|
|
|
|
|
|
|
|
|
|
if (!buffer)
|
|
|
|
|
{
|
|
|
|
|
/* No buffer given - allocate a new one. */
|
|
|
|
|
length = 256;
|
2007-08-22 12:55:07 +02:00
|
|
|
|
buffer = mem_alloc (length);
|
2006-08-14 16:40:07 +02:00
|
|
|
|
*addr_of_buffer = buffer;
|
|
|
|
|
if (!buffer)
|
|
|
|
|
{
|
|
|
|
|
*length_of_buffer = 0;
|
|
|
|
|
if (max_length)
|
|
|
|
|
*max_length = 0;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
*length_of_buffer = length;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (length < 4)
|
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
/* This should never happen. If it does, the function has been
|
2006-08-14 16:40:07 +02:00
|
|
|
|
called with wrong arguments. */
|
|
|
|
|
errno = EINVAL;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
length -= 3; /* Reserve 3 bytes for CR,LF,EOL. */
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
p = buffer;
|
|
|
|
|
while ((c = es_getc_unlocked (stream)) != EOF)
|
|
|
|
|
{
|
|
|
|
|
if (nbytes == length)
|
|
|
|
|
{
|
|
|
|
|
/* Enlarge the buffer. */
|
|
|
|
|
if (maxlen && length > maxlen)
|
|
|
|
|
{
|
|
|
|
|
/* We are beyond our limit: Skip the rest of the line. */
|
|
|
|
|
while (c != '\n' && (c=es_getc_unlocked (stream)) != EOF)
|
|
|
|
|
;
|
|
|
|
|
*p++ = '\n'; /* Always append a LF (we reserved some space). */
|
|
|
|
|
nbytes++;
|
|
|
|
|
if (max_length)
|
|
|
|
|
*max_length = 0; /* Indicate truncation. */
|
|
|
|
|
break; /* the while loop. */
|
|
|
|
|
}
|
|
|
|
|
length += 3; /* Adjust for the reserved bytes. */
|
|
|
|
|
length += length < 1024? 256 : 1024;
|
2007-08-22 12:55:07 +02:00
|
|
|
|
*addr_of_buffer = mem_realloc (buffer, length);
|
2006-08-14 16:40:07 +02:00
|
|
|
|
if (!*addr_of_buffer)
|
|
|
|
|
{
|
|
|
|
|
int save_errno = errno;
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (buffer);
|
2006-08-14 16:40:07 +02:00
|
|
|
|
*length_of_buffer = *max_length = 0;
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
errno = save_errno;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
buffer = *addr_of_buffer;
|
|
|
|
|
*length_of_buffer = length;
|
|
|
|
|
length -= 3;
|
|
|
|
|
p = buffer + nbytes;
|
|
|
|
|
}
|
|
|
|
|
*p++ = c;
|
|
|
|
|
nbytes++;
|
|
|
|
|
if (c == '\n')
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
*p = 0; /* Make sure the line is a string. */
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return nbytes;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Wrapper around free() to match the memory allocation system used
|
|
|
|
|
by estream. Should be used for all buffers returned to the caller
|
|
|
|
|
by libestream. */
|
|
|
|
|
void
|
|
|
|
|
es_free (void *a)
|
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
mem_free (a);
|
2006-08-14 16:40:07 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
int
|
|
|
|
|
es_vfprintf (estream_t ES__RESTRICT stream, const char *ES__RESTRICT format,
|
|
|
|
|
va_list ap)
|
|
|
|
|
{
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
ret = es_print (stream, format, ap);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2007-03-19 15:35:04 +01:00
|
|
|
|
static int
|
|
|
|
|
es_fprintf_unlocked (estream_t ES__RESTRICT stream,
|
2007-05-15 18:10:48 +02:00
|
|
|
|
const char *ES__RESTRICT format, ...)
|
2007-03-19 15:35:04 +01:00
|
|
|
|
{
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
va_list ap;
|
|
|
|
|
va_start (ap, format);
|
|
|
|
|
ret = es_print (stream, format, ap);
|
|
|
|
|
va_end (ap);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
int
|
|
|
|
|
es_fprintf (estream_t ES__RESTRICT stream,
|
|
|
|
|
const char *ES__RESTRICT format, ...)
|
|
|
|
|
{
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
va_list ap;
|
|
|
|
|
va_start (ap, format);
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
ret = es_print (stream, format, ap);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
va_end (ap);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
2007-08-22 12:55:07 +02:00
|
|
|
|
|
2005-01-26 23:24:59 +01:00
|
|
|
|
static int
|
|
|
|
|
tmpfd (void)
|
|
|
|
|
{
|
2007-08-22 12:55:07 +02:00
|
|
|
|
#ifdef HAVE_W32_SYSTEM
|
|
|
|
|
int attempts, n;
|
|
|
|
|
char buffer[MAX_PATH+9+12+1];
|
|
|
|
|
char *name, *p;
|
|
|
|
|
HANDLE file;
|
|
|
|
|
int pid = GetCurrentProcessId ();
|
|
|
|
|
unsigned int value;
|
|
|
|
|
int i;
|
|
|
|
|
|
|
|
|
|
n = GetTempPath (MAX_PATH+1, buffer);
|
|
|
|
|
if (!n || n > MAX_PATH || strlen (buffer) > MAX_PATH)
|
|
|
|
|
{
|
|
|
|
|
errno = ENOENT;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
p = buffer + strlen (buffer);
|
|
|
|
|
strcpy (p, "_estream");
|
|
|
|
|
p += 8;
|
|
|
|
|
/* We try to create the directory but don't care about an error as
|
|
|
|
|
it may already exist and the CreateFile would throw an error
|
|
|
|
|
anyway. */
|
|
|
|
|
CreateDirectory (buffer, NULL);
|
|
|
|
|
*p++ = '\\';
|
|
|
|
|
name = p;
|
|
|
|
|
for (attempts=0; attempts < 10; attempts++)
|
|
|
|
|
{
|
|
|
|
|
p = name;
|
|
|
|
|
value = (GetTickCount () ^ ((pid<<16) & 0xffff0000));
|
|
|
|
|
for (i=0; i < 8; i++)
|
|
|
|
|
{
|
|
|
|
|
*p++ = tohex (((value >> 28) & 0x0f));
|
|
|
|
|
value <<= 4;
|
|
|
|
|
}
|
|
|
|
|
strcpy (p, ".tmp");
|
|
|
|
|
file = CreateFile (buffer,
|
|
|
|
|
GENERIC_READ | GENERIC_WRITE,
|
|
|
|
|
0,
|
|
|
|
|
NULL,
|
|
|
|
|
CREATE_NEW,
|
|
|
|
|
FILE_ATTRIBUTE_TEMPORARY | FILE_FLAG_DELETE_ON_CLOSE,
|
|
|
|
|
NULL);
|
|
|
|
|
if (file != INVALID_HANDLE_VALUE)
|
|
|
|
|
{
|
|
|
|
|
int fd = _open_osfhandle ((long)file, 0);
|
|
|
|
|
if (fd == -1)
|
|
|
|
|
{
|
|
|
|
|
CloseHandle (file);
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
return fd;
|
|
|
|
|
}
|
|
|
|
|
Sleep (1); /* One ms as this is the granularity of GetTickCount. */
|
|
|
|
|
}
|
|
|
|
|
errno = ENOENT;
|
|
|
|
|
return -1;
|
|
|
|
|
#else /*!HAVE_W32_SYSTEM*/
|
2005-01-26 23:24:59 +01:00
|
|
|
|
FILE *fp;
|
|
|
|
|
int fp_fd;
|
|
|
|
|
int fd;
|
|
|
|
|
|
|
|
|
|
fp = NULL;
|
|
|
|
|
fd = -1;
|
|
|
|
|
|
|
|
|
|
fp = tmpfile ();
|
|
|
|
|
if (! fp)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
fp_fd = fileno (fp);
|
|
|
|
|
fd = dup (fp_fd);
|
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (fp)
|
|
|
|
|
fclose (fp);
|
|
|
|
|
|
|
|
|
|
return fd;
|
2007-08-22 12:55:07 +02:00
|
|
|
|
#endif /*!HAVE_W32_SYSTEM*/
|
2005-01-26 23:24:59 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
estream_t
|
|
|
|
|
es_tmpfile (void)
|
|
|
|
|
{
|
2007-06-25 13:54:43 +02:00
|
|
|
|
unsigned int modeflags;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
int create_called;
|
|
|
|
|
estream_t stream;
|
|
|
|
|
void *cookie;
|
|
|
|
|
int err;
|
|
|
|
|
int fd;
|
|
|
|
|
|
|
|
|
|
create_called = 0;
|
|
|
|
|
stream = NULL;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
modeflags = O_RDWR | O_TRUNC | O_CREAT;
|
2005-01-26 23:24:59 +01:00
|
|
|
|
cookie = NULL;
|
|
|
|
|
|
|
|
|
|
fd = tmpfd ();
|
|
|
|
|
if (fd == -1)
|
|
|
|
|
{
|
|
|
|
|
err = -1;
|
|
|
|
|
goto out;
|
|
|
|
|
}
|
|
|
|
|
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_func_fd_create (&cookie, fd, modeflags, 0);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
if (err)
|
|
|
|
|
goto out;
|
|
|
|
|
|
|
|
|
|
create_called = 1;
|
2007-06-25 13:54:43 +02:00
|
|
|
|
err = es_create (&stream, cookie, fd, estream_functions_fd, modeflags);
|
2005-01-26 23:24:59 +01:00
|
|
|
|
|
|
|
|
|
out:
|
|
|
|
|
|
|
|
|
|
if (err)
|
|
|
|
|
{
|
|
|
|
|
if (create_called)
|
|
|
|
|
es_func_fd_destroy (cookie);
|
|
|
|
|
else if (fd != -1)
|
|
|
|
|
close (fd);
|
|
|
|
|
stream = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return stream;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
es_setvbuf (estream_t ES__RESTRICT stream,
|
|
|
|
|
char *ES__RESTRICT buf, int type, size_t size)
|
|
|
|
|
{
|
|
|
|
|
int err;
|
|
|
|
|
|
|
|
|
|
if (((type == _IOFBF) || (type == _IOLBF) || (type == _IONBF))
|
|
|
|
|
&& (! ((! size) && (type != _IONBF))))
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
err = es_set_buffering (stream, buf, type, size);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
errno = EINVAL;
|
|
|
|
|
err = -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
es_setbuf (estream_t ES__RESTRICT stream, char *ES__RESTRICT buf)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
es_set_buffering (stream, buf, buf ? _IOFBF : _IONBF, BUFSIZ);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
es_opaque_set (estream_t stream, void *opaque)
|
|
|
|
|
{
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
es_opaque_ctrl (stream, opaque, NULL);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void *
|
|
|
|
|
es_opaque_get (estream_t stream)
|
|
|
|
|
{
|
|
|
|
|
void *opaque;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
es_opaque_ctrl (stream, NULL, &opaque);
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return opaque;
|
|
|
|
|
}
|
2006-08-14 16:40:07 +02:00
|
|
|
|
|
2007-03-19 15:35:04 +01:00
|
|
|
|
/* Print a BUFFER to STREAM while replacing all control characters and
|
|
|
|
|
the characters in DELIMITERS by standard C escape sequences.
|
|
|
|
|
Returns 0 on success or -1 on error. If BYTES_WRITTEN is not NULL
|
|
|
|
|
the number of bytes actually written are stored at this
|
|
|
|
|
address. */
|
|
|
|
|
int
|
|
|
|
|
es_write_sanitized (estream_t ES__RESTRICT stream,
|
|
|
|
|
const void * ES__RESTRICT buffer, size_t length,
|
|
|
|
|
const char * delimiters,
|
|
|
|
|
size_t * ES__RESTRICT bytes_written)
|
|
|
|
|
{
|
|
|
|
|
const unsigned char *p = buffer;
|
|
|
|
|
size_t count = 0;
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
for (; length; length--, p++, count++)
|
|
|
|
|
{
|
|
|
|
|
if (*p < 0x20
|
2008-06-26 21:09:07 +02:00
|
|
|
|
|| *p == 0x7f
|
2007-03-19 15:35:04 +01:00
|
|
|
|
|| (delimiters
|
|
|
|
|
&& (strchr (delimiters, *p) || *p == '\\')))
|
|
|
|
|
{
|
|
|
|
|
es_putc_unlocked ('\\', stream);
|
|
|
|
|
count++;
|
|
|
|
|
if (*p == '\n')
|
|
|
|
|
{
|
|
|
|
|
es_putc_unlocked ('n', stream);
|
|
|
|
|
count++;
|
|
|
|
|
}
|
|
|
|
|
else if (*p == '\r')
|
|
|
|
|
{
|
|
|
|
|
es_putc_unlocked ('r', stream);
|
|
|
|
|
count++;
|
|
|
|
|
}
|
|
|
|
|
else if (*p == '\f')
|
|
|
|
|
{
|
|
|
|
|
es_putc_unlocked ('f', stream);
|
|
|
|
|
count++;
|
|
|
|
|
}
|
|
|
|
|
else if (*p == '\v')
|
|
|
|
|
{
|
|
|
|
|
es_putc_unlocked ('v', stream);
|
|
|
|
|
count++;
|
|
|
|
|
}
|
|
|
|
|
else if (*p == '\b')
|
|
|
|
|
{
|
|
|
|
|
es_putc_unlocked ('b', stream);
|
|
|
|
|
count++;
|
|
|
|
|
}
|
|
|
|
|
else if (!*p)
|
|
|
|
|
{
|
|
|
|
|
es_putc_unlocked('0', stream);
|
|
|
|
|
count++;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
es_fprintf_unlocked (stream, "x%02x", *p);
|
|
|
|
|
count += 3;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
es_putc_unlocked (*p, stream);
|
|
|
|
|
count++;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (bytes_written)
|
|
|
|
|
*bytes_written = count;
|
|
|
|
|
ret = es_ferror_unlocked (stream)? -1 : 0;
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Write LENGTH bytes of BUFFER to STREAM as a hex encoded string.
|
|
|
|
|
RESERVED must be 0. Returns 0 on success or -1 on error. If
|
|
|
|
|
BYTES_WRITTEN is not NULL the number of bytes actually written are
|
|
|
|
|
stored at this address. */
|
|
|
|
|
int
|
|
|
|
|
es_write_hexstring (estream_t ES__RESTRICT stream,
|
|
|
|
|
const void *ES__RESTRICT buffer, size_t length,
|
|
|
|
|
int reserved, size_t *ES__RESTRICT bytes_written )
|
|
|
|
|
{
|
|
|
|
|
int ret;
|
|
|
|
|
const unsigned char *s;
|
|
|
|
|
size_t count = 0;
|
|
|
|
|
|
2008-10-20 15:53:23 +02:00
|
|
|
|
(void)reserved;
|
|
|
|
|
|
2007-03-19 15:35:04 +01:00
|
|
|
|
#define tohex(n) ((n) < 10 ? ((n) + '0') : (((n) - 10) + 'A'))
|
|
|
|
|
|
|
|
|
|
if (!length)
|
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
|
|
ESTREAM_LOCK (stream);
|
|
|
|
|
|
|
|
|
|
for (s = buffer; length; s++, length--)
|
|
|
|
|
{
|
|
|
|
|
es_putc_unlocked ( tohex ((*s>>4)&15), stream);
|
|
|
|
|
es_putc_unlocked ( tohex (*s&15), stream);
|
|
|
|
|
count += 2;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (bytes_written)
|
|
|
|
|
*bytes_written = count;
|
|
|
|
|
ret = es_ferror_unlocked (stream)? -1 : 0;
|
|
|
|
|
|
|
|
|
|
ESTREAM_UNLOCK (stream);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
|
|
|
|
|
#undef tohex
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#ifdef GNUPG_MAJOR_VERSION
|
|
|
|
|
/* Special estream function to print an UTF8 string in the native
|
|
|
|
|
encoding. The interface is the same as es_write_sanitized, however
|
|
|
|
|
only one delimiter may be supported.
|
|
|
|
|
|
2007-05-15 18:10:48 +02:00
|
|
|
|
THIS IS NOT A STANDARD ESTREAM FUNCTION AND ONLY USED BY GNUPG!. */
|
2007-03-19 15:35:04 +01:00
|
|
|
|
int
|
|
|
|
|
es_write_sanitized_utf8_buffer (estream_t stream,
|
|
|
|
|
const void *buffer, size_t length,
|
|
|
|
|
const char *delimiters, size_t *bytes_written)
|
|
|
|
|
{
|
|
|
|
|
const char *p = buffer;
|
|
|
|
|
size_t i;
|
|
|
|
|
|
|
|
|
|
/* We can handle plain ascii simpler, so check for it first. */
|
|
|
|
|
for (i=0; i < length; i++ )
|
|
|
|
|
{
|
|
|
|
|
if ( (p[i] & 0x80) )
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
if (i < length)
|
|
|
|
|
{
|
|
|
|
|
int delim = delimiters? *delimiters : 0;
|
|
|
|
|
char *buf;
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
|
|
/*(utf8 conversion already does the control character quoting). */
|
|
|
|
|
buf = utf8_to_native (p, length, delim);
|
|
|
|
|
if (bytes_written)
|
|
|
|
|
*bytes_written = strlen (buf);
|
|
|
|
|
ret = es_fputs (buf, stream);
|
|
|
|
|
xfree (buf);
|
|
|
|
|
return i;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
return es_write_sanitized (stream, p, length, delimiters, bytes_written);
|
|
|
|
|
}
|
|
|
|
|
#endif /*GNUPG_MAJOR_VERSION*/
|