Update hardening flags.
[m6w6/libmemcached] / libtest / server_container.h
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 #pragma once
38
39 #include <cassert>
40 #include <cstdio>
41 #include <cstring>
42 #include <netdb.h>
43 #include <netinet/in.h>
44 #include <string>
45 #include <unistd.h>
46 #include <vector>
47
48 namespace libtest {
49
50 class server_startup_st
51 {
52 private:
53 uint64_t _magic;
54 std::string server_list;
55 bool _socket;
56 bool _sasl;
57 uint32_t _count;
58 std::string _username;
59 std::string _password;
60
61 public:
62
63 uint8_t udp;
64 std::vector<Server *> servers;
65
66 server_startup_st();
67 ~server_startup_st();
68
69 bool validate();
70
71 bool start_socket_server(const std::string& server_type, const in_port_t try_port, int argc, const char *argv[], const bool opt_startup_message= true);
72 bool start_server(const std::string& server_type, const in_port_t try_port, int argc, const char *argv[], const bool opt_startup_message= true);
73
74 uint32_t count() const
75 {
76 return servers.size();
77 }
78
79 void restart();
80
81 std::string option_string() const;
82
83 const std::string& password() const
84 {
85 return _password;
86 }
87
88 const std::string& username() const
89 {
90 return _username;
91 }
92
93 bool socket()
94 {
95 return _socket;
96 }
97
98 bool sasl()
99 {
100 return _sasl;
101 }
102
103 void set_socket()
104 {
105 _socket= true;
106 }
107
108 void set_sasl(const std::string& username_arg, const std::string& password_arg)
109 {
110 _sasl= true;
111 _username= username_arg;
112 _password= password_arg;
113 }
114
115
116 // Just remove everything after shutdown
117 void clear();
118
119 bool shutdown();
120 bool shutdown(uint32_t number_of_host);
121
122 bool check() const;
123
124 void push_server(Server *);
125 Server* last();
126 Server *pop_server();
127
128 unsigned long int servers_to_run() const
129 {
130 return _servers_to_run;
131 }
132
133 void set_servers_to_run(unsigned long int arg)
134 {
135 _servers_to_run= arg;
136 }
137
138 private:
139 unsigned long int _servers_to_run;
140 };
141
142 bool server_startup(server_startup_st&, const std::string&, in_port_t try_port, int argc, const char *argv[], const bool opt_startup_message= true);
143
144 } // namespace libtest