From 5e8400b3b148d78861152bd551800b1e93a58155 Mon Sep 17 00:00:00 2001
From: dimitri staessens <dimitri.staessens@intec.ugent.be>
Date: Thu, 27 Oct 2016 15:25:08 +0200
Subject: irmd: Fix cleanup of pending flows

---
 src/irmd/irm_flow.c | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'src/irmd/irm_flow.c')

diff --git a/src/irmd/irm_flow.c b/src/irmd/irm_flow.c
index dc5d22d8..ac854832 100644
--- a/src/irmd/irm_flow.c
+++ b/src/irmd/irm_flow.c
@@ -26,6 +26,7 @@
 
 #include <stdlib.h>
 #include <stdbool.h>
+#include <assert.h>
 
 struct irm_flow * irm_flow_create()
 {
@@ -59,6 +60,8 @@ struct irm_flow * irm_flow_create()
 
 void irm_flow_destroy(struct irm_flow * f)
 {
+        assert(f);
+
         pthread_mutex_lock(&f->state_lock);
 
         if (f->state == FLOW_DESTROY) {
@@ -91,6 +94,8 @@ enum flow_state irm_flow_get_state(struct irm_flow * f)
 {
         enum flow_state state;
 
+        assert(f);
+
         pthread_mutex_lock(&f->state_lock);
 
         state = f->state;
@@ -102,6 +107,10 @@ enum flow_state irm_flow_get_state(struct irm_flow * f)
 
 void irm_flow_set_state(struct irm_flow * f, enum flow_state state)
 {
+        assert(f);
+        assert(state != FLOW_NULL);
+        assert(state != FLOW_DESTROY);
+
         pthread_mutex_lock(&f->state_lock);
 
         f->state = state;
@@ -112,6 +121,10 @@ void irm_flow_set_state(struct irm_flow * f, enum flow_state state)
 
 enum flow_state irm_flow_wait_state(struct irm_flow * f, enum flow_state state)
 {
+        assert(f);
+        assert(state != FLOW_NULL);
+        assert(state != FLOW_DESTROY);
+
         pthread_mutex_lock(&f->state_lock);
 
         while (!(f->state == state || f->state == FLOW_DESTROY))
-- 
cgit v1.2.3