From d89e09b3087dbaf09301648868d0d8ac340da04c Mon Sep 17 00:00:00 2001
From: akr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Wed, 18 Feb 2009 12:27:13 +0000
Subject: * ext/socket/ancdata.c (bsock_recvmsg_internal): reduce code on  
 environments which have no control message.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@22401 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 ext/socket/ancdata.c | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

(limited to 'ext/socket/ancdata.c')

diff --git a/ext/socket/ancdata.c b/ext/socket/ancdata.c
index 04c1b20889..68bb272ef3 100644
--- a/ext/socket/ancdata.c
+++ b/ext/socket/ancdata.c
@@ -1102,27 +1102,33 @@ bsock_recvmsg_internal(int argc, VALUE *argv, VALUE sock, int nonblock)
     rb_io_t *fptr;
     VALUE vmaxdatlen, vmaxctllen, vflags;
     int grow_buffer;
-    size_t maxdatlen, maxctllen;
+    size_t maxdatlen;
     int flags, orig_flags;
     struct msghdr mh;
     struct iovec iov;
-#if defined(HAVE_ST_MSG_CONTROL)
-    struct cmsghdr *cmh;
-#endif
     struct sockaddr_storage namebuf;
     char datbuf0[4096], *datbuf;
-    char ctlbuf0[4096], *ctlbuf;
     VALUE dat_str = Qnil;
-    VALUE ctl_str = Qnil;
     VALUE ret;
     ssize_t ss;
+#if defined(HAVE_ST_MSG_CONTROL)
+    struct cmsghdr *cmh;
+    size_t maxctllen;
+    char ctlbuf0[4096], *ctlbuf;
+    VALUE ctl_str = Qnil;
+#endif
 
     rb_secure(4);
 
     rb_scan_args(argc, argv, "03", &vmaxdatlen, &vflags, &vmaxctllen);
 
     maxdatlen = NIL_P(vmaxdatlen) ? sizeof(datbuf0) : NUM2SIZET(vmaxdatlen);
+#if defined(HAVE_ST_MSG_CONTROL)
     maxctllen = NIL_P(vmaxctllen) ? sizeof(ctlbuf0) : NUM2SIZET(vmaxctllen);
+#else
+    if (!NIL_P(vmaxctllen))
+        rb_raise(rb_eArgError, "control message not supported");
+#endif
     flags = NIL_P(vflags) ? 0 : NUM2INT(vflags);
 #ifdef MSG_DONTWAIT
     if (nonblock)
@@ -1159,6 +1165,7 @@ bsock_recvmsg_internal(int argc, VALUE *argv, VALUE sock, int nonblock)
         datbuf = RSTRING_PTR(dat_str);
     }
 
+#if defined(HAVE_ST_MSG_CONTROL)
     if (maxctllen <= sizeof(ctlbuf0))
         ctlbuf = ctlbuf0;
     else {
@@ -1168,6 +1175,7 @@ bsock_recvmsg_internal(int argc, VALUE *argv, VALUE sock, int nonblock)
             rb_str_resize(ctl_str, maxctllen);
         ctlbuf = RSTRING_PTR(ctl_str);
     }
+#endif
 
     memset(&mh, 0, sizeof(mh));
 
-- 
cgit v1.2.3