| /* |
| * buf.c: memory buffers for libxml2 |
| * |
| * new buffer structures and entry points to simplify the maintenance |
| * of libxml2 and ensure we keep good control over memory allocations |
| * and stay 64 bits clean. |
| * The new entry point use the xmlBuf opaque structure and |
| * xmlBuf...() counterparts to the old xmlBuf...() functions |
| * |
| * See Copyright for the status of this software. |
| * |
| * Author: Daniel Veillard |
| */ |
| |
| #define IN_LIBXML |
| #include "libxml.h" |
| |
| #include <string.h> |
| #include <limits.h> |
| |
| #include <libxml/parser.h> |
| #include <libxml/tree.h> |
| |
| #include "private/buf.h" |
| |
| #ifndef SIZE_MAX |
| #define SIZE_MAX ((size_t) -1) |
| #endif |
| |
| #define BUF_FLAG_OOM (1u << 0) |
| #define BUF_FLAG_OVERFLOW (1u << 1) |
| #define BUF_FLAG_STATIC (1u << 2) |
| |
| #define BUF_ERROR(buf) ((buf)->flags & (BUF_FLAG_OOM | BUF_FLAG_OVERFLOW)) |
| #define BUF_STATIC(buf) ((buf)->flags & BUF_FLAG_STATIC) |
| |
| /** |
| * A buffer structure. The base of the structure is somehow compatible |
| * with struct _xmlBuffer to limit risks on application which accessed |
| * directly the input->buf->buffer structures. |
| */ |
| |
| struct _xmlBuf { |
| xmlChar *content; /* The buffer content UTF8 */ |
| xmlChar *mem; /* Start of the allocation */ |
| size_t use; /* The buffer size used */ |
| size_t size; /* The buffer size, excluding terminating 0 */ |
| size_t maxSize; /* The maximum buffer size */ |
| unsigned flags; /* flags */ |
| }; |
| |
| /** |
| * Handle an out of memory condition |
| * To be improved... |
| * |
| * @param buf the buffer |
| */ |
| static void |
| xmlBufMemoryError(xmlBufPtr buf) |
| { |
| if (!BUF_ERROR(buf)) |
| buf->flags |= BUF_FLAG_OOM; |
| } |
| |
| /** |
| * Handle a buffer overflow error |
| * To be improved... |
| * |
| * @param buf the buffer |
| */ |
| static void |
| xmlBufOverflowError(xmlBufPtr buf) |
| { |
| if (!BUF_ERROR(buf)) |
| buf->flags |= BUF_FLAG_OVERFLOW; |
| } |
| |
| /** |
| * Create a buffer. |
| * |
| * @param size initial buffer size |
| * @returns the new structure |
| */ |
| xmlBuf * |
| xmlBufCreate(size_t size) { |
| xmlBufPtr ret; |
| |
| if (size == SIZE_MAX) |
| return(NULL); |
| |
| ret = xmlMalloc(sizeof(*ret)); |
| if (ret == NULL) |
| return(NULL); |
| |
| ret->use = 0; |
| ret->flags = 0; |
| ret->size = size; |
| ret->maxSize = SIZE_MAX - 1; |
| |
| ret->mem = xmlMalloc(ret->size + 1); |
| if (ret->mem == NULL) { |
| xmlFree(ret); |
| return(NULL); |
| } |
| ret->content = ret->mem; |
| ret->content[0] = 0; |
| |
| return(ret); |
| } |
| |
| /** |
| * Create a buffer initialized with memory. |
| * |
| * If `isStatic` is set, uses the memory area directly as backing store. |
| * The memory must be zero-terminated and not be modified for the |
| * lifetime of the buffer. A static buffer can't be grown, modified or |
| * detached, but it can be shrunk. |
| * |
| * @param mem a memory area |
| * @param size size of the buffer excluding terminator |
| * @param isStatic whether the memory area is static |
| * @returns a new buffer. |
| */ |
| xmlBuf * |
| xmlBufCreateMem(const xmlChar *mem, size_t size, int isStatic) { |
| xmlBufPtr ret; |
| |
| if (mem == NULL) |
| return(NULL); |
| |
| ret = xmlMalloc(sizeof(*ret)); |
| if (ret == NULL) |
| return(NULL); |
| |
| if (isStatic) { |
| /* Check that memory is zero-terminated */ |
| if (mem[size] != 0) { |
| xmlFree(ret); |
| return(NULL); |
| } |
| ret->flags = BUF_FLAG_STATIC; |
| ret->mem = (xmlChar *) mem; |
| } else { |
| ret->flags = 0; |
| ret->mem = xmlMalloc(size + 1); |
| if (ret->mem == NULL) { |
| xmlFree(ret); |
| return(NULL); |
| } |
| memcpy(ret->mem, mem, size); |
| ret->mem[size] = 0; |
| } |
| |
| ret->use = size; |
| ret->size = size; |
| ret->maxSize = SIZE_MAX - 1; |
| ret->content = ret->mem; |
| |
| return(ret); |
| } |
| |
| /** |
| * Extract buffer content. |
| * |
| * Return the content of the buffer as an `xmlChar` string, |
| * clearing the buffer. |
| * |
| * This doesn't work with static buffers as they can't be reset. |
| * |
| * @param buf the buffer |
| * @returns the buffer content |
| */ |
| xmlChar * |
| xmlBufDetach(xmlBuf *buf) { |
| xmlChar *ret; |
| |
| if ((buf == NULL) || (BUF_ERROR(buf)) || (BUF_STATIC(buf))) |
| return(NULL); |
| |
| if (buf->content != buf->mem) { |
| ret = xmlStrndup(buf->content, buf->use); |
| xmlFree(buf->mem); |
| } else { |
| ret = buf->mem; |
| } |
| |
| buf->content = NULL; |
| buf->mem = NULL; |
| buf->size = 0; |
| buf->use = 0; |
| |
| return ret; |
| } |
| |
| /** |
| * Free a buffer. |
| * |
| * @param buf the buffer to free |
| */ |
| void |
| xmlBufFree(xmlBuf *buf) { |
| if (buf == NULL) |
| return; |
| |
| if (!BUF_STATIC(buf)) |
| xmlFree(buf->mem); |
| xmlFree(buf); |
| } |
| |
| /** |
| * Empty a buffer. |
| * |
| * @param buf the buffer |
| */ |
| void |
| xmlBufEmpty(xmlBuf *buf) { |
| if ((buf == NULL) || (BUF_ERROR(buf)) || (BUF_STATIC(buf))) |
| return; |
| if (buf->mem == NULL) |
| return; |
| |
| buf->use = 0; |
| buf->size += buf->content - buf->mem; |
| buf->content = buf->mem; |
| buf->content[0] = 0; |
| } |
| |
| /** |
| * Discard bytes at the start of a buffer. |
| * |
| * NOTE that the return value differs from #xmlBufferShrink |
| * as it will return 0 on error instead of -1 due to size_t being |
| * used as the return type. |
| * |
| * @deprecated Internal function, don't use. |
| * |
| * @param buf the buffer |
| * @param len the number of bytes to remove |
| * @returns the number of bytes removed or 0 in case of failure |
| */ |
| size_t |
| xmlBufShrink(xmlBuf *buf, size_t len) { |
| if ((buf == NULL) || (BUF_ERROR(buf))) |
| return(0); |
| if (len == 0) |
| return(0); |
| |
| if (len > buf->use) |
| return(0); |
| |
| buf->use -= len; |
| buf->content += len; |
| buf->size -= len; |
| |
| return(len); |
| } |
| |
| /** |
| * Grow a buffer. |
| * |
| * Increase the capacity of a buffer. `len` is the amount of |
| * free space required after the current end of the buffer. |
| * |
| * Assumes `len > buf->size - buf->use`. |
| * |
| * @param buf the buffer |
| * @param len number of extra bytes to allocate |
| * @returns 0 on success, -1 in case of error |
| */ |
| static int |
| xmlBufGrowInternal(xmlBufPtr buf, size_t len) { |
| size_t size; |
| size_t start; |
| xmlChar *newbuf; |
| |
| /* |
| * If there's enough space at the start of the buffer, |
| * move the contents. |
| */ |
| start = buf->content - buf->mem; |
| if (len <= start + buf->size - buf->use) { |
| memmove(buf->mem, buf->content, buf->use + 1); |
| buf->size += start; |
| buf->content = buf->mem; |
| return(0); |
| } |
| |
| if (len > buf->maxSize - buf->use) { |
| xmlBufOverflowError(buf); |
| return(-1); |
| } |
| |
| if (buf->size > (size_t) len) { |
| if (buf->size <= buf->maxSize / 2) |
| size = buf->size * 2; |
| else |
| size = buf->maxSize; |
| } else { |
| size = buf->use + len; |
| if (size <= buf->maxSize - 100) |
| size += 100; |
| } |
| |
| if (buf->content == buf->mem) { |
| newbuf = xmlRealloc(buf->mem, size + 1); |
| if (newbuf == NULL) { |
| xmlBufMemoryError(buf); |
| return(-1); |
| } |
| } else { |
| newbuf = xmlMalloc(size + 1); |
| if (newbuf == NULL) { |
| xmlBufMemoryError(buf); |
| return(-1); |
| } |
| if (buf->content != NULL) |
| memcpy(newbuf, buf->content, buf->use + 1); |
| xmlFree(buf->mem); |
| } |
| |
| buf->mem = newbuf; |
| buf->content = newbuf; |
| buf->size = size; |
| |
| return(0); |
| } |
| |
| /** |
| * Grow a buffer. |
| * |
| * Increase the capacity of a buffer. `len` is the amount of |
| * free space required after the current end of the buffer. |
| * |
| * @param buf the buffer |
| * @param len number of extra bytes to allocate |
| * @returns 0 on success, -1 in case of error |
| */ |
| int |
| xmlBufGrow(xmlBuf *buf, size_t len) { |
| if ((buf == NULL) || (BUF_ERROR(buf)) || (BUF_STATIC(buf))) |
| return(-1); |
| |
| if (len <= buf->size - buf->use) |
| return(0); |
| |
| if (xmlBufGrowInternal(buf, len) < 0) |
| return(-1); |
| |
| return(0); |
| } |
| |
| /** |
| * Get pointer into buffer content. |
| * |
| * @param buf the buffer |
| * @returns the internal content or NULL in case of error |
| */ |
| xmlChar * |
| xmlBufContent(const xmlBuf *buf) |
| { |
| if ((!buf) || (BUF_ERROR(buf))) |
| return NULL; |
| |
| return(buf->content); |
| } |
| |
| /** |
| * Return a pointer to the end of the buffer content. |
| * |
| * @param buf the buffer |
| * @returns the end of the internal content or NULL in case of error |
| */ |
| xmlChar * |
| xmlBufEnd(xmlBuf *buf) |
| { |
| if ((!buf) || (BUF_ERROR(buf))) |
| return NULL; |
| |
| return(&buf->content[buf->use]); |
| } |
| |
| /** |
| * Increase the size of the buffer content. |
| * |
| * If data was appended by writing directly into the content |
| * array, this function increases the buffer size. |
| * |
| * @param buf the buffer |
| * @param len number of bytes to add |
| * @returns 0 on success, -1 in case of error |
| */ |
| int |
| xmlBufAddLen(xmlBuf *buf, size_t len) { |
| if ((buf == NULL) || (BUF_ERROR(buf)) || (BUF_STATIC(buf))) |
| return(-1); |
| if (len > buf->size - buf->use) |
| return(-1); |
| buf->use += len; |
| buf->content[buf->use] = 0; |
| return(0); |
| } |
| |
| /** |
| * Return the size of the buffer content. |
| * |
| * @param buf the buffer |
| * @returns size of buffer content in bytes |
| */ |
| size_t |
| xmlBufUse(xmlBuf *buf) |
| { |
| if ((!buf) || (BUF_ERROR(buf))) |
| return 0; |
| |
| return(buf->use); |
| } |
| |
| /** |
| * Return the size of available space at the end of a buffer. |
| * |
| * @param buf the buffer |
| * @returns available space in bytes |
| */ |
| size_t |
| xmlBufAvail(xmlBuf *buf) |
| { |
| if ((!buf) || (BUF_ERROR(buf))) |
| return 0; |
| |
| return(buf->size - buf->use); |
| } |
| |
| /** |
| * Tell if a buffer is empty |
| * |
| * @param buf the buffer |
| * @returns 0 if no, 1 if yes and -1 in case of error |
| */ |
| int |
| xmlBufIsEmpty(xmlBuf *buf) |
| { |
| if ((!buf) || (BUF_ERROR(buf))) |
| return(-1); |
| |
| return(buf->use == 0); |
| } |
| |
| /** |
| * Append data to a buffer. |
| * |
| * If `len` is -1, `str` is expected to be zero-terminated. |
| * |
| * @param buf the buffer |
| * @param str bytes to add |
| * @param len number of bytes |
| * @returns 0 if successful, -1 in case of error. |
| */ |
| int |
| xmlBufAdd(xmlBuf *buf, const xmlChar *str, size_t len) { |
| if ((buf == NULL) || (BUF_ERROR(buf)) || (BUF_STATIC(buf))) |
| return(-1); |
| if (len == 0) |
| return(0); |
| if (str == NULL) |
| return(-1); |
| |
| if (len > buf->size - buf->use) { |
| if (xmlBufGrowInternal(buf, len) < 0) |
| return(-1); |
| } |
| |
| memmove(&buf->content[buf->use], str, len); |
| buf->use += len; |
| buf->content[buf->use] = 0; |
| |
| return(0); |
| } |
| |
| /** |
| * Append a zero-terminated string to a buffer. |
| * |
| * @param buf the buffer |
| * @param str string (optional) |
| * @returns 0 if successful, -1 in case of error. |
| */ |
| int |
| xmlBufCat(xmlBuf *buf, const xmlChar *str) { |
| if (str == NULL) |
| return(0); |
| return(xmlBufAdd(buf, str, strlen((const char *) str))); |
| } |
| |
| /** |
| * Helper routine to switch from the old buffer structures in use |
| * in various APIs. It creates a wrapper xmlBuf which will be |
| * used for internal processing until xmlBufBackToBuffer is |
| * issued. |
| * |
| * @param buffer incoming old buffer to convert to a new one |
| * @returns a new xmlBuf unless the call failed and NULL is returned |
| */ |
| xmlBuf * |
| xmlBufFromBuffer(xmlBuffer *buffer) { |
| xmlBufPtr ret; |
| |
| if (buffer == NULL) |
| return(NULL); |
| |
| ret = xmlMalloc(sizeof(xmlBuf)); |
| if (ret == NULL) |
| return(NULL); |
| |
| ret->use = buffer->use; |
| ret->flags = 0; |
| ret->maxSize = SIZE_MAX - 1; |
| |
| if (buffer->content == NULL) { |
| ret->size = 50; |
| ret->mem = xmlMalloc(ret->size + 1); |
| ret->content = ret->mem; |
| if (ret->mem == NULL) |
| xmlBufMemoryError(ret); |
| else |
| ret->content[0] = 0; |
| } else { |
| ret->size = buffer->size - 1; |
| ret->content = buffer->content; |
| if (buffer->alloc == XML_BUFFER_ALLOC_IO) |
| ret->mem = buffer->contentIO; |
| else |
| ret->mem = buffer->content; |
| } |
| |
| return(ret); |
| } |
| |
| /** |
| * Function to be called once internal processing had been done to |
| * update back the buffer provided by the user. This can lead to |
| * a failure in case the size accumulated in the xmlBuf is larger |
| * than what an xmlBuffer can support on 64 bits (INT_MAX) |
| * The xmlBuf `buf` wrapper is deallocated by this call in any case. |
| * |
| * @param buf new buffer wrapping the old one |
| * @param ret old buffer |
| * @returns 0 on success, -1 on error. |
| */ |
| int |
| xmlBufBackToBuffer(xmlBuf *buf, xmlBuffer *ret) { |
| if ((buf == NULL) || (ret == NULL)) |
| return(-1); |
| |
| if ((BUF_ERROR(buf)) || (BUF_STATIC(buf)) || |
| (buf->use >= INT_MAX)) { |
| xmlBufFree(buf); |
| ret->content = NULL; |
| ret->contentIO = NULL; |
| ret->use = 0; |
| ret->size = 0; |
| return(-1); |
| } |
| |
| ret->use = buf->use; |
| if (buf->size >= INT_MAX) { |
| /* Keep the buffer but provide a truncated size value. */ |
| ret->size = INT_MAX; |
| } else { |
| ret->size = buf->size + 1; |
| } |
| ret->alloc = XML_BUFFER_ALLOC_IO; |
| ret->content = buf->content; |
| ret->contentIO = buf->mem; |
| xmlFree(buf); |
| return(0); |
| } |
| |
| /** |
| * Update pointers in the parser input struct. |
| * |
| * Update `base`, `cur` and `end` to point into the buffer. |
| * This is required after the content array was reallocated. |
| * |
| * @param buf a buffer |
| * @param input a parser input |
| * @returns 0 on success, -1 in case of error. |
| */ |
| int |
| xmlBufResetInput(xmlBuf *buf, xmlParserInput *input) { |
| return(xmlBufUpdateInput(buf, input, 0)); |
| } |
| |
| /** |
| * Update pointers in the parser input struct. |
| * |
| * Update `base`, `cur` and `end` to point into the buffer. |
| * This is required after the content array was reallocated. |
| * |
| * @param buf a buffer |
| * @param input a parser input |
| * @param pos the `cur` position relative to the start of the |
| * buffer |
| * @returns 0 on success, -1 in case of error. |
| */ |
| int |
| xmlBufUpdateInput(xmlBuf *buf, xmlParserInput *input, size_t pos) { |
| if ((buf == NULL) || (input == NULL)) |
| return(-1); |
| input->base = buf->content; |
| input->cur = input->base + pos; |
| input->end = &buf->content[buf->use]; |
| return(0); |
| } |
| |
| /************************************************************************ |
| * * |
| * Old buffer implementation * |
| * * |
| ************************************************************************/ |
| |
| /** |
| * Set the buffer allocation scheme. |
| * |
| * @deprecated No-op, allocation schemes were removed. |
| * |
| * @param scheme allocation method to use |
| */ |
| void |
| xmlSetBufferAllocationScheme(xmlBufferAllocationScheme scheme ATTRIBUTE_UNUSED) { |
| } |
| |
| /** |
| * Get the buffer allocation scheme. |
| * |
| * @deprecated Allocation schemes were removed. |
| * |
| * @returns the current allocation scheme |
| */ |
| xmlBufferAllocationScheme |
| xmlGetBufferAllocationScheme(void) { |
| return(XML_BUFFER_ALLOC_EXACT); |
| } |
| |
| /** |
| * Create a buffer. |
| * |
| * The default initial size is 256. |
| * |
| * @returns the new structure. |
| */ |
| xmlBuffer * |
| xmlBufferCreate(void) { |
| xmlBufferPtr ret; |
| |
| ret = xmlMalloc(sizeof(*ret)); |
| if (ret == NULL) |
| return(NULL); |
| |
| ret->use = 0; |
| ret->size = 256; |
| ret->alloc = XML_BUFFER_ALLOC_IO; |
| ret->contentIO = xmlMalloc(ret->size); |
| if (ret->contentIO == NULL) { |
| xmlFree(ret); |
| return(NULL); |
| } |
| ret->content = ret->contentIO; |
| ret->content[0] = 0; |
| |
| return(ret); |
| } |
| |
| /** |
| * Create a buffer with an initial size. |
| * |
| * @param size initial size of buffer |
| * @returns the new structure. |
| */ |
| xmlBuffer * |
| xmlBufferCreateSize(size_t size) { |
| xmlBufferPtr ret; |
| |
| if (size >= INT_MAX) |
| return(NULL); |
| |
| ret = xmlMalloc(sizeof(*ret)); |
| if (ret == NULL) |
| return(NULL); |
| |
| ret->use = 0; |
| ret->alloc = XML_BUFFER_ALLOC_IO; |
| ret->size = (size ? size + 1 : 0); /* +1 for ending null */ |
| |
| if (ret->size) { |
| ret->contentIO = xmlMalloc(ret->size); |
| if (ret->contentIO == NULL) { |
| xmlFree(ret); |
| return(NULL); |
| } |
| ret->content = ret->contentIO; |
| ret->content[0] = 0; |
| } else { |
| ret->contentIO = NULL; |
| ret->content = NULL; |
| } |
| |
| return(ret); |
| } |
| |
| /** |
| * Extract buffer content. |
| * |
| * Return the contents of the buffer as an `xmlChar` string, |
| * clearing the buffer. |
| * |
| * This doesn't work with static buffers as they can't be reset. |
| * |
| * @param buf the buffer |
| * @returns the buffer content |
| */ |
| xmlChar * |
| xmlBufferDetach(xmlBuffer *buf) { |
| xmlChar *ret; |
| |
| if (buf == NULL) |
| return(NULL); |
| |
| if ((buf->alloc == XML_BUFFER_ALLOC_IO) && |
| (buf->content != buf->contentIO)) { |
| ret = xmlStrndup(buf->content, buf->use); |
| xmlFree(buf->contentIO); |
| } else { |
| ret = buf->content; |
| } |
| |
| buf->contentIO = NULL; |
| buf->content = NULL; |
| buf->size = 0; |
| buf->use = 0; |
| |
| return ret; |
| } |
| |
| /** |
| * Create a static buffer. |
| * |
| * The memory must be zero-terminated and not be modified for the |
| * lifetime of the buffer. A static buffer can't be grown, modified or |
| * detached, but it can be shrunk. |
| * |
| * @param mem the memory area |
| * @param size the size in bytes |
| * @returns a new buffer |
| */ |
| xmlBuffer * |
| xmlBufferCreateStatic(void *mem, size_t size) { |
| xmlBufferPtr buf = xmlBufferCreateSize(size); |
| |
| xmlBufferAdd(buf, mem, size); |
| return(buf); |
| } |
| |
| /** |
| * Set the allocation scheme of a buffer. |
| * |
| * For libxml2 before 2.14, it is recommended to set this to |
| * XML_BUFFER_ALLOC_DOUBLE_IT. Has no effect on 2.14 or later. |
| * |
| * @param buf the buffer to tune |
| * @param scheme allocation scheme to use |
| */ |
| void |
| xmlBufferSetAllocationScheme(xmlBuffer *buf ATTRIBUTE_UNUSED, |
| xmlBufferAllocationScheme scheme ATTRIBUTE_UNUSED) { |
| } |
| |
| /** |
| * Free a buffer. |
| * |
| * @param buf the buffer to free |
| */ |
| void |
| xmlBufferFree(xmlBuffer *buf) { |
| if (buf == NULL) |
| return; |
| |
| if (buf->alloc == XML_BUFFER_ALLOC_IO) |
| xmlFree(buf->contentIO); |
| else |
| xmlFree(buf->content); |
| |
| xmlFree(buf); |
| } |
| |
| /** |
| * Empty a buffer. |
| * |
| * @param buf the buffer |
| */ |
| void |
| xmlBufferEmpty(xmlBuffer *buf) { |
| if (buf == NULL) |
| return; |
| if (buf->content == NULL) |
| return; |
| |
| buf->use = 0; |
| |
| if (buf->alloc == XML_BUFFER_ALLOC_IO) { |
| buf->size += buf->content - buf->contentIO; |
| buf->content = buf->contentIO; |
| buf->content[0] = 0; |
| } else { |
| buf->content[0] = 0; |
| } |
| } |
| |
| /** |
| * Discard bytes at the start of a buffer. |
| * |
| * @deprecated Internal function, don't use. |
| * |
| * @param buf the buffer |
| * @param len the number of bytes to remove |
| * @returns the number of bytes removed, or -1 in case of failure. |
| */ |
| int |
| xmlBufferShrink(xmlBuffer *buf, unsigned int len) { |
| if (buf == NULL) |
| return(-1); |
| if (len == 0) |
| return(0); |
| if (len > buf->use) |
| return(-1); |
| |
| buf->use -= len; |
| |
| if (buf->alloc == XML_BUFFER_ALLOC_IO) { |
| buf->content += len; |
| buf->size -= len; |
| } else { |
| memmove(buf->content, &buf->content[len], buf->use + 1); |
| } |
| |
| return(len); |
| } |
| |
| /** |
| * Grow a buffer. |
| * |
| * @deprecated Internal function, don't use. |
| * |
| * @param buf the buffer |
| * @param len number of extra bytes to allocate |
| * @returns the new available space or -1 in case of error |
| */ |
| int |
| xmlBufferGrow(xmlBuffer *buf, unsigned int len) { |
| unsigned int size; |
| xmlChar *newbuf; |
| |
| if (buf == NULL) |
| return(-1); |
| |
| if (len < buf->size - buf->use) |
| return(0); |
| if (len >= INT_MAX - buf->use) |
| return(-1); |
| |
| if (buf->size > (size_t) len) { |
| if (buf->size <= INT_MAX / 2) |
| size = buf->size * 2; |
| else |
| size = INT_MAX; |
| } else { |
| size = buf->use + len + 1; |
| if (size <= INT_MAX - 100) |
| size += 100; |
| } |
| |
| if ((buf->alloc == XML_BUFFER_ALLOC_IO) && |
| (buf->content != buf->contentIO)) { |
| newbuf = xmlMalloc(size); |
| if (newbuf == NULL) |
| return(-1); |
| if (buf->content != NULL) |
| memcpy(newbuf, buf->content, buf->use + 1); |
| xmlFree(buf->contentIO); |
| } else { |
| newbuf = xmlRealloc(buf->content, size); |
| if (newbuf == NULL) |
| return(-1); |
| } |
| |
| if (buf->alloc == XML_BUFFER_ALLOC_IO) |
| buf->contentIO = newbuf; |
| buf->content = newbuf; |
| buf->size = size; |
| |
| return(buf->size - buf->use - 1); |
| } |
| |
| /** |
| * Dump a buffer to a `FILE`. |
| * |
| * @param file the output file |
| * @param buf the buffer |
| * @returns the number of bytes written |
| */ |
| int |
| xmlBufferDump(FILE *file, xmlBuffer *buf) { |
| size_t ret; |
| |
| if (buf == NULL) |
| return(0); |
| if (buf->content == NULL) |
| return(0); |
| if (file == NULL) |
| file = stdout; |
| ret = fwrite(buf->content, 1, buf->use, file); |
| return(ret > INT_MAX ? INT_MAX : ret); |
| } |
| |
| /** |
| * Get pointer into buffer content. |
| * |
| * @param buf the buffer |
| * @returns the internal content |
| */ |
| const xmlChar * |
| xmlBufferContent(const xmlBuffer *buf) |
| { |
| if(!buf) |
| return NULL; |
| |
| return buf->content; |
| } |
| |
| /** |
| * Get the size of the buffer content. |
| * |
| * @param buf the buffer |
| * @returns the size of the buffer content in bytes |
| */ |
| int |
| xmlBufferLength(const xmlBuffer *buf) |
| { |
| if(!buf) |
| return 0; |
| |
| return buf->use; |
| } |
| |
| /** |
| * Resize a buffer to a minimum size. |
| * |
| * @deprecated Internal function, don't use. |
| * |
| * @param buf the buffer to resize |
| * @param size the desired size |
| * @returns 1 on succes, 0 in case of error |
| */ |
| int |
| xmlBufferResize(xmlBuffer *buf, unsigned int size) |
| { |
| int res; |
| |
| if (buf == NULL) |
| return(0); |
| if (size < buf->size) |
| return(1); |
| res = xmlBufferGrow(buf, size - buf->use); |
| |
| return(res < 0 ? 0 : 1); |
| } |
| |
| /** |
| * Append bytes to a buffer. |
| * |
| * If `len` is -1, `str` is assumed to be zero-terminated. |
| * |
| * @param buf the buffer |
| * @param str bytes to add |
| * @param len number of bytes |
| * @returns an xmlParserErrors code. |
| */ |
| int |
| xmlBufferAdd(xmlBuffer *buf, const xmlChar *str, int len) { |
| if ((buf == NULL) || (str == NULL)) |
| return(XML_ERR_ARGUMENT); |
| if (len < 0) |
| len = xmlStrlen(str); |
| if (len == 0) |
| return(XML_ERR_OK); |
| |
| /* Note that both buf->size and buf->use can be zero here. */ |
| if ((unsigned) len >= buf->size - buf->use) { |
| if (xmlBufferGrow(buf, len) < 0) |
| return(XML_ERR_NO_MEMORY); |
| } |
| |
| memmove(&buf->content[buf->use], str, len); |
| buf->use += len; |
| buf->content[buf->use] = 0; |
| return(XML_ERR_OK); |
| } |
| |
| /** |
| * Prepend bytes to a buffer. |
| * |
| * If `len` is -1, `str` is assumed to be zero-terminated. |
| * |
| * @param buf the buffer |
| * @param str bytes to prepend |
| * @param len number of bytes |
| * @returns an xmlParserErrors code. |
| */ |
| int |
| xmlBufferAddHead(xmlBuffer *buf, const xmlChar *str, int len) { |
| unsigned start = 0; |
| |
| if ((buf == NULL) || (str == NULL)) |
| return(XML_ERR_ARGUMENT); |
| if (len < 0) |
| len = xmlStrlen(str); |
| if (len == 0) |
| return(XML_ERR_OK); |
| |
| if (buf->alloc == XML_BUFFER_ALLOC_IO) { |
| start = buf->content - buf->contentIO; |
| |
| /* |
| * We can add it in the space previously shrunk |
| */ |
| if ((unsigned) len <= start) { |
| buf->content -= len; |
| memmove(&buf->content[0], str, len); |
| buf->use += len; |
| buf->size += len; |
| return(0); |
| } |
| if ((unsigned) len < buf->size + start - buf->use) { |
| memmove(&buf->contentIO[len], buf->content, buf->use + 1); |
| memmove(buf->contentIO, str, len); |
| buf->content = buf->contentIO; |
| buf->use += len; |
| buf->size += start; |
| return(0); |
| } |
| } |
| |
| if ((unsigned) len >= buf->size - buf->use) { |
| if (xmlBufferGrow(buf, len) < 0) |
| return(-1); |
| } |
| |
| memmove(&buf->content[len], buf->content, buf->use + 1); |
| memmove(buf->content, str, len); |
| buf->use += len; |
| return (0); |
| } |
| |
| /** |
| * Append a zero-terminated string to a buffer. |
| * |
| * @param buf the buffer |
| * @param str string to add |
| * @returns an xmlParserErrors code. |
| */ |
| int |
| xmlBufferCat(xmlBuffer *buf, const xmlChar *str) { |
| return(xmlBufferAdd(buf, str, -1)); |
| } |
| |
| /** |
| * Append a zero-terminated C string to a buffer. |
| * |
| * @param buf the buffer |
| * @param str string to add |
| * @returns an xmlParserErrors code. |
| */ |
| int |
| xmlBufferCCat(xmlBuffer *buf, const char *str) { |
| return(xmlBufferAdd(buf, (const xmlChar *) str, -1)); |
| } |
| |
| /** |
| * Append a zero-terminated `xmlChar` string to a buffer. |
| * |
| * @param buf the XML buffer |
| * @param string the string to add |
| */ |
| void |
| xmlBufferWriteCHAR(xmlBuffer *buf, const xmlChar *string) { |
| xmlBufferAdd(buf, string, -1); |
| } |
| |
| /** |
| * Append a zero-terminated C string to a buffer. |
| * |
| * Same as #xmlBufferCCat. |
| * |
| * @param buf the buffer |
| * @param string the string to add |
| */ |
| void |
| xmlBufferWriteChar(xmlBuffer *buf, const char *string) { |
| xmlBufferAdd(buf, (const xmlChar *) string, -1); |
| } |
| |
| /** |
| * Append a quoted string to a buffer. |
| * |
| * Append a string quoted with single or double quotes. If the |
| * string contains both single and double quotes, double quotes |
| * are escaped with `"`. |
| * |
| * @param buf the buffer |
| * @param string the string to add |
| */ |
| void |
| xmlBufferWriteQuotedString(xmlBuffer *buf, const xmlChar *string) { |
| const xmlChar *cur, *base; |
| if (buf == NULL) |
| return; |
| if (xmlStrchr(string, '\"')) { |
| if (xmlStrchr(string, '\'')) { |
| xmlBufferCCat(buf, "\""); |
| base = cur = string; |
| while(*cur != 0){ |
| if(*cur == '"'){ |
| if (base != cur) |
| xmlBufferAdd(buf, base, cur - base); |
| xmlBufferAdd(buf, BAD_CAST """, 6); |
| cur++; |
| base = cur; |
| } |
| else { |
| cur++; |
| } |
| } |
| if (base != cur) |
| xmlBufferAdd(buf, base, cur - base); |
| xmlBufferCCat(buf, "\""); |
| } |
| else{ |
| xmlBufferCCat(buf, "\'"); |
| xmlBufferCat(buf, string); |
| xmlBufferCCat(buf, "\'"); |
| } |
| } else { |
| xmlBufferCCat(buf, "\""); |
| xmlBufferCat(buf, string); |
| xmlBufferCCat(buf, "\""); |
| } |
| } |
| |