llvm/test/tools/llvm-ml/macro_function.asm

   1 ; RUN: llvm-ml -filetype=s %s /Fo - | FileCheck %s
   2
   3 .code
   4
   5 identity MACRO arg
   6   exitm <arg>
   7 endm
   8
   9 argument_test PROC
  10 ; CHECK-LABEL: argument_test:
  11
  12   mov eax, identity(2)
  13 ; CHECK: mov eax, 2
  14
  15   ret
  16 argument_test ENDP
  17
  18 argument_with_parens_test PROC
  19 ; CHECK-LABEL: argument_with_parens_test:
  20
  21   mov eax, identity((3))
  22 ; CHECK: mov eax, 3
  23   mov eax, identity(((4-1)-1))
  24 ; CHECK: mov eax, 2
  25
  26   ret
  27 argument_with_parens_test ENDP
  28
  29 offsetof MACRO structure, field
  30   EXITM <structure.&field>
  31 ENDM
  32
  33 S1 STRUCT
  34   W byte 0
  35   X byte 0
  36   Y byte 0
  37 S1 ENDS
  38
  39 substitutions_test PROC
  40 ; CHECK-LABEL: substitutions_test:
  41
  42   mov eax, offsetof(S1, X)
  43 ; CHECK: mov eax, 1
  44   mov eax, offsetof(S1, Y)
  45 ; CHECK: mov eax, 2
  46
  47   ret
  48 substitutions_test ENDP
  49
  50 repeated_invocations_test PROC
  51 ; CHECK-LABEL: repeated_invocations_test:
  52
  53   mov eax, identity(identity(1))
  54 ; CHECK: mov eax, 1
  55
  56   ret
  57 repeated_invocations_test ENDP
  58
  59 factorial MACRO n
  60   IF n LE 1
  61     EXITM <(1)>
  62   ELSE
  63     EXITM <(n)*factorial(n-1)>
  64   ENDIF
  65 ENDM
  66
  67 ; NOTE: This version is more sensitive to unintentional end-of-statement tokens.
  68 factorial2 MACRO n
  69   IF n LE 1
  70     EXITM <(1)>
  71   ELSE
  72     EXITM <(n)*(factorial(n-1))>
  73   ENDIF
  74 ENDM
  75
  76 string_recursive_test PROC
  77 ; CHECK-LABEL: string_recursive_test:
  78
  79   mov eax, factorial(5)
  80 ; CHECK: mov eax, 120
  81   mov eax, factorial2(4)
  82 ; CHECK: mov eax, 24
  83   mov eax, 11 + factorial(6) - 11
  84 ; CHECK: mov eax, 720
  85
  86   ret
  87 string_recursive_test ENDP
  88
  89 fibonacci MACRO n
  90   IF n LE 2
  91     EXITM %1
  92   ELSE
  93     EXITM %fibonacci(n-1)+fibonacci(n-2)
  94   ENDIF
  95 ENDM
  96
  97 expr_recursive_test PROC
  98 ; CHECK-LABEL: expr_recursive_test:
  99
 100   mov eax, fibonacci(10)
 101 ; CHECK: mov eax, 55
 102
 103   ret
 104 expr_recursive_test ENDP
 105
 106 custom_strcat MACRO arg1, arg2
 107   EXITM <arg1&arg2>
 108 ENDM
 109
 110 expand_as_directive_test custom_strcat(P, ROC)
 111 ; CHECK-LABEL: expand_as_directive_test:
 112
 113   ret
 114 expand_as_directive_test ENDP
 115
 116 end