forgot to conditionally include malloc.h for alloca on watcom and msvc

[dosdemo] / src / polyfill.c

diff --git a/src/polyfill.c b/src/polyfill.c
index 122b234..80bfee5 100644 (file)
--- a/src/polyfill.c
+++ b/src/polyfill.c
@@ -2,7 +2,11 @@
 #include <stdlib.h>
 #include <string.h>
 #include <assert.h>
+#if defined(__WATCOMC__) || defined(_MSC_VER)
+#include <malloc.h>
+#else
 #include <alloca.h>
+#endif
 #include "polyfill.h"
 #include "gfxutil.h"
 #include "demo.h"