src/hci/readline.c

   1 /*
   2  * Copyright (C) 2006 Michael Brown <mbrown@fensystems.co.uk>.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License as
   6  * published by the Free Software Foundation; either version 2 of the
   7  * License, or any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17  */
  18
  19 #include <string.h>
  20 #include <malloc.h>
  21 #include <console.h>
  22 #include <gpxe/editstring.h>
  23 #include <readline/readline.h>
  24
  25 /** @file
  26  *
  27  * Minimal readline
  28  *
  29  */
  30
  31 #define READLINE_MAX 256
  32
  33 /**
  34  * Synchronise console with edited string
  35  *
  36  * @v string            Editable string
  37  */
  38 static void sync_console ( struct edit_string *string ) {
  39         unsigned int mod_start = string->mod_start;
  40         unsigned int mod_end = string->mod_end;
  41         unsigned int cursor = string->last_cursor;
  42         size_t len = strlen ( string->buf );
  43
  44         /* Expand region back to old cursor position if applicable */
  45         if ( mod_start > string->last_cursor )
  46                 mod_start = string->last_cursor;
  47
  48         /* Expand region forward to new cursor position if applicable */
  49         if ( mod_end < string->cursor )
  50                 mod_end = string->cursor;
  51
  52         /* Backspace to start of region */
  53         while ( cursor > mod_start ) {
  54                 putchar ( '\b' );
  55                 cursor--;
  56         }
  57
  58         /* Print modified region */
  59         while ( cursor < mod_end ) {
  60                 putchar ( ( cursor >= len ) ? ' ' : string->buf[cursor] );
  61                 cursor++;
  62         }
  63
  64         /* Backspace to new cursor position */
  65         while ( cursor > string->cursor ) {
  66                 putchar ( '\b' );
  67                 cursor--;
  68         }
  69 }
  70
  71 /**
  72  * Read line from console
  73  *
  74  * @v prompt            Prompt string
  75  * @ret line            Line read from console (excluding terminating newline)
  76  *
  77  * The returned line is allocated with malloc(); the caller must
  78  * eventually call free() to release the storage.
  79  */
  80 char * readline ( const char *prompt ) {
  81         char buf[READLINE_MAX];
  82         struct edit_string string = {
  83                 .buf = buf,
  84                 .len = sizeof ( buf ),
  85                 .cursor = 0,
  86         };
  87         int key;
  88         char *line;
  89
  90         if ( prompt )
  91                 printf ( "%s", prompt );
  92
  93         buf[0] = '\0';
  94         while ( 1 ) {
  95                 key = edit_string ( &string, getchar() );
  96                 sync_console ( &string );
  97                 switch ( key ) {
  98                 case 0x0d: /* Carriage return */
  99                 case 0x0a: /* Line feed */
 100                         putchar ( '\n' );
 101                         line = strdup ( buf );
 102                         if ( ! line )
 103                                 printf ( "Out of memory\n" );
 104                         return line;
 105                 case 0x03: /* Ctrl-C */
 106                         putchar ( '\n' );
 107                         return NULL;
 108                 default:
 109                         /* Do nothing */
 110                         break;
 111                 }
 112         }
 113 }