From 46fe014378ffbdc49fdf1b80e715ee845661a4a6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ond=C5=99ej=20Kuzn=C3=ADk?= <ondra@mistotebe.net>
Date: Fri, 23 Jun 2017 11:09:03 +0100
Subject: [PATCH] Make sure operation stays alive when we process it

---
 servers/lloadd/upstream.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/servers/lloadd/upstream.c b/servers/lloadd/upstream.c
index 3b00ca5446..7557ca48de 100644
--- a/servers/lloadd/upstream.c
+++ b/servers/lloadd/upstream.c
@@ -401,6 +401,7 @@ handle_one_response( Connection *c )
         client = op->o_client;
         if ( client ) {
             CONNECTION_LOCK(client);
+            op->o_client_refcnt++;
             CONNECTION_UNLOCK_INCREF(client);
         }
         ldap_pvt_thread_mutex_unlock( &operation_mutex );
@@ -408,6 +409,10 @@ handle_one_response( Connection *c )
         if ( client ) {
             rc = handler( op, ber );
             CONNECTION_LOCK_DECREF(client);
+            op->o_client_refcnt--;
+            if ( !op->o_client_refcnt ) {
+                operation_destroy_from_client( op );
+            }
             CLIENT_UNLOCK_OR_DESTROY(client);
         } else {
             ber_free( ber, 1 );
@@ -415,7 +420,7 @@ handle_one_response( Connection *c )
 
         CONNECTION_LOCK_DECREF(c);
         op->o_upstream_refcnt--;
-        if ( !client || !op->o_upstream_live ) {
+        if ( !client || !op->o_upstream_refcnt ) {
             operation_destroy_from_upstream( op );
         }
     } else {