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
|
#include "system.h"
#include <dbus/dbus.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <rpm/rpmlog.h>
#include <rpm/rpmts.h>
#include "lib/rpmplugin.h"
static int lock_fd = -1;
static int inhibit(void)
{
DBusError err;
DBusConnection *bus = NULL;
DBusMessage *msg = NULL;
DBusMessage *reply = NULL;
int fd = -1;
dbus_error_init(&err);
bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, &err);
if (bus) {
msg = dbus_message_new_method_call("org.freedesktop.login1",
"/org/freedesktop/login1",
"org.freedesktop.login1.Manager",
"Inhibit");
}
if (msg) {
const char *what = "idle:sleep:shutdown";
const char *mode = "block";
const char *who = "RPM";
const char *reason = "Transaction running";
dbus_message_append_args(msg,
DBUS_TYPE_STRING, &what,
DBUS_TYPE_STRING, &who,
DBUS_TYPE_STRING, &reason,
DBUS_TYPE_STRING, &mode,
DBUS_TYPE_INVALID);
reply = dbus_connection_send_with_reply_and_block(bus, msg, -1, &err);
dbus_message_unref(msg);
}
if (reply) {
dbus_message_get_args(reply, &err,
DBUS_TYPE_UNIX_FD, &fd,
DBUS_TYPE_INVALID);
dbus_message_unref(reply);
}
if (dbus_error_is_set(&err)
&& !dbus_error_has_name(&err, DBUS_ERROR_NO_SERVER)
&& !dbus_error_has_name(&err, DBUS_ERROR_FILE_NOT_FOUND)) {
rpmlog(RPMLOG_WARNING,
"Unable to get systemd shutdown inhibition lock: %s\n",
err.message);
dbus_error_free(&err);
}
if (bus) {
dbus_connection_close(bus);
dbus_connection_unref(bus);
}
return fd;
}
static rpmRC systemd_inhibit_init(rpmPlugin plugin, rpmts ts)
{
struct stat st;
if (lstat("/run/systemd/system/", &st) == 0) {
if (S_ISDIR(st.st_mode)) {
return RPMRC_OK;
}
}
return RPMRC_NOTFOUND;
}
static rpmRC systemd_inhibit_tsm_pre(rpmPlugin plugin, rpmts ts)
{
if (rpmtsFlags(ts) & (RPMTRANS_FLAG_TEST|RPMTRANS_FLAG_BUILD_PROBS))
return RPMRC_OK;
lock_fd = inhibit();
if (lock_fd >= 0) {
rpmlog(RPMLOG_DEBUG, "System shutdown blocked (fd %d)\n", lock_fd);
}
return RPMRC_OK;
}
static rpmRC systemd_inhibit_tsm_post(rpmPlugin plugin, rpmts ts, int res)
{
if (lock_fd >= 0) {
close(lock_fd);
lock_fd = -1;
rpmlog(RPMLOG_DEBUG, "System shutdown unblocked\n");
}
return RPMRC_OK;
}
struct rpmPluginHooks_s systemd_inhibit_hooks = {
.init = systemd_inhibit_init,
.tsm_pre = systemd_inhibit_tsm_pre,
.tsm_post = systemd_inhibit_tsm_post,
};
|
