From 35326727916ab3dcfbf57bc6ab05c31a97aebd74 Mon Sep 17 00:00:00 2001
From: Adam Dickmeiss <adam@indexdata.dk>
Date: Thu, 4 Mar 2010 14:01:41 +0100
Subject: [PATCH] Lock client while modifying session

---
 src/client.c     |   37 +++++++------------------------------
 src/client.h     |    1 +
 src/connection.c |    7 ++++++-
 3 files changed, 14 insertions(+), 31 deletions(-)

diff --git a/src/client.c b/src/client.c
index 5975b69..b4d323e 100644
--- a/src/client.c
+++ b/src/client.c
@@ -107,17 +107,6 @@ static const char *client_states[] = {
     "Client_Disconnected"
 };
 
-static void client_enter(struct client *cl)
-{
-    yaz_mutex_enter(cl->mutex);
-}
-
-static void client_leave(struct client *cl)
-{
-    yaz_mutex_leave(cl->mutex);
-}
-
-
 const char *client_get_state_str(struct client *cl)
 {
     return client_states[cl->state];
@@ -404,7 +393,7 @@ static void ingest_raw_record(struct client *cl, ZOOM_record rec)
     client_show_raw_dequeue(cl);
 }
 
-static void search_response(struct client *cl)
+void client_search_response(struct client *cl)
 {
     struct connection *co = cl->connection;
     struct session *se = cl->session;
@@ -428,12 +417,7 @@ static void search_response(struct client *cl)
     }
 }
 
-void client_search_response(struct client *cl)
-{
-    search_response(cl);
-}
-
-static void record_response(struct client *cl)
+void client_record_response(struct client *cl)
 {
     struct connection *co = cl->connection;
     ZOOM_connection link = connection_get_link(co);
@@ -512,11 +496,6 @@ static void record_response(struct client *cl)
     }
 }
 
-void client_record_response(struct client *cl)
-{
-    record_response(cl);
-}
-
 void client_start_search(struct client *cl)
 {
     struct session_database *sdb = client_get_database(cl);
@@ -635,12 +614,7 @@ int client_destroy(struct client *c)
             c->pquery = 0;
             xfree(c->cqlquery);
             c->cqlquery = 0;
-            c->hits = 12345678;
 
-#if 0            
-            if (c->connection)
-                connection_release(c->connection);
-#endif       
             ZOOM_resultset_destroy(c->resultset);
             yaz_mutex_destroy(&c->mutex);
             xfree(c);
@@ -813,8 +787,10 @@ int client_parse_query(struct client *cl, const char *query)
 
 void client_remove_from_session(struct client *c)
 {
-    struct session *se = c->session;
-    
+    struct session *se;
+    client_incref(c);
+
+    se = c->session;
     assert(se);
     if (se)
     {
@@ -828,6 +804,7 @@ void client_remove_from_session(struct client *c)
         c->session = 0;
         c->next = 0;
     }
+    client_destroy(c);
 }
 
 void client_set_session(struct client *cl, struct session *se)
diff --git a/src/client.h b/src/client.h
index e03efac..7eedec8 100644
--- a/src/client.h
+++ b/src/client.h
@@ -90,6 +90,7 @@ const char *client_get_url(struct client *cl);
 void client_set_maxrecs(struct client *cl, int v);
 void client_set_startrecs(struct client *cl, int v);
 void client_remove_from_session(struct client *c);
+void client_incref(struct client *c);
 
 #endif
 
diff --git a/src/connection.c b/src/connection.c
index 196a397..3fb061b 100644
--- a/src/connection.c
+++ b/src/connection.c
@@ -159,16 +159,20 @@ static struct connection *connection_create(struct client *cl,
 
 static void non_block_events(struct connection *co)
 {
-    struct client *cl = co->client;
     IOCHAN iochan = co->iochan;
     ZOOM_connection link = co->link;
     while (1)
     {
+        struct client *cl = co->client;
         int ev;
         int r = ZOOM_event_nonblock(1, &link);
         if (!r)
             break;
+        if (!cl)
+            continue;
         ev = ZOOM_connection_last_event(link);
+        
+        client_incref(cl);
 #if 0
         yaz_log(YLOG_LOG, "ZOOM_EVENT_%s", ZOOM_get_event_str(ev));
 #endif
@@ -214,6 +218,7 @@ static void non_block_events(struct connection *co)
             yaz_log(YLOG_LOG, "Unhandled event (%d) from %s",
                     ev, client_get_url(cl));
         }
+        client_destroy(cl);
     }
 }
 
-- 
1.7.10.4