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 6509b181 2022-12-30 thomas #include <stdio.h>
19 6509b181 2022-12-30 thomas #include <stdlib.h>
21 6509b181 2022-12-30 thomas #include "tmpl.h"
24 6509b181 2022-12-30 thomas tp_urlescape(struct template *tp, const char *str)
27 6509b181 2022-12-30 thomas char tmp[4];
29 6509b181 2022-12-30 thomas if (str == NULL)
30 6509b181 2022-12-30 thomas return (0);
32 6509b181 2022-12-30 thomas for (; *str; ++str) {
33 6509b181 2022-12-30 thomas if (iscntrl((unsigned char)*str) ||
34 6509b181 2022-12-30 thomas isspace((unsigned char)*str) ||
35 6509b181 2022-12-30 thomas *str == '\'' || *str == '"' || *str == '\\') {
36 6509b181 2022-12-30 thomas r = snprintf(tmp, sizeof(tmp), "%%%2X", *str);
37 6509b181 2022-12-30 thomas if (r < 0 || (size_t)r >= sizeof(tmp))
38 6509b181 2022-12-30 thomas return (0);
39 6509b181 2022-12-30 thomas if (tp->tp_puts(tp, tmp) == -1)
40 6509b181 2022-12-30 thomas return (-1);
42 6509b181 2022-12-30 thomas if (tp->tp_putc(tp, *str) == -1)
43 6509b181 2022-12-30 thomas return (-1);
47 6509b181 2022-12-30 thomas return (0);
51 6509b181 2022-12-30 thomas tp_htmlescape(struct template *tp, const char *str)
55 6509b181 2022-12-30 thomas if (str == NULL)
56 6509b181 2022-12-30 thomas return (0);
58 6509b181 2022-12-30 thomas for (; *str; ++str) {
59 6509b181 2022-12-30 thomas switch (*str) {
61 6509b181 2022-12-30 thomas r = tp->tp_puts(tp, "<");
64 6509b181 2022-12-30 thomas r = tp->tp_puts(tp, ">");
67 6509b181 2022-12-30 thomas r = tp->tp_puts(tp, "&");
70 6509b181 2022-12-30 thomas r = tp->tp_puts(tp, """);
72 6509b181 2022-12-30 thomas case '\'':
73 6509b181 2022-12-30 thomas r = tp->tp_puts(tp, "'");
76 6509b181 2022-12-30 thomas r = tp->tp_putc(tp, *str);
80 6509b181 2022-12-30 thomas if (r == -1)
81 6509b181 2022-12-30 thomas return (-1);
84 6509b181 2022-12-30 thomas return (0);
87 6509b181 2022-12-30 thomas struct template *
88 6509b181 2022-12-30 thomas template(void *arg, tmpl_puts putsfn, tmpl_putc putcfn)
90 6509b181 2022-12-30 thomas struct template *tp;
92 6509b181 2022-12-30 thomas if ((tp = calloc(1, sizeof(*tp))) == NULL)
93 6509b181 2022-12-30 thomas return (NULL);
95 6509b181 2022-12-30 thomas tp->tp_arg = arg;
96 6509b181 2022-12-30 thomas tp->tp_escape = tp_htmlescape;
97 6509b181 2022-12-30 thomas tp->tp_puts = putsfn;
98 6509b181 2022-12-30 thomas tp->tp_putc = putcfn;
100 6509b181 2022-12-30 thomas return (tp);
104 6509b181 2022-12-30 thomas template_free(struct template *tp)
106 6509b181 2022-12-30 thomas free(tp->tp_tmp);
107 6509b181 2022-12-30 thomas free(tp);