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 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157
|
/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
/***
Copyright 2012 David Strauss <david@davidstrauss.net>
python-systemd is free software; you can redistribute it and/or modify it
under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation; either version 2.1 of the License, or
(at your option) any later version.
python-systemd is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with python-systemd; If not, see <http://www.gnu.org/licenses/>.
***/
#include <Python.h>
#include <alloca.h>
#define SD_JOURNAL_SUPPRESS_LOCATION
#include "systemd/sd-journal.h"
#include "macro.h"
PyDoc_STRVAR(journal_sendv__doc__,
"sendv('FIELD=value', 'FIELD=value', ...) -> None\n\n"
"Send an entry to the journal."
);
static PyObject *journal_sendv(PyObject *self, PyObject *args) {
struct iovec *iov = NULL;
int argc;
int i, r;
PyObject *ret = NULL;
PyObject **encoded;
/* Allocate an array for the argument strings */
argc = PyTuple_Size(args);
encoded = alloca0(argc * sizeof(PyObject*));
/* Allocate sufficient iovector space for the arguments. */
iov = alloca(argc * sizeof(struct iovec));
/* Iterate through the Python arguments and fill the iovector. */
for (i = 0; i < argc; ++i) {
PyObject *item = PyTuple_GetItem(args, i);
char *stritem;
Py_ssize_t length;
if (PyUnicode_Check(item)) {
encoded[i] = PyUnicode_AsEncodedString(item, "utf-8", "strict");
if (encoded[i] == NULL)
goto out;
item = encoded[i];
}
if (PyBytes_AsStringAndSize(item, &stritem, &length))
goto out;
iov[i].iov_base = stritem;
iov[i].iov_len = length;
}
/* Send the iovector to the journal. */
r = sd_journal_sendv(iov, argc);
if (r < 0) {
errno = -r;
PyErr_SetFromErrno(PyExc_IOError);
goto out;
}
/* End with success. */
Py_INCREF(Py_None);
ret = Py_None;
out:
for (i = 0; i < argc; ++i)
Py_XDECREF(encoded[i]);
return ret;
}
PyDoc_STRVAR(journal_stream_fd__doc__,
"stream_fd(identifier, priority, level_prefix) -> fd\n\n"
"Open a stream to journal by calling sd_journal_stream_fd(3)."
);
static PyObject* journal_stream_fd(PyObject *self, PyObject *args) {
const char* identifier;
int priority, level_prefix;
int fd;
if (!PyArg_ParseTuple(args, "sii:stream_fd",
&identifier, &priority, &level_prefix))
return NULL;
fd = sd_journal_stream_fd(identifier, priority, level_prefix);
if (fd < 0) {
errno = -fd;
return PyErr_SetFromErrno(PyExc_IOError);
}
return PyLong_FromLong(fd);
}
static PyMethodDef methods[] = {
{ "sendv", journal_sendv, METH_VARARGS, journal_sendv__doc__ },
{ "stream_fd", journal_stream_fd, METH_VARARGS, journal_stream_fd__doc__ },
{ NULL, NULL, 0, NULL } /* Sentinel */
};
#if PY_MAJOR_VERSION < 3
DISABLE_WARNING_MISSING_PROTOTYPES;
PyMODINIT_FUNC init_journal(void) {
PyObject *m;
m = Py_InitModule("_journal", methods);
if (m == NULL)
return;
PyModule_AddStringConstant(m, "__version__", PACKAGE_VERSION);
}
REENABLE_WARNING;
#else
static struct PyModuleDef module = {
PyModuleDef_HEAD_INIT,
"_journal", /* name of module */
NULL, /* module documentation, may be NULL */
-1, /* size of per-interpreter state of the module */
methods
};
DISABLE_WARNING_MISSING_PROTOTYPES;
PyMODINIT_FUNC PyInit__journal(void) {
PyObject *m;
m = PyModule_Create(&module);
if (m == NULL)
return NULL;
if (PyModule_AddStringConstant(m, "__version__", PACKAGE_VERSION)) {
Py_DECREF(m);
return NULL;
}
return m;
}
REENABLE_WARNING;
#endif
|