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  *  Copyright (C) 2006-2009 Brian Aker All rights reserved.
   7  *
   8  *  Redistribution and use in source and binary forms, with or without
   9  *  modification, are permitted provided that the following conditions are
  10  *  met:
  11  *
  12  *      * Redistributions of source code must retain the above copyright
  13  *  notice, this list of conditions and the following disclaimer.
  14  *
  15  *      * Redistributions in binary form must reproduce the above
  16  *  copyright notice, this list of conditions and the following disclaimer
  17  *  in the documentation and/or other materials provided with the
  18  *  distribution.
  19  *
  20  *      * The names of its contributors may not be used to endorse or
  21  *  promote products derived from this software without specific prior
  22  *  written permission.
  23  *
  24  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  25  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  26  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  27  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  28  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  30  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  31  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  32  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  34  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35  *
  36  */
  37
  38 #include <config.h>
  39 #include <iostream>
  40
  41 #include <libtest/server.h>
  42 #include <libtest/killpid.h>
  43
  44
  45 std::ostream& operator<<(std::ostream& output, const server_st &arg)
  46 {
  47   if (arg.is_socket())
  48   {
  49     output << arg.hostname;
  50   }
  51   else
  52   {
  53     output << arg.hostname << ":" << arg.port();
  54   }
  55   return output;  // for multiple << operators
  56 }
  57
  58 server_st::~server_st()
  59 {
  60   if (not _used)
  61   {
  62     kill();
  63   }
  64 }
  65
  66 void server_st::reset_pid()
  67 {
  68   pid_file[0]= 0;
  69   _pid= -1;
  70 }
  71
  72 bool server_st::kill()
  73 {
  74   if (not has_pid() and pid_file[0] == 0)
  75   {
  76     return true;
  77   }
  78
  79   if (has_pid())
  80   {
  81     kill_pid(pid());
  82     if (pid_file[0])
  83     {
  84       unlink(pid_file); // If this happens we may be dealing with a dead server that left its pid file.
  85     }
  86     reset_pid();
  87
  88     return true;
  89   }
  90   else if (pid_file[0])
  91   {
  92     kill_file(pid_file);
  93     reset_pid();
  94
  95     return true;
  96   }
  97
  98   return false;
  99 }