X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Fdynamic-buffer.h;h=3435c846c3bf60054a86339383d96aa8ccb42a97;hp=19097f927e219bdb4e7eb69defe1f4e12db1f447;hb=e6887944f09a8985efb521ea638b4dfe557ff4ed;hpb=5966417eba59c9014c1f67ae37257e7266e647b5 diff --git a/src/common/dynamic-buffer.h b/src/common/dynamic-buffer.h index 19097f927..3435c846c 100644 --- a/src/common/dynamic-buffer.h +++ b/src/common/dynamic-buffer.h @@ -1,18 +1,8 @@ /* - * Copyright (C) 2017 - Jérémie Galarneau + * Copyright (C) 2017 Jérémie Galarneau * - * This program is free software; you can redistribute it and/or modify it - * under the terms of the GNU Lesser General Public License, version 2.1 only, - * as published by the Free Software Foundation. + * SPDX-License-Identifier: LGPL-2.1-only * - * This program 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 Lesser General Public License - * for more details. - * - * You should have received a copy of the GNU Lesser General Public License - * along with this program; if not, write to the Free Software Foundation, - * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #ifndef LTTNG_DYNAMIC_BUFFER_H @@ -20,43 +10,94 @@ #include #include +#include + +struct lttng_buffer_view; struct lttng_dynamic_buffer { char *data; + /* size is the buffer's currently used capacity. */ size_t size; - size_t capacity; + /* + * capacity shall not be accessed by users directly, it is meant for + * internal use only. + */ + size_t _capacity; }; +/* + * Initialize a dynamic buffer. This performs no allocation and is meant + * to be used instead of memset or explicit initialization of the buffer. + */ +LTTNG_HIDDEN void lttng_dynamic_buffer_init(struct lttng_dynamic_buffer *buffer); +/* + * Append the content of a raw memory buffer to the end of a dynamic buffer + * (after its current "size"). The dynamic buffer's size is increased by + * "len", and its capacity is adjusted automatically. + */ +LTTNG_HIDDEN int lttng_dynamic_buffer_append(struct lttng_dynamic_buffer *buffer, const void *buf, size_t len); +/* + * Performs the same action as lttng_dynamic_buffer_append(), but using another + * dynamic buffer as the source buffer. The source buffer's size is used in lieu + * of "len". + */ +LTTNG_HIDDEN int lttng_dynamic_buffer_append_buffer(struct lttng_dynamic_buffer *dst_buffer, - struct lttng_dynamic_buffer *src_buffer); + const struct lttng_dynamic_buffer *src_buffer); + +/* + * Performs the same action as lttng_dynamic_buffer_append(), but using a + * buffer view as the source buffer. The source buffer's size is used in lieu + * of "len". + */ +LTTNG_HIDDEN +int lttng_dynamic_buffer_append_view(struct lttng_dynamic_buffer *buffer, + const struct lttng_buffer_view *view); /* * Set the buffer's size to new_size. The capacity of the buffer will - * be expanded (if necessary) to accomodate new_size. Areas acquired by - * an enlarging new_size _will be zeroed_. + * be expanded (if necessary) to accommodates new_size. Areas acquired by + * a size increase will be zeroed. * * Be careful to expand the buffer's size _before_ calling out external * APIs (e.g. read(3)) which may populate the buffer as setting the size - * _after_ will zero-out the result of the operation. + * after will zero-out the result of the operation. + * + * Shrinking a buffer does not zero the old content. If the buffer may contain + * sensititve information, it must be cleared manually _before_ changing the + * size. + * + * NOTE: It is striclty _invalid_ to access memory after _size_, regardless + * of prior calls to set_capacity(). */ +LTTNG_HIDDEN int lttng_dynamic_buffer_set_size(struct lttng_dynamic_buffer *buffer, size_t new_size); /* - * Set the buffer's capacity to accomodate the new_capacity, allocating memory - * as necessary. The buffer's content is preserved. + * Set the buffer's capacity to accommodates the new_capacity, allocating memory + * as necessary. The buffer's content is preserved. Setting a buffer's capacity + * is meant as a _hint_ to the underlying buffer and is only optimization; no + * guarantee is offered that subsequent calls to append or set_size will succeed. * * If the current size > new_capacity, the operation will fail. */ +LTTNG_HIDDEN int lttng_dynamic_buffer_set_capacity(struct lttng_dynamic_buffer *buffer, size_t new_capacity); /* Release any memory used by the dynamic buffer. */ +LTTNG_HIDDEN void lttng_dynamic_buffer_reset(struct lttng_dynamic_buffer *buffer); +/* Get the space left in the buffer before a new resize is needed. */ +LTTNG_HIDDEN +size_t lttng_dynamic_buffer_get_capacity_left( + struct lttng_dynamic_buffer *buffer); + #endif /* LTTNG_DYNAMIC_BUFFER_H */