コード例 #1
0
ファイル: posix_proc.py プロジェクト: N3X15/ambuild
  def spawn(cls, channel):
    pid = pid_t()
    file_actions = ctypes.pointer(posix_spawn_file_actions_t())
    res = posix_spawn_file_actions_init(file_actions)
    check_errno(res)

    exe = ctypes.c_char_p(util.str2b(sys.executable))

    # Bind 
    remap = (
      (channel.fd, kStartFd),
    )
    for fromfd, tofd in remap:
      res = posix_spawn_file_actions_adddup2(file_actions, fromfd, tofd)
      check_errno(res)

    envp = (ctypes.c_char_p * (len(os.environ) + 1))()
    for index, key in enumerate(os.environ.keys()):
      export = key + '=' + os.environ[key]
      envp[index] = ctypes.c_char_p(util.str2b(export))
    envp[len(os.environ)] = None

    # Create argv.
    argv = [
      sys.executable,
      '-c',
      'import sys; from ambuild2.ipc.posix_proc import child_main; child_main("{0}")'.format(channel.name)
    ]

    c_argv = (ctypes.c_char_p * (len(argv) + 1))()
    for index, arg in enumerate(argv):
      c_argv[index] = ctypes.c_char_p(util.str2b(arg))
    c_argv[len(argv)] = None

    res = posix_spawnp(
      ctypes.pointer(pid),
      exe,
      file_actions,
      None,
      ctypes.cast(c_argv, ctypes.c_void_p),
      ctypes.cast(envp, ctypes.c_void_p)
    )
    check_errno(res)

    posix_spawn_file_actions_destroy(file_actions)

    return cls(pid.value)
コード例 #2
0
ファイル: posix_proc.py プロジェクト: xtfvje/ambuild
  def spawn(cls, channel):
    pid = pid_t()
    file_actions = ctypes.pointer(posix_spawn_file_actions_t())
    res = posix_spawn_file_actions_init(file_actions)
    check_errno(res)

    exe = ctypes.c_char_p(util.str2b(sys.executable))

    # Bind 
    remap = (
      (channel.fd, kStartFd),
    )
    for fromfd, tofd in remap:
      res = posix_spawn_file_actions_adddup2(file_actions, fromfd, tofd)
      check_errno(res)

    envp = (ctypes.c_char_p * (len(os.environ) + 1))()
    for index, key in enumerate(os.environ.keys()):
      export = key + '=' + os.environ[key]
      envp[index] = ctypes.c_char_p(util.str2b(export))
    envp[len(os.environ)] = None

    # Create argv.
    argv = [
      sys.executable,
      '-c',
      'import sys; from ambuild2.ipc.posix_proc import child_main; child_main("{0}")'.format(channel.name)
    ]

    c_argv = (ctypes.c_char_p * (len(argv) + 1))()
    for index, arg in enumerate(argv):
      c_argv[index] = ctypes.c_char_p(util.str2b(arg))
    c_argv[len(argv)] = None

    res = posix_spawnp(
      ctypes.pointer(pid),
      exe,
      file_actions,
      None,
      ctypes.cast(c_argv, ctypes.c_void_p),
      ctypes.cast(envp, ctypes.c_void_p)
    )
    check_errno(res)

    posix_spawn_file_actions_destroy(file_actions)

    return cls(pid.value)
コード例 #3
0
ファイル: winapi.py プロジェクト: N3X15/ambuild
  def spawn(cls, channel):
    eval = 'from ambuild2.ipc.windows import child_main; child_main()'
    argv = [sys.executable, '-c', eval]
    argv += ['--name', '{0}'.format(channel.name)]

    # This is pretty hacky. Maybe we should just pass in the pipe name instead.
    argv += ['--pipe', '{0}'.format(channel.path)]

    cmdline = ' '.join(['"{0}"'.format(arg) for arg in argv])
    cmdline = util.str2b(cmdline)
    cmdline_buffer = ctypes.create_string_buffer(cmdline)

    startup_info = StartupInfo()
    startup_info.dwFlags |= STARTF_USESTDHANDLES
    startup_info.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE)
    startup_info.hStdError = GetStdHandle(STD_ERROR_HANDLE)
    startup_info.hStdInput = GetStdHandle(STD_INPUT_HANDLE)

    proc_info = ProcessInformation()

    executable = sys_executable()

    rval = fnCreateProcessA(
      util.str2b(executable),
      cmdline_buffer,
      None,
      None,
      1,    # bInheritHandles
      0,    # dwCreationFlags
      None, # lpEnvironment
      None, # lpCurrentDirectory
      ctypes.pointer(startup_info),
      ctypes.pointer(proc_info)
    )
    if not rval:
      raise WinError()

    CloseHandle(proc_info.hThread)

    return cls(proc_info.hProcess, proc_info.dwProcessId)
コード例 #4
0
ファイル: winapi.py プロジェクト: N3X15/ambuild
def OpenPipe(path):
  rval = fnCreateFileA(
    util.str2b(path),
    GENERIC_READ | GENERIC_WRITE,
    0,
    None,
    OPEN_EXISTING,
    FILE_FLAG_OVERLAPPED,
    None
  )
  if rval == INVALID_HANDLE_VALUE:
    raise WinError()
  return handle_t(rval)
コード例 #5
0
    def spawn(cls, channel):
        eval = 'from ambuild2.ipc.windows import child_main; child_main()'
        argv = [sys.executable, '-c', eval]
        argv += ['--name', '{0}'.format(channel.name)]

        # This is pretty hacky. Maybe we should just pass in the pipe name instead.
        argv += ['--pipe', '{0}'.format(channel.path)]

        cmdline = ' '.join(['"{0}"'.format(arg) for arg in argv])
        cmdline = util.str2b(cmdline)
        cmdline_buffer = ctypes.create_string_buffer(cmdline)

        startup_info = StartupInfo()
        startup_info.dwFlags |= STARTF_USESTDHANDLES
        startup_info.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE)
        startup_info.hStdError = GetStdHandle(STD_ERROR_HANDLE)
        startup_info.hStdInput = GetStdHandle(STD_INPUT_HANDLE)

        proc_info = ProcessInformation()

        executable = sys_executable()

        rval = fnCreateProcessA(
            util.str2b(executable),
            cmdline_buffer,
            None,
            None,
            1,  # bInheritHandles
            0,  # dwCreationFlags
            None,  # lpEnvironment
            None,  # lpCurrentDirectory
            ctypes.pointer(startup_info),
            ctypes.pointer(proc_info))
        if not rval:
            raise WinError()

        CloseHandle(proc_info.hThread)

        return cls(proc_info.hProcess, proc_info.dwProcessId)
コード例 #6
0
def CreateNamedPipe():
    global pipe_counter_

    pipe_name = r'\\.\pipe\ambuild-{0}-{1}'.format(os.getpid(), pipe_counter_)
    pipe_counter_ += 1

    flags = PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED | FILE_FLAG_FIRST_PIPE_INSTANCE

    rval = fnCreateNamedPipeA(util.str2b(pipe_name), flags,
                              PIPE_TYPE_BYTE | PIPE_TYPE_READMODE_BYTE, 1,
                              4096, 4096, 5000, None)
    if rval == INVALID_HANDLE_VALUE:
        raise WinError()
    return handle_t(rval), pipe_name
コード例 #7
0
ファイル: winapi.py プロジェクト: N3X15/ambuild
def CreateNamedPipe():
  global pipe_counter_

  pipe_name = r'\\.\pipe\ambuild-{0}-{1}'.format(os.getpid(), pipe_counter_)
  pipe_counter_ += 1

  flags = PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED | FILE_FLAG_FIRST_PIPE_INSTANCE

  rval = fnCreateNamedPipeA(
    util.str2b(pipe_name),
    flags,
    PIPE_TYPE_BYTE | PIPE_TYPE_READMODE_BYTE,
    1,
    4096,
    4096,
    5000,
    None
  )
  if rval == INVALID_HANDLE_VALUE:
    raise WinError()
  return handle_t(rval), pipe_name
コード例 #8
0
def OpenPipe(path):
    rval = fnCreateFileA(util.str2b(path), GENERIC_READ | GENERIC_WRITE, 0,
                         None, OPEN_EXISTING, FILE_FLAG_OVERLAPPED, None)
    if rval == INVALID_HANDLE_VALUE:
        raise WinError()
    return handle_t(rval)