X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libtest%2Fsignal.h;h=f54633c767e952872f261ead089d8fc15a1051fd;hb=6dab4c45effc85493d9e097d0df3446e202f3e8e;hp=5cc823acc325aa2577625c742d399fd93e25083d;hpb=ad94287c37250fcb1f7881705f0effb37bae7da5;p=awesomized%2Flibmemcached

diff --git a/libtest/signal.h b/libtest/signal.h
index 5cc823ac..f54633c7 100644
--- a/libtest/signal.h
+++ b/libtest/signal.h
@@ -25,6 +25,7 @@
 
 #include <pthread.h>
 #include <semaphore.h>
+#include <signal.h>
 
 enum shutdown_t {
   SHUTDOWN_RUNNING,
@@ -41,10 +42,12 @@ class SignalThread {
   volatile shutdown_t __shutdown;
   pthread_mutex_t shutdown_mutex;
   pthread_t thread;
+  sigset_t original_set;
 
 public:
 
   SignalThread();
+  ~SignalThread();
 
   void test();
   void post();
@@ -55,8 +58,6 @@ public:
     return sigwait(&set, &sig);
   }
 
-  ~SignalThread();
-
   void set_shutdown(shutdown_t arg);
   bool is_shutdown();
   shutdown_t get_shutdown();