From 1d0657b29ee72e842e0d1ada61ea0177b3159a8e Mon Sep 17 00:00:00 2001
From: Jonathan Dieter <jdieter@lesbg.com>
Date: Mon, 12 Jan 2015 13:47:51 +0200
Subject: [PATCH 4/4] Return error rather than crashing if we can't allocate
 memory

Signed-off-by: Jonathan Dieter <jdieter@lesbg.com>
---
 deltarpmmodule.c | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/deltarpmmodule.c b/deltarpmmodule.c
index cf333b6..40ea60f 100644
--- a/deltarpmmodule.c
+++ b/deltarpmmodule.c
@@ -46,6 +46,11 @@ PyObject *createDict(struct deltarpm d)
   if (d.seq) {
     char *tmp = calloc(d.seql * 2 + 1, sizeof(char));
     int i;
+
+    if(tmp == NULL) {
+      PyErr_SetFromErrno(PyExc_SystemError);
+      return NULL;
+    }
     for (i = 0; i < d.seql; i++) {
       char buf[3];
       
@@ -54,6 +59,10 @@ PyObject *createDict(struct deltarpm d)
     }
     o = PyString_FromString(tmp);
     free(tmp);
+    if(o == NULL) {
+      PyErr_SetFromErrno(PyExc_SystemError);
+      return NULL;
+    }
     PyDict_SetItemString(dict, "seq", o);
     Py_DECREF(o);
   } else {
-- 
2.1.0