/* $Id$ */ #include "bbs.h" #define OBUFSIZE 2048 #define IBUFSIZE 128 #ifdef DEBUG #define register #endif /* realXbuf is Xbuf+3 because hz convert library requires buf[-2]. */ static unsigned char real_outbuf[OBUFSIZE+6] = " ", real_inbuf[IBUFSIZE+6] = " "; static unsigned char *outbuf = real_outbuf + 3, *inbuf = real_inbuf + 3; static int obufsize = 0, ibufsize = 0; static int icurrchar = 0; /* ----------------------------------------------------- */ /* convert routines */ /* ----------------------------------------------------- */ #ifdef CONVERT extern read_write_type write_type; extern read_write_type read_type; extern convert_type input_type; inline static ssize_t input_wrapper(void *buf, ssize_t count) { /* input_wrapper is a special case. * because we may do nothing, * a if-branch is better than a function-pointer call. */ if(input_type) return (*input_type)(buf, count); else return count; } inline static int read_wrapper(int fd, void *buf, size_t count) { return (*read_type)(fd, buf, count); } inline static int write_wrapper(int fd, void *buf, size_t count) { return (*write_type)(fd, buf, count); } #endif /* ----------------------------------------------------- */ /* output routines */ /* ----------------------------------------------------- */ void oflush(void) { if (obufsize) { STATINC(STAT_SYSWRITESOCKET); #ifdef CONVERT write_wrapper(1, outbuf, obufsize); #else write(1, outbuf, obufsize); #endif obufsize = 0; } } void init_buf(void) { memset(inbuf, 0, IBUFSIZE); } void output(const char *s, int len) { /* Invalid if len >= OBUFSIZE */ assert(len OBUFSIZE) { STATINC(STAT_SYSWRITESOCKET); #ifdef CONVERT write_wrapper(1, outbuf, obufsize); #else write(1, outbuf, obufsize); #endif obufsize = 0; } memcpy(outbuf + obufsize, s, len); obufsize += len; } int ochar(int c) { if (obufsize > OBUFSIZE - 1) { /* suppose one byte data doesn't need to be converted. */ write(1, outbuf, obufsize); obufsize = 0; } outbuf[obufsize++] = c; return 0; } /* ----------------------------------------------------- */ /* input routines */ /* ----------------------------------------------------- */ static int i_newfd = 0; static struct timeval i_to, *i_top = NULL; static int (*flushf) () = NULL; void add_io(int fd, int timeout) { i_newfd = fd; if (timeout) { i_to.tv_sec = timeout; i_to.tv_usec = 16384; /* Ptt: 改成16384 避免不按時for loop吃cpu * time 16384 約每秒64次 */ i_top = &i_to; } else i_top = NULL; } int num_in_buf(void) { return ibufsize - icurrchar; } /* * dogetch() is not reentrant-safe. SIGUSR[12] might happen at any time, and * dogetch() might be called again, and then ibufsize/icurrchar/inbuf might * be inconsistent. We try to not segfault here... */ static int dogetch(void) { ssize_t len; static time4_t lastact; if (ibufsize <= icurrchar) { if (flushf) (*flushf) (); refresh(); if (i_newfd) { struct timeval timeout; fd_set readfds; if (i_top) timeout = *i_top; /* copy it because select() might * change it */ FD_ZERO(&readfds); FD_SET(0, &readfds); FD_SET(i_newfd, &readfds); /* jochang: modify first argument of select from FD_SETSIZE */ /* since we are only waiting input from fd 0 and i_newfd(>0) */ STATINC(STAT_SYSSELECT); while ((len = select(i_newfd + 1, &readfds, NULL, NULL, i_top ? &timeout : NULL)) < 0) { if (errno != EINTR) abort_bbs(0); /* raise(SIGHUP); */ } if (len == 0){ #ifdef OUTTA_TIMER now = SHM->GV2.e.now; #else now = time(0); #endif return I_TIMEOUT; } if (i_newfd && FD_ISSET(i_newfd, &readfds)){ #ifdef OUTTA_TIMER now = SHM->GV2.e.now; #else now = time(0); #endif return I_OTHERDATA; } } #ifdef NOKILLWATERBALL if( currutmp && currutmp->msgcount && !reentrant_write_request ) write_request(1); #endif STATINC(STAT_SYSREADSOCKET); do { len = tty_read(inbuf, IBUFSIZE); /* tty_read will handle abort_bbs. * len <= 0: read more */ #ifdef CONVERT if(len > 0) len = input_wrapper(inbuf, len); #endif } while (len <= 0); ibufsize = len; icurrchar = 0; } if (currutmp) { #ifdef OUTTA_TIMER now = SHM->GV2.e.now; #else now = time(0); #endif /* 3 秒內超過兩 byte 才算 active, anti-antiidle. * 不過方向鍵等組合鍵不止 1 byte */ if (now - lastact < 3) currutmp->lastact = now; lastact = now; } return (unsigned char)inbuf[icurrchar++]; } #ifdef DEBUG /* * These are for terminal keys debug */ void _debug_print_ibuffer() { static int y = 0; int i = 0; move(y % b_lines, 0); for (i = 0; i < t_columns; i++) outc(' '); move(y % b_lines, 0); prints("%d. Current Buffer: %d/%d, ", y+1, icurrchar, ibufsize); outs(ANSI_COLOR(1) "[" ANSI_RESET); for (i = 0; i < ibufsize; i++) { int c = (unsigned char)inbuf[i]; if(c < ' ') { prints(ANSI_COLOR(1;33) "0x%02x" ANSI_RESET, c); } else { outc(c); } } outs(ANSI_COLOR(1) "]" ANSI_RESET); y++; move(y % b_lines, 0); for (i = 0; i < t_columns; i++) outc(' '); } int _debug_check_keyinput() { int dbcsaware = 0; int flExit = 0; clear(); while(!flExit) { int i = 0; move(b_lines, 0); for(i=0; i 0) dogetch(); } return 0; } #endif static int water_which_flag = 0; int igetch(void) { register int ch, mode = 0, last = 0; while (1) { ch = dogetch(); /* for escape codes, check * http://support.dell.com/support/edocs/systems/pe2650/en/ug/5g387ad0.htm */ if (mode == 0 && ch == KEY_ESC) mode = 1; else if (mode == 1) { /* Escape sequence */ if (ch == '[' || ch == 'O') { mode = 2; last = ch; } #if 0 /* some user complained about this since they wanna * do Esc-N paste in vedit. * Before anyone to explain what this is for, * this will be commented. */ else if (ch == '1' || ch == '4') /* what is this!? */ { mode = 3; last = ch; } #endif else { KEY_ESC_arg = ch; return KEY_ESC; } } else if (mode == 2) { /* ^[ or ^O, * ordered by frequency */ if(ch >= 'A' && ch <= 'D') /* Cursor key */ { return KEY_UP + (ch - 'A'); } else if (ch >= '1' && ch <= '6') /* Ins Del Home End PgUp PgDn */ { mode = 3; last = ch; continue; } else if(ch == 'O') { mode = 4; continue; } else if(ch == 'Z') { return KEY_STAB; } else if (ch == '0') { if (dogetch() == 'Z') return KEY_STAB; else return KEY_UNKNOWN; } else if (last == 'O') { /* ^[O ... */ if (ch >= 'A' && ch <= 'D') return KEY_UP + (ch - 'A'); if (ch >= 'P' && ch <= 'S') // vt100 k1-4 return KEY_F1 + (ch - 'P'); if (ch >= 'T' && ch <= '[') // putty vt100+ F5-F12 return KEY_F5 + (ch - 'T'); if (ch >= 't' && ch <= 'z') // vt100 F5-F11 return KEY_F5 + (ch - 't'); if (ch >= 'p' && ch <= 's') // Old (num or fn)kbd 4 keys return KEY_F1 + (ch - 'p'); else if (ch == 'a') // DELL spec return KEY_F12; } else return KEY_UNKNOWN; } else if (mode == 3) { /* ^[[1-6] */ /* ~: Ins Del Home End PgUp PgDn */ if(ch == '~') return KEY_HOME + (last - '1'); else if (last == '1') { if (ch >= '1' && ch <= '6') { dogetch(); /* must be '~' */ return KEY_F1 + ch - '1'; } else if (ch >= '7' && ch <= '9') { dogetch(); /* must be '~' */ return KEY_F6 + ch - '7'; } else return KEY_UNKNOWN; } else if (last == '2') { if (ch >= '0' && ch <= '4') { dogetch(); /* hope you are '~' */ return KEY_F9 + ch - '0'; } else return KEY_UNKNOWN; } } else // here is switch for default keys switch (ch) { // XXX: indent error #ifdef DEBUG case Ctrl('Q'):{ struct rusage ru; getrusage(RUSAGE_SELF, &ru); vmsg("sbrk: %d KB, idrss: %d KB, isrss: %d KB", ((int)sbrk(0) - 0x8048000) / 1024, (int)ru.ru_idrss, (int)ru.ru_isrss); } continue; #endif case Ctrl('L'): redoscr(); continue; case Ctrl('U'): if (currutmp != NULL && currutmp->mode != EDITING && currutmp->mode != LUSERS && currutmp->mode) { void *screen0; int oldroll = roll; int y, x, my_newfd; // FIXME reveal heap data or crash if screen resize getyx(&y, &x); screen0=malloc(screen_backupsize(t_lines, big_picture)); screen_backup(t_lines, big_picture, screen0); my_newfd = i_newfd; i_newfd = 0; t_users(); i_newfd = my_newfd; screen_restore(t_lines, big_picture, screen0); free(screen0); roll = oldroll; move(y, x); redoscr(); continue; } return ch; case KEY_TAB: if (WATERMODE(WATER_ORIG) || WATERMODE(WATER_NEW)) if (currutmp != NULL && watermode > 0) { watermode = (watermode + water_which->count) % water_which->count + 1; t_display_new(); continue; } return ch; case Ctrl('R'): if (currutmp == NULL) return (ch); if (currutmp->msgs[0].pid && WATERMODE(WATER_OFO) && wmofo == NOTREPLYING) { int y, x, my_newfd; void *screen0; screen0=malloc(screen_backupsize(t_lines, big_picture)); screen_backup(t_lines, big_picture, screen0); getyx(&y, &x); my_newfd = i_newfd; i_newfd = 0; my_write2(); screen_restore(t_lines, big_picture, screen0); free(screen0); i_newfd = my_newfd; move(y, x); redoscr(); continue; } else if (!WATERMODE(WATER_OFO)) { if (watermode > 0) { watermode = (watermode + water_which->count) % water_which->count + 1; t_display_new(); continue; } else if (currutmp->mode == 0 && (currutmp->chatid[0] == 2 || currutmp->chatid[0] == 3) && water_which->count != 0 && watermode == 0) { /* 第二次按 Ctrl-R */ watermode = 1; t_display_new(); continue; } else if (watermode == -1 && currutmp->msgs[0].pid) { /* 第一次按 Ctrl-R (必須先被丟過水球) */ void *screen0; int y, x, my_newfd; getyx(&y, &x); screen0=malloc(screen_backupsize(t_lines, big_picture)); screen_backup(t_lines, big_picture, screen0); /* 如果正在talk的話先不處理對方送過來的封包 (不去select) */ my_newfd = i_newfd; i_newfd = 0; show_call_in(0, 0); watermode = 0; #ifndef PLAY_ANGEL my_write(currutmp->msgs[0].pid, "水球丟過去： ", currutmp->msgs[0].userid, WATERBALL_GENERAL, NULL); #else switch (currutmp->msgs[0].msgmode) { case MSGMODE_TALK: case MSGMODE_WRITE: my_write(currutmp->msgs[0].pid, "水球丟過去： ", currutmp->msgs[0].userid, WATERBALL_GENERAL, NULL); break; case MSGMODE_FROMANGEL: my_write(currutmp->msgs[0].pid, "再問他一次： ", currutmp->msgs[0].userid, WATERBALL_ANGEL, NULL); break; case MSGMODE_TOANGEL: my_write(currutmp->msgs[0].pid, "回答小主人： ", currutmp->msgs[0].userid, WATERBALL_ANSWER, NULL); break; } #endif i_newfd = my_newfd; /* 還原螢幕 */ screen_restore(t_lines, big_picture, screen0); free(screen0); move(y, x); redoscr(); continue; } } return ch; case Ctrl('T'): if (WATERMODE(WATER_ORIG) || WATERMODE(WATER_NEW)) { if (watermode > 0) { if (watermode > 1) watermode--; else watermode = water_which->count; t_display_new(); continue; } } return ch; case Ctrl('F'): if (WATERMODE(WATER_NEW)) { if (watermode > 0) { if (water_which_flag == (int)water_usies) water_which_flag = 0; else water_which_flag = (water_which_flag + 1) % (int)(water_usies + 1); if (water_which_flag == 0) water_which = &water[0]; else water_which = swater[water_which_flag - 1]; watermode = 1; t_display_new(); continue; } } return ch; case Ctrl('G'): if (WATERMODE(WATER_NEW)) { if (watermode > 0) { water_which_flag = (water_which_flag + water_usies) % (water_usies + 1); if (water_which_flag == 0) water_which = &water[0]; else water_which = swater[water_which_flag - 1]; watermode = 1; t_display_new(); continue; } } return ch; case Ctrl('J'): /* Ptt 把 \n 拿掉 */ #ifdef PLAY_ANGEL /* Seams some telnet client still send CR LF when changing lines. CallAngel(); */ #endif continue; default: return ch; } } // should not reach here. just to make compiler happy. return 0; } /* * wait user input anything for f seconds. * if f < 0, then wait forever. * Return 1 if anything available. */ int wait_input(float f, int flDoRefresh) { int sel = 0; fd_set readfds; struct timeval tv, *ptv = &tv; if(num_in_buf() > 0) // for EINTR return 1; FD_ZERO(&readfds); FD_SET(0, &readfds); if(flDoRefresh) refresh(); if(f > 0) { tv.tv_sec = (long) f; tv.tv_usec = (f - (long)f) * 1000000L; } else ptv = NULL; #ifdef STATINC STATINC(STAT_SYSSELECT); #endif do { if(num_in_buf() > 0) return 1; sel = select(1, &readfds, NULL, NULL, ptv); } while (sel < 0 && errno == EINTR); /* EINTR, interrupted. I don't care! */ if(sel == 0) return 0; return 1; } /** * 根據 mode 來 strip 字串 str，並把結果存到 buf * @param buf * @param str * @param mode enum {STRIP_ALL = 0, ONLY_COLOR, NO_RELOAD}; * STRIP_ALL: ?? * ONLY_COLOR: ?? * NO_RELOAD: 不 strip (?) */ int strip_ansi(char *buf, const char *str, int mode) { register int count = 0; static const char EscapeFlag[] = { /* 0 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 10 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ,0, 0, 0, 0, 0, /* 20 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 30 */ 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 0, 0, /* 0~9 ;= */ /* 40 */ 0, 2, 2, 2, 2, 0, 0, 0, 2, 2, 2, 2, 0, 0, 0, 0, /* ABCDHIJK */ /* 50 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 60 */ 0, 0, 0, 0, 0, 0, 2, 0, 2, 0, 0, 0, 2, 2, 0, 0, /* fhlm */ /* 70 */ 0, 0, 0, 2, 0, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* su */ /* 80 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 90 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* A0 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* B0 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* C0 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* D0 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* E0 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* F0 */ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, }; #define isEscapeParam(X) (EscapeFlag[(int)(X)] & 1) #define isEscapeCommand(X) (EscapeFlag[(int)(X)] & 2) for(; *str; ++str) if( *str != ESC_CHR ){ if( buf ) *buf++ = *str; ++count; }else{ const char* p = str + 1; if( *p != '[' ){ ++str; continue; } while(isEscapeParam(*++p)); if( (mode == NO_RELOAD && isEscapeCommand(*p)) || (mode == ONLY_COLOR && *p == 'm' )){ register int len = p - str + 1; if( buf ){ strncpy(buf, str, len); buf += len; } count += len; } str = p; } if( buf ) *buf = 0; return count; } void strip_nonebig5(unsigned char *str, int maxlen) { int i; int len=0; for(i=0;i= 0) { if(sts == DBCS_LEADING) sts = DBCS_TRAILING; else if (*s >= 0x80) { sts = DBCS_LEADING; } else { sts = DBCS_ASCII; } s++, pos--; } return sts; } #else #define dbcs_off (1) #endif #define MAXLASTCMD 12 int oldgetdata(int line, int col, const char *prompt, char *buf, int len, int echo) { register int ch, i; int clen; int x = col, y = line; int dirty_line = 0; /* if this line contains ansi escapes, we have to dirty entire line. */ static char lastcmd[MAXLASTCMD][80]; unsigned char occupy_msg = 0; #ifdef DBCSAWARE_GETDATA unsigned int dbcsincomplete = 0; #endif strip_ansi(buf, buf, STRIP_ALL); if(line == b_lines-msg_occupied) occupy_msg=1, msg_occupied ++; if (prompt) { x += strip_ansi(NULL, prompt, STRIP_ALL); if(strlen(prompt) + col != x) dirty_line = 1; if(!echo || !dirty_line) { move(line, col); clrtoeol(); outs(prompt); } } if (!echo) { len--; clen = 0; while ((ch = igetch()) != '\r') { if (ch == '\177' || ch == Ctrl('H')) { if (!clen) { bell(); continue; } clen--; continue; } if (!isprint(ch)) { continue; } if (clen >= len) { continue; } buf[clen++] = ch; } buf[clen] = '\0'; outc('\n'); oflush(); } else { int cmdpos = 0; int currchar = 0; len--; buf[len] = '\0'; clen = currchar = strlen(buf); if(!dirty_line) { standout(); for(i=0; i<=len; i++) outc(' '); standend(); move(y, x); edit_outs(buf); } while (1) { if(dirty_line) { move(line, col); clrtoeol(); outs(prompt); standout(); for(i=0; i<=len; i++) { if(i <= clen) outc(buf[i]); else outc(' '); } // edit_outs(buf); standend(); } move(y, x + currchar); if ((ch = igetch()) == '\r') break; switch (ch) { case KEY_DOWN: case Ctrl('N'): case KEY_UP: case Ctrl('P'): strlcpy(lastcmd[cmdpos], buf, sizeof(lastcmd[0])); if (ch == KEY_UP || ch == Ctrl('P')) cmdpos++; else cmdpos += MAXLASTCMD - 1; cmdpos %= MAXLASTCMD; strlcpy(buf, lastcmd[cmdpos], len+1); if(!dirty_line) { move(y, x); /* clrtoeof */ for (i = 0; i <= clen; i++) outc(' '); move(y, x); edit_outs(buf); } clen = currchar = strlen(buf); break; case KEY_LEFT: if (currchar > 0) { --currchar; #ifdef DBCSAWARE_GETDATA if(currchar > 0 && ISDBCSAWARE() && getDBCSstatus(buf, currchar) == DBCS_TRAILING) currchar --; #endif } break; case KEY_RIGHT: if (buf[currchar]) { ++currchar; #ifdef DBCSAWARE_GETDATA if(buf[currchar] && ISDBCSAWARE() && getDBCSstatus(buf, currchar) == DBCS_TRAILING) currchar++; #endif } break; case '\177': case Ctrl('H'): if (currchar) { #ifdef DBCSAWARE_GETDATA int dbcs_off = 1; if (ISDBCSAWARE() && getDBCSstatus(buf, currchar-1) == DBCS_TRAILING) dbcs_off = 2; #endif currchar -= dbcs_off; clen -= dbcs_off; for (i = currchar; i <= clen; i++) buf[i] = buf[i + dbcs_off]; if(!dirty_line) { move(y, x + clen); outc(' '); #ifdef DBCSAWARE_GETDATA while(--dbcs_off > 0) outc(' '); #endif move(y, x); edit_outs(buf); } } break; case Ctrl('Y'): currchar = 0; case Ctrl('K'): /* we shoud be able to avoid DBCS issues in ^K mode */ buf[currchar] = '\0'; if(!dirty_line) { move(y, x + currchar); for (i = currchar; i < clen; i++) outc(' '); } clen = currchar; break; case Ctrl('D'): case KEY_DEL: if (buf[currchar]) { #ifdef DBCSAWARE_GETDATA int dbcs_off = 1; if (ISDBCSAWARE() && buf[currchar+1] && getDBCSstatus(buf, currchar+1) == DBCS_TRAILING) dbcs_off = 2; #endif clen -= dbcs_off; for (i = currchar; i <= clen; i++) buf[i] = buf[i + dbcs_off]; if(!dirty_line) { move(y, x + clen); outc(' '); #ifdef DBCSAWARE_GETDATA while(--dbcs_off > 0) outc(' '); #endif move(y, x); edit_outs(buf); } } break; case Ctrl('A'): case KEY_HOME: currchar = 0; break; case Ctrl('E'): case KEY_END: currchar = clen; break; default: if (isprint2(ch) && clen < len && x + clen < scr_cols) { #ifdef DBCSAWARE_GETDATA if(ISDBCSAWARE()) { /* to prevent single byte input */ if(dbcsincomplete) { dbcsincomplete = 0; } else if (ch >= 0x80) { dbcsincomplete = 1; if(clen + 2 > len) { /* we can't print this. ignore and eat key. */ igetch(); dbcsincomplete = 0; break; } } else { /* nothing, normal key. */ } } #endif for (i = clen + 1; i > currchar; i--) buf[i] = buf[i - 1]; buf[currchar] = ch; if(!dirty_line) { move(y, x + currchar); edit_outs(buf + currchar); } currchar++; clen++; } break; } /* end case */ } /* end while */ if (clen > 1) { strlcpy(lastcmd[0], buf, sizeof(lastcmd[0])); memmove(lastcmd+1, lastcmd, (MAXLASTCMD-1)*sizeof(lastcmd[0])); } /* why return here? because some code then outs.*/ // outc('\n'); move(y+1, 0); refresh(); } if ((echo == LCECHO) && isupper((int)buf[0])) buf[0] = tolower(buf[0]); if(occupy_msg) msg_occupied --; return clen; } /* Ptt */ int getdata_buf(int line, int col, const char *prompt, char *buf, int len, int echo) { return oldgetdata(line, col, prompt, buf, len, echo); } int getdata_str(int line, int col, const char *prompt, char *buf, int len, int echo, const char *defaultstr) { strlcpy(buf, defaultstr, len); return oldgetdata(line, col, prompt, buf, len, echo); } int getdata(int line, int col, const char *prompt, char *buf, int len, int echo) { buf[0] = 0; return oldgetdata(line, col, prompt, buf, len, echo); } /* vim:sw=4 */