btrfs-progs/task-utils.c
Zhao Lei f3fbb9a398 btrfs-progs: Check periodic.timer_fd's value before use
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>
2015-11-16 14:23:45 +01:00

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;
}
}