Synnefo » snf-ganeti

def KillProcess(pid, signal=signal.SIGTERM, timeout=30):

  """Kill a process given by its pid.

  @type pid: int

  @param pid: The PID to terminate.

  @type signal: int

  @param signal: The signal to send, by default SIGTERM

  @type timeout: int

  @param timeout: The timeout after which, if the process is still alive,

                  a SIGKILL will be sent. If not positive, no such checking

                  will be done

  """

  if pid <= 0:

    # kill with pid=0 == suicide

    raise errors.ProgrammerError("Invalid pid given '%s'" % pid)

  if not IsProcessAlive(pid):

    return

  os.kill(pid, signal)

  if timeout <= 0:

    return

  end = time.time() + timeout

  while time.time() < end and IsProcessAlive(pid):

    time.sleep(0.1)

  if IsProcessAlive(pid):

    os.kill(pid, signal.SIGKILL)

from ganeti.errors import LockError, UnitParseError, GenericError, \

     ProgrammerError

  def testKill(self):

    pid_file = self.f_dpn('child')

    r_fd, w_fd = os.pipe()

    new_pid = os.fork()

    if new_pid == 0: #child

      utils.WritePidFile('child')

      os.write(w_fd, 'a')

      signal.pause()

      os._exit(0)

      return

    # else we are in the parent

    # wait until the child has written the pid file

    os.read(r_fd, 1)

    read_pid = utils.ReadPidFile(pid_file)

    self.failUnlessEqual(read_pid, new_pid)

    self.failUnless(utils.IsProcessAlive(new_pid))

    utils.KillProcess(new_pid)

    self.failIf(utils.IsProcessAlive(new_pid))

    utils.RemovePidFile('child')

    self.failUnlessRaises(ProgrammerError, utils.KillProcess, 0)