xref: /freebsd/contrib/libedit/eln.c (revision 5bb3134a8c21cb87b30e135ef168483f0333dabb)
1 /*	$NetBSD: eln.c,v 1.36 2021/08/15 10:08:41 christos Exp $	*/
2 
3 /*-
4  * Copyright (c) 2009 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26  * POSSIBILITY OF SUCH DAMAGE.
27  */
28 #include "config.h"
29 #if !defined(lint) && !defined(SCCSID)
30 __RCSID("$NetBSD: eln.c,v 1.36 2021/08/15 10:08:41 christos Exp $");
31 #endif /* not lint && not SCCSID */
32 
33 #include <errno.h>
34 #include <stdarg.h>
35 #include <stdio.h>
36 #include <stdlib.h>
37 
38 #include "el.h"
39 
40 int
41 el_getc(EditLine *el, char *cp)
42 {
43 	int num_read;
44 	wchar_t wc = 0;
45 
46 	num_read = el_wgetc(el, &wc);
47 	*cp = '\0';
48 	if (num_read <= 0)
49 		return num_read;
50 	num_read = wctob(wc);
51 	if (num_read == EOF) {
52 		errno = ERANGE;
53 		return -1;
54 	} else {
55 		*cp = (char)num_read;
56 		return 1;
57 	}
58 }
59 
60 
61 void
62 el_push(EditLine *el, const char *str)
63 {
64 	/* Using multibyte->wide string decoding works fine under single-byte
65 	 * character sets too, and Does The Right Thing. */
66 	el_wpush(el, ct_decode_string(str, &el->el_lgcyconv));
67 }
68 
69 
70 const char *
71 el_gets(EditLine *el, int *nread)
72 {
73 	const wchar_t *tmp;
74 
75 	tmp = el_wgets(el, nread);
76 	if (tmp != NULL) {
77 	    int i;
78 	    size_t nwread = 0;
79 
80 	    for (i = 0; i < *nread; i++)
81 		nwread += ct_enc_width(tmp[i]);
82 	    *nread = (int)nwread;
83 	}
84 	return ct_encode_string(tmp, &el->el_lgcyconv);
85 }
86 
87 
88 int
89 el_parse(EditLine *el, int argc, const char *argv[])
90 {
91 	int ret;
92 	const wchar_t **wargv;
93 
94 	wargv = (void *)ct_decode_argv(argc, argv, &el->el_lgcyconv);
95 	if (!wargv)
96 		return -1;
97 	ret = el_wparse(el, argc, wargv);
98 	el_free(wargv);
99 
100 	return ret;
101 }
102 
103 
104 int
105 el_set(EditLine *el, int op, ...)
106 {
107 	va_list ap;
108 	int ret;
109 
110 	if (!el)
111 		return -1;
112 	va_start(ap, op);
113 
114 	switch (op) {
115 	case EL_PROMPT:         /* el_pfunc_t */
116 	case EL_RPROMPT: {
117 		el_pfunc_t p = va_arg(ap, el_pfunc_t);
118 		ret = prompt_set(el, p, 0, op, 0);
119 		break;
120 	}
121 
122 	case EL_RESIZE: {
123 		el_zfunc_t p = va_arg(ap, el_zfunc_t);
124 		void *arg = va_arg(ap, void *);
125 		ret = ch_resizefun(el, p, arg);
126 		break;
127 	}
128 
129 	case EL_ALIAS_TEXT: {
130 		el_afunc_t p = va_arg(ap, el_afunc_t);
131 		void *arg = va_arg(ap, void *);
132 		ret = ch_aliasfun(el, p, arg);
133 		break;
134 	}
135 
136 	case EL_PROMPT_ESC:
137 	case EL_RPROMPT_ESC: {
138 		el_pfunc_t p = va_arg(ap, el_pfunc_t);
139 		int c = va_arg(ap, int);
140 
141 		ret = prompt_set(el, p, c, op, 0);
142 		break;
143 	}
144 
145 	case EL_TERMINAL:       /* const char * */
146 		ret = el_wset(el, op, va_arg(ap, char *));
147 		break;
148 
149 	case EL_EDITOR:		/* const wchar_t * */
150 		ret = el_wset(el, op, ct_decode_string(va_arg(ap, char *),
151 		    &el->el_lgcyconv));
152 		break;
153 
154 	case EL_SIGNAL:         /* int */
155 	case EL_EDITMODE:
156 	case EL_SAFEREAD:
157 	case EL_UNBUFFERED:
158 	case EL_PREP_TERM:
159 		ret = el_wset(el, op, va_arg(ap, int));
160 		break;
161 
162 	case EL_BIND:   /* const char * list -> const wchar_t * list */
163 	case EL_TELLTC:
164 	case EL_SETTC:
165 	case EL_ECHOTC:
166 	case EL_SETTY: {
167 		const char *argv[20];
168 		int i;
169 		const wchar_t **wargv;
170 		for (i = 1; i < (int)__arraycount(argv) - 1; ++i)
171 			if ((argv[i] = va_arg(ap, const char *)) == NULL)
172 			    break;
173 		argv[0] = argv[i] = NULL;
174 		wargv = (void *)ct_decode_argv(i + 1, argv, &el->el_lgcyconv);
175 		if (!wargv) {
176 		    ret = -1;
177 		    goto out;
178 		}
179 		/*
180 		 * AFAIK we can't portably pass through our new wargv to
181 		 * el_wset(), so we have to reimplement the body of
182 		 * el_wset() for these ops.
183 		 */
184 		switch (op) {
185 		case EL_BIND:
186 			wargv[0] = L"bind";
187 			ret = map_bind(el, i, wargv);
188 			break;
189 		case EL_TELLTC:
190 			wargv[0] = L"telltc";
191 			ret = terminal_telltc(el, i, wargv);
192 			break;
193 		case EL_SETTC:
194 			wargv[0] = L"settc";
195 			ret = terminal_settc(el, i, wargv);
196 			break;
197 		case EL_ECHOTC:
198 			wargv[0] = L"echotc";
199 			ret = terminal_echotc(el, i, wargv);
200 			break;
201 		case EL_SETTY:
202 			wargv[0] = L"setty";
203 			ret = tty_stty(el, i, wargv);
204 			break;
205 		default:
206 			ret = -1;
207 		}
208 		el_free(wargv);
209 		break;
210 	}
211 
212 	/* XXX: do we need to change el_func_t too? */
213 	case EL_ADDFN: {          /* const char *, const char *, el_func_t */
214 		const char *args[2];
215 		el_func_t func;
216 		wchar_t **wargv;
217 
218 		args[0] = va_arg(ap, const char *);
219 		args[1] = va_arg(ap, const char *);
220 		func = va_arg(ap, el_func_t);
221 
222 		wargv = ct_decode_argv(2, args, &el->el_lgcyconv);
223 		if (!wargv) {
224 		    ret = -1;
225 		    goto out;
226 		}
227 		/* XXX: The two strdup's leak */
228 		ret = map_addfunc(el, wcsdup(wargv[0]), wcsdup(wargv[1]),
229 		    func);
230 		el_free(wargv);
231 		break;
232 	}
233 	case EL_HIST: {           /* hist_fun_t, const char * */
234 		hist_fun_t fun = va_arg(ap, hist_fun_t);
235 		void *ptr = va_arg(ap, void *);
236 		ret = hist_set(el, fun, ptr);
237 		el->el_flags |= NARROW_HISTORY;
238 		break;
239 	}
240 
241 	case EL_GETCFN:         /* el_rfunc_t */
242 		ret = el_wset(el, op, va_arg(ap, el_rfunc_t));
243 		break;
244 
245 	case EL_CLIENTDATA:     /* void * */
246 		ret = el_wset(el, op, va_arg(ap, void *));
247 		break;
248 
249 	case EL_SETFP: {          /* int, FILE * */
250 		int what = va_arg(ap, int);
251 		FILE *fp = va_arg(ap, FILE *);
252 		ret = el_wset(el, op, what, fp);
253 		break;
254 	}
255 
256 	case EL_REFRESH:
257 		re_clear_display(el);
258 		re_refresh(el);
259 		terminal__flush(el);
260 		ret = 0;
261 		break;
262 
263 	default:
264 		ret = -1;
265 		break;
266 	}
267 
268 out:
269 	va_end(ap);
270 	return ret;
271 }
272 
273 
274 int
275 el_get(EditLine *el, int op, ...)
276 {
277 	va_list ap;
278 	int ret;
279 
280 	if (!el)
281 		return -1;
282 
283 	va_start(ap, op);
284 
285 	switch (op) {
286 	case EL_PROMPT:         /* el_pfunc_t * */
287 	case EL_RPROMPT: {
288 		el_pfunc_t *p = va_arg(ap, el_pfunc_t *);
289 		ret = prompt_get(el, p, 0, op);
290 		break;
291 	}
292 
293 	case EL_PROMPT_ESC: /* el_pfunc_t *, char **/
294 	case EL_RPROMPT_ESC: {
295 		el_pfunc_t *p = va_arg(ap, el_pfunc_t *);
296 		char *c = va_arg(ap, char *);
297 		wchar_t wc = 0;
298 		ret = prompt_get(el, p, &wc, op);
299 		*c = (char)wc;
300 		break;
301 	}
302 
303 	case EL_EDITOR: {
304 		const char **p = va_arg(ap, const char **);
305 		const wchar_t *pw;
306 		ret = el_wget(el, op, &pw);
307 		*p = ct_encode_string(pw, &el->el_lgcyconv);
308 		if (!el->el_lgcyconv.csize)
309 			ret = -1;
310 		break;
311 	}
312 
313 	case EL_TERMINAL:       /* const char ** */
314 		ret = el_wget(el, op, va_arg(ap, const char **));
315 		break;
316 
317 	case EL_SIGNAL:         /* int * */
318 	case EL_EDITMODE:
319 	case EL_SAFEREAD:
320 	case EL_UNBUFFERED:
321 	case EL_PREP_TERM:
322 		ret = el_wget(el, op, va_arg(ap, int *));
323 		break;
324 
325 	case EL_GETTC: {
326 		char *argv[3];
327 		static char gettc[] = "gettc";
328 		argv[0] = gettc;
329 		argv[1] = va_arg(ap, char *);
330 		argv[2] = va_arg(ap, void *);
331 		ret = terminal_gettc(el, 3, argv);
332 		break;
333 	}
334 
335 	case EL_GETCFN:         /* el_rfunc_t */
336 		ret = el_wget(el, op, va_arg(ap, el_rfunc_t *));
337 		break;
338 
339 	case EL_CLIENTDATA:     /* void ** */
340 		ret = el_wget(el, op, va_arg(ap, void **));
341 		break;
342 
343 	case EL_GETFP: {          /* int, FILE ** */
344 		int what = va_arg(ap, int);
345 		FILE **fpp = va_arg(ap, FILE **);
346 		ret = el_wget(el, op, what, fpp);
347 		break;
348 	}
349 
350 	default:
351 		ret = -1;
352 		break;
353 	}
354 
355 	va_end(ap);
356 	return ret;
357 }
358 
359 
360 const LineInfo *
361 el_line(EditLine *el)
362 {
363 	const LineInfoW *winfo = el_wline(el);
364 	LineInfo *info = &el->el_lgcylinfo;
365 	size_t offset;
366 	const wchar_t *p;
367 
368 	info->buffer   = ct_encode_string(winfo->buffer, &el->el_lgcyconv);
369 
370 	offset = 0;
371 	for (p = winfo->buffer; p < winfo->cursor; p++)
372 		offset += ct_enc_width(*p);
373 	info->cursor = info->buffer + offset;
374 
375 	offset = 0;
376 	for (p = winfo->buffer; p < winfo->lastchar; p++)
377 		offset += ct_enc_width(*p);
378 	info->lastchar = info->buffer + offset;
379 
380 	return info;
381 }
382 
383 
384 int
385 el_insertstr(EditLine *el, const char *str)
386 {
387 	return el_winsertstr(el, ct_decode_string(str, &el->el_lgcyconv));
388 }
389