git.m6w6.name Git - awesomized/libmemcached/blob - libtest/server.cc

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  Libmemcached library
   4  *
   5  *  Copyright (C) 2011 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 <config.h>
  38 #include <iostream>
  39 #include <cstdlib>
  40 #include <cassert>
  41
  42 #include <libtest/server.h>
  43 #include <libtest/killpid.h>
  44
  45
  46 std::ostream& operator<<(std::ostream& output, const server_st &arg)
  47 {
  48   if (arg.is_socket())
  49   {
  50     output << arg.hostname();
  51   }
  52   else
  53   {
  54     output << arg.hostname() << ":" << arg.port();
  55   }
  56   return output;  // for multiple << operators
  57 }
  58
  59 static void global_sleep(void)
  60 {
  61   static struct timespec global_sleep_value= { 0, 50000 };
  62
  63 #ifdef WIN32
  64   sleep(1);
  65 #else
  66   nanosleep(&global_sleep_value, NULL);
  67 #endif
  68 }
  69
  70 server_st::server_st(in_port_t port_arg, test_server_getpid *get_pid_arg, test_server_ping *ping_arg) :
  71   _used(false),
  72   _pid(-1),
  73   _port(port_arg),
  74   __get_pid(get_pid_arg),
  75   __ping(ping_arg),
  76   _hostname("localhost")
  77 {
  78   pid_file[0]= 0;
  79 }
  80
  81 server_st::server_st(const std::string &socket_file, test_server_getpid *get_pid_arg, test_server_ping *ping_arg) :
  82   _used(false),
  83   _pid(-1),
  84   _port(0),
  85   __get_pid(get_pid_arg),
  86   __ping(ping_arg),
  87   _hostname(socket_file)
  88 {
  89   pid_file[0]= 0;
  90 }
  91
  92
  93 server_st::~server_st()
  94 {
  95   if (has_pid())
  96   {
  97     kill();
  98   }
  99 }
 100
 101 bool server_st::start()
 102 {
 103   assert(not _command.empty());
 104   assert(not has_pid());
 105
 106   if (has_pid())
 107     return false;
 108
 109   if (system(_command.c_str()) == -1)
 110     return false;
 111
 112   int count= 30;
 113   while (not ping() and --count)
 114   {
 115     global_sleep();
 116   }
 117
 118   if (count == 0)
 119   {
 120     return false;
 121   }
 122
 123   _pid= get_pid();
 124
 125   return has_pid();
 126 }
 127
 128 void server_st::reset_pid()
 129 {
 130   pid_file[0]= 0;
 131   _pid= -1;
 132 }
 133
 134 pid_t server_st::pid()
 135 {
 136   if (not has_pid())
 137   {
 138     _pid= get_pid();
 139   }
 140
 141   return _pid;
 142 }
 143
 144
 145 bool server_st::kill()
 146 {
 147   if (is_used())
 148     return false;
 149
 150   if ((_pid= get_pid()))
 151   {
 152     kill_pid(_pid);
 153     if (pid_file[0])
 154     {
 155       unlink(pid_file); // If this happens we may be dealing with a dead server that left its pid file.
 156     }
 157     reset_pid();
 158
 159     return true;
 160   }
 161 #if 0
 162   else if (pid_file[0])
 163   {
 164     kill_file(pid_file);
 165     reset_pid();
 166
 167     return true;
 168   }
 169 #endif
 170
 171   return false;
 172 }
 173
 174 void server_startup_st::push_server(server_st *arg)
 175 {
 176   servers.push_back(arg);
 177 }
 178
 179 void server_startup_st::shutdown()
 180 {
 181   for (std::vector<server_st *>::iterator iter= servers.begin(); iter != servers.end(); iter++)
 182   {
 183     if ((*iter)->is_used())
 184       continue;
 185
 186     (*iter)->kill();
 187   }
 188 }
 189
 190 server_startup_st::~server_startup_st()
 191 {
 192   for (std::vector<server_st *>::iterator iter= servers.begin(); iter != servers.end(); iter++)
 193   {
 194     delete *iter;
 195   }
 196   servers.clear();
 197 }
 198
 199 void server_shutdown(server_startup_st *construct)
 200 {
 201   if (not construct)
 202     return;
 203
 204   construct->shutdown();
 205 }