1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab: 2 * 3 * Data Differential YATL (i.e. libtest) library 4 * 5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/ 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions are 9 * met: 10 * 11 * * Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 14 * * Redistributions in binary form must reproduce the above 15 * copyright notice, this list of conditions and the following disclaimer 16 * in the documentation and/or other materials provided with the 17 * distribution. 18 * 19 * * The names of its contributors may not be used to endorse or 20 * promote products derived from this software without specific prior 21 * written permission. 22 * 23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 34 * 35 */ 36 37 #pragma once 38 39 #include <pthread.h> 40 #include <semaphore.h> 41 #include <signal.h> 42 43 enum shutdown_t { 44 SHUTDOWN_RUNNING, 45 SHUTDOWN_GRACEFUL, 46 SHUTDOWN_FORCED 47 }; 48 49 namespace libtest { 50 51 class SignalThread { 52 sigset_t set; 53 sem_t lock; 54 uint64_t magic_memory; 55 volatile shutdown_t __shutdown; 56 pthread_mutex_t shutdown_mutex; 57 pthread_t thread; 58 sigset_t original_set; 59 60 public: 61 62 SignalThread(); 63 ~SignalThread(); 64 65 void test(); 66 void post(); 67 bool setup(); 68 bool unblock(); 69 wait(int & sig)70 int wait(int& sig) 71 { 72 return sigwait(&set, &sig); 73 } 74 75 void set_shutdown(shutdown_t arg); 76 bool is_shutdown(); 77 shutdown_t get_shutdown(); 78 }; 79 80 } // namespace libtest 81