libedit: Add bapt patches to improve unicode
[dragonfly.git] / contrib / libedit / src / el.c
1 /*      $NetBSD: el.c,v 1.73 2014/06/18 18:12:28 christos Exp $ */
2
3 /*-
4  * Copyright (c) 1992, 1993
5  *      The Regents of the University of California.  All rights reserved.
6  *
7  * This code is derived from software contributed to Berkeley by
8  * Christos Zoulas of Cornell University.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  * 3. Neither the name of the University nor the names of its contributors
19  *    may be used to endorse or promote products derived from this software
20  *    without specific prior written permission.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  */
34
35 #include "config.h"
36 #if !defined(lint) && !defined(SCCSID)
37 #if 0
38 static char sccsid[] = "@(#)el.c        8.2 (Berkeley) 1/3/94";
39 #else
40 __RCSID("$NetBSD: el.c,v 1.73 2014/06/18 18:12:28 christos Exp $");
41 #endif
42 #endif /* not lint && not SCCSID */
43
44 /*
45  * el.c: EditLine interface functions
46  */
47 #include <sys/types.h>
48 #include <sys/param.h>
49 #include <string.h>
50 #include <stdlib.h>
51 #include <stdarg.h>
52 #include <ctype.h>
53 #include <locale.h>
54 #include <langinfo.h>
55 #include "el.h"
56
57 #ifndef HAVE_SECURE_GETENV
58 #       ifdef HAVE___SECURE_GETENV
59 #               define secure_getenv __secure_getenv
60 #               define HAVE_SECURE_GETENV 1
61 #       else
62 #               ifdef HAVE_ISSETUGID
63 #                       include <unistd.h>
64 #               else
65 #                       undef issetugid
66 #                       define issetugid() 1
67 #               endif
68 #       endif
69 #endif
70
71 #ifndef HAVE_SECURE_GETENV
72 char *secure_getenv(char const *name)
73 {
74         if (issetugid())
75                 return 0;
76         return getenv(name);
77 }
78 #endif
79
80 /* el_init():
81  *      Initialize editline and set default parameters.
82  */
83 public EditLine *
84 el_init(const char *prog, FILE *fin, FILE *fout, FILE *ferr)
85 {
86     return el_init_fd(prog, fin, fout, ferr, fileno(fin), fileno(fout),
87         fileno(ferr));
88 }
89
90 public EditLine *
91 el_init_fd(const char *prog, FILE *fin, FILE *fout, FILE *ferr,
92     int fdin, int fdout, int fderr)
93 {
94         EditLine *el = el_malloc(sizeof(*el));
95
96         if (el == NULL)
97                 return NULL;
98
99         memset(el, 0, sizeof(EditLine));
100
101         el->el_infile = fin;
102         el->el_outfile = fout;
103         el->el_errfile = ferr;
104
105         el->el_infd = fdin;
106         el->el_outfd = fdout;
107         el->el_errfd = fderr;
108
109         el->el_prog = Strdup(ct_decode_string(prog, &el->el_scratch));
110         if (el->el_prog == NULL) {
111                 el_free(el);
112                 return NULL;
113         }
114
115         /*
116          * Initialize all the modules. Order is important!!!
117          */
118         el->el_flags = 0;
119 #ifdef WIDECHAR
120         if (setlocale(LC_CTYPE, "") != NULL){
121                 if (strcmp(nl_langinfo(CODESET), "UTF-8") == 0)
122                         el->el_flags |= CHARSET_IS_UTF8;
123         }
124 #endif
125
126         if (terminal_init(el) == -1) {
127                 el_free(el->el_prog);
128                 el_free(el);
129                 return NULL;
130         }
131         (void) keymacro_init(el);
132         (void) map_init(el);
133         if (tty_init(el) == -1)
134                 el->el_flags |= NO_TTY;
135         (void) ch_init(el);
136         (void) search_init(el);
137         (void) hist_init(el);
138         (void) prompt_init(el);
139         (void) sig_init(el);
140         (void) read_init(el);
141
142         return el;
143 }
144
145
146 /* el_end():
147  *      Clean up.
148  */
149 public void
150 el_end(EditLine *el)
151 {
152
153         if (el == NULL)
154                 return;
155
156         el_reset(el);
157
158         terminal_end(el);
159         keymacro_end(el);
160         map_end(el);
161         tty_end(el);
162         ch_end(el);
163         search_end(el);
164         hist_end(el);
165         prompt_end(el);
166         sig_end(el);
167
168         el_free(el->el_prog);
169 #ifdef WIDECHAR
170         el_free(el->el_scratch.cbuff);
171         el_free(el->el_scratch.wbuff);
172         el_free(el->el_lgcyconv.cbuff);
173         el_free(el->el_lgcyconv.wbuff);
174 #endif
175         el_free(el);
176 }
177
178
179 /* el_reset():
180  *      Reset the tty and the parser
181  */
182 public void
183 el_reset(EditLine *el)
184 {
185
186         tty_cookedmode(el);
187         ch_reset(el, 0);                /* XXX: Do we want that? */
188 }
189
190
191 /* el_set():
192  *      set the editline parameters
193  */
194 public int
195 FUN(el,set)(EditLine *el, int op, ...)
196 {
197         va_list ap;
198         int rv = 0;
199
200         if (el == NULL)
201                 return -1;
202         va_start(ap, op);
203
204         switch (op) {
205         case EL_PROMPT:
206         case EL_RPROMPT: {
207                 el_pfunc_t p = va_arg(ap, el_pfunc_t);
208
209                 rv = prompt_set(el, p, 0, op, 1);
210                 break;
211         }
212
213         case EL_RESIZE: {
214                 el_zfunc_t p = va_arg(ap, el_zfunc_t);
215                 void *arg = va_arg(ap, void *);
216                 rv = ch_resizefun(el, p, arg);
217                 break;
218         }
219
220         case EL_ALIAS_TEXT: {
221                 el_afunc_t p = va_arg(ap, el_afunc_t);
222                 void *arg = va_arg(ap, void *);
223                 rv = ch_aliasfun(el, p, arg);
224                 break;
225         }
226
227         case EL_PROMPT_ESC:
228         case EL_RPROMPT_ESC: {
229                 el_pfunc_t p = va_arg(ap, el_pfunc_t);
230                 int c = va_arg(ap, int);
231
232                 rv = prompt_set(el, p, c, op, 1);
233                 break;
234         }
235
236         case EL_TERMINAL:
237                 rv = terminal_set(el, va_arg(ap, char *));
238                 break;
239
240         case EL_EDITOR:
241                 rv = map_set_editor(el, va_arg(ap, Char *));
242                 break;
243
244         case EL_SIGNAL:
245                 if (va_arg(ap, int))
246                         el->el_flags |= HANDLE_SIGNALS;
247                 else
248                         el->el_flags &= ~HANDLE_SIGNALS;
249                 break;
250
251         case EL_BIND:
252         case EL_TELLTC:
253         case EL_SETTC:
254         case EL_ECHOTC:
255         case EL_SETTY:
256         {
257                 const Char *argv[20];
258                 int i;
259
260                 for (i = 1; i < (int)__arraycount(argv); i++)
261                         if ((argv[i] = va_arg(ap, Char *)) == NULL)
262                                 break;
263
264                 switch (op) {
265                 case EL_BIND:
266                         argv[0] = STR("bind");
267                         rv = map_bind(el, i, argv);
268                         break;
269
270                 case EL_TELLTC:
271                         argv[0] = STR("telltc");
272                         rv = terminal_telltc(el, i, argv);
273                         break;
274
275                 case EL_SETTC:
276                         argv[0] = STR("settc");
277                         rv = terminal_settc(el, i, argv);
278                         break;
279
280                 case EL_ECHOTC:
281                         argv[0] = STR("echotc");
282                         rv = terminal_echotc(el, i, argv);
283                         break;
284
285                 case EL_SETTY:
286                         argv[0] = STR("setty");
287                         rv = tty_stty(el, i, argv);
288                         break;
289
290                 default:
291                         rv = -1;
292                         EL_ABORT((el->el_errfile, "Bad op %d\n", op));
293                         break;
294                 }
295                 break;
296         }
297
298         case EL_ADDFN:
299         {
300                 Char *name = va_arg(ap, Char *);
301                 Char *help = va_arg(ap, Char *);
302                 el_func_t func = va_arg(ap, el_func_t);
303
304                 rv = map_addfunc(el, name, help, func);
305                 break;
306         }
307
308         case EL_HIST:
309         {
310                 hist_fun_t func = va_arg(ap, hist_fun_t);
311                 void *ptr = va_arg(ap, void *);
312
313                 rv = hist_set(el, func, ptr);
314                 if (!(el->el_flags & CHARSET_IS_UTF8))
315                         el->el_flags &= ~NARROW_HISTORY;
316                 break;
317         }
318
319         case EL_EDITMODE:
320                 if (va_arg(ap, int))
321                         el->el_flags &= ~EDIT_DISABLED;
322                 else
323                         el->el_flags |= EDIT_DISABLED;
324                 rv = 0;
325                 break;
326
327         case EL_GETCFN:
328         {
329                 el_rfunc_t rc = va_arg(ap, el_rfunc_t);
330                 rv = el_read_setfn(el, rc);
331                 el->el_flags &= ~NARROW_READ;
332                 break;
333         }
334
335         case EL_CLIENTDATA:
336                 el->el_data = va_arg(ap, void *);
337                 break;
338
339         case EL_UNBUFFERED:
340                 rv = va_arg(ap, int);
341                 if (rv && !(el->el_flags & UNBUFFERED)) {
342                         el->el_flags |= UNBUFFERED;
343                         read_prepare(el);
344                 } else if (!rv && (el->el_flags & UNBUFFERED)) {
345                         el->el_flags &= ~UNBUFFERED;
346                         read_finish(el);
347                 }
348                 rv = 0;
349                 break;
350
351         case EL_PREP_TERM:
352                 rv = va_arg(ap, int);
353                 if (rv)
354                         (void) tty_rawmode(el);
355                 else
356                         (void) tty_cookedmode(el);
357                 rv = 0;
358                 break;
359
360         case EL_SETFP:
361         {
362                 FILE *fp;
363                 int what;
364
365                 what = va_arg(ap, int);
366                 fp = va_arg(ap, FILE *);
367
368                 rv = 0;
369                 switch (what) {
370                 case 0:
371                         el->el_infile = fp;
372                         el->el_infd = fileno(fp);
373                         break;
374                 case 1:
375                         el->el_outfile = fp;
376                         el->el_outfd = fileno(fp);
377                         break;
378                 case 2:
379                         el->el_errfile = fp;
380                         el->el_errfd = fileno(fp);
381                         break;
382                 default:
383                         rv = -1;
384                         break;
385                 }
386                 break;
387         }
388
389         case EL_REFRESH:
390                 re_clear_display(el);
391                 re_refresh(el);
392                 terminal__flush(el);
393                 break;
394
395         default:
396                 rv = -1;
397                 break;
398         }
399
400         va_end(ap);
401         return rv;
402 }
403
404
405 /* el_get():
406  *      retrieve the editline parameters
407  */
408 public int
409 FUN(el,get)(EditLine *el, int op, ...)
410 {
411         va_list ap;
412         int rv;
413
414         if (el == NULL)
415                 return -1;
416
417         va_start(ap, op);
418
419         switch (op) {
420         case EL_PROMPT:
421         case EL_RPROMPT: {
422                 el_pfunc_t *p = va_arg(ap, el_pfunc_t *);
423                 rv = prompt_get(el, p, 0, op);
424                 break;
425         }
426         case EL_PROMPT_ESC:
427         case EL_RPROMPT_ESC: {
428                 el_pfunc_t *p = va_arg(ap, el_pfunc_t *);
429                 Char *c = va_arg(ap, Char *);
430
431                 rv = prompt_get(el, p, c, op);
432                 break;
433         }
434
435         case EL_EDITOR:
436                 rv = map_get_editor(el, va_arg(ap, const Char **));
437                 break;
438
439         case EL_SIGNAL:
440                 *va_arg(ap, int *) = (el->el_flags & HANDLE_SIGNALS);
441                 rv = 0;
442                 break;
443
444         case EL_EDITMODE:
445                 *va_arg(ap, int *) = !(el->el_flags & EDIT_DISABLED);
446                 rv = 0;
447                 break;
448
449         case EL_TERMINAL:
450                 terminal_get(el, va_arg(ap, const char **));
451                 rv = 0;
452                 break;
453
454         case EL_GETTC:
455         {
456                 static char name[] = "gettc";
457                 char *argv[20];
458                 int i;
459
460                 for (i = 1; i < (int)__arraycount(argv); i++)
461                         if ((argv[i] = va_arg(ap, char *)) == NULL)
462                                 break;
463
464                 argv[0] = name;
465                 rv = terminal_gettc(el, i, argv);
466                 break;
467         }
468
469         case EL_GETCFN:
470                 *va_arg(ap, el_rfunc_t *) = el_read_getfn(el);
471                 rv = 0;
472                 break;
473
474         case EL_CLIENTDATA:
475                 *va_arg(ap, void **) = el->el_data;
476                 rv = 0;
477                 break;
478
479         case EL_UNBUFFERED:
480                 *va_arg(ap, int *) = (el->el_flags & UNBUFFERED) != 0;
481                 rv = 0;
482                 break;
483
484         case EL_GETFP:
485         {
486                 int what;
487                 FILE **fpp;
488
489                 what = va_arg(ap, int);
490                 fpp = va_arg(ap, FILE **);
491                 rv = 0;
492                 switch (what) {
493                 case 0:
494                         *fpp = el->el_infile;
495                         break;
496                 case 1:
497                         *fpp = el->el_outfile;
498                         break;
499                 case 2:
500                         *fpp = el->el_errfile;
501                         break;
502                 default:
503                         rv = -1;
504                         break;
505                 }
506                 break;
507         }
508         default:
509                 rv = -1;
510                 break;
511         }
512         va_end(ap);
513
514         return rv;
515 }
516
517
518 /* el_line():
519  *      Return editing info
520  */
521 public const TYPE(LineInfo) *
522 FUN(el,line)(EditLine *el)
523 {
524
525         return (const TYPE(LineInfo) *)(void *)&el->el_line;
526 }
527
528
529 /* el_source():
530  *      Source a file
531  */
532 public int
533 el_source(EditLine *el, const char *fname)
534 {
535         FILE *fp;
536         size_t len;
537         char *ptr;
538         char *path = NULL;
539         const Char *dptr;
540         int error = 0;
541
542         fp = NULL;
543         if (fname == NULL) {
544                 static const char elpath[] = "/.editrc";
545                 size_t plen = sizeof(elpath);
546
547                 if ((ptr = secure_getenv("HOME")) == NULL)
548                         return -1;
549                 plen += strlen(ptr);
550                 if ((path = el_malloc(plen * sizeof(*path))) == NULL)
551                         return -1;
552                 (void)snprintf(path, plen, "%s%s", ptr, elpath);
553                 fname = path;
554         }
555         if (fp == NULL)
556                 fp = fopen(fname, "r");
557         if (fp == NULL) {
558                 el_free(path);
559                 return -1;
560         }
561
562         while ((ptr = fgetln(fp, &len)) != NULL) {
563                 if (*ptr == '\n')
564                         continue;       /* Empty line. */
565                 dptr = ct_decode_string(ptr, &el->el_scratch);
566                 if (!dptr)
567                         continue;
568                 if (len > 0 && dptr[len - 1] == '\n')
569                         --len;
570
571                 /* loop until first non-space char or EOL */
572                 while (*dptr != '\0' && Isspace(*dptr))
573                         dptr++;
574                 if (*dptr == '#')
575                         continue;   /* ignore, this is a comment line */
576                 if ((error = parse_line(el, dptr)) == -1)
577                         break;
578         }
579
580         el_free(path);
581         (void) fclose(fp);
582         return error;
583 }
584
585
586 /* el_resize():
587  *      Called from program when terminal is resized
588  */
589 public void
590 el_resize(EditLine *el)
591 {
592         int lins, cols;
593         sigset_t oset, nset;
594
595         (void) sigemptyset(&nset);
596         (void) sigaddset(&nset, SIGWINCH);
597         (void) sigprocmask(SIG_BLOCK, &nset, &oset);
598
599         /* get the correct window size */
600         if (terminal_get_size(el, &lins, &cols))
601                 terminal_change_size(el, lins, cols);
602
603         (void) sigprocmask(SIG_SETMASK, &oset, NULL);
604 }
605
606
607 /* el_beep():
608  *      Called from the program to beep
609  */
610 public void
611 el_beep(EditLine *el)
612 {
613
614         terminal_beep(el);
615 }
616
617
618 /* el_editmode()
619  *      Set the state of EDIT_DISABLED from the `edit' command.
620  */
621 protected int
622 /*ARGSUSED*/
623 el_editmode(EditLine *el, int argc, const Char **argv)
624 {
625         const Char *how;
626
627         if (argv == NULL || argc != 2 || argv[1] == NULL)
628                 return -1;
629
630         how = argv[1];
631         if (Strcmp(how, STR("on")) == 0) {
632                 el->el_flags &= ~EDIT_DISABLED;
633                 tty_rawmode(el);
634         } else if (Strcmp(how, STR("off")) == 0) {
635                 tty_cookedmode(el);
636                 el->el_flags |= EDIT_DISABLED;
637         }
638         else {
639                 (void) fprintf(el->el_errfile, "edit: Bad value `" FSTR "'.\n",
640                     how);
641                 return -1;
642         }
643         return 0;
644 }