From 7dd0e1ec7abda0f8f9609c10f988792b26a1768e Mon Sep 17 00:00:00 2001
From: Ned Batchelder <ned@nedbatchelder.com>
Date: Mon, 23 Nov 2009 06:41:44 -0500
Subject: One more fine point of error handling in the C extension.

---
 coverage/tracer.c | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'coverage')

diff --git a/coverage/tracer.c b/coverage/tracer.c
index eddbc856..304190e0 100644
--- a/coverage/tracer.c
+++ b/coverage/tracer.c
@@ -138,6 +138,7 @@ Tracer_init(Tracer *self, PyObject *args, PyObject *kwds)
     self->data_stack = PyMem_Malloc(STACK_DELTA*sizeof(DataStackEntry));
     if (self->data_stack == NULL) {
         STATS( self->stats.errors++; )
+        PyErr_NoMemory();
         return -1;
     }
     self->data_stack_alloc = STACK_DELTA;
@@ -301,6 +302,7 @@ Tracer_trace(Tracer *self, PyFrameObject *frame, int what, PyObject *arg)
             DataStackEntry * bigger_data_stack = PyMem_Realloc(self->data_stack, bigger * sizeof(DataStackEntry));
             if (bigger_data_stack == NULL) {
                 STATS( self->stats.errors++; )
+                PyErr_NoMemory();
                 self->depth--;
                 return -1;
             }
-- 
cgit v1.2.1