ld64-409.12.tar.gz

[apple/ld64.git] / unit-tests / test-cases / dtrace-old-probes / main.c

diff --git a/unit-tests/test-cases/dtrace-old-probes/main.c b/unit-tests/test-cases/dtrace-old-probes/main.c
index 811449a764ca065cfdb4f91f3f40708ad8006f81..bd02587579c15534561688135f8fb01d78494eac 100644 (file)
--- a/unit-tests/test-cases/dtrace-old-probes/main.c
+++ b/unit-tests/test-cases/dtrace-old-probes/main.c
@@ -13,12 +13,21 @@
 #define DTRACE_LAB(p, n)               \
    "__dtrace_probe$" DTRACE_TOSTRING(%=__LINE__) DTRACE_STRINGIFY(_##p##___##n)
 
+#if (defined __x86_64__ || defined __arm64__)
+#define DTRACE_LABEL(p, n)              \
+      ".section __DATA, __data\n\t"     \
+      ".globl " DTRACE_LAB(p, n) "\n\t" \
+      DTRACE_LAB(p, n) ":" ".quad 1f""\n\t"    \
+       ".text" "\n\t"                   \
+        "1:"
+#else
 #define DTRACE_LABEL(p, n)             \
       ".section __DATA, __data\n\t"    \
       ".globl " DTRACE_LAB(p, n) "\n\t"        \
        DTRACE_LAB(p, n) ":\n\t" ".long 1f""\n\t"       \
        ".text" "\n\t"                  \
        "1:"
+#endif
 
 #define DTRACE_CALL(p,n)       \
        DTRACE_LABEL(p,n)       \