Initial commit of Pollcat library
This commit is contained in:
196
tests/struct.c
Normal file
196
tests/struct.c
Normal file
@@ -0,0 +1,196 @@
|
||||
/** Pollcat test stub
|
||||
*
|
||||
* This file is part of the Pollcat Library.
|
||||
* Copyright (C) 2022 Expatria Technologies Inc.
|
||||
* Contact: Morgan Hughes <morgan@expatria.ca>
|
||||
*
|
||||
* The Pollcat Library is free software: you can redistribute it and/or modify it under
|
||||
* the terms of the the GNU Lesser General Public License as published by the Free
|
||||
* Software Foundation; either version 3 of the License, or (at your option) any later
|
||||
* version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful, but WITHOUT ANY
|
||||
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
|
||||
* PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
||||
*
|
||||
* You should have received copies of the GNU General Public License and the GNU Lesser
|
||||
* General Public License along with the Pollcat Library. If not, see
|
||||
* https://www.gnu.org/licenses/
|
||||
*
|
||||
* vim:ts=4:noexpandtab
|
||||
*/
|
||||
#include <assert.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <stdarg.h>
|
||||
#include <string.h>
|
||||
#include <fcntl.h>
|
||||
#include <ctype.h>
|
||||
#include <errno.h>
|
||||
#include <unistd.h>
|
||||
#include <signal.h>
|
||||
#include <termios.h>
|
||||
|
||||
#include <pollcat.h>
|
||||
|
||||
|
||||
static void stdin_service (struct pollcat *cat)
|
||||
{
|
||||
char buff[16];
|
||||
int ret;
|
||||
int idx;
|
||||
|
||||
if ( !(pollcat_revents(fileno(stdin)) & POLLIN) )
|
||||
return;
|
||||
|
||||
memset(buff, 0, sizeof(buff));
|
||||
if ( (ret = read(fileno(stdin), buff, sizeof(buff) - 1)) < 0 )
|
||||
{
|
||||
fprintf(stderr, "recv(%d): %s", ret, strerror(errno));
|
||||
pollcat_loop_exit();
|
||||
return;
|
||||
}
|
||||
|
||||
fprintf(stderr, "recv(%d): '", ret);
|
||||
for ( idx = 0; idx < ret; idx++ )
|
||||
if ( isprint(buff[idx]) )
|
||||
fputc(buff[idx], stderr);
|
||||
else
|
||||
fprintf(stderr, "\\x%02x", buff[idx]);
|
||||
fputs("'\n", stderr);
|
||||
|
||||
if ( ret )
|
||||
switch ( tolower(buff[0]) )
|
||||
{
|
||||
case 'q':
|
||||
case '\033':
|
||||
case '\03':
|
||||
fprintf(stderr, "shutting down\n");
|
||||
pollcat_loop_exit();
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
const struct pollcat_ops stdin_ops =
|
||||
{
|
||||
.service = stdin_service,
|
||||
};
|
||||
|
||||
struct pollcat stdin_cat =
|
||||
{
|
||||
.ops = &stdin_ops,
|
||||
};
|
||||
|
||||
|
||||
struct pollcat_timer timer = { 0 };
|
||||
static void callback (void *arg)
|
||||
{
|
||||
printf("timer callback\n");
|
||||
pollcat_timer_attach(&timer, 5000, callback, NULL);
|
||||
}
|
||||
|
||||
|
||||
/** control for main loop for orderly shutdown */
|
||||
static void signal_fatal (int signum)
|
||||
{
|
||||
fprintf(stderr, "Fatal signal %d, exit loop\n", signum);
|
||||
pollcat_loop_exit();
|
||||
}
|
||||
|
||||
static void stop (const char *fmt, ...);
|
||||
|
||||
extern char *__progname;
|
||||
static void usage (void)
|
||||
{
|
||||
printf("Usage: %s -v [-t timeout]\n"
|
||||
"-v Verbose (set all modules to debug, enable trace)\n"
|
||||
"-t timeout Timeout for poll() calls (ms)\n"
|
||||
"",
|
||||
__progname);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
int main (int argc, char **argv)
|
||||
{
|
||||
int ret;
|
||||
|
||||
// basic arguments parsing
|
||||
int opt;
|
||||
while ( (opt = getopt(argc, argv, "vt:")) != -1 )
|
||||
switch ( opt )
|
||||
{
|
||||
case 'v':
|
||||
break;
|
||||
|
||||
case 't':
|
||||
pollcat_time_base = atoi(optarg);
|
||||
break;
|
||||
|
||||
default:
|
||||
usage();
|
||||
}
|
||||
|
||||
signal(SIGTERM, signal_fatal);
|
||||
signal(SIGINT, signal_fatal);
|
||||
|
||||
// reopen stdout/stderr before making stdin non-blocking
|
||||
char tty[256];
|
||||
memset(tty, 0, sizeof(tty));
|
||||
if ( readlink("/proc/self/fd/1", tty, sizeof(tty)) > 0 )
|
||||
{
|
||||
if ( !freopen(tty, "w", stdout) || !freopen(tty, "w", stderr) )
|
||||
perror(tty);
|
||||
setbuf(stderr, NULL);
|
||||
}
|
||||
|
||||
// set stdin nonblocking
|
||||
int flags = fcntl(fileno(stdin), F_GETFL);
|
||||
if ( flags < 0 )
|
||||
stop("fcntl(0, F_GETFL)");
|
||||
if ( fcntl(fileno(stdin), F_SETFL, flags | O_NONBLOCK) < 0 )
|
||||
stop("fcntl(0, F_SETFL)");
|
||||
|
||||
// save old terminal modes and make raw
|
||||
struct termios tio_old;
|
||||
if ( tcgetattr(fileno(stdin), &tio_old) )
|
||||
stop("tcgetattr(0, &tio_old)");
|
||||
|
||||
struct termios tio_new = tio_old;
|
||||
tio_new.c_lflag &= ~(ECHO|ICANON);
|
||||
if ( tcsetattr(fileno(stdin), TCSANOW, &tio_new) )
|
||||
stop("tcsetattr(0, TCSANOW, &tio_new)");
|
||||
|
||||
pollcat_fd_add(fileno(stdin), POLLIN);
|
||||
pollcat_iter_add(&stdin_cat);
|
||||
|
||||
pollcat_timer_attach(&timer, 5000, callback, NULL);
|
||||
|
||||
while ( pollcat_loop(&ret) )
|
||||
{
|
||||
}
|
||||
|
||||
pollcat_iter_remove(&stdin_cat);
|
||||
pollcat_fd_remove(fileno(stdin));
|
||||
|
||||
if ( fcntl(0, F_SETFL, flags) < 0 )
|
||||
stop("fcntl(0, F_SETFL)");
|
||||
|
||||
if ( tcsetattr(0, TCSANOW, &tio_old) )
|
||||
stop("tcsetattr(0, TCSANOW, &tio_old)");
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void stop (const char *fmt, ...)
|
||||
{
|
||||
int err = errno;
|
||||
va_list ap;
|
||||
|
||||
va_start(ap, fmt);
|
||||
vfprintf(stderr, fmt, ap);
|
||||
va_end(ap);
|
||||
|
||||
fprintf(stderr, ": %s\n", strerror(err));
|
||||
exit(1);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user