From 5aa2d52cee15c153e43e0bb47455df732750bf12 Mon Sep 17 00:00:00 2001
From: Pavel Odintsov <pavel.odintsov@gmail.com>
Date: Sun, 2 Apr 2023 16:20:03 +0100
Subject: [PATCH] Made sflow plugin cross platform. Addresses include
 duplication

---
 src/fastnetmon_types.hpp             | 5 +++++
 src/sflow_plugin/sflow_collector.cpp | 8 +++++---
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/fastnetmon_types.hpp b/src/fastnetmon_types.hpp
index 899f369..085df7d 100644
--- a/src/fastnetmon_types.hpp
+++ b/src/fastnetmon_types.hpp
@@ -1,7 +1,12 @@
 #ifndef FASTNETMON_TYPES_H
 #define FASTNETMON_TYPES_H
 
+#ifdef _WIN32
+#include <winsock2.h>
+#else
 #include <netinet/in.h> // struct in6_addr
+#endif
+
 #include <stdint.h> // uint32_t
 #include <sys/time.h> // struct timeval
 
diff --git a/src/sflow_plugin/sflow_collector.cpp b/src/sflow_plugin/sflow_collector.cpp
index 69e8785..c50e2f3 100644
--- a/src/sflow_plugin/sflow_collector.cpp
+++ b/src/sflow_plugin/sflow_collector.cpp
@@ -8,16 +8,18 @@
 #include "../libsflow/libsflow.hpp"
 #include "sflow_collector.hpp"
 
+#ifdef _WIN32
+#include <winsock2.h>
+#else
 #include <arpa/inet.h>
 #include <netinet/in.h>
+#endif
+
 #include <sys/socket.h>
 
-// UDP server
-#include <netinet/in.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <sys/socket.h>
 
 #include "../fast_library.hpp"
 #include "../fastnetmon_plugin.hpp"