buffer.h 6.05 KB
Newer Older
Vicent Marti committed
1
/*
Edward Thomson committed
2
 * Copyright (C) the libgit2 contributors. All rights reserved.
Vicent Marti committed
3 4 5 6
 *
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
 */
7 8 9 10
#ifndef INCLUDE_buffer_h__
#define INCLUDE_buffer_h__

#include "common.h"
11
#include "git2/strarray.h"
12
#include "git2/buffer.h"
13

14 15 16 17 18
/* typedef struct {
 *  	char   *ptr;
 *  	size_t asize, size;
 * } git_buf;
 */
19

20 21
extern char git_buf__initbuf[];
extern char git_buf__oom[];
22

23
/* Use to initialize buffer structure when git_buf is on stack */
24
#define GIT_BUF_INIT { git_buf__initbuf, 0, 0 }
25

26 27 28 29 30
GIT_INLINE(bool) git_buf_is_allocated(const git_buf *buf)
{
	return (buf->ptr != NULL && buf->asize > 0);
}

31 32 33 34 35 36
/**
 * Initialize a git_buf structure.
 *
 * For the cases where GIT_BUF_INIT cannot be used to do static
 * initialization.
 */
37
extern void git_buf_init(git_buf *buf, size_t initial_size);
38 39

/**
40
 * Attempt to grow the buffer to hold at least `target_size` bytes.
41
 *
42
 * If the allocation fails, this will return an error.  If `mark_oom` is true,
43 44
 * this will mark the buffer as invalid for future operations; if false,
 * existing buffer content will be preserved, but calling code must handle
45 46 47
 * that buffer was not expanded.  If `preserve_external` is true, then any
 * existing data pointed to be `ptr` even if `asize` is zero will be copied
 * into the newly allocated buffer.
48
 */
49 50
extern int git_buf_try_grow(
	git_buf *buf, size_t target_size, bool mark_oom, bool preserve_external);
51

52 53 54 55 56 57 58 59 60
/**
 * Sanitizes git_buf structures provided from user input.  Users of the
 * library, when providing git_buf's, may wish to provide a NULL ptr for
 * ease of handling.  The buffer routines, however, expect a non-NULL ptr
 * always.  This helper method simply handles NULL input, converting to a
 * git_buf__initbuf.
 */
extern void git_buf_sanitize(git_buf *buf);

61 62 63
extern void git_buf_swap(git_buf *buf_a, git_buf *buf_b);
extern char *git_buf_detach(git_buf *buf);
extern void git_buf_attach(git_buf *buf, char *ptr, size_t asize);
64 65

/**
66 67
 * Test if there have been any reallocation failures with this git_buf.
 *
68 69
 * Any function that writes to a git_buf can fail due to memory allocation
 * issues.  If one fails, the git_buf will be marked with an OOM error and
70 71 72
 * further calls to modify the buffer will fail.  Check git_buf_oom() at the
 * end of your sequence and it will be true if you ran out of memory at any
 * point with that buffer.
73 74
 *
 * @return false if no error, true if allocation error
75
 */
76 77 78 79
GIT_INLINE(bool) git_buf_oom(const git_buf *buf)
{
	return (buf->ptr == git_buf__oom);
}
80 81

/*
82 83 84 85 86 87
 * Functions below that return int value error codes will return 0 on
 * success or -1 on failure (which generally means an allocation failed).
 * Using a git_buf where the allocation has failed with result in -1 from
 * all further calls using that buffer.  As a result, you can ignore the
 * return code of these functions and call them in a series then just call
 * git_buf_oom at the end.
88 89 90
 */
int git_buf_sets(git_buf *buf, const char *string);
int git_buf_putc(git_buf *buf, char c);
91
int git_buf_putcn(git_buf *buf, char c, size_t len);
92 93 94
int git_buf_put(git_buf *buf, const char *data, size_t len);
int git_buf_puts(git_buf *buf, const char *string);
int git_buf_printf(git_buf *buf, const char *format, ...) GIT_FORMAT_PRINTF(2, 3);
95
int git_buf_vprintf(git_buf *buf, const char *format, va_list ap);
96
void git_buf_clear(git_buf *buf);
97
void git_buf_consume(git_buf *buf, const char *end);
98
void git_buf_truncate(git_buf *buf, size_t len);
99
void git_buf_shorten(git_buf *buf, size_t amount);
100
void git_buf_rtruncate_at_char(git_buf *path, char separator);
101

102
/** General join with separator */
103
int git_buf_join_n(git_buf *buf, char separator, int nbuf, ...);
104
/** Fast join of two strings - first may legally point into `buf` data */
105
int git_buf_join(git_buf *buf, char separator, const char *str_a, const char *str_b);
106 107
/** Fast join of three strings - cannot reference `buf` data */
int git_buf_join3(git_buf *buf, char separator, const char *str_a, const char *str_b, const char *str_c);
108 109 110

/**
 * Join two strings as paths, inserting a slash between as needed.
111
 * @return 0 on success, -1 on failure
112
 */
113
GIT_INLINE(int) git_buf_joinpath(git_buf *buf, const char *a, const char *b)
114 115 116
{
	return git_buf_join(buf, '/', a, b);
}
117

118
GIT_INLINE(const char *) git_buf_cstr(const git_buf *buf)
119 120 121 122
{
	return buf->ptr;
}

123 124 125 126 127
GIT_INLINE(size_t) git_buf_len(const git_buf *buf)
{
	return buf->size;
}

128
void git_buf_copy_cstr(char *data, size_t datasize, const git_buf *buf);
129 130 131

#define git_buf_PUTS(buf, str) git_buf_put(buf, str, sizeof(str) - 1)

132
GIT_INLINE(ssize_t) git_buf_rfind_next(const git_buf *buf, char ch)
133
{
134
	ssize_t idx = (ssize_t)buf->size - 1;
135 136 137 138 139
	while (idx >= 0 && buf->ptr[idx] == ch) idx--;
	while (idx >= 0 && buf->ptr[idx] != ch) idx--;
	return idx;
}

140
GIT_INLINE(ssize_t) git_buf_rfind(const git_buf *buf, char ch)
141 142 143 144 145 146
{
	ssize_t idx = (ssize_t)buf->size - 1;
	while (idx >= 0 && buf->ptr[idx] != ch) idx--;
	return idx;
}

147
GIT_INLINE(ssize_t) git_buf_find(const git_buf *buf, char ch)
148
{
149 150
	void *found = memchr(buf->ptr, ch, buf->size);
	return found ? (ssize_t)((const char *)found - buf->ptr) : -1;
151 152
}

153 154 155
/* Remove whitespace from the end of the buffer */
void git_buf_rtrim(git_buf *buf);

156 157
int git_buf_cmp(const git_buf *a, const git_buf *b);

158 159 160
/* Write data as base64 encoded in buffer */
int git_buf_put_base64(git_buf *buf, const char *data, size_t len);

161 162 163
/* Write data as "base85" encoded in buffer */
int git_buf_put_base85(git_buf *buf, const char *data, size_t len);

164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
/*
 * Insert, remove or replace a portion of the buffer.
 *
 * @param buf The buffer to work with
 *
 * @param where The location in the buffer where the transformation
 * should be applied.
 *
 * @param nb_to_remove The number of chars to be removed. 0 to not
 * remove any character in the buffer.
 *
 * @param data A pointer to the data which should be inserted.
 *
 * @param nb_to_insert The number of chars to be inserted. 0 to not
 * insert any character from the buffer.
 *
 * @return 0 or an error code.
 */
int git_buf_splice(
	git_buf *buf,
	size_t where,
	size_t nb_to_remove,
	const char *data,
	size_t nb_to_insert);

189
#endif