#include "alarm.h"
#include <stdlib.h>
#include <unistd.h>
#include <sys/time.h>
#include <errno.h>
#include <signal.h>
typedef struct {
int sec;
void *ptr;
any_t fun;
}alarm_t;
static int inited;
static alarm_t *jobs[MAXALARM];
static struct sigaction oldact;
static struct itimerval olditv;
static void moduler_load(void);
static void alrm_handler(int s);
static void moduler_unload(void);
static int get_pos(void);
void alarm_destroy(int tb);
int alarm_init(int times,any_t any,void *s)
{
alarm_t *p =NULL;
int pos;
if(inited==0){
moduler_load();
inited=1;
}
p = malloc(sizeof(*p));
if (NULL == p)
return -ENOMEM;
p->sec=times;
p->ptr=s;
p->fun=any;
pos=get_pos();
if(pos<0){
free(p);
return -ENOSPC;
}
jobs[pos]=p;
return p;
}
static void moduler_load(void)
{
struct sigaction act;
struct itimerval itv;
act.sa_handler = alrm_handler;
act.sa_flags=0;
sigemptyset(&act.sa_mask);
sigaction(SIGALRM, &act, &oldact);
itv.it_interval.tv_sec = 1;
itv.it_interval.tv_usec = 0;
itv.it_value.tv_sec = 1;
itv.it_value.tv_usec = 0;
setitimer(ITIMER_REAL, &itv, &olditv);
atexit(moduler_unload);
}
static void alrm_handler(int s)
{
for (int i = 0; i < MAXALARM; i++) {
if (jobs[i] != NULL) {
if (jobs[i]->sec >= 1) {
jobs[i]->sec -=1;
} else {
jobs[i]->fun(jobs[i]->ptr);
alarm_destroy(i);
}
}
}
}
static void moduler_unload(void)
{
sigaction(SIGALRM, &oldact, NULL);
setitimer(ITIMER_REAL, &olditv, NULL);
}
static int get_pos(void)
{
for (int i = 0; i < MAXALARM; i++) {
if (jobs[i] == NULL)
return i;
}
return -1;
}
void alarm_destroy(int tb)
{
free(jobs[tb]);
jobs[tb] = NULL;
}
\\\\\\\\\\\\\\\\\\
#ifndef __ALARM_H
#define __ALARM_H
#define MAXALARM 1024
typedef void (*any_t)(void *s);
int alarm_init(int times,any_t any,void *s);
void alarm_destroy(int tb);
#endif
\\\\\\\\
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include "alarm.h"
static void any1(void *s)
{
printf("%s",(char *)s);
fflush(NULL);
}
static void any2(void *s)
{
printf("%s",(char *)s);
fflush(NULL);
}
static void any3(void *s)
{
printf("%s",(char *)s);
fflush(NULL);
}
int main(void)
{
int t1,t2, t3;
t1=alarm_init(3,any1,"hello");
t2=alarm_init(2,any2,"world");
t3=alarm_init(5,any3,"apue");
while(1){
write(1,"*",1);
sleep(1);
}
exit(0);
}