mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/kdave/btrfs-progs.git
synced 2024-11-15 08:14:21 +08:00
150 lines
3.0 KiB
C
150 lines
3.0 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 <time.h>
|
|
|
|
#include "common/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, time_t *start_time, u64 *item_count)
|
|
{
|
|
int ret;
|
|
|
|
if (!info)
|
|
return -1;
|
|
|
|
if (!info->threadfn)
|
|
return -1;
|
|
|
|
if (start_time)
|
|
*start_time = time(NULL);
|
|
if (item_count)
|
|
*item_count = 0;
|
|
|
|
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) {
|
|
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 * 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;
|
|
}
|
|
}
|