Add XCBPollForReply and deprecate XCBGetRequestRead and XCBGetQueuedRequestRead.
[free-sw/xcb/libxcb] / src / xcb_in.c
1 /* Copyright (C) 2001-2004 Bart Massey and Jamey Sharp.
2  *
3  * Permission is hereby granted, free of charge, to any person obtaining a
4  * copy of this software and associated documentation files (the "Software"),
5  * to deal in the Software without restriction, including without limitation
6  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
7  * and/or sell copies of the Software, and to permit persons to whom the
8  * Software is furnished to do so, subject to the following conditions:
9  * 
10  * The above copyright notice and this permission notice shall be included in
11  * all copies or substantial portions of the Software.
12  * 
13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16  * AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
17  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
18  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
19  * 
20  * Except as contained in this notice, the names of the authors or their
21  * institutions shall not be used in advertising or otherwise to promote the
22  * sale, use or other dealings in this Software without prior written
23  * authorization from the authors.
24  */
25
26 /* Stuff that reads stuff from the server. */
27
28 #include <assert.h>
29 #include <string.h>
30 #include <stdlib.h>
31 #include <unistd.h>
32 #include <stdio.h>
33 #include <errno.h>
34
35 #include "xcb.h"
36 #include "xcbext.h"
37 #include "xcbint.h"
38
39 #define XCBError 0
40 #define XCBReply 1
41
42 struct event_list {
43     XCBGenericEvent *event;
44     struct event_list *next;
45 };
46
47 struct reply_list {
48     void *reply;
49     struct reply_list *next;
50 };
51
52 typedef struct pending_reply {
53     unsigned int request;
54     enum workarounds workaround;
55     int flags;
56     struct pending_reply *next;
57 } pending_reply;
58
59 typedef struct reader_list {
60     unsigned int request;
61     pthread_cond_t *data;
62     struct reader_list *next;
63 } reader_list;
64
65 static void wake_up_next_reader(XCBConnection *c)
66 {
67     int pthreadret;
68     if(c->in.readers)
69         pthreadret = pthread_cond_signal(c->in.readers->data);
70     else
71         pthreadret = pthread_cond_signal(&c->in.event_cond);
72     assert(pthreadret == 0);
73 }
74
75 static int read_packet(XCBConnection *c)
76 {
77     XCBGenericRep genrep;
78     int length = 32;
79     void *buf;
80     pending_reply *pend = 0;
81     struct event_list *event;
82
83     /* Wait for there to be enough data for us to read a whole packet */
84     if(c->in.queue_len < length)
85         return 0;
86
87     /* Get the response type, length, and sequence number. */
88     memcpy(&genrep, c->in.queue, sizeof(genrep));
89
90     /* Compute 32-bit sequence number of this packet. */
91     if((genrep.response_type & 0x7f) != XCBKeymapNotify)
92     {
93         int lastread = c->in.request_read;
94         c->in.request_read = (lastread & 0xffff0000) | genrep.sequence;
95         if(c->in.request_read < lastread)
96             c->in.request_read += 0x10000;
97         if(c->in.request_read > c->in.request_expected)
98             c->in.request_expected = c->in.request_read;
99
100         if(c->in.request_read != lastread)
101         {
102             if(c->in.current_reply)
103             {
104                 _xcb_map_put(c->in.replies, lastread, c->in.current_reply);
105                 c->in.current_reply = 0;
106                 c->in.current_reply_tail = &c->in.current_reply;
107             }
108             c->in.request_completed = c->in.request_read - 1;
109         }
110         if(genrep.response_type != XCBReply) /* error or event */
111             c->in.request_completed = c->in.request_read; /* XXX: does event/error imply no more replies? */
112
113         while(c->in.pending_replies && c->in.pending_replies->request <= c->in.request_completed)
114         {
115             pending_reply *oldpend = c->in.pending_replies;
116             c->in.pending_replies = oldpend->next;
117             if(!oldpend->next)
118                 c->in.pending_replies_tail = &c->in.pending_replies;
119             free(oldpend);
120         }
121     }
122
123     if(genrep.response_type == XCBError || genrep.response_type == XCBReply)
124     {
125         pend = c->in.pending_replies;
126         if(pend && pend->request != c->in.request_read)
127             pend = 0;
128     }
129
130     /* For reply packets, check that the entire packet is available. */
131     if(genrep.response_type == XCBReply)
132     {
133         if(pend && pend->workaround == WORKAROUND_GLX_GET_FB_CONFIGS_BUG)
134         {
135             CARD32 *p = (CARD32 *) c->in.queue;
136             genrep.length = p[2] * p[3] * 2;
137         }
138         length += genrep.length * 4;
139     }
140
141     buf = malloc(length + (genrep.response_type == XCBReply ? 0 : sizeof(CARD32)));
142     if(!buf)
143         return 0;
144     if(_xcb_in_read_block(c, buf, length) <= 0)
145     {
146         free(buf);
147         return 0;
148     }
149     if(pend && (pend->flags & XCB_REQUEST_DISCARD_REPLY))
150     {
151         free(buf);
152         return 1;
153     }
154
155     if(genrep.response_type != XCBReply)
156         ((XCBGenericEvent *) buf)->full_sequence = c->in.request_read;
157
158     /* reply, or checked error */
159     if( genrep.response_type == XCBReply ||
160        (genrep.response_type == XCBError && pend && (pend->flags & XCB_REQUEST_CHECKED)))
161     {
162         reader_list *reader;
163         struct reply_list *cur = malloc(sizeof(struct reply_list));
164         if(!cur)
165             return 0;
166         cur->reply = buf;
167         cur->next = 0;
168         *c->in.current_reply_tail = cur;
169         c->in.current_reply_tail = &cur->next;
170         for(reader = c->in.readers; reader && reader->request <= c->in.request_read; reader = reader->next)
171             if(reader->request == c->in.request_read)
172             {
173                 pthread_cond_signal(reader->data);
174                 break;
175             }
176         return 1;
177     }
178
179     /* event, or unchecked error */
180     event = malloc(sizeof(struct event_list));
181     if(!event)
182     {
183         free(buf);
184         return 0;
185     }
186     event->event = buf;
187     event->next = 0;
188     *c->in.events_tail = event;
189     c->in.events_tail = &event->next;
190     pthread_cond_signal(&c->in.event_cond);
191     return 1; /* I have something for you... */
192 }
193
194 static XCBGenericEvent *get_event(XCBConnection *c)
195 {
196     struct event_list *cur = c->in.events;
197     XCBGenericEvent *ret;
198     if(!c->in.events)
199         return 0;
200     ret = cur->event;
201     c->in.events = cur->next;
202     if(!cur->next)
203         c->in.events_tail = &c->in.events;
204     free(cur);
205     return ret;
206 }
207
208 static void free_reply_list(struct reply_list *head)
209 {
210     while(head)
211     {
212         struct reply_list *cur = head;
213         head = cur->next;
214         free(cur->reply);
215         free(cur);
216     }
217 }
218
219 static int read_block(const int fd, void *buf, const size_t len)
220 {
221     int done = 0;
222     while(done < len)
223     {
224         int ret = read(fd, ((char *) buf) + done, len - done);
225         if(ret > 0)
226             done += ret;
227         if(ret < 0 && errno == EAGAIN)
228         {
229             fd_set fds;
230             FD_ZERO(&fds);
231             FD_SET(fd, &fds);
232             do {
233                 ret = select(fd + 1, &fds, 0, 0, 0);
234             } while (ret == -1 && errno == EINTR);
235         }
236         if(ret <= 0)
237             return ret;
238     }
239     return len;
240 }
241
242 static int poll_for_reply(XCBConnection *c, unsigned int request, void **reply, XCBGenericError **error)
243 {
244     struct reply_list *head;
245
246     if(!request)
247         head = 0;
248     else if(c->in.request_completed >= request)
249     {
250         head = _xcb_map_remove(c->in.replies, request);
251         if(head && head->next)
252             _xcb_map_put(c->in.replies, request, head->next);
253     }
254     else if(c->in.request_read == request && c->in.current_reply)
255     {
256         head = c->in.current_reply;
257         c->in.current_reply = head->next;
258         if(!head->next)
259             c->in.current_reply_tail = &c->in.current_reply;
260     }
261     else
262         /* Would block: do nothing. */
263         return 0;
264
265     if(error)
266         *error = 0;
267     *reply = 0;
268
269     if(head)
270     {
271         if(((XCBGenericRep *) head->reply)->response_type == XCBError)
272         {
273             if(error)
274                 *error = head->reply;
275             else
276                 free(head->reply);
277         }
278         else
279             *reply = head->reply;
280
281         free(head);
282     }
283
284     return 1;
285 }
286
287 /* Public interface */
288
289 void *XCBWaitForReply(XCBConnection *c, unsigned int request, XCBGenericError **e)
290 {
291     pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
292     reader_list reader;
293     reader_list **prev_reader;
294     struct reply_list *head;
295     void *ret = 0;
296     if(e)
297         *e = 0;
298
299     /* If an error occurred when issuing the request, fail immediately. */
300     if(!request)
301         return 0;
302
303     pthread_mutex_lock(&c->iolock);
304
305     /* If this request has not been written yet, write it. */
306     if(!_xcb_out_flush_to(c, request))
307         goto done; /* error */
308
309     for(prev_reader = &c->in.readers; *prev_reader && (*prev_reader)->request <= request; prev_reader = &(*prev_reader)->next)
310         if((*prev_reader)->request == request)
311             goto done; /* error */
312
313     reader.request = request;
314     reader.data = &cond;
315     reader.next = *prev_reader;
316     *prev_reader = &reader;
317
318     /* If this request has not completed yet and has no reply waiting,
319      * wait for one. */
320     while(c->in.request_completed < request &&
321             !(c->in.request_read == request && c->in.current_reply))
322         if(!_xcb_conn_wait(c, &cond, 0, 0))
323             goto done;
324
325     if(c->in.request_read != request)
326     {
327         head = _xcb_map_remove(c->in.replies, request);
328         if(head && head->next)
329             _xcb_map_put(c->in.replies, request, head->next);
330     }
331     else
332     {
333         head = c->in.current_reply;
334         if(head)
335         {
336             c->in.current_reply = head->next;
337             if(!head->next)
338                 c->in.current_reply_tail = &c->in.current_reply;
339         }
340     }
341
342     if(head)
343     {
344         ret = head->reply;
345         free(head);
346
347         if(((XCBGenericRep *) ret)->response_type == XCBError)
348         {
349             if(e)
350                 *e = ret;
351             else
352                 free(ret);
353             ret = 0;
354         }
355     }
356
357 done:
358     for(prev_reader = &c->in.readers; *prev_reader && (*prev_reader)->request <= request; prev_reader = &(*prev_reader)->next)
359         if(*prev_reader == &reader)
360         {
361             *prev_reader = (*prev_reader)->next;
362             break;
363         }
364     pthread_cond_destroy(&cond);
365
366     wake_up_next_reader(c);
367     pthread_mutex_unlock(&c->iolock);
368     return ret;
369 }
370
371 int XCBPollForReply(XCBConnection *c, unsigned int request, void **reply, XCBGenericError **error)
372 {
373     int ret;
374     assert(reply != 0);
375     pthread_mutex_lock(&c->iolock);
376     ret = poll_for_reply(c, request, reply, error);
377     pthread_mutex_unlock(&c->iolock);
378     return ret;
379 }
380
381 XCBGenericEvent *XCBWaitEvent(XCBConnection *c)
382 {
383     return XCBWaitForEvent(c);
384 }
385
386 XCBGenericEvent *XCBWaitForEvent(XCBConnection *c)
387 {
388     XCBGenericEvent *ret;
389     pthread_mutex_lock(&c->iolock);
390     /* get_event returns 0 on empty list. */
391     while(!(ret = get_event(c)))
392         if(!_xcb_conn_wait(c, &c->in.event_cond, 0, 0))
393             break;
394
395     wake_up_next_reader(c);
396     pthread_mutex_unlock(&c->iolock);
397     return ret;
398 }
399
400 XCBGenericEvent *XCBPollForEvent(XCBConnection *c, int *error)
401 {
402     XCBGenericEvent *ret = 0;
403     pthread_mutex_lock(&c->iolock);
404     if(error)
405         *error = 0;
406     /* FIXME: follow X meets Z architecture changes. */
407     if(_xcb_in_read(c))
408         ret = get_event(c);
409     else if(error)
410         *error = -1;
411     else
412     {
413         fprintf(stderr, "XCBPollForEvent: I/O error occured, but no handler provided.\n");
414         abort();
415     }
416     pthread_mutex_unlock(&c->iolock);
417     return ret;
418 }
419
420 unsigned int XCBGetRequestRead(XCBConnection *c)
421 {
422     unsigned int ret;
423     pthread_mutex_lock(&c->iolock);
424     /* FIXME: follow X meets Z architecture changes. */
425     _xcb_in_read(c);
426     ret = c->in.request_read;
427     pthread_mutex_unlock(&c->iolock);
428     return ret;
429 }
430
431 /* Private interface */
432
433 int _xcb_in_init(_xcb_in *in)
434 {
435     if(pthread_cond_init(&in->event_cond, 0))
436         return 0;
437     in->reading = 0;
438
439     in->queue_len = 0;
440
441     in->request_read = 0;
442     in->request_completed = 0;
443
444     in->replies = _xcb_map_new();
445     if(!in->replies)
446         return 0;
447
448     in->current_reply_tail = &in->current_reply;
449     in->events_tail = &in->events;
450     in->pending_replies_tail = &in->pending_replies;
451
452     return 1;
453 }
454
455 void _xcb_in_destroy(_xcb_in *in)
456 {
457     pthread_cond_destroy(&in->event_cond);
458     free_reply_list(in->current_reply);
459     _xcb_map_delete(in->replies, (void (*)(void *)) free_reply_list);
460     while(in->events)
461     {
462         struct event_list *e = in->events;
463         in->events = e->next;
464         free(e->event);
465         free(e);
466     }
467     while(in->pending_replies)
468     {
469         pending_reply *pend = in->pending_replies;
470         in->pending_replies = pend->next;
471         free(pend);
472     }
473 }
474
475 int _xcb_in_expect_reply(XCBConnection *c, unsigned int request, enum workarounds workaround, int flags)
476 {
477     if(workaround != WORKAROUND_NONE || flags != 0)
478     {
479         pending_reply *pend = malloc(sizeof(pending_reply));
480         if(!pend)
481             return 0;
482         pend->request = request;
483         pend->workaround = workaround;
484         pend->flags = flags;
485         pend->next = 0;
486         *c->in.pending_replies_tail = pend;
487         c->in.pending_replies_tail = &pend->next;
488     }
489     return 1;
490 }
491
492 int _xcb_in_read(XCBConnection *c)
493 {
494     int n = read(c->fd, c->in.queue + c->in.queue_len, sizeof(c->in.queue) - c->in.queue_len);
495     if(n > 0)
496         c->in.queue_len += n;
497     while(read_packet(c))
498         /* empty */;
499     return (n > 0) || (n < 0 && errno == EAGAIN);
500 }
501
502 int _xcb_in_read_block(XCBConnection *c, void *buf, int len)
503 {
504     int done = c->in.queue_len;
505     if(len < done)
506         done = len;
507
508     memcpy(buf, c->in.queue, done);
509     c->in.queue_len -= done;
510     memmove(c->in.queue, c->in.queue + done, c->in.queue_len);
511
512     if(len > done)
513     {
514         int ret = read_block(c->fd, (char *) buf + done, len - done);
515         if(ret <= 0)
516             return ret;
517     }
518
519     return len;
520 }