[TSan] Fix dl_iterate_phdr callback for the case when info->dlpi_name is overwritten by user.

llvm-svn: 241876
This commit is contained in:
Alexey Samsonov 2015-07-10 00:03:54 +00:00
parent d880520bc2
commit f40c1076c9
2 changed files with 10 additions and 4 deletions

View File

@ -2153,6 +2153,10 @@ struct dl_iterate_phdr_data {
void *data; void *data;
}; };
static bool IsAppNotRodata(uptr addr) {
return IsAppMem(addr) && *(u64*)MemToShadow(addr) != kShadowRodata;
}
static int dl_iterate_phdr_cb(__sanitizer_dl_phdr_info *info, SIZE_T size, static int dl_iterate_phdr_cb(__sanitizer_dl_phdr_info *info, SIZE_T size,
void *data) { void *data) {
dl_iterate_phdr_data *cbdata = (dl_iterate_phdr_data *)data; dl_iterate_phdr_data *cbdata = (dl_iterate_phdr_data *)data;
@ -2161,13 +2165,13 @@ static int dl_iterate_phdr_cb(__sanitizer_dl_phdr_info *info, SIZE_T size,
// inside of dynamic linker, so we "unpoison" it here in order to not // inside of dynamic linker, so we "unpoison" it here in order to not
// produce false reports. Ignoring malloc/free in dlopen/dlclose is not enough // produce false reports. Ignoring malloc/free in dlopen/dlclose is not enough
// because some libc functions call __libc_dlopen. // because some libc functions call __libc_dlopen.
bool reset = info && IsAppMem((uptr)info->dlpi_name) && if (info && IsAppNotRodata((uptr)info->dlpi_name))
*(u64*)MemToShadow((uptr)info->dlpi_name) != kShadowRodata;
if (reset)
MemoryResetRange(cbdata->thr, cbdata->pc, (uptr)info->dlpi_name, MemoryResetRange(cbdata->thr, cbdata->pc, (uptr)info->dlpi_name,
internal_strlen(info->dlpi_name)); internal_strlen(info->dlpi_name));
int res = cbdata->cb(info, size, cbdata->data); int res = cbdata->cb(info, size, cbdata->data);
if (reset) // Perform the check one more time in case info->dlpi_name was overwritten
// by user callback.
if (info && IsAppNotRodata((uptr)info->dlpi_name))
MemoryResetRange(cbdata->thr, cbdata->pc, (uptr)info->dlpi_name, MemoryResetRange(cbdata->thr, cbdata->pc, (uptr)info->dlpi_name,
internal_strlen(info->dlpi_name)); internal_strlen(info->dlpi_name));
return res; return res;

View File

@ -18,6 +18,8 @@ int exported_var = 0;
#include <string> #include <string>
static int callback(struct dl_phdr_info *info, size_t size, void *data) { static int callback(struct dl_phdr_info *info, size_t size, void *data) {
if (info->dlpi_name[0] == '\0')
info->dlpi_name = "/proc/self/exe";
return !strcmp(info->dlpi_name, "non existent module"); return !strcmp(info->dlpi_name, "non existent module");
} }