学步园

今天看到一份shell源码，其中有用到不输入回车即可收到字符的功能，在这份源码中，它的作用是在shell中接收上下键，执行历史命令。

原始代码：

/* 如何实现不用回车,就能读取一个字符，此程序可以用来查看键盘的字符编码 */<br /> /*<br /> 以“上方向键”为例，3个字符分别是27<esc>, 79<O>,和65<A>。A><br /> 而上面的getch()函数中的read()函数1次只从标准输入读入1个字符，<br /> 所以在getch()读入1个字符返回后再次被调用时，标准输入的缓冲区中还有2个字符，<br /> 因此read()函数不会阻塞（尽管我们这时没有再按下任何键），而是读出第2个字符，<br /> 然后返回，然后又继续读入第3个字符。<br /> 所以按下1个方向键，getch()函数实际上被调用了3次。<br /> */</p> <p>#include<stdio.h><br /> #include<termio.h><br /> #include<stdlib.h></p> <p>static struct termios tty;<br /> void set_key();<br /> void reset_key();<br /> char getch();</p> <p>main()<br /> {<br /> char a;<br /> set_key();<br /> while(1)<br /> {<br /> a=getch();<br /> printf("%d%c/n",a,a);<br /> if(a=='0')<br /> break;<br /> }<br /> reset_key();<br /> }<br /> void set_key()<br /> {<br /> struct termios new_tty;<br /> tcgetattr(0,&tty);<br /> new_tty=tty;<br /> new_tty.c_lflag&=~(ICANON|ECHO|ISIG);<br /> new_tty.c_cc[VTIME]=0;<br /> new_tty.c_cc[VMIN]=1;<br /> tcsetattr(0,TCSANOW,&new_tty);<br /> return;<br /> }<br /> void reset_key()<br /> {<br /> tcsetattr(0,TCSANOW,&tty);<br /> return;<br /> }<br /> char getch()<br /> {<br /> int i;<br /> char ch;<br /> switch(i=read(0,&ch,1))<br /> {<br /> case -1:<br /> printf("error/n");<br /> exit(3);<br /> case 0:<br /> printf("error0/n");<br /> exit(4);<br /> }<br /> return(ch);<br /> }</p> <p>

升级版：

/* 改进版不用输入回车，自动获取字符, 这段代码的功能是屏蔽掉其它按键的输入，只接收ESC,<br /> 回车以及四个方向键的输入，并且显示用户的输入。如果用户按下ESC键，程序就会结束。*/</p> <p>#include <stdio.h><br /> #include <termio.h><br /> static struct termios tty;<br /> enum Key{None, KeyESCClick, KeyEnterClick,<br /> KeyUpClick, KeyDownClick,<br /> KeyLeftClick, KeyRightClick};<br /> void set_key();<br /> void reset_key();<br /> enum Key getch();</p> <p>int main(int argc, char *argv[])<br /> {<br /> enum Key a;<br /> set_key();<br /> while(1)<br /> {<br /> a=getch();<br /> switch(a)<br /> {<br /> case KeyESCClick:<br /> printf("esc/n");<br /> reset_key();<br /> return 0;<br /> case KeyEnterClick:<br /> printf("enter/n");<br /> break;<br /> case KeyUpClick:<br /> printf("up/n");<br /> break;<br /> case KeyDownClick:<br /> printf("down/n");<br /> break;<br /> case KeyLeftClick:<br /> printf("left/n");<br /> break;<br /> case KeyRightClick:<br /> printf("right/n");<br /> break;<br /> default:<br /> break;<br /> }<br /> }<br /> reset_key();<br /> return 0;<br /> }<br /> void set_key()<br /> {<br /> struct termios new_tty;<br /> tcgetattr(0,&tty);<br /> new_tty=tty;<br /> new_tty.c_lflag&=~(ICANON|ECHO|ISIG);<br /> new_tty.c_cc[VTIME]=0;<br /> new_tty.c_cc[VMIN]=1;<br /> tcsetattr(0,TCSANOW,&new_tty);<br /> return;<br /> }<br /> void reset_key()<br /> {<br /> tcsetattr(0,TCSANOW,&tty);<br /> return;<br /> }<br /> /*<br /> 修改了getch函数，由于上下左右方向键的字符编码特殊，所以在read中如果一次读取只有一个字符，那么就是普通字符，<br /> 如果是三个字符，那么就是方向键，再通过判断最后一个字符的内容就可以确定是方向键中的那一个。<br /> */<br /> enum Key getch()<br /> {<br /> int i;<br /> char ch[3];<br /> while((i = read(0, ch, 3)) != -1)<br /> {<br /> printf("get char num: %d/n", i);<br /> switch(i)<br /> {<br /> case 1:<br /> switch(ch[0])<br /> {<br /> case 0x1b:<br /> return KeyESCClick;<br /> case 0x0a:<br /> return KeyEnterClick;<br /> }<br /> break;<br /> case 3:<br /> switch(ch[2])<br /> {<br /> case 'A':<br /> return KeyUpClick;<br /> case 'B':<br /> return KeyDownClick;<br /> case 'C':<br /> return KeyRightClick;<br /> case 'D':<br /> return KeyLeftClick;<br /> }<br /> }<br /> }<br /> return None;<br /> }<br />