Simplify: Always use writev. (In _xcb_out_flush, convert the output queue to a single...
[free-sw/xcb/libxcb] / src / xcb_out.c
index 442fcf5..e7d5323 100644 (file)
 #include <string.h>
 #include <errno.h>
 
+#ifndef __GNUC__
+# if HAVE_ALLOCA_H
+#  include <alloca.h>
+# else
+#  ifdef _AIX
+ #pragma alloca
+#  endif
+# endif
+#endif
+
 #include "xcb.h"
 #include "xcbext.h"
 #include "xcbint.h"
@@ -48,18 +58,6 @@ static int force_sequence_wrap(XCBConnection *c)
     return ret;
 }
 
-static int _xcb_write(const int fd, char (*buf)[], int *count)
-{
-    int n = write(fd, *buf, *count);
-    if(n > 0)
-    {
-        *count -= n;
-        if(*count)
-            memmove(*buf, *buf + n, *count);
-    }
-    return n;
-}
-
 static int _xcb_writev(const int fd, struct iovec *vec, int count)
 {
     int n = writev(fd, vec, count);
@@ -105,9 +103,11 @@ CARD32 XCBGetMaximumRequestLength(XCBConnection *c)
 
 int XCBSendRequest(XCBConnection *c, unsigned int *request, struct iovec *vector, const XCBProtocolRequest *req)
 {
+    static const char pad[3];
     int ret;
     int i;
-    struct iovec prefix[2];
+    struct iovec *padded;
+    int padlen = 0;
     CARD16 shortlen = 0;
     CARD32 longlen = 0;
     enum workarounds workaround = WORKAROUND_NONE;
@@ -151,17 +151,37 @@ int XCBSendRequest(XCBConnection *c, unsigned int *request, struct iovec *vector
         longlen = 0;
     }
 
+    padded =
+#ifdef HAVE_ALLOCA
+        alloca
+#else
+        malloc
+#endif
+        ((req->count * 2 + 3) * sizeof(struct iovec));
     /* set the length field. */
     ((CARD16 *) vector[0].iov_base)[1] = shortlen;
     if(!shortlen)
     {
-        prefix[0].iov_base = vector[0].iov_base;
-        prefix[0].iov_len = sizeof(CARD32);
+        padded[0].iov_base = vector[0].iov_base;
+        padded[0].iov_len = sizeof(CARD32);
         vector[0].iov_base = ((char *) vector[0].iov_base) + sizeof(CARD32);
         vector[0].iov_len -= sizeof(CARD32);
         ++longlen;
-        prefix[1].iov_base = &longlen;
-        prefix[1].iov_len = sizeof(CARD32);
+        padded[1].iov_base = &longlen;
+        padded[1].iov_len = sizeof(CARD32);
+        padlen = 2;
+    }
+
+    for(i = 0; i < req->count; ++i)
+    {
+        if(!vector[i].iov_len)
+            continue;
+        padded[padlen].iov_base = vector[i].iov_base;
+        padded[padlen++].iov_len = vector[i].iov_len;
+        if(!XCB_PAD(vector[i].iov_len))
+            continue;
+        padded[padlen].iov_base = (caddr_t) pad;
+        padded[padlen++].iov_len = XCB_PAD(vector[i].iov_len);
     }
 
     /* get a sequence number and arrange for delivery. */
@@ -169,6 +189,9 @@ int XCBSendRequest(XCBConnection *c, unsigned int *request, struct iovec *vector
     if(req->isvoid && !force_sequence_wrap(c))
     {
         pthread_mutex_unlock(&c->iolock);
+#ifndef HAVE_ALLOCA
+        free(padded);
+#endif
         return -1;
     }
 
@@ -177,13 +200,11 @@ int XCBSendRequest(XCBConnection *c, unsigned int *request, struct iovec *vector
     if(!req->isvoid)
         _xcb_in_expect_reply(c, *request, workaround);
 
-    if(!shortlen)
-        ret = _xcb_out_write_block(c, prefix, 2);
-    else
-        ret = 1;
-    if(ret > 0)
-        ret = _xcb_out_write_block(c, vector, req->count);
+    ret = _xcb_out_write_block(c, padded, padlen);
     pthread_mutex_unlock(&c->iolock);
+#ifndef HAVE_ALLOCA
+    free(padded);
+#endif
 
     return ret;
 }
@@ -223,89 +244,59 @@ void _xcb_out_destroy(_xcb_out *out)
 {
     pthread_cond_destroy(&out->cond);
     pthread_mutex_destroy(&out->reqlenlock);
-    free(out->vec);
 }
 
+/* precondition: there must be something for us to write. */
 int _xcb_out_write(XCBConnection *c)
 {
     int n;
-    if(c->out.vec_len)
-        n = _xcb_writev(c->fd, c->out.vec, c->out.vec_len);
-    else
-        n = _xcb_write(c->fd, &c->out.queue, &c->out.queue_len);
-
-    /* XXX: should "nothing was written" be considered failure or
-     * success for this function? it's not an I/O error, but... */
-    n = (n > 0) || (n < 0 && errno == EAGAIN);
-
-    if(c->out.vec_len)
-    {
-        int i;
-        for(i = 0; i < c->out.vec_len; ++i)
-            if(c->out.vec[i].iov_len)
-                return n;
-        c->out.vec_len = 0;
-    }
-    return n;
+    assert(!c->out.queue_len);
+    n = _xcb_writev(c->fd, c->out.vec, c->out.vec_len);
+    while(c->out.vec_len && !c->out.vec[0].iov_len)
+        ++c->out.vec, --c->out.vec_len;
+    if(!c->out.vec_len)
+        c->out.vec = 0;
+    return (n > 0) || (n < 0 && errno == EAGAIN);
 }
 
 int _xcb_out_write_block(XCBConnection *c, struct iovec *vector, size_t count)
 {
-    static const char pad[3];
-    int i;
-    int len = 0;
-
-    for(i = 0; i < count; ++i)
-        len += XCB_CEIL(vector[i].iov_len);
-
-    /* Is the queue about to overflow? */
-    if(c->out.queue_len + len < sizeof(c->out.queue))
+    while(count && c->out.queue_len + vector[0].iov_len < sizeof(c->out.queue))
     {
-        /* No, this will fit. */
-        for(i = 0; i < count; ++i)
-        {
-            memcpy(c->out.queue + c->out.queue_len, vector[i].iov_base, vector[i].iov_len);
-            if(vector[i].iov_len & 3)
-                memset(c->out.queue + c->out.queue_len + vector[i].iov_len, 0, XCB_PAD(vector[i].iov_len));
-            c->out.queue_len += XCB_CEIL(vector[i].iov_len);
-        }
-        return len;
+        memcpy(c->out.queue + c->out.queue_len, vector[0].iov_base, vector[0].iov_len);
+        c->out.queue_len += vector[0].iov_len;
+        ++vector, --count;
     }
+    if(!count)
+        return 1;
+
+    memmove(vector + 1, vector, count++ * sizeof(struct iovec));
+    vector[0].iov_base = c->out.queue;
+    vector[0].iov_len = c->out.queue_len;
+    c->out.queue_len = 0;
 
     assert(!c->out.vec_len);
     assert(!c->out.vec);
-    c->out.vec = malloc(sizeof(struct iovec) * (1 + count * 2));
-    if(!c->out.vec)
-        return -1;
-    if(c->out.queue_len)
-    {
-        c->out.vec[c->out.vec_len].iov_base = c->out.queue;
-        c->out.vec[c->out.vec_len++].iov_len = c->out.queue_len;
-        c->out.queue_len = 0;
-    }
-    for(i = 0; i < count; ++i)
-    {
-        if(!vector[i].iov_len)
-            continue;
-        c->out.vec[c->out.vec_len].iov_base = vector[i].iov_base;
-        c->out.vec[c->out.vec_len++].iov_len = vector[i].iov_len;
-        if(!XCB_PAD(vector[i].iov_len))
-            continue;
-        c->out.vec[c->out.vec_len].iov_base = (void *) pad;
-        c->out.vec[c->out.vec_len++].iov_len = XCB_PAD(vector[i].iov_len);
-    }
-    if(!_xcb_out_flush(c))
-        len = -1;
-    free(c->out.vec);
-    c->out.vec = 0;
-
-    return len;
+    c->out.vec_len = count;
+    c->out.vec = vector;
+    return _xcb_out_flush(c);
 }
 
 int _xcb_out_flush(XCBConnection *c)
 {
     int ret = 1;
-    while(ret && (c->out.queue_len || c->out.vec_len))
+    struct iovec vec;
+    if(c->out.queue_len)
+    {
+        assert(!c->out.vec_len);
+        assert(!c->out.vec);
+        vec.iov_base = c->out.queue;
+        vec.iov_len = c->out.queue_len;
+        c->out.vec = &vec;
+        c->out.vec_len = 1;
+        c->out.queue_len = 0;
+    }
+    while(ret && c->out.vec_len)
         ret = _xcb_conn_wait(c, /*should_write*/ 1, &c->out.cond);
     c->out.request_written = c->out.request;
     pthread_cond_broadcast(&c->out.cond);