commit 8ece42da7a8a131507f4228c70283f035f8d66df
author Vitaly Buka <vitalybuka@google.com> Mon Nov 08 19:22:24 2021 -0800
committer Copybara-Service <copybara-worker@google.com> Tue Nov 09 18:26:05 2021 -0800
tree 552edc1e3f5e4b03b0a48096567110c6871614d1
parent 939322905a0aba7e864483c31334fa0627a82308

[dfsan] Dfsan version of D113328

Depends on D113328.

Differential Revision: https://reviews.llvm.org/D113454

GitOrigin-RevId: ffd9c123e759e5811c1f643ab0ec21a23962a7af

lib/dfsan/dfsan_custom.cpp

diff --git a/lib/dfsan/dfsan_custom.cpp b/lib/dfsan/dfsan_custom.cpp
index 2ecbc07..217bd35 100644
--- a/lib/dfsan/dfsan_custom.cpp
+++ b/lib/dfsan/dfsan_custom.cpp
@@ -756,6 +756,7 @@
   DFsanThread *t = (DFsanThread *)arg;
   SetCurrentThread(t);
   t->Init();
+  SetSigProcMask(&t->starting_sigset_, nullptr);
   return t->ThreadStart();
 }
 
@@ -776,6 +777,7 @@
   DFsanThread *t =
       DFsanThread::Create(start_routine_trampoline,
                           (thread_callback_t)start_routine, arg, track_origins);
+  ScopedBlockSignals block(&t->starting_sigset_);
   int res = pthread_create(thread, attr, DFsanThreadStartFunc, t);
 
   if (attr == &myattr)

lib/dfsan/dfsan_thread.h

diff --git a/lib/dfsan/dfsan_thread.h b/lib/dfsan/dfsan_thread.h
index 6952963..1c33a18 100644
--- a/lib/dfsan/dfsan_thread.h
+++ b/lib/dfsan/dfsan_thread.h
@@ -15,6 +15,7 @@
 
 #include "dfsan_allocator.h"
 #include "sanitizer_common/sanitizer_common.h"
+#include "sanitizer_common/sanitizer_posix.h"
 
 namespace __dfsan {
 
@@ -45,6 +46,7 @@
   DFsanThreadLocalMallocStorage &malloc_storage() { return malloc_storage_; }
 
   int destructor_iterations_;
+  __sanitizer_sigset_t starting_sigset_;
 
  private:
   void SetThreadStackAndTls();