libtest: fix UB with va_start()
[awesomized/libmemcached] / libtest / exception / 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 "libtest/exception.hpp"
40 #include <cstdarg>
41
42 namespace libtest {
43
44 #pragma GCC diagnostic ignored "-Wformat-nonliteral"
45
46 fatal::fatal(const char *file_arg, int line_arg, const char *func_arg, ...) :
47 libtest::exception(file_arg, line_arg, func_arg)
48 {
49 va_list args;
50 va_start(args, func_arg);
51 init(args);
52 va_end(args);
53 }
54
55 fatal::fatal( const fatal& other ) :
56 libtest::exception(other)
57 {
58 }
59
60 static bool _disabled= false;
61 static uint32_t _counter= 0;
62
63 bool fatal::is_disabled() throw()
64 {
65 return _disabled;
66 }
67
68 void fatal::disable() throw()
69 {
70 _counter= 0;
71 _disabled= true;
72 }
73
74 void fatal::enable() throw()
75 {
76 _counter= 0;
77 _disabled= false;
78 }
79
80 uint32_t fatal::disabled_counter() throw()
81 {
82 return _counter;
83 }
84
85 void fatal::increment_disabled_counter() throw()
86 {
87 _counter++;
88 }
89
90 #pragma GCC diagnostic ignored "-Wformat-nonliteral"
91 disconnected::disconnected(const char *file_arg, int line_arg, const char *func_arg,
92 const std::string& instance, const unsigned port, ...) :
93 libtest::exception(file_arg, line_arg, func_arg),
94 _port(port)
95 {
96 va_list args;
97 va_start(args, port);
98 const char *format= va_arg(args, const char *);
99 char last_error[BUFSIZ];
100 (void)vsnprintf(last_error, sizeof(last_error), format, args);
101 va_end(args);
102
103 char buffer_error[BUFSIZ];
104 int error_length= snprintf(buffer_error, sizeof(buffer_error), "%s:%u %s", instance.c_str(), uint32_t(port), last_error);
105
106 if (error_length > 0)
107 {
108 what(size_t(error_length), buffer_error);
109 }
110 }
111
112 disconnected::disconnected(const disconnected& other):
113 libtest::exception(other),
114 _port(other._port)
115 {
116 strncpy(_instance, other._instance, BUFSIZ);
117 }
118
119 } // namespace libtest