denyhosts/clamscan/libclamav/jsparse/textbuf.h
2022-10-22 18:41:00 +08:00

68 lines
2.0 KiB
C

/*
* Copyright (C) 2013-2022 Cisco Systems, Inc. and/or its affiliates. All rights reserved.
* Copyright (C) 2008-2013 Sourcefire, Inc.
*
* Authors: Török Edvin
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2 as
* published by the Free Software Foundation.
*
* 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301, USA.
*/
#ifndef TEXTBUF_H
#define TEXTBUF_H
struct text_buffer {
char *data;
size_t pos;
size_t capacity;
};
static inline int textbuffer_ensure_capacity(struct text_buffer *txtbuf, size_t len)
{
if (txtbuf->pos + len > txtbuf->capacity) {
char *d;
unsigned capacity = MAX(txtbuf->pos + len, txtbuf->capacity + 4096);
d = cli_realloc(txtbuf->data, capacity);
if (!d)
return -1;
txtbuf->capacity = capacity;
txtbuf->data = d;
}
return 0;
}
static inline int textbuffer_append_len(struct text_buffer *txtbuf, const char *s, size_t len)
{
if (textbuffer_ensure_capacity(txtbuf, len) == -1)
return -1;
memcpy(&txtbuf->data[txtbuf->pos], s, len);
txtbuf->pos += len;
return 0;
}
static inline int textbuffer_append(struct text_buffer *txtbuf, const char *s)
{
size_t len = strlen(s);
return textbuffer_append_len(txtbuf, s, len);
}
static inline int textbuffer_putc(struct text_buffer *txtbuf, const char c)
{
if (textbuffer_ensure_capacity(txtbuf, 1) == -1)
return -1;
txtbuf->data[txtbuf->pos++] = c;
return 0;
}
#endif