utils.py

来自「xen虚拟机源代码安装包」· Python 代码 · 共 51 行

import traceback
import sys
import os

def exception_string(e):
        (ty,v,tb) = sys.exc_info()
        return traceback.format_exception_only(ty,v)

def daemonize(prog, args, stdin_tmpfile=None):
    """Runs a program as a daemon with the list of arguments.  Returns the PID
    of the daemonized program, or returns 0 on error.
    """
    r, w = os.pipe()
    pid = os.fork()

    if pid == 0:
        os.close(r)
        w = os.fdopen(w, 'w')
        os.setsid()
        try:
            pid2 = os.fork()
        except:
            pid2 = None
        if pid2 == 0:
            os.chdir("/")
            null_fd = os.open("/dev/null", os.O_RDWR)
            if stdin_tmpfile is not None:
                os.dup2(stdin_tmpfile.fileno(), 0)
            else:
                os.dup2(null_fd, 0)
            os.dup2(null_fd, 1)
            os.dup2(null_fd, 2)
            for fd in range(3, 256):
                try:
                    os.close(fd)
                except:
                    pass
            os.execvp(prog, args)
            os._exit(1)
        else:
            w.write(str(pid2 or 0))
            w.close()
            os._exit(0)
    os.close(w)
    r = os.fdopen(r)
    daemon_pid = int(r.read())
    r.close()
    os.waitpid(pid, 0)
    return daemon_pid