small fix
[enigma2.git] / lib / base / ebase.cpp
index ffd35eceae6cc6cb5d4adfac1c42bf37e5e4585e..93bfd3425b5425360d2e28568239b0f293a9087a 100644 (file)
@@ -24,7 +24,7 @@ void eSocketNotifier::start()
                stop();
 
        context.addSocketNotifier(this);
-       state=1;
+       state=2;  // running but not in poll yet
 }
 
 void eSocketNotifier::stop()
@@ -45,6 +45,7 @@ void eTimer::start(long msek, bool singleShot)
        bSingleShot = singleShot;
        interval = msek;
        gettimeofday(&nextActivation, 0);
+       nextActivation.tv_sec -= context.getTimeOffset();
 //     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 );
@@ -59,6 +60,7 @@ void eTimer::startLongTimer( int seconds )
        bActive = bSingleShot = true;
        interval = 0;
        gettimeofday(&nextActivation, 0);
+       nextActivation.tv_sec -= context.getTimeOffset();
 //     eDebug("this = %p\nnow sec = %d, usec = %d\nadd %d sec", this, nextActivation.tv_sec, nextActivation.tv_usec, seconds);
        if ( seconds > 0 )
                nextActivation.tv_sec += seconds;
@@ -116,139 +118,137 @@ 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());
+       notifiers[fd]=sn;
 }
 
 void eMainloop::removeSocketNotifier(eSocketNotifier *sn)
 {
-       for (std::multimap<int,eSocketNotifier*>::iterator i = notifiers.find(sn->getFD());
-                       i != notifiers.end();
-                       ++i)
-               if (i->second == sn)
-                       return notifiers.erase(i);
+       int fd = sn->getFD();
+       std::map<int,eSocketNotifier*>::iterator i(notifiers.find(fd));
+       if (i != notifiers.end())
+               return notifiers.erase(i);
        eFatal("removed socket notifier which is not present");
 }
 
-int eMainloop::processOneEvent(unsigned int user_timeout)
+int eMainloop::processOneEvent(unsigned int twisted_timeout, PyObject **res, ePyObject additional)
 {
        int return_reason = 0;
                /* get current time */
-       timeval now;
-       gettimeofday(&now, 0);
-       m_now_is_invalid = 0;
-       
-       int poll_timeout = -1; /* infinite in case of empty timer list */
-       
-       if (m_timer_list)
+
+       if (additional && !PyDict_Check(additional))
+               eFatal("additional, but it's not dict");
+
+       if (additional && !res)
+               eFatal("additional, but no res");
+
+       long poll_timeout = -1; /* infinite in case of empty timer list */
+
+       if (!m_timer_list.empty() || twisted_timeout > 0)
        {
-               singleLock s(recalcLock);
-               poll_timeout = timeval_to_usec(m_timer_list.begin()->getNextActivation() - now);
-                       /* if current timer already passed, don't delay infinite. */
-               if (poll_timeout < 0)
-                       poll_timeout = 0;
-                       
-                       /* convert us to ms */
-               poll_timeout /= 1000;
+               applyTimeOffset();
+               if (!m_timer_list.empty())
+               {
+                       /* process all timers which are ready. first remove them out of the list. */
+                       while (!m_timer_list.empty() && (poll_timeout = timeout_usec( m_timer_list.begin()->getNextActivation() ) ) <= 0 )
+                       {
+                               m_timer_list.begin()->activate();
+                               applyTimeOffset();
+                       }
+                       if (poll_timeout < 0)
+                               poll_timeout = 0;
+                       else /* convert us to ms */
+                               poll_timeout /= 1000;
+               }
        }
-       
-       if ((user_timeout > 0) && (poll_timeout > user_timeout))
+
+       if ((twisted_timeout > 0) && (poll_timeout > 0) && ((unsigned int)poll_timeout > twisted_timeout))
        {
-               poll_timeout = user_timeout;
+               poll_timeout = twisted_timeout;
                return_reason = 1;
        }
-       
-       int ret = 0;
-       
-       if (poll_timeout)
+
+       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)
        {
+               it->second->state = 1; // running and in poll
+               pfd[i].fd = it->first;
+               pfd[i].events = it->second->getRequested();
+       }
 
-               std::multimap<int,eSocketNotifier*>::iterator it;
-               std::map<int,int> fd_merged;
-               std::map<int,int>::const_iterator fd_merged_it;
-               
-               for (it = notifiers.begin(); it != notifiers.end(); ++it)
-                       fd_merged[it->first] |= it->second->getRequested();
-               
-               fd_merged_it = fd_merged.begin();
-               
-               int fdcount = fd_merged.size();
-
-                       // build the poll aray
-               pollfd* pfd = new pollfd[fdcount];  // make new pollfd array
-               
-               for (int i=0; i < fdcount; i++, fd_merged_it++)
-               {
-                       pfd[i].fd = fd_merged_it->first;
-                       pfd[i].events = fd_merged_it->second;
+       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);
                }
+       }
+
+       if (this == eApp)
+               Py_BEGIN_ALLOW_THREADS
+               ret = ::poll(pfd, fdcount, poll_timeout);
+               Py_END_ALLOW_THREADS
+       else
+               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)
                {
-                       return_reason = 0;
-                       for (int i=0; i < fdcount ; i++)
+                       if (pfd[i].revents)
                        {
-                               it = notifiers.begin();
-                               
-                               int handled = 0;
-                               
-                               std::multimap<int,eSocketNotifier*>::iterator 
-                                       l = notifiers.lower_bound(pfd[i].fd),
-                                       u = notifiers.upper_bound(pfd[i].fd);
-                               
-                               ePtrList<eSocketNotifier> n;
-                               
-                               for (; l != u; ++l)
-                                       n.push_back(l->second);
-                               
-                               for (ePtrList<eSocketNotifier>::iterator li(n.begin()); li != n.end(); ++li)
+                               it = notifiers.find(pfd[i].fd);
+                               if (it != notifiers.end()
+                                       && it->second->state == 1) // added and in poll
                                {
-                                       int req = li->getRequested();
-                                       
-                                       handled |= req;
-                               
+                                       int req = it->second->getRequested();
                                        if (pfd[i].revents & req)
-                                               (*li)->activate(pfd[i].revents);
+                                               it->second->activate(pfd[i].revents & req);
+                                       pfd[i].revents &= ~req;
                                }
-                               if ((pfd[i].revents&~handled) & (POLLERR|POLLHUP|POLLNVAL))
-                                               eFatal("poll: unhandled POLLERR/HUP/NVAL for fd %d(%d)", pfd[i].fd, pfd[i].revents);
+                               if (pfd[i].revents & (POLLERR|POLLHUP|POLLNVAL))
+                                       eDebug("poll: unhandled POLLERR/HUP/NVAL for fd %d(%d)", pfd[i].fd, pfd[i].revents);
                        }
-                       
-                       ret = 1; /* poll did not timeout. */
-               } else if (ret < 0)
+               }
+               for (; i < fdcount; ++i)
                {
-                               /* when we got a signal, we get EINTR. */
-                       if (errno != EINTR)
-                               eDebug("poll made error (%m)");
-                       else
+                       if (pfd[i].revents)
                        {
-                               return_reason = 2;
-                               ret = -1; /* don't assume the timeout has passed when we got a signal */
+                               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);
                        }
                }
-               delete [] pfd;
        }
-       
-               /* when we not processed anything, check timers. */
-       if (!ret)
+       else if (ret < 0)
        {
-                       /* we know that this time has passed. */
-               now += poll_timeout;
-               
-               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 0;
-
-                       /* process all timers which are ready. first remove them out of the list. */
-               while ((!m_timer_list.empty()) && (m_timer_list.begin()->getNextActivation() <= now))
-                       m_timer_list.begin()->activate();
+                       /* when we got a signal, we get EINTR. */
+               if (errno != EINTR)
+                       eDebug("poll made error (%m)");
+               else
+                       return_reason = 2; /* don't assume the timeout has passed when we got a signal */
        }
-       
+
        return return_reason;
 }
 
@@ -262,16 +262,44 @@ void eMainloop::removeTimer(eTimer* e)
        m_timer_list.remove(e);
 }
 
-int eMainloop::iterate(unsigned int user_timeout)
+int eMainloop::iterate(unsigned int twisted_timeout, PyObject **res, ePyObject dict)
 {
        int ret = 0;
-       
+
+       if (twisted_timeout)
+       {
+               gettimeofday(&m_twisted_timer, 0);
+               m_twisted_timer += twisted_timeout;
+       }
+
+               /* TODO: this code just became ugly. fix that. */
        do
-       { 
-               if (app_quit_now) break; 
-               ret = processOneEvent(user_timeout);
-       } while (ret == 0);
-       
+       {
+               if (m_interrupt_requested)
+               {
+                       m_interrupt_requested = 0;
+                       return 0;
+               }
+
+               if (app_quit_now)
+                       return -1;
+
+               int to = 0;
+               if (twisted_timeout)
+               {
+                       timeval now, timeout;
+                       gettimeofday(&now, 0);
+                       m_twisted_timer += time_offset;  // apply pending offset
+                       if (m_twisted_timer<=now) // timeout
+                               return 0;
+                       timeout = m_twisted_timer - now;
+                       to = timeout.tv_sec * 1000 + timeout.tv_usec / 1000;
+                       // remove pending offset .. it is re-applied in next call of processOneEvent.. applyTimeOffset
+                       m_twisted_timer -= time_offset;  
+               }
+               ret = processOneEvent(to, res, dict);
+       } while ( !ret && !(res && *res) );
+
        return ret;
 }
 
@@ -282,6 +310,32 @@ int eMainloop::runLoop()
        return retval;
 }
 
+void eMainloop::reset()
+{
+       app_quit_now=false;
+}
+
+PyObject *eMainloop::poll(ePyObject timeout, ePyObject dict)
+{
+       PyObject *res=0;
+
+       if (app_quit_now)
+               Py_RETURN_NONE;
+
+       int twisted_timeout = (timeout == Py_None) ? 0 : PyInt_AsLong(timeout);
+
+       iterate(twisted_timeout, &res, dict);
+       if (res)
+               return res;
+
+       return PyList_New(0); /* return empty list on timeout */
+}
+
+void eMainloop::interruptPoll()
+{
+       m_interrupt_requested = 1;
+}
+
 void eMainloop::quit(int ret)
 {
        retval = ret;
@@ -290,13 +344,33 @@ void eMainloop::quit(int ret)
 
 void eMainloop::addTimeOffset(int offset)
 {
-       for (ePtrList<eMainloop>::iterator it(eMainloop::existing_loops)
-               ;it != eMainloop::existing_loops.end(); ++it)
+       for (ePtrList<eMainloop>::iterator it(existing_loops.begin()); it != existing_loops.end(); ++it )
+               it->addInstanceTimeOffset(offset);
+}
+
+void eMainloop::addInstanceTimeOffset(int offset)
+{
+       singleLock s(recalcLock);
+       if (m_timer_list.empty())
+               time_offset=0;
+       else
+       {
+               if ( time_offset )
+                       eDebug("time_offset %d avail.. add new offset %d than new is %d",
+                       time_offset, offset, time_offset+offset);
+               time_offset+=offset;
+       }
+}
+
+void eMainloop::applyTimeOffset()
+{
+       singleLock s(recalcLock);
+       if ( time_offset )
        {
-               singleLock s(it->recalcLock);
-               it->m_now_is_invalid = 1;
-               for (ePtrList<eTimer>::iterator tit = it->m_timer_list.begin(); tit != it->m_timer_list.end(); ++tit )
-                       tit->addTimeOffset(offset);
+               for (ePtrList<eTimer>::iterator it(m_timer_list.begin()); it != m_timer_list.end(); ++it )
+                       it->addTimeOffset( time_offset );
+               m_twisted_timer += time_offset;
+               time_offset=0;
        }
 }