Updated DHT_test with WSA stuff.

1 files changed, 12 insertions, 0 deletions

diff --git a/testing/DHT_test.c b/testing/DHT_test.c
index a7c61e7f..29c25e0a 100644
--- a/testing/DHT_test.c
+++ b/testing/DHT_test.c
@@ -26,6 +26,14 @@
 
 int main(int argc, char *argv[])
 {
+    #ifdef WIN32
+    WSADATA wsaData;
+    if(WSAStartup(MAKEWORD(2,2), &wsaData) != NO_ERROR)
+    {
+        return -1;
+    }
+    #endif
+    
     if (argc < 3) {
         printf("usage %s ip port\n", argv[0]);
         exit(0);
@@ -82,5 +90,9 @@ int main(int argc, char *argv[])
         }
         c_sleep(100);
     }
+    
+    #ifdef WIN32
+    WSACleanup();
+    #endif
     return 0;   
 }
\ No newline at end of file