Merge pull request #3 from romanwue/windows_support

Add support for windows

Author: plusangel

CMakeLists.txt

@@ -3,14 +3,18 @@ project(ntp_client)
 
 set(CMAKE_CXX_STANDARD 14)
 
-file(GLOB_RECURSE SOURCES src/*.cpp)
+file(GLOB_RECURSE SOURCES src/*.cpp include/*.h*)
 include_directories(include)
 
 add_library(${PROJECT_NAME} STATIC ${SOURCES})
 
 add_executable(ntp_example main.cpp)
 target_link_libraries(ntp_example ${PROJECT_NAME})
 
+if (WIN32)
+    target_link_libraries(${PROJECT_NAME} ws2_32)
+endif()
+
 # Instalación
 INSTALL(TARGETS ${PROJECT_NAME}
         LIBRARY

include/ntp_client.hpp

@@ -2,7 +2,13 @@
 #define NTPCLIENT_H
 
 #include <string>
-#include <netinet/in.h>
+#ifdef _WIN32
+#   include <WinSock2.h>
+using Socket = SOCKET;
+#else
+#   include <netinet/in.h>
+using Socket = int;
+#endif
 
 struct NTPPacket
 {
@@ -89,7 +95,7 @@ struct NTPClient
     size_t port_;
 
     /// @brief Socket file descriptor
-    int socket_fd;
+    Socket socket_fd;
 
     /// @brief Server address data structure
     struct sockaddr_in socket_client;

src/ntp_client.cpp

@@ -5,15 +5,24 @@
 
 #include <stdio.h>
 #include <sys/types.h>
-#include <sys/socket.h>
-#include <arpa/inet.h>
-#include <netdb.h>
+#ifdef _WIN32
+#   include <WinSock2.h>
+#   define close(X) closesocket(X)
+#else
+#   include <sys/socket.h>
+#   include <arpa/inet.h>
+#   include <netdb.h>
+#   include <unistd.h>
+#endif
 #include <time.h>
 #include <string.h>
-#include <unistd.h>
 
 NTPClient::NTPClient(std::string hostname, uint16_t port) : hostname_(hostname), port_(port)
 {
+#ifdef _WIN32
+    WSADATA wsaData = { 0 };
+    (void)WSAStartup(MAKEWORD(2, 2), &wsaData);
+#endif;
 }
 
 void NTPClient::build_connection()
@@ -41,7 +50,14 @@ void NTPClient::build_connection()
     socket_client.sin_addr.s_addr = inet_addr(ntp_server_ip.c_str());
 }
 
-NTPClient::~NTPClient() { close(socket_fd); }
+NTPClient::~NTPClient()
+{
+    close_socket();
+
+#ifdef _WIN32
+    WSACleanup();
+#endif
+}
 
 uint64_t NTPClient::request_time()
 {
@@ -60,7 +76,11 @@ uint64_t NTPClient::request_time()
     packet.li_vn_mode = 0x1b;
 
     std::cout << "Sending request\n";
-    response = write(socket_fd, (char *)&packet, sizeof(NTPPacket));
+#ifdef _WIN32
+    response = sendto(socket_fd, (char*)&packet, sizeof(NTPPacket), 0, (struct sockaddr*)&socket_client, sizeof(socket_client));
+#else
+    response = write(socket_fd, (char*)&packet, sizeof(NTPPacket));
+#endif
 
     if (response < 0)
     {
@@ -69,7 +89,11 @@ uint64_t NTPClient::request_time()
     }
 
     std::cout << "Reading request\n";
+#ifdef _WIN32
+    response = recv(socket_fd, (char*)&packet, sizeof(NTPPacket), 0);
+#else
     response = read(socket_fd, (char *)&packet, sizeof(NTPPacket));
+#endif
 
     if (response < 0)
     {