lib/base/ebase.cpp

   1 #include <lib/base/ebase.h>
   2
   3 #include <fcntl.h>
   4 #include <unistd.h>
   5 #include <errno.h>
   6
   7 #include <lib/base/eerror.h>
   8 #include <lib/base/elock.h>
   9
  10 eSocketNotifier::eSocketNotifier(eMainloop *context, int fd, int requested, bool startnow): context(*context), fd(fd), state(0), requested(requested)
  11 {
  12         if (startnow)
  13                 start();
  14 }
  15
  16 eSocketNotifier::~eSocketNotifier()
  17 {
  18         stop();
  19 }
  20
  21 void eSocketNotifier::start()
  22 {
  23         if (state)
  24                 stop();
  25
  26         context.addSocketNotifier(this);
  27         state=1;
  28 }
  29
  30 void eSocketNotifier::stop()
  31 {
  32         if (state)
  33                 context.removeSocketNotifier(this);
  34
  35         state=0;
  36 }
  37
  38                                         // timer
  39 void eTimer::start(long msek, bool singleShot)
  40 {
  41         if (bActive)
  42                 stop();
  43
  44         bActive = true;
  45         bSingleShot = singleShot;
  46         interval = msek;
  47         gettimeofday(&nextActivation, 0);
  48 //      eDebug("this = %p\nnow sec = %d, usec = %d\nadd %d msec", this, nextActivation.tv_sec, nextActivation.tv_usec, msek);
  49         nextActivation += (msek<0 ? 0 : msek);
  50 //      eDebug("next Activation sec = %d, usec = %d", nextActivation.tv_sec, nextActivation.tv_usec );
  51         context.addTimer(this);
  52 }
  53
  54 void eTimer::startLongTimer( int seconds )
  55 {
  56         if (bActive)
  57                 stop();
  58
  59         bActive = bSingleShot = true;
  60         interval = 0;
  61         gettimeofday(&nextActivation, 0);
  62 //      eDebug("this = %p\nnow sec = %d, usec = %d\nadd %d sec", this, nextActivation.tv_sec, nextActivation.tv_usec, seconds);
  63         if ( seconds > 0 )
  64                 nextActivation.tv_sec += seconds;
  65 //      eDebug("next Activation sec = %d, usec = %d", nextActivation.tv_sec, nextActivation.tv_usec );
  66         context.addTimer(this);
  67 }
  68
  69 void eTimer::stop()
  70 {
  71         if (bActive)
  72         {
  73                 bActive=false;
  74                 context.removeTimer(this);
  75         }
  76 }
  77
  78 void eTimer::changeInterval(long msek)
  79 {
  80         if (bActive)  // Timer is running?
  81         {
  82                 context.removeTimer(this);       // then stop
  83                 nextActivation -= interval;  // sub old interval
  84         }
  85         else
  86                 bActive=true; // then activate Timer
  87
  88         interval = msek;                                                // set new Interval
  89         nextActivation += interval;             // calc nextActivation
  90
  91         context.addTimer(this);                         // add Timer to context TimerList
  92 }
  93
  94 void eTimer::activate()   // Internal Funktion... called from eApplication
  95 {
  96         context.removeTimer(this);
  97
  98         if (!bSingleShot)
  99         {
 100                 nextActivation += interval;
 101                 context.addTimer(this);
 102         }
 103         else
 104                 bActive=false;
 105
 106         /*emit*/ timeout();
 107 }
 108
 109 void eTimer::addTimeOffset( int offset )
 110 {
 111         nextActivation.tv_sec += offset;
 112 }
 113
 114 // mainloop
 115 ePtrList<eMainloop> eMainloop::existing_loops;
 116
 117 void eMainloop::addSocketNotifier(eSocketNotifier *sn)
 118 {
 119         notifiers.insert(std::pair<int,eSocketNotifier*> (sn->getFD(), sn));
 120 }
 121
 122 void eMainloop::removeSocketNotifier(eSocketNotifier *sn)
 123 {
 124         for (std::multimap<int,eSocketNotifier*>::iterator i = notifiers.find(sn->getFD());
 125                         i != notifiers.end();
 126                         ++i)
 127                 if (i->second == sn)
 128                         return notifiers.erase(i);
 129         eFatal("removed socket notifier which is not present");
 130 }
 131
 132 int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, PyObject *additional)
 133 {
 134         int return_reason = 0;
 135                 /* get current time */
 136         timeval now;
 137         gettimeofday(&now, 0);
 138         m_now_is_invalid = 0;
 139
 140         if (additional && !PyDict_Check(additional))
 141                 eFatal("additional, but it's not dict");
 142
 143         if (additional && !res)
 144                 eFatal("additional, but no res");
 145
 146         int poll_timeout = -1; /* infinite in case of empty timer list */
 147
 148         if (m_timer_list)
 149         {
 150                 singleLock s(recalcLock);
 151                 poll_timeout = timeval_to_usec(m_timer_list.begin()->getNextActivation() - now);
 152                         /* if current timer already passed, don't delay infinite. */
 153                 if (poll_timeout < 0)
 154                         poll_timeout = 0;
 155
 156                         /* convert us to ms */
 157                 poll_timeout /= 1000;
 158         }
 159
 160         if ((user_timeout > 0) && (poll_timeout > 0) && ((unsigned int)poll_timeout > user_timeout))
 161         {
 162                 poll_timeout = user_timeout;
 163                 return_reason = 1;
 164         }
 165
 166         int ret = 0;
 167
 168         if (poll_timeout)
 169         {
 170                 std::multimap<int,eSocketNotifier*>::iterator it;
 171                 std::map<int,int> fd_merged;
 172                 std::map<int,int>::const_iterator fd_merged_it;
 173
 174                 for (it = notifiers.begin(); it != notifiers.end(); ++it)
 175                         fd_merged[it->first] |= it->second->getRequested();
 176
 177                 fd_merged_it = fd_merged.begin();
 178
 179                 int nativecount, fdcount;
 180
 181                 nativecount = fdcount = fd_merged.size();
 182
 183                 if (additional)
 184                 {
 185                         additional = PyDict_Items(additional);
 186                         fdcount += PyList_Size(additional);
 187                 }
 188
 189                         // build the poll aray
 190                 pollfd* pfd = new pollfd[fdcount];  // make new pollfd array
 191
 192                 for (int i=0; i < nativecount; i++, fd_merged_it++)
 193                 {
 194                         pfd[i].fd = fd_merged_it->first;
 195                         pfd[i].events = fd_merged_it->second;
 196                 }
 197
 198                 if (additional)
 199                 {
 200                         for (int i=0; i < PyList_Size(additional); ++i)
 201                         {
 202                                 PyObject *it = PyList_GET_ITEM(additional, i);
 203                                 if (!PyTuple_Check(it))
 204                                         eFatal("poll item is not a tuple");
 205                                 if (PyTuple_Size(it) != 2)
 206                                         eFatal("poll tuple size is not 2");
 207                                 int fd = PyObject_AsFileDescriptor(PyTuple_GET_ITEM(it, 0));
 208                                 if (fd == -1)
 209                                         eFatal("poll tuple not a filedescriptor");
 210                                 pfd[nativecount + i].fd = fd;
 211                                 pfd[nativecount + i].events = PyInt_AsLong(PyTuple_GET_ITEM(it, 1));
 212                         }
 213                 }
 214
 215                 ret = ::poll(pfd, fdcount, poll_timeout);
 216
 217                         /* ret > 0 means that there are some active poll entries. */
 218                 if (ret > 0)
 219                 {
 220                         return_reason = 0;
 221                         for (int i=0; i < nativecount ; i++)
 222                         {
 223                                 it = notifiers.begin();
 224
 225                                 int handled = 0;
 226
 227                                 std::multimap<int,eSocketNotifier*>::iterator
 228                                         l = notifiers.lower_bound(pfd[i].fd),
 229                                         u = notifiers.upper_bound(pfd[i].fd);
 230
 231                                 ePtrList<eSocketNotifier> n;
 232
 233                                 for (; l != u; ++l)
 234                                         n.push_back(l->second);
 235
 236                                 for (ePtrList<eSocketNotifier>::iterator li(n.begin()); li != n.end(); ++li)
 237                                 {
 238                                         int req = li->getRequested();
 239
 240                                         handled |= req;
 241
 242                                         if (pfd[i].revents & req)
 243                                                 (*li)->activate(pfd[i].revents);
 244                                 }
 245                                 if ((pfd[i].revents&~handled) & (POLLERR|POLLHUP|POLLNVAL))
 246                                         eDebug("poll: unhandled POLLERR/HUP/NVAL for fd %d(%d)", pfd[i].fd, pfd[i].revents);
 247                         }
 248
 249                         for (int i = nativecount; i < fdcount; ++i)
 250                         {
 251                                 if (pfd[i].revents)
 252                                 {
 253                                         if (!*res)
 254                                                 *res = PyList_New(0);
 255                                         PyObject *it = PyTuple_New(2);
 256                                         PyTuple_SET_ITEM(it, 0, PyInt_FromLong(pfd[i].fd));
 257                                         PyTuple_SET_ITEM(it, 1, PyInt_FromLong(pfd[i].revents));
 258                                         PyList_Append(*res, it);
 259                                         Py_DECREF(it);
 260                                 }
 261                         }
 262
 263                         ret = 1; /* poll did not timeout. */
 264                 } else if (ret < 0)
 265                 {
 266                                 /* when we got a signal, we get EINTR. */
 267                         if (errno != EINTR)
 268                                 eDebug("poll made error (%m)");
 269                         else
 270                         {
 271                                 return_reason = 2;
 272                                 ret = -1; /* don't assume the timeout has passed when we got a signal */
 273                         }
 274                 }
 275                 delete [] pfd;
 276                 Py_XDECREF(additional);
 277         }
 278
 279                 /* when we not processed anything, check timers. */
 280         if (!ret)
 281         {
 282                         /* we know that this time has passed. */
 283                 now += poll_timeout;
 284
 285                 singleLock s(recalcLock);
 286
 287                         /* this will never change while we have the recalcLock */
 288                         /* we can savely return here, the timer will be re-checked soon. */
 289                 if (m_now_is_invalid)
 290                         return 0;
 291
 292                         /* process all timers which are ready. first remove them out of the list. */
 293                 while ((!m_timer_list.empty()) && (m_timer_list.begin()->getNextActivation() <= now))
 294                         m_timer_list.begin()->activate();
 295         }
 296
 297         return return_reason;
 298 }
 299
 300 void eMainloop::addTimer(eTimer* e)
 301 {
 302         m_timer_list.insert_in_order(e);
 303 }
 304
 305 void eMainloop::removeTimer(eTimer* e)
 306 {
 307         m_timer_list.remove(e);
 308 }
 309
 310 int eMainloop::iterate(unsigned int user_timeout, PyObject **res, PyObject *dict)
 311 {
 312         int ret = 0;
 313
 314         timeval user_timer;
 315         gettimeofday(&user_timer, 0);
 316         user_timer += user_timeout;
 317
 318                 /* TODO: this code just became ugly. fix that. */
 319         do
 320         {
 321                 if (m_interrupt_requested)
 322                 {
 323                         m_interrupt_requested = 0;
 324                         return 0;
 325                 }
 326                 if (app_quit_now) return -1;
 327                 timeval now, timeout;
 328                 gettimeofday(&now, 0);
 329                 timeout = user_timer - now;
 330
 331                 if (user_timeout && (user_timer <= now))
 332                         return 0;
 333
 334                 int to = 0;
 335                 if (user_timeout)
 336                         to = timeout.tv_sec * 1000 + timeout.tv_usec / 1000;
 337
 338                 ret = processOneEvent(to, res, dict);
 339                 if (res && *res)
 340                         return ret;
 341         } while (ret == 0);
 342
 343         return ret;
 344 }
 345
 346 int eMainloop::runLoop()
 347 {
 348         while (!app_quit_now)
 349                 iterate();
 350         return retval;
 351 }
 352
 353 PyObject *eMainloop::poll(PyObject *timeout, PyObject *dict)
 354 {
 355         PyObject *res = 0;
 356
 357         if (app_quit_now)
 358         {
 359                 Py_INCREF(Py_None);
 360                 return Py_None;
 361         }
 362
 363         int user_timeout = (timeout == Py_None) ? 0 : PyInt_AsLong(timeout);
 364
 365         iterate(user_timeout, &res, dict);
 366
 367         if (!res) /* return empty list on timeout */
 368                 res = PyList_New(0);
 369
 370         return res;
 371 }
 372
 373 void eMainloop::interruptPoll()
 374 {
 375         m_interrupt_requested = 1;
 376 }
 377
 378 void eMainloop::quit(int ret)
 379 {
 380         retval = ret;
 381         app_quit_now = true;
 382 }
 383
 384 void eMainloop::addTimeOffset(int offset)
 385 {
 386         for (ePtrList<eMainloop>::iterator it(eMainloop::existing_loops)
 387                 ;it != eMainloop::existing_loops.end(); ++it)
 388         {
 389                 singleLock s(it->recalcLock);
 390                 it->m_now_is_invalid = 1;
 391                 for (ePtrList<eTimer>::iterator tit = it->m_timer_list.begin(); tit != it->m_timer_list.end(); ++tit )
 392                         tit->addTimeOffset(offset);
 393         }
 394 }
 395
 396 eApplication* eApp = 0;