lib/libedit/eln.c

   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
  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
  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 *
  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
  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
 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
 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 *
 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
 398 el_insertstr(EditLine *el, const char *str)
 399 {
 400         return el_winsertstr(el, ct_decode_string(str, &el->el_lgcyconv));
 401 }