summaryrefslogtreecommitdiffstats
path: root/libglusterfs/src/strfd.c
blob: f5b7b94bfa10b117c2cdef22bae1ddf03d08edc8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
   Copyright (c) 2014 Red Hat, Inc. <http://www.redhat.com>
   This file is part of GlusterFS.

   This file is licensed to you under your choice of the GNU Lesser
   General Public License, version 3 or any later version (LGPLv3 or
   later), or the GNU General Public License, version 2 (GPLv2), in all
   cases as published by the Free Software Foundation.
*/

#ifndef _CONFIG_H
#define _CONFIG_H
#include "config.h"
#endif

#include <stdarg.h>

#include "mem-types.h"
#include "mem-pool.h"
#include "strfd.h"
#include "common-utils.h"


strfd_t *
strfd_open ()
{
	strfd_t *strfd = NULL;

	strfd = GF_CALLOC(1, sizeof(*strfd), gf_common_mt_strfd_t);

	return strfd;
}


int
strvprintf (strfd_t *strfd, const char *fmt, va_list ap)
{
	char *str = NULL;
	int size = 0;

	size = vasprintf (&str, fmt, ap);

	if (size < 0)
		return size;

	if (!strfd->alloc_size) {
		strfd->data = GF_CALLOC (max(size + 1, 4096), 1,
					 gf_common_mt_strfd_data_t);
		if (!strfd->data) {
			free (str); /* NOT GF_FREE */
			return -1;
		}
		strfd->alloc_size = max(size + 1, 4096);
	}

	if (strfd->alloc_size <= (strfd->size + size)) {
		char *tmp_ptr = NULL;
		int new_size = max ((strfd->alloc_size * 2),
				    gf_roundup_next_power_of_two (strfd->size + size + 1));
		tmp_ptr = GF_REALLOC (strfd->data, new_size);
		if (!tmp_ptr) {
			free (str); /* NOT GF_FREE */
			return -1;
		}
		strfd->alloc_size = new_size;
		strfd->data = tmp_ptr;
	}

	// Copy the trailing '\0', but do not account for it in ->size.
	// This allows safe use of strfd->data as a string.
	memcpy (strfd->data + strfd->size, str, size + 1);
	strfd->size += size;

	free (str); /* NOT GF_FREE */

	return size;
}


int
strprintf (strfd_t *strfd, const char *fmt, ...)
{
	int ret = 0;
	va_list ap;

	va_start (ap, fmt);
	ret = strvprintf (strfd, fmt, ap);
	va_end (ap);

	return ret;
}


int
strfd_close (strfd_t *strfd)
{
	GF_FREE (strfd->data);
	GF_FREE (strfd);

	return 0;
}