Merge remote branch 'origin/mdb.master'

[openldap] / libraries / librewrite / session.c

diff --git a/libraries/librewrite/session.c b/libraries/librewrite/session.c
index 4fa58f876c5664e0f04af10563b938e7d30f8963..bc0627d38e80c5898081b4a3400d1176457bdcbd 100644 (file)
--- a/libraries/librewrite/session.c
+++ b/libraries/librewrite/session.c
@@ -1,7 +1,7 @@
 /* $OpenLDAP$ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *
- * Copyright 2000-2005 The OpenLDAP Foundation.
+ * Copyright 2000-2011 The OpenLDAP Foundation.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -256,6 +256,7 @@ rewrite_session_var_get(
 {
        struct rewrite_session *session;
        struct rewrite_var *var;
+       int rc = REWRITE_SUCCESS;
 
        assert( info != NULL );
        assert( cookie != NULL );
@@ -279,27 +280,22 @@ rewrite_session_var_get(
 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
        
        var = rewrite_var_find( session->ls_vars, name );
-       if ( var == NULL ) {
-               
-#ifdef USE_REWRITE_LDAP_PVT_THREADS
-               ldap_pvt_thread_rdwr_runlock( &session->ls_vars_mutex );
-#endif /* USE_REWRITE_LDAP_PVT_THREADS */
-
-               rewrite_session_return( info, session );
-
-               return REWRITE_ERR;
-       } else {
+       if ( var != NULL ) {
                value->bv_val = strdup( var->lv_value.bv_val );
                value->bv_len = var->lv_value.bv_len;
        }
-       
+
+       if ( var == NULL || value->bv_val == NULL ) {
+               rc = REWRITE_ERR;
+       }
+
 #ifdef USE_REWRITE_LDAP_PVT_THREADS
         ldap_pvt_thread_rdwr_runlock( &session->ls_vars_mutex );
 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 
        rewrite_session_return( info, session );
-       
-       return REWRITE_SUCCESS;
+
+       return rc;
 }
 
 static void
@@ -324,6 +320,9 @@ rewrite_session_clean( void *v_session )
 static void
 rewrite_session_free( void *v_session )
 {
+       struct rewrite_session  *session = (struct rewrite_session *)v_session;
+
+       ldap_pvt_thread_mutex_lock( &session->ls_mutex );
        rewrite_session_clean( v_session );
        free( v_session );
 }