1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2013 Zbigniew Jędrzejewski-Szmek <zbyszek@in.waw.pl>
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
22 #define PY_SSIZE_T_CLEAN
23 #pragma GCC diagnostic push
24 #pragma GCC diagnostic ignored "-Wredundant-decls"
26 #pragma GCC diagnostic pop
30 #include <sys/socket.h>
32 #include "systemd/sd-daemon.h"
36 PyDoc_STRVAR(module__doc__,
37 "Python interface to the libsystemd-daemon library.\n\n"
38 "Provides _listen_fds, notify, booted, and is_* functions\n"
39 "which wrap sd_listen_fds, sd_notify, sd_booted, sd_is_* and\n"
40 "useful for socket activation and checking if the system is\n"
41 "running under systemd."
44 PyDoc_STRVAR(booted__doc__,
45 "booted() -> bool\n\n"
46 "Return True iff this system is running under systemd.\n"
47 "Wraps sd_daemon_booted(3)."
50 static PyObject* booted(PyObject *self, PyObject *args) {
55 if (set_error(r, NULL, NULL) < 0)
58 return PyBool_FromLong(r);
61 PyDoc_STRVAR(notify__doc__,
62 "notify(status, unset_environment=False) -> bool\n\n"
63 "Send a message to the init system about a status change.\n"
64 "Wraps sd_notify(3).");
66 static PyObject* notify(PyObject *self, PyObject *args, PyObject *keywds) {
71 static const char* const kwlist[] = {
76 #if PY_MAJOR_VERSION >=3 && PY_MINOR_VERSION >= 3
77 if (!PyArg_ParseTupleAndKeywords(args, keywds, "s|p:notify",
78 (char**) kwlist, &msg, &unset))
82 if (!PyArg_ParseTupleAndKeywords(args, keywds, "s|O:notify",
83 (char**) kwlist, &msg, &obj))
86 unset = PyObject_IsTrue(obj);
91 r = sd_notify(unset, msg);
92 if (set_error(r, NULL, NULL) < 0)
95 return PyBool_FromLong(r);
99 PyDoc_STRVAR(listen_fds__doc__,
100 "_listen_fds(unset_environment=True) -> int\n\n"
101 "Return the number of descriptors passed to this process by the init system\n"
102 "as part of the socket-based activation logic.\n"
103 "Wraps sd_listen_fds(3)."
106 static PyObject* listen_fds(PyObject *self, PyObject *args, PyObject *keywds) {
110 static const char* const kwlist[] = {"unset_environment", NULL};
111 #if PY_MAJOR_VERSION >=3 && PY_MINOR_VERSION >= 3
112 if (!PyArg_ParseTupleAndKeywords(args, keywds, "|p:_listen_fds",
113 (char**) kwlist, &unset))
116 PyObject *obj = NULL;
117 if (!PyArg_ParseTupleAndKeywords(args, keywds, "|O:_listen_fds",
118 (char**) kwlist, &obj))
121 unset = PyObject_IsTrue(obj);
126 r = sd_listen_fds(unset);
127 if (set_error(r, NULL, NULL) < 0)
130 return long_FromLong(r);
133 PyDoc_STRVAR(is_fifo__doc__,
134 "_is_fifo(fd, path) -> bool\n\n"
135 "Returns True iff the descriptor refers to a FIFO or a pipe.\n"
136 "Wraps sd_is_fifo(3)."
140 static PyObject* is_fifo(PyObject *self, PyObject *args) {
143 const char *path = NULL;
145 #if PY_MAJOR_VERSION >=3 && PY_MINOR_VERSION >= 1
146 if (!PyArg_ParseTuple(args, "i|O&:_is_fifo",
147 &fd, Unicode_FSConverter, &path))
150 if (!PyArg_ParseTuple(args, "i|z:_is_fifo", &fd, &path))
154 r = sd_is_fifo(fd, path);
155 if (set_error(r, path, NULL) < 0)
158 return PyBool_FromLong(r);
162 PyDoc_STRVAR(is_mq__doc__,
163 "_is_mq(fd, path) -> bool\n\n"
164 "Returns True iff the descriptor refers to a POSIX message queue.\n"
168 static PyObject* is_mq(PyObject *self, PyObject *args) {
171 const char *path = NULL;
173 #if PY_MAJOR_VERSION >=3 && PY_MINOR_VERSION >= 1
174 if (!PyArg_ParseTuple(args, "i|O&:_is_mq",
175 &fd, Unicode_FSConverter, &path))
178 if (!PyArg_ParseTuple(args, "i|z:_is_mq", &fd, &path))
182 r = sd_is_mq(fd, path);
183 if (set_error(r, path, NULL) < 0)
186 return PyBool_FromLong(r);
191 PyDoc_STRVAR(is_socket__doc__,
192 "_is_socket(fd, family=AF_UNSPEC, type=0, listening=-1) -> bool\n\n"
193 "Returns True iff the descriptor refers to a socket.\n"
194 "Wraps sd_is_socket(3).\n\n"
195 "Constants for `family` are defined in the socket module."
198 static PyObject* is_socket(PyObject *self, PyObject *args) {
200 int fd, family = AF_UNSPEC, type = 0, listening = -1;
202 if (!PyArg_ParseTuple(args, "i|iii:_is_socket",
203 &fd, &family, &type, &listening))
206 r = sd_is_socket(fd, family, type, listening);
207 if (set_error(r, NULL, NULL) < 0)
210 return PyBool_FromLong(r);
214 PyDoc_STRVAR(is_socket_inet__doc__,
215 "_is_socket_inet(fd, family=AF_UNSPEC, type=0, listening=-1, port=0) -> bool\n\n"
216 "Wraps sd_is_socket_inet(3).\n\n"
217 "Constants for `family` are defined in the socket module."
220 static PyObject* is_socket_inet(PyObject *self, PyObject *args) {
222 int fd, family = AF_UNSPEC, type = 0, listening = -1, port = 0;
224 if (!PyArg_ParseTuple(args, "i|iiii:_is_socket_inet",
225 &fd, &family, &type, &listening, &port))
228 if (port < 0 || port > INT16_MAX) {
229 set_error(-EINVAL, NULL, "port must fit into uint16_t");
233 r = sd_is_socket_inet(fd, family, type, listening, (uint16_t) port);
234 if (set_error(r, NULL, NULL) < 0)
237 return PyBool_FromLong(r);
241 PyDoc_STRVAR(is_socket_unix__doc__,
242 "_is_socket_unix(fd, type, listening, path) -> bool\n\n"
243 "Wraps sd_is_socket_unix(3)."
246 static PyObject* is_socket_unix(PyObject *self, PyObject *args) {
248 int fd, type = 0, listening = -1;
250 Py_ssize_t length = 0;
252 #if PY_MAJOR_VERSION >=3 && PY_MINOR_VERSION >= 1
253 _cleanup_Py_DECREF_ PyObject *_path = NULL;
254 if (!PyArg_ParseTuple(args, "i|iiO&:_is_socket_unix",
255 &fd, &type, &listening, Unicode_FSConverter, &_path))
258 assert(PyBytes_Check(_path));
259 if (PyBytes_AsStringAndSize(_path, &path, &length))
263 if (!PyArg_ParseTuple(args, "i|iiz#:_is_socket_unix",
264 &fd, &type, &listening, &path, &length))
268 r = sd_is_socket_unix(fd, type, listening, path, length);
269 if (set_error(r, path, NULL) < 0)
272 return PyBool_FromLong(r);
276 static PyMethodDef methods[] = {
277 { "booted", booted, METH_NOARGS, booted__doc__},
278 { "notify", (PyCFunction) notify, METH_VARARGS | METH_KEYWORDS, notify__doc__},
279 { "_listen_fds", (PyCFunction) listen_fds, METH_VARARGS | METH_KEYWORDS, listen_fds__doc__},
280 { "_is_fifo", is_fifo, METH_VARARGS, is_fifo__doc__},
281 { "_is_mq", is_mq, METH_VARARGS, is_mq__doc__},
282 { "_is_socket", is_socket, METH_VARARGS, is_socket__doc__},
283 { "_is_socket_inet", is_socket_inet, METH_VARARGS, is_socket_inet__doc__},
284 { "_is_socket_unix", is_socket_unix, METH_VARARGS, is_socket_unix__doc__},
285 { NULL, NULL, 0, NULL } /* Sentinel */
288 #if PY_MAJOR_VERSION < 3
290 DISABLE_WARNING_MISSING_PROTOTYPES;
291 PyMODINIT_FUNC init_daemon(void) {
294 m = Py_InitModule3("_daemon", methods, module__doc__);
298 PyModule_AddIntConstant(m, "LISTEN_FDS_START", SD_LISTEN_FDS_START);
299 PyModule_AddStringConstant(m, "__version__", PACKAGE_VERSION);
305 static struct PyModuleDef module = {
306 PyModuleDef_HEAD_INIT,
307 "_daemon", /* name of module */
308 module__doc__, /* module documentation, may be NULL */
309 0, /* size of per-interpreter state of the module */
313 DISABLE_WARNING_MISSING_PROTOTYPES;
314 PyMODINIT_FUNC PyInit__daemon(void) {
317 m = PyModule_Create(&module);
321 if (PyModule_AddIntConstant(m, "LISTEN_FDS_START", SD_LISTEN_FDS_START) ||
322 PyModule_AddStringConstant(m, "__version__", PACKAGE_VERSION)) {