projects/smsdaemon/kbhit.cpp

/*****************************************************************************
kbhit() and getch() for Linux/UNIX
Chris Giese <geezer@execpc.com> http://my.execpc.com/~geezer
Release date: ?
This code is public domain (no copyright).
You can do whatever you want with it.
*****************************************************************************/
#if !defined(linux)
#include <conio.h> /* kbhit(), getch() */

#else
#include <sys/time.h> /* struct timeval, select() */
/* ICANON, ECHO, TCSANOW, struct termios */
#include <termios.h> /* tcgetattr(), tcsetattr() */
#include <stdlib.h> /* atexit(), exit() */
#include <unistd.h> /* read() */
#include <stdio.h> /* printf() */

#include <string.h>

#include "kbhit.h"

static struct termios g_old_kbd_mode;
/*****************************************************************************
*****************************************************************************/
void cooked(void)
{
        tcsetattr(0, TCSANOW, &g_old_kbd_mode);
}
/*****************************************************************************
*****************************************************************************/
void raw(void)
{
        static char init;
        /**/
        struct termios new_kbd_mode;

        if (init)
                return;
        /* put keyboard (stdin, actually) in raw, unbuffered mode */
        tcgetattr(0, &g_old_kbd_mode);
        memcpy(&new_kbd_mode, &g_old_kbd_mode, sizeof(struct termios));
        new_kbd_mode.c_lflag &= ~(ICANON | ECHO);
        new_kbd_mode.c_cc[VTIME] = 0;
        new_kbd_mode.c_cc[VMIN] = 1;
        tcsetattr(0, TCSANOW, &new_kbd_mode);
        /* when we exit, go back to normal, "cooked" mode */
        atexit(cooked);

        init = 1;
}
/*****************************************************************************
*****************************************************************************/
int kbhit(void)
{
        struct timeval timeout;
        fd_set read_handles;
        int status;

        raw();
        /* check stdin (fd 0) for activity */
        FD_ZERO(&read_handles);
        FD_SET(0, &read_handles);
        timeout.tv_sec = timeout.tv_usec = 0;
        status = select(0 + 1, &read_handles, NULL, NULL, &timeout);
        if (status < 0)
        {
                printf("select() failed in kbhit()\n");
                exit(1);
        }
        return status;
}
/*****************************************************************************
*****************************************************************************/
int getch(void)
{
        unsigned char temp;

        raw();
        /* stdin = fd 0 */
        if (read(0, &temp, 1) != 1)
                return 0;
        return temp;
}
#endif
1	torben	26	/*****************************************************************************
2			kbhit() and getch() for Linux/UNIX
3			Chris Giese <geezer@execpc.com> http://my.execpc.com/~geezer
4			Release date: ?
5			This code is public domain (no copyright).
6			You can do whatever you want with it.
7			*****************************************************************************/
8			#if !defined(linux)
9			#include <conio.h> /* kbhit(), getch() */
10
11			#else
12			#include <sys/time.h> /* struct timeval, select() */
13			/* ICANON, ECHO, TCSANOW, struct termios */
14			#include <termios.h> /* tcgetattr(), tcsetattr() */
15			#include <stdlib.h> /* atexit(), exit() */
16			#include <unistd.h> /* read() */
17			#include <stdio.h> /* printf() */
18
19			#include <string.h>
20
21			#include "kbhit.h"
22
23			static struct termios g_old_kbd_mode;
24			/*****************************************************************************
25			*****************************************************************************/
26			void cooked(void)
27			{
28			tcsetattr(0, TCSANOW, &g_old_kbd_mode);
29			}
30			/*****************************************************************************
31			*****************************************************************************/
32			void raw(void)
33			{
34			static char init;
35	torben	196	/**/
36	torben	26	struct termios new_kbd_mode;
37
38	torben	196	if (init)
39	torben	26	return;
40	torben	196	/* put keyboard (stdin, actually) in raw, unbuffered mode */
41	torben	26	tcgetattr(0, &g_old_kbd_mode);
42			memcpy(&new_kbd_mode, &g_old_kbd_mode, sizeof(struct termios));
43			new_kbd_mode.c_lflag &= ~(ICANON \| ECHO);
44			new_kbd_mode.c_cc[VTIME] = 0;
45			new_kbd_mode.c_cc[VMIN] = 1;
46			tcsetattr(0, TCSANOW, &new_kbd_mode);
47	torben	196	/* when we exit, go back to normal, "cooked" mode */
48	torben	26	atexit(cooked);
49
50			init = 1;
51			}
52			/*****************************************************************************
53			*****************************************************************************/
54			int kbhit(void)
55			{
56			struct timeval timeout;
57			fd_set read_handles;
58			int status;
59
60			raw();
61	torben	196	/* check stdin (fd 0) for activity */
62	torben	26	FD_ZERO(&read_handles);
63			FD_SET(0, &read_handles);
64			timeout.tv_sec = timeout.tv_usec = 0;
65			status = select(0 + 1, &read_handles, NULL, NULL, &timeout);
66	torben	196	if (status < 0)
67	torben	26	{
68			printf("select() failed in kbhit()\n");
69			exit(1);
70			}
71			return status;
72			}
73			/*****************************************************************************
74			*****************************************************************************/
75			int getch(void)
76			{
77			unsigned char temp;
78
79			raw();
80	torben	196	/* stdin = fd 0 */
81			if (read(0, &temp, 1) != 1)
82	torben	26	return 0;
83			return temp;
84			}
85			#endif