2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4 * Copyright 2000-2014 The OpenLDAP Foundation.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted only as authorized by the OpenLDAP
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>.
16 * This work was initially developed by Pierangelo Masarati for
17 * inclusion in OpenLDAP Software.
22 #include "rewrite-int.h"
25 * Compares two cookies
33 const struct rewrite_session *s1, *s2;
35 s1 = ( const struct rewrite_session * )c1;
36 s2 = ( const struct rewrite_session * )c2;
40 assert( s1->ls_cookie != NULL );
41 assert( s2->ls_cookie != NULL );
43 return ( ( s1->ls_cookie < s2->ls_cookie ) ? -1 :
44 ( ( s1->ls_cookie > s2->ls_cookie ) ? 1 : 0 ) );
56 struct rewrite_session *s1, *s2;
58 s1 = ( struct rewrite_session * )c1;
59 s2 = ( struct rewrite_session * )c2;
63 assert( s1->ls_cookie != NULL );
64 assert( s2->ls_cookie != NULL );
66 assert( s1->ls_cookie != s2->ls_cookie );
68 return ( ( s1->ls_cookie == s2->ls_cookie ) ? -1 : 0 );
74 struct rewrite_session *
76 struct rewrite_info *info,
80 struct rewrite_session *session, tmp;
83 assert( info != NULL );
84 assert( cookie != NULL );
86 #ifdef USE_REWRITE_LDAP_PVT_THREADS
87 ldap_pvt_thread_rdwr_wlock( &info->li_cookies_mutex );
88 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
90 tmp.ls_cookie = ( void * )cookie;
91 session = ( struct rewrite_session * )avl_find( info->li_cookies,
92 ( caddr_t )&tmp, rewrite_cookie_cmp );
95 #ifdef USE_REWRITE_LDAP_PVT_THREADS
96 ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
97 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
101 session = calloc( sizeof( struct rewrite_session ), 1 );
102 if ( session == NULL ) {
105 session->ls_cookie = ( void * )cookie;
106 session->ls_count = 1;
108 #ifdef USE_REWRITE_LDAP_PVT_THREADS
109 if ( ldap_pvt_thread_mutex_init( &session->ls_mutex ) ) {
113 if ( ldap_pvt_thread_rdwr_init( &session->ls_vars_mutex ) ) {
114 ldap_pvt_thread_mutex_destroy( &session->ls_mutex );
118 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
120 rc = avl_insert( &info->li_cookies, ( caddr_t )session,
121 rewrite_cookie_cmp, rewrite_cookie_dup );
122 info->li_num_cookies++;
124 #ifdef USE_REWRITE_LDAP_PVT_THREADS
125 ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
126 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
129 #ifdef USE_REWRITE_LDAP_PVT_THREADS
130 ldap_pvt_thread_rdwr_destroy( &session->ls_vars_mutex );
131 ldap_pvt_thread_mutex_destroy( &session->ls_mutex );
132 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
144 struct rewrite_session *
145 rewrite_session_find(
146 struct rewrite_info *info,
150 struct rewrite_session *session, tmp;
152 assert( info != NULL );
153 assert( cookie != NULL );
155 tmp.ls_cookie = ( void * )cookie;
156 #ifdef USE_REWRITE_LDAP_PVT_THREADS
157 ldap_pvt_thread_rdwr_rlock( &info->li_cookies_mutex );
158 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
159 session = ( struct rewrite_session * )avl_find( info->li_cookies,
160 ( caddr_t )&tmp, rewrite_cookie_cmp );
161 #ifdef USE_REWRITE_LDAP_PVT_THREADS
163 ldap_pvt_thread_mutex_lock( &session->ls_mutex );
166 ldap_pvt_thread_rdwr_runlock( &info->li_cookies_mutex );
167 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
176 rewrite_session_return(
177 struct rewrite_info *info,
178 struct rewrite_session *session
181 assert( session != NULL );
183 ldap_pvt_thread_mutex_unlock( &session->ls_mutex );
187 * Defines and inits a var with session scope
190 rewrite_session_var_set_f(
191 struct rewrite_info *info,
198 struct rewrite_session *session;
199 struct rewrite_var *var;
201 assert( info != NULL );
202 assert( cookie != NULL );
203 assert( name != NULL );
204 assert( value != NULL );
206 session = rewrite_session_find( info, cookie );
207 if ( session == NULL ) {
208 session = rewrite_session_init( info, cookie );
209 if ( session == NULL ) {
213 #ifdef USE_REWRITE_LDAP_PVT_THREADS
214 ldap_pvt_thread_mutex_lock( &session->ls_mutex );
215 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
218 #ifdef USE_REWRITE_LDAP_PVT_THREADS
219 ldap_pvt_thread_rdwr_wlock( &session->ls_vars_mutex );
220 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
222 var = rewrite_var_find( session->ls_vars, name );
224 assert( var->lv_value.bv_val != NULL );
226 (void)rewrite_var_replace( var, value, flags );
229 var = rewrite_var_insert_f( &session->ls_vars, name, value, flags );
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 );
239 #ifdef USE_REWRITE_LDAP_PVT_THREADS
240 ldap_pvt_thread_rdwr_wunlock( &session->ls_vars_mutex );
241 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
243 rewrite_session_return( info, session );
245 return REWRITE_SUCCESS;
249 * Gets a var with session scope
252 rewrite_session_var_get(
253 struct rewrite_info *info,
259 struct rewrite_session *session;
260 struct rewrite_var *var;
261 int rc = REWRITE_SUCCESS;
263 assert( info != NULL );
264 assert( cookie != NULL );
265 assert( name != NULL );
266 assert( value != NULL );
268 value->bv_val = NULL;
271 if ( cookie == NULL ) {
275 session = rewrite_session_find( info, cookie );
276 if ( session == NULL ) {
280 #ifdef USE_REWRITE_LDAP_PVT_THREADS
281 ldap_pvt_thread_rdwr_rlock( &session->ls_vars_mutex );
282 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
284 var = rewrite_var_find( session->ls_vars, name );
286 value->bv_val = strdup( var->lv_value.bv_val );
287 value->bv_len = var->lv_value.bv_len;
290 if ( var == NULL || value->bv_val == NULL ) {
294 #ifdef USE_REWRITE_LDAP_PVT_THREADS
295 ldap_pvt_thread_rdwr_runlock( &session->ls_vars_mutex );
296 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
298 rewrite_session_return( info, session );
304 rewrite_session_clean( void *v_session )
306 struct rewrite_session *session = (struct rewrite_session *)v_session;
308 #ifdef USE_REWRITE_LDAP_PVT_THREADS
309 ldap_pvt_thread_rdwr_wlock( &session->ls_vars_mutex );
310 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
312 rewrite_var_delete( session->ls_vars );
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 */
323 rewrite_session_free( void *v_session )
325 struct rewrite_session *session = (struct rewrite_session *)v_session;
327 ldap_pvt_thread_mutex_lock( &session->ls_mutex );
328 rewrite_session_clean( v_session );
336 rewrite_session_delete(
337 struct rewrite_info *info,
341 struct rewrite_session *session, tmp = { 0 };
343 assert( info != NULL );
344 assert( cookie != NULL );
346 session = rewrite_session_find( info, cookie );
348 if ( session == NULL ) {
349 return REWRITE_SUCCESS;
352 if ( --session->ls_count > 0 ) {
353 rewrite_session_return( info, session );
354 return REWRITE_SUCCESS;
357 rewrite_session_clean( session );
359 #ifdef USE_REWRITE_LDAP_PVT_THREADS
360 ldap_pvt_thread_rdwr_wlock( &info->li_cookies_mutex );
361 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
363 assert( info->li_num_cookies > 0 );
364 info->li_num_cookies--;
367 * There is nothing to delete in the return value
369 tmp.ls_cookie = ( void * )cookie;
370 avl_delete( &info->li_cookies, ( caddr_t )&tmp, rewrite_cookie_cmp );
374 #ifdef USE_REWRITE_LDAP_PVT_THREADS
375 ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
376 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
378 return REWRITE_SUCCESS;
382 * Destroys the cookie tree
385 rewrite_session_destroy(
386 struct rewrite_info *info
391 assert( info != NULL );
393 #ifdef USE_REWRITE_LDAP_PVT_THREADS
394 ldap_pvt_thread_rdwr_wlock( &info->li_cookies_mutex );
395 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
398 * Should call per-session destruction routine ...
401 count = avl_free( info->li_cookies, rewrite_session_free );
402 info->li_cookies = NULL;
405 fprintf( stderr, "count = %d; num_cookies = %d\n",
406 count, info->li_num_cookies );
409 assert( count == info->li_num_cookies );
410 info->li_num_cookies = 0;
412 #ifdef USE_REWRITE_LDAP_PVT_THREADS
413 ldap_pvt_thread_rdwr_wunlock( &info->li_cookies_mutex );
414 #endif /* USE_REWRITE_LDAP_PVT_THREADS */
416 return REWRITE_SUCCESS;