| /* linenoise.c -- guerrilla line editing library against the idea that a |
| * line editing lib needs to be 20,000 lines of C code. |
| * |
| * You can find the latest source code at: |
| * |
| * http://github.com/antirez/linenoise |
| * |
| * Does a number of crazy assumptions that happen to be true in 99.9999% of |
| * the 2010 UNIX computers around. |
| * |
| * Copyright (c) 2010, Salvatore Sanfilippo <antirez at gmail dot com> |
| * All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are met: |
| * |
| * * Redistributions of source code must retain the above copyright notice, |
| * this list of conditions and the following disclaimer. |
| * * Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * * Neither the name of Redis nor the names of its contributors may be used |
| * to endorse or promote products derived from this software without |
| * specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
| * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE |
| * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
| * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
| * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
| * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
| * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
| * POSSIBILITY OF SUCH DAMAGE. |
| * |
| * References: |
| * - http://invisible-island.net/xterm/ctlseqs/ctlseqs.html |
| * - http://www.3waylabs.com/nw/WWW/products/wizcon/vt220.html |
| * |
| * Todo list: |
| * - Switch to gets() if $TERM is something we can't support. |
| * - Filter bogus Ctrl+<char> combinations. |
| * - Win32 support |
| * |
| * Bloat: |
| * - Completion? |
| * - History search like Ctrl+r in readline? |
| * |
| * List of escape sequences used by this program, we do everything just |
| * with three sequences. In order to be so cheap we may have some |
| * flickering effect with some slow terminal, but the lesser sequences |
| * the more compatible. |
| * |
| * CHA (Cursor Horizontal Absolute) |
| * Sequence: ESC [ n G |
| * Effect: moves cursor to column n |
| * |
| * EL (Erase Line) |
| * Sequence: ESC [ n K |
| * Effect: if n is 0 or missing, clear from cursor to end of line |
| * Effect: if n is 1, clear from beginning of line to cursor |
| * Effect: if n is 2, clear entire line |
| * |
| * CUF (CUrsor Forward) |
| * Sequence: ESC [ n C |
| * Effect: moves cursor forward of n chars |
| * |
| */ |
| |
| #include <termios.h> |
| #include <unistd.h> |
| #include <stdlib.h> |
| #include <stdio.h> |
| #include <errno.h> |
| #include <string.h> |
| #include <stdlib.h> |
| #include <sys/types.h> |
| #include <sys/ioctl.h> |
| #include <unistd.h> |
| |
| #define LINENOISE_MAX_LINE 4096 |
| static char *unsupported_term[] = {"dumb","cons25",NULL}; |
| |
| static struct termios orig_termios; /* in order to restore at exit */ |
| static int rawmode = 0; /* for atexit() function to check if restore is needed*/ |
| static int atexit_registered = 0; /* register atexit just 1 time */ |
| static int history_max_len = 100; |
| static int history_len = 0; |
| char **history = NULL; |
| |
| static void linenoiseAtExit(void); |
| int linenoiseHistoryAdd(const char *line); |
| |
| static int isUnsupportedTerm(void) { |
| char *term = getenv("TERM"); |
| int j; |
| |
| if (term == NULL) return 0; |
| for (j = 0; unsupported_term[j]; j++) |
| if (!strcasecmp(term,unsupported_term[j])) return 1; |
| return 0; |
| } |
| |
| static void freeHistory(void) { |
| if (history) { |
| int j; |
| |
| for (j = 0; j < history_len; j++) |
| free(history[j]); |
| free(history); |
| } |
| } |
| |
| static int enableRawMode(int fd) { |
| struct termios raw; |
| |
| if (!isatty(STDIN_FILENO)) goto fatal; |
| if (!atexit_registered) { |
| atexit(linenoiseAtExit); |
| atexit_registered = 1; |
| } |
| if (tcgetattr(fd,&orig_termios) == -1) goto fatal; |
| |
| raw = orig_termios; /* modify the original mode */ |
| /* input modes: no break, no CR to NL, no parity check, no strip char, |
| * no start/stop output control. */ |
| raw.c_iflag &= ~(BRKINT | ICRNL | INPCK | ISTRIP | IXON); |
| /* output modes - disable post processing */ |
| raw.c_oflag &= ~(OPOST); |
| /* control modes - set 8 bit chars */ |
| raw.c_cflag |= (CS8); |
| /* local modes - choing off, canonical off, no extended functions, |
| * no signal chars (^Z,^C) */ |
| raw.c_lflag &= ~(ECHO | ICANON | IEXTEN | ISIG); |
| /* control chars - set return condition: min number of bytes and timer. |
| * We want read to return every single byte, without timeout. */ |
| raw.c_cc[VMIN] = 1; raw.c_cc[VTIME] = 0; /* 1 byte, no timer */ |
| |
| /* put terminal in raw mode */ |
| if (tcsetattr(fd,TCSADRAIN,&raw) < 0) goto fatal; |
| rawmode = 1; |
| return 0; |
| |
| fatal: |
| errno = ENOTTY; |
| return -1; |
| } |
| |
| static void disableRawMode(int fd) { |
| /* Don't even check the return value as it's too late. */ |
| if (rawmode && tcsetattr(fd,TCSADRAIN,&orig_termios) != -1) |
| rawmode = 0; |
| } |
| |
| /* At exit we'll try to fix the terminal to the initial conditions. */ |
| static void linenoiseAtExit(void) { |
| disableRawMode(STDIN_FILENO); |
| freeHistory(); |
| } |
| |
| static int getColumns(void) { |
| struct winsize ws; |
| |
| if (ioctl(1, TIOCGWINSZ, &ws) == -1) return 4096; |
| if (ws.ws_col == 0) { |
| return 4096; |
| } |
| return ws.ws_col; |
| } |
| |
| static int effectiveLen(const char* prompt) { |
| int col = 0; |
| char c; |
| // TODO: Handle escape sequences. |
| while ( (c = *prompt++) != 0 ) { |
| if (c == '\n') { |
| col = 0; |
| } else { |
| col++; |
| } |
| } |
| return col; |
| } |
| |
| static void refreshLine(int fd, const char *prompt, char *buf, size_t len, size_t pos, size_t cols) { |
| char seq[64]; |
| size_t plen = effectiveLen(prompt); |
| |
| while((plen+pos) >= cols) { |
| buf++; |
| len--; |
| pos--; |
| } |
| while (plen+len > cols) { |
| len--; |
| } |
| |
| /* Cursor to left edge */ |
| snprintf(seq,64,"\x1b[0G"); |
| if (write(fd,seq,strlen(seq)) == -1) return; |
| /* Write the prompt and the current buffer content */ |
| if (write(fd,prompt,strlen(prompt)) == -1) return; |
| if (write(fd,buf,len) == -1) return; |
| /* Erase to right */ |
| snprintf(seq,64,"\x1b[0K"); |
| if (write(fd,seq,strlen(seq)) == -1) return; |
| /* Move cursor to original position. */ |
| snprintf(seq,64,"\x1b[0G\x1b[%dC", (int)(pos+plen)); |
| if (write(fd,seq,strlen(seq)) == -1) return; |
| } |
| |
| static int linenoisePrompt(int fd, char *buf, size_t buflen, const char *prompt) { |
| size_t plen = strlen(prompt); |
| size_t pos = 0; |
| size_t len = 0; |
| size_t cols = getColumns(); |
| int history_index = 0; |
| |
| buf[0] = '\0'; |
| buflen--; /* Make sure there is always space for the nulterm */ |
| |
| /* The latest history entry is always our current buffer, that |
| * initially is just an empty string. */ |
| linenoiseHistoryAdd(""); |
| |
| if (write(fd,prompt,plen) == -1) return -1; |
| while(1) { |
| char c; |
| int nread; |
| char seq[2]; |
| |
| nread = read(fd,&c,1); |
| if (nread <= 0) return len; |
| switch(c) { |
| case 10: /* line feed. */ |
| case 13: /* enter */ |
| history_len--; |
| return len; |
| case 4: /* ctrl-d */ |
| history_len--; |
| return (len == 0) ? -1 : (int)len; |
| case 3: /* ctrl-c */ |
| errno = EAGAIN; |
| return -1; |
| case 127: /* backspace */ |
| case 8: /* ctrl-h */ |
| if (pos > 0 && len > 0) { |
| memmove(buf+pos-1,buf+pos,len-pos); |
| pos--; |
| len--; |
| buf[len] = '\0'; |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| } |
| break; |
| case 20: /* ctrl-t */ |
| if (pos > 0 && pos < len) { |
| int aux = buf[pos-1]; |
| buf[pos-1] = buf[pos]; |
| buf[pos] = aux; |
| if (pos != len-1) pos++; |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| } |
| break; |
| case 2: /* ctrl-b */ |
| goto left_arrow; |
| case 6: /* ctrl-f */ |
| goto right_arrow; |
| case 16: /* ctrl-p */ |
| seq[1] = 65; |
| goto up_down_arrow; |
| case 14: /* ctrl-n */ |
| seq[1] = 66; |
| goto up_down_arrow; |
| break; |
| case 27: /* escape sequence */ |
| if (read(fd,seq,2) == -1) break; |
| if (seq[0] == 91 && seq[1] == 68) { |
| left_arrow: |
| /* left arrow */ |
| if (pos > 0) { |
| pos--; |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| } |
| } else if (seq[0] == 91 && seq[1] == 67) { |
| right_arrow: |
| /* right arrow */ |
| if (pos != len) { |
| pos++; |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| } |
| } else if (seq[0] == 91 && (seq[1] == 65 || seq[1] == 66)) { |
| up_down_arrow: |
| /* up and down arrow: history */ |
| if (history_len > 1) { |
| /* Update the current history entry before to |
| * overwrite it with tne next one. */ |
| free(history[history_len-1-history_index]); |
| history[history_len-1-history_index] = strdup(buf); |
| /* Show the new entry */ |
| history_index += (seq[1] == 65) ? 1 : -1; |
| if (history_index < 0) { |
| history_index = 0; |
| break; |
| } else if (history_index >= history_len) { |
| history_index = history_len-1; |
| break; |
| } |
| strncpy(buf,history[history_len-1-history_index],buflen); |
| buf[buflen] = '\0'; |
| len = pos = strlen(buf); |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| } |
| } |
| break; |
| default: |
| if (len < buflen) { |
| if (len == pos) { |
| buf[pos] = c; |
| pos++; |
| len++; |
| buf[len] = '\0'; |
| if (plen+len < cols) { |
| /* Avoid a full update of the line in the |
| * trivial case. */ |
| if (write(fd,&c,1) == -1) return -1; |
| } else { |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| } |
| } else { |
| memmove(buf+pos+1,buf+pos,len-pos); |
| buf[pos] = c; |
| len++; |
| pos++; |
| buf[len] = '\0'; |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| } |
| } |
| break; |
| case 21: /* Ctrl+u, delete the whole line. */ |
| buf[0] = '\0'; |
| pos = len = 0; |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| break; |
| case 11: /* Ctrl+k, delete from current to end of line. */ |
| buf[pos] = '\0'; |
| len = pos; |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| break; |
| case 1: /* Ctrl+a, go to the start of the line */ |
| pos = 0; |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| break; |
| case 5: /* ctrl+e, go to the end of the line */ |
| pos = len; |
| refreshLine(fd,prompt,buf,len,pos,cols); |
| break; |
| } |
| } |
| return len; |
| } |
| |
| static int linenoiseRaw(char *buf, size_t buflen, const char *prompt) { |
| int fd = STDIN_FILENO; |
| int count; |
| |
| if (buflen == 0) { |
| errno = EINVAL; |
| return -1; |
| } |
| if (!isatty(STDIN_FILENO)) { |
| if (fgets(buf, buflen, stdin) == NULL) return -1; |
| count = strlen(buf); |
| if (count && buf[count-1] == '\n') { |
| count--; |
| buf[count] = '\0'; |
| } |
| } else { |
| if (enableRawMode(fd) == -1) return -1; |
| count = linenoisePrompt(fd, buf, buflen, prompt); |
| disableRawMode(fd); |
| } |
| return count; |
| } |
| |
| char *linenoise(const char *prompt) { |
| char buf[LINENOISE_MAX_LINE]; |
| int count; |
| |
| if (isUnsupportedTerm()) { |
| size_t len; |
| |
| printf("%s",prompt); |
| fflush(stdout); |
| if (fgets(buf,LINENOISE_MAX_LINE,stdin) == NULL) return NULL; |
| len = strlen(buf); |
| while(len && (buf[len-1] == '\n' || buf[len-1] == '\r')) { |
| len--; |
| buf[len] = '\0'; |
| } |
| return strdup(buf); |
| } else { |
| count = linenoiseRaw(buf,LINENOISE_MAX_LINE,prompt); |
| if (count == -1) return NULL; |
| return strdup(buf); |
| } |
| } |
| |
| /* Using a circular buffer is smarter, but a bit more complex to handle. */ |
| int linenoiseHistoryAdd(const char *line) { |
| char *linecopy; |
| |
| if (history_max_len == 0) return 0; |
| if (history == 0) { |
| history = malloc(sizeof(char*)*history_max_len); |
| if (history == NULL) return 0; |
| memset(history,0,(sizeof(char*)*history_max_len)); |
| } |
| linecopy = strdup(line); |
| if (!linecopy) return 0; |
| if (history_len == history_max_len) { |
| memmove(history,history+1,sizeof(char*)*(history_max_len-1)); |
| history_len--; |
| } |
| history[history_len] = linecopy; |
| history_len++; |
| return 1; |
| } |
| |
| int linenoiseHistorySetMaxLen(int len) { |
| char **new; |
| |
| if (len < 1) return 0; |
| if (history) { |
| int tocopy = history_len; |
| |
| new = malloc(sizeof(char*)*len); |
| if (new == NULL) return 0; |
| if (len < tocopy) tocopy = len; |
| memcpy(new,history+(history_max_len-tocopy), sizeof(char*)*tocopy); |
| free(history); |
| history = new; |
| } |
| history_max_len = len; |
| if (history_len > history_max_len) |
| history_len = history_max_len; |
| return 1; |
| } |