elf/tst-audit6.c

   1 /* Test case for x86-64 preserved registers in dynamic linker.  */
   2
   3 #include <stdlib.h>
   4 #include <string.h>
   5 #include <cpuid.h>
   6 #include <emmintrin.h>
   7
   8 extern __m128i audit_test (__m128i, __m128i, __m128i, __m128i,
   9                            __m128i, __m128i, __m128i, __m128i);
  10
  11
  12 static int
  13 avx_enabled (void)
  14 {
  15   unsigned int eax, ebx, ecx, edx;
  16
  17   if (__get_cpuid (1, &eax, &ebx, &ecx, &edx) == 0
  18       || (ecx & (bit_AVX | bit_OSXSAVE)) != (bit_AVX | bit_OSXSAVE))
  19     return 0;
  20
  21   /* Check the OS has AVX and SSE saving enabled.  */
  22   asm ("xgetbv" : "=a" (eax), "=d" (edx) : "c" (0));
  23
  24   return (eax & 6) == 6;
  25 }
  26
  27
  28 int
  29 main (void)
  30 {
  31   /* Run AVX test only if AVX is supported.  */
  32   if (avx_enabled ())
  33     {
  34       __m128i xmm = _mm_setzero_si128 ();
  35       __m128i ret = audit_test (xmm, xmm, xmm, xmm, xmm, xmm, xmm, xmm);
  36
  37       xmm = _mm_set1_epi32 (0x98abcdef);
  38       if (memcmp (&xmm, &ret, sizeof (ret)))
  39         abort ();
  40     }
  41   return 0;
  42 }