1 /*
2  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include "telescope.h"
18 
19 #include <stdlib.h>
20 #include <string.h>
21 
22 int
23 parser_append(struct parser *p, const char *buf, size_t len)
24 {
25 	size_t newlen;
26 	char *t;
27 
28 	newlen = len + p->len;
29 	if ((t = calloc(1, newlen)) == NULL)
30 		return 0;
31 	memcpy(t, p->buf, p->len);
32 	memcpy(t + p->len, buf, len);
33 	free(p->buf);
34 	p->buf = t;
35 	p->len = newlen;
36 	return 1;
37 }
38 
39 int
40 parser_set_buf(struct parser *p, const char *buf, size_t len)
41 {
42 	char *tmp;
43 
44 	if (len == 0) {
45 		p->len = 0;
46 		free(p->buf);
47 		p->buf = NULL;
48 		return 1;
49 	}
50 
51 	/*
52 	 * p->buf and buf can (and probably almost always will)
53 	 * overlap!
54 	 */
55 
56 	if ((tmp = calloc(1, len)) == NULL)
57 		return 0;
58 	memcpy(tmp, buf, len);
59 	free(p->buf);
60 	p->buf = tmp;
61 	p->len = len;
62 	return 1;
63 }
64 
65 int
66 parser_foreach_line(struct parser *p, const char *buf, size_t size,
67     parsechunkfn fn)
68 {
69 	char		*b, *e;
70 	unsigned int	 ch;
71 	size_t		 i, l, len;
72 
73 	if (!parser_append(p, buf, size))
74 		return 0;
75 	b = p->buf;
76 	len = p->len;
77 
78 	/* drop every "funny" ASCII character */
79 	for (i = 0; i < len; ) {
80 		ch = b[i];
81 		if ((ch >= ' ' || ch == '\n' || ch == '\t')
82 		    && ch != 127) { /* del */
83 			++i;
84 			continue;
85 		}
86 		memmove(&b[i], &b[i+1], len - i);
87 		len--;
88 	}
89 
90 	while (len > 0) {
91 		if ((e = memmem((char*)b, len, "\n", 1)) == NULL)
92 			break;
93 		l = e - b;
94 
95 		if (!fn(p, b, l))
96 			return 0;
97 
98 		len -= l;
99 		b += l;
100 
101 		if (len > 0) {
102 			/* skip \n */
103 			len--;
104 			b++;
105 		}
106 	}
107 
108 	return parser_set_buf(p, b, len);
109 }