1 files changed, 10 insertions, 2 deletions

diff --git a/update.sh b/update.sh
index 2e1f5c7..99a293a 100755
--- a/update.sh
+++ b/update.sh
@@ -167,7 +167,11 @@ gen_asm_stdout() {
         .section .note.GNU-stack,"",%progbits
         #endif
         EOF
-        $MV $3.tmp $3
+        if [ $1 = "masm" ]; then
+                cpp -I./crypto $3.tmp > $3
+        else
+                $MV $3.tmp $3
+        fi
 }
 gen_asm() {
         CC=true perl $asm_src/$2 $1 $3.tmp
@@ -176,7 +180,11 @@ gen_asm() {
         .section .note.GNU-stack,"",%progbits
         #endif
         EOF
-        $MV $3.tmp $3
+        if [ $1 = "masm" ]; then
+                cpp -I./crypto $3.tmp > $3
+        else
+                $MV $3.tmp $3
+        fi
 }
 
 echo generating arm ASM source for elf