Branch data Line data Source code
1 : : /* SPDX-License-Identifier: LGPL-2.1+ */
2 : :
3 : : #include <errno.h>
4 : : #include <getopt.h>
5 : : #include <stdio.h>
6 : : #include <stdlib.h>
7 : : #include <unistd.h>
8 : :
9 : : #include "sd-daemon.h"
10 : :
11 : : #include "alloc-util.h"
12 : : #include "env-util.h"
13 : : #include "format-util.h"
14 : : #include "log.h"
15 : : #include "main-func.h"
16 : : #include "parse-util.h"
17 : : #include "pretty-print.h"
18 : : #include "string-util.h"
19 : : #include "strv.h"
20 : : #include "user-util.h"
21 : : #include "util.h"
22 : :
23 : : static bool arg_ready = false;
24 : : static pid_t arg_pid = 0;
25 : : static const char *arg_status = NULL;
26 : : static bool arg_booted = false;
27 : : static uid_t arg_uid = UID_INVALID;
28 : : static gid_t arg_gid = GID_INVALID;
29 : :
30 : 12 : static int help(void) {
31 : 12 : _cleanup_free_ char *link = NULL;
32 : : int r;
33 : :
34 : 12 : r = terminal_urlify_man("systemd-notify", "1", &link);
35 [ - + ]: 12 : if (r < 0)
36 : 0 : return log_oom();
37 : :
38 : 12 : printf("%s [OPTIONS...] [VARIABLE=VALUE...]\n\n"
39 : : "Notify the init system about service status updates.\n\n"
40 : : " -h --help Show this help\n"
41 : : " --version Show package version\n"
42 : : " --ready Inform the init system about service start-up completion\n"
43 : : " --pid[=PID] Set main PID of daemon\n"
44 : : " --uid=USER Set user to send from\n"
45 : : " --status=TEXT Set status text\n"
46 : : " --booted Check if the system was booted up with systemd\n"
47 : : "\nSee the %s for details.\n"
48 : : , program_invocation_short_name
49 : : , link
50 : : );
51 : :
52 : 12 : return 0;
53 : : }
54 : :
55 : 16 : static int parse_argv(int argc, char *argv[]) {
56 : :
57 : : enum {
58 : : ARG_READY = 0x100,
59 : : ARG_VERSION,
60 : : ARG_PID,
61 : : ARG_STATUS,
62 : : ARG_BOOTED,
63 : : ARG_UID,
64 : : };
65 : :
66 : : static const struct option options[] = {
67 : : { "help", no_argument, NULL, 'h' },
68 : : { "version", no_argument, NULL, ARG_VERSION },
69 : : { "ready", no_argument, NULL, ARG_READY },
70 : : { "pid", optional_argument, NULL, ARG_PID },
71 : : { "status", required_argument, NULL, ARG_STATUS },
72 : : { "booted", no_argument, NULL, ARG_BOOTED },
73 : : { "uid", required_argument, NULL, ARG_UID },
74 : : {}
75 : : };
76 : :
77 : : int c, r;
78 : :
79 [ - + ]: 16 : assert(argc >= 0);
80 [ - + ]: 16 : assert(argv);
81 : :
82 [ + - ]: 16 : while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0) {
83 : :
84 [ + - - - : 16 : switch (c) {
- - - +
- ]
85 : :
86 : 12 : case 'h':
87 : 12 : return help();
88 : :
89 : 0 : case ARG_VERSION:
90 : 0 : return version();
91 : :
92 : 0 : case ARG_READY:
93 : 0 : arg_ready = true;
94 : 0 : break;
95 : :
96 : 0 : case ARG_PID:
97 : :
98 [ # # ]: 0 : if (optarg) {
99 [ # # ]: 0 : if (parse_pid(optarg, &arg_pid) < 0)
100 [ # # ]: 0 : return log_error_errno(SYNTHETIC_ERRNO(EINVAL),
101 : : "Failed to parse PID %s.", optarg);
102 : : } else
103 : 0 : arg_pid = getppid();
104 : :
105 : 0 : break;
106 : :
107 : 0 : case ARG_STATUS:
108 : 0 : arg_status = optarg;
109 : 0 : break;
110 : :
111 : 0 : case ARG_BOOTED:
112 : 0 : arg_booted = true;
113 : 0 : break;
114 : :
115 : 0 : case ARG_UID: {
116 : 0 : const char *u = optarg;
117 : :
118 : 0 : r = get_user_creds(&u, &arg_uid, &arg_gid, NULL, NULL, 0);
119 [ # # ]: 0 : if (r == -ESRCH) /* If the user doesn't exist, then accept it anyway as numeric */
120 : 0 : r = parse_uid(u, &arg_uid);
121 [ # # ]: 0 : if (r < 0)
122 [ # # ]: 0 : return log_error_errno(r, "Can't resolve user %s: %m", optarg);
123 : :
124 : 0 : break;
125 : : }
126 : :
127 : 4 : case '?':
128 : 4 : return -EINVAL;
129 : :
130 : 0 : default:
131 : 0 : assert_not_reached("Unhandled option");
132 : : }
133 : : }
134 : :
135 [ # # ]: 0 : if (optind >= argc &&
136 [ # # ]: 0 : !arg_ready &&
137 [ # # ]: 0 : !arg_status &&
138 [ # # ]: 0 : !arg_pid &&
139 [ # # ]: 0 : !arg_booted) {
140 : 0 : help();
141 : 0 : return -EINVAL;
142 : : }
143 : :
144 : 0 : return 1;
145 : : }
146 : :
147 : 16 : static int run(int argc, char* argv[]) {
148 : 16 : _cleanup_free_ char *status = NULL, *cpid = NULL, *n = NULL;
149 : 16 : _cleanup_strv_free_ char **final_env = NULL;
150 : : char* our_env[4];
151 : 16 : unsigned i = 0;
152 : : int r;
153 : :
154 : 16 : log_show_color(true);
155 : 16 : log_parse_environment();
156 : 16 : log_open();
157 : :
158 : 16 : r = parse_argv(argc, argv);
159 [ + - ]: 16 : if (r <= 0)
160 : 16 : return r;
161 : :
162 [ # # ]: 0 : if (arg_booted)
163 : 0 : return sd_booted() <= 0;
164 : :
165 [ # # ]: 0 : if (arg_ready)
166 : 0 : our_env[i++] = (char*) "READY=1";
167 : :
168 [ # # ]: 0 : if (arg_status) {
169 : 0 : status = strjoin("STATUS=", arg_status);
170 [ # # ]: 0 : if (!status)
171 : 0 : return log_oom();
172 : :
173 : 0 : our_env[i++] = status;
174 : : }
175 : :
176 [ # # ]: 0 : if (arg_pid > 0) {
177 [ # # ]: 0 : if (asprintf(&cpid, "MAINPID="PID_FMT, arg_pid) < 0)
178 : 0 : return log_oom();
179 : :
180 : 0 : our_env[i++] = cpid;
181 : : }
182 : :
183 : 0 : our_env[i++] = NULL;
184 : :
185 : 0 : final_env = strv_env_merge(2, our_env, argv + optind);
186 [ # # ]: 0 : if (!final_env)
187 : 0 : return log_oom();
188 : :
189 [ # # ]: 0 : if (strv_isempty(final_env))
190 : 0 : return 0;
191 : :
192 : 0 : n = strv_join(final_env, "\n");
193 [ # # ]: 0 : if (!n)
194 : 0 : return log_oom();
195 : :
196 : : /* If this is requested change to the requested UID/GID. Note that we only change the real UID here, and leave
197 : : the effective UID in effect (which is 0 for this to work). That's because we want the privileges to fake the
198 : : ucred data, and sd_pid_notify() uses the real UID for filling in ucred. */
199 : :
200 [ # # # # ]: 0 : if (arg_gid != GID_INVALID &&
201 : 0 : setregid(arg_gid, (gid_t) -1) < 0)
202 [ # # ]: 0 : return log_error_errno(errno, "Failed to change GID: %m");
203 : :
204 [ # # # # ]: 0 : if (arg_uid != UID_INVALID &&
205 : 0 : setreuid(arg_uid, (uid_t) -1) < 0)
206 [ # # ]: 0 : return log_error_errno(errno, "Failed to change UID: %m");
207 : :
208 [ # # ]: 0 : r = sd_pid_notify(arg_pid ? arg_pid : getppid(), false, n);
209 [ # # ]: 0 : if (r < 0)
210 [ # # ]: 0 : return log_error_errno(r, "Failed to notify init system: %m");
211 [ # # ]: 0 : if (r == 0)
212 [ # # ]: 0 : return log_error_errno(SYNTHETIC_ERRNO(EOPNOTSUPP),
213 : : "No status data could be sent: $NOTIFY_SOCKET was not set");
214 : 0 : return 0;
215 : : }
216 : :
217 : 16 : DEFINE_MAIN_FUNCTION(run);
|
