From a732616ef83edb02065d4324f7e11abdf7be493d Mon Sep 17 00:00:00 2001
From: Taylor Barnes <taylor.a.barnes@gmail.com>
Date: Thu, 27 Jun 2024 13:22:18 +0000
Subject: [PATCH] Add Windows support for poll function

---
 MDI_Library/mdi_tcp.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/MDI_Library/mdi_tcp.c b/MDI_Library/mdi_tcp.c
index a6a5ab1..60f718c 100644
--- a/MDI_Library/mdi_tcp.c
+++ b/MDI_Library/mdi_tcp.c
@@ -10,13 +10,13 @@
   #include <sys/socket.h>
   #include <netdb.h>
   #include <unistd.h>
+  #include <poll.h>
 #endif
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <signal.h>
 #include <errno.h>
-#include <poll.h>
 #include "mdi.h"
 #include "mdi_tcp.h"
 #include "mdi_global.h"
@@ -474,7 +474,11 @@ int tcp_check_for_connection(int* flag) {
 
   if ( this_code->intra_rank == 0 ) { // Running on rank 0
 
+#ifdef _WIN32
+    ret = WSAPoll(&poll_args, 1, 0);
+#else
     ret = poll(&poll_args, 1, 0);
+#endif
     if (ret > 0) {
       if ( poll_args.revents & POLLIN ) {
         *flag = 1;