osdep/getch2.c

   1 /* GyS-TermIO v2.0 (for GySmail v3)          (C) 1999 A'rpi/ESP-team */
   2
   3 #include "config.h"
   4
   5 //#define USE_TERMCAP
   6 #if !defined(__OS2__) && !defined(__MORPHOS__)
   7 #define USE_IOCTL
   8 #endif
   9
  10 #define MAX_KEYS 64
  11 #define BUF_LEN 256
  12
  13 #include <stdio.h>
  14 #include <stdlib.h>
  15 #include <string.h>
  16 #include <sys/time.h>
  17 #include <sys/types.h>
  18 #ifdef USE_IOCTL
  19 #include <sys/ioctl.h>
  20 #endif
  21
  22 #ifdef HAVE_TERMIOS
  23 #ifdef HAVE_TERMIOS_H
  24 #include <termios.h>
  25 #endif
  26 #ifdef HAVE_SYS_TERMIOS_H
  27 #include <sys/termios.h>
  28 #endif
  29 #endif
  30
  31 #include <unistd.h>
  32
  33 #include "keycodes.h"
  34
  35 #ifdef HAVE_TERMIOS
  36 static struct termios tio_orig;
  37 #endif
  38 static int getch2_len=0;
  39 static char getch2_buf[BUF_LEN];
  40
  41 int screen_width=80;
  42 int screen_height=24;
  43 char * erase_to_end_of_line = NULL;
  44
  45 typedef struct {
  46   int len;
  47   int code;
  48   char chars[8];
  49 } keycode_st;
  50 static keycode_st getch2_keys[MAX_KEYS];
  51 static int getch2_key_db=0;
  52
  53 #ifdef USE_TERMCAP
  54
  55 #if 0
  56 #include <termcap.h>
  57 #else
  58   extern int tgetent (char *BUFFER, char *TERMTYPE);
  59   extern int tgetnum (char *NAME);
  60   extern int tgetflag (char *NAME);
  61   extern char *tgetstr (char *NAME, char **AREA);
  62 #endif
  63
  64 static char term_buffer[4096];
  65 static char term_buffer2[4096];
  66 static char *term_p=term_buffer2;
  67
  68 static void termcap_add(char *id,int code){
  69 char *p=tgetstr(id,&term_p);
  70   if(!p) return;
  71   if(getch2_key_db>=MAX_KEYS) return;
  72   getch2_keys[getch2_key_db].len=strlen(p);
  73   strncpy(getch2_keys[getch2_key_db].chars,p,8);
  74   getch2_keys[getch2_key_db].code=code;
  75   ++getch2_key_db;
  76 /*  printf("%s=%s\n",id,p); */
  77 }
  78
  79 static int success=0;
  80
  81 int load_termcap(char *termtype){
  82   if(!termtype) termtype=getenv("TERM");
  83   if(!termtype) termtype="unknown";
  84   success=tgetent(term_buffer, termtype);
  85   if(success<0){ printf("Could not access the 'termcap' data base.\n"); return 0; }
  86   if(success==0){ printf("Terminal type `%s' is not defined.\n", termtype);return 0;}
  87
  88   screen_width=tgetnum("co");
  89   screen_height=tgetnum("li");
  90   if(screen_width<1 || screen_width>255) screen_width=80;
  91   if(screen_height<1 || screen_height>255) screen_height=24;
  92   erase_to_end_of_line= tgetstr("cd",&term_p);
  93
  94   termcap_add("kP",KEY_PGUP);
  95   termcap_add("kN",KEY_PGDWN);
  96   termcap_add("kh",KEY_HOME);
  97   termcap_add("kH",KEY_END);
  98   termcap_add("kI",KEY_INS);
  99   termcap_add("kD",KEY_DEL);
 100   termcap_add("kb",KEY_BS);
 101   termcap_add("kl",KEY_LEFT);
 102   termcap_add("kd",KEY_DOWN);
 103   termcap_add("ku",KEY_UP);
 104   termcap_add("kr",KEY_RIGHT);
 105   termcap_add("k0",KEY_F+0);
 106   termcap_add("k1",KEY_F+1);
 107   termcap_add("k2",KEY_F+2);
 108   termcap_add("k3",KEY_F+3);
 109   termcap_add("k4",KEY_F+4);
 110   termcap_add("k5",KEY_F+5);
 111   termcap_add("k6",KEY_F+6);
 112   termcap_add("k7",KEY_F+7);
 113   termcap_add("k8",KEY_F+8);
 114   termcap_add("k9",KEY_F+9);
 115   termcap_add("k;",KEY_F+10);
 116   return getch2_key_db;
 117 }
 118
 119 #endif
 120
 121 void get_screen_size(void){
 122 #ifdef USE_IOCTL
 123   struct winsize ws;
 124   if (ioctl(0, TIOCGWINSZ, &ws) < 0 || !ws.ws_row || !ws.ws_col) return;
 125 /*  printf("Using IOCTL\n"); */
 126   screen_width=ws.ws_col;
 127   screen_height=ws.ws_row;
 128 #endif
 129 }
 130
 131 int getch2(int time){
 132   int len=0;
 133   int code=0;
 134   int i=0;
 135
 136   while(!getch2_len || (getch2_len==1 && getch2_buf[0]==27)){
 137     fd_set rfds;
 138     struct timeval tv;
 139     int retval;
 140     /* Watch stdin (fd 0) to see when it has input. */
 141     FD_ZERO(&rfds); FD_SET(0,&rfds);
 142     /* Wait up to 'time' microseconds. */
 143     tv.tv_sec=time/1000; tv.tv_usec = (time%1000)*1000;
 144     retval=select(1, &rfds, NULL, NULL, &tv);
 145     if(retval<=0) return -1;
 146     /* Data is available now. */
 147     retval=read(0,&getch2_buf[getch2_len],BUF_LEN-getch2_len);
 148     if(retval<1) return -1;
 149     getch2_len+=retval;
 150   }
 151
 152     /* First find in the TERMCAP database: */
 153     for(i=0;i<getch2_key_db;i++){
 154       if((len=getch2_keys[i].len)<=getch2_len)
 155         if(memcmp(getch2_keys[i].chars,getch2_buf,len)==0){
 156           code=getch2_keys[i].code; goto found;
 157         }
 158     }
 159     len=1;code=getch2_buf[0];
 160     /* Check the well-known codes... */
 161     if(code!=27){
 162       if(code=='A'-64){ code=KEY_HOME; goto found;}
 163       if(code=='E'-64){ code=KEY_END; goto found;}
 164       if(code=='D'-64){ code=KEY_DEL; goto found;}
 165       if(code=='H'-64){ code=KEY_BS; goto found;}
 166       if(code=='U'-64){ code=KEY_PGUP; goto found;}
 167       if(code=='V'-64){ code=KEY_PGDWN; goto found;}
 168       if(code==8 || code==127){ code=KEY_BS; goto found;}
 169       if(code==10 || code==13){
 170         if(getch2_len>1){
 171           int c=getch2_buf[1];
 172           if(c==10 || c==13) if(c!=code) len=2;
 173         }
 174         code=KEY_ENTER;
 175         goto found;
 176       }
 177     } else if(getch2_len>1){
 178       int c=getch2_buf[1];
 179       if(c==27){ code=KEY_ESC; len=2; goto found;}
 180       if(c>='0' && c<='9'){ code=c-'0'+KEY_F; len=2; goto found;}
 181       if(getch2_len>=4 && c=='[' && getch2_buf[2]=='['){
 182         int c=getch2_buf[3];
 183         if(c>='A' && c<'A'+12){ code=KEY_F+1+c-'A';len=4;goto found;}
 184       }
 185       if(c=='[' || c=='O') if(getch2_len>=3){
 186         int c=getch2_buf[2];
 187         static short int ctable[]={ KEY_UP,KEY_DOWN,KEY_RIGHT,KEY_LEFT,0,
 188                        KEY_END,KEY_PGDWN,KEY_HOME,KEY_PGUP,0,0,KEY_INS,0,0,0,
 189                        KEY_F+1,KEY_F+2,KEY_F+3,KEY_F+4};
 190         if(c>='A' && c<='S')
 191           if(ctable[c-'A']){ code=ctable[c-'A']; len=3; goto found;}
 192       }
 193       if(getch2_len>=4 && c=='[' && getch2_buf[3]=='~'){
 194         int c=getch2_buf[2];
 195         int ctable[8]={KEY_HOME,KEY_INS,KEY_DEL,KEY_END,KEY_PGUP,KEY_PGDWN,KEY_HOME,KEY_END};
 196         if(c>='1' && c<='8'){ code=ctable[c-'1']; len=4; goto found;}
 197       }
 198       if(getch2_len>=5 && c=='[' && getch2_buf[4]=='~'){
 199         int i=getch2_buf[2]-'0';
 200         int j=getch2_buf[3]-'0';
 201         if(i>=0 && i<=9 && j>=0 && j<=9){
 202           static short int ftable[20]={
 203             11,12,13,14,15, 17,18,19,20,21,
 204             23,24,25,26,28, 29,31,32,33,34 };
 205           int a=i*10+j;
 206           for(i=0;i<20;i++) if(ftable[i]==a){ code=KEY_F+1+i;len=5;goto found;}
 207         }
 208       }
 209     }
 210 found:
 211   if((getch2_len-=len)>0){
 212     int i;
 213     for(i=0;i<getch2_len;i++) getch2_buf[i]=getch2_buf[len+i];
 214   }
 215   return code;
 216 }
 217
 218 static int getch2_status=0;
 219
 220 void getch2_enable(void){
 221 #ifdef HAVE_TERMIOS
 222 struct termios tio_new;
 223 #if defined(__NetBSD__) || defined(__svr4__) || defined(__CYGWIN__) || defined(__OS2__) || defined(__GLIBC__) || defined(_AIX)
 224     tcgetattr(0,&tio_orig);
 225 #elif defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__bsdi__) || defined(__APPLE__) || defined(__DragonFly__)
 226     ioctl(0,TIOCGETA,&tio_orig);
 227 #else
 228     ioctl(0,TCGETS,&tio_orig);
 229 #endif
 230     tio_new=tio_orig;
 231     tio_new.c_lflag &= ~(ICANON|ECHO); /* Clear ICANON and ECHO. */
 232     tio_new.c_cc[VMIN] = 1;
 233     tio_new.c_cc[VTIME] = 0;
 234 #if defined(__NetBSD__) || defined(__svr4__) || defined(__CYGWIN__) || defined(__OS2__) || defined(__GLIBC__) || defined(_AIX)
 235     tcsetattr(0,TCSANOW,&tio_new);
 236 #elif defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__bsdi__) || defined(__APPLE__) || defined(__DragonFly__)
 237     ioctl(0,TIOCSETA,&tio_new);
 238 #else
 239     ioctl(0,TCSETS,&tio_new);
 240 #endif
 241 #endif
 242     getch2_status=1;
 243 }
 244
 245 void getch2_disable(void){
 246     if(!getch2_status) return; // already disabled / never enabled
 247 #ifdef HAVE_TERMIOS
 248 #if defined(__NetBSD__) || defined(__svr4__) || defined(__CYGWIN__) || defined(__OS2__) || defined(__GLIBC__) || defined(_AIX)
 249     tcsetattr(0,TCSANOW,&tio_orig);
 250 #elif defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__bsdi__) || defined(__APPLE__) || defined(__DragonFly__)
 251     ioctl(0,TIOCSETA,&tio_orig);
 252 #else
 253     ioctl(0,TCSETS,&tio_orig);
 254 #endif
 255 #endif
 256     getch2_status=0;
 257 }
 258