Open stdin as binary on Windows

alesimula · Oct 29, 2021 · f250daf · f250daf
1 parent dc36ab4
commit f250daf
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/main.cpp b/main.cpp
@@ -26,6 +26,12 @@
 #include <boost/property_tree/xml_parser.hpp>
 #include <boost/program_options.hpp>
 
+#if defined(WIN32) || defined(_WIN32) || defined(__WIN32) && !defined(__CYGWIN__)
+#define PLATFORM_WINDOWS
+#include <io.h>
+#include <fcntl.h>
+#endif
+
 #include <unzip.h>
 
 namespace boost_pt = boost::property_tree;
@@ -125,6 +131,12 @@ void process_file(const std::string& input_filename,
    if (!input_filename.empty()) {
        input_file_stream.open(input_filename, std::ios::binary);
    }
+   else {
+       //TODO Linux?
+       #ifdef PLATFORM_WINDOWS
+       _setmode(_fileno(stdin), _O_BINARY);
+       #endif
+   }
    std::istream& ifs = !input_filename.empty()
            ? input_file_stream
            : std::cin;