Merge in util update
[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
29 enum shutdown_t {
30 SHUTDOWN_RUNNING,
31 SHUTDOWN_GRACEFUL,
32 SHUTDOWN_FORCED
33 };
34
35 namespace libtest {
36
37 class SignalThread {
38 sigset_t set;
39 sem_t lock;
40 uint64_t magic_memory;
41 volatile shutdown_t __shutdown;
42 pthread_mutex_t shutdown_mutex;
43 pthread_t thread;
44
45 public:
46
47 SignalThread();
48
49 void test();
50 void post();
51 bool setup();
52
53 int wait(int& sig)
54 {
55 return sigwait(&set, &sig);
56 }
57
58 ~SignalThread();
59
60 void set_shutdown(shutdown_t arg);
61 bool is_shutdown();
62 shutdown_t get_shutdown();
63 };
64
65 } // namespace libtest