fix ASSERT in the correct way

[enigma2.git] / lib / base / ebase.cpp

diff --git a/lib/base/ebase.cpp b/lib/base/ebase.cpp
index 480c8ea8d2a8d48e70ff7e6b8e5ddc0bb697aef7..f40bf39e78187bf8b45e527554165fff4fde739e 100644 (file)
--- a/lib/base/ebase.cpp
+++ b/lib/base/ebase.cpp
@@ -118,16 +118,19 @@ void eMainloop::addSocketNotifier(eSocketNotifier *sn)
 {
        int fd = sn->getFD();
        ASSERT(notifiers.find(fd) == notifiers.end());
-       notifiers[fd]=sn;
+       ASSERT(new_notifiers.find(fd) == new_notifiers.end());
+       new_notifiers[fd]=sn;
 }
 
 void eMainloop::removeSocketNotifier(eSocketNotifier *sn)
 {
-       for (std::map<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);
+       i = new_notifiers.find(fd);
+       if (i != new_notifiers.end())
+               return new_notifiers.erase(i);
        eFatal("removed socket notifier which is not present");
 }
 
@@ -164,7 +167,13 @@ int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, ePyObj
                poll_timeout = user_timeout;
                return_reason = 1;
        }
-       
+
+       for (std::map<int, eSocketNotifier*>::iterator it(new_notifiers.begin()); it != new_notifiers.end();)
+       {
+               notifiers[it->first]=it->second;
+               new_notifiers.erase(it++);
+       }
+
        int nativecount=notifiers.size(),
                fdcount=nativecount,
                ret=0;
@@ -325,10 +334,7 @@ PyObject *eMainloop::poll(ePyObject timeout, ePyObject dict)
        PyObject *res=0;
        
        if (app_quit_now)
-       {
-               Py_INCREF(Py_None);
-               return Py_None;
-       }
+               Py_RETURN_NONE;
        
        int user_timeout = (timeout == Py_None) ? 0 : PyInt_AsLong(timeout);