aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/include/winapi/process.hpp
blob: b79d7c295bc24049e0054a43dc599f8197e2c770 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
// Copyright (c) 2020 Egor Tensin <Egor.Tensin@gmail.com>
// This file is part of the "winapi-common" project.
// For details, see https://github.com/egor-tensin/winapi-common.
// Distributed under the MIT License.

#pragma once

#include "cmd_line.hpp"
#include "handle.hpp"
#include "process_io.hpp"
#include "resource.hpp"

#include <boost/config.hpp>
#include <boost/optional.hpp>

#include <windows.h>

#include <cstdint>
#include <string>
#include <utility>

namespace winapi {

struct ProcessParameters {
    enum ConsoleCreationMode {
        ConsoleNone,
        ConsoleInherit,
        ConsoleNew,
    };

    explicit ProcessParameters(const CommandLine& cmd_line) : cmd_line(cmd_line) {}

    // VS 2013 won't generate these automatically.
    ProcessParameters(ProcessParameters&&) BOOST_NOEXCEPT_OR_NOTHROW;
    ProcessParameters& operator=(ProcessParameters) BOOST_NOEXCEPT_OR_NOTHROW;
    void swap(ProcessParameters& other) BOOST_NOEXCEPT_OR_NOTHROW;
    ProcessParameters(const ProcessParameters&) = delete;

    CommandLine cmd_line;
    boost::optional<process::IO> io;
    ConsoleCreationMode console_mode = ConsoleNew;
};

inline void swap(ProcessParameters& a, ProcessParameters& b) BOOST_NOEXCEPT_OR_NOTHROW {
    a.swap(b);
}

struct ShellParameters : ProcessParameters {
    static ShellParameters runas(const CommandLine& cmd_line) {
        ShellParameters params{cmd_line};
        params.verb = "runas";
        return params;
    }

    explicit ShellParameters(const CommandLine& cmd_line) : ProcessParameters(cmd_line) {}

    // VS 2013 won't generate these automatically.
    ShellParameters(ShellParameters&&) BOOST_NOEXCEPT_OR_NOTHROW;
    ShellParameters& operator=(ShellParameters) BOOST_NOEXCEPT_OR_NOTHROW;
    void swap(ShellParameters& other) BOOST_NOEXCEPT_OR_NOTHROW;
    ShellParameters(const ShellParameters&) = delete;

    boost::optional<std::string> verb;
};

inline void swap(ShellParameters& a, ShellParameters& b) BOOST_NOEXCEPT_OR_NOTHROW {
    a.swap(b);
}

class Process {
public:
    using ID = DWORD;

    static Process create(ProcessParameters);
    static Process create(const CommandLine&);
    static Process create(const CommandLine&, process::IO);

    static Process shell(const ShellParameters&);
    static Process shell(const CommandLine&);

    static Process current();
    static Process open(ID, DWORD permissions = default_permissions());
    static Process open_r(ID);

    static DWORD default_permissions();
    static DWORD read_permissions();

    // VS 2013 won't generate these automatically.
    Process(Process&&) BOOST_NOEXCEPT_OR_NOTHROW;
    Process& operator=(Process) BOOST_NOEXCEPT_OR_NOTHROW;
    void swap(Process& other) BOOST_NOEXCEPT_OR_NOTHROW;
    Process(const Process&) = delete;

    ID get_id() const { return m_id; }
    const Handle& get_handle() const { return m_handle; }

    bool is_running() const;
    void wait() const;
    void terminate(int ec = 0) const;
    void shut_down(int ec = 0) const;
    int get_exit_code() const;

    std::string get_exe_path() const;

    static Resource get_resource(uint32_t id);
    static std::string get_resource_string(uint32_t id);

private:
    explicit Process(Handle&& handle);
    Process(ID, Handle&& handle);

    static HMODULE get_exe_module();

    ID m_id;
    Handle m_handle;
};

inline void swap(Process& a, Process& b) BOOST_NOEXCEPT_OR_NOTHROW {
    a.swap(b);
}

} // namespace winapi

namespace std {

template <>
inline void swap(winapi::ProcessParameters& a,
                 winapi::ProcessParameters& b) BOOST_NOEXCEPT_OR_NOTHROW {
    a.swap(b);
}

template <>
inline void swap(winapi::ShellParameters& a, winapi::ShellParameters& b) BOOST_NOEXCEPT_OR_NOTHROW {
    a.swap(b);
}

template <>
inline void swap(winapi::Process& a, winapi::Process& b) BOOST_NOEXCEPT_OR_NOTHROW {
    a.swap(b);
}

} // namespace std