[sanitizer] Intercept getgroups.

llvm-svn: 188167
This commit is contained in:
Evgeniy Stepanov 2013-08-12 11:01:40 +00:00
parent 173cf4077f
commit 98c39a184e
5 changed files with 28 additions and 1 deletions

View File

@ -2275,6 +2275,15 @@ TEST(MemorySanitizer, getgrnam_r) {
EXPECT_NOT_POISONED(grp.gr_gid);
}
TEST(MemorySanitizer, getgroups) {
int n = getgroups(0, 0);
gid_t *gids = new gid_t[n];
int res = getgroups(n, gids);
ASSERT_EQ(n, res);
for (int i = 0; i < n; ++i)
EXPECT_NOT_POISONED(gids[i]);
}
template<class T>
static bool applySlt(T value, T shadow) {
__msan_partial_poison(&value, &shadow, sizeof(T));

View File

@ -1923,6 +1923,20 @@ INTERCEPTOR(int, scandir64, char *dirp, __sanitizer_dirent64 ***namelist,
#define INIT_SCANDIR64
#endif
#if SANITIZER_INTERCEPT_GETGROUPS
INTERCEPTOR(int, getgroups, int size, u32 *lst) {
void *ctx;
COMMON_INTERCEPTOR_ENTER(ctx, getgroups, size, lst);
int res = REAL(getgroups)(size, lst);
if (res && lst)
COMMON_INTERCEPTOR_WRITE_RANGE(ctx, lst, res * sizeof(*lst));
return res;
}
#define INIT_GETGROUPS INTERCEPT_FUNCTION(getgroups);
#else
#define INIT_GETGROUPS
#endif
#define SANITIZER_COMMON_INTERCEPTORS_INIT \
INIT_STRCMP; \
INIT_STRNCMP; \
@ -1987,4 +2001,5 @@ INTERCEPTOR(int, scandir64, char *dirp, __sanitizer_dirent64 ***namelist,
INIT_STRERROR; \
INIT_STRERROR_R; \
INIT_SCANDIR; \
INIT_SCANDIR64;
INIT_SCANDIR64; \
INIT_GETGROUPS;

View File

@ -115,5 +115,6 @@
# define SANITIZER_INTERCEPT_STRERROR_R SI_NOT_WINDOWS
# define SANITIZER_INTERCEPT_SCANDIR SI_LINUX_NOT_ANDROID
# define SANITIZER_INTERCEPT_SCANDIR64 SI_LINUX_NOT_ANDROID
# define SANITIZER_INTERCEPT_GETGROUPS SI_NOT_WINDOWS
#endif // #ifndef SANITIZER_PLATFORM_INTERCEPTORS_H

View File

@ -358,6 +358,7 @@ void StatOutput(u64 *stat) {
name[StatInt_strerror_r] = " strerror_r ";
name[StatInt_scandir] = " scandir ";
name[StatInt_scandir64] = " scandir64 ";
name[StatInt_getgroups] = " getgroups ";
name[StatAnnotation] = "Dynamic annotations ";
name[StatAnnotateHappensBefore] = " HappensBefore ";

View File

@ -353,6 +353,7 @@ enum StatType {
StatInt_strerror_r,
StatInt_scandir,
StatInt_scandir64,
StatInt_getgroups,
// Dynamic annotations.
StatAnnotation,