1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-25 23:21:54 +03:00
samba-mirror/source3/lib/cbuf.c
Volker Lendecke 6a91ba4f7e lib: Remove an #include "includes.h"
Signed-off-by: Volker Lendecke <vl@samba.org>
Reviewed-by: Jeremy Allison <jra@samba.org>
2018-07-24 20:36:49 +02:00

329 lines
5.6 KiB
C

/*
* Samba Unix/Linux SMB client library
*
* Copyright (C) Gregor Beck 2010
*
* This program 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 3 of the License, or
* (at your option) any later version.
*
* 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, see <http://www.gnu.org/licenses/>.
*/
/**
* @file cbuf.c
* @author Gregor Beck <gb@sernet.de>
* @date Aug 2010
*
* @brief A talloced character buffer.
*
*/
#include "replace.h"
#include "system/locale.h"
#include "cbuf.h"
#include <talloc.h>
#include <assert.h>
#include "lib/util/byteorder.h"
struct cbuf {
char* buf;
size_t pos;
size_t size;
};
cbuf* cbuf_clear(cbuf* b)
{
cbuf_setpos(b, 0);
return b;
}
cbuf* cbuf_new(const void* ctx)
{
cbuf* s = talloc(ctx, cbuf);
if (s == NULL)
return NULL;
s->size = 32;
s->buf = (char *)talloc_size(s, s->size);
if (s->size && (s->buf == NULL)) {
talloc_free(s);
return NULL;
}
return cbuf_clear(s);
}
cbuf* cbuf_copy(const cbuf* b)
{
cbuf* s = talloc(talloc_parent(b), cbuf);
if (s == NULL) {
return NULL;
}
s->buf = (char *)talloc_memdup(s, b->buf, b->size); /* only up to pos? */
/* XXX shallow did not work, because realloc */
/* fails with multiple references */
/* s->buf = talloc_reference(s, b->buf); */
if (s->buf == NULL) {
cbuf_delete(s);
return NULL;
}
s->size = b->size;
s->pos = b->pos;
return s;
}
void cbuf_delete(cbuf* b)
{
talloc_free(b);
}
#define SWAP(A,B,T) do { \
T tmp = A; A = B; B = tmp; \
} while(0)
void cbuf_swap(cbuf* b1, cbuf* b2)
{
if (b1 == b2) {
return;
}
talloc_reparent(b1, b2, b1->buf);
talloc_reparent(b2, b1, b2->buf);
SWAP(b1->buf, b2->buf, char*);
SWAP(b1->pos, b2->pos, size_t);
SWAP(b1->size, b2->size, size_t);
}
cbuf* cbuf_takeover(cbuf* b1, cbuf* b2)
{
talloc_reparent(b2, b1, b2->buf);
b1->buf = b2->buf;
b1->pos = b2->pos;
b1->size = b2->size;
cbuf_delete(b2);
return b1;
}
cbuf* cbuf_swapptr(cbuf* b, char** ptr, size_t len)
{
void* p = talloc_parent(*ptr);
SWAP(b->buf, *ptr, char*);
talloc_steal(b, b->buf);
talloc_steal(p, *ptr);
b->size = talloc_get_size(b->buf);
b->pos = (len == -1) ? strlen(b->buf) : len;
assert(b->pos <= b->size);
return b;
}
cbuf* cbuf_resize(cbuf* b, size_t size)
{
char* save_buf = b->buf;
b->buf = talloc_realloc(b, b->buf, char, size);
if (b->buf == NULL) {
talloc_free(save_buf);
b->size = 0;
} else {
b->size = size;
}
b->pos = MIN(b->pos, b->size);
return b->buf ? b : NULL;
}
char* cbuf_reserve(cbuf* b, size_t len)
{
if(b->size < b->pos + len)
cbuf_resize(b, MAX(2*b->size, b->pos + len));
return b->buf ? b->buf + b->pos : NULL;
}
int cbuf_puts(cbuf* b, const char* str, size_t len)
{
char* dst;
if (b == NULL)
return 0;
if (len == -1) {
len=strlen(str);
}
dst = cbuf_reserve(b, len+1);
if (dst == NULL)
return -1;
memcpy(dst, str, len);
dst[len] = '\0'; /* just to ease debugging */
b->pos += len;
assert(b->pos < b->size);
return len;
}
int cbuf_putc(cbuf* b, char c) {
char* dst;
if (b == NULL)
return 0;
dst = cbuf_reserve(b, 2);
if (dst == NULL) {
return -1;
}
dst[0] = c;
dst[1] = '\0'; /* just to ease debugging */
b->pos++;
assert(b->pos < b->size);
return 1;
}
int cbuf_putdw(cbuf* b, uint32_t u) {
char* dst;
static const size_t LEN = sizeof(uint32_t);
if (b == NULL)
return 0;
dst = cbuf_reserve(b, LEN);
if (dst == NULL) {
return -1;
}
SIVAL(dst, 0, u);
b->pos += LEN;
assert(b->pos <= b->size); /* no NULL termination*/
return LEN;
}
size_t cbuf_getpos(const cbuf* b) {
assert(b->pos <= b->size);
return b->pos;
}
void cbuf_setpos(cbuf* b, size_t pos) {
assert(pos <= b->size);
b->pos = pos;
if (pos < b->size)
b->buf[pos] = '\0'; /* just to ease debugging */
}
char* cbuf_gets(cbuf* b, size_t idx) {
assert(idx <= b->pos);
if (cbuf_reserve(b, 1) == NULL)
return NULL;
b->buf[b->pos] = '\0';
return b->buf + idx;
}
int cbuf_printf(cbuf* b, const char* fmt, ...)
{
va_list args, args2;
int len;
char* dst = b->buf + b->pos;
const int avail = b->size - b->pos;
assert(avail >= 0);
va_start(args, fmt);
va_copy(args2, args);
len = vsnprintf(dst, avail, fmt, args);
if (len >= avail) {
dst = cbuf_reserve(b, len+1);
len = (dst != NULL) ? vsnprintf(dst, len+1, fmt, args2) : -1;
}
if (len > 0) {
b->pos += len;
}
va_end(args);
va_end(args2);
assert(b->pos <= b->size);
return len;
}
int cbuf_print_quoted_string(cbuf* ost, const char* s)
{
int n = 1;
cbuf_putc(ost,'"');
while(true) {
switch (*s) {
case '\0':
cbuf_putc(ost, '"');
return n+1;
case '"':
case '\\':
cbuf_putc(ost, '\\');
n++;
FALL_THROUGH;
default:
cbuf_putc(ost, *s);
n++;
}
s++;
}
}
int cbuf_print_quoted(cbuf* ost, const char* s, size_t len)
{
int n = 1;
int ret;
cbuf_reserve(ost, len+2);
cbuf_putc(ost,'"');
while(len--) {
switch (*s) {
case '"':
case '\\':
ret = cbuf_printf(ost, "\\%c", *s);
break;
default:
if (isprint(*s) && ((*s == ' ') || !isspace(*s))) {
ret = cbuf_putc(ost, *s);
} else {
ret = cbuf_printf(ost,
"\\%02x",
(unsigned char)*s);
}
}
s++;
if (ret == -1) {
return -1;
}
n += ret;
}
ret = cbuf_putc(ost,'"');
return (ret == -1) ? -1 : (n + ret);
}