ipc.c 49.9 KB
Newer Older
1 2 3
/*
 * File      : ipc.c
 * This file is part of RT-Thread RTOS
D
dzzxzz 已提交
4
 * COPYRIGHT (C) 2006 - 2012, RT-Thread Development Team
5 6 7
 *
 * The license and distribution terms for this file may be
 * found in the file LICENSE in this distribution or at
B
bernard.xiong 已提交
8
 * http://www.rt-thread.org/license/LICENSE
9 10 11 12 13
 *
 * Change Logs:
 * Date           Author       Notes
 * 2006-03-14     Bernard      the first version
 * 2006-04-25     Bernard      implement semaphore
14
 * 2006-05-03     Bernard      add RT_IPC_DEBUG
15 16 17 18 19 20 21 22 23 24
 *                             modify the type of IPC waiting time to rt_int32_t
 * 2006-05-10     Bernard      fix the semaphore take bug and add IPC object
 * 2006-05-12     Bernard      implement mailbox and message queue
 * 2006-05-20     Bernard      implement mutex
 * 2006-05-23     Bernard      implement fast event
 * 2006-05-24     Bernard      implement event
 * 2006-06-03     Bernard      fix the thread timer init bug
 * 2006-06-05     Bernard      fix the mutex release bug
 * 2006-06-07     Bernard      fix the message queue send bug
 * 2006-08-04     Bernard      add hook support
B
bernard.xiong 已提交
25
 * 2009-05-21     Yi.qiu       fix the sem release bug
26 27
 * 2009-07-18     Bernard      fix the event clear bug
 * 2009-09-09     Bernard      remove fast event and fix ipc release bug
28
 * 2009-10-10     Bernard      change semaphore and mutex value to unsigned value
B
bernard.xiong 已提交
29
 * 2009-10-25     Bernard      change the mb/mq receive timeout to 0 if the
30
 *                             re-calculated delta tick is a negative number.
B
bernard.xiong 已提交
31
 * 2009-12-16     Bernard      fix the rt_ipc_object_suspend issue when IPC flag
32
 *                             is RT_IPC_FLAG_PRIO
33
 * 2010-01-20     mbbill       remove rt_ipc_object_decrease function.
34
 * 2010-04-20     Bernard      move memcpy outside interrupt disable in mq
qiuyiuestc's avatar
qiuyiuestc 已提交
35
 * 2010-10-26     yi.qiu       add module support in rt_mp_delete and rt_mq_delete
36
 * 2010-11-10     Bernard      add IPC reset command implementation.
37
 * 2011-12-18     Bernard      add more parameter checking in message queue
38 39 40 41 42 43
 */

#include <rtthread.h>
#include <rthw.h>

#ifdef RT_USING_HOOK
D
dzzxzz 已提交
44 45 46
extern void (*rt_object_trytake_hook)(struct rt_object *object);
extern void (*rt_object_take_hook)(struct rt_object *object);
extern void (*rt_object_put_hook)(struct rt_object *object);
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
#endif

/**
 * @addtogroup IPC
 */

/*@{*/

/**
 * This function will initialize an IPC object
 *
 * @param ipc the IPC object
 *
 * @return the operation status, RT_EOK on successful
 */
rt_inline rt_err_t rt_ipc_object_init(struct rt_ipc_object *ipc)
{
	/* init ipc object */
	rt_list_init(&(ipc->suspend_thread));

	return RT_EOK;
}

/**
71 72
 * This function will suspend a thread to a specified list. IPC object or some double-queue
 * object (mailbox etc.) contains this kind of list.
73
 *
B
bernard.xiong@gmail.com 已提交
74
 * @param list the IPC suspended thread list
75
 * @param thread the thread object to be suspended
B
bernard.xiong@gmail.com 已提交
76
 * @param flag the IPC object flag, which shall be RT_IPC_FLAG_FIFO/RT_IPC_FLAG_PRIO.
77 78 79
 *
 * @return the operation status, RT_EOK on successful
 */
80
rt_inline rt_err_t rt_ipc_list_suspend(rt_list_t *list, struct rt_thread *thread, rt_uint8_t flag)
81 82 83 84
{
	/* suspend thread */
	rt_thread_suspend(thread);

85
	switch (flag)
86 87
	{
	case RT_IPC_FLAG_FIFO:
88
		rt_list_insert_before(list, &(thread->tlist));
89 90 91 92
		break;

	case RT_IPC_FLAG_PRIO:
		{
D
dzzxzz 已提交
93 94
			struct rt_list_node *n;
			struct rt_thread *sthread;
95 96

			/* find a suitable position */
97
			for (n = list->next; n != list; n = n->next)
98 99 100 101
			{
				sthread = rt_list_entry(n, struct rt_thread, tlist);

				/* find out */
102 103 104 105 106 107
				if (thread->current_priority < sthread->current_priority)
				{
					/* insert this thread before the sthread */
					rt_list_insert_before(&(sthread->tlist), &(thread->tlist));
					break;
				}
108
			}
B
bernard.xiong 已提交
109 110

			/* not found a suitable position, append to the end of suspend_thread list */
111 112
			if (n == list)
				rt_list_insert_before(list, &(thread->tlist));
113 114 115 116 117 118 119 120
		}
		break;
	}

	return RT_EOK;
}

/**
121
 * This function will resume the first thread in the list of a IPC object:
122 123 124
 * - remove the thread from suspend queue of IPC object
 * - put the thread into system ready queue
 *
125
 * @param list the thread list
126 127 128
 *
 * @return the operation status, RT_EOK on successful
 */
D
dzzxzz 已提交
129
rt_inline rt_err_t rt_ipc_list_resume(rt_list_t *list)
130
{
D
dzzxzz 已提交
131
	struct rt_thread *thread;
132 133

	/* get thread entry */
134
	thread = rt_list_entry(list->next, struct rt_thread, tlist);
135

D
dzzxzz 已提交
136
	RT_DEBUG_LOG(RT_DEBUG_IPC, ("resume thread:%s\n", thread->name));
137 138 139 140 141 142 143 144

	/* resume it */
	rt_thread_resume(thread);

	return RT_EOK;
}

/**
145 146
 * This function will resume all suspended threads in a list, including
 * suspend list of IPC object and private list of mailbox etc.
147
 *
148
 * @param list of the threads to resume
149 150 151
 *
 * @return the operation status, RT_EOK on successful
 */
D
dzzxzz 已提交
152
rt_inline rt_err_t rt_ipc_list_resume_all(rt_list_t *list)
153
{
D
dzzxzz 已提交
154
	struct rt_thread *thread;
155 156 157
	register rt_ubase_t temp;

	/* wakeup all suspend threads */
158
	while (!rt_list_isempty(list))
159 160 161 162 163
	{
		/* disable interrupt */
		temp = rt_hw_interrupt_disable();

		/* get next suspend thread */
164
		thread = rt_list_entry(list->next, struct rt_thread, tlist);
165 166 167 168 169 170 171 172
		/* set error code to RT_ERROR */
		thread->error = -RT_ERROR;

		/*
		 * resume thread
		 * In rt_thread_resume function, it will remove current thread from
		 * suspend list
		 */
173
		rt_thread_resume(thread);
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193

		/* enable interrupt */
		rt_hw_interrupt_enable(temp);
	}

	return RT_EOK;
}

#ifdef RT_USING_SEMAPHORE
/**
 * This function will initialize a semaphore and put it under control of resource
 * management.
 *
 * @param sem the semaphore object
 * @param name the name of semaphore
 * @param value the init value of semaphore
 * @param flag the flag of semaphore
 *
 * @return the operation status, RT_EOK on successful
 */
D
dzzxzz 已提交
194
rt_err_t rt_sem_init(rt_sem_t sem, const char *name, rt_uint32_t value, rt_uint8_t flag)
195 196 197 198 199 200 201 202 203 204
{
	RT_ASSERT(sem != RT_NULL);

	/* init object */
	rt_object_init(&(sem->parent.parent), RT_Object_Class_Semaphore, name);

	/* init ipc object */
	rt_ipc_object_init(&(sem->parent));

	/* set init value */
D
dzzxzz 已提交
205
	sem->value = value;
206 207 208 209 210 211

	/* set parent */
	sem->parent.parent.flag = flag;

	return RT_EOK;
}
212
RTM_EXPORT(rt_sem_init);
213 214 215 216 217 218 219 220 221 222

/**
 * This function will detach a semaphore from resource management
 *
 * @param sem the semaphore object
 *
 * @return the operation status, RT_EOK on successful
 *
 * @see rt_sem_delete
 */
D
dzzxzz 已提交
223
rt_err_t rt_sem_detach(rt_sem_t sem)
224 225 226 227
{
	RT_ASSERT(sem != RT_NULL);

	/* wakeup all suspend threads */
228
	rt_ipc_list_resume_all(&(sem->parent.suspend_thread));
229 230 231 232 233 234

	/* detach semaphore object */
	rt_object_detach(&(sem->parent.parent));

	return RT_EOK;
}
235
RTM_EXPORT(rt_sem_detach);
236 237 238 239 240 241 242 243 244 245 246 247 248

#ifdef RT_USING_HEAP
/**
 * This function will create a semaphore from system resource
 *
 * @param name the name of semaphore
 * @param value the init value of semaphore
 * @param flag the flag of semaphore
 *
 * @return the created semaphore, RT_NULL on error happen
 *
 * @see rt_sem_init
 */
D
dzzxzz 已提交
249
rt_sem_t rt_sem_create(const char *name, rt_uint32_t value, rt_uint8_t flag)
250 251 252
{
	rt_sem_t sem;

253
	RT_DEBUG_NOT_IN_INTERRUPT;
254

255
	/* allocate object */
D
dzzxzz 已提交
256
	sem = (rt_sem_t)rt_object_allocate(RT_Object_Class_Semaphore, name);
257 258
	if (sem == RT_NULL)
		return sem;
259 260 261 262 263

	/* init ipc object */
	rt_ipc_object_init(&(sem->parent));

	/* set init value */
264
	sem->value = value;
265 266 267 268 269 270

	/* set parent */
	sem->parent.parent.flag = flag;

	return sem;
}
271
RTM_EXPORT(rt_sem_create);
272 273 274 275 276 277 278 279 280 281

/**
 * This function will delete a semaphore object and release the memory
 *
 * @param sem the semaphore object
 *
 * @return the error code
 *
 * @see rt_sem_detach
 */
D
dzzxzz 已提交
282
rt_err_t rt_sem_delete(rt_sem_t sem)
283
{
284
	RT_DEBUG_NOT_IN_INTERRUPT;
285

286 287 288
	RT_ASSERT(sem != RT_NULL);

	/* wakeup all suspend threads */
289
	rt_ipc_list_resume_all(&(sem->parent.suspend_thread));
290 291 292 293 294 295

	/* delete semaphore object */
	rt_object_delete(&(sem->parent.parent));

	return RT_EOK;
}
296
RTM_EXPORT(rt_sem_delete);
297 298 299 300 301 302 303 304 305 306 307
#endif

/**
 * This function will take a semaphore, if the semaphore is unavailable, the
 * thread shall wait for a specified time.
 *
 * @param sem the semaphore object
 * @param time the waiting time
 *
 * @return the error code
 */
D
dzzxzz 已提交
308
rt_err_t rt_sem_take(rt_sem_t sem, rt_int32_t time)
309 310
{
	register rt_base_t temp;
D
dzzxzz 已提交
311
	struct rt_thread *thread;
312 313 314

	RT_ASSERT(sem != RT_NULL);

315
	RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(sem->parent.parent)));
316 317 318 319

	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

320 321 322 323
	RT_DEBUG_LOG(RT_DEBUG_IPC, ("thread %s take sem:%s, which value is: %d\n",
                                rt_thread_self()->name,
                                ((struct rt_object *)sem)->name,
                                sem->value));
324

325 326 327 328 329 330 331 332 333 334 335
	if (sem->value > 0)
	{
		/* semaphore is available */
		sem->value --;

		/* enable interrupt */
		rt_hw_interrupt_enable(temp);
	}
	else
	{
		/* no waiting, return with timeout */
336
		if (time == 0)
337 338
		{
			rt_hw_interrupt_enable(temp);
D
dzzxzz 已提交
339

340 341 342 343
			return -RT_ETIMEOUT;
		}
		else
		{
344 345
			/* current context checking */
			RT_DEBUG_NOT_IN_INTERRUPT;
346

347 348 349 350 351 352 353
			/* semaphore is unavailable, push to suspend list */
			/* get current thread */
			thread = rt_thread_self();

			/* reset thread error number */
			thread->error = RT_EOK;

354 355
			RT_DEBUG_LOG(RT_DEBUG_IPC, ("sem take: suspend thread - %s\n",
                                        thread->name));
356 357

			/* suspend thread */
358 359
			rt_ipc_list_suspend(&(sem->parent.suspend_thread),
				thread, sem->parent.parent.flag);
360 361 362 363

			/* has waiting time, start thread timer */
			if (time > 0)
			{
364 365
				RT_DEBUG_LOG(RT_DEBUG_IPC, ("set thread:%s to timer list\n",
                                            thread->name));
366

367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384
				/* reset the timeout of thread timer and start it */
				rt_timer_control(&(thread->thread_timer), RT_TIMER_CTRL_SET_TIME, &time);
				rt_timer_start(&(thread->thread_timer));
			}

			/* enable interrupt */
			rt_hw_interrupt_enable(temp);

			/* do schedule */
			rt_schedule();

			if (thread->error != RT_EOK)
			{
				return thread->error;
			}
		}
	}

385
	RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(sem->parent.parent)));
386 387 388

	return RT_EOK;
}
389
RTM_EXPORT(rt_sem_take);
390 391 392 393 394 395 396 397 398 399 400 401

/**
 * This function will try to take a semaphore and immediately return
 *
 * @param sem the semaphore object
 *
 * @return the error code
 */
rt_err_t rt_sem_trytake(rt_sem_t sem)
{
	return rt_sem_take(sem, 0);
}
402
RTM_EXPORT(rt_sem_trytake);
403 404 405 406 407 408 409 410 411 412 413 414

/**
 * This function will release a semaphore, if there are threads suspended on
 * semaphore, it will be waked up.
 *
 * @param sem the semaphore object
 *
 * @return the error code
 */
rt_err_t rt_sem_release(rt_sem_t sem)
{
	register rt_base_t temp;
415
	register rt_bool_t need_schedule;
416

417
	RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(sem->parent.parent)));
418

419 420
	need_schedule = RT_FALSE;

421 422 423
	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

424 425 426 427
	RT_DEBUG_LOG(RT_DEBUG_IPC, ("thread %s releases sem:%s, which value is: %d\n",
                                rt_thread_self()->name,
                                ((struct rt_object *)sem)->name,
                                sem->value));
428

D
dzzxzz 已提交
429
	if (!rt_list_isempty(&sem->parent.suspend_thread))
430
	{
431
		/* resume the suspended thread */
432
		rt_ipc_list_resume(&(sem->parent.suspend_thread));
433
		need_schedule = RT_TRUE;
434
	}
D
dzzxzz 已提交
435 436
	else
		sem->value ++; /* increase value */
437 438 439 440

	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

441
	/* resume a thread, re-schedule */
442 443
	if (need_schedule == RT_TRUE)
		rt_schedule();
444

445 446
	return RT_EOK;
}
447
RTM_EXPORT(rt_sem_release);
448 449 450 451 452 453 454 455 456 457

/**
 * This function can get or set some extra attributions of a semaphore object.
 *
 * @param sem the semaphore object
 * @param cmd the execution command
 * @param arg the execution argument
 *
 * @return the error code
 */
D
dzzxzz 已提交
458
rt_err_t rt_sem_control(rt_sem_t sem, rt_uint8_t cmd, void *arg)
459
{
460 461 462 463
	rt_ubase_t level;
	RT_ASSERT(sem != RT_NULL);

	if (cmd == RT_IPC_CMD_RESET)
464 465 466 467
	{
		rt_uint32_t value;

		/* get value */
468
		value = (rt_uint32_t)arg;
469 470 471 472
		/* disable interrupt */
		level = rt_hw_interrupt_disable();

		/* resume all waiting thread */
473
		rt_ipc_list_resume_all(&sem->parent.suspend_thread);
474 475

		/* set new value */
476
		sem->value = (rt_uint16_t)value;
477 478 479 480

		/* enable interrupt */
		rt_hw_interrupt_enable(level);

481 482
		rt_schedule();

483 484 485 486
		return RT_EOK;
	}

	return -RT_ERROR;
487
}
488
RTM_EXPORT(rt_sem_control);
489 490 491 492 493 494 495 496 497 498 499 500 501
#endif /* end of RT_USING_SEMAPHORE */

#ifdef RT_USING_MUTEX
/**
 * This function will initialize a mutex and put it under control of resource
 * management.
 *
 * @param mutex the mutex object
 * @param name the name of mutex
 * @param flag the flag of mutex
 *
 * @return the operation status, RT_EOK on successful
 */
D
dzzxzz 已提交
502
rt_err_t rt_mutex_init(rt_mutex_t mutex, const char *name, rt_uint8_t flag)
503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521
{
	RT_ASSERT(mutex != RT_NULL);

	/* init object */
	rt_object_init(&(mutex->parent.parent), RT_Object_Class_Mutex, name);

	/* init ipc object */
	rt_ipc_object_init(&(mutex->parent));

	mutex->value = 1;
	mutex->owner = RT_NULL;
	mutex->original_priority = 0xFF;
	mutex->hold  = 0;

	/* set flag */
	mutex->parent.parent.flag = flag;

	return RT_EOK;
}
522
RTM_EXPORT(rt_mutex_init);
523 524 525 526 527 528 529 530 531 532

/**
 * This function will detach a mutex from resource management
 *
 * @param mutex the mutex object
 *
 * @return the operation status, RT_EOK on successful
 *
 * @see rt_mutex_delete
 */
D
dzzxzz 已提交
533
rt_err_t rt_mutex_detach(rt_mutex_t mutex)
534 535 536 537
{
	RT_ASSERT(mutex != RT_NULL);

	/* wakeup all suspend threads */
538
	rt_ipc_list_resume_all(&(mutex->parent.suspend_thread));
539 540 541 542 543 544

	/* detach semaphore object */
	rt_object_detach(&(mutex->parent.parent));

	return RT_EOK;
}
545
RTM_EXPORT(rt_mutex_detach);
546 547 548 549 550 551 552 553 554 555 556 557

#ifdef RT_USING_HEAP
/**
 * This function will create a mutex from system resource
 *
 * @param name the name of mutex
 * @param flag the flag of mutex
 *
 * @return the created mutex, RT_NULL on error happen
 *
 * @see rt_mutex_init
 */
D
dzzxzz 已提交
558
rt_mutex_t rt_mutex_create(const char *name, rt_uint8_t flag)
559
{
D
dzzxzz 已提交
560
	struct rt_mutex *mutex;
561

562
	RT_DEBUG_NOT_IN_INTERRUPT;
563

564
	/* allocate object */
D
dzzxzz 已提交
565
	mutex = (rt_mutex_t)rt_object_allocate(RT_Object_Class_Mutex, name);
566 567
	if (mutex == RT_NULL)
		return mutex;
568 569 570 571 572 573 574 575 576 577 578 579 580 581

	/* init ipc object */
	rt_ipc_object_init(&(mutex->parent));

	mutex->value = 1;
	mutex->owner = RT_NULL;
	mutex->original_priority = 0xFF;
	mutex->hold  = 0;

	/* set flag */
	mutex->parent.parent.flag = flag;

	return mutex;
}
582
RTM_EXPORT(rt_mutex_create);
583 584 585 586 587 588 589 590 591 592

/**
 * This function will delete a mutex object and release the memory
 *
 * @param mutex the mutex object
 *
 * @return the error code
 *
 * @see rt_mutex_detach
 */
D
dzzxzz 已提交
593
rt_err_t rt_mutex_delete(rt_mutex_t mutex)
594
{
595
	RT_DEBUG_NOT_IN_INTERRUPT;
596

597 598 599
	RT_ASSERT(mutex != RT_NULL);

	/* wakeup all suspend threads */
600
	rt_ipc_list_resume_all(&(mutex->parent.suspend_thread));
601 602 603 604 605 606

	/* delete semaphore object */
	rt_object_delete(&(mutex->parent.parent));

	return RT_EOK;
}
607
RTM_EXPORT(rt_mutex_delete);
608 609 610 611 612 613 614 615 616 617 618
#endif

/**
 * This function will take a mutex, if the mutex is unavailable, the
 * thread shall wait for a specified time.
 *
 * @param mutex the mutex object
 * @param time the waiting time
 *
 * @return the error code
 */
D
dzzxzz 已提交
619
rt_err_t rt_mutex_take(rt_mutex_t mutex, rt_int32_t time)
620 621
{
	register rt_base_t temp;
D
dzzxzz 已提交
622
	struct rt_thread *thread;
623

624
	/* this function must not be used in interrupt even if time = 0 */
625
	RT_DEBUG_NOT_IN_INTERRUPT;
626

627 628 629 630 631 632 633 634
	RT_ASSERT(mutex != RT_NULL);

	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

	/* get current thread */
	thread = rt_thread_self();

635
	RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(mutex->parent.parent)));
636

637
	RT_DEBUG_LOG(RT_DEBUG_IPC,
638 639
                 ("mutex_take: current thread %s, mutex value: %d, hold: %d\n",
                  thread->name, mutex->value, mutex->hold));
640

641 642 643 644 645 646 647 648 649 650
	/* reset thread error */
	thread->error = RT_EOK;

	if (mutex->owner == thread)
	{
		/* it's the same thread */
		mutex->hold ++;
	}
	else
	{
651 652
		/* The value of mutex is 1 in initial status. Therefore, if the
		 * value is great than 0, it indicates the mutex is avaible.
653
		 */
654 655 656 657 658 659 660 661 662 663 664 665 666
		if (mutex->value > 0)
		{
			/* mutex is available */
			mutex->value --;

			/* set mutex owner and original priority */
			mutex->owner = thread;
			mutex->original_priority = thread->current_priority;
			mutex->hold ++;
		}
		else
		{
			/* no waiting, return with timeout */
D
dzzxzz 已提交
667
			if (time == 0)
668 669 670 671 672 673 674 675 676 677 678 679
			{
				/* set error as timeout */
				thread->error = -RT_ETIMEOUT;

				/* enable interrupt */
				rt_hw_interrupt_enable(temp);

				return -RT_ETIMEOUT;
			}
			else
			{
				/* mutex is unavailable, push to suspend list */
680 681
				RT_DEBUG_LOG(RT_DEBUG_IPC, ("mutex_take: suspend thread: %s\n",
                                            thread->name));
682

683 684 685 686 687 688 689 690 691
				/* change the owner thread priority of mutex */
				if (thread->current_priority < mutex->owner->current_priority)
				{
					/* change the owner thread priority */
					rt_thread_control(mutex->owner, RT_THREAD_CTRL_CHANGE_PRIORITY,
						&thread->current_priority);
				}

				/* suspend current thread */
692 693
				rt_ipc_list_suspend(&(mutex->parent.suspend_thread),
				thread, mutex->parent.parent.flag);
694 695 696 697

				/* has waiting time, start thread timer */
				if (time > 0)
				{
698
					RT_DEBUG_LOG(RT_DEBUG_IPC,
699 700
                                 ("mutex_take: start the timer of thread:%s\n",
                                  thread->name));
701

702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719
					/* reset the timeout of thread timer and start it */
					rt_timer_control(&(thread->thread_timer), RT_TIMER_CTRL_SET_TIME, &time);
					rt_timer_start(&(thread->thread_timer));
				}

				/* enable interrupt */
				rt_hw_interrupt_enable(temp);

				/* do schedule */
				rt_schedule();

				if (thread->error != RT_EOK)
				{
					/* return error */
					return thread->error;
				}
				else
				{
720
					/* the mutex is taken successfully. */
721 722 723 724 725 726 727 728 729 730
					/* disable interrupt */
					temp = rt_hw_interrupt_disable();
				}
			}
		}
	}

	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

731
	RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(mutex->parent.parent)));
732 733 734

	return RT_EOK;
}
735
RTM_EXPORT(rt_mutex_take);
736 737 738 739 740 741 742 743 744 745 746 747

/**
 * This function will release a mutex, if there are threads suspended on mutex,
 * it will be waked up.
 *
 * @param mutex the mutex object
 *
 * @return the error code
 */
rt_err_t rt_mutex_release(rt_mutex_t mutex)
{
	register rt_base_t temp;
D
dzzxzz 已提交
748
	struct rt_thread *thread;
749
	rt_bool_t need_schedule;
750

751
	need_schedule = RT_FALSE;
752

753 754 755
	/* get current thread */
	thread = rt_thread_self();

756 757 758
	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

759
	RT_DEBUG_LOG(RT_DEBUG_IPC,
760 761
                 ("mutex_release:current thread %s, mutex value: %d, hold: %d\n",
                  thread->name, mutex->value, mutex->hold));
762

763
	RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(mutex->parent.parent)));
764

765
	/* mutex only can be released by owner */
766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781
	if (thread != mutex->owner)
	{
		thread->error = -RT_ERROR;

		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

		return -RT_ERROR;
	}

	/* decrease hold */
	mutex->hold --;
	/* if no hold */
	if (mutex->hold == 0)
	{
		/* change the owner thread to original priority */
782
		if (mutex->original_priority != mutex->owner->current_priority)
783 784
		{
			rt_thread_control(mutex->owner, RT_THREAD_CTRL_CHANGE_PRIORITY,
785
				&(mutex->original_priority));
786 787
		}

788
		/* wakeup suspended thread */
D
dzzxzz 已提交
789
		if (!rt_list_isempty(&mutex->parent.suspend_thread))
790
		{
791
			/* get suspended thread */
792 793
			thread = rt_list_entry(mutex->parent.suspend_thread.next, struct rt_thread, tlist);

794 795
			RT_DEBUG_LOG(RT_DEBUG_IPC, ("mutex_release: resume thread: %s\n",
                                        thread->name));
796

797 798 799 800
			/* set new owner and priority */
			mutex->owner = thread;
			mutex->original_priority = thread->current_priority;
			mutex->hold ++;
801 802

			/* resume thread */
803
			rt_ipc_list_resume(&(mutex->parent.suspend_thread));
804 805

			need_schedule = RT_TRUE;
806
		}
807 808
		else
		{
809 810 811
			/* increase value */
			mutex->value ++;

B
bernard.xiong 已提交
812 813
			/* clear owner */
			mutex->owner = RT_NULL;
814
			mutex->original_priority = 0xff;
815
		}
816 817 818 819 820
	}

	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

821
	/* perform a schedule */
822 823
	if (need_schedule == RT_TRUE)
		rt_schedule();
B
bernard.xiong 已提交
824

825 826
	return RT_EOK;
}
827
RTM_EXPORT(rt_mutex_release);
828 829 830 831 832 833 834 835 836 837

/**
 * This function can get or set some extra attributions of a mutex object.
 *
 * @param mutex the mutex object
 * @param cmd the execution command
 * @param arg the execution argument
 *
 * @return the error code
 */
D
dzzxzz 已提交
838
rt_err_t rt_mutex_control(rt_mutex_t mutex, rt_uint8_t cmd, void *arg)
839
{
840
	return -RT_ERROR;
841
}
842
RTM_EXPORT(rt_mutex_control);
843 844 845 846 847 848 849 850 851 852 853 854 855
#endif /* end of RT_USING_MUTEX */

#ifdef RT_USING_EVENT
/**
 * This function will initialize an event and put it under control of resource
 * management.
 *
 * @param event the event object
 * @param name the name of event
 * @param flag the flag of event
 *
 * @return the operation status, RT_EOK on successful
 */
D
dzzxzz 已提交
856
rt_err_t rt_event_init(rt_event_t event, const char *name, rt_uint8_t flag)
857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873
{
	RT_ASSERT(event != RT_NULL);

	/* init object */
	rt_object_init(&(event->parent.parent), RT_Object_Class_Event, name);

	/* set parent flag */
	event->parent.parent.flag = flag;

	/* init ipc object */
	rt_ipc_object_init(&(event->parent));

	/* init event */
	event->set = 0;

	return RT_EOK;
}
874
RTM_EXPORT(rt_event_init);
875 876 877 878 879 880 881 882 883 884 885 886 887 888

/**
 * This function will detach an event object from resource management
 *
 * @param event the event object
 *
 * @return the operation status, RT_EOK on successful
 */
rt_err_t rt_event_detach(rt_event_t event)
{
	/* parameter check */
	RT_ASSERT(event != RT_NULL);

	/* resume all suspended thread */
889
	rt_ipc_list_resume_all(&(event->parent.suspend_thread));
890 891 892 893 894 895

	/* detach event object */
	rt_object_detach(&(event->parent.parent));

	return RT_EOK;
}
896
RTM_EXPORT(rt_event_detach);
897 898 899 900 901 902 903 904 905 906

#ifdef RT_USING_HEAP
/**
 * This function will create an event object from system resource
 *
 * @param name the name of event
 * @param flag the flag of event
 *
 * @return the created event, RT_NULL on error happen
 */
D
dzzxzz 已提交
907
rt_event_t rt_event_create(const char *name, rt_uint8_t flag)
908 909 910
{
	rt_event_t event;

911
	RT_DEBUG_NOT_IN_INTERRUPT;
912

913
	/* allocate object */
D
dzzxzz 已提交
914
	event = (rt_event_t)rt_object_allocate(RT_Object_Class_Event, name);
915 916
	if (event == RT_NULL)
		return event;
917 918 919 920 921 922 923 924 925 926 927 928

	/* set parent */
	event->parent.parent.flag = flag;

	/* init ipc object */
	rt_ipc_object_init(&(event->parent));

	/* init event */
	event->set = 0;

	return event;
}
929
RTM_EXPORT(rt_event_create);
930 931 932 933 934 935 936 937

/**
 * This function will delete an event object and release the memory
 *
 * @param event the event object
 *
 * @return the error code
 */
D
dzzxzz 已提交
938
rt_err_t rt_event_delete(rt_event_t event)
939 940 941 942
{
	/* parameter check */
	RT_ASSERT(event != RT_NULL);

943
	RT_DEBUG_NOT_IN_INTERRUPT;
944

945
	/* resume all suspended thread */
946
	rt_ipc_list_resume_all(&(event->parent.suspend_thread));
947 948 949 950 951 952

	/* delete event object */
	rt_object_delete(&(event->parent.parent));

	return RT_EOK;
}
953
RTM_EXPORT(rt_event_delete);
954 955 956 957 958 959 960 961 962 963 964 965 966
#endif

/**
 * This function will send an event to the event object, if there are threads
 * suspended on event object, it will be waked up.
 *
 * @param event the event object
 * @param set the event set
 *
 * @return the error code
 */
rt_err_t rt_event_send(rt_event_t event, rt_uint32_t set)
{
D
dzzxzz 已提交
967 968
	struct rt_list_node *n;
	struct rt_thread *thread;
969 970
	register rt_ubase_t level;
	register rt_base_t status;
B
bernard.xiong 已提交
971
	rt_bool_t need_schedule;
972 973 974

	/* parameter check */
	RT_ASSERT(event != RT_NULL);
975 976
	if (set == 0)
		return -RT_ERROR;
977

B
bernard.xiong 已提交
978
	need_schedule = RT_FALSE;
979
	RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(event->parent.parent)));
980 981 982 983 984 985

	/* disable interrupt */
	level = rt_hw_interrupt_disable();

	/* set event */
	event->set |= set;
986

D
dzzxzz 已提交
987
	if (!rt_list_isempty(&event->parent.suspend_thread))
988 989 990 991 992 993 994 995 996 997 998 999 1000
	{
		/* search thread list to resume thread */
		n = event->parent.suspend_thread.next;
		while (n != &(event->parent.suspend_thread))
		{
			/* get thread */
			thread = rt_list_entry(n, struct rt_thread, tlist);

			status = -RT_ERROR;
			if (thread->event_info & RT_EVENT_FLAG_AND)
			{
				if ((thread->event_set & event->set) == thread->event_set)
				{
1001
					/* received an AND event */
1002 1003 1004 1005 1006 1007 1008
					status = RT_EOK;
				}
			}
			else if (thread->event_info & RT_EVENT_FLAG_OR)
			{
				if (thread->event_set & event->set)
				{
1009 1010 1011 1012
					/* save recieved event set */
					thread->event_set = thread->event_set & event->set;

					/* received an OR event */
1013 1014 1015 1016
					status = RT_EOK;
				}
			}

B
bernard.xiong 已提交
1017
			/* move node to the next */
1018 1019 1020 1021 1022
			n = n->next;

			/* condition is satisfied, resume thread */
			if (status == RT_EOK)
			{
1023 1024 1025 1026
				/* clear event */
				if (thread->event_info & RT_EVENT_FLAG_CLEAR)
					event->set &= ~thread->event_set;

1027 1028 1029
				/* resume thread, and thread list breaks out */
				rt_thread_resume(thread);

B
bernard.xiong 已提交
1030 1031
				/* need do a scheduling */
				need_schedule = RT_TRUE;
1032 1033 1034 1035 1036 1037 1038 1039
			}
		}
	}

	/* enable interrupt */
	rt_hw_interrupt_enable(level);

	/* do a schedule */
B
bernard.xiong 已提交
1040
	if (need_schedule == RT_TRUE)
1041
		rt_schedule();
1042 1043 1044

	return RT_EOK;
}
1045
RTM_EXPORT(rt_event_send);
1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058

/**
 * This function will receive an event from event object, if the event is unavailable,
 * the thread shall wait for a specified time.
 *
 * @param event the fast event object
 * @param set the interested event set
 * @param option the receive option
 * @param timeout the waiting time
 * @param recved the received event
 *
 * @return the error code
 */
D
dzzxzz 已提交
1059
rt_err_t rt_event_recv(rt_event_t event, rt_uint32_t set, rt_uint8_t option, rt_int32_t timeout, rt_uint32_t *recved)
1060
{
D
dzzxzz 已提交
1061
	struct rt_thread *thread;
1062 1063 1064
	register rt_ubase_t level;
	register rt_base_t status;

1065
	RT_DEBUG_NOT_IN_INTERRUPT;
1066

1067 1068
	/* parameter check */
	RT_ASSERT(event != RT_NULL);
1069 1070
	if (set == 0)
		return -RT_ERROR;
1071 1072 1073

	/* init status */
	status = -RT_ERROR;
B
bernard.xiong 已提交
1074 1075 1076 1077
	/* get current thread */
	thread = rt_thread_self();
	/* reset thread error */
	thread->error = RT_EOK;
1078

1079
	RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(event->parent.parent)));
1080 1081 1082 1083 1084 1085 1086

	/* disable interrupt */
	level = rt_hw_interrupt_disable();

	/* check event set */
	if (option & RT_EVENT_FLAG_AND)
	{
1087 1088
		if ((event->set & set) == set)
			status = RT_EOK;
1089 1090 1091
	}
	else if (option & RT_EVENT_FLAG_OR)
	{
1092 1093
		if (event->set & set)
			status = RT_EOK;
1094 1095 1096 1097 1098
	}

	if (status == RT_EOK)
	{
		/* set received event */
1099
		*recved = (event->set & set);
1100 1101

		/* received event */
1102 1103
		if (option & RT_EVENT_FLAG_CLEAR)
			event->set &= ~set;
1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116
	}
	else if (timeout == 0)
	{
		/* no waiting */
		thread->error = -RT_ETIMEOUT;
	}
	else
	{
		/* fill thread event info */
		thread->event_set  = set;
		thread->event_info = option;

		/* put thread to suspended thread list */
D
dzzxzz 已提交
1117
		rt_ipc_list_suspend(&(event->parent.suspend_thread), thread, event->parent.parent.flag);
1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134

		/* if there is a waiting timeout, active thread timer */
		if (timeout > 0)
		{
			/* reset the timeout of thread timer and start it */
			rt_timer_control(&(thread->thread_timer), RT_TIMER_CTRL_SET_TIME, &timeout);
			rt_timer_start(&(thread->thread_timer));
		}

		/* enable interrupt */
		rt_hw_interrupt_enable(level);

		/* do a schedule */
		rt_schedule();

		if (thread->error != RT_EOK)
		{
D
dzzxzz 已提交
1135
			/* return error */
1136 1137 1138
			return thread->error;
		}

1139
		/* received an event, disable interrupt to protect */
1140 1141
		level = rt_hw_interrupt_disable();

1142 1143
		/* set received event */
		*recved = thread->event_set;
1144 1145 1146 1147 1148
	}

	/* enable interrupt */
	rt_hw_interrupt_enable(level);

1149
	RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(event->parent.parent)));
1150 1151 1152

	return thread->error;
}
1153
RTM_EXPORT(rt_event_recv);
1154 1155 1156 1157 1158 1159 1160 1161 1162 1163

/**
 * This function can get or set some extra attributions of an event object.
 *
 * @param event the event object
 * @param cmd the execution command
 * @param arg the execution argument
 *
 * @return the error code
 */
D
dzzxzz 已提交
1164
rt_err_t rt_event_control(rt_event_t event, rt_uint8_t cmd, void *arg)
1165
{
1166 1167 1168 1169 1170 1171 1172 1173 1174
	rt_ubase_t level;
	RT_ASSERT(event != RT_NULL);

	if (cmd == RT_IPC_CMD_RESET)
	{
		/* disable interrupt */
		level = rt_hw_interrupt_disable();

		/* resume all waiting thread */
1175
		rt_ipc_list_resume_all(&event->parent.suspend_thread);
1176 1177 1178 1179 1180 1181 1182

		/* init event set */
		event->set = 0;

		/* enable interrupt */
		rt_hw_interrupt_enable(level);

1183 1184
		rt_schedule();

1185 1186 1187 1188
		return RT_EOK;
	}

	return -RT_ERROR;
1189
}
1190
RTM_EXPORT(rt_event_control); 
1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205
#endif /* end of RT_USING_EVENT */

#ifdef RT_USING_MAILBOX
/**
 * This function will initialize a mailbox and put it under control of resource
 * management.
 *
 * @param mb the mailbox object
 * @param name the name of mailbox
 * @param msgpool the begin address of buffer to save received mail
 * @param size the size of mailbox
 * @param flag the flag of mailbox
 *
 * @return the operation status, RT_EOK on successful
 */
D
dzzxzz 已提交
1206
rt_err_t rt_mb_init(rt_mailbox_t mb, const char *name, void *msgpool, rt_size_t size, rt_uint8_t flag)
1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219
{
	RT_ASSERT(mb != RT_NULL);

	/* init object */
	rt_object_init(&(mb->parent.parent), RT_Object_Class_MailBox, name);

	/* set parent flag */
	mb->parent.parent.flag = flag;

	/* init ipc object */
	rt_ipc_object_init(&(mb->parent));

	/* init mailbox */
D
dzzxzz 已提交
1220 1221 1222 1223 1224
	mb->msg_pool   = msgpool;
	mb->size       = size;
	mb->entry      = 0;
	mb->in_offset  = 0;
	mb->out_offset = 0;
1225

1226
	/* init an additional list of sender suspend thread */
1227 1228
	rt_list_init(&(mb->suspend_sender_thread));

1229 1230
	return RT_EOK;
}
1231
RTM_EXPORT(rt_mb_init);
1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245

/**
 * This function will detach a mailbox from resource management
 *
 * @param mb the mailbox object
 *
 * @return the operation status, RT_EOK on successful
 */
rt_err_t rt_mb_detach(rt_mailbox_t mb)
{
	/* parameter check */
	RT_ASSERT(mb != RT_NULL);

	/* resume all suspended thread */
1246 1247 1248
	rt_ipc_list_resume_all(&(mb->parent.suspend_thread));
	/* also resume all mailbox private suspended thread */
	rt_ipc_list_resume_all(&(mb->suspend_sender_thread));
1249 1250 1251 1252 1253 1254

	/* detach mailbox object */
	rt_object_detach(&(mb->parent.parent));

	return RT_EOK;
}
1255
RTM_EXPORT(rt_mb_detach);
1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266

#ifdef RT_USING_HEAP
/**
 * This function will create a mailbox object from system resource
 *
 * @param name the name of mailbox
 * @param size the size of mailbox
 * @param flag the flag of mailbox
 *
 * @return the created mailbox, RT_NULL on error happen
 */
D
dzzxzz 已提交
1267
rt_mailbox_t rt_mb_create(const char *name, rt_size_t size, rt_uint8_t flag)
1268 1269 1270
{
	rt_mailbox_t mb;

1271
	RT_DEBUG_NOT_IN_INTERRUPT;
1272

1273
	/* allocate object */
D
dzzxzz 已提交
1274
	mb = (rt_mailbox_t)rt_object_allocate(RT_Object_Class_MailBox, name);
1275 1276
	if (mb == RT_NULL)
		return mb;
1277 1278 1279 1280 1281 1282 1283 1284

	/* set parent */
	mb->parent.parent.flag = flag;

	/* init ipc object */
	rt_ipc_object_init(&(mb->parent));

	/* init mailbox */
D
dzzxzz 已提交
1285 1286
	mb->size     = size;
	mb->msg_pool = rt_malloc(mb->size * sizeof(rt_uint32_t));
1287 1288 1289 1290 1291 1292 1293
	if (mb->msg_pool == RT_NULL)
	{
		/* delete mailbox object */
		rt_object_delete(&(mb->parent.parent));

		return RT_NULL;
	}
1294
	mb->entry      = 0;
D
dzzxzz 已提交
1295 1296
	mb->in_offset  = 0;
	mb->out_offset = 0;
1297

1298 1299 1300
	/* init an additional list of sender suspend thread */
	rt_list_init(&(mb->suspend_sender_thread));

1301 1302
	return mb;
}
1303
RTM_EXPORT(rt_mb_create);
1304 1305 1306 1307 1308 1309 1310 1311

/**
 * This function will delete a mailbox object and release the memory
 *
 * @param mb the mailbox object
 *
 * @return the error code
 */
D
dzzxzz 已提交
1312
rt_err_t rt_mb_delete(rt_mailbox_t mb)
M
mbbill@gmail.com 已提交
1313
{
1314
	RT_DEBUG_NOT_IN_INTERRUPT;
1315

1316 1317 1318 1319
	/* parameter check */
	RT_ASSERT(mb != RT_NULL);

	/* resume all suspended thread */
1320
	rt_ipc_list_resume_all(&(mb->parent.suspend_thread));
1321

1322 1323
	/* also resume all mailbox private suspended thread */
	rt_ipc_list_resume_all(&(mb->suspend_sender_thread));
1324

qiuyiuestc's avatar
qiuyiuestc 已提交
1325
#if defined(RT_USING_MODULE) && defined(RT_USING_SLAB)
qiuyiuestc's avatar
qiuyiuestc 已提交
1326
	/* the mb object belongs to an application module */
D
dzzxzz 已提交
1327
	if (mb->parent.parent.flag & RT_OBJECT_FLAG_MODULE)
qiuyiuestc's avatar
qiuyiuestc 已提交
1328 1329 1330 1331
		rt_module_free(mb->parent.parent.module_id, mb->msg_pool);
	else
#endif

1332 1333 1334 1335 1336 1337 1338 1339
	/* free mailbox pool */
	rt_free(mb->msg_pool);

	/* delete mailbox object */
	rt_object_delete(&(mb->parent.parent));

	return RT_EOK;
}
1340
RTM_EXPORT(rt_mb_delete);
1341 1342 1343
#endif

/**
1344 1345
 * This function will send a mail to mailbox object. If the mailbox is full,
 * current thread will be suspended until timeout.
1346 1347 1348
 *
 * @param mb the mailbox object
 * @param value the mail
1349
 * @param timeout the waiting time
1350 1351 1352
 *
 * @return the error code
 */
D
dzzxzz 已提交
1353
rt_err_t rt_mb_send_wait(rt_mailbox_t mb, rt_uint32_t value, rt_int32_t timeout)
1354
{
D
dzzxzz 已提交
1355
	struct rt_thread *thread;
1356
	register rt_ubase_t temp;
1357
	rt_uint32_t tick_delta;
1358 1359

	/* parameter check */
1360
	RT_ASSERT(mb != RT_NULL);
1361

B
bernard.xiong@gmail.com 已提交
1362
	/* initialize delta tick */
1363
	tick_delta = 0;
B
bernard.xiong@gmail.com 已提交
1364 1365 1366
	/* get current thread */
	thread = rt_thread_self();

1367
	RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(mb->parent.parent)));
1368 1369 1370 1371

	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

1372 1373 1374 1375
	/* for non-blocking call */
	if (mb->entry == mb->size && timeout == 0)
	{
		rt_hw_interrupt_enable(temp);
D
dzzxzz 已提交
1376

1377 1378 1379
		return -RT_EFULL;
	}

1380
	/* mailbox is full */
1381
	while (mb->entry == mb->size)
1382
	{
1383 1384 1385 1386 1387 1388
		/* reset error number in thread */
		thread->error = RT_EOK;

		/* no waiting, return timeout */
		if (timeout == 0)
		{
D
dzzxzz 已提交
1389 1390
			/* enable interrupt */
			rt_hw_interrupt_enable(temp);
D
dzzxzz 已提交
1391

D
dzzxzz 已提交
1392 1393
			return -RT_EFULL;
		}
B
bernard.xiong@gmail.com 已提交
1394

B
bernard.xiong@gmail.com 已提交
1395
		RT_DEBUG_NOT_IN_INTERRUPT;
1396
		/* suspend current thread */
D
dzzxzz 已提交
1397
		rt_ipc_list_suspend(&(mb->suspend_sender_thread), thread, mb->parent.parent.flag);
1398 1399 1400 1401 1402 1403 1404

		/* has waiting time, start thread timer */
		if (timeout > 0)
		{
			/* get the start tick of timer */
			tick_delta = rt_tick_get();

1405 1406
			RT_DEBUG_LOG(RT_DEBUG_IPC, ("mb_send_wait: start timer of thread:%s\n",
                                        thread->name));
1407

1408 1409 1410 1411 1412
			/* reset the timeout of thread timer and start it */
			rt_timer_control(&(thread->thread_timer), RT_TIMER_CTRL_SET_TIME, &timeout);
			rt_timer_start(&(thread->thread_timer));
		}

1413 1414 1415
		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

1416 1417 1418 1419 1420 1421
		/* re-schedule */
		rt_schedule();

		/* resume from suspend state */
		if (thread->error != RT_EOK)
		{
D
dzzxzz 已提交
1422
			/* return error */
1423 1424 1425 1426 1427 1428
			return thread->error;
		}

		/* disable interrupt */
		temp = rt_hw_interrupt_disable();

B
bernard.xiong@gmail.com 已提交
1429
		/* if it's not waiting forever and then re-calculate timeout tick */
1430 1431 1432 1433
		if (timeout > 0)
		{
			tick_delta = rt_tick_get() - tick_delta;
			timeout -= tick_delta;
1434 1435
			if (timeout < 0)
				timeout = 0;
1436
		}
1437 1438 1439 1440 1441 1442
	}

	/* set ptr */
	mb->msg_pool[mb->in_offset] = value;
	/* increase input offset */
	++ mb->in_offset;
1443 1444
	if (mb->in_offset >= mb->size)
		mb->in_offset = 0;
1445 1446 1447
	/* increase message entry */
	mb->entry ++;

1448
	/* resume suspended thread */
B
bernard.xiong@gmail.com 已提交
1449
	if (!rt_list_isempty(&mb->parent.suspend_thread))
1450
	{
1451
		rt_ipc_list_resume(&(mb->parent.suspend_thread));
1452 1453 1454 1455 1456

		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

		rt_schedule();
D
dzzxzz 已提交
1457

1458 1459 1460 1461 1462 1463 1464 1465
		return RT_EOK;
	}

	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

	return RT_EOK;
}
1466
RTM_EXPORT(rt_mb_send_wait);
1467

1468 1469 1470 1471 1472 1473 1474 1475 1476 1477
/**
 * This function will send a mail to mailbox object, if there are threads suspended
 * on mailbox object, it will be waked up. This function will return immediately, if
 * you want blocking send, use rt_mb_send_wait instead.
 *
 * @param mb the mailbox object
 * @param value the mail
 *
 * @return the error code
 */
D
dzzxzz 已提交
1478
rt_err_t rt_mb_send(rt_mailbox_t mb, rt_uint32_t value)
1479
{
D
dzzxzz 已提交
1480
	return rt_mb_send_wait(mb, value, 0);
1481
}
1482
RTM_EXPORT(rt_mb_send);
1483

1484 1485 1486 1487 1488 1489 1490 1491 1492 1493
/**
 * This function will receive a mail from mailbox object, if there is no mail in
 * mailbox object, the thread shall wait for a specified time.
 *
 * @param mb the mailbox object
 * @param value the received mail will be saved in
 * @param timeout the waiting time
 *
 * @return the error code
 */
D
dzzxzz 已提交
1494
rt_err_t rt_mb_recv(rt_mailbox_t mb, rt_uint32_t *value, rt_int32_t timeout)
1495
{
D
dzzxzz 已提交
1496
	struct rt_thread *thread;
1497
	register rt_ubase_t temp;
1498
	rt_uint32_t tick_delta;
1499 1500 1501 1502

	/* parameter check */
	RT_ASSERT(mb != RT_NULL);

B
bernard.xiong@gmail.com 已提交
1503 1504 1505 1506 1507
	/* initialize delta tick */
	tick_delta = 0;
	/* get current thread */
	thread = rt_thread_self();

1508
	RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(mb->parent.parent)));
1509 1510 1511 1512

	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

1513 1514 1515 1516
	/* for non-blocking call */
	if (mb->entry == 0 && timeout == 0)
	{
		rt_hw_interrupt_enable(temp);
D
dzzxzz 已提交
1517
		
1518 1519 1520
		return -RT_ETIMEOUT;
	}

1521
	/* mailbox is empty */
1522
	while (mb->entry == 0)
1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533
	{
		/* reset error number in thread */
		thread->error = RT_EOK;

		/* no waiting, return timeout */
		if (timeout == 0)
		{
			/* enable interrupt */
			rt_hw_interrupt_enable(temp);

			thread->error = -RT_ETIMEOUT;
D
dzzxzz 已提交
1534
			
1535 1536 1537
			return -RT_ETIMEOUT;
		}

B
bernard.xiong@gmail.com 已提交
1538
		RT_DEBUG_NOT_IN_INTERRUPT;
1539
		/* suspend current thread */
D
dzzxzz 已提交
1540
		rt_ipc_list_suspend(&(mb->parent.suspend_thread), thread, mb->parent.parent.flag);
1541 1542 1543 1544

		/* has waiting time, start thread timer */
		if (timeout > 0)
		{
1545 1546 1547
			/* get the start tick of timer */
			tick_delta = rt_tick_get();

1548 1549
			RT_DEBUG_LOG(RT_DEBUG_IPC, ("mb_recv: start timer of thread:%s\n",
                                        thread->name));
1550

1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561
			/* reset the timeout of thread timer and start it */
			rt_timer_control(&(thread->thread_timer), RT_TIMER_CTRL_SET_TIME, &timeout);
			rt_timer_start(&(thread->thread_timer));
		}

		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

		/* re-schedule */
		rt_schedule();

1562
		/* resume from suspend state */
1563 1564
		if (thread->error != RT_EOK)
		{
D
dzzxzz 已提交
1565
			/* return error */
1566 1567 1568 1569 1570
			return thread->error;
		}

		/* disable interrupt */
		temp = rt_hw_interrupt_disable();
1571

B
bernard.xiong@gmail.com 已提交
1572
		/* if it's not waiting forever and then re-calculate timeout tick */
1573 1574 1575 1576
		if (timeout > 0)
		{
			tick_delta = rt_tick_get() - tick_delta;
			timeout -= tick_delta;
1577 1578
			if (timeout < 0)
				timeout = 0;
1579
		}
1580 1581 1582 1583 1584 1585
	}

	/* fill ptr */
	*value = mb->msg_pool[mb->out_offset];

	/* increase output offset */
D
dzzxzz 已提交
1586
	++ mb->out_offset;
1587 1588
	if (mb->out_offset >= mb->size)
		mb->out_offset = 0;
1589 1590 1591
	/* decrease message entry */
	mb->entry --;

1592
	/* resume suspended thread */
B
bernard.xiong@gmail.com 已提交
1593
	if (!rt_list_isempty(&(mb->suspend_sender_thread)))
1594 1595 1596 1597 1598 1599
	{
		rt_ipc_list_resume(&(mb->suspend_sender_thread));

		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

1600 1601
		RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(mb->parent.parent)));

1602 1603 1604 1605 1606
		rt_schedule();

		return RT_EOK;
	}

1607 1608 1609
	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

1610
	RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(mb->parent.parent)));
1611 1612 1613

	return RT_EOK;
}
1614
RTM_EXPORT(rt_mb_recv);
1615 1616 1617 1618 1619 1620 1621 1622 1623 1624

/**
 * This function can get or set some extra attributions of a mailbox object.
 *
 * @param mb the mailbox object
 * @param cmd the execution command
 * @param arg the execution argument
 *
 * @return the error code
 */
D
dzzxzz 已提交
1625
rt_err_t rt_mb_control(rt_mailbox_t mb, rt_uint8_t cmd, void *arg)
1626
{
1627 1628 1629 1630 1631 1632 1633 1634 1635
	rt_ubase_t level;
	RT_ASSERT(mb != RT_NULL);

	if (cmd == RT_IPC_CMD_RESET)
	{
		/* disable interrupt */
		level = rt_hw_interrupt_disable();

		/* resume all waiting thread */
1636
		rt_ipc_list_resume_all(&(mb->parent.suspend_thread));
1637 1638 1639
		/* also resume all mailbox private suspended thread */
		rt_ipc_list_resume_all(&(mb->suspend_sender_thread));

1640
		/* re-init mailbox */
D
dzzxzz 已提交
1641 1642 1643
		mb->entry      = 0;
		mb->in_offset  = 0;
		mb->out_offset = 0;
1644

1645 1646 1647
		/* enable interrupt */
		rt_hw_interrupt_enable(level);

1648 1649
		rt_schedule();

1650 1651 1652 1653
		return RT_EOK;
	}

	return -RT_ERROR;
1654
}
1655
RTM_EXPORT(rt_mb_control); 
1656 1657 1658 1659 1660
#endif /* end of RT_USING_MAILBOX */

#ifdef RT_USING_MESSAGEQUEUE
struct rt_mq_message
{
D
dzzxzz 已提交
1661
	struct rt_mq_message *next;
1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676
};

/**
 * This function will initialize a message queue and put it under control of resource
 * management.
 *
 * @param mq the message object
 * @param name the name of message queue
 * @param msgpool the beginning address of buffer to save messages
 * @param msg_size the maximum size of message
 * @param pool_size the size of buffer to save messages
 * @param flag the flag of message queue
 *
 * @return the operation status, RT_EOK on successful
 */
D
dzzxzz 已提交
1677
rt_err_t rt_mq_init(rt_mq_t mq, const char *name, void *msgpool, rt_size_t msg_size, rt_size_t pool_size, rt_uint8_t flag)
1678
{
D
dzzxzz 已提交
1679
	struct rt_mq_message *head;
1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694
	register rt_base_t temp;

	/* parameter check */
	RT_ASSERT(mq != RT_NULL);

	/* init object */
	rt_object_init(&(mq->parent.parent), RT_Object_Class_MessageQueue, name);

	/* set parent flag */
	mq->parent.parent.flag = flag;

	/* init ipc object */
	rt_ipc_object_init(&(mq->parent));

	/* set messasge pool */
D
dzzxzz 已提交
1695
	mq->msg_pool = msgpool;
1696 1697

	/* get correct message size */
1698
	mq->msg_size = RT_ALIGN(msg_size, RT_ALIGN_SIZE);
D
dzzxzz 已提交
1699
	mq->max_msgs = pool_size / (mq->msg_size + sizeof(struct rt_mq_message));
1700 1701 1702 1703 1704 1705 1706 1707 1708

	/* init message list */
	mq->msg_queue_head = RT_NULL;
	mq->msg_queue_tail = RT_NULL;

	/* init message empty list */
	mq->msg_queue_free = RT_NULL;
	for (temp = 0; temp < mq->max_msgs; temp ++)
	{
D
dzzxzz 已提交
1709
		head = (struct rt_mq_message *)((rt_uint8_t *)mq->msg_pool +
1710 1711 1712 1713 1714 1715
			temp * (mq->msg_size + sizeof(struct rt_mq_message)));
		head->next = mq->msg_queue_free;
		mq->msg_queue_free = head;
	}

	/* the initial entry is zero */
D
dzzxzz 已提交
1716
	mq->entry = 0;
1717 1718 1719

	return RT_EOK;
}
1720
RTM_EXPORT(rt_mq_init);
1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734

/**
 * This function will detach a message queue object from resource management
 *
 * @param mq the message queue object
 *
 * @return the operation status, RT_EOK on successful
 */
rt_err_t rt_mq_detach(rt_mq_t mq)
{
	/* parameter check */
	RT_ASSERT(mq != RT_NULL);

	/* resume all suspended thread */
1735
	rt_ipc_list_resume_all(&mq->parent.suspend_thread);
1736 1737 1738 1739 1740 1741

	/* detach message queue object */
	rt_object_detach(&(mq->parent.parent));

	return RT_EOK;
}
1742
RTM_EXPORT(rt_mq_detach);
1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754

#ifdef RT_USING_HEAP
/**
 * This function will create a message queue object from system resource
 *
 * @param name the name of message queue
 * @param msg_size the size of message
 * @param max_msgs the maximum number of message in queue
 * @param flag the flag of message queue
 *
 * @return the created message queue, RT_NULL on error happen
 */
D
dzzxzz 已提交
1755
rt_mq_t rt_mq_create(const char *name, rt_size_t msg_size, rt_size_t max_msgs, rt_uint8_t flag)
1756
{
D
dzzxzz 已提交
1757 1758
	struct rt_messagequeue *mq;
	struct rt_mq_message *head;
1759 1760
	register rt_base_t temp;

1761
	RT_DEBUG_NOT_IN_INTERRUPT;
1762

1763
	/* allocate object */
D
dzzxzz 已提交
1764
	mq = (rt_mq_t)rt_object_allocate(RT_Object_Class_MessageQueue, name);
1765 1766
	if (mq == RT_NULL)
		return mq;
1767 1768 1769 1770 1771 1772 1773 1774 1775 1776

	/* set parent */
	mq->parent.parent.flag = flag;

	/* init ipc object */
	rt_ipc_object_init(&(mq->parent));

	/* init message queue */

	/* get correct message size */
D
dzzxzz 已提交
1777 1778
	mq->msg_size = RT_ALIGN(msg_size, RT_ALIGN_SIZE);
	mq->max_msgs = max_msgs;
1779 1780

	/* allocate message pool */
D
dzzxzz 已提交
1781
	mq->msg_pool = rt_malloc((mq->msg_size + sizeof(struct rt_mq_message))* mq->max_msgs);
1782 1783 1784
	if (mq->msg_pool == RT_NULL)
	{
		rt_mq_delete(mq);
D
dzzxzz 已提交
1785
		
1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803
		return RT_NULL;
	}

	/* init message list */
	mq->msg_queue_head = RT_NULL;
	mq->msg_queue_tail = RT_NULL;

	/* init message empty list */
	mq->msg_queue_free = RT_NULL;
	for (temp = 0; temp < mq->max_msgs; temp ++)
	{
		head = (struct rt_mq_message*)((rt_uint8_t*)mq->msg_pool +
			temp * (mq->msg_size + sizeof(struct rt_mq_message)));
		head->next = mq->msg_queue_free;
		mq->msg_queue_free = head;
	}

	/* the initial entry is zero */
D
dzzxzz 已提交
1804
	mq->entry = 0;
1805 1806 1807

	return mq;
}
1808
RTM_EXPORT(rt_mq_create);
1809 1810 1811 1812 1813 1814 1815 1816

/**
 * This function will delete a message queue object and release the memory
 *
 * @param mq the message queue object
 *
 * @return the error code
 */
D
dzzxzz 已提交
1817
rt_err_t rt_mq_delete(rt_mq_t mq)
1818
{
1819
	RT_DEBUG_NOT_IN_INTERRUPT;
1820

1821 1822 1823 1824
	/* parameter check */
	RT_ASSERT(mq != RT_NULL);

	/* resume all suspended thread */
1825
	rt_ipc_list_resume_all(&(mq->parent.suspend_thread));
1826

qiuyiuestc's avatar
qiuyiuestc 已提交
1827
#if defined(RT_USING_MODULE) && defined(RT_USING_SLAB)
qiuyiuestc's avatar
qiuyiuestc 已提交
1828
	/* the mq object belongs to an application module */
D
dzzxzz 已提交
1829
	if (mq->parent.parent.flag & RT_OBJECT_FLAG_MODULE)
qiuyiuestc's avatar
qiuyiuestc 已提交
1830 1831 1832 1833 1834
		rt_module_free(mq->parent.parent.module_id, mq->msg_pool);
	else
#endif

	/* free message queue pool */
1835 1836
	rt_free(mq->msg_pool);

qiuyiuestc's avatar
qiuyiuestc 已提交
1837
	/* delete message queue object */
1838 1839 1840 1841
	rt_object_delete(&(mq->parent.parent));

	return RT_EOK;
}
1842
RTM_EXPORT(rt_mq_delete);
1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854
#endif

/**
 * This function will send a message to message queue object, if there are threads
 * suspended on message queue object, it will be waked up.
 *
 * @param mq the message queue object
 * @param buffer the message
 * @param size the size of buffer
 *
 * @return the error code
 */
D
dzzxzz 已提交
1855
rt_err_t rt_mq_send(rt_mq_t mq, void *buffer, rt_size_t size)
1856 1857
{
	register rt_ubase_t temp;
D
dzzxzz 已提交
1858
	struct rt_mq_message *msg;
1859

1860 1861 1862 1863
	RT_ASSERT(mq != RT_NULL);
	RT_ASSERT(buffer != RT_NULL);
	RT_ASSERT(size != 0);

1864
	/* greater than one message size */
1865 1866
	if (size > mq->msg_size)
		return -RT_ERROR;
1867

1868
	RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(mq->parent.parent)));
1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885

	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

	/* get a free list, there must be an empty item */
	msg = (struct rt_mq_message*)mq->msg_queue_free;
	/* message queue is full */
	if (msg == RT_NULL)
	{
		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

		return -RT_EFULL;
	}
	/* move free list pointer */
	mq->msg_queue_free = msg->next;

1886 1887 1888
	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

1889 1890
	/* the msg is the new tailer of list, the next shall be NULL */
	msg->next = RT_NULL;
1891 1892 1893
	/* copy buffer */
	rt_memcpy(msg + 1, buffer, size);

1894 1895
	/* disable interrupt */
	temp = rt_hw_interrupt_disable();
1896 1897 1898 1899
	/* link msg to message queue */
	if (mq->msg_queue_tail != RT_NULL)
	{
		/* if the tail exists, */
D
dzzxzz 已提交
1900
		((struct rt_mq_message *)mq->msg_queue_tail)->next = msg;
1901 1902 1903 1904 1905
	}

	/* set new tail */
	mq->msg_queue_tail = msg;
	/* if the head is empty, set head */
D
dzzxzz 已提交
1906 1907
	if (mq->msg_queue_head == RT_NULL)
		mq->msg_queue_head = msg;
1908 1909 1910 1911

	/* increase message entry */
	mq->entry ++;

1912
	/* resume suspended thread */
B
bernard.xiong@gmail.com 已提交
1913
	if (!rt_list_isempty(&mq->parent.suspend_thread))
1914
	{
1915
		rt_ipc_list_resume(&(mq->parent.suspend_thread));
1916 1917 1918 1919 1920

		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

		rt_schedule();
D
dzzxzz 已提交
1921

1922 1923 1924 1925 1926 1927 1928 1929
		return RT_EOK;
	}

	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

	return RT_EOK;
}
1930
RTM_EXPORT(rt_mq_send);
1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942

/**
 * This function will send urgently a message to message queue object, which means
 * the message will be inserted to the head of message queue. If there are threads
 * suspended on message queue object, it will be waked up.
 *
 * @param mq the message queue object
 * @param buffer the message
 * @param size the size of buffer
 *
 * @return the error code
 */
D
dzzxzz 已提交
1943
rt_err_t rt_mq_urgent(rt_mq_t mq, void *buffer, rt_size_t size)
1944 1945
{
	register rt_ubase_t temp;
D
dzzxzz 已提交
1946
	struct rt_mq_message *msg;
1947

1948 1949 1950 1951
	RT_ASSERT(mq != RT_NULL);
	RT_ASSERT(buffer != RT_NULL);
	RT_ASSERT(size != 0);

1952
	/* greater than one message size */
1953 1954
	if (size > mq->msg_size)
		return -RT_ERROR;
1955

1956
	RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(mq->parent.parent)));
1957 1958 1959 1960 1961

	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

	/* get a free list, there must be an empty item */
D
dzzxzz 已提交
1962
	msg = (struct rt_mq_message *)mq->msg_queue_free;
1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973
	/* message queue is full */
	if (msg == RT_NULL)
	{
		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

		return -RT_EFULL;
	}
	/* move free list pointer */
	mq->msg_queue_free = msg->next;

1974 1975 1976
	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

1977 1978 1979
	/* copy buffer */
	rt_memcpy(msg + 1, buffer, size);

1980 1981 1982
	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

1983 1984 1985 1986 1987
	/* link msg to the beginning of message queue */
	msg->next = mq->msg_queue_head;
	mq->msg_queue_head = msg;

	/* if there is no tail */
1988 1989
	if (mq->msg_queue_tail == RT_NULL)
		mq->msg_queue_tail = msg;
1990 1991 1992 1993

	/* increase message entry */
	mq->entry ++;

1994
	/* resume suspended thread */
D
dzzxzz 已提交
1995
	if (!rt_list_isempty(&mq->parent.suspend_thread))
1996
	{
1997
		rt_ipc_list_resume(&(mq->parent.suspend_thread));
1998 1999 2000 2001 2002

		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

		rt_schedule();
D
dzzxzz 已提交
2003

2004 2005 2006 2007 2008 2009 2010 2011
		return RT_EOK;
	}

	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

	return RT_EOK;
}
2012
RTM_EXPORT(rt_mq_urgent);
2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024

/**
 * This function will receive a message from message queue object, if there is no
 * message in message queue object, the thread shall wait for a specified time.
 *
 * @param mq the message queue object
 * @param buffer the received message will be saved in
 * @param size the size of buffer
 * @param timeout the waiting time
 *
 * @return the error code
 */
D
dzzxzz 已提交
2025
rt_err_t rt_mq_recv(rt_mq_t mq, void *buffer, rt_size_t size, rt_int32_t timeout)
2026
{
D
dzzxzz 已提交
2027
	struct rt_thread *thread;
2028
	register rt_ubase_t temp;
D
dzzxzz 已提交
2029
	struct rt_mq_message *msg;
2030
	rt_uint32_t tick_delta;
2031

2032 2033 2034 2035
	RT_ASSERT(mq != RT_NULL);
	RT_ASSERT(buffer != RT_NULL);
	RT_ASSERT(size != 0);

B
bernard.xiong@gmail.com 已提交
2036 2037 2038 2039
	/* initialize delta tick */
	tick_delta = 0;
	/* get current thread */
	thread = rt_thread_self();
2040
	RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(mq->parent.parent)));
2041

2042 2043 2044 2045 2046 2047 2048
	/* disable interrupt */
	temp = rt_hw_interrupt_disable();

	/* for non-blocking call */
	if (mq->entry == 0 && timeout == 0)
	{
		rt_hw_interrupt_enable(temp);
D
dzzxzz 已提交
2049

2050 2051 2052
		return -RT_ETIMEOUT;
	}

2053 2054
	/* message queue is empty */
	while (mq->entry == 0)
2055
	{
2056 2057
		RT_DEBUG_NOT_IN_INTERRUPT;

2058 2059 2060 2061 2062 2063 2064 2065 2066 2067
		/* reset error number in thread */
		thread->error = RT_EOK;

		/* no waiting, return timeout */
		if (timeout == 0)
		{
			/* enable interrupt */
			rt_hw_interrupt_enable(temp);

			thread->error = -RT_ETIMEOUT;
D
dzzxzz 已提交
2068

2069 2070 2071 2072
			return -RT_ETIMEOUT;
		}

		/* suspend current thread */
D
dzzxzz 已提交
2073
		rt_ipc_list_suspend(&(mq->parent.suspend_thread), thread, mq->parent.parent.flag);
2074 2075 2076 2077

		/* has waiting time, start thread timer */
		if (timeout > 0)
		{
2078 2079
			/* get the start tick of timer */
			tick_delta = rt_tick_get();
B
bernard.xiong 已提交
2080

2081 2082
			RT_DEBUG_LOG(RT_DEBUG_IPC, ("set thread:%s to timer list\n",
                                        thread->name));
2083

2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097
			/* reset the timeout of thread timer and start it */
			rt_timer_control(&(thread->thread_timer), RT_TIMER_CTRL_SET_TIME, &timeout);
			rt_timer_start(&(thread->thread_timer));
		}

		/* enable interrupt */
		rt_hw_interrupt_enable(temp);

		/* re-schedule */
		rt_schedule();

		/* recv message */
		if (thread->error != RT_EOK)
		{
D
dzzxzz 已提交
2098
			/* return error */
2099 2100 2101 2102 2103
			return thread->error;
		}

		/* disable interrupt */
		temp = rt_hw_interrupt_disable();
2104

B
bernard.xiong@gmail.com 已提交
2105
		/* if it's not waiting forever and then re-calculate timeout tick */
2106 2107 2108 2109
		if (timeout > 0)
		{
			tick_delta = rt_tick_get() - tick_delta;
			timeout -= tick_delta;
2110 2111
			if (timeout < 0)
				timeout = 0;
2112
		}
2113 2114 2115
	}

	/* get message from queue */
D
dzzxzz 已提交
2116
	msg = (struct rt_mq_message *)mq->msg_queue_head;
2117 2118 2119 2120

	/* move message queue head */
	mq->msg_queue_head = msg->next;
	/* reach queue tail, set to NULL */
2121 2122
	if (mq->msg_queue_tail == msg)
		mq->msg_queue_tail = RT_NULL;
2123

2124 2125 2126 2127 2128 2129
	/* decrease message entry */
	mq->entry --;

	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

2130
	/* copy message */
D
dzzxzz 已提交
2131
	rt_memcpy(buffer, msg + 1, size > mq->msg_size ? mq->msg_size : size);
2132

2133 2134
	/* disable interrupt */
	temp = rt_hw_interrupt_disable();
2135
	/* put message to free list */
D
dzzxzz 已提交
2136
	msg->next = (struct rt_mq_message *)mq->msg_queue_free;
2137 2138 2139 2140
	mq->msg_queue_free = msg;
	/* enable interrupt */
	rt_hw_interrupt_enable(temp);

2141
	RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(mq->parent.parent)));
2142 2143 2144

	return RT_EOK;
}
2145
RTM_EXPORT(rt_mq_recv);  
2146 2147 2148 2149 2150 2151 2152 2153 2154 2155

/**
 * This function can get or set some extra attributions of a message queue object.
 *
 * @param mq the message queue object
 * @param cmd the execution command
 * @param arg the execution argument
 *
 * @return the error code
 */
D
dzzxzz 已提交
2156
rt_err_t rt_mq_control(rt_mq_t mq, rt_uint8_t cmd, void *arg)
2157
{
2158
	rt_ubase_t level;
D
dzzxzz 已提交
2159
	struct rt_mq_message *msg;
2160 2161 2162 2163 2164 2165 2166 2167 2168

	RT_ASSERT(mq != RT_NULL);

	if (cmd == RT_IPC_CMD_RESET)
	{
		/* disable interrupt */
		level = rt_hw_interrupt_disable();

		/* resume all waiting thread */
2169
		rt_ipc_list_resume_all(&mq->parent.suspend_thread);
2170 2171 2172 2173 2174

		/* release all message in the queue */
		while (mq->msg_queue_head != RT_NULL)
		{
			/* get message from queue */
D
dzzxzz 已提交
2175
			msg = (struct rt_mq_message *)mq->msg_queue_head;
2176 2177 2178 2179

			/* move message queue head */
			mq->msg_queue_head = msg->next;
			/* reach queue tail, set to NULL */
2180 2181
			if (mq->msg_queue_tail == msg)
				mq->msg_queue_tail = RT_NULL;
2182 2183

			/* put message to free list */
D
dzzxzz 已提交
2184
			msg->next = (struct rt_mq_message *)mq->msg_queue_free;
2185 2186 2187 2188 2189 2190 2191 2192 2193
			mq->msg_queue_free = msg;
		}

		/* clean entry */
		mq->entry = 0;

		/* enable interrupt */
		rt_hw_interrupt_enable(level);

2194 2195
		rt_schedule();

2196 2197 2198 2199
		return RT_EOK;
	}

	return -RT_ERROR;
2200
}
2201
RTM_EXPORT(rt_mq_control);  
2202
#endif /* end of RT_USING_MESSAGEQUEUE */
D
dzzxzz 已提交
2203

2204
/*@}*/