Signed-off-by: Silvio Fricke <silvio.fri...@gmail.com> --- task-util.c | 121 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ task-util.h | 33 +++++++++++++++++ 2 files changed, 154 insertions(+) create mode 100644 task-util.c create mode 100644 task-util.h
diff --git a/task-util.c b/task-util.c new file mode 100644 index 0000000..9268df7 --- /dev/null +++ b/task-util.c @@ -0,0 +1,121 @@ + +#include <pthread.h> +#include <sys/timerfd.h> +#include <stdio.h> +#include <stdlib.h> +#include <unistd.h> + +#include "task-util.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 == 0) + pthread_detach(info->id); + else + info->id = -1; + + return ret; +} + +void task_stop(struct task_info *info) +{ + if (!info) + return; + + if (info->periodic.timer_fd) + close(info->periodic.timer_fd); + + if (info->id > 0) + pthread_cancel(info->id); + + 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) + 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; + + ret = read(info->periodic.timer_fd, &missed, sizeof (missed)); + if (ret == -1) { + perror ("read timer"); + 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); + } +} diff --git a/task-util.h b/task-util.h new file mode 100644 index 0000000..95f7b5b --- /dev/null +++ b/task-util.h @@ -0,0 +1,33 @@ + +#ifndef __PROGRESS_ +#define __PROGRESS_ + +#include <pthread.h> + +struct periodic_info { + int timer_fd; + unsigned long long wakeups_missed; +}; + +struct task_info { + struct periodic_info periodic; + pthread_t id; + void *private_data; + void *(*threadfn)(void *); + int (*postfn)(void *); +}; + +/* task life cycle */ +struct task_info *task_init(void *(*threadfn)(void *), int (*postfn)(void *), + void *thread_private); +int task_start(struct task_info *info); +void task_stop(struct task_info *info); +void task_deinit(struct task_info *info); + +/* periodic life cycle */ +int task_period_start(struct task_info *info, unsigned int period_ms); +void task_period_wait(struct task_info *info); +void task_period_stop(struct task_info *info); + +#endif /* __PROGRESS_ */ + -- 2.1.2 -- To unsubscribe from this list: send the line "unsubscribe linux-btrfs" in the body of a message to majord...@vger.kernel.org More majordomo info at http://vger.kernel.org/majordomo-info.html