00ecb345128c8bccd5b2b3a740f2fe764617e4b2
[enigma2.git] / 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 > user_timeout))
161         {
162                 poll_timeout = user_timeout;
163                 return_reason = 1;
164         }
165         
166         int ret = 0;
167         
168         
169         if (poll_timeout)
170         {
171                 std::multimap<int,eSocketNotifier*>::iterator it;
172                 std::map<int,int> fd_merged;
173                 std::map<int,int>::const_iterator fd_merged_it;
174                 
175                 for (it = notifiers.begin(); it != notifiers.end(); ++it)
176                         fd_merged[it->first] |= it->second->getRequested();
177                 
178                 fd_merged_it = fd_merged.begin();
179                 
180                 int nativecount, fdcount;
181                 
182                 nativecount = fdcount = fd_merged.size();
183                 
184                 if (additional) 
185                 {
186                         additional = PyDict_Items(additional);
187                         fdcount += PyList_Size(additional);
188                 }
189
190                         // build the poll aray
191                 pollfd* pfd = new pollfd[fdcount];  // make new pollfd array
192                 
193                 for (int i=0; i < nativecount; i++, fd_merged_it++)
194                 {
195                         pfd[i].fd = fd_merged_it->first;
196                         pfd[i].events = fd_merged_it->second;
197                 }
198                 
199                 if (additional)
200                 {
201                         for (int i=0; i < PyList_Size(additional); ++i)
202                         {
203                                 PyObject *it = PyList_GET_ITEM(additional, i);
204                                 if (!PyTuple_Check(it))
205                                         eFatal("poll item is not a tuple");
206                                 if (PyTuple_Size(it) != 2)
207                                         eFatal("poll tuple size is not 2");
208                                 int fd = PyObject_AsFileDescriptor(PyTuple_GET_ITEM(it, 0));
209                                 if (fd == -1)
210                                         eFatal("poll tuple not a filedescriptor");
211                                 pfd[nativecount + i].fd = fd;
212                                 pfd[nativecount + i].events = PyInt_AsLong(PyTuple_GET_ITEM(it, 1));
213                         }
214                 }
215
216                 ret = ::poll(pfd, fdcount, poll_timeout);
217                 
218                         /* ret > 0 means that there are some active poll entries. */
219                 if (ret > 0)
220                 {
221                         return_reason = 0;
222                         for (int i=0; i < nativecount ; i++)
223                         {
224                                 it = notifiers.begin();
225                                 
226                                 int handled = 0;
227                                 
228                                 std::multimap<int,eSocketNotifier*>::iterator 
229                                         l = notifiers.lower_bound(pfd[i].fd),
230                                         u = notifiers.upper_bound(pfd[i].fd);
231                                 
232                                 ePtrList<eSocketNotifier> n;
233                                 
234                                 for (; l != u; ++l)
235                                         n.push_back(l->second);
236                                 
237                                 for (ePtrList<eSocketNotifier>::iterator li(n.begin()); li != n.end(); ++li)
238                                 {
239                                         int req = li->getRequested();
240                                         
241                                         handled |= req;
242                                 
243                                         if (pfd[i].revents & req)
244                                                 (*li)->activate(pfd[i].revents);
245                                 }
246                                 if ((pfd[i].revents&~handled) & (POLLERR|POLLHUP|POLLNVAL))
247                                         eDebug("poll: unhandled POLLERR/HUP/NVAL for fd %d(%d)", pfd[i].fd, pfd[i].revents);
248                         }
249                         
250                         for (int i = nativecount; i < fdcount; ++i)
251                         {
252                                 if (pfd[i].revents)
253                                 {
254                                         if (!*res)
255                                                 *res = PyList_New(0);
256                                         PyObject *it = PyTuple_New(2);
257                                         PyTuple_SET_ITEM(it, 0, PyInt_FromLong(pfd[i].fd));
258                                         PyTuple_SET_ITEM(it, 1, PyInt_FromLong(pfd[i].revents));
259                                         PyList_Append(*res, it);
260                                         Py_DECREF(it);
261                                 }
262                         }
263                         
264                         ret = 1; /* poll did not timeout. */
265                 } else if (ret < 0)
266                 {
267                                 /* when we got a signal, we get EINTR. */
268                         if (errno != EINTR)
269                                 eDebug("poll made error (%m)");
270                         else
271                         {
272                                 return_reason = 2;
273                                 ret = -1; /* don't assume the timeout has passed when we got a signal */
274                         }
275                 }
276                 delete [] pfd;
277                 Py_XDECREF(additional);
278         }
279         
280                 /* when we not processed anything, check timers. */
281         if (!ret)
282         {
283                         /* we know that this time has passed. */
284                 now += poll_timeout;
285                 
286                 singleLock s(recalcLock);
287
288                         /* this will never change while we have the recalcLock */
289                         /* we can savely return here, the timer will be re-checked soon. */
290                 if (m_now_is_invalid)
291                         return 0;
292
293                         /* process all timers which are ready. first remove them out of the list. */
294                 while ((!m_timer_list.empty()) && (m_timer_list.begin()->getNextActivation() <= now))
295                         m_timer_list.begin()->activate();
296         }
297         
298         return return_reason;
299 }
300
301 void eMainloop::addTimer(eTimer* e)
302 {
303         m_timer_list.insert_in_order(e);
304 }
305
306 void eMainloop::removeTimer(eTimer* e)
307 {
308         m_timer_list.remove(e);
309 }
310
311 int eMainloop::iterate(unsigned int user_timeout, PyObject **res, PyObject *dict)
312 {
313         int ret = 0;
314         
315         do
316         { 
317                 if (app_quit_now) return -1;
318                 ret = processOneEvent(user_timeout, res, dict);
319                 if (res && *res)
320                         return ret;
321         } while (ret == 0);
322         
323         return ret;
324 }
325
326 int eMainloop::runLoop()
327 {
328         while (!app_quit_now)
329                 iterate();
330         return retval;
331 }
332
333 PyObject *eMainloop::poll(PyObject *timeout, PyObject *dict)
334 {
335         PyObject *res = 0;
336         int user_timeout = (timeout == Py_None) ? 0 : PyInt_AsLong(timeout);
337         
338         iterate(user_timeout, &res, dict);
339         
340         if (!res) /* return empty list on timeout */
341                 res = PyList_New(0);
342         
343         return res;
344 }
345
346 void eMainloop::quit(int ret)
347 {
348         retval = ret;
349         app_quit_now = true;
350 }
351
352 void eMainloop::addTimeOffset(int offset)
353 {
354         for (ePtrList<eMainloop>::iterator it(eMainloop::existing_loops)
355                 ;it != eMainloop::existing_loops.end(); ++it)
356         {
357                 singleLock s(it->recalcLock);
358                 it->m_now_is_invalid = 1;
359                 for (ePtrList<eTimer>::iterator tit = it->m_timer_list.begin(); tit != it->m_timer_list.end(); ++tit )
360                         tit->addTimeOffset(offset);
361         }
362 }
363
364 eApplication* eApp = 0;