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