summaryrefslogtreecommitdiff
path: root/src/html/houdini_html_u.c
blob: b88b9d1471eb5127149eea845b992ee8147f2c9a (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
#include <assert.h>
#include <stdio.h>
#include <string.h>

#include "buffer.h"
#include "houdini.h"
#include "utf8.h"
#include "html_unescape.h"

size_t
houdini_unescape_ent(strbuf *ob, const uint8_t *src, size_t size)
{
	size_t i = 0;

	if (size > 3 && src[0] == '#') {
		int codepoint = 0;

		if (_isdigit(src[1])) {
			for (i = 1; i < size && _isdigit(src[i]); ++i) {
				int cp = (codepoint * 10) + (src[i] - '0');

				if (cp < codepoint)
					return 0;

				codepoint = cp;
			}
		}

		else if (src[1] == 'x' || src[1] == 'X') {
			for (i = 2; i < size && _isxdigit(src[i]); ++i) {
				int cp = (codepoint * 16) + ((src[i] | 32) % 39 - 9);

				if (cp < codepoint)
					return 0;

				codepoint = cp;
			}
		}

		if (i < size && src[i] == ';' && codepoint) {
			utf8proc_encode_char(codepoint, ob);
			return i + 1;
		}
	}

	else {
		if (size > MAX_WORD_LENGTH)
			size = MAX_WORD_LENGTH;

		for (i = MIN_WORD_LENGTH; i < size; ++i) {
			if (src[i] == ' ')
				break;

			if (src[i] == ';') {
				const struct html_ent *entity = find_entity((char *)src, i);

				if (entity != NULL) {
					strbuf_put(ob, entity->utf8, entity->utf8_len);
					return i + 1;
				}

				break;
			}
		}
	}

	return 0;
}

int
houdini_unescape_html(strbuf *ob, const uint8_t *src, size_t size)
{
	size_t  i = 0, org, ent;

	while (i < size) {
		org = i;
		while (i < size && src[i] != '&')
			i++;

		if (likely(i > org)) {
			if (unlikely(org == 0)) {
				if (i >= size)
					return 0;

				strbuf_grow(ob, HOUDINI_UNESCAPED_SIZE(size));
			}

			strbuf_put(ob, src + org, i - org);
		}

		/* escaping */
		if (i >= size)
			break;

		i++;

		ent = houdini_unescape_ent(ob, src + i, size - i);
		i += ent;

		/* not really an entity */
		if (ent == 0)
			strbuf_putc(ob, '&');
	}

	return 1;
}

void houdini_unescape_html_f(strbuf *ob, const uint8_t *src, size_t size)
{
	if (!houdini_unescape_html(ob, src, size))
		strbuf_put(ob, src, size);
}