darwin: Use read(2) rather than recv(2)
[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 #if USE_POLL
39 #include <poll.h>
40 #endif
41 #ifndef _WIN32
42 #include <sys/select.h>
43 #include <sys/socket.h>
44 #endif
45
46 #ifdef _WIN32
47 #include "xcb_windefs.h"
48 #endif /* _WIN32 */
49
50 #define XCB_ERROR 0
51 #define XCB_REPLY 1
52 #define XCB_XGE_EVENT 35
53
54 /* required for compiling for Win32 using MinGW */
55 #ifndef MSG_WAITALL
56 #define MSG_WAITALL 0
57 #endif
58
59 struct event_list {
60     xcb_generic_event_t *event;
61     struct event_list *next;
62 };
63
64 struct reply_list {
65     void *reply;
66     struct reply_list *next;
67 };
68
69 typedef struct pending_reply {
70     uint64_t first_request;
71     uint64_t last_request;
72     enum workarounds workaround;
73     int flags;
74     struct pending_reply *next;
75 } pending_reply;
76
77 typedef struct reader_list {
78     uint64_t request;
79     pthread_cond_t *data;
80     struct reader_list *next;
81 } reader_list;
82
83 static void remove_finished_readers(reader_list **prev_reader, uint64_t completed)
84 {
85     while(*prev_reader && XCB_SEQUENCE_COMPARE((*prev_reader)->request, <=, completed))
86     {
87         /* If you don't have what you're looking for now, you never
88          * will. Wake up and leave me alone. */
89         pthread_cond_signal((*prev_reader)->data);
90         *prev_reader = (*prev_reader)->next;
91     }
92 }
93
94 static int read_packet(xcb_connection_t *c)
95 {
96     xcb_generic_reply_t genrep;
97     int length = 32;
98     int eventlength = 0; /* length after first 32 bytes for GenericEvents */
99     void *buf;
100     pending_reply *pend = 0;
101     struct event_list *event;
102
103     /* Wait for there to be enough data for us to read a whole packet */
104     if(c->in.queue_len < length)
105         return 0;
106
107     /* Get the response type, length, and sequence number. */
108     memcpy(&genrep, c->in.queue, sizeof(genrep));
109
110     /* Compute 32-bit sequence number of this packet. */
111     if((genrep.response_type & 0x7f) != XCB_KEYMAP_NOTIFY)
112     {
113         uint64_t lastread = c->in.request_read;
114         c->in.request_read = (lastread & UINT64_C(0xffffffffffff0000)) | genrep.sequence;
115         if(XCB_SEQUENCE_COMPARE(c->in.request_read, <, lastread))
116             c->in.request_read += 0x10000;
117         if(XCB_SEQUENCE_COMPARE(c->in.request_read, >, c->in.request_expected))
118             c->in.request_expected = c->in.request_read;
119
120         if(c->in.request_read != lastread)
121         {
122             if(c->in.current_reply)
123             {
124                 _xcb_map_put(c->in.replies, lastread, c->in.current_reply);
125                 c->in.current_reply = 0;
126                 c->in.current_reply_tail = &c->in.current_reply;
127             }
128             c->in.request_completed = c->in.request_read - 1;
129         }
130
131         while(c->in.pending_replies && 
132               c->in.pending_replies->workaround != WORKAROUND_EXTERNAL_SOCKET_OWNER &&
133               XCB_SEQUENCE_COMPARE (c->in.pending_replies->last_request, <=, c->in.request_completed))
134         {
135             pending_reply *oldpend = c->in.pending_replies;
136             c->in.pending_replies = oldpend->next;
137             if(!oldpend->next)
138                 c->in.pending_replies_tail = &c->in.pending_replies;
139             free(oldpend);
140         }
141
142         if(genrep.response_type == XCB_ERROR)
143             c->in.request_completed = c->in.request_read;
144
145         remove_finished_readers(&c->in.readers, c->in.request_completed);
146     }
147
148     if(genrep.response_type == XCB_ERROR || genrep.response_type == XCB_REPLY)
149     {
150         pend = c->in.pending_replies;
151         if(pend &&
152            !(XCB_SEQUENCE_COMPARE(pend->first_request, <=, c->in.request_read) &&
153              (pend->workaround == WORKAROUND_EXTERNAL_SOCKET_OWNER ||
154               XCB_SEQUENCE_COMPARE(c->in.request_read, <=, pend->last_request))))
155             pend = 0;
156     }
157
158     /* For reply packets, check that the entire packet is available. */
159     if(genrep.response_type == XCB_REPLY)
160     {
161         if(pend && pend->workaround == WORKAROUND_GLX_GET_FB_CONFIGS_BUG)
162         {
163             uint32_t *p = (uint32_t *) c->in.queue;
164             genrep.length = p[2] * p[3] * 2;
165         }
166         length += genrep.length * 4;
167     }
168
169     /* XGE events may have sizes > 32 */
170     if ((genrep.response_type & 0x7f) == XCB_XGE_EVENT)
171         eventlength = genrep.length * 4;
172
173     buf = malloc(length + eventlength +
174             (genrep.response_type == XCB_REPLY ? 0 : sizeof(uint32_t)));
175     if(!buf)
176     {
177         _xcb_conn_shutdown(c, XCB_CONN_CLOSED_MEM_INSUFFICIENT);
178         return 0;
179     }
180
181     if(_xcb_in_read_block(c, buf, length) <= 0)
182     {
183         free(buf);
184         return 0;
185     }
186
187     /* pull in XGE event data if available, append after event struct */
188     if (eventlength)
189     {
190         if(_xcb_in_read_block(c, &((xcb_generic_event_t*)buf)[1], eventlength) <= 0)
191         {
192             free(buf);
193             return 0;
194         }
195     }
196
197     if(pend && (pend->flags & XCB_REQUEST_DISCARD_REPLY))
198     {
199         free(buf);
200         return 1;
201     }
202
203     if(genrep.response_type != XCB_REPLY)
204         ((xcb_generic_event_t *) buf)->full_sequence = c->in.request_read;
205
206     /* reply, or checked error */
207     if( genrep.response_type == XCB_REPLY ||
208        (genrep.response_type == XCB_ERROR && pend && (pend->flags & XCB_REQUEST_CHECKED)))
209     {
210         struct reply_list *cur = malloc(sizeof(struct reply_list));
211         if(!cur)
212         {
213             _xcb_conn_shutdown(c, XCB_CONN_CLOSED_MEM_INSUFFICIENT);
214             free(buf);
215             return 0;
216         }
217         cur->reply = buf;
218         cur->next = 0;
219         *c->in.current_reply_tail = cur;
220         c->in.current_reply_tail = &cur->next;
221         if(c->in.readers && c->in.readers->request == c->in.request_read)
222             pthread_cond_signal(c->in.readers->data);
223         return 1;
224     }
225
226     /* event, or unchecked error */
227     event = malloc(sizeof(struct event_list));
228     if(!event)
229     {
230         _xcb_conn_shutdown(c, XCB_CONN_CLOSED_MEM_INSUFFICIENT);
231         free(buf);
232         return 0;
233     }
234     event->event = buf;
235     event->next = 0;
236     *c->in.events_tail = event;
237     c->in.events_tail = &event->next;
238     pthread_cond_signal(&c->in.event_cond);
239     return 1; /* I have something for you... */
240 }
241
242 static xcb_generic_event_t *get_event(xcb_connection_t *c)
243 {
244     struct event_list *cur = c->in.events;
245     xcb_generic_event_t *ret;
246     if(!c->in.events)
247         return 0;
248     ret = cur->event;
249     c->in.events = cur->next;
250     if(!cur->next)
251         c->in.events_tail = &c->in.events;
252     free(cur);
253     return ret;
254 }
255
256 static void free_reply_list(struct reply_list *head)
257 {
258     while(head)
259     {
260         struct reply_list *cur = head;
261         head = cur->next;
262         free(cur->reply);
263         free(cur);
264     }
265 }
266
267 static int read_block(const int fd, void *buf, const ssize_t len)
268 {
269     int done = 0;
270     while(done < len)
271     {
272 #ifdef __APPLE__
273         int ret = read(fd, ((char *) buf) + done, len - done);
274 #else
275         int ret = recv(fd, ((char *) buf) + done, len - done,MSG_WAITALL);
276 #endif
277         if(ret > 0)
278             done += ret;
279 #ifndef _WIN32
280         if(ret < 0 && errno == EAGAIN)
281 #else
282         if(ret == SOCKET_ERROR && WSAGetLastError() == WSAEWOULDBLOCK)
283 #endif /* !_Win32 */
284         {
285 #if USE_POLL
286             struct pollfd pfd;
287             pfd.fd = fd;
288             pfd.events = POLLIN;
289             pfd.revents = 0;
290             do {
291                 ret = poll(&pfd, 1, -1);
292             } while (ret == -1 && errno == EINTR);
293 #else
294             fd_set fds;
295             FD_ZERO(&fds);
296             FD_SET(fd, &fds);
297
298             /* Initializing errno here makes sure that for Win32 this loop will execute only once */
299             errno = 0;  
300             do {
301                 ret = select(fd + 1, &fds, 0, 0, 0);
302             } while (ret == -1 && errno == EINTR);
303 #endif /* USE_POLL */
304         }
305         if(ret <= 0)
306             return ret;
307     }
308     return len;
309 }
310
311 static int poll_for_reply(xcb_connection_t *c, uint64_t request, void **reply, xcb_generic_error_t **error)
312 {
313     struct reply_list *head;
314
315     /* If an error occurred when issuing the request, fail immediately. */
316     if(!request)
317         head = 0;
318     /* We've read requests past the one we want, so if it has replies we have
319      * them all and they're in the replies map. */
320     else if(XCB_SEQUENCE_COMPARE(request, <, c->in.request_read))
321     {
322         head = _xcb_map_remove(c->in.replies, request);
323         if(head && head->next)
324             _xcb_map_put(c->in.replies, request, head->next);
325     }
326     /* We're currently processing the responses to the request we want, and we
327      * have a reply ready to return. So just return it without blocking. */
328     else if(request == c->in.request_read && c->in.current_reply)
329     {
330         head = c->in.current_reply;
331         c->in.current_reply = head->next;
332         if(!head->next)
333             c->in.current_reply_tail = &c->in.current_reply;
334     }
335     /* We know this request can't have any more replies, and we've already
336      * established it doesn't have a reply now. Don't bother blocking. */
337     else if(request == c->in.request_completed)
338         head = 0;
339     /* We may have more replies on the way for this request: block until we're
340      * sure. */
341     else
342         return 0;
343
344     if(error)
345         *error = 0;
346     *reply = 0;
347
348     if(head)
349     {
350         if(((xcb_generic_reply_t *) head->reply)->response_type == XCB_ERROR)
351         {
352             if(error)
353                 *error = head->reply;
354             else
355                 free(head->reply);
356         }
357         else
358             *reply = head->reply;
359
360         free(head);
361     }
362
363     return 1;
364 }
365
366 static void insert_reader(reader_list **prev_reader, reader_list *reader, uint64_t request, pthread_cond_t *cond)
367 {
368     while(*prev_reader && XCB_SEQUENCE_COMPARE((*prev_reader)->request, <=, request))
369         prev_reader = &(*prev_reader)->next;
370     reader->request = request;
371     reader->data = cond;
372     reader->next = *prev_reader;
373     *prev_reader = reader;
374 }
375
376 static void remove_reader(reader_list **prev_reader, reader_list *reader)
377 {
378     while(*prev_reader && XCB_SEQUENCE_COMPARE((*prev_reader)->request, <=, reader->request))
379         if(*prev_reader == reader)
380         {
381             *prev_reader = (*prev_reader)->next;
382             break;
383         }
384 }
385
386 static void *wait_for_reply(xcb_connection_t *c, uint64_t request, xcb_generic_error_t **e)
387 {
388     void *ret = 0;
389
390     /* If this request has not been written yet, write it. */
391     if(c->out.return_socket || _xcb_out_flush_to(c, request))
392     {
393         pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
394         reader_list reader;
395
396         insert_reader(&c->in.readers, &reader, request, &cond);
397
398         while(!poll_for_reply(c, request, &ret, e))
399             if(!_xcb_conn_wait(c, &cond, 0, 0))
400                 break;
401
402         remove_reader(&c->in.readers, &reader);
403         pthread_cond_destroy(&cond);
404     }
405
406     _xcb_in_wake_up_next_reader(c);
407     return ret;
408 }
409
410 static uint64_t widen(xcb_connection_t *c, unsigned int request)
411 {
412     uint64_t widened_request = (c->out.request & UINT64_C(0xffffffff00000000)) | request;
413     if(widened_request > c->out.request)
414         widened_request -= UINT64_C(1) << 32;
415     return widened_request;
416 }
417
418 /* Public interface */
419
420 void *xcb_wait_for_reply(xcb_connection_t *c, unsigned int request, xcb_generic_error_t **e)
421 {
422     void *ret;
423     if(e)
424         *e = 0;
425     if(c->has_error)
426         return 0;
427
428     pthread_mutex_lock(&c->iolock);
429     ret = wait_for_reply(c, widen(c, request), e);
430     pthread_mutex_unlock(&c->iolock);
431     return ret;
432 }
433
434 static void insert_pending_discard(xcb_connection_t *c, pending_reply **prev_next, uint64_t seq)
435 {
436     pending_reply *pend;
437     pend = malloc(sizeof(*pend));
438     if(!pend)
439     {
440         _xcb_conn_shutdown(c, XCB_CONN_CLOSED_MEM_INSUFFICIENT);
441         return;
442     }
443
444     pend->first_request = seq;
445     pend->last_request = seq;
446     pend->workaround = 0;
447     pend->flags = XCB_REQUEST_DISCARD_REPLY;
448     pend->next = *prev_next;
449     *prev_next = pend;
450
451     if(!pend->next)
452         c->in.pending_replies_tail = &pend->next;
453 }
454
455 static void discard_reply(xcb_connection_t *c, uint64_t request)
456 {
457     void *reply;
458     pending_reply **prev_pend;
459
460     /* Free any replies or errors that we've already read. Stop if
461      * xcb_wait_for_reply would block or we've run out of replies. */
462     while(poll_for_reply(c, request, &reply, 0) && reply)
463         free(reply);
464
465     /* If we've proven there are no more responses coming, we're done. */
466     if(XCB_SEQUENCE_COMPARE(request, <=, c->in.request_completed))
467         return;
468
469     /* Walk the list of pending requests. Mark the first match for deletion. */
470     for(prev_pend = &c->in.pending_replies; *prev_pend; prev_pend = &(*prev_pend)->next)
471     {
472         if(XCB_SEQUENCE_COMPARE((*prev_pend)->first_request, >, request))
473             break;
474
475         if((*prev_pend)->first_request == request)
476         {
477             /* Pending reply found. Mark for discard: */
478             (*prev_pend)->flags |= XCB_REQUEST_DISCARD_REPLY;
479             return;
480         }
481     }
482
483     /* Pending reply not found (likely due to _unchecked request). Create one: */
484     insert_pending_discard(c, prev_pend, request);
485 }
486
487 void xcb_discard_reply(xcb_connection_t *c, unsigned int sequence)
488 {
489     if(c->has_error)
490         return;
491
492     /* If an error occurred when issuing the request, fail immediately. */
493     if(!sequence)
494         return;
495
496     pthread_mutex_lock(&c->iolock);
497     discard_reply(c, widen(c, sequence));
498     pthread_mutex_unlock(&c->iolock);
499 }
500
501 int xcb_poll_for_reply(xcb_connection_t *c, unsigned int request, void **reply, xcb_generic_error_t **error)
502 {
503     int ret;
504     if(c->has_error)
505     {
506         *reply = 0;
507         if(error)
508             *error = 0;
509         return 1; /* would not block */
510     }
511     assert(reply != 0);
512     pthread_mutex_lock(&c->iolock);
513     ret = poll_for_reply(c, widen(c, request), reply, error);
514     pthread_mutex_unlock(&c->iolock);
515     return ret;
516 }
517
518 xcb_generic_event_t *xcb_wait_for_event(xcb_connection_t *c)
519 {
520     xcb_generic_event_t *ret;
521     if(c->has_error)
522         return 0;
523     pthread_mutex_lock(&c->iolock);
524     /* get_event returns 0 on empty list. */
525     while(!(ret = get_event(c)))
526         if(!_xcb_conn_wait(c, &c->in.event_cond, 0, 0))
527             break;
528
529     _xcb_in_wake_up_next_reader(c);
530     pthread_mutex_unlock(&c->iolock);
531     return ret;
532 }
533
534 static xcb_generic_event_t *poll_for_next_event(xcb_connection_t *c, int queued)
535 {
536     xcb_generic_event_t *ret = 0;
537     if(!c->has_error)
538     {
539         pthread_mutex_lock(&c->iolock);
540         /* FIXME: follow X meets Z architecture changes. */
541         ret = get_event(c);
542         if(!ret && !queued && c->in.reading == 0 && _xcb_in_read(c)) /* _xcb_in_read shuts down the connection on error */
543             ret = get_event(c);
544         pthread_mutex_unlock(&c->iolock);
545     }
546     return ret;
547 }
548
549 xcb_generic_event_t *xcb_poll_for_event(xcb_connection_t *c)
550 {
551     return poll_for_next_event(c, 0);
552 }
553
554 xcb_generic_event_t *xcb_poll_for_queued_event(xcb_connection_t *c)
555 {
556     return poll_for_next_event(c, 1);
557 }
558
559 xcb_generic_error_t *xcb_request_check(xcb_connection_t *c, xcb_void_cookie_t cookie)
560 {
561     uint64_t request;
562     xcb_generic_error_t *ret = 0;
563     void *reply;
564     if(c->has_error)
565         return 0;
566     pthread_mutex_lock(&c->iolock);
567     request = widen(c, cookie.sequence);
568     if(XCB_SEQUENCE_COMPARE(request, >=, c->in.request_expected)
569        && XCB_SEQUENCE_COMPARE(request, >, c->in.request_completed))
570     {
571         _xcb_out_send_sync(c);
572         _xcb_out_flush_to(c, c->out.request);
573     }
574     reply = wait_for_reply(c, request, &ret);
575     assert(!reply);
576     pthread_mutex_unlock(&c->iolock);
577     return ret;
578 }
579
580 /* Private interface */
581
582 int _xcb_in_init(_xcb_in *in)
583 {
584     if(pthread_cond_init(&in->event_cond, 0))
585         return 0;
586     in->reading = 0;
587
588     in->queue_len = 0;
589
590     in->request_read = 0;
591     in->request_completed = 0;
592
593     in->replies = _xcb_map_new();
594     if(!in->replies)
595         return 0;
596
597     in->current_reply_tail = &in->current_reply;
598     in->events_tail = &in->events;
599     in->pending_replies_tail = &in->pending_replies;
600
601     return 1;
602 }
603
604 void _xcb_in_destroy(_xcb_in *in)
605 {
606     pthread_cond_destroy(&in->event_cond);
607     free_reply_list(in->current_reply);
608     _xcb_map_delete(in->replies, (void (*)(void *)) free_reply_list);
609     while(in->events)
610     {
611         struct event_list *e = in->events;
612         in->events = e->next;
613         free(e->event);
614         free(e);
615     }
616     while(in->pending_replies)
617     {
618         pending_reply *pend = in->pending_replies;
619         in->pending_replies = pend->next;
620         free(pend);
621     }
622 }
623
624 void _xcb_in_wake_up_next_reader(xcb_connection_t *c)
625 {
626     int pthreadret;
627     if(c->in.readers)
628         pthreadret = pthread_cond_signal(c->in.readers->data);
629     else
630         pthreadret = pthread_cond_signal(&c->in.event_cond);
631     assert(pthreadret == 0);
632 }
633
634 int _xcb_in_expect_reply(xcb_connection_t *c, uint64_t request, enum workarounds workaround, int flags)
635 {
636     pending_reply *pend = malloc(sizeof(pending_reply));
637     assert(workaround != WORKAROUND_NONE || flags != 0);
638     if(!pend)
639     {
640         _xcb_conn_shutdown(c, XCB_CONN_CLOSED_MEM_INSUFFICIENT);
641         return 0;
642     }
643     pend->first_request = pend->last_request = request;
644     pend->workaround = workaround;
645     pend->flags = flags;
646     pend->next = 0;
647     *c->in.pending_replies_tail = pend;
648     c->in.pending_replies_tail = &pend->next;
649     return 1;
650 }
651
652 void _xcb_in_replies_done(xcb_connection_t *c)
653 {
654     struct pending_reply *pend;
655     if (c->in.pending_replies_tail != &c->in.pending_replies)
656     {
657         pend = container_of(c->in.pending_replies_tail, struct pending_reply, next);
658         if(pend->workaround == WORKAROUND_EXTERNAL_SOCKET_OWNER)
659         {
660             pend->last_request = c->out.request;
661             pend->workaround = WORKAROUND_NONE;
662         }
663     }
664 }
665
666 int _xcb_in_read(xcb_connection_t *c)
667 {
668 #ifdef __APPLE__
669     int n = read(c->fd, c->in.queue + c->in.queue_len, sizeof(c->in.queue) - c->in.queue_len);
670 #else
671     int n = recv(c->fd, c->in.queue + c->in.queue_len, sizeof(c->in.queue) - c->in.queue_len,MSG_WAITALL);
672 #endif
673     if(n > 0)
674         c->in.queue_len += n;
675     while(read_packet(c))
676         /* empty */;
677 #ifndef _WIN32
678     if((n > 0) || (n < 0 && errno == EAGAIN))
679 #else
680     if((n > 0) || (n < 0 && WSAGetLastError() == WSAEWOULDBLOCK))
681 #endif /* !_WIN32 */
682         return 1;
683     _xcb_conn_shutdown(c, XCB_CONN_ERROR);
684     return 0;
685 }
686
687 int _xcb_in_read_block(xcb_connection_t *c, void *buf, int len)
688 {
689     int done = c->in.queue_len;
690     if(len < done)
691         done = len;
692
693     memcpy(buf, c->in.queue, done);
694     c->in.queue_len -= done;
695     memmove(c->in.queue, c->in.queue + done, c->in.queue_len);
696
697     if(len > done)
698     {
699         int ret = read_block(c->fd, (char *) buf + done, len - done);
700         if(ret <= 0)
701         {
702             _xcb_conn_shutdown(c, XCB_CONN_ERROR);
703             return ret;
704         }
705     }
706
707     return len;
708 }