restore sigprocmask in setjmp/longjmp test.
authorYabin Cui <yabinc@google.com>
Tue, 2 Dec 2014 17:57:45 +0000 (09:57 -0800)
committerYabin Cui <yabinc@google.com>
Tue, 2 Dec 2014 18:35:25 +0000 (10:35 -0800)
Bug: 18571921
Change-Id: Ib8ca7a36abd3ccc729c20c0e32d45b1b33069f65

tests/setjmp_test.cpp
tests/stack_unwinding_test.cpp

index 065c783312af2e04e471cd47df8e85201b55882a..2df0135d99c0c30f89430222d0144dfa2c0c32ca 100644 (file)
@@ -74,7 +74,8 @@ TEST(setjmp, _setjmp_signal_mask) {
   // _setjmp/_longjmp do not save/restore the signal mask.
   sigset_t ss1(SigSetOf(SIGUSR1));
   sigset_t ss2(SigSetOf(SIGUSR2));
-  sigprocmask(SIG_SETMASK, &ss1, NULL);
+  sigset_t original_set;
+  sigprocmask(SIG_SETMASK, &ss1, &original_set);
   jmp_buf jb;
   if (_setjmp(jb) == 0) {
     sigprocmask(SIG_SETMASK, &ss2, NULL);
@@ -85,6 +86,7 @@ TEST(setjmp, _setjmp_signal_mask) {
     sigprocmask(SIG_SETMASK, NULL, &ss);
     EXPECT_TRUE(sigismember(&ss, SIGUSR2));
   }
+  sigprocmask(SIG_SETMASK, &original_set, NULL);
 }
 
 TEST(setjmp, setjmp_signal_mask) {
@@ -93,7 +95,8 @@ TEST(setjmp, setjmp_signal_mask) {
   // behavior unspecified, so any code that cares needs to use sigsetjmp.
   sigset_t ss1(SigSetOf(SIGUSR1));
   sigset_t ss2(SigSetOf(SIGUSR2));
-  sigprocmask(SIG_SETMASK, &ss1, NULL);
+  sigset_t original_set;
+  sigprocmask(SIG_SETMASK, &ss1, &original_set);
   jmp_buf jb;
   if (setjmp(jb) == 0) {
     sigprocmask(SIG_SETMASK, &ss2, NULL);
@@ -110,13 +113,15 @@ TEST(setjmp, setjmp_signal_mask) {
     EXPECT_TRUE(sigismember(&ss, SIGUSR2));
 #endif
   }
+  sigprocmask(SIG_SETMASK, &original_set, NULL);
 }
 
 TEST(setjmp, sigsetjmp_0_signal_mask) {
   // sigsetjmp(0)/siglongjmp do not save/restore the signal mask.
   sigset_t ss1(SigSetOf(SIGUSR1));
   sigset_t ss2(SigSetOf(SIGUSR2));
-  sigprocmask(SIG_SETMASK, &ss1, NULL);
+  sigset_t original_set;
+  sigprocmask(SIG_SETMASK, &ss1, &original_set);
   sigjmp_buf sjb;
   if (sigsetjmp(sjb, 0) == 0) {
     sigprocmask(SIG_SETMASK, &ss2, NULL);
@@ -127,13 +132,15 @@ TEST(setjmp, sigsetjmp_0_signal_mask) {
     sigprocmask(SIG_SETMASK, NULL, &ss);
     EXPECT_TRUE(sigismember(&ss, SIGUSR2));
   }
+  sigprocmask(SIG_SETMASK, &original_set, NULL);
 }
 
 TEST(setjmp, sigsetjmp_1_signal_mask) {
   // sigsetjmp(1)/siglongjmp does save/restore the signal mask.
   sigset_t ss1(SigSetOf(SIGUSR1));
   sigset_t ss2(SigSetOf(SIGUSR2));
-  sigprocmask(SIG_SETMASK, &ss1, NULL);
+  sigset_t original_set;
+  sigprocmask(SIG_SETMASK, &ss1, &original_set);
   sigjmp_buf sjb;
   if (sigsetjmp(sjb, 1) == 0) {
     sigprocmask(SIG_SETMASK, &ss2, NULL);
@@ -144,4 +151,5 @@ TEST(setjmp, sigsetjmp_1_signal_mask) {
     sigprocmask(SIG_SETMASK, NULL, &ss);
     EXPECT_TRUE(sigismember(&ss, SIGUSR1));
   }
+  sigprocmask(SIG_SETMASK, &original_set, NULL);
 }
index 3fc45c537a2459a5a845235d0cf9d8578711999e..3d3f22df50718a3f73c357400995b5cc529d3e33 100644 (file)
@@ -73,6 +73,7 @@ TEST(stack_unwinding, easy) {
   ASSERT_EQ(count + 1, deeper_count);
 }
 
+static volatile bool signal_handler_run = false;
 static int killer_count = 0;
 static int handler_count = 0;
 static int handler_one_deeper_count = 0;
@@ -83,6 +84,7 @@ static void noinline UnwindSignalHandler(int) {
 
   handler_one_deeper_count = unwind_one_frame_deeper();
   ASSERT_EQ(handler_count + 1, handler_one_deeper_count);
+  signal_handler_run = true;
 }
 
 TEST(stack_unwinding, unwind_through_signal_frame) {
@@ -90,8 +92,9 @@ TEST(stack_unwinding, unwind_through_signal_frame) {
   ScopedSignalHandler ssh(SIGUSR1, UnwindSignalHandler);
 
   _Unwind_Backtrace(FrameCounter, &killer_count);
-
+  signal_handler_run = false;
   ASSERT_EQ(0, kill(getpid(), SIGUSR1));
+  while (!signal_handler_run) {}
 }
 
 // On LP32, the SA_SIGINFO flag gets you __restore_rt instead of __restore.
@@ -100,6 +103,7 @@ TEST(stack_unwinding, unwind_through_signal_frame_SA_SIGINFO) {
   ScopedSignalHandler ssh(SIGUSR1, UnwindSignalHandler, SA_SIGINFO);
 
   _Unwind_Backtrace(FrameCounter, &killer_count);
-
+  signal_handler_run = false;
   ASSERT_EQ(0, kill(getpid(), SIGUSR1));
+  while (!signal_handler_run) {}
 }