Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
8272cce96c2c0ddf4839d2f03c1b8066662c2c78
[simgrid.git] / src / simix / smx_synchro.c
1 /*      $Id$     */
2
3 /* Copyright (c) 2007 Arnaud Legrand, Bruno Donnassolo.
4    All rights reserved.                                          */
5
6 /* This program is free software; you can redistribute it and/or modify it
7  * under the terms of the license (GNU LGPL) which comes with this package. */
8
9 #include "private.h"
10 #include "xbt/log.h"
11
12
13 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simix_synchro, simix,
14                                 "Logging specific to SIMIX (synchronization)");
15
16
17 /****************************** Synchronization *******************************/
18
19 /*********************************** Mutex ************************************/
20
21 /**
22  * \brief Initialize a mutex.
23  *
24  * Allocs and creates the data for the mutex. It have to be called before the utilisation of the mutex.
25  * \return A mutex
26  */
27 smx_mutex_t SIMIX_mutex_init()
28 {
29   smx_mutex_t m = xbt_new0(s_smx_mutex_t, 1);
30   s_smx_process_t p;            /* useful to initialize sleeping swag */
31   /* structures initialization */
32   m->refcount = 0;
33   m->sleeping = xbt_swag_new(xbt_swag_offset(p, synchro_hookup));
34   return m;
35 }
36
37 /**
38  * \brief Locks a mutex.
39  *
40  * Tries to lock a mutex, if the mutex isn't used yet, the process can continue its execution, else it'll be blocked here. You have to call #SIMIX_mutex_unlock to free the mutex.
41  * \param mutex The mutex
42  */
43 void SIMIX_mutex_lock(smx_mutex_t mutex)
44 {
45   smx_process_t self = SIMIX_process_self();
46   xbt_assert0((mutex != NULL), "Invalid parameters");
47
48
49   if (mutex->refcount) {
50     /* somebody using the mutex, block */
51     xbt_swag_insert(self, mutex->sleeping);
52     self->mutex = mutex;
53     /* wait for some process make the unlock and wake up me from mutex->sleeping */
54     SIMIX_process_yield();
55     self->mutex = NULL;
56
57     /* verify if the process was suspended */
58     while (self->suspended) {
59       SIMIX_process_yield();
60     }
61
62     mutex->refcount = 1;
63   } else {
64     /* mutex free */
65     mutex->refcount = 1;
66   }
67   return;
68 }
69
70 /**
71  * \brief Tries to lock a mutex.
72  *
73  * Tries to lock a mutex, return 1 if the mutex is free, 0 else. This function does not block the process if the mutex is used.
74  * \param mutex The mutex
75  * \return 1 - mutex free, 0 - mutex used
76  */
77 int SIMIX_mutex_trylock(smx_mutex_t mutex)
78 {
79   xbt_assert0((mutex != NULL), "Invalid parameters");
80
81   if (mutex->refcount)
82     return 0;
83   else {
84     mutex->refcount = 1;
85     return 1;
86   }
87 }
88
89 /**
90  * \brief Unlocks a mutex.
91  *
92  * Unlocks the mutex and wakes up a process blocked on it. If there are no process sleeping, only sets the mutex as free.
93  * \param mutex The mutex
94  */
95 void SIMIX_mutex_unlock(smx_mutex_t mutex)
96 {
97   smx_process_t p;              /*process to wake up */
98
99   xbt_assert0((mutex != NULL), "Invalid parameters");
100
101   if (xbt_swag_size(mutex->sleeping) > 0) {
102     p = xbt_swag_extract(mutex->sleeping);
103     mutex->refcount = 0;
104     xbt_swag_insert(p, simix_global->process_to_run);
105   } else {
106     /* nobody to wake up */
107     mutex->refcount = 0;
108   }
109   return;
110 }
111
112 /**
113  * \brief Destroys a mutex.
114  *
115  * Destroys and frees the mutex's memory. 
116  * \param mutex A mutex
117  */
118 void SIMIX_mutex_destroy(smx_mutex_t mutex)
119 {
120   if (mutex == NULL)
121     return;
122   else {
123     xbt_swag_free(mutex->sleeping);
124     xbt_free(mutex);
125     return;
126   }
127 }
128
129 /******************************** Conditional *********************************/
130
131 /**
132  * \brief Initialize a condition.
133  *
134  * Allocates and creates the data for the condition.
135  * It have to be called before the use of the condition.
136  * \return A condition
137  */
138 smx_cond_t SIMIX_cond_init()
139 {
140   smx_cond_t cond = xbt_new0(s_smx_cond_t, 1);
141   s_smx_process_t p;
142
143   cond->sleeping = xbt_swag_new(xbt_swag_offset(p, synchro_hookup));
144   cond->actions = xbt_fifo_new();
145   cond->mutex = NULL;
146   return cond;
147 }
148
149 /**
150  * \brief Signalizes a condition.
151  *
152  * Signalizes a condition and wakes up a sleeping process. If there are no process sleeping, no action is done.
153  * \param cond A condition
154  */
155 void SIMIX_cond_signal(smx_cond_t cond)
156 {
157   smx_process_t proc = NULL;
158   DEBUG1("Signal condition %p", cond);
159   xbt_assert0((cond != NULL), "Invalid parameters");
160
161
162   if (xbt_swag_size(cond->sleeping) >= 1) {
163     proc = xbt_swag_extract(cond->sleeping);
164     xbt_swag_insert(proc, simix_global->process_to_run);
165   }
166
167   return;
168 }
169
170 /**
171  * \brief Waits on a condition.
172  *
173  * Blocks a process until the signal is called. This functions frees the mutex associated and locks it after its execution.
174  * \param cond A condition
175  * \param mutex A mutex
176  */
177 void SIMIX_cond_wait(smx_cond_t cond, smx_mutex_t mutex)
178 {
179   smx_action_t act_sleep;
180
181   DEBUG1("Wait condition %p", cond);
182
183   /* If there is a mutex unlock it */
184   if(mutex != NULL){
185     cond->mutex = mutex;
186     SIMIX_mutex_unlock(mutex);
187   }
188   
189   /* Always create an action null in case there is a host failure */
190   act_sleep = SIMIX_action_sleep(SIMIX_host_self(), -1);
191   SIMIX_action_set_name(act_sleep,bprintf("Wait condition %p", cond));
192   SIMIX_process_self()->waiting_action = act_sleep;
193   SIMIX_register_action_to_condition(act_sleep, cond);
194   __SIMIX_cond_wait(cond);
195   SIMIX_process_self()->waiting_action = NULL;
196   SIMIX_unregister_action_to_condition(act_sleep, cond);
197   SIMIX_action_destroy(act_sleep);
198
199   /* get the mutex again if necessary */
200   if(mutex != NULL)
201     SIMIX_mutex_lock(cond->mutex);
202
203   return;
204 }
205
206 xbt_fifo_t SIMIX_cond_get_actions(smx_cond_t cond)
207 {
208   xbt_assert0((cond != NULL), "Invalid parameters");
209   return cond->actions;
210 }
211
212 void __SIMIX_cond_wait(smx_cond_t cond)
213 {
214   smx_process_t self = SIMIX_process_self();
215   xbt_assert0((cond != NULL), "Invalid parameters");
216
217   /* process status */
218
219   self->cond = cond;
220   xbt_swag_insert(self, cond->sleeping);
221   SIMIX_process_yield();
222   self->cond = NULL;
223   while (self->suspended) {
224     SIMIX_process_yield();
225   }
226   return;
227 }
228
229 /**
230  * \brief Waits on a condition with timeout.
231  *
232  * Same behavior of #SIMIX_cond_wait, but waits a maximum time and throws an timeout_error if it happens.
233  * \param cond A condition
234  * \param mutex A mutex
235  * \param max_duration Timeout time
236  */
237 void SIMIX_cond_wait_timeout(smx_cond_t cond, smx_mutex_t mutex,
238                              double max_duration)
239 {
240   smx_action_t act_sleep;
241
242   DEBUG1("Timed wait condition %p", cond);
243
244   /* If there is a mutex unlock it */
245   if(mutex != NULL){
246     cond->mutex = mutex;
247     SIMIX_mutex_unlock(mutex);
248   }
249
250   if (max_duration >= 0) {
251     act_sleep = SIMIX_action_sleep(SIMIX_host_self(), max_duration);
252     SIMIX_action_set_name(act_sleep,bprintf("Timed wait condition %p (max_duration:%f)", cond,max_duration));
253     SIMIX_register_action_to_condition(act_sleep, cond);
254     SIMIX_process_self()->waiting_action = act_sleep;
255     __SIMIX_cond_wait(cond);
256     SIMIX_process_self()->waiting_action = NULL;
257     SIMIX_unregister_action_to_condition(act_sleep, cond);
258     if (SIMIX_action_get_state(act_sleep) == SURF_ACTION_DONE) {
259       SIMIX_action_destroy(act_sleep);
260       THROW1(timeout_error, 0, "Condition timeout after %f",max_duration);
261     } else {
262       SIMIX_action_destroy(act_sleep);
263     }
264
265   } else
266     __SIMIX_cond_wait(cond);
267
268   /* get the mutex again if necessary */
269   if(mutex != NULL)
270     SIMIX_mutex_lock(cond->mutex);
271 }
272
273 /**
274  * \brief Broadcasts a condition.
275  *
276  * Signalizes a condition and wakes up ALL sleping process. If there are no process sleeping, no action is done.
277  * \param cond A condition
278  */
279 void SIMIX_cond_broadcast(smx_cond_t cond)
280 {
281   smx_process_t proc = NULL;
282   smx_process_t proc_next = NULL;
283
284   xbt_assert0((cond != NULL), "Invalid parameters");
285
286   DEBUG1("Broadcast condition %p", cond);
287   xbt_swag_foreach_safe(proc, proc_next, cond->sleeping) {
288     xbt_swag_remove(proc, cond->sleeping);
289     xbt_swag_insert(proc, simix_global->process_to_run);
290   }
291 }
292
293 /**
294  * \brief Destroys a contidion.
295  *
296  * Destroys and frees the condition's memory. 
297  * \param cond A condition
298  */
299 void SIMIX_cond_destroy(smx_cond_t cond)
300 {
301   DEBUG1("Destroy condition %p", cond);
302   if (cond == NULL)
303     return;
304   else {
305     xbt_fifo_item_t item = NULL;
306     smx_action_t action = NULL;
307
308     xbt_assert0(xbt_swag_size(cond->sleeping) == 0,
309                 "Cannot destroy conditional since someone is still using it");
310     xbt_swag_free(cond->sleeping);
311
312     DEBUG1("%d actions registered", xbt_fifo_size(cond->actions));
313     __SIMIX_cond_display_actions(cond);
314     xbt_fifo_foreach(cond->actions, item, action, smx_action_t) {
315       SIMIX_unregister_action_to_condition(action, cond);
316     }
317     __SIMIX_cond_display_actions(cond);
318
319     xbt_fifo_free(cond->actions);
320     xbt_free(cond);
321     return;
322   }
323 }
324
325 void SIMIX_cond_display_info(smx_cond_t cond)
326 {
327   if (cond == NULL)
328     return;
329   else {
330     smx_process_t process = NULL;
331
332     INFO0("Blocked process on this condition:");
333     xbt_swag_foreach(process, cond->sleeping) {
334       INFO2("\t %s running on host %s", process->name,
335             process->smx_host->name);
336     }
337   }
338 }
339
340 /* ************************** Semaphores ************************************** */
341 #define SMX_SEM_NOLIMIT 99999
342 /** @brief Initialize a semaphore */
343 smx_sem_t SIMIX_sem_init(int capacity) {
344   smx_sem_t sem = xbt_new0(s_smx_sem_t, 1);
345   s_smx_process_t p;
346
347   sem->sleeping = xbt_swag_new(xbt_swag_offset(p, synchro_hookup));
348   sem->actions = xbt_fifo_new();
349   sem->capacity = capacity;
350   return sem;
351 }
352 /** @brief Destroys a semaphore */
353 void SIMIX_sem_destroy(smx_sem_t sem) {
354   DEBUG1("Destroy semaphore %p", sem);
355   if (sem == NULL)
356     return;
357
358   xbt_fifo_item_t item = NULL;
359   smx_action_t action = NULL;
360
361   xbt_assert0(xbt_swag_size(sem->sleeping) == 0,
362       "Cannot destroy semaphore since someone is still using it");
363   xbt_swag_free(sem->sleeping);
364
365   DEBUG1("%d actions registered", xbt_fifo_size(sem->actions));
366   xbt_fifo_foreach(sem->actions, item, action, smx_action_t) {
367     SIMIX_unregister_action_to_semaphore(action, sem);
368   }
369
370   xbt_fifo_free(sem->actions);
371   xbt_free(sem);
372 }
373
374 /** @brief release the semaphore
375  *
376  * The first locked process on this semaphore is unlocked.
377  * If no one was blocked, the semaphore capacity is increased by 1.
378  * */
379 void SIMIX_sem_release(smx_sem_t sem) {
380   if (xbt_swag_size(sem->sleeping) >= 1) {
381     smx_process_t proc = xbt_swag_extract(sem->sleeping);
382     xbt_swag_insert(proc, simix_global->process_to_run);
383   } else if (sem->capacity != SMX_SEM_NOLIMIT) {
384     sem->capacity++;
385   }
386 }
387 /** @brief make sure the semaphore will never be blocking again
388  *
389  * This function is not really in the semaphore spirit. It makes
390  * sure that the semaphore will never be blocking anymore.
391  *
392  * Releasing and acquiring the semaphore after calling this
393  * function is a noop. Such "broken" semaphores are useful to
394  * implement something between condition variables (with broadcast)
395  * and semaphore (with memory). It's like a semaphore signaled for ever.
396  *
397  * There is no way to reset the semaphore to a more regular state afterward.
398  * */
399 void SIMIX_sem_release_forever(smx_sem_t sem) {
400   smx_process_t proc = NULL;
401   smx_process_t proc_next = NULL;
402
403   DEBUG1("Broadcast semaphore %p", sem);
404   xbt_swag_foreach_safe(proc, proc_next, sem->sleeping) {
405     xbt_swag_remove(proc, sem->sleeping);
406     xbt_swag_insert(proc, simix_global->process_to_run);
407   }
408 }
409
410 static inline void SIMIX_sem_block_onto(smx_sem_t sem) {
411   smx_process_t self = SIMIX_process_self();
412
413   /* process status */
414   self->sem = sem;
415   xbt_swag_insert(self, sem->sleeping);
416   SIMIX_process_yield();
417   self->sem = NULL;
418   while (self->suspended)
419     SIMIX_process_yield();
420 }
421
422 /** @brief Returns true if acquiring this semaphore would block */
423 int SIMIX_sem_would_block(smx_sem_t sem) {
424   return (sem->capacity>0);
425 }
426
427 /**
428  * \brief Waits on a semaphore
429  *
430  * If the capacity>0, decrease the capacity.
431  *
432  * If capacity==0, locks the current process
433  * until someone call SIMIX_sem_release() on this semaphore
434  */
435 void SIMIX_sem_acquire(smx_sem_t sem) {
436   smx_action_t act_sleep;
437
438   DEBUG1("Wait semaphore %p", sem);
439
440   if (sem->capacity == SMX_SEM_NOLIMIT)
441     return; /* don't even decrease it if wide open */
442
443   /* If capacity sufficient, decrease it */
444   if (sem->capacity>0) {
445     sem->capacity--;
446     return;
447   }
448
449   /* Always create an action null in case there is a host failure */
450   act_sleep = SIMIX_action_sleep(SIMIX_host_self(), -1);
451   SIMIX_action_set_name(act_sleep,bprintf("Locked in semaphore %p", sem));
452   SIMIX_process_self()->waiting_action = act_sleep;
453   SIMIX_register_action_to_semaphore(act_sleep, sem);
454   SIMIX_sem_block_onto(sem);
455   SIMIX_process_self()->waiting_action = NULL;
456   SIMIX_unregister_action_to_semaphore(act_sleep, sem);
457   SIMIX_action_destroy(act_sleep);
458 }
459 /**
460  * \brief Tries to acquire a semaphore before a timeout
461  *
462  * Same behavior of #SIMIX_sem_acquire, but waits a maximum time and throws an timeout_error if it happens.
463  */
464 void SIMIX_sem_acquire_timeout(smx_sem_t sem, double max_duration) {
465   smx_action_t act_sleep;
466
467   DEBUG2("Timed wait semaphore %p (timeout:%f)", sem,max_duration);
468
469   if (sem->capacity == SMX_SEM_NOLIMIT)
470     return; /* don't even decrease it if wide open */
471
472   /* If capacity sufficient, decrease it */
473   if (sem->capacity>0) {
474     sem->capacity--;
475     return;
476   }
477
478   if (max_duration >= 0) {
479     act_sleep = SIMIX_action_sleep(SIMIX_host_self(), max_duration);
480     SIMIX_action_set_name(act_sleep,bprintf("Timed wait semaphore %p (max_duration:%f)", sem,max_duration));
481     SIMIX_register_action_to_semaphore(act_sleep, sem);
482     SIMIX_process_self()->waiting_action = act_sleep;
483     SIMIX_sem_block_onto(sem);
484     SIMIX_process_self()->waiting_action = NULL;
485     SIMIX_unregister_action_to_semaphore(act_sleep, sem);
486     if (SIMIX_action_get_state(act_sleep) == SURF_ACTION_DONE) {
487       SIMIX_action_destroy(act_sleep);
488       THROW1(timeout_error, 0, "Semaphore acquire timeouted after %f",max_duration);
489     } else {
490       SIMIX_action_destroy(act_sleep);
491     }
492
493   } else
494     SIMIX_sem_block_onto(sem);
495 }