osdep/getch2.c

   1 /*
   2  * GyS-TermIO v2.0 (for GySmail v3)
   3  * a very small replacement of ncurses library
   4  *
   5  * copyright (C) 1999 A'rpi/ESP-team
   6  *
   7  * This file is part of MPlayer.
   8  *
   9  * MPlayer is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * MPlayer is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License along
  20  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  21  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  22  */
  23
  24 #include "config.h"
  25
  26 //#define HAVE_TERMCAP
  27 #if !defined(__OS2__) && !defined(__MORPHOS__)
  28 #define CONFIG_IOCTL
  29 #endif
  30
  31 #define MAX_KEYS 64
  32 #define BUF_LEN 256
  33
  34 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <string.h>
  37 #include <sys/time.h>
  38 #include <sys/types.h>
  39 #ifdef CONFIG_IOCTL
  40 #include <sys/ioctl.h>
  41 #endif
  42
  43 #ifdef HAVE_TERMIOS
  44 #ifdef HAVE_TERMIOS_H
  45 #include <termios.h>
  46 #endif
  47 #ifdef HAVE_SYS_TERMIOS_H
  48 #include <sys/termios.h>
  49 #endif
  50 #endif
  51
  52 #if defined(HAVE_LANGINFO) && defined(CONFIG_ICONV)
  53 #include <locale.h>
  54 #include <langinfo.h>
  55 #endif
  56
  57 #include <unistd.h>
  58 #include <fcntl.h>
  59
  60 #include "mp_fifo.h"
  61 #include "keycodes.h"
  62 #include "getch2.h"
  63
  64 #ifdef HAVE_TERMIOS
  65 static struct termios tio_orig;
  66 #endif
  67 static int getch2_len=0;
  68 static char getch2_buf[BUF_LEN];
  69
  70 int screen_width=80;
  71 int screen_height=24;
  72 char * erase_to_end_of_line = NULL;
  73
  74 typedef struct {
  75   int len;
  76   int code;
  77   char chars[8];
  78 } keycode_st;
  79 static keycode_st getch2_keys[MAX_KEYS];
  80 static int getch2_key_db=0;
  81
  82 #ifdef HAVE_TERMCAP
  83
  84 #if 0
  85 #include <termcap.h>
  86 #else
  87 int tgetent(char *BUFFER, char *TERMTYPE);
  88 int tgetnum(char *NAME);
  89 int tgetflag(char *NAME);
  90 char *tgetstr(char *NAME, char **AREA);
  91 #endif
  92
  93 static char term_buffer[4096];
  94 static char term_buffer2[4096];
  95 static char *term_p=term_buffer2;
  96
  97 static void termcap_add(char *id,int code){
  98 char *p=tgetstr(id,&term_p);
  99   if(!p) return;
 100   if(getch2_key_db>=MAX_KEYS) return;
 101   getch2_keys[getch2_key_db].len=strlen(p);
 102   strncpy(getch2_keys[getch2_key_db].chars,p,8);
 103   getch2_keys[getch2_key_db].code=code;
 104   ++getch2_key_db;
 105 /*  printf("%s=%s\n",id,p); */
 106 }
 107
 108 static int success=0;
 109
 110 int load_termcap(char *termtype){
 111   if(!termtype) termtype=getenv("TERM");
 112   if(!termtype) termtype="unknown";
 113   success=tgetent(term_buffer, termtype);
 114   if(success<0){ printf("Could not access the 'termcap' data base.\n"); return 0; }
 115   if(success==0){ printf("Terminal type `%s' is not defined.\n", termtype);return 0;}
 116
 117   screen_width=tgetnum("co");
 118   screen_height=tgetnum("li");
 119   if(screen_width<1 || screen_width>255) screen_width=80;
 120   if(screen_height<1 || screen_height>255) screen_height=24;
 121   erase_to_end_of_line= tgetstr("ce",&term_p);
 122
 123   termcap_add("kP",KEY_PGUP);
 124   termcap_add("kN",KEY_PGDWN);
 125   termcap_add("kh",KEY_HOME);
 126   termcap_add("kH",KEY_END);
 127   termcap_add("kI",KEY_INS);
 128   termcap_add("kD",KEY_DEL);
 129   termcap_add("kb",KEY_BS);
 130   termcap_add("kl",KEY_LEFT);
 131   termcap_add("kd",KEY_DOWN);
 132   termcap_add("ku",KEY_UP);
 133   termcap_add("kr",KEY_RIGHT);
 134   termcap_add("k0",KEY_F+0);
 135   termcap_add("k1",KEY_F+1);
 136   termcap_add("k2",KEY_F+2);
 137   termcap_add("k3",KEY_F+3);
 138   termcap_add("k4",KEY_F+4);
 139   termcap_add("k5",KEY_F+5);
 140   termcap_add("k6",KEY_F+6);
 141   termcap_add("k7",KEY_F+7);
 142   termcap_add("k8",KEY_F+8);
 143   termcap_add("k9",KEY_F+9);
 144   termcap_add("k;",KEY_F+10);
 145   return getch2_key_db;
 146 }
 147
 148 #endif
 149
 150 void get_screen_size(void){
 151 #ifdef CONFIG_IOCTL
 152   struct winsize ws;
 153   if (ioctl(0, TIOCGWINSZ, &ws) < 0 || !ws.ws_row || !ws.ws_col) return;
 154 /*  printf("Using IOCTL\n"); */
 155   screen_width=ws.ws_col;
 156   screen_height=ws.ws_row;
 157 #endif
 158 }
 159
 160 void getch2(struct mp_fifo *fifo)
 161 {
 162     int retval = read(0, &getch2_buf[getch2_len], BUF_LEN-getch2_len);
 163     if (retval < 1)
 164         return;
 165     getch2_len += retval;
 166
 167     while (getch2_len > 0 && (getch2_len > 1 || getch2_buf[0] != 27)) {
 168         int i, len, code;
 169
 170         /* First find in the TERMCAP database: */
 171         for (i = 0; i < getch2_key_db; i++) {
 172             if ((len = getch2_keys[i].len) <= getch2_len)
 173                 if(memcmp(getch2_keys[i].chars, getch2_buf, len) == 0) {
 174                     code = getch2_keys[i].code;
 175                     goto found;
 176                 }
 177         }
 178         /* We always match some keypress here, with length 1 if nothing else.
 179          * Since some of the cases explicitly test remaining buffer length
 180          * having a keycode only partially read in the buffer could incorrectly
 181          * use the first byte as an independent character.
 182          * However the buffer is big enough that this shouldn't happen too
 183          * easily, and it's been this way for years without many complaints.
 184          * I see no simple fix as there's no easy test which would tell
 185          * whether a string must be part of a longer keycode. */
 186         len = 1;
 187         code = getch2_buf[0];
 188         /* Check the well-known codes... */
 189         if (code != 27) {
 190             if (code == 'A'-64) code = KEY_HOME;
 191             else if (code == 'E'-64) code = KEY_END;
 192             else if (code == 'D'-64) code = KEY_DEL;
 193             else if (code == 'H'-64) code = KEY_BS;
 194             else if (code == 'U'-64) code = KEY_PGUP;
 195             else if (code == 'V'-64) code = KEY_PGDWN;
 196             else if (code == 8 || code==127) code = KEY_BS;
 197             else if (code == 10 || code==13) {
 198                 if (getch2_len > 1) {
 199                     int c = getch2_buf[1];
 200                     if ((c == 10 || c == 13) && (c != code))
 201                         len = 2;
 202                 }
 203                 code = KEY_ENTER;
 204             }
 205         }
 206         else if (getch2_len > 1) {
 207             int c = getch2_buf[1];
 208             if (c == 27) {
 209                 code = KEY_ESC;
 210                 len = 2;
 211                 goto found;
 212             }
 213             if (c >= '0' && c <= '9') {
 214                 code = c-'0'+KEY_F;
 215                 len = 2;
 216                 goto found;
 217             }
 218             if (getch2_len >= 4 && c == '[' && getch2_buf[2] == '[') {
 219                 int c = getch2_buf[3];
 220                 if (c >= 'A' && c < 'A'+12) {
 221                     code = KEY_F+1 + c-'A';
 222                     len = 4;
 223                     goto found;
 224                 }
 225             }
 226             if ((c == '[' || c == 'O') && getch2_len >= 3) {
 227                 int c = getch2_buf[2];
 228                 const short ctable[] = {
 229                     KEY_UP, KEY_DOWN, KEY_RIGHT, KEY_LEFT, 0,
 230                     KEY_END, KEY_PGDWN, KEY_HOME, KEY_PGUP, 0, 0, KEY_INS, 0, 0, 0,
 231                     KEY_F+1, KEY_F+2, KEY_F+3, KEY_F+4};
 232                 if (c >= 'A' && c <= 'S')
 233                     if (ctable[c - 'A']) {
 234                         code = ctable[c - 'A'];
 235                         len = 3;
 236                         goto found;
 237                     }
 238             }
 239             if (getch2_len >= 4 && c == '[' && getch2_buf[3] == '~') {
 240                 int c = getch2_buf[2];
 241                 const int ctable[8] = {KEY_HOME, KEY_INS, KEY_DEL, KEY_END, KEY_PGUP, KEY_PGDWN, KEY_HOME, KEY_END};
 242                 if (c >= '1' && c <= '8') {
 243                     code = ctable[c - '1'];
 244                     len = 4;
 245                     goto found;
 246                 }
 247             }
 248             if (getch2_len >= 5 && c == '[' && getch2_buf[4] == '~') {
 249                 int i = getch2_buf[2] - '0';
 250                 int j = getch2_buf[3] - '0';
 251                 if (i >= 0 && i <= 9 && j >= 0 && j <= 9) {
 252                     const short ftable[20] = {
 253                         11,12,13,14,15, 17,18,19,20,21,
 254                         23,24,25,26,28, 29,31,32,33,34 };
 255                     int a = i*10 + j;
 256                     for (i = 0; i < 20; i++)
 257                         if (ftable[i] == a) {
 258                             code = KEY_F+1 + i;
 259                             len = 5;
 260                             goto found;
 261                         }
 262                 }
 263             }
 264         }
 265     found:
 266         getch2_len -= len;
 267         for (i = 0; i < getch2_len; i++)
 268             getch2_buf[i] = getch2_buf[len+i];
 269         mplayer_put_key(fifo, code);
 270     }
 271 }
 272
 273 static int getch2_status=0;
 274
 275 void getch2_enable(void){
 276 #ifdef HAVE_TERMIOS
 277 struct termios tio_new;
 278     tcgetattr(0,&tio_orig);
 279     tio_new=tio_orig;
 280     tio_new.c_lflag &= ~(ICANON|ECHO); /* Clear ICANON and ECHO. */
 281     tio_new.c_cc[VMIN] = 0;
 282     tio_new.c_cc[VTIME] = 0;
 283     tcsetattr(0,TCSANOW,&tio_new);
 284 #endif
 285     /* Setting VMIN above should already make terminal non-blocking; but
 286      * that won't work if stdin is not a real terminal. */
 287     int flags = fcntl(0, F_GETFL);
 288     if (flags != -1)
 289         fcntl(0, F_SETFL, flags | O_NONBLOCK);
 290     getch2_status=1;
 291 }
 292
 293 void getch2_disable(void){
 294     if(!getch2_status) return; // already disabled / never enabled
 295 #ifdef HAVE_TERMIOS
 296     tcsetattr(0,TCSANOW,&tio_orig);
 297 #endif
 298     getch2_status=0;
 299 }
 300
 301 #ifdef CONFIG_ICONV
 302 char* get_term_charset(void)
 303 {
 304     char* charset = NULL;
 305 #ifdef HAVE_LANGINFO
 306     setlocale(LC_CTYPE, "");
 307     charset = strdup(nl_langinfo(CODESET));
 308     setlocale(LC_CTYPE, "C");
 309 #endif
 310     return charset;
 311 }
 312 #endif