fix dos build

[retroray] / libs / treestor / src / treestor.c

diff --git a/libs/treestor/src/treestor.c b/libs/treestor/src/treestor.c
index 115c2b4..67fdcf6 100644 (file)
--- a/libs/treestor/src/treestor.c
+++ b/libs/treestor/src/treestor.c
@@ -5,13 +5,13 @@
 #include <assert.h>
 #include "treestor.h"
 
-#ifdef WIN32
+#if defined(__WATCOMC__) || defined(_WIN32) || defined(__DJGPP__)
 #include <malloc.h>
 #else
 #include <alloca.h>
 #endif
 
-#ifdef _MSC_VER
+#if defined(_MSC_VER) || defined(__WATCOMC__)
 #define snprintf _snprintf
 #endif