util.h 6.89 KB
Newer Older
Vicent Marti committed
1
/*
schu committed
2
 * Copyright (C) 2009-2012 the libgit2 contributors
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_util_h__
#define INCLUDE_util_h__

#define ARRAY_SIZE(x) (sizeof(x)/sizeof(x[0]))
Vicent Marti committed
11
#define bitsizeof(x) (CHAR_BIT * sizeof(x))
Vicent Marti committed
12
#define MSB(x, bits) ((x) & (~0ULL << (bitsizeof(x) - (bits))))
13 14 15
#ifndef min
# define min(a,b) ((a) < (b) ? (a) : (b))
#endif
16

17
/*
18 19 20
 * Custom memory allocation wrappers
 * that set error code and error message
 * on allocation failure
Vicent Marti committed
21
 */
22 23 24
GIT_INLINE(void *) git__malloc(size_t len)
{
	void *ptr = malloc(len);
25
	if (!ptr) giterr_set_oom();
26 27 28 29 30 31
	return ptr;
}

GIT_INLINE(void *) git__calloc(size_t nelem, size_t elsize)
{
	void *ptr = calloc(nelem, elsize);
32
	if (!ptr) giterr_set_oom();
33 34 35 36 37 38
	return ptr;
}

GIT_INLINE(char *) git__strdup(const char *str)
{
	char *ptr = strdup(str);
39
	if (!ptr) giterr_set_oom();
40 41 42
	return ptr;
}

43 44
GIT_INLINE(char *) git__strndup(const char *str, size_t n)
{
45
	size_t length = 0;
46 47
	char *ptr;

48 49
	while (length < n && str[length])
		++length;
50

51
	ptr = (char*)malloc(length + 1);
52
	if (!ptr) {
53
		giterr_set_oom();
54 55
		return NULL;
	}
56

57 58 59
	if (length)
		memcpy(ptr, str, length);

60
	ptr[length] = '\0';
61 62 63 64

	return ptr;
}

65 66 67
GIT_INLINE(void *) git__realloc(void *ptr, size_t size)
{
	void *new_ptr = realloc(ptr, size);
68
	if (!new_ptr) giterr_set_oom();
69 70
	return new_ptr;
}
71

72 73
#define git__free(ptr) free(ptr)

74 75 76 77 78 79
#define STRCMP_CASESELECT(IGNORE_CASE, STR1, STR2) \
	((IGNORE_CASE) ? strcasecmp((STR1), (STR2)) : strcmp((STR1), (STR2)))

#define CASESELECT(IGNORE_CASE, ICASE, CASE) \
	((IGNORE_CASE) ? (ICASE) : (CASE))

80
extern int git__prefixcmp(const char *str, const char *prefix);
81
extern int git__prefixcmp_icase(const char *str, const char *prefix);
82
extern int git__suffixcmp(const char *str, const char *suffix);
83

84 85 86 87 88
GIT_INLINE(int) git__signum(int val)
{
	return ((val > 0) - (val < 0));
}

89 90
extern int git__strtol32(int32_t *n, const char *buff, const char **end_buf, int base);
extern int git__strtol64(int64_t *n, const char *buff, const char **end_buf, int base);
91

92
extern void git__hexdump(const char *buffer, size_t n);
93 94
extern uint32_t git__hash(const void *key, int len, uint32_t seed);

95
/** @return true if p fits into the range of a size_t */
96
GIT_INLINE(int) git__is_sizet(git_off_t p)
97 98
{
	size_t r = (size_t)p;
99
	return p == (git_off_t)r;
100 101
}

102 103 104 105 106 107 108
/* 32-bit cross-platform rotl */
#ifdef _MSC_VER /* use built-in method in MSVC */
#	define git__rotl(v, s) (uint32_t)_rotl(v, s)
#else /* use bitops in GCC; with o2 this gets optimized to a rotl instruction */
#	define git__rotl(v, s) (uint32_t)(((uint32_t)(v) << (s)) | ((uint32_t)(v) >> (32 - (s))))
#endif

109
extern char *git__strtok(char **end, const char *sep);
110
extern char *git__strsep(char **end, const char *sep);
111

112
extern void git__strntolower(char *str, size_t len);
113 114
extern void git__strtolower(char *str);

115 116 117
GIT_INLINE(const char *) git__next_line(const char *s)
{
	while (*s && *s != '\n') s++;
Russell Belfer committed
118
	while (*s == '\n' || *s == '\r') s++;
119 120 121
	return s;
}

122
extern void git__tsort(void **dst, size_t size, int (*cmp)(const void *, const void *));
123

124 125 126 127 128
/**
 * @param position If non-NULL, this will be set to the position where the
 * 		element is or would be inserted if not found.
 * @return pos (>=0) if found or -1 if not found
 */
129 130 131 132 133 134
extern int git__bsearch(
	void **array,
	size_t array_len,
	const void *key,
	int (*compare)(const void *, const void *),
	size_t *position);
135

136 137
extern int git__strcmp_cb(const void *a, const void *b);

138 139 140 141 142
extern int git__strcmp(const char *a, const char *b);
extern int git__strcasecmp(const char *a, const char *b);
extern int git__strncmp(const char *a, const char *b, size_t sz);
extern int git__strncasecmp(const char *a, const char *b, size_t sz);

143 144 145 146 147 148 149 150 151 152 153 154 155 156
typedef struct {
	short refcount;
	void *owner;
} git_refcount;

typedef void (*git_refcount_freeptr)(void *r);

#define GIT_REFCOUNT_INC(r) { \
	((git_refcount *)(r))->refcount++; \
}

#define GIT_REFCOUNT_DEC(_r, do_free) { \
	git_refcount *r = (git_refcount *)(_r); \
	r->refcount--; \
157
	if (r->refcount <= 0 && r->owner == NULL) { do_free(_r); } \
158 159 160 161 162 163 164 165
}

#define GIT_REFCOUNT_OWN(r, o) { \
	((git_refcount *)(r))->owner = o; \
}

#define GIT_REFCOUNT_OWNER(r) (((git_refcount *)(r))->owner)

nulltoken committed
166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
static signed char from_hex[] = {
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 00 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 10 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 20 */
 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, -1, -1, -1, -1, -1, -1, /* 30 */
-1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 40 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 50 */
-1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 60 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 70 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 80 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 90 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* a0 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* b0 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* c0 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* d0 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* e0 */
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* f0 */
};

GIT_INLINE(int) git__fromhex(char h)
{
	return from_hex[(unsigned char) h];
}
189

schu committed
190 191 192 193 194 195 196 197 198
GIT_INLINE(int) git__ishex(const char *str)
{
	unsigned i;
	for (i=0; i<strlen(str); i++)
		if (git__fromhex(str[i]) < 0)
			return 0;
	return 1;
}

199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215
GIT_INLINE(size_t) git__size_t_bitmask(size_t v)
{
	v--;
	v |= v >> 1;
	v |= v >> 2;
	v |= v >> 4;
	v |= v >> 8;
	v |= v >> 16;

	return v;
}

GIT_INLINE(size_t) git__size_t_powerof2(size_t v)
{
	return git__size_t_bitmask(v) + 1;
}

216 217 218 219 220 221 222 223 224 225
GIT_INLINE(bool) git__isupper(int c)
{
    return (c >= 'A' && c <= 'Z');
}

GIT_INLINE(bool) git__isalpha(int c)
{
    return ((c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z'));
}

nulltoken committed
226 227 228 229 230
GIT_INLINE(bool) git__isdigit(int c)
{
    return (c >= '0' && c <= '9');
}

231 232
GIT_INLINE(bool) git__isspace(int c)
{
233
    return (c == ' ' || c == '\t' || c == '\n' || c == '\f' || c == '\r' || c == '\v' || c == 0x85 /* Unicode CR+LF */);
234 235
}

236 237 238 239 240
GIT_INLINE(bool) git__iswildcard(int c)
{
	return (c == '*' || c == '?' || c == '[');
}

241 242 243 244 245 246 247 248 249
/*
 * Parse a string value as a boolean, just like Core Git
 * does.
 *
 * Valid values for true are: 'true', 'yes', 'on'
 * Valid values for false are: 'false', 'no', 'off'
 */
extern int git__parse_bool(int *out, const char *value);

250 251 252 253 254 255 256 257 258 259
/*
 * Parse a string into a value as a git_time_t.
 *
 * Sample valid input:
 * - "yesterday"
 * - "July 17, 2003"
 * - "2003-7-17 08:23"
 */
int git__date_parse(git_time_t *out, const char *date);

260 261 262 263 264 265 266 267 268
/*
 * Unescapes a string in-place.
 * 
 * Edge cases behavior:
 * - "jackie\" -> "jacky\"
 * - "chan\\" -> "chan\"
 */
extern size_t git__unescape(char *str);

269
#endif /* INCLUDE_util_h__ */