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