git.sur5r.net Git - openldap/blob - libraries/librewrite/session.c

   1 /* $OpenLDAP$ */
   2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   3  *
   4  * Copyright 2000-2016 The OpenLDAP Foundation.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted only as authorized by the OpenLDAP
   9  * Public License.
  10  *
  11  * A copy of this license is available in the file LICENSE in the
  12  * top-level directory of the distribution or, alternatively, at
  13  * <http://www.OpenLDAP.org/license.html>.
  14  */
  15 /* ACKNOWLEDGEMENT:
  16  * This work was initially developed by Pierangelo Masarati for
  17  * inclusion in OpenLDAP Software.
  18  */
  19
  20 #include <portable.h>
  21
  22 #include "rewrite-int.h"
  23
  24 /*
  25  * Compares two cookies
  26  */
  27 static int
  28 rewrite_cookie_cmp(
  29                 const void *c1,
  30                 const void *c2
  31 )
  32 {
  33         const struct rewrite_session *s1, *s2;
  34
  35         s1 = ( const struct rewrite_session * )c1;
  36         s2 = ( const struct rewrite_session * )c2;
  37
  38         assert( s1 != NULL );
  39         assert( s2 != NULL );
  40         assert( s1->ls_cookie != NULL );
  41         assert( s2->ls_cookie != NULL );
  42
  43         return ( ( s1->ls_cookie < s2->ls_cookie ) ? -1 :
  44                         ( ( s1->ls_cookie > s2->ls_cookie ) ? 1 : 0 ) );
  45 }
  46
  47 /*
  48  * Duplicate cookies?
  49  */
  50 static int
  51 rewrite_cookie_dup(
  52                 void *c1,
  53                 void *c2
  54 )
  55 {
  56         struct rewrite_session *s1, *s2;
  57
  58         s1 = ( struct rewrite_session * )c1;
  59         s2 = ( struct rewrite_session * )c2;
  60
  61         assert( s1 != NULL );
  62         assert( s2 != NULL );
  63         assert( s1->ls_cookie != NULL );
  64         assert( s2->ls_cookie != NULL );
  65
  66         assert( s1->ls_cookie != s2->ls_cookie );
  67
  68         return ( ( s1->ls_cookie == s2->ls_cookie ) ? -1 : 0 );
  69 }
  70
  71 /*
  72  * Inits a session
  73  */
  74 struct rewrite_session *
  75 rewrite_session_init(
  76                 struct rewrite_info *info,
  77                 const void *cookie
  78 )
  79 {
  80         struct rewrite_session  *session, tmp;
  81         int                     rc;
  82
  83         assert( info != NULL );
  84         assert( cookie != NULL );
  85
  86 #ifdef USE_REWRITE_LDAP_PVT_THREADS
  87         ldap_pvt_thread_rdwr_wlock( &info->li_cookies_mutex );
  88 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
  89
  90         tmp.ls_cookie = ( void * )cookie;
  91         session = ( struct rewrite_session * )avl_find( info->li_cookies,
  92                         ( caddr_t )&tmp, rewrite_cookie_cmp );
  93         if ( session ) {
  94                 session->ls_count++;
  95 #ifdef USE_REWRITE_LDAP_PVT_THREADS
  96                 ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
  97 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
  98                 return session;
  99         }
 100
 101         session = calloc( sizeof( struct rewrite_session ), 1 );
 102         if ( session == NULL ) {
 103 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 104                 ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
 105 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 106                 return NULL;
 107         }
 108         session->ls_cookie = ( void * )cookie;
 109         session->ls_count = 1;
 110
 111 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 112         if ( ldap_pvt_thread_mutex_init( &session->ls_mutex ) ) {
 113                 free( session );
 114                 ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
 115                 return NULL;
 116         }
 117         if ( ldap_pvt_thread_rdwr_init( &session->ls_vars_mutex ) ) {
 118                 ldap_pvt_thread_mutex_destroy( &session->ls_mutex );
 119                 free( session );
 120                 ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
 121                 return NULL;
 122         }
 123 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 124
 125         rc = avl_insert( &info->li_cookies, ( caddr_t )session,
 126                         rewrite_cookie_cmp, rewrite_cookie_dup );
 127         info->li_num_cookies++;
 128
 129 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 130         ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
 131 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 132
 133         if ( rc != 0 ) {
 134 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 135                 ldap_pvt_thread_rdwr_destroy( &session->ls_vars_mutex );
 136                 ldap_pvt_thread_mutex_destroy( &session->ls_mutex );
 137 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 138
 139                 free( session );
 140                 return NULL;
 141         }
 142
 143         return session;
 144 }
 145
 146 /*
 147  * Fetches a session
 148  */
 149 struct rewrite_session *
 150 rewrite_session_find(
 151                 struct rewrite_info *info,
 152                 const void *cookie
 153 )
 154 {
 155         struct rewrite_session *session, tmp;
 156
 157         assert( info != NULL );
 158         assert( cookie != NULL );
 159
 160         tmp.ls_cookie = ( void * )cookie;
 161 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 162         ldap_pvt_thread_rdwr_rlock( &info->li_cookies_mutex );
 163 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 164         session = ( struct rewrite_session * )avl_find( info->li_cookies,
 165                         ( caddr_t )&tmp, rewrite_cookie_cmp );
 166 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 167         if ( session ) {
 168                 ldap_pvt_thread_mutex_lock( &session->ls_mutex );
 169                 session->ls_count++;
 170         }
 171         ldap_pvt_thread_rdwr_runlock( &info->li_cookies_mutex );
 172 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 173
 174         return session;
 175 }
 176
 177 /*
 178  * Returns a session
 179  */
 180 void
 181 rewrite_session_return(
 182                 struct rewrite_info *info,
 183                 struct rewrite_session *session
 184 )
 185 {
 186         assert( session != NULL );
 187         session->ls_count--;
 188         ldap_pvt_thread_mutex_unlock( &session->ls_mutex );
 189 }
 190
 191 /*
 192  * Defines and inits a var with session scope
 193  */
 194 int
 195 rewrite_session_var_set_f(
 196                 struct rewrite_info *info,
 197                 const void *cookie,
 198                 const char *name,
 199                 const char *value,
 200                 int flags
 201 )
 202 {
 203         struct rewrite_session *session;
 204         struct rewrite_var *var;
 205
 206         assert( info != NULL );
 207         assert( cookie != NULL );
 208         assert( name != NULL );
 209         assert( value != NULL );
 210
 211         session = rewrite_session_find( info, cookie );
 212         if ( session == NULL ) {
 213                 session = rewrite_session_init( info, cookie );
 214                 if ( session == NULL ) {
 215                         return REWRITE_ERR;
 216                 }
 217
 218 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 219                 ldap_pvt_thread_mutex_lock( &session->ls_mutex );
 220 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 221         }
 222
 223 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 224         ldap_pvt_thread_rdwr_wlock( &session->ls_vars_mutex );
 225 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 226
 227         var = rewrite_var_find( session->ls_vars, name );
 228         if ( var != NULL ) {
 229                 assert( var->lv_value.bv_val != NULL );
 230
 231                 (void)rewrite_var_replace( var, value, flags );
 232
 233         } else {
 234                 var = rewrite_var_insert_f( &session->ls_vars, name, value, flags );
 235                 if ( var == NULL ) {
 236 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 237                         ldap_pvt_thread_rdwr_wunlock( &session->ls_vars_mutex );
 238 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 239                         rewrite_session_return( info, session );
 240                         return REWRITE_ERR;
 241                 }
 242         }
 243
 244 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 245         ldap_pvt_thread_rdwr_wunlock( &session->ls_vars_mutex );
 246 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 247
 248         rewrite_session_return( info, session );
 249
 250         return REWRITE_SUCCESS;
 251 }
 252
 253 /*
 254  * Gets a var with session scope
 255  */
 256 int
 257 rewrite_session_var_get(
 258                 struct rewrite_info *info,
 259                 const void *cookie,
 260                 const char *name,
 261                 struct berval *value
 262 )
 263 {
 264         struct rewrite_session *session;
 265         struct rewrite_var *var;
 266         int rc = REWRITE_SUCCESS;
 267
 268         assert( info != NULL );
 269         assert( cookie != NULL );
 270         assert( name != NULL );
 271         assert( value != NULL );
 272
 273         value->bv_val = NULL;
 274         value->bv_len = 0;
 275
 276         if ( cookie == NULL ) {
 277                 return REWRITE_ERR;
 278         }
 279
 280         session = rewrite_session_find( info, cookie );
 281         if ( session == NULL ) {
 282                 return REWRITE_ERR;
 283         }
 284
 285 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 286         ldap_pvt_thread_rdwr_rlock( &session->ls_vars_mutex );
 287 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 288
 289         var = rewrite_var_find( session->ls_vars, name );
 290         if ( var != NULL ) {
 291                 value->bv_val = strdup( var->lv_value.bv_val );
 292                 value->bv_len = var->lv_value.bv_len;
 293         }
 294
 295         if ( var == NULL || value->bv_val == NULL ) {
 296                 rc = REWRITE_ERR;
 297         }
 298
 299 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 300         ldap_pvt_thread_rdwr_runlock( &session->ls_vars_mutex );
 301 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 302
 303         rewrite_session_return( info, session );
 304
 305         return rc;
 306 }
 307
 308 static void
 309 rewrite_session_clean( void *v_session )
 310 {
 311         struct rewrite_session  *session = (struct rewrite_session *)v_session;
 312
 313 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 314         ldap_pvt_thread_rdwr_wlock( &session->ls_vars_mutex );
 315 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 316
 317         rewrite_var_delete( session->ls_vars );
 318
 319 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 320         ldap_pvt_thread_rdwr_wunlock( &session->ls_vars_mutex );
 321         ldap_pvt_thread_rdwr_destroy( &session->ls_vars_mutex );
 322         ldap_pvt_thread_mutex_unlock( &session->ls_mutex );
 323         ldap_pvt_thread_mutex_destroy( &session->ls_mutex );
 324 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 325 }
 326
 327 static void
 328 rewrite_session_free( void *v_session )
 329 {
 330         struct rewrite_session  *session = (struct rewrite_session *)v_session;
 331
 332         ldap_pvt_thread_mutex_lock( &session->ls_mutex );
 333         rewrite_session_clean( v_session );
 334         free( v_session );
 335 }
 336
 337 /*
 338  * Deletes a session
 339  */
 340 int
 341 rewrite_session_delete(
 342                 struct rewrite_info *info,
 343                 const void *cookie
 344 )
 345 {
 346         struct rewrite_session *session, tmp = { 0 };
 347
 348         assert( info != NULL );
 349         assert( cookie != NULL );
 350
 351         session = rewrite_session_find( info, cookie );
 352
 353         if ( session == NULL ) {
 354                 return REWRITE_SUCCESS;
 355         }
 356
 357         if ( --session->ls_count > 0 ) {
 358                 rewrite_session_return( info, session );
 359                 return REWRITE_SUCCESS;
 360         }
 361
 362         rewrite_session_clean( session );
 363
 364 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 365         ldap_pvt_thread_rdwr_wlock( &info->li_cookies_mutex );
 366 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 367
 368         assert( info->li_num_cookies > 0 );
 369         info->li_num_cookies--;
 370
 371         /*
 372          * There is nothing to delete in the return value
 373          */
 374         tmp.ls_cookie = ( void * )cookie;
 375         avl_delete( &info->li_cookies, ( caddr_t )&tmp, rewrite_cookie_cmp );
 376
 377         free( session );
 378
 379 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 380         ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
 381 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 382
 383         return REWRITE_SUCCESS;
 384 }
 385
 386 /*
 387  * Destroys the cookie tree
 388  */
 389 int
 390 rewrite_session_destroy(
 391                 struct rewrite_info *info
 392 )
 393 {
 394         int count;
 395
 396         assert( info != NULL );
 397
 398 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 399         ldap_pvt_thread_rdwr_wlock( &info->li_cookies_mutex );
 400 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 401
 402         /*
 403          * Should call per-session destruction routine ...
 404          */
 405
 406         count = avl_free( info->li_cookies, rewrite_session_free );
 407         info->li_cookies = NULL;
 408
 409 #if 0
 410         fprintf( stderr, "count = %d; num_cookies = %d\n",
 411                         count, info->li_num_cookies );
 412 #endif
 413
 414         assert( count == info->li_num_cookies );
 415         info->li_num_cookies = 0;
 416
 417 #ifdef USE_REWRITE_LDAP_PVT_THREADS
 418         ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
 419 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
 420
 421         return REWRITE_SUCCESS;
 422 }
 423