about summary refs log tree commit homepage
path: root/ext/raindrops/linux_inet_diag.c
diff options
context:
space:
mode:
authorEric Wong <normalperson@yhbt.net>2011-03-12 08:30:40 +0000
committerEric Wong <normalperson@yhbt.net>2011-03-12 08:30:52 +0000
commite6b5e940fdb8e2ded70a80132af6c280db02e8d8 (patch)
treee35bb2676c51fb241ac054d51df632c3e734eaad /ext/raindrops/linux_inet_diag.c
parentd4faac5480f6416cf92301745a9a9572bc865061 (diff)
downloadraindrops-e6b5e940fdb8e2ded70a80132af6c280db02e8d8.tar.gz
Be stricter about invalid inputs.
Diffstat (limited to 'ext/raindrops/linux_inet_diag.c')
-rw-r--r--ext/raindrops/linux_inet_diag.c19
1 files changed, 9 insertions, 10 deletions
diff --git a/ext/raindrops/linux_inet_diag.c b/ext/raindrops/linux_inet_diag.c
index 2358fc0..280bfc7 100644
--- a/ext/raindrops/linux_inet_diag.c
+++ b/ext/raindrops/linux_inet_diag.c
@@ -315,15 +315,15 @@ static void parse_addr(struct sockaddr_storage *inet, VALUE addr)
         if (*host_ptr == '[') { /* ipv6 address format (rfc2732) */
                 rbracket = memchr(host_ptr + 1, ']', host_len - 1);
 
-                if (rbracket) {
-                        if (rbracket[1] == ':') {
-                                colon = rbracket + 1;
-                                host_ptr++;
-                                *rbracket = 0;
-                        } else {
-                                rbracket = NULL;
-                        }
-                }
+                if (rbracket == NULL)
+                        rb_raise(rb_eArgError, "']' not found in IPv6 addr=%s",
+                                 host_ptr);
+                if (rbracket[1] != ':')
+                        rb_raise(rb_eArgError, "':' not found in IPv6 addr=%s",
+                                 host_ptr);
+                colon = rbracket + 1;
+                host_ptr++;
+                *rbracket = 0;
         } else { /* ipv4 */
                 colon = memchr(host_ptr, ':', host_len);
         }
@@ -337,7 +337,6 @@ static void parse_addr(struct sockaddr_storage *inet, VALUE addr)
         hints.ai_flags = AI_NUMERICHOST | AI_NUMERICSERV;
 
         *colon = 0;
-        if (rbracket) *rbracket = 0;
         rc = getaddrinfo(host_ptr, colon + 1, &hints, &res);
         *colon = ':';
         if (rbracket) *rbracket = ']';