From 7b61129b2ed89d2cc3ca5560f55c26c6c347a215 Mon Sep 17 00:00:00 2001
From: Angelo Mantellini <manangel@cisco.com>
Date: Thu, 31 Jan 2019 10:36:54 +0100
Subject: [HICN-20] This source upgrade allows to compile ping_client,
 ping_server and hiperf (utils folder) in Windows.

Change-Id: I8253aa9aa640644b0daffd95dff202956371d814
Signed-off-by: Angelo Mantellini <manangel@cisco.com>
---
 utils/src/hiperf.cc | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

(limited to 'utils/src/hiperf.cc')

diff --git a/utils/src/hiperf.cc b/utils/src/hiperf.cc
index 221a8746b..65ee315f5 100644
--- a/utils/src/hiperf.cc
+++ b/utils/src/hiperf.cc
@@ -15,7 +15,9 @@
 
 #include <hicn/transport/interfaces/socket_consumer.h>
 #include <hicn/transport/interfaces/socket_producer.h>
+#ifndef _WIN32
 #include <hicn/transport/utils/daemonizator.h>
+#endif
 #include <hicn/transport/utils/literals.h>
 
 #include <fstream>
@@ -25,6 +27,10 @@
 #include <mcheck.h>
 #endif
 
+#ifdef _WIN32
+#include <hicn/transport/portability/win_portability.h>
+#endif
+
 namespace transport {
 
 namespace interface {
@@ -492,8 +498,10 @@ void usage() {
             << std::endl;
   std::cerr << "usage: hiperf [-S|-C] [options] [prefix|name]" << std::endl;
   std::cerr << "Server or Client:" << std::endl;
+#ifndef _WIN32
   std::cerr << "-D\t\t\t\t\t"
             << "Run as a daemon" << std::endl;
+#endif
   std::cerr << std::endl;
   std::cerr << "Server specific:" << std::endl;
   std::cerr << "-s\t<content_size>\t\t\tSize of the content to publish"
@@ -538,8 +546,14 @@ void usage() {
 }
 
 int main(int argc, char *argv[]) {
+
+#ifndef _WIN32
   // Common
   bool daemon = false;
+#else
+  WSADATA wsaData = { 0 };
+  WSAStartup(MAKEWORD(2, 2), &wsaData);
+#endif
 
   // -1 server, 0 undefined, 1 client
   int role = 0;
@@ -554,12 +568,17 @@ int main(int argc, char *argv[]) {
   ServerConfiguration server_configuration;
 
   int opt;
+#ifndef _WIN32
   while ((opt = getopt(argc, argv, "DSCf:b:d:W:c:vs:rmlk:y:p:hi:x")) != -1) {
     switch (opt) {
       // Common
       case 'D':
         daemon = true;
         break;
+#else
+  while ((opt = getopt(argc, argv, "SCf:b:d:W:c:vs:rmlk:y:p:hi:x")) != -1) {
+    switch (opt) {
+#endif
       case 'f':
         log_file = optarg;
         break;
@@ -685,15 +704,27 @@ int main(int argc, char *argv[]) {
   }
 
   if (log_file) {
+#ifndef _WIN32
     int fd = open(log_file, O_WRONLY | O_APPEND | O_CREAT, S_IWUSR | S_IRUSR);
     dup2(fd, STDOUT_FILENO);
     dup2(STDOUT_FILENO, STDERR_FILENO);
     close(fd);
+#else
+    int fd = _open(log_file, _O_WRONLY | _O_APPEND | _O_CREAT, _S_IWRITE | _S_IREAD);
+    _dup2(fd, STDOUT_FILENO);
+    _dup2(STDOUT_FILENO, STDERR_FILENO);
+    _close(fd);
+#endif
+    dup2(fd, STDOUT_FILENO);
+    dup2(STDOUT_FILENO, STDERR_FILENO);
+    close(fd);
   }
 
+#ifndef _WIN32
   if (daemon) {
     utils::Daemonizator::daemonize(false);
   }
+#endif
 
   if (role > 0) {
     HIperfClient c(client_configuration);
@@ -712,6 +743,9 @@ int main(int argc, char *argv[]) {
 
   std::cout << "Bye bye" << std::endl;
 
+#ifdef _WIN32
+  WSACleanup();
+#endif
   return 0;
 }
 
-- 
cgit 1.2.3-korg