chiark / gitweb /
systemd-python: introduce error setting helper
[elogind.git] / src / python-systemd / _reader.c
index 9d7ce2da2738a43086818f7cae0dfb6ec25eea5e..65ca06b97ff48a800522f134cd341fb569b3e9c1 100644 (file)
@@ -30,6 +30,20 @@ typedef struct {
 } Journal;
 static PyTypeObject JournalType;
 
+static int set_error(int r, const char* path, const char* invalid_message) {
+    if (r >= 0)
+        return r;
+    if (r == -EINVAL && invalid_message)
+        PyErr_SetString(PyExc_ValueError, invalid_message);
+    else if (r == -ENOMEM)
+        PyErr_SetString(PyExc_MemoryError, "Not enough memory");
+    else {
+        errno = -r;
+        PyErr_SetFromErrnoWithFilename(PyExc_OSError, path);
+    }
+    return 1;
+}
+
 static void
 Journal_dealloc(Journal* self)
 {
@@ -57,7 +71,7 @@ Journal_init(Journal *self, PyObject *args, PyObject *keywds)
     char *path=NULL;
 
     static char *kwlist[] = {"flags", "path", NULL};
-    if (! PyArg_ParseTupleAndKeywords(args, keywds, "|is", kwlist,
+    if (! PyArg_ParseTupleAndKeywords(args, keywds, "|iz", kwlist,
                                       &flags, &path))
         return 1;
 
@@ -69,16 +83,8 @@ Journal_init(Journal *self, PyObject *args, PyObject *keywds)
         r = sd_journal_open(&self->j, flags);
     }
     Py_END_ALLOW_THREADS
-    if (r < 0) {
-        errno = -r;
-        PyObject *errtype = r == -EINVAL ? PyExc_ValueError :
-                            r == -ENOMEM ? PyExc_MemoryError :
-                            PyExc_OSError;
-        PyErr_SetFromErrnoWithFilename(errtype, path);
-        return -1;
-    }
 
-    return 0;
+    return set_error(r, path, "Invalid flags or path");
 }
 
 PyDoc_STRVAR(Journal_get_next__doc__,
@@ -97,7 +103,7 @@ Journal_get_next(Journal *self, PyObject *args)
         return NULL;
     }
 
-    int r;
+    int r = -EINVAL;
     Py_BEGIN_ALLOW_THREADS
     if (skip == 1LL) {
         r = sd_journal_next(self->j);
@@ -110,13 +116,11 @@ Journal_get_next(Journal *self, PyObject *args)
     }
     Py_END_ALLOW_THREADS
 
-    if (r < 0) {
-        errno = -r;
-        PyErr_SetFromErrno(PyExc_OSError);
+    set_error(r, NULL, NULL);
+    if (r < 0)
         return NULL;
-    }else if ( r == 0) { //EOF
+    else if (r == 0) /* EOF */
         return PyDict_New();
-    }
 
     PyObject *dict;
     dict = PyDict_New();
@@ -217,116 +221,24 @@ Journal_get_previous(Journal *self, PyObject *args)
 }
 
 PyDoc_STRVAR(Journal_add_match__doc__,
-"add_match(match, ..., field=value, ...) -> None\n\n"
+"add_match(match) -> None\n\n"
 "Add a match to filter journal log entries. All matches of different\n"
-"field are combined in logical AND, and matches of the same field\n"
+"fields are combined in logical AND, and matches of the same field\n"
 "are automatically combined in logical OR.\n"
-"Matches can be passed as strings \"field=value\", or keyword\n"
-"arguments field=\"value\".");
+"Match is string of form \"field=value\".");
 static PyObject *
 Journal_add_match(Journal *self, PyObject *args, PyObject *keywds)
 {
-    Py_ssize_t arg_match_len;
-    char *arg_match;
-    int i, r;
-    for (i = 0; i < PySequence_Size(args); i++) {
-#if PY_MAJOR_VERSION >=3
-        PyObject *arg;
-        arg = PySequence_Fast_GET_ITEM(args, i);
-        if (PyUnicode_Check(arg)) {
-#if PY_MINOR_VERSION >=3
-            arg_match = PyUnicode_AsUTF8AndSize(arg, &arg_match_len);
-#else
-            PyObject *temp;
-            temp = PyUnicode_AsUTF8String(arg);
-            PyBytes_AsStringAndSize(temp, &arg_match, &arg_match_len);
-            Py_DECREF(temp);
-#endif
-        }else if (PyBytes_Check(arg)) {
-            PyBytes_AsStringAndSize(arg, &arg_match, &arg_match_len);
-        }else{
-            PyErr_SetString(PyExc_TypeError, "expected bytes or string");
-        }
-#else
-        PyString_AsStringAndSize(PySequence_Fast_GET_ITEM(args, i), &arg_match, &arg_match_len);
-#endif
-        if (PyErr_Occurred())
-            return NULL;
-        r = sd_journal_add_match(self->j, arg_match, arg_match_len);
-        if (r < 0) {
-            errno = -r;
-            PyObject *errtype = r == -EINVAL ? PyExc_ValueError :
-                                r == -ENOMEM ? PyExc_MemoryError :
-                                PyExc_OSError;
-            PyErr_SetFromErrno(errtype);
-            return NULL;
-        }
-    }
-
-    if (! keywds)
-        Py_RETURN_NONE;
-
-    PyObject *key, *value;
-    Py_ssize_t pos=0, match_key_len, match_value_len;
+    char *match;
     int match_len;
-    char *match_key, *match_value;
-    void *match;
-    while (PyDict_Next(keywds, &pos, &key, &value)) {
-#if PY_MAJOR_VERSION >=3
-        if (PyUnicode_Check(key)) {
-#if PY_MINOR_VERSION >=3
-            match_key = PyUnicode_AsUTF8AndSize(key, &match_key_len);
-#else
-            PyObject *temp2;
-            temp2 = PyUnicode_AsUTF8String(key);
-            PyBytes_AsStringAndSize(temp2, &match_key, &match_key_len);
-            Py_DECREF(temp2);
-#endif
-        }else if (PyBytes_Check(key)) {
-            PyBytes_AsStringAndSize(key, &match_key, &match_key_len);
-        }else{
-            PyErr_SetString(PyExc_TypeError, "expected bytes or string");
-        }
-        if (PyUnicode_Check(value)) {
-#if PY_MINOR_VERSION >=3
-            match_value = PyUnicode_AsUTF8AndSize(value, &match_value_len);
-#else
-            PyObject *temp3;
-            temp3 = PyUnicode_AsUTF8String(value);
-            PyBytes_AsStringAndSize(temp3, &match_value, &match_value_len);
-            Py_DECREF(temp3);
-#endif
-        }else if (PyBytes_Check(value)) {
-            PyBytes_AsStringAndSize(value, &match_value, &match_value_len);
-        }else{
-            PyErr_SetString(PyExc_TypeError, "expected bytes or string");
-        }
-#else
-        PyString_AsStringAndSize(key, &match_key, &match_key_len);
-        PyString_AsStringAndSize(value, &match_value, &match_value_len);
-#endif
-        if (PyErr_Occurred())
-            return NULL;
-
-        match_len = match_key_len + 1 + match_value_len;
-        match = malloc(match_len);
-        memcpy(match, match_key, match_key_len);
-        memcpy(match + match_key_len, "=", 1);
-        memcpy(match + match_key_len + 1, match_value, match_value_len);
+    if (! PyArg_ParseTuple(args, "s#", &match, &match_len))
+        return NULL;
 
-        r = sd_journal_add_match(self->j, match, match_len);
-        free(match);
-        if (r == -EINVAL) {
-            PyErr_SetString(PyExc_ValueError, "Invalid match");
-            return NULL;
-        }else if (r == -ENOMEM) {
-            PyErr_SetString(PyExc_MemoryError, "Not enough memory");
-            return NULL;
-        }else if (r < 0) {
-            PyErr_SetString(PyExc_RuntimeError, "Error adding match");
+    int r;
+    r = sd_journal_add_match(self->j, match, match_len);
+    set_error(r, NULL, "Invalid match");
+    if (r < 0)
             return NULL;
-        }
-    }
 
     Py_RETURN_NONE;
 }
@@ -340,13 +252,9 @@ Journal_add_disjunction(Journal *self, PyObject *args)
 {
     int r;
     r = sd_journal_add_disjunction(self->j);
-    if (r < 0) {
-        errno = -r;
-        PyObject *errtype = r == -ENOMEM ? PyExc_MemoryError :
-                            PyExc_OSError;
-        PyErr_SetFromErrno(errtype);
+    set_error(r, NULL, NULL);
+    if (r < 0)
         return NULL;
-    }
     Py_RETURN_NONE;
 }
 
@@ -384,11 +292,9 @@ Journal_seek(Journal *self, PyObject *args, PyObject *keywds)
         Py_BEGIN_ALLOW_THREADS
         r = sd_journal_seek_head(self->j);
         Py_END_ALLOW_THREADS
-        if (r < 0) {
-            errno = -r;
-            PyErr_SetFromErrno(PyExc_OSError);
+        if (set_error(r, NULL, NULL))
             return NULL;
-        }
+
         if (offset > 0LL) {
             result = PyObject_CallMethod((PyObject *)self, "get_next", "L", offset);
         }
@@ -399,11 +305,9 @@ Journal_seek(Journal *self, PyObject *args, PyObject *keywds)
         Py_BEGIN_ALLOW_THREADS
         r = sd_journal_seek_tail(self->j);
         Py_END_ALLOW_THREADS
-        if (r < 0) {
-            errno = -r;
-            PyErr_SetFromErrno(PyExc_OSError);
+        if (set_error(r, NULL, NULL))
             return NULL;
-        }
+
         if (offset < 0LL) {
             result = PyObject_CallMethod((PyObject *)self, "get_next", "L", offset);
         }else{
@@ -413,8 +317,7 @@ Journal_seek(Journal *self, PyObject *args, PyObject *keywds)
         PyErr_SetString(PyExc_ValueError, "Invalid value for whence");
     }
 
-    if (result)
-        Py_DECREF(result);
+    Py_XDECREF(result);
     if (PyErr_Occurred())
         return NULL;
     Py_RETURN_NONE;
@@ -423,15 +326,17 @@ Journal_seek(Journal *self, PyObject *args, PyObject *keywds)
 PyDoc_STRVAR(Journal_seek_realtime__doc__,
 "seek_realtime(realtime) -> None\n\n"
 "Seek to nearest matching journal entry to `realtime`. Argument\n"
-"`realtime` can be an integer unix timestamp in usecs or a "
-"datetime instance.");
+"`realtime` can must be an integer unix timestamp.");
 static PyObject *
 Journal_seek_realtime(Journal *self, PyObject *args)
 {
-    uint64_t timestamp;
-    if (! PyArg_ParseTuple(args, "K", &timestamp))
+    double timedouble;
+    if (! PyArg_ParseTuple(args, "d", &timedouble))
         return NULL;
 
+    uint64_t timestamp;
+    timestamp = (uint64_t) (timedouble * 1.0E6);
+
     if ((int64_t) timestamp < 0LL) {
         PyErr_SetString(PyExc_ValueError, "Time must be positive integer");
         return NULL;
@@ -441,19 +346,15 @@ Journal_seek_realtime(Journal *self, PyObject *args)
     Py_BEGIN_ALLOW_THREADS
     r = sd_journal_seek_realtime_usec(self->j, timestamp);
     Py_END_ALLOW_THREADS
-    if (r < 0) {
-        errno = -r;
-        PyErr_SetFromErrno(PyExc_OSError);
+    if (set_error(r, NULL, NULL))
         return NULL;
-    }
     Py_RETURN_NONE;
 }
 
 PyDoc_STRVAR(Journal_seek_monotonic__doc__,
 "seek_monotonic(monotonic[, bootid]) -> None\n\n"
 "Seek to nearest matching journal entry to `monotonic`. Argument\n"
-"`monotonic` is an timestamp from boot in secs, or a\n"
-"timedelta instance.\n"
+"`monotonic` is an timestamp from boot in seconds.\n"
 "Argument `bootid` is a string representing which boot the\n"
 "monotonic time is reference to. Defaults to current bootid.");
 static PyObject *
@@ -476,37 +377,24 @@ Journal_seek_monotonic(Journal *self, PyObject *args)
     int r;
     if (bootid) {
         r = sd_id128_from_string(bootid, &sd_id);
-        if (r == -EINVAL) {
-            PyErr_SetString(PyExc_ValueError, "Invalid bootid");
+        if (set_error(r, NULL, "Invalid bootid"))
             return NULL;
-        }else if (r < 0) {
-            errno = -r;
-            PyErr_SetFromErrno(PyExc_OSError);
-            return NULL;
-        }
-    }else{
+    } else {
+        Py_BEGIN_ALLOW_THREADS
         r = sd_id128_get_boot(&sd_id);
-        if (r == -EIO) {
-            PyErr_SetString(PyExc_IOError, "Error getting current boot ID");
-            return NULL;
-        }else if (r < 0) {
-            errno = -r;
-            PyErr_SetFromErrno(PyExc_OSError);
+        Py_END_ALLOW_THREADS
+        if (set_error(r, NULL, NULL))
             return NULL;
-        }
     }
 
     Py_BEGIN_ALLOW_THREADS
     r = sd_journal_seek_monotonic_usec(self->j, sd_id, timestamp);
     Py_END_ALLOW_THREADS
-    if (r < 0) {
-        errno = -r;
-        PyErr_SetFromErrno(PyExc_OSError);
+    if (set_error(r, NULL, NULL))
         return NULL;
-    }
     Py_RETURN_NONE;
 }
+
 PyDoc_STRVAR(Journal_wait__doc__,
 "wait([timeout]) -> Change state (integer)\n\n"
 "Waits until there is a change in the journal. Argument `timeout`\n"
@@ -525,19 +413,11 @@ Journal_wait(Journal *self, PyObject *args, PyObject *keywds)
 
     int r;
     Py_BEGIN_ALLOW_THREADS
-    if ( timeout == 0LL) {
-        r = sd_journal_wait(self->j, (uint64_t) -1);
-    }else{
-        r = sd_journal_wait(self->j, timeout * 1E6);
-    }
+    r = sd_journal_wait(self->j, timeout ==0 ? (uint64_t) -1 : timeout * 1E6);
     Py_END_ALLOW_THREADS
-    if (r < 0) {
-        errno = -r;
-        PyObject *errtype = r == -ENOMEM ? PyExc_MemoryError :
-                            PyExc_OSError;
-        PyErr_SetFromErrno(errtype);
+    if (set_error(r, NULL, NULL))
         return NULL;
-    }
+
 #if PY_MAJOR_VERSION >=3
     return PyLong_FromLong(r);
 #else
@@ -559,14 +439,8 @@ Journal_seek_cursor(Journal *self, PyObject *args)
     Py_BEGIN_ALLOW_THREADS
     r = sd_journal_seek_cursor(self->j, cursor);
     Py_END_ALLOW_THREADS
-    if (r < 0) {
-        errno = -r;
-        PyObject *errtype = r == -EINVAL ? PyExc_ValueError :
-                            r == -ENOMEM ? PyExc_MemoryError :
-                            PyExc_OSError;
-        PyErr_SetFromErrno(errtype);
+    if (set_error(r, NULL, "Invalid cursor"))
         return NULL;
-    }
     Py_RETURN_NONE;
 }
 
@@ -584,6 +458,8 @@ Journal_iternext(PyObject *self)
     Py_ssize_t dict_size;
 
     dict = PyObject_CallMethod(self, "get_next", "");
+    if (PyErr_Occurred())
+        return NULL;
     dict_size = PyDict_Size(dict);
     if ((int64_t) dict_size > 0LL) {
         return dict;
@@ -594,7 +470,6 @@ Journal_iternext(PyObject *self)
     }
 }
 
-#ifdef SD_JOURNAL_FOREACH_UNIQUE
 PyDoc_STRVAR(Journal_query_unique__doc__,
 "query_unique(field) -> a set of values\n\n"
 "Returns a set of unique values in journal for given `field`.\n"
@@ -610,14 +485,8 @@ Journal_query_unique(Journal *self, PyObject *args)
     Py_BEGIN_ALLOW_THREADS
     r = sd_journal_query_unique(self->j, query);
     Py_END_ALLOW_THREADS
-    if (r < 0) {
-        errno = -r;
-        PyObject *errtype = r == -EINVAL ? PyExc_ValueError :
-                            r == -ENOMEM ? PyExc_MemoryError :
-                            PyExc_OSError;
-        PyErr_SetFromErrno(errtype);
+    if (set_error(r, NULL, "Invalid field name"))
         return NULL;
-    }
 
     const void *uniq;
     size_t uniq_len;
@@ -640,7 +509,6 @@ Journal_query_unique(Journal *self, PyObject *args)
     Py_DECREF(key);
     return value_set;
 }
-#endif //def SD_JOURNAL_FOREACH_UNIQUE
 
 static PyObject *
 Journal_get_data_threshold(Journal *self, void *closure)
@@ -650,11 +518,8 @@ Journal_get_data_threshold(Journal *self, void *closure)
     int r;
 
     r = sd_journal_get_data_threshold(self->j, &cvalue);
-    if (r < 0) {
-        errno = -r;
-        PyErr_SetFromErrno(PyExc_OSError);
+    if (set_error(r, NULL, NULL))
         return NULL;
-    }
 
 #if PY_MAJOR_VERSION >=3
     value = PyLong_FromSize_t(cvalue);
@@ -685,12 +550,7 @@ Journal_set_data_threshold(Journal *self, PyObject *value, void *closure)
 #else
     r = sd_journal_set_data_threshold(self->j, (size_t) PyInt_AsLong(value));
 #endif
-    if (r < 0) {
-        errno = -r;
-        PyErr_SetFromErrno(PyExc_OSError);
-        return NULL;
-    }
-    return 0;
+    return set_error(r, NULL, NULL);
 }
 
 static PyGetSetDef Journal_getseters[] = {
@@ -723,16 +583,14 @@ static PyMethodDef Journal_methods[] = {
     Journal_wait__doc__},
     {"seek_cursor", (PyCFunction)Journal_seek_cursor, METH_VARARGS,
     Journal_seek_cursor__doc__},
-#ifdef SD_JOURNAL_FOREACH_UNIQUE
     {"query_unique", (PyCFunction)Journal_query_unique, METH_VARARGS,
     Journal_query_unique__doc__},
-#endif
     {NULL}  /* Sentinel */
 };
 
 static PyTypeObject JournalType = {
     PyVarObject_HEAD_INIT(NULL, 0)
-    "_reader.Journal",           /*tp_name*/
+    "_reader._Journal",               /*tp_name*/
     sizeof(Journal),                  /*tp_basicsize*/
     0,                                /*tp_itemsize*/
     (destructor)Journal_dealloc,      /*tp_dealloc*/