WebSVN – HelenOS – Diff – /branches/rcu/kernel/generic/src/time/timeout.c


/*
 * Copyright (C) 2001-2004 Jakub Jermar
 * Copyright (C) 2007 Vojtech Mencl
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
 
/** @addtogroup time
 * @{
 */
 
/**
 * @file
 * @brief   Timeout management functions.
 */
 
#include <time/timeout.h>
#include <arch/types.h>
#include <config.h>
#include <panic.h>
#include <synch/spinlock.h>
#include <func.h>
#include <cpu.h>
#include <arch/asm.h>
#include <arch.h>
 
 
/** Initialize timeouts
 *
 * Initialize kernel timeouts.
 *
 */
void timeout_init(void)
{
    spinlock_initialize(&CPU->timeoutlock, "timeout_lock");
 
#if defined CONFIG_TIMEOUT_AVL_TREE
    avltree_create(&CPU->timeout_active_tree);
#elif defined CONFIG_TIMEOUT_EXTAVL_TREE
    extavltree_create(&CPU->timeout_active_tree);
#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE
    extavlreltree_create(&CPU->timeout_active_tree);
#else
    list_initialize(&CPU->timeout_active_head);
#endif
}
 
 
/** Reinitialize timeout
 *
 * Initialize all members except the lock.
 *
 * @param t Timeout to be initialized.
 *
 */
void timeout_reinitialize(timeout_t *t)
{
    t->cpu = NULL;
    t->handler = NULL;
    t->arg = NULL;
 
#if defined CONFIG_TIMEOUT_AVL_TREE
    avltree_node_initialize(&t->node);
#elif defined CONFIG_TIMEOUT_EXTAVL_TREE
    extavltree_node_initialize(&t->node);
#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE
    extavlreltree_node_initialize(&t->node);
#else
    t->ticks = 0;
    link_initialize(&t->link);
#endif
}
 
 
/** Initialize timeout
 *
 * Initialize all members including the lock.
 *
 * @param t Timeout to be initialized.
 *
 */
void timeout_initialize(timeout_t *t)
{
    spinlock_initialize(&t->lock, "timeout_t_lock");
    timeout_reinitialize(t);
}
 
#if defined CONFIG_TIMEOUT_AVL_TREE || \
    defined CONFIG_TIMEOUT_EXTAVL_TREE || \
    defined CONFIG_TIMEOUT_EXTAVLREL_TREE
 
/** Register timeout
 *
 * Insert timeout handler f (with argument arg)
 * to timeout ExtAVL tree and make it execute in
 * time microseconds (or slightly more).
 *
 * @param t    Timeout structure.
 * @param time Number of usec in the future to execute
 *             the handler.
 * @param f    Timeout handler function.
 * @param arg  Timeout handler argument.
 *
 */
void timeout_register(timeout_t *t, uint64_t time, timeout_handler_t f, void *arg)
{
    ipl_t ipl;
 
    ipl = interrupts_disable();
    spinlock_lock(&CPU->timeoutlock);
    spinlock_lock(&t->lock);
 
    if (t->cpu)
        panic("t->cpu != 0");
   
    t->cpu = CPU;
    t->handler = f;
    t->arg = arg;
    t->node.key = us2ticks(time);
 
    /*
     * Put timeout into tree structure.
     */
#if defined CONFIG_TIMEOUT_AVL_TREE
    avltree_insert(&CPU->timeout_active_tree,&t->node);
#elif defined CONFIG_TIMEOUT_EXTAVL_TREE
    extavltree_insert(&CPU->timeout_active_tree,&t->node);
#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE
    extavlreltree_insert(&CPU->timeout_active_tree,&t->node);
#endif
 
    spinlock_unlock(&t->lock);
    spinlock_unlock(&CPU->timeoutlock);
    interrupts_restore(ipl);
}
 
 
/** Unregister timeout
 *
 * Remove timeout from timeout ExtAVL tree structure.
 *
 * @param t Timeout to unregister.
 *
 * @return true on success, false on failure.
 */
bool timeout_unregister(timeout_t *t)
{
    ipl_t ipl;
 
grab_locks:
    ipl = interrupts_disable();
    spinlock_lock(&t->lock);
    if (!t->cpu) {
        spinlock_unlock(&t->lock);
        interrupts_restore(ipl);
        return false;
    }
    if (!spinlock_trylock(&t->cpu->timeoutlock)) {
        spinlock_unlock(&t->lock);
        interrupts_restore(ipl);       
        goto grab_locks;
    }
    /*
     * Now we know for sure that t hasn't been activated yet
     * and is lurking in t->cpu->timeout_active_head queue.
     */
 
    /*
     * Delete timeout from tree structure.
     */
#if defined CONFIG_TIMEOUT_AVL_TREE
    avltree_delete(&CPU->timeout_active_tree,&t->node);
#elif defined CONFIG_TIMEOUT_EXTAVL_TREE
    extavltree_delete(&CPU->timeout_active_tree,&t->node);
#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE
    extavlreltree_delete(&CPU->timeout_active_tree,&t->node);
#endif
 
    spinlock_unlock(&t->cpu->timeoutlock);
 
    timeout_reinitialize(t);
    spinlock_unlock(&t->lock);
 
    interrupts_restore(ipl);
    return true;
}
 
#else
 
/** Register timeout
 *
 * Insert timeout handler f (with argument arg)
 * to timeout list and make it execute in
 * time microseconds (or slightly more).
 *
 * @param t    Timeout structure.
 * @param time Number of usec in the future to execute
 *             the handler.
 * @param f    Timeout handler function.
 * @param arg  Timeout handler argument.
 *
 */
void timeout_register(timeout_t *t, uint64_t time, timeout_handler_t f, void *arg)
{
    timeout_t *hlp = NULL;
    link_t *l, *m;
    ipl_t ipl;
    uint64_t sum;
 
    ipl = interrupts_disable();
    spinlock_lock(&CPU->timeoutlock);
    spinlock_lock(&t->lock);
 
    if (t->cpu)
        panic("t->cpu != 0");
 
    t->cpu = CPU;
    t->ticks = us2ticks(time);
   
    t->handler = f;
    t->arg = arg;
 
    /*
     * Insert t into the active timeouts list according to t->ticks.
     */
    sum = 0;
    l = CPU->timeout_active_head.next;
    while (l != &CPU->timeout_active_head) {
        hlp = list_get_instance(l, timeout_t, link);
        spinlock_lock(&hlp->lock);
        if (t->ticks < sum + hlp->ticks) {
            spinlock_unlock(&hlp->lock);
            break;
        }
        sum += hlp->ticks;
        spinlock_unlock(&hlp->lock);
        l = l->next;
    }
 
    m = l->prev;
    list_prepend(&t->link, m); /* avoid using l->prev */
 
    /*
     * Adjust t->ticks according to ticks accumulated in h's predecessors.
     */
    t->ticks -= sum;
 
    /*
     * Decrease ticks of t's immediate succesor by t->ticks.
     */
    if (l != &CPU->timeout_active_head) {
        spinlock_lock(&hlp->lock);
        hlp->ticks -= t->ticks;
        spinlock_unlock(&hlp->lock);
    }
 
    spinlock_unlock(&t->lock);
    spinlock_unlock(&CPU->timeoutlock);
    interrupts_restore(ipl);
}
 
 
/** Unregister timeout
 *
 * Remove timeout from timeout list.
 *
 * @param t Timeout to unregister.
 *
 * @return true on success, false on failure.
 */
bool timeout_unregister(timeout_t *t)
{
    timeout_t *hlp;
    link_t *l;
    ipl_t ipl;
 
grab_locks:
    ipl = interrupts_disable();
    spinlock_lock(&t->lock);
    if (!t->cpu) {
        spinlock_unlock(&t->lock);
        interrupts_restore(ipl);
        return false;
    }
    if (!spinlock_trylock(&t->cpu->timeoutlock)) {
        spinlock_unlock(&t->lock);
        interrupts_restore(ipl);       
        goto grab_locks;
    }
   
    /*
     * Now we know for sure that t hasn't been activated yet
     * and is lurking in t->cpu->timeout_active_head queue.
     */
 
    l = t->link.next;
    if (l != &t->cpu->timeout_active_head) {
        hlp = list_get_instance(l, timeout_t, link);
        spinlock_lock(&hlp->lock);
        hlp->ticks += t->ticks;
        spinlock_unlock(&hlp->lock);
    }
   
    list_remove(&t->link);
    spinlock_unlock(&t->cpu->timeoutlock);
 
    timeout_reinitialize(t);
    spinlock_unlock(&t->lock);
 
    interrupts_restore(ipl);
    return true;
}
 
#endif
/** @}
 */
 

Rev 2421	Rev 2431
1	/*	1	/*
2	* Copyright (C) 2001-2004 Jakub Jermar	2	* Copyright (C) 2001-2004 Jakub Jermar
-		3	* Copyright (C) 2007 Vojtech Mencl
3	* All rights reserved.	4	* All rights reserved.
4	*	5	*
5	* Redistribution and use in source and binary forms, with or without	6	* Redistribution and use in source and binary forms, with or without
6	* modification, are permitted provided that the following conditions	7	* modification, are permitted provided that the following conditions
7	* are met:	8	* are met:
8	*	9	*
9	* - Redistributions of source code must retain the above copyright	10	* - Redistributions of source code must retain the above copyright
10	* notice, this list of conditions and the following disclaimer.	11	* notice, this list of conditions and the following disclaimer.
11	* - Redistributions in binary form must reproduce the above copyright	12	* - Redistributions in binary form must reproduce the above copyright
12	* notice, this list of conditions and the following disclaimer in the	13	* notice, this list of conditions and the following disclaimer in the
13	* documentation and/or other materials provided with the distribution.	14	* documentation and/or other materials provided with the distribution.
14	* - The name of the author may not be used to endorse or promote products	15	* - The name of the author may not be used to endorse or promote products
15	* derived from this software without specific prior written permission.	16	* derived from this software without specific prior written permission.
16	*	17	*
17	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR	18	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES	19	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.	20	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,	21	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT	22	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,	23	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY	24	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT	25	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF	26	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.	27	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27	*/	28	*/
28		29
29	/** @addtogroup time	30	/** @addtogroup time
30	* @{	31	* @{
31	*/	32	*/
32		33
33	/**	34	/**
34	* @file	35	* @file
35	* @brief Timeout management functions.	36	* @brief Timeout management functions.
36	*/	37	*/
37		38
38	#include <time/timeout.h>	39	#include <time/timeout.h>
39	#include <arch/types.h>	40	#include <arch/types.h>
40	#include <config.h>	41	#include <config.h>
41	#include <panic.h>	42	#include <panic.h>
42	#include <synch/spinlock.h>	43	#include <synch/spinlock.h>
43	#include <func.h>	44	#include <func.h>
44	#include <cpu.h>	45	#include <cpu.h>
45	#include <arch/asm.h>	46	#include <arch/asm.h>
46	#include <arch.h>	47	#include <arch.h>
47		48
48		49
49	/** Initialize timeouts	50	/** Initialize timeouts
50	*	51	*
51	* Initialize kernel timeouts.	52	* Initialize kernel timeouts.
52	*	53	*
53	*/	54	*/
54	void timeout_init(void)	55	void timeout_init(void)
55	{	56	{
56	spinlock_initialize(&CPU->timeoutlock, "timeout_lock");	57	spinlock_initialize(&CPU->timeoutlock, "timeout_lock");
57		58
58	#if defined CONFIG_TIMEOUT_AVL_TREE	59	#if defined CONFIG_TIMEOUT_AVL_TREE
59	avltree_create(&CPU->timeout_active_tree);	60	avltree_create(&CPU->timeout_active_tree);
60	#elif defined CONFIG_TIMEOUT_EXTAVL_TREE	61	#elif defined CONFIG_TIMEOUT_EXTAVL_TREE
61	extavltree_create(&CPU->timeout_active_tree);	62	extavltree_create(&CPU->timeout_active_tree);
62	#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE	63	#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE
63	extavlreltree_create(&CPU->timeout_active_tree);	64	extavlreltree_create(&CPU->timeout_active_tree);
64	#else	65	#else
65	list_initialize(&CPU->timeout_active_head);	66	list_initialize(&CPU->timeout_active_head);
66	#endif	67	#endif
67	}	68	}
68		69
69		70
70	/** Reinitialize timeout	71	/** Reinitialize timeout
71	*	72	*
72	* Initialize all members except the lock.	73	* Initialize all members except the lock.
73	*	74	*
74	* @param t Timeout to be initialized.	75	* @param t Timeout to be initialized.
75	*	76	*
76	*/	77	*/
77	void timeout_reinitialize(timeout_t *t)	78	void timeout_reinitialize(timeout_t *t)
78	{	79	{
79	t->cpu = NULL;	80	t->cpu = NULL;
80	t->handler = NULL;	81	t->handler = NULL;
81	t->arg = NULL;	82	t->arg = NULL;
82		83
83	#if defined CONFIG_TIMEOUT_AVL_TREE	84	#if defined CONFIG_TIMEOUT_AVL_TREE
84	avltree_node_initialize(&t->node);	85	avltree_node_initialize(&t->node);
85	#elif defined CONFIG_TIMEOUT_EXTAVL_TREE	86	#elif defined CONFIG_TIMEOUT_EXTAVL_TREE
86	extavltree_node_initialize(&t->node);	87	extavltree_node_initialize(&t->node);
87	#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE	88	#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE
88	extavlreltree_node_initialize(&t->node);	89	extavlreltree_node_initialize(&t->node);
89	#else	90	#else
90	t->ticks = 0;	91	t->ticks = 0;
91	link_initialize(&t->link);	92	link_initialize(&t->link);
92	#endif	93	#endif
93	}	94	}
94		95
95		96
96	/** Initialize timeout	97	/** Initialize timeout
97	*	98	*
98	* Initialize all members including the lock.	99	* Initialize all members including the lock.
99	*	100	*
100	* @param t Timeout to be initialized.	101	* @param t Timeout to be initialized.
101	*	102	*
102	*/	103	*/
103	void timeout_initialize(timeout_t *t)	104	void timeout_initialize(timeout_t *t)
104	{	105	{
105	spinlock_initialize(&t->lock, "timeout_t_lock");	106	spinlock_initialize(&t->lock, "timeout_t_lock");
106	timeout_reinitialize(t);	107	timeout_reinitialize(t);
107	}	108	}
108		109
109	#if defined CONFIG_TIMEOUT_AVL_TREE \|\| \	110	#if defined CONFIG_TIMEOUT_AVL_TREE \|\| \
110	defined CONFIG_TIMEOUT_EXTAVL_TREE \|\| \	111	defined CONFIG_TIMEOUT_EXTAVL_TREE \|\| \
111	defined CONFIG_TIMEOUT_EXTAVLREL_TREE	112	defined CONFIG_TIMEOUT_EXTAVLREL_TREE
112		113
113	/** Register timeout	114	/** Register timeout
114	*	115	*
115	* Insert timeout handler f (with argument arg)	116	* Insert timeout handler f (with argument arg)
116	* to timeout ExtAVL tree and make it execute in	117	* to timeout ExtAVL tree and make it execute in
117	* time microseconds (or slightly more).	118	* time microseconds (or slightly more).
118	*	119	*
119	* @param t Timeout structure.	120	* @param t Timeout structure.
120	* @param time Number of usec in the future to execute	121	* @param time Number of usec in the future to execute
121	* the handler.	122	* the handler.
122	* @param f Timeout handler function.	123	* @param f Timeout handler function.
123	* @param arg Timeout handler argument.	124	* @param arg Timeout handler argument.
124	*	125	*
125	*/	126	*/
126	void timeout_register(timeout_t t, uint64_t time, timeout_handler_t f, void arg)	127	void timeout_register(timeout_t t, uint64_t time, timeout_handler_t f, void arg)
127	{	128	{
128	ipl_t ipl;	129	ipl_t ipl;
129		130
130	ipl = interrupts_disable();	131	ipl = interrupts_disable();
131	spinlock_lock(&CPU->timeoutlock);	132	spinlock_lock(&CPU->timeoutlock);
132	spinlock_lock(&t->lock);	133	spinlock_lock(&t->lock);
133		134
134	if (t->cpu)	135	if (t->cpu)
135	panic("t->cpu != 0");	136	panic("t->cpu != 0");
136		137
137	t->cpu = CPU;	138	t->cpu = CPU;
138	t->handler = f;	139	t->handler = f;
139	t->arg = arg;	140	t->arg = arg;
140	t->node.key = us2ticks(time);	141	t->node.key = us2ticks(time);
141		142
142	/*	143	/*
143	* Put timeout into tree structure.	144	* Put timeout into tree structure.
144	*/	145	*/
145	#if defined CONFIG_TIMEOUT_AVL_TREE	146	#if defined CONFIG_TIMEOUT_AVL_TREE
146	avltree_insert(&CPU->timeout_active_tree,&t->node);	147	avltree_insert(&CPU->timeout_active_tree,&t->node);
147	#elif defined CONFIG_TIMEOUT_EXTAVL_TREE	148	#elif defined CONFIG_TIMEOUT_EXTAVL_TREE
148	extavltree_insert(&CPU->timeout_active_tree,&t->node);	149	extavltree_insert(&CPU->timeout_active_tree,&t->node);
149	#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE	150	#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE
150	extavlreltree_insert(&CPU->timeout_active_tree,&t->node);	151	extavlreltree_insert(&CPU->timeout_active_tree,&t->node);
151	#endif	152	#endif
152		153
153	spinlock_unlock(&t->lock);	154	spinlock_unlock(&t->lock);
154	spinlock_unlock(&CPU->timeoutlock);	155	spinlock_unlock(&CPU->timeoutlock);
155	interrupts_restore(ipl);	156	interrupts_restore(ipl);
156	}	157	}
157		158
158		159
159	/** Unregister timeout	160	/** Unregister timeout
160	*	161	*
161	* Remove timeout from timeout ExtAVL tree structure.	162	* Remove timeout from timeout ExtAVL tree structure.
162	*	163	*
163	* @param t Timeout to unregister.	164	* @param t Timeout to unregister.
164	*	165	*
165	* @return true on success, false on failure.	166	* @return true on success, false on failure.
166	*/	167	*/
167	bool timeout_unregister(timeout_t *t)	168	bool timeout_unregister(timeout_t *t)
168	{	169	{
169	ipl_t ipl;	170	ipl_t ipl;
170		171
171	grab_locks:	172	grab_locks:
172	ipl = interrupts_disable();	173	ipl = interrupts_disable();
173	spinlock_lock(&t->lock);	174	spinlock_lock(&t->lock);
174	if (!t->cpu) {	175	if (!t->cpu) {
175	spinlock_unlock(&t->lock);	176	spinlock_unlock(&t->lock);
176	interrupts_restore(ipl);	177	interrupts_restore(ipl);
177	return false;	178	return false;
178	}	179	}
179	if (!spinlock_trylock(&t->cpu->timeoutlock)) {	180	if (!spinlock_trylock(&t->cpu->timeoutlock)) {
180	spinlock_unlock(&t->lock);	181	spinlock_unlock(&t->lock);
181	interrupts_restore(ipl);	182	interrupts_restore(ipl);
182	goto grab_locks;	183	goto grab_locks;
183	}	184	}
184	/*	185	/*
185	* Now we know for sure that t hasn't been activated yet	186	* Now we know for sure that t hasn't been activated yet
186	* and is lurking in t->cpu->timeout_active_head queue.	187	* and is lurking in t->cpu->timeout_active_head queue.
187	*/	188	*/
188		189
189	/*	190	/*
190	* Delete timeout from tree structure.	191	* Delete timeout from tree structure.
191	*/	192	*/
192	#if defined CONFIG_TIMEOUT_AVL_TREE	193	#if defined CONFIG_TIMEOUT_AVL_TREE
193	avltree_delete(&CPU->timeout_active_tree,&t->node);	194	avltree_delete(&CPU->timeout_active_tree,&t->node);
194	#elif defined CONFIG_TIMEOUT_EXTAVL_TREE	195	#elif defined CONFIG_TIMEOUT_EXTAVL_TREE
195	extavltree_delete(&CPU->timeout_active_tree,&t->node);	196	extavltree_delete(&CPU->timeout_active_tree,&t->node);
196	#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE	197	#elif defined CONFIG_TIMEOUT_EXTAVLREL_TREE
197	extavlreltree_delete(&CPU->timeout_active_tree,&t->node);	198	extavlreltree_delete(&CPU->timeout_active_tree,&t->node);
198	#endif	199	#endif
199		200
200	spinlock_unlock(&t->cpu->timeoutlock);	201	spinlock_unlock(&t->cpu->timeoutlock);
201		202
202	timeout_reinitialize(t);	203	timeout_reinitialize(t);
203	spinlock_unlock(&t->lock);	204	spinlock_unlock(&t->lock);
204		205
205	interrupts_restore(ipl);	206	interrupts_restore(ipl);
206	return true;	207	return true;
207	}	208	}
208		209
209	#else	210	#else
210		211
211	/** Register timeout	212	/** Register timeout
212	*	213	*
213	* Insert timeout handler f (with argument arg)	214	* Insert timeout handler f (with argument arg)
214	* to timeout list and make it execute in	215	* to timeout list and make it execute in
215	* time microseconds (or slightly more).	216	* time microseconds (or slightly more).
216	*	217	*
217	* @param t Timeout structure.	218	* @param t Timeout structure.
218	* @param time Number of usec in the future to execute	219	* @param time Number of usec in the future to execute
219	* the handler.	220	* the handler.
220	* @param f Timeout handler function.	221	* @param f Timeout handler function.
221	* @param arg Timeout handler argument.	222	* @param arg Timeout handler argument.
222	*	223	*
223	*/	224	*/
224	void timeout_register(timeout_t t, uint64_t time, timeout_handler_t f, void arg)	225	void timeout_register(timeout_t t, uint64_t time, timeout_handler_t f, void arg)
225	{	226	{
226	timeout_t *hlp = NULL;	227	timeout_t *hlp = NULL;
227	link_t l, m;	228	link_t l, m;
228	ipl_t ipl;	229	ipl_t ipl;
229	uint64_t sum;	230	uint64_t sum;
230		231
231	ipl = interrupts_disable();	232	ipl = interrupts_disable();
232	spinlock_lock(&CPU->timeoutlock);	233	spinlock_lock(&CPU->timeoutlock);
233	spinlock_lock(&t->lock);	234	spinlock_lock(&t->lock);
234		235
235	if (t->cpu)	236	if (t->cpu)
236	panic("t->cpu != 0");	237	panic("t->cpu != 0");
237		238
238	t->cpu = CPU;	239	t->cpu = CPU;
239	t->ticks = us2ticks(time);	240	t->ticks = us2ticks(time);
240		241
241	t->handler = f;	242	t->handler = f;
242	t->arg = arg;	243	t->arg = arg;
243		244
244	/*	245	/*
245	* Insert t into the active timeouts list according to t->ticks.	246	* Insert t into the active timeouts list according to t->ticks.
246	*/	247	*/
247	sum = 0;	248	sum = 0;
248	l = CPU->timeout_active_head.next;	249	l = CPU->timeout_active_head.next;
249	while (l != &CPU->timeout_active_head) {	250	while (l != &CPU->timeout_active_head) {
250	hlp = list_get_instance(l, timeout_t, link);	251	hlp = list_get_instance(l, timeout_t, link);
251	spinlock_lock(&hlp->lock);	252	spinlock_lock(&hlp->lock);
252	if (t->ticks < sum + hlp->ticks) {	253	if (t->ticks < sum + hlp->ticks) {
253	spinlock_unlock(&hlp->lock);	254	spinlock_unlock(&hlp->lock);
254	break;	255	break;
255	}	256	}
256	sum += hlp->ticks;	257	sum += hlp->ticks;
257	spinlock_unlock(&hlp->lock);	258	spinlock_unlock(&hlp->lock);
258	l = l->next;	259	l = l->next;
259	}	260	}
260		261
261	m = l->prev;	262	m = l->prev;
262	list_prepend(&t->link, m); /* avoid using l->prev */	263	list_prepend(&t->link, m); /* avoid using l->prev */
263		264
264	/*	265	/*
265	* Adjust t->ticks according to ticks accumulated in h's predecessors.	266	* Adjust t->ticks according to ticks accumulated in h's predecessors.
266	*/	267	*/
267	t->ticks -= sum;	268	t->ticks -= sum;
268		269
269	/*	270	/*
270	* Decrease ticks of t's immediate succesor by t->ticks.	271	* Decrease ticks of t's immediate succesor by t->ticks.
271	*/	272	*/
272	if (l != &CPU->timeout_active_head) {	273	if (l != &CPU->timeout_active_head) {
273	spinlock_lock(&hlp->lock);	274	spinlock_lock(&hlp->lock);
274	hlp->ticks -= t->ticks;	275	hlp->ticks -= t->ticks;
275	spinlock_unlock(&hlp->lock);	276	spinlock_unlock(&hlp->lock);
276	}	277	}
277		278
278	spinlock_unlock(&t->lock);	279	spinlock_unlock(&t->lock);
279	spinlock_unlock(&CPU->timeoutlock);	280	spinlock_unlock(&CPU->timeoutlock);
280	interrupts_restore(ipl);	281	interrupts_restore(ipl);
281	}	282	}
282		283
283		284
284	/** Unregister timeout	285	/** Unregister timeout
285	*	286	*
286	* Remove timeout from timeout list.	287	* Remove timeout from timeout list.
287	*	288	*
288	* @param t Timeout to unregister.	289	* @param t Timeout to unregister.
289	*	290	*
290	* @return true on success, false on failure.	291	* @return true on success, false on failure.
291	*/	292	*/
292	bool timeout_unregister(timeout_t *t)	293	bool timeout_unregister(timeout_t *t)
293	{	294	{
294	timeout_t *hlp;	295	timeout_t *hlp;
295	link_t *l;	296	link_t *l;
296	ipl_t ipl;	297	ipl_t ipl;
297		298
298	grab_locks:	299	grab_locks:
299	ipl = interrupts_disable();	300	ipl = interrupts_disable();
300	spinlock_lock(&t->lock);	301	spinlock_lock(&t->lock);
301	if (!t->cpu) {	302	if (!t->cpu) {
302	spinlock_unlock(&t->lock);	303	spinlock_unlock(&t->lock);
303	interrupts_restore(ipl);	304	interrupts_restore(ipl);
304	return false;	305	return false;
305	}	306	}
306	if (!spinlock_trylock(&t->cpu->timeoutlock)) {	307	if (!spinlock_trylock(&t->cpu->timeoutlock)) {
307	spinlock_unlock(&t->lock);	308	spinlock_unlock(&t->lock);
308	interrupts_restore(ipl);	309	interrupts_restore(ipl);
309	goto grab_locks;	310	goto grab_locks;
310	}	311	}
311		312
312	/*	313	/*
313	* Now we know for sure that t hasn't been activated yet	314	* Now we know for sure that t hasn't been activated yet
314	* and is lurking in t->cpu->timeout_active_head queue.	315	* and is lurking in t->cpu->timeout_active_head queue.
315	*/	316	*/
316		317
317	l = t->link.next;	318	l = t->link.next;
318	if (l != &t->cpu->timeout_active_head) {	319	if (l != &t->cpu->timeout_active_head) {
319	hlp = list_get_instance(l, timeout_t, link);	320	hlp = list_get_instance(l, timeout_t, link);
320	spinlock_lock(&hlp->lock);	321	spinlock_lock(&hlp->lock);
321	hlp->ticks += t->ticks;	322	hlp->ticks += t->ticks;
322	spinlock_unlock(&hlp->lock);	323	spinlock_unlock(&hlp->lock);
323	}	324	}
324		325
325	list_remove(&t->link);	326	list_remove(&t->link);
326	spinlock_unlock(&t->cpu->timeoutlock);	327	spinlock_unlock(&t->cpu->timeoutlock);
327		328
328	timeout_reinitialize(t);	329	timeout_reinitialize(t);
329	spinlock_unlock(&t->lock);	330	spinlock_unlock(&t->lock);
330		331
331	interrupts_restore(ipl);	332	interrupts_restore(ipl);
332	return true;	333	return true;
333	}	334	}
334		335
335	#endif	336	#endif
336	/** @}	337	/** @}
337	*/	338	*/
338		339

Subversion Repositories HelenOS

(root)/branches/rcu/kernel/generic/src/time/timeout.c – Rev 2421 → 2431