Navigation


source: freeDiameter/freeDiameter/p_expiry.c @ 16:013ce9851131

Last change on this file since 16:013ce9851131 was 16:013ce9851131, checked in by Sebastien Decugis <sdecugis@nict.go.jp>, 12 years ago

Started including TLS code

File size: 7.2 KB
Line 
1/*********************************************************************************************************
2* Software License Agreement (BSD License)                                                               *
3* Author: Sebastien Decugis <sdecugis@nict.go.jp>                                                        *
4*                                                                                                        *
5* Copyright (c) 2009, WIDE Project and NICT                                                              *
6* All rights reserved.                                                                                   *
7*                                                                                                        *
8* Redistribution and use of this software in source and binary forms, with or without modification, are  *
9* permitted provided that the following conditions are met:                                              *
10*                                                                                                        *
11* * Redistributions of source code must retain the above                                                 *
12*   copyright notice, this list of conditions and the                                                    *
13*   following disclaimer.                                                                                *
14*                                                                                                        *
15* * Redistributions in binary form must reproduce the above                                              *
16*   copyright notice, this list of conditions and the                                                    *
17*   following disclaimer in the documentation and/or other                                               *
18*   materials provided with the distribution.                                                            *
19*                                                                                                        *
20* * Neither the name of the WIDE Project or NICT nor the                                                 *
21*   names of its contributors may be used to endorse or                                                  *
22*   promote products derived from this software without                                                  *
23*   specific prior written permission of WIDE Project and                                                *
24*   NICT.                                                                                                *
25*                                                                                                        *
26* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED *
27* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A *
28* PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR *
29* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT     *
30* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS    *
31* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR *
32* TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF   *
33* ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.                                                             *
34*********************************************************************************************************/
35
36#include "fD.h"
37
38/* Delay for garbage collection of expired threads, in seconds */
39#define GC_TIME         60
40
41static pthread_t       exp_thr;
42static pthread_t       gc_thr;
43static struct fd_list  exp_list = FD_LIST_INITIALIZER( exp_list );
44static pthread_cond_t  exp_cnd  = PTHREAD_COND_INITIALIZER;
45static pthread_mutex_t exp_mtx  = PTHREAD_MUTEX_INITIALIZER;
46
47static void * gc_th_fct(void * arg)
48{
49        fd_log_threadname ( "Peers/garbage" );
50        TRACE_ENTRY( "" );
51       
52        do {
53                struct fd_list * li, purge = FD_LIST_INITIALIZER(purge);
54               
55                pthread_testcancel();
56                sleep(GC_TIME);
57               
58                /* Now check in the peers list if any peer can be deleted */
59                CHECK_FCT_DO( pthread_rwlock_wrlock(&fd_g_peers_rw), goto error );
60               
61                for (li = fd_g_peers.next; li != &fd_g_peers; li = li->next) {
62                        struct fd_peer * peer = (struct fd_peer *)li;
63                       
64                        if (peer->p_hdr.info.pi_state != STATE_ZOMBIE)
65                                continue;
66                       
67                        if (peer->p_hdr.info.pi_flags.exp == PI_EXP_NONE)
68                                continue; /* This peer was not supposed to expire, keep it in the list */
69                       
70                        /* Ok, the peer was expired, let's remove it */
71                        li = li->prev; /* to avoid breaking the loop */
72                        fd_list_unlink(&peer->p_hdr.chain);
73                        fd_list_insert_before(&purge, &peer->p_hdr.chain);
74                }
75
76                CHECK_FCT_DO( pthread_rwlock_unlock(&fd_g_peers_rw), goto error );
77               
78                /* Now delete peers that are in the purge list */
79                while (!FD_IS_LIST_EMPTY(&purge)) {
80                        struct fd_peer * peer = (struct fd_peer *)(purge.next);
81                        fd_list_unlink(&peer->p_hdr.chain);
82                        TRACE_DEBUG(INFO, "Garbage Collect: delete zombie peer '%s'", peer->p_hdr.info.pi_diamid);
83                        CHECK_FCT_DO( fd_peer_free(&peer), /* Continue... what else to do ? */ );
84                }
85        } while (1);
86       
87error:
88        TRACE_DEBUG(INFO, "An error occurred in peers module! GC thread is terminating...");
89        ASSERT(0);
90        CHECK_FCT_DO(fd_event_send(fd_g_config->cnf_main_ev, FDEV_TERMINATE, NULL), );
91        return NULL;
92}
93
94
95static void * exp_th_fct(void * arg)
96{
97        fd_log_threadname ( "Peers/expire" );
98        TRACE_ENTRY( "" );
99       
100        CHECK_POSIX_DO( pthread_mutex_lock(&exp_mtx),  goto error );
101        pthread_cleanup_push( fd_cleanup_mutex, &exp_mtx );
102       
103        do {
104                struct timespec now;
105                struct fd_peer * first;
106               
107                /* Check if there are expiring sessions available */
108                if (FD_IS_LIST_EMPTY(&exp_list)) {
109                        /* Just wait for a change or cancelation */
110                        CHECK_POSIX_DO( pthread_cond_wait( &exp_cnd, &exp_mtx ), goto error );
111                        /* Restart the loop on wakeup */
112                        continue;
113                }
114               
115                /* Get the pointer to the peer that expires first */
116                first = (struct fd_peer *)(exp_list.next->o);
117                ASSERT( CHECK_PEER(first) );
118               
119                /* Get the current time */
120                CHECK_SYS_DO(  clock_gettime(CLOCK_REALTIME, &now),  goto error  );
121
122                /* If first peer is not expired, we just wait until it happens */
123                if ( TS_IS_INFERIOR( &now, &first->p_exp_timer ) ) {
124                       
125                        CHECK_POSIX_DO2(  pthread_cond_timedwait( &exp_cnd, &exp_mtx, &first->p_exp_timer ), 
126                                        ETIMEDOUT, /* ETIMEDOUT is a normal error, continue */,
127                                        /* on other error, */ goto error );
128       
129                        /* on wakeup, loop */
130                        continue;
131                }
132               
133                /* Now, the first peer in the list is expired; signal it */
134                fd_list_unlink( &first->p_expiry );
135                CHECK_FCT_DO( fd_event_send(first->p_events, FDEVP_TERMINATE, NULL), goto error );
136               
137        } while (1);
138       
139        pthread_cleanup_pop( 1 );
140error:
141        TRACE_DEBUG(INFO, "An error occurred in peers module! Expiry thread is terminating...");
142        ASSERT(0);
143        CHECK_FCT_DO(fd_event_send(fd_g_config->cnf_main_ev, FDEV_TERMINATE, NULL), );
144        return NULL;
145}
146
147/* Initialize peers expiry mechanism */
148int fd_p_expi_init(void)
149{
150        TRACE_ENTRY();
151        CHECK_FCT( pthread_create( &exp_thr, NULL, exp_th_fct, NULL ) );
152        CHECK_FCT( pthread_create( &gc_thr,  NULL, gc_th_fct,  NULL ) );
153        return 0;
154}
155
156/* Finish peers expiry mechanism */
157int fd_p_expi_fini(void)
158{
159        CHECK_FCT_DO( fd_thr_term(&exp_thr), );
160        CHECK_POSIX( pthread_mutex_lock(&exp_mtx) );
161       
162        while (!FD_IS_LIST_EMPTY(&exp_list)) {
163                struct fd_peer * peer = (struct fd_peer *)(exp_list.next->o);
164                fd_list_unlink(&peer->p_expiry );
165        }
166       
167        CHECK_POSIX( pthread_mutex_unlock(&exp_mtx) );
168        CHECK_FCT_DO( fd_thr_term(&gc_thr), );
169        return 0;
170}
171
172/* Add / requeue a peer in the expiry list */
173int fd_p_expi_update(struct fd_peer * peer )
174{
175        TRACE_ENTRY("%p", peer);
176        CHECK_PARAMS( CHECK_PEER(peer) );
177       
178        CHECK_POSIX( pthread_mutex_lock(&exp_mtx) );
179       
180        fd_list_unlink(&peer->p_expiry );
181       
182        /* if peer expires */
183        if (peer->p_hdr.info.pi_flags.exp) {
184                struct fd_list * li;
185               
186                /* update the p_exp_timer value */
187                CHECK_SYS(  clock_gettime(CLOCK_REALTIME, &peer->p_exp_timer)  );
188                peer->p_exp_timer.tv_sec += peer->p_hdr.info.pi_lft;
189               
190                /* add to the expiry list in appropriate position (probably around the end) */
191                for (li = exp_list.prev; li != &exp_list; li = li->prev) {
192                        struct fd_peer * p = (struct fd_peer *)(li->o);
193                        if (TS_IS_INFERIOR( &p->p_exp_timer, &peer->p_exp_timer ) )
194                                break;
195                }
196               
197                fd_list_insert_after(li, &peer->p_expiry);
198               
199                /* signal the expiry thread if we added in first position */
200                if (li == &exp_list) {
201                        CHECK_POSIX( pthread_cond_signal(&exp_cnd) );
202                }
203        }
204       
205        CHECK_POSIX( pthread_mutex_unlock(&exp_mtx) );
206        return 0;
207}
208
Note: See TracBrowser for help on using the repository browser.