Merge lp:~tangent-org/libmemcached/1.0-build/ Build: jenkins-Libmemcached-192
[awesomized/libmemcached] / libtest / alarm.cc
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 #include "libtest/yatlcon.h"
38
39 #include <libtest/common.h>
40
41 #include <sys/time.h>
42 #include <cstdlib>
43
44 namespace libtest {
45
46 static const struct timeval default_it_value= { 600, 0 };
47 static const struct timeval default_it_interval= { 0, 0 };
48 static const struct itimerval defualt_timer= { default_it_interval, default_it_value };
49
50 static const struct itimerval cancel_timer= { default_it_interval, default_it_interval };
51
52
53 void set_alarm()
54 {
55 if (setitimer(ITIMER_VIRTUAL, &defualt_timer, NULL) == -1)
56 {
57 Error << "setitimer() failed";
58 }
59 }
60
61 void set_alarm(long tv_sec, long tv_usec)
62 {
63 // For the moment use any value to YATL_ALARM to cancel alarming.
64 if (getenv("YATL_ALARM"))
65 {
66 errno= 0;
67 tv_sec= strtol(getenv("YATL_ALARM"), (char **) NULL, 10);
68
69 if (errno != 0)
70 {
71 fatal_message("Bad value for YATL_ALARM");
72 }
73 else if (tv_sec == 0)
74 {
75 cancel_alarm();
76 }
77 }
78
79 #if defined(TARGET_OS_OSX) && TARGET_OS_OSX
80 struct timeval it_value= { time_t(tv_sec), suseconds_t(tv_usec) };
81 #else
82 struct timeval it_value= { tv_sec, tv_usec };
83 #endif
84
85 struct itimerval timer= { default_it_interval, it_value };
86
87 if (setitimer(ITIMER_VIRTUAL, &timer, NULL) == -1)
88 {
89 Error << "setitimer() failed";
90 }
91 }
92
93 void cancel_alarm()
94 {
95 if (setitimer(ITIMER_VIRTUAL, &cancel_timer, NULL) == -1)
96 {
97 Error << "setitimer() failed";
98 }
99 }
100
101 } // namespace libtest
102