fix ASSERT in the correct way
[enigma2.git] / lib / base / ebase.cpp
index 7f84cc91196fb45fac7f9f13a3900a1379804578..f40bf39e78187bf8b45e527554165fff4fde739e 100644 (file)
@@ -45,8 +45,8 @@ void eTimer::start(long msek, bool singleShot)
        bSingleShot = singleShot;
        interval = msek;
        gettimeofday(&nextActivation, 0);
-       nextActivation += (msek<0 ? 0 : msek);
 //     eDebug("this = %p\nnow sec = %d, usec = %d\nadd %d msec", this, nextActivation.tv_sec, nextActivation.tv_usec, msek);
+       nextActivation += (msek<0 ? 0 : msek);
 //     eDebug("next Activation sec = %d, usec = %d", nextActivation.tv_sec, nextActivation.tv_usec );
        context.addTimer(this);
 }
@@ -116,54 +116,40 @@ ePtrList<eMainloop> eMainloop::existing_loops;
 
 void eMainloop::addSocketNotifier(eSocketNotifier *sn)
 {
-       notifiers.insert(std::pair<int,eSocketNotifier*> (sn->getFD(), sn));
+       int fd = sn->getFD();
+       ASSERT(notifiers.find(fd) == notifiers.end());
+       ASSERT(new_notifiers.find(fd) == new_notifiers.end());
+       new_notifiers[fd]=sn;
 }
 
 void eMainloop::removeSocketNotifier(eSocketNotifier *sn)
 {
-       notifiers.erase(sn->getFD());
+       int fd = sn->getFD();
+       std::map<int,eSocketNotifier*>::iterator i(notifiers.find(fd));
+       if (i != notifiers.end())
+               return notifiers.erase(i);
+       i = new_notifiers.find(fd);
+       if (i != new_notifiers.end())
+               return new_notifiers.erase(i);
+       eFatal("removed socket notifier which is not present");
 }
 
-void eMainloop::processOneEvent()
+int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, ePyObject additional)
 {
-               /* notes:
-                 - we should use epoll(4)
-                 - timer are checked twice. there was a strong reason for it, but i can't remember. (FIXME)
-                 - for each time, we gettimeofday() and check wether the timer should fire.
-                   we should do this all better - we know how long the poll last, so we know which
-                   timers should fire. Problem is that a timer handler could have required so
-                   much time that another timer fired.
-
-                   A probably structure could look
-
-                   while (1)
-                   {
-                           time = gettimeofday()
-                           timeout = calculate_pending_timers(time);
-
-                     doPoll(timeout or infinite);
-
-                       if (poll_had_results)
-                               handle_poll_handler();
-                       else
-                                   fire_timers(time + timeout)
-                         }
-
-                         the gettimeofday() call is required because fire_timers could last more
-                         than nothing.
-
-                         when poll did no timeout, we don't handle timers, as this will be done
-                         in the next iteration (without adding overhead - we had to get the new
-                         time anyway
-               */
-
+       int return_reason = 0;
                /* get current time */
        timeval now;
        gettimeofday(&now, 0);
        m_now_is_invalid = 0;
-       
+               
+       if (additional && !PyDict_Check(additional))
+               eFatal("additional, but it's not dict");
+               
+       if (additional && !res)
+               eFatal("additional, but no res");
+               
        int poll_timeout = -1; /* infinite in case of empty timer list */
-       
+               
        if (m_timer_list)
        {
                singleLock s(recalcLock);
@@ -176,68 +162,113 @@ void eMainloop::processOneEvent()
                poll_timeout /= 1000;
        }
        
-       int ret = 0;
+       if ((user_timeout > 0) && (poll_timeout > 0) && ((unsigned int)poll_timeout > user_timeout))
+       {
+               poll_timeout = user_timeout;
+               return_reason = 1;
+       }
 
-       if (poll_timeout)
+       for (std::map<int, eSocketNotifier*>::iterator it(new_notifiers.begin()); it != new_notifiers.end();)
        {
-                       // build the poll aray
-               int fdcount = notifiers.size();
-               pollfd* pfd = new pollfd[fdcount];  // make new pollfd array
+               notifiers[it->first]=it->second;
+               new_notifiers.erase(it++);
+       }
 
-               std::map<int,eSocketNotifier*>::iterator it(notifiers.begin());
-               for (int i=0; i < fdcount; i++, it++)
-               {
-                       pfd[i].fd = it->first;
-                       pfd[i].events = it->second->getRequested();
+       int nativecount=notifiers.size(),
+               fdcount=nativecount,
+               ret=0;
+
+       if (additional)
+               fdcount += PyDict_Size(additional);
+               
+               // build the poll aray
+       pollfd pfd[fdcount];  // make new pollfd array
+       std::map<int,eSocketNotifier*>::iterator it = notifiers.begin();
+       int i=0;
+       for (; i < nativecount; ++i, ++it)
+       {
+               pfd[i].fd = it->first;
+               pfd[i].events = it->second->getRequested();
+       }
+       
+       if (additional)
+       {
+               PyObject *key, *val;
+               int pos=0;
+               while (PyDict_Next(additional, &pos, &key, &val)) {
+                       pfd[i].fd = PyObject_AsFileDescriptor(key);
+                       pfd[i++].events = PyInt_AsLong(val);
                }
-
-               ret = poll(pfd, fdcount, poll_timeout);
-
+       }
+       
+       ret = ::poll(pfd, fdcount, poll_timeout);
                        /* ret > 0 means that there are some active poll entries. */
-               if (ret > 0)
+       if (ret > 0)
+       {
+               int i=0;
+               return_reason = 0;
+               for (; i < nativecount ; ++i)
                {
-                       for (int i=0; i < fdcount ; i++)
+                       if (pfd[i].revents)
                        {
-                               if (notifiers.find(pfd[i].fd) == notifiers.end())
-                                       continue;
-                               
-                               int req = notifiers[pfd[i].fd]->getRequested();
-                               
-                               if (pfd[i].revents & req)
+                               int handled = 0;
+                               it = notifiers.find(pfd[i].fd);
+                               if (it != notifiers.end())
                                {
-                                       notifiers[pfd[i].fd]->activate(pfd[i].revents);
-                               
-                                       if (!--ret)
-                                               break;
-                               } else if (pfd[i].revents & (POLLERR|POLLHUP|POLLNVAL))
-                                       eFatal("poll: unhandled POLLERR/HUP/NVAL for fd %d(%d) -> FIX YOUR CODE", pfd[i].fd,pfd[i].revents);
+                                       int req = it->second->getRequested();
+                                       handled |= req;
+                                       if (pfd[i].revents & req)
+                                               it->second->activate(pfd[i].revents);
+                               }
+                               pfd[i].revents &= ~handled;
+                               if (pfd[i].revents & (POLLERR|POLLHUP|POLLNVAL))
+                                       eDebug("poll: unhandled POLLERR/HUP/NVAL for fd %d(%d)", pfd[i].fd, pfd[i].revents);
+                       }
+               }
+               for (; i < fdcount; ++i)
+               {
+                       if (pfd[i].revents)
+                       {
+                               if (!*res)
+                                       *res = PyList_New(0);
+                               ePyObject it = PyTuple_New(2);
+                               PyTuple_SET_ITEM(it, 0, PyInt_FromLong(pfd[i].fd));
+                               PyTuple_SET_ITEM(it, 1, PyInt_FromLong(pfd[i].revents));
+                               PyList_Append(*res, it);
+                               Py_DECREF(it);
                        }
-               } else if (ret < 0)
+               }
+                       
+               ret = 1; /* poll did not timeout. */
+       } else if (ret < 0)
+       {
+                       /* when we got a signal, we get EINTR. */
+               if (errno != EINTR)
+                       eDebug("poll made error (%m)");
+               else
                {
-                               /* when we got a signal, we get EINTR. we do not care, 
-                                  because we check current time in timers anyway. */
-                       if (errno != EINTR)
-                               eDebug("poll made error (%m)");
-                       else
-                               ret = 0;
+                       return_reason = 2;
+                       ret = -1; /* don't assume the timeout has passed when we got a signal */
                }
-               delete [] pfd;
        }
        
                /* when we not processed anything, check timers. */
-       if (!ret)
+       if (!m_timer_list.empty())
        {
+                       /* we know that this time has passed. */
                singleLock s(recalcLock);
 
-                       /* this will never change while we have the recalcLock */
-                       /* we can savely return here, the timer will be re-checked soon. */
-               if (m_now_is_invalid)
-                       return;
+               if (ret || m_now_is_invalid)
+                       gettimeofday(&now, 0);
+               else
+                       now += poll_timeout;
 
                        /* process all timers which are ready. first remove them out of the list. */
-               while ((!m_timer_list.empty()) && (m_timer_list.begin()->getNextActivation() < now))
+               while ((!m_timer_list.empty()) && (m_timer_list.begin()->getNextActivation() <= now))
                        m_timer_list.begin()->activate();
        }
+       
+       return return_reason;
 }
 
 void eMainloop::addTimer(eTimer* e)
@@ -250,47 +281,80 @@ void eMainloop::removeTimer(eTimer* e)
        m_timer_list.remove(e);
 }
 
-int eMainloop::exec()
+int eMainloop::iterate(unsigned int user_timeout, PyObject **res, ePyObject dict)
 {
-       if (!loop_level)
+       int ret = 0;
+       
+       timeval user_timer;
+       gettimeofday(&user_timer, 0);
+       user_timer += user_timeout;
+
+               /* TODO: this code just became ugly. fix that. */
+       do
        {
-               app_quit_now = false;
-               app_exit_loop = false;
-               enter_loop();
-       }
-       return retval;
+               if (m_interrupt_requested)
+               {
+                       m_interrupt_requested = 0;
+                       return 0;
+               }
+               if (app_quit_now) return -1;
+               timeval now, timeout;
+               gettimeofday(&now, 0);
+               timeout = user_timer - now;
+               
+               if (user_timeout && (user_timer <= now))
+                       return 0;
+               
+               int to = 0;
+               if (user_timeout)
+                       to = timeout.tv_sec * 1000 + timeout.tv_usec / 1000;
+               
+               ret = processOneEvent(to, res, dict);
+               if (res && *res)
+                       return ret;
+       } while (ret == 0);
+       
+       return ret;
 }
 
-void eMainloop::enter_loop()
+int eMainloop::runLoop()
 {
-       loop_level++;
-       // Status der vorhandenen Loop merken
-       bool old_exit_loop = app_exit_loop;
-
-       app_exit_loop = false;
-
-       while (!app_exit_loop && !app_quit_now)
-               processOneEvent();
+       while (!app_quit_now)
+               iterate();
+       return retval;
+}
 
-       // wiederherstellen der vorherigen app_exit_loop
-       app_exit_loop = old_exit_loop;
+void eMainloop::reset()
+{
+       app_quit_now=false;
+}
 
-       --loop_level;
+PyObject *eMainloop::poll(ePyObject timeout, ePyObject dict)
+{
+       PyObject *res=0;
+       
+       if (app_quit_now)
+               Py_RETURN_NONE;
+       
+       int user_timeout = (timeout == Py_None) ? 0 : PyInt_AsLong(timeout);
 
-       if (!loop_level)
-       {
-               // do something here on exit the last loop
-       }
+       iterate(user_timeout, &res, dict);
+       ePyObject ret(res);
+       
+       if (!ret) /* return empty list on timeout */
+               return PyList_New(0);
+       
+       return ret;
 }
 
-void eMainloop::exit_loop()  // call this to leave the current loop
+void eMainloop::interruptPoll()
 {
-       app_exit_loop = true;
+       m_interrupt_requested = 1;
 }
 
-void eMainloop::quit( int ret )   // call this to leave all loops
+void eMainloop::quit(int ret)
 {
-       retval=ret;
+       retval = ret;
        app_quit_now = true;
 }