Merge lp:~tangent-org/libmemcached/1.0-build/ Build: jenkins-Libmemcached-187
[m6w6/libmemcached] / libtest / fatal.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 #include <libtest/common.h>
39 #include <cstdarg>
40
41 namespace libtest {
42
43 #pragma GCC diagnostic ignored "-Wformat-nonliteral"
44
45 fatal::fatal(const char *file_arg, int line_arg, const char *func_arg, ...) :
46 __test_result(file_arg, line_arg, func_arg),
47 _error_message(NULL),
48 _error_message_size(0)
49 {
50 va_list args;
51 va_start(args, func_arg);
52 const char *format= va_arg(args, const char *);
53 _error_message_size= vasprintf(&_error_message, format, args);
54 assert(_error_message_size != -1);
55 if (_error_message_size > 0)
56 {
57 _error_message_size++;
58 }
59 va_end(args);
60 }
61
62 fatal::fatal( const fatal& other ) :
63 __test_result(other),
64 _error_message_size(other._error_message_size)
65 {
66 _error_message= (char*) malloc(_error_message_size);
67 if (_error_message)
68 {
69 memcpy(_error_message, other._error_message, _error_message_size);
70 }
71 else
72 {
73 _error_message_size= -1;
74 }
75 }
76
77 fatal::~fatal() throw()
78 {
79 if ((_error_message_size > 0) and _error_message)
80 {
81 free(_error_message);
82 _error_message= NULL;
83 }
84 }
85
86 static bool _disabled= false;
87 static uint32_t _counter= 0;
88
89 bool fatal::is_disabled()
90 {
91 return _disabled;
92 }
93
94 void fatal::disable()
95 {
96 _counter= 0;
97 _disabled= true;
98 }
99
100 void fatal::enable()
101 {
102 _counter= 0;
103 _disabled= false;
104 }
105
106 uint32_t fatal::disabled_counter()
107 {
108 return _counter;
109 }
110
111 void fatal::increment_disabled_counter()
112 {
113 _counter++;
114 }
115
116 #pragma GCC diagnostic ignored "-Wformat-nonliteral"
117 disconnected::disconnected(const char *file_arg, int line_arg, const char *func_arg,
118 const std::string& instance, const in_port_t port, ...) :
119 std::runtime_error(func_arg),
120 _port(port),
121 _line(line_arg),
122 _file(file_arg),
123 _func(func_arg)
124 {
125 va_list args;
126 va_start(args, port);
127 const char *format= va_arg(args, const char *);
128 char last_error[BUFSIZ];
129 (void)vsnprintf(last_error, sizeof(last_error), format, args);
130 va_end(args);
131
132 snprintf(_error_message, sizeof(_error_message), "%s:%u %s", instance.c_str(), uint32_t(port), last_error);
133 }
134
135 } // namespace libtest