2 96b8c570 2022-12-30 thomas * Copyright (c) 2022 Omar Polo <op@openbsd.org>
4 6509b181 2022-12-30 thomas * Permission to use, copy, modify, and distribute this software for any
5 6509b181 2022-12-30 thomas * purpose with or without fee is hereby granted, provided that the above
6 6509b181 2022-12-30 thomas * copyright notice and this permission notice appear in all copies.
8 6509b181 2022-12-30 thomas * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 6509b181 2022-12-30 thomas * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 6509b181 2022-12-30 thomas * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 6509b181 2022-12-30 thomas * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 6509b181 2022-12-30 thomas * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 6509b181 2022-12-30 thomas * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 6509b181 2022-12-30 thomas * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 6509b181 2022-12-30 thomas #include <ctype.h>
18 d8bf4f25 2023-09-14 thomas #include <stdarg.h>
19 6509b181 2022-12-30 thomas #include <stdio.h>
20 6509b181 2022-12-30 thomas #include <stdlib.h>
21 d8bf4f25 2023-09-14 thomas #include <string.h>
23 efc1d990 2023-11-25 thomas #include "got_compat.h"
24 6509b181 2022-12-30 thomas #include "tmpl.h"
27 d8bf4f25 2023-09-14 thomas tp_write(struct template *tp, const char *str, size_t len)
29 d8bf4f25 2023-09-14 thomas size_t avail;
31 d8bf4f25 2023-09-14 thomas while (len > 0) {
32 d8bf4f25 2023-09-14 thomas avail = tp->tp_cap - tp->tp_len;
33 d8bf4f25 2023-09-14 thomas if (avail == 0) {
34 d8bf4f25 2023-09-14 thomas if (template_flush(tp) == -1)
35 d8bf4f25 2023-09-14 thomas return (-1);
36 d8bf4f25 2023-09-14 thomas avail = tp->tp_cap;
39 d8bf4f25 2023-09-14 thomas if (len < avail)
40 d8bf4f25 2023-09-14 thomas avail = len;
42 d8bf4f25 2023-09-14 thomas memcpy(tp->tp_buf + tp->tp_len, str, avail);
43 d8bf4f25 2023-09-14 thomas tp->tp_len += avail;
44 d8bf4f25 2023-09-14 thomas str += avail;
45 d8bf4f25 2023-09-14 thomas len -= avail;
48 d8bf4f25 2023-09-14 thomas return (0);
52 d8bf4f25 2023-09-14 thomas tp_writes(struct template *tp, const char *str)
54 d8bf4f25 2023-09-14 thomas return (tp_write(tp, str, strlen(str)));
58 d8bf4f25 2023-09-14 thomas tp_writef(struct template *tp, const char *fmt, ...)
60 d8bf4f25 2023-09-14 thomas va_list ap;
61 d8bf4f25 2023-09-14 thomas char *str;
64 d8bf4f25 2023-09-14 thomas va_start(ap, fmt);
65 d8bf4f25 2023-09-14 thomas r = vasprintf(&str, fmt, ap);
66 d8bf4f25 2023-09-14 thomas va_end(ap);
67 d8bf4f25 2023-09-14 thomas if (r == -1)
68 d8bf4f25 2023-09-14 thomas return (-1);
69 d8bf4f25 2023-09-14 thomas r = tp_write(tp, str, r);
70 d8bf4f25 2023-09-14 thomas free(str);
71 d8bf4f25 2023-09-14 thomas return (r);
75 6509b181 2022-12-30 thomas tp_urlescape(struct template *tp, const char *str)
78 6509b181 2022-12-30 thomas char tmp[4];
80 6509b181 2022-12-30 thomas if (str == NULL)
81 6509b181 2022-12-30 thomas return (0);
83 6509b181 2022-12-30 thomas for (; *str; ++str) {
84 6509b181 2022-12-30 thomas if (iscntrl((unsigned char)*str) ||
85 6509b181 2022-12-30 thomas isspace((unsigned char)*str) ||
86 6509b181 2022-12-30 thomas *str == '\'' || *str == '"' || *str == '\\') {
87 6509b181 2022-12-30 thomas r = snprintf(tmp, sizeof(tmp), "%%%2X", *str);
88 6509b181 2022-12-30 thomas if (r < 0 || (size_t)r >= sizeof(tmp))
89 6509b181 2022-12-30 thomas return (0);
90 d8bf4f25 2023-09-14 thomas if (tp_write(tp, tmp, r) == -1)
91 6509b181 2022-12-30 thomas return (-1);
93 d8bf4f25 2023-09-14 thomas if (tp_write(tp, str, 1) == -1)
94 6509b181 2022-12-30 thomas return (-1);
98 6509b181 2022-12-30 thomas return (0);
102 6509b181 2022-12-30 thomas tp_htmlescape(struct template *tp, const char *str)
106 6509b181 2022-12-30 thomas if (str == NULL)
107 6509b181 2022-12-30 thomas return (0);
109 6509b181 2022-12-30 thomas for (; *str; ++str) {
110 6509b181 2022-12-30 thomas switch (*str) {
111 6509b181 2022-12-30 thomas case '<':
112 d8bf4f25 2023-09-14 thomas r = tp_write(tp, "<", 4);
114 6509b181 2022-12-30 thomas case '>':
115 d8bf4f25 2023-09-14 thomas r = tp_write(tp, ">", 4);
117 6509b181 2022-12-30 thomas case '&':
118 d8bf4f25 2023-09-14 thomas r = tp_write(tp, "&", 5);
120 6509b181 2022-12-30 thomas case '"':
121 d8bf4f25 2023-09-14 thomas r = tp_write(tp, """, 6);
123 6509b181 2022-12-30 thomas case '\'':
124 d8bf4f25 2023-09-14 thomas r = tp_write(tp, "'", 6);
127 d8bf4f25 2023-09-14 thomas r = tp_write(tp, str, 1);
131 6509b181 2022-12-30 thomas if (r == -1)
132 6509b181 2022-12-30 thomas return (-1);
135 6509b181 2022-12-30 thomas return (0);
138 6509b181 2022-12-30 thomas struct template *
139 d8bf4f25 2023-09-14 thomas template(void *arg, tmpl_write writefn, char *buf, size_t siz)
141 6509b181 2022-12-30 thomas struct template *tp;
143 6509b181 2022-12-30 thomas if ((tp = calloc(1, sizeof(*tp))) == NULL)
144 6509b181 2022-12-30 thomas return (NULL);
146 6509b181 2022-12-30 thomas tp->tp_arg = arg;
147 d8bf4f25 2023-09-14 thomas tp->tp_write = writefn;
148 d8bf4f25 2023-09-14 thomas tp->tp_buf = buf;
149 d8bf4f25 2023-09-14 thomas tp->tp_cap = siz;
151 6509b181 2022-12-30 thomas return (tp);
155 d8bf4f25 2023-09-14 thomas template_flush(struct template *tp)
157 d8bf4f25 2023-09-14 thomas if (tp->tp_len == 0)
158 d8bf4f25 2023-09-14 thomas return (0);
160 d8bf4f25 2023-09-14 thomas if (tp->tp_write(tp->tp_arg, tp->tp_buf, tp->tp_len) == -1)
161 d8bf4f25 2023-09-14 thomas return (-1);
162 d8bf4f25 2023-09-14 thomas tp->tp_len = 0;
163 d8bf4f25 2023-09-14 thomas return (0);
167 6509b181 2022-12-30 thomas template_free(struct template *tp)
169 6509b181 2022-12-30 thomas free(tp->tp_tmp);
170 6509b181 2022-12-30 thomas free(tp);
