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