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	/*****************************************************************************
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	/**/
36	struct termios new_kbd_mode;
37
38	if (init)
39	return;
40	/* put keyboard (stdin, actually) in raw, unbuffered mode */
41	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	/* when we exit, go back to normal, "cooked" mode */
48	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	/* check stdin (fd 0) for activity */
62	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	if (status < 0)
67	{
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	/* stdin = fd 0 */
81	if (read(0, &temp, 1) != 1)
82	return 0;
83	return temp;
84	}
85	#endif