[sanitizer] Add callbacks for epoll_pwait2

[vitalybuka]

Summary

Depends on D107207.

[morehouse]

compiler-rt/test/msan/Linux/syscalls.cpp

143

Should we leave the pwait test in addition to pwait2?

[vitalybuka]

restore test

compiler-rt/test/msan/Linux/syscalls.cpp

143

Thanks, I probably screwed it on rebase .

[yingcong-wu]

compiler-rt/lib/sanitizer_common/sanitizer_common_syscalls.inc

2135

Shouldn't this be PRE_READ(timeout, sizeof(sanitizer_kernel_timespec)) ? sizeof(timeout) is just a size of the pointer.

compiler-rt/include/sanitizer/linux_syscall_hooks.h

#define __sanitizer_syscall_pre_epoll_pwait(epfd, events, maxevents, timeout,  \
   __sanitizer_syscall_pre_impl_epoll_pwait(                                    \
       (long)(epfd), (long)(events), (long)(maxevents), (long)(timeout),        \
       (long)(sigmask), (long)(sigsetsize))
 #define __sanitizer_syscall_post_epoll_pwait(res, epfd, events, maxevents,     \
                                              timeout, sigmask, sigsetsize)     \
   __sanitizer_syscall_post_impl_epoll_pwait(                                   \
       res, (long)(epfd), (long)(events), (long)(maxevents), (long)(timeout),   \
       (long)(sigmask), (long)(sigsetsize))
+#define __sanitizer_syscall_pre_epoll_pwait2(epfd, events, maxevents, timeout, \
+                                             sigmask, sigsetsize)              \
+  __sanitizer_syscall_pre_impl_epoll_pwait2(                                   \
+      (long)(epfd), (long)(events), (long)(maxevents), (long)(timeout),        \
+      (long)(sigmask), (long)(sigsetsize))
+#define __sanitizer_syscall_post_epoll_pwait2(res, epfd, events, maxevents,    \
+                                              timeout, sigmask, sigsetsize)    \
+  __sanitizer_syscall_post_impl_epoll_pwait2(                                  \
+      res, (long)(epfd), (long)(events), (long)(maxevents), (long)(timeout),   \
+      (long)(sigmask), (long)(sigsetsize))
 #define __sanitizer_syscall_pre_gethostname(name, len)                         \
   __sanitizer_syscall_pre_impl_gethostname((long)(name), (long)(len))
 #define __sanitizer_syscall_post_gethostname(res, name, len)                   \
   __sanitizer_syscall_post_impl_gethostname(res, (long)(name), (long)(len))
 #define __sanitizer_syscall_pre_sethostname(name, len)                         \
   __sanitizer_syscall_pre_impl_sethostname((long)(name), (long)(len))
 #define __sanitizer_syscall_post_sethostname(res, name, len)                   \
   __sanitizer_syscall_post_impl_sethostname(res, (long)(name), (long)(len))

 void __sanitizer_syscall_post_impl_epoll_wait(long res, long epfd, long events,
                                               long maxevents, long timeout);
 void __sanitizer_syscall_pre_impl_epoll_pwait(long epfd, long events,
                                               long maxevents, long timeout,
                                               long sigmask, long sigsetsize);
 void __sanitizer_syscall_post_impl_epoll_pwait(long res, long epfd, long events,
                                                long maxevents, long timeout,
                                                long sigmask, long sigsetsize);
+void __sanitizer_syscall_pre_impl_epoll_pwait2(long epfd, long events,
+                                               long maxevents, long timeout,
+                                               long sigmask, long sigsetsize);
+void __sanitizer_syscall_post_impl_epoll_pwait2(long res, long epfd,
+                                                long events, long maxevents,
+                                                long timeout, long sigmask,
+                                                long sigsetsize);
 void __sanitizer_syscall_pre_impl_gethostname(long name, long len);
 void __sanitizer_syscall_post_impl_gethostname(long res, long name, long len);
 void __sanitizer_syscall_pre_impl_sethostname(long name, long len);
 void __sanitizer_syscall_post_impl_sethostname(long res, long name, long len);
 void __sanitizer_syscall_pre_impl_setdomainname(long name, long len);
 void __sanitizer_syscall_post_impl_setdomainname(long res, long name, long len);
 void __sanitizer_syscall_pre_impl_newuname(long name);
 void __sanitizer_syscall_post_impl_newuname(long res, long name);

compiler-rt/lib/sanitizer_common/sanitizer_common_syscalls.inc

 (long res, long epfd, void *events, long maxevents, long timeout,
  const void *sigmask, long sigsetsize) {
   if (res >= 0) {
     if (events)
       POST_WRITE(events, res * struct_epoll_event_sz);
   }
 }
 
+PRE_SYSCALL(epoll_pwait2)
+(long epfd, void *events, long maxevents,
+ const sanitizer_kernel_timespec *timeout, const kernel_sigset_t *sigmask,
+ long sigsetsize) {
+  if (timeout)
+    PRE_READ(timeout, sizeof(timeout));
+  if (sigmask)
+    PRE_READ(sigmask, sigsetsize);
+}
+
+POST_SYSCALL(epoll_pwait2)
+(long res, long epfd, void *events, long maxevents,
+ const sanitizer_kernel_timespec *timeout, const void *sigmask,
+ long sigsetsize) {
+  if (res >= 0) {
+    if (events)
+      POST_WRITE(events, res * struct_epoll_event_sz);
+  }
+}
+
 PRE_SYSCALL(gethostname)(void *name, long len) {}
 
 POST_SYSCALL(gethostname)(long res, void *name, long len) {
   if (res >= 0) {
     if (name)
       POST_WRITE(name, __sanitizer::internal_strlen((const char *)name) + 1);
   }
 }

compiler-rt/test/msan/Linux/syscalls.cpp

int main(int argc, char *argv[]) {
   __sanitizer_syscall_post_epoll_wait(max_events, 0, buf, max_events, 0);
   assert(__msan_test_shadow(buf, sizeof(buf)) == max_events * sizeof(epoll_event));
 
   __msan_poison(buf, sizeof(buf));
   sigset_t sigset = {};
   __sanitizer_syscall_pre_epoll_pwait(0, buf, max_events, 0, &sigset, sizeof(sigset));
   __sanitizer_syscall_post_epoll_pwait(max_events, 0, buf, max_events, 0, &sigset, sizeof(sigset));
   assert(__msan_test_shadow(buf, sizeof(buf)) == max_events * sizeof(epoll_event));
 
+  __msan_poison(buf, sizeof(buf));
+  sigset = {};
+  timespec timespec = {};
+  __sanitizer_syscall_pre_epoll_pwait2(0, buf, max_events, &timespec,
+                                       &sigset, sizeof(sigset));
+  __sanitizer_syscall_post_epoll_pwait2(max_events, 0, buf, max_events,
+                                        &timespec, &sigset, sizeof(sigset));
+  assert(__msan_test_shadow(buf, sizeof(buf)) == max_events * sizeof(epoll_event));
+
   return 0;
 }