Updating for 1.0.2 release
[awesomized/libmemcached] / libtest / signal.h
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * libtest
4 *
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 3 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22
23
24 #pragma once
25
26 #include <pthread.h>
27 #include <semaphore.h>
28 #include <signal.h>
29
30 enum shutdown_t {
31 SHUTDOWN_RUNNING,
32 SHUTDOWN_GRACEFUL,
33 SHUTDOWN_FORCED
34 };
35
36 namespace libtest {
37
38 class SignalThread {
39 sigset_t set;
40 sem_t lock;
41 uint64_t magic_memory;
42 volatile shutdown_t __shutdown;
43 pthread_mutex_t shutdown_mutex;
44 pthread_t thread;
45
46 public:
47
48 SignalThread();
49
50 void test();
51 void post();
52 bool setup();
53
54 int wait(int& sig)
55 {
56 return sigwait(&set, &sig);
57 }
58
59 ~SignalThread();
60
61 void set_shutdown(shutdown_t arg);
62 bool is_shutdown();
63 shutdown_t get_shutdown();
64 };
65
66 } // namespace libtest