summaryrefslogtreecommitdiff
path: root/src/chunk.h
blob: 364918de34ee8784a593e007a63cc1d38aaefde9 (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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#ifndef CMARK_CHUNK_H
#define CMARK_CHUNK_H

#include <string.h>
#include <stdlib.h>
#include <assert.h>
#include "cmark_ctype.h"
#include "buffer.h"

#define CMARK_CHUNK_EMPTY { NULL, 0, 0 }

typedef struct {
	unsigned char *data;
	bufsize_t len;
	bufsize_t alloc;  // also implies a NULL-terminated string
} cmark_chunk;

static inline void cmark_chunk_free(cmark_chunk *c)
{
	if (c->alloc)
		free(c->data);

	c->data = NULL;
	c->alloc = 0;
	c->len = 0;
}

static inline void cmark_chunk_ltrim(cmark_chunk *c)
{
	assert(!c->alloc);

	while (c->len && cmark_isspace(c->data[0])) {
		c->data++;
		c->len--;
	}
}

static inline void cmark_chunk_rtrim(cmark_chunk *c)
{
	while (c->len > 0) {
		if (!cmark_isspace(c->data[c->len - 1]))
			break;

		c->len--;
	}
}

static inline void cmark_chunk_trim(cmark_chunk *c)
{
	cmark_chunk_ltrim(c);
	cmark_chunk_rtrim(c);
}

static inline bufsize_t cmark_chunk_strchr(cmark_chunk *ch, int c, bufsize_t offset)
{
	const unsigned char *p = (unsigned char *)memchr(ch->data + offset, c, ch->len - offset);
	return p ? (bufsize_t)(p - ch->data) : ch->len;
}

static inline const char *cmark_chunk_to_cstr(cmark_chunk *c)
{
	unsigned char *str;

	if (c->alloc) {
		return (char *)c->data;
	}
	str = (unsigned char *)malloc(c->len + 1);
	if(str != NULL) {
		if(c->len > 0) {
			memcpy(str, c->data, c->len);
		}
		str[c->len] = 0;
	}
	c->data  = str;
	c->alloc = 1;

	return (char *)str;
}

static inline void cmark_chunk_set_cstr(cmark_chunk *c, const char *str)
{
	if (c->alloc) {
		free(c->data);
	}
	if (str == NULL) {
		c->len   = 0;
		c->data  = NULL;
		c->alloc = 0;
	} else {
		c->len   = strlen(str);
		c->data  = (unsigned char *)malloc(c->len + 1);
		c->alloc = 1;
		memcpy(c->data, str, c->len + 1);
	}
}

static inline cmark_chunk cmark_chunk_literal(const char *data)
{
	cmark_chunk c = {(unsigned char *)data, data ? strlen(data) : 0, 0};
	return c;
}

static inline cmark_chunk cmark_chunk_dup(const cmark_chunk *ch, bufsize_t pos, bufsize_t len)
{
	cmark_chunk c = {ch->data + pos, len, 0};
	return c;
}

static inline cmark_chunk cmark_chunk_buf_detach(cmark_strbuf *buf)
{
	cmark_chunk c;

	c.len = buf->size;
	c.data = cmark_strbuf_detach(buf);
	c.alloc = 1;

	return c;
}

#endif