1 014c66b6 2023-06-25 op /*
  2 014c66b6 2023-06-25 op  * Copyright (c) 2022 Omar Polo <op@openbsd.org>
  3 014c66b6 2023-06-25 op  *
  4 014c66b6 2023-06-25 op  * Permission to use, copy, modify, and distribute this software for any
  5 014c66b6 2023-06-25 op  * purpose with or without fee is hereby granted, provided that the above
  6 014c66b6 2023-06-25 op  * copyright notice and this permission notice appear in all copies.
  7 014c66b6 2023-06-25 op  *
  8 014c66b6 2023-06-25 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 014c66b6 2023-06-25 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 014c66b6 2023-06-25 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 014c66b6 2023-06-25 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 014c66b6 2023-06-25 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 014c66b6 2023-06-25 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 014c66b6 2023-06-25 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 014c66b6 2023-06-25 op  */
 16 014c66b6 2023-06-25 op 
 17 014c66b6 2023-06-25 op #include <ctype.h>
 18 014c66b6 2023-06-25 op #include <stdio.h>
 19 014c66b6 2023-06-25 op #include <stdlib.h>
 20 014c66b6 2023-06-25 op 
 21 014c66b6 2023-06-25 op #include "tmpl.h"
 22 014c66b6 2023-06-25 op 
 23 014c66b6 2023-06-25 op int
 24 014c66b6 2023-06-25 op tp_urlescape(struct template *tp, const char *str)
 25 014c66b6 2023-06-25 op {
 26 014c66b6 2023-06-25 op 	int	 r;
 27 014c66b6 2023-06-25 op 	char	 tmp[4];
 28 014c66b6 2023-06-25 op 
 29 014c66b6 2023-06-25 op 	if (str == NULL)
 30 014c66b6 2023-06-25 op 		return (0);
 31 014c66b6 2023-06-25 op 
 32 014c66b6 2023-06-25 op 	for (; *str; ++str) {
 33 014c66b6 2023-06-25 op 		if (iscntrl((unsigned char)*str) ||
 34 014c66b6 2023-06-25 op 		    isspace((unsigned char)*str) ||
 35 014c66b6 2023-06-25 op 		    *str == '\'' || *str == '"' || *str == '\\') {
 36 014c66b6 2023-06-25 op 			r = snprintf(tmp, sizeof(tmp), "%%%2X", *str);
 37 014c66b6 2023-06-25 op 			if (r < 0  || (size_t)r >= sizeof(tmp))
 38 014c66b6 2023-06-25 op 				return (0);
 39 014c66b6 2023-06-25 op 			if (tp->tp_puts(tp, tmp) == -1)
 40 014c66b6 2023-06-25 op 				return (-1);
 41 014c66b6 2023-06-25 op 		} else {
 42 014c66b6 2023-06-25 op 			if (tp->tp_putc(tp, *str) == -1)
 43 014c66b6 2023-06-25 op 				return (-1);
 44 014c66b6 2023-06-25 op 		}
 45 014c66b6 2023-06-25 op 	}
 46 014c66b6 2023-06-25 op 
 47 014c66b6 2023-06-25 op 	return (0);
 48 014c66b6 2023-06-25 op }
 49 014c66b6 2023-06-25 op 
 50 014c66b6 2023-06-25 op int
 51 014c66b6 2023-06-25 op tp_htmlescape(struct template *tp, const char *str)
 52 014c66b6 2023-06-25 op {
 53 014c66b6 2023-06-25 op 	int r;
 54 014c66b6 2023-06-25 op 
 55 014c66b6 2023-06-25 op 	if (str == NULL)
 56 014c66b6 2023-06-25 op 		return (0);
 57 014c66b6 2023-06-25 op 
 58 014c66b6 2023-06-25 op 	for (; *str; ++str) {
 59 014c66b6 2023-06-25 op 		switch (*str) {
 60 014c66b6 2023-06-25 op 		case '<':
 61 014c66b6 2023-06-25 op 			r = tp->tp_puts(tp, "&lt;");
 62 014c66b6 2023-06-25 op 			break;
 63 014c66b6 2023-06-25 op 		case '>':
 64 014c66b6 2023-06-25 op 			r = tp->tp_puts(tp, "&gt;");
 65 014c66b6 2023-06-25 op 			break;
 66 014c66b6 2023-06-25 op 		case '&':
 67 014c66b6 2023-06-25 op 			r = tp->tp_puts(tp, "&amp;");
 68 014c66b6 2023-06-25 op 			break;
 69 014c66b6 2023-06-25 op 		case '"':
 70 014c66b6 2023-06-25 op 			r = tp->tp_puts(tp, "&quot;");
 71 014c66b6 2023-06-25 op 			break;
 72 014c66b6 2023-06-25 op 		case '\'':
 73 014c66b6 2023-06-25 op 			r = tp->tp_puts(tp, "&apos;");
 74 014c66b6 2023-06-25 op 			break;
 75 014c66b6 2023-06-25 op 		default:
 76 014c66b6 2023-06-25 op 			r = tp->tp_putc(tp, *str);
 77 014c66b6 2023-06-25 op 			break;
 78 014c66b6 2023-06-25 op 		}
 79 014c66b6 2023-06-25 op 
 80 014c66b6 2023-06-25 op 		if (r == -1)
 81 014c66b6 2023-06-25 op 			return (-1);
 82 014c66b6 2023-06-25 op 	}
 83 014c66b6 2023-06-25 op 
 84 014c66b6 2023-06-25 op 	return (0);
 85 014c66b6 2023-06-25 op }
 86 014c66b6 2023-06-25 op 
 87 014c66b6 2023-06-25 op struct template *
 88 014c66b6 2023-06-25 op template(void *arg, tmpl_puts putsfn, tmpl_putc putcfn)
 89 014c66b6 2023-06-25 op {
 90 014c66b6 2023-06-25 op 	struct template *tp;
 91 014c66b6 2023-06-25 op 
 92 014c66b6 2023-06-25 op 	if ((tp = calloc(1, sizeof(*tp))) == NULL)
 93 014c66b6 2023-06-25 op 		return (NULL);
 94 014c66b6 2023-06-25 op 
 95 014c66b6 2023-06-25 op 	tp->tp_arg = arg;
 96 014c66b6 2023-06-25 op 	tp->tp_escape = tp_htmlescape;
 97 014c66b6 2023-06-25 op 	tp->tp_puts = putsfn;
 98 014c66b6 2023-06-25 op 	tp->tp_putc = putcfn;
 99 014c66b6 2023-06-25 op 
100 014c66b6 2023-06-25 op 	return (tp);
101 014c66b6 2023-06-25 op }
102 014c66b6 2023-06-25 op 
103 014c66b6 2023-06-25 op void
104 014c66b6 2023-06-25 op template_free(struct template *tp)
105 014c66b6 2023-06-25 op {
106 014c66b6 2023-06-25 op 	free(tp->tp_tmp);
107 014c66b6 2023-06-25 op 	free(tp);
108 014c66b6 2023-06-25 op }