xref: /freebsd/contrib/libedit/eln.c (revision 732a02b4e77866604a120a275c082bb6221bd2ff)
1 /*	$NetBSD: eln.c,v 1.35 2019/04/26 16:56:57 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.35 2019/04/26 16:56:57 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_UNBUFFERED:
157 	case EL_PREP_TERM:
158 		ret = el_wset(el, op, va_arg(ap, int));
159 		break;
160 
161 	case EL_BIND:   /* const char * list -> const wchar_t * list */
162 	case EL_TELLTC:
163 	case EL_SETTC:
164 	case EL_ECHOTC:
165 	case EL_SETTY: {
166 		const char *argv[20];
167 		int i;
168 		const wchar_t **wargv;
169 		for (i = 1; i < (int)__arraycount(argv) - 1; ++i)
170 			if ((argv[i] = va_arg(ap, const char *)) == NULL)
171 			    break;
172 		argv[0] = argv[i] = NULL;
173 		wargv = (void *)ct_decode_argv(i + 1, argv, &el->el_lgcyconv);
174 		if (!wargv) {
175 		    ret = -1;
176 		    goto out;
177 		}
178 		/*
179 		 * AFAIK we can't portably pass through our new wargv to
180 		 * el_wset(), so we have to reimplement the body of
181 		 * el_wset() for these ops.
182 		 */
183 		switch (op) {
184 		case EL_BIND:
185 			wargv[0] = L"bind";
186 			ret = map_bind(el, i, wargv);
187 			break;
188 		case EL_TELLTC:
189 			wargv[0] = L"telltc";
190 			ret = terminal_telltc(el, i, wargv);
191 			break;
192 		case EL_SETTC:
193 			wargv[0] = L"settc";
194 			ret = terminal_settc(el, i, wargv);
195 			break;
196 		case EL_ECHOTC:
197 			wargv[0] = L"echotc";
198 			ret = terminal_echotc(el, i, wargv);
199 			break;
200 		case EL_SETTY:
201 			wargv[0] = L"setty";
202 			ret = tty_stty(el, i, wargv);
203 			break;
204 		default:
205 			ret = -1;
206 		}
207 		el_free(wargv);
208 		break;
209 	}
210 
211 	/* XXX: do we need to change el_func_t too? */
212 	case EL_ADDFN: {          /* const char *, const char *, el_func_t */
213 		const char *args[2];
214 		el_func_t func;
215 		wchar_t **wargv;
216 
217 		args[0] = va_arg(ap, const char *);
218 		args[1] = va_arg(ap, const char *);
219 		func = va_arg(ap, el_func_t);
220 
221 		wargv = ct_decode_argv(2, args, &el->el_lgcyconv);
222 		if (!wargv) {
223 		    ret = -1;
224 		    goto out;
225 		}
226 		/* XXX: The two strdup's leak */
227 		ret = map_addfunc(el, wcsdup(wargv[0]), wcsdup(wargv[1]),
228 		    func);
229 		el_free(wargv);
230 		break;
231 	}
232 	case EL_HIST: {           /* hist_fun_t, const char * */
233 		hist_fun_t fun = va_arg(ap, hist_fun_t);
234 		void *ptr = va_arg(ap, void *);
235 		ret = hist_set(el, fun, ptr);
236 		el->el_flags |= NARROW_HISTORY;
237 		break;
238 	}
239 
240 	case EL_GETCFN:         /* el_rfunc_t */
241 		ret = el_wset(el, op, va_arg(ap, el_rfunc_t));
242 		break;
243 
244 	case EL_CLIENTDATA:     /* void * */
245 		ret = el_wset(el, op, va_arg(ap, void *));
246 		break;
247 
248 	case EL_SETFP: {          /* int, FILE * */
249 		int what = va_arg(ap, int);
250 		FILE *fp = va_arg(ap, FILE *);
251 		ret = el_wset(el, op, what, fp);
252 		break;
253 	}
254 
255 	case EL_REFRESH:
256 		re_clear_display(el);
257 		re_refresh(el);
258 		terminal__flush(el);
259 		ret = 0;
260 		break;
261 
262 	default:
263 		ret = -1;
264 		break;
265 	}
266 
267 out:
268 	va_end(ap);
269 	return ret;
270 }
271 
272 
273 int
274 el_get(EditLine *el, int op, ...)
275 {
276 	va_list ap;
277 	int ret;
278 
279 	if (!el)
280 		return -1;
281 
282 	va_start(ap, op);
283 
284 	switch (op) {
285 	case EL_PROMPT:         /* el_pfunc_t * */
286 	case EL_RPROMPT: {
287 		el_pfunc_t *p = va_arg(ap, el_pfunc_t *);
288 		ret = prompt_get(el, p, 0, op);
289 		break;
290 	}
291 
292 	case EL_PROMPT_ESC: /* el_pfunc_t *, char **/
293 	case EL_RPROMPT_ESC: {
294 		el_pfunc_t *p = va_arg(ap, el_pfunc_t *);
295 		char *c = va_arg(ap, char *);
296 		wchar_t wc = 0;
297 		ret = prompt_get(el, p, &wc, op);
298 		*c = (char)wc;
299 		break;
300 	}
301 
302 	case EL_EDITOR: {
303 		const char **p = va_arg(ap, const char **);
304 		const wchar_t *pw;
305 		ret = el_wget(el, op, &pw);
306 		*p = ct_encode_string(pw, &el->el_lgcyconv);
307 		if (!el->el_lgcyconv.csize)
308 			ret = -1;
309 		break;
310 	}
311 
312 	case EL_TERMINAL:       /* const char ** */
313 		ret = el_wget(el, op, va_arg(ap, const char **));
314 		break;
315 
316 	case EL_SIGNAL:         /* int * */
317 	case EL_EDITMODE:
318 	case EL_UNBUFFERED:
319 	case EL_PREP_TERM:
320 		ret = el_wget(el, op, va_arg(ap, int *));
321 		break;
322 
323 	case EL_GETTC: {
324 		char *argv[3];
325 		static char gettc[] = "gettc";
326 		argv[0] = gettc;
327 		argv[1] = va_arg(ap, char *);
328 		argv[2] = va_arg(ap, void *);
329 		ret = terminal_gettc(el, 3, argv);
330 		break;
331 	}
332 
333 	case EL_GETCFN:         /* el_rfunc_t */
334 		ret = el_wget(el, op, va_arg(ap, el_rfunc_t *));
335 		break;
336 
337 	case EL_CLIENTDATA:     /* void ** */
338 		ret = el_wget(el, op, va_arg(ap, void **));
339 		break;
340 
341 	case EL_GETFP: {          /* int, FILE ** */
342 		int what = va_arg(ap, int);
343 		FILE **fpp = va_arg(ap, FILE **);
344 		ret = el_wget(el, op, what, fpp);
345 		break;
346 	}
347 
348 	default:
349 		ret = -1;
350 		break;
351 	}
352 
353 	va_end(ap);
354 	return ret;
355 }
356 
357 
358 const LineInfo *
359 el_line(EditLine *el)
360 {
361 	const LineInfoW *winfo = el_wline(el);
362 	LineInfo *info = &el->el_lgcylinfo;
363 	size_t offset;
364 	const wchar_t *p;
365 
366 	info->buffer   = ct_encode_string(winfo->buffer, &el->el_lgcyconv);
367 
368 	offset = 0;
369 	for (p = winfo->buffer; p < winfo->cursor; p++)
370 		offset += ct_enc_width(*p);
371 	info->cursor = info->buffer + offset;
372 
373 	offset = 0;
374 	for (p = winfo->buffer; p < winfo->lastchar; p++)
375 		offset += ct_enc_width(*p);
376 	info->lastchar = info->buffer + offset;
377 
378 	return info;
379 }
380 
381 
382 int
383 el_insertstr(EditLine *el, const char *str)
384 {
385 	return el_winsertstr(el, ct_decode_string(str, &el->el_lgcyconv));
386 }
387