From c42d3d9f7ac086b5e012d9c2ee64f3b024c9ed17 Mon Sep 17 00:00:00 2001
From: Markus Makela <markus.makela@mariadb.com>
Date: Tue, 23 Jun 2015 17:05:31 +0300
Subject: [PATCH] Added missing NULL checks.

---
 server/core/dcb.c     |  3 +++
 server/core/service.c | 16 ++++++++++++----
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/server/core/dcb.c b/server/core/dcb.c
index 7f3651953..07bb1dc93 100644
--- a/server/core/dcb.c
+++ b/server/core/dcb.c
@@ -2423,6 +2423,9 @@ static bool dcb_set_state_nomutex(
                 
         case DCB_STATE_NOPOLLING:
                 switch (new_state) {
+		    /** Stopped services which are restarting will go from
+		     *  DCB_STATE_NOPOLLING to DCB_STATE_LISTENING.*/
+			case DCB_STATE_LISTENING:
 			case DCB_STATE_ZOMBIE: /*< fall through */
 				dcb->state = new_state;
 			case DCB_STATE_POLLING: /*< ok to try but state can't change */
diff --git a/server/core/service.c b/server/core/service.c
index 7966d33ec..24d51542d 100644
--- a/server/core/service.c
+++ b/server/core/service.c
@@ -534,10 +534,14 @@ int		listeners = 0;
 	port = service->ports;
 	while (port)
 	{
-	    if(port->listener && poll_remove_dcb(port->listener) == 0)
+	    if(port->listener &&
+	     port->listener->session->state == SESSION_STATE_LISTENER)
 	    {
-		port->listener->session->state = SESSION_STATE_LISTENER_STOPPED;
-		listeners++;
+		if(poll_remove_dcb(port->listener) == 0)
+		{
+		    port->listener->session->state = SESSION_STATE_LISTENER_STOPPED;
+		    listeners++;
+		}
 	    }
 	    port = port->next;
 	}
@@ -563,11 +567,15 @@ int		listeners = 0;
 	port = service->ports;
 	while (port)
 	{
-		if(port->listener && poll_add_dcb(port->listener) == 0)
+	    if(port->listener &&
+	     port->listener->session->state == SESSION_STATE_LISTENER_STOPPED)
+	    {
+		if(poll_add_dcb(port->listener) == 0)
 		{
 		    port->listener->session->state = SESSION_STATE_LISTENER;
 		    listeners++;
 		}
+	    }
 	    port = port->next;
 	}
 	service->state = SERVICE_STATE_STARTED;