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)
{
- 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);
+ i = new_notifiers.find(fd);
+ if (i != new_notifiers.end())
+ return new_notifiers.erase(i);
eFatal("removed socket notifier which is not present");
}
-int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, PyObject *additional)
+int eMainloop::processOneEvent(unsigned int user_timeout, PyObject **res, ePyObject additional)
{
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);
poll_timeout = user_timeout;
return_reason = 1;
}
-
- int ret = 0;
-
- if (poll_timeout)
+
+ for (std::map<int, eSocketNotifier*>::iterator it(new_notifiers.begin()); it != new_notifiers.end();)
{
- 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 nativecount, fdcount;
-
- nativecount = fdcount = fd_merged.size();
-
- if (additional)
- {
- additional = PyDict_Items(additional);
- fdcount += PyList_Size(additional);
- }
+ notifiers[it->first]=it->second;
+ new_notifiers.erase(it++);
+ }
- // build the poll aray
- pollfd* pfd = new pollfd[fdcount]; // make new pollfd array
-
- for (int i=0; i < nativecount; i++, fd_merged_it++)
- {
- pfd[i].fd = fd_merged_it->first;
- pfd[i].events = fd_merged_it->second;
- }
-
- if (additional)
- {
- for (int i=0; i < PyList_Size(additional); ++i)
- {
- PyObject *it = PyList_GET_ITEM(additional, i);
- if (!PyTuple_Check(it))
- eFatal("poll item is not a tuple");
- if (PyTuple_Size(it) != 2)
- eFatal("poll tuple size is not 2");
- int fd = PyObject_AsFileDescriptor(PyTuple_GET_ITEM(it, 0));
- if (fd == -1)
- eFatal("poll tuple not a filedescriptor");
- pfd[nativecount + i].fd = fd;
- pfd[nativecount + i].events = PyInt_AsLong(PyTuple_GET_ITEM(it, 1));
- }
- }
+ int nativecount=notifiers.size(),
+ fdcount=nativecount,
+ ret=0;
- ret = ::poll(pfd, fdcount, poll_timeout);
+ 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 > 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 < nativecount ; 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())
{
- int req = li->getRequested();
-
+ int req = it->second->getRequested();
handled |= req;
-
if (pfd[i].revents & req)
- (*li)->activate(pfd[i].revents);
+ it->second->activate(pfd[i].revents);
}
- if ((pfd[i].revents&~handled) & (POLLERR|POLLHUP|POLLNVAL))
+ 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 (int i = nativecount; i < fdcount; ++i)
+ }
+ for (; i < fdcount; ++i)
+ {
+ if (pfd[i].revents)
{
- if (pfd[i].revents)
- {
- if (!*res)
- *res = PyList_New(0);
- PyObject *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);
- }
+ 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);
}
+ }
- ret = 1; /* poll did not timeout. */
- } 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. */
- if (errno != EINTR)
- eDebug("poll made error (%m)");
- else
- {
- return_reason = 2;
- ret = -1; /* don't assume the timeout has passed when we got a signal */
- }
+ return_reason = 2;
+ ret = -1; /* don't assume the timeout has passed when we got a signal */
}
- delete [] pfd;
- Py_XDECREF(additional);
}
/* when we not processed anything, check timers. */
- if (!ret)
+ if (!m_timer_list.empty())
{
/* 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;
+ 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))
m_timer_list.remove(e);
}
-int eMainloop::iterate(unsigned int user_timeout, PyObject **res, PyObject *dict)
+int eMainloop::iterate(unsigned int user_timeout, PyObject **res, ePyObject dict)
{
int ret = 0;
app_quit_now=false;
}
-PyObject *eMainloop::poll(PyObject *timeout, PyObject *dict)
+PyObject *eMainloop::poll(ePyObject timeout, ePyObject dict)
{
- PyObject *res = 0;
+ 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);
iterate(user_timeout, &res, dict);
+ ePyObject ret(res);
- if (!res) /* return empty list on timeout */
- res = PyList_New(0);
+ if (!ret) /* return empty list on timeout */
+ return PyList_New(0);
- return res;
+ return ret;
}
void eMainloop::interruptPoll()