Permalink
Cannot retrieve contributors at this time
Name already in use
A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
glibc/elf/tst-auditmod4b.c
Go to fileThis commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
201 lines (170 sloc)
4.95 KB
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
/* Verify that changing AVX registers in audit library won't affect | |
function parameter passing/return. */ | |
#include <dlfcn.h> | |
#include <stdint.h> | |
#include <stdio.h> | |
#include <stdlib.h> | |
#include <string.h> | |
#include <unistd.h> | |
#include <bits/wordsize.h> | |
#include <gnu/lib-names.h> | |
unsigned int | |
la_version (unsigned int v) | |
{ | |
setlinebuf (stdout); | |
printf ("version: %u\n", v); | |
char buf[20]; | |
sprintf (buf, "%u", v); | |
return v; | |
} | |
void | |
la_activity (uintptr_t *cookie, unsigned int flag) | |
{ | |
if (flag == LA_ACT_CONSISTENT) | |
printf ("activity: consistent\n"); | |
else if (flag == LA_ACT_ADD) | |
printf ("activity: add\n"); | |
else if (flag == LA_ACT_DELETE) | |
printf ("activity: delete\n"); | |
else | |
printf ("activity: unknown activity %u\n", flag); | |
} | |
char * | |
la_objsearch (const char *name, uintptr_t *cookie, unsigned int flag) | |
{ | |
char buf[100]; | |
const char *flagstr; | |
if (flag == LA_SER_ORIG) | |
flagstr = "LA_SET_ORIG"; | |
else if (flag == LA_SER_LIBPATH) | |
flagstr = "LA_SER_LIBPATH"; | |
else if (flag == LA_SER_RUNPATH) | |
flagstr = "LA_SER_RUNPATH"; | |
else if (flag == LA_SER_CONFIG) | |
flagstr = "LA_SER_CONFIG"; | |
else if (flag == LA_SER_DEFAULT) | |
flagstr = "LA_SER_DEFAULT"; | |
else if (flag == LA_SER_SECURE) | |
flagstr = "LA_SER_SECURE"; | |
else | |
{ | |
sprintf (buf, "unknown flag %d", flag); | |
flagstr = buf; | |
} | |
printf ("objsearch: %s, %s\n", name, flagstr); | |
return (char *) name; | |
} | |
unsigned int | |
la_objopen (struct link_map *l, Lmid_t lmid, uintptr_t *cookie) | |
{ | |
printf ("objopen: %ld, %s\n", lmid, l->l_name); | |
return 3; | |
} | |
void | |
la_preinit (uintptr_t *cookie) | |
{ | |
printf ("preinit\n"); | |
} | |
unsigned int | |
la_objclose (uintptr_t *cookie) | |
{ | |
printf ("objclose\n"); | |
return 0; | |
} | |
uintptr_t | |
la_symbind64 (Elf64_Sym *sym, unsigned int ndx, uintptr_t *refcook, | |
uintptr_t *defcook, unsigned int *flags, const char *symname) | |
{ | |
printf ("symbind64: symname=%s, st_value=%#lx, ndx=%u, flags=%u\n", | |
symname, (long int) sym->st_value, ndx, *flags); | |
return sym->st_value; | |
} | |
#include <tst-audit.h> | |
#ifdef __AVX__ | |
#include <immintrin.h> | |
#include <cpuid.h> | |
static int avx = -1; | |
static inline int | |
__attribute ((always_inline)) | |
check_avx (void) | |
{ | |
if (avx == -1) | |
{ | |
unsigned int eax, ebx, ecx, edx; | |
if (__get_cpuid (1, &eax, &ebx, &ecx, &edx) | |
&& (ecx & bit_AVX)) | |
avx = 1; | |
else | |
avx = 0; | |
} | |
return avx; | |
} | |
#else | |
#include <emmintrin.h> | |
#endif | |
ElfW(Addr) | |
pltenter (ElfW(Sym) *sym, unsigned int ndx, uintptr_t *refcook, | |
uintptr_t *defcook, La_regs *regs, unsigned int *flags, | |
const char *symname, long int *framesizep) | |
{ | |
printf ("pltenter: symname=%s, st_value=%#lx, ndx=%u, flags=%u\n", | |
symname, (long int) sym->st_value, ndx, *flags); | |
#ifdef __AVX__ | |
if (check_avx () && strcmp (symname, "audit_test") == 0) | |
{ | |
__m256i zero = _mm256_setzero_si256 (); | |
if (memcmp (®s->lr_vector[0], &zero, sizeof (zero)) | |
|| memcmp (®s->lr_vector[1], &zero, sizeof (zero)) | |
|| memcmp (®s->lr_vector[2], &zero, sizeof (zero)) | |
|| memcmp (®s->lr_vector[3], &zero, sizeof (zero)) | |
|| memcmp (®s->lr_vector[4], &zero, sizeof (zero)) | |
|| memcmp (®s->lr_vector[5], &zero, sizeof (zero)) | |
|| memcmp (®s->lr_vector[6], &zero, sizeof (zero)) | |
|| memcmp (®s->lr_vector[7], &zero, sizeof (zero))) | |
abort (); | |
for (int i = 0; i < 8; i++) | |
regs->lr_vector[i].ymm[0] | |
= (La_x86_64_ymm) _mm256_set1_epi32 (i + 1); | |
__m256i ymm = _mm256_set1_epi32 (-1); | |
asm volatile ("vmovdqa %0, %%ymm0" : : "x" (ymm) : "xmm0" ); | |
asm volatile ("vmovdqa %0, %%ymm1" : : "x" (ymm) : "xmm1" ); | |
asm volatile ("vmovdqa %0, %%ymm2" : : "x" (ymm) : "xmm2" ); | |
asm volatile ("vmovdqa %0, %%ymm3" : : "x" (ymm) : "xmm3" ); | |
asm volatile ("vmovdqa %0, %%ymm4" : : "x" (ymm) : "xmm4" ); | |
asm volatile ("vmovdqa %0, %%ymm5" : : "x" (ymm) : "xmm5" ); | |
asm volatile ("vmovdqa %0, %%ymm6" : : "x" (ymm) : "xmm6" ); | |
asm volatile ("vmovdqa %0, %%ymm7" : : "x" (ymm) : "xmm7" ); | |
*framesizep = 1024; | |
} | |
#endif | |
return sym->st_value; | |
} | |
unsigned int | |
pltexit (ElfW(Sym) *sym, unsigned int ndx, uintptr_t *refcook, | |
uintptr_t *defcook, const La_regs *inregs, La_retval *outregs, | |
const char *symname) | |
{ | |
printf ("pltexit: symname=%s, st_value=%#lx, ndx=%u, retval=%tu\n", | |
symname, (long int) sym->st_value, ndx, | |
(ptrdiff_t) outregs->int_retval); | |
#ifdef __AVX__ | |
if (check_avx () && strcmp (symname, "audit_test") == 0) | |
{ | |
__m256i zero = _mm256_setzero_si256 (); | |
if (memcmp (&outregs->lrv_vector0, &zero, sizeof (zero))) | |
abort (); | |
for (int i = 0; i < 8; i++) | |
{ | |
__m256i ymm = _mm256_set1_epi32 (i + 1); | |
if (memcmp (&inregs->lr_vector[i], &ymm, sizeof (ymm)) != 0) | |
abort (); | |
} | |
outregs->lrv_vector0.ymm[0] | |
= (La_x86_64_ymm) _mm256_set1_epi32 (0x12349876); | |
__m256i ymm = _mm256_set1_epi32 (-1); | |
asm volatile ("vmovdqa %0, %%ymm0" : : "x" (ymm) : "xmm0" ); | |
asm volatile ("vmovdqa %0, %%ymm1" : : "x" (ymm) : "xmm1" ); | |
} | |
#endif | |
return 0; | |
} |