/* * 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"task-utils.h"structtask_info*task_init(void*(*threadfn)(void*),int(*postfn)(void*),void*thread_private){structtask_info*info=calloc(1,sizeof(structtask_info));if(!info)returnNULL;info->private_data=thread_private;info->threadfn=threadfn;info->postfn=postfn;returninfo;}inttask_start(structtask_info*info,time_t*start_time,u64*item_count){intret;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;returnret;}voidtask_stop(structtask_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);}voidtask_deinit(structtask_info*info){if(!info)return;free(info);}inttask_period_start(structtask_info*info,unsignedintperiod_ms){unsignedintns;unsignedintsec;structitimerspecitval;if(!info)return-1;info->periodic.timer_fd=timerfd_create(CLOCK_MONOTONIC,0);if(info->periodic.timer_fd==-1){info->periodic.timer_fd=0;returninfo->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;returntimerfd_settime(info->periodic.timer_fd,0,&itval,NULL);};voidtask_period_wait(structtask_info*info){unsignedlonglongmissed;intret;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);}voidtask_period_stop(structtask_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;}}