diff options
Diffstat (limited to 'task-utils.c')
-rw-r--r-- | task-utils.c | 143 |
1 files changed, 143 insertions, 0 deletions
diff --git a/task-utils.c b/task-utils.c new file mode 100644 index 00000000..12b00027 --- /dev/null +++ b/task-utils.c @@ -0,0 +1,143 @@ +/* + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public + * License v2 as published by the Free Software Foundation. + * + * 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 a copy of the GNU General Public + * License along with this program; if not, write to the + * Free Software Foundation, Inc., 59 Temple Place - Suite 330, + * Boston, MA 021110-1307, USA. + */ + +#include <pthread.h> +#include <sys/timerfd.h> +#include <stdio.h> +#include <stdlib.h> +#include <unistd.h> + +#include "task-utils.h" + +struct task_info *task_init(void *(*threadfn)(void *), int (*postfn)(void *), + void *thread_private) +{ + struct task_info *info = calloc(1, sizeof(struct task_info)); + + if (!info) + return NULL; + + info->private_data = thread_private; + info->threadfn = threadfn; + info->postfn = postfn; + + return info; +} + +int task_start(struct task_info *info) +{ + int ret; + + if (!info) + return -1; + + if (!info->threadfn) + return -1; + + ret = pthread_create(&info->id, NULL, info->threadfn, + info->private_data); + + if (ret) + info->id = 0; + + return ret; +} + +void task_stop(struct task_info *info) +{ + if (!info) + return; + + if (info->id > 0) { + pthread_cancel(info->id); + pthread_join(info->id, NULL); + info->id = 0; + } + + if (info->periodic.timer_fd) { + close(info->periodic.timer_fd); + info->periodic.timer_fd = 0; + } + + if (info->postfn) + info->postfn(info->private_data); +} + +void task_deinit(struct task_info *info) +{ + if (!info) + return; + + free(info); +} + +int task_period_start(struct task_info *info, unsigned int period_ms) +{ + unsigned int ns; + unsigned int sec; + struct itimerspec itval; + + if (!info) + return -1; + + info->periodic.timer_fd = timerfd_create(CLOCK_MONOTONIC, 0); + if (info->periodic.timer_fd == -1) { + info->periodic.timer_fd = 0; + return info->periodic.timer_fd; + } + + info->periodic.wakeups_missed = 0; + + sec = period_ms / 1000; + ns = (period_ms - (sec * 1000)) * 1000; + itval.it_interval.tv_sec = sec; + itval.it_interval.tv_nsec = ns; + itval.it_value.tv_sec = sec; + itval.it_value.tv_nsec = ns; + + return timerfd_settime(info->periodic.timer_fd, 0, &itval, NULL); +}; + +void task_period_wait(struct task_info *info) +{ + unsigned long long missed; + int ret; + + if (!info) + return; + + if (info->periodic.timer_fd == 0) + return; + + ret = read(info->periodic.timer_fd, &missed, sizeof (missed)); + if (ret == -1) + return; + + if (missed > 0) + info->periodic.wakeups_missed += (missed - 1); +} + +void task_period_stop(struct task_info *info) +{ + if (!info) + return; + + if (info->periodic.timer_fd) { + timerfd_settime(info->periodic.timer_fd, 0, NULL, NULL); + close(info->periodic.timer_fd); + info->periodic.timer_fd = -1; + } +} |