rpc fixes

* set THIS properly so memory accounting works fine.
* also check for few pointers before dereferencing them.

Signed-off-by: Amar Tumballi <amar@gluster.com>
Signed-off-by: Anand V. Avati <avati@dev.gluster.com>

BUG: 1317 ()
URL: http://bugs.gluster.com/cgi-bin/bugzilla3/show_bug.cgi?id=1317

1 files changed, 20 insertions, 6 deletions

diff --git a/rpc/rpc-transport/socket/src/socket.c b/rpc/rpc-transport/socket/src/socket.c
index b017462a507..e544332c0b7 100644
--- a/rpc/rpc-transport/socket/src/socket.c
+++ b/rpc/rpc-transport/socket/src/socket.c
@@ -1497,8 +1497,12 @@ socket_event_handler (int fd, int idx, void *data,
         int               ret = 0;
 
         this = data;
+        THIS = this->xl;
         priv = this->private;
 
+        if (!priv)
+                return 0;
+
         pthread_mutex_lock (&priv->lock);
         {
                 priv->idx = idx;
@@ -1542,9 +1546,13 @@ socket_server_event_handler (int fd, int idx, void *data,
 	glusterfs_ctx_t         *ctx = NULL;
 
         this = data;
+        THIS = this->xl;
         priv = this->private;
 	ctx  = this->ctx;
 
+        if (!priv)
+                return 0;
+
         pthread_mutex_lock (&priv->lock);
         {
                 priv->idx = idx;
@@ -1584,7 +1592,6 @@ socket_server_event_handler (int fd, int idx, void *data,
                         if (!new_trans)
                                 goto unlock;
 
-                        new_trans->fini = this->fini;
                         new_trans->name = gf_strdup (this->name);
 
                         memcpy (&new_trans->peerinfo.sockaddr, &new_sockaddr,
@@ -1611,6 +1618,7 @@ socket_server_event_handler (int fd, int idx, void *data,
                         new_trans->init = this->init;
                         new_trans->fini = this->fini;
                         new_trans->ctx  = ctx;
+                        new_trans->xl   = this->xl;
                         new_trans->mydata = this->mydata;
                         new_trans->notify = this->notify;
                         new_priv = new_trans->private;
@@ -2323,13 +2331,19 @@ fini (rpc_transport_t *this)
 {
         socket_private_t *priv = this->private;
 
-        gf_log (this->name, GF_LOG_TRACE,
-                "transport %p destroyed", this);
+        this->private = NULL;
+        if (priv) {
+                gf_log (this->name, GF_LOG_TRACE,
+                        "transport %p destroyed", this);
 
-        pthread_mutex_destroy (&priv->lock);
+                pthread_mutex_destroy (&priv->lock);
+                GF_FREE (priv);
+        }
 
-        GF_FREE (this->name);
-        GF_FREE (priv);
+        if (this->name) {
+                GF_FREE (this->name);
+                this->name = NULL;
+        }
 }