mirror of
https://github.com/kdave/btrfs-progs
synced 2024-12-24 23:22:27 +00:00
f3fbb9a398
periodic.timer_fd's value is 0 on inititlize-failed case, if no value-checking before read(), the code will run as read(STDIN). This patch fixed above case. Signed-off-by: Zhao Lei <zhaolei@cn.fujitsu.com> Signed-off-by: David Sterba <dsterba@suse.com>
144 lines
2.9 KiB
C
144 lines
2.9 KiB
C
/*
|
|
* 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;
|
|
}
|
|
}
|