Update hardening flags.
[m6w6/libmemcached] / libtest / blobslap_worker.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
38 #include <config.h>
39 #include <libtest/common.h>
40
41 #include <libtest/blobslap_worker.h>
42
43 #include <cassert>
44 #include <cerrno>
45 #include <cstdio>
46 #include <cstdlib>
47 #include <cstring>
48 #include <iostream>
49 #include <signal.h>
50 #include <sys/types.h>
51 #include <sys/wait.h>
52 #include <unistd.h>
53
54 #include <libgearman/gearman.h>
55
56 #ifndef __INTEL_COMPILER
57 #pragma GCC diagnostic ignored "-Wold-style-cast"
58 #endif
59
60 namespace libtest {
61
62 class BlobslapWorker : public Server
63 {
64 private:
65 public:
66 BlobslapWorker(in_port_t port_arg) :
67 Server("localhost", port_arg, "benchmark/blobslap_worker", true)
68 {
69 set_pid_file();
70 }
71
72 pid_t get_pid(bool error_is_ok)
73 {
74 if (pid_file().empty())
75 {
76 Error << "pid_file was empty";
77 return -1;
78 }
79
80 Wait wait(pid_file(), 0);
81
82 if (error_is_ok and not wait.successful())
83 {
84 Error << "Pidfile was not found:" << pid_file();
85 return -1;
86 }
87
88 std::stringstream error_message;
89 pid_t ret= get_pid_from_file(pid_file(), error_message);
90
91 if (error_is_ok and is_pid_valid(ret) == false)
92 {
93 Error << error_message.str();
94 }
95
96 return ret;
97 }
98
99 bool ping()
100 {
101 if (pid_file().empty())
102 {
103 Error << "No pid file available";
104 return false;
105 }
106
107 Wait wait(pid_file(), 0);
108 if (not wait.successful())
109 {
110 Error << "Pidfile was not found:" << pid_file();
111 return false;
112 }
113
114 std::stringstream error_message;
115 pid_t local_pid= get_pid_from_file(pid_file(), error_message);
116 if (is_pid_valid(local_pid) == false)
117 {
118 Error << error_message.str();
119 return false;
120 }
121
122 // Use kill to determine is the process exist
123 if (::kill(local_pid, 0) == 0)
124 {
125 return true;
126 }
127
128 return false;
129 }
130
131 const char *name()
132 {
133 return "blobslap_worker";
134 };
135
136 bool has_port_option() const
137 {
138 return true;
139 }
140
141 bool has_log_file_option() const
142 {
143 return true;
144 }
145
146 bool is_libtool()
147 {
148 return true;
149 }
150
151 bool build(size_t argc, const char *argv[]);
152 };
153
154
155 #include <sstream>
156
157 bool BlobslapWorker::build(size_t argc, const char *argv[])
158 {
159 std::stringstream arg_buffer;
160
161 for (size_t x= 0 ; x < argc ; x++)
162 {
163 add_option(argv[x]);
164 }
165
166 return true;
167 }
168
169 Server *build_blobslap_worker(in_port_t try_port)
170 {
171 return new BlobslapWorker(try_port);
172 }
173
174 } // namespace libtest