DNSClientHA/Timer.cpp at main · Confiared/DNSClientHA · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include "Timer.hpp"

#include <iostream>
#include <sys/timerfd.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <time.h>

char buff_temp[sizeof(uint64_t)];

Timer::Timer() :
    msec(0)
{
    this->kind=EpollObject::Kind::Kind_Timer;
}

Timer::~Timer()
{
    if(fd!=-1)
    {
        #ifdef DEBUGFASTCGI
        std::cerr << "close() fd: " << fd << " " << this << " " << __FILE__ << ":" << __LINE__ << std::endl;
        #endif
        ::close(fd);
    }
    //mostly bad usage for my own upper class
    //abort();
}

bool Timer::start(const unsigned int &msec)
{
    if(fd!=-1)
        return false;
    if(msec<1)
        return false;
    if((fd=::timerfd_create(CLOCK_REALTIME,TFD_NONBLOCK)) < 0)
    {
        std::cerr << "Timer creation error" << std::endl;
        return false;
    }

    timespec now;
    if (clock_gettime(CLOCK_REALTIME, &now) == -1)
    {
        std::cerr << "clock_gettime error" << std::endl;
        return false;
    }
    itimerspec new_value;
    new_value.it_value.tv_sec = now.tv_sec + 0;
    new_value.it_value.tv_nsec = now.tv_nsec + 0;
    if(new_value.it_value.tv_nsec>999999999)
    {
        new_value.it_value.tv_nsec-=1000000000;
        new_value.it_value.tv_sec++;
    }
    new_value.it_interval.tv_sec = msec/1000;
    new_value.it_interval.tv_nsec = (msec%1000)*1000000;
    if(new_value.it_interval.tv_nsec>999999999)
    {
        new_value.it_interval.tv_nsec-=1000000000;
        new_value.it_interval.tv_sec++;
    }

    const int &result=::timerfd_settime(fd, TFD_TIMER_ABSTIME, &new_value, NULL);
    if(result<0)
    {
        //settime error: 22: Invalid argument
        std::cerr << "settime error: " << errno << ": " << strerror(errno) << std::endl;
        return false;
    }
    epoll_event event;
    memset(&event,0,sizeof(event));
    event.data.ptr = this;
    event.events = EPOLLIN;// | EPOLLONESHOT: broke
    #ifdef DEBUGFASTCGI
    std::cerr << "EPOLL_CTL_ADD: " << event.data.ptr << " " << __FILE__ << ":" << __LINE__ << std::endl;
    #endif
    if(epoll_ctl(epollfd,EPOLL_CTL_ADD,fd,&event) < 0)
    {
        std::cerr << "epoll_ctl error" << std::endl;
        return false;
    }
    return true;
}

void Timer::validateTheTimer()
{
    if(::read(fd, buff_temp, sizeof(uint64_t))!=sizeof(uint64_t))
    {}
}

void Timer::parseEvent(const epoll_event &event)
{
    (void)event;
}