X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Futil%2Fbio.c;h=62f4904f7d19367ab973161cbbaaf6e56551f045;hb=82e765caeb53a1de54676738169dab98ca76c47e;hp=b29d75eb9e0988b5a51491a617c6d5e03bddf16a;hpb=502af2167f7c218366666ca4944bd7cc54b5b19a;p=oweals%2Fgnunet.git diff --git a/src/util/bio.c b/src/util/bio.c index b29d75eb9..62f4904f7 100644 --- a/src/util/bio.c +++ b/src/util/bio.c @@ -1,10 +1,10 @@ /* This file is part of GNUnet. - (C) 2006, 2009 Christian Grothoff (and other contributing authors) + Copyright (C) 2006, 2009, 2013 GNUnet e.V. GNUnet is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published - by the Free Software Foundation; either version 2, or (at your + by the Free Software Foundation; either version 3, or (at your option) any later version. GNUnet is distributed in the hope that it will be useful, but @@ -14,8 +14,8 @@ You should have received a copy of the GNU General Public License along with GNUnet; see the file COPYING. If not, write to the - Free Software Foundation, Inc., 59 Temple Place - Suite 330, - Boston, MA 02111-1307, USA. + Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, + Boston, MA 02110-1301, USA. */ /** * @file util/bio.c @@ -23,23 +23,63 @@ * @author Christian Grothoff */ #include "platform.h" -#include "gnunet_bio_lib.h" -#include "gnunet_disk_lib.h" +#include "gnunet_util_lib.h" +#define LOG(kind,...) GNUNET_log_from (kind, "util",__VA_ARGS__) + +#ifndef PATH_MAX +/** + * Assumed maximum path length (for source file names). + */ +#define PATH_MAX 4096 +#endif + + +/** + * Size for I/O buffers. + */ #define BIO_BUFFER_SIZE 65536 +/** + * Maximum size allowed for meta data written/read from disk. + * File-sharing limits to 64k, so this should be rather generous. + */ #define MAX_META_DATA (1024 * 1024) + /** * Handle for buffered reading. */ struct GNUNET_BIO_ReadHandle { + /** + * Underlying file abstraction. + */ struct GNUNET_DISK_FileHandle *fd; + + /** + * Error message, NULL if there were no errors. + */ char *emsg; + + /** + * I/O buffer. Allocated at the end of the struct, do not free! + */ char *buffer; + + /** + * Number of bytes available in read @e buffer. + */ size_t have; + + /** + * Total size of @e buffer. + */ size_t size; + + /** + * Current read offset in @e buffer. + */ off_t pos; }; @@ -73,10 +113,11 @@ GNUNET_BIO_read_open (const char *fn) * * @param h file handle * @param emsg set to the error message - * @return GNUNET_OK on success, GNUNET_SYSERR otherwise + * @return #GNUNET_OK on success, #GNUNET_SYSERR otherwise */ int -GNUNET_BIO_read_close (struct GNUNET_BIO_ReadHandle *h, char **emsg) +GNUNET_BIO_read_close (struct GNUNET_BIO_ReadHandle *h, + char **emsg) { int err; @@ -98,18 +139,19 @@ GNUNET_BIO_read_close (struct GNUNET_BIO_ReadHandle *h, char **emsg) * @param what describes what is being read (for error message creation) * @param result the buffer to write the result to * @param len the number of bytes to read - * @return GNUNET_OK on success, GNUNET_SYSERR on failure + * @return #GNUNET_OK on success, #GNUNET_SYSERR on failure */ int GNUNET_BIO_read (struct GNUNET_BIO_ReadHandle *h, - const char *what, void *result, size_t len) + const char *what, + void *result, size_t len) { char *dst = result; size_t min; size_t pos; ssize_t ret; - if (h->emsg != NULL) + if (NULL != h->emsg) return GNUNET_SYSERR; pos = 0; do @@ -120,7 +162,7 @@ GNUNET_BIO_read (struct GNUNET_BIO_ReadHandle *h, { if (min > len - pos) min = len - pos; - memcpy (&dst[pos], &h->buffer[h->pos], min); + GNUNET_memcpy (&dst[pos], &h->buffer[h->pos], min); h->pos += min; pos += min; } @@ -129,16 +171,20 @@ GNUNET_BIO_read (struct GNUNET_BIO_ReadHandle *h, GNUNET_assert (h->have == h->pos); /* fill buffer */ ret = GNUNET_DISK_file_read (h->fd, h->buffer, h->size); - if (ret == -1) + if (-1 == ret) { GNUNET_asprintf (&h->emsg, - _("Error reading `%s': %s"), what, STRERROR (errno)); + _("Error reading `%s': %s"), + what, + STRERROR (errno)); return GNUNET_SYSERR; } - if (ret == 0) + if (0 == ret) { GNUNET_asprintf (&h->emsg, - _("Error reading `%s': %s"), what, _("End of file")); + _("Error reading `%s': %s"), + what, + _("End of file")); return GNUNET_SYSERR; } h->pos = 0; @@ -157,13 +203,16 @@ GNUNET_BIO_read (struct GNUNET_BIO_ReadHandle *h, * @param line line number in the source file * @param result the buffer to write the result to * @param len the number of bytes to read - * @return GNUNET_OK on success, GNUNET_SYSERR on failure + * @return #GNUNET_OK on success, #GNUNET_SYSERR on failure */ int GNUNET_BIO_read_fn (struct GNUNET_BIO_ReadHandle *h, - const char *file, int line, void *result, size_t len) + const char *file, + int line, + void *result, + size_t len) { - char what[1024]; + char what[PATH_MAX + 1024]; GNUNET_snprintf (what, sizeof (what), "%s:%d", file, line); return GNUNET_BIO_read (h, what, result, len); @@ -177,12 +226,14 @@ GNUNET_BIO_read_fn (struct GNUNET_BIO_ReadHandle *h, * @param what describes what is being read (for error message creation) * @param result the buffer to store a pointer to the (allocated) string to * (note that *result could be set to NULL as well) - * @param maxLen maximum allowed length for the string - * @return GNUNET_OK on success, GNUNET_SYSERR on failure + * @param max_length maximum allowed length for the string + * @return #GNUNET_OK on success, #GNUNET_SYSERR on failure */ int GNUNET_BIO_read_string (struct GNUNET_BIO_ReadHandle *h, - const char *what, char **result, size_t maxLen) + const char *what, + char **result, + size_t max_length) { char *buf; uint32_t big; @@ -193,22 +244,21 @@ GNUNET_BIO_read_string (struct GNUNET_BIO_ReadHandle *h, GNUNET_asprintf (&h->emsg, _("Error reading length of string `%s'"), what); return GNUNET_SYSERR; } - if (big == 0) + if (0 == big) { *result = NULL; return GNUNET_OK; } - if (big > maxLen) + if (big > max_length) { - GNUNET_asprintf (&h->emsg, - _("String `%s' longer than allowed (%u > %u)"), - what, big, maxLen); + GNUNET_asprintf (&h->emsg, _("String `%s' longer than allowed (%u > %u)"), + what, big, max_length); return GNUNET_SYSERR; } buf = GNUNET_malloc (big); *result = buf; buf[--big] = '\0'; - if (big == 0) + if (0 == big) return GNUNET_OK; if (GNUNET_OK != GNUNET_BIO_read (h, what, buf, big)) { @@ -226,7 +276,7 @@ GNUNET_BIO_read_string (struct GNUNET_BIO_ReadHandle *h, * @param h handle to an open file * @param what describes what is being read (for error message creation) * @param result the buffer to store a pointer to the (allocated) metadata - * @return GNUNET_OK on success, GNUNET_SYSERR on failure + * @return #GNUNET_OK on success, #GNUNET_SYSERR on failure */ int GNUNET_BIO_read_meta_data (struct GNUNET_BIO_ReadHandle *h, @@ -277,11 +327,11 @@ GNUNET_BIO_read_meta_data (struct GNUNET_BIO_ReadHandle *h, * @param file name of the source file * @param line line number in the source file * @param i address of 32-bit integer to read - * @return GNUNET_OK on success, GNUNET_SYSERR on error + * @return #GNUNET_OK on success, #GNUNET_SYSERR on error */ int -GNUNET_BIO_read_int32__ (struct GNUNET_BIO_ReadHandle *h, - const char *file, int line, int32_t * i) +GNUNET_BIO_read_int32__ (struct GNUNET_BIO_ReadHandle *h, const char *file, + int line, int32_t * i) { int32_t big; @@ -299,11 +349,13 @@ GNUNET_BIO_read_int32__ (struct GNUNET_BIO_ReadHandle *h, * @param file name of the source file * @param line line number in the source file * @param i address of 64-bit integer to read - * @return GNUNET_OK on success, GNUNET_SYSERR on error + * @return #GNUNET_OK on success, #GNUNET_SYSERR on error */ int GNUNET_BIO_read_int64__ (struct GNUNET_BIO_ReadHandle *h, - const char *file, int line, int64_t * i) + const char *file, + int line, + int64_t *i) { int64_t big; @@ -319,9 +371,24 @@ GNUNET_BIO_read_int64__ (struct GNUNET_BIO_ReadHandle *h, */ struct GNUNET_BIO_WriteHandle { + /** + * Underlying file handle. + */ struct GNUNET_DISK_FileHandle *fd; + + /** + * I/O buffer. Do not free, allocated at the end of the struct. + */ char *buffer; + + /** + * Number of bytes already in @e buffer. + */ size_t have; + + /** + * Total size of @e buffer. + */ size_t size; }; @@ -339,9 +406,8 @@ GNUNET_BIO_write_open (const char *fn) struct GNUNET_BIO_WriteHandle *h; fd = GNUNET_DISK_file_open (fn, - GNUNET_DISK_OPEN_WRITE | - GNUNET_DISK_OPEN_TRUNCATE | - GNUNET_DISK_OPEN_CREATE, + GNUNET_DISK_OPEN_WRITE | GNUNET_DISK_OPEN_TRUNCATE + | GNUNET_DISK_OPEN_CREATE, GNUNET_DISK_PERM_USER_READ | GNUNET_DISK_PERM_USER_WRITE); if (NULL == fd) @@ -350,7 +416,6 @@ GNUNET_BIO_write_open (const char *fn) h->buffer = (char *) &h[1]; h->size = BIO_BUFFER_SIZE; h->fd = fd; - return h; } @@ -359,48 +424,60 @@ GNUNET_BIO_write_open (const char *fn) * Close an open file for writing. * * @param h file handle - * @return GNUNET_OK on success, GNUNET_SYSERR otherwise + * @return #GNUNET_OK on success, #GNUNET_SYSERR otherwise */ int GNUNET_BIO_write_close (struct GNUNET_BIO_WriteHandle *h) { - ssize_t wrt; int ret; - if (NULL == h->fd) - { - ret = GNUNET_SYSERR; - } - else - { - wrt = GNUNET_DISK_file_write (h->fd, h->buffer, h->have); - if (wrt == h->have) - ret = GNUNET_OK; - else - ret = GNUNET_SYSERR; + ret = GNUNET_SYSERR; + if ( (NULL != h->fd) && (GNUNET_OK == (ret = GNUNET_BIO_flush (h))) ) GNUNET_DISK_file_close (h->fd); - } GNUNET_free (h); return ret; } +/** + * Force a buffered writer to flush its buffer + * + * @param h the writer handle + * @return #GNUNET_OK upon success. Upon failure #GNUNET_SYSERR is returned and + * the file is closed + */ +int +GNUNET_BIO_flush (struct GNUNET_BIO_WriteHandle *h) +{ + ssize_t ret; + + ret = GNUNET_DISK_file_write (h->fd, h->buffer, h->have); + if (ret != h->have) + { + GNUNET_DISK_file_close (h->fd); + h->fd = NULL; + return GNUNET_SYSERR; /* error */ + } + h->have = 0; + return GNUNET_OK; +} + + /** * Write a buffer to a file. * * @param h handle to open file * @param buffer the data to write * @param n number of bytes to write - * @return GNUNET_OK on success, GNUNET_SYSERR on error + * @return #GNUNET_OK on success, #GNUNET_SYSERR on error */ int -GNUNET_BIO_write (struct GNUNET_BIO_WriteHandle *h, - const void *buffer, size_t n) +GNUNET_BIO_write (struct GNUNET_BIO_WriteHandle *h, const void *buffer, + size_t n) { const char *src = buffer; size_t min; size_t pos; - ssize_t ret; if (NULL == h->fd) return GNUNET_SYSERR; @@ -411,20 +488,14 @@ GNUNET_BIO_write (struct GNUNET_BIO_WriteHandle *h, min = h->size - h->have; if (min > n - pos) min = n - pos; - memcpy (&h->buffer[h->have], &src[pos], min); + GNUNET_memcpy (&h->buffer[h->have], &src[pos], min); pos += min; h->have += min; if (pos == n) return GNUNET_OK; /* done */ GNUNET_assert (h->have == h->size); - ret = GNUNET_DISK_file_write (h->fd, h->buffer, h->size); - if (ret != h->size) - { - GNUNET_DISK_file_close (h->fd); - h->fd = NULL; + if (GNUNET_OK != GNUNET_BIO_flush (h)) return GNUNET_SYSERR; /* error */ - } - h->have = 0; } while (pos < n); /* should always be true */ GNUNET_break (0); @@ -437,10 +508,11 @@ GNUNET_BIO_write (struct GNUNET_BIO_WriteHandle *h, * * @param h handle to open file * @param s string to write (can be NULL) - * @return GNUNET_OK on success, GNUNET_SYSERR on error + * @return #GNUNET_OK on success, #GNUNET_SYSERR on error */ int -GNUNET_BIO_write_string (struct GNUNET_BIO_WriteHandle *h, const char *s) +GNUNET_BIO_write_string (struct GNUNET_BIO_WriteHandle *h, + const char *s) { uint32_t slen; @@ -458,7 +530,7 @@ GNUNET_BIO_write_string (struct GNUNET_BIO_WriteHandle *h, const char *s) * * @param h handle to open file * @param m metadata to write - * @return GNUNET_OK on success, GNUNET_SYSERR on error + * @return #GNUNET_OK on success, #GNUNET_SYSERR on error */ int GNUNET_BIO_write_meta_data (struct GNUNET_BIO_WriteHandle *h, @@ -470,10 +542,9 @@ GNUNET_BIO_write_meta_data (struct GNUNET_BIO_WriteHandle *h, if (m == NULL) return GNUNET_BIO_write_int32 (h, 0); buf = NULL; - size = GNUNET_CONTAINER_meta_data_serialize (m, - &buf, - MAX_META_DATA, - GNUNET_CONTAINER_META_DATA_SERIALIZE_PART); + size = + GNUNET_CONTAINER_meta_data_serialize (m, &buf, MAX_META_DATA, + GNUNET_CONTAINER_META_DATA_SERIALIZE_PART); if (size == -1) { GNUNET_free (buf); @@ -494,11 +565,12 @@ GNUNET_BIO_write_meta_data (struct GNUNET_BIO_WriteHandle *h, * Write an (u)int32_t. * * @param h hande to open file - * @param i address of 32-bit integer to write - * @return GNUNET_OK on success, GNUNET_SYSERR on error + * @param i 32-bit integer to write + * @return #GNUNET_OK on success, #GNUNET_SYSERR on error */ int -GNUNET_BIO_write_int32 (struct GNUNET_BIO_WriteHandle *h, int32_t i) +GNUNET_BIO_write_int32 (struct GNUNET_BIO_WriteHandle *h, + int32_t i) { int32_t big; @@ -511,11 +583,12 @@ GNUNET_BIO_write_int32 (struct GNUNET_BIO_WriteHandle *h, int32_t i) * Write an (u)int64_t. * * @param h hande to open file - * @param i address of 64-bit integer to write - * @return GNUNET_OK on success, GNUNET_SYSERR on error + * @param i 64-bit integer to write + * @return #GNUNET_OK on success, #GNUNET_SYSERR on error */ int -GNUNET_BIO_write_int64 (struct GNUNET_BIO_WriteHandle *h, int64_t i) +GNUNET_BIO_write_int64 (struct GNUNET_BIO_WriteHandle *h, + int64_t i) { int64_t big;