CDRIVER-536 hostnames are normalized to lowercase

Author: samantharitter

src/mongoc/Makefile.am

@@ -77,6 +77,7 @@ INST_H_FILES = \
 	src/mongoc/mongoc-thread-private.h \
 	src/mongoc/mongoc-trace.h \
 	src/mongoc/mongoc-uri.h \
+	src/mongoc/mongoc-uri-private.h \
 	src/mongoc/mongoc-util-private.h \
 	src/mongoc/mongoc-version.h \
 	src/mongoc/mongoc-write-command-private.h \

src/mongoc/mongoc-uri-private.h

@@ -0,0 +1,39 @@
+/*
+ * Copyright 2015 MongoDB, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef MONGOC_URI_PRIVATE_H
+#define MONGOC_URI_PRIVATE_H
+
+#if !defined (MONGOC_I_AM_A_DRIVER) && !defined (MONGOC_COMPILATION)
+#error "Only <mongoc.h> can be included directly."
+#endif
+
+#include "mongoc-uri.h"
+
+
+BSON_BEGIN_DECLS
+
+
+void
+mongoc_uri_lowercase_hostname (const char *src,
+                               char       *buf /* OUT */,
+                               int         len);
+
+
+BSON_BEGIN_DECLS
+
+
+#endif /* MONGOC_URI_PRIVATE_H */

src/mongoc/mongoc-uri.c

@@ -23,7 +23,7 @@
 #include "mongoc-host-list-private.h"
 #include "mongoc-log.h"
 #include "mongoc-socket.h"
-#include "mongoc-uri.h"
+#include "mongoc-uri-private.h"
 
 
 #if defined(_WIN32) && !defined(strcasecmp)
@@ -55,6 +55,24 @@ mongoc_uri_do_unescape (char **str)
    }
 }
 
+void
+mongoc_uri_lowercase_hostname (const char *src,
+                               char *buf /* OUT */,
+                               int len)
+{
+   bson_unichar_t c;
+   const char *iter;
+   char *buf_iter;
+
+   /* TODO: this code only accepts ascii, and assumes that lowercased
+      chars are the same width as originals */
+   for (iter = src, buf_iter = buf;
+        iter && *iter && (c = bson_utf8_get_char(iter)) && buf_iter - buf < len;
+        iter = bson_utf8_next_char(iter), buf_iter++) {
+      assert(c < 128);
+      *buf_iter = tolower(c);
+   }
+}
 
 static void
 mongoc_uri_append_host (mongoc_uri_t  *uri,
@@ -65,7 +83,7 @@ mongoc_uri_append_host (mongoc_uri_t  *uri,
    mongoc_host_list_t *link_;
 
    link_ = bson_malloc0(sizeof *link_);
-   bson_strncpy (link_->host, host, sizeof link_->host);
+   mongoc_uri_lowercase_hostname(host, link_->host, sizeof link_->host);
    if (strchr (host, ':')) {
       bson_snprintf (link_->host_and_port, sizeof link_->host_and_port,
                      "[%s]:%hu", host, port);

tests/test-mongoc-uri.c

@@ -41,6 +41,14 @@ test_mongoc_uri_new (void)
    ASSERT(uri);
    mongoc_uri_destroy(uri);
 
+   /* should normalize to lowercase */
+   uri = mongoc_uri_new ("mongodb://cRaZyHoStNaMe");
+   assert (uri);
+   hosts = mongoc_uri_get_hosts (uri);
+   assert (hosts);
+   ASSERT_CMPSTR (hosts->host, "crazyhostname");
+   mongoc_uri_destroy (uri);
+
    uri = mongoc_uri_new("mongodb://localhost/?");
    ASSERT(uri);
    mongoc_uri_destroy(uri);