ares test: fix win32 build errors with virtual socket function tests

The added api requires both some typedefs not previously imported
into the test build + the test code did not fully deal with
socket differences on windows.

test/ares-test-internal.cc

@@ -2,7 +2,10 @@
 #include "dns-proto.h"
 
 #include <stdio.h>
+
+#ifdef HAVE_UNISTD_H
 #include <unistd.h>
+#endif
 #include <fcntl.h>
 
 extern "C" {
@@ -388,15 +391,9 @@ TEST_F(DefaultChannelTest, SaveInvalidChannel) {
   channel_->nservers = saved;
 }
 
-// TODO: This should not really be in this file, but we need ares config
-// flags, and here they are available.
-const struct ares_socket_functions VirtualizeIO::default_functions = {
-  [](int af, int type, int protocol, void *) {
-    auto s = ::socket(af, type, protocol);
-    if (s < 0) {
-      return s;
-    }
-    auto res = [s] {
+// Need to put this in own function due to nested lambda bug
+// in VS2013. (C2888)
+static int configure_socket(ares_socket_t s) {
   // transposed from ares-process, simplified non-block setter.
 #if defined(USE_BLOCKING_SOCKETS)
   return 0; /* returns success */
@@ -428,28 +425,41 @@ const struct ares_socket_functions VirtualizeIO::default_functions = {
 #else
 #  error "no non-blocking method was found/used/set"
 #endif
-    }();
-    if (res != 0) {
-      ::close(s);
-      return -1;
+}
+
+// TODO: This should not really be in this file, but we need ares config
+// flags, and here they are available.
+const struct ares_socket_functions VirtualizeIO::default_functions = {
+  [](int af, int type, int protocol, void *) -> ares_socket_t {
+    auto s = ::socket(af, type, protocol);
+    if (s == ARES_SOCKET_BAD) {
+      return s;
+    }
+    if (configure_socket(s) != 0) {
+      sclose(s);
+      return ares_socket_t(-1);
     }
     return s;
   },
   [](ares_socket_t s, void * p) {
-    return ::close(s);
+    return :: sclose(s);
   },
   [](ares_socket_t s, const struct sockaddr * addr, socklen_t len, void *) {
     return ::connect(s, addr, len);
   },
-  [](ares_socket_t s, void * dst, size_t len, int flags, struct sockaddr * addr, socklen_t * alen, void *) {
+  [](ares_socket_t s, void * dst, size_t len, int flags, struct sockaddr * addr, socklen_t * alen, void *) -> ssize_t {
 #ifdef HAVE_RECVFROM
-    return ::recvfrom(s, dst, len, flags, addr, alen);
+    return ::recvfrom(s, reinterpret_cast<RECV_TYPE_ARG2>(dst), len, flags, addr, alen);
 #else
     return sread(s, dst, len);
 #endif
   },
   [](ares_socket_t s, const struct iovec * vec, int len, void *) {
+#ifdef _WIN32
+    return ares_writev(s, vec, len);
+#else
     return :: writev(s, vec, len);
+#endif
   }
 };
 

test/ares-test.cc

@@ -19,11 +19,9 @@
 
 #ifdef WIN32
 #define BYTE_CAST (char *)
-#define sclose(x) closesocket(x)
 #define mkdir_(d, p) mkdir(d)
 #else
 #define BYTE_CAST
-#define sclose(x) close(x)
 #define mkdir_(d, p) mkdir(d, p)
 #endif
 

test/ares-test.h

@@ -2,13 +2,13 @@
 #ifndef ARES_TEST_H
 #define ARES_TEST_H
 
-#include "ares.h"
-
 #include "dns-proto.h"
-
 // Include ares internal file for DNS protocol constants
 #include "nameser.h"
 
+#include "ares_setup.h"
+#include "ares.h"
+
 #include "gtest/gtest.h"
 #include "gmock/gmock.h"
 

test/dns-proto.cc

@@ -1,6 +1,7 @@
 #include "dns-proto.h"
 
 // Include ares internal file for DNS protocol details
+#include "ares_setup.h"
 #include "ares.h"
 #include "ares_dns.h"