shared/strbuf.c - master - niceSOFT/kmod

shared/strbuf.c
122 строки2 KB
Emil Velikov
shared/strbuf: make strbuf_reserve_extra public
10 авг 2026, 16:49
10 авг 2026, 16:4944cdb6f
// SPDX-License-Identifier: LGPL-2.1-or-later
/*
 * Copyright (C) 2011-2013 ProFUSION embedded systems
 * Copyright (C) 2014 Intel Corporation. All rights reserved.
 */

#include <assert.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include <sys/param.h>

#include "macro.h"
#include "strbuf.h"
#include "util.h"

#define BUF_STEP 128

static bool buf_realloc(struct strbuf *buf, size_t sz)
{
	void *tmp = realloc(buf->heap ? buf->bytes : NULL, sz);

	if (tmp == NULL)
		return false;

	if (!buf->heap)
		memcpy(tmp, buf->bytes, MIN(buf->size, sz));

	buf->heap = true;
	buf->bytes = tmp;
	buf->size = sz;

	return true;
}

bool strbuf_reserve_extra(struct strbuf *buf, size_t n)
{
	if (n < buf->size - buf->used)
		return true;

	/* Reserve extra space for the \0 byte */
	if (uaddsz_overflow(n, 1, &n) || uaddsz_overflow(buf->used, n, &n) ||
	    n > SIZE_MAX - BUF_STEP)
		return false;

	n = ALIGN(n, BUF_STEP);

	return buf_realloc(buf, n);
}

void strbuf_init(struct strbuf *buf)
{
	buf->bytes = NULL;
	buf->size = 0;
	buf->used = 0;
	buf->heap = true;
}

void strbuf_release(struct strbuf *buf)
{
	if (buf->heap)
		free(buf->bytes);
}

const char *strbuf_str(struct strbuf *buf)
{
	if (!buf->used)
		return "";

	if (buf->bytes[buf->used - 1])
		buf->bytes[buf->used] = '\0';
	return buf->bytes;
}

bool strbuf_pushchar(struct strbuf *buf, char ch)
{
	if (!strbuf_reserve_extra(buf, 1))
		return false;
	buf->bytes[buf->used] = ch;
	buf->used++;
	return true;
}

size_t strbuf_pushmem(struct strbuf *buf, const char *src, size_t sz)
{
	assert(src != NULL);
	assert(buf != NULL);

	if (sz == 0)
		return 0;

	if (!strbuf_reserve_extra(buf, sz))
		return 0;

	memcpy(buf->bytes + buf->used, src, sz);
	buf->used += sz;

	return sz;
}

void strbuf_popchar(struct strbuf *buf)
{
	assert(buf->used > 0);
	buf->used--;
}

void strbuf_popchars(struct strbuf *buf, size_t n)
{
	assert(buf->used >= n);
	buf->used -= n;
}

void strbuf_shrink_to(struct strbuf *buf, size_t sz)
{
	assert(buf->used >= sz);
	buf->used = sz;
}

void strbuf_clear(struct strbuf *buf)
{
	buf->used = 0;
}